1
0
Fork 0
mirror of https://gitlab.com/sortix/sortix.git synced 2023-02-13 20:55:38 -05:00
sortix--sortix/libc/format.cpp
2013-12-17 14:30:27 +01:00

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/>.
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;
}