如何调整以下代码以防止缓冲区溢出？

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终止的字符串。