[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]
[Qemu-devel] [PATCH V14 09/12] filter-rewriter: track connection and par
From: |
Zhang Chen |
Subject: |
[Qemu-devel] [PATCH V14 09/12] filter-rewriter: track connection and parse packet |
Date: |
Thu, 8 Sep 2016 17:46:11 +0800 |
We use net/colo.h to track connection and parse packet
Signed-off-by: Zhang Chen <address@hidden>
Signed-off-by: Li Zhijian <address@hidden>
Signed-off-by: Wen Congyang <address@hidden>
---
net/colo.c | 14 ++++++++++++++
net/colo.h | 1 +
net/filter-rewriter.c | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
3 files changed, 65 insertions(+)
diff --git a/net/colo.c b/net/colo.c
index e517521..dc4e4a4 100644
--- a/net/colo.c
+++ b/net/colo.c
@@ -115,6 +115,20 @@ void fill_connection_key(Packet *pkt, ConnectionKey *key)
}
}
+void reverse_connection_key(ConnectionKey *key)
+{
+ struct in_addr tmp_ip;
+ uint16_t tmp_port;
+
+ tmp_ip = key->src;
+ key->src = key->dst;
+ key->dst = tmp_ip;
+
+ tmp_port = key->src_port;
+ key->src_port = key->dst_port;
+ key->dst_port = tmp_port;
+}
+
Connection *connection_new(ConnectionKey *key)
{
Connection *conn = g_slice_new(Connection);
diff --git a/net/colo.h b/net/colo.h
index 9a7d5e0..6720a3a 100644
--- a/net/colo.h
+++ b/net/colo.h
@@ -68,6 +68,7 @@ uint32_t connection_key_hash(const void *opaque);
int connection_key_equal(const void *opaque1, const void *opaque2);
int parse_packet_early(Packet *pkt);
void fill_connection_key(Packet *pkt, ConnectionKey *key);
+void reverse_connection_key(ConnectionKey *key);
Connection *connection_new(ConnectionKey *key);
void connection_destroy(void *opaque);
Connection *connection_get(GHashTable *connection_track_table,
diff --git a/net/filter-rewriter.c b/net/filter-rewriter.c
index de29f07..1d49d04 100644
--- a/net/filter-rewriter.c
+++ b/net/filter-rewriter.c
@@ -44,6 +44,20 @@ static void filter_rewriter_flush(NetFilterState *nf)
}
}
+/*
+ * Return 1 on success, if return 0 means the pkt
+ * is not TCP packet
+ */
+static int is_tcp_packet(Packet *pkt)
+{
+ if (!parse_packet_early(pkt) &&
+ pkt->ip->ip_p == IPPROTO_TCP) {
+ return 1;
+ } else {
+ return 0;
+ }
+}
+
static ssize_t colo_rewriter_receive_iov(NetFilterState *nf,
NetClientState *sender,
unsigned flags,
@@ -51,11 +65,47 @@ static ssize_t colo_rewriter_receive_iov(NetFilterState *nf,
int iovcnt,
NetPacketSent *sent_cb)
{
+ RewriterState *s = FILTER_COLO_REWRITER(nf);
+ Connection *conn;
+ ConnectionKey key = { 0 };
+ Packet *pkt;
+ ssize_t size = iov_size(iov, iovcnt);
+ char *buf = g_malloc0(size);
+
+ iov_to_buf(iov, iovcnt, 0, buf, size);
+ pkt = packet_new(buf, size);
+
/*
* if we get tcp packet
* we will rewrite it to make secondary guest's
* connection established successfully
*/
+ if (pkt && is_tcp_packet(pkt)) {
+
+ fill_connection_key(pkt, &key);
+
+ if (sender == nf->netdev) {
+ /*
+ * We need make tcp TX and RX packet
+ * into one connection.
+ */
+ reverse_connection_key(&key);
+ }
+ conn = connection_get(s->connection_track_table,
+ &key,
+ NULL);
+
+ if (sender == nf->netdev) {
+ /* NET_FILTER_DIRECTION_TX */
+ /* handle_primary_tcp_pkt */
+ } else {
+ /* NET_FILTER_DIRECTION_RX */
+ /* handle_secondary_tcp_pkt */
+ }
+ }
+
+ packet_destroy(pkt, NULL);
+ pkt = NULL;
return 0;
}
--
2.7.4
- [Qemu-devel] [PATCH V14 00/12] Introduce COLO-Proxy, Zhang Chen, 2016/09/08
- [Qemu-devel] [PATCH V14 05/12] colo-compare: track connection and enqueue packet, Zhang Chen, 2016/09/08
- [Qemu-devel] [PATCH V14 04/12] Jhash: add linux kernel jhashtable in qemu, Zhang Chen, 2016/09/08
- [Qemu-devel] [PATCH V14 01/12] qemu-char: Add qemu_chr_add_handlers_full() for GMaincontext, Zhang Chen, 2016/09/08
- [Qemu-devel] [PATCH V14 06/12] colo-compare: introduce packet comparison thread, Zhang Chen, 2016/09/08
- [Qemu-devel] [PATCH V14 03/12] net/colo.c: add colo.c to define and handle packet, Zhang Chen, 2016/09/08
- [Qemu-devel] [PATCH V14 02/12] colo-compare: introduce colo compare initialization, Zhang Chen, 2016/09/08
- [Qemu-devel] [PATCH V14 07/12] colo-compare: add TCP, UDP, ICMP packet comparison, Zhang Chen, 2016/09/08
- [Qemu-devel] [PATCH V14 09/12] filter-rewriter: track connection and parse packet,
Zhang Chen <=
- [Qemu-devel] [PATCH V14 08/12] filter-rewriter: introduce filter-rewriter initialization, Zhang Chen, 2016/09/08
- [Qemu-devel] [PATCH V14 10/12] filter-rewriter: rewrite tcp packet to keep secondary connection, Zhang Chen, 2016/09/08
- [Qemu-devel] [PATCH V14 12/12] docs: Add documentation for COLO-proxy, Zhang Chen, 2016/09/08
- [Qemu-devel] [PATCH V14 11/12] MAINTAINERS: add maintainer for COLO-proxy, Zhang Chen, 2016/09/08
- Re: [Qemu-devel] [PATCH V14 00/12] Introduce COLO-Proxy, Jason Wang, 2016/09/22