yeahuh / rpms / qemu-kvm

Forked from rpms/qemu-kvm 2 years ago
Clone
76daa3
From 41adc9eecf696d607037b330b02991e7a80d8321 Mon Sep 17 00:00:00 2001
76daa3
From: Eric Blake <eblake@redhat.com>
76daa3
Date: Fri, 9 Jun 2017 21:56:08 +0200
76daa3
Subject: [PATCH 03/13] nbd: Fix regression on resiliency to port scan
76daa3
76daa3
RH-Author: Eric Blake <eblake@redhat.com>
76daa3
Message-id: <20170609215608.27053-3-eblake@redhat.com>
76daa3
Patchwork-id: 75569
76daa3
O-Subject: [RHEV-7.4 qemu-kvm-rhev PATCH 2/2] nbd: Fix regression on resiliency to port scan
76daa3
Bugzilla: 1447948
76daa3
RH-Acked-by: Paolo Bonzini <pbonzini@redhat.com>
76daa3
RH-Acked-by: Thomas Huth <thuth@redhat.com>
76daa3
RH-Acked-by: Laszlo Ersek <lersek@redhat.com>
76daa3
76daa3
Back in qemu 2.5, qemu-nbd was immune to port probes (a transient
76daa3
server would not quit, regardless of how many probe connections
76daa3
came and went, until a connection actually negotiated).  But we
76daa3
broke that in commit ee7d7aa when removing the return value to
76daa3
nbd_client_new(), although that patch also introduced a bug causing
76daa3
an assertion failure on a client that fails negotiation.  We then
76daa3
made it worse during refactoring in commit 1a6245a (a segfault
76daa3
before we could even assert); the (masked) assertion was cleaned
76daa3
up in d3780c2 (still in 2.6), and just recently we finally fixed
76daa3
the segfault ("nbd: Fully intialize client in case of failed
76daa3
negotiation").  But that still means that ever since we added
76daa3
TLS support to qemu-nbd, we have been vulnerable to an ill-timed
76daa3
port-scan being able to cause a denial of service by taking down
76daa3
qemu-nbd before a real client has a chance to connect.
76daa3
76daa3
Since negotiation is now handled asynchronously via coroutines,
76daa3
we no longer have a synchronous point of return by re-adding a
76daa3
return value to nbd_client_new().  So this patch instead wires
76daa3
things up to pass the negotiation status through the close_fn
76daa3
callback function.
76daa3
76daa3
Simple test across two terminals:
76daa3
$ qemu-nbd -f raw -p 30001 file
76daa3
$ nmap 127.0.0.1 -p 30001 && \
76daa3
  qemu-io -c 'r 0 512' -f raw nbd://localhost:30001
76daa3
76daa3
Note that this patch does not change what constitutes successful
76daa3
negotiation (thus, a client must enter transmission phase before
76daa3
that client can be considered as a reason to terminate the server
76daa3
when the connection ends).  Perhaps we may want to tweak things
76daa3
in a later patch to also treat a client that uses NBD_OPT_ABORT
76daa3
as being a 'successful' negotiation (the client correctly talked
76daa3
the NBD protocol, and informed us it was not going to use our
76daa3
export after all), but that's a discussion for another day.
76daa3
76daa3
Fixes: https://bugzilla.redhat.com/show_bug.cgi?id=1451614
76daa3
76daa3
Signed-off-by: Eric Blake <eblake@redhat.com>
76daa3
Message-Id: <20170608222617.20376-1-eblake@redhat.com>
76daa3
(cherry picked from commit ???)
76daa3
https://bugzilla.redhat.com/show_bug.cgi?id=1447948
76daa3
Signed-off-by: Miroslav Rezanina <mrezanin@redhat.com>
76daa3
---
76daa3
 blockdev-nbd.c      |  6 +++++-
76daa3
 include/block/nbd.h |  2 +-
76daa3
 nbd/server.c        | 24 +++++++++++++++---------
76daa3
 qemu-nbd.c          |  4 ++--
76daa3
 4 files changed, 23 insertions(+), 13 deletions(-)
76daa3
76daa3
diff --git a/blockdev-nbd.c b/blockdev-nbd.c
76daa3
index 8a11807..8d7284a 100644
76daa3
--- a/blockdev-nbd.c
76daa3
+++ b/blockdev-nbd.c
76daa3
@@ -27,6 +27,10 @@ typedef struct NBDServerData {
76daa3
 
76daa3
 static NBDServerData *nbd_server;
76daa3
 
76daa3
+static void nbd_blockdev_client_closed(NBDClient *client, bool ignored)
76daa3
+{
76daa3
+    nbd_client_put(client);
76daa3
+}
76daa3
 
76daa3
 static gboolean nbd_accept(QIOChannel *ioc, GIOCondition condition,
76daa3
                            gpointer opaque)
76daa3
@@ -46,7 +50,7 @@ static gboolean nbd_accept(QIOChannel *ioc, GIOCondition condition,
76daa3
     qio_channel_set_name(QIO_CHANNEL(cioc), "nbd-server");
76daa3
     nbd_client_new(NULL, cioc,
76daa3
                    nbd_server->tlscreds, NULL,
76daa3
-                   nbd_client_put);
76daa3
+                   nbd_blockdev_client_closed);
76daa3
     object_unref(OBJECT(cioc));
76daa3
     return TRUE;
76daa3
 }
76daa3
diff --git a/include/block/nbd.h b/include/block/nbd.h
76daa3
index 3e373f0..b69c30d 100644
76daa3
--- a/include/block/nbd.h
76daa3
+++ b/include/block/nbd.h
76daa3
@@ -160,7 +160,7 @@ void nbd_client_new(NBDExport *exp,
76daa3
                     QIOChannelSocket *sioc,
76daa3
                     QCryptoTLSCreds *tlscreds,
76daa3
                     const char *tlsaclname,
76daa3
-                    void (*close)(NBDClient *));
76daa3
+                    void (*close_fn)(NBDClient *, bool));
76daa3
 void nbd_client_get(NBDClient *client);
