vsnprintf 的bug

jeppeter

1. #include <stdio.h>
   
2. #include <stdarg.h>
   
3. #include <stdlib.h>
   
4. #include <string.h>
   
5. 
   
6. int simple_assert(const char* file, int lineno, const char* resultstr, const char* fmt, ...)
   
7. {
   
8.         va_list ap;
   
9.         char* fmtstr = NULL;
   
10.         int fmtsize = 20;
    
11.         int ret = 0;
    
12.         va_start(ap, fmt);
    
13. try_again:
    
14.         if (fmtstr) {
    
15.                 free(fmtstr);
    
16.         }
    
17.         fmtstr = NULL;
    
18.         fmtstr = (char*)malloc(fmtsize);
    
19.         if (fmtstr == NULL) {
    
20.                 fprintf(stderr, "%s:%d can not alloc(%d)\n", file, lineno, fmtsize);
    
21.                 abort();
    
22.         }
    
23. 
    
24.         fprintf(stderr, "fmt (%s)(", fmt);
    
25.         vfprintf(stderr, fmt, ap);
    
26.         fprintf(stderr, ")\n");
    
27.         fflush(stderr);
    
28.         ret = vsnprintf(fmtstr, fmtsize - 10, fmt, ap);
    
29.         if (ret < 0 || ret >= (fmtsize - 10)) {
    
30.                 fprintf(stderr, "expand size %d => %d\n", fmtsize, fmtsize << 1);
    
31.                 fmtsize <<= 1;
    
32.                 goto try_again;
    
33.         }
    
34.         va_end(ap);
    
35.         ret = strcmp(resultstr, fmtstr);
    
36.         if (ret != 0) {
    
37.                 fprintf(stderr, "%s:%d (%s) != (%s)\n", file, lineno, resultstr, fmtstr);
    
38.                 abort();
    
39.         }
    
40. 
    
41.         if (fmtstr) {
    
42.                 free(fmtstr);
    
43.         }
    
44.         fmtstr = NULL;
    
45.         fmtsize = 0;
    
46.         return 0;
    
47. }
    
48. 
    
49. #define SIMPLE_ASSERT(res,...) simple_assert(__FILE__,__LINE__,res,__VA_ARGS__)
    
50. 
    
51. 
    
52. int main(void)
    
53. {
    
54.         SIMPLE_ASSERT("to display this help information", "%s %s", "to display", "this help information");
    
55.         return 0;
    
56. 
    
57. }

复制代码

这个是测试vsnprintf 的代码，会引起 segment fault
但这里已经对内存加以限制了，但还是会出现错误

ljmmail

int main(void)
{
        SIMPLE_ASSERT("to display this help information %s %s", "to display", "this help information");
        return 0;

}

程序调用格式有问题，换上面格式看看。

ljmmail

1. #include <stdio.h>
   
2. #include <stdarg.h>
   
3. #include <stdlib.h>
   
4. #include <string.h>
   
5. 
   
6. int simple_assert(const char* file, int lineno, const char* resultstr, const char* fmt, ...)
   
7. {
   
8.     va_list ap;
   
9.     char* fmtstr = NULL;
   
10.     int fmtsize ;
    
11.     int ret = 0;
    
12. 
    
13.         // 通过一次输出，确定 fmtsize 长度
    
14.     va_start(ap, fmt);
    
15.     fprintf(stdout, "fmt (%s) = (", fmt);
    
16.     fmtsize = vfprintf(stdout, fmt, ap);
    
17.     fprintf(stdout, ") [%d]\n", fmtsize);
    
18.     va_end(ap);
    
19.     // 一次的 va_start() 函数调用，也仅仅是提供一次 vsprintf() 函数的支持
    
20. 
    
21.         fmtsize += 3;
    
22. try_again:
    
23.         if( fmtstr ) free(fmtstr);
    
24. 
    
25.     fmtstr = (char*)calloc(fmtsize, 1 );
    
26.     if (fmtstr == NULL) {
    
27.         fprintf(stderr, "%s:%d can not alloc(%d)\n", file, lineno, fmtsize);
    
28.         abort();
    
29.     }
    
30. 
    
31.     va_start(ap, fmt);
    
32.     ret = vsnprintf(fmtstr, fmtsize-1, fmt, ap);
    
33.     va_end(ap);
    
34.     if (ret < 0 || ret >= (fmtsize - 1)) {
    
35.         fmtsize <<= 1;
    
36.         fprintf(stderr, "expand size %d, %d => %d\n", ret, fmtsize, fmtsize);
    
37.         goto try_again;
    
38.     }
    
39.     fprintf(stdout, "fmtstr = [%d][%d][%s]\n", ret, fmtsize, fmtstr);
    
40.     ret = strncmp(resultstr, fmtstr, strlen(resultstr) );
    
41.     if (ret != 0) {
    
42.         fprintf(stderr, "%s:%d (%s) != (%s)\n", file, lineno, resultstr, fmtstr);
    
43.     }
    
44.     else
    
45.     {
    
46.             fprintf(stderr, "[%s] 与 fmtstr 匹配\n", resultstr);
    
47.     }
    
48. 
    
49.     free(fmtstr);
    
50.     return 0;
    
51. }
    
52. 
    
53. #define SIMPLE_ASSERT(res,...) simple_assert(__FILE__,__LINE__,res,__VA_ARGS__)
    
