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