input.c

/*
 * Copyright (c) 2010 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.
 */

#include <stdio.h>
#include <stdlib.h>
#include "mytypes.h"
#include "os/os.h"
#include "strtab.h"

#include "input.h"

#define INPUT_BUFFER_SIZE 256

static int input_init_file(input_t *input, const char *fname);
static void input_init_interactive(input_t *input);
static void input_init_string(input_t *input, const char *str);

int input_new_file(input_t **input, const char *fname)
{
        *input = malloc(sizeof(input_t));
        if (*input == NULL)
                return ENOMEM;

        return input_init_file(*input, fname);
}

int input_new_interactive(input_t **input)
{
        *input = malloc(sizeof(input_t));
        if (*input == NULL)
                return ENOMEM;

        input_init_interactive(*input);
        return EOK;
}

int input_new_string(input_t **input, const char *str)
{
        *input = malloc(sizeof(input_t));
        if (*input == NULL)
                return ENOMEM;

        input_init_string(*input, str);
        return EOK;
}

static int input_init_file(input_t *input, const char *fname)
{
        FILE *f;

        f = fopen(fname, "rt");
        if (f == NULL)
                return ENOENT;

        input->buffer = malloc(INPUT_BUFFER_SIZE);
        if (input->buffer == NULL) {
                printf("Memory allocation failed.\n");
                exit(1);
        }

        input->name = os_str_dup(fname);
        input->str = NULL;
        input->line_no = 0;
        input->fin = f;
        return EOK;
}

static void input_init_interactive(input_t *input)
{
        input->buffer = malloc(INPUT_BUFFER_SIZE);
        if (input->buffer == NULL) {
                printf("Memory allocation failed.\n");
                exit(1);
        }

        input->name = "<user-input>";
        input->str = NULL;
        input->line_no = 0;
        input->fin = NULL;
}

static void input_init_string(input_t *input, const char *str)
{
        input->buffer = malloc(INPUT_BUFFER_SIZE);
        if (input->buffer == NULL) {
                printf("Memory allocation failed.\n");
                exit(1);
        }

        input->name = "<builtin>";
        input->str = str;
        input->line_no = 0;
        input->fin = NULL;
}

int input_get_line(input_t *input, char **line)
{
        const char *sp;
        char *dp;
        char *line_p;
        size_t cnt;

        if (input->fin != NULL) {
                /* Reading from file. */
                if (fgets(input->buffer, INPUT_BUFFER_SIZE, input->fin) == NULL)
                        input->buffer[0] = '\0';

                if (ferror(input->fin))
                        return EIO;

                *line = input->buffer;
        } else if (input->str != NULL) {
                /* Reading from a string constant. */

                /* Copy one line. */
                sp = input->str;
                dp = input->buffer;
                cnt = 0;
                while (*sp != '\n' && *sp != '\0' &&
                    cnt < INPUT_BUFFER_SIZE - 2) {
                        *dp++ = *sp++;
                }

                /* Advance to start of next line. */
                if (*sp == '\n')
                        *dp++ = *sp++;

                *dp = '\0';
                input->str = sp;
                *line = input->buffer;
        } else {
                /* Interactive mode */
                if (input->line_no == 0)
                        printf("sbi> ");
                else
                        printf("...  ");

                fflush(stdout);
                if (os_input_line(&line_p) != EOK)
                        return EIO;

                *line = line_p;
        }

        ++input->line_no;
        return EOK;
}

int input_get_line_no(input_t *input)
{
        return input->line_no;
}

---