net/colo.c

   1 /*
   2  * COarse-grain LOck-stepping Virtual Machines for Non-stop Service (COLO)
   3  * (a.k.a. Fault Tolerance or Continuous Replication)
   4  *
   5  * Copyright (c) 2016 HUAWEI TECHNOLOGIES CO., LTD.
   6  * Copyright (c) 2016 FUJITSU LIMITED
   7  * Copyright (c) 2016 Intel Corporation
   8  *
   9  * Author: Zhang Chen <[email protected]>
  10  *
  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.
  13  */
  14
  15 #include "qemu/osdep.h"
  16 #include "trace.h"
  17 #include "net/colo.h"
  18
  19 uint32_t connection_key_hash(const void *opaque)
  20 {
  21     const ConnectionKey *key = opaque;
  22     uint32_t a, b, c;
  23
  24     /* Jenkins hash */
  25     a = b = c = JHASH_INITVAL + sizeof(*key);
  26     a += key->src.s_addr;
  27     b += key->dst.s_addr;
  28     c += (key->src_port | key->dst_port << 16);
  29     __jhash_mix(a, b, c);
  30
  31     a += key->ip_proto;
  32     __jhash_final(a, b, c);
  33
  34     return c;
  35 }
  36
  37 int connection_key_equal(const void *key1, const void *key2)
  38 {
  39     return memcmp(key1, key2, sizeof(ConnectionKey)) == 0;
  40 }
  41
  42 int parse_packet_early(Packet *pkt)
  43 {
  44     int network_length;
  45     static const uint8_t vlan[] = {0x81, 0x00};
  46     uint8_t *data = pkt->data;
  47     uint16_t l3_proto;
  48     ssize_t l2hdr_len = eth_get_l2_hdr_length(data);
  49
  50     if (pkt->size < ETH_HLEN) {
  51         trace_colo_proxy_main("pkt->size < ETH_HLEN");
  52         return 1;
  53     }
  54
  55     /*
  56      * TODO: support vlan.
  57      */
  58     if (!memcmp(&data[12], vlan, sizeof(vlan))) {
  59         trace_colo_proxy_main("COLO-proxy don't support vlan");
  60         return 1;
  61     }
  62
  63     pkt->network_header = data + l2hdr_len;
  64
  65     const struct iovec l2vec = {
  66         .iov_base = (void *) data,
  67         .iov_len = l2hdr_len
  68     };
  69     l3_proto = eth_get_l3_proto(&l2vec, 1, l2hdr_len);
  70
  71     if (l3_proto != ETH_P_IP) {
  72         return 1;
  73     }
  74
  75     network_length = pkt->ip->ip_hl * 4;
  76     if (pkt->size < l2hdr_len + network_length) {
  77         trace_colo_proxy_main("pkt->size < network_header + network_length");
  78         return 1;
  79     }
  80     pkt->transport_header = pkt->network_header + network_length;
  81
  82     return 0;
  83 }
  84
  85 void fill_connection_key(Packet *pkt, ConnectionKey *key)
  86 {
  87     uint32_t tmp_ports;
  88
  89     memset(key, 0, sizeof(*key));
  90     key->ip_proto = pkt->ip->ip_p;
  91
  92     switch (key->ip_proto) {
  93     case IPPROTO_TCP:
  94     case IPPROTO_UDP:
  95     case IPPROTO_DCCP:
  96     case IPPROTO_ESP:
  97     case IPPROTO_SCTP:
  98     case IPPROTO_UDPLITE:
  99         tmp_ports = *(uint32_t *)(pkt->transport_header);
 100         key->src = pkt->ip->ip_src;
 101         key->dst = pkt->ip->ip_dst;
 102         key->src_port = ntohs(tmp_ports & 0xffff);
 103         key->dst_port = ntohs(tmp_ports >> 16);
 104         break;
 105     case IPPROTO_AH:
 106         tmp_ports = *(uint32_t *)(pkt->transport_header + 4);
 107         key->src = pkt->ip->ip_src;
 108         key->dst = pkt->ip->ip_dst;
 109         key->src_port = ntohs(tmp_ports & 0xffff);
 110         key->dst_port = ntohs(tmp_ports >> 16);
 111         break;
 112     default:
 113         break;
 114     }
 115 }
 116
 117 void reverse_connection_key(ConnectionKey *key)
 118 {
 119     struct in_addr tmp_ip;
 120     uint16_t tmp_port;
 121
 122     tmp_ip = key->src;
 123     key->src = key->dst;
 124     key->dst = tmp_ip;
 125
 126     tmp_port = key->src_port;
 127     key->src_port = key->dst_port;
 128     key->dst_port = tmp_port;
 129 }
 130
 131 Connection *connection_new(ConnectionKey *key)
 132 {
 133     Connection *conn = g_slice_new(Connection);
 134
 135     conn->ip_proto = key->ip_proto;
 136     conn->processing = false;
 137     conn->offset = 0;
 138     conn->syn_flag = 0;
 139     g_queue_init(&conn->primary_list);
 140     g_queue_init(&conn->secondary_list);
 141
 142     return conn;
 143 }
 144
 145 void connection_destroy(void *opaque)
 146 {
 147     Connection *conn = opaque;
 148
 149     g_queue_foreach(&conn->primary_list, packet_destroy, NULL);
 150     g_queue_clear(&conn->primary_list);
 151     g_queue_foreach(&conn->secondary_list, packet_destroy, NULL);
 152     g_queue_clear(&conn->secondary_list);
 153     g_slice_free(Connection, conn);
 154 }
 155
 156 Packet *packet_new(const void *data, int size)
 157 {
 158     Packet *pkt = g_slice_new(Packet);
 159
 160     pkt->data = g_memdup(data, size);
 161     pkt->size = size;
 162     pkt->creation_ms = qemu_clock_get_ms(QEMU_CLOCK_HOST);
 163
 164     return pkt;
 165 }
 166
 167 void packet_destroy(void *opaque, void *user_data)
 168 {
 169     Packet *pkt = opaque;
 170
 171     g_free(pkt->data);
 172     g_slice_free(Packet, pkt);
 173 }
 174
 175 /*
 176  * Clear hashtable, stop this hash growing really huge
 177  */
 178 void connection_hashtable_reset(GHashTable *connection_track_table)
 179 {
 180     g_hash_table_remove_all(connection_track_table);
 181 }
 182
 183 /* if not found, create a new connection and add to hash table */
 184 Connection *connection_get(GHashTable *connection_track_table,
 185                            ConnectionKey *key,
 186                            GQueue *conn_list)
 187 {
 188     Connection *conn = g_hash_table_lookup(connection_track_table, key);
 189
 190     if (conn == NULL) {
 191         ConnectionKey *new_key = g_memdup(key, sizeof(*key));
 192
 193         conn = connection_new(key);
 194
 195         if (g_hash_table_size(connection_track_table) > HASHTABLE_MAX_SIZE) {
 196             trace_colo_proxy_main("colo proxy connection hashtable full,"
 197                                   " clear it");
 198             connection_hashtable_reset(connection_track_table);
 199             /*
 200              * clear the conn_list
 201              */
 202             while (!g_queue_is_empty(conn_list)) {
 203                 connection_destroy(g_queue_pop_head(conn_list));
 204             }
 205         }
 206
 207         g_hash_table_insert(connection_track_table, new_key, conn);
 208     }
 209
 210     return conn;
 211 }