浏览代码

migration/rdma: Split qemu_fopen_rdma() into input/output functions

This is how everything else in QEMUFile is structured.
As a bonus they are three less lines of code.

Reviewed-by: Peter Xu <peterx@redhat.com>
Message-ID: <20230530183941.7223-17-quintela@redhat.com>
Signed-off-by: Juan Quintela <quintela@redhat.com>
Juan Quintela 2 年之前
父节点
当前提交
697c4c86ab
共有 3 个文件被更改,包括 19 次插入33 次删除
  1. 0 12
      migration/qemu-file.c
  2. 0 1
      migration/qemu-file.h
  3. 19 20
      migration/rdma.c

+ 0 - 12
migration/qemu-file.c

@@ -100,18 +100,6 @@ int qemu_file_shutdown(QEMUFile *f)
     return 0;
     return 0;
 }
 }
 
 
-bool qemu_file_mode_is_not_valid(const char *mode)
-{
-    if (mode == NULL ||
-        (mode[0] != 'r' && mode[0] != 'w') ||
-        mode[1] != 'b' || mode[2] != 0) {
-        fprintf(stderr, "qemu_fopen: Argument validity check failed\n");
-        return true;
-    }
-
-    return false;
-}
-
 static QEMUFile *qemu_file_new_impl(QIOChannel *ioc, bool is_writable)
 static QEMUFile *qemu_file_new_impl(QIOChannel *ioc, bool is_writable)
 {
 {
     QEMUFile *f;
     QEMUFile *f;

+ 0 - 1
migration/qemu-file.h

@@ -102,7 +102,6 @@ uint64_t qemu_file_transferred_noflush(QEMUFile *f);
  */
  */
 void qemu_put_buffer_async(QEMUFile *f, const uint8_t *buf, size_t size,
 void qemu_put_buffer_async(QEMUFile *f, const uint8_t *buf, size_t size,
                            bool may_free);
                            bool may_free);
-bool qemu_file_mode_is_not_valid(const char *mode);
 
 
 #include "migration/qemu-file-types.h"
 #include "migration/qemu-file-types.h"
 
 

+ 19 - 20
migration/rdma.c

@@ -4053,27 +4053,26 @@ static void qio_channel_rdma_register_types(void)
 
 
 type_init(qio_channel_rdma_register_types);
 type_init(qio_channel_rdma_register_types);
 
 
-static QEMUFile *qemu_fopen_rdma(RDMAContext *rdma, const char *mode)
+static QEMUFile *rdma_new_input(RDMAContext *rdma)
 {
 {
-    QIOChannelRDMA *rioc;
+    QIOChannelRDMA *rioc = QIO_CHANNEL_RDMA(object_new(TYPE_QIO_CHANNEL_RDMA));
 
 
-    if (qemu_file_mode_is_not_valid(mode)) {
-        return NULL;
-    }
+    rioc->file = qemu_file_new_input(QIO_CHANNEL(rioc));
+    rioc->rdmain = rdma;
+    rioc->rdmaout = rdma->return_path;
+    qemu_file_set_hooks(rioc->file, &rdma_read_hooks);
+
+    return rioc->file;
+}
 
 
-    rioc = QIO_CHANNEL_RDMA(object_new(TYPE_QIO_CHANNEL_RDMA));
+static QEMUFile *rdma_new_output(RDMAContext *rdma)
+{
+    QIOChannelRDMA *rioc = QIO_CHANNEL_RDMA(object_new(TYPE_QIO_CHANNEL_RDMA));
 
 
-    if (mode[0] == 'w') {
-        rioc->file = qemu_file_new_output(QIO_CHANNEL(rioc));
-        rioc->rdmaout = rdma;
-        rioc->rdmain = rdma->return_path;
-        qemu_file_set_hooks(rioc->file, &rdma_write_hooks);
-    } else {
-        rioc->file = qemu_file_new_input(QIO_CHANNEL(rioc));
-        rioc->rdmain = rdma;
-        rioc->rdmaout = rdma->return_path;
-        qemu_file_set_hooks(rioc->file, &rdma_read_hooks);
-    }
+    rioc->file = qemu_file_new_output(QIO_CHANNEL(rioc));
+    rioc->rdmaout = rdma;
+    rioc->rdmain = rdma->return_path;
+    qemu_file_set_hooks(rioc->file, &rdma_write_hooks);
 
 
     return rioc->file;
     return rioc->file;
 }
 }
@@ -4099,9 +4098,9 @@ static void rdma_accept_incoming_migration(void *opaque)
         return;
         return;
     }
     }
 
 
-    f = qemu_fopen_rdma(rdma, "rb");
+    f = rdma_new_input(rdma);
     if (f == NULL) {
     if (f == NULL) {
-        fprintf(stderr, "RDMA ERROR: could not qemu_fopen_rdma\n");
+        fprintf(stderr, "RDMA ERROR: could not open RDMA for input\n");
         qemu_rdma_cleanup(rdma);
         qemu_rdma_cleanup(rdma);
         return;
         return;
     }
     }
@@ -4224,7 +4223,7 @@ void rdma_start_outgoing_migration(void *opaque,
 
 
     trace_rdma_start_outgoing_migration_after_rdma_connect();
     trace_rdma_start_outgoing_migration_after_rdma_connect();
 
 
-    s->to_dst_file = qemu_fopen_rdma(rdma, "wb");
+    s->to_dst_file = rdma_new_output(rdma);
     migrate_fd_connect(s, NULL);
     migrate_fd_connect(s, NULL);
     return;
     return;
 return_path_err:
 return_path_err: