简单的 socket 代码

TCP 编程

客户端代码

将键盘输入的字符发送到服务端，并将从服务端接收到的字符输出到终端

#!/usr/python3

import socket

def socket_client():
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    #获取主机名
    host = socket.gethostname()
    port = 9999

    #建立连接
    s.connect((host, port))

    #接受欢迎消息
    print(s.recv(1024).decode('utf-8'))
    while True:
        data = input()
        #发送数据
        s.send(data.encode('utf-8'))
        print(s.recv(1024).decode('utf-8'))
    s.send(b'exit')
    s.close()

def main():
    socket_client()

if __name__ == '__main__':
    main()

注意：若服务器需要同时响应多个客户端请求，每个连接都需要一个新进程或新线程处理，否则服务器一次只能服务一个客户端 

服务端代码

将从客户端接收到的字符前面加上 HELLO, 再发送回给客户端

#!/usr/bin/python3

import socket
import threading
import time

def tcp_link(sock, addr):
    print('Accept new connection from %s %s...' % addr)
    sock.send('欢迎学习 python 网络编程'.encode('utf-8'))
    while True:
        data = sock.recv(1024)
        time.sleep(1)
        if not data or data.decode('utf-8') == 'exit':
            break
        sock.send(('HELLO, %s' % data.decode('utf-8')).encode('utf-8'))
    sock.close()
    print('Connection from %s : %s closed.' % addr)


def socket_server():
    #创建一个 socket 
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)#SOCK_STREAM 指定使用面向流的 TCP 协议
    #其中 AF_INET 指定使用 IPv4 协议，若要使用 IPv6 则指定该参数为 AF_INET6

    #获取本地主机名
    host = socket.gethostname()

    #获取本地主机名
    host = socket.gethostname()

    #设置端口号
    port = 9999

    #监听端口
    s.bind((host, port))

    #设置最大连接数，超过后排队
    s.listen(5)

    while True:
        #接受一个新连接
        sock, addr = s.accept()
        # 创建新线程处理TCP连接
        t = threading.Thread(target = tcp_link, args = (sock, addr))
        t.start()    


def main():
    socket_server()


if __name__ == '__main__':
    main()

UDP 编程

 客户端代码

将键盘输入的字符发送到服务端，并将从服务端接收到的字符输出到终端

#!/usr/bin/python3

import socket

def socket_udp_client():
    host = socket.gethostname()
    port = 9999
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    while True:
        data = input()

        #发送数据
        s.sendto(data.encode('utf-8'), (host, port))

        #接收数据
        print(s.recv(1024).decode('utf-8'))


def main():
    socket_udp_client()


if __name__ == '__main__':
    main()

服务端代码

将从客户端接收到的字符前面加上 hello, 再发送回给客户端

#!/usr/bin/python3

import socket

def socket_udp_server():
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) #SOCK_DGRAM 参数指定 socket 的类型是 udp
    host = socket.gethostname()
    port = 9999

    #绑定端口
    s.bind((host, port))

    #不需要调用 listen 方法，直接接收来自任何客户端的数据

    while True:
        #接收数据
        data, addr = s.recvfrom(1024)
        #recvfrom 方法返回数据和客户端的地址与端口。这样，服务器收到数据后，直接调用 sendto 就可以把数据用 udp 发送给客户端
        print('received from %s: %s.' % addr)
        s.sendto(b'hello, %s' % data, addr)


def main():
    socket_udp_server()


if __name__ == '__main__':
    main()