diff options
Diffstat (limited to 'src/timers.c')
| -rw-r--r-- | src/timers.c | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/src/timers.c b/src/timers.c index 47d7854..01933a8 100644 --- a/src/timers.c +++ b/src/timers.c @@ -76,6 +76,15 @@ static void queued_expired_kill_ephemerals(struct work_struct *work) peer_put(peer); } +static void expired_send_persistent_keepalive(unsigned long ptr) +{ + struct wireguard_peer *peer = (struct wireguard_peer *)ptr; + + if (unlikely(!peer->persistent_keepalive_interval)) + return; + socket_send_buffer_to_peer(peer, NULL, 0, 0); +} + void timers_data_sent(struct wireguard_peer *peer) { if (likely(peer->timer_send_keepalive.data)) @@ -121,6 +130,12 @@ void timers_ephemeral_key_created(struct wireguard_peer *peer) do_gettimeofday(&peer->walltime_last_handshake); } +void timers_any_packet_sent(struct wireguard_peer *peer) +{ + if (peer->persistent_keepalive_interval && likely(peer->timer_persistent_keepalive.data)) + mod_timer(&peer->timer_persistent_keepalive, jiffies + HZ * peer->persistent_keepalive_interval); +} + void timers_init_peer(struct wireguard_peer *peer) { init_timer(&peer->timer_retransmit_handshake); @@ -139,6 +154,10 @@ void timers_init_peer(struct wireguard_peer *peer) peer->timer_kill_ephemerals.function = expired_kill_ephemerals; peer->timer_kill_ephemerals.data = (unsigned long)peer; + init_timer(&peer->timer_persistent_keepalive); + peer->timer_persistent_keepalive.function = expired_send_persistent_keepalive; + peer->timer_persistent_keepalive.data = (unsigned long)peer; + INIT_WORK(&peer->clear_peer_work, queued_expired_kill_ephemerals); } @@ -160,6 +179,10 @@ void timers_uninit_peer(struct wireguard_peer *peer) del_timer(&peer->timer_kill_ephemerals); peer->timer_kill_ephemerals.data = 0; } + if (peer->timer_persistent_keepalive.data) { + del_timer(&peer->timer_persistent_keepalive); + peer->timer_persistent_keepalive.data = 0; + } } void timers_uninit_peer_wait(struct wireguard_peer *peer) { |