Bootstrap

解决字符串格式化时内存不足的问题

我们知道使用sprintf(), _snprintf()可变长函数对字符串格式化时,经常会出现不知道分配多少内存的问题,如果分配的内存过小,则字符串格式化后不完整,如果分配过长又浪费内存,所以我们对格式化函数作修改,让其动态分配内存:

#include <vector>
#include <iostream>
#include <cstdarg>
#include <stdio.h>
using namespace std;

int mysprintf(const char* fmt, ...)
{
    int l_iRet = 0;
    std::string var_str;
    //int len = _vscprintf(fmt, ap);
    int l_nDatalen = 1024;
    int l_retLen = 0;

    //由于linux下vsnprintf不能多次使用va_list,因而把va_list放在循环中
    //动态申请内存
    do
    {
        va_list    ap;
        va_start(ap, fmt);
        std::vector<char> buf(l_nDatalen + 10);
        l_retLen = _vsnprintf(&buf.front(), l_nDatalen, fmt, ap);
        l_nDatalen*=4;
        va_end(ap);
        if(l_retLen>=0)
        {
            var_str.assign(buf.begin(), buf.begin() + l_retLen);
            break;
        }
    } while (l_retLen<0);

    cout<<var_str;
    return l_iRet;
}

int main()
{
    mysprintf("test %d, %s", 0, "dkddkdkdlldl");
    return 0;
}

;