python learning Network Programming.py

Socket

# 用一个 Socke t表示“打开了一个网络连接”
# 打开一个 Socket 需要知道目标计算机的IP地址和端口号，再指定协议类型即可。

# TCP
# 主动发起连接的是客户端，被动相应的是服务器

# 1. 创建一个基于 TCP 连接的socket

import socket

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# AF_INET 指 IPv4 协议。SOCK_STREAM 指面向流的 TCP 协议。
# 初始化了一个 socket 实例，还没有建立连接

# 2. 连接到服务器
# connect 接受的参数是一个 tuple：(ip, port)。输入域名可以通过DNS自动转换为IP地址。

s.connect(('www.sina.com.cn',80))

# 3. 发送数据 send
# 下列数据的意思是：请求返回首页的内容
# TCP连接创建的是双向通道，双方都可以同时给对方发数据。但是谁先发谁后发，怎么协调，要根据具体的协议来决定。例如，HTTP协议规定客户端必须先发请求给服务器，服务器收到后才发数据给客户端。

s.send(b'GET / HTTP/1.1\r\nHost: www.sina.com.cn\r\nConnection: close\r\n\r\n')

# 4. 接受服务器发来的数据
# 发送完 HTTP 格式请求后，接下来就可以接收新浪服务器返回的数据了：

buffer = []
while True:
    d = s.recv(1024) # 1KB
    if d:
        buffer.append(d)
    else: # 数据读取完毕或者连接中断，返回 empty string
        break

data = b''.join(buffer) # 这句语句的意思就是使用空字节把 buffer 这个字节列表连接在一起，成为一个新的字节串

# b''是一个空字节，join是连接列表的函数，buffer是一个不断接受到的串的列表
# 例如把[b'ab',b'cd',b'ef']变成 b'abcdef'

# 5. 接受完数据，关闭 socket

s.close()

# 接收到的数据包括HTTP头和网页本身
# 我们只需要把HTTP头和网页分离一下，把HTTP头打印出来，网页内容保存到文件：

header, html = data.split(b'\r\n\r\n',1)
print(header.decode('utf-8'))

with open('sina.html','wb') as file:
    file.write(html)

服务器

# 1. 服务器进程绑定一个端口，监听来自客户端的连接
# 四元组来唯一确定一个 socket
# 因为服务器要同时相应多个客户端的请求，所以，每次连接都需要一个新的进程或者线程来处理。

# 我们来编写一个简单的服务器程序，它接收客户端连接，把客户端发过来的字符串加上Hello再发回去。

import socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 监听端口
# 绑到本地地址，端口号9999
# 127.0.0.1是一个特殊的IP地址，表示本机地址，如果绑定到这个地址，客户端必须同时在本机运行才能连接，也就是说，外部的计算机无法连接进来。
s.bind(('127.0.0.1', 9999))

# 调用listen()方法开始监听端口，传入的参数指定等待连接的最大数量：
s.listen(5)
print('Waiting for connection...')

# 服务器程序通过一个永久循环来接受来自客户端的连接
# accept()会等待并返回一个指示客户端连接的socket 和客户端 ip 地址:
import threading
import time
def tcplink(sock, addr):
    print('Accept new connection from %s...' % addr)
    sock.send(b'Welcome!')
    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 closed.' % addr)

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

客户端

import socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 建立连接:
s.connect(('127.0.0.1', 9999))
# 接收欢迎消息:
print(s.recv(1024).decode('utf-8'))
for data in [b'Chang', b'KangKang', b'Wangbin']:
    # 发送数据:
    s.send(data)
    print(s.recv(1024).decode('utf-8'))
s.send(b'exit')
s.close()

UDP server

import socket

s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.bind(('127.0.0.1',9999))
print('Bind UDP on 9999....')
while True:
    # UDP: 不需要调用listen()方法，而是直接接收来自任何客户端的数据：
    data, addr = s.recvfrom(1024) # recvfrom()方法返回数据和客户端的地址与端口
    print('Received from %s:%s' % addr) # 把数据用UDP发给客户端
    s.sendto(b'Hello, %s!'% data, addr)

UDP client

import socket
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
for data in [b'zhao', b'Chang']:
    s.sendto(data, ('127.0.0.1',9999))
    print(s.recv(1024).decode('utf-8'))
s.close()