1 /* 2 * Run a program using the supplied arguments. 3 * 4 * Copyright (C) 2024 Paul Boddie <paul@boddie.org.uk> 5 * 6 * This program is free software; you can redistribute it and/or 7 * modify it under the terms of the GNU General Public License as 8 * published by the Free Software Foundation; either version 2 of 9 * the License, or (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, write to the Free Software 18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, 19 * Boston, MA 02110-1301, USA 20 */ 21 22 #include <errno.h> 23 #include <fcntl.h> 24 #include <port.h> 25 #include <process.h> 26 #include <stdio.h> 27 #include <sys/types.h> 28 29 30 31 /* Transfer size for communication. */ 32 33 static const offset_t TO_TRANSFER = 1024; 34 35 /* Transfer data between streams. */ 36 37 static void transfer(FILE *input, FILE *output, char *buffer, int *start, size_t *remaining) 38 { 39 size_t nwritten; 40 41 /* Transfer any previously read data before trying to read more. */ 42 43 while (*remaining || (*start = 0, *remaining = fread(buffer, sizeof(char), TO_TRANSFER, input))) 44 { 45 /* Write the data, stopping if no data can be written. */ 46 47 nwritten = fwrite(buffer + *start, sizeof(char), *remaining, output); 48 49 if (!nwritten) 50 return; 51 52 /* Record the remaining amount. */ 53 54 *remaining -= nwritten; 55 *start += nwritten; 56 } 57 } 58 59 60 61 /* List objects in the filesystem image. */ 62 63 int main(int argc, char *argv[]) 64 { 65 FILE *input, *output, *error; 66 pid_t pid; 67 port_t port; 68 event_source_t source; 69 event_values_t result; 70 71 /* Transfer buffers and status. */ 72 73 char input_buffer[TO_TRANSFER]; 74 char output_buffer[TO_TRANSFER]; 75 char error_buffer[TO_TRANSFER]; 76 int input_start = 0, output_start = 0, error_start = 0; 77 size_t input_remaining = 0, output_remaining = 0, error_remaining = 0; 78 79 if (argc < 2) 80 { 81 printf("Usage: test_popenv <program> <argument>...\n"); 82 return 1; 83 } 84 85 printf("Running: %s\n", argv[1]); 86 87 pid = popenv(argc - 1, (const char **) argv + 1, &input, &output, &error); 88 89 if (pid == -1) 90 { 91 printf("Error: %d\n", errno); 92 return 1; 93 } 94 95 printf("Process: %d\n", pid); 96 97 /* Prevent stdin from blocking to be able to read as much data as is 98 available and to monitor events on it manually. */ 99 100 if (stdin != NULL) 101 fcntl(fileno(stdin), F_SETFL, O_NONBLOCK | fcntl(fileno(stdin), F_GETFL)); 102 103 /* Monitor the streams and the process. */ 104 105 port = port_create(); 106 107 port_subscribe(port, port_process_source(pid), PORT_PROCESS_ALL); 108 109 if (stdin != NULL) 110 { 111 port_subscribe(port, port_file_source(fileno(stdin)), PORT_INPUT | PORT_CLOSE); 112 port_subscribe(port, port_file_source(fileno(input)), PORT_OUTPUT | PORT_CLOSE); 113 } 114 115 port_subscribe(port, port_file_source(fileno(output)), PORT_INPUT | PORT_CLOSE); 116 117 if (error != NULL) 118 port_subscribe(port, port_file_source(fileno(error)), PORT_INPUT | PORT_CLOSE); 119 120 /* Wait for events on the streams and the process. */ 121 122 while (1) 123 { 124 if (port_wait(port, &source)) 125 { 126 printf("Error waiting for notifications.\n"); 127 return 1; 128 } 129 130 /* Handle input to this process and its transfer to the created 131 process. */ 132 133 if ((stdin != NULL) && (input != NULL)) 134 { 135 /* Attempt to transfer input. */ 136 137 if ((source == port_file_source(fileno(stdin))) || 138 input_remaining || (source == port_file_source(fileno(input)))) 139 transfer(stdin, input, input_buffer, &input_start, &input_remaining); 140 141 /* If no input remains and stdin has been closed, close the input 142 pipe. */ 143 144 if (!input_remaining && port_source_event_flags(port_file_source(fileno(stdin))) & PORT_CLOSE) 145 { 146 fclose(input); 147 input = NULL; 148 } 149 } 150 151 /* Handle output from the process. */ 152 153 if (output_remaining || (source == port_file_source(fileno(output)))) 154 transfer(output, stdout, output_buffer, &output_start, &output_remaining); 155 156 if ((error != NULL) && (error_remaining || (source == port_file_source(fileno(error))))) 157 transfer(error, stderr, error_buffer, &error_start, &error_remaining); 158 159 /* Handle process termination. */ 160 161 else if (source == port_process_source(pid)) 162 { 163 transfer(output, stdout, output_buffer, &output_start, &output_remaining); 164 165 if (error != NULL) 166 transfer(error, stderr, error_buffer, &error_start, &error_remaining); 167 168 result = port_source_event_values(source); 169 break; 170 } 171 } 172 173 /* Close any open pipes. */ 174 175 if (input != NULL) 176 fclose(input); 177 178 fclose(output); 179 180 if (error != NULL) 181 fclose(error); 182 183 printf("Result: %ld\n", result.val); 184 return 0; 185 } 186 187 /* vim: tabstop=4 expandtab shiftwidth=4 188 */