]> gitweb.michael.orlitzky.com - hath.git/blobdiff - src/Cidr.hs
Use min_address to simplify some of the min_host cases.
[hath.git] / src / Cidr.hs
index 7bb3ac2c41229b286f67f39a8ce58c24274171b7..41ea9ee9a711b6af63c928419f77a7363fbaf20e 100644 (file)
@@ -2,13 +2,26 @@ module Cidr
 ( Cidr(..),
   cidr_from_string,
   cidr_tests,
-  combine_all
+  combine_all,
+  contains,
+  contains_proper,
+  max_octet1,
+  max_octet2,
+  max_octet3,
+  max_octet4,
+  min_octet1,
+  min_octet2,
+  min_octet3,
+  min_octet4,
+  prop_all_cidrs_contain_themselves,
+  prop_contains_proper_intransitive
 ) where
 
 import Data.List (nubBy)
 import Test.HUnit
 import Test.QuickCheck
 
+import qualified Bit as B
 import IPv4Address
 import ListUtils
 import Maskable
@@ -74,6 +87,189 @@ cidr_from_string s
 
 
 
+min_host :: Cidr -> IPv4Address
+min_host Cidr.None                 = IPv4Address.None
+min_host (Cidr IPv4Address.None _) = IPv4Address.None
+min_host (Cidr _ Maskbits.None)    = IPv4Address.None
+min_host (Cidr addr mask)
+    | mask == ThirtyTwo = addr
+    | mask == ThirtyOne = addr { octet4 = apply_mask oct4 Seven }
+    | mask == Thirty = addr { octet4 = apply_mask oct4 Six }
+    | mask == TwentyNine = addr { octet4 = apply_mask oct4 Five }
+    | mask == TwentyEight = addr { octet4 = apply_mask oct4 Four }
+    | mask == TwentySeven = addr { octet4 = apply_mask oct4 Three }
+    | mask == TwentySix = addr { octet4 = apply_mask oct4 Two }
+    | mask == TwentyFive = addr { octet4 = apply_mask oct4 One }
+    | mask == TwentyFour = addr { octet4 = min_octet }
+    | mask == TwentyThree = addr { octet3 = apply_mask oct3 Seven,
+                                   octet4 = min_octet }
+    | mask == TwentyTwo = addr { octet3 = apply_mask oct3 Six,
+                                 octet4 = min_octet }
+    | mask == TwentyOne = addr { octet3 = apply_mask oct3 Five,
+                                 octet4 = min_octet }
+    | mask == Twenty = addr { octet3 = apply_mask oct3 Four,
+                              octet4 = min_octet }
+    | mask == Nineteen = addr { octet3 = apply_mask oct3 Three,
+                                octet4 = min_octet }
+    | mask == Eighteen = addr { octet3 = apply_mask oct3 Two,
+                                octet4 = min_octet }
+    | mask == Seventeen = addr { octet3 = apply_mask oct3 One,
+                                 octet4 = min_octet }
+    | mask == Sixteen = addr { octet3 = min_octet,
+                               octet4 = min_octet }
+    | mask == Fifteen = addr { octet2 = apply_mask oct2 Seven,
+                               octet3 = min_octet,
+                               octet4 = min_octet }
+    | mask == Fourteen = addr { octet2 = apply_mask oct2 Six,
+                                octet3 = min_octet,
+                                octet4 = min_octet }
+    | mask == Thirteen = addr { octet2 = apply_mask oct2 Five,
+                                octet3 = min_octet,
+                                octet4 = min_octet }
+    | mask == Twelve = addr { octet2 = apply_mask oct2 Four,
+                              octet3 = min_octet,
+                              octet4 = min_octet }
+    | mask == Eleven = addr { octet2 = apply_mask oct2 Three,
+                              octet3 = min_octet,
+                              octet4 = min_octet }
+    | mask == Ten = addr { octet2 = apply_mask oct2 Two,
+                           octet3 = min_octet,
+                           octet4 = min_octet }
+    | mask == Nine = addr { octet2 = apply_mask oct2 One,
+                            octet3 = min_octet,
+                            octet4 = min_octet }
+    | mask == Eight = min_address { octet1 = oct1 }
+    | mask == Seven = min_address { octet1 = apply_mask oct1 Seven }
+    | mask == Six = min_address { octet1 = apply_mask oct1 Six }
+    | mask == Five = min_address { octet1 = apply_mask oct1 Five }
+    | mask == Four = min_address { octet1 = apply_mask oct1 Four }
+    | mask == Three = min_address { octet1 = apply_mask oct1 Three }
+    | mask == Two = min_address { octet1 = apply_mask oct1 Two }
+    | mask == One = min_address { octet1 = apply_mask oct1 One }
+    | mask == Zero = min_address
+    | otherwise = addr
+    where
+      oct1 = (octet1 addr)
+      oct2 = (octet2 addr)
+      oct3 = (octet3 addr)
+      oct4 = (octet4 addr)
+
+
+
+max_host :: Cidr -> IPv4Address
+max_host Cidr.None                 = IPv4Address.None
+max_host (Cidr IPv4Address.None _) = IPv4Address.None
+max_host (Cidr _ Maskbits.None)    = IPv4Address.None
+max_host (Cidr addr mask)
+    | mask == ThirtyTwo = addr
+    | mask == ThirtyOne = addr { octet4 = oct4 { b8 = B.One } }
+    | mask == Thirty = addr { octet4 = oct4 { b7 = B.One, b8 = B.One } }
+    | mask == TwentyNine = addr { octet4 = oct4 { b6 = B.One,
+                                                  b7 = B.One,
+                                                  b8 = B.One } }
+    | mask == TwentyEight = addr { octet4 = oct4 { b5 = B.One,
+                                                   b6 = B.One,
+                                                   b7 = B.One,
+                                                   b8 = B.One } }
+    | mask == TwentySeven = addr { octet4 = oct4 { b4 = B.One,
+                                                   b5 = B.One,
+                                                   b6 = B.One,
+                                                   b7 = B.One,
+                                                   b8 = B.One } }
+    | mask == TwentySix = addr { octet4 = oct4 { b3 = B.One,
+                                                 b4 = B.One,
+                                                 b5 = B.One,
+                                                 b6 = B.One,
+                                                 b7 = B.One,
+                                                 b8 = B.One } }
+    | mask == TwentyFive = addr { octet4 = oct4 { b2 = B.One,
+                                                  b3 = B.One,
+                                                  b4 = B.One,
+                                                  b5 = B.One,
+                                                  b6 = B.One,
+                                                  b7 = B.One,
+                                                  b8 = B.One } }
+    | mask == TwentyFour = addr { octet4 = max_octet }
+    | mask == TwentyThree = ipv4address_from_octets oct1 oct2 (Octet a17 a18 a19 a20 a21 a22 a23 B.One) (max_octet)
+    | mask == TwentyTwo = ipv4address_from_octets oct1 oct2 (Octet a17 a18 a19 a20 a21 a22 B.One B.One) (max_octet)
+    | mask == TwentyOne = ipv4address_from_octets oct1 oct2 (Octet a17 a18 a19 a20 a21 B.One B.One B.One) (max_octet)
+    | mask == Twenty = ipv4address_from_octets oct1 oct2 (Octet a17 a18 a19 a20 B.One B.One B.One B.One) (max_octet)
+    | mask == Nineteen = ipv4address_from_octets oct1 oct2 (Octet a17 a18 a19 B.One B.One B.One B.One B.One) (max_octet)
+    | mask == Eighteen = ipv4address_from_octets oct1 oct2 (Octet a17 a18 B.One B.One B.One B.One B.One B.One) (max_octet)
+    | mask == Seventeen = ipv4address_from_octets oct1 oct2 (Octet a17 B.One B.One B.One B.One B.One B.One B.One) (max_octet)
+    | mask == Sixteen = ipv4address_from_octets oct1 oct2 (max_octet) (max_octet)
+    | mask == Fifteen = ipv4address_from_octets oct1 (Octet a9 a10 a11 a12 a13 a14 a15 B.One) (max_octet) (max_octet)
+    | mask == Fourteen = ipv4address_from_octets oct1 (Octet a9 a10 a11 a12 a13 a14 B.One B.One) (max_octet) (max_octet)
+    | mask == Thirteen = ipv4address_from_octets oct1 (Octet a9 a10 a11 a12 a13 B.One B.One B.One) (max_octet) (max_octet)
+    | mask == Twelve = ipv4address_from_octets oct1 (Octet a9 a10 a11 a12 B.One B.One B.One B.One) (max_octet) (max_octet)
+    | mask == Eleven = ipv4address_from_octets oct1 (Octet a9 a10 a11 B.One B.One B.One B.One B.One) (max_octet) (max_octet)
+    | mask == Ten = ipv4address_from_octets oct1 (Octet a9 a10 B.One B.One B.One B.One B.One B.One) (max_octet) (max_octet)
+    | mask == Nine = ipv4address_from_octets oct1 (Octet a9 B.One B.One B.One B.One B.One B.One B.One) (max_octet) (max_octet)
+    | mask == Eight = ipv4address_from_octets oct1 (max_octet) (max_octet) (max_octet)
+    | mask == Seven = ipv4address_from_octets (Octet a1 a2 a3 a4 a5 a6 a7 B.One) (max_octet) (max_octet) (max_octet)
+    | mask == Six = ipv4address_from_octets (Octet a1 a2 a3 a4 a5 a6 B.One B.One) (max_octet) (max_octet) (max_octet)
+    | mask == Five = ipv4address_from_octets (Octet a1 a2 a3 a4 a5 B.One B.One B.One) (max_octet) (max_octet) (max_octet)
+    | mask == Four = ipv4address_from_octets (Octet a1 a2 a3 a4 B.One B.One B.One B.One) (max_octet) (max_octet) (max_octet)
+    | mask == Three = ipv4address_from_octets (Octet a1 a2 a3 B.One B.One B.One B.One B.One) (max_octet) (max_octet) (max_octet)
+    | mask == Two = ipv4address_from_octets (Octet a1 a2 B.One B.One B.One B.One B.One B.One) (max_octet) (max_octet) (max_octet)
+    | mask == One = ipv4address_from_octets (Octet a1 B.One B.One B.One B.One B.One B.One B.One) (max_octet) (max_octet) (max_octet)
+    | mask == Zero = ipv4address_from_octets (max_octet) (max_octet) (max_octet) (max_octet)
+    | otherwise = addr
+    where
+      a1  = (b1 oct1)
+      a2  = (b2 oct1)
+      a3  = (b3 oct1)
+      a4  = (b4 oct1)
+      a5  = (b5 oct1)
+      a6  = (b6 oct1)
+      a7  = (b7 oct1)
+      a9  = (b1 oct2)
+      a10 = (b2 oct2)
+      a11 = (b3 oct2)
+      a12 = (b4 oct2)
+      a13 = (b5 oct2)
+      a14 = (b6 oct2)
+      a15 = (b7 oct2)
+      a17 = (b1 oct3)
+      a18 = (b2 oct3)
+      a19 = (b3 oct3)
+      a20 = (b4 oct3)
+      a21 = (b5 oct3)
+      a22 = (b6 oct3)
+      a23 = (b7 oct3)
+      oct1 = (octet1 addr)
+      oct2 = (octet2 addr)
+      oct3 = (octet3 addr)
+      oct4 = (octet4 addr)
+
+
+
+min_octet1 :: Cidr -> Octet
+min_octet1 cidr = octet1 (min_host cidr)
+
+min_octet2 :: Cidr -> Octet
+min_octet2 cidr = octet2 (min_host cidr)
+
+min_octet3 :: Cidr -> Octet
+min_octet3 cidr = octet3 (min_host cidr)
+
+min_octet4 :: Cidr -> Octet
+min_octet4 cidr = octet4 (min_host cidr)
+
+max_octet1 :: Cidr -> Octet
+max_octet1 cidr = octet1 (max_host cidr)
+
+max_octet2 :: Cidr -> Octet
+max_octet2 cidr = octet2 (max_host cidr)
+
+max_octet3 :: Cidr -> Octet
+max_octet3 cidr = octet3 (max_host cidr)
+
+max_octet4 :: Cidr -> Octet
+max_octet4 cidr = octet4 (max_host cidr)
+
+
+
 -- Return true if the first argument (a CIDR range) contains the
 -- second (another CIDR range). There are a lot of ways we can be fed
 -- junk here. For lack of a better alternative, just return False when
