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) 

你可能感兴趣的:(python,sip)