Files
RTOS/DeviceKit/drv_device.c
T

73 lines
2.4 KiB
C
Raw Normal View History

2026-05-19 11:49:22 +08:00
#include <drv_device.h>
#include <drv_rbtree.h>
#include "os_align.h"
/* ------------------------------------------------------------------------------------------------------------------ */
/* */
#define NODE_BLOCK_SIZE (DRV_RBTREE_NODE_SIZE * OS_CFG_DEVICE_MAX)
/* ------------------------------------------------------------------------------------------------------------------ */
/* */
static volatile uint8_t s_drv_device_id_cnt;
OS_ALIGNED(OS_ALIGN_SIZE)
static uint8_t s_drv_device_rbtree_node_block[NODE_BLOCK_SIZE];
static drv_rbtree_t s_drv_rbtree;
/* ------------------------------------------------------------------------------------------------------------------ */
/* */
void drv_device_system_init(void){
s_drv_device_id_cnt = 0;
drv_rbtree_init(&s_drv_rbtree, s_drv_device_rbtree_node_block, NODE_BLOCK_SIZE);
}
void drv_device_register(drv_device_t* self,
const char* name,
void* handle,
os_err_t (*init)(void* handle, void* args),
os_err_t (*destroy)(void* handle, void* args),
os_err_t (*open)(void* handle, void* args),
os_err_t (*close)(void* handle, void* args),
os_err_t (*read)(void* handle, uint8_t * buf, os_size_t buf_size, os_size_t * rd_size, void* args),
os_err_t (*write)(void* handle, const uint8_t * data, os_size_t data_size, os_size_t * wr_size, void* args),
void* args)
{
self->handle = handle;
self->init = init;
self->destroy = destroy;
self->open = open;
self->close = close;
self->read = read;
self->write = write;
self->args = args;
if(name){
os_size_t name_len = strlen(name);
name_len = OS_MIN(name_len, OS_ARRAY_SIZE(self->name)-1);
strncpy(self->name, name, name_len);
self->name[name_len] = '\0';
}else{
snprintf(self->name, OS_ARRAY_SIZE(self->name), "drv_%d", s_drv_device_id_cnt++);
}
drv_rbtree_insert(&s_drv_rbtree, self);
}
void drv_device_unregister(drv_device_t* self){
if(!self) return;
drv_rbtree_remove(&s_drv_rbtree, self->name);
}
drv_device_t* drv_device_find(const char* name){
drv_rbtree_node_t* p = drv_rbtree_find(&s_drv_rbtree, name);
if(p==NULL) return NULL;
return &p->device;
}