python send 案例_python--socket实例

模拟SSH--client:

# Author:Tim Gu

import socket

client = socket.socket()

client.connect(('localhost',6969))

while True:

cmd = input(">>").strip()

if len(cmd) == 0:continue

client.send(cmd.encode("utf-8"))

cmd_res_size = client.recv(1024)

client.send(b"start_trans")

sev_file_size = int(cmd_res_size.decode())

#print('sev_file_size',sev_file_size,type(sev_file_size))

received_size = 0

received_data = b''

while received_size < sev_file_size:

data = client.recv(1024)

#print("收到数据！")

received_size += len(data)

received_data += data

#print('cli_file_size',received_data)

else:

#print("receive done...",received_size)

print(received_data.decode())

client.close()

模拟SSH--server:

# Author:Tim Gu

import socket,os,time

server = socket.socket()

server.bind(('localhost',6969))

server.listen()

while True:

print("beging waiting request...")

conn,addr = server.accept()

print("new conn:",addr)

while True:

print("request comming!",addr)

data = conn.recv(1024)

if not data:

print("客户端已断开")

break

print("执行指令：",data)

cmd = data.decode()

cmd_res = os.popen(cmd).read()

print("before send",len(cmd_res.encode('utf-8')))

if len(cmd_res) == 0:

com_res = "cmd has no output..."

conn.send(str(len(cmd_res.encode('utf-8'))).encode('utf-8'))

#time.sleep(0.5)

client_ack = conn.recv(1024) #wait client confirm

conn.send(cmd_res.encode("utf-8"))

print("send done")

server.close()

socketserver简易server：

# Author:Tim Gu

import socket

import socketserver

class MyTCPHandler(socketserver.BaseRequestHandler):

def handle(self):

while True:

try:

print("开始监听！")

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

self.request.send(self.data.upper())

except (WindowsError,ConnectionRefusedError) as e:

print(e)

break

if __name__ == "__main__":

HOST,PORT="localhost",9999

server = socketserver.ThreadingTCPServer((HOST,PORT),MyTCPHandler)

server.serve_forever()

socketserver简易client:

# Author:Tim Gu

import socket

def myquit():

choose = input("是否退出？'y' or 'n':").strip()

if choose == 'y':

exit("退出！")

else:

pass

client = socket.socket()

client.connect(('localhost',9999))

while True:

msg = input(">>").strip()

if len(msg) == 0:continue

client.send(msg.encode('utf-8'))

data = client.recv(10240)

print("recv:",data.decode())

myquit()

client.close()

传文件server:

# Author:Tim Gu

import socket,os,time,hashlib

server = socket.socket()

server.bind(('localhost',6969))

server.listen()

while True:

print("beging waiting request...")

conn,addr = server.accept()

print("new conn:",addr)

while True:

print("request comming!",addr)

data = conn.recv(1024)

if not data:

print("客户端已断开")

break

cmd,filename = data.decode().split()

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('utf-8')) #send file size

conn.recv(1024) #wait for ack

for line in f:

m.update(line)

conn.send(line)

print("file md5",m.hexdigest())

f.close()

conn.send(m.hexdigest().encode())

print("send done")

server.close()

传文件client:

# Author:Tim Gu

import socket,hashlib

client = socket.socket()

client.connect(('localhost',6969))

while True:

cmd = input(">>").strip()

if len(cmd) == 0:continue

if cmd.startswith('get'):

client.send(cmd.encode('utf-8'))

server_response = client.recv(1024)

print('server response:',server_response)

client.send(b'start to trans')

file_total_size = int(server_response.decode())

received_size = 0

file_name = cmd.split()[1]

f = open(file_name+".new",'wb')

m = hashlib.md5()

while received_size < file_total_size:

if file_total_size - received_size >= 1024:

size = 1024

else:

size = file_total_size - received_size

data = client.recv(size)

received_size += len(data)

m.update(data)

f.write(data)

#print(file_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)

client.close()

本周作业：

client:

import socket

import os,json

import hashlib

class FtpClient(object):

def __init__(self):

self.client = socket.socket()

def cmd_help(self,args):

msg = '''

ls

pwd

cd

get filename

put filename

'''

print(msg)

def connect(self,ip,port):

self.client.connect((ip,port))

def interactive(self):

self.auth()

while True:

cmd = input(">>").strip()

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]

os.path.isfile(filename)

filesize = os.stat(filename).st_size

msg_dic = {

'action': 'put',

'filename': filename,

'filesize': filesize,

'overridden': True

}

self.client.send(json.dumps(msg_dic).encode('utf-8'))

server_response = self.client.recv(1024)

f = open(filename,'rb')

m = hashlib.md5()

for line in f:

m.update(line)

self.client.send(line)

else:

print("file upload success...")

f.close()

self.client.send(m.hexdigest().encode())

sev_md5_check = self.client.recv(1024)

print(sev_md5_check.decode())

else:

print('file is not exist')

def cmd_get(self):

pass

def auth(self):

pass

class Account(object):

def __init__(self):

pass

ftp = FtpClient()

ftp.connect('localhost',9999)

ftp.interactive()

server:

# Author:Tim Gu

import socket,hashlib

import socketserver,json,os

class MyTCPHandler(socketserver.BaseRequestHandler):

def put(self,*args):

cmd_dic = args[0]

filename = cmd_dic['filename']

filesize = cmd_dic['filesize']

if os.path.isfile(filename):

f = open(filename + ".new",'wb')

else:

f = open(filename,'wb')

self.request.send(b'200,ok')

recived_size = 0

m = hashlib.md5()

while recived_size < filesize:

data = self.request.recv(1024)

m.update(data)

f.write(data)

recived_size += len(data)

else:

f.close()

print("file [%s] has uploaded..."%filename)

recived_md5 = m.hexdigest()

source_md5 = self.request.recv(1024).decode()

if recived_md5 == source_md5:

self.request.send('MD5校验成功'.encode('utf-8'))

else:

self.request.send('MD5校验失败'.encode('utf-8'))

def handle(self):

while True:

try:

print("开始监听！")

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

cmd_dic = json.loads(self.data.decode())

action = cmd_dic['action']

if hasattr(self,action):

func = getattr(self,action)

func(cmd_dic)

except ConnectionRefusedError as e:

print(e)

break

if __name__ == "__main__":

HOST,PORT="localhost",9999

server = socketserver.ThreadingTCPServer((HOST,PORT),MyTCPHandler)

server.serve_forever()