使用va_arg

时间:2016-02-25 07:41:22

标签: c string segmentation-fault variadic-functions

我正在创建一个类似于printf的函数,它需要接受一个字符串和参数,如:

form("Integer %d, String %s", 54, "STRING");

并创建一个字符串"Integer 54, String STRING"。 我正在使用stdarg.h库,因为我的函数需要根据字符串具有可变数量的参数。

问题是我得到Segmentation fault。我发现只有当我strlenstrcpy使用字符串(char *)传递给va_arg时才会发生这种情况。这是我的代码:

#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <string.h>

char* form(char *format, ...)
{
    va_list ap;char sign;int br=0,lasti,memo=0;char* help;
    int ints; float floats; double doubles; char chars; char* strings;
    va_start(ap,format);
    char* result=(char*)calloc(100,strlen(format));
    strcpy(result,format);
    for(int i=0; result[i] ;i++)
        if(result[i]=='%')
        {   
            switch (result[i+1])    
            {
                case 'd': {
                    ints=va_arg(ap,int);
                    int b,save=ints,dec=1,j;char *p=result+i;
                    for(b=0;save;b++) {save/=10;dec*=10;}
                    for(dec/=10,j=0; dec ; j++) { p[j]=((ints/dec)%10)+0x30; dec/=10; }
                    strcpy(result+i+b,format+i-memo+2);memo+=b-2;
                } break;

                case 'f': {
                    floats=va_arg(ap,double);

                } break;

                case 'l': {
                    doubles=va_arg(ap,double);

                } break;

                case 'c': {
                    chars=va_arg(ap,int);
                    result[i]=chars;
                } break;

                case 's': {
                    strings=va_arg(ap,char*);
                    strcpy(result+i+strlen(strings),format+i-memo+2);memo+=(strlen(strings)-2);
                } break;

                default: printf("Unknown type.\n"); break;
            }
            i=0;
        }

    return result;
}

int main()
{
    char a[100];
    scanf("%s",a);
    char*s=form("treci %s peti",a);
    printf("%s", s);

    printf("\n");
    free(s);
    return 0;
}

strlen或strcpy导致段错误的唯一方法是,如果字符串不是以null结尾,而是我的。那么这里有什么问题,我该如何解决?

编辑:添加了代码。

2 个答案:

答案 0 :(得分:4)

仅针对%s案例进行了测试

char* form(char *format, ...)
{
    va_list ap;char sign;int br=0,lasti,memo=0;char* help;
    int ints; float floats; double doubles; char chars; char* strings;
    va_start(ap,format);
    char* result=(char*)calloc(100,strlen(format));
//    strcpy(result,format);
    for(int i=0; format[i] ;i++)
    {
        if(format[i]=='%')
        {
            switch (format[i+1])
            {
                case 'd': {
                    ints=va_arg(ap,int);
                    int b,save=ints,dec=1,j;char *p=result+i;
                    for(b=0;save;b++) {save/=10;dec*=10;}
                    for(dec/=10,j=0; dec ; j++) { p[j]=((ints/dec)%10)+0x30; dec/=10; }
                    strcpy(result+i+b,format+i-memo+2);memo+=b-2;
                } break;

                case 'f': {
                    floats=va_arg(ap,double);

                } break;

                case 'l': {
                    doubles=va_arg(ap,double);

                } break;

                case 'c': {
                    chars=va_arg(ap,int);
                    result[i]=chars;
                } break;

                case 's': {
                    strings=va_arg(ap,char*);
                    strcpy(result+memo,strings);
                    memo+=strlen(strings);
                } break;

                default: printf("Unknown type.\n"); break;
            }
            i++;
        }
        else
        {
            result[memo++] = format[i];
        }
    }

    return result;
}

许多问题:

  1. strcpy(result+i+strlen(strings),format+i-memo+2);是应对格式的结果。您需要的是将作为变量参数传递的字符串复制到结果字符串。 Futhermore result+i+strlen(strings)没有任何意义:你为什么写这个?
  2. 每次找到格式说明符i=0时都有%:为什么?这是分段错误的根本原因。每次找到%时循环都会重新启动,并且会请求一个新的变量参数,但是还没有其他参数可以被检索。
  3. 你必须有2个不同的索引来解析输入格式sting并索引结果字符串,因为输出字符串可能会有不同的长度。

答案 1 :(得分:1)

使用vsnprintf非常简单:

char *form(const char *format, ...)
{
    va_list va;
    va_start(va, format);

    int result = vsnprintf(NUll, 0, format, va);

    // Error checking
    if (result < 0)
        return NULL;

    // Here result is the number of bytes we need to allocate (excluding terminator)
    char *string = malloc(result + 1);

    // Now do the actual formatting
    vsnprintf(string, result + 1, format, va);

    return string;
}

重要提示:请记住free您返回的字符串。