MentOS/libc/src/stdio.c

/// @file stdio.c
/// @brief Standard I/0 functions.
/// @copyright (c) 2014-2023 This file is distributed under the MIT License.
/// See LICENSE.md for details.

#include <sys/errno.h>
#include "stdio.h"
#include "ctype.h"
#include "string.h"
#include "stdbool.h"
#include "sys/unistd.h"
#include "strerror.h"

void putchar(int character)
{
    write(STDOUT_FILENO, &character, 1);
}

void puts(const char *str)
{
    write(STDOUT_FILENO, str, strlen(str));
}

int getchar(void)
{
    char c;
    while (read(STDIN_FILENO, &c, 1) == 0)
        continue;

    return c;
}

char *gets(char *str)
{
    // Check the input string.
    if (str == NULL)
        return NULL;
    // Buffer for reading input.
    char buffer[GETS_BUFFERSIZE];
    memset(buffer, '\0', GETS_BUFFERSIZE);
    // Char pointer to the buffer.
    char *cptr = buffer;
    // Character storage and counter to prevent overflow.
    int ch, counter = 0;
    // Read until we find a newline or we exceed the buffer size.
    while (((ch = getchar()) != '\n') && (counter++ < GETS_BUFFERSIZE)) {
        // If we encounter EOF, stop.
        if (ch == EOF) {
            // EOF at start of line return NULL.
            if (cptr == str) {
                return NULL;
            }
            break;
        }
        // Backspace key
        if (ch == '\b') {
            if (counter > 0) {
                counter--;
                --cptr;
                putchar('\b');
            }
        } else {
            // The character is stored at address, and the pointer is incremented.
            *cptr++ = ch;
        }
    }
    // Add the null-terminating character.
    *cptr = '\0';
    // Copy the string we have read.
    strcpy(str, buffer);
    // Return a pointer to the original string.
    return str;
}

int atoi(const char *str)
{
    // Check the input string.
    if (str == NULL)
        return 0;
    // Initialize sign, the result variable, and two indices.
    int sign = (str[0] == '-') ? -1 : +1, result = 0, i;
    // Find where the number ends.
    for (i = (sign == -1) ? 1 : 0; (str[i] != '\0') && isdigit(str[i]); ++i)
        result = (result * 10) + str[i] - '0';
    return sign * result;
}

long strtol(const char *str, char **endptr, int base)
{
    const char *s;
    long acc, cutoff;
    int c;
    int neg, any, cutlim;
    /*
     * Skip white space and pick up leading +/- sign if any.
     * If base is 0, allow 0x for hex and 0 for octal, else
     * assume decimal; if base is already 16, allow 0x.
     */
    s = str;
    do {
        c = (unsigned char)*s++;
    } while (isspace(c));
    if (c == '-') {
        neg = 1;
        c   = *s++;
    } else {
        neg = 0;
        if (c == '+')
            c = *s++;
    }
    if ((base == 0 || base == 16) &&
        c == '0' && (*s == 'x' || *s == 'X')) {
        c = s[1];
        s += 2;
        base = 16;
    }
    if (base == 0)
        base = c == '0' ? 8 : 10;
    /*
     * Compute the cutoff value between legal numbers and illegal
     * numbers.  That is the largest legal value, divided by the
     * base.  An input number that is greater than this value, if
     * followed by a legal input character, is too big.  One that
     * is equal to this value may be valid or not; the limit
     * between valid and invalid numbers is then based on the last
     * digit.  For instance, if the range for longs is
     * [-2147483648..2147483647] and the input base is 10,
     * cutoff will be set to 214748364 and cutlim to either
     * 7 (neg==0) or 8 (neg==1), meaning that if we have accumulated
     * a value > 214748364, or equal but the next digit is > 7 (or 8),
     * the number is too big, and we will return a range error.
     *
     * Set any if any `digits' consumed; make it negative to indicate
     * overflow.
     */
    cutoff = neg ? LONG_MIN : LONG_MAX;
    cutlim = cutoff % base;
    cutoff /= base;
    if (neg) {
        if (cutlim > 0) {
            cutlim -= base;
            cutoff += 1;
        }
        cutlim = -cutlim;
    }
    for (acc = 0, any = 0;; c = (unsigned char)*s++) {
        if (isdigit(c))
            c -= '0';
        else if (isalpha(c))
            c -= isupper(c) ? 'A' - 10 : 'a' - 10;
        else
            break;
        if (c >= base)
            break;
        if (any < 0)
            continue;
        if (neg) {
            if (acc < cutoff || (acc == cutoff && c > cutlim)) {
                any   = -1;
                acc   = LONG_MIN;
                errno = ERANGE;
            } else {
                any = 1;
                acc *= base;
                acc -= c;
            }
        } else {
            if (acc > cutoff || (acc == cutoff && c > cutlim)) {
                any   = -1;
                acc   = LONG_MAX;
                errno = ERANGE;
            } else {
                any = 1;
                acc *= base;
                acc += c;
            }
        }
    }
    if (endptr != 0)
        *endptr = (char *)(any ? s - 1 : str);
    return (acc);
}

int fgetc(int fd)
{
    unsigned char c;
    if (read(fd, &c, 1) <= 0)
        return EOF;
    return c;
}

char *fgets(char *buf, int n, int fd)
{
    int c;
    char *p;

    /* get max bytes or upto a newline */
    for (p = buf, n--; n > 0; n--) {
        if ((c = fgetc(fd)) == EOF)
            break;
        *p++ = c;
        if (c == '\n')
            break;
    }
    *p = 0;
    if (p == buf || c == EOF)
        return NULL;
    return (p);
}

void perror(const char *s)
{
    if (s) {
        puts(s);
        putchar(':');
        putchar(' ');
    }
    puts(strerror(errno));
}