Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use get_protocol method in autotest

Best Python code snippet using autotest_python

utils.py

Source:utils.py

...32        	config[sec][key] = val33    return config34def packetIsIP(message) :35    pkt = packet.Packet(message.data)36    ip = pkt.get_protocol(ipv4.ipv4)37    if ip is not None :38        return True39    return False40def packetIsARP(message) :41    pkt = packet.Packet(message.data)42    a = pkt.get_protocol(arp.arp)43    if a is not None :44        return True45    return False46def packetIsRequestARP(message) :47    pkt = packet.Packet(message.data)48    a = pkt.get_protocol(arp.arp)49    if a.opcode == arp.ARP_REQUEST :50        return True51    return False52def packetIsReplyARP(message) :53    pkt = packet.Packet(message.data)54    a = pkt.get_protocol(arp.arp)55    if a.opcode == arp.ARP_REPLY :56	return True57    return False58def packetIsTCP(message) :59    pkt = packet.Packet(message.data)60    ip = pkt.get_protocol(ipv4.ipv4)61    if ip is not None and ip.proto == 6 :62	return True63    return False64def packetDstIp(message, ipaddr) :65    if packetIsIP(message):66	pkt = packet.Packet(message.data)67	ip = pkt.get_protocol(ipv4.ipv4)68    	if not cmp(ip.dst, ipaddr):69		return True70    return False71def packetSrcIp(message, ipaddr) :72    if packetIsIP(message):73        pkt = packet.Packet(message.data)74        ip = pkt.get_protocol(ipv4.ipv4)75        if not cmp(ip.src, ipaddr):76                return True77    return False78def packetDstTCPPort(message, tcpport) :79    if packetIsTCP(message) :80	pkt = packet.Packet(message.data)81        dsttcp = pkt.get_protocol(tcp.tcp)82	if dsttcp.dst_port == tcpport :83		return True84    return False85def packetSrcTCPPort(message, tcpport) :86    if packetIsTCP(message) :87        pkt = packet.Packet(message.data)88        srctcp = pkt.get_protocol(tcp.tcp)89	if srctcp.src_port == tcpport :90                return True91    return False92def packetArpDstIp(message, ipaddr) :93    if packetIsARP(message):94        pkt = packet.Packet(message.data)95        a = pkt.get_protocol(arp.arp)96        if not cmp(a.dst_ip, ipaddr):97                return True98    return False99def packetArpSrcIp(message, ipaddr) :100    if packetIsARP(message):101        pkt = packet.Packet(message.data)102        a = pkt.get_protocol(arp.arp)103        if not cmp(a.src_ip, ipaddr):104                return True105    return False106def createArpRequest(message, ip):107    if not packetIsARP(message):108    	print("Packet is not ARP")109    	return110    pkt = packet.Packet(message.data)111    origarp = pkt.get_protocol(arp.arp)112    a = arp.arp(113    	hwtype=origarp.hwtype,114    	proto=origarp.proto,115	src_mac=origarp.src_mac,116    	dst_mac=origarp.dst_mac,117	hlen=origarp.hlen,118    	opcode=arp.ARP_REQUEST,119    	plen=origarp.plen,120	src_ip=origarp.src_ip,121	dst_ip=ip122	)123    e = ethernet.ethernet(124	dst=mac.BROADCAST_STR,125	src=origarp.src_mac,126	ethertype=ether.ETH_TYPE_ARP)    127    p = packet.Packet()128    p.add_protocol(e)129    p.add_protocol(a)130    p.serialize()131    return p132def createArpReply(message, ip):133    if not packetIsARP(message):134        print("Packet is not ARP")135        return136    pkt = packet.Packet(message.data)137    origarp = pkt.get_protocol(arp.arp)138    a = arp.arp(139        hwtype=origarp.hwtype,140        proto=origarp.proto,141        src_mac=origarp.src_mac,142        dst_mac=origarp.dst_mac,143        hlen=origarp.hlen,144        opcode=arp.ARP_REPLY,145        plen=origarp.plen,146        src_ip=ip,147        dst_ip=origarp.dst_ip148        )149    e = ethernet.ethernet(150        dst=origarp.dst_mac,151        src=origarp.src_mac,152        ethertype=ether.ETH_TYPE_ARP)153    p = packet.Packet()154    p.add_protocol(e)155    p.add_protocol(a)156    p.serialize()157    return p158def ipv4_to_int(string):159	ip = string.split('.')160       	assert len(ip) == 4161       	i = 0162       	for b in ip:163    		b = int(b)164        	i = (i << 8) | b165        return i166def findIPInServerList(l, ip_address):167        for i in range (0, len(l)):168	  if (l[i][0]==ip_address):169            return i170          else:171            continue172 	return -1173def sendPacketOut( msg, actions, buffer_id=0xffffffff, data=None ):174    datapath = msg.datapath175    parser = datapath.ofproto_parser176    if buffer_id == 0xffffffff :177        out = parser.OFPPacketOut(178            datapath=datapath, buffer_id=buffer_id, in_port=msg.in_port,179            actions=actions, data=data)180        datapath.send_msg(out)181    else :182        out = parser.OFPPacketOut(183            datapath=datapath, buffer_id=buffer_id, in_port=msg.in_port,184            actions=actions)185        datapath.send_msg(out)186def getFullMatch( msg ):187    datapath = msg.datapath188    parser = datapath.ofproto_parser189    190    in_port=None191    dl_src=None192    dl_dst=None193    dl_vlan=None194    dl_vlan_pcp=None195    dl_type=None196    nw_tos=None197    nw_proto=None198    nw_src=None199    nw_dst=None200    tp_src=None201    tp_dst=None202    203    in_port = msg.in_port204    pkt = packet.Packet(msg.data)205    eth = pkt.get_protocol(ethernet.ethernet)206    dl_src = eth.src207    dl_dst = eth.dst208    dl_type = eth.ethertype209    vl = pkt.get_protocol(vlan.vlan)210    if vl is not None :211        dl_vlan = vl.vid212        dl_vlan_pcp = vl.pcp213        dl_type = vl.ethertype214    215    ip = pkt.get_protocol(ipv4.ipv4)216    if ip is not None :217        nw_src = ip.src218        nw_dst = ip.dst219        nw_proto = ip.proto220        nw_tos = ip.tos221        t = pkt.get_protocol(tcp.tcp)222        if t is not None :223            tp_src = t.src_port224            tp_dst = t.dst_port225        u = pkt.get_protocol(udp.udp)   226        if u is not None :227            tp_src = u.src_port228            tp_dst = u.dst_port229    230        ic = pkt.get_protocol(icmp.icmp)231        if ic is not None :232            tp_src = ic.type233            tp_dst = ic.code234    235    a = pkt.get_protocol(arp.arp)236    if a is not None :237        nw_src = a.src_ip238        nw_dst = a.dst_ip239        nw_proto = a.opcode240    match = parser.OFPMatch( 241        dl_src=mac.haddr_to_bin(dl_src), 242        dl_dst=mac.haddr_to_bin(dl_dst), 243        dl_vlan=dl_vlan, 244        dl_vlan_pcp=dl_vlan_pcp, 245        dl_type=dl_type, 246        nw_tos=nw_tos, 247        nw_proto=nw_proto, 248        nw_src=ipv4_to_int(nw_src), 249        nw_dst=ipv4_to_int(nw_dst), ...

protocols.py

Source:protocols.py

...11model_0 = {12    "qc_options": wb97xd,13    "charges": ddec6,14    "virtual_sites": None,15    "non_bonded": get_protocol(protocol_name="0"),16}17model_1a = {18    "qc_options": b3lyp,19    "charges": ddec6,20    "virtual_sites": None,21    "non_bonded": get_protocol(protocol_name="1a"),22}23model_1b = {24    "qc_options": QCOptions(method="HF", basis="6-31G(d)", program="gaussian"),25    "charges": DDECCharges(ddec_version=6, solvent_settings=None),26    "virtual_sites": None,27    "non_bonded": get_protocol(protocol_name="1b"),28}29model_2a = {30    "qc_options": wb97xd,31    "charges": DDECCharges(ddec_version=6, solvent_settings=SolventGaussian(epsilon=2)),32    "virtual_sites": None,33    "non_bonded": get_protocol(protocol_name="2a"),34}35model_2b = {36    "qc_options": wb97xd,37    "charges": DDECCharges(38        ddec_version=6, solvent_settings=SolventGaussian(epsilon=10)39    ),40    "virtual_sites": None,41    "non_bonded": get_protocol(protocol_name="2b"),42}43model_2c = {44    "qc_options": wb97xd,45    "charges": DDECCharges(46        ddec_version=6, solvent_settings=SolventGaussian(epsilon=20)47    ),48    "virtual_sites": None,49    "non_bonded": get_protocol(protocol_name="2c"),50}51model_3a = {52    "qc_options": wb97xd,53    "charges": DDECCharges(ddec_version=3),54    "virtual_sites": None,55    "non_bonded": get_protocol(protocol_name="3a"),56}57model_3b = {58    "qc_options": QCOptions(method="B3LYP-D3BJ", basis="DZVP", program="psi4"),59    "charges": MBISCharges(),60    "virtual_sites": None,61    "non_bonded": get_protocol(protocol_name="3b"),62}63model_4a = {64    "qc_options": wb97xd,65    "charges": ddec6,66    "virtual_sites": None,67    "non_bonded": get_protocol(protocol_name="4a"),68}69model_4b = {70    "qc_options": wb97xd,71    "charges": ddec6,72    "virtual_sites": None,73    "non_bonded": get_protocol(protocol_name="4b"),74}75model_5a = {76    "qc_options": wb97xd,77    "charges": ddec6,78    "non_bonded": get_protocol(protocol_name="5a"),79}80model_5b = {81    "qc_options": wb97xd,82    "charges": ddec6,83    "non_bonded": get_protocol(protocol_name="5b"),84}85model_5c = {86    "qc_options": wb97xd,87    "charges": DDECCharges(ddec_version=3),88    "non_bonded": get_protocol(protocol_name="5c"),89}90model_5d = {91    "qc_options": b3lyp,92    "charges": ddec6,93    "non_bonded": get_protocol(protocol_name="5d"),94}95model_5e = {96    "qc_options": QCOptions(method="B3LYP-D3BJ", basis="DZVP", program="psi4"),97    "charges": MBISCharges(),98    "non_bonded": get_protocol(protocol_name="5e"),99}100workflow_protocols = {101    "0": model_0,102    "1a": model_1a,103    "1b": model_1b,104    "2a": model_2a,105    "2b": model_2b,106    "2c": model_2c,107    "3a": model_3a,108    "3b": model_3b,109    "4a": model_4a,110    "4b": model_4b,111    "5a": model_5a,112    "5b": model_5b,...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.