205 lines
		
	
	
		
			4.8 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			205 lines
		
	
	
		
			4.8 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  * Copyright (c) 2003, 2007-14 Matteo Frigo
 | |
|  * Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology
 | |
|  *
 | |
|  * This program is free software; you can redistribute it and/or modify
 | |
|  * it under the terms of the GNU General Public License as published by
 | |
|  * the Free Software Foundation; either version 2 of the License, or
 | |
|  * (at your option) any later version.
 | |
|  *
 | |
|  * This program is distributed in the hope that it will be useful,
 | |
|  * but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
|  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | |
|  * GNU General Public License for more details.
 | |
|  *
 | |
|  * You should have received a copy of the GNU General Public License
 | |
|  * along with this program; if not, write to the Free Software
 | |
|  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 | |
|  *
 | |
|  */
 | |
| 
 | |
| 
 | |
| #include "kernel/ifftw.h"
 | |
| #include <string.h>
 | |
| #include <stddef.h>
 | |
| #include <stdarg.h>
 | |
| #include <stdio.h>
 | |
| 
 | |
| #ifdef USE_CTYPE
 | |
| #include <ctype.h>
 | |
| #else
 | |
| /* Screw ctype. On linux, the is* functions call a routine that gets
 | |
|    the ctype map in the current locale.  Because this operation is
 | |
|    expensive, the map is cached on a per-thread basis.  I am not
 | |
|    willing to link this crap with FFTW.  Not over my dead body.
 | |
| 
 | |
|    Sic transit gloria mundi.
 | |
| */
 | |
| #undef isspace
 | |
| #define isspace(x) ((x) >= 0 && (x) <= ' ')
 | |
| #undef isdigit
 | |
| #define isdigit(x) ((x) >= '0' && (x) <= '9')
 | |
| #undef isupper
 | |
| #define isupper(x) ((x) >= 'A' && (x) <= 'Z')
 | |
| #undef islower
 | |
| #define islower(x) ((x) >= 'a' && (x) <= 'z')
 | |
| #endif
 | |
| 
 | |
| static int mygetc(scanner *sc)
 | |
| {
 | |
|      if (sc->ungotc != EOF) {
 | |
| 	  int c = sc->ungotc;
 | |
| 	  sc->ungotc = EOF;
 | |
| 	  return c;
 | |
|      }
 | |
|      return(sc->getchr(sc));
 | |
| }
 | |
| 
 | |
| #define GETCHR(sc) mygetc(sc)
 | |
| 
 | |
| static void myungetc(scanner *sc, int c)
 | |
| {
 | |
|      sc->ungotc = c;
 | |
| }
 | |
| 
 | |
| #define UNGETCHR(sc, c) myungetc(sc, c)
 | |
| 
 | |
| static void eat_blanks(scanner *sc)
 | |
| {
 | |
|      int ch;
 | |
|      while (ch = GETCHR(sc), isspace(ch))
 | |
|           ;
 | |
|      UNGETCHR(sc, ch);
 | |
| }
 | |
| 
 | |
| static void mygets(scanner *sc, char *s, int maxlen)
 | |
| {
 | |
|      char *s0 = s;
 | |
|      int ch;
 | |
| 
 | |
|      A(maxlen > 0);
 | |
|      while ((ch = GETCHR(sc)) != EOF && !isspace(ch)
 | |
| 	    && ch != ')' && ch != '(' && s < s0 + maxlen)
 | |
| 	  *s++ = (char)(ch & 0xFF);
 | |
|      *s = 0;
 | |
|      UNGETCHR(sc, ch);
 | |
| }
 | |
| 
 | |
| static long getlong(scanner *sc, int base, int *ret)
 | |
| {
 | |
|      int sign = 1, ch, count;
 | |
|      long x = 0;     
 | |
| 
 | |
|      ch = GETCHR(sc);
 | |
|      if (ch == '-' || ch == '+') {
 | |
| 	  sign = ch == '-' ? -1 : 1;
 | |
| 	  ch = GETCHR(sc);
 | |
|      }
 | |
|      for (count = 0; ; ++count) {
 | |
| 	  if (isdigit(ch)) 
 | |
| 	       ch -= '0';
 | |
| 	  else if (isupper(ch))
 | |
| 	       ch -= 'A' - 10;
 | |
| 	  else if (islower(ch))
 | |
| 	       ch -= 'a' - 10;
 | |
| 	  else
 | |
| 	       break;
 | |
| 	  x = x * base + ch;
 | |
| 	  ch = GETCHR(sc);
 | |
|      }
 | |
|      x *= sign;
 | |
|      UNGETCHR(sc, ch);
 | |
|      *ret = count > 0;
 | |
|      return x;
 | |
| }
 | |
