Optarg和命令行参数

时间:2013-11-02 23:28:59

标签: c malloc command-line-arguments

我从用户那里获得命令行参数。

然后我为命令切换案例,例如:

    case 'f':
         *file_name = optarg;
         break;

我不确定我是否需要malloc作为指针,因为我不完全理解optarg。

这是file_name的声明方式:

char **file_name;

我应该

int length = strlen(optarg); // This gives a warning about types when compiling.

然后malloc为字符串长度+ 1?

如何针对此类问题进行malloc?请记住,用户输入的文件名为** argv。

编辑:这就是我调用此函数并仍然出现分段错误的方法。

int main(int argc, char **argv)
{
   char **file_name;
   parser(argc, argvm file_name);
}

void parser(int argc, char **argv, char **file_name)
{
  // Switch cases.
}

2 个答案:

答案 0 :(得分:5)

'optarg'只是指向argv []中元素的指针。因此,不分配内存并复制'optarg'指向的值是安全的。

假设您的程序使用以下参数调用:

myapp -a "hello" -b "world"

您的代码是:

#include <stdio.h>
#include <getopt.h>

void parse_options(int argc, char* argv[], char ** first_arg, char ** second_arg)
{
  const char* opt_string = "a:b:";
  int opt = -1;
  opt = getopt(argc, argv, opt_string);
  while (opt != -1) {
    switch(opt) {
      case 'a':
        *first_arg = optarg; /* points to argv[2]="hello" */
        break;
      case 'b':
        *second_arg = optarg; /* points to argv[4]="world" */
        break;
      default:
        break;
      }
    opt = getopt(argc, argv, opt_string);
  }
}

int main(int argc, char* argv[])
{
  char* first = 0;
  char* second = 0;
  parse_options(argc, argv, &first, &second);
  printf("first=%s, second=%s\n", first, second);
  return 0;
}

我的输出:

freebsd% gcc -Wall main.c
freebsd% ./a.out -a hello -b world
first=hello, second=world

答案 1 :(得分:1)

你说你有:

char **file_name;
...
switch (opt)
{
case 'f':
     *file_name = optarg;
     break;
...
}

代码崩溃是因为您没有分配空间或初始化变量file_name

你需要做什么取决于你想要发生什么。通常,您会将file_name的定义更改为:

char *file_name = 0;
...
switch (opt)
{
case 'f':
     file_name = optarg;
     break;
...
}

这允许您在循环/切换后检测是否提供了文件名,如果没有,则可以提供默认值或报告错误。在这种情况下,您可以检测先前是否提供了文件名,如果是,则检测对象。

替代方案是您希望在命令行上允许多次使用-f。然后你需要建立一个指针数组,并保持计数。你可以写:

char **file_name = 0;
size_t num_files = 0;
size_t max_files = 0;
...
switch (opt)
{
case 'f':
     if (num_files == max_files)
     {
         size_t new_files = (max_files + 2) * 2;
         void  *new_space = realloc(file_name, new_files * sizeof(*file_name));
         if (new_space == 0)
             ...report out of memory error...
         file_name = new_space;
         max_files = new_files;
     }
     file_name[num_files++] = optarg;
     break;
...
}

这使用realloc()的奇数属性,如果传入的指针为NULL,则模拟malloc()。或者,您可以在循环外执行初始malloc()分配,并在循环内执行({相同)realloc()。大小的计算第一次分配4个条目,然后是12个,然后是28个等等。如果您担心过度分配,可以在完成循环后再次使用realloc()来调整大小正确的大小,但它可能不会影响很多。

循环之后,您有一个可以处理的文件列表:

for (size_t i = 0; i < num_files; i++)
    process_file(file_name[i]);
相关问题