2 * Copyright (c) 2008 Kungliga Tekniska Högskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
17 * 3. Neither the name of the Institute nor the names of its contributors
18 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
21 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35 #include "system/filesys.h"
36 #include "system/wait.h"
44 #ifdef HAVE_BSD_LIBUTIL_H
45 #include <bsd/libutil.h>
46 #elif defined HAVE_LIBUTIL_H
52 #endif /* STREAMPTY */
58 enum { CMD_EXPECT
= 0, CMD_SEND
, CMD_PASSWORD
} type
;
68 static struct command
*commands
, **next
= &commands
;
70 static sig_atomic_t alarmset
= 0;
72 static int opt_timeout
= 10;
73 static int opt_verbose
;
77 static char line
[256] = { 0 };
79 static void caught_signal(int signo
)
85 static void open_pty(void)
88 printf("implement open_pty\n");
91 #if defined(HAVE_OPENPTY) || defined(__linux) || defined(__osf__) /* XXX */
92 if(openpty(&master
, &slave
, line
, 0, 0) == 0)
94 #endif /* HAVE_OPENPTY .... */
106 for(q
= clone
; *q
; q
++){
107 master
= open(*q
, O_RDWR
);
115 strlcpy(line
, ptsname(master
), sizeof(line
));
116 slave
= open(line
, O_RDWR
);
118 errx(1, "failed to open slave when using %s", *q
);
119 ioctl(slave
, I_PUSH
, "ptem");
120 ioctl(slave
, I_PUSH
, "ldterm");
126 #endif /* STREAMSPTY */
128 /* more cases, like open /dev/ptmx, etc */
137 static char *iscmd(const char *buf
, const char *s
)
139 size_t len
= strlen(s
);
141 if (strncmp(buf
, s
, len
) != 0) {
145 return strdup(buf
+ len
);
148 /*******************************************************************
149 A write wrapper that will deal with EINTR.
150 ********************************************************************/
152 static ssize_t
sys_write(int fd
, const void *buf
, size_t count
)
157 ret
= write(fd
, buf
, count
);
158 #if defined(EWOULDBLOCK)
159 } while (ret
== -1 && (errno
== EINTR
|| errno
== EAGAIN
|| errno
== EWOULDBLOCK
));
161 } while (ret
== -1 && (errno
== EINTR
|| errno
== EAGAIN
));
166 static void parse_configuration(const char *fn
)
171 unsigned int lineno
= 0;
174 cmd
= fopen(fn
, "r");
176 err(1, "open: %s", fn
);
178 while (fgets(s
, sizeof(s
), cmd
) != NULL
) {
180 s
[strcspn(s
, "#\n")] = '\0';
183 c
= calloc(1, sizeof(*c
));
191 if ((str
= iscmd(s
, "expect ")) != NULL
) {
192 c
->type
= CMD_EXPECT
;
194 } else if ((str
= iscmd(s
, "send ")) != NULL
) {
197 } else if ((str
= iscmd(s
, "password ")) != NULL
) {
198 c
->type
= CMD_PASSWORD
;
201 errx(1, "Invalid command on line %d: %s", lineno
, s
);
207 /* A wrapper to close als file descriptors above the given fd */
208 static int sys_closefrom(int fd
)
210 int num
= getdtablesize();
216 for (; fd
<= num
; fd
++) {
228 static int eval_parent(pid_t pid
)
235 for (c
= commands
; c
!= NULL
; c
= c
->next
) {
239 printf("[expecting %s]\n", c
->str
);
243 while((sret
= read(master
, &in
, sizeof(in
))) > 0) {
246 if (c
->str
[len
] != in
) {
251 if (c
->str
[len
] == '\0') {
256 if (alarmset
== SIGALRM
) {
257 errx(1, "timeout waiting for %s (line %u)",
259 } else if (alarmset
) {
260 errx(1, "got a signal %d waiting for %s (line %u)",
261 (int)alarmset
, c
->str
, c
->lineno
);
265 errx(1, "end command while waiting for %s (line %u)",
272 const char *msg
= (c
->type
== CMD_PASSWORD
) ? "****" : c
->str
;
275 printf("[send %s]\n", msg
);
278 len
= strlen(c
->str
);
281 if (c
->str
[i
] == '\\' && i
< len
- 1) {
296 "unknown control char %c (line %u)",
300 if (sys_write(master
, &ctrl
, 1) != 1) {
301 errx(1, "command refused input (line %u)", c
->lineno
);
304 if (sys_write(master
, &c
->str
[i
], 1) != 1) {
305 errx(1, "command refused input (line %u)", c
->lineno
);
317 while(read(master
, &in
, sizeof(in
)) > 0) {
322 printf("[end of program]\n");
326 * Fetch status from child
331 ret
= waitpid(pid
, &status
, 0);
336 if (WIFEXITED(status
) && WEXITSTATUS(status
)) {
337 return WEXITSTATUS(status
);
338 } else if (WIFSIGNALED(status
)) {
339 printf("killed by signal: %d\n", WTERMSIG(status
));
350 struct poptOption long_options
[] = {
352 {"timeout", 't', POPT_ARG_INT
, &opt_timeout
, 't'},
353 {"verbose", 'v', POPT_ARG_NONE
, &opt_verbose
, 'v'},
357 int main(int argc
, const char **argv
)
362 const char *instruction_file
;
365 char * const *program_args
;
367 pc
= poptGetContext("texpect",
371 POPT_CONTEXT_POSIXMEHARDER
);
374 poptPrintHelp(pc
, stderr
, 0);
378 while ((optidx
= poptGetNextOpt(pc
)) != -1) {
382 instruction_file
= poptGetArg(pc
);
383 args
= poptGetArgs(pc
);
384 program_args
= (char * const *)discard_const_p(char *, args
);
385 program
= program_args
[0];
390 printf("Using instruction_file: %s\n", instruction_file
);
391 printf("Executing '%s' ", program
);
392 for (i
= 0; program_args
&& program_args
[i
] != NULL
; i
++) {
393 printf("'%s' ", program_args
[i
]);
398 parse_configuration(instruction_file
);
405 err(1, "Failed to fork");
411 dup2(slave
, STDIN_FILENO
);
412 dup2(slave
, STDOUT_FILENO
);
413 dup2(slave
, STDERR_FILENO
);
415 sys_closefrom(STDERR_FILENO
+ 1);
417 /* texpect <expect_instructions> <progname> [<args>] */
418 execvp(program
, program_args
);
419 err(1, "Failed to exec: %s", program
);
425 sa
.sa_handler
= caught_signal
;
427 sigemptyset (&sa
.sa_mask
);
429 sigaction(SIGALRM
, &sa
, NULL
);
432 return eval_parent(pid
);