C語言內存問題如何解決
大家好，我是雜燴君。
C 語言內存問題，難在于定位，定位到了就好解決了。
這篇筆記我們來聊聊踩內存。踩內存，通過字面理解即可。本來是操作這一塊內存，因為設計失誤操作到了相鄰內存，篡改了相鄰內存的數(shù)據(jù)。
踩內存，輕則導致功能異常，重則導致程序崩潰死機。
內存，粗略地分：
- 靜態(tài)存儲區(qū)
- 動態(tài)存儲區(qū)
存儲于相同存儲區(qū)的變量才有互踩內存的可能。
靜態(tài)存儲區(qū)踩內存
分享一個之前在實際項目中遇到的問題。
在Linux中，一個進程默認可以打開的文件數(shù)為1024個，fd的范圍為0~1023。
項目中使用了串口，串口fd為static全局變量，某次這個fd突然變?yōu)橐粋€超范圍得值，顯然被踩了。
出問題的代碼如：
float arr[5];
int count = 8;
for (size_t i = 0; i < count; i++)
{
arr[i] = xxx;
}
操作同屬于靜態(tài)存儲區(qū)的arr數(shù)組出現(xiàn)了數(shù)組越界操作，踩了后面幾個連續(xù)變量，fd也踩了。
實際中，純靠log打印調試很難定位fd的相鄰變量，需要花比較多的時間。
在Linux中，這個問題我們可以通過生成生成map文件來查看，在CMakeLists.txt中生成map文件的代碼如：
set(CMAKE_EXE_LINKER_FLAGS "-Wl,-Map=output.map") # 生成map文件
set(CMAKE_C_FLAGS "-fdata-sections") # 把static變量地址輸出到map文件
set(CMAKE_CXX_FLAGS "-fdata-sections")
動態(tài)存儲區(qū)踩內存
動態(tài)堆內存踩內存典型例子：malloc與strcpy搭配使用不當導致緩沖區(qū)溢出。
#include < stdio.h >
#include < stdlib.h >
#include < unistd.h >
#include < string.h >
int main (void)
{
char *str = "hello";
int str_len = strlen(str);
///< 此時str_len = 5
printf("str_len = %d\\n", str_len);
///< 申請5字節(jié)的堆內存
char *ptr = (char*)malloc(str_len);
if (NULL == ptr)
{
printf("malloc error\\n");
exit(EXIT_FAILURE);
}
///< 定義一個指針p_a指向ptr向后偏移5字節(jié)的地址, 并在這個地址里寫入整數(shù)20
char *p_a = ptr + 5;
*p_a = 20;
printf("*p_a = %d\\n", *p_a);
///< 拷貝字符串str到ptr指向的地址
strcpy(ptr, str);
///< 打印結果：a指向的地方被踩了
printf("ptr = %s\\n", ptr);
printf("*p_a = %d\\n", *p_a);
///< 釋放對應內存
if (ptr)
{
free(ptr);
ptr = NULL;
}
return 0;
}
運行結果：
顯然，經(jīng)過strcpy操作之后，數(shù)據(jù)a的值被篡改了。
原因：忽略了strcpy操作會把字符串結束符一同拷貝到目的緩沖區(qū)。
如果相鄰的空間里沒有存放其它業(yè)務數(shù)據(jù)，那么踩了也不會出現(xiàn)問題，如果正好存放了重要數(shù)據(jù)，這時候可能會出現(xiàn)大bug，而且可能是偶現(xiàn)的，不好復現(xiàn)定位。
針對這種情況，我們可以借助一些工具來定位問題，比如：
- dmalloc
- valgrind
valgrind的簡單使用可閱讀往期筆記：工具 | Valgrind仿真調試工具的使用
當然，我們也可以在我們的代碼里進行一些嘗試。針對這類問題，分享一個檢測思路：
我們在申請內存時，在申請內存的前后增加兩塊標識區(qū)（紅區(qū)），里面寫入固定數(shù)據(jù)。申請、釋放內存的時候去檢測這兩塊標識區(qū)有沒有被破壞（檢測操作堆內存時是否踩到高壓紅區(qū)）。
為了能定位到后面的標識區(qū)，在增加一塊len區(qū)用來存儲實際申請的空間的長度。
此處，我們定義：
- 前紅區(qū)（before_ red_area）：4字節(jié)。寫入固定數(shù)據(jù)0x11223344。
- 后紅區(qū)（after_ red_area）：4字節(jié)。寫入固定數(shù)據(jù)0x55667788。
- 長度區(qū)（len_area）：4字節(jié)。存儲數(shù)據(jù)存儲區(qū)的長度。
自定義申請內存函數(shù)
除了數(shù)據(jù)存儲區(qū)之外，多申請12個字節(jié)。自定義申請內存的函數(shù)自然是要兼容malloc的使用方法。malloc原型：
void *malloc(size_t __size);
自定義申請內存的函數(shù)：
void *Malloc(size_t __size);
返回值自然要返回數(shù)據(jù)存儲區(qū)的地址。具體實現(xiàn)：
#define BEFORE_RED_AREA_LEN (4) ///< 前紅區(qū)長度
#define AFTER_RED_AREA_LEN (4) ///< 后紅區(qū)長度
#define LEN_AREA_LEN (4) ///< 長度區(qū)長度
#define BEFORE_RED_AREA_DATA (0x11223344u) ///< 前紅區(qū)數(shù)據(jù)
#define AFTER_RED_AREA_DATA (0x55667788u) ///< 后紅區(qū)數(shù)據(jù)
void *Malloc(size_t __size)
{
///< 申請內存：4 + 4 + __size + 4
void *ptr = malloc(BEFORE_RED_AREA_LEN + AFTER_RED_AREA_LEN + __size + LEN_AREA_LEN);
if (NULL == ptr)
{
printf("[%s]malloc error\\n", __FUNCTION__);
return NULL;
}
///< 往前紅區(qū)地址寫入固定值
*((unsigned int*)(ptr)) = BEFORE_RED_AREA_DATA;
///< 往長度區(qū)地址寫入長度
*((unsigned int*)(ptr + BEFORE_RED_AREA_LEN)) = __size;
///< 往后紅區(qū)地址寫入固定值
*((unsigned int*)(ptr + BEFORE_RED_AREA_LEN + LEN_AREA_LEN + __size)) = AFTER_RED_AREA_DATA;
///< 返回數(shù)據(jù)區(qū)地址
void *data_area_ptr = (ptr + BEFORE_RED_AREA_LEN + LEN_AREA_LEN);
return data_area_ptr;
}
自定義檢測內存函數(shù)
申請完內存并往內存里寫入數(shù)據(jù)后，檢測本該寫入到數(shù)據(jù)存儲區(qū)的數(shù)據(jù)有沒有寫到紅區(qū)。這種內存檢測方法我們是用在開發(fā)調試階段的，所以檢測內存，我們可以使用斷言，一旦觸發(fā)斷言，直接終止程序報錯。
檢測前后紅區(qū)里的數(shù)據(jù)有沒有被踩：
void CheckMem(void *ptr, size_t __size)
{
void *data_area_ptr = ptr;
///< 檢測是否踩了前紅區(qū)
printf("[%s]before_red_area_data = 0x%x\\n", __FUNCTION__, *((unsigned int*)(data_area_ptr - LEN_AREA_LEN - BEFORE_RED_AREA_LEN)));
assert(*((unsigned int*)(data_area_ptr - LEN_AREA_LEN - BEFORE_RED_AREA_LEN)) == BEFORE_RED_AREA_DATA);
///< 檢測是否踩了長度區(qū)
printf("[%s]len_area_data = 0x%x\\n", __FUNCTION__, *((unsigned int*)(data_area_ptr - LEN_AREA_LEN)));
assert(*((unsigned int*)(data_area_ptr - LEN_AREA_LEN)) == __size);
///< 檢測是否踩了后紅區(qū)
printf("[%s]after_red_area_data = 0x%x\\n", __FUNCTION__, *((unsigned int*)(data_area_ptr + __size)));
assert(*((unsigned int*)(data_area_ptr + __size)) == AFTER_RED_AREA_DATA);
}
自定義釋放內存函數(shù)
要釋放所有前面申請內存。釋放前同樣要進行檢測：
void Free(void *ptr)
{
void *all_area_ptr = ptr - LEN_AREA_LEN - BEFORE_RED_AREA_LEN;
///< 檢測是否踩了前紅區(qū)
printf("[%s]before_red_area_data = 0x%x\\n", __FUNCTION__, *((unsigned int*)(all_area_ptr)));
assert(*((unsigned int*)(all_area_ptr)) == BEFORE_RED_AREA_DATA);
///< 讀取長度區(qū)內容
size_t __size = *((unsigned int*)(all_area_ptr + BEFORE_RED_AREA_LEN));
///< 檢測是否踩了后紅區(qū)
printf("[%s]before_red_area_data = 0x%x\\n", __FUNCTION__, *((unsigned int*)(all_area_ptr + BEFORE_RED_AREA_LEN + LEN_AREA_LEN + __size)));
assert(*((unsigned int*)(all_area_ptr + BEFORE_RED_AREA_LEN + LEN_AREA_LEN + __size)) == AFTER_RED_AREA_DATA);
///< 釋放所有區(qū)域內存
free(all_area_ptr);
}
我們使用這種方法檢測上面的 malloc與strcpy搭配使用不當導致緩沖區(qū)溢出 的例子：
可以看到，這個例子踩了后紅區(qū)，把后紅區(qū)數(shù)據(jù)修改為了 0x55667700 ，觸發(fā)斷言程序終止。
測試代碼：
// 公眾號：嵌入式大雜燴
#include < stdio.h >
#include < stdlib.h >
#include < unistd.h >
#include < string.h >
#include < assert.h >
#define BEFORE_RED_AREA_LEN (4) ///< 前紅區(qū)長度
#define AFTER_RED_AREA_LEN (4) ///< 后紅區(qū)長度
#define LEN_AREA_LEN (4) ///< 長度區(qū)長度
#define BEFORE_RED_AREA_DATA (0x11223344u) ///< 前紅區(qū)數(shù)據(jù)
#define AFTER_RED_AREA_DATA (0x55667788u) ///< 后紅區(qū)數(shù)據(jù)
void *Malloc(size_t __size)
{
///< 申請內存：4 + 4 + __size + 4
void *ptr = malloc(BEFORE_RED_AREA_LEN + AFTER_RED_AREA_LEN + __size + LEN_AREA_LEN);
if (NULL == ptr)
{
printf("[%s]malloc error\\n", __FUNCTION__);
return NULL;
}
///< 往前紅區(qū)地址寫入固定值
*((unsigned int*)(ptr)) = BEFORE_RED_AREA_DATA;
///< 往長度區(qū)地址寫入長度
*((unsigned int*)(ptr + BEFORE_RED_AREA_LEN)) = __size;
///< 往后紅區(qū)地址寫入固定值
*((unsigned int*)(ptr + BEFORE_RED_AREA_LEN + LEN_AREA_LEN + __size)) = AFTER_RED_AREA_DATA;
///< 返回數(shù)據(jù)區(qū)地址
void *data_area_ptr = (ptr + BEFORE_RED_AREA_LEN + LEN_AREA_LEN);
return data_area_ptr;
}
void CheckMem(void *ptr, size_t __size)
{
void *data_area_ptr = ptr;
///< 檢測是否踩了前紅區(qū)
printf("[%s]before_red_area_data = 0x%x\\n", __FUNCTION__, *((unsigned int*)(data_area_ptr - LEN_AREA_LEN - BEFORE_RED_AREA_LEN)));
assert(*((unsigned int*)(data_area_ptr - LEN_AREA_LEN - BEFORE_RED_AREA_LEN)) == BEFORE_RED_AREA_DATA);
///< 檢測是否踩了長度區(qū)
printf("[%s]len_area_data = 0x%x\\n", __FUNCTION__, *((unsigned int*)(data_area_ptr - LEN_AREA_LEN)));
assert(*((unsigned int*)(data_area_ptr - LEN_AREA_LEN)) == __size);
///< 檢測是否踩了后紅區(qū)
printf("[%s]after_red_area_data = 0x%x\\n", __FUNCTION__, *((unsigned int*)(data_area_ptr + __size)));
assert(*((unsigned int*)(data_area_ptr + __size)) == AFTER_RED_AREA_DATA);
}
void Free(void *ptr)
{
void *all_area_ptr = ptr - LEN_AREA_LEN - BEFORE_RED_AREA_LEN;
///< 檢測是否踩了前紅區(qū)
printf("[%s]before_red_area_data = 0x%x\\n", __FUNCTION__, *((unsigned int*)(all_area_ptr)));
assert(*((unsigned int*)(all_area_ptr)) == BEFORE_RED_AREA_DATA);
///< 讀取長度區(qū)內容
size_t __size = *((unsigned int*)(all_area_ptr + BEFORE_RED_AREA_LEN));
///< 檢測是否踩了后紅區(qū)
printf("[%s]before_red_area_data = 0x%x\\n", __FUNCTION__, *((unsigned int*)(all_area_ptr + BEFORE_RED_AREA_LEN + LEN_AREA_LEN + __size)));
assert(*((unsigned int*)(all_area_ptr + BEFORE_RED_AREA_LEN + LEN_AREA_LEN + __size)) == AFTER_RED_AREA_DATA);
///< 釋放所有區(qū)域內存
free(all_area_ptr);
}
int main (void)
{
char *str = "hello";