def test_5(self): # CONN/C2 resp = b'\x05\x00\x14\x03\x10\x00\x00\x00\x2c\x00\x00\x00\x00\x00' + \ b'\x00\x00\x00\x00\x03\x00\x06\x00\x00\x00\x01\x00\x00\x00' + \ b'\x00\x00\x00\x00\x00\x00\x01\x00\x02\x00\x00\x00\xc0\xd4' + \ b'\x01\x00' packet = rpch.RTSHeader(resp) packet.dump() pduData = packet['pduData'] numberOfCommands = packet['NumberOfCommands'] server_cmds = [] while numberOfCommands > 0: numberOfCommands -= 1 cmd_type = unpack('<L', pduData[:4])[0] cmd = rpch.COMMANDS[cmd_type](pduData) server_cmds.append(cmd) pduData = pduData[len(cmd):] for cmd in server_cmds: cmd.dump() connectionTimeout = rpch.ConnectionTimeout() connectionTimeout['ConnectionTimeout'] = 120000 self.assertEqual(server_cmds[2].getData(), connectionTimeout.getData()) receiveWindowSize = rpch.ReceiveWindowSize() receiveWindowSize['ReceiveWindowSize'] = 65536 self.assertEqual(server_cmds[1].getData(), receiveWindowSize.getData()) self.assertEqual(server_cmds[0].getData(), rpch.Version().getData())
def test_2(self): # CONN/A1 resp = b'\x05\x00\x14\x03\x10\x00\x00\x00\x4c\x00\x00\x00\x00\x00' + \ b'\x00\x00\x00\x00\x04\x00\x06\x00\x00\x00\x01\x00\x00\x00' + \ b'\x03\x00\x00\x00\xb0\xf6\xaf\x3d\x77\x62\x98\x07\x9b\x21' + \ b'\x54\x6e\xec\xf4\x22\x53\x03\x00\x00\x00\x3a\x24\x7a\x37' + \ b'\x6d\xc1\xed\x2c\x68\x5d\x34\x35\x13\x46\x43\x25\x00\x00' + \ b'\x00\x00\x00\x00\x04\x00' packet = rpch.RTSHeader(resp) packet.dump() pduData = packet['pduData'] numberOfCommands = packet['NumberOfCommands'] self.assertTrue(numberOfCommands == 4) self.assertTrue(packet['Flags'] == rpch.RTS_FLAG_NONE) self.assertTrue(packet['frag_len'] == 76) self.assertTrue(len(pduData) == 56) server_cmds = [] while numberOfCommands > 0: numberOfCommands -= 1 cmd_type = unpack('<L', pduData[:4])[0] cmd = rpch.COMMANDS[cmd_type](pduData) server_cmds.append(cmd) pduData = pduData[len(cmd):] for cmd in server_cmds: cmd.dump() self.assertTrue(server_cmds[0].getData() == rpch.Version().getData()) receiveWindowSize = rpch.ReceiveWindowSize() receiveWindowSize['ReceiveWindowSize'] = 262144 self.assertTrue( server_cmds[3].getData() == receiveWindowSize.getData()) cookie = rpch.Cookie() cookie['Cookie'] = b'\xb0\xf6\xaf=wb\x98\x07\x9b!Tn\xec\xf4"S' self.assertTrue(server_cmds[1].getData() == cookie.getData())