54. 
    
55. 
    
56. int main(void)
    
57. {
    
58.     SIMPLE_ASSERT("to display this help information", "%s %s", "to display", "this help information");
    
59.     return 0;
    
60. 
    
61. }

复制代码

使用上述代码，测试看看。

jeppeter

1. #include <stdio.h>
   
2. #include <stdarg.h>
   
3. #include <stdlib.h>
   
4. #include <string.h>
   
5. 
   
6. void debug_buffer(const char* file, int lineno, void* pbuffer, int bufsize, const char* fmt, ...)
   
7. {
   
8.         va_list ap;
   
9.         int i;
   
10.         unsigned char* ptr;
    
11.         fprintf(stderr, "%s:%d buffer(%p) size(%d)", file, lineno, pbuffer, bufsize);
    
12.         if (fmt) {
    
13.                 fprintf(stderr, " " );
    
14.                 va_start(ap, fmt);
    
15.                 vfprintf(stderr, fmt, ap);
    
16.                 va_end(ap);
    
17.         }
    
18.         ptr = (unsigned char*)pbuffer;
    
19.         for (i = 0; i < bufsize; i++) {
    
20.                 if ((i % 16) == 0) {
    
21.                         fprintf(stderr, "\n0x%08x",i );
    
22.                 }
    
23.                 fprintf(stderr, " 0x%02x",ptr[i] );
    
24.         }
    
25.         fprintf(stderr, "\n");
    
26.         fflush(stderr);
    
27.         return;
    
28. }
    
29. 
    
30. #define DEBUG_BUFFER(ptr,size,...)  do{debug_buffer(__FILE__,__LINE__,ptr,size,__VA_ARGS__);}while(0)
    
31. 
    
32. int simple_assertv(const char* file, int lineno, const char* resultstr, const char* fmt, va_list ap)
    
33. {
    
34.         va_list oldap;
    
35.         char* fmtstr = NULL;
    
36.         int fmtsize = 32;
    
37.         int ret = 0;
    
38.         memset(&oldap,0,sizeof(oldap));
    
39.         memcpy(&oldap, ap, sizeof(oldap));
    
40.         //DEBUG_BUFFER(&oldap,sizeof(oldap),"oldap pointer");
    
41.         //DEBUG_BUFFER(ap,sizeof(oldap),"ap pointer");
    
42. 
    
43.         fprintf(stderr, "fmt (%s)(", fmt);
    
44.         vfprintf(stderr, fmt, ap);
    
45.         fprintf(stderr, ")\n");
    
46.         fflush(stderr);
    
47. 
    
48. try_again:
    
49.         memcpy(ap, &oldap, sizeof(oldap));
    
50.         //DEBUG_BUFFER(ap,sizeof(oldap),"newap pointer");
    
51.         if (fmtstr) {
    
52.                 free(fmtstr);
    
53.         }
    
54.         fmtstr = NULL;
    
55.         fmtstr = (char*)malloc(fmtsize);
    
56.         if (fmtstr == NULL) {
    
57.                 fprintf(stderr, "%s:%d can not alloc(%d)\n", file, lineno, fmtsize);
    
58.                 abort();
    
59.         }
    
60. 
    
61.         ret = vsnprintf(fmtstr, fmtsize - 10, fmt, ap);
    
62.         if (ret < 0 || ret >= (fmtsize - 10)) {
    
63.                 fprintf(stderr, "expand size %d => %d\n", fmtsize, fmtsize << 1);
    
64.                 fmtsize <<= 1;
    
65.                 goto try_again;
    
66.         }
    
67.         va_end(ap);
    
68.         ret = strcmp(resultstr, fmtstr);
    
69.         if (ret != 0) {
    
70.                 fprintf(stderr, "%s:%d (%s) != (%s)\n", file, lineno, resultstr, fmtstr);
    
71.                 abort();
    
72.         }
    
73. 
    
74.         if (fmtstr) {
    
75.                 free(fmtstr);
    
76.         }
    
77.         fmtstr = NULL;
    
78.         fmtsize = 0;
    
79.         return 0;
    
80. }
    
81. 
    
82. int simple_assert(const char* file, int lineno, const char* resultstr, const char* fmt, ...)
    
83. {
    
84.         va_list ap;
    
85.         if (fmt) {
    
86.                 va_start(ap, fmt);
    
87.         }
    
88.         //DEBUG_BUFFER(ap,sizeof(ap),"initliaze ap");
    
89.         return simple_assertv(file, lineno, resultstr, fmt, ap);
    
90. }
    
91. 
    
92. #define SIMPLE_ASSERT(res,...) simple_assert(__FILE__,__LINE__,res,__VA_ARGS__)
    
93. 
    
94. 
    
95. int main(void)
    
96. {
    
97.         SIMPLE_ASSERT("to display this help information", "%s %s", "to display", "this help information");
    
98.         return 0;
    
99. 
    
100. }

复制代码

回复 3# ljmmail

谢谢你的回复，你的方法是可以，但因为如果真的就是输入一个ap，那就需要保留这个值，我自己调整了一下，使用memcpy保留值，可以不改变参数输入的情况下，使代码不出现问题。

lxyscls

回复 4# jeppeter

莫干傻事，好好用人家的接口