forked from Bananymous/banan-os
LibC: Implement almost POSIX compliant *scanf functions
Only wchar strings and floating point values are not parsed
This commit is contained in:
parent
de4fdcd898
commit
606a7cb313
|
|
@ -11,6 +11,7 @@ set(LIBC_SOURCES
|
|||
malloc.cpp
|
||||
printf_impl.cpp
|
||||
pwd.cpp
|
||||
scanf_impl.cpp
|
||||
signal.cpp
|
||||
stdio.cpp
|
||||
stdlib.cpp
|
||||
|
|
|
|||
|
|
@ -0,0 +1,5 @@
|
|||
#pragma once
|
||||
|
||||
#include <stdarg.h>
|
||||
|
||||
int scanf_impl(const char* format, va_list arguments, int (*getc_fun)(void*), void* data);
|
||||
|
|
@ -0,0 +1,473 @@
|
|||
#include <BAN/Assert.h>
|
||||
#include <BAN/Math.h>
|
||||
#include <BAN/Traits.h>
|
||||
|
||||
#include <ctype.h>
|
||||
#include <scanf_impl.h>
|
||||
#include <stddef.h>
|
||||
#include <stdint.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
enum class LengthModifier
|
||||
{
|
||||
none,
|
||||
hh,
|
||||
h,
|
||||
l,
|
||||
ll,
|
||||
j,
|
||||
z,
|
||||
t,
|
||||
L,
|
||||
};
|
||||
|
||||
struct Conversion
|
||||
{
|
||||
bool suppress = false;
|
||||
int field_width = -1;
|
||||
bool allocate = false;
|
||||
LengthModifier length = LengthModifier::none;
|
||||
char conversion = '\0';
|
||||
};
|
||||
|
||||
Conversion parse_conversion_specifier(const char*& format)
|
||||
{
|
||||
Conversion conversion;
|
||||
|
||||
if (*format == '*')
|
||||
{
|
||||
conversion.suppress = true;
|
||||
format++;
|
||||
}
|
||||
|
||||
if (isdigit(*format))
|
||||
{
|
||||
conversion.field_width = 0;
|
||||
while (isdigit(*format))
|
||||
{
|
||||
conversion.field_width = (conversion.field_width * 10) + (*format - '0');
|
||||
format++;
|
||||
}
|
||||
}
|
||||
|
||||
if (*format == 'm')
|
||||
{
|
||||
conversion.allocate = true;
|
||||
format++;
|
||||
}
|
||||
|
||||
if (*format == 'h')
|
||||
{
|
||||
conversion.length = LengthModifier::h;
|
||||
format++;
|
||||
if (*format == 'h')
|
||||
{
|
||||
conversion.length = LengthModifier::hh;
|
||||
format++;
|
||||
}
|
||||
}
|
||||
else if (*format == 'l')
|
||||
{
|
||||
conversion.length = LengthModifier::l;
|
||||
format++;
|
||||
if (*format == 'l')
|
||||
{
|
||||
conversion.length = LengthModifier::ll;
|
||||
format++;
|
||||
}
|
||||
}
|
||||
else if (*format == 'j')
|
||||
{
|
||||
conversion.length = LengthModifier::j;
|
||||
format++;
|
||||
}
|
||||
else if (*format == 'z')
|
||||
{
|
||||
conversion.length = LengthModifier::z;
|
||||
format++;
|
||||
}
|
||||
else if (*format == 't')
|
||||
{
|
||||
conversion.length = LengthModifier::t;
|
||||
format++;
|
||||
}
|
||||
else if (*format == 'L')
|
||||
{
|
||||
conversion.length = LengthModifier::L;
|
||||
format++;
|
||||
}
|
||||
|
||||
conversion.conversion = *format;
|
||||
format++;
|
||||
|
||||
return conversion;
|
||||
}
|
||||
|
||||
template<int BASE>
|
||||
using BASE_TYPE = BAN::integral_constant<int, BASE>;
|
||||
|
||||
template<bool UNSIGNED>
|
||||
using IS_UNSIGNED = BAN::integral_constant<bool, UNSIGNED>;
|
||||
|
||||
int scanf_impl(const char* format, va_list arguments, int (*__getc_fun)(void*), void* data)
|
||||
{
|
||||
static constexpr int DONE = -1;
|
||||
static constexpr int NONE = -2;
|
||||
|
||||
int nread = 0;
|
||||
int conversions = 0;
|
||||
int in = NONE;
|
||||
|
||||
enum class ConversionResult
|
||||
{
|
||||
NONE,
|
||||
SUCCESS,
|
||||
INPUT_FAILURE,
|
||||
MATCH_FAILURE,
|
||||
};
|
||||
|
||||
auto get_input =
|
||||
[&](bool advance) -> void
|
||||
{
|
||||
if (in == DONE)
|
||||
return;
|
||||
if (advance || in == NONE)
|
||||
{
|
||||
in = __getc_fun(data);
|
||||
nread++;
|
||||
}
|
||||
};
|
||||
|
||||
auto parse_integer_internal =
|
||||
[&get_input, &in]<int BASE, typename T>(BASE_TYPE<BASE>, bool negative, int width, T* out) -> ConversionResult
|
||||
{
|
||||
constexpr auto is_base_digit =
|
||||
[](char c) -> bool
|
||||
{
|
||||
c = tolower(c);
|
||||
if ('0' <= c && c <= '9')
|
||||
return c - '0' < BASE;
|
||||
if ('a' <= c && c <= 'z')
|
||||
return c - 'a' + 10 < BASE;
|
||||
return false;
|
||||
};
|
||||
constexpr auto get_base_digit = [](char c) -> T { if (c <= '9') return c - '0'; return tolower(c) - 'a' + 10; };
|
||||
|
||||
if (!is_base_digit(in))
|
||||
return ConversionResult::MATCH_FAILURE;
|
||||
|
||||
*out = 0;
|
||||
while (width-- && is_base_digit(in))
|
||||
{
|
||||
*out = (*out * BASE) + get_base_digit(in);
|
||||
get_input(true);
|
||||
}
|
||||
|
||||
if (negative)
|
||||
*out = -*out;
|
||||
|
||||
return ConversionResult::SUCCESS;
|
||||
};
|
||||
|
||||
auto parse_integer_typed =
|
||||
[&parse_integer_internal, &arguments, &get_input, &in]<int BASE, typename T>(BASE_TYPE<BASE>, bool suppress, int width, T*) -> ConversionResult
|
||||
{
|
||||
T dummy;
|
||||
T* out = suppress ? &dummy : va_arg(arguments, T*);
|
||||
|
||||
bool negative = (in == '-');
|
||||
if (in == '-' || in == '+')
|
||||
{
|
||||
get_input(true);
|
||||
if (--width == 0)
|
||||
return ConversionResult::MATCH_FAILURE;
|
||||
}
|
||||
|
||||
if constexpr(BASE == 0)
|
||||
{
|
||||
if (in != '0')
|
||||
return parse_integer_internal(BASE_TYPE<10>{}, negative, width, out);
|
||||
else
|
||||
{
|
||||
get_input(true);
|
||||
if (--width == 0)
|
||||
{
|
||||
*out = 0;
|
||||
return ConversionResult::SUCCESS;
|
||||
}
|
||||
if ('0' <= in && in <= '7')
|
||||
return parse_integer_internal(BASE_TYPE<8>{}, negative, width, out);
|
||||
if (tolower(in) == 'x')
|
||||
{
|
||||
get_input(true);
|
||||
if (--width == 0)
|
||||
return ConversionResult::MATCH_FAILURE;
|
||||
return parse_integer_internal(BASE_TYPE<16>{}, negative, width, out);
|
||||
}
|
||||
*out = 0;
|
||||
return ConversionResult::SUCCESS;
|
||||
}
|
||||
}
|
||||
|
||||
if constexpr(BASE == 16)
|
||||
{
|
||||
if (in == '0')
|
||||
{
|
||||
get_input(true);
|
||||
width--;
|
||||
if (tolower(in) == 'x')
|
||||
{
|
||||
get_input(true);
|
||||
width--;
|
||||
}
|
||||
if (width <= 0)
|
||||
{
|
||||
*out = 0;
|
||||
return ConversionResult::SUCCESS;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return parse_integer_internal(BASE_TYPE<BASE>{}, negative, width, out);
|
||||
};
|
||||
|
||||
auto parse_integer =
|
||||
[&parse_integer_typed, &get_input, &in]<int BASE, bool UNSIGNED>(BASE_TYPE<BASE>, IS_UNSIGNED<UNSIGNED>, bool suppress, int width, LengthModifier length) -> ConversionResult
|
||||
{
|
||||
get_input(false);
|
||||
while (isspace(in))
|
||||
get_input(true);
|
||||
if (in == DONE)
|
||||
return ConversionResult::INPUT_FAILURE;
|
||||
|
||||
if (width == -1)
|
||||
width = __INT_MAX__;
|
||||
|
||||
switch (length)
|
||||
{
|
||||
case LengthModifier::none: return parse_integer_typed(BASE_TYPE<BASE>{}, suppress, width, BAN::either_or_t<UNSIGNED, unsigned int*, int*> {});
|
||||
case LengthModifier::hh: return parse_integer_typed(BASE_TYPE<BASE>{}, suppress, width, BAN::either_or_t<UNSIGNED, unsigned char*, signed char*> {});
|
||||
case LengthModifier::h: return parse_integer_typed(BASE_TYPE<BASE>{}, suppress, width, BAN::either_or_t<UNSIGNED, unsigned short*, short*> {});
|
||||
case LengthModifier::l: return parse_integer_typed(BASE_TYPE<BASE>{}, suppress, width, BAN::either_or_t<UNSIGNED, unsigned long*, long*> {});
|
||||
case LengthModifier::ll: return parse_integer_typed(BASE_TYPE<BASE>{}, suppress, width, BAN::either_or_t<UNSIGNED, unsigned long long*, long long*> {});
|
||||
case LengthModifier::j: return parse_integer_typed(BASE_TYPE<BASE>{}, suppress, width, BAN::either_or_t<UNSIGNED, uintmax_t*, intmax_t*> {});
|
||||
case LengthModifier::z: return parse_integer_typed(BASE_TYPE<BASE>{}, suppress, width, BAN::either_or_t<UNSIGNED, size_t*, ssize_t*> {});
|
||||
case LengthModifier::t: return parse_integer_typed(BASE_TYPE<BASE>{}, suppress, width, BAN::either_or_t<UNSIGNED, size_t*, ptrdiff_t*> {});
|
||||
static_assert(sizeof(size_t) == sizeof(ptrdiff_t));
|
||||
default:
|
||||
return ConversionResult::MATCH_FAILURE;
|
||||
}
|
||||
};
|
||||
|
||||
auto parse_string =
|
||||
[&arguments, &get_input, &in](uint8_t* mask, bool exclude, bool suppress, bool allocate, int min_len, int max_len, bool terminate) -> ConversionResult
|
||||
{
|
||||
char* temp_dummy;
|
||||
char** outp = nullptr;
|
||||
if (suppress)
|
||||
;
|
||||
else if (allocate)
|
||||
{
|
||||
outp = va_arg(arguments, char**);
|
||||
*outp = (char*)malloc(BUFSIZ);
|
||||
if (*outp == nullptr)
|
||||
return ConversionResult::MATCH_FAILURE;
|
||||
}
|
||||
else
|
||||
{
|
||||
temp_dummy = va_arg(arguments, char*);
|
||||
outp = &temp_dummy;
|
||||
}
|
||||
|
||||
const uint8_t xor_mask = exclude ? 0xFF : 0x00;
|
||||
|
||||
get_input(false);
|
||||
if (in == DONE)
|
||||
{
|
||||
if (allocate)
|
||||
free(*outp);
|
||||
*outp = nullptr;
|
||||
return ConversionResult::INPUT_FAILURE;
|
||||
}
|
||||
|
||||
int len = 0;
|
||||
while (len < max_len && in != DONE && ((mask[in / 8] ^ xor_mask) & (1 << (in % 8))))
|
||||
{
|
||||
len++;
|
||||
if (!suppress)
|
||||
{
|
||||
(*outp)[len - 1] = in;
|
||||
if (allocate && len % BUFSIZ == 0)
|
||||
{
|
||||
char* newp = (char*)realloc(*outp, len + BUFSIZ);
|
||||
if (newp == nullptr)
|
||||
{
|
||||
free(*outp);
|
||||
*outp = nullptr;
|
||||
return ConversionResult::MATCH_FAILURE;
|
||||
}
|
||||
*outp = newp;
|
||||
}
|
||||
}
|
||||
get_input(true);
|
||||
}
|
||||
if (len < min_len)
|
||||
{
|
||||
if (allocate)
|
||||
free(*outp);
|
||||
*outp = nullptr;
|
||||
return ConversionResult::MATCH_FAILURE;
|
||||
}
|
||||
|
||||
if (!suppress && terminate)
|
||||
(*outp)[len] = '\0';
|
||||
|
||||
return ConversionResult::SUCCESS;
|
||||
};
|
||||
|
||||
while (isspace(*format) || isprint(*format))
|
||||
{
|
||||
if (*format == '%')
|
||||
{
|
||||
format++;
|
||||
auto conversion = parse_conversion_specifier(format);
|
||||
|
||||
ConversionResult result = ConversionResult::NONE;
|
||||
switch (conversion.conversion)
|
||||
{
|
||||
case 'd': result = parse_integer(BASE_TYPE<10>{}, IS_UNSIGNED<false>{}, conversion.suppress, conversion.field_width, conversion.length); break;
|
||||
case 'i': result = parse_integer(BASE_TYPE<0> {}, IS_UNSIGNED<false>{}, conversion.suppress, conversion.field_width, conversion.length); break;
|
||||
case 'o': result = parse_integer(BASE_TYPE<8> {}, IS_UNSIGNED<true> {}, conversion.suppress, conversion.field_width, conversion.length); break;
|
||||
case 'u': result = parse_integer(BASE_TYPE<10>{}, IS_UNSIGNED<true> {}, conversion.suppress, conversion.field_width, conversion.length); break;
|
||||
case 'x': result = parse_integer(BASE_TYPE<16>{}, IS_UNSIGNED<true> {}, conversion.suppress, conversion.field_width, conversion.length); break;
|
||||
case 'X': result = parse_integer(BASE_TYPE<16>{}, IS_UNSIGNED<true> {}, conversion.suppress, conversion.field_width, conversion.length); break;
|
||||
case 'p': result = parse_integer(BASE_TYPE<16>{}, IS_UNSIGNED<true> {}, conversion.suppress, conversion.field_width, LengthModifier::j); break;
|
||||
case 'S':
|
||||
conversion.length = LengthModifier::l;
|
||||
// fall through
|
||||
case 's':
|
||||
{
|
||||
int width = conversion.field_width;
|
||||
if (width == -1)
|
||||
width = __INT_MAX__;
|
||||
uint8_t mask[0x100 / 8] {};
|
||||
for (int i = 0; i < 0x100; i++)
|
||||
if (isspace(i))
|
||||
mask[i / 8] |= 1 << (i % 8);
|
||||
get_input(false);
|
||||
while (isspace(in))
|
||||
get_input(true);
|
||||
result = parse_string(mask, true, conversion.suppress, conversion.allocate, 1, width, true);
|
||||
break;
|
||||
}
|
||||
case 'C':
|
||||
conversion.length = LengthModifier::l;
|
||||
// fall through
|
||||
case 'c':
|
||||
{
|
||||
int width = conversion.field_width;
|
||||
if (width == -1)
|
||||
width = 1;
|
||||
uint8_t mask[0x100 / 8] {};
|
||||
result = parse_string(mask, true, conversion.suppress, conversion.allocate, width, width, false);
|
||||
break;
|
||||
}
|
||||
case '[':
|
||||
{
|
||||
int width = conversion.field_width;
|
||||
if (width == -1)
|
||||
width = __INT_MAX__;
|
||||
|
||||
bool exclude = (*format == '^');
|
||||
if (exclude)
|
||||
format++;
|
||||
|
||||
uint8_t mask[0x100 / 8] {};
|
||||
if (*format == ']')
|
||||
{
|
||||
mask[']' / 8] |= 1 << (']' % 8);
|
||||
format++;
|
||||
}
|
||||
|
||||
bool first = true;
|
||||
while (*format && *format != ']')
|
||||
{
|
||||
if (!first && *format == '-' && *(format + 1) != ']')
|
||||
{
|
||||
int min = BAN::Math::min(*(format - 1), *(format + 1));
|
||||
int max = BAN::Math::max(*(format - 1), *(format + 1));
|
||||
for (int i = min; i <= max; i++)
|
||||
mask[i / 8] |= 1 << (i % 8);
|
||||
format += 2;
|
||||
}
|
||||
else
|
||||
{
|
||||
mask[*format / 8] |= 1 << (*format % 8);
|
||||
format++;
|
||||
}
|
||||
first = false;
|
||||
}
|
||||
|
||||
if (*format == ']')
|
||||
result = parse_string(mask, exclude, conversion.suppress, conversion.allocate, 1, width, true);
|
||||
else
|
||||
result = ConversionResult::MATCH_FAILURE;
|
||||
format++;
|
||||
break;
|
||||
}
|
||||
case 'n':
|
||||
if (!conversion.suppress)
|
||||
*va_arg(arguments, int*) = nread - (in != NONE);
|
||||
conversion.suppress = true; // Dont count this as conversion
|
||||
result = ConversionResult::SUCCESS;
|
||||
break;
|
||||
case '%':
|
||||
get_input(false);
|
||||
if (in == DONE)
|
||||
{
|
||||
result = ConversionResult::INPUT_FAILURE;
|
||||
break;
|
||||
}
|
||||
if (in != '%')
|
||||
{
|
||||
result = ConversionResult::MATCH_FAILURE;
|
||||
break;
|
||||
}
|
||||
get_input(true);
|
||||
result = ConversionResult::SUCCESS;
|
||||
break;
|
||||
default:
|
||||
result = ConversionResult::MATCH_FAILURE;
|
||||
break;
|
||||
}
|
||||
|
||||
ASSERT(result != ConversionResult::NONE);
|
||||
|
||||
if (result == ConversionResult::INPUT_FAILURE && conversions == 0)
|
||||
return EOF;
|
||||
if (result != ConversionResult::SUCCESS)
|
||||
return conversions;
|
||||
if (!conversion.suppress)
|
||||
conversions++;
|
||||
}
|
||||
else if (isspace(*format))
|
||||
{
|
||||
get_input(false);
|
||||
while (isspace(in))
|
||||
get_input(true);
|
||||
format++;
|
||||
}
|
||||
else
|
||||
{
|
||||
get_input(false);
|
||||
if (in != *format)
|
||||
break;
|
||||
get_input(true);
|
||||
format++;
|
||||
}
|
||||
}
|
||||
|
||||
return conversions;
|
||||
}
|
||||
|
|
@ -1,6 +1,7 @@
|
|||
#include <bits/printf.h>
|
||||
#include <errno.h>
|
||||
#include <fcntl.h>
|
||||
#include <scanf_impl.h>
|
||||
#include <stdarg.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
|
@ -438,8 +439,14 @@ int rename(const char*, const char*);
|
|||
// TODO
|
||||
void rewind(FILE*);
|
||||
|
||||
// TODO
|
||||
int scanf(const char*, ...);
|
||||
int scanf(const char* format, ...)
|
||||
{
|
||||
va_list arguments;
|
||||
va_start(arguments, format);
|
||||
int ret = vscanf(format, arguments);
|
||||
va_end(arguments);
|
||||
return ret;
|
||||
}
|
||||
|
||||
// TODO
|
||||
void setbuf(FILE*, char*);
|
||||
|
|
@ -465,8 +472,14 @@ int sprintf(char* buffer, const char* format, ...)
|
|||
return ret;
|
||||
}
|
||||
|
||||
// TODO
|
||||
int sscanf(const char*, const char*, ...);
|
||||
int sscanf(const char* s, const char* format, ...)
|
||||
{
|
||||
va_list arguments;
|
||||
va_start(arguments, format);
|
||||
int ret = vsscanf(s, format, arguments);
|
||||
va_end(arguments);
|
||||
return ret;
|
||||
}
|
||||
|
||||
// TODO
|
||||
char* tempnam(const char*, const char*);
|
||||
|
|
@ -485,16 +498,20 @@ int vfprintf(FILE* file, const char* format, va_list arguments)
|
|||
return printf_impl(format, arguments, [](int c, void* file) { return fputc(c, (FILE*)file); }, file);
|
||||
}
|
||||
|
||||
// TODO
|
||||
int vfscanf(FILE*, const char*, va_list);
|
||||
int vfscanf(FILE* file, const char* format, va_list arguments)
|
||||
{
|
||||
return scanf_impl(format, arguments, [](void* file) { return fgetc(static_cast<FILE*>(file)); }, file);
|
||||
}
|
||||
|
||||
int vprintf(const char* format, va_list arguments)
|
||||
{
|
||||
return vfprintf(stdout, format, arguments);
|
||||
}
|
||||
|
||||
// TODO
|
||||
int vscanf(const char*, va_list);
|
||||
int vscanf(const char* format, va_list arguments)
|
||||
{
|
||||
return vfscanf(stdin, format, arguments);
|
||||
}
|
||||
|
||||
int vsnprintf(char* buffer, size_t max_size, const char* format, va_list arguments)
|
||||
{
|
||||
|
|
@ -548,5 +565,16 @@ int vsprintf(char* buffer, const char* format, va_list arguments)
|
|||
return ret;
|
||||
}
|
||||
|
||||
// TODO
|
||||
int vsscanf(const char*, const char*, va_list);
|
||||
int vsscanf(const char* s, const char* format, va_list arguments)
|
||||
{
|
||||
return scanf_impl(format, arguments,
|
||||
[](void* data) -> int
|
||||
{
|
||||
char ret = **static_cast<char**>(data);
|
||||
(*static_cast<char**>(data))++;
|
||||
if (ret == '\0')
|
||||
return -1;
|
||||
return ret;
|
||||
}, &s
|
||||
);
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue