Usermode/libc - scanf() and many other cleanups
[tpg/acess2.git] / Usermode / Libraries / libc.so_src / fileIO.c
index c1a47ff..af8f007 100644 (file)
@@ -1,6 +1,7 @@
 /*\r
-AcessOS Basic C Library\r
-*/\r
+ * AcessOS Basic C Library\r
+ * stdio.c\r
+ */\r
 #include "config.h"\r
 #include <acess/sys.h>\r
 #include <stdlib.h>\r
@@ -9,6 +10,8 @@ AcessOS Basic C Library
 #include "lib.h"\r
 #include "stdio_int.h"\r
 \r
+#define WRITE_STR(_fd, _str)   write(_fd, _str, sizeof(_str))\r
+\r
 #define DEBUG_BUILD    0\r
 \r
 // === CONSTANTS ===\r
@@ -16,89 +19,117 @@ AcessOS Basic C Library
 #define        _stdout 1\r
 \r
 // === PROTOTYPES ===\r
-EXPORT void    itoa(char *buf, unsigned long num, int base, int minLength, char pad);\r
+EXPORT void    itoa(char *buf, uint64_t num, size_t base, int minLength, char pad, int bSigned);\r
 struct sFILE   *get_file_struct();\r
 \r
 // === GLOBALS ===\r
 struct sFILE   _iob[STDIO_MAX_STREAMS];        // IO Buffer\r
-struct sFILE   *stdin = &_iob[0];      // Standard Input\r
-struct sFILE   *stdout = &_iob[1];     // Standard Output\r
-struct sFILE   *stderr = &_iob[2];     // Standard Error\r
+struct sFILE   *stdin; // Standard Input\r
+struct sFILE   *stdout;        // Standard Output\r
+struct sFILE   *stderr;        // Standard Error\r
+///\note Initialised in SoMain\r
 \r
 // === CODE ===\r
