source: mainline/uspace/lib/libc/generic/io/stdio.c@ 6b21292

/*
 * Copyright (c) 2008 Jiri Svoboda
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup libc
 * @{
 */
/**
 * @file
 * @brief ANSI C Stream I/O.
 */ 

#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>
#include <bool.h>
#include <stdio.h>

FILE *stdin, *stdout, *stderr;

/**
 * Open a stream.
 *
 * @param file_name     Name of the file to open.
 * @param mode          Mode string, (r|w|a)[b|t][+].
 */
FILE *fopen(const char *file_name, const char *mode)
{
        FILE *f;
        int flags;
        bool plus;
        const char *mp;

        /* Parse mode except first character. */

        mp = mode;
        if (*mp++ == '\0') {
                errno = EINVAL;
                return NULL;
        }

        if (*mp == 'b' || *mp == 't') ++mp;

        if (*mp == '+') {
                ++mp;
                plus = true;
        } else {
                plus = false;
        }

        if (*mp != '\0') {
                errno = EINVAL;
                return NULL;
        }

        /* 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 NULL;
                }
                flags = (O_APPEND | O_CREAT) | (plus ? O_RDWR : O_WRONLY);
        default:
                errno = EINVAL;
                return NULL;
        }

        /* Open file. */
        
        f = malloc(sizeof(FILE));
        if (f == NULL) {
                errno = ENOMEM;
                return NULL;
        }

        f->fd = open(file_name, flags, 0666);
        if (f->fd < 0) {
                free(f);
                return NULL; /* errno was set by open() */
        }

        f->error = 0;
        f->eof = 0;

        return f;
}

/** Close a stream.
 *
 * @param f     Pointer to stream.
 * @return      0 on success, EOF on error.
 */
int fclose(FILE *f)
{
        int rc;

        rc = close(f->fd);
        free(f);

        if (rc != 0)
                return EOF; /* errno was set by close() */

        return 0;
}

/** Read from a stream.
 *
 * @param buf   Destination buffer.
 * @param size  Size of each record.
 * @param nmemb Number of records to read.
 * @param f     Pointer to the stream.
 */
size_t fread(void *buf, size_t size, size_t nmemb, FILE *f)
{
        size_t left, done, n;

        left = size * nmemb;
        done = 0;

        while (left > 0 && !f->error && !f->eof) {
                n = read(f->fd, buf + done, left);

                if (n < 0) {
                        f->error = 1;
                } else if (n == 0) {
                        f->eof = 1;
                } else {
                        left -= n;
                        done += n;
                }
        }

        return done / size;
}


/** Write to a stream.
 *
 * @param buf   Source buffer.
 * @param size  Size of each record.
 * @param nmemb Number of records to write.
 * @param f     Pointer to the stream.
 */
size_t fwrite(const void *buf, size_t size, size_t nmemb, FILE *f)
{
        size_t left, done, n;

        left = size * nmemb;
        done = 0;

        while (left > 0 && !f->error) {
                n = write(f->fd, buf + done, left);

                if (n <= 0) {
                        f->error = 1;
                } else {
                        left -= n;
                        done += n;
                }
        } 

        return done / size;
}

/** Return the end-of-file indicator of a stream. */
int feof(FILE *f)
{
        return f->eof;
}

/** Return the error indicator of a stream. */
int ferror(FILE *f)
{
        return f->error;
}

/** Clear the error and end-of-file indicators of a stream. */
void clearerr(FILE *f)
{
        f->eof = 0;
        f->error = 0;
}

/** Read character from a stream. */
int fgetc(FILE *f)
{
        unsigned char c;
        size_t n;

        n = fread(&c, sizeof(c), 1, f);
        if (n < 1) return EOF;

        return (int) c;
}

/** Write character to a stream. */
int fputc(int c, FILE *f)
{
        unsigned char cc;
        size_t n;

        cc = (unsigned char) c;
        n = fwrite(&cc, sizeof(cc), 1, f);
        if (n < 1) return EOF;

        return (int) cc;
}

/** Write string to a stream. */
int fputs(const char *s, FILE *f)
{
        int rc;

        rc = 0;

        while (*s && rc >= 0) {
                rc = fputc(*s++, f);
        }

        if (rc < 0) return EOF;

        return 0;
}

/** Seek to position in stream. */
int fseek(FILE *f, long offset, int origin)
{
        off_t rc;

        rc = lseek(f->fd, offset, origin);
        if (rc == (off_t) (-1)) {
                /* errno has been set by lseek. */
                return -1;
        }

        f->eof = 0;

        return 0;
}

/** @}
 */