1 /* 2 * migration yank functions 3 * 4 * Copyright (c) Lukas Straub <lukasstraub2@web.de> 5 * 6 * This work is licensed under the terms of the GNU GPL, version 2 or later. 7 * See the COPYING file in the top-level directory. 8 */ 9 10 #include "qemu/osdep.h" 11 #include "qapi/error.h" 12 #include "io/channel.h" 13 #include "yank_functions.h" 14 #include "qemu/yank.h" 15 #include "io/channel-socket.h" 16 #include "io/channel-tls.h" 17 #include "qemu-file.h" 18 19 void migration_yank_iochannel(void *opaque) 20 { 21 QIOChannel *ioc = QIO_CHANNEL(opaque); 22 23 qio_channel_shutdown(ioc, QIO_CHANNEL_SHUTDOWN_BOTH, NULL); 24 } 25 26 /* Return whether yank is supported on this ioc */ 27 static bool migration_ioc_yank_supported(QIOChannel *ioc) 28 { 29 return object_dynamic_cast(OBJECT(ioc), TYPE_QIO_CHANNEL_SOCKET) || 30 object_dynamic_cast(OBJECT(ioc), TYPE_QIO_CHANNEL_TLS); 31 } 32 33 void migration_ioc_register_yank(QIOChannel *ioc) 34 { 35 if (migration_ioc_yank_supported(ioc)) { 36 yank_register_function(MIGRATION_YANK_INSTANCE, 37 migration_yank_iochannel, 38 ioc); 39 } 40 } 41 42 void migration_ioc_unregister_yank(QIOChannel *ioc) 43 { 44 if (migration_ioc_yank_supported(ioc)) { 45 yank_unregister_function(MIGRATION_YANK_INSTANCE, 46 migration_yank_iochannel, 47 ioc); 48 } 49 } 50 51 void migration_ioc_unregister_yank_from_file(QEMUFile *file) 52 { 53 QIOChannel *ioc = qemu_file_get_ioc(file); 54 55 if (ioc) { 56 /* 57 * For migration qemufiles, we'll always reach here. Though we'll skip 58 * calls from e.g. savevm/loadvm as they don't use yank. 59 */ 60 migration_ioc_unregister_yank(ioc); 61 } 62 } 63