drivers/net/wireless/quantenna/qtnfmac/util.c

   1 /*
   2  * Copyright (c) 2015-2016 Quantenna Communications, Inc.
   3  * All rights reserved.
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU General Public License
   7  * as published by the Free Software Foundation; either version 2
   8  * of the License, or (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  */
  16
  17 #include "util.h"
  18
  19 void qtnf_sta_list_init(struct qtnf_sta_list *list)
  20 {
  21         if (unlikely(!list))
  22                 return;
  23
  24         INIT_LIST_HEAD(&list->head);
  25         atomic_set(&list->size, 0);
  26 }
  27
  28 struct qtnf_sta_node *qtnf_sta_list_lookup(struct qtnf_sta_list *list,
  29                                            const u8 *mac)
  30 {
  31         struct qtnf_sta_node *node;
  32
  33         if (unlikely(!mac))
  34                 return NULL;
  35
  36         list_for_each_entry(node, &list->head, list) {
  37                 if (ether_addr_equal(node->mac_addr, mac))
  38                         return node;
  39         }
  40
  41         return NULL;
  42 }
  43
  44 struct qtnf_sta_node *qtnf_sta_list_lookup_index(struct qtnf_sta_list *list,
  45                                                  size_t index)
  46 {
  47         struct qtnf_sta_node *node;
  48
  49         if (qtnf_sta_list_size(list) <= index)
  50                 return NULL;
  51
  52         list_for_each_entry(node, &list->head, list) {
  53                 if (index-- == 0)
  54                         return node;
  55         }
  56
  57         return NULL;
  58 }
  59
  60 struct qtnf_sta_node *qtnf_sta_list_add(struct qtnf_vif *vif,
  61                                         const u8 *mac)
  62 {
  63         struct qtnf_sta_list *list = &vif->sta_list;
  64         struct qtnf_sta_node *node;
  65
  66         if (unlikely(!mac))
  67                 return NULL;
  68
  69         node = qtnf_sta_list_lookup(list, mac);
  70
  71         if (node)
  72                 goto done;
  73
  74         node = kzalloc(sizeof(*node), GFP_KERNEL);
  75         if (unlikely(!node))
  76                 goto done;
  77
  78         ether_addr_copy(node->mac_addr, mac);
  79         list_add_tail(&node->list, &list->head);
  80         atomic_inc(&list->size);
  81         ++vif->generation;
  82
  83 done:
  84         return node;
  85 }
  86
  87 bool qtnf_sta_list_del(struct qtnf_vif *vif, const u8 *mac)
  88 {
  89         struct qtnf_sta_list *list = &vif->sta_list;
  90         struct qtnf_sta_node *node;
  91         bool ret = false;
  92
  93         node = qtnf_sta_list_lookup(list, mac);
  94
  95         if (node) {
  96                 list_del(&node->list);
  97                 atomic_dec(&list->size);
  98                 kfree(node);
  99                 ++vif->generation;
 100                 ret = true;
 101         }
 102
 103         return ret;
 104 }
 105
 106 void qtnf_sta_list_free(struct qtnf_sta_list *list)
 107 {
 108         struct qtnf_sta_node *node, *tmp;
 109
 110         atomic_set(&list->size, 0);
 111
 112         list_for_each_entry_safe(node, tmp, &list->head, list) {
 113                 list_del(&node->list);
 114                 kfree(node);
 115         }
 116
 117         INIT_LIST_HEAD(&list->head);
 118 }