3 * - By John Hodge (thePowersGang)
6 * - Kernel-land C Library
9 #include <hal_proc.h> // For MM_*
13 #define RANDOM_SEED 0xACE55052
14 #define RANDOM_A 0x00731ADE
15 #define RANDOM_C 12345
16 #define RANDOM_SPRUCE 0xf12b039
20 unsigned long long strtoull(const char *str, char **end, int base);
21 unsigned long strtoul(const char *str, char **end, int base);
22 signed long long strtoll(const char *str, char **end, int base);
23 signed long strtol(const char *str, char **end, int base);
24 int atoi(const char *string);
25 int ParseInt(const char *string, int *Val);
26 void itoa(char *buf, Uint64 num, int base, int minLength, char pad);
27 int vsnprintf(char *__s, size_t __maxlen, const char *__format, va_list args);
28 int snprintf(char *__s, size_t __n, const char *__format, ...);
29 int sprintf(char *__s, const char *__format, ...);
30 int strucmp(const char *Str1, const char *Str2);
31 char *strchr(const char *__s, int __c);
32 int strpos(const char *Str, char Ch);
33 size_t strlen(const char *__s);
34 char *strcpy(char *__str1, const char *__str2);
35 char *strncpy(char *__str1, const char *__str2, size_t max);
36 char *strcat(char *dest, const char *source);
37 int strcmp(const char *str1, const char *str2);
38 int strncmp(const char *str1, const char *str2, size_t num);
39 char *_strdup(const char *File, int Line, const char *Str);
41 void *memmove(void *__dest, const void *__src, size_t len);
43 int CheckString(char *String);
44 int CheckMem(void *Mem, int NumBytes);
67 EXPORT(_strdup); // Takes File/Line too
75 // - Import userland stroi.c file
77 #define _SysDebug(f,v...) Log_Debug("libc", f ,## v)
78 #include "../../Usermode/Libraries/libc.so_src/strtoi.c"
80 int ParseInt(const char *string, int *Val)
84 const char *orig_string = string;
86 //Log("atoi: (string='%s')", string);
88 // Clear non-numeric characters
89 while( !('0' <= *string && *string <= '9') && *string != '-' ) string++;
90 if( *string == '-' ) {
92 while( !('0' <= *string && *string <= '9') ) string++;
104 if('0' <= *string && *string <= '9') {
106 ret += *string - '0';
108 else if('A' <= *string && *string <= 'F') {
110 ret += *string - 'A' + 10;
112 else if('a' <= *string && *string <= 'f') {
114 ret += *string - 'a' + 10;
122 for( ; '0' <= *string && *string <= '7'; string ++ )
125 ret += *string - '0';
131 for( ; '0' <= *string && *string <= '9'; string++)
134 ret += *string - '0';
137 if( ret == 0 ) return 0;
142 //Log("atoi: RETURN %i", ret);
146 return string - orig_string;
149 static const char cUCDIGITS[] = "0123456789ABCDEF";
151 * \fn void itoa(char *buf, Uint64 num, int base, int minLength, char pad)
152 * \brief Convert an integer into a character string
154 void itoa(char *buf, Uint64 num, int base, int minLength, char pad)
161 ASSERTR(base >= 2, );
162 ASSERTR(base <= 16, );
168 while(num > base-1) {
169 num = DivMod64U(num, base, &rem); // Shift `num` and get remainder
171 if( rem >= base && base != 16 ) {
172 Debug("rem(%llx) >= base(%x), num=%llx", rem, base, num);
175 tmpBuf[pos] = cUCDIGITS[ rem ];
178 tmpBuf[pos++] = cUCDIGITS[ num ]; // Last digit of `num`
183 while(minLength-- > 0) buf[i++] = pad;
184 while(pos-- > 0) buf[i++] = tmpBuf[pos]; // Reverse the order of characters
189 * \brief Append a character the the vsnprintf output
191 #define PUTCH(ch) do { \
192 if(pos < __maxlen && __s) { \
199 #define GETVAL() do {\
200 if(isLongLong) val = va_arg(args, Uint64);\
201 else val = va_arg(args, unsigned int);\
204 * \brief VArg String Number Print Formatted
206 int vsnprintf(char *__s, const size_t __maxlen, const char *__format, va_list args)
209 int minSize = 0, precision = -1, len;
210 char tmpBuf[34]; // For Integers
211 const char *p = NULL;
212 int isLongLong = 0, isLong;
218 while((c = *__format++) != 0)
220 // Non control character
221 if(c != '%') { PUTCH(c); continue; }
227 if(c == '%') { PUTCH('%'); continue; }
229 // Pointer - Done first for debugging
231 Uint ptr = va_arg(args, Uint);
232 PUTCH('*'); PUTCH('0'); PUTCH('x');
233 for( len = BITS/4; len -- && ((ptr>>(len*4))&15) == 0; )
240 PUTCH( cUCDIGITS[ (ptr>>(len*4))&15 ] );
247 // - Padding Side Flag
262 if(c == '*') { // Dynamic length
263 minSize = va_arg(args, unsigned int);
266 else if('1' <= c && c <= '9')
269 while('0' <= c && c <= '9')
284 if(c == '*') { // Dynamic length
285 precision = va_arg(args, unsigned int);
288 else if('1' <= c && c <= '9')
291 while('0' <= c && c <= '9')
294 precision += c - '0';
300 // - Default, Long or LongLong?
302 if(c == 'l') // Long is actually the default on x86
312 // - Now get the format code
319 if( isLongLong && val >> 63 ) {
321 if( val == LLONG_MIN )
326 else if( !isLongLong && (val >> 31) ) {
328 val = (~val & 0xFFFFFFFF)+1;
330 itoa(tmpBuf, val, 10, minSize, pad);
332 case 'u': // Unsigned
334 itoa(tmpBuf, val, 10, minSize, pad);
336 case 'P': // Physical Address
339 if(sizeof(tPAddr) > 4) isLongLong = 1;
341 itoa(tmpBuf, val, 16, minSize, pad);
345 isLongLong = 1; // TODO: Handle non-x86 64-bit archs
347 itoa(tmpBuf, val, 16, minSize, pad);
350 case 'x': // Lower case hex
352 itoa(tmpBuf, val, 16, minSize, pad);
356 itoa(tmpBuf, val, 8, minSize, pad);
360 itoa(tmpBuf, val, 2, minSize, pad);
364 val = va_arg(args, unsigned int);
369 // String - Null Terminated Array
372 Uint16 *p16 = va_arg(args, Uint16*);
374 while( *p16 && precision-- ) {
376 p16 += ReadUTF16(p16, &cp);
377 tmp[WriteUTF8(tmp, cp)] = 0;
378 for(int i = 0; tmp[i] && i<5; i ++)
383 p = va_arg(args, char*); // Get Argument
384 if( p && !CheckString(p) ) goto invalString; // Avoid #PFs
388 if( !bPadLeft ) while(len++ < minSize) PUTCH(pad);
389 while(*p && precision--) { PUTCH(*p); p++;}
390 if( bPadLeft ) while(len++ < minSize) PUTCH(pad);
393 PUTCH('(');PUTCH('i');PUTCH('n');PUTCH('v');PUTCH('a'); PUTCH('l');PUTCH(':');
394 PUTCH('*');PUTCH('0');PUTCH('x');
396 for( len = BITS/4; len -- && ((val>>(len*4))&15) == 0; )
403 PUTCH( cUCDIGITS[ (val>>(len*4))&15 ] );
407 case 'C': // Non-Null Terminated Character Array
408 p = va_arg(args, char*);
409 if( !CheckMem(p, minSize) ) {
413 if(!p) goto printString;
419 else if(*p == '\\') {
439 if(__s && pos < __maxlen)
448 int snprintf(char *__s, size_t __n, const char *__format, ...)
453 va_start(args, __format);
454 ret = vsnprintf(__s, __n, __format, args);
462 int sprintf(char *__s, const char *__format, ...)
467 va_start(args, __format);
468 ret = vsnprintf(__s, -1, __format, args);
481 return isalpha(c) || isdigit(c);
485 return isupper(c) || islower(c);
489 return (0 <= c && c < 128);
493 if(c == '\t') return 1;
494 if(c == ' ') return 1;
499 // TODO: Check iscntrl
500 if(c < ' ') return 1;
505 return ('0' <= c && c <= '9');
509 // TODO: Check isgraph
514 return ('a' <= c && c <= 'z');
518 if( ' ' <= c && c <= 0x7F ) return 1;
534 if(c == ' ') return 1;
535 if(c == '\t') return 1;
536 if(c == '\v') return 1;
537 if(c == '\n') return 1;
538 if(c == '\r') return 1;
543 return ('A' <= c && c <= 'Z');
547 return isdigit(c) || ('a' <= c && c <= 'f') || ('A' <= c && c <= 'F');
566 * \fn int strucmp(const char *Str1, const char *Str2)
567 * \brief Compare \a Str1 and \a Str2 case-insensitively
569 int strucmp(const char *Str1, const char *Str2)
571 while(*Str1 && tolower(*Str1) == tolower(*Str2))
573 return tolower(*Str1) - tolower(*Str2);
577 * \brief Locate a byte in a string
579 char *strchr(const char *__s, int __c)
581 for( ; *__s; __s ++ )
583 if( *__s == __c ) return (char*)__s;
588 char *strrchr(const char *__s, int __c)
590 size_t ofs = strlen(__s);
591 while(--ofs && __s[ofs] != __c);
592 if( __s[ofs] == __c )
593 return (char*)__s + ofs;
598 * \fn int strpos(const char *Str, char Ch)
599 * \brief Search a string for an ascii character
601 int strpos(const char *Str, char Ch)
604 for(pos=0;Str[pos];pos++)
606 if(Str[pos] == Ch) return pos;
612 * \fn size_t strlen(const char *__str)
613 * \brief Get the length of string
615 size_t strlen(const char *__str)
618 while(*__str++) ret++;
623 * \brief Copy a string to a new location
625 char *strcpy(char *__str1, const char *__str2)
628 *__str1++ = *__str2++;
629 *__str1 = '\0'; // Terminate String
634 * \brief Copy a string to a new location
635 * \note Copies at most `max` chars
637 char *strncpy(char *__str1, const char *__str2, size_t __max)
639 while(*__str2 && __max-- >= 1)
640 *__str1++ = *__str2++;
642 *__str1 = '\0'; // Terminate String
647 * \brief Append a string to another
649 char *strcat(char *__dest, const char *__src)
654 *__dest++ = *__src++;
660 * \brief Append at most \a n chars to a string from another
661 * \note At most n+1 chars are written (the dest is always zero terminated)
663 char *strncat(char *__dest, const char *__src, size_t n)
666 while(*__src && n-- >= 1)
667 *__dest++ = *__src++;
673 * \fn int strcmp(const char *str1, const char *str2)
674 * \brief Compare two strings return the difference between
675 * the first non-matching characters.
677 int strcmp(const char *str1, const char *str2)
679 while(*str1 && *str1 == *str2)
681 return *str1 - *str2;
685 * \fn int strncmp(const char *Str1, const char *Str2, size_t num)
686 * \brief Compare strings \a Str1 and \a Str2 to a maximum of \a num characters
688 int strncmp(const char *Str1, const char *Str2, size_t num)
690 if(num == 0) return 0; // TODO: Check what should officially happen here
691 while(--num && *Str1 && *Str1 == *Str2)
698 * \fn char *strdup(const char *Str)
699 * \brief Duplicates a string
701 char *strdup(const char *Str)
704 ret = malloc(strlen(Str)+1);
705 if( !ret ) return NULL;
712 * \fn char *_strdup(const char *File, int Line, const char *Str)
713 * \brief Duplicates a string
715 char *_strdup(const char *File, int Line, const char *Str)
718 ret = Heap_Allocate(File, Line, strlen(Str)+1);
719 if( !ret ) return NULL;
727 * \brief Pseudo random number generator
732 static Uint state = RANDOM_SEED;
734 // Get the next state value
735 giRandomState = (RANDOM_A*state + RANDOM_C);
736 // Check if it has changed, and if it hasn't, change it
737 if(state == old) state += RANDOM_SPRUCE;
740 // http://en.wikipedia.org/wiki/Xorshift
742 static Uint32 x = 123456789;
743 static Uint32 y = 362436069;
744 static Uint32 z = 521288629;
745 static Uint32 w = 88675123;
750 return w = w ^ (w >> 19) ^ t ^ (t >> 8);
754 void *memmove(void *__dest, const void *__src, size_t len)
757 const char *src = __src;
760 if( len == 0 || dest == src )
763 if( (tVAddr)dest > (tVAddr)src + len )
764 return memcpy(dest, src, len);
765 if( (tVAddr)dest + len < (tVAddr)src )
766 return memcpy(dest, src, len);
768 // NOTE: Assumes memcpy works forward
769 if( (tVAddr)dest < (tVAddr)src )
770 return memcpy(dest, src, len);
774 if( (tVAddr)dest < (tVAddr)src )
775 block_size = (tVAddr)src - (tVAddr)dest;
777 block_size = (tVAddr)dest - (tVAddr)src;
781 while(len >= block_size)
783 memcpy(dest, src, block_size);
788 memcpy(dest, src, len);
791 for( int i = len; i--; )
800 // NOTE: Strictly not libc, but lib.c is used by userland code too and hence these two
803 * \name Memory Validation
807 * \brief Checks if a string resides fully in valid memory
809 int CheckString(const char *String)
814 addr = (tVAddr)String;
816 if( !MM_GetPhysAddr( (void*)addr ) )
820 bUser = MM_IsUser( addr );
822 while( *(char*)addr )
824 if( (addr & (PAGE_SIZE-1)) == 0 )
826 if(bUser && !MM_IsUser(addr) )
828 if(!bUser && !MM_GetPhysAddr((void*)addr) )
837 * \brief Check if a sized memory region is valid memory
838 * \return Boolean success
840 int CheckMem(const void *Mem, int NumBytes)
842 return MM_IsValidBuffer( (tVAddr)Mem, NumBytes );