python实现一个简单的网络聊天程序

一、Linux Socket

1.Linux Socke基本上就是BSD Socket（伯克利套接字）

伯克利套接字的应用编程接口（API）是采用C语言的进程间通信的库，经常用在计算机网络间的通信。BSD Socket的应用编程接口已经是网络套接字的抽象标准。大多数其他程序语言使用一种相似的编程接口。由于伯克利套接字是第一个socket，大多数程序员很熟悉它们，所以大量系统把伯克利套接字作为其主要的网络API。

主要的头文件如下，不同的系统可能具体不同。

<sys/socket.h> BSD socket 核心函数和数据结构。

<netinet/in.h> AF_INET 和AF_INET6 地址家族和他们对应的协议家族PF_INET 和PF_INET6。在互联网编程中广泛使用，包括IP地址以及TCP和UDP端口号。

<sys/un.h> PF_UNIX/PF_LOCAL 地址家族。用于运行在一台计算机上的程序间的本地通信，不用在网络中。

<arpa/inet.h> 和IP地址相关的一些函数。

<netdb.h> 把协议名和主机名转化成数字的一些函数。

2.API函数

这些是伯克利套接字提供的库函数：

（1）socket() 创造某种类型的套接字，分配一些系统资源，用返回的整数识别。

（2）bind() 一般是用在服务器这边，和一个套接字地址结构相连，比如说是一个特定的本地端口号和一个IP地址。

（3）listen()用在服务器一边，导致一个绑定的TCP套接字进入监听状态。

（4）connect() 用在客户机这边，给套接字分配一个空闲的端口号。比如说一个TCP套接字，它会试图建立一个新的TCP连接。

（5）accept() 用在服务器这边。从客户机那接受请求试图创造一个新的TCP连接，并把一个套接字和这个连接相联系起来。

（6）send() and recv()， or write() and read()，or sendto() and recvfrom()用来接收和发送数据。

（7）close() 关闭连接，系统释放资源。

（8）gethostbyname() and gethostbyaddr()用来解析主机名和地址。

（9）select() 、poll() 处理多个连续的读、写饿错误状态。

（11）getsockopt() 得到对应socket的选项值。

（12）setsockopt() 设置对应socket的选项值。

二、Python实现网络通信

　　网络上的两个程序通过一个双向的通信连接实现数据的交换，这个连接的一端称为一个socket。python中socket模块为操作系统的socket实现提供了一个python接口。

1.socket的相关函数：

（1）socket()：用于创建与指定的服务提供者绑定socket。

　　函数原型为：socket=socket.socket(familly,type)

　　参数说明：

　　familly：指定协议的地址家族，可为AF_INET或AF_UNIX。AF_INET家族包括Internet地址，AF_UNIX家族用于同一台机器上的进程间通信。

　　type：指定套接字的类型。

（2）bind()：bind()函数可以将本地地址与一个socket绑定在一起.

　　函数原型为：socket.bind( address )

　　参数address是一个双元素元组，格式是(host,port)。host代表主机，port代表端口号。

（3）listen()：listen()函数可以将socket设置为监听接入连接的状态。

　　函数原型为：listen(backlog)

　　参数backlog指定等待连接队列的最大长度。

（4）connect()：connect()函数用于连接到address处的socket。

　　函数原型为：socket.connect(address)

　　参数address是一个双元素元组，格式是(host,port)。host代表主机，port代表端口号

（5）accept()：在服务器端调用listen()函数监听接入连接后，可以调用accept()函数来等待接受连接请求。

　　函数原型为：(connection,address)=socket.accept()

　　用accept()方法后，socket会进入waiting状态。客户请求连接时，accept()方法会建立连接并返回服务器。accept()方法返回一个含有两个元素的元组(connection,address)。第一个元素connection是新的socket对象，服务器必须通过它与客户通信；第二个元素address是客户的Internet地址。

（6）recv()：调用recv()函数可以从已连接的socket中接收数据。

　　函数原型为：buf = sock.recv(size)

　　参数sock是接收数据的socket对象，参数size指定接收数据的缓冲区的大小。recv()的函数的返回接收的数据。

（7）send()：调用send()函数可以在已连接的socket上发送数据。

　　函数原型为：sock.recv(buf)

　　参数sock是在已连接的socket上发送数据。参数buf是也要已连接的Socket上发送数据。

（8）close()：close ()函数用于关闭一个socket，释放其所占用的所有资源。

　　函数原型为：sock.closesocket();

　　参数s表示要关闭的socket。

可以在Linux系统下查看socket模块：

　　通过比较可以看出，再socket模块中并没有上述的bind()，listen()，accept()等函数。

　　这是因为它们都是继承自_socket模块。而_socket是在链接库里的，也就是说它不是用python实现的，而是socket操作的C实现，这个是非常底层的操作。socket.py是用py代码把C实现的模块的封装起来之后的模块，供人使用，在不同系统_socket的位置不一样。

再查看_socket模块：

这下可以看到我们在编程中需要用到的那些函数了。

2.TCP/编程实现：

　　socket编程需要两端,一般来说,需要一个服务端(Server),一个客户端(Client)。如图：

TCP服务端编程：

服务端连接步骤：

（1）创建socket对象

（2）使用bind方法,绑定IP地址,Address和端口Port

（3）使用listen开始监听,在上面以绑定的地址上

（4）使用accept开始等待连接进来,获取用于传送数据的socket对象和addr。注意使用accept会发生阻塞,惯用放在新的线程里面

（5）接收数据recv

代码：

import socket

sk = socket.socket() # 默认是AF_INET、SOCK_STREAM
address = ('127.0.0.1', 9000)
sk.bind(address) # 将主机号与端口绑定到套接字
sk.listen(3) # 设置并启动TCP监听器
print('waitting......')

while True:
    conn, addr = sk.accept() # 被动接受TCP连接，一直等待连接到达
    print('连接到达',addr)
    while True:
        data = conn.recv(1024) # 接收TCP消息，并制定最大长度
        if not data:
            print('连接已断开！')
            conn.close()
            break
        print(str(data, 'utf8'))
        inp = input('>>')
        conn.send(bytes(inp, 'utf8'))  #向客户端回送信息

TCP客户端编程：

客户端连接步骤:

（1）创建Socket对象

（2）连接到远端服务端的IP和port端口, connect方法

（3）传输数据：send发送数据；recv接收数据, 会阻塞

（4）关闭连接,释放资源

代码：

import  socket
sk = socket.socket() # 默认是AF_INET、SOCK_STREAM
address = ('127.0.0.1', 9000)
sk.connect(address) #连接服务端
while True:
    inp = input('>>')
    if inp == 'exit':
        break
    sk.send(bytes(inp, 'utf8')) #向服务端发送信息
    data = sk.recv(1024)
    print(str(data, 'utf8'))
else:
    sk.close()

运行结果：

服务端：

客户端：