1/* 2 * Copyright (C) 2004-2005 Kay Sievers <kay.sievers@vrfy.org> 3 * 4 * This program is free software; you can redistribute it and/or modify it 5 * under the terms of the GNU General Public License as published by the 6 * Free Software Foundation version 2 of the License. 7 * 8 * This program is distributed in the hope that it will be useful, but 9 * WITHOUT ANY WARRANTY; without even the implied warranty of 10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 11 * General Public License for more details. 12 * 13 * You should have received a copy of the GNU General Public License along 14 * with this program; if not, write to the Free Software Foundation, Inc., 15 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. 16 * 17 */ 18 19 20#include <stdlib.h> 21#include <stdio.h> 22#include <stddef.h> 23#include <unistd.h> 24#include <fcntl.h> 25#include <errno.h> 26#include <ctype.h> 27#include <sys/socket.h> 28#include <sys/un.h> 29#include <sys/wait.h> 30#include <sys/select.h> 31 32#include "udev.h" 33 34extern char **environ; 35 36int pass_env_to_socket(const char *sockname, const char *devpath, const char *action) 37{ 38 int sock; 39 struct sockaddr_un saddr; 40 socklen_t addrlen; 41 char buf[2048]; 42 size_t bufpos = 0; 43 int i; 44 ssize_t count; 45 int retval = 0; 46 47 dbg("pass environment to socket '%s'", sockname); 48 sock = socket(AF_LOCAL, SOCK_DGRAM, 0); 49 memset(&saddr, 0x00, sizeof(struct sockaddr_un)); 50 saddr.sun_family = AF_LOCAL; 51 /* abstract namespace only */ 52 strcpy(&saddr.sun_path[1], sockname); 53 addrlen = offsetof(struct sockaddr_un, sun_path) + strlen(saddr.sun_path+1) + 1; 54 55 bufpos = snprintf(buf, sizeof(buf)-1, "%s@%s", action, devpath); 56 bufpos++; 57 for (i = 0; environ[i] != NULL && bufpos < sizeof(buf); i++) { 58 bufpos += strlcpy(&buf[bufpos], environ[i], sizeof(buf) - bufpos-1); 59 bufpos++; 60 } 61 62 count = sendto(sock, &buf, bufpos, 0, (struct sockaddr *)&saddr, addrlen); 63 if (count < 0) 64 retval = -1; 65 info("passed %zi bytes to socket '%s', ", count, sockname); 66 67 close(sock); 68 return retval; 69} 70 71int run_program(const char *command, const char *subsystem, 72 char *result, size_t ressize, size_t *reslen, int log) 73{ 74 int retval = 0; 75 int status; 76 int outpipe[2] = {-1, -1}; 77 int errpipe[2] = {-1, -1}; 78 pid_t pid; 79 char arg[PATH_SIZE]; 80 char program[PATH_SIZE]; 81 char *argv[(sizeof(arg) / 2) + 1]; 82 int devnull; 83 int i; 84 85 /* build argv from comand */ 86 strlcpy(arg, command, sizeof(arg)); 87 i = 0; 88 if (strchr(arg, ' ') != NULL) { 89 char *pos = arg; 90 91 while (pos != NULL) { 92 if (pos[0] == '\'') { 93 /* don't separate if in apostrophes */ 94 pos++; 95 argv[i] = strsep(&pos, "\'"); 96 while (pos != NULL && pos[0] == ' ') 97 pos++; 98 } else { 99 argv[i] = strsep(&pos, " "); 100 } 101 dbg("arg[%i] '%s'", i, argv[i]); 102 i++; 103 } 104 argv[i] = NULL; 105 } else { 106 argv[0] = arg; 107 argv[1] = NULL; 108 } 109 info("'%s'", command); 110 111 /* prepare pipes from child to parent */ 112 if (result || log) { 113 if (pipe(outpipe) != 0) { 114 err("pipe failed: %s", strerror(errno)); 115 return -1; 116 } 117 } 118 if (log) { 119 if (pipe(errpipe) != 0) { 120 err("pipe failed: %s", strerror(errno)); 121 return -1; 122 } 123 } 124 125 /* allow programs in /lib/udev called without the path */ 126 if (strchr(argv[0], '/') == NULL) { 127 strlcpy(program, "/lib/udev/", sizeof(program)); 128 strlcat(program, argv[0], sizeof(program)); 129 argv[0] = program; 130 } 131 132 pid = fork(); 133 switch(pid) { 134 case 0: 135 /* child closes parent ends of pipes */ 136 if (outpipe[READ_END] > 0) 137 close(outpipe[READ_END]); 138 if (errpipe[READ_END] > 0) 139 close(errpipe[READ_END]); 140 141 /* discard child output or connect to pipe */ 142 devnull = open("/dev/null", O_RDWR); 143 if (devnull > 0) { 144 dup2(devnull, STDIN_FILENO); 145 if (outpipe[WRITE_END] < 0) 146 dup2(devnull, STDOUT_FILENO); 147 if (errpipe[WRITE_END] < 0) 148 dup2(devnull, STDERR_FILENO); 149 close(devnull); 150 } else 151 err("open /dev/null failed: %s", strerror(errno)); 152 if (outpipe[WRITE_END] > 0) { 153 dup2(outpipe[WRITE_END], STDOUT_FILENO); 154 close(outpipe[WRITE_END]); 155 } 156 if (errpipe[WRITE_END] > 0) { 157 dup2(errpipe[WRITE_END], STDERR_FILENO); 158 close(errpipe[WRITE_END]); 159 } 160 execv(argv[0], argv); 161 if (errno == ENOENT || errno == ENOTDIR) { 162 /* may be on a filesytem which is not mounted right now */ 163 info("program '%s' not found", argv[0]); 164 } else { 165 /* other problems */ 166 err("exec of program '%s' failed", argv[0]); 167 } 168 _exit(1); 169 case -1: 170 err("fork of '%s' failed: %s", argv[0], strerror(errno)); 171 return -1; 172 default: 173 /* read from child if requested */ 174 if (outpipe[READ_END] > 0 || errpipe[READ_END] > 0) { 175 ssize_t count; 176 size_t respos = 0; 177 178 /* parent closes child ends of pipes */ 179 if (outpipe[WRITE_END] > 0) 180 close(outpipe[WRITE_END]); 181 if (errpipe[WRITE_END] > 0) 182 close(errpipe[WRITE_END]); 183 184 /* read child output */ 185 while (outpipe[READ_END] > 0 || errpipe[READ_END] > 0) { 186 int fdcount; 187 fd_set readfds; 188 189 FD_ZERO(&readfds); 190 if (outpipe[READ_END] > 0) 191 FD_SET(outpipe[READ_END], &readfds); 192 if (errpipe[READ_END] > 0) 193 FD_SET(errpipe[READ_END], &readfds); 194 fdcount = select(UDEV_MAX(outpipe[READ_END], errpipe[READ_END])+1, &readfds, NULL, NULL, NULL); 195 if (fdcount < 0) { 196 if (errno == EINTR) 197 continue; 198 retval = -1; 199 break; 200 } 201 202 /* get stdout */ 203 if (outpipe[READ_END] > 0 && FD_ISSET(outpipe[READ_END], &readfds)) { 204 char inbuf[1024]; 205 char *pos; 206 char *line; 207 208 count = read(outpipe[READ_END], inbuf, sizeof(inbuf)-1); 209 if (count <= 0) { 210 close(outpipe[READ_END]); 211 outpipe[READ_END] = -1; 212 if (count < 0) { 213 err("stdin read failed: %s", strerror(errno)); 214 retval = -1; 215 } 216 continue; 217 } 218 inbuf[count] = '\0'; 219 220 /* store result for rule processing */ 221 if (result) { 222 if (respos + count < ressize) { 223 memcpy(&result[respos], inbuf, count); 224 respos += count; 225 } else { 226 err("ressize %ld too short", (long)ressize); 227 retval = -1; 228 } 229 } 230 pos = inbuf; 231 while ((line = strsep(&pos, "\n"))) 232 if (pos || line[0] != '\0') 233 info("'%s' (stdout) '%s'", argv[0], line); 234 } 235 236 /* get stderr */ 237 if (errpipe[READ_END] > 0 && FD_ISSET(errpipe[READ_END], &readfds)) { 238 char errbuf[1024]; 239 char *pos; 240 char *line; 241 242 count = read(errpipe[READ_END], errbuf, sizeof(errbuf)-1); 243 if (count <= 0) { 244 close(errpipe[READ_END]); 245 errpipe[READ_END] = -1; 246 if (count < 0) 247 err("stderr read failed: %s", strerror(errno)); 248 continue; 249 } 250 errbuf[count] = '\0'; 251 pos = errbuf; 252 while ((line = strsep(&pos, "\n"))) 253 if (pos || line[0] != '\0') 254 info("'%s' (stderr) '%s'", argv[0], line); 255 } 256 } 257 if (outpipe[READ_END] > 0) 258 close(outpipe[READ_END]); 259 if (errpipe[READ_END] > 0) 260 close(errpipe[READ_END]); 261 262 /* return the childs stdout string */ 263 if (result) { 264 result[respos] = '\0'; 265 dbg("result='%s'", result); 266 if (reslen) 267 *reslen = respos; 268 } 269 } 270 waitpid(pid, &status, 0); 271 if (WIFEXITED(status)) { 272 info("'%s' returned with status %i", argv[0], WEXITSTATUS(status)); 273 if (WEXITSTATUS(status) != 0) 274 retval = -1; 275 } else { 276 err("'%s' abnormal exit", argv[0]); 277 retval = -1; 278 } 279 } 280 281 return retval; 282} 283