函數(shù)中隨處return，是造成我們資源泄露和程序死鎖的主要根源。很多同志寫過類似的代碼，函數(shù)中創(chuàng)建了和引用了多個資源，中間使用的過程中出錯了，程序return，經(jīng)典的代碼是這樣的：

void?fun() ? {Lock(mutex);mem?=?malloc(size);if?(null?==?mem){ ?????????return;?//死鎖 ????} ????fh?=?fopen(“test.txt”); ????if?(fh) ?????{ ?????????return;?//死鎖+內(nèi)存泄露 ????} ????if?(fwrite(“abc”,?fh)?<?0) ?????{ ?????????return;?//死鎖+內(nèi)存泄露+句柄泄露 ????} ????fclose(fh); ?????free(mem); ?????Unlock(mtex); }

上述代碼中的資源泄露是顯而易見的，編程新手卻很容易寫出這樣的代碼，但問題卻不容易發(fā)現(xiàn)，因?yàn)楫惓Ａ鞒掏ǔ２粫l(fā)生，而一旦發(fā)生了，就是一場噩夢：不好跟蹤也不好重現(xiàn)。對吃過這種虧編程老手而言，處理起來會格外小心，而是有了如下版本的代碼（我們的工程中類似的代碼隨處可見）：

void?fun() ? {Lock(mutex);mem?=?malloc(size); ????if?(null?==?mem) ?????{ ?????????Unlock(mutex);?//鎖解除了，很好 ????????return; ?????} ??fh?=?fopen(“test.txt”); ????if?(fh) ?????{ ?????????Free(mem); ?????????Unlock(mutext);?//錯誤處理加了一行 ????????return; ?????} ???if?(fread(mem,?size,?fh)?<?0) ?????{ ?????????Free(mem); ?????????Flcose(fh); ?????????Unlock(mutext);?//錯誤處理又加了一行… ????????return; ?????} ??fclose(fh); ?????free(mem); ?????Unlock(mtex); }

?

這個版本的代碼沒什么邏輯問題，在每個異常環(huán)節(jié)，把該處理的都處理完了。看起來非常完美，卻依然存有大坑。首先，我們的異常處理流程越來越長，重復(fù)的代碼像火車車廂一樣越串越長，代碼維護(hù)非常不方便，某天其中一個異常處理流程要修改了，我們需要修同步多處，萬一漏了某處（事實(shí)上不是萬一，而是經(jīng)常），問題沒有修改徹底；更為嚴(yán)重的是，冗于的異常處理會讓大家感到疲勞，某一天函數(shù)中又要加入新的資源引用或異常處理邏輯時，在先驅(qū)的引導(dǎo)下，我們的人依然還是會犯同樣的錯誤：要么什么都不管提前返回，或者處理不充分就走了，結(jié)果還是同樣的資源泄露。

如何解決，Linux內(nèi)核代碼給了我們很好的解決方案，合理地使用goto語句，將異常處理定向到統(tǒng)一的地方，既解決了資源泄露隱患，也保持了正常流程代碼的簡潔性，這是改進(jìn)后的版本：

void?fun() ?
{ ?????
? ? Lock(mutex);???
? ? mem?=?malloc(size); ????
? ? if?(null?==?mem) ?????
? ? { ?????????
? ? ? ? goto?ERR_EXIT1; ?????
? ? } ????
? ? fh?=?fopen(“test.txt”); ????
? ? if?(fh) ?????
? ? { ?????????
? ? ? ? goto?ERR_EXIT2; ?????
? ? } ????
? ? if?(fread(mem,?size,?fh)?<?0) ?????
? ? { ????????
? ? ? ? goto?ERR_EXIT3; ?????
? ? } ??
? ? ERR_EXIT3: ?????
? ? fclose(fh); ?
? ? ERR_EXIT2: ?????
? ? free(mem); ?
? ? ERR_EXIT1: ?????
? ? Unlock(mtex); ?
}

總結(jié)：以后新寫C代碼的異常處理，統(tǒng)一采用上面的模版。C++/java有更好的異常處理機(jī)制，上述方案只供參考，但解決思路應(yīng)該是一致的，隨處return與冗余異常處理應(yīng)該被嚴(yán)厲禁止。

?

總結(jié)

以上是生活随笔為你收集整理的return导致内存泄露。C函数中不要有多次return，以及其解决方法。goto的用法。的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。