Ignore:
Timestamp:
Jul 7, 2008 8:14:41 PM (11 years ago)
Author:
bennylp
Message:

Added mod_recvfrom.py testing module and some registrar test scenarios

File:
1 edited

Legend:

Unmodified
Added
Removed
  • pjproject/trunk/pjsip-apps/src/test-pjsua/inc_sip.py

    r2084 r2110  
    5050        #return re.split("[;& ]", s) 
    5151 
     52def get_header(msg, hname): 
     53        headers = msg.splitlines() 
     54        for hdr in headers: 
     55                hfields = hdr.split(": ", 2) 
     56                if hfields[0]==hname: 
     57                        return hfields[1] 
     58        return None 
    5259 
    5360class Dialog: 
     
    6673        trace_enabled = True 
    6774        last_request = "" 
    68         def __init__(self, dst_addr, dst_port=5060, tcp=False, trace=True): 
     75        def __init__(self, dst_addr, dst_port=5060, tcp=False, trace=True, local_port=0): 
    6976                self.dst_addr = dst_addr 
    7077                self.dst_port = dst_port 
     
    7683                else: 
    7784                        self.sock = socket(AF_INET, SOCK_DGRAM) 
    78                         self.sock.bind(("127.0.0.1", 0)) 
     85                        self.sock.bind(("127.0.0.1", local_port)) 
    7986                 
    8087                self.local_ip, self.local_port = self.sock.getsockname() 
     
    114121                return msg 
    115122 
     123        def create_response(self, request, code, reason, to_tag=""): 
     124                response = "SIP/2.0 " + str(code) + " " + reason + "\r\n" 
     125                lines = request.splitlines() 
     126                for line in lines: 
     127                        hdr = line.split(":", 1)[0] 
     128                        if hdr in ["Via", "From", "To", "CSeq", "Call-ID"]: 
     129                                if hdr=="To" and to_tag!="": 
     130                                        line = line + ";tag=" + to_tag 
     131                                elif hdr=="Via": 
     132                                        line = line + ";received=127.0.0.1" 
     133                                response = response + line + "\r\n" 
     134                return response 
     135 
    116136        def create_invite(self, sdp, extra_headers=""): 
    117137                self.inv_branch = str(random.random()) 
     
    124144                return self.create_req("BYE", "", extra_headers) 
    125145 
    126         def send_msg(self, msg): 
     146        def send_msg(self, msg, dst_addr=None): 
    127147                if (is_request(msg)): 
    128148                        self.last_request = msg.split(" ", 1)[0] 
    129                 self.trace("============== TX MSG ============= \n" + msg) 
    130                 self.sock.sendto(msg, 0, (self.dst_addr, self.dst_port)) 
    131  
    132         def wait_msg(self, timeout): 
     149                if not dst_addr: 
     150                        dst_addr = (self.dst_addr, self.dst_port) 
     151                self.trace("============== TX MSG to " + str(dst_addr) + " ============= \n" + msg) 
     152                self.sock.sendto(msg, 0, dst_addr) 
     153 
     154        def wait_msg_from(self, timeout): 
    133155                endtime = time.time() + timeout 
    134156                msg = "" 
     157                src_addr = None 
    135158                while time.time() < endtime: 
    136159                        readset = select([self.sock], [], [], timeout) 
     
    144167                                continue 
    145168                        try: 
    146                                 msg = self.sock.recv(2048) 
     169                                msg, src_addr = self.sock.recvfrom(2048) 
    147170                        except: 
    148171                                print "recv() exception: ", sys.exc_info()[0] 
     
    150173 
    151174                if msg=="": 
    152                         return "" 
     175                        return "", None 
    153176                if self.last_request=="INVITE" and self.rem_tag=="": 
    154177                        self.rem_tag = get_tag(msg, "To") 
     
    157180                                self.rem_tag = ";tag=" + self.rem_tag 
    158181                        self.trace("=== rem_tag:" + self.rem_tag) 
    159                 self.trace("=========== RX MSG ===========\n" + msg) 
    160                 return msg 
     182                self.trace("=========== RX MSG from " + str(src_addr) +  " ===========\n" + msg) 
     183                return (msg, src_addr) 
    161184         
     185        def wait_msg(self, timeout): 
     186                return self.wait_msg_from(timeout)[0] 
     187                 
    162188        # Send request and wait for final response 
    163189        def send_request_wait(self, msg, timeout): 
     
    237263                self.inst_param.enable_buffer = enable_buffer  
    238264 
     265 
     266class RecvfromTransaction: 
     267        # The test title for this transaction 
     268        title = "" 
     269        # Optinal list of pjsua command and optional expect patterns  
     270        # to be invoked to make pjsua send a request 
     271        # Sample: 
     272        #       (to make call and wait for INVITE to be sent) 
     273        #       cmds = [ ["m"], ["sip:127.0.0.1", "INVITE sip:"]  ] 
     274        cmds = [] 
     275        # Check if the CSeq must be greater than last Cseq? 
     276        check_cseq = True 
     277        # List of RE patterns that must exists in incoming request 
     278        include = [] 
     279        # List of RE patterns that MUST NOT exist in incoming request 
     280        exclude = [] 
     281        # Response code to send 
     282        resp_code = 0 
     283        # Additional list of headers to be sent on the response 
     284        # Note: no need to add CRLF on the header 
     285        resp_hdr = [] 
     286        # Message body. This should include the Content-Type header too. 
     287        # Sample: 
     288        #       body = """Content-Type: application/sdp\r\n 
     289        #       \r\n 
     290        #       v=0\r\n 
     291        #       ... 
     292        #       """ 
     293        body = None 
     294        # Pattern to be expected on pjsua when receiving the response 
     295        expect = "" 
     296         
     297        def __init__(self, title, resp_code, check_cseq=True, 
     298                        include=[], exclude=[], cmds=[], resp_hdr=[], resp_body=None, expect=""): 
     299                self.title = title 
     300                self.cmds = cmds 
     301                self.include = include 
     302                self.exclude = exclude 
     303                self.resp_code = resp_code 
     304                self.resp_hdr = resp_hdr 
     305                self.resp_body = resp_body 
     306                self.expect = expect 
     307                         
     308 
     309class RecvfromCfg: 
     310        # Test name 
     311        name = "" 
     312        # pjsua InstanceParam 
     313        inst_param = None 
     314        # List of RecvfromTransaction 
     315        transaction = None 
     316        # Use TCP? 
     317        tcp = False 
     318 
     319        # Note: 
     320        #  Any "$PORT" string in the pjsua_args will be replaced 
     321        #  by server port 
     322        def __init__(self, name, pjsua_args, transaction, tcp=False): 
     323                self.name = name 
     324                self.inst_param = cfg.InstanceParam("pjsua", pjsua_args) 
     325                self.transaction = transaction 
     326                self.tcp=tcp 
     327 
     328 
     329 
     330 
Note: See TracChangeset for help on using the changeset viewer.