summaryrefslogtreecommitdiff
path: root/kernel/libc/stdio/print.c
blob: e8b245f058d6502b0d8b6b8bfd7c7a3d8af50ca1 (plain)
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include <assert.h>
#include <drivers/serial.h>
#include <stdio.h>

#define TAB_SIZE 8

const char HEX_SET[0x10] = {'0', '1', '2', '3', '4', '5', '6', '7',
                            '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'};

inline void putc(const char c) {
  write_serial(c);
}

int kprint_hex(u64 num) {
  int c = 2;

  if (num == 0) {
    putc('0');
    c++;
    return c;
  }

  char str[16] = {0};
  int i = 0;
  for (; num != 0 && i < 16; i++, num /= 16) {
    str[i] = HEX_SET[(num % 16)];
  }

  c += i;
  for (i--; i >= 0; i--) {
    putc(str[i]);
  }

  return c;
}

int kprint_int(int num) {
  int c = 0;
  if (0 == num) {
    putc('0');
    c++;
    return c;
  }
  char str[10];
  int i = 0;
  for (; num != 0 && i < 10; i++, num /= 10) {
    str[i] = (num % 10) + '0';
  }

  c += i;
  for (i--; i >= 0; i--) {
    putc(str[i]);
  }
  return c;
}

int vkprintf(const char *format, va_list list) {
  int c = 0;
  const char *s = format;
  for (; *s; s++) {
    if ('%' != *s) {
      putc(*s);
      c++;
      continue;
    }

    char flag = *(s + 1);
    if ('\0' == flag) {
      break;
    }

    switch (flag) {
    case 'c':
      putc((char)va_arg(list, int));
      c++;
      break;
    case 'd':
      c += kprint_int(va_arg(list, int));
      break;
    case 's':
      for (char *string = va_arg(list, char *); *string; putc(*string++), c++)
        ;
      break;
    case 'x':
      c += kprint_hex(va_arg(list, const u32));
      break;
    case '%':
      putc('%');
      c++;
      break;
    default:
      ASSERT_NOT_REACHED;
      break;
    }
    s++;
  }
  return c;
}

int kprintf(const char *format, ...) {
  va_list list;
  va_start(list, format);
  int rc = vkprintf(format, list);
  va_end(list);
  return rc;
}