Changeset 559

Timestamp:

04/27/07 03:05:51 (16 months ago)

Author:

hyriand

Message:

Preliminary IP blocked framework

Location:

museek+/trunk/sources/Museek

Files:

• Museek.cc (modified) (1 diff)
• Museek.hh (modified) (1 diff)
• PeerListener.cc (modified) (1 diff)

museek+/trunk/sources/Museek/Museek.cc

@@ -161 +161 @@
         return false;
 }
+bool Museek::is_ip_blocked(const std::string& ip) const {
+        return false;
+}
+
 void Museek::mu_set_user_warnings(bool uw) {
         mUserWarnings = uw;

museek+/trunk/sources/Museek/Museek.hh

@@ -110 +110 @@
         virtual bool is_banned(const std::string& user) const;
         virtual bool is_trusted(const std::string& user) const;
+        virtual bool is_ip_blocked(const std::string& ip) const;
 
         inline bool have_buddy_shares() const { return mBuddySharesHave; }

museek+/trunk/sources/Museek/PeerListener.cc

@@ -127 +127 @@
                 return false;
         
+        sockaddr_in addr;
+        socklen_t addrlen = sizeof(addr);
+        if(! getpeername(sock, (sockaddr*)&addr, &addrlen)) {
+                unsigned int ip = ntohl(addr.sin_addr.s_addr);
+                char ip_c_str[16];
+                snprintf(ip_c_str, 16, "%i.%i.%i.%i", (ip >> 24) & 0xff, (ip >> 16) & 0xff, (ip >> 8) & 0xff, ip & 0xff);
+                std::string ip_str(ip_c_str);
+                if(mMuseek->is_ip_blocked(ip_str)) {
+                        DEBUG("Blocking incoming connection from %s\n", ip_c_str);
+                        close(sock);
+                        return true;
+                }
+        }
+        
         new PreConnection(this, sock);