Python實現簡單的udp打洞（P2P）
| NAT的分類：在STUN協議中，根據內部終端的地址(LocalIP:LocalPort)到NAT出口的公網地址(PublicIP:PublicPort)的影射方式，把NAT分為四種類型（英文原文詳見rfc3489標準：）： |
| 由於存在有4種NAT的型別，所以在方案設計中，必須要考慮到用哪種軟體或者裝置來模擬NAT環境。我在實現過程中嘗試過了很多選擇，包括Linux下的iptables，海蜘蛛的軟路由，使用CISCO的IOS模擬，window2003自帶的NAT服務，還有VMWARE自帶的NAT網路環境。最終發現window2003和VMWARE是支援full cone nat的。 |
#!/usr/bin/python
#coding:utf-8
import socket, sys, SocketServer, threading, thread, time
SERVER_PORT = 1234
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind(('', SERVER_PORT))
user_list = []
def server_handle():
whileTrue:
cli_date, cli_pub_add = sock.recvfrom(8192)
now_user = []
headder = []
cli_str = {}
headder = cli_date.split('\t')
for one_line in headder:
str = {}
str = one_line
args = str.split(':')
cli_str[args[0]] = args[1]
if cli_str['type'] == 'login' :
del cli_str['type']
now_user = cli_str
now_user['cli_pub_ip'] = cli_pub_add[0]
now_user['cli_pub_port'] = cli_pub_add[1]
user_list.append(now_user)
toclient = 'info#%s login in successful , the info from server'%now_user['user_name']
sock.sendto(toclient,cli_pub_add)
print'-'*100
print"%s 已經登入,公網IP:%s 埠:%d\n"%(now_user['user_name'],now_user['cli_pub_ip'],now_user['cli_pub_port'])
print"以下是已經登入的使用者列表"
for one_user in user_list:
print'使用者名稱:%s 公網ip:%s 公網埠:%s 私網ip:%s 私網埠:%s'%(one_user['user_name'],one_user['cli_pub_ip'],one_user['cli_pub_port'],one_user['private_ip'],one_user['private_port'])
elif cli_str['type'] == 'alive':
pass
elif cli_str['type'] == 'logout' :
pass
elif cli_str['type'] == 'getalluser' :
print'-'*100
for one_user in user_list :
toclient = 'getalluser#username:%s pub_ip:%s pub_port:%s pri_ip:%s pri_port:%s'%(one_user['user_name'],one_user['cli_pub_ip'],one_user['cli_pub_port'],one_user['private_ip'],one_user['private_port'])
sock.sendto(toclient,cli_pub_add)
if __name__ == '__main__':
thread.start_new_thread(server_handle, ())
print'伺服器程序已啟動，等待客戶連線'
whileTrue:
for one_user in user_list:
toclient = 'keepconnect#111'
sock.sendto(toclient,(one_user['cli_pub_ip'],one_user['cli_pub_port']))
time.sleep(1)
#!/usr/bin/python
#coding:utf-8
import socket, SocketServer, threading, thread, time
CLIENT_PORT = 4321
SERVER_IP = "200.0.0.128"
SERVER_PORT = 1234
user_list = {}
local_ip = socket.gethostbyname(socket.gethostname())
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
def server_handle():
print'客戶端執行緒已經啟動 , 等待其它客戶端連線'
whileTrue:
data, addr = sock.recvfrom(8192)
data_str = data.split('#')
data_type = data_str[0]
data_info = data_str[1]
if data_type == 'info' :
del data_str[0]
print data_info
if data_type == 'getalluser' :
data_sp = data_info.split(' ')
user_name = data_sp[0].split(':')[1]
del data_sp[0]
user_list[user_name] = {}
for one_line in data_sp:
arg = one_line.split(':')
user_list[user_name][arg[0]] = arg[1]
if data_type == 'echo' :
print data_info
if data_type == 'keepconnect':
messeg = 'type:alive'
sock.sendto(messeg, addr)
if __name__ == '__main__':
thread.start_new_thread(server_handle, ())
time.sleep(0.1)
cmd = raw_input('輸入指令>>')
whileTrue:
args = cmd.split(' ')
if args[0] == 'login':
user_name = args[1]
local_uname = args[1]
address = "private_ip:%s private_port:%d" % (local_ip, CLIENT_PORT)
headder = "type:login\tuser_name:%s\tprivate_ip:%s\tprivate_port:%d" % (user_name,local_ip,CLIENT_PORT)
sock.sendto(headder, (SERVER_IP, SERVER_PORT))
elif args[0] == 'getalluser':
headder = "type:getalluser\tuser_name:al"
sock.sendto(headder,(SERVER_IP,SERVER_PORT))
print'獲取使用者列表中。。。'
time.sleep(1)
for one_user in user_list:
print'username:%s pub_ip:%s pub_port:%s pri_ip:%s pri_port:%s'%(one_user,user_list[one_user]['pub_ip'],user_list[one_user]['pub_port'],user_list[one_user]['pri_ip'],user_list[one_user]['pri_port'])
elif args[0] == 'connect':
user_name = args[1]
to_user_ip = user_list[user_name]['pub_ip']
to_user_port = int(user_list[user_name]['pub_port'])
elif args[0] =='echo':
m = ' '.join(args[1:])
messeg = 'echo#from %s:%s'%(local_uname,m)
sock.sendto(messeg, (to_user_ip, to_user_port))
time.sleep(0.1)
cmd = raw_input('輸入指令>>')