C - 尝试在linux中构建一个简单的shell,并且在循环中重新分配strtok,realloc

时间:2015-03-20 19:25:14

标签: c linux realloc strtok

尝试在linux中构建shell实现以进行分配并遇到问题。在循环运行3/4次后,我收到以下错误

*`./1':realloc()出错:下一个大小无效:0x000000000132c150 * 中止(核心倾销)

我已经将部分代码留了很长时间,我知道它们没有任何问题。如果您需要其他部分,我也可以添加它们。 在此先感谢:)

注意:argc和argv是全局变量

int argc = 0;
char **argv = NULL; 

int main(void)
{   
    while(1)
    {
        catch_signal();
        printDate();
        readCommand();

        if(strcmp(argv[0], "cd") == 0)
        {
            changeDirectory();
        }
        else
        {
            executeCommand();
        }
        //free(argv);
    }   
}

void readCommand()
{
    char *buffer = NULL;
    char *token = " ";
    char *p;

    size_t len = 0;
    int i = 0;

    getline(&buffer, &len, stdin);
    p = strtok(buffer, "\n");
    p = strtok(buffer, token);

    while(p)
    {
        argv = realloc(argv, sizeof(char *) * ++i);

        argv[i - 1] = p;
        p = strtok(NULL, token);
    }
    argc = i;
    argv[argc] = NULL;
    free(p);
}

void executeCommand()
{
    int status;
    pid_t pid;

    if((pid = fork()) < 0)
    {
        printf("Error: forking child process failed.");
    }
    else if(pid == 0)
    {
       if(execvp(argv[0], argv) < 0)
       {
           printf("error");
           exit(1);
       }
    }
    else
    {
       while(wait(&status) != pid);
    }
}

1 个答案:

答案 0 :(得分:2)

您正在重新分配argv数组1指针太短。最终的NULL指针没有足够的空间。考虑使用以下代码替换while(p)循环:

if (!p) {
    /* deal with empty command line */
}

while (p) {
    argv = realloc(argv, sizeof(char *) * (i + 2));
    argv[i++] = p;
    p = strtok(NULL, token);
}
如果用户输入了实际命令,则

p应以NULL开头。但是如果缓冲区包含空字符串或只是空字符串呢?在这些情况下,strtok会直接返回NULL。你应该忽略这样的命令行。