1/* $NetBSD: director.c,v 1.10 2012/06/03 23:19:11 joerg Exp $ */ 2 3/*- 4 * Copyright 2009 Brett Lymn <blymn@NetBSD.org> 5 * 6 * All rights reserved. 7 * 8 * This code has been donated to The NetBSD Foundation by the Author. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. The name of the author may not be used to endorse or promote products 16 * derived from this software withough specific prior written permission 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 19 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 20 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 21 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 22 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 23 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 * 29 * 30 */ 31 32#include <sys/param.h> 33#include <sys/stat.h> 34#include <sys/mman.h> 35#include <fcntl.h> 36#include <unistd.h> 37#include <ctype.h> 38#include <termios.h> 39#include <signal.h> 40#include <stdio.h> 41#include <stdlib.h> 42#include <string.h> 43#include <util.h> 44#include <err.h> 45#include "returns.h" 46 47void yyparse(void); 48#define DEF_TERMPATH "." 49#define DEF_TERM "atf" 50#define DEF_SLAVE "./slave" 51 52const char *def_check_path = "./"; /* default check path */ 53const char *def_include_path = "./"; /* default include path */ 54 55extern size_t nvars; /* In testlang_conf.y */ 56saved_data_t saved_output; /* In testlang_conf.y */ 57int cmdpipe[2]; /* command pipe between director and slave */ 58int slvpipe[2]; /* reply pipe back from slave */ 59int master; /* pty to the slave */ 60int verbose; /* control verbosity of tests */ 61const char *check_path; /* path to prepend to check files for output 62 validation */ 63const char *include_path; /* path to prepend to include files */ 64char *cur_file; /* name of file currently being read */ 65 66void init_parse_variables(int); /* in testlang_parse.y */ 67 68/* 69 * Handle the slave exiting unexpectedly, try to recover the exit message 70 * and print it out. 71 */ 72static void 73slave_died(int param) 74{ 75 char last_words[256]; 76 size_t count; 77 78 fprintf(stderr, "ERROR: Slave has exited\n"); 79 if (saved_output.count > 0) { 80 fprintf(stderr, "output from slave: "); 81 for (count = 0; count < saved_output.count; count ++) { 82 if (isprint((unsigned char)saved_output.data[count])) 83 fprintf(stderr, "%c", saved_output.data[count]); 84 } 85 fprintf(stderr, "\n"); 86 } 87 88 if ((count = read(master, &last_words, 255)) > 0) { 89 last_words[count] = '\0'; 90 fprintf(stderr, "slave exited with message \"%s\"\n", 91 last_words); 92 } 93 94 exit(2); 95} 96 97 98static void 99usage(void) 100{ 101 fprintf(stderr, "Usage: %s [-v] [-I include-path] [-C check-path] " 102 "[-T terminfo-file] [-s pathtoslave] [-t term] " 103 "commandfile\n", getprogname()); 104 fprintf(stderr, " where:\n"); 105 fprintf(stderr, " -v enables verbose test output\n"); 106 fprintf(stderr, " -T is a directory containing the terminfo.cdb " 107 "file, or a file holding the terminfo description n"); 108 fprintf(stderr, " -s is the path to the slave executable\n"); 109 fprintf(stderr, " -t is value to set TERM to for the test\n"); 110 fprintf(stderr, " -I is the directory to include files\n"); 111 fprintf(stderr, " -C is the directory for config files\n"); 112 fprintf(stderr, " commandfile is a file of test directives\n"); 113 exit(1); 114} 115 116 117int 118main(int argc, char *argv[]) 119{ 120 extern char *optarg; 121 extern int optind; 122 const char *termpath, *term, *slave; 123 int ch; 124 pid_t slave_pid; 125 extern FILE *yyin; 126 char *arg1, *arg2, *arg3, *arg4; 127 struct termios term_attr; 128 struct stat st; 129 130 termpath = term = slave = NULL; 131 verbose = 0; 132 133 while ((ch = getopt(argc, argv, "vC:I:p:s:t:T:")) != -1) { 134 switch(ch) { 135 case 'I': 136 include_path = optarg; 137 break; 138 case 'C': 139 check_path = optarg; 140 break; 141 case 'T': 142 termpath = optarg; 143 break; 144 case 'p': 145 termpath = optarg; 146 break; 147 case 's': 148 slave = optarg; 149 break; 150 case 't': 151 term = optarg; 152 break; 153 case 'v': 154 verbose = 1; 155 break; 156 case '?': 157 default: 158 usage(); 159 break; 160 } 161 } 162 163 argc -= optind; 164 argv += optind; 165 if (argc < 1) 166 usage(); 167 168 if (termpath == NULL) 169 termpath = DEF_TERMPATH; 170 171 if (slave == NULL) 172 slave = DEF_SLAVE; 173 174 if (term == NULL) 175 term = DEF_TERM; 176 177 if (check_path == NULL) 178 check_path = getenv("CHECK_PATH"); 179 if ((check_path == NULL) || (check_path[0] == '\0')) { 180 warn("$CHECK_PATH not set, defaulting to %s", def_check_path); 181 check_path = def_check_path; 182 } 183 184 if (include_path == NULL) 185 include_path = getenv("INCLUDE_PATH"); 186 if ((include_path == NULL) || (include_path[0] == '\0')) { 187 warn("$INCLUDE_PATH not set, defaulting to %s", 188 def_include_path); 189 include_path = def_include_path; 190 } 191 192 signal(SIGCHLD, slave_died); 193 194 if (setenv("TERM", term, 1) != 0) 195 err(2, "Failed to set TERM variable"); 196 197 if (stat(termpath, &st) == -1) 198 err(1, "Cannot stat %s", termpath); 199 200 if (S_ISDIR(st.st_mode)) { 201 char tinfo[MAXPATHLEN]; 202 int l = snprintf(tinfo, sizeof(tinfo), "%s/%s", termpath, 203 "terminfo.cdb"); 204 if (stat(tinfo, &st) == -1) 205 err(1, "Cannot stat `%s'", tinfo); 206 if (l >= 4) 207 tinfo[l - 4] = '\0'; 208 if (setenv("TERMINFO", tinfo, 1) != 0) 209 err(1, "Failed to set TERMINFO variable"); 210 } else { 211 int fd; 212 char *tinfo; 213 if ((fd = open(termpath, O_RDONLY)) == -1) 214 err(1, "Cannot open `%s'", termpath); 215 if ((tinfo = mmap(NULL, (size_t)st.st_size, PROT_READ, MAP_FILE, 216 fd, 0)) == MAP_FAILED) 217 err(1, "Cannot map `%s'", termpath); 218 if (setenv("TERMINFO", tinfo, 1) != 0) 219 err(1, "Failed to set TERMINFO variable"); 220 close(fd); 221 munmap(tinfo, (size_t)st.st_size); 222 } 223 224 if (pipe(cmdpipe) < 0) 225 err(1, "Command pipe creation failed"); 226 227 if (pipe(slvpipe) < 0) 228 err(1, "Slave pipe creation failed"); 229 230 /* 231 * Create default termios settings for later use 232 */ 233 memset(&term_attr, 0, sizeof(term_attr)); 234 term_attr.c_iflag = TTYDEF_IFLAG; 235 term_attr.c_oflag = TTYDEF_OFLAG; 236 term_attr.c_cflag = TTYDEF_CFLAG; 237 term_attr.c_lflag = TTYDEF_LFLAG; 238 cfsetspeed(&term_attr, TTYDEF_SPEED); 239 term_attr.c_cc[VERASE] = '\b'; 240 term_attr.c_cc[VKILL] = '\025'; /* ^U */ 241 242 if ((slave_pid = forkpty(&master, NULL, &term_attr, NULL)) < 0) 243 err(1, "Fork of pty for slave failed\n"); 244 245 if (slave_pid == 0) { 246 /* slave side, just exec the slave process */ 247 if (asprintf(&arg1, "%d", cmdpipe[0]) < 0) 248 err(1, "arg1 conversion failed"); 249 250 if (asprintf(&arg2, "%d", cmdpipe[1]) < 0) 251 err(1, "arg2 conversion failed"); 252 253 if (asprintf(&arg3, "%d", slvpipe[0]) < 0) 254 err(1, "arg3 conversion failed"); 255 256 if (asprintf(&arg4, "%d", slvpipe[1]) < 0) 257 err(1, "arg4 conversion failed"); 258 259 if (execl(slave, slave, arg1, arg2, arg3, arg4, NULL) < 0) 260 err(1, "Exec of slave %s failed", slave); 261 262 /* NOT REACHED */ 263 } 264 265 fcntl(master, F_SETFL, O_NONBLOCK); 266 267 if ((yyin = fopen(argv[0], "r")) == NULL) 268 err(1, "Cannot open command file %s", argv[0]); 269 270 if ((cur_file = strdup(argv[0])) == NULL) 271 err(2, "Failed to alloc memory for test file name"); 272 273 init_parse_variables(1); 274 275 yyparse(); 276 fclose(yyin); 277 278 exit(0); 279} 280