David Marchand | a75eb03 | 2014-09-08 11:17:48 +0200 | [diff] [blame] | 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 | |
Peter Maydell | ccd241b | 2016-01-29 17:50:04 +0000 | [diff] [blame] | 9 | #include "qemu/osdep.h" |
David Marchand | a75eb03 | 2014-09-08 11:17:48 +0200 | [diff] [blame] | 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 */ |
Marc-André Lureau | 45b00c4 | 2015-06-24 13:33:32 +0200 | [diff] [blame] | 56 | args->unix_sock_path = optarg; |
David Marchand | a75eb03 | 2014-09-08 11:17:48 +0200 | [diff] [blame] | 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; |
Marc-André Lureau | f7a199b | 2015-09-24 12:55:01 +0200 | [diff] [blame] | 182 | printf("receive notification from peer_id=%" PRId64 " vector=%u\n", |
| 183 | peer->id, vect); |
David Marchand | a75eb03 | 2014-09-08 11:17:48 +0200 | [diff] [blame] | 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 | } |