-/**\r
- * \fn FILE *freopen(FILE *fp, char *file, char *mode)\r
- */\r
-EXPORT FILE *freopen(FILE *fp, char *file, char *mode)\r
+int _fopen_modetoflags(const char *mode)\r
 {\r
-        int    openFlags = 0;\r
-        int    i;\r
-       \r
-       // Sanity Check Arguments\r
-       if(!fp || !file || !mode)       return NULL;\r
-       \r
-       if(fp->Flags) {\r
-               fflush(fp);\r
-               close(fp->FD);\r
-       }\r
+       int flags = 0;\r
        \r
        // Get main mode\r
-       switch(mode[0])\r
+       switch(*mode)\r
        {\r
-       case 'r':       fp->Flags = FILE_FLAG_MODE_READ;        break;\r
-       case 'w':       fp->Flags = FILE_FLAG_MODE_WRITE;       break;\r
-       case 'a':       fp->Flags = FILE_FLAG_MODE_APPEND;      break;\r
-       case 'x':       fp->Flags = FILE_FLAG_MODE_EXEC;        break;\r
+       case 'r':       flags = FILE_FLAG_MODE_READ;    break;\r
+       case 'w':       flags = FILE_FLAG_MODE_WRITE;   break;\r
+       case 'a':       flags = FILE_FLAG_MODE_APPEND;  break;\r
+       case 'x':       flags = FILE_FLAG_MODE_EXEC;    break;  // Acess addon\r
        default:\r
-               return NULL;\r
+               return -1;\r
        }\r
+       mode ++;\r
+\r
        // Get Modifiers\r
-       for(i=1;mode[i];i++)\r
+       for( ; *mode; mode ++ )\r
        {\r
-               switch(mode[i])\r
+               switch(*mode)\r
                {\r
-               case '+':       fp->Flags |= FILE_FLAG_M_EXT;\r
+               case 'b':       flags |= FILE_FLAG_M_BINARY;    break;\r
+               case '+':       flags |= FILE_FLAG_M_EXT;       break;\r
+               default:\r
+                       return -1;\r
                }\r
        }\r
        \r
+       return flags;\r
+}\r
+\r
+/**\r
+ * \fn FILE *freopen(char *file, char *mode, FILE *fp)\r
+ */\r
+EXPORT FILE *freopen(const char *file, const char *mode, FILE *fp)\r
+{\r
+        int    openFlags = 0;\r
+       \r
+       // Sanity Check Arguments\r
+       if(!fp || !file || !mode)       return NULL;\r
+       \r
+       if(fp->FD != -1) {\r
+               fflush(fp);\r
+       }\r
+\r
+       // Get stdio flags\r
+       fp->Flags = _fopen_modetoflags(mode);\r
+       if(fp->Flags == -1)\r
+               return NULL;\r
+       \r
        // Get Open Flags\r
-       switch(mode[0])\r
+       switch(fp->Flags & FILE_FLAG_MODE_MASK)\r
        {\r
        // Read\r
-       case 'r':       openFlags = OPENFLAG_READ;\r
+       case FILE_FLAG_MODE_READ:\r
+               openFlags = OPENFLAG_READ;\r
                if(fp->Flags & FILE_FLAG_M_EXT)\r
                        openFlags |= OPENFLAG_WRITE;\r
                break;\r
        // Write\r
-       case 'w':       openFlags = OPENFLAG_WRITE;\r
+       case FILE_FLAG_MODE_WRITE:\r
+               openFlags = OPENFLAG_WRITE;\r
+               if(fp->Flags & FILE_FLAG_M_EXT)\r
+                       openFlags |= OPENFLAG_READ;\r
+               break;\r
+       // Execute\r
+       case FILE_FLAG_MODE_APPEND:\r
+               openFlags = OPENFLAG_APPEND;\r
                if(fp->Flags & FILE_FLAG_M_EXT)\r
                        openFlags |= OPENFLAG_READ;\r
                break;\r
        // Execute\r
-       case 'x':       openFlags = OPENFLAG_EXEC;\r
+       case FILE_FLAG_MODE_EXEC:\r
+               openFlags = OPENFLAG_EXEC;\r
                break;\r
        }\r
-       \r
+\r
        //Open File\r
-       fp->FD = reopen(fp->FD, file, openFlags);\r
+       if(fp->FD != -1)\r
+               fp->FD = reopen(fp->FD, file, openFlags);\r
+       else\r
+               fp->FD = open(file, openFlags);\r
        if(fp->FD == -1) {\r
                fp->Flags = 0;\r
                return NULL;\r
        }\r
        \r
-       if(mode[0] == 'a') {\r
+       if( (fp->Flags & FILE_FLAG_MODE_MASK) == FILE_FLAG_MODE_APPEND ) {\r
                seek(fp->FD, 0, SEEK_END);      //SEEK_END\r
        }\r
        \r
        return fp;\r
 }\r
 /**\r
- \fn FILE *fopen(char *file, char *mode)\r
+ \fn FILE *fopen(const char *file, const char *mode)\r
  \brief Opens a file and returns the pointer\r
  \param file   String - Filename to open\r
  \param mode   Mode to open in\r
 */\r
-EXPORT FILE *fopen(char *file, char *mode)\r
+EXPORT FILE *fopen(const char *file, const char *mode)\r
 {\r
        FILE    *retFile;\r
        \r
@@ -108,45 +139,141 @@ EXPORT FILE *fopen(char *file, char *mode)
        // Create Return Structure\r
        retFile = get_file_struct();\r
        \r
-       return freopen(retFile, file, mode);\r
+       return freopen(file, mode, retFile);\r
 }\r
 \r
-EXPORT void fclose(FILE *fp)\r
+EXPORT FILE *fmemopen(void *buffer, size_t length, const char *mode)\r
 {\r
-       close(fp->FD);\r
-       free(fp);\r
+       FILE    *ret;\r
+       \r
+       if( !buffer || !mode )  return NULL;\r
+       \r
+       ret = get_file_struct();\r
+       \r
+       ret->FD = -2;\r
+       ret->Flags = _fopen_modetoflags(mode);\r
+       if(ret->Flags == -1) {\r
+               ret->Flags = 0;\r
+               return NULL;\r
+       }\r
+       \r
+       ret->Buffer = buffer;\r
+       ret->BufferStart = 0;\r
+       ret->BufferSize = length;\r
+       \r
+       return ret;\r
+}\r
+\r
+EXPORT int fclose(FILE *fp)\r
+{\r
+       fflush(fp);\r
+       if( fp->FD != -1 ) {\r
+               close(fp->FD);\r
+       }\r
+       fp->Flags = 0;\r
+       fp->FD = -1;\r
+       return 0;\r
 }\r
 \r
 EXPORT void fflush(FILE *fp)\r
 {\r
-       ///\todo Implement\r
+       if( !fp || fp->FD == -1 )\r
+               return ;\r
+       \r
+       if( !(fp->Flags & FILE_FLAG_DIRTY) )\r
+               return ;\r
+       \r
+       // Nothing to do for memory files\r
+       if( fp->FD == -2 )\r
+               return ;\r
 }\r
 \r
+EXPORT void clearerr(FILE *fp)\r
+{\r
+       if( !fp || fp->FD == -1 )\r
+               return ;\r
+       \r
+       // TODO: Impliment clearerr()\r
+}\r
+\r
+EXPORT int feof(FILE *fp)\r
+{\r
+       if( !fp || fp->FD == -1 )\r
+               return 0;\r
+       return !!(fp->Flags & FILE_FLAG_EOF);\r
+}\r
+\r
+EXPORT int ferror(FILE *fp)\r
+{\r
+       if( !fp || fp->FD == -1 )\r
+               return 0;\r
+       return 0;\r
+}\r
+EXPORT int fileno(FILE *stream)\r
+{\r
+       return stream->FD;\r
+}\r
+\r
+EXPORT off_t ftell(FILE *fp)\r
+{\r
+       if(!fp || fp->FD == -1) return -1;\r
+\r
+       if( fp->FD == -2 )\r
+               return fp->Pos; \r
+       else\r
+               return tell(fp->FD);\r
+}\r
+\r
+EXPORT int fseek(FILE *fp, long int amt, int whence)\r
+{\r
+       if(!fp || fp->FD == -1) return -1;\r
+\r
+       if( fp->FD == -2 ) {\r
+               switch(whence)\r
+               {\r
+               case SEEK_CUR:\r
+                       fp->Pos += amt;\r
+                       break;\r
+               case SEEK_SET:\r
+                       fp->Pos = amt;\r
+                       break;\r
+               case SEEK_END:\r
+                       if( fp->BufferSize < (size_t)amt )\r
+                               fp->Pos = 0;\r
+                       else\r
+                               fp->Pos = fp->BufferSize - amt;\r
+                       break;\r
+               }\r
+               if(fp->Pos > (off_t)fp->BufferSize) {\r
+                       fp->Pos = fp->BufferSize;\r
+                       fp->Flags |= FILE_FLAG_EOF;\r
+               }\r
+               return 0;\r
+       }\r
+       else\r
+               return seek(fp->FD, amt, whence);\r
+}\r
+\r
+\r
 /**\r
- * \fn int fprintfv(FILE *fp, const char *format, va_list args)\r
+ * \fn EXPORT int vfprintf(FILE *fp, const char *format, va_list args)\r
  * \brief Print to a file from a variable argument list\r
  */\r
-EXPORT int fprintfv(FILE *fp, const char *format, va_list args)\r
+EXPORT int vfprintf(FILE *fp, const char *format, va_list args)\r
 {\r
-       va_list tmpList = args;\r
+       va_list tmpList;\r
         int    size;\r
-       char    *buf;\r
-        \r
+\r
        if(!fp || !format)      return -1;\r
+\r
+       va_copy(tmpList, args);\r
        \r
-       size = ssprintfv((char*)format, tmpList);\r
-       \r
-       buf = (char*)malloc(size+1);\r
-       buf[size] = '\0';\r
-       \r
-       // Print\r
-       sprintfv(buf, (char*)format, args);\r
+       size = vsnprintf(NULL, 0, (char*)format, tmpList);\r
+       char    buf[size+1];\r
+       vsnprintf(buf, size+1, (char*)format, args);\r
        \r
        // Write to stream\r
-       write(fp->FD, size+1, buf);\r
-       \r
-       // Free buffer\r
-       free(buf);\r
+       fwrite(buf, size, 1, fp);\r
        \r
        // Return written byte count\r
        return size;\r
@@ -163,32 +290,83 @@ EXPORT int fprintf(FILE *fp, const char *format, ...)
        \r
        // Get Size\r
        va_start(args, format);\r
-       ret = fprintfv(fp, (char*)format, args);\r
+       ret = vfprintf(fp, (char*)format, args);\r
        va_end(args);\r
        \r
        return ret;\r
 }\r
 \r
 /**\r
+ * \fn EXPORT size_t fwrite(void *ptr, size_t size, size_t num, FILE *fp)\r
+ * \brief Write to a stream\r
  */\r
 EXPORT size_t fwrite(void *ptr, size_t size, size_t num, FILE *fp)\r
 {\r
-        int    ret;\r
-       if(!fp || !fp->FD)      return -1;\r
+       size_t  ret;\r
        \r
-       ret = write(fp->FD, size*num, ptr);\r
+       if(!fp || fp->FD == -1)\r
+               return -1;\r
+\r
+       if( fp->FD == -2 ) {\r
+               size_t  avail = (fp->BufferSize - fp->Pos) / size;\r
+               if( avail == 0 )\r
+                       fp->Flags |= FILE_FLAG_EOF;\r
+               if( num > avail )       num = avail;\r
+               size_t  bytes = num * size;\r
+               memcpy((char*)fp->Buffer + fp->Pos, ptr, bytes);\r
+               fp->Pos += bytes;\r
+               ret = num;\r
+       }\r
+       else {  \r
+               ret = write(fp->FD, ptr, size*num);\r
+               ret /= size;\r
+       }\r
        \r
        return ret;\r
 }\r
 \r
+/**\r
+ * \fn EXPORT size_t fread(void *ptr, size_t size, size_t num, FILE *fp)\r
+ * \brief Read from a stream\r
+ */\r
+EXPORT size_t fread(void *ptr, size_t size, size_t num, FILE *fp)\r
+{\r
+       size_t  ret;\r
+       \r
+       if(!fp || fp->FD == -1)\r
+               return -1;\r
+\r
+       if( fp->FD == -2 ) {\r
+               size_t  avail = (fp->BufferSize - fp->Pos) / size;\r
+               if( avail == 0 )\r
+                       fp->Flags |= FILE_FLAG_EOF;\r
+               if( num > avail )       num = avail;\r
+               size_t  bytes = num * size;\r
+               memcpy(ptr, (char*)fp->Buffer + fp->Pos, bytes);\r
+               fp->Pos += bytes;\r
+               ret = num;\r
+       }\r
+       else {\r
+               ret = read(fp->FD, ptr, size*num);\r
+               ret /= size;\r
+       }\r
+               \r
+       return ret;\r
+}\r
+\r
 /**\r
  * \fn EXPORT int fputc(int c, FILE *fp)\r
  * \brief Write a single character to the stream\r
  */\r
 EXPORT int fputc(int c, FILE *fp)\r
 {\r
-       if(!fp || !fp->FD)      return -1;\r
-       return write(fp->FD, 1, &c);\r
+       return fwrite(&c, 1, 1, fp);\r
+}\r
+\r
+EXPORT int putchar(int c)\r
+{\r
+       c &= 0xFF;\r
+       return write(_stdout, &c, 1);\r
 }\r
 \r
 /**\r
@@ -197,9 +375,16 @@ EXPORT int fputc(int c, FILE *fp)
  */\r
 EXPORT int fgetc(FILE *fp)\r
 {\r
-        int    ret = 0;\r
-       if(!fp) return -1;\r
-       if(read(fp->FD, 1, &ret) == -1) return -1;\r
+       char    ret = 0;\r
+       if( fread(&ret, 1, 1, fp) == (size_t)-1 )\r
+               return -1;\r
+       return ret;\r
+}\r
+\r
+EXPORT int getchar(void)\r
+{\r
+       char    ret = 0;\r
+       if(read(_stdin, &ret, 1) != 1)  return -1;\r
        return ret;\r
 }\r
 \r
@@ -213,242 +398,266 @@ FILE *get_file_struct()
         int    i;\r
        for(i=0;i<STDIO_MAX_STREAMS;i++)\r
        {\r
-               if(_iob[i].Flags == 0)  return &_iob[i];\r
+               if(_iob[i].Flags & FILE_FLAG_ALLOC)\r
+                       continue ;\r
+               _iob[i].Flags |= FILE_FLAG_ALLOC;\r
+               _iob[i].FD = -1;\r
+               _iob[i].Pos = 0;\r
+               return &_iob[i];\r
        }\r
        return NULL;\r
 }\r
 \r
