1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17#include "qemu/osdep.h"
18
19#include "qemu-common.h"
20#include "qemu/error-report.h"
21#include "qapi/error.h"
22#include "channel.h"
23#include "socket.h"
24#include "migration.h"
25#include "qemu-file.h"
26#include "io/channel-socket.h"
27#include "io/net-listener.h"
28#include "trace.h"
29
30
31static SocketAddress *tcp_build_address(const char *host_port, Error **errp)
32{
33 SocketAddress *saddr;
34
35 saddr = g_new0(SocketAddress, 1);
36 saddr->type = SOCKET_ADDRESS_TYPE_INET;
37
38 if (inet_parse(&saddr->u.inet, host_port, errp)) {
39 qapi_free_SocketAddress(saddr);
40 return NULL;
41 }
42
43 return saddr;
44}
45
46
47static SocketAddress *unix_build_address(const char *path)
48{
49 SocketAddress *saddr;
50
51 saddr = g_new0(SocketAddress, 1);
52 saddr->type = SOCKET_ADDRESS_TYPE_UNIX;
53 saddr->u.q_unix.path = g_strdup(path);
54
55 return saddr;
56}
57
58
59struct SocketConnectData {
60 MigrationState *s;
61 char *hostname;
62};
63
64static void socket_connect_data_free(void *opaque)
65{
66 struct SocketConnectData *data = opaque;
67 if (!data) {
68 return;
69 }
70 g_free(data->hostname);
71 g_free(data);
72}
73
74static void socket_outgoing_migration(QIOTask *task,
75 gpointer opaque)
76{
77 struct SocketConnectData *data = opaque;
78 QIOChannel *sioc = QIO_CHANNEL(qio_task_get_source(task));
79 Error *err = NULL;
80
81 if (qio_task_propagate_error(task, &err)) {
82 trace_migration_socket_outgoing_error(error_get_pretty(err));
83 } else {
84 trace_migration_socket_outgoing_connected(data->hostname);
85 }
86 migration_channel_connect(data->s, sioc, data->hostname, err);
87 object_unref(OBJECT(sioc));
88}
89
90static void socket_start_outgoing_migration(MigrationState *s,
91 SocketAddress *saddr,
92 Error **errp)
93{
94 QIOChannelSocket *sioc = qio_channel_socket_new();
95 struct SocketConnectData *data = g_new0(struct SocketConnectData, 1);
96
97 data->s = s;
98 if (saddr->type == SOCKET_ADDRESS_TYPE_INET) {
99 data->hostname = g_strdup(saddr->u.inet.host);
100 }
101
102 qio_channel_set_name(QIO_CHANNEL(sioc), "migration-socket-outgoing");
103 qio_channel_socket_connect_async(sioc,
104 saddr,
105 socket_outgoing_migration,
106 data,
107 socket_connect_data_free,
108 NULL);
109 qapi_free_SocketAddress(saddr);
110}
111
112void tcp_start_outgoing_migration(MigrationState *s,
113 const char *host_port,
114 Error **errp)
115{
116 Error *err = NULL;
117 SocketAddress *saddr = tcp_build_address(host_port, &err);
118 if (!err) {
119 socket_start_outgoing_migration(s, saddr, &err);
120 }
121 error_propagate(errp, err);
122}
123
124void unix_start_outgoing_migration(MigrationState *s,
125 const char *path,
126 Error **errp)
127{
128 SocketAddress *saddr = unix_build_address(path);
129 socket_start_outgoing_migration(s, saddr, errp);
130}
131
132
133static void socket_accept_incoming_migration(QIONetListener *listener,
134 QIOChannelSocket *cioc,
135 gpointer opaque)
136{
137 trace_migration_socket_incoming_accepted();
138
139 qio_channel_set_name(QIO_CHANNEL(cioc), "migration-socket-incoming");
140 migration_channel_process_incoming(QIO_CHANNEL(cioc));
141
142 if (migration_has_all_channels()) {
143
144 qio_net_listener_disconnect(listener);
145
146 object_unref(OBJECT(listener));
147 }
148}
149
150
151static void socket_start_incoming_migration(SocketAddress *saddr,
152 Error **errp)
153{
154 QIONetListener *listener = qio_net_listener_new();
155
156 qio_net_listener_set_name(listener, "migration-socket-listener");
157
158 if (qio_net_listener_open_sync(listener, saddr, errp) < 0) {
159 object_unref(OBJECT(listener));
160 return;
161 }
162
163 qio_net_listener_set_client_func(listener,
164 socket_accept_incoming_migration,
165 NULL, NULL);
166}
167
168void tcp_start_incoming_migration(const char *host_port, Error **errp)
169{
170 Error *err = NULL;
171 SocketAddress *saddr = tcp_build_address(host_port, &err);
172 if (!err) {
173 socket_start_incoming_migration(saddr, &err);
174 }
175 qapi_free_SocketAddress(saddr);
176 error_propagate(errp, err);
177}
178
179void unix_start_incoming_migration(const char *path, Error **errp)
180{
181 SocketAddress *saddr = unix_build_address(path);
182 socket_start_incoming_migration(saddr, errp);
183 qapi_free_SocketAddress(saddr);
184}
185