本文轉(zhuǎn)自公眾號，歡迎關(guān)注

https://mp.weixin.qq.com/s/mBoGSf_u9edFF01U_OZT9g

一.前言

lwIP為基礎(chǔ)結(jié)構(gòu)提供了專用的內(nèi)存池管理,比如netconn,protocol控制塊,包緩存等。在memp.c下實現(xiàn)。

LWIP的內(nèi)存池有兩種方式實現(xiàn),通過宏MEMP_MEM_MALLOC配置,默認(rèn)opt.h中配置為0.

配置為1使用mem_malloc/mem_free mem.c

配置為0使用單獨實現(xiàn)memp.c。

我們這里重點講后者。

二.相關(guān)源碼

src/core/memp.c

src/include/lwip/memp.h

src/include/lwip/priv/memp_std.h

src/include/lwip/priv/memp_priv.h

三.源碼分析

3.1數(shù)據(jù)結(jié)構(gòu)

內(nèi)存池的關(guān)鍵數(shù)據(jù)結(jié)構(gòu)是struct memp_desc對應(yīng)內(nèi)存池節(jié)點，一個類型的內(nèi)存池是一個節(jié)點，

多個類型的內(nèi)存池可以作為鏈表一起管理。

內(nèi)存池最基本的數(shù)據(jù)結(jié)構(gòu)是由宏LWIP_MEMPOOL_DECLARE定義的,

該宏在memp.h根據(jù)MEMP_MEM_MALLOC的配置實現(xiàn)為

#define LWIP_MEMPOOL_DECLARE(name,num,size,desc)

和

#define LWIP_MEMPOOL_DECLARE(name,num,size,desc)

前者是使用mem_malloc/mem_free實現(xiàn)內(nèi)存池時使用,后者是單獨實現(xiàn)時使用，我們重點關(guān)注后者。

其中LWIP_DECLARE_MEMORY_ALIGNED可以用戶實現(xiàn),如果用戶未定義,arch.h默認(rèn)是

#ifndef LWIP_DECLARE_MEMORY_ALIGNED

即定義了一個數(shù)組variable_name,其大小是LWIP_MEM_ALIGN_BUFFER(size)

LWIP_MEM_ALIGN_BUFFER可以用戶實現(xiàn)，如果用戶未定義，arch.h中默認(rèn)為

#ifndef LWIP_MEM_ALIGN_BUFFER

即人為的放大了,保證對齊之后始終能保證有足夠size的空間。

比如size是8,但是數(shù)組的基地址則可能是任意地址,比如是0x0001,要保證地址4字節(jié)對齊，

那么只能往后移動實際用的地址是0x0004,那么前面就浪費了3字節(jié),此時8+(4-1)多分配3字節(jié)則浪費了這3字節(jié)也能保證剩余8字節(jié)可用8+(4-1)-3=8。

如果地址是0x0002則浪費2字節(jié)，可用8+(4-1)-2>8大于2字節(jié),其他情況類似。

其中MEM_ALIGNMENT在opt中默認(rèn)為1，用戶在lwipopts.h中可以配置。

繼續(xù)來看宏LWIP_MEMPOOL_DECLARE_STATS_INSTANCE和LWIP_MEMPOOL_DECLARE_STATS_REFERENCE，memp_priv.h中

如果定義了宏MEMP_STATS即使能統(tǒng)計信息,則LWIP_MEMPOOL_DECLARE_STATS_INSTANCE定義了結(jié)構(gòu)體變量static struct stats_mem name;

LWIP_MEMPOOL_DECLARE_STATS_REFERENCE即&name,該變量地址

否則都是空

#if MEMP_STATS

其中stats.h中truct stats_mem如下

/** Memory stats */

繼續(xù)看DECLARE_LWIP_MEMPOOL_DESC，memp_priv.h中

如果定義了宏LWIP_DEBUG，MEMP_OVERFLOW_CHECK，LWIP_STATS_DISPLAY三者之一，實際就是成員賦值為desc，否則為空。

#if defined(LWIP_DEBUG) || MEMP_OVERFLOW_CHECK || LWIP_STATS_DISPLAY

繼續(xù)看LWIP_MEM_ALIGN_SIZE，在arch.h中默認(rèn)實現(xiàn)，即向上對齊，MEM_ALIGNMENT對齊必須是2的指數(shù)。

#ifndef LWIP_MEM_ALIGN_SIZE

最終LWIP_MEMPOOL_DECLARE(name,num,size,desc)展開后為

u8_t memp_memory_name_base[LWIP_MEM_ALIGN_BUFFER(((num) * (MEMP_SIZE + MEMP_ALIGN_SIZE(size))))]

static struct stats_mem memp_stats_name;

static struct memp*memp_tab_name;

const struct memp_desc memp_name= {

desc,

&memp_stats_name,

LWIP_MEM_ALIGN_SIZE(size),

num,

memp_memory_ name_base,

&memp_tab_name

};

即定義一個數(shù)組作為存儲，一個統(tǒng)計變量，一個static struct memp指針變量，一個結(jié)構(gòu)變量static struct memp。

其中struct memp

#if !MEMP_MEM_MALLOC || MEMP_OVERFLOW_CHECK

其中struct memp_desc

/** Memory pool descriptor */

3.2內(nèi)存池接口

LWIP_MEMPOOL_DECLARE(my_private_pool, 10, sizeof(foo), "Some description")

即上面數(shù)據(jù)結(jié)構(gòu)說明的，使用該宏定義節(jié)點相關(guān)存儲數(shù)組，變量等。

LWIP_MEMPOOL_INIT(my_private_pool) ->memp_init_pool

將節(jié)點的存儲鏈表形式串起來。

*desc->tab指向最后一個item。

