181114socke编程

其他 2019-02-19 19:41:20 阅读次数: 0

一、Socket Families 地址簇

  • socket.AF_UNIX
  • socket.AF_INET

  • socket.AF_INET6

    二、Socker Types

  • socket.SOCK_STREAM #tcp
  • socket.SOCK_DGRAM #udp
  • socket.SOCK_RAM
  • socket.SOCK_RDM

  • socket.SOCK_SEQPACKET

    三、socket的收发实例

  • client
# Author:Li Dongfei
import socket
client = socket.socket()  #声明socket类型，同时生成socket连接对象
client.connect(('localhost',6969))
client.send(b'hello python')
data = client.recv(1024)
print("recv:",data)
client.close()
  • server
# Author:Li Dongfei
import socket
server = socket.socket()
server.bind(('localhost',6969))  #绑定要监听的端口
server.listen()  #监听
print("开始接收")
conn,addr = server.accept()  #等待接收数据
print(conn,addr)  #conn是客户端连过来而在服务器端为其生成的一个连接实例
print("数据来了")
data = conn.recv(1024)
print("recv:",data)
conn.send(data.upper())
server.close()

发送中文示例

client.send('你好'.encode("utf-8"))

四、实现连续通信

  • client
# Author:Li Dongfei
import socket
client = socket.socket()  #声明socket类型，同时生成socket连接对象
client.connect(('localhost',6969))

while True:
    msg = input(">>: ")
    client.send(msg.encode("utf-8"))
    data = client.recv(1024)
    print("recv:",data.decode())

client.close()
  • server
# Author:Li Dongfei
import socket
server = socket.socket()
server.bind(('localhost',6969))  #绑定要监听的端口
server.listen()  #监听
print("开始接收")
conn, addr = server.accept()  # 等待接收数据
print("数据来了")
while True:
    data = conn.recv(1024)
    print("recv:",data.decode())
    conn.send(data.upper())
server.close()

五、实现客户端断开服务器不断

  • server
# Author:Li Dongfei
import socket
server = socket.socket()
server.bind(('localhost',6969))  #绑定要监听的端口
server.listen(5)  #监听,最大允许5个连接
while True:
    print("开始接收..")
    conn,addr = server.accept()  # 等待接收数据
    print("数据来了")
    while True:
        data = conn.recv(1024)
        print("recv:",data.decode())
        if not data:
            print("client has lost...")
            break
        conn.send(data.upper())
server.close()
  • client
# Author:Li Dongfei
import socket
client = socket.socket()  #声明socket类型，同时生成socket连接对象
client.connect(('localhost',6969))

while True:
    msg = input(">>: ")
    if len(msg) == 0:continue
    client.send(msg.encode("utf-8"))
    data = client.recv(1024)
    print("recv:",data.decode())

client.close()

六、ssh服务器和ssh客户端

  • client
# Author:Li Dongfei
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)  #接收命令结果的长度
    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("utf-8"))
    else:
        print("received_size:",received_size)

client.close()
  • server
# Author:Li Dongfei
import socket,os
server = socket.socket()
server.bind(('localhost',9999))
server.listen(5)

while True:
    conn,addr = server.accept()
    print("new conn:",addr)
    while True:
        data = conn.recv(1024)
        if not data:
            print("客户端已断开")
            break
        print("执行指令: ",data)
        cmd_res = os.popen(data.decode()).read()
        if len(cmd_res) == 0:
            cmd_res = "cmd has no output..."
        conn.send(str(len(cmd_res.encode())).encode("utf-8"))  #先发数据大小给客户端
        conn.send(cmd_res.encode("utf-8"))

server.close()

七、最终版本

  • sock_server_ssh.py
# Author:Li Dongfei
import socket,os
server = socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
    conn, addr = server.accept()
    print("new conn:", 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"))
        conn.send(cmd_res.encode("utf-8"))
        print("send done")
server.close()
  • sock_client_ssh
# Author:Li Dongfei
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)
    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())
    else:
        print("cmd res receive done.",received_size)
    #cmd_res = client.recv(1024)
    #print(cmd_res.decode())
client.close()

八、socket粘包

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

