2 ( ipv4address_from_octets,
7 most_sig_bit_different,
15 import Test.QuickCheck
21 data IPv4Address = None | IPv4Address { octet1 :: Octet,
28 instance Show IPv4Address where
29 show IPv4Address.None = "None"
30 show addr = concat [(show oct1) ++ ".",
41 instance Arbitrary IPv4Address where
43 oct1 <- arbitrary :: Gen Octet
44 oct2 <- arbitrary :: Gen Octet
45 oct3 <- arbitrary :: Gen Octet
46 oct4 <- arbitrary :: Gen Octet
47 return (IPv4Address oct1 oct2 oct3 oct4)
49 coarbitrary _ = variant 0
53 instance Maskable IPv4Address where
54 apply_mask _ Maskbits.None = IPv4Address.None
56 | mask == ThirtyTwo = ipv4address_from_octets oct1 oct2 oct3 oct4
57 | mask == ThirtyOne = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Seven)
58 | mask == Thirty = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Six)
59 | mask == TwentyNine = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Five)
60 | mask == TwentyEight = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Four)
61 | mask == TwentySeven = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Three)
62 | mask == TwentySix = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Two)
63 | mask == TwentyFive = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 One)
64 | mask == TwentyFour = ipv4address_from_octets oct1 oct2 oct3 (min_octet)
65 | mask == TwentyThree = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Seven) (min_octet)
66 | mask == TwentyTwo = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Six) (min_octet)
67 | mask == TwentyOne = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Five) (min_octet)
68 | mask == Twenty = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Four) (min_octet)
69 | mask == Nineteen = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Three) (min_octet)
70 | mask == Eighteen = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Two) (min_octet)
71 | mask == Seventeen = ipv4address_from_octets oct1 oct2 (apply_mask oct3 One) (min_octet)
72 | mask == Sixteen = ipv4address_from_octets oct1 oct2 (min_octet) (min_octet)
73 | mask == Fifteen = ipv4address_from_octets oct1 (apply_mask oct2 Seven) (min_octet) (min_octet)
74 | mask == Fourteen = ipv4address_from_octets oct1 (apply_mask oct2 Six) (min_octet) (min_octet)
75 | mask == Thirteen = ipv4address_from_octets oct1 (apply_mask oct2 Five) (min_octet) (min_octet)
76 | mask == Twelve = ipv4address_from_octets oct1 (apply_mask oct2 Four) (min_octet) (min_octet)
77 | mask == Eleven = ipv4address_from_octets oct1 (apply_mask oct2 Three) (min_octet) (min_octet)
78 | mask == Ten = ipv4address_from_octets oct1 (apply_mask oct2 Two) (min_octet) (min_octet)
79 | mask == Nine = ipv4address_from_octets oct1 (apply_mask oct2 One) (min_octet) (min_octet)
80 | mask == Eight = ipv4address_from_octets oct1 (min_octet) (min_octet) (min_octet)
81 | mask == Seven = ipv4address_from_octets (apply_mask oct1 Seven) (min_octet) (min_octet) (min_octet)
82 | mask == Six = ipv4address_from_octets (apply_mask oct1 Six) (min_octet) (min_octet) (min_octet)
83 | mask == Five = ipv4address_from_octets (apply_mask oct1 Five) (min_octet) (min_octet) (min_octet)
84 | mask == Four = ipv4address_from_octets (apply_mask oct1 Four) (min_octet) (min_octet) (min_octet)
85 | mask == Three = ipv4address_from_octets (apply_mask oct1 Three) (min_octet) (min_octet) (min_octet)
86 | mask == Two = ipv4address_from_octets (apply_mask oct1 Two) (min_octet) (min_octet) (min_octet)
87 | mask == One = ipv4address_from_octets (apply_mask oct1 One) (min_octet) (min_octet) (min_octet)
88 | mask == Zero = ipv4address_from_octets (min_octet) (min_octet) (min_octet) (min_octet)
89 | otherwise = IPv4Address.None
97 -- We don't export our constructor so this function is the only
98 -- way to construct an address from octets. As a result, we can
99 -- return IPv4Address.None in response to being passed one of more
100 -- Octet.None octets.
101 ipv4address_from_octets :: Octet -> Octet -> Octet -> Octet -> IPv4Address
102 ipv4address_from_octets oct1 oct2 oct3 oct4
103 | or [oct1 == Octet.None,
106 oct4 == Octet.None] = IPv4Address.None
107 | otherwise = IPv4Address oct1 oct2 oct3 oct4
110 -- The minimum possible IPv4 address, 0.0.0.0.
111 min_address :: IPv4Address
112 min_address = IPv4Address min_octet min_octet min_octet min_octet
115 -- The maximum possible IPv4 address, 255.255.255.255.
116 max_address :: IPv4Address
117 max_address = IPv4Address max_octet max_octet max_octet max_octet
120 -- Given two addresses, find the number of the most significant bit
121 -- where they differ. If the addresses are the same, return
123 most_sig_bit_different :: IPv4Address -> IPv4Address -> Maskbits
124 most_sig_bit_different addr1 addr2
125 | addr1 == addr2 = Maskbits.Zero
126 | m1 /= n1 = Maskbits.One
136 | m11 /= n11 = Eleven
137 | m12 /= n12 = Twelve
138 | m13 /= n13 = Thirteen
139 | m14 /= n14 = Fourteen
140 | m15 /= n15 = Fifteen
141 | m16 /= n16 = Sixteen
142 | m17 /= n17 = Seventeen
143 | m18 /= n18 = Eighteen
144 | m19 /= n19 = Nineteen
145 | m20 /= n20 = Twenty
146 | m21 /= n21 = TwentyOne
147 | m22 /= n22 = TwentyTwo
148 | m23 /= n23 = TwentyThree
149 | m24 /= n24 = TwentyFour
150 | m25 /= n25 = TwentyFive
151 | m26 /= n26 = TwentySix
152 | m27 /= n27 = TwentySeven
153 | m28 /= n28 = TwentyEight
154 | m29 /= n29 = TwentyNine
155 | m30 /= n30 = Thirty
156 | m31 /= n31 = ThirtyOne
157 | m32 /= n32 = ThirtyTwo
158 | otherwise = Maskbits.Zero
192 oct1a = (octet1 addr1)
193 oct2a = (octet2 addr1)
194 oct3a = (octet3 addr1)
195 oct4a = (octet4 addr1)
228 oct1b = (octet1 addr2)
229 oct2b = (octet2 addr2)
230 oct3b = (octet3 addr2)
231 oct4b = (octet4 addr2)
236 mk_testaddr :: Int -> Int -> Int -> Int -> IPv4Address
237 mk_testaddr a b c d =
238 IPv4Address oct1 oct2 oct3 oct4
240 oct1 = octet_from_int a
241 oct2 = octet_from_int b
242 oct3 = octet_from_int c
243 oct4 = octet_from_int d
246 test_most_sig_bit_different1 :: Test
247 test_most_sig_bit_different1 =
248 TestCase $ assertEqual "10.1.1.0 and 10.1.0.0 differ in bit 24" TwentyFour (most_sig_bit_different (mk_testaddr 10 1 1 0) (mk_testaddr 10 1 0 0))
251 test_most_sig_bit_different2 :: Test
252 test_most_sig_bit_different2 =
253 TestCase $ assertEqual "10.1.2.0 and 10.1.1.0 differ in bit 23" TwentyThree (most_sig_bit_different (mk_testaddr 10 1 2 0) (mk_testaddr 10 1 1 0))
256 ipv4address_tests :: [Test]
257 ipv4address_tests = [ test_most_sig_bit_different1,
258 test_most_sig_bit_different2 ]