diff options
Diffstat (limited to 'vendor/voclient/voapps/session_cmd.c')
| -rw-r--r-- | vendor/voclient/voapps/session_cmd.c | 171 |
1 files changed, 171 insertions, 0 deletions
diff --git a/vendor/voclient/voapps/session_cmd.c b/vendor/voclient/voapps/session_cmd.c new file mode 100644 index 00000000..41b70451 --- /dev/null +++ b/vendor/voclient/voapps/session_cmd.c @@ -0,0 +1,171 @@ +/** + * SESSION_CMD -- Session Manager test client. + * + * @file session_cmd.c + * @author Mike Fitzpatrick + * @date 6/03/12 + * + * @brief Session Manager test client. + */ + +#include <stdio.h> +#include <stdlib.h> +#include <unistd.h> +#include <string.h> +#include <ctype.h> +#include <fcntl.h> +#include <time.h> + +#include <netdb.h> +#include <sys/errno.h> +#include <sys/socket.h> +#include <netinet/in.h> +#include <arpa/inet.h> + +#include "samp.h" /* SAMP interface */ +#include "voApps.h" /* voApps interface */ + + +#define SZ_BUF 1024 +#define SELWIDTH 4 + + +/* Utility socket routines. + */ +extern int vos_uploadFiles (int sock, char *cmd); +extern int vos_openClientSocket (char *host, int port, int retry); +extern int vos_sockRead (int fd, void *vptr, int nbytes); +extern int vos_sockWrite (int fd, void *vptr, int nbytes); +extern int vos_sockWriteHdr (int fd, int len, char *name, int type, + int mode, char *to); +extern int vos_sockReadHdr (int fd, int *len, char *name, int *type, + int *mode); + +extern int vos_openSession (char *host, int port, char *session_name); +extern int vos_closeSession (int sock); + +extern void vos_setNonBlock (int sock); +extern struct hostent *vos_getHostByName (char *lhost); + + + +/**************************************************************************** + * Program entry point. + */ +int +main (int argc, char **argv) +{ + int i, j, rc, debug=0, len=0, nw=0, cb_sock=0, nfiles=0; + int mgr_port = SESS_DEFPORT; + char *mgr_host = SESS_DEFHOST; + char *sname = "test"; + char buf[SZ_BUF]; + + fd_set allset, fds; + + + /* Process cmdline args. + */ + for (i=1; i < argc; i++) { + for (j=1; j < argc; j++) { + switch ( argv[i][j] ) { + case 'd': debug++; break; + + default: + fprintf (stderr, "Invalid option '%c'\n", argv[i][0]); + return (1); + } + } + } + + + + /* Open a client connection to the VOSAMP Session Manager. + */ + if ((cb_sock = vos_openSession (mgr_host, mgr_port, sname)) == 0) { + fprintf (stderr, "Cannot open connection to session manager\n"); + return (1); + } + + /* Take commands from the stdin, write results to stdout. + */ + printf ("vosession> "); fflush (stdout); + + FD_ZERO (&allset); + FD_SET (cb_sock, &allset); vos_setNonBlock (cb_sock); + FD_SET (fileno(stdin), &allset); vos_setNonBlock (fileno(stdin)); + + + /* Process input either from the stdin or the session manager. + */ + while (1) { + /* Initialize the input file descriptor set. + */ + memcpy (&fds, &allset, sizeof(allset)); + rc = select (SELWIDTH, &fds, NULL, NULL, NULL); + + for (i=0; i < SELWIDTH+1; i++) { + if (FD_ISSET(i, &fds)) { + if (i == fileno(stdin)) { + if (fgets (buf, SZ_BUF, stdin) == NULL) + goto done_; /* EOF, time to quit */ + buf[strlen(buf)-1] = '\0'; /* kill newline */ + + } else { + int nread, nbytes, type, mode; + char fname[SZ_FNAME]; + + if (vos_sockReadHdr (i, &nbytes, fname, &type, &mode)) + nread = vos_sockRead (i, buf, nbytes); + else + goto done_; + + switch (type) { + case SAMP_QUIT: + goto done_; + case SAMP_RESULT: + printf ("\r\n%s\n", buf); + goto next_cmd; + case SAMP_CMD: + printf ("Got cmd: '%s'\n", buf); + goto next_cmd; + case SAMP_TEST: + printf ("Got test cmd: '%s'\n", buf); + goto next_cmd; + case SAMP_RELAY: + printf ("Got relayed cmd: '%s'\n", buf); + goto next_cmd; + default: + printf ("Got msg type: %d\n", type); + goto next_cmd; + } + } + } + } + + /* If we want to quit, let the vos_closeSession() send the 'quit'. + */ + if (strncasecmp (buf, "quit", 4) == 0) + break; + + /* Upload any local files to the session manager. + */ + nfiles = vos_uploadFiles (cb_sock, buf); + if (debug) + if (nfiles) fprintf (stderr, "sent %d data files....\n", nfiles); + + /* Send the command string. + */ + len = strlen (buf); + if (vos_sockWriteHdr(cb_sock, len, NULL, SAMP_CMD, SAMP_NOTIFY, "smgr")) + nw = vos_sockWrite (cb_sock, buf, len); + +next_cmd: + memset (buf, 0, SZ_BUF); + printf ("vosession> "); fflush (stdout); + } + +done_: + vos_closeSession (cb_sock); + return (0); +} |