40行代码实现sip注册

最近一段时间一直在学习python，并且用python编写了一个基于sipp工具的自动化测试平台，基本功能良好。对python的兴趣也与日俱增，因此决定每周练手使用python实现一个测试功能。既然对sip熟悉，那就先从sip开始吧。

话说socket真是一个万能的消息收发机制，尤其是对于测试人员来说，学会一门语言，并且把socket玩6，也许会给你的测试工作带来很大的便利。python 的socket机制很强大，使用也很简单，如下代码就是通过简单的socket交互，实现sip消息的发送和接收。

import os
import socket
import time

host = '172.31.225.220'
sport = '5060'

# 创建socket对象
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

# 进行socekt配置，使其支持端口复用，否则发送方绑定5066，则无法使用该端口进行接收
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.bind(('172.31.225.112', 5066))

try:
    port = int(sport)
except ValueError:
    port = socket.getservbyname(sport, 'udp')
s.connect((host, port))

# 获取当前路径，并读取当前目录下的reg.msg文件内，作为发送的data。
# 本例子中的reg.msg为完整的REGISTER消息
path = os.getcwd()
file = os.path.join(path, 'reg.msg')

# 由于python在进行明文读取，默认会将回车换行强制转换为/n，为了防止sip server解析错误，
# 在读取时增加了newline参数，并设置为空，则不进行转换，windows下读取为/r/n
with open(file, "r+", newline='') as sm:
    data = sm.read()
    
# 进行循环收发，需要将reg.msg中的[users]字段替换为1001-1005
for i in range(1001, 1006):
    data = data.replace(r"[users]", str(i))
    print(data)
    s.sendall(data.encode(encoding='utf-8'))
    while 1:
        buf = s.recv(2048)
        print(buf.decode('utf-8'))
        break
    time.sleep(1)