尝试在 C 中使用 execvp() 并在 unix 中使用用户输入

Question

我正在尝试制作一个程序，提示用户输入命令，然后使用 exec 执行该命令。

例如，如果他们给我“ls -la”，我就必须执行该命令。我已经尝试过以下代码：

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

int main()
{

    int ret, num_args;

    printf("Enter number of arguments (Example: \"ls -la\" has 1 argument): ");
    scanf("%d", &num_args);

    char *cmd[num_args];

    printf("Enter command name: ");
    scanf("%s", &cmd[0]);

    int i;
    for (i = 0; i < num_args; i++)
    {
            printf("Enter parameter: ");
            scanf("%s", &cmd[i]);
    }

    execvp(cmd[0], cmd);
}

但是，当我尝试以下运行时，它给了我一个“分段错误”

$ ./a.out 
Enter number of arguments (Example: "ls -la" has 1 argument): 2
Enter command name: ls
Enter parameter: -la
Enter parameter: .
Segmentation fault
$

有什么想法吗？

Answer 1

如果您的实现支持它，您应该使用更安全的 getline() 而不是 scanf() 或 fgets()。 getline() 将安全地处理长行和 NULL 字符。它将分配足够的内存来容纳整行。 getline() 可以分配内存，因此您稍后必须自己释放它。

这是 glibc getline()文档。

这是使用 getline 的快速修改（它仍然需要工作、错误检查，我还没有完全检查它的正确性）：

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

int main()
{

    printf("Enter number of arguments (Example: \"ls -la\" has 1 argument): \n");

    char *num = NULL;
    size_t sz = 0;
    getline(&num, &sz, stdin);

    int num_args;
    sscanf(num, "%d", &num_args);

    char *cmd[num_args+2];
    memset(cmd, 0, sizeof(char*) * (num_args+2));

    printf("Enter command name: \n");


    int len = getline(&cmd[0], &sz, stdin); 

    cmd[0][len-1] = '\0';

    int i;
    for (i = 1; i < num_args+1; i++)
    {
        printf("Enter parameter: \n");
        sz = 0;
        len = getline(&cmd[i], &sz, stdin);
        cmd[i][len-1] = '\0';
    }

    return execvp(cmd[0], cmd);

}

Answer 2

您需要为字符串分配内存。以下行仅分配 num_args 值指向 char 的指针：

char *cmd[num_args];

首先，您将获得 num_args + 1 字符串（不要忘记命令本身是 cmd[0]）。最简单的方法是将内存静态分配为字符缓冲区数组：

const unsigned int MAX_LEN = 512; // Arbitrary number
char cmd[num_args + 1][MAX_LEN];

但是，现在您不能使用 scanf 读取一行，因为用户可以输入比字符缓冲区长的字符串。相反，您必须使用 fgets，它可以限制用户可以输入的字符数：

fgets(cmd[i], MAX_LEN, stdin);

请记住，fgets 还会读取换行符，因此请确保剥去所有出现的杂散的东西（但不要假设它们在那里）。

Answer 3

此外，您还需要在传递给 execvp 的 argv 中添加一个条目，该条目必须为 (char *)NULL 以让它知道它已到达列表末尾。

Answer 4

您实际上尚未为 cmd 数组指向的字符串分配任何内存。

Answer 5

查看 scanf() 的手册页。它可以做的最巧妙的事情之一是动态自动分配字符串缓冲区，您需要提供一个指向字符串的指针，而不是仅仅传递字符串并提供 %as 格式。

char *my_string;
scanf("%as", &my_string);

然后你就不需要费心去预分配，不需要费心缓冲区溢出等等。只要记住在使用完它之后 free() 它即可。