module.c 50.0 KB
Newer Older
qiuyiuestc's avatar
qiuyiuestc 已提交
1 2 3
/*
 * File      : module.c
 * This file is part of RT-Thread RTOS
D
dzzxzz 已提交
4
 * COPYRIGHT (C) 2006 - 2012, RT-Thread Development Team
qiuyiuestc's avatar
qiuyiuestc 已提交
5 6 7 8 9 10
 *
 * The license and distribution terms for this file may be
 * found in the file LICENSE in this distribution or at
 * http://www.rt-thread.org/license/LICENSE
 *
 * Change Logs:
D
dzzxzz 已提交
11 12 13 14 15 16
 * Date           Author       Notes
 * 2010-01-09     Bernard      first version
 * 2010-04-09     yi.qiu       implement based on first version
 * 2010-10-23     yi.qiu       implement module memory allocator
 * 2011-05-25     yi.qiu       implement module hook function
 * 2011-06-23     yi.qiu       rewrite module memory allocator
17
 * 2012-11-23     Bernard      using RT_DEBUG_LOG instead of rt_kprintf.
18 19
 * 2012-11-28     Bernard      remove rt_current_module and user 
 *                             can use rt_module_unload to remove a module.
qiuyiuestc's avatar
qiuyiuestc 已提交
20
 */
qiuyiuestc's avatar
qiuyiuestc 已提交
21

22
#include <rthw.h>
qiuyiuestc's avatar
qiuyiuestc 已提交
23
#include <rtthread.h>
qiuyiuestc's avatar
qiuyiuestc 已提交
24
#include <rtm.h>
qiuyiuestc's avatar
qiuyiuestc 已提交
25

qiuyiuestc's avatar
qiuyiuestc 已提交
26
#ifdef RT_USING_MODULE
27
#include "module.h"
qiuyiuestc's avatar
qiuyiuestc 已提交
28

29 30 31 32 33 34 35 36 37 38 39
#define elf_module        ((Elf32_Ehdr *)module_ptr)
#define shdr              ((Elf32_Shdr *)((rt_uint8_t *)module_ptr + elf_module->e_shoff))
#define phdr              ((Elf32_Phdr *)((rt_uint8_t *)module_ptr + elf_module->e_phoff))

#define IS_PROG(s)        (s.sh_type == SHT_PROGBITS)
#define IS_NOPROG(s)      (s.sh_type == SHT_NOBITS)
#define IS_REL(s)         (s.sh_type == SHT_REL)
#define IS_RELA(s)        (s.sh_type == SHT_RELA)
#define IS_ALLOC(s)       (s.sh_flags == SHF_ALLOC)
#define IS_AX(s)          ((s.sh_flags & SHF_ALLOC) && (s.sh_flags & SHF_EXECINSTR))
#define IS_AW(s)          ((s.sh_flags & SHF_ALLOC) && (s.sh_flags & SHF_WRITE))
qiuyiuestc's avatar
qiuyiuestc 已提交
40

41
#ifdef RT_USING_SLAB
42
#define PAGE_COUNT_MAX    256
qiuyiuestc's avatar
qiuyiuestc 已提交
43

qiuyiuestc's avatar
qiuyiuestc 已提交
44 45 46
/* module memory allocator */
struct rt_mem_head
{
47 48
    rt_size_t size;                /* size of memory block */
    struct rt_mem_head *next;      /* next valid memory block */
qiuyiuestc's avatar
qiuyiuestc 已提交
49 50
};

qiuyiuestc's avatar
qiuyiuestc 已提交
51 52
struct rt_page_info
{
53 54
    rt_uint32_t *page_ptr;
    rt_uint32_t npage;
qiuyiuestc's avatar
qiuyiuestc 已提交
55 56 57
};

static void *rt_module_malloc_page(rt_size_t npages);
58 59 60
static void rt_module_free_page(rt_module_t module,
                                void       *page_ptr,
                                rt_size_t   npages);
qiuyiuestc's avatar
qiuyiuestc 已提交
61

qiuyiuestc's avatar
qiuyiuestc 已提交
62
static struct rt_semaphore mod_sem;
63 64
#endif

65
static struct rt_module_symtab *_rt_module_symtab_begin = RT_NULL;
66
static struct rt_module_symtab *_rt_module_symtab_end   = RT_NULL;
qiuyiuestc's avatar
qiuyiuestc 已提交
67

68 69 70
/**
 * @ingroup SystemInit
 *
B
bernard.xiong@gmail.com 已提交
71
 * This function will initialize system module
72 73 74
 */
void rt_system_module_init(void)
{
qiuyiuestc's avatar
qiuyiuestc 已提交
75
#ifdef __GNUC__
76 77
    extern int __rtmsymtab_start;
    extern int __rtmsymtab_end;
78

79 80
    _rt_module_symtab_begin = (struct rt_module_symtab *)&__rtmsymtab_start;
    _rt_module_symtab_end   = (struct rt_module_symtab *)&__rtmsymtab_end;
D
dzzxzz 已提交
81
#elif defined (__CC_ARM)
82 83
    extern int RTMSymTab$$Base;
    extern int RTMSymTab$$Limit;
D
dzzxzz 已提交
84

85
    _rt_module_symtab_begin = (struct rt_module_symtab *)&RTMSymTab$$Base;
86
    _rt_module_symtab_end   = (struct rt_module_symtab *)&RTMSymTab$$Limit;
qiuyiuestc's avatar
qiuyiuestc 已提交
87
#endif
88

89
#ifdef RT_USING_SLAB
90 91
    /* initialize heap semaphore */
    rt_sem_init(&mod_sem, "module", 1, RT_IPC_FLAG_FIFO);
92
#endif
93
}
B
Bernard Xiong 已提交
94
INIT_COMPONENT_EXPORT(rt_system_module_init);
95

D
dzzxzz 已提交
96
static rt_uint32_t rt_module_symbol_find(const char *sym_str)
97
{
98 99
    /* find in kernel symbol table */
    struct rt_module_symtab *index;
100 101 102 103

    for (index = _rt_module_symtab_begin;
         index != _rt_module_symtab_end;
         index ++)
104 105 106 107 108 109
    {
        if (rt_strcmp(index->name, sym_str) == 0)
            return (rt_uint32_t)index->addr;
    }

    return 0;
110
}
qiuyiuestc's avatar
qiuyiuestc 已提交
111

qiuyiuestc's avatar
qiuyiuestc 已提交
112 113 114
/**
 * This function will return self module object
 *
qiuyiuestc's avatar
qiuyiuestc 已提交
115
 * @return the self module object
qiuyiuestc's avatar
qiuyiuestc 已提交
116
 */
D
dzzxzz 已提交
117
rt_module_t rt_module_self(void)
qiuyiuestc's avatar
qiuyiuestc 已提交
118
{
119
    rt_thread_t tid;
qiuyiuestc's avatar
qiuyiuestc 已提交
120

121 122 123
    tid = rt_thread_self();
    if (tid == RT_NULL)
        return RT_NULL;
qiuyiuestc's avatar
qiuyiuestc 已提交
124

125 126
    /* return current module */
    return (rt_module_t)tid->module_id;
qiuyiuestc's avatar
qiuyiuestc 已提交
127 128
}

129 130 131
static int rt_module_arm_relocate(struct rt_module *module,
                                  Elf32_Rel        *rel,
                                  Elf32_Addr        sym_val)
