1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16#include "qemu-common.h"
17#include "qemu_socket.h"
18#include "migration.h"
19#include "qemu-char.h"
20#include "buffered_file.h"
21#include "block.h"
22
23
24
25#ifdef DEBUG_MIGRATION_UNIX
26#define DPRINTF(fmt, ...) \
27 do { printf("migration-unix: " fmt, ## __VA_ARGS__); } while (0)
28#else
29#define DPRINTF(fmt, ...) \
30 do { } while (0)
31#endif
32
33static int unix_errno(MigrationState *s)
34{
35 return errno;
36}
37
38static int unix_write(MigrationState *s, const void * buf, size_t size)
39{
40 return write(s->fd, buf, size);
41}
42
43static int unix_close(MigrationState *s)
44{
45 int r = 0;
46 DPRINTF("unix_close\n");
47 if (s->fd != -1) {
48 if (close(s->fd) < 0) {
49 r = -errno;
50 }
51 s->fd = -1;
52 }
53 return r;
54}
55
56static void unix_wait_for_connect(void *opaque)
57{
58 MigrationState *s = opaque;
59 int val, ret;
60 socklen_t valsize = sizeof(val);
61
62 DPRINTF("connect completed\n");
63 do {
64 ret = getsockopt(s->fd, SOL_SOCKET, SO_ERROR, (void *) &val, &valsize);
65 } while (ret == -1 && errno == EINTR);
66
67 if (ret < 0) {
68 migrate_fd_error(s);
69 return;
70 }
71
72 qemu_set_fd_handler2(s->fd, NULL, NULL, NULL, NULL);
73
74 if (val == 0)
75 migrate_fd_connect(s);
76 else {
77 DPRINTF("error connecting %d\n", val);
78 migrate_fd_error(s);
79 }
80}
81
82int unix_start_outgoing_migration(MigrationState *s, const char *path)
83{
84 struct sockaddr_un addr;
85 int ret;
86
87 addr.sun_family = AF_UNIX;
88 snprintf(addr.sun_path, sizeof(addr.sun_path), "%s", path);
89 s->get_error = unix_errno;
90 s->write = unix_write;
91 s->close = unix_close;
92
93 s->fd = qemu_socket(PF_UNIX, SOCK_STREAM, 0);
94 if (s->fd == -1) {
95 DPRINTF("Unable to open socket");
96 return -errno;
97 }
98
99 socket_set_nonblock(s->fd);
100
101 do {
102 ret = connect(s->fd, (struct sockaddr *)&addr, sizeof(addr));
103 if (ret == -1) {
104 ret = -errno;
105 }
106 if (ret == -EINPROGRESS || ret == -EWOULDBLOCK) {
107 qemu_set_fd_handler2(s->fd, NULL, NULL, unix_wait_for_connect, s);
108 return 0;
109 }
110 } while (ret == -EINTR);
111
112 if (ret < 0) {
113 DPRINTF("connect failed\n");
114 migrate_fd_error(s);
115 return ret;
116 }
117 migrate_fd_connect(s);
118 return 0;
119}
120
121static void unix_accept_incoming_migration(void *opaque)
122{
123 struct sockaddr_un addr;
124 socklen_t addrlen = sizeof(addr);
125 int s = (intptr_t)opaque;
126 QEMUFile *f;
127 int c;
128
129 do {
130 c = qemu_accept(s, (struct sockaddr *)&addr, &addrlen);
131 } while (c == -1 && errno == EINTR);
132
133 DPRINTF("accepted migration\n");
134
135 if (c == -1) {
136 fprintf(stderr, "could not accept migration connection\n");
137 goto out2;
138 }
139
140 f = qemu_fopen_socket(c);
141 if (f == NULL) {
142 fprintf(stderr, "could not qemu_fopen socket\n");
143 goto out;
144 }
145
146 process_incoming_migration(f);
147 qemu_fclose(f);
148out:
149 close(c);
150out2:
151 qemu_set_fd_handler2(s, NULL, NULL, NULL, NULL);
152 close(s);
153}
154
155int unix_start_incoming_migration(const char *path)
156{
157 struct sockaddr_un addr;
158 int s;
159 int ret;
160
161 DPRINTF("Attempting to start an incoming migration\n");
162
163 s = qemu_socket(PF_UNIX, SOCK_STREAM, 0);
164 if (s == -1) {
165 fprintf(stderr, "Could not open unix socket: %s\n", strerror(errno));
166 return -errno;
167 }
168
169 memset(&addr, 0, sizeof(addr));
170 addr.sun_family = AF_UNIX;
171 snprintf(addr.sun_path, sizeof(addr.sun_path), "%s", path);
172
173 unlink(addr.sun_path);
174 if (bind(s, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
175 ret = -errno;
176 fprintf(stderr, "bind(unix:%s): %s\n", addr.sun_path, strerror(errno));
177 goto err;
178 }
179 if (listen(s, 1) == -1) {
180 fprintf(stderr, "listen(unix:%s): %s\n", addr.sun_path,
181 strerror(errno));
182 ret = -errno;
183 goto err;
184 }
185
186 qemu_set_fd_handler2(s, NULL, unix_accept_incoming_migration, NULL,
187 (void *)(intptr_t)s);
188
189 return 0;
190
191err:
192 close(s);
193 return ret;
194}
195
