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);
							 | 
						||
| 
								 | 
							
								}
							 |