如何綁架C語言標(biāo)準(zhǔn)庫函數(shù)？

C語言動(dòng)態(tài)內(nèi)存可以使用 malloc 標(biāo)準(zhǔn)庫函數(shù)，而 C++ 會(huì)使用 new 關(guān)鍵字申請(qǐng)，裸機(jī)情況下一般沒有問題，但是在 RTOS 情況下，就需要考慮資源保護(hù)的情況，比如多個(gè)任務(wù)同時(shí)調(diào)用 malloc 函數(shù)申請(qǐng)內(nèi)存，如果說不用關(guān)中斷或互斥量之類的手段進(jìn)行保護(hù)的，有可能同一個(gè)內(nèi)存塊被兩個(gè)任務(wù)申請(qǐng)，動(dòng)態(tài)內(nèi)存內(nèi)部維護(hù)的一些全局變量也可能出現(xiàn)問題。

因此，如果在 RTOS （不應(yīng)該在中斷使用）中使用 malloc 函數(shù)，一定要進(jìn)行保護(hù)，像一般的操作系統(tǒng)如 freeRTOS、RT-Thread 都有動(dòng)態(tài)內(nèi)存的實(shí)現(xiàn)，也強(qiáng)烈建議大家使用操作系統(tǒng)提供的動(dòng)態(tài)內(nèi)存函數(shù)，這樣安全性有一定保證。

但是有時(shí)候我們因?yàn)槟承┰?，需要維護(hù)老的項(xiàng)目，而老的項(xiàng)目大量的使用 malloc 函數(shù)和 new，此時(shí)又該如何呢？

笨辦法是一個(gè)個(gè)替換，如果代碼龐大的話，工作量不小，并且不能很好的保證修改不會(huì)有問題。

這里魚鷹介紹一種比較合適的方法，就是重定向函數(shù)（函數(shù)名不變），類似 weak（困惑多年，為什么 printf 可以重定向？）。

我們只需要隨便在一個(gè) C 文件中加入以下代碼即可完成：

?

#include "stdlib.h"


#if defined(__CC_ARM)
void *$Super$$malloc(size_t);
void *$Sub$$malloc(size_t size)
{
    // mutex_lock()
    void *addr = $Super$$malloc(size);
    // mutex_unlock()
    return addr;
}
void $Super$$free(void *);
void $Sub$$free(void *addr)
{
    // mutex_lock()
    $Super$$free(addr);
    // mutex_unlock()
}
#endif

?

當(dāng)然這個(gè)是在 MDK 環(huán)境下才有如此功能。這樣上層代碼就不用修改了，所有調(diào)用 malloc 和 free 的地方都會(huì)調(diào)用到上面的 sub 函數(shù)，這樣你可以加入互斥量或者跟蹤打印，方便調(diào)試。

C++ 版（C++自帶重寫功能，注意：很多 STL 模板是沒有保護(hù)的，如 push_back() 慎用）：

?

#include 
void *operator new(size_t size) throw(std::bad_alloc)
{
??? void *addr= malloc(size);
??? return addr;
}
void operator delete(void *addr) throw()
{
    free(addr);
}
void *operator new[](size_t size) throw(std::bad_alloc)
{
??? void *addr= malloc(size);
??? return addr;
}
void operator delete[](void *addr) throw()
{
    free(addr);
}?
void *operator new(std::size_t size, const std::nothrow_t& nothrow)
{
??? void *addr= malloc(size);
??? return addr;
}
void *operator new[](std::size_t size, const std::nothrow_t& nothrow)
{
??? void *addr= malloc(size);
??? return addr;
}

?

希望對(duì)各位道友有幫助。

　　審核編輯：湯梓紅