cmdline: big rework and clean of cmdline library

[libcmdline.git] / src / lib / cmdline_parse_num.c

diff --git a/src/lib/cmdline_parse_num.c b/src/lib/cmdline_parse_num.c
index 1332efc..31ca78d 100644 (file)
--- a/src/lib/cmdline_parse_num.c
+++ b/src/lib/cmdline_parse_num.c
@@ -71,14 +71,6 @@
 //#define debug_printf(args...) printf(args)
 #define debug_printf(args...) do {} while(0)
 
-struct cmdline_token_ops cmdline_token_num_ops = {
-       .parse = cmdline_parse_num,
-       .complete_get_nb = NULL,
-       .complete_get_elt = NULL,
-       .get_help = cmdline_get_help_num,
-};
-
-
 enum num_parse_state_t {
        START,
        DEC_NEG,
@@ -102,42 +94,84 @@ enum num_parse_state_t {
 
 /* Keep it sync with enum in .h */
 static const char * num_help[] = {
-       "UINT8", "UINT16", "UINT32",
-       "INT8", "INT16", "INT32",
+       "UINT8", "UINT16", "UINT32", "UINT64",
+       "INT8", "INT16", "INT32", "INT64",
 #ifdef CMDLINE_HAVE_FLOAT
        "FLOAT",
 #endif
 };
 
-static inline int
-add_to_res(unsigned int c, uint32_t *res, unsigned int base)
+static int
+add_to_res(unsigned int c, uint64_t *res, unsigned int base)
 {
        /* overflow */
-       if ( (UINT32_MAX - c) / base < *res ) {
+       if ( (UINT64_MAX - c) / base < *res ) {
                return -1;
        }
 
-       *res = *res * base + c ;
+       *res = (uint64_t) (*res * base + c);
        return 0;
 }
 
+static int
+check_res_size(struct cmdline_token_num_data *nd, unsigned ressize)
+{
+       switch (nd->type) {
+               case INT8:
+               case UINT8:
+                       if (ressize < sizeof(int8_t))
+                               return -1;
+                       break;
+               case INT16:
+               case UINT16:
+                       if (ressize < sizeof(int16_t))
+                               return -1;
+                       break;
+               case INT32:
+               case UINT32:
+                       if (ressize < sizeof(int32_t))
+                               return -1;
+                       break;
+               case INT64:
+               case UINT64:
+                       if (ressize < sizeof(int64_t))
+                               return -1;
+                       break;
+#ifdef CMDLINE_HAVE_FLOAT
+               case FLOAT:
+                       if (ressize < sizeof(float))
+                               return -1;
+                       break;
+#endif
+               default:
+                       return -1;
+       }
+       return 0;
+}
 
 /* parse an int or a float */
-int
-cmdline_parse_num(cmdline_parse_token_hdr_t *tk, const char *srcbuf, void *res)
+static int
+cmdline_parse_num(cmdline_parse_token_hdr_t *tk, const char *srcbuf,
+                 void *res, unsigned ressize)
 {
        struct cmdline_token_num_data nd;
        enum num_parse_state_t st = START;
        const char * buf = srcbuf;
        char c = *buf;
-       uint32_t res1 = 0;
+       uint64_t res1 = 0;
 #ifdef CMDLINE_HAVE_FLOAT
-       uint32_t res2 = 0, res3 = 1;
+       uint64_t res2 = 0, res3 = 1;
 #endif
 
        memcpy(&nd, &((struct cmdline_token_num *)tk)->num_data, sizeof(nd));
 
-       while ( st != ERROR && c && ! cmdline_isendoftoken(c) ) {
+       /* check that we have enough room in res */
+       if (res) {
+               if (check_res_size(&nd, ressize) < 0)
+                       return -1;
+       }
+
+       while (st != ERROR && c != '\0') {
                debug_printf("%c %x -> ", c, c);
                switch (st) {
                case START:
@@ -389,6 +423,11 @@ cmdline_parse_num(cmdline_parse_token_hdr_t *tk, const char *srcbuf, void *res)
                                *(uint32_t *)res = (uint32_t) res1;
                        return (buf-srcbuf);
                }
+               else if ( nd.type == UINT64 ) {
+                       if (res)
+                               *(uint64_t *)res = res1;
+                       return (buf-srcbuf);
+               }
 #ifdef CMDLINE_HAVE_FLOAT
                else if ( nd.type == FLOAT ) {
                        if (res)
@@ -464,8 +503,8 @@ cmdline_parse_num(cmdline_parse_token_hdr_t *tk, const char *srcbuf, void *res)
 
 
 /* parse an int or a float */
-int
-cmdline_get_help_num(cmdline_parse_token_hdr_t *tk, char *dstbuf, unsigned int size)
+static int
+cmdline_help_num(cmdline_parse_token_hdr_t *tk, char *dstbuf, unsigned int size)
 {
        struct cmdline_token_num_data nd;
 
@@ -479,3 +518,12 @@ cmdline_get_help_num(cmdline_parse_token_hdr_t *tk, char *dstbuf, unsigned int s
        dstbuf[size-1] = '\0';
        return 0;
 }
+
+
+struct cmdline_token_ops cmdline_token_num_ops = {
+       .parse = cmdline_parse_num,
+       .complete_start = NULL,
+       .complete_iterate = NULL,
+       .complete_end = NULL,
+       .help = cmdline_help_num,
+};