网络编程socket

2022-11-06 16:01:20

socket

socket解释

socket通常也称作"套接字"，用于描述IP地址和端口，是一个通信链的句柄。

应用程序通常通过"套接字"向网络发出请求或者应答网络请求。说白了就是一种通信机制。

socket起源于Unix，而Unix/Linux基本哲学之一就是“一切皆文件”，对于文件用【打开】【读写】【关闭】模式来操作。

socket就是该模式的一个实现，socket即是一种特殊的文件，一些socket函数就是对其进行的操作（读/写IO、打开、关闭）

socket和file的区别：

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

更多功能

sk.bind(address)

　　s.bind(address) 将套接字绑定到地址。address地址的格式取决于地址族。在AF_INET下，以元组（host,port）的形式表示地址。

sk.listen(backlog)

　　开始监听传入连接。backlog指定在拒绝连接之前，可以挂起的最大连接数量。

backlog等于5，表示内核已经接到了连接请求，但服务器还没有调用accept进行处理的连接个数最大为5
这个值不能无限大，因为要在内核中维护连接队列

sk.setblocking(bool)

　　是否阻塞（默认True），如果设置False，那么accept和recv时一旦无数据，则报错。

sk.accept()

　　接受连接并返回（conn,address）,其中conn是新的套接字对象，可以用来接收和发送数据。address是连接客户端的地址。

　　接收TCP 客户的连接（阻塞式）等待连接的到来

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表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如 client 连接最多等待5s ）

sk.getpeername()

　　返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。

sk.getsockname()

　　返回套接字自己的地址。通常是一个元组(ipaddr,port)

sk.fileno()

　　套接字的文件描述符

s.bind(address) 将套接字绑定到地址。address地址的格式取决于地址族。在AF_INET下，以元组（host,port）的形式表示地址。

s.listen(backlog)   开始监听传入连接。backlog指定在拒绝连接之前，操作系统可以挂起的最大连接数量。该值至少为1，大部分应用程序设为5就可以了。

s.connect(address)  连接到address处的套接字。一般，address的格式为元组（hostname,port），如果连接同一台机器上的服务器，可以将hostname设为‘localhost’。如果连接出错，返回socket.error错误。

s.connect_ex(adddress)  功能与connect(address)相同，但是成功返回0，失败返回errno的值。

s.accept() 接受连接并返回（conn,address）,其中conn是新的套接字对象，可以用来接收和发送数据。address是连接客户端的地址。

s.close()  关闭套接字。

s.fileno()  返回套接字的文件描述符。

s.getpeername() 返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。

s.getsockname()  返回套接字自己的地址。通常是一个元组(ipaddr,port)

s.getsockopt(level,optname[.buflen]) 返回套接字选项的值。

s.gettimeout() 返回当前超时期的值，单位是秒，如果没有设置超时期，则返回None。

s.recv(bufsize[,flag])  接受套接字的数据。数据以字符串形式返回，bufsize指定要接收的最大数据量。flag提供有关消息的其他信息，通常可以忽略。

s.recvfrom(bufsize[.flag])  与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。

s.send(string[,flag])  将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。

s.sendall(string[,flag])  将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。

s.sendto(string[,flag],address)  将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数。该函数主要用于UDP协议。

s.setblocking(flag)  如果flag为0，则将套接字设为非阻塞模式，否则将套接字设为阻塞模式（默认值）。非阻塞模式下，如果调用recv()没有发现任何数据，或send()调用无法立即发送数据，那么将引起socket.error异常。

s.setsockopt(level,optname,value)   设置给定套接字选项的值。

s.settimeout(timeout)   设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如connect()）普通的非套接字实例的函数

getdefaulttimeout()返回默认的套接字超时时间（以秒为单位）。None表示不设置任何超时时间。

gethostbyname(hostname)   将主机名（如“www.baidu.com”）转换为IPv4地址，IP地址将以字符串的形式返回，如“8.8.8.8”。不支持IPv6

gethostname() 返回本地机器的主机名。

更多功能

建立连接

单一的服务端与客户端建立连接

server 服务端

import socket

sk = socket.socket()

sk.bind(('127.0.0.1', 9999,))

sk.listen(5)

while True:

    conn,address = sk.accept()

    print(address,conn)

client 客户端

import socket

obj = socket.socket()

obj.connect(('127.0.0.1', 9999,))

obj.close()

实例：

（聊天机器人）

