WebSVN – HelenOS – Diff – /branches/tracing/uspace/lib/libc/generic/io/io.c

 /** @addtogroup libc
  * @{
  */
 /** @file
  */
-#include <libc.h>
-#include <unistd.h>
 #include <stdio.h>
+#include <unistd.h>
-#include <io/io.h>
+#include <fcntl.h>
+#include <assert.h>
 #include <string.h>
 #include <errno.h>
-#include <console.h>
+#include <bool.h>
+#include <malloc.h>
+#include <io/klog.h>
+#include <vfs/vfs.h>
+#include <ipc/devmap.h>
+#include <adt/list.h>
+static void _fflushbuf(FILE *stream);
+static FILE stdin_null = {
+    .fd = -1,
+    .error = true,
+    .eof = true,
+    .klog = false,
+    .phone = -1,
+    .btype = _IONBF,
+    .buf = NULL,
+    .buf_size = 0,
+    .buf_head = NULL
+};
+static FILE stdout_klog = {
+    .fd = -1,
+    .error = false,
+    .eof = false,
+    .klog = true,
+    .phone = -1,
+    .btype = _IOLBF,
+    .buf = NULL,
+    .buf_size = BUFSIZ,
+    .buf_head = NULL
+};
+static FILE stderr_klog = {
+    .fd = -1,
+    .error = false,
+    .eof = false,
+    .klog = true,
+    .phone = -1,
+    .btype = _IONBF,
+    .buf = NULL,
+    .buf_size = 0,
+    .buf_head = NULL
+};
+FILE *stdin = NULL;
+FILE *stdout = NULL;
+FILE *stderr = NULL;
-const static char nl = '\n';
+static LIST_INITIALIZE(files);
+void __stdio_init(int filc, fdi_node_t *filv[])
+{
+    if (filc > 0) {
+        stdin = fopen_node(filv[0], "r");
+    } else {
+        stdin = &stdin_null;
+        list_append(&stdin->link, &files);
+    }
+    if (filc > 1) {
+        stdout = fopen_node(filv[1], "w");
+    } else {
-int puts(const char *str)
+        stdout = &stdout_klog;
+        list_append(&stdout->link, &files);
+    }
+    if (filc > 2) {
+        stderr = fopen_node(filv[2], "w");
+    } else {
+        stderr = &stderr_klog;
+        list_append(&stderr->link, &files);
+    }
+}
+void __stdio_done(void)
+{
+    link_t *link = files.next;
+    while (link != &files) {
+        FILE *file = list_get_instance(link, FILE, link);
+        fclose(file);
+        link = files.next;
+    }
+}
+static bool parse_mode(const char *mode, int *flags)
+{
+    /* Parse mode except first character. */
+    const char *mp = mode;
+    if (*mp++ == 0) {
+        errno = EINVAL;
-    size_t count;
+        return false;
+    }
-    if (str == NULL)
+    if ((*mp == 'b') || (*mp == 't'))
-        return putnchars("(NULL)", 6);
+        mp++;
+    bool plus;
-    for (count = 0; str[count] != 0; count++);
+    if (*mp == '+') {
+        mp++;
+        plus = true;
+    } else
+        plus = false;
-    if (console_write((void *) str, count) == count) {
+    if (*mp != 0) {
-        if (console_write(&nl, 1) == 1)
+        errno = EINVAL;
-            return 0;
+        return false;
+    }
+    /* Parse first character of mode and determine flags for open(). */
+    switch (mode[0]) {
+    case 'r':
+        *flags = plus ? O_RDWR : O_RDONLY;
+        break;
+    case 'w':
+        *flags = (O_TRUNC | O_CREAT) | (plus ? O_RDWR : O_WRONLY);
+        break;
+    case 'a':
+        /* TODO: a+ must read from beginning, append to the end. */
+        if (plus) {
+            errno = ENOTSUP;
+            return false;
+        }
+        *flags = (O_APPEND | O_CREAT) | (plus ? O_RDWR : O_WRONLY);
+    default:
+        errno = EINVAL;
+        return false;
+    }
-    return EOF;
+    return true;
+}
-/** Put count chars from buffer to stdout without adding newline
+/** Set stream buffer. */
- * @param buf Buffer with size at least count bytes - NULL pointer NOT allowed!
+void setvbuf(FILE *stream, void *buf, int mode, size_t size)
+{
+    stream->btype = mode;
- * @param count
+    stream->buf = buf;
+    stream->buf_size = size;
- * @return 0 on succes, EOF on fail
+    stream->buf_head = stream->buf;
- */
+}
-int putnchars(const char *buf, size_t count)
+static void _setvbuf(FILE *stream)
+{
-    if (console_write((void *) buf, count) == count)
+    /* FIXME: Use more complex rules for setting buffering options. */
-        return 0;
+    switch (stream->fd) {
+    case 1:
+        setvbuf(stream, NULL, _IOLBF, BUFSIZ);
-    return EOF;
+        break;
+    case 0:
+    case 2:
+        setvbuf(stream, NULL, _IONBF, 0);
+        break;
+    default:
+        setvbuf(stream, NULL, _IOFBF, BUFSIZ);
+    }
+}
+/** Allocate stream buffer. */
+static int _fallocbuf(FILE *stream)
+{
+    assert(stream->buf == NULL);
+    stream->buf = malloc(stream->buf_size);
+    if (stream->buf == NULL) {
+        errno = ENOMEM;
+        return -1;
+    }
+    stream->buf_head = stream->buf;
+    return 0;
+}
+/** Open a stream.
+ *
-/** Same as puts, but does not print newline at end
+ * @param path Path of the file to open.
+ * @param mode Mode string, (r|w|a)[b|t][+].
+ *
  */
-int putstr(const char *str)
+FILE *fopen(const char *path, const char *mode)
+{
-    size_t count;
+    int flags;
+    if (!parse_mode(mode, &flags))
+        return NULL;
+    /* Open file. */
+    FILE *stream = malloc(sizeof(FILE));
+    if (stream == NULL) {
+        errno = ENOMEM;
+        return NULL;
+    }
+    stream->fd = open(path, flags, 0666);
+    if (stream->fd < 0) {
+        /* errno was set by open() */
+        free(stream);
+        return NULL;
+    }
+    stream->error = false;
+    stream->eof = false;
+    stream->klog = false;
+    stream->phone = -1;
-    if (str == NULL)
+    _setvbuf(stream);
+    list_append(&stream->link, &files);
-        return putnchars("(NULL)", 6);
+    return stream;
+}
-    for (count = 0; str[count] != 0; count++);
+FILE *fdopen(int fd, const char *mode)
+{
+    /* Open file. */
-    if (console_write((void *) str, count) == count)
+    FILE *stream = malloc(sizeof(FILE));
+    if (stream == NULL) {
+        errno = ENOMEM;
-        return 0;
+        return NULL;
+    }
+    stream->fd = fd;
+    stream->error = false;
+    stream->eof = false;
+    stream->klog = false;
+    stream->phone = -1;
+    _setvbuf(stream);
+    list_append(&stream->link, &files);
-    return EOF;
+    return stream;
+}
-int putchar(int c)
+FILE *fopen_node(fdi_node_t *node, const char *mode)
+{
+    int flags;
+    if (!parse_mode(mode, &flags))
+        return NULL;
+    /* Open file. */
+    FILE *stream = malloc(sizeof(FILE));
-    char buf[STR_BOUNDS(1)];
+    if (stream == NULL) {
+        errno = ENOMEM;
+        return NULL;
+    }
+    stream->fd = open_node(node, flags);
+    if (stream->fd < 0) {
+        /* errno was set by open_node() */
+        free(stream);
+        return NULL;
+    }
+    stream->error = false;
-    size_t offs;
+    stream->eof = false;
+    stream->klog = false;
+    stream->phone = -1;
+    _setvbuf(stream);
+    list_append(&stream->link, &files);
+    return stream;
+}
+int fclose(FILE *stream)
+{
-    offs = 0;
+    int rc = 0;
+    fflush(stream);
+    if (stream->phone >= 0)
+        ipc_hangup(stream->phone);
+    if (stream->fd >= 0)
+        rc = close(stream->fd);
+    list_remove(&stream->link);
+    if ((stream != &stdin_null)
+        && (stream != &stdout_klog)
-    if (chr_encode(c, buf, &offs, STR_BOUNDS(1)) != EOK)
+        && (stream != &stderr_klog))
+        free(stream);
+    stream = NULL;
+    if (rc != 0) {
+        /* errno was set by close() */
         return EOF;
+    }
+    return 0;
+}
+/** Read from a stream.
+ *
+ * @param buf    Destination buffer.
+ * @param size   Size of each record.
+ * @param nmemb  Number of records to read.
+ * @param stream Pointer to the stream.
+ *
+ */
+size_t fread(void *buf, size_t size, size_t nmemb, FILE *stream)
+{
+    size_t left = size * nmemb;
+    size_t done = 0;
+    /* Make sure no data is pending write. */
+    _fflushbuf(stream);
+    while ((left > 0) && (!stream->error) && (!stream->eof)) {
-    if (console_write((void *) buf, offs) == offs)
+        ssize_t rd = read(stream->fd, buf + done, left);
+        if (rd < 0)
+            stream->error = true;
+        else if (rd == 0)
+            stream->eof = true;
+        else {
-        return c;
+            left -= rd;
+            done += rd;
+        }
+    }
+    return (done / size);
+}
+static size_t _fwrite(const void *buf, size_t size, size_t nmemb, FILE *stream)
+{
+    size_t left = size * nmemb;
-    return EOF;
+    size_t done = 0;
+    while ((left > 0) && (!stream->error)) {
+        ssize_t wr;
+        if (stream->klog)
+            wr = klog_write(buf + done, left);
+        else
+            wr = write(stream->fd, buf + done, left);
+        if (wr <= 0)
+            stream->error = true;
+        else {
+            left -= wr;
+            done += wr;
+        }
+    }
+    return (done / size);
+}
+/** Drain stream buffer, do not sync stream. */
+static void _fflushbuf(FILE *stream)
+{
-int getchar(void)
+    size_t bytes_used;
+    if ((!stream->buf) || (stream->btype == _IONBF) || (stream->error))
+        return;
+    bytes_used = stream->buf_head - stream->buf;
+    if (bytes_used == 0)
+        return;
+    (void) _fwrite(stream->buf, 1, bytes_used, stream);
+    stream->buf_head = stream->buf;
+}
+/** Write to a stream.
+ *
+ * @param buf    Source buffer.
+ * @param size   Size of each record.
+ * @param nmemb  Number of records to write.
+ * @param stream Pointer to the stream.
+ *
+ */
+size_t fwrite(const void *buf, size_t size, size_t nmemb, FILE *stream)
+{
-    unsigned char c;
+    uint8_t *data;
+    size_t bytes_left;
+    size_t now;
+    size_t buf_free;
+    size_t total_written;
+    size_t i;
+    uint8_t b;
+    bool need_flush;
+    /* If not buffered stream, write out directly. */
+    if (stream->btype == _IONBF) {
+        now = _fwrite(buf, size, nmemb, stream);
+        fflush(stream);
+        return now;
+    }
+    /* Perform lazy allocation of stream buffer. */
+    if (stream->buf == NULL) {
+        if (_fallocbuf(stream) != 0)
+            return 0; /* Errno set by _fallocbuf(). */
+    }
+    data = (uint8_t *) buf;
+    bytes_left = size * nmemb;
+    total_written = 0;
+    need_flush = false;
+    while ((!stream->error) && (bytes_left > 0)) {
+        buf_free = stream->buf_size - (stream->buf_head - stream->buf);
+        if (bytes_left > buf_free)
+            now = buf_free;
+        else
+            now = bytes_left;
+        for (i = 0; i < now; i++) {
+            b = data[i];
+            stream->buf_head[i] = b;
+            if ((b == '\n') && (stream->btype == _IOLBF))
+                need_flush = true;
+        }
+        buf += now;
+        stream->buf_head += now;
+        buf_free -= now;
+        bytes_left -= now;
+        total_written += now;
+        if (buf_free == 0) {
+            /* Only need to drain buffer. */
+            _fflushbuf(stream);
+            need_flush = false;
+        }
+    }
-    console_flush();
+    if (need_flush)
+        fflush(stream);
+    return (total_written / size);
+}
+int fputc(wchar_t c, FILE *stream)
+{
+    char buf[STR_BOUNDS(1)];
+    size_t sz = 0;
-    if (read_stdin((void *) &c, 1) == 1)
+    if (chr_encode(c, buf, &sz, STR_BOUNDS(1)) == EOK) {
+        size_t wr = fwrite(buf, sz, 1, stream);
+        if (wr < sz)
-        return c;
+            return EOF;
+        return (int) c;
+    }
     return EOF;
+}
-int fflush(FILE *f)
+int putchar(wchar_t c)
+{
-    /* Dummy implementation */
+    return fputc(c, stdout);
+}
+int fputs(const char *str, FILE *stream)
+{
+    return fwrite(str, str_size(str), 1, stream);
+}
+int puts(const char *str)
+{
+    return fputs(str, stdout);
+}
+int fgetc(FILE *stream)
+{
-    (void) f;
+    char c;
+    /* This could be made faster by only flushing when needed. */
+    if (stdout)
+        fflush(stdout);
+    if (stderr)
-    console_flush();
+        fflush(stderr);
+    if (fread(&c, sizeof(char), 1, stream) < sizeof(char))
+        return EOF;
+    return (int) c;
+}
+int getchar(void)
+{
+    return fgetc(stdin);
+}
+int fseek(FILE *stream, long offset, int origin)
+{
+    off_t rc = lseek(stream->fd, offset, origin);
+    if (rc == (off_t) (-1)) {
+        /* errno has been set by lseek. */
+        return -1;
+    }
+    stream->eof = false;
     return 0;
+}
+void rewind(FILE *stream)
+{
+    (void) fseek(stream, 0, SEEK_SET);
+}
+int fflush(FILE *stream)
+{
+    _fflushbuf(stream);
+    if (stream->klog) {
+        klog_update();
+        return EOK;
+    }
+    if (stream->fd >= 0)
+        return fsync(stream->fd);
+    return ENOENT;
+}
+int feof(FILE *stream)
+{
+    return stream->eof;
+}
+int ferror(FILE *stream)
+{
+    return stream->error;
+}
+int fphone(FILE *stream)
+{
+    if (stream->fd >= 0) {
+        if (stream->phone < 0)
+            stream->phone = fd_phone(stream->fd);
+        return stream->phone;
+    }
+    return -1;
+}
+int fnode(FILE *stream, fdi_node_t *node)
+{
+    if (stream->fd >= 0)
+        return fd_node(stream->fd, node);
+    return ENOENT;
+}
 /** @}
  */

Subversion Repositories HelenOS

(root)/branches/tracing/uspace/lib/libc/generic/io/io.c – Rev 4377 → 4692