187 lines
4.5 KiB
C
187 lines
4.5 KiB
C
#include <stdarg.h>
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
#include <lib/print.h>
|
|
#include <lib/blib.h>
|
|
#include <lib/config.h>
|
|
#include <lib/term.h>
|
|
#include <lib/libc.h>
|
|
#include <sys/cpu.h>
|
|
|
|
static int e9_output = -1;
|
|
|
|
static const char *base_digits = "0123456789abcdef";
|
|
|
|
#define PRINT_BUF_MAX 512
|
|
|
|
static void prn_str(char *print_buf, size_t *print_buf_i, const char *string) {
|
|
size_t i;
|
|
|
|
for (i = 0; string[i]; i++) {
|
|
if (*print_buf_i == (PRINT_BUF_MAX - 1))
|
|
break;
|
|
print_buf[(*print_buf_i)++] = string[i];
|
|
}
|
|
|
|
print_buf[*print_buf_i] = 0;
|
|
}
|
|
|
|
static void prn_nstr(char *print_buf, size_t *print_buf_i, const char *string, size_t len) {
|
|
size_t i;
|
|
|
|
for (i = 0; i < len; i++) {
|
|
if (*print_buf_i == (PRINT_BUF_MAX - 1))
|
|
break;
|
|
print_buf[(*print_buf_i)++] = string[i];
|
|
}
|
|
|
|
print_buf[*print_buf_i] = 0;
|
|
}
|
|
|
|
static void prn_char(char *print_buf, size_t *print_buf_i, char c) {
|
|
if (*print_buf_i < (PRINT_BUF_MAX - 1)) {
|
|
print_buf[(*print_buf_i)++] = c;
|
|
}
|
|
|
|
print_buf[*print_buf_i] = 0;
|
|
}
|
|
|
|
static void prn_i(char *print_buf, size_t *print_buf_i, int64_t x) {
|
|
int i;
|
|
char buf[20] = {0};
|
|
|
|
if (!x) {
|
|
prn_char(print_buf, print_buf_i, '0');
|
|
return;
|
|
}
|
|
|
|
int sign = x < 0;
|
|
if (sign) x = -x;
|
|
|
|
for (i = 18; x; i--) {
|
|
buf[i] = (x % 10) + 0x30;
|
|
x = x / 10;
|
|
}
|
|
if (sign)
|
|
buf[i] = '-';
|
|
else
|
|
i++;
|
|
|
|
prn_str(print_buf, print_buf_i, buf + i);
|
|
}
|
|
|
|
static void prn_ui(char *print_buf, size_t *print_buf_i, uint64_t x) {
|
|
int i;
|
|
char buf[21] = {0};
|
|
|
|
if (!x) {
|
|
prn_char(print_buf, print_buf_i, '0');
|
|
return;
|
|
}
|
|
|
|
for (i = 19; x; i--) {
|
|
buf[i] = (x % 10) + 0x30;
|
|
x = x / 10;
|
|
}
|
|
|
|
i++;
|
|
prn_str(print_buf, print_buf_i, buf + i);
|
|
}
|
|
|
|
static void prn_x(char *print_buf, size_t *print_buf_i, uint64_t x) {
|
|
int i;
|
|
char buf[17] = {0};
|
|
|
|
if (!x) {
|
|
prn_str(print_buf, print_buf_i, "0x0");
|
|
return;
|
|
}
|
|
|
|
for (i = 15; x; i--) {
|
|
buf[i] = base_digits[(x % 16)];
|
|
x = x / 16;
|
|
}
|
|
|
|
i++;
|
|
prn_str(print_buf, print_buf_i, "0x");
|
|
prn_str(print_buf, print_buf_i, buf + i);
|
|
}
|
|
|
|
void print(const char *fmt, ...) {
|
|
va_list args;
|
|
|
|
va_start(args, fmt);
|
|
vprint(fmt, args);
|
|
va_end(args);
|
|
}
|
|
|
|
static char print_buf[PRINT_BUF_MAX];
|
|
|
|
void vprint(const char *fmt, va_list args) {
|
|
if (config_ready && e9_output == -1) {
|
|
e9_output = config_get_value(print_buf, 0, PRINT_BUF_MAX, "E9_OUTPUT") &&
|
|
!strcmp(print_buf, "yes");
|
|
}
|
|
|
|
size_t print_buf_i = 0;
|
|
|
|
for (;;) {
|
|
while (*fmt && *fmt != '%')
|
|
prn_char(print_buf, &print_buf_i, *fmt++);
|
|
|
|
if (!*fmt++)
|
|
goto out;
|
|
|
|
switch (*fmt++) {
|
|
case 's': {
|
|
char *str = (char *)va_arg(args, const char *);
|
|
if (!str)
|
|
prn_str(print_buf, &print_buf_i, "(null)");
|
|
else
|
|
prn_str(print_buf, &print_buf_i, str); }
|
|
break;
|
|
case 'S': {
|
|
char *str = (char *)va_arg(args, const char *);
|
|
size_t str_len = va_arg(args, size_t);
|
|
if (!str)
|
|
prn_str(print_buf, &print_buf_i, "(null)");
|
|
else
|
|
prn_nstr(print_buf, &print_buf_i, str, str_len); }
|
|
break;
|
|
case 'd':
|
|
prn_i(print_buf, &print_buf_i, (int64_t)va_arg(args, int32_t));
|
|
break;
|
|
case 'u':
|
|
prn_ui(print_buf, &print_buf_i, (uint64_t)va_arg(args, uint32_t));
|
|
break;
|
|
case 'x':
|
|
prn_x(print_buf, &print_buf_i, (uint64_t)va_arg(args, uint32_t));
|
|
break;
|
|
case 'D':
|
|
prn_i(print_buf, &print_buf_i, va_arg(args, int64_t));
|
|
break;
|
|
case 'U':
|
|
prn_ui(print_buf, &print_buf_i, va_arg(args, uint64_t));
|
|
break;
|
|
case 'X':
|
|
prn_x(print_buf, &print_buf_i, va_arg(args, uint64_t));
|
|
break;
|
|
case 'c': {
|
|
char c = (char)va_arg(args, int);
|
|
prn_char(print_buf, &print_buf_i, c); }
|
|
break;
|
|
default:
|
|
prn_char(print_buf, &print_buf_i, '?');
|
|
break;
|
|
}
|
|
}
|
|
|
|
out:
|
|
term_write(print_buf, print_buf_i);
|
|
|
|
if (e9_output == 1) {
|
|
for (size_t i = 0; i < print_buf_i; i++)
|
|
outb(0xe9, print_buf[i]);
|
|
}
|
|
}
|