2 ( ipv4address_from_octets,
13 most_sig_bit_different,
21 import Test.QuickCheck
23 import qualified Bit as B
28 data IPv4Address = None | IPv4Address { octet1 :: Octet,
35 instance Show IPv4Address where
36 show IPv4Address.None = "None"
37 show addr = concat [(show oct1) ++ ".",
48 instance Arbitrary IPv4Address where
50 oct1 <- arbitrary :: Gen Octet
51 oct2 <- arbitrary :: Gen Octet
52 oct3 <- arbitrary :: Gen Octet
53 oct4 <- arbitrary :: Gen Octet
54 return (IPv4Address oct1 oct2 oct3 oct4)
56 coarbitrary _ = variant 0
60 instance Maskable IPv4Address where
61 apply_mask _ Maskbits.None = IPv4Address.None
63 | mask == ThirtyTwo = ipv4address_from_octets oct1 oct2 oct3 oct4
64 | mask == ThirtyOne = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Seven)
65 | mask == Thirty = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Six)
66 | mask == TwentyNine = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Five)
67 | mask == TwentyEight = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Four)
68 | mask == TwentySeven = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Three)
69 | mask == TwentySix = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 Two)
70 | mask == TwentyFive = ipv4address_from_octets oct1 oct2 oct3 (apply_mask oct4 One)
71 | mask == TwentyFour = ipv4address_from_octets oct1 oct2 oct3 (min_octet)
72 | mask == TwentyThree = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Seven) (min_octet)
73 | mask == TwentyTwo = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Six) (min_octet)
74 | mask == TwentyOne = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Five) (min_octet)
75 | mask == Twenty = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Four) (min_octet)
76 | mask == Nineteen = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Three) (min_octet)
77 | mask == Eighteen = ipv4address_from_octets oct1 oct2 (apply_mask oct3 Two) (min_octet)
78 | mask == Seventeen = ipv4address_from_octets oct1 oct2 (apply_mask oct3 One) (min_octet)
79 | mask == Sixteen = ipv4address_from_octets oct1 oct2 (min_octet) (min_octet)
80 | mask == Fifteen = ipv4address_from_octets oct1 (apply_mask oct2 Seven) (min_octet) (min_octet)
81 | mask == Fourteen = ipv4address_from_octets oct1 (apply_mask oct2 Six) (min_octet) (min_octet)
82 | mask == Thirteen = ipv4address_from_octets oct1 (apply_mask oct2 Five) (min_octet) (min_octet)
83 | mask == Twelve = ipv4address_from_octets oct1 (apply_mask oct2 Four) (min_octet) (min_octet)
84 | mask == Eleven = ipv4address_from_octets oct1 (apply_mask oct2 Three) (min_octet) (min_octet)
85 | mask == Ten = ipv4address_from_octets oct1 (apply_mask oct2 Two) (min_octet) (min_octet)
86 | mask == Nine = ipv4address_from_octets oct1 (apply_mask oct2 One) (min_octet) (min_octet)
87 | mask == Eight = ipv4address_from_octets oct1 (min_octet) (min_octet) (min_octet)
88 | mask == Seven = ipv4address_from_octets (apply_mask oct1 Seven) (min_octet) (min_octet) (min_octet)
89 | mask == Six = ipv4address_from_octets (apply_mask oct1 Six) (min_octet) (min_octet) (min_octet)
90 | mask == Five = ipv4address_from_octets (apply_mask oct1 Five) (min_octet) (min_octet) (min_octet)
91 | mask == Four = ipv4address_from_octets (apply_mask oct1 Four) (min_octet) (min_octet) (min_octet)
92 | mask == Three = ipv4address_from_octets (apply_mask oct1 Three) (min_octet) (min_octet) (min_octet)
93 | mask == Two = ipv4address_from_octets (apply_mask oct1 Two) (min_octet) (min_octet) (min_octet)
94 | mask == One = ipv4address_from_octets (apply_mask oct1 One) (min_octet) (min_octet) (min_octet)
95 | mask == Zero = ipv4address_from_octets (min_octet) (min_octet) (min_octet) (min_octet)
96 | otherwise = IPv4Address.None
104 -- We don't export our constructor so this function is the only
105 -- way to construct an address from octets. As a result, we can
106 -- return IPv4Address.None in response to being passed one of more
107 -- Octet.None octets.
108 ipv4address_from_octets :: Octet -> Octet -> Octet -> Octet -> IPv4Address
109 ipv4address_from_octets oct1 oct2 oct3 oct4
110 | or [oct1 == Octet.None,
113 oct4 == Octet.None] = IPv4Address.None
114 | otherwise = IPv4Address oct1 oct2 oct3 oct4
117 min_address :: IPv4Address -> Maskbits -> IPv4Address
118 min_address IPv4Address.None _ = IPv4Address.None
119 min_address _ Maskbits.None = IPv4Address.None
120 min_address addr mask
121 | mask == ThirtyTwo = addr
122 | mask == ThirtyOne = addr { octet4 = apply_mask oct4 Seven }
123 | mask == Thirty = addr { octet4 = apply_mask oct4 Six }
124 | mask == TwentyNine = addr { octet4 = apply_mask oct4 Five }
125 | mask == TwentyEight = addr { octet4 = apply_mask oct4 Four }
126 | mask == TwentySeven = addr { octet4 = apply_mask oct4 Three }
127 | mask == TwentySix = addr { octet4 = apply_mask oct4 Two }
128 | mask == TwentyFive = addr { octet4 = apply_mask oct4 One }
129 | mask == TwentyFour = addr { octet4 = min_octet }
130 | mask == TwentyThree = addr { octet3 = apply_mask oct3 Seven,
132 | mask == TwentyTwo = addr { octet3 = apply_mask oct3 Six,
134 | mask == TwentyOne = addr { octet3 = apply_mask oct3 Five,
136 | mask == Twenty = addr { octet3 = apply_mask oct3 Four,
138 | mask == Nineteen = addr { octet3 = apply_mask oct3 Three,
140 | mask == Eighteen = addr { octet3 = apply_mask oct3 Two,
142 | mask == Seventeen = addr { octet3 = apply_mask oct3 One,
144 | mask == Sixteen = addr { octet3 = min_octet,
146 | mask == Fifteen = addr { octet2 = apply_mask oct2 Seven,
149 | mask == Fourteen = addr { octet2 = apply_mask oct2 Six,
152 | mask == Thirteen = addr { octet2 = apply_mask oct2 Five,
155 | mask == Twelve = addr { octet2 = apply_mask oct2 Four,
158 | mask == Eleven = addr { octet2 = apply_mask oct2 Three,
161 | mask == Ten = addr { octet2 = apply_mask oct2 Two,
164 | mask == Nine = addr { octet2 = apply_mask oct2 One,
167 | mask == Eight = addr { octet2 = min_octet,
170 | mask == Seven = addr { octet1 = apply_mask oct1 Seven,
174 | mask == Six = addr { octet1 = apply_mask oct1 Six,
178 | mask == Five = addr { octet1 = apply_mask oct1 Five,
182 | mask == Four = addr { octet1 = apply_mask oct1 Four,
186 | mask == Three = addr { octet1 = apply_mask oct1 Three,
190 | mask == Two = addr { octet1 = apply_mask oct1 Two,
194 | mask == One = addr { octet1 = apply_mask oct1 One,
198 | mask == Zero = addr { octet1 = min_octet,
211 max_address :: IPv4Address -> Maskbits -> IPv4Address
212 max_address IPv4Address.None _ = IPv4Address.None
213 max_address _ Maskbits.None = IPv4Address.None
214 max_address addr mask
215 | mask == ThirtyTwo = IPv4Address oct1 oct2 oct3 oct4
216 | mask == ThirtyOne = IPv4Address oct1 oct2 oct3 (Octet a25 a26 a27 a28 a29 a30 a31 B.One)
217 | mask == Thirty = IPv4Address oct1 oct2 oct3 (Octet a25 a26 a27 a28 a29 a30 B.One B.One)
218 | mask == TwentyNine = IPv4Address oct1 oct2 oct3 (Octet a25 a26 a27 a28 a29 B.One B.One B.One)
219 | mask == TwentyEight = IPv4Address oct1 oct2 oct3 (Octet a25 a26 a27 a28 B.One B.One B.One B.One)
220 | mask == TwentySeven = IPv4Address oct1 oct2 oct3 (Octet a25 a26 a27 B.One B.One B.One B.One B.One)
221 | mask == TwentySix = IPv4Address oct1 oct2 oct3 (Octet a25 a26 B.One B.One B.One B.One B.One B.One)
222 | mask == TwentyFive = IPv4Address oct1 oct2 oct3 (Octet a25 B.One B.One B.One B.One B.One B.One B.One)
223 | mask == TwentyFour = IPv4Address oct1 oct2 oct3 (max_octet)
224 | mask == TwentyThree = IPv4Address oct1 oct2 (Octet a17 a18 a19 a20 a21 a22 a23 B.One) (max_octet)
225 | mask == TwentyTwo = IPv4Address oct1 oct2 (Octet a17 a18 a19 a20 a21 a22 B.One B.One) (max_octet)
226 | mask == TwentyOne = IPv4Address oct1 oct2 (Octet a17 a18 a19 a20 a21 B.One B.One B.One) (max_octet)
227 | mask == Twenty = IPv4Address oct1 oct2 (Octet a17 a18 a19 a20 B.One B.One B.One B.One) (max_octet)
228 | mask == Nineteen = IPv4Address oct1 oct2 (Octet a17 a18 a19 B.One B.One B.One B.One B.One) (max_octet)
229 | mask == Eighteen = IPv4Address oct1 oct2 (Octet a17 a18 B.One B.One B.One B.One B.One B.One) (max_octet)
230 | mask == Seventeen = IPv4Address oct1 oct2 (Octet a17 B.One B.One B.One B.One B.One B.One B.One) (max_octet)
231 | mask == Sixteen = IPv4Address oct1 oct2 (max_octet) (max_octet)
232 | mask == Fifteen = IPv4Address oct1 (Octet a9 a10 a11 a12 a13 a14 a15 B.One) (max_octet) (max_octet)
233 | mask == Fourteen = IPv4Address oct1 (Octet a9 a10 a11 a12 a13 a14 B.One B.One) (max_octet) (max_octet)
234 | mask == Thirteen = IPv4Address oct1 (Octet a9 a10 a11 a12 a13 B.One B.One B.One) (max_octet) (max_octet)
235 | mask == Twelve = IPv4Address oct1 (Octet a9 a10 a11 a12 B.One B.One B.One B.One) (max_octet) (max_octet)
236 | mask == Eleven = IPv4Address oct1 (Octet a9 a10 a11 B.One B.One B.One B.One B.One) (max_octet) (max_octet)
237 | mask == Ten = IPv4Address oct1 (Octet a9 a10 B.One B.One B.One B.One B.One B.One) (max_octet) (max_octet)
238 | mask == Nine = IPv4Address oct1 (Octet a9 B.One B.One B.One B.One B.One B.One B.One) (max_octet) (max_octet)
239 | mask == Eight = IPv4Address oct1 (max_octet) (max_octet) (max_octet)
240 | mask == Seven = IPv4Address (Octet a1 a2 a3 a4 a5 a6 a7 B.One) (max_octet) (max_octet) (max_octet)
241 | mask == Six = IPv4Address (Octet a1 a2 a3 a4 a5 a6 B.One B.One) (max_octet) (max_octet) (max_octet)
242 | mask == Five = IPv4Address (Octet a1 a2 a3 a4 a5 B.One B.One B.One) (max_octet) (max_octet) (max_octet)
243 | mask == Four = IPv4Address (Octet a1 a2 a3 a4 B.One B.One B.One B.One) (max_octet) (max_octet) (max_octet)
244 | mask == Three = IPv4Address (Octet a1 a2 a3 B.One B.One B.One B.One B.One) (max_octet) (max_octet) (max_octet)
245 | mask == Two = IPv4Address (Octet a1 a2 B.One B.One B.One B.One B.One B.One) (max_octet) (max_octet) (max_octet)
246 | mask == One = IPv4Address (Octet a1 B.One B.One B.One B.One B.One B.One B.One) (max_octet) (max_octet) (max_octet)
247 | mask == Zero = IPv4Address (max_octet) (max_octet) (max_octet) (max_octet)
285 min_octet1 :: IPv4Address -> Maskbits -> Octet
286 min_octet1 addr mask = octet1 (min_address addr mask)
288 min_octet2 :: IPv4Address -> Maskbits -> Octet
289 min_octet2 addr mask = octet2 (min_address addr mask)
291 min_octet3 :: IPv4Address -> Maskbits -> Octet
292 min_octet3 addr mask = octet3 (min_address addr mask)
294 min_octet4 :: IPv4Address -> Maskbits -> Octet
295 min_octet4 addr mask = octet4 (min_address addr mask)
297 max_octet1 :: IPv4Address -> Maskbits -> Octet
298 max_octet1 addr mask = octet1 (max_address addr mask)
300 max_octet2 :: IPv4Address -> Maskbits -> Octet
301 max_octet2 addr mask = octet2 (max_address addr mask)
303 max_octet3 :: IPv4Address -> Maskbits -> Octet
304 max_octet3 addr mask = octet3 (max_address addr mask)
306 max_octet4 :: IPv4Address -> Maskbits -> Octet
307 max_octet4 addr mask = octet4 (max_address addr mask)
310 -- Given two addresses, find the number of the most significant bit
311 -- where they differ. If the addresses are the same, return
313 most_sig_bit_different :: IPv4Address -> IPv4Address -> Maskbits
314 most_sig_bit_different addr1 addr2
315 | addr1 == addr2 = Maskbits.Zero
316 | m1 /= n1 = Maskbits.One
326 | m11 /= n11 = Eleven
327 | m12 /= n12 = Twelve
328 | m13 /= n13 = Thirteen
329 | m14 /= n14 = Fourteen
330 | m15 /= n15 = Fifteen
331 | m16 /= n16 = Sixteen
332 | m17 /= n17 = Seventeen
333 | m18 /= n18 = Eighteen
334 | m19 /= n19 = Nineteen
335 | m20 /= n20 = Twenty
336 | m21 /= n21 = TwentyOne
337 | m22 /= n22 = TwentyTwo
338 | m23 /= n23 = TwentyThree
339 | m24 /= n24 = TwentyFour
340 | m25 /= n25 = TwentyFive
341 | m26 /= n26 = TwentySix
342 | m27 /= n27 = TwentySeven
343 | m28 /= n28 = TwentyEight
344 | m29 /= n29 = TwentyNine
345 | m30 /= n30 = Thirty
346 | m31 /= n31 = ThirtyOne
347 | m32 /= n32 = ThirtyTwo
348 | otherwise = Maskbits.Zero
382 oct1a = (octet1 addr1)
383 oct2a = (octet2 addr1)
384 oct3a = (octet3 addr1)
385 oct4a = (octet4 addr1)
418 oct1b = (octet1 addr2)
419 oct2b = (octet2 addr2)
420 oct3b = (octet3 addr2)
421 oct4b = (octet4 addr2)
426 mk_testaddr :: Int -> Int -> Int -> Int -> IPv4Address
427 mk_testaddr a b c d =
428 IPv4Address oct1 oct2 oct3 oct4
430 oct1 = octet_from_int a
431 oct2 = octet_from_int b
432 oct3 = octet_from_int c
433 oct4 = octet_from_int d
436 test_most_sig_bit_different1 :: Test
437 test_most_sig_bit_different1 =
438 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))
441 test_most_sig_bit_different2 :: Test
442 test_most_sig_bit_different2 =
443 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))
446 ipv4address_tests :: [Test]
447 ipv4address_tests = [ test_most_sig_bit_different1,
448 test_most_sig_bit_different2 ]