cp/shell/cmd_helpers.c

   1 /*
   2  * (C) Copyright 2007-2010  Josef 'Jeff' Sipek <jeffpc@josefsipek.net>
   3  *
   4  * This file is released under the GPLv2.  See the COPYING file for more
   5  * details.
   6  */
   7
   8 static char* __extract_dec(char *str, u64 *val)
   9 {
  10         u64 res;
  11         u64 tmp;
  12         int len;
  13
  14         if (!str || !val)
  15                 return ERR_PTR(-EINVAL);
  16
  17         res = 0;
  18         len = -1;
  19
  20         for (; str && *str != '\0'; str++, len++) {
  21                 if (*str >= '0' && *str <= '9')
  22                         tmp = *str - '0';
  23                 else if (*str == ' ' || *str == '\t')
  24                         break;
  25                 else
  26                         return ERR_PTR(-EINVAL);
  27
  28                 res = (res * 10) + tmp;
  29         }
  30
  31         if (len == -1)
  32                 return ERR_PTR(-EINVAL);
  33
  34         *val = res;
  35
  36         return str;
  37 }
  38
  39 static char* __extract_hex(char *str, u64 *val)
  40 {
  41         u64 res;
  42         u64 tmp;
  43         int len;
  44
  45         if (!str || !val)
  46                 return ERR_PTR(-EINVAL);
  47
  48         res = 0;
  49         len = -1;
  50
  51         for (; str && *str != '\0'; str++, len++) {
  52                 if (*str >= '0' && *str <= '9')
  53                         tmp = *str - '0';
  54                 else if (*str >= 'A' && *str <= 'F')
  55                         tmp = *str - 'A' + 10;
  56                 else if (*str >= 'a' && *str <= 'f')
  57                         tmp = *str - 'a' + 10;
  58                 else if (*str == ' ' || *str == '\t')
  59                         break;
  60                 else
  61                         return ERR_PTR(-EINVAL);
  62
  63                 res = (res << 4) | tmp;
  64         }
  65
  66         if (len == -1)
  67                 return ERR_PTR(-EINVAL);
  68
  69         *val = res;
  70
  71         return str;
  72 }
  73
  74 static char* __consume_ws(char *str)
  75 {
  76         if (!str)
  77                 return str;
  78
  79         /* consume any extra whitespace */
  80         while(*str == ' ' || *str == '\t')
  81                 str++;
  82
  83         return str;
  84 }