]>
Commit | Line | Data |
---|---|---|
34c9dd8e AL |
1 | /* |
2 | * QEMU live migration | |
3 | * | |
4 | * Copyright IBM, Corp. 2008 | |
5 | * | |
6 | * Authors: | |
7 | * Anthony Liguori <aliguori@us.ibm.com> | |
8 | * | |
9 | * This work is licensed under the terms of the GNU GPL, version 2. See | |
10 | * the COPYING file in the top-level directory. | |
11 | * | |
6b620ca3 PB |
12 | * Contributions after 2012-01-13 are licensed under the terms of the |
13 | * GNU GPL, version 2 or (at your option) any later version. | |
34c9dd8e AL |
14 | */ |
15 | ||
16 | #include "qemu-common.h" | |
1de7afc9 | 17 | #include "qemu/sockets.h" |
caf71f86 | 18 | #include "migration/migration.h" |
557ec5a0 | 19 | #include "migration/qemu-file.h" |
737e150e | 20 | #include "block/block.h" |
34c9dd8e AL |
21 | |
22 | //#define DEBUG_MIGRATION_TCP | |
23 | ||
34c9dd8e | 24 | #ifdef DEBUG_MIGRATION_TCP |
d0f2c4c6 | 25 | #define DPRINTF(fmt, ...) \ |
34c9dd8e AL |
26 | do { printf("migration-tcp: " fmt, ## __VA_ARGS__); } while (0) |
27 | #else | |
d0f2c4c6 | 28 | #define DPRINTF(fmt, ...) \ |
34c9dd8e AL |
29 | do { } while (0) |
30 | #endif | |
31 | ||
22f00a44 | 32 | static int socket_errno(MigrationState *s) |
34c9dd8e | 33 | { |
8ad9fa5d | 34 | return socket_error(); |
34c9dd8e AL |
35 | } |
36 | ||
22f00a44 | 37 | static int socket_write(MigrationState *s, const void * buf, size_t size) |
34c9dd8e | 38 | { |
065e2813 | 39 | return send(s->fd, buf, size, 0); |
34c9dd8e AL |
40 | } |
41 | ||
22f00a44 | 42 | static int tcp_close(MigrationState *s) |
34c9dd8e | 43 | { |
61a5872f | 44 | int r = 0; |
d0f2c4c6 | 45 | DPRINTF("tcp_close\n"); |
6c360136 PB |
46 | if (closesocket(s->fd) < 0) { |
47 | r = -socket_error(); | |
34c9dd8e | 48 | } |
61a5872f | 49 | return r; |
34c9dd8e AL |
50 | } |
51 | ||
233aa5c2 | 52 | static void tcp_wait_for_connect(int fd, void *opaque) |
34c9dd8e | 53 | { |
22f00a44 | 54 | MigrationState *s = opaque; |
34c9dd8e | 55 | |
233aa5c2 OW |
56 | if (fd < 0) { |
57 | DPRINTF("migrate connect error\n"); | |
58 | s->fd = -1; | |
065e2813 | 59 | migrate_fd_error(s); |
233aa5c2 OW |
60 | } else { |
61 | DPRINTF("migrate connect success\n"); | |
62 | s->fd = fd; | |
dd217b87 | 63 | socket_set_block(s->fd); |
065e2813 | 64 | migrate_fd_connect(s); |
34c9dd8e | 65 | } |
34c9dd8e AL |
66 | } |
67 | ||
f37afb5a | 68 | void tcp_start_outgoing_migration(MigrationState *s, const char *host_port, Error **errp) |
34c9dd8e | 69 | { |
065e2813 AL |
70 | s->get_error = socket_errno; |
71 | s->write = socket_write; | |
72 | s->close = tcp_close; | |
34c9dd8e | 73 | |
f37afb5a | 74 | s->fd = inet_nonblocking_connect(host_port, tcp_wait_for_connect, s, errp); |
34c9dd8e AL |
75 | } |
76 | ||
77 | static void tcp_accept_incoming_migration(void *opaque) | |
78 | { | |
79 | struct sockaddr_in addr; | |
80 | socklen_t addrlen = sizeof(addr); | |
e0efb993 | 81 | int s = (intptr_t)opaque; |
34c9dd8e | 82 | QEMUFile *f; |
511c0231 | 83 | int c; |
34c9dd8e AL |
84 | |
85 | do { | |
40ff6d7e | 86 | c = qemu_accept(s, (struct sockaddr *)&addr, &addrlen); |
c1d36665 | 87 | } while (c == -1 && socket_error() == EINTR); |
a6ef2909 | 88 | qemu_set_fd_handler2(s, NULL, NULL, NULL, NULL); |
09bac73c | 89 | closesocket(s); |
34c9dd8e | 90 | |
d0f2c4c6 | 91 | DPRINTF("accepted migration\n"); |
34c9dd8e AL |
92 | |
93 | if (c == -1) { | |
94 | fprintf(stderr, "could not accept migration connection\n"); | |
a6ef2909 | 95 | goto out; |
34c9dd8e AL |
96 | } |
97 | ||
c1d36665 | 98 | f = qemu_fopen_socket(c); |
34c9dd8e AL |
99 | if (f == NULL) { |
100 | fprintf(stderr, "could not qemu_fopen socket\n"); | |
101 | goto out; | |
102 | } | |
103 | ||
511c0231 | 104 | process_incoming_migration(f); |
ab52a824 PB |
105 | return; |
106 | ||
34c9dd8e | 107 | out: |
09bac73c | 108 | closesocket(c); |
34c9dd8e AL |
109 | } |
110 | ||
43eaae28 | 111 | void tcp_start_incoming_migration(const char *host_port, Error **errp) |
34c9dd8e | 112 | { |
34c9dd8e AL |
113 | int s; |
114 | ||
d5c5dacc | 115 | s = inet_listen(host_port, NULL, 256, SOCK_STREAM, 0, errp); |
d5c5dacc | 116 | if (s < 0) { |
43eaae28 | 117 | return; |
ee86c61f | 118 | } |
34c9dd8e AL |
119 | |
120 | qemu_set_fd_handler2(s, NULL, tcp_accept_incoming_migration, NULL, | |
e0efb993 | 121 | (void *)(intptr_t)s); |
34c9dd8e | 122 | } |