2 * Copyright 6WIND S.A., 2014
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
9 #include "qemu/osdep.h"
10 #include "qemu-common.h"
12 #include "ivshmem-server.h"
14 #define IVSHMEM_SERVER_DEFAULT_VERBOSE 0
15 #define IVSHMEM_SERVER_DEFAULT_FOREGROUND 0
16 #define IVSHMEM_SERVER_DEFAULT_PID_FILE "/var/run/ivshmem-server.pid"
17 #define IVSHMEM_SERVER_DEFAULT_UNIX_SOCK_PATH "/tmp/ivshmem_socket"
18 #define IVSHMEM_SERVER_DEFAULT_SHM_PATH "ivshmem"
19 #define IVSHMEM_SERVER_DEFAULT_SHM_SIZE (4*1024*1024)
20 #define IVSHMEM_SERVER_DEFAULT_N_VECTORS 1
22 /* used to quit on signal SIGTERM */
23 static int ivshmem_server_quit
;
25 /* arguments given by the user */
26 typedef struct IvshmemServerArgs
{
30 const char *unix_socket_path
;
38 ivshmem_server_usage(const char *progname
)
40 printf("Usage: %s [OPTION]...\n"
41 " -h: show this help\n"
43 " -F: foreground mode (default is to daemonize)\n"
44 " -p <pid-file>: path to the PID file (used in daemon mode only)\n"
45 " default " IVSHMEM_SERVER_DEFAULT_PID_FILE
"\n"
46 " -S <unix-socket-path>: path to the unix socket to listen to\n"
47 " default " IVSHMEM_SERVER_DEFAULT_UNIX_SOCK_PATH
"\n"
48 " -M <shm-name>: POSIX shared memory object to use\n"
49 " default " IVSHMEM_SERVER_DEFAULT_SHM_PATH
"\n"
50 " -m <dir-name>: where to create shared memory\n"
51 " -l <size>: size of shared memory in bytes\n"
52 " suffixes K, M and G can be used, e.g. 1K means 1024\n"
54 " -n <nvectors>: number of vectors\n"
56 progname
, IVSHMEM_SERVER_DEFAULT_SHM_SIZE
,
57 IVSHMEM_SERVER_DEFAULT_N_VECTORS
);
61 ivshmem_server_help(const char *progname
)
63 fprintf(stderr
, "Try '%s -h' for more information.\n", progname
);
66 /* parse the program arguments, exit on error */
68 ivshmem_server_parse_args(IvshmemServerArgs
*args
, int argc
, char *argv
[])
74 while ((c
= getopt(argc
, argv
, "hvFp:S:m:M:l:n:")) != -1) {
78 ivshmem_server_usage(argv
[0]);
82 case 'v': /* verbose */
86 case 'F': /* foreground */
90 case 'p': /* pid file */
91 args
->pid_file
= optarg
;
94 case 'S': /* unix socket path */
95 args
->unix_socket_path
= optarg
;
98 case 'M': /* shm name */
99 case 'm': /* dir name */
100 args
->shm_path
= optarg
;
101 args
->use_shm_open
= c
== 'M';
104 case 'l': /* shm size */
105 parse_option_size("shm_size", optarg
, &args
->shm_size
, &err
);
107 error_report_err(err
);
108 ivshmem_server_help(argv
[0]);
113 case 'n': /* number of vectors */
114 if (parse_uint_full(optarg
, &v
, 0) < 0) {
115 fprintf(stderr
, "cannot parse n_vectors\n");
116 ivshmem_server_help(argv
[0]);
123 ivshmem_server_usage(argv
[0]);
129 if (args
->n_vectors
> IVSHMEM_SERVER_MAX_VECTORS
) {
130 fprintf(stderr
, "too many requested vectors (max is %d)\n",
131 IVSHMEM_SERVER_MAX_VECTORS
);
132 ivshmem_server_help(argv
[0]);
136 if (args
->verbose
== 1 && args
->foreground
== 0) {
137 fprintf(stderr
, "cannot use verbose in daemon mode\n");
138 ivshmem_server_help(argv
[0]);
143 /* wait for events on listening server unix socket and connected client
146 ivshmem_server_poll_events(IvshmemServer
*server
)
151 while (!ivshmem_server_quit
) {
155 ivshmem_server_get_fds(server
, &fds
, &maxfd
);
157 ret
= select(maxfd
, &fds
, NULL
, NULL
, NULL
);
160 if (errno
== EINTR
) {
164 fprintf(stderr
, "select error: %s\n", strerror(errno
));
171 if (ivshmem_server_handle_fds(server
, &fds
, maxfd
) < 0) {
172 fprintf(stderr
, "ivshmem_server_handle_fds() failed\n");
181 ivshmem_server_quit_cb(int signum
)
183 ivshmem_server_quit
= 1;
187 main(int argc
, char *argv
[])
189 IvshmemServer server
;
190 struct sigaction sa
, sa_quit
;
191 IvshmemServerArgs args
= {
192 .verbose
= IVSHMEM_SERVER_DEFAULT_VERBOSE
,
193 .foreground
= IVSHMEM_SERVER_DEFAULT_FOREGROUND
,
194 .pid_file
= IVSHMEM_SERVER_DEFAULT_PID_FILE
,
195 .unix_socket_path
= IVSHMEM_SERVER_DEFAULT_UNIX_SOCK_PATH
,
196 .shm_path
= IVSHMEM_SERVER_DEFAULT_SHM_PATH
,
197 .use_shm_open
= true,
198 .shm_size
= IVSHMEM_SERVER_DEFAULT_SHM_SIZE
,
199 .n_vectors
= IVSHMEM_SERVER_DEFAULT_N_VECTORS
,
203 /* parse arguments, will exit on error */
204 ivshmem_server_parse_args(&args
, argc
, argv
);
206 /* Ignore SIGPIPE, see this link for more info:
207 * http://www.mail-archive.com/libevent-users@monkey.org/msg01606.html */
208 sa
.sa_handler
= SIG_IGN
;
210 if (sigemptyset(&sa
.sa_mask
) == -1 ||
211 sigaction(SIGPIPE
, &sa
, 0) == -1) {
212 perror("failed to ignore SIGPIPE; sigaction");
216 sa_quit
.sa_handler
= ivshmem_server_quit_cb
;
217 sa_quit
.sa_flags
= 0;
218 if (sigemptyset(&sa_quit
.sa_mask
) == -1 ||
219 sigaction(SIGTERM
, &sa_quit
, 0) == -1) {
220 perror("failed to add SIGTERM handler; sigaction");
224 /* init the ivshms structure */
225 if (ivshmem_server_init(&server
, args
.unix_socket_path
,
226 args
.shm_path
, args
.use_shm_open
,
227 args
.shm_size
, args
.n_vectors
, args
.verbose
) < 0) {
228 fprintf(stderr
, "cannot init server\n");
232 /* start the ivshmem server (open shm & unix socket) */
233 if (ivshmem_server_start(&server
) < 0) {
234 fprintf(stderr
, "cannot bind\n");
238 /* daemonize if asked to */
239 if (!args
.foreground
) {
242 if (qemu_daemon(1, 1) < 0) {
243 fprintf(stderr
, "cannot daemonize: %s\n", strerror(errno
));
248 fp
= fopen(args
.pid_file
, "w");
250 fprintf(stderr
, "cannot write pid file: %s\n", strerror(errno
));
254 fprintf(fp
, "%d\n", (int) getpid());
258 ivshmem_server_poll_events(&server
);
259 fprintf(stdout
, "server disconnected\n");
263 ivshmem_server_close(&server
);