oci/zeronet
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix P2P port check

Browse source
This commit is contained in:
grez911 2017-08-18 11:29:41 +03:00
parent c0a96983a7
commit 1fc2a917d4
2 changed files with 17 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/File/FileRequest.py
View file

@ -463,9 +463,9 @@ class FileRequest(object):
with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as sock: with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as sock:
sock.settimeout(5) sock.settimeout(5)
if sock.connect_ex((self.connection.ip, params["port"])) == 0: if sock.connect_ex((self.connection.ip, params["port"])) == 0:
self.response("open %s" % self.connection.ip) self.response({"status": "open", "ip_external": self.connection.ip})
else: else:
self.response("closed %s" % self.connection.ip) self.response({"status": "closed", "ip_external": self.connection.ip})
# Unknown command # Unknown command
def actionUnknown(self, cmd, params): def actionUnknown(self, cmd, params):

32
src/File/FileServer.py
View file

@ -97,48 +97,40 @@ class FileServer(ConnectionServer):
def testOpenportP2P(self, port=None): def testOpenportP2P(self, port=None):
self.log.info("Checking port %s using P2P..." % port) self.log.info("Checking port %s using P2P..." % port)
site = SiteManager.site_manager.get("1HeLLo4uzjaLetFx6NH3PMwFP3qbRbTf3D") site = SiteManager.site_manager.get(config.homepage)
peers = [] peers = []
res = None res = None
if not site: # First run, has no any peers if not site: # First run, has no any peers
return self.testOpenportPortchecker(port) # Fallback to centralized service return self.testOpenportPortchecker(port) # Fallback to centralized service
for peer in site.peers.values(): # Get all non-onion peers for peer in site.peers.values():
if peer.ip.endswith(".onion"): if not peer.ip.endswith(".onion"): # Get all non-onion peers
peers.append(peer) peers.append(peer)
if len(peers) < 3: # Not enough peers if len(peers) < 3: # Not enough peers
return self.testOpenportPortchecker(port) # Fallback to centralized service return self.testOpenportPortchecker(port) # Fallback to centralized service
for retry in range(1, 3): # Try 3 peers for retry in range(0, 3): # Try 3 peers
random_peer = random.choice(peers) random_peer = random.choice(peers)
with gevent.Timeout(10.0, False): # 10 sec timeout, don't raise exception with gevent.Timeout(10.0, False): # 10 sec timeout, don't raise exception
random_peer.connect() random_peer.connect()
res = random_peer.request("checkport", {"port": port}) res = random_peer.request("checkport", {"port": port})
if res is not None:
break # All fine, exit from for loop break # All fine, exit from for loop
if res is None: # Nobody answered if res is None: # Nobody answered
return self.testOpenportPortchecker(port) # Fallback to centralized service return self.testOpenportPortchecker(port) # Fallback to centralized service
message = res["body"] if res["status"] == "closed":
if "open" not in message:
if config.tor != "always": if config.tor != "always":
self.log.info("[BAD :(] Port closed %s" % message) self.log.info("[BAD :(] %s says that your port %s is closed" % (random_peer.ip, port))
if port == self.port: if port == self.port:
self.port_opened = False # Self port, update port_opened status self.port_opened = False # Self port, update port_opened status
match = message.split(" ", 1)[1] # Try find my external ip in message config.ip_external = res["ip_external"]
if match: # Found my ip in message
config.ip_external = match
SiteManager.peer_blacklist.append((config.ip_external, self.port)) # Add myself to peer blacklist SiteManager.peer_blacklist.append((config.ip_external, self.port)) # Add myself to peer blacklist
return {"result": False}
else: else:
config.ip_external = False self.log.info("[OK :)] %s says that your port %s is open" % (random_peer.ip, port))
return {"result": False, "message": message}
else:
self.log.info("[OK :)] Port open: %s" % message)
if port == self.port: # Self port, update port_opened status if port == self.port: # Self port, update port_opened status
self.port_opened = True self.port_opened = True
match = message.split(" ", 1)[1] # Try find my external ip in message config.ip_external = res["ip_external"]
if match: # Found my ip in message
config.ip_external = match
SiteManager.peer_blacklist.append((config.ip_external, self.port)) # Add myself to peer blacklist SiteManager.peer_blacklist.append((config.ip_external, self.port)) # Add myself to peer blacklist
else: return {"result": True}
config.ip_external = False
return {"result": True, "message": message}
def testOpenportPortchecker(self, port=None): def testOpenportPortchecker(self, port=None):
self.log.info("Checking port %s using portchecker.co..." % port) self.log.info("Checking port %s using portchecker.co..." % port)