Usermode/libc #6 Fix string.h functions, add some more unit tests
[tpg/acess2.git] / Usermode / Libraries / libc.so_src / stdio.c
index 7082f23..55fd521 100644 (file)
@@ -14,6 +14,9 @@
 \r
 #define DEBUG_BUILD    0\r
 \r
+#define LOG_WARN(f,...)        _SysDebug("WARN: %s: "f, __func__ ,## __VA_ARGS__)\r
+#define LOG_NOTICE(f,...)      _SysDebug("NOTE: %s: "f, __func__ ,## __VA_ARGS__)\r
+\r
 // === CONSTANTS ===\r
 #define        _stdin  0\r
 #define        _stdout 1\r
@@ -27,9 +30,9 @@ struct sFILE  *get_file_struct();
 \r
 // === GLOBALS ===\r
 struct sFILE   _iob[STDIO_MAX_STREAMS];        // IO Buffer\r
-struct sFILE   *stdin; // Standard Input\r
-struct sFILE   *stdout;        // Standard Output\r
-struct sFILE   *stderr;        // Standard Error\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
 ///\note Initialised in SoMain\r
 static const int STDIN_BUFSIZ = 512;\r
 static const int STDOUT_BUFSIZ = 512;\r
@@ -38,19 +41,16 @@ static const int STDOUT_BUFSIZ = 512;
 void _stdio_init(void)\r
 {\r
        // Init FileIO Pointers\r
-       stdin = &_iob[0];\r
        stdin->FD = 0;\r
-       stdin->Flags = FILE_FLAG_ALLOC|FILE_FLAG_MODE_READ|FILE_FLAG_LINEBUFFERED;\r
+       stdin->Flags = FILE_FLAG_ALLOC|FILE_FLAG_MODE_READ|FILE_FLAG_LINEBUFFERED|FILE_FLAG_OURBUFFER;\r
        stdin->Buffer = malloc(STDIN_BUFSIZ);\r
        stdin->BufferSpace = STDIN_BUFSIZ;\r
 \r
-       stdout = &_iob[1];\r
        stdout->FD = 1;\r
-       stdout->Flags = FILE_FLAG_ALLOC|FILE_FLAG_MODE_WRITE|FILE_FLAG_LINEBUFFERED;\r
+       stdout->Flags = FILE_FLAG_ALLOC|FILE_FLAG_MODE_WRITE|FILE_FLAG_LINEBUFFERED|FILE_FLAG_OURBUFFER;\r
        stdout->Buffer = malloc(STDOUT_BUFSIZ);\r
        stdout->BufferSpace = STDOUT_BUFSIZ;\r
        \r
-       stderr = &_iob[2];\r
        stderr->FD = 2;\r
        stderr->Flags = FILE_FLAG_ALLOC|FILE_FLAG_MODE_WRITE;\r
 }\r
@@ -151,6 +151,16 @@ EXPORT FILE *freopen(const char *file, const char *mode, FILE *fp)
                return NULL;\r
        }\r
        \r
+       // Default to buffered\r
+       // - Disabled until fseek() is fixed\r
+       #if 0\r
+       fp->BufferOfs = 0;\r
+       fp->BufferPos = 0;\r
+       fp->BufferSpace = BUFSIZ;\r
+       fp->Buffer = malloc( fp->BufferSpace );\r
+       fp->Flags |= FILE_FLAG_OURBUFFER;\r
+       #endif\r
+       \r
        if( (fp->Flags & FILE_FLAG_MODE_MASK) == FILE_FLAG_MODE_APPEND ) {\r
                _SysSeek(fp->FD, 0, SEEK_END);  //SEEK_END\r
        }\r
@@ -213,6 +223,33 @@ EXPORT FILE *open_memstream(char **bufferptr, size_t *lengthptr)
        return ret;\r
 }\r
 \r
