3 * - By John Hodge (thePowersGang)
6 * - Kernel-land C Library
9 #include <hal_proc.h> // For MM_*
12 #define RANDOM_SEED 0xACE55052
13 #define RANDOM_A 0x00731ADE
14 #define RANDOM_C 12345
15 #define RANDOM_SPRUCE 0xf12b039
19 int atoi(const char *string);
20 int ParseInt(const char *string, int *Val);
21 void itoa(char *buf, Uint64 num, int base, int minLength, char pad);
22 int vsnprintf(char *__s, size_t __maxlen, const char *__format, va_list args);
23 int snprintf(char *__s, size_t __n, const char *__format, ...);
24 int sprintf(char *__s, const char *__format, ...);
25 int strucmp(const char *Str1, const char *Str2);
26 char *strchr(const char *__s, int __c);
27 int strpos(const char *Str, char Ch);
28 size_t strlen(const char *__s);
29 char *strcpy(char *__str1, const char *__str2);
30 char *strncpy(char *__str1, const char *__str2, size_t max);
31 char *strcat(char *dest, const char *source);
32 int strcmp(const char *str1, const char *str2);
33 int strncmp(const char *str1, const char *str2, size_t num);
34 char *_strdup(const char *File, int Line, const char *Str);
36 void *memmove(void *__dest, const void *__src, size_t len);
38 int CheckString(char *String);
39 int CheckMem(void *Mem, int NumBytes);
62 EXPORT(_strdup); // Takes File/Line too
71 * \brief Convert a string into an integer
73 int atoi(const char *string)
76 ParseInt(string, &ret);
79 int ParseInt(const char *string, int *Val)
83 const char *orig_string = string;
85 //Log("atoi: (string='%s')", string);
87 // Clear non-numeric characters
88 while( !('0' <= *string && *string <= '9') && *string != '-' ) string++;
89 if( *string == '-' ) {
91 while( !('0' <= *string && *string <= '9') ) string++;
103 if('0' <= *string && *string <= '9') {
105 ret += *string - '0';
107 else if('A' <= *string && *string <= 'F') {
109 ret += *string - 'A' + 10;
111 else if('a' <= *string && *string <= 'f') {
113 ret += *string - 'a' + 10;
121 for( ; '0' <= *string && *string <= '7'; string ++ )
124 ret += *string - '0';
130 for( ; '0' <= *string && *string <= '9'; string++)
133 ret += *string - '0';
136 if( ret == 0 ) return 0;
141 //Log("atoi: RETURN %i", ret);
145 return string - orig_string;
148 static const char cUCDIGITS[] = "0123456789ABCDEF";
150 * \fn void itoa(char *buf, Uint64 num, int base, int minLength, char pad)
151 * \brief Convert an integer into a character string
153 void itoa(char *buf, Uint64 num, int base, int minLength, char pad)
163 if(base > 16 || base < 2) {
169 while(num > base-1) {
170 num = DivMod64U(num, base, &rem); // Shift `num` and get remainder
171 tmpBuf[pos] = cUCDIGITS[ rem ];
174 tmpBuf[pos++] = cUCDIGITS[ num ]; // Last digit of `num`
179 while(minLength-- > 0) buf[i++] = pad;
180 while(pos-- > 0) buf[i++] = tmpBuf[pos]; // Reverse the order of characters
185 * \brief Append a character the the vsnprintf output
187 #define PUTCH(ch) do { \
188 if(pos < __maxlen) { \
189 if(__s) __s[pos] = ch; \
195 #define GETVAL() do {\
196 if(isLongLong) val = va_arg(args, Uint64);\
197 else val = va_arg(args, unsigned int);\
200 * \brief VArg String Number Print Formatted
202 int vsnprintf(char *__s, size_t __maxlen, const char *__format, va_list args)
205 int minSize = 0, precision = -1, len;
206 char tmpBuf[34]; // For Integers
207 const char *p = NULL;
214 while((c = *__format++) != 0)
216 // Non control character
217 if(c != '%') { PUTCH(c); continue; }
223 if(c == '%') { PUTCH('%'); continue; }
225 // Pointer - Done first for debugging
227 Uint ptr = va_arg(args, Uint);
228 PUTCH('*'); PUTCH('0'); PUTCH('x');
229 for( len = BITS/4; len -- && ((ptr>>(len*4))&15) == 0; )
236 PUTCH( cUCDIGITS[ (ptr>>(len*4))&15 ] );
240 // - Padding Side Flag
255 if(c == '*') { // Dynamic length
256 minSize = va_arg(args, unsigned int);
259 else if('1' <= c && c <= '9')
262 while('0' <= c && c <= '9')
277 if(c == '*') { // Dynamic length
278 precision = va_arg(args, unsigned int);
281 else if('1' <= c && c <= '9')
284 while('0' <= c && c <= '9')
287 precision += c - '0';
293 // - Default, Long or LongLong?
295 if(c == 'l') // Long is actually the default on x86
304 // - Now get the format code
311 if( isLongLong && val >> 63 ) {
315 else if( !isLongLong && val >> 31 ) {
319 itoa(tmpBuf, val, 10, minSize, pad);
321 case 'u': // Unsigned
323 itoa(tmpBuf, val, 10, minSize, pad);
325 case 'P': // Physical Address
328 if(sizeof(tPAddr) > 4) isLongLong = 1;
330 itoa(tmpBuf, val, 16, minSize, pad);
334 isLongLong = 1; // TODO: Handle non-x86 64-bit archs
336 itoa(tmpBuf, val, 16, minSize, pad);
339 case 'x': // Lower case hex
341 itoa(tmpBuf, val, 16, minSize, pad);
345 itoa(tmpBuf, val, 8, minSize, pad);
349 itoa(tmpBuf, val, 2, minSize, pad);
353 val = va_arg(args, unsigned int);
358 // String - Null Terminated Array
360 p = va_arg(args, char*); // Get Argument
361 if( !p || !CheckString(p) ) p = "(inval)"; // Avoid #PFs
365 if( !bPadLeft ) while(len++ < minSize) PUTCH(pad);
366 while(*p && precision--) { PUTCH(*p); p++;}
367 if( bPadLeft ) while(len++ < minSize) PUTCH(pad);
370 case 'C': // Non-Null Terminated Character Array
371 p = va_arg(args, char*);
372 if( !CheckMem(p, minSize) ) continue; // No #PFs please
373 if(!p) goto printString;
389 if(__s && pos != __maxlen)
398 int snprintf(char *__s, size_t __n, const char *__format, ...)
403 va_start(args, __format);
404 ret = vsnprintf(__s, __n, __format, args);
412 int sprintf(char *__s, const char *__format, ...)
417 va_start(args, __format);
418 ret = vsnprintf(__s, -1, __format, args);
431 return isalpha(c) || isdigit(c);
435 return isupper(c) || islower(c);
439 return (0 <= c && c < 128);
443 if(c == '\t') return 1;
444 if(c == ' ') return 1;
449 // TODO: Check iscntrl
450 if(c < ' ') return 1;
455 return ('0' <= c && c <= '9');
459 // TODO: Check isgraph
464 return ('a' <= c && c <= 'z');
468 if( ' ' <= c && c <= 0x7F ) return 1;
484 if(c == ' ') return 1;
485 if(c == '\t') return 1;
486 if(c == '\v') return 1;
487 if(c == '\n') return 1;
488 if(c == '\r') return 1;
493 return ('a' <= c && c <= 'z');
497 return isdigit(c) || ('a' <= c && c <= 'f') || ('A' <= c && c <= 'F');
516 * \fn int strucmp(const char *Str1, const char *Str2)
517 * \brief Compare \a Str1 and \a Str2 case-insensitively
519 int strucmp(const char *Str1, const char *Str2)
521 while(*Str1 && tolower(*Str1) == tolower(*Str2))
523 return tolower(*Str1) - tolower(*Str2);
527 * \brief Locate a byte in a string
529 char *strchr(const char *__s, int __c)
531 for( ; *__s; __s ++ )
533 if( *__s == __c ) return (char*)__s;
538 char *strrchr(const char *__s, int __c)
540 size_t ofs = strlen(__s);
541 while(--ofs && __s[ofs] != __c);
542 if( __s[ofs] == __c )
543 return (char*)__s + ofs;
548 * \fn int strpos(const char *Str, char Ch)
549 * \brief Search a string for an ascii character
551 int strpos(const char *Str, char Ch)
554 for(pos=0;Str[pos];pos++)
556 if(Str[pos] == Ch) return pos;
562 * \fn size_t strlen(const char *__str)
563 * \brief Get the length of string
565 size_t strlen(const char *__str)
568 while(*__str++) ret++;
573 * \brief Copy a string to a new location
575 char *strcpy(char *__str1, const char *__str2)
578 *__str1++ = *__str2++;
579 *__str1 = '\0'; // Terminate String
584 * \brief Copy a string to a new location
585 * \note Copies at most `max` chars
587 char *strncpy(char *__str1, const char *__str2, size_t __max)
589 while(*__str2 && __max-- >= 1)
590 *__str1++ = *__str2++;
592 *__str1 = '\0'; // Terminate String
597 * \brief Append a string to another
599 char *strcat(char *__dest, const char *__src)
604 *__dest++ = *__src++;
610 * \brief Append at most \a n chars to a string from another
611 * \note At most n+1 chars are written (the dest is always zero terminated)
613 char *strncat(char *__dest, const char *__src, size_t n)
616 while(*__src && n-- >= 1)
617 *__dest++ = *__src++;
623 * \fn int strcmp(const char *str1, const char *str2)
624 * \brief Compare two strings return the difference between
625 * the first non-matching characters.
627 int strcmp(const char *str1, const char *str2)
629 while(*str1 && *str1 == *str2)
631 return *str1 - *str2;
635 * \fn int strncmp(const char *Str1, const char *Str2, size_t num)
636 * \brief Compare strings \a Str1 and \a Str2 to a maximum of \a num characters
638 int strncmp(const char *Str1, const char *Str2, size_t num)
640 if(num == 0) return 0; // TODO: Check what should officially happen here
641 while(--num && *Str1 && *Str1 == *Str2)
648 * \fn char *strdup(const char *Str)
649 * \brief Duplicates a string
651 char *strdup(const char *Str)
654 ret = malloc(strlen(Str)+1);
655 if( !ret ) return NULL;
662 * \fn char *_strdup(const char *File, int Line, const char *Str)
663 * \brief Duplicates a string
665 char *_strdup(const char *File, int Line, const char *Str)
668 ret = Heap_Allocate(File, Line, strlen(Str)+1);
669 if( !ret ) return NULL;
677 * \brief Pseudo random number generator
682 static Uint state = RANDOM_SEED;
684 // Get the next state value
685 giRandomState = (RANDOM_A*state + RANDOM_C);
686 // Check if it has changed, and if it hasn't, change it
687 if(state == old) state += RANDOM_SPRUCE;
690 // http://en.wikipedia.org/wiki/Xorshift
692 static Uint32 x = 123456789;
693 static Uint32 y = 362436069;
694 static Uint32 z = 521288629;
695 static Uint32 w = 88675123;
700 return w = w ^ (w >> 19) ^ t ^ (t >> 8);
704 void *memmove(void *__dest, const void *__src, size_t len)
707 const char *src = __src;
710 if( len == 0 || dest == src )
713 if( (tVAddr)dest > (tVAddr)src + len )
714 return memcpy(dest, src, len);
715 if( (tVAddr)dest + len < (tVAddr)src )
716 return memcpy(dest, src, len);
718 // NOTE: Assumes memcpy works forward
719 if( (tVAddr)dest < (tVAddr)src )
720 return memcpy(dest, src, len);
724 if( (tVAddr)dest < (tVAddr)src )
725 block_size = (tVAddr)src - (tVAddr)dest;
727 block_size = (tVAddr)dest - (tVAddr)src;
731 while(len >= block_size)
733 memcpy(dest, src, block_size);
738 memcpy(dest, src, len);
741 for( int i = len; i--; )
750 // NOTE: Strictly not libc, but lib.c is used by userland code too and hence these two
753 * \name Memory Validation
757 * \brief Checks if a string resides fully in valid memory
759 int CheckString(const char *String)
764 addr = (tVAddr)String;
766 if( !MM_GetPhysAddr( (void*)addr ) )
770 bUser = MM_IsUser( addr );
772 while( *(char*)addr )
774 if( (addr & (PAGE_SIZE-1)) == 0 )
776 if(bUser && !MM_IsUser(addr) )
778 if(!bUser && !MM_GetPhysAddr((void*)addr) )
787 * \brief Check if a sized memory region is valid memory
788 * \return Boolean success
790 int CheckMem(const void *Mem, int NumBytes)
792 return MM_IsValidBuffer( (tVAddr)Mem, NumBytes );