1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
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);
}
|
