Merge pull request #4 from clowwindy/ipv6

Ipv6
This commit is contained in:
clowwindy 2012-05-10 19:55:36 -07:00
commit 174d50bc6b
2 changed files with 65 additions and 6 deletions

View file

@ -20,7 +20,7 @@
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
SERVER = 'myserver_ip_or_hostname'
SERVER = '::1'
REMOTE_PORT = 8499
PORT = 1080
KEY = "foobar!"
@ -34,6 +34,32 @@ import threading
import time
import SocketServer
def socket_create_connection(address, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
source_address=None):
"""python 2.7 socket.create_connection"""
host, port = address
err = None
for res in socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM):
af, socktype, proto, canonname, sa = res
sock = None
try:
sock = socket.socket(af, socktype, proto)
if timeout is not socket._GLOBAL_DEFAULT_TIMEOUT:
sock.settimeout(timeout)
if source_address:
sock.bind(source_address)
sock.connect(sa)
return sock
except socket.error as _:
err = _
if sock is not None:
sock.close()
if err is not None:
raise err
else:
raise error("getaddrinfo returns an empty list")
def get_table(key):
m = hashlib.md5()
m.update(key)
@ -95,8 +121,7 @@ class Socks5Server(SocketServer.StreamRequestHandler):
def handle(self):
try:
sock = self.connection
remote = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
remote.connect((SERVER, REMOTE_PORT))
remote = socket_create_connection((SERVER, REMOTE_PORT))
self.handle_tcp(sock, remote)
except socket.error:
lock_print('socket error')

View file

@ -23,12 +23,45 @@
PORT = 8499
KEY = "foobar!"
try:
import gevent, gevent.monkey
gevent.monkey.patch_all(dns=gevent.version_info[0]>=1)
except ImportError:
gevent = None
import socket
import select
import SocketServer
import struct
import string
import hashlib
import sys
def socket_create_connection(address, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
source_address=None):
"""python 2.7 socket.create_connection"""
host, port = address
err = None
for res in socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM):
af, socktype, proto, canonname, sa = res
sock = None
try:
sock = socket.socket(af, socktype, proto)
if timeout is not socket._GLOBAL_DEFAULT_TIMEOUT:
sock.settimeout(timeout)
if source_address:
sock.bind(source_address)
sock.connect(sa)
return sock
except error as _:
err = _
if sock is not None:
sock.close()
if err is not None:
raise err
else:
raise error("getaddrinfo returns an empty list")
def get_table(key):
m = hashlib.md5()
@ -90,8 +123,7 @@ class Socks5Server(SocketServer.StreamRequestHandler):
reply = "\x05\x00\x00\x01"
try:
if mode == 1:
remote = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
remote.connect((addr, port[0]))
remote = socket_create_connection((addr, port[0]))
local = remote.getsockname()
reply += socket.inet_aton(local[0]) + struct.pack(">H",
local[1])
@ -106,11 +138,13 @@ class Socks5Server(SocketServer.StreamRequestHandler):
if reply[1] == '\x00':
if mode == 1:
self.handle_tcp(sock, remote)
except socket.error:
except socket.error as e:
print 'socket error'
def main():
if '-6' in sys.argv[1:]:
ThreadingTCPServer.address_family = socket.AF_INET6
server = ThreadingTCPServer(('', PORT), Socks5Server)
server.allow_reuse_address = True
print "starting server at port %d ..." % PORT