python3之路 基础-socket

网络传输原理

socket起源于Unix，而Unix/Linux基本哲学之一就是“一切皆文件”，对于文件用【打开】【读写】【关闭】模式来操作。socket就是该模式的一个实现，socket即是一种特殊的文件，一些socket函数就是对其进行的操作（读/写IO、打开、关闭）

socket概念

socket通常也称作"套接字"，用于描述IP地址和端口，是一个通信链的句柄，应用程序通常通过"套接字"向网络发出请求或者应答网络请求。
socket起源于Unix，而Unix/Linux基本哲学之一就是“一切皆文件”，对于文件用【打开】【读写】【关闭】模式来操作。socket就是该模式的一个实现，socket即是一种特殊的文件，一些socket函数就是对其进行的操作（读/写IO、打开、关闭）

socket和file的区别：

• file模块是针对某个指定文件进行【打开】【读写】【关闭】
• socket模块是针对 服务器端 和 客户端Socket 进行【打开】【读写】【关闭】

socket本质上就是在2台网络互通的电脑之间，架设一个通道，两台电脑通过这个通道来实现数据的互相传递。 我们知道网络 通信 都 是基于 ip+port 方能定位到目标的具体机器上的具体服务，操作系统有0-65535个端口，每个端口都可以独立对外提供服务，如果 把一个公司比做一台电脑 ，那公司的总机号码就相当于ip地址， 每个员工的分机号就相当于端口， 你想找公司某个人，必须 先打电话到总机，然后再转分机 。

基本原理

建立一个socket必须至少有2端， 一个服务端，一个客户端， 服务端被动等待并接收请求，客户端主动发起请求， 连接建立之后，双方可以互发数据。

Socket对象

sk = socket.socket(socket.AF_INET,socket.SOCK_STREAM,0)

参数	描述
socket.AF_INET	IPv4（默认）
socket.AF_INET6	IPv6
socket.AF_UNIX	只能够用于单一的Unix系统进程间通信

参数二：类型

参数	描述
socket.SOCK_STREAM	流式socket , for TCP （默认）
socket.SOCK_DGRAM	数据报式socket , for UDP
socket.SOCK_RAW	原始套接字，普通的套接字无法处理ICMP、IGMP等网络报文，而SOCK_RAW可以；其次，SOCK_RAW也可以处理特殊的IPv4报文；此外，利用原始套接字，可以通过IP_HDRINCL套接字选项由用户构造IP头。
socket.SOCK_RDM	是一种可靠的UDP形式，即保证交付数据报但不保证顺序。SOCK_RAM用来提供对原始协议的低级访问，在需要执行某些特殊操作时使用，如发送ICMP报文。SOCK_RAM通常仅限于高级用户或管理员运行的程序使用。
socket.SOCK_SEQPACKET	可靠的连续数据包服务

Socket类方法

方法	描述
s.bind(address)	将套接字绑定到地址。address地址的格式取决于地址族。在AF_INET下，以元组（host,port）的形式表示地址。
sk.listen(backlog)	开始监听传入连接。backlog指定在拒绝连接之前，可以挂起的最大连接数量。
sk.setblocking(bool)	是否阻塞（默认True），如果设置False，那么accept和recv时一旦无数据，则报错。
sk.accept()	接受连接并返回（conn,address）,其中conn是新的套接字对象，可以用来接收和发送数据。address是连接客户端的地址。
sk.connect(address)	连接到address处的套接字。一般，address的格式为元组（hostname,port）,如果连接出错，返回socket.error错误。
sk.connect_ex(address)	同上，只不过会有返回值，连接成功时返回 0 ，连接失败时候返回编码，例如：10061
sk.close()	关闭套接字连接
sk.recv(bufsize[,flag])	接受套接字的数据。数据以字符串形式返回，bufsize指定最多可以接收的数量。flag提供有关消息的其他信息，通常可以忽略。
sk.recvfrom(bufsize[.flag])	与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。
sk.send(string[,flag])	将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。即：可能未将指定内容全部发送。
sk.sendall(string[,flag])	将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。内部通过递归调用send，将所有内容发送出去。
sk.sendto(string[,flag],address)	将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数。该函数主要用于UDP协议。
sk.settimeout(timeout)	设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。
sk.getpeername()	返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。
sk.getsockname()	返回套接字自己的地址。通常是一个元组(ipaddr,port)
sk.fileno()	套接字的文件描述符

