drivers/net/xen-netback/hash.c

   1 /*
   2  * Copyright (c) 2016 Citrix Systems Inc.
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License version 2
   6  * as published by the Free Softare Foundation; or, when distributed
   7  * separately from the Linux kernel or incorporated into other
   8  * software packages, subject to the following license:
   9  *
  10  * Permission is hereby granted, free of charge, to any person obtaining a copy
  11  * of this source file (the "Software"), to deal in the Software without
  12  * restriction, including without limitation the rights to use, copy, modify,
  13  * merge, publish, distribute, sublicense, and/or sell copies of the Software,
  14  * and to permit persons to whom the Software is furnished to do so, subject to
  15  * the following conditions:
  16  *
  17  * The above copyright notice and this permission notice shall be included in
  18  * all copies or substantial portions of the Software.
  19  *
  20  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  21  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  22  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  23  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  24  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  25  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  26  * IN THE SOFTWARE.
  27  */
  28
  29 #define XEN_NETIF_DEFINE_TOEPLITZ
  30
  31 #include "common.h"
  32 #include <linux/vmalloc.h>
  33 #include <linux/rculist.h>
  34
  35 static void xenvif_add_hash(struct xenvif *vif, const u8 *tag,
  36                             unsigned int len, u32 val)
  37 {
  38         struct xenvif_hash_cache_entry *new, *entry, *oldest;
  39         unsigned long flags;
  40         bool found;
  41
  42         new = kmalloc(sizeof(*entry), GFP_ATOMIC);
  43         if (!new)
  44                 return;
  45
  46         memcpy(new->tag, tag, len);
  47         new->len = len;
  48         new->val = val;
  49
  50         spin_lock_irqsave(&vif->hash.cache.lock, flags);
  51
  52         found = false;
  53         oldest = NULL;
  54         list_for_each_entry_rcu(entry, &vif->hash.cache.list, link,
  55                                 lockdep_is_held(&vif->hash.cache.lock)) {
  56                 /* Make sure we don't add duplicate entries */
  57                 if (entry->len == len &&
  58                     memcmp(entry->tag, tag, len) == 0)
  59                         found = true;
  60                 if (!oldest || entry->seq < oldest->seq)
  61                         oldest = entry;
  62         }
  63
  64         if (!found) {
  65                 new->seq = atomic_inc_return(&vif->hash.cache.seq);
  66                 list_add_rcu(&new->link, &vif->hash.cache.list);
  67
  68                 if (++vif->hash.cache.count > xenvif_hash_cache_size) {
  69                         list_del_rcu(&oldest->link);
  70                         vif->hash.cache.count--;
  71                         kfree_rcu(oldest, rcu);
  72                 }
  73         }
  74
  75         spin_unlock_irqrestore(&vif->hash.cache.lock, flags);
  76
  77         if (found)
  78                 kfree(new);
  79 }
  80
  81 static u32 xenvif_new_hash(struct xenvif *vif, const u8 *data,
  82                            unsigned int len)
  83 {
  84         u32 val;
  85
  86         val = xen_netif_toeplitz_hash(vif->hash.key,
  87                                       sizeof(vif->hash.key),
  88                                       data, len);
  89
  90         if (xenvif_hash_cache_size != 0)
  91                 xenvif_add_hash(vif, data, len, val);
  92
  93         return val;
  94 }
  95
  96 static void xenvif_flush_hash(struct xenvif *vif)
  97 {
  98         struct xenvif_hash_cache_entry *entry;
  99         unsigned long flags;
 100
 101         if (xenvif_hash_cache_size == 0)
 102                 return;
 103
 104         spin_lock_irqsave(&vif->hash.cache.lock, flags);
 105
 106         list_for_each_entry_rcu(entry, &vif->hash.cache.list, link,
 107                                 lockdep_is_held(&vif->hash.cache.lock)) {
 108                 list_del_rcu(&entry->link);
 109                 vif->hash.cache.count--;
 110                 kfree_rcu(entry, rcu);
 111         }
 112
 113         spin_unlock_irqrestore(&vif->hash.cache.lock, flags);
 114 }
 115
 116 static u32 xenvif_find_hash(struct xenvif *vif, const u8 *data,
 117                             unsigned int len)
 118 {
 119         struct xenvif_hash_cache_entry *entry;
 120         u32 val;
 121         bool found;
 122
 123         if (len >= XEN_NETBK_HASH_TAG_SIZE)
 124                 return 0;
 125
 126         if (xenvif_hash_cache_size == 0)
 127                 return xenvif_new_hash(vif, data, len);
 128
 129         rcu_read_lock();
 130
 131         found = false;
 132
 133         list_for_each_entry_rcu(entry, &vif->hash.cache.list, link) {
 134                 if (entry->len == len &&
 135                     memcmp(entry->tag, data, len) == 0) {
 136                         val = entry->val;
 137                         entry->seq = atomic_inc_return(&vif->hash.cache.seq);
 138                         found = true;
 139                         break;
 140                 }
 141         }
 142
 143         rcu_read_unlock();
 144
 145         if (!found)
 146                 val = xenvif_new_hash(vif, data, len);
 147
 148         return val;
 149 }
 150
 151 void xenvif_set_skb_hash(struct xenvif *vif, struct sk_buff *skb)
 152 {
 153         struct flow_keys flow;
 154         u32 hash = 0;
 155         enum pkt_hash_types type = PKT_HASH_TYPE_NONE;
 156         u32 flags = vif->hash.flags;
 157         bool has_tcp_hdr;
 158
 159         /* Quick rejection test: If the network protocol doesn't
 160          * correspond to any enabled hash type then there's no point
 161          * in parsing the packet header.
 162          */
 163         switch (skb->protocol) {
 164         case htons(ETH_P_IP):
 165                 if (flags & (XEN_NETIF_CTRL_HASH_TYPE_IPV4_TCP |
 166                              XEN_NETIF_CTRL_HASH_TYPE_IPV4))
 167                         break;
 168
 169                 goto done;
 170
 171         case htons(ETH_P_IPV6):
 172                 if (flags & (XEN_NETIF_CTRL_HASH_TYPE_IPV6_TCP |
 173                              XEN_NETIF_CTRL_HASH_TYPE_IPV6))
 174                         break;
 175
 176                 goto done;
 177
 178         default:
 179                 goto done;
 180         }
 181
 182         memset(&flow, 0, sizeof(flow));
 183         if (!skb_flow_dissect_flow_keys(skb, &flow, 0))
 184                 goto done;
 185
 186         has_tcp_hdr = (flow.basic.ip_proto == IPPROTO_TCP) &&
 187                       !(flow.control.flags & FLOW_DIS_IS_FRAGMENT);
 188
 189         switch (skb->protocol) {
 190         case htons(ETH_P_IP):
 191                 if (has_tcp_hdr &&
 192                     (flags & XEN_NETIF_CTRL_HASH_TYPE_IPV4_TCP)) {
 193                         u8 data[12];
 194
 195                         memcpy(&data[0], &flow.addrs.v4addrs.src, 4);
 196                         memcpy(&data[4], &flow.addrs.v4addrs.dst, 4);
 197                         memcpy(&data[8], &flow.ports.src, 2);
 198                         memcpy(&data[10], &flow.ports.dst, 2);
 199
 200                         hash = xenvif_find_hash(vif, data, sizeof(data));
 201                         type = PKT_HASH_TYPE_L4;
 202                 } else if (flags & XEN_NETIF_CTRL_HASH_TYPE_IPV4) {
 203                         u8 data[8];
 204
 205                         memcpy(&data[0], &flow.addrs.v4addrs.src, 4);
 206                         memcpy(&data[4], &flow.addrs.v4addrs.dst, 4);
 207
 208                         hash = xenvif_find_hash(vif, data, sizeof(data));
 209                         type = PKT_HASH_TYPE_L3;
 210                 }
 211
 212                 break;
 213
 214         case htons(ETH_P_IPV6):
 215                 if (has_tcp_hdr &&
 216                     (flags & XEN_NETIF_CTRL_HASH_TYPE_IPV6_TCP)) {
 217                         u8 data[36];
 218
 219                         memcpy(&data[0], &flow.addrs.v6addrs.src, 16);
 220                         memcpy(&data[16], &flow.addrs.v6addrs.dst, 16);
 221                         memcpy(&data[32], &flow.ports.src, 2);
 222                         memcpy(&data[34], &flow.ports.dst, 2);
 223
 224                         hash = xenvif_find_hash(vif, data, sizeof(data));
 225                         type = PKT_HASH_TYPE_L4;
 226                 } else if (flags & XEN_NETIF_CTRL_HASH_TYPE_IPV6) {
 227                         u8 data[32];
 228
 229                         memcpy(&data[0], &flow.addrs.v6addrs.src, 16);
 230                         memcpy(&data[16], &flow.addrs.v6addrs.dst, 16);
 231
 232                         hash = xenvif_find_hash(vif, data, sizeof(data));
 233                         type = PKT_HASH_TYPE_L3;
 234                 }
 235
 236                 break;
 237         }
 238
 239 done:
 240         if (type == PKT_HASH_TYPE_NONE)
 241                 skb_clear_hash(skb);
 242         else
 243                 __skb_set_sw_hash(skb, hash, type == PKT_HASH_TYPE_L4);
 244 }
 245
 246 u32 xenvif_set_hash_alg(struct xenvif *vif, u32 alg)
 247 {
 248         switch (alg) {
 249         case XEN_NETIF_CTRL_HASH_ALGORITHM_NONE:
 250         case XEN_NETIF_CTRL_HASH_ALGORITHM_TOEPLITZ:
 251                 break;
 252
 253         default:
 254                 return XEN_NETIF_CTRL_STATUS_INVALID_PARAMETER;
 255         }
 256
 257         vif->hash.alg = alg;
 258
 259         return XEN_NETIF_CTRL_STATUS_SUCCESS;
 260 }
 261
 262 u32 xenvif_get_hash_flags(struct xenvif *vif, u32 *flags)
 263 {
 264         if (vif->hash.alg == XEN_NETIF_CTRL_HASH_ALGORITHM_NONE)
 265                 return XEN_NETIF_CTRL_STATUS_NOT_SUPPORTED;
 266
 267         *flags = XEN_NETIF_CTRL_HASH_TYPE_IPV4 |
 268                  XEN_NETIF_CTRL_HASH_TYPE_IPV4_TCP |
 269                  XEN_NETIF_CTRL_HASH_TYPE_IPV6 |
 270                  XEN_NETIF_CTRL_HASH_TYPE_IPV6_TCP;
 271
 272         return XEN_NETIF_CTRL_STATUS_SUCCESS;
 273 }
 274
 275 u32 xenvif_set_hash_flags(struct xenvif *vif, u32 flags)
 276 {
 277         if (flags & ~(XEN_NETIF_CTRL_HASH_TYPE_IPV4 |
 278                       XEN_NETIF_CTRL_HASH_TYPE_IPV4_TCP |
 279                       XEN_NETIF_CTRL_HASH_TYPE_IPV6 |
 280                       XEN_NETIF_CTRL_HASH_TYPE_IPV6_TCP))
 281                 return XEN_NETIF_CTRL_STATUS_INVALID_PARAMETER;
 282
 283         if (vif->hash.alg == XEN_NETIF_CTRL_HASH_ALGORITHM_NONE)
 284                 return XEN_NETIF_CTRL_STATUS_INVALID_PARAMETER;
 285
 286         vif->hash.flags = flags;
 287
 288         return XEN_NETIF_CTRL_STATUS_SUCCESS;
 289 }
 290
 291 u32 xenvif_set_hash_key(struct xenvif *vif, u32 gref, u32 len)
 292 {
 293         u8 *key = vif->hash.key;
 294         struct gnttab_copy copy_op = {
 295                 .source.u.ref = gref,
 296                 .source.domid = vif->domid,
 297                 .dest.u.gmfn = virt_to_gfn(key),
 298                 .dest.domid = DOMID_SELF,
 299                 .dest.offset = xen_offset_in_page(key),
 300                 .len = len,
 301                 .flags = GNTCOPY_source_gref
 302         };
 303
 304         if (len > XEN_NETBK_MAX_HASH_KEY_SIZE)
 305                 return XEN_NETIF_CTRL_STATUS_INVALID_PARAMETER;
 306
 307         if (copy_op.len != 0) {
 308                 gnttab_batch_copy(&copy_op, 1);
 309
 310                 if (copy_op.status != GNTST_okay)
 311                         return XEN_NETIF_CTRL_STATUS_INVALID_PARAMETER;
 312         }
 313
 314         /* Clear any remaining key octets */
 315         if (len < XEN_NETBK_MAX_HASH_KEY_SIZE)
 316                 memset(key + len, 0, XEN_NETBK_MAX_HASH_KEY_SIZE - len);
 317
 318         xenvif_flush_hash(vif);
 319
 320         return XEN_NETIF_CTRL_STATUS_SUCCESS;
 321 }
 322
 323 u32 xenvif_set_hash_mapping_size(struct xenvif *vif, u32 size)
 324 {
 325         if (size > XEN_NETBK_MAX_HASH_MAPPING_SIZE)
 326                 return XEN_NETIF_CTRL_STATUS_INVALID_PARAMETER;
 327
 328         vif->hash.size = size;
 329         memset(vif->hash.mapping[vif->hash.mapping_sel], 0,
 330                sizeof(u32) * size);
 331
 332         return XEN_NETIF_CTRL_STATUS_SUCCESS;
 333 }
 334
 335 u32 xenvif_set_hash_mapping(struct xenvif *vif, u32 gref, u32 len,
 336                             u32 off)
 337 {
 338         u32 *mapping = vif->hash.mapping[!vif->hash.mapping_sel];
 339         unsigned int nr = 1;
 340         struct gnttab_copy copy_op[2] = {{
 341                 .source.u.ref = gref,
 342                 .source.domid = vif->domid,
 343                 .dest.domid = DOMID_SELF,
 344                 .len = len * sizeof(*mapping),
 345                 .flags = GNTCOPY_source_gref
 346         }};
 347
 348         if ((off + len < off) || (off + len > vif->hash.size) ||
 349             len > XEN_PAGE_SIZE / sizeof(*mapping))
 350                 return XEN_NETIF_CTRL_STATUS_INVALID_PARAMETER;
 351
 352         copy_op[0].dest.u.gmfn = virt_to_gfn(mapping + off);
 353         copy_op[0].dest.offset = xen_offset_in_page(mapping + off);
 354         if (copy_op[0].dest.offset + copy_op[0].len > XEN_PAGE_SIZE) {
 355                 copy_op[1] = copy_op[0];
 356                 copy_op[1].source.offset = XEN_PAGE_SIZE - copy_op[0].dest.offset;
 357                 copy_op[1].dest.u.gmfn = virt_to_gfn(mapping + off + len);
 358                 copy_op[1].dest.offset = 0;
 359                 copy_op[1].len = copy_op[0].len - copy_op[1].source.offset;
 360                 copy_op[0].len = copy_op[1].source.offset;
 361                 nr = 2;
 362         }
 363
 364         memcpy(mapping, vif->hash.mapping[vif->hash.mapping_sel],
 365                vif->hash.size * sizeof(*mapping));
 366
 367         if (copy_op[0].len != 0) {
 368                 gnttab_batch_copy(copy_op, nr);
 369
 370                 if (copy_op[0].status != GNTST_okay ||
 371                     copy_op[nr - 1].status != GNTST_okay)
 372                         return XEN_NETIF_CTRL_STATUS_INVALID_PARAMETER;
 373         }
 374
 375         while (len-- != 0)
 376                 if (mapping[off++] >= vif->num_queues)
 377                         return XEN_NETIF_CTRL_STATUS_INVALID_PARAMETER;
 378
 379         vif->hash.mapping_sel = !vif->hash.mapping_sel;
 380
 381         return XEN_NETIF_CTRL_STATUS_SUCCESS;
 382 }
 383
 384 #ifdef CONFIG_DEBUG_FS
 385 void xenvif_dump_hash_info(struct xenvif *vif, struct seq_file *m)
 386 {
 387         unsigned int i;
 388
 389         switch (vif->hash.alg) {
 390         case XEN_NETIF_CTRL_HASH_ALGORITHM_TOEPLITZ:
 391                 seq_puts(m, "Hash Algorithm: TOEPLITZ\n");
 392                 break;
 393
 394         case XEN_NETIF_CTRL_HASH_ALGORITHM_NONE:
 395                 seq_puts(m, "Hash Algorithm: NONE\n");
 396                 fallthrough;
 397         default:
 398                 return;
 399         }
 400
 401         if (vif->hash.flags) {
 402                 seq_puts(m, "\nHash Flags:\n");
 403
 404                 if (vif->hash.flags & XEN_NETIF_CTRL_HASH_TYPE_IPV4)
 405                         seq_puts(m, "- IPv4\n");
 406                 if (vif->hash.flags & XEN_NETIF_CTRL_HASH_TYPE_IPV4_TCP)
 407                         seq_puts(m, "- IPv4 + TCP\n");
 408                 if (vif->hash.flags & XEN_NETIF_CTRL_HASH_TYPE_IPV6)
 409                         seq_puts(m, "- IPv6\n");
 410                 if (vif->hash.flags & XEN_NETIF_CTRL_HASH_TYPE_IPV6_TCP)
 411                         seq_puts(m, "- IPv6 + TCP\n");
 412         }
 413
 414         seq_puts(m, "\nHash Key:\n");
 415
 416         for (i = 0; i < XEN_NETBK_MAX_HASH_KEY_SIZE; ) {
 417                 unsigned int j, n;
 418
 419                 n = 8;
 420                 if (i + n >= XEN_NETBK_MAX_HASH_KEY_SIZE)
 421                         n = XEN_NETBK_MAX_HASH_KEY_SIZE - i;
 422
 423                 seq_printf(m, "[%2u - %2u]: ", i, i + n - 1);
 424
 425                 for (j = 0; j < n; j++, i++)
 426                         seq_printf(m, "%02x ", vif->hash.key[i]);
 427
 428                 seq_puts(m, "\n");
 429         }
 430
 431         if (vif->hash.size != 0) {
 432                 const u32 *mapping = vif->hash.mapping[vif->hash.mapping_sel];
 433
 434                 seq_puts(m, "\nHash Mapping:\n");
 435
 436                 for (i = 0; i < vif->hash.size; ) {
 437                         unsigned int j, n;
 438
 439                         n = 8;
 440                         if (i + n >= vif->hash.size)
 441                                 n = vif->hash.size - i;
 442
 443                         seq_printf(m, "[%4u - %4u]: ", i, i + n - 1);
 444
 445                         for (j = 0; j < n; j++, i++)
 446                                 seq_printf(m, "%4u ", mapping[i]);
 447
 448                         seq_puts(m, "\n");
 449                 }
 450         }
 451 }
 452 #endif /* CONFIG_DEBUG_FS */
 453
 454 void xenvif_init_hash(struct xenvif *vif)
 455 {
 456         if (xenvif_hash_cache_size == 0)
 457                 return;
 458
 459         BUG_ON(vif->hash.cache.count);
 460
 461         spin_lock_init(&vif->hash.cache.lock);
 462         INIT_LIST_HEAD(&vif->hash.cache.list);
 463 }
 464
 465 void xenvif_deinit_hash(struct xenvif *vif)
 466 {
 467         xenvif_flush_hash(vif);
 468 }