Files
mkrtos-real/mkrtos_user/user/drv/STM32F205/block/main.c
2025-03-09 23:53:18 +08:00

134 lines
2.9 KiB
C

#include "flash.h"
#include "u_share_mem.h"
#include <assert.h>
#include <blk_drv_cli.h>
#include <blk_drv_svr.h>
#include <errno.h>
#include <mk_dtb_parse.h>
#include <ns_cli.h>
#include <rpc_prot.h>
#include <stdio.h>
#include <sys/stat.h>
#include <u_fast_ipc.h>
#include <u_hd_man.h>
#include <u_sleep.h>
#include <u_sys.h>
#include <u_task.h>
#include <u_thread.h>
#include "u_hd_man.h"
#define STACK_COM_ITME_SIZE (2048)
ATTR_ALIGN(8)
uint8_t stack_coms[STACK_COM_ITME_SIZE];
uint8_t msg_buf_coms[MSG_BUG_LEN];
static obj_handler_t com_th_obj;
void fast_ipc_init(void)
{
com_th_obj = handler_alloc();
assert(com_th_obj != HANDLER_INVALID);
u_fast_ipc_init(stack_coms, msg_buf_coms, 1, STACK_COM_ITME_SIZE, &com_th_obj);
}
static blk_drv_t blk_drv;
int blk_drv_write(obj_handler_t obj, int len, int inx)
{
int ret = -1;
addr_t addr = 0;
umword_t size = 0;
msg_tag_t tag;
if (len == 0)
{
ret = flash_erase_sector(inx);
}
else
{
tag = share_mem_map(obj, vma_addr_create(VPAGE_PROT_RWX, 0, 0), &addr, &size);
if (msg_tag_get_val(tag) < 0)
{
handler_free_umap(obj);
printf("net write error.\n");
return msg_tag_get_val(tag);
}
ret = flash_write_sector(inx, (void *)addr, len / 4);
handler_free_umap(obj);
}
if (ret < 0)
{
return ret;
}
return ret * 4;
}
int blk_drv_read(obj_handler_t obj, int len, int inx)
{
int ret = -1;
addr_t addr = 0;
umword_t size = 0;
uint32_t _err;
msg_tag_t tag = share_mem_map(obj, vma_addr_create(VPAGE_PROT_RWX, 0, 0), &addr, &size);
if (msg_tag_get_val(tag) < 0)
{
handler_free_umap(obj);
printf("net write error.\n");
return msg_tag_get_val(tag);
}
ret = flash_read_sector(inx, (void *)addr, len / 4);
handler_free_umap(obj);
if (ret < 0)
{
return ret;
}
return ret * 4;
}
int blk_drv_map(obj_handler_t *hd)
{
return -ENOSYS;
}
int blk_drv_info(blk_drv_info_t *info)
{
umword_t mem_addr;
int blk_size;
int blk_nr;
int ret;
ret = flash_get_sector_size(&mem_addr, &blk_size, &blk_nr);
if (ret < 0)
{
return ret;
}
info->blk_nr = blk_nr;
info->blk_size = blk_size;
info->blk_start_addr = mem_addr;
return 0;
}
int main(int argc, char *argv[])
{
obj_handler_t hd;
int ret;
if (argc < 2)
{
printf("please set dev path.\n");
printf("example:block /block");
return -1;
}
task_set_obj_name(TASK_THIS, TASK_THIS, "tk_blk");
task_set_obj_name(TASK_THIS, THREAD_MAIN, "th_blk");
printf("%s init..\n", argv[0]);
fast_ipc_init();
flash_init();
blk_drv_init(&blk_drv);
ret = rpc_meta_init_def(TASK_THIS, &hd);
assert(ret >= 0);
ns_register(argv[1], hd, 0);
meta_reg_svr_obj(&blk_drv.svr, BLK_DRV_PROT);
while (1)
{
// rpc_loop();
u_sleep_ms(U_SLEEP_ALWAYS);
}
}