libc/stdio/doscanf.c

   1 /*
   2  *  ZeX/OS
   3  *  Copyright (C) 2008  Tomas 'ZeXx86' Jedrzejek (zexx86@gmail.com)
   4  *  Copyright (C) 2009  Tomas 'ZeXx86' Jedrzejek (zexx86@gmail.com)
   5  *
   6  *  This program is free software: you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation, either version 3 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20
  21 #include <_printf.h> /* fnptr_t */
  22 #include <string.h> /* strlen() */
  23 #include <stdarg.h> /* va_list, va_arg() */
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26
  27 static int scanf_getint (int *n)
  28 {
  29         char *str = (char *) malloc (sizeof (char) * 128);
  30
  31         if (!str)
  32                 return 0;
  33
  34         unsigned i = 0;
  35
  36         for (;;) {
  37                 schedule ();
  38
  39                 char c = getch ();
  40
  41                 if (!c)
  42                         continue;
  43
  44                 if (c == '\n' || i > 127)
  45                         break;
  46
  47                 putchar (c);
  48
  49                 if (c == ' ' && i)
  50                         break;
  51
  52                 str[i] = c;
  53
  54                 if (c == '\b' && i)
  55                         i --;
  56                 else
  57                         i ++;
  58         }
  59
  60         if (!i)
  61                 return 0;
  62
  63         str[i] = '\0';
  64
  65         i = 0;
  66
  67         *n = atoi (str);
  68
  69         free (str);
  70
  71         return 1;
  72 }
  73
  74 /* TODO: implement real floatint point format */
  75 static int scanf_getfloat (float *n)
  76 {
  77         char *str = (char *) malloc (sizeof (char) * 128);
  78
  79         if (!str)
  80                 return 0;
  81
  82         unsigned i = 0;
  83
  84         for (;;) {
  85                 schedule ();
  86
  87                 char c = getch ();
  88
  89                 if (!c)
  90                         continue;
  91
  92                 if (c == '\n' || i > 127)
  93                         break;
  94
  95                 putchar (c);
  96
  97                 if (c == ' ' && i)
  98                         break;
  99
 100                 str[i] = c;
 101
 102                 if (c == '\b' && i)
 103                         i --;
 104                 else
 105                         i ++;
 106         }
 107
 108         if (!i)
 109                 return 0;
 110
 111         str[i] = '\0';
 112
 113         i = 0;
 114
 115         *n = (float) atoi (str);
 116
 117         free (str);
 118
 119         return 1;
 120 }
 121
 122 static int scanf_gethex (int *n)
 123 {
 124         char *str = (char *) malloc (sizeof (char) * 128);
 125
 126         if (!str)
 127                 return 0;
 128
 129         unsigned i = 0;
 130
 131         for (;;) {
 132                 schedule ();
 133
 134                 char c = getch ();
 135
 136                 if (!c)
 137                         continue;
 138
 139                 if (c == '\n' || i > 127)
 140                         break;
 141
 142                 putchar (c);
 143
 144                 if (c == ' ' && i)
 145                         break;
 146
 147                 str[i] = c;
 148
 149                 if (c == '\b' && i)
 150                         i --;
 151                 else
 152                         i ++;
 153         }
 154
 155         if (!i)
 156                 return 0;
 157
 158         str[i] = '\0';
 159
 160         i = 0;
 161
 162         char *endptr;
 163
 164         *n = strtol (str, &endptr, 16);
 165
 166         free (str);
 167
 168         return 1;
 169 }
 170
 171 static int scanf_getstring (unsigned char *n)
 172 {
 173         unsigned i = 0;
 174
 175         for (;;) {
 176                 schedule ();
 177
 178                 char c = getch ();
 179
 180                 if (!c)
 181                         continue;
 182
 183                 if (c == '\n')
 184                         break;
 185
 186                 putchar (c);
 187
 188                 if (c == ' ' && i)
 189                         break;
 190
 191                 n[i] = c;
 192
 193                 if (c == '\b' && i)
 194                         i --;
 195                 else
 196                         i ++;
 197         }
 198
 199         if (!i)
 200                 return 0;
 201
 202         n[i] = '\0';
 203
 204         return 1;
 205 }
 206
 207 static int scanf_getchar (unsigned char *n)
 208 {
 209         unsigned i = 0;
 210
 211         for (;;) {
 212                 schedule ();
 213
 214                 char c = getch ();
 215
 216                 if (!c)
 217                         continue;
 218
 219                 if (c == '\n')
 220                         break;
 221
 222                 putchar (c);
 223
 224                 if (c == ' ' && i)
 225                         break;
 226
 227                 if (i == 0)
 228                         n[i] = c;
 229
 230                 if (c == '\b' && i)
 231                         i --;
 232                 else
 233                         i ++;
 234         }
 235
 236         if (!i)
 237                 return 0;
 238
 239         return 1;
 240 }
 241
 242 int vscanf (const char *fmt, va_list args)
 243 {
 244         unsigned count = 0;
 245         int *num = 0;
 246         float *fp = 0;
 247         unsigned char *buf = 0;
 248
 249         for (; *fmt; fmt ++) {
 250                 /* probably argument is here */
 251                 if (*fmt == '%') {
 252                         fmt ++;
 253
 254                         if (!*fmt)
 255                                 break;
 256
 257                         if (*fmt == 'd' || *fmt == 'i' || *fmt == 'u') {
 258                                 num = va_arg (args, int *);
 259
 260                                 if (scanf_getint (num))
 261                                         count ++;
 262
 263                                 continue;
 264                         }
 265
 266                         if (*fmt == 'x' || *fmt == 'X') {
 267                                 num = va_arg (args, int *);
 268
 269                                 if (scanf_gethex (num))
 270                                         count ++;
 271
 272                                 continue;
 273                         }
 274
 275                         if (*fmt == 'c') {
 276                                 buf = va_arg (args, unsigned char *);
 277
 278                                 if (scanf_getchar (buf))
 279                                         count ++;
 280
 281                                 continue;
 282                         }
 283
 284                         if (*fmt == 'f') {
 285                                 fp = va_arg (args, float *);
 286
 287                                 if (scanf_getfloat (fp))
 288                                         count ++;
 289
 290                                 continue;
 291                         }
 292
 293                         if (*fmt == 's') {
 294                                 buf = va_arg (args, unsigned char *);
 295
 296                                 if (scanf_getstring (buf))
 297                                         count ++;
 298
 299                                 continue;
 300                         }
 301
 302                         continue;
 303                 }
 304
 305                 count ++;
 306
 307         }
 308
 309         putchar ('\n');
 310
 311         return count;
 312 }
 313