void* my_new_mem = LWIP_MEMPOOL_ALLOC(my_private_pool);->memp_malloc_pool

從鏈表中取出一個item。

LWIP_MEMPOOL_FREE(my_private_pool, my_new_mem);->memp_free_pool

Item插入到鏈表中

這里管理內(nèi)存池實際可以用上述的鏈表形式，也可以用bitmap形式。

上述用鏈表形式每次malloc(出鏈表)，free(入鏈表)都是堆鏈表尾,*desc->tab操作,執(zhí)行時間固定。

而bitmap需要使用for循環(huán)去查詢空閑bit執(zhí)行時間不固定，當(dāng)然也可以使用類似ucOS優(yōu)先級調(diào)度的查表法來優(yōu)化。

四.內(nèi)部使用的內(nèi)存池

數(shù)據(jù)結(jié)構(gòu)中分析了LWIP_MEMPOOL_DECLARE宏，實際就是定義內(nèi)存節(jié)點相關(guān)的變量(存儲數(shù)組,統(tǒng)計變量,描述結(jié)構(gòu)等)

用戶可以直接使用內(nèi)存池接口。

而lwip內(nèi)部也使用了內(nèi)存池進(jìn)行管理,見memp.c/memp.h

4.1內(nèi)部使用內(nèi)存池

memp.h中

/* run once with empty definition to handle all custom includes in lwippools.h */

第一個#include "lwip/priv/memp_std.h"前因為LWIP_MEMPOOL為空，所以include memp_std.h進(jìn)來所有的LWIP_MEMPOOL為空，此時相當(dāng)于只是include進(jìn)來memp_std.h中包含的頭文件。

后一個#include "lwip/priv/memp_std.h"前LWIP_MEMPOOL(name,num,size,desc) MEMP_##name,

所以include memp_std.h后相當(dāng)于，定義了MEMP_xxx的枚舉

展開就是(當(dāng)然要對應(yīng)的宏使能才會有)

typedef enum {

RAW_PCB_RAW_PCB,

UDP_PCB_UDP_PCB,

......

MEMP_MAX

} memp_t;

注意每次memp_std.h中都undef了相關(guān)宏，所以不影響后續(xù)使用。

這里是一個小的編程技巧，一個頭文件不同的宏配置下展開為不同的內(nèi)容。

而memp.c中

#define LWIP_MEMPOOL(name,num,size,desc) LWIP_MEMPOOL_DECLARE(name,num,size,desc)

第一個包含memp_std.h 根據(jù)LWIP_MEMPOOL_DECLARE，定義了各個內(nèi)存池節(jié)點

后一個memp_std.h則根據(jù)LWIP_MEMPOOL(name,num,size,desc) & name展開

即定義了一個結(jié)構(gòu)體數(shù)組，數(shù)組的成員即上面定義的內(nèi)存池節(jié)點。

const struct memp_desc *const memp_pools[MEMP_MAX] = {

&RAW_PCB,

......

};

所有使用到內(nèi)建內(nèi)存池總結(jié)如下

4.2內(nèi)部使用內(nèi)存池接口

以下都是調(diào)用內(nèi)存池接口,不再贅述。

memp_free

memp_free_pool

memp_malloc

memp_malloc_pool

memp_init

五.堆使用內(nèi)存池實現(xiàn)

前面我們看到定義宏MEMP_MEM_MALLOC,memp.c內(nèi)存池可以使用mem.c堆實現(xiàn).

反過來定義宏MEM_USE_POOLS,mem.c也可以用memp.c內(nèi)存池實現(xiàn).所以可以看出LWIP的堆管理實現(xiàn)方式比較靈活的,可以根據(jù)實際應(yīng)用配置。

在opt.h中MEM_USE_POOLS是默認(rèn)配置為0的,可以在lwipopts.h中修改配置。

#if !defined MEM_USE_POOLS || defined __DOXYGEN__

如果MEMP_USE_CUSTOM_POOLS配置為1，則MEMP_USE_CUSTOM_POOLS也要配置為1

此時

memp_std.h中

/*

可以看出，用戶必須提供lwippools.h文件，申明對應(yīng)的內(nèi)存池節(jié)點。

即在原來memp_t和memp_pools的基礎(chǔ)上后面繼續(xù)添加節(jié)點。

內(nèi)容如下

LWIP_MALLOC_MEMPOOL_START

此時mem_malloc則從

memp_pools中

MEMP_POOL_FIRST~MEMP_POOL_LAST處

搜搜，找到有節(jié)點有空閑空間,大小滿足所需大小的即止。

即如下獲取MEMP_POOL_HELPER_FIRST和MEMP_POOL_HELPER_LAST

#if MEM_USE_POOLS && MEMP_USE_CUSTOM_POOLS

六.總結(jié)

內(nèi)存池適合分配大小固定的動態(tài)內(nèi)存分配，比如用于協(xié)議包等的緩存等；

其算法簡單，執(zhí)行時間固定，比較可靠。但是其存儲是單獨分配的靜態(tài)數(shù)組，相當(dāng)于需要固定分配一部分空間，不管對應(yīng)的程序運行還是不運行，比較浪費空間。

當(dāng)然也可以定義宏MEMP_MEM_MALLOC使用mem_malloc堆的方式實現(xiàn)，而mem_malloc進(jìn)一步可以配置使用LWIP的實現(xiàn)還是使用系統(tǒng)的malloc(配置宏MEM_LIBC_MALLOC)。

此時和系統(tǒng)堆共用，這樣存儲利用率更高。

LWIP內(nèi)部實現(xiàn)了堆和內(nèi)存池管理可以直接使用,也可以配置使用系統(tǒng)的堆管理，非常靈活，移植性也非常好。

審核編輯 黃宇