2 * COarse-grain LOck-stepping Virtual Machines for Non-stop Service (COLO)
3 * (a.k.a. Fault Tolerance or Continuous Replication)
5 * Copyright (c) 2016 HUAWEI TECHNOLOGIES CO., LTD.
6 * Copyright (c) 2016 FUJITSU LIMITED
7 * Copyright (c) 2016 Intel Corporation
9 * Author: Zhang Chen <zhangchen.fnst@cn.fujitsu.com>
11 * This work is licensed under the terms of the GNU GPL, version 2 or
12 * later. See the COPYING file in the top-level directory.
15 #include "qemu/osdep.h"
20 uint32_t connection_key_hash(const void *opaque)
22 const ConnectionKey *key = opaque;
26 a = b = c = JHASH_INITVAL + sizeof(*key);
29 c += (key->src_port | key->dst_port << 16);
33 __jhash_final(a, b, c);
38 int connection_key_equal(const void *key1, const void *key2)
40 return memcmp(key1, key2, sizeof(ConnectionKey)) == 0;
43 int parse_packet_early(Packet *pkt)
46 static const uint8_t vlan[] = {0x81, 0x00};
47 uint8_t *data = pkt->data + pkt->vnet_hdr_len;
49 ssize_t l2hdr_len = eth_get_l2_hdr_length(data);
51 if (pkt->size < ETH_HLEN + pkt->vnet_hdr_len) {
52 trace_colo_proxy_main("pkt->size < ETH_HLEN");
59 if (!memcmp(&data[12], vlan, sizeof(vlan))) {
60 trace_colo_proxy_main("COLO-proxy don't support vlan");
64 pkt->network_header = data + l2hdr_len;
66 const struct iovec l2vec = {
67 .iov_base = (void *) data,
70 l3_proto = eth_get_l3_proto(&l2vec, 1, l2hdr_len);
72 if (l3_proto != ETH_P_IP) {
76 network_length = pkt->ip->ip_hl * 4;
77 if (pkt->size < l2hdr_len + network_length + pkt->vnet_hdr_len) {
78 trace_colo_proxy_main("pkt->size < network_header + network_length");
81 pkt->transport_header = pkt->network_header + network_length;
86 void extract_ip_and_port(uint32_t tmp_ports, ConnectionKey *key,
87 Packet *pkt, bool reverse)
90 key->src = pkt->ip->ip_dst;
91 key->dst = pkt->ip->ip_src;
92 key->src_port = ntohs(tmp_ports & 0xffff);
93 key->dst_port = ntohs(tmp_ports >> 16);
95 key->src = pkt->ip->ip_src;
96 key->dst = pkt->ip->ip_dst;
97 key->src_port = ntohs(tmp_ports >> 16);
98 key->dst_port = ntohs(tmp_ports & 0xffff);
102 void fill_connection_key(Packet *pkt, ConnectionKey *key, bool reverse)
104 uint32_t tmp_ports = 0;
106 key->ip_proto = pkt->ip->ip_p;
108 switch (key->ip_proto) {
114 case IPPROTO_UDPLITE:
115 tmp_ports = *(uint32_t *)(pkt->transport_header);
118 tmp_ports = *(uint32_t *)(pkt->transport_header + 4);
124 extract_ip_and_port(tmp_ports, key, pkt, reverse);
127 Connection *connection_new(ConnectionKey *key)
129 Connection *conn = g_slice_new0(Connection);
131 conn->ip_proto = key->ip_proto;
132 conn->processing = false;
133 conn->tcp_state = TCPS_CLOSED;
134 g_queue_init(&conn->primary_list);
135 g_queue_init(&conn->secondary_list);
140 void connection_destroy(void *opaque)
142 Connection *conn = opaque;
144 g_queue_foreach(&conn->primary_list, packet_destroy, NULL);
145 g_queue_clear(&conn->primary_list);
146 g_queue_foreach(&conn->secondary_list, packet_destroy, NULL);
147 g_queue_clear(&conn->secondary_list);
148 g_slice_free(Connection, conn);
151 Packet *packet_new(const void *data, int size, int vnet_hdr_len)
153 Packet *pkt = g_slice_new0(Packet);
155 pkt->data = g_memdup(data, size);
157 pkt->creation_ms = qemu_clock_get_ms(QEMU_CLOCK_HOST);
158 pkt->vnet_hdr_len = vnet_hdr_len;
164 * packet_new_nocopy will not copy data, so the caller can't release
165 * the data. And it will be released in packet_destroy.
167 Packet *packet_new_nocopy(void *data, int size, int vnet_hdr_len)
169 Packet *pkt = g_slice_new0(Packet);
173 pkt->creation_ms = qemu_clock_get_ms(QEMU_CLOCK_HOST);
174 pkt->vnet_hdr_len = vnet_hdr_len;
179 void packet_destroy(void *opaque, void *user_data)
181 Packet *pkt = opaque;
184 g_slice_free(Packet, pkt);
187 void packet_destroy_partial(void *opaque, void *user_data)
189 Packet *pkt = opaque;
191 g_slice_free(Packet, pkt);
195 * Clear hashtable, stop this hash growing really huge
197 void connection_hashtable_reset(GHashTable *connection_track_table)
199 g_hash_table_remove_all(connection_track_table);
202 /* if not found, create a new connection and add to hash table */
203 Connection *connection_get(GHashTable *connection_track_table,
207 Connection *conn = g_hash_table_lookup(connection_track_table, key);
210 ConnectionKey *new_key = g_memdup(key, sizeof(*key));
212 conn = connection_new(key);
214 if (g_hash_table_size(connection_track_table) > HASHTABLE_MAX_SIZE) {
215 trace_colo_proxy_main("colo proxy connection hashtable full,"
217 connection_hashtable_reset(connection_track_table);
219 * clear the conn_list
221 while (!g_queue_is_empty(conn_list)) {
222 connection_destroy(g_queue_pop_head(conn_list));
226 g_hash_table_insert(connection_track_table, new_key, conn);
232 bool connection_has_tracked(GHashTable *connection_track_table,
235 Connection *conn = g_hash_table_lookup(connection_track_table, key);
237 return conn ? true : false;