]> gitweb.michael.orlitzky.com - hath.git/blobdiff - src/Cidr.hs
Moved the min/max address/octetN functions out of IPv4Address and in to Cidr.
[hath.git] / src / Cidr.hs
index 98c42c7e6bdab38a0f2d1da47c142aba445c7182..06704fb5e37479b8bade2a9420a4b824cfeb60bc 100644 (file)
@@ -5,6 +5,14 @@ module Cidr
   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
@@ -13,6 +21,7 @@ import Data.List (nubBy)
 import Test.HUnit
 import Test.QuickCheck
 
+import qualified Bit as B
 import IPv4Address
 import ListUtils
 import Maskable
@@ -78,6 +87,200 @@ cidr_from_string s
 
 
 
+min_address :: Cidr -> IPv4Address
+min_address (Cidr IPv4Address.None _) = IPv4Address.None
+min_address (Cidr _ Maskbits.None)    = IPv4Address.None
+min_address (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 = addr { octet2 = min_octet,
+                             octet3 = min_octet,
+                             octet4 = min_octet }
+    | mask == Seven = addr { octet1 = apply_mask oct1 Seven,
+                             octet2 = min_octet,
+                             octet3 = min_octet,
+                             octet4 = min_octet }
+    | mask == Six = addr { octet1 = apply_mask oct1 Six,
+                           octet2 = min_octet,
+                           octet3 = min_octet,
+                           octet4 = min_octet }
+    | mask == Five = addr { octet1 = apply_mask oct1 Five,
+                            octet2 = min_octet,
+                            octet3 = min_octet,
+                            octet4 = min_octet }
+    | mask == Four = addr { octet1 = apply_mask oct1 Four,
+                            octet2 = min_octet,
+                            octet3 = min_octet,
+                            octet4 = min_octet }
+    | mask == Three = addr { octet1 = apply_mask oct1 Three,
+                             octet2 = min_octet,
+                             octet3 = min_octet,
+                             octet4 = min_octet }
+    | mask == Two = addr { octet1 = apply_mask oct1 Two,
+                           octet2 = min_octet,
+                           octet3 = min_octet,
+                           octet4 = min_octet }
+    | mask == One = addr { octet1 = apply_mask oct1 One,
+                           octet2 = min_octet,
+                           octet3 = min_octet,
+                           octet4 = min_octet }
+    | mask == Zero = addr { octet1 = min_octet,
+                            octet2 = min_octet,
+                            octet3 = min_octet,
+                            octet4 = min_octet }
+    | otherwise = addr
+    where
+      oct1 = (octet1 addr)
+      oct2 = (octet2 addr)
+      oct3 = (octet3 addr)
+      oct4 = (octet4 addr)
+
+
+
+max_address :: Cidr -> IPv4Address
+max_address (Cidr IPv4Address.None _) = IPv4Address.None
+max_address (Cidr _ Maskbits.None)    = IPv4Address.None
+max_address (Cidr addr mask)
+    | mask == ThirtyTwo = ipv4address_from_octets oct1 oct2 oct3 oct4
+    | mask == ThirtyOne = ipv4address_from_octets oct1 oct2 oct3 (Octet a25 a26 a27 a28 a29 a30 a31 B.One)
+    | mask == Thirty = ipv4address_from_octets oct1 oct2 oct3 (Octet a25 a26 a27 a28 a29 a30 B.One B.One)
+    | mask == TwentyNine = ipv4address_from_octets oct1 oct2 oct3 (Octet a25 a26 a27 a28 a29 B.One B.One B.One)
+    | mask == TwentyEight = ipv4address_from_octets oct1 oct2 oct3 (Octet a25 a26 a27 a28 B.One B.One B.One B.One)
+    | mask == TwentySeven = ipv4address_from_octets oct1 oct2 oct3 (Octet a25 a26 a27 B.One B.One B.One B.One B.One)
+    | mask == TwentySix = ipv4address_from_octets oct1 oct2 oct3 (Octet a25 a26 B.One B.One B.One B.One B.One B.One)
+    | mask == TwentyFive = ipv4address_from_octets oct1 oct2 oct3 (Octet a25 B.One B.One B.One B.One B.One B.One B.One)
+    | mask == TwentyFour = ipv4address_from_octets oct1 oct2 oct3 (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)
+      a25 = (b1 oct4)
+      a26 = (b2 oct4)
+      a27 = (b3 oct4)
+      a28 = (b4 oct4)
+      a29 = (b5 oct4)
+      a30 = (b6 oct4)
+      a31 = (b7 oct4)
+      oct1 = (octet1 addr)
+      oct2 = (octet2 addr)
+      oct3 = (octet3 addr)
+      oct4 = (octet4 addr)
+
+
+
+min_octet1 :: Cidr -> Octet
+min_octet1 cidr = octet1 (min_address cidr)
+
+min_octet2 :: Cidr -> Octet
+min_octet2 cidr = octet2 (min_address cidr)
+
+min_octet3 :: Cidr -> Octet
+min_octet3 cidr = octet3 (min_address cidr)
+
+min_octet4 :: Cidr -> Octet
+min_octet4 cidr = octet4 (min_address cidr)
+
+max_octet1 :: Cidr -> Octet
+max_octet1 cidr = octet1 (max_address cidr)
+
+max_octet2 :: Cidr -> Octet
+max_octet2 cidr = octet2 (max_address cidr)
+
+max_octet3 :: Cidr -> Octet
+max_octet3 cidr = octet3 (max_address cidr)
+
+max_octet4 :: Cidr -> Octet
+max_octet4 cidr = octet4 (max_address 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