#! /usr/bin/env python#-*- coding:utf-8 -*-# sip reg test # E-Mail : Mike_Zhang@live.comimport sys,socket,time,tracebackimport uuidimport hashlibsvrIp,svrPort = "192.168.168.85",5060transportType = "udp"localIp,localPort = "192.168.168.168",17061uid,passwd = "1000","1234"g_branch,g_callId = uuid.uuid1(),uuid.uuid1()def getRegHeader(seqNum):    retStr = "REGISTER sip:{remote_ip} SIP/2.0\r\n"    retStr += "Via: SIP/2.0/{transport} {local_ip}:{local_port};branch={branch}\r\n"    retStr += "Max-Forwards: 70\r\n"    retStr += "From: {uid} <sip:{uid}@{remote_ip}:{remote_port}>;tag={call_number}\r\n"    retStr += "To: {uid} <sip:{uid}@{remote_ip}:{remote_port}>\r\n"    retStr += "Call-ID: {call_id}\r\n"    retStr += "User-Agent: fs testing\r\n"    retStr += "CSeq: %d REGISTER\r\n"%seqNum    retStr += "Contact: sip:{uid}@{local_ip}:{local_port}\r\n"    retStr += "Expires: 3600\r\n"    return retStrdef formatRegHeader(patternStr):    retstr = patternStr.format(        remote_ip=svrIp,        transport=transportType,        local_ip=localIp,        local_port=localPort,        branch=g_branch,        uid=uid,        remote_port=svrPort,        call_number=uid,        call_id=g_callId)    return retstrclass SipRegObj(object):    def __init__(self,uid,passwd,        realm="",nonce="",algorithm="MD5",qop="auth"):        self.uid = uid # "1000"        self.password = passwd # "1234"        self.realm = realm        self.nonce = nonce        self.algorithm = algorithm        self.qop = qop        self.uri = ""        self.method = "REGISTER"        self.cnonce = ""        self.nc = "00000001"          self.svrIp = svrIp        #self.transportType = "udp"    def getReponse(self):            md5 = hashlib.md5        ha1 = md5("%s:%s:%s"%(self.uid,self.realm,self.password)).hexdigest()        ha2 = md5("%s:%s"%(self.method,self.uri)).hexdigest()            reponse = md5("%s:%s:%s:%s:%s:%s"%(ha1,self.nonce,self.nc,self.cnonce,self.qop,ha2)).hexdigest()        return reponse    def getAuthStr(self):            self.uri,self.cnonce = ("sip:%s"%self.svrIp),uuid.uuid1()        retStr = ""        response = self.getReponse()        retStr += ‘Authorization: Digest username="%s",realm="%s",nonce="%s",uri="%s",response="%s",cnonce="%s",nc=%s,qop=%s,algorithm=%s\r\n‘%(            self.uid,self.realm,self.nonce,self.uri,response,self.cnonce,self.nc,self.qop,self.algorithm)        retStr += ‘Content-Length: 0\r\n‘        return retStr    def genChallengeMsg(self):        # gen sip reg challenge message        str1=getRegHeader(1) + "Content-Length: 0\r\n"                return formatRegHeader(str1)            def genAuthMsg(self,retstr1):        # parse data        data1 = retstr1.split("\r\n")[-4].split(": Digest")[1]        data1 = str(data1).replace(‘MD5‘,‘"MD5"‘)        tmpList = data1.split(",")        for item in tmpList :            #print item            arrtmp = item.split("=")            setattr(self,arrtmp[0].strip(),arrtmp[1].strip(‘"‘))                             # gen sip reg message with auth        str2=getRegHeader(2)        str2 = formatRegHeader(str2)                  str2 += self.getAuthStr()        return str2if __name__ == "__main__":    treg = SipRegObj(uid,passwd)    client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)    client.bind((localIp,localPort))    dstHost = (svrIp,svrPort)    client.sendto(treg.genChallengeMsg(),dstHost)    print time.time(),‘ : send success‘    # get response (401 msg)    retstr1 = client.recv(1024)    print retstr1    client.sendto(treg.genAuthMsg(retstr1),dstHost)    print time.time(),‘ : send success‘    retstr2 = client.recv(1024)    print "return str : \r\n",retstr2    time.sleep(30)