@@ -123,7 +319,7 @@ contains (Cidr addr1 mbits1) (Cidr addr2 mbits2)
 
 contains_proper :: Cidr -> Cidr -> Bool
 contains_proper cidr1 cidr2 =
-    (cidr1 `contains` cidr2) && (not (cidr1 == cidr2))
+    (cidr1 `contains` cidr2) && (not (cidr2 `contains` cidr1))
 
 
 -- A CIDR range is redundant (with respect to the given list) if
@@ -260,7 +456,7 @@ test_adjacent4 =
 
 test_combine_contained1 :: Test
 test_combine_contained1 =
-    TestCase $ assertEqual "10.0.0.0/8, 10.1.0.0/16, and 10.1.1.0/24 combine to 10.0.0.0/8" expected_cidrs (combine_all test_cidrs)
+    TestCase $ assertEqual "10.0.0.0/8, 10.1.0.0/16, and 10.1.1.0/24 combine to 10.0.0.0/8" expected_cidrs (combine_contained test_cidrs)
     where
       cidr1 = cidr_from_string "10.0.0.0/8"
       cidr2 = cidr_from_string "10.1.0.0/16"
@@ -271,9 +467,32 @@ test_combine_contained1 =
 
 test_combine_contained2 :: Test
 test_combine_contained2 =
