# Copyright (c) 2013 by Gilbert Ramirez <gram@alumni.rice.edu>
#
# SPDX-License-Identifier: GPL-2.0-or-later

import unittest
import fixtures
from suite_dfilter.dfiltertest import *


@fixtures.uses_fixtures
class case_ipv4(unittest.TestCase):
    trace_file = "nfs.pcap"

    def test_eq_1(self, checkDFilterCount):
        dfilter = "ip.src == 172.25.100.14"
        checkDFilterCount(dfilter, 1)

    def test_eq_2(self, checkDFilterCount):
        dfilter = "ip.src == 255.255.255.255"
        checkDFilterCount(dfilter, 0)

    def test_ne_1(self, checkDFilterCount):
        dfilter = "ip.src != 172.25.100.14"
        checkDFilterCount(dfilter, 1)

    def test_ne_2(self, checkDFilterCount):
        dfilter = "ip.src != 255.255.255.255"
        checkDFilterCount(dfilter, 2)

    def test_gt_1(self, checkDFilterCount):
        dfilter = "ip.dst > 198.95.230.200"
        checkDFilterCount(dfilter, 0)

    def test_gt_2(self, checkDFilterCount):
        dfilter = "ip.dst > 198.95.230.20"
        checkDFilterCount(dfilter, 0)

    def test_gt_3(self, checkDFilterCount):
        dfilter = "ip.dst > 198.95.230.10"
        checkDFilterCount(dfilter, 1)

    def test_ge_1(self, checkDFilterCount):
        dfilter = "ip.dst >= 198.95.230.200"
        checkDFilterCount(dfilter, 0)

    def test_ge_2(self, checkDFilterCount):
        dfilter = "ip.dst >= 198.95.230.20"
        checkDFilterCount(dfilter, 1)

    def test_ge_3(self, checkDFilterCount):
        dfilter = "ip.dst >= 198.95.230.10"
        checkDFilterCount(dfilter, 1)

    def test_lt_1(self, checkDFilterCount):
        dfilter = "ip.src < 172.25.100.140"
        checkDFilterCount(dfilter, 1)

    def test_lt_2(self, checkDFilterCount):
        dfilter = "ip.src < 172.25.100.14"
        checkDFilterCount(dfilter, 0)

    def test_lt_3(self, checkDFilterCount):
        dfilter = "ip.src < 172.25.100.10"
        checkDFilterCount(dfilter, 0)

    def test_le_1(self, checkDFilterCount):
        dfilter = "ip.src <= 172.25.100.140"
        checkDFilterCount(dfilter, 1)

    def test_le_2(self, checkDFilterCount):
        dfilter = "ip.src <= 172.25.100.14"
        checkDFilterCount(dfilter, 1)

    def test_le_3(self, checkDFilterCount):
        dfilter = "ip.src <= 172.25.100.10"
        checkDFilterCount(dfilter, 0)

    def test_cidr_eq_1(self, checkDFilterCount):
        dfilter = "ip.src == 172.25.100.14/32"
        checkDFilterCount(dfilter, 1)

    def test_cidr_eq_2(self, checkDFilterCount):
        dfilter = "ip.src == 172.25.100.0/24"
        checkDFilterCount(dfilter, 1)

    def test_cidr_eq_3(self, checkDFilterCount):
        dfilter = "ip.src == 172.25.0.0/16"
        checkDFilterCount(dfilter, 1)

    def test_cidr_eq_4(self, checkDFilterCount):
        dfilter = "ip.src == 172.0.0.0/8"
        checkDFilterCount(dfilter, 1)

    def test_cidr_ne_1(self, checkDFilterCount):
        dfilter = "ip.src != 172.25.100.14/32"
        checkDFilterCount(dfilter, 1)

    def test_cidr_ne_2(self, checkDFilterCount):
        dfilter = "ip.src != 172.25.100.0/24"
        checkDFilterCount(dfilter, 1)

    def test_cidr_ne_3(self, checkDFilterCount):
        dfilter = "ip.src != 172.25.0.0/16"
        checkDFilterCount(dfilter, 1)

    def test_cidr_ne_4(self, checkDFilterCount):
        dfilter = "ip.src != 200.0.0.0/8"
        checkDFilterCount(dfilter, 2)

    def test_slice_1(self, checkDFilterCount):
         dfilter = "ip.src[0:2] == ac:19"
         checkDFilterCount(dfilter, 1)

    def test_slice_2(self, checkDFilterCount):
         dfilter = "ip.src[0:2] == 00:00"
         checkDFilterCount(dfilter, 0)

    def test_slice_3(self, checkDFilterCount):
         dfilter = "ip.src[2:2] == 64:0e"
         checkDFilterCount(dfilter, 1)

    def test_slice_4(self, checkDFilterCount):
         dfilter = "ip.src[2:2] == ff:ff"
         checkDFilterCount(dfilter, 0)

    def test_count_1(self, checkDFilterCount):
         dfilter = "count(ip.src) == 1"
         checkDFilterCount(dfilter, 2)

    def test_count_2(self, checkDFilterCount):
         dfilter = "count(ip.addr) == 2"
         checkDFilterCount(dfilter, 2)