src/bin/getopts.c - master - githubmirror/libsmbios

src/bin/getopts.c
179 строк6 KB
Michael E Brown
fix compiler warning.
15 май 2009, 00:43
15 май 2009, 00:43fc135b8
/* getopts.c - Command line argument parser
 *
 * Whom: Steve Mertz <steve@dragon-ware.com>
 * Date: 20010111
 * Why:  Because I couldn't find one that I liked. So I wrote this one.
 *
*/
/*
 * Copyright (c) 2001, 2002, Steve Mertz <steve@dragon-ware.com>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer.
 *
 * Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *
 * Neither the name of Dragon Ware nor the names of its contributors may be
 * used to endorse or promote products derived from this software without
 * specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <libintl.h>

#include "getopts.h"

int option_index = 1;
/* int getopts_usage()
 *
 *  Returns: 1 - Successful
 */
int getopts_usage(char *progName, struct options opts[])
{
  int count;

  printf("Usage: %s [options]\n\n", progName);
  printf("  --help,\t-h\t\t\tDisplays this information\n");
  for (count = 0; opts[count].description; count++)
    {
      char *cmd = NULL;
      if (opts[count].name && opts[count].shortName)
        {
          cmd = calloc(1, strlen(opts[count].name) + strlen(opts[count].shortName) + 15);
          if (opts[count].args)
            {
              sprintf(cmd, "--%s,\t-%s <args>\t\t", opts[count].name, opts[count].shortName);
            }
          else
            {
              sprintf(cmd, "--%s,\t-%s\t\t\t", opts[count].name, opts[count].shortName);
            }
        }
      else if (opts[count].name)
        {
          cmd = calloc(1, strlen(opts[count].name) + 15);
          if (opts[count].args)
            {
              sprintf(cmd, "--%s <args>\t\t\t", opts[count].name);
            }
          else
            {
              sprintf(cmd, "--%s\t\t\t", opts[count].name);
            }
        }
      else if (opts[count].shortName)
        {
          cmd = calloc(1, strlen(opts[count].shortName) + 15);
          if (opts[count].args)
            {
              sprintf(cmd, "\t\t-%s <args>\t\t", opts[count].shortName);
            }
          else
            {
              sprintf(cmd, "\t\t-%s\t\t\t", opts[count].shortName);
            }
        }
      if(cmd)
        {
          printf("  %s%s\n", cmd, gettext(opts[count].description));
          free(cmd);
        }
    }
  return 1;
}

/* int getopts()
 *
 * Returns: -1 - Couldn't allocate memory.  Please handle me.
 *          0  - No arguements to parse
 *          #  - The number in the struct for the matched arg.
 *
*/
int getopts(int argc, char **argv, struct options opts[], char **args)
{
  int count1 = 0;
  size_t sizeOfArgs = 0;

  if (argc == 1 || option_index == argc)
    return 0;

/* Search for '-h' or '--help' first.  Then we can just exit */
  for (count1 = 1; count1 < argc; count1++)
    {
      if (!strcmp(argv[count1], "-h") || !strcmp(argv[count1], "--help"))
        {
useage:
          getopts_usage(argv[0], opts);
          exit(0);
        }
    }
/* End of -h --help section */
  *args = NULL;
  if (option_index <= argc)
    {
      for (count1 = 0; opts[count1].description; count1++)
        {
          if ((opts[count1].name && !strcmp(opts[count1].name,
              (argv[option_index]+2))) ||
              (opts[count1].shortName && !strcmp(opts[count1].shortName,
              (argv[option_index]+1))))
            {
              if (opts[count1].args)
                {
                  option_index++;
                  if (option_index >= argc)
                    goto useage;
/* This grossness that follows is to supporte having a '-' in the
   argument.  It's all squished together like this because I use
   an 80 char wide screen for my coding.  If you don't like it, help
   yourself to fixing it so you do. */
                  if (*argv[option_index] == '-')
                    {
                      int optionSeeker;
                      for (optionSeeker = 0; opts[optionSeeker].description;
                           optionSeeker++)
                        {
                          if ((opts[optionSeeker].name &&
                               !strcmp(opts[optionSeeker].name,
                                       (argv[option_index]+2))) ||
                               (opts[optionSeeker].shortName &&
                               !strcmp(opts[optionSeeker].shortName,
                                       (argv[option_index]+1))))
                            {
                              goto useage;
                            }
                        }
/* End of gross hack for supporting '-' in arguments. */
                    }
                  sizeOfArgs = strlen(argv[option_index]);
                  if ((*args = calloc(1, sizeOfArgs+1)) == NULL)
                    return -1;
                  strncpy(*args, argv[option_index], sizeOfArgs);
                }
              option_index++;
              return opts[count1].number;
            }
        }
    }
  return 0;
}