+EXPORT FILE *fdopen(int fd, const char *mode)\r
+{\r
+       FILE    *ret;\r
+       \r
+       if( fd < 0 || !mode )   return NULL;\r
+       \r
+       ret = get_file_struct();\r
+       \r
+       ret->FD = fd;\r
+       ret->Flags = _fopen_modetoflags(mode);\r
+       if(ret->Flags == -1) {\r
+               ret->Flags = 0;\r
+               return NULL;\r
+       }\r
+       \r
+       ret->Buffer = NULL;\r
+       ret->BufferPos = 0;\r
+       ret->BufferSpace = 0;\r
+       \r
+       return ret;\r
+}\r
+\r
+EXPORT FILE *tmpfile(void)\r
+{\r
+       return NULL;\r
+}\r
+\r
 EXPORT int fclose(FILE *fp)\r
 {\r
        if( !(fp->Flags & FILE_FLAG_ALLOC) )\r
@@ -220,6 +257,10 @@ EXPORT int fclose(FILE *fp)
        fflush(fp);\r
        if( fp->FD >= 0 ) {\r
                _SysClose(fp->FD);\r
+               if( fp->Buffer && (fp->Flags & FILE_FLAG_OURBUFFER) ) {\r
+                       free(fp->Buffer);\r
+               }\r
+               fp->Buffer = NULL;\r
        }\r
        fp->Flags = 0;\r
        fp->FD = FD_NOTOPEN;\r
@@ -262,9 +303,13 @@ EXPORT int setvbuf(FILE *fp, char *buffer, int mode, size_t size)
                }\r
                // Allocate a buffer if one was not provided\r
                if( !buffer ) {\r
+                       fp->Flags |= FILE_FLAG_OURBUFFER;\r
                        buffer = malloc(size);\r
                        assert(buffer);\r
                }\r
+               else {\r
+                       fp->Flags &= ~FILE_FLAG_OURBUFFER;\r
+               }\r
                \r
                // Set buffer pointer and size\r
                fp->Buffer = buffer;\r
@@ -300,9 +345,12 @@ int _fflush_int(FILE *fp)
        case FILE_FLAG_MODE_APPEND:\r
                _SysSeek(fp->FD, fp->BufferOfs, SEEK_SET);\r
                len = _SysWrite(fp->FD, fp->Buffer, fp->BufferPos);\r
-               if( len < fp->BufferPos )\r
+               if( len != fp->BufferPos )\r
                        ret = 1;\r
-               fp->BufferPos -= len;\r
+               if( len <= fp->BufferPos )\r
+               {\r
+                       fp->BufferPos -= len;\r
+               }\r
                fp->BufferOfs = _SysTell(fp->FD);\r
                break;\r
                \r
@@ -312,7 +360,13 @@ int _fflush_int(FILE *fp)
                len = _SysWrite(fp->FD, fp->Buffer, fp->BufferPos);\r
                if( len != fp->BufferPos )\r
                        ret = 1;\r
-               fp->BufferPos -= len;\r
+               if( len <= fp->BufferPos )\r
+               {\r
+                       fp->BufferPos -= len;\r
+               }\r
+               //else {\r
+               //      _SysDebug("Flush of %i failed, %s", fp->FD, strerror(errno));\r
+               //}\r
                break;\r
        default:\r
                break;\r
@@ -320,22 +374,23 @@ int _fflush_int(FILE *fp)
        return ret;\r
 }\r
 \r
-EXPORT void fflush(FILE *fp)\r
+EXPORT int fflush(FILE *fp)\r
 {\r
        if( !fp || fp->FD == FD_NOTOPEN )\r
-               return ;\r
+               return EBADF;\r
        \r
        // Nothing to do for memory files\r
        if( fp->FD == FD_MEMFILE )\r
-               return ;\r
+               return 0;\r
        // Memory streams, update pointers\r
        if( fp->FD == FD_MEMSTREAM ) {\r
                *fp->BufPtr = fp->Buffer;\r
                *fp->LenPtr = fp->BufferPos;\r
-               return ;\r
+               return 0;\r
        }\r
        \r
        _fflush_int(fp);\r
+       return 0;\r
 }\r
 \r
 EXPORT void clearerr(FILE *fp)\r
@@ -425,7 +480,7 @@ int _fseek_memstream(FILE *fp, long int amt, int whence)
        return 0;\r
 }\r
 \r