类似以上代码会将两条send的数据放到一个包中发出
解决方法

  • server
# Author:Li Dongfei
import socket,os
server = socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
    conn, addr = server.accept()
    print("new conn:", 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"))
        client_ack = conn.recv(1024)
        print("ack from client",client_ack.decode())
        conn.send(cmd_res.encode("utf-8"))
        print("send done")
server.close()
  • client
# Author:Li Dongfei
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())
    else:
        print("cmd res receive done.",received_size)
    #cmd_res = client.recv(1024)
    #print(cmd_res.decode())
client.close()

九、ftp server

  1. 读取文件名
  2. 检测文件是否存在
  3. 打开文件
  4. 检测文件大小
  5. 发送文件大小给客户端
  6. 等客户端确认
  7. 开始边读边发数据
  8. 发送md5
  • server
# Author:Li Dongfei
import socket,os,hashlib
server = socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
    conn, addr = server.accept()
    print("new conn:", addr)
    while True:
        data = conn.recv(1024)
        if not data:
            print("客户端已断开")
            break
        cmd,filename = data.decode().split()
        print(filename)
        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.update(line)
                conn.send(line)
            print("file md5:",m.hexdigest())
            f.close()
            conn.send(m.hexdigest().encode())
        print("send done.")
server.close()
  • client
# Author:Li Dongfei
import socket,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,"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.",file_total_size,received_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()

十、socketserver

  • 基本写法
# Author:Li Dongfei
import socketserver
class MyTCPHandler(socketserver.BaseRequestHandler):
    def handle(self):
        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
                self.request.send(self.data.upper())
            except ConnectionResetError as e:
                print("err",e)
                break
if __name__ == "__main__":
    HOST, PORT = "localhost", 9999
    server = socketserver.TCPServer((HOST, PORT), MyTCPHandler)
    server.serve_forever()
  • client
# Author:Li Dongfei
import socket
client = socket.socket()  #声明socket类型，同时生成socket连接对象
client.connect(('localhost',9999))

while True:
    msg = input(">>: ")
    client.send(msg.encode("utf-8"))
    data = client.recv(1024)
    print("recv:",data.decode())

十一、socket多线程并发

# Author:Li Dongfei
import socketserver
class MyTCPHandler(socketserver.BaseRequestHandler):
    def handle(self):
        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
                self.request.send(self.data.upper())
            except ConnectionResetError as e:
                print("err",e)
                break
if __name__ == "__main__":
    HOST, PORT = "localhost", 9999
    server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)  #多线程
    server.serve_forever()

猜你喜欢

转载自www.cnblogs.com/L-dongf/p/10402929.html
今日推荐
火速冲上 GitHub 热榜 —— 开源编程语言、框架哪有这么可爱？
北京人形机器人创新中心发布全球首个纯电驱拟人奔跑的全尺寸人形机器人“天工”
LFOSSA 源来如此公开课 | 掌握云原生未来：CNCF 认证全面攻略与备考秘籍
国产云输入法——仅华为无云端数据上传安全问题
开源日报 | 工业开源项目OGG 1.0；姐姐，你要和我一起配置火狐吗；苹果AI遥遥落后？Fedora 40
开放签电子签章：停止新增，优化体验，前进更进（五一假期前工作）
开源日报 | 中学生开源前端动画引擎；全球首个Llama3 8B中文版开源模型；联想电脑恐出局；Linus讽刺AI炒作
周排行
浏览器对同一域名进行请求的最大并发连接数
React Hook之自定义Hook
【转】MyBatis缓存机制
-Java-泛型
自动化测试常用脚本-发送邮件
LeetCode#859: Buddy Strings
java、Python处理字符串
第二篇の博客
Hadoop伪分布式环境安装
SQL Server进阶（十一）临时表、表变量
每日归档
更多
2024-04-27(56)
2024-04-26(39)
2024-04-25(22)
2024-04-24(36)
2024-04-23(26)
2024-04-22(39)
2024-04-21(0)
2024-04-20(6)
2024-04-19(5)
2024-04-18(0)

代码天地 © 2018 codetd.com

境外服务器   最新电视剧2022