76daa3
 void nbd_client_put(NBDClient *client);
76daa3
 
76daa3
diff --git a/nbd/server.c b/nbd/server.c
76daa3
index edfda84..a98bb21 100644
76daa3
--- a/nbd/server.c
76daa3
+++ b/nbd/server.c
76daa3
@@ -81,7 +81,7 @@ static QTAILQ_HEAD(, NBDExport) exports = QTAILQ_HEAD_INITIALIZER(exports);
76daa3
 
76daa3
 struct NBDClient {
76daa3
     int refcount;
76daa3
-    void (*close)(NBDClient *client);
76daa3
+    void (*close_fn)(NBDClient *client, bool negotiated);
76daa3
 
76daa3
     bool no_zeroes;
76daa3
     NBDExport *exp;
76daa3
@@ -796,7 +796,7 @@ void nbd_client_put(NBDClient *client)
76daa3
     }
76daa3
 }
76daa3
 
76daa3
-static void client_close(NBDClient *client)
76daa3
+static void client_close(NBDClient *client, bool negotiated)
76daa3
 {
76daa3
     if (client->closing) {
76daa3
         return;
76daa3
@@ -811,8 +811,8 @@ static void client_close(NBDClient *client)
76daa3
                          NULL);
76daa3
 
76daa3
     /* Also tell the client, so that they release their reference.  */
76daa3
-    if (client->close) {
76daa3
-        client->close(client);
76daa3
+    if (client->close_fn) {
76daa3
+        client->close_fn(client, negotiated);
76daa3
     }
76daa3
 }
76daa3
 
76daa3
@@ -993,7 +993,7 @@ void nbd_export_close(NBDExport *exp)
76daa3
 
76daa3
     nbd_export_get(exp);
76daa3
     QTAILQ_FOREACH_SAFE(client, &exp->clients, next, next) {
76daa3
-        client_close(client);
76daa3
+        client_close(client, true);
76daa3
     }
76daa3
     nbd_export_set_name(exp, NULL);
76daa3
     nbd_export_set_description(exp, NULL);
76daa3
@@ -1355,7 +1355,7 @@ done:
76daa3
 
76daa3
 out:
76daa3
     nbd_request_put(req);
76daa3
-    client_close(client);
76daa3
+    client_close(client, true);
76daa3
     nbd_client_put(client);
76daa3
 }
76daa3
 
76daa3
@@ -1381,7 +1381,7 @@ static coroutine_fn void nbd_co_client_start(void *opaque)
76daa3
     qemu_co_mutex_init(&client->send_lock);
76daa3
 
76daa3
     if (nbd_negotiate(data)) {
76daa3
-        client_close(client);
76daa3
+        client_close(client, false);
76daa3
         goto out;
76daa3
     }
76daa3
 
76daa3
@@ -1391,11 +1391,17 @@ out:
76daa3
     g_free(data);
76daa3
 }
76daa3
 
76daa3
+/*
76daa3
+ * Create a new client listener on the given export @exp, using the
76daa3
+ * given channel @sioc.  Begin servicing it in a coroutine.  When the
76daa3
+ * connection closes, call @close_fn with an indication of whether the
76daa3
+ * client completed negotiation.
76daa3
+ */
76daa3
 void nbd_client_new(NBDExport *exp,
76daa3
                     QIOChannelSocket *sioc,
76daa3
                     QCryptoTLSCreds *tlscreds,
76daa3
                     const char *tlsaclname,
76daa3
-                    void (*close_fn)(NBDClient *))
76daa3
+                    void (*close_fn)(NBDClient *, bool))
76daa3
 {
76daa3
     NBDClient *client;
76daa3
     NBDClientNewData *data = g_new(NBDClientNewData, 1);
76daa3
@@ -1412,7 +1418,7 @@ void nbd_client_new(NBDExport *exp,
76daa3
     object_ref(OBJECT(client->sioc));
76daa3
     client->ioc = QIO_CHANNEL(sioc);
76daa3
     object_ref(OBJECT(client->ioc));
76daa3
-    client->close = close_fn;
76daa3
+    client->close_fn = close_fn;
76daa3
 
76daa3
     data->client = client;
76daa3
     data->co = qemu_coroutine_create(nbd_co_client_start, data);
76daa3
diff --git a/qemu-nbd.c b/qemu-nbd.c
76daa3
index b44764e..483dd77 100644
76daa3
--- a/qemu-nbd.c
76daa3
+++ b/qemu-nbd.c
76daa3
@@ -335,10 +335,10 @@ static void nbd_export_closed(NBDExport *exp)
76daa3
 
76daa3
 static void nbd_update_server_watch(void);
76daa3
 
76daa3
-static void nbd_client_closed(NBDClient *client)
76daa3
+static void nbd_client_closed(NBDClient *client, bool negotiated)
76daa3
 {
76daa3
     nb_fds--;
76daa3
-    if (nb_fds == 0 && !persistent && state == RUNNING) {
76daa3
+    if (negotiated && nb_fds == 0 && !persistent && state == RUNNING) {
76daa3
         state = TERMINATE;
76daa3
     }
76daa3
     nbd_update_server_watch();
76daa3
-- 
76daa3
1.8.3.1
76daa3