| 
 | |
| /* vscan is mostly scanf-like, with our additional format specifiers,
 | |
|    but with a few twists.  It returns simply 0 or 1 indicating whether
 | |
|    the match was successful. '(' and ')' in the format string match
 | |
|    those characters preceded by any whitespace.  Finally, if a
 | |
|    character match fails, it will ungetchr() the last character back
 | |
|    onto the stream. */
 | |
| static int vscan(scanner *sc, const char *format, va_list ap)
 | |
| {
 | |
|      const char *s = format;
 | |
|      char c;
 | |
|      int ch = 0;
 | |
|      int fmt_len;
 | |
| 
 | |
|      while ((c = *s++)) {
 | |
| 	  fmt_len = 0;
 | |
|           switch (c) {
 | |
| 	      case '%':
 | |
| 	  getformat:
 | |
| 		   switch ((c = *s++)) {
 | |
| 		       case 's': {
 | |
| 			    char *x = va_arg(ap, char *);
 | |
| 			    mygets(sc, x, fmt_len);
 | |
| 			    break;
 | |
| 		       }
 | |
| 		       case 'd': {
 | |
| 			    int *x = va_arg(ap, int *);
 | |
| 			    *x = (int) getlong(sc, 10, &ch);
 | |
| 			    if (!ch) return 0;
 | |
| 			    break;
 | |
| 		       }
 | |
| 		       case 'x': {
 | |
| 			    int *x = va_arg(ap, int *);
 | |
| 			    *x = (int) getlong(sc, 16, &ch);
 | |
| 			    if (!ch) return 0;
 | |
| 			    break;
 | |
| 		       }
 | |
| 		       case 'M': {
 | |
| 			    md5uint *x = va_arg(ap, md5uint *);
 | |
| 			    *x = (md5uint)
 | |
| 				    (0xFFFFFFFF & getlong(sc, 16, &ch));
 | |
| 			    if (!ch) return 0;
 | |
| 			    break;
 | |
| 		       }
 | |
| 		       case '*': {
 | |
| 			    if ((fmt_len = va_arg(ap, int)) <= 0) return 0;
 | |
| 			    goto getformat;
 | |
| 		       }
 | |
| 		       default:
 | |
| 			    A(0 /* unknown format */);
 | |
| 			    break;
 | |
| 		   }
 | |
| 		   break;
 | |
| 	      default:
 | |
| 		   if (isspace(c) || c == '(' || c == ')')
 | |
| 			eat_blanks(sc);
 | |
| 		   if (!isspace(c) && (ch = GETCHR(sc)) != c) {
 | |
| 			UNGETCHR(sc, ch);
 | |
| 			return 0;
 | |
| 		   }
 | |
| 		   break;
 | |
|           }
 | |
|      }
 | |
|      return 1;
 | |
| }
 | |
| 
 | |
| static int scan(scanner *sc, const char *format, ...)
 | |
| {
 | |
|      int ret;
 | |
|      va_list ap;
 | |
|      va_start(ap, format);
 | |
|      ret = vscan(sc, format, ap);
 | |
|      va_end(ap);
 | |
|      return ret;
 | |
| }
 | |
| 
 | |
| scanner *X(mkscanner)(size_t size, int (*getchr)(scanner *sc))
 | |
| {
 | |
|      scanner *s = (scanner *)MALLOC(size, OTHER);
 | |
|      s->scan = scan;
 | |
|      s->vscan = vscan;
 | |
|      s->getchr = getchr;
 | |
|      s->ungotc = EOF;
 | |
|      return s;
 | |
| }
 | |
| 
 | |
| void X(scanner_destroy)(scanner *sc)
 | |
| {
 | |
|      X(ifree)(sc);
 | |
| }
 | 
