diff mbox series

[V6,6/6] net/net.c: Add handler for COLO passthrough connection

Message ID 20210420151537.64360-7-chen.zhang@intel.com
State New
Headers show
Series Passthrough specific network traffic in COLO | expand

Commit Message

Zhang, Chen April 20, 2021, 3:15 p.m. UTC
Use connection protocol,src port,dst port,src ip,dst ip as the key
to bypass certain network traffic in COLO compare.

Signed-off-by: Zhang Chen <chen.zhang@intel.com>
---
 net/net.c | 160 +++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 158 insertions(+), 2 deletions(-)

Comments

Lukas Straub May 17, 2021, 8:38 p.m. UTC | #1
On Tue, 20 Apr 2021 23:15:37 +0800
Zhang Chen <chen.zhang@intel.com> wrote:

> Use connection protocol,src port,dst port,src ip,dst ip as the key
> to bypass certain network traffic in COLO compare.
> 
> Signed-off-by: Zhang Chen <chen.zhang@intel.com>
> ---
>  net/net.c | 160 +++++++++++++++++++++++++++++++++++++++++++++++++++++-
>  1 file changed, 158 insertions(+), 2 deletions(-)
> 
> diff --git a/net/net.c b/net/net.c
> index 2a6e5f3886..9b0de0f332 100644
> --- a/net/net.c
> +++ b/net/net.c
> @@ -56,6 +56,8 @@
>  #include "sysemu/sysemu.h"
>  #include "net/filter.h"
>  #include "qapi/string-output-visitor.h"
> +#include "net/colo-compare.h"
> +#include "qom/object_interfaces.h"
>  
>  /* Net bridge is currently not supported for W32. */
>  #if !defined(_WIN32)
> @@ -1196,14 +1198,168 @@ void qmp_netdev_del(const char *id, Error **errp)
>      }
>  }
>  
> +static CompareState *colo_passthrough_check(IPFlowSpec *spec, Error **errp)
> +{
> +    Object *container;
> +    Object *obj;
> +    CompareState *s;
> +
> +    if (!spec->object_name) {
> +        error_setg(errp, QERR_INVALID_PARAMETER_VALUE, "object-name",
> +                   "Need input colo-compare object name");
> +        return NULL;
> +    }
> +
> +    container = object_get_objects_root();
> +    obj = object_resolve_path_component(container, spec->object_name);
> +    if (!obj) {
> +        error_setg(errp, "colo-compare '%s' not found", spec->object_name);
> +        return NULL;
> +    }
> +
> +    s = COLO_COMPARE(obj);
> +
> +    if (!getprotobyname(spec->protocol)) {
> +        error_setg(errp, "COLO pass through get wrong protocol");
> +        return NULL;
> +    }
> +
> +    if ((spec->source->host && !qemu_isdigit(spec->source->host[0])) ||
> +        (spec->destination->host &&
> +        !qemu_isdigit(spec->destination->host[0]))) {
> +        error_setg(errp, "COLO pass through get wrong IP");
> +        return NULL;
> +    }
> +
> +    if (atoi(spec->source->port) > 65536 || atoi(spec->source->port) < 0 ||
> +        atoi(spec->destination->port) > 65536 ||
> +        atoi(spec->destination->port) < 0) {
> +        error_setg(errp, "COLO pass through get wrong port");
> +        return NULL;
> +    }
> +
> +    return s;
> +}
> +
> +static void compare_passthrough_add(CompareState *s,
> +                                    IPFlowSpec *spec,
> +                                    Error **errp)
> +{
> +    COLOPassthroughEntry *pass = NULL, *next = NULL, *origin = NULL;
> +
> +    pass = g_new0(COLOPassthroughEntry, 1);
> +
> +    pass->l4_protocol = getprotobyname(spec->protocol);
> +    pass->src_port = atoi(spec->source->port);
> +    pass->dst_port = atoi(spec->destination->port);
> +
> +    if (!inet_aton(spec->source->host, &pass->src_ip)) {
> +        pass->src_ip.s_addr = 0;
> +    }
> +
> +    if (!inet_aton(spec->destination->host, &pass->dst_ip)) {
> +        pass->dst_ip.s_addr = 0;
> +    }
> +
> +    qemu_mutex_lock(&s->passthroughlist_mutex);
> +    if (!QLIST_EMPTY(&s->passthroughlist)) {
> +        QLIST_FOREACH_SAFE(origin, &s->passthroughlist, node, next) {
> +            if ((pass->l4_protocol->p_proto == origin->l4_protocol->p_proto) &&
> +                (pass->src_port == origin->src_port) &&
> +                (pass->dst_port == origin->dst_port) &&
> +                (pass->src_ip.s_addr == origin->src_ip.s_addr) &&
> +                (pass->dst_ip.s_addr == origin->dst_ip.s_addr)) {
> +                error_setg(errp, "The pass through connection already exists");
> +                g_free(pass);
> +                qemu_mutex_unlock(&s->passthroughlist_mutex);
> +                return;
> +            }
> +        }
> +    }

I think this searching for a existing passthrough rule should move into
a function. The function can then be used in compare_passthrough_del
too.

> +    QLIST_INSERT_HEAD(&s->passthroughlist, pass, node);
> +    qemu_mutex_unlock(&s->passthroughlist_mutex);
> +}
> +
> +static void compare_passthrough_del(CompareState *s,
> +                                    IPFlowSpec *spec,
> +                                    Error **errp)
> +{
> +    COLOPassthroughEntry *pass = NULL, *next = NULL, *origin = NULL;
> +
> +    pass = g_new0(COLOPassthroughEntry, 1);
> +
> +    pass->l4_protocol = getprotobyname(spec->protocol);
> +    pass->src_port = atoi(spec->source->port);
> +    pass->dst_port = atoi(spec->destination->port);
> +
> +    if (!inet_aton(spec->source->host, &pass->src_ip)) {
> +        pass->src_ip.s_addr = 0;
> +    }
> +
> +    if (!inet_aton(spec->destination->host, &pass->dst_ip)) {
> +        pass->dst_ip.s_addr = 0;
> +    }
> +
> +    qemu_mutex_lock(&s->passthroughlist_mutex);
> +    if (!QLIST_EMPTY(&s->passthroughlist)) {
> +        QLIST_FOREACH_SAFE(origin, &s->passthroughlist, node, next) {
> +            if ((pass->l4_protocol->p_proto == origin->l4_protocol->p_proto) &&
> +                (pass->src_port == origin->src_port) &&
> +                (pass->dst_port == origin->dst_port) &&
> +                (pass->src_ip.s_addr == origin->src_ip.s_addr) &&
> +                (pass->dst_ip.s_addr == origin->dst_ip.s_addr)) {
> +                QLIST_REMOVE(origin, node);
> +                g_free(origin);
> +                g_free(pass);
> +                qemu_mutex_unlock(&s->passthroughlist_mutex);
> +                return;
> +            }
> +        }
> +        error_setg(errp, "The pass through list can't find the connection");
> +    } else {
> +        error_setg(errp, "The pass through connection list is empty");
> +    }
> +
> +    g_free(pass);
> +    qemu_mutex_unlock(&s->passthroughlist_mutex);
> +}
> +
> +
>  void qmp_colo_passthrough_add(IPFlowSpec *spec, Error **errp)
>  {
> -    /* TODO implement setup passthrough rule */
> +    CompareState *s;
> +    Error *err = NULL;
> +
> +    s = colo_passthrough_check(spec, &err);
> +    if (err) {
> +        error_propagate(errp, err);
> +        return;
> +    }
> +
> +    compare_passthrough_add(s, spec, &err);
> +    if (err) {
> +        error_propagate(errp, err);
> +        return;
> +    }
>  }
>  
>  void qmp_colo_passthrough_del(IPFlowSpec *spec, Error **errp)
>  {
> -    /* TODO implement delete passthrough rule */
> +    CompareState *s;
> +    Error *err = NULL;
> +
> +    s = colo_passthrough_check(spec, &err);
> +    if (err) {
> +        error_propagate(errp, err);
> +        return;
> +    }
> +
> +    compare_passthrough_del(s, spec, &err);
> +    if (err) {
> +        error_propagate(errp, err);
> +        return;
> +    }
>  }
>  
>  static void netfilter_print_info(Monitor *mon, NetFilterState *nf)



