From e914ec3fa2e3faad5806c40b5c6476524e15e4fc Mon Sep 17 00:00:00 2001
From: ChillerDragon <chillerdragon@gmail.com>
Date: Sat, 12 Nov 2022 12:01:38 +0100
Subject: [PATCH] Send broken close message to clients after 1 second

Aka starting to work on timeouting clients
---
 lib/game_server.rb      | 18 ++++++++++++++++++
 lib/teeworlds_server.rb | 23 +++++++++++++++++++++--
 2 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/lib/game_server.rb b/lib/game_server.rb
index ab9c3c4..a6d1aae 100644
--- a/lib/game_server.rb
+++ b/lib/game_server.rb
@@ -71,4 +71,22 @@ class GameServer
     # we do nothing for now
     # TODO: do something
   end
+
+  def disconnect_client(client, reason = nil)
+    @server.send_ctrl_close(client, reason)
+  end
+
+  def on_tick
+    now = Time.now
+    timeout_ids = []
+    @server.clients.each do |id, client|
+      diff = now - client.last_recv_time
+      timeout_ids.push(id) if diff > 1
+    end
+
+    timeout_ids.each do |id|
+      disconnect_client(@server.clients[id], 'Timeout')
+      @server.clients.delete(id)
+    end
+  end
 end
diff --git a/lib/teeworlds_server.rb b/lib/teeworlds_server.rb
index f476b64..c62f838 100644
--- a/lib/teeworlds_server.rb
+++ b/lib/teeworlds_server.rb
@@ -15,12 +15,20 @@ require_relative 'game_server'
 require_relative 'message'
 
 class Client
-  attr_accessor :id, :addr, :vital_sent
+  attr_accessor :id, :addr, :vital_sent, :last_recv_time, :token
 
   def initialize(attr = {})
     @id = attr[:id]
     @addr = attr[:addr]
     @vital_sent = 0
+    @token = attr[:token]
+    unless @token.size == 4
+      raise "Invalid client token size\n" \
+            "got=#{@token.size} expected=4\n" \
+            "#{str_hex(@token)}"
+    end
+
+    @last_recv_time = Time.now
   end
 
   # TODO: use or remove
@@ -36,6 +44,8 @@ class Client
 end
 
 class TeeworldsServer
+  attr_accessor :clients
+
   def initialize(options = {})
     @verbose = options[:verbose] || false
     @ip = '127.0.0.1'
@@ -132,6 +142,14 @@ class TeeworldsServer
     end
   end
 
+  def send_ctrl_close(client, reason)
+    msg = [NET_CTRLMSG_CLOSE]
+    msg += Packer.pack_str(reason) unless reason.nil?
+    @netbase.peer_token = client.token
+    @netbase.send_packet(msg, 0, control: true, addr: client.addr)
+    # @netbase.peer_token = @server_token
+  end
+
   def send_ctrl_with_token(addr, token)
     msg = [NET_CTRLMSG_TOKEN] + str_bytes(@server_token)
     @netbase.peer_token = token
@@ -203,7 +221,7 @@ class TeeworldsServer
       puts 'server full drop packet. TODO: tell the client'
       return
     end
-    client = Client.new(id:, addr: packet.addr)
+    client = Client.new(id:, addr: packet.addr, token: packet.payload[...4])
     @clients[id] = client
     @netbase.send_packet([NET_CTRLMSG_ACCEPT], 0, control: true, addr: packet.addr)
   end
@@ -254,6 +272,7 @@ class TeeworldsServer
         @current_game_tick += 1
         do_snapshot
       end
+      @game_server.on_tick
     end
 
     begin