-EXPORT int putchar(int ch)\r
-{\r
-       return write(_stdout, 1, (char*)&ch);\r
-}\r
-\r
-EXPORT int     puts(const char *str)\r
+EXPORT int puts(const char *str)\r
 {\r
         int    len;\r
        \r
        if(!str)        return 0;\r
        len = strlen(str);\r
        \r
-       len = write(_stdout, len, (char*)str);\r
-       write(_stdout, 1, "\n");\r
+       len = write(_stdout, str, len);\r
+       write(_stdout, "\n", 1);\r
        return len;\r
 }\r
 \r
+EXPORT int vsprintf(char * __s, const char *__format, va_list __args)\r
+{\r
+       return vsnprintf(__s, 0x7FFFFFFF, __format, __args);\r
+}\r
+\r
 //sprintfv\r
 /**\r
\fn EXPORT void sprintfv(char *buf, const char *format, va_list args)\r
- \brief Prints a formatted string to a buffer\r
\param buf    Pointer - Destination Buffer\r
\param format String - Format String\r
\param args   VarArgs List - Arguments\r
-*/\r
-EXPORT void sprintfv(char *buf, const char *format, va_list args)\r
* \fn EXPORT void vsnprintf(char *buf, const char *format, va_list args)\r
\brief Prints a formatted string to a buffer\r
* \param buf  Pointer - Destination Buffer\r
* \param format       String - Format String\r
* \param args VarArgs List - Arguments\r
+ */\r
+EXPORT int vsnprintf(char *buf, size_t __maxlen, const char *format, va_list args)\r
 {\r
-       char    tmp[33];\r
-        int    c, arg, minSize;\r
-        int    pos = 0;\r
+       char    tmp[65];\r
+        int    c, minSize, precision, len;\r
+       size_t  pos = 0;\r
        char    *p;\r
        char    pad;\r
+       uint64_t        arg;\r
+        int    bLongLong, bPadLeft;\r
+\r
+       void _addchar(char ch)\r
+       {\r
+               if(buf && pos < __maxlen)       buf[pos] = ch;\r
+               pos ++;\r
+       }\r
 \r
        tmp[32] = '\0';\r
        \r
        while((c = *format++) != 0)\r
        {\r
-               //SysDebug("c = '%c'\n", c);\r
+               // Non-control character\r
                if (c != '%') {\r
-                       buf[pos++] = c;\r
+                       _addchar(c);\r
                        continue;\r
                }\r
                \r
+               // Control Character\r
                c = *format++;\r
-               if(c == '%') {\r
-                       buf[pos++] = '%';\r
+               if(c == '%') {  // Literal %\r
+                       _addchar('%');\r
                        continue;\r
                }\r
                \r
-               // Padding\r
+               bPadLeft = 0;\r
+               bLongLong = 0;\r
+               minSize = 0;\r
+               precision = -1;\r
+               pad = ' ';\r
+               \r
+               // Padding Character\r
                if(c == '0') {\r
                        pad = '0';\r
                        c = *format++;\r
-               } else\r
-                       pad = ' ';\r
-               minSize = 0;\r
-               if('1' <= c && c <= '9')\r
-               {\r
-                       while('0' <= c && c <= '9')\r
+               }\r
+               // Padding length\r
+               if( c == '*' ) {\r
+                       // Variable length\r
+                       minSize = va_arg(args, size_t);\r
+                       c = *format++;\r
+               }\r
+               else {\r
+                       if('1' <= c && c <= '9')\r
                        {\r
-                               minSize *= 10;\r
-                               minSize += c - '0';\r
-                               c = *format++;\r
+                               minSize = 0;\r
+                               while('0' <= c && c <= '9')\r
+                               {\r
+                                       minSize *= 10;\r
+                                       minSize += c - '0';\r
+                                       c = *format++;\r
+                               }\r
                        }\r
                }\r
-       \r
-               p = tmp;\r
-       \r
-               // Get Argument\r
-               arg = va_arg(args, int);\r
-               // Get Type\r
-               switch (c) {\r
-               case 'd':\r
-               case 'i':\r
-                       if(arg < 0) {\r
-                               buf[pos++] = '-';\r
-                               arg = -arg;\r
-                       }\r
-                       itoa(tmp, arg, 10, minSize, pad);\r
-                       goto sprintf_puts;\r
-               //      break;\r
-               case 'u':\r
-                       itoa(tmp, arg, 10, minSize, pad);\r
-                       goto sprintf_puts;\r
-               //      break;\r
-               case 'x':\r
-                       itoa(tmp, arg, 16, minSize, pad);\r
-                       goto sprintf_puts;\r
-               //      break;\r
-               case 'o':\r
-                       itoa(tmp, arg, 8, minSize, pad);\r
-                       goto sprintf_puts;\r
-               //      break;\r
-               case 'b':\r
-                       itoa(tmp, arg, 2, minSize, pad);\r
-                       goto sprintf_puts;\r
-               //      break;\r
 \r
-               case 's':\r
-                       p = (void*)arg;\r
-               sprintf_puts:\r
-                       if(!p)  p = "(null)";\r
-                       while(*p)       buf[pos++] = *p++;\r
-                       break;\r
-\r
-               default:\r
-                       buf[pos++] = arg;\r
-                       break;\r
+               // Precision\r
+               if(c == '.') {\r
+                       c = *format++;\r
+                       if(c == '*') {\r
+                               precision = va_arg(args, size_t);\r
+                               c = *format++;\r
+                       }\r
+                       else if('1' <= c && c <= '9')\r
+                       {\r
+                               precision = 0;\r
+                               while('0' <= c && c <= '9')\r
+                               {\r
+                                       precision *= 10;\r
+                                       precision += c - '0';\r
+                                       c = *format++;\r
+                               }\r
+                       }\r
                }\r
-    }\r
-       buf[pos++] = '\0';\r
-}\r
-/*\r
-ssprintfv\r
-- Size, Stream, Print Formated, Variable Argument List\r
-*/\r
-/**\r
- \fn EXPORT int ssprintfv(char *format, va_list args)\r
- \brief Gets the total character count from a formatted string\r
- \param format String - Format String\r
- \param args   VarArgs - Argument List\r
-*/\r
-EXPORT int ssprintfv(char *format, va_list args)\r
-{\r
-       char    tmp[33];\r
-        int    c, arg, minSize;\r
-        int    len = 0;\r
-       char    *p;\r
-       char    pad;\r
-\r
-       tmp[32] = '\0';\r
        \r
-       while((c = *format++) != 0)\r
-       {\r
-               if (c != '%') {\r
-                       len++;\r
-                       continue;\r
-               }\r
-               \r
-               c = *format++;\r
-               \r
-               // Literal '%'\r
-               if(c == '%') {\r
-                       len++;\r
-                       continue;\r
-               }\r
-               \r
-               // Padding\r
-               if(c == '0') {\r
-                       pad = '0';\r
-                       c = *format++;\r
-               } else\r
-                       pad = ' ';\r
-               minSize = 0;\r
-               if('1' <= c && c <= '9')\r
+               // Check for long long\r
+               bLongLong = 0;\r
+               if(c == 'l')\r
                {\r
-                       while('0' <= c && c <= '9')\r
-                       {\r
-                               minSize *= 10;\r
-                               minSize += c - '0';\r
-                               c = *format++;\r
+                       c = *format++;\r
+                       if(c == 'l') {\r
+                               bLongLong = 1;\r
                        }\r
                }\r
                \r
+               // Just help things along later\r
                p = tmp;\r
-               arg = va_arg(args, int);\r
-               switch (c) {                    \r
-               case 'd':\r
-               case 'i':\r
-                       if(arg < 0) {\r
-                               len ++;\r
-                               arg = -arg;\r
-                       }\r
-                       itoa(tmp, arg, 10, minSize, pad);\r
+               \r
+               // Get Type\r
+               switch( c )\r
+               {\r
+               // Signed Integer\r
+               case 'd':       case 'i':\r
+                       // Get Argument\r
+                       if(bLongLong)   arg = va_arg(args, int64_t);\r
+                       else                    arg = va_arg(args, int32_t);\r
+                       itoa(tmp, arg, 10, minSize, pad, 1);\r
+                       precision = -1;\r
                        goto sprintf_puts;\r
+               \r
+               // Unsigned Integer\r
                case 'u':\r
-                       itoa(tmp, arg, 10, minSize, pad);\r
+                       // Get Argument\r
+                       if(bLongLong)   arg = va_arg(args, uint64_t);\r
+                       else                    arg = va_arg(args, uint32_t);\r
+                       itoa(tmp, arg, 10, minSize, pad, 0);\r
+                       precision = -1;\r
+                       goto sprintf_puts;\r
+               \r
+               // Pointer\r
+               case 'p':\r
+                       _addchar('*');\r
+                       _addchar('0');\r
+                       _addchar('x');\r
+                       arg = va_arg(args, intptr_t);\r
+                       itoa(tmp, arg, 16, minSize, pad, 0);\r
+                       precision = -1;\r
                        goto sprintf_puts;\r
+               // Unsigned Hexadecimal\r
                case 'x':\r
-                       itoa(tmp, arg, 16, minSize, pad);\r
+                       if(bLongLong)   arg = va_arg(args, uint64_t);\r
+                       else                    arg = va_arg(args, uint32_t);\r
+                       itoa(tmp, arg, 16, minSize, pad, 0);\r
+                       precision = -1;\r
                        goto sprintf_puts;\r
+               \r
+               // Unsigned Octal\r
                case 'o':\r
-                       itoa(tmp, arg, 8, minSize, pad);\r
-                       p = tmp;\r
+                       if(bLongLong)   arg = va_arg(args, uint64_t);\r
+                       else                    arg = va_arg(args, uint32_t);\r
+                       itoa(tmp, arg, 8, minSize, pad, 0);\r
+                       precision = -1;\r
                        goto sprintf_puts;\r
+               \r
+               // Unsigned binary\r
                case 'b':\r
-                       itoa(tmp, arg, 2, minSize, pad);\r
+                       if(bLongLong)   arg = va_arg(args, uint64_t);\r
+                       else                    arg = va_arg(args, uint32_t);\r
+                       itoa(tmp, arg, 2, minSize, pad, 0);\r
+                       precision = -1;\r
                        goto sprintf_puts;\r
 \r
+               // String\r
                case 's':\r
-                       p = (char*)arg;\r
+                       p = va_arg(args, char*);\r
                sprintf_puts:\r
                        if(!p)  p = "(null)";\r
-                       while(*p)       len++, p++;\r
+                       //_SysDebug("vsnprintf: p = '%s'", p);\r
+                       if(precision >= 0)\r
+                               len = strnlen(p, precision);\r
+                       else\r
+                               len = strlen(p);\r
+                       if(bPadLeft)    while(minSize > len++)  _addchar(pad);\r
+                       while( *p ) {\r
+                               if(precision >= 0 && precision -- == 0)\r
+                                       break;\r
+                               _addchar(*p++);\r
+                       }\r
+                       if(!bPadLeft)   while(minSize > len++)  _addchar(pad);\r
                        break;\r
 \r
+               // Unknown, just treat it as a character\r
                default:\r
-                       len ++;\r
+                       arg = va_arg(args, uint32_t);\r
+                       _addchar(arg);\r
                        break;\r
                }\r
-    }\r
-       return len;\r
+       }\r
+       _addchar('\0');\r
+       pos --;\r
+       \r
+       //_SysDebug("vsnprintf: buf = '%s'", buf);\r
+       \r
+       return pos;\r
 }\r
 \r
 const char cUCDIGITS[] = "0123456789ABCDEF";\r
 /**\r
- * \fn static void itoa(char *buf, unsigned long num, int base, int minLength, char pad)\r
+ * \fn static void itoa(char *buf, uint64_t num, int base, int minLength, char pad, int bSigned)\r
  * \brief Convert an integer into a character string\r
+ * \param buf  Destination Buffer\r
+ * \param num  Number to convert\r
+ * \param base Base-n number output\r
+ * \param minLength    Minimum length of output\r
+ * \param pad  Padding used to ensure minLength\r
+ * \param bSigned      Signed number output?\r
  */\r
-EXPORT void itoa(char *buf, unsigned long num, int base, int minLength, char pad)\r
+EXPORT void itoa(char *buf, uint64_t num, size_t base, int minLength, char pad, int bSigned)\r
 {\r
-       char    tmpBuf[32];\r
+       char    tmpBuf[64];\r
         int    pos=0, i;\r
 \r
        if(!buf)        return;\r
-       if(base > 16) {\r
+       if(base > 16 || base < 2) {\r
                buf[0] = 0;\r
                return;\r
        }\r
        \r
+       if(bSigned && (int64_t)num < 0)\r
+       {\r
+               num = -num;\r
+               bSigned = 1;\r
+       } else\r
+               bSigned = 0;\r
+       \r
+       // Encode into reversed string\r
        while(num > base-1) {\r
-               tmpBuf[pos] = cUCDIGITS[ num % base ];\r
-               num = (long) num / base;                //Shift {number} right 1 digit\r
-               pos++;\r
+               tmpBuf[pos++] = cUCDIGITS[ num % base ];\r
+               num = (uint64_t) num / (uint64_t)base;          // Shift {number} right 1 digit\r
        }\r
 \r
-       tmpBuf[pos++] = cUCDIGITS[ num % base ];                //Last digit of {number}\r
+       tmpBuf[pos++] = cUCDIGITS[ num % base ];                // Last digit of {number}\r
+       if(bSigned)     tmpBuf[pos++] = '-';    // Append sign symbol if needed\r
+       \r
        i = 0;\r
        minLength -= pos;\r
        while(minLength-- > 0)  buf[i++] = pad;\r
-       while(pos-- > 0)                buf[i++] = tmpBuf[pos]; //Reverse the order of characters\r
+       while(pos-- > 0)                buf[i++] = tmpBuf[pos]; // Reverse the order of characters\r
        buf[i] = 0;\r
 }\r
 \r
@@ -458,46 +667,52 @@ EXPORT void itoa(char *buf, unsigned long num, int base, int minLength, char pad
  */\r
 EXPORT int printf(const char *format, ...)\r
 {\r
-       #if 0\r
         int    size;\r
-       char    *buf;\r
        va_list args;\r
        \r
+       // Get final size\r
        va_start(args, format);\r
-       size = ssprintfv((char*)format, args);\r
+       size = vsnprintf(NULL, 0, (char*)format, args);\r
        va_end(args);\r
-       \r
-       buf = (char*)malloc(size+1);\r
-       buf[size] = '\0';\r
-       \r
+       char buf[size+1];\r
+       // Fill Buffer\r
        va_start(args, format);\r
-       sprintfv(buf, (char*)format, args);\r
+       vsnprintf(buf, size+1, (char*)format, args);\r
        va_end(args);\r
        \r
+       // Send to stdout\r
+       write(_stdout, buf, size+1);\r
        \r
-       write(_stdout, size+1, buf);\r
-       \r
+       // Free buffer\r
        free(buf);\r
+       // Return\r
        return size;\r
-       #endif\r
-       \r
+}\r
+\r
+/**\r
+ * \fn EXPORT int sprintf(const char *buf, char *format, ...)\r
+ * \brief Print a formatted string to a buffer\r
+ */\r
+EXPORT int sprintf(char *buf, const char *format, ...)\r
+{\r
         int    ret;\r
        va_list args;\r
        va_start(args, format);\r
-       ret = fprintfv(stdout, (char*)format, args);\r
+       ret = vsprintf((char*)buf, (char*)format, args);\r
        va_end(args);\r
        return ret;\r
 }\r
 \r
 /**\r
- * \fn EXPORT int sprintf(const char *buf, char *format, ...)\r
+ * \fn EXPORT int snprintf(const char *buf, size_t maxlen, char *format, ...)\r
  * \brief Print a formatted string to a buffer\r
  */\r
-EXPORT int sprintf(char *buf, const char *format, ...)\r
+EXPORT int snprintf(char *buf, size_t maxlen, const char *format, ...)\r
 {\r
+        int    ret;\r
        va_list args;\r
        va_start(args, format);\r
-       sprintfv((char*)buf, (char*)format, args);\r
+       ret = vsnprintf((char*)buf, maxlen, (char*)format, args);\r
        va_end(args);\r
-       return 1;\r
+       return ret;\r
 }\r

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