i386: Fix pkg_id offset for EPYC cpu models
[qemu.git] / contrib / ivshmem-client / main.c
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 }