Changeset 187 for binary-improvements/7dtd-server-fixes/src/NetConnections/Servers/Telnet/Telnet.cs

Timestamp:

Sep 10, 2014, 8:09:28 PM (10 years ago)

Author:

alloc

Message:

fixes

File:

• binary-improvements/7dtd-server-fixes/src/NetConnections/Servers/Telnet/Telnet.cs (modified) (3 diffs)

binary-improvements/7dtd-server-fixes/src/NetConnections/Servers/Telnet/Telnet.cs

@@ -9 +9 @@
         public class Telnet : IServer
         {
+                private const int MAX_LOGIN_ATTEMPTS = 10;
+                private const int BLOCK_TIME_SECONDS = 10;
+
+                private class LoginAttempts
+                {
+                        private int count = 0;
+                        private DateTime lastAttempt = new DateTime (0);
+
+                        public bool LogAttempt ()
+                        {
+                                lastAttempt = DateTime.Now;
+                                count++;
+                                return count < MAX_LOGIN_ATTEMPTS;
+                        }
+
+                        public bool IsBanned ()
+                        {
+                                if ((DateTime.Now - lastAttempt).TotalSeconds > BLOCK_TIME_SECONDS)
+                                        count = 0;
+                                return count >= MAX_LOGIN_ATTEMPTS;
+                        }
+                }
+
                 private TcpListener listener = null;
                 private bool authEnabled = false;
                 private List<TelnetConnection> connections = new List<TelnetConnection> ();
+                private Dictionary<int, LoginAttempts> loginAttemptsPerIP = new Dictionary<int, LoginAttempts> ();
 
                 public Telnet ()
@@ -39 +63 @@
                 }
 
-                public void FailedLogins ()
+                public bool RegisterFailedLogin (int addressHash)
                 {
+                        lock (loginAttemptsPerIP) {
+                                LoginAttempts la = loginAttemptsPerIP [addressHash];
+                                return la.LogAttempt ();
+                        }
                 }
 
@@ -46 +74 @@
                 {
                         if (listener.Server.IsBound) {
-                                TelnetConnection c = new TelnetConnection (this, listener.EndAcceptTcpClient (asyncResult), authEnabled);
-                                connections.Add (c);
+                                TcpClient client = listener.EndAcceptTcpClient (asyncResult);
+
+                                EndPoint endpoint = client.Client.RemoteEndPoint;
+                                int addressHash = -1;
+                                if (endpoint is IPEndPoint) {
+                                        addressHash = ((IPEndPoint)endpoint).Address.GetHashCode ();
+                                        //Log.Out ("Hash: " + endpointAddressHash);
+                                } else {
+                                        Log.Out ("EndPoint is not an IPEndPoint but: " + endpoint.GetType ().ToString ());
+                                }
+
+                                lock (loginAttemptsPerIP) {
+                                        LoginAttempts la = null;
+                                        if (loginAttemptsPerIP.ContainsKey(addressHash))
+                                                la = loginAttemptsPerIP [addressHash];
+                                        if (la == null) {
+                                                la = new LoginAttempts ();
+                                                loginAttemptsPerIP [addressHash] = la;
+                                        }
+                                        if (!la.IsBanned ()) {
+                                                TelnetConnection con = new TelnetConnection (this, client, authEnabled);
+                                                connections.Add (con);
+                                        } else {
+                                                client.Close ();
+                                                Log.Out ("Telnet connection not accepted for too many login attempts: " + endpoint);
+                                        }
+                                }
                                 listener.BeginAcceptTcpClient (new AsyncCallback (AcceptClient), null);
                         }