服务端：接受客户端的请求并返回给客户端数据。完成对话（聊天机器人）

import socket

sk = socket.socket()

sk.bind(('127.0.0.1', 9999,))

sk.listen(5)

while True:

    conn,address = sk.accept()

    conn.sendall(bytes('欢迎致电老男孩', 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(ret_str + "你好", encoding='utf-8'))

客户端：发送内容服务端会返回数据

import socket

obj = socket.socket()

obj.connect(('127.0.0.1', 9999,))

ret_bytes = obj.recv(1024)

ret_str = str(ret_bytes, encoding='utf')

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 = str(obj.recv(1024), encoding='utf-8')

        print(ret)

obj.close()

FTP上传

服务端

　　发送返回值，防止粘包

import socket

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

sk.bind(('127.0.0.1',9999,))  #绑定ip和端口，是元组

sk.listen(5)  #监听链接;数值是表示一次性最多监听5个

while True:

    #连接，客户端地址信息

    conn,address =sk.accept()  #在此阻塞，等待客户端的发送请求。accept阻塞

    conn.sendall(bytes("欢迎登录老男孩FTP", encoding='utf-8'))  #sendall表示发送数据，给客户端（只能发送字节）

    #先接收文件大小，然后再开始接收

    file_size = str(conn.recv(1024), encoding='utf-8')  #recv接受数据

    conn.sendall(bytes("ack",encoding='utf-8'))   ###发送一个返回值，进行一次交互，防止粘包

    # print(file_size)

    total_size = int(file_size)  #文件总大小

    has_recv = 0                 #已经接收的文件大小

    f = open('new.png','wb')

    #接收文件内容，直到获取完毕

    while True:

        if total_size == has_recv:

            break

        data = conn.recv(1024)  #接收

        f.write(data)

        has_recv += len(data)

    f.close()  #关闭服务端

客户端

　　接受服务端发来的返回值，防止粘包

import socket

import os

obj = socket.socket()  #创建客户端套接字

obj.connect(('127.0.0.1', 9999))  #尝试连接服务器

ret_bytes = obj.recv(1024)  #在此阻塞，等待服务端发送数据。recv接收数据

ret_str = str(ret_bytes, encoding='utf-8')

print(ret_str)

size = os.stat('f.png').st_size

obj.sendall(bytes(str(size),encoding='utf-8')) #发送

obj.recv(1024)   ####接收服务端发来的返回值，防止粘包。

with open('f.png', 'rb') as f:

    for line in f:

        obj.sendall(line)  #发送数据

obj.close()  #关闭客户端

IO多路复用

I/O多路复用指：通过一种机制，可以监视多个描述符，一旦某个描述符就绪（一般是读就绪或者写就绪），能够通知程序进行相应的读写操作。

服务端

import socket

sk1 = socket.socket()

sk1.bind(('127.0.0.1',8801))

sk1.listen()

sk2 = socket.socket()

sk2.bind(('127.0.0.1',8802))

sk2.listen()

sk3 = socket.socket()

sk3.bind(('127.0.0.1',8803))

sk3.listen()

inputs = [sk1,sk2,sk3, ]

import select

while True:

    # #[sk1,sk2,sk3],selsct内部自动监听sk1,sk2,sk3三个对象，一旦某个句柄发生变化

    # #如果有连接 sk1

    # #r_list = [sk1,sk2,sk3]

    # r_list,w_list,e_list = select.select(inputs,[],[],1)

    # for sk in r_list:

    #     #每一个连接对象

    #     conn,address = sk.accept()

    #     conn.sendall(bytes("hello",encoding="utf-8"))

    #     conn.close()

    r_list, w_list, e_list = select.select(inputs, [], inputs, 1)

    print('正在监听的socket对象%d' % len(inputs))

    print(r_list)

    for sk1_or_conn in r_list:

        #每一个连接对象

        if sk1_or_conn == sk1:

            #表示有新用户来连接

            conn, address = sk1_or_conn.accept()

            inputs.append(conn)

        else:

            #有老用户发消息

            try:

                data_bytes = sk1_or_conn.recv(1024)

                data_str = str(data_bytes,encoding="utf-8")

                sk1_or_conn.sendall(bytes(data_str+"好",encoding="utf-8"))

            except Exception as ex:

                inputs.remove(sk1_or_conn)

客户端1

import socket

obj = socket.socket()

obj.connect(("127.0.0.1", 8801))

content = str(obj.recv(1024),encoding="utf-8")

print(content)

obj.close()

客户端2

import socket

obj = socket.socket()

obj.connect(("127.0.0.1", 8802))

content = str(obj.recv(1024),encoding="utf-8")

print(content)

obj.close()

读写分离

服务端

import socket

sk1 = socket.socket()

sk1.bind(('127.0.0.1',8801))

sk1.listen()

sk2 = socket.socket()

sk2.bind(('127.0.0.1',8802))

sk2.listen()

sk3 = socket.socket()

sk3.bind(('127.0.0.1',8803))

sk3.listen()

inputs = [sk1,sk2,sk3, ]

outputs = []

message_dict = {}

import select

while True:

    r_list, w_list, e_list = select.select(inputs, outputs, inputs, 1)

    print('正在监听的socket对象%d' % len(inputs))

    print(r_list)

    for sk1_or_conn in r_list:

        #每一个连接对象

        if sk1_or_conn == sk1:

            #表示有新用户来连接

            conn, address = sk1_or_conn.accept()

            inputs.append(conn)

            message_dict[conn] = []

        else:

            #有老用户发消息

            try:

                data_str = str(data_bytes, encoding="utf-8")

                message_dict[sk1_or_conn].append(data_str)

                outputs.append(sk1_or_conn)

    for conn in w_list:

        recv_str = message_dict[conn][0]

        del message_dict[conn][0]

        conn.sendall(bytes(recv_str + "好",encoding="utf-8"))

        outputs.remove(conn)

    for sk in e_list:

        inputs.remove(sk)

客户端

import socket

obj = socket.socket()

obj.connect(("127.0.0.1", 8801))

while True:

    inp = input('>>>')

    obj.sendall(bytes(inp,encoding="utf-8"))

    ret = str(obj.recv(1024),encoding="utf-8")

    print(ret)

obj.close()

socketserver

从而实现并发处理多个客户端请求的Socket服务端
　使用：
　　1，创建类，必须继承(socketserver.BaseRequestHandler)
　　2，handle方法
　　3，server_forever 循环着等待着客户端连接

服务端

　　并发处理多个客户端请求　　

import socketserver  #进行并发处理，接受多个客户端的请求

class MyServer(socketserver.BaseRequestHandler):

    def handle(self):

        conn = self.request

        conn.sendall(bytes('欢迎致电老男孩', 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(ret_str + " 你好", encoding='utf-8'))

if __name__ == '__main__':

    server = socketserver.ThreadingTCPServer(('127.0.0.1', 9999),MyServer)  #MYserver就是创建的类

    server.serve_forever()  #循环着等待着客户端连接

客户端

　　多进程的与服务端连接

import socket

obj = socket.socket()

obj.connect(('127.0.0.1', 9999,))

ret_bytes = obj.recv(1024)

ret_str = str(ret_bytes, encoding='utf')

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 = str(obj.recv(1024), encoding='utf-8')

        print(ret)

obj.close()

ThredingTCPServer分析

ThreadingTCPServer实现的Soket服务器内部会为每个client创建一个 “线程”，该线程用来和客户端进行交互。

1,ThreadingTCPServer基础

创建一个继承自 SocketServer.BaseRequestHandler 的类
类中必须定义一个名称为 handle 的方法
启动ThreadingTCPServer

2,ThreadingTCPServer源码剖析

内部调用流程为：

启动服务端程序
执行 TCPServer.__init__ 方法，创建服务端Socket对象并绑定 IP 和端口
执行 BaseServer.__init__ 方法，将自定义的继承自SocketServer.BaseRequestHandler 的类 MyRequestHandle赋值给self.RequestHandlerClass
执行 BaseServer.server_forever 方法，While 循环一直监听是否有客户端请求到达 ...
当客户端连接到达服务器
执行 ThreadingMixIn.process_request 方法，创建一个 “线程” 用来处理请求
执行 ThreadingMixIn.process_request_thread 方法
执行 BaseServer.finish_request 方法，执行 self.RequestHandlerClass() 即：执行自定义 MyRequestHandler 的构造方法（自动调用基类BaseRequestHandler的构造方法，在该构造方法中又会调用 MyRequestHandler的handle方法）

文件续传

服务端

import socket

import os

BASE_DIR=os.path.dirname(os.path.abspath(__file__))  #文件名的上一级目录

home=os.path.join(BASE_DIR,"home/yuan")  #在拼接路径

ip_port=("127.0.0.1",8998)

sk=socket.socket()

sk.bind(ip_port)

sk.listen(5)

while True:

    print("waiting ....")

    conn,addr=sk.accept()

    conn.sendall(bytes("欢迎登录","utf8"))

    flag=True

    while flag:

        client_bytes=conn.recv(1024)  #拿到客户端发来的文件名文件大小文件路径这三个参数

        client_str=str(client_bytes,"utf8")  #把获得的字节转换字符串

        func,file_name,file_byte_size,target_path=client_str.split("|",3)  #分解发来的三个参数

        file_byte_size=int(file_byte_size)  #

        path=os.path.join(home,file_name)  #拼接的路径

        has_received=0

        if os.path.exists(path):  #判断是否有文件

            conn.sendall(bytes("","utf8"))  #返回一个值做一次交互，表示文件存在

            is_continue=str(conn.recv(1024),"utf8") #等待客户端操作

            if is_continue=="":

                has_file_size=os.stat(path).st_size  #得到上次上次传送的文件大小

                conn.sendall(bytes(str(has_file_size),"utf8"))  #服务端发送给客户端，告诉上次传送的文件大小

                has_received+=has_file_size  #把之前上传的文件与续传的文件进行想加

                f=open(path,"ab")  #打开文件进行追加

            else:

                f=open(path,"wb")  #否则就进行写入

        else:

            conn.sendall(bytes("","utf8"))  #

            f=open(path,"wb")

        while has_received<file_byte_size:  #判断发送过的文件与要上传文件进行比较，如果小于文件的大小就一直进行上传直到上传完成

            try:

                data=conn.recv(1024) #对出现客户端进行中断的处理

                if not data:

                    raise Exception

            except Exception:

                flag=False

                break

            f.write(data)

            has_received+=len(data)

        print("ending")

        f.close()

客户端

import socket

import re,os,sys

ip_port=("127.0.0.1",8998)

sk=socket.socket()

sk.connect(ip_port)

print(str(sk.recv(1024),"utf8"))

#def bar(num=1, sum=100):

#    rate = float(num) / float(sum)

#    rate_num = int(rate * 100)

#  temp = '\r%d %%' % (rate_num, )

#    sys.stdout.write(temp)

  #  sys.stdout.flush()

while True:

    inp=input("please input:").strip()  #拿到输入值

    func,file_path=inp.split("|",1)  #用|分开方法名和路径

    local_path,target_path=re.split("\s*",file_path,1)  #通过正则的\s*（空格） 一个或是多个空格分开本地路径和目标路径

    file_byte_size=os.stat(local_path).st_size  #获取文件大小

    file_name=os.path.basename(local_path)  #拿到文件名称

    post_info="post|%s|%s|%s"%(file_name,file_byte_size,target_path)  #拼接文件名、文件大小和文件路径

    sk.sendall(bytes(post_info,"utf8"))  #拼接后上传到服务端

    result_exist=str(sk.recv(1024),"utf8")  #接受服务端传过来的文件存在的返回值

    has_sent=0

    if result_exist=="":

        inp=input("文件存在，是否续传？Y/N").strip()

        if inp.upper()=="Y":  #

            sk.sendall(bytes("","utf8"))  #告诉服务端进行断点续传

            result_continue_pos=str(sk.recv(1024),"utf8")  #

            print(result_continue_pos)

            has_sent=int(result_continue_pos)  #已经上传文件的大小

        else:

            sk.sendall(bytes("","utf8"))  #不进行续传

    file_obj=open(local_path,"rb")  #在绝对路径下读取内容

    file_obj.seek(has_sent)  #寻找到指针的位置（上次传到哪就跳到哪，如果没有传就从开始进行读取）

    while has_sent<file_byte_size:

        data=file_obj.read(1024)

        sk.sendall(data)  #发送给服务端

        has_sent+=len(data)  #一次最多传送1024，超过那就进行累加

        #进度条

        def bar(num=1, sum=100):

            rate = float(num) / float(sum)

            rate_num = int(rate * 100)

            temp = '\r%d %%' % (rate_num, )

            sys.stdout.write(temp)

            sys.stdout.flush()

        bar(has_sent,file_byte_size)

    file_obj.close()

    print("上传成功")

详细请参考：http://www.cnblogs.com/wupeiqi/articles/5040823.html

码农公寓

socket

socket解释

更多功能

建立连接

IO多路复用

读写分离

socketserver

ThredingTCPServer分析

文件续传

相关文章