网络编程 socket

socket 又称为 “套接字”，应用程序通常通过“套接字”向网络发出请求或应答网络请求。使主机间或者一台服务器上的进程可以通信。

实例说明：

客户端：

import socket

sk = socket.socket() #创建套接字

addr = ('127.0.0.1',8888)
sk.bind(addr) # 以元组的形式绑定地址到套接字（ip，port）

sk.listen(3) # 打开tcp监听，表示在结束连接前队列中支持挂起的最大连接数
conn,address=sk.accept() # 被动接受客户端的连接

inp = input(">>>")
conn.send(bytes(inp,'utf8'))  # 发送tcp数据，python3支持字节数据的传输
conn.close() # 关闭单个（本次）套接字的连接。sk.close()关闭所有套接字的连接

 服务端：

import socket

sk = socket.socket()  # 创建套接字
addr = ('127.0.0.1',8888)
sk.connect(addr) # 以元组的形式主动连接server端

data=sk.recv(1024) # 接收TCP数据，1024是指定的最大数据量
print(str(data,'utf8'))

sk.close()  # 关闭套接字

socket对象的内置方法：

函数	描述
服务器端套接字
s.bind()	绑定地址（host,port）到套接字， 在AF_INET下,以元组（host,port）的形式表示地址。
s.listen()	开始TCP监听。backlog指定在拒绝连接之前，操作系统可以挂起的最大连接数量。该值至少为1，大部分应用程序设为5就可以了。
s.accept()	被动接受TCP客户端连接,(阻塞式)等待连接的到来
客户端套接字
s.connect()	主动初始化TCP服务器连接，。一般address的格式为元组（hostname,port），如果连接出错，返回socket.error错误。
s.connect_ex()	connect()函数的扩展版本,出错时返回出错码,而不是抛出异常
公共用途的套接字函数
s.recv()	接收TCP数据，数据以字符串形式返回，bufsize指定要接收的最大数据量。flag提供有关消息的其他信息，通常可以忽略。
s.send()	发送TCP数据，将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。
s.sendall()	完整发送TCP数据，完整发送TCP数据。将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。
s.recvfrom()	接收UDP数据，与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。
s.sendto()	发送UDP数据，将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数。
s.close()	关闭套接字
s.getpeername()	返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。
s.getsockname()	返回套接字自己的地址。通常是一个元组(ipaddr,port)
s.setsockopt(level,optname,value)	设置给定套接字选项的值。
s.getsockopt(level,optname[.buflen])	返回套接字选项的值。
s.settimeout(timeout)	设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如connect()）
s.gettimeout()	返回当前超时期的值，单位是秒，如果没有设置超时期，则返回None。
s.fileno()	返回套接字的文件描述符。
s.setblocking(flag)	如果flag为0，则将套接字设为非阻塞模式，否则将套接字设为阻塞模式（默认值）。非阻塞模式下，如果调用recv()没有发现任何数据，或send()调用无法立即发送数据，那么将引起socket.error异常。
s.makefile()	创建一个与该套接字相关连的文件

循环聊天：

# Author : Adair
# -*- coding:utf-8 -*-


import socket

sk = socket.socket() #创建套接字

addr = ('127.0.0.1',8888)
sk.bind(addr) # 以元组的形式绑定地址到套接字（ip，port）

sk.listen(3) # 打开tcp监听，表示在结束连接前队列中支持挂起的最大连接数


while True:
    conn, address = sk.accept()  # 被动接受客户端的连接
    data = conn.recv(1024)
    if not data:
        continue
    print(str(data,"utf8"))

    inp = input(">>>")
    if inp == "exit": break
    conn.send(bytes(inp,'utf8'))  # 发送tcp数据，python3支持字节数据的传输

conn.close() # 关闭单个（本次）套接字的连接。sk.close()关闭所有套接字的连接

# Author : Adair
# -*- coding:utf-8 -*-
import socket

sk = socket.socket()  # 创建套接字
addr = ('127.0.0.1',8888)
sk.connect(addr) # 以元组的形式主动连接server端

while True:
    inp = input(">>>")
    if inp == "exit":break
    sk.send(bytes(inp,"UTF8"))

    data=sk.recv(1024) # 接收TCP数据，1024是指定的最大数据量
    if not data: print("[ x ] The other party has been taken offline!");break
    print(str(data,'utf8'))

sk.close()  # 关闭套接字

执行远程命令：

# Author : Adair
# -*- coding:utf-8 -*-


import socket,os

sk = socket.socket() #创建套接字

addr = ('127.0.0.1',8888)
sk.bind(addr) # 以元组的形式绑定地址到套接字（ip，port）

sk.listen(3) # 打开tcp监听，表示在结束连接前队列中支持挂起的最大连接数

