add more test (3)

[cmore.git] / cmd.c

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

#include "debug.h"

#include "cmd.h"

#define BUFFERSIZE 4096

static char *_read_stream (FILE *sd)
{
    char *buffer = NULL;
    size_t size = 0;
    do {
        size += BUFFERSIZE + (size == 0);
        buffer = (char *) realloc (buffer, size);
        memset (buffer + size - BUFFERSIZE - 1, 0, BUFFERSIZE + 1);
    } while (fread (buffer + size - BUFFERSIZE - 1, 1, BUFFERSIZE, sd) > 0);

    /* check size */
    if (buffer[0] == '\0') {
        free (buffer);
        buffer = NULL;
    }

    return buffer;
}

char *exec_cmd (char *cmd)
{
    int status = -1;
    char *buffer = NULL;

    VERBOSE (DEBUG, fprintf (stdout, "exec command: %s\n", cmd));
    int saved_stderr = dup (STDERR_FILENO);
    close (STDERR_FILENO);
    FILE *fp = popen (cmd, "r");
    if (fp != NULL) {
        buffer = _read_stream (fp);
        status = pclose(fp);
    }
    dup2 (saved_stderr, STDERR_FILENO);

    if ((status == -1) || (buffer == NULL)) {
        free (buffer);
        buffer = NULL;
    }

    return buffer;
}

char *load_file (char *name)
{
    int status = -1;
    char *buffer = NULL;

    VERBOSE (DEBUG, fprintf (stdout, "open file: %s\n", name));
    FILE *fd = fopen (name, "r");
    if (fd != NULL) {
        buffer = _read_stream (fd);
        status = fclose (fd);
    }

    if (status == -1) {
        free (buffer);
        buffer = NULL;
    }

    return buffer;
}

char *read_stdin (void)
{
    char *buffer = NULL;

    VERBOSE (DEBUG, fprintf (stdout, "read stdin\n"));
    buffer = _read_stream (stdin);

    return buffer;
}

char **split_lines (char *buffer, int max)
{
    int n = 0;

    char **lines = NULL;
    int i, j;
    char *line = (char *) calloc (max + 1, sizeof (char));
    for (i = j = 0; buffer[i] != '\0'; i++) {
        if (buffer[i] >= ' ') {
            line[j++] = buffer[i];
        }
        if ((j == max) || (buffer[i] == '\n')) {
            lines = (char **) realloc (lines, (n + 2) * sizeof (char *));
            lines[n] = line;
            lines[n + 1] = NULL;
            line = calloc (max + 1, sizeof (char));
            n++;
            j = 0;
        }
    }
    if (line[0] != '\0') {
        lines = (char **) realloc (lines, (n + 2) * sizeof (char *));
        lines[n++] = line;
    } else {
        free (line);
    }
    VERBOSE (DEBUG, fprintf (stdout, "split lines: %d\n", n));

    return lines;
}

void free_lines (char **lines)
{
    int i = 0;

    if (!lines) {
        return;
    }

    while (lines[i]) {
        free (lines[i++]);
    }
    free (lines);
}

/* vim: set ts=4 sw=4 et: */