qiuyiuestc's avatar
qiuyiuestc 已提交
132
{
133 134 135 136 137 138 139 140 141 142 143
    Elf32_Addr *where, tmp;
    Elf32_Sword addend, offset;
    rt_uint32_t upper, lower, sign, j1, j2;

    where = (Elf32_Addr *)((rt_uint8_t *)module->module_space + rel->r_offset);
    switch (ELF32_R_TYPE(rel->r_info))
    {
    case R_ARM_NONE:
        break;
    case R_ARM_ABS32:
        *where += (Elf32_Addr)sym_val;
144 145
        RT_DEBUG_LOG(RT_DEBUG_MODULE, ("R_ARM_ABS32: %x -> %x\n",
                                       where, *where));
146 147 148 149 150 151 152 153 154 155 156
        break;
    case R_ARM_PC24:
    case R_ARM_PLT32:
    case R_ARM_CALL:
    case R_ARM_JUMP24:
        addend = *where & 0x00ffffff;
        if (addend & 0x00800000)
            addend |= 0xff000000;
        tmp = sym_val - (Elf32_Addr)where + (addend << 2);
        tmp >>= 2;
        *where = (*where & 0xff000000) | (tmp & 0x00ffffff);
157 158
        RT_DEBUG_LOG(RT_DEBUG_MODULE, ("R_ARM_PC24: %x -> %x\n",
                                       where, *where));
159 160 161
        break;
    case R_ARM_REL32:
        *where += sym_val - (Elf32_Addr)where;
162 163 164
        RT_DEBUG_LOG(RT_DEBUG_MODULE,
                     ("R_ARM_REL32: %x -> %x, sym %x, offset %x\n",
                      where, *where, sym_val, rel->r_offset));
165 166 167 168 169 170 171 172
        break;
    case R_ARM_V4BX:
        *where &= 0xf000000f;
        *where |= 0x01a0f000;
        break;
    case R_ARM_GLOB_DAT:
    case R_ARM_JUMP_SLOT:
        *where = (Elf32_Addr)sym_val;
173 174
        RT_DEBUG_LOG(RT_DEBUG_MODULE, ("R_ARM_JUMP_SLOT: 0x%x -> 0x%x 0x%x\n",
                                       where, *where, sym_val));
175 176 177
        break;
#if 0        /* To do */
    case R_ARM_GOT_BREL:
178
        temp   = (Elf32_Addr)sym_val;
179
        *where = (Elf32_Addr)&temp;
180 181
        RT_DEBUG_LOG(RT_DEBUG_MODULE, ("R_ARM_GOT_BREL: 0x%x -> 0x%x 0x%x\n",
                                       where, *where, sym_val));
182
        break;
B
bernard.xiong@gmail.com 已提交
183
#endif
184 185
    case R_ARM_RELATIVE:
        *where = (Elf32_Addr)sym_val + *where;
186 187
        RT_DEBUG_LOG(RT_DEBUG_MODULE, ("R_ARM_RELATIVE: 0x%x -> 0x%x 0x%x\n",
                                       where, *where, sym_val));
188 189 190
        break;
    case R_ARM_THM_CALL:
    case R_ARM_THM_JUMP24:
191 192 193 194 195 196 197 198 199 200 201
        upper  = *(rt_uint16_t *)where;
        lower  = *(rt_uint16_t *)((Elf32_Addr)where + 2);

        sign   = (upper >> 10) & 1;
        j1     = (lower >> 13) & 1;
        j2     = (lower >> 11) & 1;
        offset = (sign << 24) |
                 ((~(j1 ^ sign) & 1) << 23) |
                 ((~(j2 ^ sign) & 1) << 22) |
                 ((upper & 0x03ff) << 12) |
                 ((lower & 0x07ff) << 1);
202
        if (offset & 0x01000000)
203
            offset -= 0x02000000;
204 205
        offset += sym_val - (Elf32_Addr)where;

206 207 208
        if (!(offset & 1) ||
            offset <= (rt_int32_t)0xff000000 ||
            offset >= (rt_int32_t)0x01000000)
209
        {
210
            rt_kprintf("Module: Only Thumb addresses allowed\n");
211

212 213 214 215
            return -1;
        }

        sign = (offset >> 24) & 1;
216 217 218 219 220
        j1   = sign ^ (~(offset >> 23) & 1);
        j2   = sign ^ (~(offset >> 22) & 1);
        *(rt_uint16_t *)where = (rt_uint16_t)((upper & 0xf800) |
                                (sign << 10) |
                                ((offset >> 12) & 0x03ff));
221
        *(rt_uint16_t *)(where + 2) = (rt_uint16_t)((lower & 0xd000) |
222 223
                                      (j1 << 13) | (j2 << 11) |
                                      ((offset >> 1) & 0x07ff));
224 225 226 227 228 229 230 231
        upper = *(rt_uint16_t *)where;
        lower = *(rt_uint16_t *)((Elf32_Addr)where + 2);
        break;
    default:
        return -1;
    }

    return 0;
qiuyiuestc's avatar
qiuyiuestc 已提交
232 233
}

D
dzzxzz 已提交
234
static void rt_module_init_object_container(struct rt_module *module)
qiuyiuestc's avatar
qiuyiuestc 已提交
235
{
236
    RT_ASSERT(module != RT_NULL);
qiuyiuestc's avatar
qiuyiuestc 已提交
237

238 239 240 241
    /* initialize object container - thread */
    rt_list_init(&(module->module_object[RT_Object_Class_Thread].object_list));
    module->module_object[RT_Object_Class_Thread].object_size = sizeof(struct rt_thread);
    module->module_object[RT_Object_Class_Thread].type = RT_Object_Class_Thread;
qiuyiuestc's avatar
qiuyiuestc 已提交
242 243

#ifdef RT_USING_SEMAPHORE
244 245 246 247
    /* initialize object container - semaphore */
    rt_list_init(&(module->module_object[RT_Object_Class_Semaphore].object_list));
    module->module_object[RT_Object_Class_Semaphore].object_size = sizeof(struct rt_semaphore);
    module->module_object[RT_Object_Class_Semaphore].type = RT_Object_Class_Semaphore;
qiuyiuestc's avatar
qiuyiuestc 已提交
248 249 250
#endif

#ifdef RT_USING_MUTEX
251 252 253 254
    /* initialize object container - mutex */
    rt_list_init(&(module->module_object[RT_Object_Class_Mutex].object_list));
    module->module_object[RT_Object_Class_Mutex].object_size = sizeof(struct rt_mutex);
    module->module_object[RT_Object_Class_Mutex].type = RT_Object_Class_Mutex;
qiuyiuestc's avatar
qiuyiuestc 已提交
255 256 257
#endif

#ifdef RT_USING_EVENT
258 259 260 261
    /* initialize object container - event */
    rt_list_init(&(module->module_object[RT_Object_Class_Event].object_list));
    module->module_object[RT_Object_Class_Event].object_size = sizeof(struct rt_event);
    module->module_object[RT_Object_Class_Event].type = RT_Object_Class_Event;
qiuyiuestc's avatar
qiuyiuestc 已提交
262 263 264
#endif

#ifdef RT_USING_MAILBOX
265 266 267 268
    /* initialize object container - mailbox */
    rt_list_init(&(module->module_object[RT_Object_Class_MailBox].object_list));
    module->module_object[RT_Object_Class_MailBox].object_size = sizeof(struct rt_mailbox);
    module->module_object[RT_Object_Class_MailBox].type = RT_Object_Class_MailBox;
qiuyiuestc's avatar
qiuyiuestc 已提交
269 270 271
#endif

#ifdef RT_USING_MESSAGEQUEUE
272 273 274 275
    /* initialize object container - message queue */
    rt_list_init(&(module->module_object[RT_Object_Class_MessageQueue].object_list));
    module->module_object[RT_Object_Class_MessageQueue].object_size = sizeof(struct rt_messagequeue);
    module->module_object[RT_Object_Class_MessageQueue].type = RT_Object_Class_MessageQueue;
qiuyiuestc's avatar
qiuyiuestc 已提交
276 277
#endif

278
#ifdef RT_USING_MEMHEAP
279 280 281 282
    /* initialize object container - memory heap */
    rt_list_init(&(module->module_object[RT_Object_Class_MemHeap].object_list));
    module->module_object[RT_Object_Class_MemHeap].object_size = sizeof(struct rt_memheap);
    module->module_object[RT_Object_Class_MemHeap].type = RT_Object_Class_MemHeap;
283 284
#endif

qiuyiuestc's avatar
qiuyiuestc 已提交
285
#ifdef RT_USING_MEMPOOL
286 287 288 289
    /* initialize object container - memory pool */
    rt_list_init(&(module->module_object[RT_Object_Class_MemPool].object_list));
    module->module_object[RT_Object_Class_MemPool].object_size = sizeof(struct rt_mempool);
    module->module_object[RT_Object_Class_MemPool].type = RT_Object_Class_MemPool;
qiuyiuestc's avatar
qiuyiuestc 已提交
290 291 292
#endif

#ifdef RT_USING_DEVICE
293 294 295 296
    /* initialize object container - device */
    rt_list_init(&(module->module_object[RT_Object_Class_Device].object_list));
    module->module_object[RT_Object_Class_Device].object_size = sizeof(struct rt_device);
    module->module_object[RT_Object_Class_Device].type = RT_Object_Class_Device;
qiuyiuestc's avatar
qiuyiuestc 已提交
297 298
#endif

299 300 301 302
    /* initialize object container - timer */
    rt_list_init(&(module->module_object[RT_Object_Class_Timer].object_list));
    module->module_object[RT_Object_Class_Timer].object_size = sizeof(struct rt_timer);
    module->module_object[RT_Object_Class_Timer].type = RT_Object_Class_Timer;
qiuyiuestc's avatar
qiuyiuestc 已提交
303 304
}

qiuyiuestc's avatar
qiuyiuestc 已提交
305 306 307 308 309 310 311
#ifdef RT_USING_HOOK
static void (*rt_module_load_hook)(rt_module_t module);
static void (*rt_module_unload_hook)(rt_module_t module);

/**
 * @addtogroup Hook
 */
D
dzzxzz 已提交
312

qiuyiuestc's avatar
qiuyiuestc 已提交
313 314 315 316 317 318 319 320 321 322
/*@{*/

/**
 * This function will set a hook function, which will be invoked when module
 * be loaded to system.
 *
 * @param hook the hook function
 */
