move connect to a proper place
This commit is contained in:
parent
6e50890f4f
commit
9ffb0320bb
1 changed files with 4 additions and 2 deletions
|
@ -176,6 +176,10 @@ class UDPRelay(object):
|
||||||
return
|
return
|
||||||
self._eventloop.add(client, eventloop.MODE_IN)
|
self._eventloop.add(client, eventloop.MODE_IN)
|
||||||
|
|
||||||
|
# prevent from recv other sources
|
||||||
|
if self._is_local:
|
||||||
|
client.connect((server_addr, server_port))
|
||||||
|
|
||||||
data = data[header_length:]
|
data = data[header_length:]
|
||||||
if not data:
|
if not data:
|
||||||
return
|
return
|
||||||
|
@ -183,8 +187,6 @@ class UDPRelay(object):
|
||||||
data = encrypt.encrypt_all(self._password, self._method, 1, data)
|
data = encrypt.encrypt_all(self._password, self._method, 1, data)
|
||||||
if not data:
|
if not data:
|
||||||
return
|
return
|
||||||
# prevent from recv other sources
|
|
||||||
client.connect((server_addr, server_port))
|
|
||||||
client.sendto(data, (server_addr, server_port))
|
client.sendto(data, (server_addr, server_port))
|
||||||
|
|
||||||
def _handle_client(self, sock):
|
def _handle_client(self, sock):
|
||||||
|
|
Loading…
Reference in a new issue