标签:action name binding bre comm tar tin tput pop
socket函数编程
客户端
import socket,time
import os
server = socket.socket()
server.bind((‘localhost‘,9999))
server.listen()
while True:
conn,addr = server.accept()
print("the new :" ,addr)
while True:
data = conn.recv(1024)
if not data:
print("客户端已断开")
break
print("执行指令:",data)
cmd_res = os.popen(data.decode()).read()
print("before send",len(cmd_res))
if len(cmd_res) == 0:
cmd_res = "cmd has no output"
conn.send( str(len(cmd_res.encode())).encode("utf-8") )
#time.sleep(0.5)#避免粘包
client_ack = conn.recv(1024)#wait client to confirm
conn.send(cmd_res.encode("utf-8"))
print("send done")
服务端
import socket
client = socket.socket()
client.connect((‘localhost‘,9999))
while True:
cmd = input(">>:").strip()
if len(cmd) == 0:
continue
client.send(cmd.encode("utf-8"))
cmd_res_size = client.recv(1024)
print("命令结果大小:",cmd_res_size)
client.send("准备好接受了".encode("utf-8"))
received_size = 0
received_data = b‘‘
while received_size < int(cmd_res_size.decode()):
data = client.recv(1024)
received_size += len(data)
# print(data.decode())
# print(received_size)
received_data += data
else:
print("cmd res receive done...",received_size)
# cmd_res = client.recv(1024)
print(cmd_res.decode())
server.close()
并发客户端
import socket,time,hashlib
import os
server = socket.socket()
server.bind((‘localhost‘,9999))
server.listen()
while True:
conn,addr = server.accept()
print("the new :" ,addr)
while True:
data = conn.recv(1024)
if not data:
print("客户端已断开")
break
cmd,filename = data.decode().split()
print(fliename)
if os.path.isfile(filename):
f = open(filename,"rb")
m = hashlib.md5()
file_size = os.stat(filename).st_size
conn.send(str(file_size).encode())#send file size
conn.recv(1024)#wait for ack
for line in f:
#m.updata(line)
conn.send(line)
#print("file md5 ",m.hexdigest())
f.close()
conn.send(m.hexdigest().encode())#send md5
print("send done")
server.close()
并发服务端
import socket
import hashlib
client = socket.socket()
client.connect((‘localhost‘,9999))
while True:
cmd = input(">>:").strip()
if len(cmd) == 0:
continue
if cmd.startswith("get"):
client.send(cmd.encode())
server_response = client.recv(1024)
print("server response:",server_response)
client.send(b"ready to recv file")
file_total_size = int(server_response.decode())
received_size = 0
filename = cmd.split()[1]
f = open(filename + ".new","wb")
m = hashlib.md5()
while received_size <file_total_size:
data =client.recv(1024)
recevied_size += len(data)
m.updata(data)
f.write(data)
#print(flie_total_size,received_size)
else:
new_file_md5 = m.hexdigest()
print("file recv done",received_size,file_total_size)
f.close()
server_file_md5 = client.recv(1024)
print("server file md5:",server_file_md5)
print("client file md5:",new_file_md5)
server.close()
基本服务端
import socketserver
class MyTCPHandler(socketserver.BaseRequestHandler):
"""
The request handler class for our server.
It is instantiated once per connection to the server, and must
override the handle() method to implement communication to the
client.
"""
def handle(self):
# self.request is the TCP socket connected to the client
while True:
try:
self.data = self.request.recv(1024).strip()
print("{} wrote:".format(self.client_address[0]))
print(self.data)
# if not self.data:#客户端断了
# print(self.client_address,"断开了")
# break
# just send back the same data, but upper-cased
self.request.sendall(self.data.upper())
except ConnectionResetError as e:
print("err",e)
break
if __name__ == "__main__":
HOST, PORT = "localhost", 9999
# Create the server, binding to localhost on port 9999
server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)
# Activate the server; this will keep running until you
# interrupt the program with Ctrl-C
server.serve_forever()
FTP上传client
import socket
import hashlib
import os
import json
class FtpClient(object):
def __init__(self):
self.client = socket.socket()
def help(self):
msg = ‘‘‘
ls
pwd
cd ../..
get filename
put filename
‘‘‘
def connect(self,ip,port):
self.client.connect((ip,port))
def interactive(self):
#self.authenticate()
while True:
cmd = input(">>")
if len(cmd) == 0:
continue
cmd_str = cmd.split()[0]
if hasattr(self,"cmd_%s"%cmd_str):
func = getattr(self,"cmd_%s"%cmd_str)
func(cmd)
else:
self.help()
def cmd_put(self,*args):
cmd_split = args[0].split()
if len(cmd_split) > 1:
filename = cmd_split[1]
if os.path.isfile(filename):
filesize = os.stat(filename).st_size
msg_dic ={
"action":"put",
"filename":filename,
"size":filesize,
"overriden":True
}
self.client.send(json.dumps(msg_dic).encode("utf-8"))
print("send",json.dumps(msg_dic).encode("utf-8"))
#防止粘包,等服务器确认
server_response = self.client.recv(1024)
f = open(filename,"rb")
for line in f:
self.client.send(line)
else:
print("file upload success...")
f.close()
else:
print(filename,"is not exist")
def cmd_get(self):
pass
ftp = FtpClient()
ftp.connect("localhost",9999)
ftp.interactive()
FTP上传server
import socketserver
import json
import os
class MyTCPHandler(socketserver.BaseRequestHandler):
"""
The request handler class for our server.
It is instantiated once per connection to the server, and must
override the handle() method to implement communication to the
client.
"""
def put(self,*args):
‘‘‘接受客户端文件‘‘‘
cmd_dic = args[0]
filename = cmd_dic["filename"]
filesize = cmd_dic["size"]
if os.path.isfile(filename):
f = open(filename+".new","wb")
else:
f = open(filename,"wb")
self.request.send(b"200 ok")
received_size = 0
while received_size < filesize:
data = self.request.recv(1024)
f.write(data)
received_size += len(data)
else:
print("file [%s] has uploaded.."%filename)
def handle(self):
# self.request is the TCP socket connected to the client
while True:
try:
self.data = self.request.recv(1024).strip()
print("{} wrote:".format(self.client_address[0]))
print(self.data)
cmd_dic = json.loads(self.data.decode())
action = cmd_dic["action"]
if hasattr(self,action):
func = getattr(self,action)
func(cmd_dic)
# if not self.data:#客户端断了
# print(self.client_address,"断开了")
# break
# just send back the same data, but upper-cased
self.request.sendall(self.data.upper())
except ConnectionResetError as e:
print("err",e)
break
if __name__ == "__main__":
HOST, PORT = "localhost", 9999
# Create the server, binding to localhost on port 9999
server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)
# Activate the server; this will keep running until you
# interrupt the program with Ctrl-C
server.serve_forever()
标签:action name binding bre comm tar tin tput pop
原文地址:https://www.cnblogs.com/yuanjun93/p/10965134.html