cmdline parsing of --ipv4/6 plucked into util

2 files changed, 34 insertions, 0 deletions

diff --git a/toxcore/util.c b/toxcore/util.c
index 19d464d4..7b5ddc49 100644
--- a/toxcore/util.c
+++ b/toxcore/util.c
@@ -43,6 +43,38 @@ void id_cpy(uint8_t *dest, uint8_t *src)
     memcpy(dest, src, CLIENT_ID_SIZE);
 }
 
+int cmdline_parsefor_ipv46(int argc, char **argv, uint8_t *ipv6enabled)
+{
+    int argvoffset = 0, argi;
+    for(argi = 1; argi < argc; argi++)
+        if (!strncasecmp(argv[argi], "--ipv", 5)) {
+            if (argv[argi][5] && !argv[argi][6]) {
+                char c = argv[argi][5];
+                if (c == '4')
+                    *ipv6enabled = 0;
+                else if (c == '6')
+                    *ipv6enabled = 1;
+                else {
+                    printf("Invalid argument: %s. Try --ipv4 or --ipv6!\n", argv[argi]);
+                    return -1;
+                }
+            }
+            else {
+                printf("Invalid argument: %s. Try --ipv4 or --ipv6!\n", argv[argi]);
+                return -1;
+            }
+
+            if (argvoffset != argi - 1) {
+                printf("Argument must come first: %s.\n", argv[argi]);
+                return -1;
+            }
+
+            argvoffset++;
+        }
+
+    return argvoffset;
+};
+
 #ifdef LOGGING
 char logbuffer[512];
 static FILE *logfile = NULL;
diff --git a/toxcore/util.h b/toxcore/util.h
index 71be84ca..e7be2d51 100644
--- a/toxcore/util.h
+++ b/toxcore/util.h
@@ -16,6 +16,8 @@ uint64_t random_64b();
 bool id_eq(uint8_t *dest, uint8_t *src);
 void id_cpy(uint8_t *dest, uint8_t *src);
 
+int cmdline_parsefor_ipv46(int argc, char **argv, uint8_t *ipv6enabled);
+
 #undef LOGGING
 /* #define LOGGING */
 #ifdef LOGGING