matos/core/klibc.c

428 lines
14 KiB
C
Raw Normal View History

2021-01-25 10:17:19 +01:00
#include "alloc.h"
2018-07-20 15:41:58 +02:00
#include "klibc.h"
2019-05-17 09:57:14 +02:00
#include "serial.h"
#include "vga.h"
2018-07-20 15:41:58 +02:00
int memcmp(const void *aptr, const void *bptr, size_t size)
{
const unsigned char *a = (const unsigned char *)aptr;
const unsigned char *b = (const unsigned char *)bptr;
for (size_t i = 0; i < size; i++) {
if (a[i] < b[i])
return -1;
else if (b[i] < a[i])
return 1;
}
return 0;
2019-04-15 23:19:14 +02:00
}
2018-07-20 15:41:58 +02:00
void *memcpy(void *dst, const void *src, size_t n)
{
char *dstChar = dst;
const char *srcChar = src;
for (size_t i = 0; i < n; i++) {
*(dstChar++) = *(srcChar++);
}
return dst;
2018-07-20 15:41:58 +02:00
}
2018-08-09 22:19:34 +02:00
void *memset(void *src, int c, size_t n)
{
for (char *ptr = (char *)src; n > 0; n--, ptr++) {
*ptr = (char)c;
}
return src;
2018-08-09 22:19:34 +02:00
}
2018-11-16 10:34:03 +01:00
2021-01-23 21:21:13 +01:00
char *itoa(long long int value, char *str, int base)
2018-11-16 10:34:03 +01:00
{
char *rc;
char *ptr;
char *low;
2018-11-16 10:34:03 +01:00
// Check for supported base.
if (base < 2 || base > 36) {
2018-11-16 10:34:03 +01:00
*str = '\0';
return str;
}
rc = ptr = str;
// Set '-' for negative decimals.
if (value < 0 && base == 10) {
2018-11-16 10:34:03 +01:00
*ptr++ = '-';
}
// Remember where the numbers start.
low = ptr;
// The actual conversion.
do {
2018-11-16 10:34:03 +01:00
// Modulo is negative for negative value. This trick makes abs() unnecessary.
*ptr++ = "zyxwvutsrqponmlkjihgfedcba9876543210123456789abcdefghijklmnopqrstuvwxyz"
[35 + value % base];
2018-11-16 10:34:03 +01:00
value /= base;
} while (value);
2018-11-16 10:34:03 +01:00
// Terminating the string.
*ptr-- = '\0';
// Invert the numbers.
while (low < ptr) {
2018-11-16 10:34:03 +01:00
char tmp = *low;
*low++ = *ptr;
*ptr-- = tmp;
2018-11-16 10:34:03 +01:00
}
return rc;
}
2018-11-20 17:03:57 +01:00
/* K&R */
void reverse(char s[])
{
int c, i, j;
for (i = 0, j = strlen(s) - 1; i < j; i++, j--) {
c = s[i];
s[i] = s[j];
s[j] = c;
}
}
/* K&R */
2020-04-23 00:49:09 +02:00
int strlen(const char s[])
2018-11-20 17:03:57 +01:00
{
int i = 0;
while (s[i] != '\0')
++i;
return i;
}
/* K&R
* Returns <0 if s1<s2, 0 if s1==s2, >0 if s1>s2 */
2020-04-23 00:49:09 +02:00
int strcmp(const char s1[], const char s2[])
2018-11-20 17:03:57 +01:00
{
int i;
for (i = 0; s1[i] == s2[i]; i++) {
if (s1[i] == '\0')
return 0;
}
return s1[i] - s2[i];
}
2020-04-23 00:49:09 +02:00
unsigned int strnlen(const char *s, size_t count)
{
const char *sc;
for (sc = s; count-- && *sc != '\0'; ++sc)
/* nothing */ continue;
return sc - s;
}
char *strzcpy(register char *dst, register const char *src, register int len)
{
int i;
if (len <= 0)
return dst;
for (i = 0; i < len; i++) {
dst[i] = src[i];
if (src[i] == '\0')
return dst;
}
dst[len - 1] = '\0';
return dst;
}
2021-01-24 22:20:43 +01:00
int printf(const char *format, ...)
{
2021-01-24 22:20:43 +01:00
int ret;
va_list ap;
va_start(ap, format);
2021-01-24 22:20:43 +01:00
ret = vprintf(format, ap);
va_end(ap);
2021-01-24 22:20:43 +01:00
return ret;
}
2021-01-24 22:20:43 +01:00
int puts(const char *str)
{
2021-01-24 22:20:43 +01:00
int ret = 0;
while (*str) {
putc(*(str++));
2021-01-24 22:20:43 +01:00
ret++;
}
2021-01-24 22:20:43 +01:00
return ret;
}
2021-01-24 22:20:43 +01:00
int putc(const char c)
2020-04-23 00:49:09 +02:00
{
2021-01-25 14:00:06 +01:00
VGAPutc(c);
2021-01-24 22:20:43 +01:00
serialPutc(c);
return (unsigned char)c;
}
2021-01-24 23:51:21 +01:00
// int max is 2^(sizeof(int)*8) which is (2^3)^(sizeof(int)*8/3)
// = 8^(sizeof(int)*8/3) ~ 10^(sizeof(int)*8/3)
#define PRINT_INT(name, type) \
int print##name(type integer, char *str, size_t size) \
{ \
char num[sizeof(integer) * 3]; \
int i = 0; \
int c = 0; \
int ret = 0; \
\
if (integer < 0) { \
if (str) { \
if (size) { \
str[c++] = '-'; \
size--; \
ret++; \
} else { \
return ret; \
} \
} else { \
ret++; \
} \
} \
\
do { \
int digit = integer % 10; \
num[i++] = (digit > 0) ? digit : -digit; \
integer = integer / 10; \
} while (integer != 0); \
\
for (i = i - 1; i >= 0; i--) { \
if (str) { \
if (size) { \
str[c++] = num[i] + '0'; \
size--; \
ret++; \
} else { \
return ret; \
} \
} else { \
ret++; \
} \
} \
return ret; \
}
2021-01-23 21:21:13 +01:00
2021-01-24 23:51:21 +01:00
PRINT_INT(Int, int);
PRINT_INT(Int64, long long int);
2021-01-23 21:21:13 +01:00
2021-01-24 23:51:21 +01:00
int vsnprintf(char *str, size_t size, const char *format, va_list ap)
2021-01-23 21:21:13 +01:00
{
2021-01-24 22:20:43 +01:00
int ret = 0;
int i = 0;
2021-01-24 23:51:21 +01:00
int c = 0;
2021-01-25 10:17:19 +01:00
while (format[i] != '\0' && (size|| !str)) {
switch (format[i]) {
case '%':
switch (format[i + 1]) {
case 'i':
case 'd': {
2021-01-24 23:51:21 +01:00
int s;
int d = va_arg(ap, int);
2021-01-24 23:51:21 +01:00
if (str)
s = printInt(d, &str[c], size);
else
s = printInt(d, NULL, size);
size -= s;
c += s;
ret += s;
break;
}
2020-07-08 23:03:05 +02:00
case 'p':
case 'x': {
char val[sizeof(int) * 2];
2021-01-24 23:51:21 +01:00
unsigned int valIdx = 0;
int d = va_arg(ap, int);
itoa(d, val, 16);
2021-01-24 23:51:21 +01:00
if (str) {
while (val[valIdx]) {
if (size) {
str[c++] = val[valIdx++];
size--;
ret++;
} else {
return ret;
}
}
} else {
ret += strlen(val);
}
break;
}
case 'c': {
2021-01-24 23:51:21 +01:00
if (str) {
int ch = va_arg(ap, int);
str[c++] = ch;
size--;
}
2021-01-24 22:20:43 +01:00
ret++;
break;
}
case 's': {
2021-01-24 23:51:21 +01:00
char *stri = va_arg(ap, char *);
if (!stri)
stri = "[NULL STR]";
if (str) {
while (*stri) {
if (size) {
str[c++] = *(stri++);
size--;
ret++;
} else {
return ret;
}
}
} else {
ret += strlen(stri);
}
break;
}
case '%':
2021-01-24 23:51:21 +01:00
if (str) {
str[c++] = '%';
size--;
}
2021-01-24 22:20:43 +01:00
ret++;
break;
2021-01-23 21:21:13 +01:00
case 'l':
switch (format[i + 2]) {
case 'l':
switch (format[i + 3]) {
case 'd': {
2021-01-24 23:51:21 +01:00
int s;
2021-01-23 21:21:13 +01:00
long long int d = va_arg(ap, long long int);
2021-01-24 23:51:21 +01:00
if (str)
s = printInt64(d, &str[c], size);
else
s = printInt64(d, NULL, size);
size -= s;
c += s;
ret += s;
2021-01-23 21:21:13 +01:00
break;
}
case 'p':
case 'x': {
char val[sizeof(long long int) * 2];
2021-01-24 23:51:21 +01:00
unsigned int valIdx = 0;
long long int d = va_arg(ap, long long int);
2021-01-23 21:21:13 +01:00
itoa(d, val, 16);
2021-01-24 23:51:21 +01:00
if (str) {
while (val[valIdx]) {
if (size) {
str[c++] = val[valIdx++];
size--;
ret++;
} else {
return ret;
}
}
} else {
ret += strlen(val);
}
2021-01-23 21:21:13 +01:00
break;
}
}
i++;
break;
case 'i':
case 'd': {
2021-01-24 23:51:21 +01:00
int s;
2021-01-23 21:21:13 +01:00
long int d = va_arg(ap, long int);
2021-01-24 23:51:21 +01:00
if (str)
s = printInt64(d, &str[c], size);
else
s = printInt64(d, NULL, size);
size -= s;
c += s;
ret += s;
2021-01-23 21:21:13 +01:00
break;
}
}
i++;
break;
}
i++;
break;
default:
2021-01-24 23:51:21 +01:00
if (str) {
str[c++] = format[i];
size--;
}
2021-01-24 22:20:43 +01:00
ret++;
}
i++;
}
2021-01-24 23:51:21 +01:00
if (str) {
if (size) {
str[c++] = '\0';
size--;
} else {
if (c > 0) {
str[c - 1] = '\0';
}
}
}
return ret;
}
int vprintf(const char *fmt, va_list ap)
{
char tmp[256];
int idx = 0;
int ret = vsnprintf(tmp, sizeof(tmp), fmt, ap);
while (ret > 0 && tmp[idx]) {
putc(tmp[idx++]);
}
2021-01-24 22:20:43 +01:00
return ret;
}
2021-01-25 10:17:19 +01:00
int asprintf(char **strp, const char *fmt, ...)
{
int ret;
va_list ap;
va_start(ap, fmt);
ret = vasprintf(strp, fmt, ap);
va_end(ap);
return ret;
}
int vasprintf(char **strp, const char *fmt, va_list ap)
{
int n = 0;
size_t size = 0;
char *p = NULL;
/* Determine required size */
n = vsnprintf(p, size, fmt, ap);
if (n < 0)
return -1;
/* One extra byte for '\0' */
size = (size_t)n + 1;
p = malloc(size);
if (p == NULL)
return -1;
n = vsnprintf(p, size, fmt, ap);
if (n < 0) {
free(p);
return -1;
}
*strp = p;
return size;
}