--
Zhang, Chen May 20, 2021, 5:50 a.m. UTC | #2
> -----Original Message-----
> From: Lukas Straub <lukasstraub2@web.de>
> Sent: Tuesday, May 18, 2021 4:39 AM
> To: Zhang, Chen <chen.zhang@intel.com>
> Cc: Jason Wang <jasowang@redhat.com>; qemu-dev <qemu-
> devel@nongnu.org>; Eric Blake <eblake@redhat.com>; Dr. David Alan
> Gilbert <dgilbert@redhat.com>; Markus Armbruster <armbru@redhat.com>;
> Daniel P. Berrangé <berrange@redhat.com>; Gerd Hoffmann
> <kraxel@redhat.com>; Li Zhijian <lizhijian@cn.fujitsu.com>; Zhang Chen
> <zhangckid@gmail.com>
> Subject: Re: [PATCH V6 6/6] net/net.c: Add handler for COLO passthrough
> connection
> 
> On Tue, 20 Apr 2021 23:15:37 +0800
> Zhang Chen <chen.zhang@intel.com> wrote:
> 
> > Use connection protocol,src port,dst port,src ip,dst ip as the key to
> > bypass certain network traffic in COLO compare.
> >
> > Signed-off-by: Zhang Chen <chen.zhang@intel.com>
> > ---
> >  net/net.c | 160
> > +++++++++++++++++++++++++++++++++++++++++++++++++++++-
> >  1 file changed, 158 insertions(+), 2 deletions(-)
> >
> > diff --git a/net/net.c b/net/net.c
> > index 2a6e5f3886..9b0de0f332 100644
> > --- a/net/net.c
> > +++ b/net/net.c
> > @@ -56,6 +56,8 @@
> >  #include "sysemu/sysemu.h"
> >  #include "net/filter.h"
> >  #include "qapi/string-output-visitor.h"
> > +#include "net/colo-compare.h"
> > +#include "qom/object_interfaces.h"
> >
> >  /* Net bridge is currently not supported for W32. */  #if
> > !defined(_WIN32) @@ -1196,14 +1198,168 @@ void
> qmp_netdev_del(const
> > char *id, Error **errp)
> >      }
> >  }
> >
> > +static CompareState *colo_passthrough_check(IPFlowSpec *spec, Error
> > +**errp) {
> > +    Object *container;
> > +    Object *obj;
> > +    CompareState *s;
> > +
> > +    if (!spec->object_name) {
> > +        error_setg(errp, QERR_INVALID_PARAMETER_VALUE, "object-name",
> > +                   "Need input colo-compare object name");
> > +        return NULL;
> > +    }
> > +
> > +    container = object_get_objects_root();
> > +    obj = object_resolve_path_component(container, spec->object_name);
> > +    if (!obj) {
> > +        error_setg(errp, "colo-compare '%s' not found", spec->object_name);
> > +        return NULL;
> > +    }
> > +
> > +    s = COLO_COMPARE(obj);
> > +
> > +    if (!getprotobyname(spec->protocol)) {
> > +        error_setg(errp, "COLO pass through get wrong protocol");
> > +        return NULL;
> > +    }
> > +
> > +    if ((spec->source->host && !qemu_isdigit(spec->source->host[0])) ||
> > +        (spec->destination->host &&
> > +        !qemu_isdigit(spec->destination->host[0]))) {
> > +        error_setg(errp, "COLO pass through get wrong IP");
> > +        return NULL;
> > +    }
> > +
> > +    if (atoi(spec->source->port) > 65536 || atoi(spec->source->port) < 0 ||
> > +        atoi(spec->destination->port) > 65536 ||
> > +        atoi(spec->destination->port) < 0) {
> > +        error_setg(errp, "COLO pass through get wrong port");
> > +        return NULL;
> > +    }
> > +
> > +    return s;
> > +}
> > +
> > +static void compare_passthrough_add(CompareState *s,
> > +                                    IPFlowSpec *spec,
> > +                                    Error **errp) {
> > +    COLOPassthroughEntry *pass = NULL, *next = NULL, *origin = NULL;
> > +
> > +    pass = g_new0(COLOPassthroughEntry, 1);
> > +
> > +    pass->l4_protocol = getprotobyname(spec->protocol);
> > +    pass->src_port = atoi(spec->source->port);
> > +    pass->dst_port = atoi(spec->destination->port);
> > +
> > +    if (!inet_aton(spec->source->host, &pass->src_ip)) {
> > +        pass->src_ip.s_addr = 0;
> > +    }
> > +
> > +    if (!inet_aton(spec->destination->host, &pass->dst_ip)) {
> > +        pass->dst_ip.s_addr = 0;
> > +    }
> > +
> > +    qemu_mutex_lock(&s->passthroughlist_mutex);
> > +    if (!QLIST_EMPTY(&s->passthroughlist)) {
> > +        QLIST_FOREACH_SAFE(origin, &s->passthroughlist, node, next) {
> > +            if ((pass->l4_protocol->p_proto == origin->l4_protocol->p_proto)
> &&
> > +                (pass->src_port == origin->src_port) &&
> > +                (pass->dst_port == origin->dst_port) &&
> > +                (pass->src_ip.s_addr == origin->src_ip.s_addr) &&
> > +                (pass->dst_ip.s_addr == origin->dst_ip.s_addr)) {
> > +                error_setg(errp, "The pass through connection already exists");
> > +                g_free(pass);
> > +                qemu_mutex_unlock(&s->passthroughlist_mutex);
> > +                return;
> > +            }
> > +        }
> > +    }
> 
> I think this searching for a existing passthrough rule should move into a
> function. The function can then be used in compare_passthrough_del too.

Good idea.
I will change it in next version.

Thanks
Chen

> 
> > +    QLIST_INSERT_HEAD(&s->passthroughlist, pass, node);
> > +    qemu_mutex_unlock(&s->passthroughlist_mutex);
> > +}
> > +
> > +static void compare_passthrough_del(CompareState *s,
> > +                                    IPFlowSpec *spec,
> > +                                    Error **errp) {
> > +    COLOPassthroughEntry *pass = NULL, *next = NULL, *origin = NULL;
> > +
> > +    pass = g_new0(COLOPassthroughEntry, 1);
> > +
> > +    pass->l4_protocol = getprotobyname(spec->protocol);
> > +    pass->src_port = atoi(spec->source->port);
> > +    pass->dst_port = atoi(spec->destination->port);
> > +
> > +    if (!inet_aton(spec->source->host, &pass->src_ip)) {
> > +        pass->src_ip.s_addr = 0;
> > +    }
> > +
> > +    if (!inet_aton(spec->destination->host, &pass->dst_ip)) {
> > +        pass->dst_ip.s_addr = 0;
> > +    }
> > +
> > +    qemu_mutex_lock(&s->passthroughlist_mutex);
> > +    if (!QLIST_EMPTY(&s->passthroughlist)) {
> > +        QLIST_FOREACH_SAFE(origin, &s->passthroughlist, node, next) {
> > +            if ((pass->l4_protocol->p_proto == origin->l4_protocol->p_proto)
> &&
> > +                (pass->src_port == origin->src_port) &&
> > +                (pass->dst_port == origin->dst_port) &&
> > +                (pass->src_ip.s_addr == origin->src_ip.s_addr) &&
> > +                (pass->dst_ip.s_addr == origin->dst_ip.s_addr)) {
> > +                QLIST_REMOVE(origin, node);
> > +                g_free(origin);
> > +                g_free(pass);
> > +                qemu_mutex_unlock(&s->passthroughlist_mutex);
> > +                return;
> > +            }
> > +        }
> > +        error_setg(errp, "The pass through list can't find the connection");
> > +    } else {
> > +        error_setg(errp, "The pass through connection list is empty");
> > +    }
> > +
> > +    g_free(pass);
> > +    qemu_mutex_unlock(&s->passthroughlist_mutex);
> > +}
> > +
> > +
> >  void qmp_colo_passthrough_add(IPFlowSpec *spec, Error **errp)  {
> > -    /* TODO implement setup passthrough rule */
> > +    CompareState *s;
> > +    Error *err = NULL;
> > +
> > +    s = colo_passthrough_check(spec, &err);
> > +    if (err) {
> > +        error_propagate(errp, err);
> > +        return;
> > +    }
> > +
> > +    compare_passthrough_add(s, spec, &err);
> > +    if (err) {
> > +        error_propagate(errp, err);
> > +        return;
> > +    }
> >  }
> >
> >  void qmp_colo_passthrough_del(IPFlowSpec *spec, Error **errp)  {
> > -    /* TODO implement delete passthrough rule */
> > +    CompareState *s;
> > +    Error *err = NULL;
> > +
> > +    s = colo_passthrough_check(spec, &err);
> > +    if (err) {
> > +        error_propagate(errp, err);
> > +        return;
> > +    }
> > +
> > +    compare_passthrough_del(s, spec, &err);
> > +    if (err) {
> > +        error_propagate(errp, err);
> > +        return;
> > +    }
> >  }
> >
> >  static void netfilter_print_info(Monitor *mon, NetFilterState *nf)
> 
> 
> 
> --
diff mbox series

