1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
#include <stdarg.h>
#include <stdio.h>
#include <string.h>
#include <sys/file.h>
#include <unistd.h>
#include <log.h>
#include <util.h>
static const char *colors[] = {
[0] = ANSI_BLINK ANSI_REVERSE ANSI_BOLD ANSI_RED,
[1] = ANSI_REVERSE ANSI_BOLD ANSI_RED,
[2] = ANSI_BOLD ANSI_RED,
[3] = ANSI_RED,
[4] = ANSI_BOLD,
[5] = ANSI_BRIGHT_WHITE,
[6] = ANSI_RESET,
[7] = ANSI_ITALIC ANSI_BRIGHT_BLUE,
};
int console_lock = 0;
int print(const char *fmt, ...)
{
int loglevel = 5;
if(fmt[0] == LOG_SOH_ASCII) {
loglevel = (fmt[1] - 0x30) % 10;
fmt += 2;
}
char buf[512];
va_list ap;
va_start(ap, fmt);
vsnprintf(buf, 512, fmt, ap);
va_end(ap);
buf[512 - 1] = '\0';
size_t colon = 0;
for(; colon < strlen(buf); ++colon) {
if(buf[colon] == ':')
break;
}
/* spin lock, at the cost of architecture portability */
__asm__(".spin_lock:");
__asm__("mov rax, 1");
__asm__("xchg rax, [console_lock]");
__asm__("test rax, rax");
__asm__("jnz .spin_lock");
if(buf[colon] == ':') {
writeputs(ANSI_RESET);
writeputs(colors[loglevel]);
writeputs(ANSI_YELLOW);
write(STDOUT_FILENO, buf, colon);
writeputs(ANSI_RESET);
}
writeputs(colors[loglevel]);
if(colon && *(buf + colon)) {
writeputs(buf + colon);
} else {
writeputs(buf);
}
writeputs(ANSI_RESET);
console_lock = 0;
return 0;
}
|