while True:
    conn, address = sk.accept()  # 被动接受客户端的连接
    print("[ X ] %s has been connected. Port number:%s" %(address[0],address[1]))
    while True:
        data = conn.recv(1024)
        if not data:
            break
        CmdReturn=os.popen(str(data,'utf8')).read()
        CmdLen=bytes(str(len(CmdReturn)),'utf8') # int字符型不能直接转换成bytes类型
        print("[ X ] The orders executed are:%s,Length is:%s"%(data,CmdLen))
        conn.sendall(CmdLen)  # 只能发送bytes类型的数据
        conn.sendall(bytes(CmdReturn,'utf8'))

conn.close() # 关闭单个（本次）套接字的连接。sk.close()关闭所有套接字的连接

# Author : Adair
# -*- coding:utf-8 -*-
import socket

sk = socket.socket()  # 创建套接字
addr = ('127.0.0.1',8888)
sk.connect(addr) # 以元组的形式主动连接server端

while True:
    inp = input(">>>")
    if inp == "exit":break
    sk.send(bytes(inp,"utf8"))

    data_len=int(str(sk.recv(1024),'utf8'))# 接收的是data的字符长度
    if data_len == 0:print("[ X ] Command execution failed!")
    data=bytes() # 定义一个空的bytes，用于接收合并TCP数据
    while len(data) < int(data_len):
        data_recv=sk.recv(1024) # 接收TCP数据，1024是指定的最大数据量
        data+=data_recv
        if not data: print("[ x ] The other party has been taken offline!");break
    print(str(data,'utf8'))

sk.close()  # 关闭套接字

 上传/下载：

# Author : Adair
# -*- coding:utf-8 -*-

import  socket,os

address = ("127.0.0.1",1111)

sk = socket.socket()
sk.bind(address)
sk.listen(3)

conn,addr=sk.accept()

TargetPath = str(conn.recv(1024),'utf8')
FileSize = int(str(conn.recv(1024),'utf8'))
print(type(FileSize))
FileData= conn.recv(FileSize)




TargetFilePath,TargetFileName=os.path.split(TargetPath)
if not os.path.exists(TargetFilePath):
    os.makedirs(TargetFilePath)
print("The target directory has been created!")

with open(TargetPath,'wb') as file:
    file.write(FileData)

# Author : Adair
# -*- coding:utf-8 -*-

import socket,os

MenuList=['Upload','Download','Exit']  # 菜单
for i,element in enumerate(MenuList,start=1): # 菜单内容输出
    print("\t%s、%s"%(i,element))

def switch(para):     # 菜单项调用功能函数执行
    return {
        '1':Upload,
        '2':Download,
        '3':exit
    }.get(para,'The selected parameter does not exist')

class SocketLib(): # socket 接口
    def __init__(self,TargetPath,FileData,FileSize):

        self.TargetPath=TargetPath
        self.FileData=FileData
        self.FileSize=FileSize

        self.address = ('127.0.0.1', 1111)
        self.sk = socket.socket()
        self.sk.connect(self.address)

    def SocketSend(self):
        self.sk.send(bytes(self.TargetPath,'utf8'))
        self.sk.send(str(self.FileSize).encode())
        self.sk.send(bytes(self.FileData))




def Upload():
    SourcePath = input("Please enter the source path:")
    TargetPath = input("Please enter the target path:")
    OpenFile=open(SourcePath,'rb')
    FileData=OpenFile.read()
    FileSize=os.path.getsize(SourcePath)

    SkLib=SocketLib(TargetPath,FileData,FileSize)
    SkLib.SocketSend()
def Download():
    print("download!!!")


ActionChoice=input("Please select the action you want to operate:")
switch(ActionChoice)()
# sk.send(bytes("test","utf8"))

 

socketserver是socket的升级版，是socket的封装，他支持多线程的并发。socketserver中包含了两种类，一种为服务类，一种是服务请求类，前一种是创建连接的工程（绑定、监听、运行），后者主要处理事物逻辑（用户发过来的数据）

server：

# Author : Adair
# -*- coding:utf-8 -*-

import socketserver

class MyTCPServer(socketserver.BaseRequestHandler): # 创建一个继承类

    def handle(self):  # 重写父类的handle，此方法中处理业务逻辑
        print("******")
        while True:
            conn = self.request # self.request 为客户端的连接句柄
            print(self.client_address) # self.client_address 为客户端的ip、port
            while True:
                ClientData=conn.recv(1024)
                print(str(ClientData,"UTF8"))
                InP=input(">>>")
                conn.sendall(str(InP).encode())
            conn.close()


if __name__ == "__main__":
    server = socketserver.ThreadingTCPServer(("127.0.0.1",8090),MyTCPServer) # 选择合适的服务类型（ThreadingTCPServer）创建连接
    server.serve_forever()  # 处理无限的请求

client：

# Author : Adair
# -*- coding:utf-8 -*-

import socket

addr = ("127.0.0.1",8090)
sk = socket.socket()
sk.connect(addr)

while True:
    msg = input(">>>")
    sk.send(str(msg).encode())
    data=sk.recv(1024)
    print("data:",str(data,"utf8"))
sk.close()