1 /* Test harness for pipe-filter-gi.
2
3 Copyright (C) 2009-2021 Free Software Foundation, Inc.
4 Written by Paolo Bonzini <bonzini@gnu.org>, 2009.
5
6 This program is free software: you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
10
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
15
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <https://www.gnu.org/licenses/>. */
18
19 #include <config.h>
20
21 #include "pipe-filter.h"
22
23 #include <stdio.h>
24 #include <errno.h>
25 #include <unistd.h>
26 #include <string.h>
27 #include <signal.h>
28
29 #include "full-write.h"
30 #include "macros.h"
31
32 /* 0.1 sec pause */
33 static void
small_nap(void)34 small_nap (void)
35 {
36 usleep (100000);
37 }
38
39 static char static_buf[5];
40
41 static void *
prepare_read(size_t * num_bytes_p,void * private_data)42 prepare_read (size_t *num_bytes_p, void *private_data)
43 {
44 *num_bytes_p = sizeof (static_buf);
45 return static_buf;
46 }
47
48 /* Callback that ignores the data that has been read. */
49
50 static void
ignore_done_read(void * data_read,size_t num_bytes_read,void * private_data)51 ignore_done_read (void *data_read, size_t num_bytes_read, void *private_data)
52 {
53 }
54
55 /* Callback that outputs the data that has been read. */
56
57 static void
output_done_read(void * data_read,size_t num_bytes_read,void * private_data)58 output_done_read (void *data_read, size_t num_bytes_read, void *private_data)
59 {
60 full_write (STDOUT_FILENO, data_read, num_bytes_read);
61 }
62
63 static void
pipe_filter_gi_write_string(struct pipe_filter_gi * f,const char * string)64 pipe_filter_gi_write_string (struct pipe_filter_gi *f, const char *string)
65 {
66 pipe_filter_gi_write (f, string, strlen (string));
67 }
68
69 int
main(int argc,char ** argv)70 main (int argc, char **argv)
71 {
72 struct pipe_filter_gi *f;
73 const char *path[] = { NULL, NULL };
74
75 ASSERT (argc == 2);
76
77 /* Test writing to a nonexistent program traps sooner or later. */
78 {
79 int rc;
80
81 path[0] = "/nonexistent/blah";
82 f = pipe_filter_gi_create ("pipe-filter-test", path[0], path, true, false,
83 prepare_read, ignore_done_read, NULL);
84 small_nap ();
85 rc = pipe_filter_gi_write (f, "", 1);
86 ASSERT (rc == 127 || rc == -1);
87 rc = pipe_filter_gi_close (f);
88 ASSERT (rc == 127 || rc == -1);
89 printf ("Test 1 passed.\n");
90 fflush (stdout);
91 }
92
93 /* Test returning the exit status. */
94 {
95 path[0] = argv[1];
96 f = pipe_filter_gi_create ("pipe-filter-test", path[0], path, false, false,
97 prepare_read, ignore_done_read, NULL);
98 pipe_filter_gi_write_string (f, "1 -1");
99 ASSERT (pipe_filter_gi_close (f) == 1);
100 printf ("Test 2 passed.\n");
101 fflush (stdout);
102 }
103
104 /* Same, but test returning the status in pipe_filter_gi_write. */
105 {
106 int rc;
107
108 path[0] = argv[1];
109 f = pipe_filter_gi_create ("pipe-filter-test", path[0], path, false, false,
110 prepare_read, ignore_done_read, NULL);
111 pipe_filter_gi_write_string (f, "1 -1\n"); /* tell the child to terminate */
112 small_nap (); /* let the child terminate */
113 rc = pipe_filter_gi_write (f, " ", 1); /* write to a closed pipe */
114 ASSERT (rc == -1 && errno == EPIPE);
115 /* Closing the filter must report same error again, for consistency with
116 pipe_filter_ii_execute. The subprocess' exit status is not returned. */
117 rc = pipe_filter_gi_close (f);
118 ASSERT (rc == -1 && errno == EPIPE);
119 printf ("Test 3 passed.\n");
120 fflush (stdout);
121 }
122
123 /* Now test asynchronous I/O. */
124 {
125 path[0] = argv[1];
126 f = pipe_filter_gi_create ("pipe-filter-test", path[0], path, false, true,
127 prepare_read, output_done_read, NULL);
128 pipe_filter_gi_write_string (f, "1 50\n");
129 pipe_filter_gi_write_string (f, "51\n");
130 pipe_filter_gi_write_string (f, "100");
131 ASSERT (pipe_filter_gi_close (f) == 0);
132 printf ("Test 4 passed.\n");
133 fflush (stdout);
134 }
135
136 return 0;
137 }
138