-EXPORT int fseek(FILE *fp, long int amt, int whence)\r
+EXPORT int fseeko(FILE *fp, off_t amt, int whence)\r
 {\r
        if(!fp || fp->FD == FD_NOTOPEN) {\r
                errno = EBADF;\r
@@ -446,6 +501,11 @@ EXPORT int fseek(FILE *fp, long int amt, int whence)
        }\r
 }\r
 \r
+EXPORT int fseek(FILE *fp, long int amt, int whence)\r
+{\r
+       return fseeko(fp, amt, whence);\r
+}\r
+\r
 size_t _fwrite_unbuffered(FILE *fp, size_t size, size_t num, const void *data)\r
 {\r
        size_t  ret = 0, bytes;\r
@@ -583,6 +643,37 @@ size_t _fread_memstream(void *ptr, size_t size, size_t num, FILE *fp)
 }\r
 #endif\r
 \r
+size_t _fread_buffered(void *ptr, size_t size, FILE *fp)\r
+{\r
+       //_SysDebug("%p: %i-%i <= %i", fp,\r
+       //      (int)fp->Pos, (int)fp->BufferOfs, (int)fp->BufferPos);\r
+       if( fp->BufferPos > 0 ) {\r
+               assert( fp->Pos - fp->BufferOfs <= (int)fp->BufferPos );\r
+       }\r
+       if( fp->BufferPos == 0 || fp->Pos - fp->BufferOfs == (int)fp->BufferPos )\r
+       {\r
+               int rv = _SysRead(fp->FD, fp->Buffer, fp->BufferSpace);\r
+               if( rv <= 0 ) {\r
+                       fp->Flags |= FILE_FLAG_EOF;\r
+                       return 0;\r
+               }\r
+               \r
+               fp->BufferPos = rv;\r
+               fp->BufferOfs = fp->Pos;\r
+               //_SysDebug("%p: Buffered %i at %i", fp, rv, fp->Pos);\r
+       }\r
+       \r
+       size_t  inner_ofs = fp->Pos - fp->BufferOfs;\r
+       if(size > fp->BufferPos - inner_ofs)\r
+               size = fp->BufferPos - inner_ofs;\r
+       \r
+       //_SysDebug("%p: Read %i from %i+%i", fp, size,\r
+       //      (int)fp->BufferOfs, inner_ofs);\r
+       memcpy(ptr, fp->Buffer + inner_ofs, size);\r
+       fp->Pos += size;\r
+       return size;\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
@@ -591,36 +682,84 @@ EXPORT size_t fread(void *ptr, size_t size, size_t num, FILE *fp)
 {\r
        size_t  ret;\r
        \r
-       if(!fp || fp->FD == -1)\r
+       if(!fp || fp->FD == -1) {\r
+               LOG_WARN("bad fp %p", fp);\r
                return -1;\r
+       }\r
        if( size == 0 || num == 0 )\r
                return 0;\r
        \r
        if( _GetFileMode(fp) != FILE_FLAG_MODE_READ ) {\r
                errno = 0;\r
+               LOG_WARN("not open for read");\r
+               if( fp == stdin ) {\r
+                       LOG_WARN("BUGCHECK FAIL: stdin was not open for read");\r
+                       exit(129);\r
+               }\r
                return -1;\r
        }\r
 \r
+       // Don't read if EOF is set\r
+       if( fp->Flags & FILE_FLAG_EOF ) {\r
+               LOG_NOTICE("EOF");\r
+               return 0;\r
+       }\r
+\r
+       \r
        if( fp->FD == FD_MEMFILE ) {\r
                return _fread_memfile(ptr, size, num, fp);\r
        }\r
        else if( fp->FD == FD_MEMSTREAM ) {\r
                //return _fread_memstream(ptr, size, num, fp);\r
+               LOG_WARN("Reading from a mem stream");\r
                errno = EBADF;\r
                return 0;\r
        }\r
-       \r
+\r
        // Standard file\r
-       ret = _SysRead(fp->FD, ptr, size*num);\r
-       if( ret == (size_t)-1)\r
-               return -1;\r
-       if( ret == 0 && size*num > 0 ) {\r
-               fp->Flags |= FILE_FLAG_EOF;\r
-               return 0;\r
+       const size_t    bytes = size*num;\r
+\r
+       // TODO: Buffered reads\r
+       if( fp->BufferSpace )\r
+       {\r
+               size_t  ofs = 0;\r
+               size_t  rv;\r
+               // While not done, and buffered read succeeds\r
+               while( ofs < bytes && (rv = _fread_buffered((char*)ptr + ofs, bytes - ofs, fp)) != 0 )\r
+               {\r
+                       ofs += rv;\r
+               }\r
+               \r
+               ret = ofs;\r
+       }\r
+       else\r
+       {\r
+               ret = _SysRead(fp->FD, ptr, bytes);\r
+               if( ret == (size_t)-1)\r
+                       return -1;\r
+               if( ret == 0 && bytes > 0 ) {\r
+                       fp->Flags |= FILE_FLAG_EOF;\r
+                       return 0;\r
+               }\r
+       }\r
+\r
+       // if read was cut short        \r
+       if( ret != bytes )\r
+       {\r
+               size_t  extra = ret - (ret / size) * size;\r
+               // And it didn't fall short on a member boundary\r
+               if( extra )\r
+               {\r
+                       // Need to roll back the file pointer to the end of the last member\r
+                       _SysDebug("fread: TODO Roll back %zi bytes due to incomplete object (sz=%zi,n=%zi)",\r
+                               extra, size, num\r
+                               );\r
+               }\r
+               LOG_NOTICE("Incomplete read %i/%i bytes (object size %i)",\r
+                       ret, bytes, size);\r
        }\r
-       ret /= size;\r
        \r
-       return ret;\r
+       return ret / size;\r
 }\r
 \r
 /**\r
@@ -629,7 +768,7 @@ EXPORT size_t fread(void *ptr, size_t size, size_t num, FILE *fp)
 EXPORT int fputs(const char *s, FILE *fp)\r
 {\r
        int len = strlen(s);\r
-       return fwrite(s, 1, len, fp);\r
+       return fwrite(s, len, 1, fp);\r
 }\r
 \r
 /**\r
@@ -656,13 +795,13 @@ EXPORT char *fgets(char *s, int size, FILE *fp)
  */\r
 EXPORT int fputc(int c, FILE *fp)\r
 {\r
-       return fwrite(&c, 1, 1, fp);\r
+       unsigned char   ch = c;\r
+       return fwrite(&ch, 1, 1, fp);\r
 }\r
 \r
 EXPORT int putchar(int c)\r
 {\r
-       c &= 0xFF;\r
-       return _SysWrite(_stdout, &c, 1);\r
+       return fputc(c, stdout);\r
 }\r
 \r
 /**\r
@@ -671,7 +810,7 @@ EXPORT int putchar(int c)
  */\r
 EXPORT int fgetc(FILE *fp)\r
 {\r
-       char    ret = 0;\r
+       unsigned char   ret = 0;\r
        if( fread(&ret, 1, 1, fp) != 1 )\r
                return -1;\r
        return ret;\r
@@ -679,20 +818,17 @@ EXPORT int fgetc(FILE *fp)
 \r
 EXPORT int getchar(void)\r
 {\r
-       char    ret = 0;\r
-       if(_SysRead(_stdin, &ret, 1) != 1)      return -1;\r
-       return ret;\r
+       fflush(stdout);\r
+       return fgetc(stdin);\r
 }\r
 \r
 EXPORT int puts(const char *str)\r
 {\r
-        int    len;\r
-       \r
        if(!str)        return 0;\r
-       len = strlen(str);\r
+        int    len = strlen(str);\r
        \r
-       len = _SysWrite(_stdout, str, len);\r
-       _SysWrite(_stdout, "\n", 1);\r
+       fwrite(str, 1, len, stdout);\r
+       fwrite("\n", 1, 1, stdout);\r
        return len;\r
 }\r
 \r
@@ -703,8 +839,7 @@ EXPORT int puts(const char *str)
  */\r
 FILE *get_file_struct()\r
 {\r
-        int    i;\r
-       for(i=0;i<STDIO_MAX_STREAMS;i++)\r
+       for(int i=0;i<STDIO_MAX_STREAMS;i++)\r
        {\r
                if(_iob[i].Flags & FILE_FLAG_ALLOC)\r
                        continue ;\r

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