void rt_module_load_sethook(void (*hook)(rt_module_t module))
{
323
    rt_module_load_hook = hook;
qiuyiuestc's avatar
qiuyiuestc 已提交
324 325 326 327 328 329 330 331 332 333
}

/**
 * This function will set a hook function, which will be invoked when module
 * be unloaded from system.
 *
 * @param hook the hook function
 */
void rt_module_unload_sethook(void (*hook)(rt_module_t module))
{
334
    rt_module_unload_hook = hook;
qiuyiuestc's avatar
qiuyiuestc 已提交
335 336 337 338 339
}

/*@}*/
#endif

340 341
static struct rt_module *_load_shared_object(const char *name,
                                             void       *module_ptr)
qiuyiuestc's avatar
qiuyiuestc 已提交
342
{
343
    rt_uint8_t *ptr    = RT_NULL;
344
    rt_module_t module = RT_NULL;
345
    rt_bool_t linked   = RT_FALSE;
346 347 348 349
    rt_uint32_t index, module_size = 0;

    RT_ASSERT(module_ptr != RT_NULL);

350
    if (rt_memcmp(elf_module->e_ident, RTMMAG, SELFMAG) == 0)
351 352 353 354
    {
        /* rtmlinker finished */
        linked = RT_TRUE;
    }
355

356 357 358
    /* get the ELF image size */
    for (index = 0; index < elf_module->e_phnum; index++)
    {
359
        if (phdr[index].p_type == PT_LOAD)
360
            module_size += phdr[index].p_memsz;
361 362
    }

363 364
    if (module_size == 0)
    {
365
        rt_kprintf("Module: size error\n");
366

367
        return RT_NULL;
368
    }
369 370

    /* allocate module */
371 372 373 374
    module = (struct rt_module *)rt_object_allocate(RT_Object_Class_Module,
                                                    name);
    if (!module)
        return RT_NULL;
375 376

    module->nref = 0;
377

378 379 380 381
    /* allocate module space */
    module->module_space = rt_malloc(module_size);
    if (module->module_space == RT_NULL)
    {
382
        rt_kprintf("Module: allocate space failed.\n");
383 384 385 386 387 388 389 390 391 392 393 394 395
        rt_object_delete(&(module->parent));

        return RT_NULL;
    }

    /* zero all space */
    ptr = module->module_space;
    rt_memset(ptr, 0, module_size);

    for (index = 0; index < elf_module->e_phnum; index++)
    {
        if (phdr[index].p_type == PT_LOAD)
        {
396
            rt_memcpy(ptr + phdr[index].p_paddr,
397 398
                      (rt_uint8_t *)elf_module + phdr[index].p_offset,
                      phdr[index].p_filesz);
399 400
        }
    }
401 402 403

    /* set module entry */
    module->module_entry = module->module_space + elf_module->e_entry;
404

405 406 407 408 409 410 411 412
    /* handle relocation section */
    for (index = 0; index < elf_module->e_shnum; index ++)
    {
        rt_uint32_t i, nr_reloc;
        Elf32_Sym *symtab;
        Elf32_Rel *rel;
        rt_uint8_t *strtab;
        static rt_bool_t unsolved = RT_FALSE;
413

414 415
        if (!IS_REL(shdr[index]))
            continue;
416 417 418 419 420

        /* get relocate item */
        rel = (Elf32_Rel *)((rt_uint8_t *)module_ptr + shdr[index].sh_offset);

        /* locate .rel.plt and .rel.dyn section */
421 422 423 424 425
        symtab = (Elf32_Sym *)((rt_uint8_t *)module_ptr +
                               shdr[shdr[index].sh_link].sh_offset);
        strtab = (rt_uint8_t *)module_ptr +
                 shdr[shdr[shdr[index].sh_link].sh_link].sh_offset;
        nr_reloc = (rt_uint32_t)(shdr[index].sh_size / sizeof(Elf32_Rel));
426 427 428 429 430 431

        /* relocate every items */
        for (i = 0; i < nr_reloc; i ++)
        {
            Elf32_Sym *sym = &symtab[ELF32_R_SYM(rel->r_info)];

432 433 434
            RT_DEBUG_LOG(RT_DEBUG_MODULE, ("relocate symbol %s shndx %d\n",
                                           strtab + sym->st_name,
                                           sym->st_shndx));
435

436
            if ((sym->st_shndx != SHT_NULL) ||
437 438
                (ELF_ST_BIND(sym->st_info) == STB_LOCAL))
            {
439
                rt_module_arm_relocate(module, rel,
440
                           (Elf32_Addr)(module->module_space + sym->st_value));
441
            }
442
            else if (!linked)
443 444 445
            {
                Elf32_Addr addr;

446 447
                RT_DEBUG_LOG(RT_DEBUG_MODULE, ("relocate symbol: %s\n",
                                               strtab + sym->st_name));
448 449 450 451 452

                /* need to resolve symbol in kernel symbol table */
                addr = rt_module_symbol_find((const char *)(strtab + sym->st_name));
                if (addr == 0)
                {
453
                    rt_kprintf("Module: can't find %s in kernel symbol table\n",
454
                               strtab + sym->st_name);
455
                    unsolved = RT_TRUE;
456
                }
457 458 459 460 461 462 463 464 465 466 467
                else
                    rt_module_arm_relocate(module, rel, addr);
            }
            rel ++;
        }

        if (unsolved)
        {
            rt_object_delete(&(module->parent));

            return RT_NULL;
468
        }
469 470 471 472
    }

    /* construct module symbol table */
    for (index = 0; index < elf_module->e_shnum; index ++)
473
    {
474
        /* find .dynsym section */
475
        rt_uint8_t *shstrab;
476 477
        shstrab = (rt_uint8_t *)module_ptr +
                  shdr[elf_module->e_shstrndx].sh_offset;
478 479 480 481 482 483 484 485
        if (rt_strcmp((const char *)(shstrab + shdr[index].sh_name), ELF_DYNSYM) == 0)
            break;
    }

    /* found .dynsym section */
    if (index != elf_module->e_shnum)
    {
        int i, count = 0;
486
        Elf32_Sym  *symtab = RT_NULL;
487 488 489 490 491 492 493
        rt_uint8_t *strtab = RT_NULL;

        symtab =(Elf32_Sym *)((rt_uint8_t *)module_ptr + shdr[index].sh_offset);
        strtab = (rt_uint8_t *)module_ptr + shdr[shdr[index].sh_link].sh_offset;

        for (i=0; i<shdr[index].sh_size/sizeof(Elf32_Sym); i++)
        {
494
            if ((ELF_ST_BIND(symtab[i].st_info) == STB_GLOBAL) &&
495 496 497 498 499
                (ELF_ST_TYPE(symtab[i].st_info) == STT_FUNC))
                count ++;
        }

        module->symtab = (struct rt_module_symtab *)rt_malloc
500
                         (count * sizeof(struct rt_module_symtab));
501 502 503 504
        module->nsym = count;
        for (i=0, count=0; i<shdr[index].sh_size/sizeof(Elf32_Sym); i++)
        {
            rt_size_t length;
505 506

            if ((ELF_ST_BIND(symtab[i].st_info) != STB_GLOBAL) ||
507 508
                (ELF_ST_TYPE(symtab[i].st_info) != STT_FUNC))
                continue;
509 510 511

            length = rt_strlen((const char *)(strtab + symtab[i].st_name)) + 1;

512
            module->symtab[count].addr =
513 514 515
                (void *)(module->module_space + symtab[i].st_value);
            module->symtab[count].name = rt_malloc(length);
            rt_memset((void *)module->symtab[count].name, 0, length);
516
            rt_memcpy((void *)module->symtab[count].name,
517 518
                      strtab + symtab[i].st_name,
                      length);
519
            count ++;
520
        }
521 522 523
    }

    return module;
524 525
}

526 527
static struct rt_module* _load_relocated_object(const char *name,
                                                void       *module_ptr)
