摘要：對的興趣也與日俱增，因此決定每周練手使用實現(xiàn)一個測試功能。的機(jī)制很強(qiáng)大，使用也很簡單，如下代碼就是通過簡單的交互，實現(xiàn)消息的發(fā)送和接收。

最近一段時間一直在學(xué)習(xí)python，并且用python編寫了一個基于sipp工具的自動化測試平臺，基本功能良好。對python的興趣也與日俱增，因此決定每周練手使用python實現(xiàn)一個測試功能。既然對sip熟悉，那就先從sip開始吧。

話說socket真是一個萬能的消息收發(fā)機(jī)制，尤其是對于測試人員來說，學(xué)會一門語言，并且把socket玩6，也許會給你的測試工作帶來很大的便利。python 的socket機(jī)制很強(qiáng)大，使用也很簡單，如下代碼就是通過簡單的socket交互，實現(xiàn)sip消息的發(fā)送和接收。

import os
import socket
import time

host = "172.31.225.220"
sport = "5060"

# 創(chuàng)建socket對象
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

# 進(jìn)行socekt配置，使其支持端口復(fù)用，否則發(fā)送方綁定5066，則無法使用該端口進(jìn)行接收
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))

# 獲取當(dāng)前路徑，并讀取當(dāng)前目錄下的reg.msg文件內(nèi)，作為發(fā)送的data。
# 本例子中的reg.msg為完整的REGISTER消息
path = os.getcwd()
file = os.path.join(path, "reg.msg")

# 由于python在進(jìn)行明文讀取，默認(rèn)會將回車換行強(qiáng)制轉(zhuǎn)換為/n，為了防止sip server解析錯誤，
# 在讀取時增加了newline參數(shù)，并設(shè)置為空，則不進(jìn)行轉(zhuǎn)換，windows下讀取為/r/n
with open(file, "r+", newline="") as sm:
    data = sm.read()
    
# 進(jìn)行循環(huán)收發(fā)，需要將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)