-    TestCase $ assertEqual "192.168.1.0/24 combines with itself" [cidr1] (combine_all [cidr1, cidr1])
+    TestCase $ assertEqual "192.168.3.0/23 does not contain 192.168.1.0/24" [cidr1, cidr2] (combine_contained [cidr1, cidr2])
+    where
+      cidr1 = cidr_from_string "192.168.3.0/23"
+      cidr2 = cidr_from_string "192.168.1.0/24"
+
+
+test_combine_all1 :: Test
+test_combine_all1 =
+    TestCase $ assertEqual "10.0.0.0/24 is adjacent to 10.0.1.0/24 and 10.0.3.0/23 contains 10.0.2.0/24" expected_cidrs (combine_all test_cidrs)
     where
-      cidr1 = cidr_from_string "192.168.1.0/24"
+      cidr1 = cidr_from_string "10.0.0.0/24"
+      cidr2 = cidr_from_string "10.0.1.0/24"
+      cidr3 = cidr_from_string "10.0.2.0/24"
+      cidr4 = cidr_from_string "10.0.3.0/23"
+      cidr5 = cidr_from_string "10.0.0.0/23"
+      expected_cidrs = [cidr4, cidr5]
+      test_cidrs = [cidr1, cidr2, cidr3, cidr4]
+
+
+test_combine_all2 :: Test
+test_combine_all2 =
+    TestCase $ assertEqual "127.0.0.1/32 combines with itself recursively" expected_cidrs (combine_all test_cidrs)
+    where
+      cidr1 = cidr_from_string "127.0.0.1/32"
+      expected_cidrs = [cidr1]
+      test_cidrs = [cidr1, cidr1, cidr1, cidr1, cidr1]
 
 
 cidr_tests :: [Test]
@@ -287,5 +506,20 @@ cidr_tests = [ test_equality1,
                test_adjacent3,
                test_adjacent4,
                test_combine_contained1,
-               test_combine_contained2
+               test_combine_contained2,
+               test_combine_all1,
+               test_combine_all2
              ]
+
+
+-- QuickCheck Tests
+prop_all_cidrs_contain_themselves :: Cidr -> Bool
+prop_all_cidrs_contain_themselves cidr1 = cidr1 `contains` cidr1
+
+
+-- If cidr1 properly contains cidr2, then by definition cidr2
+-- does not properly contain cidr1.
+prop_contains_proper_intransitive :: Cidr -> Cidr -> Property
+prop_contains_proper_intransitive cidr1 cidr2 =
+    (cidr1 `contains_proper` cidr2) ==>
+      (not (cidr2 `contains_proper` cidr1))