528
{
529 530 531 532 533 534
    rt_uint32_t index, rodata_addr = 0, bss_addr = 0, data_addr = 0;
    rt_uint32_t module_addr = 0, module_size = 0;
    struct rt_module *module = RT_NULL;
    rt_uint8_t *ptr, *strtab, *shstrab;

    /* get the ELF image size */
535
    for (index = 0; index < elf_module->e_shnum; index ++)
536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564
    {
        /* text */
        if (IS_PROG(shdr[index]) && IS_AX(shdr[index]))
        {
            module_size += shdr[index].sh_size;
            module_addr = shdr[index].sh_addr;
        }
        /* rodata */
        if (IS_PROG(shdr[index]) && IS_ALLOC(shdr[index]))
        {
            module_size += shdr[index].sh_size;
        }
        /* data */
        if (IS_PROG(shdr[index]) && IS_AW(shdr[index]))
        {
            module_size += shdr[index].sh_size;
        }
        /* bss */
        if (IS_NOPROG(shdr[index]) && IS_AW(shdr[index]))
        {
            module_size += shdr[index].sh_size;
        }
    }

    /* no text, data and bss on image */
    if (module_size == 0)
        return RT_NULL;

    /* allocate module */
565 566
    module = (struct rt_module *)
             rt_object_allocate(RT_Object_Class_Module, (const char *)name);
567 568 569 570 571 572 573
    if (module == RT_NULL)
        return RT_NULL;

    /* allocate module space */
    module->module_space = rt_malloc(module_size);
    if (module->module_space == RT_NULL)
    {
574
        rt_kprintf("Module: allocate space failed.\n");
575 576 577 578 579 580 581 582 583 584
        rt_object_delete(&(module->parent));

        return RT_NULL;
    }

    /* zero all space */
    ptr = module->module_space;
    rt_memset(ptr, 0, module_size);

    /* load text and data section */
585
    for (index = 0; index < elf_module->e_shnum; index ++)
586 587 588 589
    {
        /* load text section */
        if (IS_PROG(shdr[index]) && IS_AX(shdr[index]))
        {
590 591 592
            rt_memcpy(ptr,
                      (rt_uint8_t *)elf_module + shdr[index].sh_offset,
                      shdr[index].sh_size);
593 594
            RT_DEBUG_LOG(RT_DEBUG_MODULE, ("load text 0x%x, size %d\n",
                                           ptr, shdr[index].sh_size));
595 596 597 598 599 600
            ptr += shdr[index].sh_size;
        }

        /* load rodata section */
        if (IS_PROG(shdr[index]) && IS_ALLOC(shdr[index]))
        {
601 602 603
            rt_memcpy(ptr,
                      (rt_uint8_t *)elf_module + shdr[index].sh_offset,
                      shdr[index].sh_size);
604
            rodata_addr = (rt_uint32_t)ptr;
605 606 607
            RT_DEBUG_LOG(RT_DEBUG_MODULE,
                         ("load rodata 0x%x, size %d, rodata 0x%x\n",
                          ptr, shdr[index].sh_size, *(rt_uint32_t *)data_addr));
608 609 610 611 612 613
            ptr += shdr[index].sh_size;
        }

        /* load data section */
        if (IS_PROG(shdr[index]) && IS_AW(shdr[index]))
        {
614 615 616
            rt_memcpy(ptr,
                      (rt_uint8_t *)elf_module + shdr[index].sh_offset,
                      shdr[index].sh_size);
617
            data_addr = (rt_uint32_t)ptr;
618 619 620
            RT_DEBUG_LOG(RT_DEBUG_MODULE,
                         ("load data 0x%x, size %d, data 0x%x\n",
                          ptr, shdr[index].sh_size, *(rt_uint32_t *)data_addr));
621 622 623 624 625 626 627 628
            ptr += shdr[index].sh_size;
        }

        /* load bss section */
        if (IS_NOPROG(shdr[index]) && IS_AW(shdr[index]))
        {
            rt_memset(ptr, 0, shdr[index].sh_size);
            bss_addr = (rt_uint32_t)ptr;
629 630
            RT_DEBUG_LOG(RT_DEBUG_MODULE, ("load bss 0x%x, size %d,\n",
                                           ptr, shdr[index].sh_size));
631 632 633 634
        }
    }

    /* set module entry */
635 636
    module->module_entry =
        (rt_uint8_t *)module->module_space + elf_module->e_entry - module_addr;
637 638 639 640 641 642 643

    /* handle relocation section */
    for (index = 0; index < elf_module->e_shnum; index ++)
    {
        rt_uint32_t i, nr_reloc;
        Elf32_Sym *symtab;
        Elf32_Rel *rel;
644

645 646
        if (!IS_REL(shdr[index]))
            continue;
647

648
        /* get relocate item */
649
        rel = (Elf32_Rel *)((rt_uint8_t *)module_ptr + shdr[index].sh_offset);
650 651

        /* locate .dynsym and .dynstr */
652 653 654 655 656 657 658
        symtab   = (Elf32_Sym *)((rt_uint8_t *)module_ptr +
                   shdr[shdr[index].sh_link].sh_offset);
        strtab   = (rt_uint8_t *)module_ptr +
                   shdr[shdr[shdr[index].sh_link].sh_link].sh_offset;
        shstrab  = (rt_uint8_t *)module_ptr +
                   shdr[elf_module->e_shstrndx].sh_offset;
        nr_reloc = (rt_uint32_t)(shdr[index].sh_size / sizeof(Elf32_Rel));
659 660 661 662 663

        /* relocate every items */
        for (i = 0; i < nr_reloc; i ++)
        {
            Elf32_Sym *sym = &symtab[ELF32_R_SYM(rel->r_info)];
664

665 666
            RT_DEBUG_LOG(RT_DEBUG_MODULE, ("relocate symbol: %s\n",
                                           strtab + sym->st_name));
667 668 669

            if (sym->st_shndx != STN_UNDEF)
            {
670 671
                if ((ELF_ST_TYPE(sym->st_info) == STT_SECTION) ||
                    (ELF_ST_TYPE(sym->st_info) == STT_OBJECT))
672
                {
673
                    if (rt_strncmp((const char *)(shstrab +
674 675 676
                        shdr[sym->st_shndx].sh_name), ELF_RODATA, 8) == 0)
                    {
                        /* relocate rodata section */
677
                        RT_DEBUG_LOG(RT_DEBUG_MODULE, ("rodata\n"));
678
                        rt_module_arm_relocate(module, rel,
679
                                               (Elf32_Addr)(rodata_addr + sym->st_value));
680
                    }
681 682
                    else if (rt_strncmp((const char*)
                             (shstrab + shdr[sym->st_shndx].sh_name), ELF_BSS, 5) == 0)
683 684
                    {
                        /* relocate bss section */
685
                        RT_DEBUG_LOG(RT_DEBUG_MODULE, ("bss\n"));
686
                        rt_module_arm_relocate(module, rel,
687
                                               (Elf32_Addr)bss_addr + sym->st_value);
688
                    }
689 690
                    else if (rt_strncmp((const char *)(shstrab + shdr[sym->st_shndx].sh_name),
                             ELF_DATA, 6) == 0)
691 692
                    {
                        /* relocate data section */
693
                        RT_DEBUG_LOG(RT_DEBUG_MODULE, ("data\n"));
694
                        rt_module_arm_relocate(module, rel,
695
                                               (Elf32_Addr)data_addr + sym->st_value);
696 697 698
                    }
                }
            }
699
            else if (ELF_ST_TYPE(sym->st_info) == STT_FUNC)
700 701
            {
                /* relocate function */
702
                rt_module_arm_relocate(module, rel, (Elf32_Addr)((rt_uint8_t *)
703 704 705 706 707 708
                    module->module_space - module_addr + sym->st_value));
            }
            else
            {
                Elf32_Addr addr;

709
                if (ELF32_R_TYPE(rel->r_info) != R_ARM_V4BX)
710
                {
711 712
                    RT_DEBUG_LOG(RT_DEBUG_MODULE, ("relocate symbol: %s\n",
                                                   strtab + sym->st_name));
713

714
                    /* need to resolve symbol in kernel symbol table */
715
                    addr = rt_module_symbol_find((const char *)(strtab + sym->st_name));
716 717 718
                    if (addr != (Elf32_Addr)RT_NULL)
                    {
                        rt_module_arm_relocate(module, rel, addr);
719 720
                        RT_DEBUG_LOG(RT_DEBUG_MODULE, ("symbol addr 0x%x\n",
                                                       addr));
721 722
                    }
                    else
723
                        rt_kprintf("Module: can't find %s in kernel symbol table\n",
724
                                   strtab + sym->st_name);
725 726 727 728 729 730 731 732 733 734 735 736
                }
                else
                {
                    rt_module_arm_relocate(module, rel, (Elf32_Addr)((rt_uint8_t*)
                        module->module_space - module_addr + sym->st_value));
                }
            }
            rel ++;
        }
    }

    return module;
737 738 739 740 741 742 743 744 745 746
}

/**
 * This function will load a module from memory and create a thread for it
 *
 * @param name the name of module, which shall be unique
 * @param module_ptr the memory address of module image
 *
 * @return the module object
 */
