char-socket.c 48 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569
  1. /*
  2. * QEMU System Emulator
  3. *
  4. * Copyright (c) 2003-2008 Fabrice Bellard
  5. *
  6. * Permission is hereby granted, free of charge, to any person obtaining a copy
  7. * of this software and associated documentation files (the "Software"), to deal
  8. * in the Software without restriction, including without limitation the rights
  9. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10. * copies of the Software, and to permit persons to whom the Software is
  11. * furnished to do so, subject to the following conditions:
  12. *
  13. * The above copyright notice and this permission notice shall be included in
  14. * all copies or substantial portions of the Software.
  15. *
  16. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  19. * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22. * THE SOFTWARE.
  23. */
  24. #include "qemu/osdep.h"
  25. #include "chardev/char.h"
  26. #include "io/channel-socket.h"
  27. #include "io/channel-websock.h"
  28. #include "qemu/error-report.h"
  29. #include "qemu/module.h"
  30. #include "qemu/option.h"
  31. #include "qapi/error.h"
  32. #include "qapi/clone-visitor.h"
  33. #include "qapi/qapi-visit-sockets.h"
  34. #include "qemu/yank.h"
  35. #include "chardev/char-io.h"
  36. #include "chardev/char-socket.h"
  37. static gboolean socket_reconnect_timeout(gpointer opaque);
  38. static void tcp_chr_telnet_init(Chardev *chr);
  39. static void tcp_chr_change_state(SocketChardev *s, TCPChardevState state)
  40. {
  41. switch (state) {
  42. case TCP_CHARDEV_STATE_DISCONNECTED:
  43. break;
  44. case TCP_CHARDEV_STATE_CONNECTING:
  45. assert(s->state == TCP_CHARDEV_STATE_DISCONNECTED);
  46. break;
  47. case TCP_CHARDEV_STATE_CONNECTED:
  48. assert(s->state == TCP_CHARDEV_STATE_CONNECTING);
  49. break;
  50. }
  51. s->state = state;
  52. }
  53. static void tcp_chr_reconn_timer_cancel(SocketChardev *s)
  54. {
  55. if (s->reconnect_timer) {
  56. g_source_destroy(s->reconnect_timer);
  57. g_source_unref(s->reconnect_timer);
  58. s->reconnect_timer = NULL;
  59. }
  60. }
  61. static void qemu_chr_socket_restart_timer(Chardev *chr)
  62. {
  63. SocketChardev *s = SOCKET_CHARDEV(chr);
  64. char *name;
  65. assert(s->state == TCP_CHARDEV_STATE_DISCONNECTED);
  66. assert(!s->reconnect_timer);
  67. name = g_strdup_printf("chardev-socket-reconnect-%s", chr->label);
  68. s->reconnect_timer = qemu_chr_timeout_add_ms(chr,
  69. s->reconnect_time * 1000,
  70. socket_reconnect_timeout,
  71. chr);
  72. g_source_set_name(s->reconnect_timer, name);
  73. g_free(name);
  74. }
  75. static void check_report_connect_error(Chardev *chr,
  76. Error *err)
  77. {
  78. SocketChardev *s = SOCKET_CHARDEV(chr);
  79. if (!s->connect_err_reported) {
  80. error_reportf_err(err,
  81. "Unable to connect character device %s: ",
  82. chr->label);
  83. s->connect_err_reported = true;
  84. } else {
  85. error_free(err);
  86. }
  87. qemu_chr_socket_restart_timer(chr);
  88. }
  89. static void tcp_chr_accept(QIONetListener *listener,
  90. QIOChannelSocket *cioc,
  91. void *opaque);
  92. static int tcp_chr_read_poll(void *opaque);
  93. static void tcp_chr_disconnect_locked(Chardev *chr);
  94. /* Called with chr_write_lock held. */
  95. static int tcp_chr_write(Chardev *chr, const uint8_t *buf, int len)
  96. {
  97. SocketChardev *s = SOCKET_CHARDEV(chr);
  98. if (s->state == TCP_CHARDEV_STATE_CONNECTED) {
  99. int ret = io_channel_send_full(s->ioc, buf, len,
  100. s->write_msgfds,
  101. s->write_msgfds_num);
  102. /* free the written msgfds in any cases
  103. * other than ret < 0 && errno == EAGAIN
  104. */
  105. if (!(ret < 0 && EAGAIN == errno)
  106. && s->write_msgfds_num) {
  107. g_free(s->write_msgfds);
  108. s->write_msgfds = 0;
  109. s->write_msgfds_num = 0;
  110. }
  111. if (ret < 0 && errno != EAGAIN) {
  112. if (tcp_chr_read_poll(chr) <= 0) {
  113. /* Perform disconnect and return error. */
  114. tcp_chr_disconnect_locked(chr);
  115. } /* else let the read handler finish it properly */
  116. }
  117. return ret;
  118. } else {
  119. /* Indicate an error. */
  120. errno = EIO;
  121. return -1;
  122. }
  123. }
  124. static int tcp_chr_read_poll(void *opaque)
  125. {
  126. Chardev *chr = CHARDEV(opaque);
  127. SocketChardev *s = SOCKET_CHARDEV(opaque);
  128. if (s->state != TCP_CHARDEV_STATE_CONNECTED) {
  129. return 0;
  130. }
  131. s->max_size = qemu_chr_be_can_write(chr);
  132. return s->max_size;
  133. }
  134. static void tcp_chr_process_IAC_bytes(Chardev *chr,
  135. SocketChardev *s,
  136. uint8_t *buf, int *size)
  137. {
  138. /* Handle any telnet or tn3270 client's basic IAC options.
  139. * For telnet options, it satisfies char by char mode with no echo.
  140. * For tn3270 options, it satisfies binary mode with EOR.
  141. * All IAC options will be removed from the buf and the do_opt
  142. * pointer will be used to track the state of the width of the
  143. * IAC information.
  144. *
  145. * RFC854: "All TELNET commands consist of at least a two byte sequence.
  146. * The commands dealing with option negotiation are three byte sequences,
  147. * the third byte being the code for the option referenced."
  148. * "IAC BREAK", "IAC IP", "IAC NOP" and the double IAC are two bytes.
  149. * "IAC SB", "IAC SE" and "IAC EOR" are saved to split up data boundary
  150. * for tn3270.
  151. * NOP, Break and Interrupt Process(IP) might be encountered during a TN3270
  152. * session, and NOP and IP need to be done later.
  153. */
  154. int i;
  155. int j = 0;
  156. for (i = 0; i < *size; i++) {
  157. if (s->do_telnetopt > 1) {
  158. if ((unsigned char)buf[i] == IAC && s->do_telnetopt == 2) {
  159. /* Double IAC means send an IAC */
  160. if (j != i) {
  161. buf[j] = buf[i];
  162. }
  163. j++;
  164. s->do_telnetopt = 1;
  165. } else {
  166. if ((unsigned char)buf[i] == IAC_BREAK
  167. && s->do_telnetopt == 2) {
  168. /* Handle IAC break commands by sending a serial break */
  169. qemu_chr_be_event(chr, CHR_EVENT_BREAK);
  170. s->do_telnetopt++;
  171. } else if (s->is_tn3270 && ((unsigned char)buf[i] == IAC_EOR
  172. || (unsigned char)buf[i] == IAC_SB
  173. || (unsigned char)buf[i] == IAC_SE)
  174. && s->do_telnetopt == 2) {
  175. buf[j++] = IAC;
  176. buf[j++] = buf[i];
  177. s->do_telnetopt++;
  178. } else if (s->is_tn3270 && ((unsigned char)buf[i] == IAC_IP
  179. || (unsigned char)buf[i] == IAC_NOP)
  180. && s->do_telnetopt == 2) {
  181. /* TODO: IP and NOP need to be implemented later. */
  182. s->do_telnetopt++;
  183. }
  184. s->do_telnetopt++;
  185. }
  186. if (s->do_telnetopt >= 4) {
  187. s->do_telnetopt = 1;
  188. }
  189. } else {
  190. if ((unsigned char)buf[i] == IAC) {
  191. s->do_telnetopt = 2;
  192. } else {
  193. if (j != i) {
  194. buf[j] = buf[i];
  195. }
  196. j++;
  197. }
  198. }
  199. }
  200. *size = j;
  201. }
  202. static int tcp_get_msgfds(Chardev *chr, int *fds, int num)
  203. {
  204. SocketChardev *s = SOCKET_CHARDEV(chr);
  205. int to_copy = (s->read_msgfds_num < num) ? s->read_msgfds_num : num;
  206. assert(num <= TCP_MAX_FDS);
  207. if (to_copy) {
  208. int i;
  209. memcpy(fds, s->read_msgfds, to_copy * sizeof(int));
  210. /* Close unused fds */
  211. for (i = to_copy; i < s->read_msgfds_num; i++) {
  212. close(s->read_msgfds[i]);
  213. }
  214. g_free(s->read_msgfds);
  215. s->read_msgfds = 0;
  216. s->read_msgfds_num = 0;
  217. }
  218. return to_copy;
  219. }
  220. static int tcp_set_msgfds(Chardev *chr, int *fds, int num)
  221. {
  222. SocketChardev *s = SOCKET_CHARDEV(chr);
  223. /* clear old pending fd array */
  224. g_free(s->write_msgfds);
  225. s->write_msgfds = NULL;
  226. s->write_msgfds_num = 0;
  227. if ((s->state != TCP_CHARDEV_STATE_CONNECTED) ||
  228. !qio_channel_has_feature(s->ioc,
  229. QIO_CHANNEL_FEATURE_FD_PASS)) {
  230. return -1;
  231. }
  232. if (num) {
  233. s->write_msgfds = g_new(int, num);
  234. memcpy(s->write_msgfds, fds, num * sizeof(int));
  235. }
  236. s->write_msgfds_num = num;
  237. return 0;
  238. }
  239. static ssize_t tcp_chr_recv(Chardev *chr, char *buf, size_t len)
  240. {
  241. SocketChardev *s = SOCKET_CHARDEV(chr);
  242. struct iovec iov = { .iov_base = buf, .iov_len = len };
  243. int ret;
  244. size_t i;
  245. int *msgfds = NULL;
  246. size_t msgfds_num = 0;
  247. if (qio_channel_has_feature(s->ioc, QIO_CHANNEL_FEATURE_FD_PASS)) {
  248. ret = qio_channel_readv_full(s->ioc, &iov, 1,
  249. &msgfds, &msgfds_num,
  250. 0, NULL);
  251. } else {
  252. ret = qio_channel_readv_full(s->ioc, &iov, 1,
  253. NULL, NULL,
  254. 0, NULL);
  255. }
  256. if (msgfds_num) {
  257. /* close and clean read_msgfds */
  258. for (i = 0; i < s->read_msgfds_num; i++) {
  259. close(s->read_msgfds[i]);
  260. }
  261. if (s->read_msgfds_num) {
  262. g_free(s->read_msgfds);
  263. }
  264. s->read_msgfds = msgfds;
  265. s->read_msgfds_num = msgfds_num;
  266. }
  267. for (i = 0; i < s->read_msgfds_num; i++) {
  268. int fd = s->read_msgfds[i];
  269. if (fd < 0) {
  270. continue;
  271. }
  272. /* O_NONBLOCK is preserved across SCM_RIGHTS so reset it */
  273. qemu_socket_set_block(fd);
  274. #ifndef MSG_CMSG_CLOEXEC
  275. qemu_set_cloexec(fd);
  276. #endif
  277. }
  278. if (ret == QIO_CHANNEL_ERR_BLOCK) {
  279. errno = EAGAIN;
  280. ret = -1;
  281. } else if (ret == -1) {
  282. errno = EIO;
  283. }
  284. return ret;
  285. }
  286. static GSource *tcp_chr_add_watch(Chardev *chr, GIOCondition cond)
  287. {
  288. SocketChardev *s = SOCKET_CHARDEV(chr);
  289. if (!s->ioc) {
  290. return NULL;
  291. }
  292. return qio_channel_create_watch(s->ioc, cond);
  293. }
  294. static void remove_hup_source(SocketChardev *s)
  295. {
  296. if (s->hup_source != NULL) {
  297. g_source_destroy(s->hup_source);
  298. g_source_unref(s->hup_source);
  299. s->hup_source = NULL;
  300. }
  301. }
  302. static void char_socket_yank_iochannel(void *opaque)
  303. {
  304. QIOChannel *ioc = QIO_CHANNEL(opaque);
  305. qio_channel_shutdown(ioc, QIO_CHANNEL_SHUTDOWN_BOTH, NULL);
  306. }
  307. static void tcp_chr_free_connection(Chardev *chr)
  308. {
  309. SocketChardev *s = SOCKET_CHARDEV(chr);
  310. int i;
  311. if (s->read_msgfds_num) {
  312. for (i = 0; i < s->read_msgfds_num; i++) {
  313. close(s->read_msgfds[i]);
  314. }
  315. g_free(s->read_msgfds);
  316. s->read_msgfds = NULL;
  317. s->read_msgfds_num = 0;
  318. }
  319. remove_hup_source(s);
  320. tcp_set_msgfds(chr, NULL, 0);
  321. remove_fd_in_watch(chr);
  322. if (s->registered_yank &&
  323. (s->state == TCP_CHARDEV_STATE_CONNECTING
  324. || s->state == TCP_CHARDEV_STATE_CONNECTED)) {
  325. yank_unregister_function(CHARDEV_YANK_INSTANCE(chr->label),
  326. char_socket_yank_iochannel,
  327. QIO_CHANNEL(s->sioc));
  328. }
  329. object_unref(OBJECT(s->sioc));
  330. s->sioc = NULL;
  331. object_unref(OBJECT(s->ioc));
  332. s->ioc = NULL;
  333. g_free(chr->filename);
  334. chr->filename = NULL;
  335. tcp_chr_change_state(s, TCP_CHARDEV_STATE_DISCONNECTED);
  336. }
  337. static const char *qemu_chr_socket_protocol(SocketChardev *s)
  338. {
  339. if (s->is_telnet) {
  340. return "telnet";
  341. }
  342. return s->is_websock ? "websocket" : "tcp";
  343. }
  344. static char *qemu_chr_socket_address(SocketChardev *s, const char *prefix)
  345. {
  346. switch (s->addr->type) {
  347. case SOCKET_ADDRESS_TYPE_INET:
  348. return g_strdup_printf("%s%s:%s:%s%s", prefix,
  349. qemu_chr_socket_protocol(s),
  350. s->addr->u.inet.host,
  351. s->addr->u.inet.port,
  352. s->is_listen ? ",server=on" : "");
  353. break;
  354. case SOCKET_ADDRESS_TYPE_UNIX:
  355. {
  356. const char *tight = "", *abstract = "";
  357. UnixSocketAddress *sa = &s->addr->u.q_unix;
  358. #ifdef CONFIG_LINUX
  359. if (sa->has_abstract && sa->abstract) {
  360. abstract = ",abstract=on";
  361. if (sa->has_tight && sa->tight) {
  362. tight = ",tight=on";
  363. }
  364. }
  365. #endif
  366. return g_strdup_printf("%sunix:%s%s%s%s", prefix, sa->path,
  367. abstract, tight,
  368. s->is_listen ? ",server=on" : "");
  369. break;
  370. }
  371. case SOCKET_ADDRESS_TYPE_FD:
  372. return g_strdup_printf("%sfd:%s%s", prefix, s->addr->u.fd.str,
  373. s->is_listen ? ",server=on" : "");
  374. break;
  375. case SOCKET_ADDRESS_TYPE_VSOCK:
  376. return g_strdup_printf("%svsock:%s:%s", prefix,
  377. s->addr->u.vsock.cid,
  378. s->addr->u.vsock.port);
  379. default:
  380. abort();
  381. }
  382. }
  383. static void update_disconnected_filename(SocketChardev *s)
  384. {
  385. Chardev *chr = CHARDEV(s);
  386. g_free(chr->filename);
  387. if (s->addr) {
  388. chr->filename = qemu_chr_socket_address(s, "disconnected:");
  389. } else {
  390. chr->filename = g_strdup("disconnected:socket");
  391. }
  392. }
  393. /* NB may be called even if tcp_chr_connect has not been
  394. * reached, due to TLS or telnet initialization failure,
  395. * so can *not* assume s->state == TCP_CHARDEV_STATE_CONNECTED
  396. * This must be called with chr->chr_write_lock held.
  397. */
  398. static void tcp_chr_disconnect_locked(Chardev *chr)
  399. {
  400. SocketChardev *s = SOCKET_CHARDEV(chr);
  401. bool emit_close = s->state == TCP_CHARDEV_STATE_CONNECTED;
  402. tcp_chr_free_connection(chr);
  403. if (s->listener) {
  404. qio_net_listener_set_client_func_full(s->listener, tcp_chr_accept,
  405. chr, NULL, chr->gcontext);
  406. }
  407. update_disconnected_filename(s);
  408. if (emit_close) {
  409. qemu_chr_be_event(chr, CHR_EVENT_CLOSED);
  410. }
  411. if (s->reconnect_time && !s->reconnect_timer) {
  412. qemu_chr_socket_restart_timer(chr);
  413. }
  414. }
  415. static void tcp_chr_disconnect(Chardev *chr)
  416. {
  417. qemu_mutex_lock(&chr->chr_write_lock);
  418. tcp_chr_disconnect_locked(chr);
  419. qemu_mutex_unlock(&chr->chr_write_lock);
  420. }
  421. static gboolean tcp_chr_read(QIOChannel *chan, GIOCondition cond, void *opaque)
  422. {
  423. Chardev *chr = CHARDEV(opaque);
  424. SocketChardev *s = SOCKET_CHARDEV(opaque);
  425. uint8_t buf[CHR_READ_BUF_LEN];
  426. int len, size;
  427. if ((s->state != TCP_CHARDEV_STATE_CONNECTED) ||
  428. s->max_size <= 0) {
  429. return TRUE;
  430. }
  431. len = sizeof(buf);
  432. if (len > s->max_size) {
  433. len = s->max_size;
  434. }
  435. size = tcp_chr_recv(chr, (void *)buf, len);
  436. if (size == 0 || (size == -1 && errno != EAGAIN)) {
  437. /* connection closed */
  438. tcp_chr_disconnect(chr);
  439. } else if (size > 0) {
  440. if (s->do_telnetopt) {
  441. tcp_chr_process_IAC_bytes(chr, s, buf, &size);
  442. }
  443. if (size > 0) {
  444. qemu_chr_be_write(chr, buf, size);
  445. }
  446. }
  447. return TRUE;
  448. }
  449. static gboolean tcp_chr_hup(QIOChannel *channel,
  450. GIOCondition cond,
  451. void *opaque)
  452. {
  453. Chardev *chr = CHARDEV(opaque);
  454. tcp_chr_disconnect(chr);
  455. return G_SOURCE_REMOVE;
  456. }
  457. static int tcp_chr_sync_read(Chardev *chr, const uint8_t *buf, int len)
  458. {
  459. SocketChardev *s = SOCKET_CHARDEV(chr);
  460. int size;
  461. int saved_errno;
  462. if (s->state != TCP_CHARDEV_STATE_CONNECTED) {
  463. return 0;
  464. }
  465. qio_channel_set_blocking(s->ioc, true, NULL);
  466. size = tcp_chr_recv(chr, (void *) buf, len);
  467. saved_errno = errno;
  468. if (s->state != TCP_CHARDEV_STATE_DISCONNECTED) {
  469. qio_channel_set_blocking(s->ioc, false, NULL);
  470. }
  471. if (size == 0) {
  472. /* connection closed */
  473. tcp_chr_disconnect(chr);
  474. }
  475. errno = saved_errno;
  476. return size;
  477. }
  478. static char *qemu_chr_compute_filename(SocketChardev *s)
  479. {
  480. struct sockaddr_storage *ss = &s->sioc->localAddr;
  481. struct sockaddr_storage *ps = &s->sioc->remoteAddr;
  482. socklen_t ss_len = s->sioc->localAddrLen;
  483. socklen_t ps_len = s->sioc->remoteAddrLen;
  484. char shost[NI_MAXHOST], sserv[NI_MAXSERV];
  485. char phost[NI_MAXHOST], pserv[NI_MAXSERV];
  486. const char *left = "", *right = "";
  487. switch (ss->ss_family) {
  488. case AF_UNIX:
  489. return g_strdup_printf("unix:%s%s",
  490. ((struct sockaddr_un *)(ss))->sun_path,
  491. s->is_listen ? ",server=on" : "");
  492. case AF_INET6:
  493. left = "[";
  494. right = "]";
  495. /* fall through */
  496. case AF_INET:
  497. getnameinfo((struct sockaddr *) ss, ss_len, shost, sizeof(shost),
  498. sserv, sizeof(sserv), NI_NUMERICHOST | NI_NUMERICSERV);
  499. getnameinfo((struct sockaddr *) ps, ps_len, phost, sizeof(phost),
  500. pserv, sizeof(pserv), NI_NUMERICHOST | NI_NUMERICSERV);
  501. return g_strdup_printf("%s:%s%s%s:%s%s <-> %s%s%s:%s",
  502. qemu_chr_socket_protocol(s),
  503. left, shost, right, sserv,
  504. s->is_listen ? ",server=on" : "",
  505. left, phost, right, pserv);
  506. default:
  507. return g_strdup_printf("unknown");
  508. }
  509. }
  510. static void update_ioc_handlers(SocketChardev *s)
  511. {
  512. Chardev *chr = CHARDEV(s);
  513. if (s->state != TCP_CHARDEV_STATE_CONNECTED) {
  514. return;
  515. }
  516. remove_fd_in_watch(chr);
  517. chr->gsource = io_add_watch_poll(chr, s->ioc,
  518. tcp_chr_read_poll,
  519. tcp_chr_read, chr,
  520. chr->gcontext);
  521. remove_hup_source(s);
  522. s->hup_source = qio_channel_create_watch(s->ioc, G_IO_HUP);
  523. g_source_set_callback(s->hup_source, (GSourceFunc)tcp_chr_hup,
  524. chr, NULL);
  525. g_source_attach(s->hup_source, chr->gcontext);
  526. }
  527. static void tcp_chr_connect(void *opaque)
  528. {
  529. Chardev *chr = CHARDEV(opaque);
  530. SocketChardev *s = SOCKET_CHARDEV(opaque);
  531. g_free(chr->filename);
  532. chr->filename = qemu_chr_compute_filename(s);
  533. tcp_chr_change_state(s, TCP_CHARDEV_STATE_CONNECTED);
  534. update_ioc_handlers(s);
  535. qemu_chr_be_event(chr, CHR_EVENT_OPENED);
  536. }
  537. static void tcp_chr_telnet_destroy(SocketChardev *s)
  538. {
  539. if (s->telnet_source) {
  540. g_source_destroy(s->telnet_source);
  541. g_source_unref(s->telnet_source);
  542. s->telnet_source = NULL;
  543. }
  544. }
  545. static void tcp_chr_update_read_handler(Chardev *chr)
  546. {
  547. SocketChardev *s = SOCKET_CHARDEV(chr);
  548. if (s->listener && s->state == TCP_CHARDEV_STATE_DISCONNECTED) {
  549. /*
  550. * It's possible that chardev context is changed in
  551. * qemu_chr_be_update_read_handlers(). Reset it for QIO net
  552. * listener if there is.
  553. */
  554. qio_net_listener_set_client_func_full(s->listener, tcp_chr_accept,
  555. chr, NULL, chr->gcontext);
  556. }
  557. if (s->telnet_source) {
  558. tcp_chr_telnet_init(CHARDEV(s));
  559. }
  560. update_ioc_handlers(s);
  561. }
  562. static gboolean tcp_chr_telnet_init_io(QIOChannel *ioc,
  563. GIOCondition cond G_GNUC_UNUSED,
  564. gpointer user_data)
  565. {
  566. SocketChardev *s = user_data;
  567. Chardev *chr = CHARDEV(s);
  568. TCPChardevTelnetInit *init = s->telnet_init;
  569. ssize_t ret;
  570. assert(init);
  571. ret = qio_channel_write(ioc, init->buf, init->buflen, NULL);
  572. if (ret < 0) {
  573. if (ret == QIO_CHANNEL_ERR_BLOCK) {
  574. ret = 0;
  575. } else {
  576. tcp_chr_disconnect(chr);
  577. goto end;
  578. }
  579. }
  580. init->buflen -= ret;
  581. if (init->buflen == 0) {
  582. tcp_chr_connect(chr);
  583. goto end;
  584. }
  585. memmove(init->buf, init->buf + ret, init->buflen);
  586. return G_SOURCE_CONTINUE;
  587. end:
  588. g_free(s->telnet_init);
  589. s->telnet_init = NULL;
  590. g_source_unref(s->telnet_source);
  591. s->telnet_source = NULL;
  592. return G_SOURCE_REMOVE;
  593. }
  594. static void tcp_chr_telnet_init(Chardev *chr)
  595. {
  596. SocketChardev *s = SOCKET_CHARDEV(chr);
  597. TCPChardevTelnetInit *init;
  598. size_t n = 0;
  599. /* Destroy existing task */
  600. tcp_chr_telnet_destroy(s);
  601. if (s->telnet_init) {
  602. /* We are possibly during a handshake already */
  603. goto cont;
  604. }
  605. s->telnet_init = g_new0(TCPChardevTelnetInit, 1);
  606. init = s->telnet_init;
  607. #define IACSET(x, a, b, c) \
  608. do { \
  609. x[n++] = a; \
  610. x[n++] = b; \
  611. x[n++] = c; \
  612. } while (0)
  613. if (!s->is_tn3270) {
  614. init->buflen = 12;
  615. /* Prep the telnet negotiation to put telnet in binary,
  616. * no echo, single char mode */
  617. IACSET(init->buf, 0xff, 0xfb, 0x01); /* IAC WILL ECHO */
  618. IACSET(init->buf, 0xff, 0xfb, 0x03); /* IAC WILL Suppress go ahead */
  619. IACSET(init->buf, 0xff, 0xfb, 0x00); /* IAC WILL Binary */
  620. IACSET(init->buf, 0xff, 0xfd, 0x00); /* IAC DO Binary */
  621. } else {
  622. init->buflen = 21;
  623. /* Prep the TN3270 negotiation based on RFC1576 */
  624. IACSET(init->buf, 0xff, 0xfd, 0x19); /* IAC DO EOR */
  625. IACSET(init->buf, 0xff, 0xfb, 0x19); /* IAC WILL EOR */
  626. IACSET(init->buf, 0xff, 0xfd, 0x00); /* IAC DO BINARY */
  627. IACSET(init->buf, 0xff, 0xfb, 0x00); /* IAC WILL BINARY */
  628. IACSET(init->buf, 0xff, 0xfd, 0x18); /* IAC DO TERMINAL TYPE */
  629. IACSET(init->buf, 0xff, 0xfa, 0x18); /* IAC SB TERMINAL TYPE */
  630. IACSET(init->buf, 0x01, 0xff, 0xf0); /* SEND IAC SE */
  631. }
  632. #undef IACSET
  633. cont:
  634. s->telnet_source = qio_channel_add_watch_source(s->ioc, G_IO_OUT,
  635. tcp_chr_telnet_init_io,
  636. s, NULL,
  637. chr->gcontext);
  638. }
  639. static void tcp_chr_websock_handshake(QIOTask *task, gpointer user_data)
  640. {
  641. Chardev *chr = user_data;
  642. SocketChardev *s = user_data;
  643. Error *err = NULL;
  644. if (qio_task_propagate_error(task, &err)) {
  645. error_reportf_err(err,
  646. "websock handshake of character device %s failed: ",
  647. chr->label);
  648. tcp_chr_disconnect(chr);
  649. } else {
  650. if (s->do_telnetopt) {
  651. tcp_chr_telnet_init(chr);
  652. } else {
  653. tcp_chr_connect(chr);
  654. }
  655. }
  656. }
  657. static void tcp_chr_websock_init(Chardev *chr)
  658. {
  659. SocketChardev *s = SOCKET_CHARDEV(chr);
  660. QIOChannelWebsock *wioc = NULL;
  661. gchar *name;
  662. wioc = qio_channel_websock_new_server(s->ioc);
  663. name = g_strdup_printf("chardev-websocket-server-%s", chr->label);
  664. qio_channel_set_name(QIO_CHANNEL(wioc), name);
  665. g_free(name);
  666. object_unref(OBJECT(s->ioc));
  667. s->ioc = QIO_CHANNEL(wioc);
  668. qio_channel_websock_handshake(wioc, tcp_chr_websock_handshake, chr, NULL);
  669. }
  670. static void tcp_chr_tls_handshake(QIOTask *task,
  671. gpointer user_data)
  672. {
  673. Chardev *chr = user_data;
  674. SocketChardev *s = user_data;
  675. Error *err = NULL;
  676. if (qio_task_propagate_error(task, &err)) {
  677. error_reportf_err(err,
  678. "TLS handshake of character device %s failed: ",
  679. chr->label);
  680. tcp_chr_disconnect(chr);
  681. } else {
  682. if (s->is_websock) {
  683. tcp_chr_websock_init(chr);
  684. } else if (s->do_telnetopt) {
  685. tcp_chr_telnet_init(chr);
  686. } else {
  687. tcp_chr_connect(chr);
  688. }
  689. }
  690. }
  691. static void tcp_chr_tls_init(Chardev *chr)
  692. {
  693. SocketChardev *s = SOCKET_CHARDEV(chr);
  694. QIOChannelTLS *tioc;
  695. gchar *name;
  696. if (s->is_listen) {
  697. tioc = qio_channel_tls_new_server(
  698. s->ioc, s->tls_creds,
  699. s->tls_authz,
  700. NULL);
  701. } else {
  702. tioc = qio_channel_tls_new_client(
  703. s->ioc, s->tls_creds,
  704. s->addr->u.inet.host,
  705. NULL);
  706. }
  707. if (tioc == NULL) {
  708. tcp_chr_disconnect(chr);
  709. return;
  710. }
  711. name = g_strdup_printf("chardev-tls-%s-%s",
  712. s->is_listen ? "server" : "client",
  713. chr->label);
  714. qio_channel_set_name(QIO_CHANNEL(tioc), name);
  715. g_free(name);
  716. object_unref(OBJECT(s->ioc));
  717. s->ioc = QIO_CHANNEL(tioc);
  718. qio_channel_tls_handshake(tioc,
  719. tcp_chr_tls_handshake,
  720. chr,
  721. NULL,
  722. chr->gcontext);
  723. }
  724. static void tcp_chr_set_client_ioc_name(Chardev *chr,
  725. QIOChannelSocket *sioc)
  726. {
  727. SocketChardev *s = SOCKET_CHARDEV(chr);
  728. char *name;
  729. name = g_strdup_printf("chardev-tcp-%s-%s",
  730. s->is_listen ? "server" : "client",
  731. chr->label);
  732. qio_channel_set_name(QIO_CHANNEL(sioc), name);
  733. g_free(name);
  734. }
  735. static int tcp_chr_new_client(Chardev *chr, QIOChannelSocket *sioc)
  736. {
  737. SocketChardev *s = SOCKET_CHARDEV(chr);
  738. if (s->state != TCP_CHARDEV_STATE_CONNECTING) {
  739. return -1;
  740. }
  741. s->ioc = QIO_CHANNEL(sioc);
  742. object_ref(OBJECT(sioc));
  743. s->sioc = sioc;
  744. object_ref(OBJECT(sioc));
  745. qio_channel_set_blocking(s->ioc, false, NULL);
  746. if (s->do_nodelay) {
  747. qio_channel_set_delay(s->ioc, false);
  748. }
  749. if (s->listener) {
  750. qio_net_listener_set_client_func_full(s->listener, NULL, NULL,
  751. NULL, chr->gcontext);
  752. }
  753. if (s->tls_creds) {
  754. tcp_chr_tls_init(chr);
  755. } else if (s->is_websock) {
  756. tcp_chr_websock_init(chr);
  757. } else if (s->do_telnetopt) {
  758. tcp_chr_telnet_init(chr);
  759. } else {
  760. tcp_chr_connect(chr);
  761. }
  762. return 0;
  763. }
  764. static int tcp_chr_add_client(Chardev *chr, int fd)
  765. {
  766. int ret;
  767. QIOChannelSocket *sioc;
  768. SocketChardev *s = SOCKET_CHARDEV(chr);
  769. if (s->state != TCP_CHARDEV_STATE_DISCONNECTED) {
  770. return -1;
  771. }
  772. sioc = qio_channel_socket_new_fd(fd, NULL);
  773. if (!sioc) {
  774. return -1;
  775. }
  776. tcp_chr_change_state(s, TCP_CHARDEV_STATE_CONNECTING);
  777. tcp_chr_set_client_ioc_name(chr, sioc);
  778. if (s->registered_yank) {
  779. yank_register_function(CHARDEV_YANK_INSTANCE(chr->label),
  780. char_socket_yank_iochannel,
  781. QIO_CHANNEL(sioc));
  782. }
  783. ret = tcp_chr_new_client(chr, sioc);
  784. object_unref(OBJECT(sioc));
  785. return ret;
  786. }
  787. static void tcp_chr_accept(QIONetListener *listener,
  788. QIOChannelSocket *cioc,
  789. void *opaque)
  790. {
  791. Chardev *chr = CHARDEV(opaque);
  792. SocketChardev *s = SOCKET_CHARDEV(chr);
  793. tcp_chr_change_state(s, TCP_CHARDEV_STATE_CONNECTING);
  794. tcp_chr_set_client_ioc_name(chr, cioc);
  795. if (s->registered_yank) {
  796. yank_register_function(CHARDEV_YANK_INSTANCE(chr->label),
  797. char_socket_yank_iochannel,
  798. QIO_CHANNEL(cioc));
  799. }
  800. tcp_chr_new_client(chr, cioc);
  801. }
  802. static int tcp_chr_connect_client_sync(Chardev *chr, Error **errp)
  803. {
  804. SocketChardev *s = SOCKET_CHARDEV(chr);
  805. QIOChannelSocket *sioc = qio_channel_socket_new();
  806. tcp_chr_change_state(s, TCP_CHARDEV_STATE_CONNECTING);
  807. tcp_chr_set_client_ioc_name(chr, sioc);
  808. if (qio_channel_socket_connect_sync(sioc, s->addr, errp) < 0) {
  809. tcp_chr_change_state(s, TCP_CHARDEV_STATE_DISCONNECTED);
  810. object_unref(OBJECT(sioc));
  811. return -1;
  812. }
  813. if (s->registered_yank) {
  814. yank_register_function(CHARDEV_YANK_INSTANCE(chr->label),
  815. char_socket_yank_iochannel,
  816. QIO_CHANNEL(sioc));
  817. }
  818. tcp_chr_new_client(chr, sioc);
  819. object_unref(OBJECT(sioc));
  820. return 0;
  821. }
  822. static void tcp_chr_accept_server_sync(Chardev *chr)
  823. {
  824. SocketChardev *s = SOCKET_CHARDEV(chr);
  825. QIOChannelSocket *sioc;
  826. info_report("QEMU waiting for connection on: %s",
  827. chr->filename);
  828. tcp_chr_change_state(s, TCP_CHARDEV_STATE_CONNECTING);
  829. sioc = qio_net_listener_wait_client(s->listener);
  830. tcp_chr_set_client_ioc_name(chr, sioc);
  831. if (s->registered_yank) {
  832. yank_register_function(CHARDEV_YANK_INSTANCE(chr->label),
  833. char_socket_yank_iochannel,
  834. QIO_CHANNEL(sioc));
  835. }
  836. tcp_chr_new_client(chr, sioc);
  837. object_unref(OBJECT(sioc));
  838. }
  839. static int tcp_chr_wait_connected(Chardev *chr, Error **errp)
  840. {
  841. SocketChardev *s = SOCKET_CHARDEV(chr);
  842. const char *opts[] = { "telnet", "tn3270", "websock", "tls-creds" };
  843. bool optset[] = { s->is_telnet, s->is_tn3270, s->is_websock, s->tls_creds };
  844. size_t i;
  845. QEMU_BUILD_BUG_ON(G_N_ELEMENTS(opts) != G_N_ELEMENTS(optset));
  846. for (i = 0; i < G_N_ELEMENTS(opts); i++) {
  847. if (optset[i]) {
  848. error_setg(errp,
  849. "'%s' option is incompatible with waiting for "
  850. "connection completion", opts[i]);
  851. return -1;
  852. }
  853. }
  854. tcp_chr_reconn_timer_cancel(s);
  855. /*
  856. * We expect states to be as follows:
  857. *
  858. * - server
  859. * - wait -> CONNECTED
  860. * - nowait -> DISCONNECTED
  861. * - client
  862. * - reconnect == 0 -> CONNECTED
  863. * - reconnect != 0 -> CONNECTING
  864. *
  865. */
  866. if (s->state == TCP_CHARDEV_STATE_CONNECTING) {
  867. if (!s->connect_task) {
  868. error_setg(errp,
  869. "Unexpected 'connecting' state without connect task "
  870. "while waiting for connection completion");
  871. return -1;
  872. }
  873. /*
  874. * tcp_chr_wait_connected should only ever be run from the
  875. * main loop thread associated with chr->gcontext, otherwise
  876. * qio_task_wait_thread has a dangerous race condition with
  877. * free'ing of the s->connect_task object.
  878. *
  879. * Acquiring the main context doesn't 100% prove we're in
  880. * the main loop thread, but it does at least guarantee
  881. * that the main loop won't be executed by another thread
  882. * avoiding the race condition with the task idle callback.
  883. */
  884. g_main_context_acquire(chr->gcontext);
  885. qio_task_wait_thread(s->connect_task);
  886. g_main_context_release(chr->gcontext);
  887. /*
  888. * The completion callback (qemu_chr_socket_connected) for
  889. * s->connect_task should have set this to NULL by the time
  890. * qio_task_wait_thread has returned.
  891. */
  892. assert(!s->connect_task);
  893. /*
  894. * NB we are *not* guaranteed to have "s->state == ..CONNECTED"
  895. * at this point as this first connect may be failed, so
  896. * allow the next loop to run regardless.
  897. */
  898. }
  899. while (s->state != TCP_CHARDEV_STATE_CONNECTED) {
  900. if (s->is_listen) {
  901. tcp_chr_accept_server_sync(chr);
  902. } else {
  903. Error *err = NULL;
  904. if (tcp_chr_connect_client_sync(chr, &err) < 0) {
  905. if (s->reconnect_time) {
  906. error_free(err);
  907. g_usleep(s->reconnect_time * 1000ULL * 1000ULL);
  908. } else {
  909. error_propagate(errp, err);
  910. return -1;
  911. }
  912. }
  913. }
  914. }
  915. return 0;
  916. }
  917. static void char_socket_finalize(Object *obj)
  918. {
  919. Chardev *chr = CHARDEV(obj);
  920. SocketChardev *s = SOCKET_CHARDEV(obj);
  921. tcp_chr_free_connection(chr);
  922. tcp_chr_reconn_timer_cancel(s);
  923. qapi_free_SocketAddress(s->addr);
  924. tcp_chr_telnet_destroy(s);
  925. g_free(s->telnet_init);
  926. if (s->listener) {
  927. qio_net_listener_set_client_func_full(s->listener, NULL, NULL,
  928. NULL, chr->gcontext);
  929. object_unref(OBJECT(s->listener));
  930. s->listener = NULL;
  931. }
  932. if (s->tls_creds) {
  933. object_unref(OBJECT(s->tls_creds));
  934. }
  935. g_free(s->tls_authz);
  936. if (s->registered_yank) {
  937. /*
  938. * In the chardev-change special-case, we shouldn't unregister the yank
  939. * instance, as it still may be needed.
  940. */
  941. if (!chr->handover_yank_instance) {
  942. yank_unregister_instance(CHARDEV_YANK_INSTANCE(chr->label));
  943. }
  944. }
  945. qemu_chr_be_event(chr, CHR_EVENT_CLOSED);
  946. }
  947. static void qemu_chr_socket_connected(QIOTask *task, void *opaque)
  948. {
  949. QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(qio_task_get_source(task));
  950. Chardev *chr = CHARDEV(opaque);
  951. SocketChardev *s = SOCKET_CHARDEV(chr);
  952. Error *err = NULL;
  953. s->connect_task = NULL;
  954. if (qio_task_propagate_error(task, &err)) {
  955. tcp_chr_change_state(s, TCP_CHARDEV_STATE_DISCONNECTED);
  956. if (s->registered_yank) {
  957. yank_unregister_function(CHARDEV_YANK_INSTANCE(chr->label),
  958. char_socket_yank_iochannel,
  959. QIO_CHANNEL(sioc));
  960. }
  961. check_report_connect_error(chr, err);
  962. goto cleanup;
  963. }
  964. s->connect_err_reported = false;
  965. tcp_chr_new_client(chr, sioc);
  966. cleanup:
  967. object_unref(OBJECT(sioc));
  968. }
  969. static void tcp_chr_connect_client_task(QIOTask *task,
  970. gpointer opaque)
  971. {
  972. QIOChannelSocket *ioc = QIO_CHANNEL_SOCKET(qio_task_get_source(task));
  973. SocketAddress *addr = opaque;
  974. Error *err = NULL;
  975. qio_channel_socket_connect_sync(ioc, addr, &err);
  976. qio_task_set_error(task, err);
  977. }
  978. static void tcp_chr_connect_client_async(Chardev *chr)
  979. {
  980. SocketChardev *s = SOCKET_CHARDEV(chr);
  981. QIOChannelSocket *sioc;
  982. tcp_chr_change_state(s, TCP_CHARDEV_STATE_CONNECTING);
  983. sioc = qio_channel_socket_new();
  984. tcp_chr_set_client_ioc_name(chr, sioc);
  985. if (s->registered_yank) {
  986. yank_register_function(CHARDEV_YANK_INSTANCE(chr->label),
  987. char_socket_yank_iochannel,
  988. QIO_CHANNEL(sioc));
  989. }
  990. /*
  991. * Normally code would use the qio_channel_socket_connect_async
  992. * method which uses a QIOTask + qio_task_set_error internally
  993. * to avoid blocking. The tcp_chr_wait_connected method, however,
  994. * needs a way to synchronize with completion of the background
  995. * connect task which can't be done with the QIOChannelSocket
  996. * async APIs. Thus we must use QIOTask directly to implement
  997. * the non-blocking concept locally.
  998. */
  999. s->connect_task = qio_task_new(OBJECT(sioc),
  1000. qemu_chr_socket_connected,
  1001. object_ref(OBJECT(chr)),
  1002. (GDestroyNotify)object_unref);
  1003. qio_task_run_in_thread(s->connect_task,
  1004. tcp_chr_connect_client_task,
  1005. s->addr,
  1006. NULL,
  1007. chr->gcontext);
  1008. }
  1009. static gboolean socket_reconnect_timeout(gpointer opaque)
  1010. {
  1011. Chardev *chr = CHARDEV(opaque);
  1012. SocketChardev *s = SOCKET_CHARDEV(opaque);
  1013. qemu_mutex_lock(&chr->chr_write_lock);
  1014. g_source_unref(s->reconnect_timer);
  1015. s->reconnect_timer = NULL;
  1016. qemu_mutex_unlock(&chr->chr_write_lock);
  1017. if (chr->be_open) {
  1018. return false;
  1019. }
  1020. tcp_chr_connect_client_async(chr);
  1021. return false;
  1022. }
  1023. static int qmp_chardev_open_socket_server(Chardev *chr,
  1024. bool is_telnet,
  1025. bool is_waitconnect,
  1026. Error **errp)
  1027. {
  1028. SocketChardev *s = SOCKET_CHARDEV(chr);
  1029. char *name;
  1030. if (is_telnet) {
  1031. s->do_telnetopt = 1;
  1032. }
  1033. s->listener = qio_net_listener_new();
  1034. name = g_strdup_printf("chardev-tcp-listener-%s", chr->label);
  1035. qio_net_listener_set_name(s->listener, name);
  1036. g_free(name);
  1037. if (s->addr->type == SOCKET_ADDRESS_TYPE_FD && !*s->addr->u.fd.str) {
  1038. goto skip_listen;
  1039. }
  1040. if (qio_net_listener_open_sync(s->listener, s->addr, 1, errp) < 0) {
  1041. object_unref(OBJECT(s->listener));
  1042. s->listener = NULL;
  1043. return -1;
  1044. }
  1045. qapi_free_SocketAddress(s->addr);
  1046. s->addr = socket_local_address(s->listener->sioc[0]->fd, errp);
  1047. skip_listen:
  1048. update_disconnected_filename(s);
  1049. if (is_waitconnect) {
  1050. tcp_chr_accept_server_sync(chr);
  1051. } else {
  1052. qio_net_listener_set_client_func_full(s->listener,
  1053. tcp_chr_accept,
  1054. chr, NULL,
  1055. chr->gcontext);
  1056. }
  1057. return 0;
  1058. }
  1059. static int qmp_chardev_open_socket_client(Chardev *chr,
  1060. int64_t reconnect,
  1061. Error **errp)
  1062. {
  1063. SocketChardev *s = SOCKET_CHARDEV(chr);
  1064. if (reconnect > 0) {
  1065. s->reconnect_time = reconnect;
  1066. tcp_chr_connect_client_async(chr);
  1067. return 0;
  1068. } else {
  1069. return tcp_chr_connect_client_sync(chr, errp);
  1070. }
  1071. }
  1072. static bool qmp_chardev_validate_socket(ChardevSocket *sock,
  1073. SocketAddress *addr,
  1074. Error **errp)
  1075. {
  1076. /* Validate any options which have a dependency on address type */
  1077. switch (addr->type) {
  1078. case SOCKET_ADDRESS_TYPE_FD:
  1079. if (sock->has_reconnect) {
  1080. error_setg(errp,
  1081. "'reconnect' option is incompatible with "
  1082. "'fd' address type");
  1083. return false;
  1084. }
  1085. if (sock->tls_creds &&
  1086. !(sock->has_server && sock->server)) {
  1087. error_setg(errp,
  1088. "'tls_creds' option is incompatible with "
  1089. "'fd' address type as client");
  1090. return false;
  1091. }
  1092. break;
  1093. case SOCKET_ADDRESS_TYPE_UNIX:
  1094. if (sock->tls_creds) {
  1095. error_setg(errp,
  1096. "'tls_creds' option is incompatible with "
  1097. "'unix' address type");
  1098. return false;
  1099. }
  1100. break;
  1101. case SOCKET_ADDRESS_TYPE_INET:
  1102. break;
  1103. case SOCKET_ADDRESS_TYPE_VSOCK:
  1104. if (sock->tls_creds) {
  1105. error_setg(errp,
  1106. "'tls_creds' option is incompatible with "
  1107. "'vsock' address type");
  1108. return false;
  1109. }
  1110. default:
  1111. break;
  1112. }
  1113. if (sock->tls_authz && !sock->tls_creds) {
  1114. error_setg(errp, "'tls_authz' option requires 'tls_creds' option");
  1115. return false;
  1116. }
  1117. /* Validate any options which have a dependency on client vs server */
  1118. if (!sock->has_server || sock->server) {
  1119. if (sock->has_reconnect) {
  1120. error_setg(errp,
  1121. "'reconnect' option is incompatible with "
  1122. "socket in server listen mode");
  1123. return false;
  1124. }
  1125. } else {
  1126. if (sock->has_websocket && sock->websocket) {
  1127. error_setg(errp, "%s", "Websocket client is not implemented");
  1128. return false;
  1129. }
  1130. if (sock->has_wait) {
  1131. error_setg(errp, "%s",
  1132. "'wait' option is incompatible with "
  1133. "socket in client connect mode");
  1134. return false;
  1135. }
  1136. }
  1137. return true;
  1138. }
  1139. static void qmp_chardev_open_socket(Chardev *chr,
  1140. ChardevBackend *backend,
  1141. bool *be_opened,
  1142. Error **errp)
  1143. {
  1144. SocketChardev *s = SOCKET_CHARDEV(chr);
  1145. ChardevSocket *sock = backend->u.socket.data;
  1146. bool do_nodelay = sock->has_nodelay ? sock->nodelay : false;
  1147. bool is_listen = sock->has_server ? sock->server : true;
  1148. bool is_telnet = sock->has_telnet ? sock->telnet : false;
  1149. bool is_tn3270 = sock->has_tn3270 ? sock->tn3270 : false;
  1150. bool is_waitconnect = sock->has_wait ? sock->wait : false;
  1151. bool is_websock = sock->has_websocket ? sock->websocket : false;
  1152. int64_t reconnect = sock->has_reconnect ? sock->reconnect : 0;
  1153. SocketAddress *addr;
  1154. s->is_listen = is_listen;
  1155. s->is_telnet = is_telnet;
  1156. s->is_tn3270 = is_tn3270;
  1157. s->is_websock = is_websock;
  1158. s->do_nodelay = do_nodelay;
  1159. if (sock->tls_creds) {
  1160. Object *creds;
  1161. creds = object_resolve_path_component(
  1162. object_get_objects_root(), sock->tls_creds);
  1163. if (!creds) {
  1164. error_setg(errp, "No TLS credentials with id '%s'",
  1165. sock->tls_creds);
  1166. return;
  1167. }
  1168. s->tls_creds = (QCryptoTLSCreds *)
  1169. object_dynamic_cast(creds,
  1170. TYPE_QCRYPTO_TLS_CREDS);
  1171. if (!s->tls_creds) {
  1172. error_setg(errp, "Object with id '%s' is not TLS credentials",
  1173. sock->tls_creds);
  1174. return;
  1175. }
  1176. object_ref(OBJECT(s->tls_creds));
  1177. if (!qcrypto_tls_creds_check_endpoint(s->tls_creds,
  1178. is_listen
  1179. ? QCRYPTO_TLS_CREDS_ENDPOINT_SERVER
  1180. : QCRYPTO_TLS_CREDS_ENDPOINT_CLIENT,
  1181. errp)) {
  1182. return;
  1183. }
  1184. }
  1185. s->tls_authz = g_strdup(sock->tls_authz);
  1186. s->addr = addr = socket_address_flatten(sock->addr);
  1187. if (!qmp_chardev_validate_socket(sock, addr, errp)) {
  1188. return;
  1189. }
  1190. qemu_chr_set_feature(chr, QEMU_CHAR_FEATURE_RECONNECTABLE);
  1191. #ifndef _WIN32
  1192. /* TODO SOCKET_ADDRESS_FD where fd has AF_UNIX */
  1193. if (addr->type == SOCKET_ADDRESS_TYPE_UNIX) {
  1194. qemu_chr_set_feature(chr, QEMU_CHAR_FEATURE_FD_PASS);
  1195. }
  1196. #endif
  1197. /*
  1198. * In the chardev-change special-case, we shouldn't register a new yank
  1199. * instance, as there already may be one.
  1200. */
  1201. if (!chr->handover_yank_instance) {
  1202. if (!yank_register_instance(CHARDEV_YANK_INSTANCE(chr->label), errp)) {
  1203. return;
  1204. }
  1205. }
  1206. s->registered_yank = true;
  1207. /* be isn't opened until we get a connection */
  1208. *be_opened = false;
  1209. update_disconnected_filename(s);
  1210. if (s->is_listen) {
  1211. if (qmp_chardev_open_socket_server(chr, is_telnet || is_tn3270,
  1212. is_waitconnect, errp) < 0) {
  1213. return;
  1214. }
  1215. } else {
  1216. if (qmp_chardev_open_socket_client(chr, reconnect, errp) < 0) {
  1217. return;
  1218. }
  1219. }
  1220. }
  1221. static void qemu_chr_parse_socket(QemuOpts *opts, ChardevBackend *backend,
  1222. Error **errp)
  1223. {
  1224. const char *path = qemu_opt_get(opts, "path");
  1225. const char *host = qemu_opt_get(opts, "host");
  1226. const char *port = qemu_opt_get(opts, "port");
  1227. const char *fd = qemu_opt_get(opts, "fd");
  1228. #ifdef CONFIG_LINUX
  1229. bool tight = qemu_opt_get_bool(opts, "tight", true);
  1230. bool abstract = qemu_opt_get_bool(opts, "abstract", false);
  1231. #endif
  1232. SocketAddressLegacy *addr;
  1233. ChardevSocket *sock;
  1234. if ((!!path + !!fd + !!host) > 1) {
  1235. error_setg(errp,
  1236. "None or one of 'path', 'fd' or 'host' option required.");
  1237. return;
  1238. }
  1239. if (host && !port) {
  1240. error_setg(errp, "chardev: socket: no port given");
  1241. return;
  1242. }
  1243. backend->type = CHARDEV_BACKEND_KIND_SOCKET;
  1244. sock = backend->u.socket.data = g_new0(ChardevSocket, 1);
  1245. qemu_chr_parse_common(opts, qapi_ChardevSocket_base(sock));
  1246. if (qemu_opt_get(opts, "delay") && qemu_opt_get(opts, "nodelay")) {
  1247. error_setg(errp, "'delay' and 'nodelay' are mutually exclusive");
  1248. return;
  1249. }
  1250. sock->has_nodelay =
  1251. qemu_opt_get(opts, "delay") ||
  1252. qemu_opt_get(opts, "nodelay");
  1253. sock->nodelay =
  1254. !qemu_opt_get_bool(opts, "delay", true) ||
  1255. qemu_opt_get_bool(opts, "nodelay", false);
  1256. /*
  1257. * We have different default to QMP for 'server', hence
  1258. * we can't just check for existence of 'server'
  1259. */
  1260. sock->has_server = true;
  1261. sock->server = qemu_opt_get_bool(opts, "server", false);
  1262. sock->has_telnet = qemu_opt_get(opts, "telnet");
  1263. sock->telnet = qemu_opt_get_bool(opts, "telnet", false);
  1264. sock->has_tn3270 = qemu_opt_get(opts, "tn3270");
  1265. sock->tn3270 = qemu_opt_get_bool(opts, "tn3270", false);
  1266. sock->has_websocket = qemu_opt_get(opts, "websocket");
  1267. sock->websocket = qemu_opt_get_bool(opts, "websocket", false);
  1268. /*
  1269. * We have different default to QMP for 'wait' when 'server'
  1270. * is set, hence we can't just check for existence of 'wait'
  1271. */
  1272. sock->has_wait = qemu_opt_find(opts, "wait") || sock->server;
  1273. sock->wait = qemu_opt_get_bool(opts, "wait", true);
  1274. sock->has_reconnect = qemu_opt_find(opts, "reconnect");
  1275. sock->reconnect = qemu_opt_get_number(opts, "reconnect", 0);
  1276. sock->tls_creds = g_strdup(qemu_opt_get(opts, "tls-creds"));
  1277. sock->tls_authz = g_strdup(qemu_opt_get(opts, "tls-authz"));
  1278. addr = g_new0(SocketAddressLegacy, 1);
  1279. if (path) {
  1280. UnixSocketAddress *q_unix;
  1281. addr->type = SOCKET_ADDRESS_TYPE_UNIX;
  1282. q_unix = addr->u.q_unix.data = g_new0(UnixSocketAddress, 1);
  1283. q_unix->path = g_strdup(path);
  1284. #ifdef CONFIG_LINUX
  1285. q_unix->has_tight = true;
  1286. q_unix->tight = tight;
  1287. q_unix->has_abstract = true;
  1288. q_unix->abstract = abstract;
  1289. #endif
  1290. } else if (host) {
  1291. addr->type = SOCKET_ADDRESS_TYPE_INET;
  1292. addr->u.inet.data = g_new(InetSocketAddress, 1);
  1293. *addr->u.inet.data = (InetSocketAddress) {
  1294. .host = g_strdup(host),
  1295. .port = g_strdup(port),
  1296. .has_to = qemu_opt_get(opts, "to"),
  1297. .to = qemu_opt_get_number(opts, "to", 0),
  1298. .has_ipv4 = qemu_opt_get(opts, "ipv4"),
  1299. .ipv4 = qemu_opt_get_bool(opts, "ipv4", 0),
  1300. .has_ipv6 = qemu_opt_get(opts, "ipv6"),
  1301. .ipv6 = qemu_opt_get_bool(opts, "ipv6", 0),
  1302. };
  1303. } else {
  1304. addr->type = SOCKET_ADDRESS_TYPE_FD;
  1305. addr->u.fd.data = g_new(String, 1);
  1306. addr->u.fd.data->str = g_strdup(fd);
  1307. }
  1308. sock->addr = addr;
  1309. }
  1310. static void
  1311. char_socket_get_addr(Object *obj, Visitor *v, const char *name,
  1312. void *opaque, Error **errp)
  1313. {
  1314. SocketChardev *s = SOCKET_CHARDEV(obj);
  1315. visit_type_SocketAddress(v, name, &s->addr, errp);
  1316. }
  1317. static bool
  1318. char_socket_get_connected(Object *obj, Error **errp)
  1319. {
  1320. SocketChardev *s = SOCKET_CHARDEV(obj);
  1321. return s->state == TCP_CHARDEV_STATE_CONNECTED;
  1322. }
  1323. static void char_socket_class_init(ObjectClass *oc, void *data)
  1324. {
  1325. ChardevClass *cc = CHARDEV_CLASS(oc);
  1326. cc->supports_yank = true;
  1327. cc->parse = qemu_chr_parse_socket;
  1328. cc->open = qmp_chardev_open_socket;
  1329. cc->chr_wait_connected = tcp_chr_wait_connected;
  1330. cc->chr_write = tcp_chr_write;
  1331. cc->chr_sync_read = tcp_chr_sync_read;
  1332. cc->chr_disconnect = tcp_chr_disconnect;
  1333. cc->get_msgfds = tcp_get_msgfds;
  1334. cc->set_msgfds = tcp_set_msgfds;
  1335. cc->chr_add_client = tcp_chr_add_client;
  1336. cc->chr_add_watch = tcp_chr_add_watch;
  1337. cc->chr_update_read_handler = tcp_chr_update_read_handler;
  1338. object_class_property_add(oc, "addr", "SocketAddress",
  1339. char_socket_get_addr, NULL,
  1340. NULL, NULL);
  1341. object_class_property_add_bool(oc, "connected", char_socket_get_connected,
  1342. NULL);
  1343. }
  1344. static const TypeInfo char_socket_type_info = {
  1345. .name = TYPE_CHARDEV_SOCKET,
  1346. .parent = TYPE_CHARDEV,
  1347. .instance_size = sizeof(SocketChardev),
  1348. .instance_finalize = char_socket_finalize,
  1349. .class_init = char_socket_class_init,
  1350. };
  1351. static void register_types(void)
  1352. {
  1353. type_register_static(&char_socket_type_info);
  1354. }
  1355. type_init(register_types);