RossComputerGuy/stardustos

/**
 * NewLand Kernel - (C) 2019 Tristan Ross
 */
#include <newland/dev/tty.h>
#include <newland/log.h>
#include <newland/time.h>
#include <newland/errno.h>
#include <limits.h>
#include <newland-config.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>

#define TIME_STRLEN 12

static char klog_buffer[NEWLAND_KLOG_SIZE] = { 0 };
static size_t klog_pos = 0;

static int print(const char* str, size_t len, int inmsg) {
    if (str[0] == KLOG_SOH_ASCII) {
        str += 2;
        len -= 2;
    }
#ifndef __nvk__
    size_t freesize = NEWLAND_KLOG_SIZE - klog_pos;
    if (!inmsg) len += TIME_STRLEN + 3;
    if (freesize < len) {
        klog_pos = 0;
    }
    char* buff = klog_buffer + klog_pos;
    if (!inmsg) {
        buff[0] = '[';
        memset(buff + 1, ' ', TIME_STRLEN);
        time_t t = boottime(NULL);
        snprintf(buff + 1, TIME_STRLEN, "%d", t);
        buff += 1 + TIME_STRLEN;
        buff[0] = ']';
        buff[1] = ' ';
        buff += 2;
    } else buff++;
    strcpy(buff, str);
    buff[len] = 0;
    buff = klog_buffer + klog_pos;
    tty_t* tty = tty_get(tty_count() - 1);
    if (tty != NULL) {
        tty->opts.write(tty, buff, len);
        if (buff[len - 1] == '\n') tty->opts.write(tty, "\r", 1);
    }
    klog_pos += len;
#endif
    return len;
}

int printk(const char* fmt, ...) {
    va_list ap;
    va_start(ap, fmt);
    char buff[512];
    int wrote = vsnprintf(buff, sizeof(buff), fmt, ap);
    if (wrote < 0) {
        va_end(ap);
        return wrote;
    }
    wrote = print(buff, strlen(buff), 0);
    va_end(ap);
    return wrote;
}