diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/device.h | 2 | ||||
| -rw-r--r-- | src/peer.c | 15 | ||||
| -rw-r--r-- | src/peer.h | 2 |
3 files changed, 4 insertions, 15 deletions
diff --git a/src/device.h b/src/device.h index 4932477..a74f58b 100644 --- a/src/device.h +++ b/src/device.h @@ -49,7 +49,7 @@ struct wireguard_device { struct routing_table peer_routing_table; struct mutex device_update_lock, socket_update_lock; struct list_head device_list, peer_list; - unsigned int device_update_gen; + unsigned int num_peers, device_update_gen; u32 fwmark; u16 incoming_port; }; @@ -20,8 +20,9 @@ struct wireguard_peer *peer_create(struct wireguard_device *wg, const u8 public_ lockdep_assert_held(&wg->device_update_lock); - if (peer_total_count(wg) >= MAX_PEERS_PER_DEVICE) + if (wg->num_peers >= MAX_PEERS_PER_DEVICE) return NULL; + ++wg->num_peers; peer = kzalloc(sizeof(struct wireguard_peer), GFP_KERNEL); if (!peer) @@ -89,6 +90,7 @@ void peer_remove(struct wireguard_peer *peer) flush_workqueue(peer->device->packet_crypt_wq); /* The first flush is for encrypt/decrypt step. */ flush_workqueue(peer->device->packet_crypt_wq); /* The second flush is for send/receive step. */ flush_workqueue(peer->device->handshake_send_wq); + --peer->device->num_peers; peer_put(peer); } @@ -127,14 +129,3 @@ void peer_remove_all(struct wireguard_device *wg) list_for_each_entry_safe (peer, temp, &wg->peer_list, peer_list) peer_remove(peer); } - -unsigned int peer_total_count(struct wireguard_device *wg) -{ - unsigned int i = 0; - struct wireguard_peer *peer; - - lockdep_assert_held(&wg->device_update_lock); - list_for_each_entry (peer, &wg->peer_list, peer_list) - ++i; - return i; -} @@ -67,6 +67,4 @@ void peer_remove_all(struct wireguard_device *wg); struct wireguard_peer *peer_lookup_by_index(struct wireguard_device *wg, u32 index); -unsigned int peer_total_count(struct wireguard_device *wg); - #endif /* _WG_PEER_H */ |