Sprinter-SDCC/libc/stdio/file.c

/*
 * file.c — minimal unbuffered FILE * implementation on top of the
 * POSIX-style fd I/O (open/read/write/lseek/close).
 *
 * No buffering: each fputc/fgetc maps to one read/write syscall.  For
 * heavy-throughput code, prefer fread/fwrite with a sizable buffer or
 * the raw fd I/O directly.
 *
 * stdin/stdout/stderr are STATIC sentinel FILEs with fd=-1 and the
 * _F_CONIN/_F_CONOUT flags set; fputc/fgetc detect them and call
 * putchar()/getchar() (which already do CR/LF mapping and ESTEX calls).
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>

/* ---- console pseudo-streams ----------------------------------------*/
static FILE _stdin  = { -1, _F_READ  | _F_CONIN  };
static FILE _stdout = { -2, _F_WRITE | _F_CONOUT };
static FILE _stderr = { -3, _F_WRITE | _F_CONOUT };
static FILE _stdaux = { -4, _F_WRITE | _F_CONOUT };
static FILE _stdprn = { -5, _F_WRITE | _F_CONOUT };

FILE *const stdin  = &_stdin;
FILE *const stdout = &_stdout;
FILE *const stderr = &_stderr;
FILE *const stdaux = &_stdaux;
FILE *const stdprn = &_stdprn;

/* ---- fopen / fclose -------------------------------------------------*/

/* Translate a fopen() mode string to the open() flags subset our
 * libc/io/open.c understands.  Supported: r, w, a, with optional "+"
 * and trailing "b" (binary — we ignore as all I/O is binary).
 */
static int mode_to_flags(const char *mode, uint8_t *file_flags)
{
    if (!mode || !*mode) {
        errno = EINVAL;
        return -1;
    }
    int  oflags = 0;
    uint8_t ff = 0;
    char base = *mode;
    int  plus = 0;
    for (const char *p = mode + 1; *p; p++) {
        if (*p == '+') plus = 1;
        /* 'b' and 't' are ignored — all I/O is binary on Sprinter. */
    }
    switch (base) {
    case 'r':
        oflags = plus ? O_RDWR : O_RDONLY;
        ff = _F_READ | (plus ? _F_WRITE : 0);
        break;
    case 'w':
        oflags = (plus ? O_RDWR : O_WRONLY) | O_CREAT | O_TRUNC;
        ff = _F_WRITE | (plus ? _F_READ : 0);
        break;
    case 'a':
        oflags = (plus ? O_RDWR : O_WRONLY) | O_CREAT | O_APPEND;
        ff = _F_WRITE | _F_APPEND | (plus ? _F_READ : 0);
        break;
    default:
        errno = EINVAL;
        return -1;
    }
    *file_flags = ff;
    return oflags;
}

FILE *fopen(const char *path, const char *mode)
{
    uint8_t ff;
    int oflags = mode_to_flags(mode, &ff);
    if (oflags < 0) return NULL;

    int fd = open(path, oflags);
    if (fd < 0) return NULL;

    FILE *fp = (FILE *)malloc(sizeof(FILE));
    if (!fp) {
        int saved = errno;
        close(fd);
        errno = saved ? saved : ENOMEM;
        return NULL;
    }
    fp->fd    = fd;
    fp->flags = ff;
    return fp;
}

int fclose(FILE *fp)
{
    if (!fp) {
        errno = EBADF;
        return EOF;
    }
    /* Don't close stdin/stdout/stderr. */
    if (fp == &_stdin || fp == &_stdout || fp == &_stderr) {
        return 0;
    }
    int r = close(fp->fd);
    free(fp);
    return r < 0 ? EOF : 0;
}

int fflush(FILE *fp)
{
    /* Unbuffered — nothing to flush. */
    (void)fp;
    return 0;
}

/* ---- char-at-a-time -------------------------------------------------*/

int fputc(int c, FILE *fp)
{
    if (!fp) { errno = EBADF; return EOF; }
    if (fp->flags & _F_CONOUT) {
        return putchar(c);
    }
    if (!(fp->flags & _F_WRITE)) { errno = EBADF; return EOF; }
    uint8_t ch = (uint8_t)c;
    if (write(fp->fd, &ch, 1) != 1) {
        fp->flags |= _F_ERROR;
        return EOF;
    }
    return (int)ch;
}

