402 lines
12 KiB
C
402 lines
12 KiB
C
#include "../stdio_p.h"
|
|
#include "../../stdlib/stdlib_p.h"
|
|
#include <stdio.h>
|
|
#include <ctype.h>
|
|
#include <string.h>
|
|
#include <stdbool.h>
|
|
|
|
void __scanf_start(struct __scanf_input *in)
|
|
{
|
|
if(in->fp)
|
|
in->buffer = fgetc(in->fp);
|
|
else {
|
|
in->buffer = (*in->str ? *in->str : EOF);
|
|
in->str += (in->buffer != EOF);
|
|
}
|
|
}
|
|
|
|
int __scanf_fetch(struct __scanf_input *in)
|
|
{
|
|
if(in->fp)
|
|
return fgetc(in->fp);
|
|
|
|
int c = *in->str;
|
|
if(c == 0)
|
|
return EOF;
|
|
in->str++;
|
|
return c;
|
|
}
|
|
|
|
void __scanf_end(struct __scanf_input *in)
|
|
{
|
|
if(in->buffer == EOF)
|
|
return;
|
|
|
|
if(in->fp)
|
|
ungetc(in->buffer, in->fp);
|
|
else
|
|
in->str--;
|
|
}
|
|
|
|
void __purge_space( struct __scanf_input * __restrict__ in )
|
|
{
|
|
while (isspace(__scanf_peek(in))) __scanf_in(in);
|
|
}
|
|
|
|
void __scanf_store_i(int64_t value, int size, va_list *args)
|
|
{
|
|
if(size == 1)
|
|
*va_arg(*args, int8_t *) = value;
|
|
else if(size == 2)
|
|
*va_arg(*args, int16_t *) = value;
|
|
else if(size == 4)
|
|
*va_arg(*args, int32_t *) = value;
|
|
else if(size == 8)
|
|
*va_arg(*args, int64_t *) = value;
|
|
}
|
|
|
|
void __scanf_store_d(long double value, int size, va_list *args)
|
|
{
|
|
if(size == sizeof(float))
|
|
*va_arg(*args, float *) = value;
|
|
else if(size == sizeof(double))
|
|
*va_arg(*args, double *) = value;
|
|
else if(size == sizeof(long double))
|
|
*va_arg(*args, long double *) = value;
|
|
}
|
|
|
|
/* STATUS OF __scanf DEVELOPMENT */
|
|
|
|
// XX = not done yet (but will be done)
|
|
// OK = OK, done and tested
|
|
// -- = not applicable
|
|
// NO = not supported (and will not be done) only for %lc as long char are not supported by gint
|
|
|
|
|
|
/*************************************************************************************************************/
|
|
/* Specifier * Explanation * num * hh * h *none* l * ll * j * z * t * L */
|
|
/*************************************************************************************************************/
|
|
/* % * Parse literal '%' * -- * -- * -- * OK * -- * -- * -- * -- * -- * -- */
|
|
/*************************************************************************************************************/
|
|
/* c * match a char or several char * OK * -- * -- * OK * NO * -- * -- * -- * -- * -- */
|
|
/*************************************************************************************************************/
|
|
/* s * match a string * OK * -- * -- * OK * NO * -- * -- * -- * -- * -- */
|
|
/*************************************************************************************************************/
|
|
/* [set] * match a set of char * OK * -- * -- * OK * -- * -- * -- * -- * -- * -- */
|
|
/*************************************************************************************************************/
|
|
/* d * match a decimal integer * OK * OK * OK * OK * OK * OK * OK * OK * OK * -- */
|
|
/*************************************************************************************************************/
|
|
/* i * match an integer * OK * OK * OK * OK * OK * OK * OK * OK * OK * -- */
|
|
/*************************************************************************************************************/
|
|
/* d * match an unsigned decimal integer * OK * OK * OK * OK * OK * OK * OK * OK * OK * -- */
|
|
/*************************************************************************************************************/
|
|
/* o * match a unsigned octal integer * OK * OK * OK * OK * OK * OK * OK * OK * OK * -- */
|
|
/*************************************************************************************************************/
|
|
/* x,X * match a unsigned hexadecimal integer * OK * OK * OK * OK * OK * OK * OK * OK * OK * -- */
|
|
/*************************************************************************************************************/
|
|
/* n * return the nb of chars read so far * -- * -- * -- * OK * -- * -- * -- * -- * -- * -- */
|
|
/*************************************************************************************************************/
|
|
/* a,A * match a floating point number * OK * -- * -- * OK * OK * -- * -- * -- * -- * OK */
|
|
/* e,E * match a floating point number * OK * -- * -- * OK * OK * -- * -- * -- * -- * OK */
|
|
/* f,F * match a floating point number * OK * -- * -- * OK * OK * -- * -- * -- * -- * OK */
|
|
/* g,G * match a floating point number * OK * -- * -- * OK * OK * -- * -- * -- * -- * OK */
|
|
/*************************************************************************************************************/
|
|
/* p * match a pointer * -- * -- * -- * OK * -- * -- * -- * -- * -- * -- */
|
|
/*************************************************************************************************************/
|
|
|
|
// %ms and %m[set] are not implemented (with memory allocation while parsing a chain or a set of characters)
|
|
|
|
struct scanf_format {
|
|
/* Maximum field width */
|
|
int field_width;
|
|
/* Size of the assigned (integer or floating-point) type, in bytes */
|
|
uint8_t size;
|
|
/* Whether to skip assignment */
|
|
bool skip;
|
|
|
|
/* Set of bytes allowed for bracket sets in %[] */
|
|
uint8_t bracket_set[32];
|
|
};
|
|
|
|
/* Allow/disallow the entire set */
|
|
static void bracket_set_init(uint8_t *set, bool allow)
|
|
{
|
|
memset(set, allow ? 0xff : 0x00, 32);
|
|
}
|
|
|
|
/* Allow/disallow a range of characters. Both ends are included. */
|
|
static void bracket_set_range(
|
|
uint8_t *set, uint8_t start, uint8_t end, bool allow)
|
|
{
|
|
for(int u = start; u <= end; u++) {
|
|
int byte = u >> 3;
|
|
int bit = 1 << (u & 7);
|
|
if(allow)
|
|
set[byte] |= bit;
|
|
else
|
|
set[byte] &= ~bit;
|
|
}
|
|
}
|
|
|
|
/* Check whether a byte is allowed by the bracket set. */
|
|
static bool bracket_set_test(uint8_t *set, int c)
|
|
{
|
|
return (c != EOF) && (set[c >> 3] & (1 << (c & 7)));
|
|
}
|
|
|
|
/* Parse a bracket set from a format string. Returns true on success. */
|
|
static bool bracket_set_parse(uint8_t *set, char const *format, int *pos)
|
|
{
|
|
int last = 0;
|
|
bool allow = true;
|
|
bracket_set_init(set, false);
|
|
|
|
/* '^' denotes a negated set */
|
|
if(format[*pos] == '^') {
|
|
allow = false;
|
|
(*pos)++;
|
|
bracket_set_init(set, true);
|
|
}
|
|
/* ']' as the first character adds ']' to the set */
|
|
if(format[*pos] == ']' ) {
|
|
bracket_set_range(set, ']', ']', allow);
|
|
(*pos)++;
|
|
}
|
|
|
|
for(; format[*pos] && format[*pos] != ']'; (*pos)++) {
|
|
/* '-' as the last character, thus explicitly in the set */
|
|
if(format[*pos] == '-' && format[*pos + 1] == ']')
|
|
bracket_set_range(set, '-', '-', allow);
|
|
/* '-' as denoting a range */
|
|
else if(format[*pos] == '-') {
|
|
(*pos)++;
|
|
bracket_set_range(set, last, format[*pos], allow);
|
|
}
|
|
/* Any other character */
|
|
else {
|
|
last = format[*pos];
|
|
bracket_set_range(set, last, last, allow);
|
|
}
|
|
}
|
|
|
|
return (format[*pos] == ']');
|
|
}
|
|
|
|
static int parse_fmt(char const *fmt, int *pos, struct scanf_format *opt)
|
|
{
|
|
opt->field_width = INT_MAX;
|
|
opt->size = sizeof(int);
|
|
opt->skip = false;
|
|
|
|
int width = 0;
|
|
char size_letter = 0;
|
|
|
|
for((*pos)++;; (*pos)++) {
|
|
int c = fmt[*pos];
|
|
|
|
if(c == '*')
|
|
opt->skip = true;
|
|
else if(strchr("hlzjtL", c)) {
|
|
if(c == 'h')
|
|
opt->size = (size_letter=='h') ? sizeof(char) : sizeof(short);
|
|
else if(c == 'l')
|
|
opt->size = (size_letter=='l') ? sizeof(long long) :
|
|
sizeof(long);
|
|
else if(c == 'z')
|
|
opt->size = sizeof(size_t);
|
|
else if(c == 'j')
|
|
opt->size = sizeof(intmax_t);
|
|
else if(c == 't')
|
|
opt->size = sizeof(ptrdiff_t);
|
|
else if(c == 'L')
|
|
opt->size = sizeof(long double);
|
|
|
|
size_letter = c;
|
|
}
|
|
else if(isdigit(c)) {
|
|
width = width * 10 + (fmt[*pos] - '0');
|
|
opt->field_width = width;
|
|
}
|
|
else if(c == '[') {
|
|
(*pos)++;
|
|
return bracket_set_parse(opt->bracket_set, fmt, pos) ? '[' : 0;
|
|
}
|
|
else if(strchr("diouxXpsn", c))
|
|
return c;
|
|
else if(strchr("aAeEfFgG", c)) {
|
|
/* Adjust interpretation of no size / 'l' size */
|
|
if(size_letter == 0)
|
|
opt->size = sizeof(float);
|
|
if(size_letter == 'l')
|
|
opt->size = sizeof(double);
|
|
return c;
|
|
}
|
|
else if(c == 'c') {
|
|
if(opt->field_width == INT_MAX)
|
|
opt->field_width = 1;
|
|
return c;
|
|
}
|
|
else return 0;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int __scanf(
|
|
struct __scanf_input * __restrict__ in,
|
|
char const * __restrict__ format,
|
|
va_list *args)
|
|
{
|
|
/* Number of successful assignments */
|
|
int validrets = 0;
|
|
|
|
__scanf_start( in );
|
|
|
|
// TODO: No __scanf_end() in any of the "return validrets"!!
|
|
|
|
for(int pos = 0; format[pos]; pos++) {
|
|
if(format[pos] == ' ') {
|
|
__purge_space(in);
|
|
continue;
|
|
}
|
|
else if(format[pos] != '%' || format[pos + 1] == '%') {
|
|
/* Expect this specific character */
|
|
if(__scanf_peek(in) != format[pos])
|
|
return validrets;
|
|
__scanf_in(in);
|
|
pos += (format[pos] == '%');
|
|
continue;
|
|
}
|
|
|
|
/* Perform a conversion */
|
|
struct scanf_format opt;
|
|
int spec = parse_fmt(format, &pos, &opt);
|
|
if(spec == 0)
|
|
return validrets;
|
|
|
|
switch(spec) {
|
|
case '[': {
|
|
int currentlength = 0;
|
|
// we need to assign the read char to the corresponding pointer
|
|
char *c = opt.skip ? NULL : va_arg(*args, char *);
|
|
for(int u=0; u<opt.field_width; u++) {
|
|
int temp = __scanf_peek(in);
|
|
if(bracket_set_test(opt.bracket_set, temp)) {
|
|
__scanf_in(in);
|
|
if(c) *c++ = temp;
|
|
currentlength++;
|
|
}
|
|
else if(temp==EOF && !currentlength && !validrets)
|
|
return EOF;
|
|
else break;
|
|
}
|
|
if(!currentlength)
|
|
return validrets;
|
|
*c = '\0';
|
|
validrets += !opt.skip;
|
|
break;
|
|
}
|
|
|
|
case 'n':
|
|
*va_arg(*args, int *) = in->bytes_read;
|
|
break;
|
|
|
|
case 'd':
|
|
case 'i':
|
|
case 'o':
|
|
case 'u':
|
|
case 'x':
|
|
case 'X': {
|
|
int f = format[pos];
|
|
int base = (f == 'd' || f == 'u') ? 10 :
|
|
(f == 'o') ? 8:
|
|
(f == 'x' || f == 'X') ? 16 : 0;
|
|
bool use_unsigned = (f == 'o' || f == 'x' || f == 'X');
|
|
|
|
long long int temp;
|
|
int err = __strto_int(in, base, NULL, &temp, use_unsigned,
|
|
opt.field_width);
|
|
if(err == EOF && validrets == 0) return EOF;
|
|
if(err) return validrets;
|
|
if(!opt.skip)
|
|
__scanf_store_i(temp, opt.size, args);
|
|
validrets++;
|
|
break;
|
|
}
|
|
|
|
case 'a':
|
|
case 'A':
|
|
case 'e':
|
|
case 'E':
|
|
case 'f':
|
|
case 'F':
|
|
case 'g':
|
|
case 'G': {
|
|
// read a double from the current input stream
|
|
// and store in the corresponding arg as a char by reference
|
|
long double temp;
|
|
int err = __strto_fp(in, NULL, NULL, &temp, opt.field_width);
|
|
if(err == EOF && validrets == 0) return EOF;
|
|
if(err) return validrets;
|
|
if(!opt.skip)
|
|
__scanf_store_d(temp, opt.size, args);
|
|
validrets++;
|
|
break;
|
|
}
|
|
|
|
case 'p': {
|
|
long int temp;
|
|
int err = 0;
|
|
if(!opt.skip) {
|
|
void *p = va_arg(*args, void *);
|
|
err = __strto_int(in, 0, p, NULL, true, opt.field_width);
|
|
validrets += (err == 0);
|
|
}
|
|
else err = __strto_int(in, 0, &temp, NULL, true, opt.field_width);
|
|
if(err) return validrets;
|
|
break;
|
|
}
|
|
|
|
case 'c': {
|
|
char *c = opt.skip ? NULL : va_arg(*args, char *);
|
|
|
|
for(int u = 0; u < opt.field_width; u++) {
|
|
int temp = __scanf_in(in);
|
|
if(temp==EOF) return EOF;
|
|
else if(c) *c++ = temp;
|
|
}
|
|
validrets += !opt.skip;
|
|
break;
|
|
}
|
|
|
|
case 's': {
|
|
char temp;
|
|
int curstrlength = 0;
|
|
__purge_space(in);
|
|
|
|
char *c = opt.skip ? NULL : va_arg(*args, char *);
|
|
for(int u = 0; u < opt.field_width; u++) {
|
|
temp = __scanf_peek(in);
|
|
if(temp==EOF && curstrlength==0) return validrets;
|
|
if(isspace(temp) || ((temp==EOF && curstrlength!=0))) {
|
|
if(c) {
|
|
*c = 0;
|
|
validrets++;
|
|
}
|
|
break;
|
|
}
|
|
else {
|
|
int temp = __scanf_in(in);
|
|
if(c)
|
|
*c++ = temp;
|
|
curstrlength++;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
__scanf_end( in );
|
|
return validrets;
|
|
}
|