Kernel/libc - Fixed passing INT_MIN to %i causing an assertion failure
[tpg/acess2.git] / KernelLand / Kernel / libc.c
1 /*
2  * Acess2 Kernel
3  * - By John Hodge (thePowersGang)
4  *
5  * libc.c
6  * - Kernel-land C Library
7  */
8 #include <acess.h>
9 #include <hal_proc.h>   // For MM_*
10 #include <utf16.h>
11
12 // === CONSTANTS ===
13 #define RANDOM_SEED     0xACE55052
14 #define RANDOM_A        0x00731ADE
15 #define RANDOM_C        12345
16 #define RANDOM_SPRUCE   0xf12b039
17
18 // === PROTOTYPES ===
19 #if 0
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);
40  int    rand(void);
41 void    *memmove(void *__dest, const void *__src, size_t len);
42
43  int    CheckString(char *String);
44  int    CheckMem(void *Mem, int NumBytes); 
45 #endif
46
47 // === EXPORTS ===
48 EXPORT(atoi);
49 EXPORT(itoa);
50 EXPORT(vsnprintf);
51 EXPORT(snprintf);
52 EXPORT(sprintf);
53 EXPORT(tolower);
54
55 EXPORT(strucmp);
56 EXPORT(strchr);
57 EXPORT(strrchr);
58 EXPORT(strpos);
59 EXPORT(strlen);
60 EXPORT(strcpy);
61 EXPORT(strncpy);
62 EXPORT(strcat);
63 EXPORT(strncat);
64 EXPORT(strcmp);
65 EXPORT(strncmp);
66 //EXPORT(strdup);
67 EXPORT(_strdup);        // Takes File/Line too
68 EXPORT(rand);
69 EXPORT(memmove);
70
71 EXPORT(CheckString);
72 EXPORT(CheckMem);
73
74 // === CODE ===
75 // - Import userland stroi.c file
76 #define _LIB_H_
77 #include "../../Usermode/Libraries/libc.so_src/strtoi.c"
78
79 int ParseInt(const char *string, int *Val)
80 {
81          int    ret = 0;
82          int    bNeg = 0;
83         const char *orig_string = string;
84         
85         //Log("atoi: (string='%s')", string);
86         
87         // Clear non-numeric characters
88         while( !('0' <= *string && *string <= '9') && *string != '-' )  string++;
89         if( *string == '-' ) {
90                 bNeg = 1;
91                 while( !('0' <= *string && *string <= '9') )    string++;
92         }
93         
94         if(*string == '0')
95         {
96                 string ++;
97                 if(*string == 'x')
98                 {
99                         // Hex
100                         string ++;
101                         for( ;; string ++ )
102                         {
103                                 if('0' <= *string && *string <= '9') {
104                                         ret *= 16;
105                                         ret += *string - '0';
106                                 }
107                                 else if('A' <= *string && *string <= 'F') {
108                                         ret *= 16;
109                                         ret += *string - 'A' + 10;
110                                 }
111                                 else if('a' <= *string && *string <= 'f') {
112                                         ret *= 16;
113                                         ret += *string - 'a' + 10;
114                                 }
115                                 else
116                                         break;
117                         }
118                 }
119                 else    // Octal
120                 {
121                         for( ; '0' <= *string && *string <= '7'; string ++ )
122                         {
123                                 ret *= 8;
124                                 ret += *string - '0';
125                         }
126                 }
127         }
128         else    // Decimal
129         {
130                 for( ; '0' <= *string && *string <= '9'; string++)
131                 {
132                         ret *= 10;
133                         ret += *string - '0';
134                 }
135                 // Error check
136                 if( ret == 0 )  return 0;
137         }
138         
139         if(bNeg)        ret = -ret;
140         
141         //Log("atoi: RETURN %i", ret);
142         
143         if(Val) *Val = ret;
144         
145         return string - orig_string;
146 }
147
148 static const char cUCDIGITS[] = "0123456789ABCDEF";
149 /**
150  * \fn void itoa(char *buf, Uint64 num, int base, int minLength, char pad)
151  * \brief Convert an integer into a character string
152  */
153 void itoa(char *buf, Uint64 num, int base, int minLength, char pad)
154 {
155         char    tmpBuf[64+1];
156          int    pos=0, i;
157         Uint64  rem;
158
159         buf[0] = 0;
160         ASSERTR(base >= 2, );
161         ASSERTR(base <= 16, );
162
163         // Sanity check
164         if(!buf)        return;
165         
166         // Convert 
167         while(num > base-1) {
168                 num = DivMod64U(num, base, &rem);       // Shift `num` and get remainder
169                 ASSERT(rem >= 0);
170                 if( rem >= base && base != 16 ) {
171                         Debug("rem(%llx) >= base(%x), num=%llx", rem, base, num);
172                 }
173                 ASSERT(rem < base);
174                 tmpBuf[pos] = cUCDIGITS[ rem ];
175                 pos++;
176         }
177         tmpBuf[pos++] = cUCDIGITS[ num ];               // Last digit of `num`
178         
179         // Put in reverse
180         i = 0;
181         minLength -= pos;
182         while(minLength-- > 0)  buf[i++] = pad;
183         while(pos-- > 0)                buf[i++] = tmpBuf[pos]; // Reverse the order of characters
184         buf[i] = 0;
185 }
186
187 /**
188  * \brief Append a character the the vsnprintf output
189  */
190 #define PUTCH(ch)       do { \
191                 if(pos < __maxlen && __s) { \
192                         __s[pos] = ch; \
193                 } else { \
194                         (void)ch;\
195                 } \
196                 pos ++; \
197         } while(0)
198 #define GETVAL()        do {\
199         if(isLongLong)  val = va_arg(args, Uint64);\
200         else    val = va_arg(args, unsigned int);\
201         }while(0)
202 /**
203  * \brief VArg String Number Print Formatted
204  */
205 int vsnprintf(char *__s, const size_t __maxlen, const char *__format, va_list args)
206 {
207         char    c, pad = ' ';
208          int    minSize = 0, precision = -1, len;
209         char    tmpBuf[34];     // For Integers
210         const char      *p = NULL;
211          int    isLongLong = 0, isLong;
212         Uint64  val;
213         size_t  pos = 0;
214         // Flags
215          int    bPadLeft = 0;
216
217         while((c = *__format++) != 0)
218         {
219                 // Non control character
220                 if(c != '%') { PUTCH(c); continue; }
221
222                 c = *__format++;
223                 if(c == '\0')   break;
224                 
225                 // Literal %
226                 if(c == '%') { PUTCH('%'); continue; }
227                 
228                 // Pointer - Done first for debugging
229                 if(c == 'p') {
230                         Uint    ptr = va_arg(args, Uint);
231                         PUTCH('*');     PUTCH('0');     PUTCH('x');
232                         for( len = BITS/4; len -- && ((ptr>>(len*4))&15) == 0; )
233                                 ;
234                         len ++;
235                         if( len == 0 )
236                                 PUTCH( '0' );
237                         else
238                                 while( len -- )
239                                         PUTCH( cUCDIGITS[ (ptr>>(len*4))&15 ] );
240                         continue ;
241                 }
242
243                 isLongLong = 0;
244                 isLong = 0;
245         
246                 // - Padding Side Flag
247                 if(c == '-') {
248                         bPadLeft = 1;
249                         c = *__format++;
250                 }
251                 
252                 // - Padding
253                 if(c == '0') {
254                         pad = '0';
255                         c = *__format++;
256                 }
257                 else
258                         pad = ' ';
259                 
260                 // - Minimum length
261                 if(c == '*') {  // Dynamic length
262                         minSize = va_arg(args, unsigned int);
263                         c = *__format++;
264                 }
265                 else if('1' <= c && c <= '9')
266                 {
267                         minSize = 0;
268                         while('0' <= c && c <= '9')
269                         {
270                                 minSize *= 10;
271                                 minSize += c - '0';
272                                 c = *__format++;
273                         }
274                 }
275                 else
276                         minSize = 0;
277                 
278                 // - Precision
279                 precision = -1;
280                 if( c == '.' ) {
281                         c = *__format++;
282                         
283                         if(c == '*') {  // Dynamic length
284                                 precision = va_arg(args, unsigned int);
285                                 c = *__format++;
286                         }
287                         else if('1' <= c && c <= '9')
288                         {
289                                 precision = 0;
290                                 while('0' <= c && c <= '9')
291                                 {
292                                         precision *= 10;
293                                         precision += c - '0';
294                                         c = *__format++;
295                                 }
296                         }
297                 }
298                 
299                 // - Default, Long or LongLong?
300                 isLongLong = 0;
301                 if(c == 'l')    // Long is actually the default on x86
302                 {
303                         isLong = 1;
304                         c = *__format++;
305                         if(c == 'l') {
306                                 c = *__format++;
307                                 isLongLong = 1;
308                         }
309                 }
310                 
311                 // - Now get the format code
312                 p = tmpBuf;
313                 switch(c)
314                 {
315                 case 'd':
316                 case 'i':
317                         GETVAL();
318                         if( isLongLong && val >> 63 ) {
319                                 PUTCH('-');
320                                 if( val == LLONG_MIN )
321                                         val = LLONG_MAX;
322                                 else
323                                         val = -val;
324                         }
325                         else if( !isLongLong && (val >> 31) ) {
326                                 PUTCH('-');
327                                 val = (~val & 0xFFFFFFFF)+1;
328                         }
329                         itoa(tmpBuf, val, 10, minSize, pad);
330                         goto printString;
331                 case 'u':       // Unsigned
332                         GETVAL();
333                         itoa(tmpBuf, val, 10, minSize, pad);
334                         goto printString;
335                 case 'P':       // Physical Address
336                         PUTCH('0');
337                         PUTCH('x');
338                         if(sizeof(tPAddr) > 4)  isLongLong = 1;
339                         GETVAL();
340                         itoa(tmpBuf, val, 16, minSize, pad);
341                         goto printString;
342                 case 'X':       // Hex
343                         if(BITS == 64)
344                                 isLongLong = 1; // TODO: Handle non-x86 64-bit archs
345                         GETVAL();
346                         itoa(tmpBuf, val, 16, minSize, pad);
347                         goto printString;
348                         
349                 case 'x':       // Lower case hex
350                         GETVAL();
351                         itoa(tmpBuf, val, 16, minSize, pad);
352                         goto printString;
353                 case 'o':       // Octal
354                         GETVAL();
355                         itoa(tmpBuf, val, 8, minSize, pad);
356                         goto printString;
357                 case 'b':
358                         GETVAL();
359                         itoa(tmpBuf, val, 2, minSize, pad);
360                         goto printString;
361
362                 case 'B':       //Boolean
363                         val = va_arg(args, unsigned int);
364                         if(val) p = "True";
365                         else    p = "False";
366                         goto printString;
367                 
368                 // String - Null Terminated Array
369                 case 's':
370                         if( isLong ) {
371                                 Uint16  *p16 = va_arg(args, Uint16*);
372                                 Uint8   tmp[5];
373                                 while( *p16 && precision-- ) {
374                                         Uint32  cp;
375                                         p16 += ReadUTF16(p16, &cp);
376                                         tmp[WriteUTF8(tmp, cp)] = 0;
377                                         for(int i = 0; tmp[i] && i<5; i ++)
378                                                 PUTCH(tmp[i]);
379                                 }
380                                 break;
381                         }
382                         p = va_arg(args, char*);        // Get Argument
383                         if( !p || !CheckString(p) )     p = "(inval)";  // Avoid #PFs  
384                 printString:
385                         if(!p)          p = "(null)";
386                         len = strlen(p);
387                         if( !bPadLeft ) while(len++ < minSize)  PUTCH(pad);
388                         while(*p && precision--) { PUTCH(*p); p++;} 
389                         if( bPadLeft )  while(len++ < minSize)  PUTCH(pad);
390                         break;
391                 
392                 case 'C':       // Non-Null Terminated Character Array
393                         p = va_arg(args, char*);
394                         if( !CheckMem(p, minSize) )     continue;       // No #PFs please
395                         if(!p)  goto printString;
396                         while(minSize--) {
397                                 PUTCH(*p);
398                                 p ++;
399                         }
400                         break;
401                 
402                 // Single Character
403                 case 'c':
404                 default:
405                         GETVAL();
406                         PUTCH( (Uint8)val );
407                         break;
408                 }
409         }
410         
411         if(__s && pos < __maxlen)
412                 __s[pos] = '\0';
413         
414         return pos;
415 }
416 #undef PUTCH
417
418 /**
419  */
420 int snprintf(char *__s, size_t __n, const char *__format, ...)
421 {
422         va_list args;
423          int    ret;
424         
425         va_start(args, __format);
426         ret = vsnprintf(__s, __n, __format, args);
427         va_end(args);
428         
429         return ret;
430 }
431
432 /**
433  */
434 int sprintf(char *__s, const char *__format, ...)
435 {
436         va_list args;
437          int    ret;
438         
439         va_start(args, __format);
440         ret = vsnprintf(__s, -1, __format, args);
441         va_end(args);
442         
443         return ret;
444 }
445
446 /*
447  * ==================
448  * ctype.h
449  * ==================
450  */
451 int isalnum(int c)
452 {
453         return isalpha(c) || isdigit(c);
454 }
455 int isalpha(int c)
456 {
457         return isupper(c) || islower(c);
458 }
459 int isascii(int c)
460 {
461         return (0 <= c && c < 128);
462 }
463 int isblank(int c)
464 {
465         if(c == '\t')   return 1;
466         if(c == ' ')    return 1;
467         return 0;
468 }
469 int iscntrl(int c)
470 {
471         // TODO: Check iscntrl
472         if(c < ' ')     return 1;
473         return 0;
474 }
475 int isdigit(int c)
476 {
477         return ('0' <= c && c <= '9');
478 }
479 int isgraph(int c)
480 {
481         // TODO: Check isgraph
482         return 0;
483 }
484 int islower(int c)
485 {
486         return ('a' <= c && c <= 'z');
487 }
488 int isprint(int c)
489 {
490         if( ' ' <= c && c <= 0x7F )     return 1;
491         return 0;
492 }
493 int ispunct(int c)
494 {
495         switch(c)
496         {
497         case '.':       case ',':
498         case '?':       case '!':
499                 return 1;
500         default:
501                 return 0;
502         }
503 }
504 int isspace(int c)
505 {
506         if(c == ' ')    return 1;
507         if(c == '\t')   return 1;
508         if(c == '\v')   return 1;
509         if(c == '\n')   return 1;
510         if(c == '\r')   return 1;
511         return 0;
512 }
513 int isupper(int c)
514 {
515         return ('A' <= c && c <= 'Z');
516 }
517 int isxdigit(int c)
518 {
519         return isdigit(c) || ('a' <= c && c <= 'f') || ('A' <= c && c <= 'F');
520 }
521
522 int toupper(int c)
523 {
524         if( islower(c) )
525                 return c - 0x20;
526         else
527                 return c;
528 }
529 int tolower(int c)
530 {
531         if( isupper(c) )
532                 return c + 0x20;
533         else
534                 return c;
535 }
536
537 /**
538  * \fn int strucmp(const char *Str1, const char *Str2)
539  * \brief Compare \a Str1 and \a Str2 case-insensitively
540  */
541 int strucmp(const char *Str1, const char *Str2)
542 {
543         while(*Str1 && tolower(*Str1) == tolower(*Str2))
544                 Str1++, Str2++;
545         return tolower(*Str1) - tolower(*Str2);
546 }
547
548 /**
549  * \brief Locate a byte in a string
550  */
551 char *strchr(const char *__s, int __c)
552 {
553         for( ; *__s; __s ++ )
554         {
555                 if( *__s == __c )       return (char*)__s;
556         }
557         return NULL;
558 }
559
560 char *strrchr(const char *__s, int __c)
561 {
562         size_t ofs = strlen(__s);
563         while(--ofs && __s[ofs] != __c);
564         if( __s[ofs] == __c )
565                 return (char*)__s + ofs;
566         return NULL;
567 }
568
569 /**
570  * \fn int strpos(const char *Str, char Ch)
571  * \brief Search a string for an ascii character
572  */
573 int strpos(const char *Str, char Ch)
574 {
575          int    pos;
576         for(pos=0;Str[pos];pos++)
577         {
578                 if(Str[pos] == Ch)      return pos;
579         }
580         return -1;
581 }
582
583 /**
584  * \fn size_t strlen(const char *__str)
585  * \brief Get the length of string
586  */
587 size_t strlen(const char *__str)
588 {
589         size_t  ret = 0;
590         while(*__str++) ret++;
591         return ret;
592 }
593
594 /**
595  * \brief Copy a string to a new location
596  */
597 char *strcpy(char *__str1, const char *__str2)
598 {
599         while(*__str2)
600                 *__str1++ = *__str2++;
601         *__str1 = '\0'; // Terminate String
602         return __str1;
603 }
604
605 /**
606  * \brief Copy a string to a new location
607  * \note Copies at most `max` chars
608  */
609 char *strncpy(char *__str1, const char *__str2, size_t __max)
610 {
611         while(*__str2 && __max-- >= 1)
612                 *__str1++ = *__str2++;
613         if(__max)
614                 *__str1 = '\0'; // Terminate String
615         return __str1;
616 }
617
618 /**
619  * \brief Append a string to another
620  */
621 char *strcat(char *__dest, const char *__src)
622 {
623         while(*__dest++);
624         __dest--;
625         while(*__src)
626                 *__dest++ = *__src++;
627         *__dest = '\0';
628         return __dest;
629 }
630
631 /**
632  * \brief Append at most \a n chars to a string from another
633  * \note At most n+1 chars are written (the dest is always zero terminated)
634  */
635 char *strncat(char *__dest, const char *__src, size_t n)
636 {
637         while(*__dest++);
638         while(*__src && n-- >= 1)
639                 *__dest++ = *__src++;
640         *__dest = '\0';
641         return __dest;
642 }
643
644 /**
645  * \fn int strcmp(const char *str1, const char *str2)
646  * \brief Compare two strings return the difference between
647  *        the first non-matching characters.
648  */
649 int strcmp(const char *str1, const char *str2)
650 {
651         while(*str1 && *str1 == *str2)
652                 str1++, str2++;
653         return *str1 - *str2;
654 }
655
656 /**
657  * \fn int strncmp(const char *Str1, const char *Str2, size_t num)
658  * \brief Compare strings \a Str1 and \a Str2 to a maximum of \a num characters
659  */
660 int strncmp(const char *Str1, const char *Str2, size_t num)
661 {
662         if(num == 0)    return 0;       // TODO: Check what should officially happen here
663         while(--num && *Str1 && *Str1 == *Str2)
664                 Str1++, Str2++;
665         return *Str1-*Str2;
666 }
667
668 #if 0
669 /**
670  * \fn char *strdup(const char *Str)
671  * \brief Duplicates a string
672  */
673 char *strdup(const char *Str)
674 {
675         char    *ret;
676         ret = malloc(strlen(Str)+1);
677         if( !ret )      return NULL;
678         strcpy(ret, Str);
679         return ret;
680 }
681 #else
682
683 /**
684  * \fn char *_strdup(const char *File, int Line, const char *Str)
685  * \brief Duplicates a string
686  */
687 char *_strdup(const char *File, int Line, const char *Str)
688 {
689         char    *ret;
690         ret = Heap_Allocate(File, Line, strlen(Str)+1);
691         if( !ret )      return NULL;
692         strcpy(ret, Str);
693         return ret;
694 }
695 #endif
696
697 /**
698  * \fn int rand()
699  * \brief Pseudo random number generator
700  */
701 int rand(void)
702 {
703         #if 0
704         static Uint     state = RANDOM_SEED;
705         Uint    old = state;
706         // Get the next state value
707         giRandomState = (RANDOM_A*state + RANDOM_C);
708         // Check if it has changed, and if it hasn't, change it
709         if(state == old)        state += RANDOM_SPRUCE;
710         return state;
711         #else
712         // http://en.wikipedia.org/wiki/Xorshift
713         // 2010-10-03
714         static Uint32   x = 123456789;
715         static Uint32   y = 362436069;
716         static Uint32   z = 521288629;
717         static Uint32   w = 88675123; 
718         Uint32  t;
719  
720         t = x ^ (x << 11);
721         x = y; y = z; z = w;
722         return w = w ^ (w >> 19) ^ t ^ (t >> 8); 
723         #endif
724 }
725
726 void *memmove(void *__dest, const void *__src, size_t len)
727 {
728         char    *dest = __dest;
729         const char      *src = __src;
730         void    *ret = __dest;
731
732         if( len == 0 || dest == src )
733                 return dest;
734         
735         if( (tVAddr)dest > (tVAddr)src + len )
736                 return memcpy(dest, src, len);
737         if( (tVAddr)dest + len < (tVAddr)src )
738                 return memcpy(dest, src, len);
739         
740         // NOTE: Assumes memcpy works forward
741         if( (tVAddr)dest < (tVAddr)src )
742                 return memcpy(dest, src, len);
743
744         #if 0
745         size_t  block_size;
746         if( (tVAddr)dest < (tVAddr)src )
747                 block_size = (tVAddr)src - (tVAddr)dest;
748         else
749                 block_size = (tVAddr)dest - (tVAddr)src;
750         
751         block_size &= ~0xF;
752         
753         while(len >= block_size)
754         {
755                 memcpy(dest, src, block_size);
756                 len -= block_size;
757                 dest += block_size;
758                 src += block_size;
759         }
760         memcpy(dest, src, len);
761         return ret;
762         #else
763         for( int i = len; i--; )
764         {
765                 dest[i] = src[i];
766         }
767         return ret;
768         #endif
769         
770 }
771
772 // NOTE: Strictly not libc, but lib.c is used by userland code too and hence these two
773 // can't be in it.
774 /**
775  * \name Memory Validation
776  * \{
777  */
778 /**
779  * \brief Checks if a string resides fully in valid memory
780  */
781 int CheckString(const char *String)
782 {
783         tVAddr  addr;
784          int    bUser;
785
786         addr = (tVAddr)String;
787
788         if( !MM_GetPhysAddr( (void*)addr ) )
789                 return 0;
790         
791         // Check 1st page
792         bUser = MM_IsUser( addr );
793         
794         while( *(char*)addr )
795         {
796                 if( (addr & (PAGE_SIZE-1)) == 0 )
797                 {
798                         if(bUser && !MM_IsUser(addr) )
799                                 return 0;
800                         if(!bUser && !MM_GetPhysAddr((void*)addr) )
801                                 return 0;
802                 }
803                 addr ++;
804         }
805         return 1;
806 }
807
808 /**
809  * \brief Check if a sized memory region is valid memory
810  * \return Boolean success
811  */
812 int CheckMem(const void *Mem, int NumBytes)
813 {
814         return MM_IsValidBuffer( (tVAddr)Mem, NumBytes );
815 }
816 /* *
817  * \}
818  */
819

UCC git Repository :: git.ucc.asn.au