转载

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)
原文  https://segmentfault.com/a/1190000005961290
正文到此结束
Loading...