1 | /* |
2 | * Copyright 6WIND S.A., 2014 |
3 | * |
4 | * This work is licensed under the terms of the GNU GPL, version 2 or |
5 | * (at your option) any later version. See the COPYING file in the |
6 | * top-level directory. |
7 | */ |
8 | |
9 | #include "qemu/osdep.h" |
10 | |
11 | #include "ivshmem-client.h" |
12 | |
13 | #define IVSHMEM_CLIENT_DEFAULT_VERBOSE 0 |
14 | #define IVSHMEM_CLIENT_DEFAULT_UNIX_SOCK_PATH "/tmp/ivshmem_socket" |
15 | |
16 | typedef struct IvshmemClientArgs { |
17 | bool verbose; |
18 | const char *unix_sock_path; |
19 | } IvshmemClientArgs; |
20 | |
21 | /* show ivshmem_client_usage and exit with given error code */ |
22 | static void |
23 | ivshmem_client_usage(const char *name, int code) |
24 | { |
25 | fprintf(stderr, "%s [opts]\n" , name); |
26 | fprintf(stderr, " -h: show this help\n" ); |
27 | fprintf(stderr, " -v: verbose mode\n" ); |
28 | fprintf(stderr, " -S <unix_sock_path>: path to the unix socket\n" |
29 | " to connect to.\n" |
30 | " default=%s\n" , IVSHMEM_CLIENT_DEFAULT_UNIX_SOCK_PATH); |
31 | exit(code); |
32 | } |
33 | |
34 | /* parse the program arguments, exit on error */ |
35 | static void |
36 | ivshmem_client_parse_args(IvshmemClientArgs *args, int argc, char *argv[]) |
37 | { |
38 | int c; |
39 | |
40 | while ((c = getopt(argc, argv, |
41 | "h" /* help */ |
42 | "v" /* verbose */ |
43 | "S:" /* unix_sock_path */ |
44 | )) != -1) { |
45 | |
46 | switch (c) { |
47 | case 'h': /* help */ |
48 | ivshmem_client_usage(argv[0], 0); |
49 | break; |
50 | |
51 | case 'v': /* verbose */ |
52 | args->verbose = 1; |
53 | break; |
54 | |
55 | case 'S': /* unix_sock_path */ |
56 | args->unix_sock_path = optarg; |
57 | break; |
58 | |
59 | default: |
60 | ivshmem_client_usage(argv[0], 1); |
61 | break; |
62 | } |
63 | } |
64 | } |
65 | |
66 | /* show command line help */ |
67 | static void |
68 | ivshmem_client_cmdline_help(void) |
69 | { |
70 | printf("dump: dump peers (including us)\n" |
71 | "int <peer> <vector>: notify one vector on a peer\n" |
72 | "int <peer> all: notify all vectors of a peer\n" |
73 | "int all: notify all vectors of all peers (excepting us)\n" ); |
74 | } |
75 | |
76 | /* read stdin and handle commands */ |
77 | static int |
78 | ivshmem_client_handle_stdin_command(IvshmemClient *client) |
79 | { |
80 | IvshmemClientPeer *peer; |
81 | char buf[128]; |
82 | char *s, *token; |
83 | int ret; |
84 | int peer_id, vector; |
85 | |
86 | memset(buf, 0, sizeof(buf)); |
87 | ret = read(0, buf, sizeof(buf) - 1); |
88 | if (ret < 0) { |
89 | return -1; |
90 | } |
91 | |
92 | s = buf; |
93 | while ((token = strsep(&s, "\n\r;" )) != NULL) { |
94 | if (!strcmp(token, "" )) { |
95 | continue; |
96 | } |
97 | if (!strcmp(token, "?" )) { |
98 | ivshmem_client_cmdline_help(); |
99 | } |
100 | if (!strcmp(token, "help" )) { |
101 | ivshmem_client_cmdline_help(); |
102 | } else if (!strcmp(token, "dump" )) { |
103 | ivshmem_client_dump(client); |
104 | } else if (!strcmp(token, "int all" )) { |
105 | ivshmem_client_notify_broadcast(client); |
106 | } else if (sscanf(token, "int %d %d" , &peer_id, &vector) == 2) { |
107 | peer = ivshmem_client_search_peer(client, peer_id); |
108 | if (peer == NULL) { |
109 | printf("cannot find peer_id = %d\n" , peer_id); |
110 | continue; |
111 | } |
112 | ivshmem_client_notify(client, peer, vector); |
113 | } else if (sscanf(token, "int %d all" , &peer_id) == 1) { |
114 | peer = ivshmem_client_search_peer(client, peer_id); |
115 | if (peer == NULL) { |
116 | printf("cannot find peer_id = %d\n" , peer_id); |
117 | continue; |
118 | } |
119 | ivshmem_client_notify_all_vects(client, peer); |
120 | } else { |
121 | printf("invalid command, type help\n" ); |
122 | } |
123 | } |
124 | |
125 | printf("cmd> " ); |
126 | fflush(stdout); |
127 | return 0; |
128 | } |
129 | |
130 | /* listen on stdin (command line), on unix socket (notifications of new |
131 | * and dead peers), and on eventfd (IRQ request) */ |
132 | static int |
133 | ivshmem_client_poll_events(IvshmemClient *client) |
134 | { |
135 | fd_set fds; |
136 | int ret, maxfd; |
137 | |
138 | while (1) { |
139 | |
140 | FD_ZERO(&fds); |
141 | FD_SET(0, &fds); /* add stdin in fd_set */ |
142 | maxfd = 1; |
143 | |
144 | ivshmem_client_get_fds(client, &fds, &maxfd); |
145 | |
146 | ret = select(maxfd, &fds, NULL, NULL, NULL); |
147 | if (ret < 0) { |
148 | if (errno == EINTR) { |
149 | continue; |
150 | } |
151 | |
152 | fprintf(stderr, "select error: %s\n" , strerror(errno)); |
153 | break; |
154 | } |
155 | if (ret == 0) { |
156 | continue; |
157 | } |
158 | |
159 | if (FD_ISSET(0, &fds) && |
160 | ivshmem_client_handle_stdin_command(client) < 0 && errno != EINTR) { |
161 | fprintf(stderr, "ivshmem_client_handle_stdin_command() failed\n" ); |
162 | break; |
163 | } |
164 | |
165 | if (ivshmem_client_handle_fds(client, &fds, maxfd) < 0) { |
166 | fprintf(stderr, "ivshmem_client_handle_fds() failed\n" ); |
167 | break; |
168 | } |
169 | } |
170 | |
171 | return ret; |
172 | } |
173 | |
174 | /* callback when we receive a notification (just display it) */ |
175 | static void |
176 | ivshmem_client_notification_cb(const IvshmemClient *client, |
177 | const IvshmemClientPeer *peer, |
178 | unsigned vect, void *arg) |
179 | { |
180 | (void)client; |
181 | (void)arg; |
182 | printf("receive notification from peer_id=%" PRId64 " vector=%u\n" , |
183 | peer->id, vect); |
184 | } |
185 | |
186 | int |
187 | main(int argc, char *argv[]) |
188 | { |
189 | struct sigaction sa; |
190 | IvshmemClient client; |
191 | IvshmemClientArgs args = { |
192 | .verbose = IVSHMEM_CLIENT_DEFAULT_VERBOSE, |
193 | .unix_sock_path = IVSHMEM_CLIENT_DEFAULT_UNIX_SOCK_PATH, |
194 | }; |
195 | |
196 | /* parse arguments, will exit on error */ |
197 | ivshmem_client_parse_args(&args, argc, argv); |
198 | |
199 | /* Ignore SIGPIPE, see this link for more info: |
200 | * http://www.mail-archive.com/libevent-users@monkey.org/msg01606.html */ |
201 | sa.sa_handler = SIG_IGN; |
202 | sa.sa_flags = 0; |
203 | if (sigemptyset(&sa.sa_mask) == -1 || |
204 | sigaction(SIGPIPE, &sa, 0) == -1) { |
205 | perror("failed to ignore SIGPIPE; sigaction" ); |
206 | return 1; |
207 | } |
208 | |
209 | ivshmem_client_cmdline_help(); |
210 | printf("cmd> " ); |
211 | fflush(stdout); |
212 | |
213 | if (ivshmem_client_init(&client, args.unix_sock_path, |
214 | ivshmem_client_notification_cb, NULL, |
215 | args.verbose) < 0) { |
216 | fprintf(stderr, "cannot init client\n" ); |
217 | return 1; |
218 | } |
219 | |
220 | while (1) { |
221 | if (ivshmem_client_connect(&client) < 0) { |
222 | fprintf(stderr, "cannot connect to server, retry in 1 second\n" ); |
223 | sleep(1); |
224 | continue; |
225 | } |
226 | |
227 | fprintf(stdout, "listen on server socket %d\n" , client.sock_fd); |
228 | |
229 | if (ivshmem_client_poll_events(&client) == 0) { |
230 | continue; |
231 | } |
232 | |
233 | /* disconnected from server, reset all peers */ |
234 | fprintf(stdout, "disconnected from server\n" ); |
235 | |
236 | ivshmem_client_close(&client); |
237 | } |
238 | |
239 | return 0; |
240 | } |
241 | |