当 strtok 返回 NULL 时，strtol 给出段错误

Question

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

const char* getfield(char* line, int num) {
        const char* tok;
        for (tok = strtok(line, ",");
                tok && *tok;
                tok = strtok(NULL, ",\n"))
        {
            if (!--num)
                return tok;
        }
        return NULL;
    }

    int main()
    {
        FILE* stream = fopen("b.csv", "r");
        char line[1024];
        char *pstr;int num;
         const char* value;

        while (fgets(line, 1024, stream))
        {
            char* tmp = strdup(line);
        //printf("Field 3 would be %s\n", getfield(tmp, 3));    
        value=getfield(tmp, 3);
        num =strtol(value,&pstr,10);
        printf("Field 3 would be %d\n", num);
        // NOTE strtok clobbers tmp
            free(tmp);
        }
    }

/* b.csv

301,36,15
302,88,75

//我的输出

Field 3 would be 15
Field 3 would be 75

*/

问题是：/* b.csv

301,36,15
 302,88,
 ,,,34

如果表被破坏如上“strtok”返回NULL，所以“strtol”给出“segfault”..如何解决它？

这里的主要问题是如果 2nd 不存在，它会将 3rd 视为第二个并给出段错误！例如，在 b.csv “,,,34” 的第三行中是否存在表示第三个值，但它的行为就像“34”是第一个值，第二个和第三个分别是 NULL !!

Answer 1

为什么不能只检查从getfield(tmp, 3); 获得的value 是否为NULL，而不是在返回NULL 时调用strtol？另一种解决方法是在getfield 中创建一个static char* not_found = ""; 并返回地址而不是NULL，然后strtol 将不会出现段错误。

更新

因为我发现strtok 在这种情况下真的很无助，所以我尝试编写与strchr 相同的代码：

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

    char* getfield(char* line, int num) {
        char* tok = line;
        char* result;
        if (line)
        {
            do
            {
                if (!--num)
                {
                    tok = strchr(line, ',');
                    if (tok == NULL)
                    {
                        tok = &line[strlen(line)];
                    }
                    size_t fieldlen = tok - line;
                    if (fieldlen)
                    {
                        result = (char*)malloc(fieldlen+1);
                        result[fieldlen] = '\0';
                        strncpy(result, line, fieldlen);
                        return result;
                    }
                    else
                    {
                        break;
                    }
                }
                tok = strchr(line, ',');
                line = tok + 1;
            } while (tok);
        }
        result = (char*)malloc(2);
        strcpy(result, "0");
        return result;
    }

    int main()
    {
        FILE* stream = fopen("b.csv", "r");
        char line[1024];
        char *pstr;int num;
        char* value;

        while (fgets(line, 1024, stream))
        {
            char* tmp = strdup(line);
            //printf("Field 3 would be %s\n", getfield(tmp, 3));    
            value=getfield(tmp, 3);
            num =strtol(value,&pstr,10);
            free(value);
            printf("Field 3 would be %d\n", num);
            // NOTE strtok clobbers tmp
            free(tmp);
        }
    }

这适用于输入文件：

    10,,30
    10,

如果没有找到，代码返回0，你可以改变它，结果是动态分配的。我希望这会有所帮助，对我的教训是 - 解析字符串时避免使用 C：D

Answer 2

而不是

num =strtol(value,&pstr,10);
printf("Field 3 would be %d\n", num);

使用这个：

if (value!=NULL) {         
    num =strtol(value,&pstr,10);
    printf("Field 3 would be %d\n", num);
}
else {
    printf("Field 3 does not exist\n");
}

你必须在调用strtol()函数之前检查value!= NULL

编辑修复getfield()函数的返回

在for循环中尝试更改：（我没有测试它但它可以解决你的问题）

 tok = strtok(NULL, ",\n")

通过

 tok = strtok(tok+strlen(tok)+1, ",\n")