回射
SERVER
#!/usr/bin/python3 #_*_ coding:utf-8 _*_ import socket,os,time import socketserver import time import threading import random import sys import socket import getopt import os import re import shutil import platform class ServerHandle(socketserver.BaseRequestHandler): def handle(self): conn = self.request conn.sendall(bytes("Hello I'm robot!",encoding="utf-8")) while True: ret_bytes = conn.recv(1024) ret_str = str(ret_bytes,encoding="utf-8") if ret_str == "q": break conn.sendall(bytes("You Send:" + ret_str,encoding="utf-8")) def daemonize(stdin='/dev/null', stdout='/dev/null', stderr='/dev/null'): pid = os.fork() if pid > 0: sys.exit(0) os.chdir("/") os.umask(0) os.setsid() pid = os.fork() if pid > 0: sys.exit(0) for f in sys.stdout, sys.stderr: f.flush() si = open(stdin, 'r') so = open(stdout, 'a+') #se = open(stderr, 'a+', 0) os.dup2(si.fileno(), sys.stdin.fileno()) os.dup2(so.fileno(), sys.stdout.fileno()) #os.dup2(se.fileno(), sys.stderr.fileno()) if __name__ == "__main__": daemonize('/dev/null', '/home/share/kitking/stdout.log', '/home/share/kitking/error.log') server = socketserver.ThreadingTCPServer(("192.168.199.XXX",6969),ServerHandle) server.serve_forever()
CLIENT
#!/usr/bin/python3 #_*_ coding:utf-8 _*_ import socket,os,time import socketserver host = "XXX.eicp.net" ip = socket.gethostbyname(host) port = 6969 obj = socket.socket() obj.connect((ip,port)) ret_bytes = obj.recv(1024) ret_str = str(ret_bytes,encoding="utf-8") print(ret_str) while True: inp = input(">>") if inp == "q": obj.sendall(bytes(inp,encoding="utf-8")) break else: obj.sendall(bytes(inp, encoding="utf-8")) ret_bytes = obj.recv(1024) ret_str = str(ret_bytes,encoding="utf-8") print(ret_str)
SSH模拟
SERVER
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()
CLIENT
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()
文件传送
SERVER
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
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()
FTP模式
SERVER
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()
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()
上传文件
import socket sk = socket.socket() sk.bind(("127.0.0.1",8080)) sk.listen(5) while True: conn,address = sk.accept() conn.sendall(bytes("欢迎光临我爱我家",encoding="utf-8")) size = conn.recv(1024) size_str = str(size,encoding="utf-8") file_size = int(size_str) conn.sendall(bytes("开始传送", encoding="utf-8")) has_size = 0 f = open("db_new.jpg","wb") while True: if file_size == has_size: break date = conn.recv(1024) f.write(date) has_size += len(date) f.close() 客户端 import socket import os obj = socket.socket() obj.connect(("127.0.0.1",8080)) ret_bytes = obj.recv(1024) ret_str = str(ret_bytes,encoding="utf-8") print(ret_str) size = os.stat("yan.jpg").st_size obj.sendall(bytes(str(size),encoding="utf-8")) obj.recv(1024) with open("yan.jpg","rb") as f: for line in f: obj.sendall(line)