/*
* Copyright (C) 2013 Peter Schiffer.
*
* This file is part of libpipeline.
*
* libpipeline is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or (at
* your option) any later version.
*
* libpipeline is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with libpipeline; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301
* USA.
*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include <sys/select.h>
#include <errno.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include "xalloc.h"
#include "xvasprintf.h"
#include "common.h"
const char *program_name = "read";
/* Must be 8194 or bigger */
#define RANDOM_STR_LEN 9000
/* Unit test for bug: https://bugzilla.redhat.com/show_bug.cgi?id=876108 */
START_TEST (test_read_long_line)
{
/* Generate long random string */
static const char *alphanum = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
"abcdefghijklmnopqrstuvwxyz";
char random_string[RANDOM_STR_LEN] = "";
unsigned i;
for (i = 0; i < RANDOM_STR_LEN - 2; i++) {
random_string[i] = alphanum[rand () % (strlen (alphanum) - 1)];
}
random_string[RANDOM_STR_LEN - 1] = '\0';
/* Write the random string to file */
char *testfile = xasprintf ("%s/test", temp_dir);
FILE *tfd = fopen (testfile, "w");
if (!tfd) {
fail ("fopen failed: %s", strerror (errno));
return;
}
fprintf (tfd, "%s\n", random_string);
fclose (tfd);
char *expected_output = xasprintf ("%s\n", random_string);
pipeline *p;
const char *line;
char *read_result = NULL, *temp = NULL;
/* File must be read twice to reproduce the bug */
p = pipeline_new ();
pipeline_want_infile (p, testfile);
pipeline_want_out (p, -1);
pipeline_start (p);
while ((line = pipeline_readline (p)) != NULL){
if (read_result) {
temp = read_result;
read_result = xasprintf ("%s%s", read_result, line);
free (temp);
} else {
read_result = xasprintf ("%s", line);
}
}
pipeline_free (p);
fail_unless (!strcmp (read_result, expected_output),
"Returned string doesn't match the input.");
free (read_result);
read_result = NULL;
p = pipeline_new ();
pipeline_want_infile (p, testfile);
pipeline_want_out (p, -1);
pipeline_start (p);
while ((line = pipeline_readline (p)) != NULL){
if (read_result) {
temp = read_result;
read_result = xasprintf ("%s%s", read_result, line);
free (temp);
} else {
read_result = xasprintf ("%s", line);
}
}
pipeline_free (p);
fail_unless (!strcmp (read_result, expected_output),
"Returned string doesn't match the input.");
free (testfile);
free (expected_output);
free (read_result);
}
END_TEST
/* Write the first character of a string quickly, followed by the rest of it
* a little later.
*/
static void slow_line_helper (void *data)
{
const char *s = data;
struct timeval timeout;
setbuf (stdout, NULL);
putchar (s[0]);
timeout.tv_sec = 0;
timeout.tv_usec = 100000;
select (0, NULL, NULL, NULL, &timeout);
fputs (s + 1, stdout);
}
START_TEST (test_read_readline_slow)
{
pipeline *p;
pipecmd *cmd;
const char *line;
p = pipeline_new ();
cmd = pipecmd_new_function ("slow_line_helper", slow_line_helper,
free, xstrdup ("a line\nsome more"));
pipeline_command (p, cmd);
pipeline_want_out (p, -1);
pipeline_start (p);
line = pipeline_readline (p);
fail_unless (!strcmp (line, "a line\n"));
pipeline_free (p);
}
END_TEST
Suite *read_suite (void)
{
Suite *s = suite_create ("Read");
TEST_CASE_WITH_FIXTURE (s, read, long_line,
temp_dir_setup, temp_dir_teardown);
TEST_CASE (s, read, readline_slow);
return s;
}
MAIN (read)