#include "prx/libc/include/FileStream.hpp" #include "prx/libc/include/general/VabiMacros.hpp" #include #include #include #include #include extern "C" { extern FileStream* __stdinp_nid_postfix; int APS5_VABI vfwscanf_nid_postfix(FileStream*, const char16_t*, void*); int APS5_VABI vfwscanf_s_nid_postfix(FileStream*, const char16_t*, void*); int APS5_VABI vwscanf_nid_postfix(const char16_t*, void*); int APS5_VABI vwscanf_s_nid_postfix(const char16_t*, void*); int APS5_VABI fwscanf_nid_postfix(FileStream*, const char16_t*, ...); int APS5_VABI fwscanf_s_nid_postfix(FileStream*, const char16_t*, ...); int APS5_VABI wscanf_nid_postfix(const char16_t*, ...); int APS5_VABI wscanf_s_nid_postfix(const char16_t*, ...); std::int32_t APS5_VABI fgetwc_nid_postfix(FileStream*); } #ifdef _WIN32 #define APS5_VA_BEGIN(last) __builtin_sysv_va_list args; __builtin_sysv_va_start(args, last) #define APS5_VA_END() __builtin_sysv_va_end(args) #else #define APS5_VA_BEGIN(last) std::va_list args; va_start(args, last) #define APS5_VA_END() va_end(args) #endif static void Require(bool value, const char* what) { if (value) return; std::fprintf(stderr, "failed: %s\n", what); std::abort(); } static int APS5_VABI ScanStream(FileStream* stream, const char16_t* format, ...) { APS5_VA_BEGIN(format); const int result = vfwscanf_nid_postfix(stream, format, args); APS5_VA_END(); return result; } static int APS5_VABI ScanStreamSized(FileStream* stream, const char16_t* format, ...) { APS5_VA_BEGIN(format); const int result = vfwscanf_s_nid_postfix(stream, format, args); APS5_VA_END(); return result; } static int APS5_VABI ScanInput(const char16_t* format, ...) { APS5_VA_BEGIN(format); const int result = vwscanf_nid_postfix(format, args); APS5_VA_END(); return result; } static int APS5_VABI ScanInputSized(const char16_t* format, ...) { APS5_VA_BEGIN(format); const int result = vwscanf_s_nid_postfix(format, args); APS5_VA_END(); return result; } static FileStream* Open(const char* text) { auto* stream = new FileStream(std::tmpfile()); Require(std::fputs(text, stream->GetHandle()) >= 0, "seed the stream"); std::rewind(stream->GetHandle()); return stream; } static void Close(FileStream* stream) { stream->Close(); delete stream; } static void CheckConsumption() { std::int32_t value = -1; std::int64_t count = -1; char16_t word[8]{}; FileStream* stream = Open(" 42 rest"); Require(ScanStream(stream, u"%d", &value) == 1 && value == 42, "an integer after leading space"); Require(fgetwc_nid_postfix(stream) == u' ', "the character after the field stays in the stream"); Require(fwscanf_nid_postfix(stream, u"%ls", word) == 1 && std::u16string(word) == u"rest", "a following fwscanf continues"); Require(fwscanf_nid_postfix(stream, u"%d", &value) == EOF, "end of file before any conversion"); Close(stream); stream = Open("0xg"); std::uint32_t hex = 7; Require(ScanStream(stream, u"%x", &hex) == 0 && hex == 7, "a hexadecimal prefix without digits is a matching failure"); Require(fgetwc_nid_postfix(stream) == u'g', "the prefix is consumed and the next character is pushed back"); Close(stream); stream = Open("12abc"); Require(ScanStream(stream, u"%d%ln", &value, &count) == 1 && value == 12 && count == 2, "%ln counts the consumed characters"); Require(fgetwc_nid_postfix(stream) == u'a', "the field stops before the first letter"); Close(stream); stream = Open("ab"); Require(ScanStream(stream, u"ac") == 0, "a literal mismatch"); Require(fgetwc_nid_postfix(stream) == u'b', "the mismatching character is pushed back"); Close(stream); stream = Open("x"); Require(ScanStream(stream, u"%d", &value) == 0, "a letter is not an integer"); Require(fgetwc_nid_postfix(stream) == u'x', "the rejected letter is pushed back"); Close(stream); stream = Open("-"); Require(ScanStream(stream, u"%d", &value) == 0, "a lone sign is a matching failure"); Require(fgetwc_nid_postfix(stream) == -1, "the sign is consumed"); Close(stream); stream = Open(" "); Require(ScanStream(stream, u"%d", &value) == EOF, "only white space"); Close(stream); stream = Open("2.5x"); double real = 0; Require(ScanStream(stream, u"%lf", &real) == 1 && real == 2.5, "a float"); Require(fgetwc_nid_postfix(stream) == u'x', "the character after the float"); Close(stream); } static void CheckWideText() { std::int32_t value = 0; char16_t letters[8]{}; FileStream* stream = Open("5 ab\xc3\xa9t\xc3\xa9"); Require(ScanStream(stream, u"%d %l[abc]", &value, letters) == 2 && value == 5 && std::u16string(letters) == u"ab", "a scanset stops at a non-ASCII character"); Require(fgetwc_nid_postfix(stream) == 0xe9, "a non-ASCII lookahead goes back as UTF-8"); Require(fwscanf_nid_postfix(stream, u"%ls", letters) == 1 && std::u16string(letters) == u"té", "the rest decodes"); Close(stream); stream = Open("hello world"); char16_t small[4]{}; Require(ScanStreamSized(stream, u"%3ls", small, 4u) == 1 && std::u16string(small) == u"hel", "vfwscanf_s %ls with a size"); Require(fwscanf_s_nid_postfix(stream, u"%ls", small, 4u) == 1 && std::u16string(small) == u"lo", "fwscanf_s %ls that fits"); Require(fwscanf_s_nid_postfix(stream, u"%ls", small, 4u) == 0 && small[0] == 0, "fwscanf_s rejects a too small buffer"); Close(stream); Require(fwscanf_nid_postfix(nullptr, u"%d", &value) == EOF, "null stream"); Require(vfwscanf_s_nid_postfix(nullptr, u"%d", nullptr) == EOF, "null stream with sizes"); } static void CheckStandardInput() { FileStream* const standardInput = __stdinp_nid_postfix; FileStream* stream = Open("-5 12 ab cd"); __stdinp_nid_postfix = stream; std::int32_t first = 0, second = 0; char16_t word[4]{}; Require(ScanInput(u"%d", &first) == 1 && first == -5, "vwscanf reads the guest standard input"); Require(wscanf_nid_postfix(u"%d", &second) == 1 && second == 12, "wscanf reads the guest standard input"); Require(ScanInputSized(u"%ls", word, 4u) == 1 && std::u16string(word) == u"ab", "vwscanf_s"); Require(wscanf_s_nid_postfix(u"%ls", word, 4u) == 1 && std::u16string(word) == u"cd", "wscanf_s"); __stdinp_nid_postfix = standardInput; Close(stream); } int main() { CheckConsumption(); CheckWideText(); CheckStandardInput(); return 0; }