1.gets()函數

問：請找出下面代碼里的問題：

#include<stdio.h>?
int?main(void)?
{?
????char?buff[10];?
????memset(buff,0,sizeof(buff));?
?
????gets(buff);?
?
????printf("\n?The?buffer?entered?is?[%s]\n",buff);?
?
????return?0;?
}?

答：上面代碼里的問題在于函數gets()的使用，這個函數從stdin接收一個字符串而不檢查它所復制的緩存的容積，這可能會導致緩存溢出。這里推薦使用標準函數fgets()代替。

2.strcpy()函數

問：下面是一個簡單的密碼保護功能，你能在不知道密碼的情況下將其破解嗎？

#include<stdio.h>?
?
int?main(int?argc,?char?*argv[])?
{?
????int?flag?=?0;?
????char?passwd[10];?
?
????memset(passwd,0,sizeof(passwd));?
?
????strcpy(passwd,?argv[1]);?
?
????if(0?==?strcmp("LinuxGeek",?passwd))?
????{?
????????flag?=?1;?
????}?
?
????if(flag)?
????{?
????????printf("\n?Password?cracked?\n");?
????}?
????else?
????{?
????????printf("\n?Incorrect?passwd?\n");?
?
????}?
????return?0;?
}?

答：破解上述加密的關鍵在于利用攻破strcpy()函數的漏洞。所以用戶在向“passwd”緩存輸入隨機密碼的時候并沒有提前檢查“passwd”的容量是否足夠。所以，如果用戶輸入一個足夠造成緩存溢出并且重寫“flag”變量默認值所存在位置的內存的長“密碼”，即使這個密碼無法通過驗證，flag驗證位也變成了非零，也就可以獲得被保護的數據了。例如：

$?./psswd?aaaaaaaaaaaaa?
?
Password?cracked?

雖然上面的密碼并不正確，但我們仍然可以通過緩存溢出繞開密碼安全保護。

要避免這樣的問題，建議使用 strncpy()函數。

作者注：最近的編譯器會在內部檢測棧溢出的可能，所以這樣往棧里存儲變量很難出現棧溢出。在我的gcc里默認就是這樣，所以我不得不使用編譯命令‘-fno-stack-protector’來實現上述方案。

3.main()的返回類型

問：下面的代碼能 編譯通過嗎？如果能，它有什么潛在的問題嗎？

#include<stdio.h>?
?
void?main(void)?
{?
????char?*ptr?=?(char*)malloc(10);?
?
????if(NULL?==?ptr)?
????{?
????????printf("\n?Malloc?failed?\n");?
????????return;?
????}?
????else?
????{?
????????//?Do?some?processing?
????????free(ptr);?
????}?
?
????return;?
}?

答：因為main()方法的返回類型，這段代碼的錯誤在大多數編譯器里會被當作警告。main()的返回類型應該是“int”而不是“void”。因為“int”返回類型會讓程序返回狀態值。這點非常重要，特別當程序是作為依賴于程序成功運行的腳本的一部分運行時。

4.內存泄露

問：下面的代碼會導致內存泄漏嗎？

#include<stdio.h>?
?
void?main(void)?
{?
????char?*ptr?=?(char*)malloc(10);?
?
????if(NULL?==?ptr)?
????{?
????????printf("\n?Malloc?failed?\n");?
????????return;?
????}?
????else?
????{?
????????//?Do?some?processing?
????}?
?
????return;?
}?

答：盡管上面的代碼并沒有釋放分配給“ptr”的內存，但并不會在程序退出后導致內存泄漏。在程序結束后，所有這個程序分配的內存都會自動被處理掉。但如果上面的代碼處于一個“while循環”中，那將會導致嚴重的內存泄漏問題！

提示：如果你想知道更多關于內存泄漏的知識和內存泄漏檢測工具，可以來看看我們在Valgrind上的文章。

5.free()函數

問：下面的程序會在用戶輸入'freeze'的時候出問題，而'zebra'則不會，為什么？

#include<stdio.h>?
?
int?main(int?argc,?char?*argv[])?
{?
????char?*ptr?=?(char*)malloc(10);?
?
????if(NULL?==?ptr)?
????{?
????????printf("\n?Malloc?failed?\n");?
????????return?-1;?
????}?
????else?if(argc?==?1)?
????{?
????????printf("\n?Usage??\n");?
????}?
????else?
????{?
????????memset(ptr,?0,?10);?
?
????????strncpy(ptr,?argv[1],?9);?
?
????????while(*ptr?!=?'z')?
????????{?
????????????if(*ptr?==?'')?
????????????????break;?
????????????else?
????????????????ptr++;?
????????}?
?
????????if(*ptr?==?'z')?
????????{?
????????????printf("\n?String?contains?'z'\n");?
????????????//?Do?some?more?processing?
????????}?
?
???????free(ptr);?
????}?
?
????return?0;?
}?

答：這里的問題在于，代碼會（通過增加“ptr”）修改while循環里“ptr”存儲的地址。當輸入“zebra”時，while循環會在執行前被終止，因此傳給free()的變量就是傳給malloc()的地址。但在“freeze”時，“ptr”存儲的地址會在while循環里被修改，因此導致傳給free()的地址出錯，也就導致了seg-fault或者崩潰。

6.使用_exit退出

問：在下面的代碼中，atexit()并沒有被調用，為什么？

#include<stdio.h>?
?
void?func(void)?
{?
????printf("\n?Cleanup?function?called?\n");?
????return;?
}?
?
int?main(void)?
{?
????int?i?=?0;?
?
????atexit(func);?
?
????for(;i<0xffffff;i++);?
?
????_exit(0);?
}?

這是因為_exit()函數的使用，該函數并沒有調用atexit()等函數清理。如果使用atexit()就應當使用exit()或者“return”與之相配合。