socket编程思路

• TCP服务端
  • 创建套接字，绑定套接字到本地IP与端口
  • 开始监听连接
  • 进入循环，不断接受客户端的连接请求
  • 然后接收传来的数据，并发送给对方数据
  • 传输完毕后，关闭套接字
• TCP客户端
  • 创建套接字，连接远端地址
  • 连接后发送数据和接收数据
  • 传输完毕后，关闭套接字

一个简单的socket实现逻辑

• 服务端

##服务端
__author__ = 'GPF'
import socket

server = socket.socket()
server.bind(("localhost", 9999))
server.listen(5)
conn, addr = server.accept()
client_data = conn.recv(1024)
print('i Receive the Data',client_data.decode().upper())
conn.close()
server.close()

• 客户端

##客户端
__author__ = 'GPF'
import socket

client = socket.socket()
client.connect(("localhost", 9999))
client.send(b"i am gpf")
res = client.recv(1024)
print(res.decode())
client.close()

实现一般linux命令的socket(防止念包)

服务端

__author__ = "GPF"
import socket, os, time

server = socket.socket()
server.bind(('0.0.0.0', 9999))
server.listen()

while True:
    try :
        conn, addr = server.accept()
        print("new conn:", addr)
        while True:
            print("等待新指令")
            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.recv(1024) # 接收客户端 发送过来的内容，内容为：'解除念包请求。。。'
            conn.send(cmd_res.encode("utf-8"))# 发送 命令 结果给客户端
            print("send done")
    except ConnectionResetError as e:
        print("err",e)
        break
server.close()

客户端

__author__ = "GPF"
import socket

client = socket.socket()
client.connect(('0.0.0.0', 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())
    received_size = 0
    received_data = b''
    while received_size < int(cmd_res_size.decode()):
        data = client.recv(1024) # 接收服务端的结果数据 一次只接收1024大小的数据
        received_size += len(data)  # 每次收到的有可能小于1024，所以必须用len判断
        received_data += data # 将接受数据拼接起来
    else:
        print("cmd res receive done...", received_size)
        print(received_data.decode())

client.close()

SocketServer的使用

创建一个SocketServer的步骤

1. 你必须自己创建一个请求处理类，并且这个类必须继承BaseRequestHandler，并且重写父类中的handle()方法
2. 你必须实例化TCPServer,并且传递Server ip 和 你上面创建的请求处理类给这个TCPServer。
3. 然后你可以调用server的handle_request()或者server_forever()方法在你的项目中
   server.handle_request()#只处理一个请求
   server.server_forever()#处理多个请求，永远执行
4. 关闭socket,socket.close;

• 简单的socketServer
  服务端

# author GPF
# 功能
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):  # 请求处理类，所有请求的交互都是在handle里执行的
        # self.request is the TCP socket connected to the client
        while True:
            try:
                self.data = self.request.recv(1024).strip()  # 每一个请求都会实例化MyTCPHandler(socketserver.BaseRequestHandler):
                print("{} wrote:".format(self.client_address[0]))
                print(self.data)
                # just send back the same data, but upper-cased
                self.request.sendall(self.data.upper())  # sendall是重复调用send.
            except ConnectionResetError as e:
                print("客户端已断开", e)
                break

if __name__ == "__main__":
    HOST, PORT = "localhost", 9999
    # server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)  # 多并发线程
    server = socketserver.TCPServer((HOST, PORT), MyTCPHandler)
    server.serve_forever()

客户端

__author__ = 'GPF'
import socket

client = socket.socket()
client.connect(("localhost", 9999))

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