int
main (int argc, char *argv[])
{
int sd,res;
struct sockaddr_in sin_buf;
struct sockaddr_in * sin = &sin_buf;
char buffer [BUFSIZ];
char dishname [BUFSIZ];
char * ptr;
isodetailor (argv[0], 1);
if ((sd = start_tcp_client ((struct sockaddr_in *) 0, 0)) == NOTOK) {
perror("start_tcp_client");
exit(-20);
}
if (get_dish_sock (sin, 0, 1) != 0)
exit (-21);
if (join_tcp_server (sd, sin) == NOTOK) {
int pid;
close_tcp_socket (sd);
fork_again:
;
switch (pid = vfork ()) {
case 0:
/* child */
close_tcp_socket (sd);
strcpy (dishname,
_isodefile (isodebinpath, "dish"));
{
int i, nfds = getdtablesize ();
for (i = fileno(stderr) + 1; i < nfds; i++)
close (i);
}
execl (dishname, "dish","-pipe",NULLCP);
fprintf (stderr, "unable to exec ");
perror (dishname);
_exit (-21);
case -1:
perror ("fork");
close_tcp_socket (sd);
exit (-22);
default:
/* parent */
for (;;) {
if ((sd = start_tcp_client ((struct sockaddr_in *) 0,
0)) == NOTOK) {
perror("start_tcp_client");
exit(-23);
}
if (join_tcp_server (sd, sin) != NOTOK)
break;
/* need to introduce a timeout !!! */
close_tcp_socket (sd);
sleep (5);
if (kill (pid, 0) == NOTOK) {
fprintf (stderr,"Trying again...\n");
goto fork_again;
}
}
break;
}
}
if ((ptr = rindex (argv[0], '/')) == NULLCP)
strcpy (buffer,argv[0]);
else
strcpy (buffer,++ptr);
argc--,argv++;
while (argc--) {
strcat (buffer, " \"");
strcat (buffer, *argv++);
strcat (buffer, "\"");
}
strcat (buffer, "\n");
if (send(sd, buffer, strlen(buffer), 0) == -1) {
perror("send");
close_tcp_socket (sd);
exit (-25);
}
for (;;) {
if ((res = recv(sd, buffer, BUFSIZ - 1, 0)) == -1) {
err_recv:
;
perror ("recv");
close_tcp_socket (sd);
exit (-26);
}
*(buffer + res) = 0;
if (res == 0) {
close_tcp_socket (sd);
exit (0);
}
if (*buffer == '2') {
if (res > 1)
write (2,&buffer[1],--res);
while ( (res = recv(sd, buffer, BUFSIZ, 0)) > 0)
write (2,buffer,res);
close_tcp_socket (sd);
exit (1);
} else if ((*buffer == '1') || (*buffer == '3')) {
int eval;
eval = (*buffer == '1' ? 0 : 2);
if (res > 1)
write (1,&buffer[1],--res);
while ( (res = recv(sd, buffer, BUFSIZ, 0)) > 0)
write (1,buffer,res);
close_tcp_socket (sd);
exit (eval);
} else { /* 'e', 'y', 'm', or 'p' */
char *cp, *ep;
char where[BUFSIZ];
cp = buffer + res - 1;
ep = buffer + sizeof buffer - 1;
while (*cp != '\n') {
++cp;
switch (res = recv (sd, cp, ep - cp, 0)) {
case NOTOK:
goto err_recv;
case OK:
fprintf (stderr,
"eof reading '%c' directive\n",
*buffer);
exit (-28);
default:
cp += res - 1;
if (cp < ep)
continue;
fprintf (stderr,
"'%c' directive exceeds %d octets\n",
*buffer, sizeof buffer - 1);
exit(-29);
}
}
*cp = NULL;
if (*buffer == 'e') {
if (system (&buffer[1]))
strcpy (where, "e");
else
getcwd (where, sizeof where);
} else if (*buffer == 'm') {
fprintf (stderr, "\n%s\n", buffer + 1);
strcpy (where, "m");
} else if (*buffer == 'y') {
fprintf (stderr,"%s",buffer + 1);
fgets (where, sizeof where, stdin);
if (cp = index (where, '\n'))
*cp = NULL;
} else { /* 'p' */
sprintf (where,
"Enter password for \"%s\": ",
buffer + 1);
sprintf (where, "p%s",
getpassword (where));
}
strcat (where, "\n");
if (send(sd, where, strlen(where), 0) == -1) {
perror("send");
close_tcp_socket (sd);
exit (-27);
}
}
}
}
int main(int argc, char **argv) {
char* port = NULL;
char* host = NULL;
int index;
int c;
int test;
unsigned int msg = 0;
packet_type type = INVALID_PACKET_TYPE;
opterr = 0;
if (argc == 1) {
fprintf(stderr, "Must specify port, packet type, server, and data\n");
return 1;
}
while ((c = getopt(argc, argv, "p:t:x:s:")) != -1) {
switch (c) {
case 'p':
if (parsePortNumber(LOWER_IP, UPPER_IP, optarg) == NULL) {
fprintf(stderr, "Invalid port %s. Must use ephemeral or dynamic port within range of 49152-65535 so as to prevent stepping on registered ports.\n", optarg);
} else {
port = optarg;
}
break;
case 't':
//Value ucp or tcp
type = parsePacketType(optarg);
break;
case 'x':
// Data to send
test = atoi(optarg);
if ((*optarg != '0' && test == 0) ||
//(test == INT_MAX || test == INT_MIN) || Need to find these consts
(test < 0)) {
fprintf(stderr, "Invalid data. Data must be a positive 32 bit integer.");
return 1;
}
msg = (unsigned int)test;
break;
case 's':
if (gethostbyname(optarg) == NULL) {
fprintf(stderr, "Invalid hostname.\n");
return 2;
}
host = optarg;
break;
case '?':
if (optopt == 't') {
fprintf(stderr, "Option -%c requires an argument. (tcp | udp)\n", optopt);
} else if (optopt == 'p') {
fprintf(stderr, "Option -%c requires an argument (-p <port number from 1-65535>)\n", optopt);
} else if (optopt == 's') {
fprintf(stderr, "Option -%c requires an argument (-s <valid hostname>)\n", optopt);
} else if (optopt == 'x') {
fprintf(stderr, "Option -%c requires an argument (-x <some positive 32 bit int>)\n", optopt);
} else if (isprint(optopt)) {
fprintf(stderr, "Unknown option `-%c'.\n", optopt);
} else {
fprintf(stderr, "Unknown option character `\\x%x'.\n", optopt);
}
return 1;
default:
abort();
}
}
for (index = optind; index < argc; index++) {
printf("Non-option argument %s, ignoring.\n", argv[index]);
}
if (type == UDP) {
start_udp_client(host, port, msg);
} else if (type == TCP) {
start_tcp_client(host, port, msg);
} else {
fprintf(stderr, "Invalid packet type, must be tcp or udp.");
}
return 0;
}