D
dzzxzz 已提交
747
rt_module_t rt_module_load(const char *name, void *module_ptr)
748
{
749 750 751 752
    rt_module_t module;

    RT_DEBUG_NOT_IN_INTERRUPT;

753
    RT_DEBUG_LOG(RT_DEBUG_MODULE, ("rt_module_load: %s ,", name));
754 755

    /* check ELF header */
756 757
    if (rt_memcmp(elf_module->e_ident, RTMMAG, SELFMAG) != 0 &&
        rt_memcmp(elf_module->e_ident, ELFMAG, SELFMAG) != 0)
758
    {
759
        rt_kprintf("Module: magic error\n");
760 761 762 763 764

        return RT_NULL;
    }

    /* check ELF class */
765
    if (elf_module->e_ident[EI_CLASS] != ELFCLASS32)
766
    {
767
        rt_kprintf("Module: ELF class error\n");
768 769 770 771

        return RT_NULL;
    }

772
    if (elf_module->e_type == ET_REL)
773 774 775
    {
        module = _load_relocated_object(name, module_ptr);
    }
776
    else if (elf_module->e_type == ET_DYN)
777 778 779 780 781
    {
        module = _load_shared_object(name, module_ptr);
    }
    else
    {
782
        rt_kprintf("Module: unsupported elf type\n");
783 784 785 786

        return RT_NULL;
    }

787
    if (module == RT_NULL)
788 789 790 791
        return RT_NULL;

    /* init module object container */
    rt_module_init_object_container(module);
792

793 794 795 796
    /* increase module reference count */
    module->nref ++;

    if (elf_module->e_entry != 0)
797
    {
798 799
        rt_uint32_t *stack_size;
        rt_uint8_t  *priority;
800

qiuyiuestc's avatar
qiuyiuestc 已提交
801
#ifdef RT_USING_SLAB
802 803
        /* init module memory allocator */
        module->mem_list = RT_NULL;
qiuyiuestc's avatar
qiuyiuestc 已提交
804

805
        /* create page array */
806 807
        module->page_array = 
            (void *)rt_malloc(PAGE_COUNT_MAX * sizeof(struct rt_page_info));
808
        module->page_cnt = 0;
809 810
#endif

811
        /* get the main thread stack size */
812
        module->stack_size = 2048;
813 814
        module->thread_priority = RT_THREAD_PRIORITY_MAX - 2;

815
        /* create module thread */
816 817 818 819 820 821 822 823
        module->module_thread =
            rt_thread_create(name,
                             (void(*)(void *))module->module_entry,
                             RT_NULL,
                             module->stack_size,
                             module->thread_priority,
                             10);

824 825
        RT_DEBUG_LOG(RT_DEBUG_MODULE, ("thread entry 0x%x\n",
                                       module->module_entry));
826 827 828

        /* set module id */
        module->module_thread->module_id = (void *)module;
829 830 831 832
        module->parent.flag = RT_MODULE_FLAG_WITHENTRY;

        /* startup module thread */
        rt_thread_startup(module->module_thread);
833
    }
834 835 836 837
    else
    {
        /* without entry point */
        module->parent.flag |= RT_MODULE_FLAG_WITHOUTENTRY;
838
    }
qiuyiuestc's avatar
qiuyiuestc 已提交
839

qiuyiuestc's avatar
qiuyiuestc 已提交
840
#ifdef RT_USING_HOOK
841 842 843 844
    if (rt_module_load_hook != RT_NULL)
    {
        rt_module_load_hook(module);
    }
qiuyiuestc's avatar
qiuyiuestc 已提交
845 846
#endif

847
    return module;
848 849 850 851
}

#ifdef RT_USING_DFS
#include <dfs_posix.h>
qiuyiuestc's avatar
qiuyiuestc 已提交
852

853
static char* _module_name(const char *path)
qiuyiuestc's avatar
qiuyiuestc 已提交
854
{
855
    const char *first, *end, *ptr;
856
    char *name;
qiuyiuestc's avatar
qiuyiuestc 已提交
857
    int size;
858

859 860 861 862 863
    ptr   = (char *)path;
    first = ptr;
    end   = path + rt_strlen(path);

    while (*ptr != '\0')
qiuyiuestc's avatar
qiuyiuestc 已提交
864
    {
865 866 867 868
        if (*ptr == '/')
            first = ptr + 1;
        if (*ptr == '.')
            end = ptr - 1;
869

870
        ptr ++;
qiuyiuestc's avatar
qiuyiuestc 已提交
871 872 873 874 875 876 877 878 879 880
    }

    size = end - first + 1;
    name = rt_malloc(size);
    rt_strncpy(name, first, size);
    name[size] = '\0';

    return name;
}

qiuyiuestc's avatar
qiuyiuestc 已提交
881
/**
B
bernard.xiong@gmail.com 已提交
882
 * This function will load a module from a file
qiuyiuestc's avatar
qiuyiuestc 已提交
883
 *
B
bernard.xiong@gmail.com 已提交
884
 * @param path the full path of application module
qiuyiuestc's avatar
qiuyiuestc 已提交
885 886 887
 *
 * @return the module object
 */
D
dzzxzz 已提交
888
rt_module_t rt_module_open(const char *path)
qiuyiuestc's avatar
qiuyiuestc 已提交
889
{
890 891 892 893
    int fd, length;
    struct rt_module *module;
    struct stat s;
    char *buffer, *offset_ptr;
894
    char *name;
895 896 897 898 899 900 901 902

    RT_DEBUG_NOT_IN_INTERRUPT;

    /* check parameters */
    RT_ASSERT(path != RT_NULL);

    if (stat(path, &s) !=0)
    {
903
        rt_kprintf("Module: access %s failed\n", path);
904 905 906 907 908 909

        return RT_NULL;
    }
    buffer = (char *)rt_malloc(s.st_size);
    if (buffer == RT_NULL)
    {
910
        rt_kprintf("Module: out of memory\n");
911 912 913 914 915 916 917 918

        return RT_NULL;
    }

    offset_ptr = buffer;
    fd = open(path, O_RDONLY, 0);
    if (fd < 0)
    {
919
        rt_kprintf("Module: open %s failed\n", path);
920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938
        rt_free(buffer);

        return RT_NULL;
    }

    do
    {
        length = read(fd, offset_ptr, 4096);
        if (length > 0)
        {
            offset_ptr += length;
        }
    }while (length > 0);

    /* close fd */
    close(fd);

    if ((rt_uint32_t)offset_ptr - (rt_uint32_t)buffer != s.st_size)
    {
939
        rt_kprintf("Module: read file failed\n");
940 941 942 943 944
        rt_free(buffer);

        return RT_NULL;
    }

945 946
    name   = _module_name(path);
    module = rt_module_load(name, (void *)buffer);
947
    rt_free(buffer);
qiuyiuestc's avatar
qiuyiuestc 已提交
948
    rt_free(name);
949 950

    return module;
qiuyiuestc's avatar
qiuyiuestc 已提交
951
}
qiuyiuestc's avatar
qiuyiuestc 已提交
952 953 954

#if defined(RT_USING_FINSH)
#include <finsh.h>
955

956
FINSH_FUNCTION_EXPORT_ALIAS(rt_module_open, exec, exec module from a file);
qiuyiuestc's avatar
qiuyiuestc 已提交
957
#endif
958

qiuyiuestc's avatar
qiuyiuestc 已提交
959 960
#endif

qiuyiuestc's avatar
qiuyiuestc 已提交
961
/**
962
 * This function will destroy a module and release its resource.
qiuyiuestc's avatar
qiuyiuestc 已提交
963
 *
964
 * @param module the module to be destroyed.
qiuyiuestc's avatar
qiuyiuestc 已提交
965 966 967
 *
 * @return the operation status, RT_EOK on OK; -RT_ERROR on error
 */
