socket官方文檔：https://docs.python.org/2/library/socket.html
socket中文詳細介紹：http://blog.csdn.net/rebelqsp/article/details/22109925
查看某端口是否被占用(如6666端口)：sudo netstat -nap | grep 6666 如果沒有查找到6666端口則說明沒有被占用（額，有點廢話了~~）
客戶端
- 創建 socket
- 連接到遠程服務器
- 發送數據
- 接收數據
- 關閉 socket
#!/usr/bin/env python # -*- coding=utf-8 -*- """ file: client.py socket client """ import socket import sys def socket_client(): try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.connect(('127.0.0.1', 6666)) except socket.error as msg: print msg sys.exit(1) print s.recv(1024) while 1: data = raw_input('please input work: ') s.send(data) print s.recv(1024) if data == 'exit': break s.close() if __name__ == '__main__': socket_client()
服務器端
- 打開 socket
- 綁定到特定的地址以及端口上
- 監聽連接
- 建立連接
- 接收/發送數據
#!/usr/bin/env python # -*- coding=utf-8 -*- """ file: service.py socket service """ import socket import threading import time import sys def socket_service(): try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 防止socket server重啟后端口被占用（socket.error: [Errno 98] Address already in use） s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) s.bind(('127.0.0.1', 6666)) s.listen(10) except socket.error as msg: print msg sys.exit(1) print 'Waiting connection...' while 1: conn, addr = s.accept() t = threading.Thread(target=deal_data, args=(conn, addr)) t.start() def deal_data(conn, addr): print 'Accept new connection from {0}'.format(addr) conn.send('Hi, Welcome to the server!') while 1: data = conn.recv(1024) print '{0} client send data is {1}'.format(addr, data) #time.sleep(1) if data == 'exit' or not data: print '{0} connection close'.format(addr) conn.send('Connection closed!') break conn.send('Hello, {0}'.format(data)) conn.close() if __name__ == '__main__': socket_service()
copy以上兩個腳本代碼后，如果你想在本地測試，直接運行這兩個腳本便是（注：先運行服務端service.py腳本）,下圖左邊是運行服務端腳本，右邊打開兩個終端與服務端socket進行通信。
一屏開多個終端是不是很屌（強烈建議安裝此工具，非常方便好用）
終端增強工具Terminator
sudo apt-get install terminator # Ubuntu
yum -y install terminator # Centos
如果你想在兩台不同機器上測試，把service.py腳本上傳到服務器(假設服務器ip是：192.168.1.2)，然后把client.py連接ip改為：192.168.1.2，service.py監聽的ip改為：192.168.1.2，接着運行這兩個腳本即可
# client.py s.connect(('192.168.1.2', 6666)) # service.py s.bind(('192.168.1.2', 6666))
利用socket進行文件、圖片等傳輸
傳輸文件主要分以下兩步：
- 將要傳輸的文件的基本信息發送到接收端（文件名、大小等其他信息）
- 發送端讀取文件內容並發送過去,接受端將緩存里面的內容寫入文件
注：以下發送文件代碼參考：http://www.mamicode.com/info-detail-1346172.html
發送端代碼
#!/usr/bin/env python # -*- coding=utf-8 -*- """ file: send.py socket client """ import socket import os import sys import struct def socket_client(): try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.connect(('127.0.0.1', 6666)) except socket.error as msg: print msg sys.exit(1) print s.recv(1024) while 1: filepath = raw_input('please input file path: ') if os.path.isfile(filepath): # 定義定義文件信息。128s表示文件名為128bytes長，l表示一個int或log文件類型，在此為文件大小 fileinfo_size = struct.calcsize('128sl') # 定義文件頭信息，包含文件名和文件大小 fhead = struct.pack('128sl', os.path.basename(filepath), os.stat(filepath).st_size) s.send(fhead) print 'client filepath: {0}'.format(filepath) fp = open(filepath, 'rb') while 1: data = fp.read(1024) if not data: print '{0} file send over...'.format(filepath) break s.send(data) s.close() break if __name__ == '__main__': socket_client()
接收端代碼
#!/usr/bin/env python # -*- coding=utf-8 -*- """ file: recv.py socket service """ import socket import threading import time import sys import os import struct def socket_service(): try: s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) s.bind(('127.0.0.1', 6666)) s.listen(10) except socket.error as msg: print msg sys.exit(1) print 'Waiting connection...' while 1: conn, addr = s.accept() t = threading.Thread(target=deal_data, args=(conn, addr)) t.start() def deal_data(conn, addr): print 'Accept new connection from {0}'.format(addr) #conn.settimeout(500) conn.send('Hi, Welcome to the server!') while 1: fileinfo_size = struct.calcsize('128sl') buf = conn.recv(fileinfo_size) if buf: filename, filesize = struct.unpack('128sl', buf) fn = filename.strip('\00') new_filename = os.path.join('./', 'new_' + fn) print 'file new name is {0}, filesize if {1}'.format(new_filename, filesize) recvd_size = 0 # 定義已接收文件的大小 fp = open(new_filename, 'wb') print 'start receiving...' while not recvd_size == filesize: if filesize - recvd_size > 1024: data = conn.recv(1024) recvd_size += len(data) else: data = conn.recv(filesize - recvd_size) recvd_size = filesize fp.write(data) fp.close() print 'end receive...' conn.close() break if __name__ == '__main__': socket_service()
打開兩個終端，分別運行recv.py和send.py腳本，輸入待傳輸文件的全路徑和文件名（如：/home/test.txt），你就會在recv.py腳本路徑下看到剛輸入的傳輸文件，且文件名為new_****。