13e87f800SKristof Provost#!/usr/bin/env python3 23e87f800SKristof Provost# 33e87f800SKristof Provost# SPDX-License-Identifier: BSD-2-Clause 43e87f800SKristof Provost# 53e87f800SKristof Provost# Copyright (c) 2021 Rubicon Communications, LLC (Netgate). 63e87f800SKristof Provost# 73e87f800SKristof Provost# Redistribution and use in source and binary forms, with or without 83e87f800SKristof Provost# modification, are permitted provided that the following conditions 93e87f800SKristof Provost# are met: 103e87f800SKristof Provost# 1. Redistributions of source code must retain the above copyright 113e87f800SKristof Provost# notice, this list of conditions and the following disclaimer. 123e87f800SKristof Provost# 2. Redistributions in binary form must reproduce the above copyright 133e87f800SKristof Provost# notice, this list of conditions and the following disclaimer in the 143e87f800SKristof Provost# documentation and/or other materials provided with the distribution. 153e87f800SKristof Provost# 163e87f800SKristof Provost# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 173e87f800SKristof Provost# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 183e87f800SKristof Provost# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 193e87f800SKristof Provost# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 203e87f800SKristof Provost# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 213e87f800SKristof Provost# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 223e87f800SKristof Provost# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 233e87f800SKristof Provost# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 243e87f800SKristof Provost# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 253e87f800SKristof Provost# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 263e87f800SKristof Provost# SUCH DAMAGE. 273e87f800SKristof Provost 283e87f800SKristof Provostimport argparse 293e87f800SKristof Provostimport logging 303e87f800SKristof Provostlogging.getLogger("scapy").setLevel(logging.CRITICAL) 313e87f800SKristof Provostimport scapy.all as sp 323e87f800SKristof Provostimport sys 333e87f800SKristof Provostimport os 343e87f800SKristof Provostcurdir = os.path.dirname(os.path.realpath(__file__)) 353e87f800SKristof Provostnetpfil_common = curdir + "/../netpfil/common" 363e87f800SKristof Provostsys.path.append(netpfil_common) 373e87f800SKristof Provostfrom sniffer import Sniffer 383e87f800SKristof Provost 393e87f800SKristof Provostdef check_pcp(args, packet): 403e87f800SKristof Provost vlan = packet.getlayer(sp.Dot1Q) 413e87f800SKristof Provost 423e87f800SKristof Provost if vlan is None: 433e87f800SKristof Provost return False 443e87f800SKristof Provost 453e87f800SKristof Provost if not packet.getlayer(sp.BOOTP): 463e87f800SKristof Provost return False 473e87f800SKristof Provost 483e87f800SKristof Provost if vlan.prio == int(args.expect_pcp[0]): 493e87f800SKristof Provost return True 503e87f800SKristof Provost 513e87f800SKristof Provost return False 523e87f800SKristof Provost 533e87f800SKristof Provostdef main(): 543e87f800SKristof Provost parser = argparse.ArgumentParser("pcp.py", 553e87f800SKristof Provost description="PCP test tool") 563e87f800SKristof Provost parser.add_argument('--recvif', nargs=1, 573e87f800SKristof Provost required=True, 583e87f800SKristof Provost help='The interface where to look for packets to check') 593e87f800SKristof Provost parser.add_argument('--expect-pcp', nargs=1, 603e87f800SKristof Provost help='The expected PCP value on VLAN packets') 613e87f800SKristof Provost 623e87f800SKristof Provost args = parser.parse_args() 633e87f800SKristof Provost 64*a39dedebSKajetan Staszkiewicz sniffer = Sniffer(args, check_pcp, args.recvif[0], timeout=20) 653e87f800SKristof Provost 663e87f800SKristof Provost sniffer.join() 673e87f800SKristof Provost 68*a39dedebSKajetan Staszkiewicz if sniffer.correctPackets: 693e87f800SKristof Provost sys.exit(0) 703e87f800SKristof Provost 713e87f800SKristof Provost sys.exit(1) 723e87f800SKristof Provost 733e87f800SKristof Provostif __name__ == '__main__': 743e87f800SKristof Provost main() 75