Move opts/ into util/opts/

util/opts/opts.c

@@ -0,0 +1,179 @@
+#include "opts.h"
+#include "util/opts/log.h"
+
+#include <stddef.h>
+#include <stdio.h>
+#include <limits.h>
+
+#define OPT_LENGTH 30
+
+static void print_help(option_t *options, const char *cmd)
+{
+  printf("Usage:\n");
+  printf("$ %s <options...>\n", cmd);
+  printf("\n");
+
+  printf("Options:\n");
+  for (int i = 0; options[i].name; i++) {
+    option_t *option = &options[i];
+    int len = 0;
+
+    len += printf("  -%s", option->name);
+    if (option->default_value) {
+      len += printf("[=%s]", option->default_value);
+    } else if (option->value_mapping) {
+      len += printf("=arg");
+    } else {
+      len += printf("=%s", option->format);
+    }
+    if (len < OPT_LENGTH) {
+      printf("%*s", OPT_LENGTH-len, " ");
+    }
+
+    printf("- %s", option->description);
+    if (option->value_mapping) {
+      printf(" Values: ");
+      for (int j = 0; option->value_mapping[j].name; j++) {
+        if (j > 0) printf(", ");
+        printf("%s", option->value_mapping[j].name);
+      }
+      printf(".");
+    }
+    printf("\n");
+  }
+  printf("\n");
+
+  printf("Configuration:\n");
+  for (int i = 0; options[i].name; i++) {
+    option_t *option = &options[i];
+    int len = 0;
+
+    if (option->value_list) {
+      char *string = option->value_list;
+      char *token;
+
+      if (!*string)
+        continue;
+
+      while ((token = strsep(&string, OPTION_VALUE_LIST_SEP)) != NULL) {
+        len = printf("  -%s=", option->name);
+        if (len < OPT_LENGTH) {
+          printf("%*s", OPT_LENGTH-len, " ");
+        }
+        printf("%s\n", token);
+      }
+    } else if (option->value_string) {
+      len += printf("  -%s=", option->name);
+      if (len < OPT_LENGTH) {
+        printf("%*s", OPT_LENGTH-len, " ");
+      }
+      printf(option->format, option->value_string);
+      printf("\n");
+    } else {
+      len += printf("  -%s=", option->name);
+      if (len < OPT_LENGTH) {
+        printf("%*s", OPT_LENGTH-len, " ");
+      }
+      if (option->value_mapping) {
+        for (int j = 0; option->value_mapping[j].name; j++) {
+          if (option->value_mapping[j].value == *option->value) {
+            printf("%s - ", option->value_mapping[j].name);
+            break;
+          }
+        }
+      }
+
+      unsigned mask = UINT_MAX >> ((sizeof(*option->value) - option->size) * 8);
+      printf(option->format, *option->value & mask);
+      printf("\n");
+    }
+  }
+  printf("\n");
+}
+
+static int parse_opt(option_t *options, const char *key)
+{
+  option_t *option = NULL;
+  const char *value = strchr(key, '=');
+
+  for (int i = 0; options[i].name; i++) {
+    if (value) {
+      if (!strncmp(key, options[i].name, value - key)) {
+        option = &options[i];
+        value++; // ignore '='
+        break;
+      }
+    } else {
+      // require exact match
+      if (!strcmp(key, options[i].name)) {
+        option = &options[i];
+        value = option->default_value;
+        break;
+      }
+    }
+  }
+
+  LOG_DEBUG(NULL, "Parsing '%s'. Got value='%s', and option='%s'", key, value, option ? option->name : NULL);
+
+  if (!option || !value) {
+    return -EINVAL;
+  }
+
+  int ret = 0;
+  if (option->value_list) {
+    char *ptr = option->value_list;
+
+    if (*ptr) {
+      strcat(ptr, OPTION_VALUE_LIST_SEP);
+      ptr += strlen(ptr);
+    }
+
+    ret = sscanf(value, option->format, ptr);
+  } else if (option->value_string) {
+    ret = sscanf(value, option->format, option->value_string);
+  } else if (option->value_mapping) {
+    for (int j = 0; option->value_mapping[j].name; j++) {
+      if (!strcmp(option->value_mapping[j].name, value)) {
+        *option->value_uint = option->value_mapping[j].value;
+        return 1;
+      }
+    }
+
+    ret = sscanf(value, option->format, option->value);
+  } else {
+    ret = sscanf(value, option->format, option->value);
+  }
+  return ret;
+}
+
+int parse_opts(option_t *options, int argc, char *argv[])
+{
+  int arg;
+
+  for (arg = 1; arg < argc; arg++) {
+    const char *key = argv[arg];
+
+    if (key[0] == '-') {
+      key++;
+      if (key[0] == '-')
+        key++;
+    } else {
+      LOG_ERROR(NULL, "The '%s' is not option (should start with - or --).", key);
+    }
+
+    if (!strcmp(key, "help")) {
+      print_help(options, argv[0]);
+      return -1;
+    }
+
+    int ret = parse_opt(options, key);
+    if (ret <= 0) {
+      LOG_ERROR(NULL, "Parsing '%s' returned '%d'.", argv[arg], ret);
+    }
+  }
+
+  return 0;
+
+error:
+  return -1;
+}