int fgetc(FILE *fp)
{
    if (!fp) { errno = EBADF; return EOF; }
    if (fp->flags & _F_CONIN) {
        return getchar();
    }
    if (!(fp->flags & _F_READ)) { errno = EBADF; return EOF; }
    uint8_t ch;
    int r = read(fp->fd, &ch, 1);
    if (r == 0) { fp->flags |= _F_EOF; return EOF; }
    if (r < 0)  { fp->flags |= _F_ERROR; return EOF; }
    return (int)ch;
}

/* ---- string-at-a-time ----------------------------------------------*/

int fputs(const char *s, FILE *fp)
{
    if (!fp || !s) { errno = EBADF; return EOF; }
    if (fp->flags & _F_CONOUT) {
        while (*s) {
            if (putchar((unsigned char)*s++) == EOF) return EOF;
        }
        return 0;
    }
    if (!(fp->flags & _F_WRITE)) { errno = EBADF; return EOF; }
    size_t n = strlen(s);
    int w = write(fp->fd, s, (uint16_t)n);
    if (w < 0 || (size_t)w != n) {
        fp->flags |= _F_ERROR;
        return EOF;
    }
    return 0;
}

char *fgets(char *buf, int n, FILE *fp)
{
    if (!buf || n < 2 || !fp) return NULL;
    int i = 0;
    while (i < n - 1) {
        int c = fgetc(fp);
        if (c == EOF) {
            if (i == 0) return NULL;
            break;
        }
        buf[i++] = (char)c;
        if (c == '\n') break;
    }
    buf[i] = '\0';
    return buf;
}

/* ---- block-at-a-time -----------------------------------------------*/

size_t fread(void *ptr, size_t size, size_t nmemb, FILE *fp)
{
    if (!ptr || !fp) return 0;
    if (size == 0 || nmemb == 0) return 0;
    if (fp->flags & _F_CONIN) {
        /* line-buffered console read — not very useful but functional. */
        char *p = (char *)ptr;
        size_t total = size * nmemb;
        for (size_t i = 0; i < total; i++) {
            int c = getchar();
            if (c == EOF) return i / size;
            p[i] = (char)c;
        }
        return nmemb;
    }
    if (!(fp->flags & _F_READ)) { errno = EBADF; return 0; }
    size_t total = size * nmemb;
    int r = read(fp->fd, ptr, (uint16_t)total);
    if (r < 0) { fp->flags |= _F_ERROR; return 0; }
    if ((size_t)r < total) fp->flags |= _F_EOF;
    return (size_t)r / size;
}

size_t fwrite(const void *ptr, size_t size, size_t nmemb, FILE *fp)
{
    if (!ptr || !fp) return 0;
    if (size == 0 || nmemb == 0) return 0;
    if (fp->flags & _F_CONOUT) {
        const char *p = (const char *)ptr;
        size_t total = size * nmemb;
        for (size_t i = 0; i < total; i++) {
            if (putchar((unsigned char)p[i]) == EOF) return i / size;
        }
        return nmemb;
    }
    if (!(fp->flags & _F_WRITE)) { errno = EBADF; return 0; }
    size_t total = size * nmemb;
    int w = write(fp->fd, ptr, (uint16_t)total);
    if (w < 0) { fp->flags |= _F_ERROR; return 0; }
    return (size_t)w / size;
}

/* ---- positioning ---------------------------------------------------*/

int fseek(FILE *fp, long off, int whence)
{
    if (!fp || (fp->flags & (_F_CONIN | _F_CONOUT))) {
        errno = EBADF;
        return -1;
    }
    long r = lseek(fp->fd, off, whence);
    if (r < 0) return -1;
    fp->flags &= (uint8_t)~_F_EOF;
    return 0;
}

long ftell(FILE *fp)
{
    if (!fp || (fp->flags & (_F_CONIN | _F_CONOUT))) {
        errno = EBADF;
        return -1L;
    }
    return lseek(fp->fd, 0L, SEEK_CUR);
}

void rewind(FILE *fp)
{
    if (!fp) return;
    fseek(fp, 0L, SEEK_SET);
    fp->flags &= (uint8_t)~(_F_EOF | _F_ERROR);
}

/* ---- status --------------------------------------------------------*/

int feof  (FILE *fp) { return fp && (fp->flags & _F_EOF)   ? 1 : 0; }
int ferror(FILE *fp) { return fp && (fp->flags & _F_ERROR) ? 1 : 0; }
void clearerr(FILE *fp) {
    if (fp) fp->flags &= (uint8_t)~(_F_EOF | _F_ERROR);
}