snprintf 導致的bug

 

以下代碼是有bug的, 不知到你看出來了否?

#include <stdio.h>
#include <unistd.h>
#include <set>
#include <vector>
#include <fstream>

using namespace std;

int main(int argc, char *argv[])
{
    const char* str_arr[] = {
        "1234567890",
        "abcdefghij",
        "oooooooooo"
    }; 
   
    char buf[10] = {0};

    int pos = 0;
    for (size_t i=0; i<sizeof(str_arr)/sizeof(char*); ++i)
    {  
        pos += snprintf(buf+pos, sizeof(buf)-pos, str_arr[i]); // bug
    }  
    printf("buf:%s\n", buf);
       
    return 0;
}

以上代碼運行會core.

存在bug的原因在於snprintf的陷阱
1) 函數的第二個參數是 size_t, 所以, 如果傳入負數, 後果很嚴重! 可能會導致寫越界
2) 函數的傳回值是欲寫入的位元組數, 而不是實際寫入的位元組數!!!

所以, 正確的寫法應該是:

int main(int argc, char *argv[])
{
const char* str_arr[] = {
"1234567890",
"abcdefghij",
"oooooooooo"
};

char buf[20] = {0};

int ret = 0;
int pos = 0;
for (size_t i=0; i<sizeof(str_arr)/sizeof(char*); ++i)
{
ret = snprintf(buf+pos, sizeof(buf)-pos, str_arr[i]);

// 如果出錯,或者傳回值>=buf容量(意味著本次snprintf被截斷)
if (ret==-1 || ret>=(int)(sizeof(buf)-pos))
{
break;
}

pos += ret;
}
printf("buf:%s\n", buf);

return 0;
}

以上代碼運行結果:
buf:1234567890abcdefghi