linux網絡編程入門三步走
阿新 • • 發佈:2018-02-09
tin cep use sea pass 內容 url more 統一 背景:每一個語言都會有入門的helloworld
內容:看完網絡編程的基礎知識,知道原來網絡編程也有小練習
概述:Echo,Chat,Proxy
內容:看完網絡編程的基礎知識,知道原來網絡編程也有小練習
概述:Echo,Chat,Proxy
網上有專門的練習。不管是UNIX網絡編程還是其他的書籍都會有相關的舉例
Echo,chat,Proxy在這裏統一再現Python的實現,主要是Python的比較簡單,稍後也會補充一下PHP的swoole
Echo:
import os,sys,platform import BaseHTTPServer from SocketServer import ThreadingMixIn import urllib,urllib2 import shutil import re import time try: from cStringIO import StringIO except ImportError: from StringIO import StringIO class SimpleHTTPRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler): def do_GET(self): f = StringIO() f.write(self.path) length = f.tell() f.seek(0) self.response(length) if f: shutil.copyfileobj(f,self.wfile) f.close() def response(self,length): self.send_response(200) self.send_header("Content-type","text/html") self.send_header("Content-Length",str(length)) self.end_headers() class ThreadingServer(ThreadingMixIn,BaseHTTPServer.HTTPServer): pass if __name__ =="__main__": serveraddr = ("",int(sys.argv[1])) print "LISTEN Port(%s)" % (int(sys.argv[1])) srvr = ThreadingServer(serveraddr,SimpleHTTPRequestHandler) srvr.serve_forever()
Chat:
import socket,select def broadcast_data(sock,message): for socket in CONNECTION_LIST: if socket != server_socket and socket !=sock: try: socket.send(message) except: socket.close() CONNECTION_LIST.remove(socket) if __name__ ==‘__main__‘: CONNECTION_LIST = [] RECV_BUFFER = 4096 HOST = ‘0.0.0.0‘ PORT = 3446 server_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM) server_socket.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1) server_socket.bind((HOST,PORT)) server_socket.listen(10) CONNECTION_LIST.append(server_socket) print "Chat server started on port {}"+str(PORT) while True: read_sockets,write_sockets,error_sockets = select.select(CONNECTION_LIST,[],[]) for sock in read_sockets: if sock ==server_socket: sockfd,addr = server_socket.accept() CONNECTION_LIST.append(sockfd) print ("Client (%s,%s) connected" % addr) broadcast_data(sockfd,"[%s:%s] entered room\n"% addr) else: try: data = sock.recv(RECV_BUFFER) if data: broadcast_data(sock,"\r"+‘<‘+str(sock.getpeername())+‘>‘+data) except: broadcast_data(sock,"Client (%s,%s) is offline"% addr) print ("Client (%s,%s) is offline"% addr) sock.close() CONNECTION_LIST.remove(sock) continue server_socket.close()
聊天的客戶端也是一樣的
Chat-Client:
import socket,select,string,sys def prompt(): sys.stdout.write("<you> ") sys.stdout.flush() if __name__=="__main__": if(len(sys.argv)<3): print "Usage: python chat_client.py host port" sys.exit() host = sys.argv[1] port = int(sys.argv[2]) s = socket.socket(socket.AF_INET,socket.SOCK_STREAM) s.settimeout(2) try: s.connect((host,port)) except: print ("Unable to connnect") sys.exit() print("Connected to remote host,Start send message") prompt() while True: rlist = [sys.stdin,s] read_list,write_list,error_list = select.select(rlist,[],[]) for sock in read_list: if sock ==s: data = sock.recv(4096) if not data: print ‘\nDisconnected from chat server‘; sys.exit() else: sys.stdout.write(data) prompt() else: msg = sys.stdin.readline() s.send(msg) prompt()
Proxy(好像當時碼的時候有小錯誤,反正就是連接兩邊的服務器,都接收過來,然後各自在轉發出去,來來回回,寫暈了)
import sys
import socket
import threading
def server_loop(local_host,local_port,remote_host,remote_port,receive_first):
server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
try:
server.bind(local_host,local_port)
except:
print ("[!!] Failed to listen on %s:%d"% (local_host,local_port))
print ("[!!] Check for other listening sockets or correct permissions")
sys.exit(0)
print ("[*] Listening on $s:%d"% (localhos_host,local_port))
server.listen(5)
while True:
client_socket.addr = server.accept()
print ("[==>] Received incoming connnection from %s:%d"% addr[0],addr[1])
proxy_thread = threading.Thread(target = proxy_handler,args =(client_socket,remote_host,remote_port,receive_first))
proxy_thread.start()
def proxy_handler(client_socket,remote_host,remote_port,receive_first):
remote_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
remote_socket.connect((remote_host,remote_port))
if receive_first:
remote_buffer = receive_from(remote_socket)
hexdump(remote_buffer)
remote_buffer = response_handler(remote_buffer)
remote_socket.send(local_buffer)
print("[==>] Sent to remote")
remote_buffer = receive_from(remote_socket)
if len(remote_buffer):
print("[<==] Sending %d bytes to local_host."%len(remote_buffer))
client.send(remote_buffer)
while True:
local_buffer = receive_from(client_socket)
if len(remote_buffer):
print ("[==>] Received %d bites from local_host" %len(local_buffer))
hexdump(local_buffer)
local_buffer = request_handler(local_buffer)
remote_socket.send(local_buffer)
print("[==>] Sent to remote")
remote_buffer = receive_from(remote_socket)
if len(remote_buffer):
print("[<== ] Received %d bytes from remote." % len(remote_buffer))
hexdump(remote_buffer)
remote_buffer = response_handler(remote_buffer)
client.socket.send(remote_buffer)
print("[<==] Sent to Sent to local_host")
if not len(local_buffer) or not len(remote_buffer):
client_socket.close()
remote_socket.close()
print("[*] No more data.Closing connections")
break
def hexdump(src,length = 16):
result =[]
digits = 4 if isinstance(str,unicode) else 2
for i in xrange(0,len(src),length):
s = src[i:i+length]
hexa = b‘ ‘.join(["%0*x"%(digits,ord(x)) for x in s])
text = b‘‘.join([x if 0x20 <= ord(0) <=0x7F else b‘.‘ for x in s])
result.append(b‘%04X %-*s %s‘%(i,length*(digits + 1),hexa,text))
print(b‘\n‘.join(result))
def receive_from(connection):
buffer = ‘‘
connection.settimeout(2)
try:
while True:
data = connection.recv(4096)
if not data:
break
buffer +=data
except:
pass
return buffer
def request_handler(buffer):
return buffer
def response_handler(buffer):
return buffer
def main():
if len(sys.argv[1:])!=5:
print("Usage:python proxy.py [localhos_host] [local_port] [remote_host] [remote_port] [receice_first]")
print("Example:python proxy.py 127.0.0.1")
sys.exit(0)
local_host = sys.argv[1]
local_port = int(sys.argv[2])
remote_host = sys.argv[3]
remote_port = sys.argv[4]
receive_first = sys.argv[5]
if ‘True‘ in receive_first:
receive_first = True
else:
receive_first = False
server_loop(local_host,local_port,remote_host,remote_port,receive_first)
main()
linux網絡編程入門三步走