968
rt_err_t rt_module_destroy(rt_module_t module)
qiuyiuestc's avatar
qiuyiuestc 已提交
969
{
970 971 972 973 974 975 976 977
    int i;
    struct rt_object *object;
    struct rt_list_node *list;

    RT_DEBUG_NOT_IN_INTERRUPT;

    /* check parameter */
    RT_ASSERT(module != RT_NULL);
978
    RT_ASSERT(module->nref == 0);
979

980 981
    RT_DEBUG_LOG(RT_DEBUG_MODULE, ("rt_module_destroy: %8.*s\n",
                                   RT_NAME_MAX, module->parent.name));
982 983 984 985

    /* module has entry point */
    if (!(module->parent.flag & RT_MODULE_FLAG_WITHOUTENTRY))
    {
qiuyiuestc's avatar
qiuyiuestc 已提交
986
#ifdef RT_USING_SEMAPHORE
987
        /* delete semaphores */
988
        list = &module->module_object[RT_Object_Class_Thread].object_list;
989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002
        while (list->next != list)
        {
            object = rt_list_entry(list->next, struct rt_object, list);
            if (rt_object_is_systemobject(object) == RT_TRUE)
            {
                /* detach static object */
                rt_sem_detach((rt_sem_t)object);
            }
            else
            {
                /* delete dynamic object */
                rt_sem_delete((rt_sem_t)object);
            }
        }
qiuyiuestc's avatar
qiuyiuestc 已提交
1003 1004 1005
#endif

#ifdef RT_USING_MUTEX
1006
        /* delete mutexs*/
1007
        list = &module->module_object[RT_Object_Class_Mutex].object_list;
1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021
        while (list->next != list)
        {
            object = rt_list_entry(list->next, struct rt_object, list);
            if (rt_object_is_systemobject(object) == RT_TRUE)
            {
                /* detach static object */
                rt_mutex_detach((rt_mutex_t)object);
            }
            else
            {
                /* delete dynamic object */
                rt_mutex_delete((rt_mutex_t)object);
            }
        }
qiuyiuestc's avatar
qiuyiuestc 已提交
1022 1023 1024
#endif

#ifdef RT_USING_EVENT
1025
        /* delete mailboxs */
1026
        list = &module->module_object[RT_Object_Class_Event].object_list;
1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040
        while (list->next != list)
        {
            object = rt_list_entry(list->next, struct rt_object, list);
            if (rt_object_is_systemobject(object) == RT_TRUE)
            {
                /* detach static object */
                rt_event_detach((rt_event_t)object);
            }
            else
            {
                /* delete dynamic object */
                rt_event_delete((rt_event_t)object);
            }
        }
qiuyiuestc's avatar
qiuyiuestc 已提交
1041 1042 1043
#endif

#ifdef RT_USING_MAILBOX
1044
        /* delete mailboxs */
1045
        list = &module->module_object[RT_Object_Class_MailBox].object_list;
1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059
        while (list->next != list)
        {
            object = rt_list_entry(list->next, struct rt_object, list);
            if (rt_object_is_systemobject(object) == RT_TRUE)
            {
                /* detach static object */
                rt_mb_detach((rt_mailbox_t)object);
            }
            else
            {
                /* delete dynamic object */
                rt_mb_delete((rt_mailbox_t)object);
            }
        }
qiuyiuestc's avatar
qiuyiuestc 已提交
1060 1061 1062
#endif

#ifdef RT_USING_MESSAGEQUEUE
1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078
        /* delete msgqueues */
        list = &module->module_object[RT_Object_Class_MessageQueue].object_list;
        while (list->next != list)
        {
            object = rt_list_entry(list->next, struct rt_object, list);
            if (rt_object_is_systemobject(object) == RT_TRUE)
            {
                /* detach static object */
                rt_mq_detach((rt_mq_t)object);
            }
            else
            {
                /* delete dynamic object */
                rt_mq_delete((rt_mq_t)object);
            }
        }
qiuyiuestc's avatar
qiuyiuestc 已提交
1079 1080 1081
#endif

#ifdef RT_USING_MEMPOOL
1082
        /* delete mempools */
1083
        list = &module->module_object[RT_Object_Class_MemPool].object_list;
1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097
        while (list->next != list)
        {
            object = rt_list_entry(list->next, struct rt_object, list);
            if (rt_object_is_systemobject(object) == RT_TRUE)
            {
                /* detach static object */
                rt_mp_detach((rt_mp_t)object);
            }
            else
            {
                /* delete dynamic object */
                rt_mp_delete((rt_mp_t)object);
            }
        }
qiuyiuestc's avatar
qiuyiuestc 已提交
1098 1099 1100
#endif

#ifdef RT_USING_DEVICE
1101
        /* delete devices */
1102
        list = &module->module_object[RT_Object_Class_Device].object_list;
1103 1104 1105 1106 1107
        while (list->next != list)
        {
            object = rt_list_entry(list->next, struct rt_object, list);
            rt_device_unregister((rt_device_t)object);
        }
qiuyiuestc's avatar
qiuyiuestc 已提交
1108 1109
#endif

1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126
        /* delete timers */
        list = &module->module_object[RT_Object_Class_Timer].object_list;
        while (list->next != list)
        {
            object = rt_list_entry(list->next, struct rt_object, list);
            if (rt_object_is_systemobject(object) == RT_TRUE)
            {
                /* detach static object */
                rt_timer_detach((rt_timer_t)object);
            }
            else
            {
                /* delete dynamic object */
                rt_timer_delete((rt_timer_t)object);
            }
        }
    }
qiuyiuestc's avatar
qiuyiuestc 已提交
1127

1128
#ifdef RT_USING_SLAB
1129 1130 1131
    if (module->page_cnt > 0)
    {
        struct rt_page_info *page = (struct rt_page_info *)module->page_array;
B
bernard.xiong@gmail.com 已提交
1132

1133
        rt_kprintf("Module: warning - memory still hasn't been free finished\n");
qiuyiuestc's avatar
qiuyiuestc 已提交
1134

1135
        while (module->page_cnt != 0)
1136 1137 1138 1139
        {
            rt_module_free_page(module, page[0].page_ptr, page[0].npage);
        }
    }
1140
#endif
D
dzzxzz 已提交
1141

1142 1143
    /* release module space memory */
    rt_free(module->module_space);
qiuyiuestc's avatar
qiuyiuestc 已提交
1144

1145
    /* release module symbol table */
1146
    for (i = 0; i < module->nsym; i ++)
1147
    {
1148
        rt_free((void *)module->symtab[i].name);
1149
    }
1150 1151
    if (module->symtab != RT_NULL)
        rt_free(module->symtab);
qiuyiuestc's avatar
qiuyiuestc 已提交
1152

qiuyiuestc's avatar
qiuyiuestc 已提交
1153
#ifdef RT_USING_SLAB
1154
    if (module->page_array != RT_NULL)
1155
        rt_free(module->page_array);
qiuyiuestc's avatar
qiuyiuestc 已提交
1156
#endif
qiuyiuestc's avatar
qiuyiuestc 已提交
1157

1158 1159
    /* delete module object */
    rt_object_delete((rt_object_t)module);
qiuyiuestc's avatar
qiuyiuestc 已提交
1160

1161
    return RT_EOK;
qiuyiuestc's avatar
qiuyiuestc 已提交
1162 1163
}

1164 1165 1166 1167 1168 1169 1170 1171 1172 1173
/**
 * This function will unload a module from memory and release resources
 *
 * @param module the module to be unloaded
 *
 * @return the operation status, RT_EOK on OK; -RT_ERROR on error
 */
rt_err_t rt_module_unload(rt_module_t module)
{
    int i;
1174
    rt_err_t result;
1175 1176 1177 1178 1179 1180
    struct rt_object *object;
    struct rt_list_node *list;

    RT_DEBUG_NOT_IN_INTERRUPT;

    /* check parameter */
1181 1182
    if (module == RT_NULL)
        return -RT_ERROR;
1183

1184
    rt_enter_critical();
1185 1186
    if (!(module->parent.flag & RT_MODULE_FLAG_WITHOUTENTRY))
    {
1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208
        /* delete all sub-threads */
        list = &module->module_object[RT_Object_Class_Thread].object_list;
        while (list->next != list)
        {
            object = rt_list_entry(list->next, struct rt_object, list);
            if (rt_object_is_systemobject(object) == RT_TRUE)
            {
                /* detach static object */
                rt_thread_detach((rt_thread_t)object);
            }
            else
            {
                /* delete dynamic object */
                rt_thread_delete((rt_thread_t)object);
            }
        }

        /* delete the main thread of module */
        if (module->module_thread != RT_NULL)
        {
            rt_thread_delete(module->module_thread);
        }
1209
    }
1210
    rt_exit_critical();
1211 1212

#ifdef RT_USING_HOOK
1213 1214 1215 1216
    if (rt_module_unload_hook != RT_NULL)
    {
        rt_module_unload_hook(module);
    }
1217 1218
#endif

1219
    return RT_EOK;
1220 1221
}

qiuyiuestc's avatar
qiuyiuestc 已提交
1222 1223 1224 1225 1226 1227 1228
/**
 * This function will find the specified module.
 *
 * @param name the name of module finding
 *
 * @return the module
 */
D
dzzxzz 已提交
1229
rt_module_t rt_module_find(const char *name)
qiuyiuestc's avatar
qiuyiuestc 已提交
1230
{
1231 1232 1233
    struct rt_object_information *information;
    struct rt_object *object;
    struct rt_list_node *node;
qiuyiuestc's avatar
qiuyiuestc 已提交
1234

1235
    extern struct rt_object_information rt_object_container[];
B
bernard.xiong 已提交
1236

1237
    RT_DEBUG_NOT_IN_INTERRUPT;
1238

1239 1240
    /* enter critical */
    rt_enter_critical();
B
bernard.xiong 已提交
1241

1242 1243
    /* try to find device object */
    information = &rt_object_container[RT_Object_Class_Module];
1244
    for (node = information->object_list.next;
1245 1246
         node != &(information->object_list);
         node = node->next)
1247 1248 1249 1250 1251 1252
    {
        object = rt_list_entry(node, struct rt_object, list);
        if (rt_strncmp(object->name, name, RT_NAME_MAX) == 0)
        {
            /* leave critical */
            rt_exit_critical();
B
bernard.xiong 已提交
1253

1254 1255 1256
            return (rt_module_t)object;
        }
    }
B
bernard.xiong 已提交
1257

1258 1259
    /* leave critical */
    rt_exit_critical();
B
bernard.xiong 已提交
1260

1261 1262
    /* not found */
    return RT_NULL;
qiuyiuestc's avatar
qiuyiuestc 已提交
1263 1264
}

qiuyiuestc's avatar
qiuyiuestc 已提交
1265 1266 1267 1268 1269 1270 1271 1272 1273
#ifdef RT_USING_SLAB
/*
 * This function will allocate the numbers page with specified size
 * in page memory.
 *
 * @param size the size of memory to be allocated.
 * @note this function is used for RT-Thread Application Module
 */
