root / migration-unix.c @ 301901b5
History | View | Annotate | Download (4.9 kB)
1 | 4951f65b | Chris Lalancette | /*
|
---|---|---|---|
2 | 4951f65b | Chris Lalancette | * QEMU live migration via Unix Domain Sockets
|
3 | 4951f65b | Chris Lalancette | *
|
4 | 4951f65b | Chris Lalancette | * Copyright Red Hat, Inc. 2009
|
5 | 4951f65b | Chris Lalancette | *
|
6 | 4951f65b | Chris Lalancette | * Authors:
|
7 | 4951f65b | Chris Lalancette | * Chris Lalancette <clalance@redhat.com>
|
8 | 4951f65b | Chris Lalancette | *
|
9 | 4951f65b | Chris Lalancette | * This work is licensed under the terms of the GNU GPL, version 2. See
|
10 | 4951f65b | Chris Lalancette | * the COPYING file in the top-level directory.
|
11 | 4951f65b | Chris Lalancette | *
|
12 | 4951f65b | Chris Lalancette | */
|
13 | 4951f65b | Chris Lalancette | |
14 | 4951f65b | Chris Lalancette | #include "qemu-common.h" |
15 | 4951f65b | Chris Lalancette | #include "qemu_socket.h" |
16 | 4951f65b | Chris Lalancette | #include "migration.h" |
17 | 4951f65b | Chris Lalancette | #include "qemu-char.h" |
18 | 4951f65b | Chris Lalancette | #include "sysemu.h" |
19 | 4951f65b | Chris Lalancette | #include "buffered_file.h" |
20 | 4951f65b | Chris Lalancette | #include "block.h" |
21 | 4951f65b | Chris Lalancette | |
22 | 4951f65b | Chris Lalancette | //#define DEBUG_MIGRATION_UNIX
|
23 | 4951f65b | Chris Lalancette | |
24 | 4951f65b | Chris Lalancette | #ifdef DEBUG_MIGRATION_UNIX
|
25 | 4951f65b | Chris Lalancette | #define dprintf(fmt, ...) \
|
26 | 4951f65b | Chris Lalancette | do { printf("migration-unix: " fmt, ## __VA_ARGS__); } while (0) |
27 | 4951f65b | Chris Lalancette | #else
|
28 | 4951f65b | Chris Lalancette | #define dprintf(fmt, ...) \
|
29 | 4951f65b | Chris Lalancette | do { } while (0) |
30 | 4951f65b | Chris Lalancette | #endif
|
31 | 4951f65b | Chris Lalancette | |
32 | 4951f65b | Chris Lalancette | static int unix_errno(FdMigrationState *s) |
33 | 4951f65b | Chris Lalancette | { |
34 | 4951f65b | Chris Lalancette | return errno;
|
35 | 4951f65b | Chris Lalancette | } |
36 | 4951f65b | Chris Lalancette | |
37 | 4951f65b | Chris Lalancette | static int unix_write(FdMigrationState *s, const void * buf, size_t size) |
38 | 4951f65b | Chris Lalancette | { |
39 | 4951f65b | Chris Lalancette | return write(s->fd, buf, size);
|
40 | 4951f65b | Chris Lalancette | } |
41 | 4951f65b | Chris Lalancette | |
42 | 4951f65b | Chris Lalancette | static int unix_close(FdMigrationState *s) |
43 | 4951f65b | Chris Lalancette | { |
44 | 4951f65b | Chris Lalancette | dprintf("unix_close\n");
|
45 | 4951f65b | Chris Lalancette | if (s->fd != -1) { |
46 | 4951f65b | Chris Lalancette | close(s->fd); |
47 | 4951f65b | Chris Lalancette | s->fd = -1;
|
48 | 4951f65b | Chris Lalancette | } |
49 | 4951f65b | Chris Lalancette | return 0; |
50 | 4951f65b | Chris Lalancette | } |
51 | 4951f65b | Chris Lalancette | |
52 | 4951f65b | Chris Lalancette | static void unix_wait_for_connect(void *opaque) |
53 | 4951f65b | Chris Lalancette | { |
54 | 4951f65b | Chris Lalancette | FdMigrationState *s = opaque; |
55 | 4951f65b | Chris Lalancette | int val, ret;
|
56 | 4951f65b | Chris Lalancette | socklen_t valsize = sizeof(val);
|
57 | 4951f65b | Chris Lalancette | |
58 | 4951f65b | Chris Lalancette | dprintf("connect completed\n");
|
59 | 4951f65b | Chris Lalancette | do {
|
60 | 4951f65b | Chris Lalancette | ret = getsockopt(s->fd, SOL_SOCKET, SO_ERROR, (void *) &val, &valsize);
|
61 | 4951f65b | Chris Lalancette | } while (ret == -1 && (s->get_error(s)) == EINTR); |
62 | 4951f65b | Chris Lalancette | |
63 | 4951f65b | Chris Lalancette | if (ret < 0) { |
64 | 4951f65b | Chris Lalancette | migrate_fd_error(s); |
65 | 4951f65b | Chris Lalancette | return;
|
66 | 4951f65b | Chris Lalancette | } |
67 | 4951f65b | Chris Lalancette | |
68 | 4951f65b | Chris Lalancette | qemu_set_fd_handler2(s->fd, NULL, NULL, NULL, NULL); |
69 | 4951f65b | Chris Lalancette | |
70 | 4951f65b | Chris Lalancette | if (val == 0) |
71 | 4951f65b | Chris Lalancette | migrate_fd_connect(s); |
72 | 4951f65b | Chris Lalancette | else {
|
73 | 4951f65b | Chris Lalancette | dprintf("error connecting %d\n", val);
|
74 | 4951f65b | Chris Lalancette | migrate_fd_error(s); |
75 | 4951f65b | Chris Lalancette | } |
76 | 4951f65b | Chris Lalancette | } |
77 | 4951f65b | Chris Lalancette | |
78 | 4951f65b | Chris Lalancette | MigrationState *unix_start_outgoing_migration(const char *path, |
79 | 4951f65b | Chris Lalancette | int64_t bandwidth_limit, |
80 | 4951f65b | Chris Lalancette | int detach)
|
81 | 4951f65b | Chris Lalancette | { |
82 | 4951f65b | Chris Lalancette | FdMigrationState *s; |
83 | 4951f65b | Chris Lalancette | struct sockaddr_un addr;
|
84 | 4951f65b | Chris Lalancette | int ret;
|
85 | 4951f65b | Chris Lalancette | |
86 | 4951f65b | Chris Lalancette | addr.sun_family = AF_UNIX; |
87 | 4951f65b | Chris Lalancette | snprintf(addr.sun_path, sizeof(addr.sun_path), "%s", path); |
88 | 4951f65b | Chris Lalancette | |
89 | 4951f65b | Chris Lalancette | s = qemu_mallocz(sizeof(*s));
|
90 | 4951f65b | Chris Lalancette | |
91 | 4951f65b | Chris Lalancette | s->get_error = unix_errno; |
92 | 4951f65b | Chris Lalancette | s->write = unix_write; |
93 | 4951f65b | Chris Lalancette | s->close = unix_close; |
94 | 4951f65b | Chris Lalancette | s->mig_state.cancel = migrate_fd_cancel; |
95 | 4951f65b | Chris Lalancette | s->mig_state.get_status = migrate_fd_get_status; |
96 | 4951f65b | Chris Lalancette | s->mig_state.release = migrate_fd_release; |
97 | 4951f65b | Chris Lalancette | |
98 | 4951f65b | Chris Lalancette | s->state = MIG_STATE_ACTIVE; |
99 | 4951f65b | Chris Lalancette | s->mon_resume = NULL;
|
100 | 4951f65b | Chris Lalancette | s->bandwidth_limit = bandwidth_limit; |
101 | 4951f65b | Chris Lalancette | s->fd = socket(PF_UNIX, SOCK_STREAM, 0);
|
102 | 4951f65b | Chris Lalancette | if (s->fd < 0) { |
103 | 4951f65b | Chris Lalancette | dprintf("Unable to open socket");
|
104 | 4951f65b | Chris Lalancette | goto err_after_alloc;
|
105 | 4951f65b | Chris Lalancette | } |
106 | 4951f65b | Chris Lalancette | |
107 | 4951f65b | Chris Lalancette | socket_set_nonblock(s->fd); |
108 | 4951f65b | Chris Lalancette | |
109 | 4951f65b | Chris Lalancette | if (!detach)
|
110 | 4951f65b | Chris Lalancette | migrate_fd_monitor_suspend(s); |
111 | 4951f65b | Chris Lalancette | |
112 | 4951f65b | Chris Lalancette | do {
|
113 | 4951f65b | Chris Lalancette | ret = connect(s->fd, (struct sockaddr *)&addr, sizeof(addr)); |
114 | 4951f65b | Chris Lalancette | if (ret == -1) |
115 | 4951f65b | Chris Lalancette | ret = -(s->get_error(s)); |
116 | 4951f65b | Chris Lalancette | |
117 | 4951f65b | Chris Lalancette | if (ret == -EINPROGRESS || ret == -EWOULDBLOCK)
|
118 | 4951f65b | Chris Lalancette | qemu_set_fd_handler2(s->fd, NULL, NULL, unix_wait_for_connect, s); |
119 | 4951f65b | Chris Lalancette | } while (ret == -EINTR);
|
120 | 4951f65b | Chris Lalancette | |
121 | 4951f65b | Chris Lalancette | if (ret < 0 && ret != -EINPROGRESS && ret != -EWOULDBLOCK) { |
122 | 4951f65b | Chris Lalancette | dprintf("connect failed\n");
|
123 | 4951f65b | Chris Lalancette | goto err_after_open;
|
124 | 4951f65b | Chris Lalancette | } else if (ret >= 0) |
125 | 4951f65b | Chris Lalancette | migrate_fd_connect(s); |
126 | 4951f65b | Chris Lalancette | |
127 | 4951f65b | Chris Lalancette | return &s->mig_state;
|
128 | 4951f65b | Chris Lalancette | |
129 | 4951f65b | Chris Lalancette | err_after_open:
|
130 | 4951f65b | Chris Lalancette | close(s->fd); |
131 | 4951f65b | Chris Lalancette | |
132 | 4951f65b | Chris Lalancette | err_after_alloc:
|
133 | 4951f65b | Chris Lalancette | qemu_free(s); |
134 | 4951f65b | Chris Lalancette | return NULL; |
135 | 4951f65b | Chris Lalancette | } |
136 | 4951f65b | Chris Lalancette | |
137 | 4951f65b | Chris Lalancette | static void unix_accept_incoming_migration(void *opaque) |
138 | 4951f65b | Chris Lalancette | { |
139 | 4951f65b | Chris Lalancette | struct sockaddr_un addr;
|
140 | 4951f65b | Chris Lalancette | socklen_t addrlen = sizeof(addr);
|
141 | 4951f65b | Chris Lalancette | int s = (unsigned long)opaque; |
142 | 4951f65b | Chris Lalancette | QEMUFile *f; |
143 | 4951f65b | Chris Lalancette | int c, ret;
|
144 | 4951f65b | Chris Lalancette | |
145 | 4951f65b | Chris Lalancette | do {
|
146 | 4951f65b | Chris Lalancette | c = accept(s, (struct sockaddr *)&addr, &addrlen);
|
147 | 4951f65b | Chris Lalancette | } while (c == -1 && socket_error() == EINTR); |
148 | 4951f65b | Chris Lalancette | |
149 | 4951f65b | Chris Lalancette | dprintf("accepted migration\n");
|
150 | 4951f65b | Chris Lalancette | |
151 | 4951f65b | Chris Lalancette | if (c == -1) { |
152 | 4951f65b | Chris Lalancette | fprintf(stderr, "could not accept migration connection\n");
|
153 | 4951f65b | Chris Lalancette | return;
|
154 | 4951f65b | Chris Lalancette | } |
155 | 4951f65b | Chris Lalancette | |
156 | 4951f65b | Chris Lalancette | f = qemu_fopen_socket(c); |
157 | 4951f65b | Chris Lalancette | if (f == NULL) { |
158 | 4951f65b | Chris Lalancette | fprintf(stderr, "could not qemu_fopen socket\n");
|
159 | 4951f65b | Chris Lalancette | goto out;
|
160 | 4951f65b | Chris Lalancette | } |
161 | 4951f65b | Chris Lalancette | |
162 | 4951f65b | Chris Lalancette | ret = qemu_loadvm_state(f); |
163 | 4951f65b | Chris Lalancette | if (ret < 0) { |
164 | 4951f65b | Chris Lalancette | fprintf(stderr, "load of migration failed\n");
|
165 | 4951f65b | Chris Lalancette | goto out_fopen;
|
166 | 4951f65b | Chris Lalancette | } |
167 | 4951f65b | Chris Lalancette | qemu_announce_self(); |
168 | 4951f65b | Chris Lalancette | dprintf("successfully loaded vm state\n");
|
169 | 4951f65b | Chris Lalancette | |
170 | 4951f65b | Chris Lalancette | /* we've successfully migrated, close the server socket */
|
171 | 4951f65b | Chris Lalancette | qemu_set_fd_handler2(s, NULL, NULL, NULL, NULL); |
172 | 4951f65b | Chris Lalancette | close(s); |
173 | 4951f65b | Chris Lalancette | |
174 | 4951f65b | Chris Lalancette | out_fopen:
|
175 | 4951f65b | Chris Lalancette | qemu_fclose(f); |
176 | 4951f65b | Chris Lalancette | out:
|
177 | 4951f65b | Chris Lalancette | close(c); |
178 | 4951f65b | Chris Lalancette | } |
179 | 4951f65b | Chris Lalancette | |
180 | 4951f65b | Chris Lalancette | int unix_start_incoming_migration(const char *path) |
181 | 4951f65b | Chris Lalancette | { |
182 | 4951f65b | Chris Lalancette | struct sockaddr_un un;
|
183 | 4951f65b | Chris Lalancette | int sock;
|
184 | 4951f65b | Chris Lalancette | |
185 | 4951f65b | Chris Lalancette | dprintf("Attempting to start an incoming migration\n");
|
186 | 4951f65b | Chris Lalancette | |
187 | 4951f65b | Chris Lalancette | sock = socket(PF_UNIX, SOCK_STREAM, 0);
|
188 | 4951f65b | Chris Lalancette | if (sock < 0) { |
189 | 4951f65b | Chris Lalancette | fprintf(stderr, "Could not open unix socket: %s\n", strerror(errno));
|
190 | 4951f65b | Chris Lalancette | return -EINVAL;
|
191 | 4951f65b | Chris Lalancette | } |
192 | 4951f65b | Chris Lalancette | |
193 | 4951f65b | Chris Lalancette | memset(&un, 0, sizeof(un)); |
194 | 4951f65b | Chris Lalancette | un.sun_family = AF_UNIX; |
195 | 4951f65b | Chris Lalancette | snprintf(un.sun_path, sizeof(un.sun_path), "%s", path); |
196 | 4951f65b | Chris Lalancette | |
197 | 4951f65b | Chris Lalancette | unlink(un.sun_path); |
198 | 4951f65b | Chris Lalancette | if (bind(sock, (struct sockaddr*) &un, sizeof(un)) < 0) { |
199 | 4951f65b | Chris Lalancette | fprintf(stderr, "bind(unix:%s): %s\n", un.sun_path, strerror(errno));
|
200 | 4951f65b | Chris Lalancette | goto err;
|
201 | 4951f65b | Chris Lalancette | } |
202 | 4951f65b | Chris Lalancette | if (listen(sock, 1) < 0) { |
203 | 4951f65b | Chris Lalancette | fprintf(stderr, "listen(unix:%s): %s\n", un.sun_path, strerror(errno));
|
204 | 4951f65b | Chris Lalancette | goto err;
|
205 | 4951f65b | Chris Lalancette | } |
206 | 4951f65b | Chris Lalancette | |
207 | 4951f65b | Chris Lalancette | qemu_set_fd_handler2(sock, NULL, unix_accept_incoming_migration, NULL, |
208 | 4951f65b | Chris Lalancette | (void *)(unsigned long)sock); |
209 | 4951f65b | Chris Lalancette | |
210 | 4951f65b | Chris Lalancette | return 0; |
211 | 4951f65b | Chris Lalancette | |
212 | 4951f65b | Chris Lalancette | err:
|
213 | 4951f65b | Chris Lalancette | close(sock); |
214 | 4951f65b | Chris Lalancette | |
215 | 4951f65b | Chris Lalancette | return -EINVAL;
|
216 | 4951f65b | Chris Lalancette | } |