git.ozlabs.org Git - ccan/blob - ccan/opt/helpers.c

   1 /* Licensed under GPLv3+ - see LICENSE file for details */
   2 #include <ccan/opt/opt.h>
   3 #include <string.h>
   4 #include <stdlib.h>
   5 #include <errno.h>
   6 #include <stdio.h>
   7 #include "private.h"
   8
   9 /* Upper bound to sprintf this simple type?  Each 3 bits < 1 digit. */
  10 #define CHAR_SIZE(type) (((sizeof(type)*CHAR_BIT + 2) / 3) + 1)
  11
  12 /* FIXME: asprintf module? */
  13 static char *arg_bad(const char *fmt, const char *arg)
  14 {
  15         char *str = malloc(strlen(fmt) + strlen(arg));
  16         sprintf(str, fmt, arg);
  17         return str;
  18 }
  19
  20 char *opt_set_bool(bool *b)
  21 {
  22         *b = true;
  23         return NULL;
  24 }
  25
  26 char *opt_set_invbool(bool *b)
  27 {
  28         *b = false;
  29         return NULL;
  30 }
  31
  32 char *opt_set_bool_arg(const char *arg, bool *b)
  33 {
  34         if (!strcasecmp(arg, "yes") || !strcasecmp(arg, "true"))
  35                 return opt_set_bool(b);
  36         if (!strcasecmp(arg, "no") || !strcasecmp(arg, "false"))
  37                 return opt_set_invbool(b);
  38
  39         return opt_invalid_argument(arg);
  40 }
  41
  42 char *opt_set_invbool_arg(const char *arg, bool *b)
  43 {
  44         char *err = opt_set_bool_arg(arg, b);
  45
  46         if (!err)
  47                 *b = !*b;
  48         return err;
  49 }
  50
  51 /* Set a char *. */
  52 char *opt_set_charp(const char *arg, char **p)
  53 {
  54         *p = (char *)arg;
  55         return NULL;
  56 }
  57
  58 /* Set an integer value, various forms.  Sets to 1 on arg == NULL. */
  59 char *opt_set_intval(const char *arg, int *i)
  60 {
  61         long l;
  62         char *err = opt_set_longval(arg, &l);
  63
  64         if (err)
  65                 return err;
  66         *i = l;
  67         /* Beware truncation... */
  68         if (*i != l)
  69                 return arg_bad("value '%s' does not fit into an integer", arg);
  70         return err;
  71 }
  72
  73 char *opt_set_uintval(const char *arg, unsigned int *ui)
  74 {
  75         int i;
  76         char *err = opt_set_intval(arg, &i);
  77
  78         if (err)
  79                 return err;
  80         if (i < 0)
  81                 return arg_bad("'%s' is negative", arg);
  82         *ui = i;
  83         return NULL;
  84 }
  85
  86 char *opt_set_longval(const char *arg, long *l)
  87 {
  88         char *endp;
  89
  90         /* This is how the manpage says to do it.  Yech. */
  91         errno = 0;
  92         *l = strtol(arg, &endp, 0);
  93         if (*endp || !arg[0])
  94                 return arg_bad("'%s' is not a number", arg);
  95         if (errno)
  96                 return arg_bad("'%s' is out of range", arg);
  97         return NULL;
  98 }
  99
 100 char *opt_set_ulongval(const char *arg, unsigned long *ul)
 101 {
 102         long int l;
 103         char *err;
 104
 105         err = opt_set_longval(arg, &l);
 106         if (err)
 107                 return err;
 108         *ul = l;
 109         if (l < 0)
 110                 return arg_bad("'%s' is negative", arg);
 111         return NULL;
 112 }
 113
 114 char *opt_inc_intval(int *i)
 115 {
 116         (*i)++;
 117         return NULL;
 118 }
 119
 120 /* Display version string. */
 121 char *opt_version_and_exit(const char *version)
 122 {
 123         printf("%s\n", version);
 124         exit(0);
 125 }
 126
 127 char *opt_usage_and_exit(const char *extra)
 128 {
 129         printf("%s", opt_usage(opt_argv0, extra));
 130         exit(0);
 131 }
 132
 133 void opt_show_bool(char buf[OPT_SHOW_LEN], const bool *b)
 134 {
 135         strncpy(buf, *b ? "true" : "false", OPT_SHOW_LEN);
 136 }
 137
 138 void opt_show_invbool(char buf[OPT_SHOW_LEN], const bool *b)
 139 {
 140         strncpy(buf, *b ? "false" : "true", OPT_SHOW_LEN);
 141 }
 142
 143 void opt_show_charp(char buf[OPT_SHOW_LEN], char *const *p)
 144 {
 145         size_t len = strlen(*p);
 146         buf[0] = '"';
 147         if (len > OPT_SHOW_LEN - 2)
 148                 len = OPT_SHOW_LEN - 2;
 149         strncpy(buf+1, *p, len);
 150         buf[1+len] = '"';
 151         if (len < OPT_SHOW_LEN - 2)
 152                 buf[2+len] = '\0';
 153 }
 154
 155 /* Set an integer value, various forms.  Sets to 1 on arg == NULL. */
 156 void opt_show_intval(char buf[OPT_SHOW_LEN], const int *i)
 157 {
 158         snprintf(buf, OPT_SHOW_LEN, "%i", *i);
 159 }
 160
 161 void opt_show_uintval(char buf[OPT_SHOW_LEN], const unsigned int *ui)
 162 {
 163         snprintf(buf, OPT_SHOW_LEN, "%u", *ui);
 164 }
 165
 166 void opt_show_longval(char buf[OPT_SHOW_LEN], const long *l)
 167 {
 168         snprintf(buf, OPT_SHOW_LEN, "%li", *l);
 169 }
 170
 171 void opt_show_ulongval(char buf[OPT_SHOW_LEN], const unsigned long *ul)
 172 {
 173         snprintf(buf, OPT_SHOW_LEN, "%lu", *ul);
 174 }