mirror of
https://gitlab.com/sortix/sortix.git
synced 2023-02-13 20:55:38 -05:00
527 lines
14 KiB
C++
527 lines
14 KiB
C++
/*******************************************************************************
|
|
|
|
Copyright(C) Jonas 'Sortie' Termansen 2011, 2012.
|
|
|
|
This file is part of the Sortix C Library.
|
|
|
|
The Sortix C Library is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or (at your
|
|
option) any later version.
|
|
|
|
The Sortix C Library 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 Lesser General Public
|
|
License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with the Sortix C Library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
stdio/format.cpp
|
|
Provides printf formatting functions that uses callbacks.
|
|
|
|
*******************************************************************************/
|
|
|
|
// Number of bugs seemingly unrelated bugs that have been traced to here:
|
|
// Countless + 2
|
|
|
|
#include <errno.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
namespace String {
|
|
|
|
static int ConvertInt32(int32_t num, char* dest, char possign)
|
|
{
|
|
int result = 0;
|
|
int32_t copy = num;
|
|
|
|
if ( num < 0 )
|
|
{
|
|
*dest++ = '-';
|
|
result++;
|
|
|
|
int offset = 0;
|
|
while ( copy < -9 ) { copy /= 10; offset++; }
|
|
result += offset;
|
|
while ( offset >= 0 )
|
|
{
|
|
dest[offset] = '0' - num % 10; num /= 10; offset--;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if ( possign )
|
|
*dest++ = possign,
|
|
result++;
|
|
|
|
int offset = 0;
|
|
while ( copy > 9 ) { copy /= 10; offset++; }
|
|
result += offset;
|
|
while ( offset >= 0 )
|
|
{
|
|
dest[offset] = '0' + num % 10; num /= 10; offset--;
|
|
}
|
|
}
|
|
|
|
return result + 1;
|
|
}
|
|
|
|
static int ConvertInt64(int64_t num, char* dest, char possign)
|
|
{
|
|
int result = 0;
|
|
int64_t copy = num;
|
|
|
|
if ( num < 0 )
|
|
{
|
|
*dest++ = '-';
|
|
result++;
|
|
|
|
int offset = 0;
|
|
while ( copy < -9 ) { copy /= 10; offset++; }
|
|
result += offset;
|
|
while ( offset >= 0 )
|
|
{
|
|
dest[offset] = '0' - num % 10; num /= 10; offset--;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if ( possign )
|
|
*dest++ = possign,
|
|
result++;
|
|
|
|
int offset = 0;
|
|
while ( copy > 9 ) { copy /= 10; offset++; }
|
|
result += offset;
|
|
while ( offset >= 0 )
|
|
{
|
|
dest[offset] = '0' + num % 10; num /= 10; offset--;
|
|
}
|
|
}
|
|
|
|
return result + 1;
|
|
}
|
|
|
|
static int ConvertUInt32(uint32_t num, char* dest)
|
|
{
|
|
int result = 0;
|
|
uint32_t copy = num;
|
|
int offset = 0;
|
|
while ( copy > 9 ) { copy /= 10; offset++; }
|
|
result += offset;
|
|
while ( offset >= 0 )
|
|
{
|
|
dest[offset] = '0' + num % 10; num /= 10; offset--;
|
|
}
|
|
|
|
return result + 1;
|
|
}
|
|
|
|
static int ConvertUInt64(uint64_t num, char* dest)
|
|
{
|
|
int result = 0;
|
|
uint64_t copy = num;
|
|
int offset = 0;
|
|
while ( copy > 9 ) { copy /= 10; offset++; }
|
|
result += offset;
|
|
while ( offset >= 0 )
|
|
{
|
|
dest[offset] = '0' + num % 10; num /= 10; offset--;
|
|
}
|
|
|
|
return result + 1;
|
|
}
|
|
|
|
static int ConvertUInt32Octal(uint32_t num, char* dest)
|
|
{
|
|
int result = 0;
|
|
uint32_t copy = num;
|
|
int offset = 0;
|
|
while ( copy > 7 ) { copy /= 8; offset++; }
|
|
result += offset;
|
|
while ( offset >= 0 )
|
|
{
|
|
dest[offset] = '0' + num % 8; num /= 8; offset--;
|
|
}
|
|
|
|
return result + 1;
|
|
}
|
|
|
|
static int ConvertUInt64Octal(uint64_t num, char* dest)
|
|
{
|
|
int result = 0;
|
|
uint64_t copy = num;
|
|
int offset = 0;
|
|
while ( copy > 7 ) { copy /= 8; offset++; }
|
|
result += offset;
|
|
while ( offset >= 0 )
|
|
{
|
|
dest[offset] = '0' + num % 8; num /= 8; offset--;
|
|
}
|
|
|
|
return result + 1;
|
|
}
|
|
|
|
static int ConvertUInt32Hex(uint32_t num, char* dest)
|
|
{
|
|
char chars[16] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
|
|
'A', 'B', 'C', 'D', 'E', 'F' };
|
|
int result = 0;
|
|
uint32_t copy = num;
|
|
int offset = 0;
|
|
while ( copy > 15 ) { copy /= 16; offset++; }
|
|
result += offset;
|
|
while ( offset >= 0 )
|
|
{
|
|
dest[offset] = chars[num % 16]; num /= 16; offset--;
|
|
}
|
|
|
|
return result + 1;
|
|
}
|
|
|
|
static int ConvertUInt64Hex(uint64_t num, char* dest)
|
|
{
|
|
char chars[16] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
|
|
'A', 'B', 'C', 'D', 'E', 'F' };
|
|
int result = 0;
|
|
uint64_t copy = num;
|
|
int offset = 0;
|
|
while ( copy > 15 ) { copy /= 16; offset++; }
|
|
result += offset;
|
|
while ( offset >= 0 )
|
|
{
|
|
dest[offset] = chars[num % 16]; num /= 16; offset--;
|
|
}
|
|
|
|
return result + 1;
|
|
}
|
|
|
|
} // namespace String
|
|
|
|
#define READY_SIZE 128
|
|
|
|
#define READY_FLUSH() \
|
|
ready[readylen] = '\0'; \
|
|
if ( 0 < readylen && callback && callback(user, ready, readylen) != readylen ) { return SIZE_MAX; } \
|
|
written += readylen; readylen = 0;
|
|
|
|
#define REPEAT_BLANKS(num) \
|
|
for ( unsigned int i = 0; i < (num); i++ ) \
|
|
{ \
|
|
if ( readylen == READY_SIZE ) { READY_FLUSH(); } \
|
|
ready[readylen++ + i - i] = blank_char; \
|
|
}
|
|
|
|
extern "C" size_t vprintf_callback(size_t (*callback)(void*, const char*, size_t),
|
|
void* user,
|
|
const char* restrict format,
|
|
va_list parameters)
|
|
{
|
|
size_t written = 0;
|
|
size_t readylen = 0;
|
|
char ready[READY_SIZE + 1];
|
|
|
|
while ( *format != '\0' )
|
|
{
|
|
char c = *format++;
|
|
|
|
if ( c != '%' )
|
|
{
|
|
print_c:
|
|
if ( READY_SIZE <= readylen ) { READY_FLUSH(); }
|
|
ready[readylen++] = c;
|
|
continue;
|
|
}
|
|
|
|
if ( *format == '%' )
|
|
{
|
|
c = *format++;
|
|
goto print_c;
|
|
}
|
|
|
|
const char* format_begun_at = format;
|
|
if ( false )
|
|
{
|
|
unsupported_conversion:
|
|
format = format_begun_at;
|
|
c = '%';
|
|
goto print_c;
|
|
}
|
|
|
|
const unsigned UNSIGNED = 0;
|
|
const unsigned INTEGER = (1<<0);
|
|
const unsigned BIT64 = (1<<1);
|
|
const unsigned HEX = (1<<2);
|
|
const unsigned OCTAL = (1<<3);
|
|
const unsigned STRING = 16;
|
|
const unsigned CHARACTER = 17;
|
|
#if defined(__x86_64__)
|
|
const unsigned WORDWIDTH = BIT64;
|
|
#else
|
|
const unsigned WORDWIDTH = 0;
|
|
#endif
|
|
|
|
// TODO: Support signed datatypes!
|
|
|
|
unsigned type = 0;
|
|
|
|
bool prepend_chars = false;
|
|
bool append_chars = false;
|
|
bool space_pad = false;
|
|
bool zero_pad = false;
|
|
bool alternate = false;
|
|
bool possign_space = false;
|
|
bool possign_plus = false;
|
|
char blank_char = ' ';
|
|
unsigned int field_width = 0;
|
|
|
|
bool scanning = true;
|
|
while ( scanning )
|
|
{
|
|
switch ( char c = *(format++) )
|
|
{
|
|
case ' ':
|
|
possign_space = true;
|
|
break;
|
|
case '+':
|
|
possign_plus = true;
|
|
break;
|
|
case '#':
|
|
alternate = true;
|
|
break;
|
|
case '-':
|
|
if ( prepend_chars || append_chars )
|
|
goto unsupported_conversion;
|
|
append_chars = space_pad = true;
|
|
prepend_chars = zero_pad = false;
|
|
blank_char = ' ';
|
|
break;
|
|
case '*':
|
|
if ( !(prepend_chars || append_chars) )
|
|
prepend_chars = true;
|
|
field_width = (unsigned int) va_arg(parameters, int);
|
|
break;
|
|
case '0':
|
|
if ( !(prepend_chars || append_chars) || (!field_width && space_pad) )
|
|
{
|
|
prepend_chars = zero_pad = true;
|
|
append_chars = space_pad = false;
|
|
blank_char = '0';
|
|
break;
|
|
}
|
|
case '1':
|
|
case '2':
|
|
case '3':
|
|
case '4':
|
|
case '5':
|
|
case '6':
|
|
case '7':
|
|
case '8':
|
|
case '9':
|
|
if ( !(prepend_chars || append_chars) )
|
|
prepend_chars = true;
|
|
field_width = field_width * 10 + c - '0';
|
|
break;
|
|
case 'p':
|
|
type = WORDWIDTH | HEX;
|
|
scanning = false;
|
|
break;
|
|
case 't':
|
|
type |= INTEGER;
|
|
case 'z':
|
|
case 'l':
|
|
if ( type & WORDWIDTH ) { type |= BIT64; }
|
|
type |= WORDWIDTH;
|
|
break;
|
|
case 'j':
|
|
type |= BIT64;
|
|
break;
|
|
case 'u':
|
|
type |= UNSIGNED;
|
|
scanning = false;
|
|
break;
|
|
case 'd':
|
|
case 'i':
|
|
type |= INTEGER;
|
|
scanning = false;
|
|
break;
|
|
case 'o':
|
|
type |= OCTAL;
|
|
scanning = false;
|
|
break;
|
|
case 'x':
|
|
case 'X':
|
|
type |= HEX;
|
|
scanning = false;
|
|
break;
|
|
case 's':
|
|
type = STRING;
|
|
scanning = false;
|
|
break;
|
|
case 'c':
|
|
type = CHARACTER;
|
|
scanning = false;
|
|
break;
|
|
default:
|
|
goto unsupported_conversion;
|
|
}
|
|
}
|
|
|
|
char possign = '\0';
|
|
if ( possign_plus )
|
|
possign = '+';
|
|
else if ( possign_space )
|
|
possign = ' ';
|
|
|
|
switch ( type )
|
|
{
|
|
case INTEGER:
|
|
{
|
|
if ( READY_SIZE - readylen < 10 ) { READY_FLUSH(); }
|
|
int32_t num = va_arg(parameters, int32_t);
|
|
size_t chars = String::ConvertInt32(num, ready + readylen, possign);
|
|
if ( prepend_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
if ( READY_SIZE - readylen < 10 ) { READY_FLUSH(); }
|
|
String::ConvertInt32(num, ready + readylen, possign);
|
|
readylen += chars;
|
|
if ( append_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
break;
|
|
}
|
|
case UNSIGNED:
|
|
{
|
|
if ( READY_SIZE - readylen < 10 ) { READY_FLUSH(); }
|
|
uint32_t num = va_arg(parameters, uint32_t);
|
|
size_t chars = String::ConvertUInt32(num, ready + readylen);
|
|
if ( prepend_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
if ( READY_SIZE - readylen < 10 ) { READY_FLUSH(); }
|
|
String::ConvertUInt32(num, ready + readylen);
|
|
readylen += chars;
|
|
if ( append_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
break;
|
|
}
|
|
case INTEGER | BIT64:
|
|
{
|
|
if ( READY_SIZE - readylen < 20 ) { READY_FLUSH(); }
|
|
int64_t num = va_arg(parameters, int64_t);
|
|
size_t chars = String::ConvertInt64(num, ready + readylen, possign);
|
|
if ( prepend_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
if ( READY_SIZE - readylen < 20 ) { READY_FLUSH(); }
|
|
String::ConvertInt64(num, ready + readylen, possign);
|
|
readylen += chars;
|
|
if ( append_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
break;
|
|
}
|
|
case UNSIGNED | BIT64:
|
|
{
|
|
if ( READY_SIZE - readylen < 20 ) { READY_FLUSH(); }
|
|
uint64_t num = va_arg(parameters, uint64_t);
|
|
size_t chars = String::ConvertUInt64(num, ready + readylen);
|
|
if ( prepend_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
if ( READY_SIZE - readylen < 20 ) { READY_FLUSH(); }
|
|
String::ConvertUInt64(num, ready + readylen);
|
|
readylen += chars;
|
|
if ( append_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
break;
|
|
}
|
|
case INTEGER | HEX:
|
|
case UNSIGNED | HEX:
|
|
{
|
|
if ( READY_SIZE - readylen < 8 ) { READY_FLUSH(); }
|
|
uint32_t num = va_arg(parameters, uint32_t);
|
|
size_t chars = String::ConvertUInt32Hex(num, ready + readylen);
|
|
size_t real_chars = chars;
|
|
if ( num && alternate ) chars += 2;
|
|
if ( prepend_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
if ( READY_SIZE - readylen < 8 + 2 ) { READY_FLUSH(); }
|
|
if ( alternate && num )
|
|
ready[readylen++] = '0',
|
|
ready[readylen++] = 'x';
|
|
String::ConvertUInt32Hex(num, ready + readylen);
|
|
readylen += real_chars;
|
|
if ( append_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
break;
|
|
}
|
|
case INTEGER | BIT64 | HEX:
|
|
case UNSIGNED | BIT64 | HEX:
|
|
{
|
|
if ( READY_SIZE - readylen < 16 ) { READY_FLUSH(); }
|
|
uint64_t num = va_arg(parameters, uint64_t);
|
|
size_t chars = String::ConvertUInt64Hex(num, ready + readylen);
|
|
size_t real_chars = chars;
|
|
if ( num && alternate ) chars += 2;
|
|
if ( prepend_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
if ( READY_SIZE - readylen < 16 + 2 ) { READY_FLUSH(); }
|
|
if ( alternate && num )
|
|
ready[readylen++] = '0',
|
|
ready[readylen++] = 'x';
|
|
String::ConvertUInt64Hex(num, ready + readylen);
|
|
readylen += real_chars;
|
|
if ( append_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
break;
|
|
}
|
|
case INTEGER | OCTAL:
|
|
case UNSIGNED | OCTAL:
|
|
{
|
|
if ( READY_SIZE - readylen < 20 ) { READY_FLUSH(); }
|
|
uint32_t num = va_arg(parameters, uint32_t);
|
|
size_t chars = String::ConvertUInt32Octal(num, ready + readylen);
|
|
size_t real_chars = chars;
|
|
if ( num && alternate ) chars += 1;
|
|
if ( prepend_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
if ( READY_SIZE - readylen < 20 + 1 ) { READY_FLUSH(); }
|
|
if ( alternate && num )
|
|
ready[readylen++] = '0';
|
|
String::ConvertUInt32Octal(num, ready + readylen);
|
|
readylen += real_chars;
|
|
if ( append_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
break;
|
|
}
|
|
case INTEGER | BIT64 | OCTAL:
|
|
case UNSIGNED | BIT64 | OCTAL:
|
|
{
|
|
if ( READY_SIZE - readylen < 40 ) { READY_FLUSH(); }
|
|
uint64_t num = va_arg(parameters, uint64_t);
|
|
size_t chars = String::ConvertUInt64Octal(num, ready + readylen);
|
|
size_t real_chars = chars;
|
|
if ( num && alternate ) chars += 1;
|
|
if ( prepend_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
if ( READY_SIZE - readylen < 40 + 1 ) { READY_FLUSH(); }
|
|
if ( alternate && num )
|
|
ready[readylen++] = '0';
|
|
String::ConvertUInt64Octal(num, ready + readylen);
|
|
readylen += real_chars;
|
|
if ( append_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
break;
|
|
}
|
|
case STRING:
|
|
{
|
|
const char* str = va_arg(parameters, const char*);
|
|
size_t len = strlen(str);
|
|
size_t chars = len;
|
|
if ( prepend_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
READY_FLUSH();
|
|
if ( callback && callback(user, str, len) != len ) { return SIZE_MAX; }
|
|
if ( append_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
written += len;
|
|
break;
|
|
}
|
|
case CHARACTER:
|
|
{
|
|
int c = va_arg(parameters, int);
|
|
size_t len = 1;
|
|
size_t chars = len;
|
|
if ( prepend_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
if ( READY_SIZE <= readylen ) { READY_FLUSH(); }
|
|
ready[readylen++] = c;
|
|
if ( append_chars && chars < field_width ) { REPEAT_BLANKS(field_width - chars); }
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
READY_FLUSH();
|
|
|
|
return written;
|
|
}
|