DragonOS/docs/kernel/core_api/data_structures.md
guanjinquan d53ddde95d
Patch add idr (#56)
* 增加了idr模块

* 增加了IDR模块,并尝试覆盖上一个错误版本.

* 增加了IDR模块

* 修改了test-idr.c文件

* 进一步完善函数注释

* 更新idr文档同时修改了test-idr的错误


* 将lz4库改为使用系统的clz函数

* idr和test-idr O1

* bugfix: 修复测试用例中的移位问题

* 修正问题

Signed-off-by: guanjinquan <1666320330@qq.com>
Co-authored-by: fslongjin <longjin@RinGoTek.cn>
2022-10-23 16:07:28 +08:00

15 KiB
Raw Blame History

内核数据结构

内核中实现了常用的几种数据结构这里是他们的api文档。


kfifo先进先出缓冲区

kfifo先进先出缓冲区定义于common/kfifo.h中。您可以使用它创建指定大小的fifo缓冲区最大大小为4GB

kfifo_alloc

int kfifo_alloc(struct kfifo_t *fifo, uint32_t size, uint64_t reserved)

描述

通过动态方式初始化kfifo缓冲队列。fifo缓冲区的buffer将由该函数进行申请。

参数

fifo

kfifo队列结构体的指针

size

缓冲区大小单位bytes

reserved

当前字段保留请将其置为0

返回值

当返回值为0时表示正常初始化成功否则返回对应的errno

kfifo_init

void kfifo_init(struct kfifo_t *fifo, void *buffer, uint32_t size)

描述

使用指定的缓冲区来初始化kfifo缓冲队列

参数

fifo

kfifo队列结构体的指针

buffer

  缓冲区基地址指针

size

缓冲区大小单位bytes

kfifo_free_alloc

void kfifo_free_alloc(struct kfifo_t* fifo)

描述

释放通过kfifo_alloc创建的fifo缓冲区. 请勿通过该函数释放其他方式创建的kfifo缓冲区。

参数

fifo

kfifo队列结构体的指针

kfifo_in

uint32_t kfifo_in(struct kfifo_t *fifo, const void *from, uint32_t size)

描述

向kfifo缓冲区推入指定大小的数据。当队列中空间不足时则不推入数据。

参数

fifo

kfifo队列结构体的指针

from

  源数据基地址指针

size

数据大小单位bytes

返回值

  返回成功被推入的数据的大小。

kfifo_out

uint32_t kfifo_out(struct kfifo_t *fifo, void *to, uint32_t size)

描述

从kfifo缓冲区取出数据并从队列中删除数据。当队列中数据量不足时则不取出。

参数

fifo

kfifo队列结构体的指针

to

  目标缓冲区基地址指针

size

数据大小单位bytes

返回值

  返回成功被取出的数据的大小。

kfifo_out_peek

uint32_t kfifo_out_peek(struct kfifo_t *fifo, void *to, uint32_t size)

描述

从kfifo缓冲区取出数据但是不从队列中删除数据。当队列中数据量不足时则不取出。

参数

fifo

kfifo队列结构体的指针

to

  目标缓冲区基地址指针

size

数据大小单位bytes

返回值

  返回成功被取出的数据的大小。

kfifo_reset

kfifo_reset(fifo)

描述

忽略kfifo队列中的所有内容并把输入和输出偏移量都归零

参数

fifo

kfifo队列结构体的指针

kfifo_reset_out

kfifo_reset_out(fifo)

描述

忽略kfifo队列中的所有内容并将输入偏移量赋值给输出偏移量

参数

fifo

kfifo队列结构体的指针

kfifo_total_size

kfifo_total_size(fifo)

描述

获取kfifo缓冲区的最大大小

参数

fifo

kfifo队列结构体的指针

返回值

  缓冲区最大大小

kfifo_size

kfifo_size(fifo)

描述

获取kfifo缓冲区当前已使用的大小

参数

fifo

kfifo队列结构体的指针

返回值

  缓冲区当前已使用的大小

kfifo_empty

kfifo_empty(fifo)

描述

判断kfifo缓冲区当前是否为空

参数

fifo

kfifo队列结构体的指针

返回值

情况 返回值
1
非空 0

kfifo_full

kfifo_full(fifo)

描述

判断kfifo缓冲区当前是否为满

参数

fifo

kfifo队列结构体的指针

返回值

情况 返回值
1
不满 0

ID Allocation

ida的主要作用是分配+管理id. 它能分配一个最小的, 未被分配出去的id. 当您需要管理某个数据结构时, 可能需要使用id来区分不同的目标. 这个时候, ida将会是很好的选择. 因为ida的十分高效, 运行常数相对数组更小, 而且提供了基本管理id需要用到的功能, 值得您试一试.

IDA定义于idr.h文件中. 您通过DECLARE_IDA(my_ida)来创建一个ida对象, 或者struct ida my_ida; ida_init(my_ida);来初始化一个ida.

ida_init

void ida_init(struct ida *ida_p)

描述

通初始化IDA, 你需要保证调用函数之前, ida的free_list为空, 否则会导致内存泄漏.

参数

ida_p

指向ida的指针

返回值

  无返回值

ida_preload

int ida_preload(struct ida *ida_p, gfp_t gfp_mask)

描述

为ida预分配空间.您可以不自行调用, 因为当ida需要空间的时候, 内部会自行使用kmalloc函数获取空间. 当然, 设计这个函数的目的是为了让您有更多的选择. 当您提前调用这个函数, 可以避免之后在开辟空间上的时间开销.

参数

ida_p

指向ida的指针

gfp_mask

保留参数, 目前尚未使用.

返回值

  如果分配成功,将返回0; 否则返回负数错误码, 有可能是内存空间不够.

ida_alloc

int ida_alloc(struct ida *ida_p, int *p_id)

描述

获取一个空闲ID. 您需要注意, 返回值是成功/错误码.

参数

ida_p

指向ida的指针

p_id

您需要传入一个int变量的指针, 如果成功分配ID, ID将会存储在该指针所指向的地址.

返回值

  如果分配成功,将返回0; 否则返回负数错误码, 有可能是内存空间不够.

ida_count

bool ida_count(struct ida *ida_p, int id)

描述

查询一个ID是否被分配.

参数

ida_p

指向ida的指针

id

您查询该ID是否被分配.

返回值

  如果分配,将返回true; 否则返回false.

ida_remove

void ida_remove(struct ida *ida_p, int id)

描述

删除一个已经分配的ID. 如果该ID不存在, 该函数不会产生异常错误, 因为在检测到该ID不存在的时候, 函数将会自动退出.

参数

ida_p

指向ida的指针

id

您要删除的id.

返回值

  无返回值.

ida_destroy

void ida_destroy(struct ida *ida_p)

描述

释放一个IDA所有的空间, 同时删除ida的所有已经分配的id.(所以您不用担心删除id之后, ida还会占用大量空间.)

参数

ida_p

指向ida的指针

返回值

  无返回值

ida_empty

void ida_empty(struct ida *ida_p)

描述

查询一个ida是否为空

参数

ida_p

指向ida的指针

返回值

ida为空则返回true否则返回false。


IDR

idr是一个基于radix-tree的ID-pointer的数据结构. 该数据结构提供了建id与数据指针绑定的功能, 它的主要功能有以下4个

  1. 获取一个ID, 并且将该ID与一个指针绑定
  2. 删除一个已分配的ID
  3. 根据ID查找对应的指针
  4. 根据ID使用新的ptr替换旧的ptr
    您可以使用DECLARE_idr(my_idr)来创建一个idr。或者您也可以使用struct idr my_idr; idr_init(my_idr);这两句话创建一个idr。 至于什么是radix-tree您可以把他简单理解为一个向上生长的多叉树在实现中我们选取了64叉树。

idr_init

void idr_init(struct idr *idp)

描述

通初始化IDR, 你需要保证调用函数之前, idr的free_list为空, 否则会导致内存泄漏.

参数

idp

指向idr的指针

返回值

  无返回值

idr_preload

int idr_preload(struct idr *idp, gfp_t gfp_mask)

描述

为idr预分配空间.您可以不自行调用, 因为当idr需要空间的时候, 内部会自行使用kmalloc函数获取空间. 当然, 设计这个函数的目的是为了让您有更多的选择. 当您提前调用这个函数, 可以避免之后在开辟空间上的时间开销.

参数

idp

指向idr的指针

gfp_mask

保留参数, 目前尚未使用.

返回值

  如果分配成功,将返回0; 否则返回负数错误码, 有可能是内存空间不够.

idr_alloc

int idr_alloc(struct idr *idp, void *ptr, int *id)

描述

获取一个空闲ID. 您需要注意, 返回值是成功/错误码. 调用这个函数需要您保证ptr是非空的即: ptr != NULL, 否则将会影响 idr_find/idr_find_next/idr_find_next_getid/...等函数的使用。(具体请看这三个函数的说明当然只会影响到您的使用体验并不会影响到idr内部函数的决策和逻辑)

参数

idp

指向ida的指针

ptr

指向数据的指针

id

您需要传入一个int变量的指针, 如果成功分配ID, ID将会存储在该指针所指向的地址.

返回值

  如果分配成功,将返回0; 否则返回负数错误码, 有可能是内存空间不够.

idr_remove

void* idr_remove(struct idr *idp, int id)

描述

删除一个id, 但是不释放对应的ptr指向的空间, 同时返回这个被删除id所对应的ptr。 如果该ID不存在, 该函数不会产生异常错误, 因为在检测到该ID不存在的时候, 函数将会自动退出并返回NULL。

参数

idp

指向idr的指针

id

您要删除的id.

返回值

如果删除成功就返回被删除id所对应的ptr否则返回NULL。注意如果这个id本来就和NULL绑定那么也会返回NULL

idr_remove_all

void idr_remove_all(struct idr *idp)

描述

删除idr的所有已经分配的id.(所以您不用担心删除id之后, idr还会占用大量空间。)

但是你需要注意的是,调用这个函数是不会释放数据指针指向的空间的。 所以您调用该函数之前, 确保IDR内部的数据指针被保存。否则当IDR删除所有ID之后 将会造成内存泄漏。

参数

idp

指向idr的指针

返回值

  无返回值

idr_destroy

void idr_destroy(struct idr *idp)

描述

释放一个IDR所有的空间, 同时删除idr的所有已经分配的id.(所以您不用担心删除id之后, ida还会占用大量空间.) - 和idr_remove_all的区别是, 释放掉所有的空间(包括free_list的预分配空间)。

参数

idp

指向idr的指针

返回值

  无返回值

idr_find

void *idr_find(struct idr *idp, int id)

描述

查询一个ID所绑定的数据指针

参数

idp

指向idr的指针

id

您查询该ID的数据指针

返回值

如果分配,将返回该ID对应的数据指针; 否则返回NULL.(注意, 返回NULL不一定代表这ID不存在有可能该ID就是与空指针绑定。) 当然,我们也提供了idr_count函数来判断id是否被分配具体请查看idr_count介绍。

idr_find_next

void *idr_find_next(struct idr *idp, int start_id)

描述

传进一个start_id返回满足 "id大于start_id的最小id" 所对应的数据指针。

参数

idp

指向idr的指针

start_id

您提供的ID限制

返回值

如果分配,将返回该ID对应的数据指针; 否则返回NULL.(注意, 返回NULL不一定代表这ID不存在有可能该ID就是与空指针绑定。) 当然,我们也提供了idr_count函数来判断id是否被分配具体请查看idr_count介绍。

idr_find_next_getid

void *idr_find_next_getid(struct idr *idp, int start_id, int *nextid)

描述

传进一个start_id返回满足 "id大于start_id的最小id" 所对应的数据指针。同时你获取到这个满足条件的最小id 即参数中的 *nextid。

参数

idp

指向idr的指针

start_id

您提供的ID限制

返回值

如果分配,将返回该ID对应的数据指针; 否则返回NULL.(注意, 返回NULL不一定代表这ID不存在有可能该ID就是与空指针绑定。) 当然,我们也提供了idr_count函数来判断id是否被分配具体请查看idr_count介绍。

idr_replace

int idr_replace(struct idr *idp, void *ptr, int id)

描述

传进一个ptr使用该ptr替换掉id所对应的Old_ptr。

参数

idp

指向idr的指针

ptr

您要替换原来的old_ptr的新指针

id

您要替换的指针所对应的id

返回值

0代表成功否则就是错误码 - 代表错误。

idr_replace_get_old

int idr_replace_get_old(struct idr *idp, void *ptr, int id, void **oldptr)

描述

传进一个ptr使用该ptr替换掉id所对应的Old_ptr同时你可以获取到old_ptr。

参数

idp

指向idr的指针

ptr

您要替换原来的old_ptr的新指针

id

您要替换的指针所对应的id

old_ptr

您需要传进该(void**)指针old_ptr将会存放在该指针所指向的地址。

返回值

0代表成功否则就是错误码 - 代表错误。

idr_empty

void idr_empty(struct idr *idp)

描述

查询一个idr是否为空

参数

idp

指向idr的指针

返回值

idr为空则返回true否则返回false。

idr_count

bool idr_count(struct idr *idp, int id)

描述

查询一个ID是否被分配.

参数

ida_p

指向idr的指针

id

您查询该ID是否被分配.

返回值

  如果分配,将返回true; 否则返回false.