如何調整以下代碼以防止緩衝區溢出？

Question

void updateConfigParams(void) { 
    char buffer [512]; 
    int i = 0; 

while ((c = readFromWireless()) != NULL) 
{ 
    buffer [ i ] = c; 
    i += 1; 
} 
writeConfigParams (buffer); 
} 

我只是在處理緩衝區溢出問題，所以請有人告訴我如何更改下面的代碼以防止緩衝區溢出，並解釋新代碼如何停止寫入緩衝區末尾。

Answer 1

void updateConfigParams(void) { 
    char buffer [512]; 
    int i = 0; 

    while (i < 512 && (c = readFromWireless()) != NULL) { 
     buffer [ i++ ] = c; 
    } 
    writeConfigParams (buffer); 
} 

Answer 2

您可以添加以下if語句來檢查緩衝區溢出條件。增加變量後我

i+=1; 
if(i > 512) 
    break; 

一旦我達到緩衝區限制，你將退出循環。

Answer 3

如果（i == sizeof（buffer）） break，則需要在while循環的開始處添加緩衝區大小檢查 ;

，並在緩衝區的末尾添加\ 0，如果writeConfigParams需要一個字符串

Answer 4

Buffer overflow是當你嘗試寫超出分配給緩衝區的內存地址。

在你的情況下，你已經分配了512字節，所以你的代碼應該確保你永遠不會脫離buffer + 511之外的參考。換句話說

buffer [i] // i should never exceed 511

你的代碼應該檢查是否停止服用一次輸入你的索引計數器達到等於緩衝區的大小。

Answer 5

我會做一些大意如下：

bool 
updateConfigParams(void) { 
    char buffer[512] = {}; 
    int i = 0; 

    while((c = readFromWireless()) != NULL) { 
    if(i == sizeof(buffer) - 1) { 
     warn("readFromWireless exceeded %zu byte limit", sizeof(buffer)); 
     return false; 
    } 
    buffer[i++] = c; 
    } 
    writeConfigParams(buffer); 
    return true; 
} 

根據你的程序的狀態，它可能更適合簡單地調用ERR（3）。重要的一點是：

• 當寫入數組時，總是確保你在進入界限。
• 將輸入收集到數組中時，始終要爲超出陣列大小的輸入做好準備。如何處理您無法接受的輸入取決於應用程序。
• 當程序員在編譯時無法阻止內部存儲 - 例如無線設備發送「太多」數據 - 通知用戶和該程序。上面，程序發出一條消息，該函數返回一個錯誤狀態。
• 部分輸入通常是可疑的，不應被接受。

編輯：根據評論，我添加了初始化到buffer。由於writeConfigParams不接受長度參數，因此可能接受NUL終止的字符串。