static void *rt_module_malloc_page(rt_size_t npages)
qiuyiuestc's avatar
qiuyiuestc 已提交
1274
{
1275 1276
    void *chunk;
    struct rt_page_info *page;
1277
    rt_module_t self_module;
1278

1279 1280
    self_module = rt_module_self();
    RT_ASSERT(self_module != RT_NULL);
qiuyiuestc's avatar
qiuyiuestc 已提交
1281

1282 1283 1284
    chunk = rt_page_alloc(npages);
    if (chunk == RT_NULL)
        return RT_NULL;
1285

1286 1287
    page = (struct rt_page_info *)self_module->page_array;
    page[self_module->page_cnt].page_ptr = chunk;
1288
    page[self_module->page_cnt].npage    = npages;
1289
    self_module->page_cnt ++;
qiuyiuestc's avatar
qiuyiuestc 已提交
1290

1291
    RT_ASSERT(self_module->page_cnt <= PAGE_COUNT_MAX);
1292 1293
    RT_DEBUG_LOG(RT_DEBUG_MODULE, ("rt_module_malloc_page 0x%x %d\n",
                                   chunk, npages));
D
dzzxzz 已提交
1294

1295
    return chunk;
qiuyiuestc's avatar
qiuyiuestc 已提交
1296
}
qiuyiuestc's avatar
qiuyiuestc 已提交
1297

qiuyiuestc's avatar
qiuyiuestc 已提交
1298
/*
D
dzzxzz 已提交
1299
 * This function will release the previously allocated memory page
qiuyiuestc's avatar
qiuyiuestc 已提交
1300 1301 1302 1303
 * by rt_malloc_page.
 *
 * @param page_ptr the page address to be released.
 * @param npages the number of page shall be released.
B
bernard.xiong@gmail.com 已提交
1304
 *
qiuyiuestc's avatar
qiuyiuestc 已提交
1305 1306
 * @note this function is used for RT-Thread Application Module
 */
1307 1308 1309
static void rt_module_free_page(rt_module_t module,
                                void       *page_ptr,
                                rt_size_t   npages)
qiuyiuestc's avatar
qiuyiuestc 已提交
1310
{
1311 1312
    int i, index;
    struct rt_page_info *page;
1313
    rt_module_t self_module;
1314

1315 1316
    self_module = rt_module_self();
    RT_ASSERT(self_module != RT_NULL);
1317

1318 1319
    RT_DEBUG_LOG(RT_DEBUG_MODULE, ("rt_module_free_page 0x%x %d\n",
                                   page_ptr, npages));
1320 1321
    rt_page_free(page_ptr, npages);

1322
    page = (struct rt_page_info *)module->page_array;
1323

1324
    for (i = 0; i < module->page_cnt; i ++)
1325 1326 1327 1328 1329
    {
        if (page[i].page_ptr == page_ptr)
        {
            if (page[i].npage == npages + 1)
            {
1330 1331 1332
                page[i].page_ptr +=
                    npages * RT_MM_PAGE_SIZE / sizeof(rt_uint32_t);
                page[i].npage    -= npages;
1333
            }
1334
            else if (page[i].npage == npages)
1335
            {
1336
                for (index = i; index < module->page_cnt-1; index ++)
1337 1338
                {
                    page[index].page_ptr = page[index + 1].page_ptr;
1339
                    page[index].npage    = page[index + 1].npage;
1340 1341
                }
                page[module->page_cnt - 1].page_ptr = RT_NULL;
1342
                page[module->page_cnt - 1].npage    = 0;
1343

1344
                module->page_cnt --;
1345 1346 1347
            }
            else
                RT_ASSERT(RT_FALSE);
1348
            self_module->page_cnt --;
1349 1350 1351 1352 1353

            return;
        }
    }

1354
    /* should not get here */
1355
    RT_ASSERT(RT_FALSE);
qiuyiuestc's avatar
qiuyiuestc 已提交
1356
}
qiuyiuestc's avatar
qiuyiuestc 已提交
1357

1358 1359 1360
/**
 * rt_module_malloc - allocate memory block in free list
 */
qiuyiuestc's avatar
qiuyiuestc 已提交
1361 1362
void *rt_module_malloc(rt_size_t size)
{
1363 1364 1365 1366
    struct rt_mem_head *b, *n, *up;
    struct rt_mem_head **prev;
    rt_uint32_t npage;
    rt_size_t nunits;
1367
    rt_module_t self_module;
1368

1369 1370
    self_module = rt_module_self();
    RT_ASSERT(self_module != RT_NULL);
1371

1372
    RT_DEBUG_NOT_IN_INTERRUPT;
1373

1374 1375 1376
    nunits = (size + sizeof(struct rt_mem_head) - 1) /
             sizeof(struct rt_mem_head)
             + 1;
qiuyiuestc's avatar
qiuyiuestc 已提交
1377

1378 1379
    RT_ASSERT(size != 0);
    RT_ASSERT(nunits != 0);
qiuyiuestc's avatar
qiuyiuestc 已提交
1380

1381
    rt_sem_take(&mod_sem, RT_WAITING_FOREVER);
qiuyiuestc's avatar
qiuyiuestc 已提交
1382

1383
    for (prev = (struct rt_mem_head **)&self_module->mem_list;
1384 1385
         (b = *prev) != RT_NULL;
         prev = &(b->next))
1386 1387 1388 1389
    {
        if (b->size > nunits)
        {
            /* split memory */
1390
            n       = b + nunits;
1391 1392 1393
            n->next = b->next;
            n->size = b->size - nunits;
            b->size = nunits;
1394
            *prev   = n;
qiuyiuestc's avatar
qiuyiuestc 已提交
1395

1396 1397
            RT_DEBUG_LOG(RT_DEBUG_MODULE, ("rt_module_malloc 0x%x, %d\n",
                                           b + 1, size));
1398
            rt_sem_release(&mod_sem);
D
dzzxzz 已提交
1399

1400 1401
            return (void *)(b + 1);
        }
qiuyiuestc's avatar
qiuyiuestc 已提交
1402

1403 1404 1405 1406
        if (b->size == nunits)
        {
            /* this node fit, remove this node */
            *prev = b->next;
qiuyiuestc's avatar
qiuyiuestc 已提交
1407

1408 1409
            RT_DEBUG_LOG(RT_DEBUG_MODULE, ("rt_module_malloc 0x%x, %d\n",
                                           b + 1, size));
B
bernard.xiong@gmail.com 已提交
1410

1411
            rt_sem_release(&mod_sem);
D
dzzxzz 已提交
1412

1413 1414 1415
            return (void *)(b + 1);
        }
    }
qiuyiuestc's avatar
qiuyiuestc 已提交
1416

1417
    /* allocate pages from system heap */
1418 1419
    npage = (size + sizeof(struct rt_mem_head) + RT_MM_PAGE_SIZE - 1) /
            RT_MM_PAGE_SIZE;
1420 1421
    if ((up = (struct rt_mem_head *)rt_module_malloc_page(npage)) == RT_NULL)
        return RT_NULL;
qiuyiuestc's avatar
qiuyiuestc 已提交
1422

1423
    up->size = npage * RT_MM_PAGE_SIZE / sizeof(struct rt_mem_head);
B
bernard.xiong@gmail.com 已提交
1424

1425
    for (prev = (struct rt_mem_head **)&self_module->mem_list;
1426 1427
         (b = *prev) != RT_NULL;
         prev = &(b->next))
1428 1429 1430 1431
    {
        if (b > up + up->size)
            break;
    }
qiuyiuestc's avatar
qiuyiuestc 已提交
1432

1433
    up->next = b;
1434
    *prev    = up;
qiuyiuestc's avatar
qiuyiuestc 已提交
1435

1436
    rt_sem_release(&mod_sem);
D
dzzxzz 已提交
1437

1438
    return rt_module_malloc(size);
qiuyiuestc's avatar
qiuyiuestc 已提交
1439 1440
}

1441 1442 1443
/**
 * rt_module_free - free memory block in free list
 */