Patch

diff --git a/net/net.c b/net/net.c
index 2a6e5f3886..9b0de0f332 100644
--- a/net/net.c
+++ b/net/net.c
@@ -56,6 +56,8 @@ 
 #include "sysemu/sysemu.h"
 #include "net/filter.h"
 #include "qapi/string-output-visitor.h"
+#include "net/colo-compare.h"
+#include "qom/object_interfaces.h"
 
 /* Net bridge is currently not supported for W32. */
 #if !defined(_WIN32)
@@ -1196,14 +1198,168 @@  void qmp_netdev_del(const char *id, Error **errp)
     }
 }
 
+static CompareState *colo_passthrough_check(IPFlowSpec *spec, Error **errp)
+{
+    Object *container;
+    Object *obj;
+    CompareState *s;
+
+    if (!spec->object_name) {
+        error_setg(errp, QERR_INVALID_PARAMETER_VALUE, "object-name",
+                   "Need input colo-compare object name");
+        return NULL;
+    }
+
+    container = object_get_objects_root();
+    obj = object_resolve_path_component(container, spec->object_name);
+    if (!obj) {
+        error_setg(errp, "colo-compare '%s' not found", spec->object_name);
+        return NULL;
+    }
+
+    s = COLO_COMPARE(obj);
+
+    if (!getprotobyname(spec->protocol)) {
+        error_setg(errp, "COLO pass through get wrong protocol");
+        return NULL;
+    }
+
+    if ((spec->source->host && !qemu_isdigit(spec->source->host[0])) ||
+        (spec->destination->host &&
+        !qemu_isdigit(spec->destination->host[0]))) {
+        error_setg(errp, "COLO pass through get wrong IP");
+        return NULL;
+    }
+
+    if (atoi(spec->source->port) > 65536 || atoi(spec->source->port) < 0 ||
+        atoi(spec->destination->port) > 65536 ||
+        atoi(spec->destination->port) < 0) {
+        error_setg(errp, "COLO pass through get wrong port");
+        return NULL;
+    }
+
+    return s;
+}
+
+static void compare_passthrough_add(CompareState *s,
+                                    IPFlowSpec *spec,
+                                    Error **errp)
+{
+    COLOPassthroughEntry *pass = NULL, *next = NULL, *origin = NULL;
+
+    pass = g_new0(COLOPassthroughEntry, 1);
+
+    pass->l4_protocol = getprotobyname(spec->protocol);
+    pass->src_port = atoi(spec->source->port);
+    pass->dst_port = atoi(spec->destination->port);
+
+    if (!inet_aton(spec->source->host, &pass->src_ip)) {
+        pass->src_ip.s_addr = 0;
+    }
+
+    if (!inet_aton(spec->destination->host, &pass->dst_ip)) {
+        pass->dst_ip.s_addr = 0;
+    }
+
+    qemu_mutex_lock(&s->passthroughlist_mutex);
+    if (!QLIST_EMPTY(&s->passthroughlist)) {
+        QLIST_FOREACH_SAFE(origin, &s->passthroughlist, node, next) {
+            if ((pass->l4_protocol->p_proto == origin->l4_protocol->p_proto) &&
+                (pass->src_port == origin->src_port) &&
+                (pass->dst_port == origin->dst_port) &&
+                (pass->src_ip.s_addr == origin->src_ip.s_addr) &&
+                (pass->dst_ip.s_addr == origin->dst_ip.s_addr)) {
+                error_setg(errp, "The pass through connection already exists");
+                g_free(pass);
+                qemu_mutex_unlock(&s->passthroughlist_mutex);
+                return;
+            }
+        }
+    }
+
+    QLIST_INSERT_HEAD(&s->passthroughlist, pass, node);
+    qemu_mutex_unlock(&s->passthroughlist_mutex);
+}
+
+static void compare_passthrough_del(CompareState *s,
+                                    IPFlowSpec *spec,
+                                    Error **errp)
+{
+    COLOPassthroughEntry *pass = NULL, *next = NULL, *origin = NULL;
+
+    pass = g_new0(COLOPassthroughEntry, 1);
+
+    pass->l4_protocol = getprotobyname(spec->protocol);
+    pass->src_port = atoi(spec->source->port);
+    pass->dst_port = atoi(spec->destination->port);
+
+    if (!inet_aton(spec->source->host, &pass->src_ip)) {
+        pass->src_ip.s_addr = 0;
+    }
+
+    if (!inet_aton(spec->destination->host, &pass->dst_ip)) {
+        pass->dst_ip.s_addr = 0;
+    }
+
+    qemu_mutex_lock(&s->passthroughlist_mutex);
+    if (!QLIST_EMPTY(&s->passthroughlist)) {
+        QLIST_FOREACH_SAFE(origin, &s->passthroughlist, node, next) {
+            if ((pass->l4_protocol->p_proto == origin->l4_protocol->p_proto) &&
+                (pass->src_port == origin->src_port) &&
+                (pass->dst_port == origin->dst_port) &&
+                (pass->src_ip.s_addr == origin->src_ip.s_addr) &&
+                (pass->dst_ip.s_addr == origin->dst_ip.s_addr)) {
+                QLIST_REMOVE(origin, node);
+                g_free(origin);
+                g_free(pass);
+                qemu_mutex_unlock(&s->passthroughlist_mutex);
+                return;
+            }
+        }
+        error_setg(errp, "The pass through list can't find the connection");
+    } else {
+        error_setg(errp, "The pass through connection list is empty");
+    }
+
+    g_free(pass);
+    qemu_mutex_unlock(&s->passthroughlist_mutex);
+}
+
+
 void qmp_colo_passthrough_add(IPFlowSpec *spec, Error **errp)
 {
-    /* TODO implement setup passthrough rule */
+    CompareState *s;
+    Error *err = NULL;
+
+    s = colo_passthrough_check(spec, &err);
+    if (err) {
+        error_propagate(errp, err);
+        return;
+    }
+
+    compare_passthrough_add(s, spec, &err);
+    if (err) {
+        error_propagate(errp, err);
+        return;
+    }
 }
 
 void qmp_colo_passthrough_del(IPFlowSpec *spec, Error **errp)
 {
-    /* TODO implement delete passthrough rule */
+    CompareState *s;
+    Error *err = NULL;
+
+    s = colo_passthrough_check(spec, &err);
+    if (err) {
+        error_propagate(errp, err);
+        return;
+    }
+
+    compare_passthrough_del(s, spec, &err);
+    if (err) {
+        error_propagate(errp, err);
+        return;
+    }
 }
 
 static void netfilter_print_info(Monitor *mon, NetFilterState *nf)