qiuyiuestc's avatar
qiuyiuestc 已提交
1444 1445
void rt_module_free(rt_module_t module, void *addr)
{
1446 1447 1448 1449 1450 1451 1452 1453
    struct rt_mem_head *b, *n, *r;
    struct rt_mem_head **prev;

    RT_DEBUG_NOT_IN_INTERRUPT;

    RT_ASSERT(addr);
    RT_ASSERT((((rt_uint32_t)addr) & (sizeof(struct rt_mem_head) -1)) == 0);

1454
    RT_DEBUG_LOG(RT_DEBUG_MODULE, ("rt_module_free 0x%x\n", addr));
1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477

    rt_sem_take(&mod_sem, RT_WAITING_FOREVER);

    n = (struct rt_mem_head *)addr - 1;
    prev = (struct rt_mem_head **)&module->mem_list;

    while ((b = *prev) != RT_NULL)
    {
        RT_ASSERT(b->size > 0);
        RT_ASSERT(b > n || b + b->size <= n);

        if (b + b->size == n && ((rt_uint32_t)n % RT_MM_PAGE_SIZE != 0))
        {
            if (b + (b->size + n->size) == b->next)
            {
                b->size += b->next->size + n->size;
                b->next = b->next->next;
            }
            else
                b->size += n->size;

            if ((rt_uint32_t)b % RT_MM_PAGE_SIZE == 0)
            {
1478 1479
                int npage =
                    b->size * sizeof(struct rt_page_info) / RT_MM_PAGE_SIZE;
1480 1481 1482 1483
                if (npage > 0)
                {
                    if ((b->size * sizeof(struct rt_page_info) % RT_MM_PAGE_SIZE) != 0)
                    {
1484 1485 1486
                        rt_size_t nunits = npage *
                                           RT_MM_PAGE_SIZE /
                                           sizeof(struct rt_mem_head);
1487
                        /* split memory */
1488
                        r       = b + nunits;
1489 1490
                        r->next = b->next;
                        r->size = b->size - nunits;
1491
                        *prev   = r;
1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514
                    }
                    else
                    {
                        *prev = b->next;
                    }

                    rt_module_free_page(module, b, npage);
                }
            }

            /* unlock */
            rt_sem_release(&mod_sem);

            return;
        }

        if (b == n + n->size)
        {
            n->size = b->size + n->size;
            n->next = b->next;

            if ((rt_uint32_t)n % RT_MM_PAGE_SIZE == 0)
            {
1515 1516
                int npage =
                    n->size * sizeof(struct rt_page_info) / RT_MM_PAGE_SIZE;
1517 1518 1519 1520
                if (npage > 0)
                {
                    if ((n->size * sizeof(struct rt_page_info) % RT_MM_PAGE_SIZE) != 0)
                    {
1521 1522 1523
                        rt_size_t nunits = npage *
                                           RT_MM_PAGE_SIZE /
                                           sizeof(struct rt_mem_head);
1524
                        /* split memory */
1525
                        r       = n + nunits;
1526 1527
                        r->next = n->next;
                        r->size = n->size - nunits;
1528
                        *prev   = r;
1529
                    }
1530 1531
                    else
                        *prev = n->next;
1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559

                    rt_module_free_page(module, n, npage);
                }
            }
            else
            {
                *prev = n;
            }

            /* unlock */
            rt_sem_release(&mod_sem);

            return;
        }
        if (b > n + n->size)
            break;

        prev = &(b->next);
    }

    if ((rt_uint32_t)n % RT_MM_PAGE_SIZE == 0)
    {
        int npage = n->size * sizeof(struct rt_page_info) / RT_MM_PAGE_SIZE;
        if (npage > 0)
        {
            rt_module_free_page(module, n, npage);
            if (n->size % RT_MM_PAGE_SIZE != 0)
            {
1560 1561
                rt_size_t nunits =
                    npage * RT_MM_PAGE_SIZE / sizeof(struct rt_mem_head);
1562
                /* split memory */
1563
                r       = n + nunits;
1564 1565
                r->next = b;
                r->size = n->size - nunits;
1566
                *prev   = r;
1567 1568 1569 1570 1571 1572 1573 1574 1575 1576
            }
            else
            {
                *prev = b;
            }
        }
    }
    else
    {
        n->next = b;
1577
        *prev   = n;
1578 1579 1580 1581
    }

    /* unlock */
    rt_sem_release(&mod_sem);
qiuyiuestc's avatar
qiuyiuestc 已提交
1582 1583
}

1584 1585 1586
/**
 * rt_module_realloc - realloc memory block in free list
 */
qiuyiuestc's avatar
qiuyiuestc 已提交
1587 1588
void *rt_module_realloc(void *ptr, rt_size_t size)
{
1589 1590
    struct rt_mem_head *b, *p, *prev, *tmpp;
    rt_size_t nunits;
1591
    rt_module_t self_module;
1592

1593 1594
    self_module = rt_module_self();
    RT_ASSERT(self_module != RT_NULL);
1595 1596 1597 1598 1599 1600 1601

    RT_DEBUG_NOT_IN_INTERRUPT;

    if (!ptr)
        return rt_module_malloc(size);
    if (size == 0)
    {
1602
        rt_module_free(self_module, ptr);
1603 1604 1605 1606

        return RT_NULL;
    }

1607 1608 1609
    nunits = (size + sizeof(struct rt_mem_head) - 1) /
             sizeof(struct rt_mem_head)
             +1;
1610 1611 1612 1613 1614 1615 1616 1617 1618
    b = (struct rt_mem_head *)ptr - 1;

    if (nunits <= b->size)
    {
        /* new size is smaller or equal then before */
        if (nunits == b->size)
            return ptr;
        else
        {
1619
            p       = b + nunits;
1620 1621
            p->size = b->size - nunits;
            b->size = nunits;
1622
            rt_module_free(self_module, (void *)(p + 1));
1623 1624 1625 1626 1627 1628 1629

            return (void *)(b + 1);
        }
    }
    else
    {
        /* more space then required */
1630
        prev = (struct rt_mem_head *)self_module->mem_list;
1631 1632 1633 1634
        for (p = prev->next;
             p != (b->size + b) && p != RT_NULL;
             prev = p, p = p->next)
        {
1635
            break;
1636
        }
1637

1638
        /* available block after ap in freelist */
1639 1640 1641
        if (p != RT_NULL &&
            (p->size >= (nunits - (b->size))) &&
            p == (b + b->size))
1642 1643
        {
            /* perfect match */
1644
            if (p->size == (nunits - (b->size)))
1645
            {
1646
                b->size    = nunits;
1647 1648
                prev->next = p->next;
            }
1649
            else  /* more space then required, split block */
1650 1651 1652
            {
                /* pointer to old header */
                tmpp = p;
1653
                p    = b + nunits;
1654 1655 1656

                /* restoring old pointer */
                p->next = tmpp->next;
1657

1658
                /* new size for p */
1659 1660
                p->size    = tmpp->size + b->size - nunits;
                b->size    = nunits;
1661 1662
                prev->next = p;
            }
1663
            self_module->mem_list = (void *)prev;
1664 1665 1666 1667 1668

            return (void *)(b + 1);
        }
        else /* allocate new memory and copy old data */
        {
1669 1670
            if ((p = rt_module_malloc(size)) == RT_NULL)
                return RT_NULL;
1671
            rt_memmove(p, (b+1), ((b->size) * sizeof(struct rt_mem_head)));
1672
            rt_module_free(self_module, (void *)(b + 1));
1673 1674 1675 1676

            return (void *)(p);
        }
    }
qiuyiuestc's avatar
qiuyiuestc 已提交
1677
}
qiuyiuestc's avatar
qiuyiuestc 已提交
1678 1679 1680

#ifdef RT_USING_FINSH
#include <finsh.h>
1681

D
dzzxzz 已提交
1682
void list_memlist(const char *name)
qiuyiuestc's avatar
qiuyiuestc 已提交
1683
{
1684 1685 1686 1687 1688 1689 1690 1691
    rt_module_t module;
    struct rt_mem_head **prev;
    struct rt_mem_head *b;

    module = rt_module_find(name);
    if (module == RT_NULL)
        return;

1692
    for (prev = (struct rt_mem_head **)&module->mem_list;
1693 1694
         (b = *prev) != RT_NULL;
         prev = &(b->next))
1695 1696 1697
    {
        rt_kprintf("0x%x--%d\n", b, b->size * sizeof(struct rt_mem_head));
    }
qiuyiuestc's avatar
qiuyiuestc 已提交
1698 1699 1700
}
FINSH_FUNCTION_EXPORT(list_memlist, list module free memory information)

D
dzzxzz 已提交
1701
void list_mempage(const char *name)
qiuyiuestc's avatar
qiuyiuestc 已提交
1702
{
1703 1704 1705
    rt_module_t module;
    struct rt_page_info *page;
    int i;
qiuyiuestc's avatar
qiuyiuestc 已提交
1706

1707 1708 1709
    module = rt_module_find(name);
    if (module == RT_NULL)
        return;
qiuyiuestc's avatar
qiuyiuestc 已提交
1710

1711
    page = (struct rt_page_info *)module->page_array;
qiuyiuestc's avatar
qiuyiuestc 已提交
1712

1713
    for (i = 0; i < module->page_cnt; i ++)
1714 1715 1716
    {
        rt_kprintf("0x%x--%d\n", page[i].page_ptr, page[i].npage);
    }
qiuyiuestc's avatar
qiuyiuestc 已提交
1717 1718 1719
}
FINSH_FUNCTION_EXPORT(list_mempage, list module using memory page information)
#endif
qiuyiuestc's avatar
qiuyiuestc 已提交
1720

qiuyiuestc's avatar
qiuyiuestc 已提交
1721
#endif
qiuyiuestc's avatar
qiuyiuestc 已提交
1722 1723

#endif