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)
51 instance Maskable IPv4Address where
52 apply_mask _ Maskbits.None _ = IPv4Address.None
53 apply_mask addr mask bit
54 | mask == ThirtyTwo = addr
55 | mask == ThirtyOne = addr { octet4 = (apply_mask oct4 Seven bit) }
56 | mask == Thirty = addr { octet4 = (apply_mask oct4 Six bit) }
57 | mask == TwentyNine = addr { octet4 = (apply_mask oct4 Five bit) }
58 | mask == TwentyEight = addr { octet4 = (apply_mask oct4 Four bit) }
59 | mask == TwentySeven = addr { octet4 = (apply_mask oct4 Three bit) }
60 | mask == TwentySix = addr { octet4 = (apply_mask oct4 Two bit) }
61 | mask == TwentyFive = addr { octet4 = (apply_mask oct4 One bit) }
62 | mask == TwentyFour = addr { octet4 = (apply_mask oct4 Zero bit) }
63 | mask == TwentyThree = addr { octet3 = (apply_mask oct3 Seven bit),
64 octet4 = (apply_mask oct4 Zero bit) }
65 | mask == TwentyTwo = addr { octet3 = (apply_mask oct3 Six bit),
66 octet4 = (apply_mask oct4 Zero bit) }
67 | mask == TwentyOne = addr { octet3 = (apply_mask oct3 Five bit),
68 octet4 = (apply_mask oct4 Zero bit) }
69 | mask == Twenty = addr { octet3 = (apply_mask oct3 Four bit),
70 octet4 = (apply_mask oct4 Zero bit) }
71 | mask == Nineteen = addr { octet3 = (apply_mask oct3 Three bit),
72 octet4 = (apply_mask oct4 Zero bit) }
73 | mask == Eighteen = addr { octet3 = (apply_mask oct3 Two bit),
74 octet4 = (apply_mask oct4 Zero bit) }
75 | mask == Seventeen = addr { octet3 = (apply_mask oct3 One bit),
76 octet4 = (apply_mask oct4 Zero bit) }
77 | mask == Sixteen = addr { octet3 = (apply_mask oct3 Zero bit),
78 octet4 = (apply_mask oct4 Zero bit) }
79 | mask == Fifteen = addr { octet2 = (apply_mask oct2 Seven bit),
80 octet3 = (apply_mask oct3 Zero bit),
81 octet4 = (apply_mask oct4 Zero bit) }
82 | mask == Fourteen = addr { octet2 = (apply_mask oct2 Six bit),
83 octet3 = (apply_mask oct3 Zero bit),
84 octet4 = (apply_mask oct4 Zero bit)}
85 | mask == Thirteen = addr { octet2 = (apply_mask oct2 Five bit),
86 octet3 = (apply_mask oct3 Zero bit),
87 octet4 = (apply_mask oct4 Zero bit)}
88 | mask == Twelve = addr { octet2 = (apply_mask oct2 Four bit),
89 octet3 = (apply_mask oct3 Zero bit),
90 octet4 = (apply_mask oct4 Zero bit)}
91 | mask == Eleven = addr { octet2 = (apply_mask oct2 Three bit),
92 octet3 = (apply_mask oct3 Zero bit),
93 octet4 = (apply_mask oct4 Zero bit)}
94 | mask == Ten = addr { octet2 = (apply_mask oct2 Two bit),
95 octet3 = (apply_mask oct3 Zero bit),
96 octet4 = (apply_mask oct4 Zero bit)}
97 | mask == Nine = addr { octet2 = (apply_mask oct2 One bit),
98 octet3 = (apply_mask oct3 Zero bit),
99 octet4 = (apply_mask oct4 Zero bit)}
100 | mask == Eight = addr { octet2 = (apply_mask oct2 Zero bit),
101 octet3 = (apply_mask oct3 Zero bit),
102 octet4 = (apply_mask oct4 Zero bit)}
103 | mask == Seven = addr { octet1 = (apply_mask oct1 Seven bit),
104 octet2 = (apply_mask oct2 Zero bit),
105 octet3 = (apply_mask oct3 Zero bit),
106 octet4 = (apply_mask oct4 Zero bit)}
107 | mask == Six = addr { octet1 = (apply_mask oct1 Six bit),
108 octet2 = (apply_mask oct2 Zero bit),
109 octet3 = (apply_mask oct3 Zero bit),
110 octet4 = (apply_mask oct4 Zero bit)}
111 | mask == Five = addr { octet1 = (apply_mask oct1 Five bit),
112 octet2 = (apply_mask oct2 Zero bit),
113 octet3 = (apply_mask oct3 Zero bit),
114 octet4 = (apply_mask oct4 Zero bit)}
115 | mask == Four = addr { octet1 = (apply_mask oct1 Four bit),
116 octet2 = (apply_mask oct2 Zero bit),
117 octet3 = (apply_mask oct3 Zero bit),
118 octet4 = (apply_mask oct4 Zero bit)}
119 | mask == Three = addr { octet1 = (apply_mask oct1 Three bit),
120 octet2 = (apply_mask oct2 Zero bit),
121 octet3 = (apply_mask oct3 Zero bit),
122 octet4 = (apply_mask oct4 Zero bit)}
123 | mask == Two = addr { octet1 = (apply_mask oct1 Two bit),
124 octet2 = (apply_mask oct2 Zero bit),
125 octet3 = (apply_mask oct3 Zero bit),
126 octet4 = (apply_mask oct4 Zero bit)}
127 | mask == One = addr { octet1 = (apply_mask oct1 One bit),
128 octet2 = (apply_mask oct2 Zero bit),
129 octet3 = (apply_mask oct3 Zero bit),
130 octet4 = (apply_mask oct4 Zero bit)}
131 | mask == Zero = addr { octet1 = (apply_mask oct1 Zero bit),
132 octet2 = (apply_mask oct2 Zero bit),
133 octet3 = (apply_mask oct3 Zero bit),
134 octet4 = (apply_mask oct4 Zero bit)}
135 | otherwise = IPv4Address.None
143 -- We don't export our constructor so this function is the only
144 -- way to construct an address from octets. As a result, we can
145 -- return IPv4Address.None in response to being passed one of more
146 -- Octet.None octets.
147 ipv4address_from_octets :: Octet -> Octet -> Octet -> Octet -> IPv4Address
148 ipv4address_from_octets oct1 oct2 oct3 oct4
149 | or [oct1 == Octet.None,
152 oct4 == Octet.None] = IPv4Address.None
153 | otherwise = IPv4Address oct1 oct2 oct3 oct4
156 -- The minimum possible IPv4 address, 0.0.0.0.
157 min_address :: IPv4Address
158 min_address = IPv4Address min_octet min_octet min_octet min_octet
161 -- The maximum possible IPv4 address, 255.255.255.255.
162 max_address :: IPv4Address
163 max_address = IPv4Address max_octet max_octet max_octet max_octet
166 -- Given two addresses, find the number of the most significant bit
167 -- where they differ. If the addresses are the same, return
169 most_sig_bit_different :: IPv4Address -> IPv4Address -> Maskbits
170 most_sig_bit_different addr1 addr2
171 | addr1 == addr2 = Maskbits.Zero
172 | m1 /= n1 = Maskbits.One
182 | m11 /= n11 = Eleven
183 | m12 /= n12 = Twelve
184 | m13 /= n13 = Thirteen
185 | m14 /= n14 = Fourteen
186 | m15 /= n15 = Fifteen
187 | m16 /= n16 = Sixteen
188 | m17 /= n17 = Seventeen
189 | m18 /= n18 = Eighteen
190 | m19 /= n19 = Nineteen
191 | m20 /= n20 = Twenty
192 | m21 /= n21 = TwentyOne
193 | m22 /= n22 = TwentyTwo
194 | m23 /= n23 = TwentyThree
195 | m24 /= n24 = TwentyFour
196 | m25 /= n25 = TwentyFive
197 | m26 /= n26 = TwentySix
198 | m27 /= n27 = TwentySeven
199 | m28 /= n28 = TwentyEight
200 | m29 /= n29 = TwentyNine
201 | m30 /= n30 = Thirty
202 | m31 /= n31 = ThirtyOne
203 | m32 /= n32 = ThirtyTwo
204 | otherwise = Maskbits.Zero
238 oct1a = (octet1 addr1)
239 oct2a = (octet2 addr1)
240 oct3a = (octet3 addr1)
241 oct4a = (octet4 addr1)
274 oct1b = (octet1 addr2)
275 oct2b = (octet2 addr2)
276 oct3b = (octet3 addr2)
277 oct4b = (octet4 addr2)
282 mk_testaddr :: Int -> Int -> Int -> Int -> IPv4Address
283 mk_testaddr a b c d =
284 IPv4Address oct1 oct2 oct3 oct4
286 oct1 = octet_from_int a
287 oct2 = octet_from_int b
288 oct3 = octet_from_int c
289 oct4 = octet_from_int d
292 test_most_sig_bit_different1 :: Test
293 test_most_sig_bit_different1 =
294 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))
297 test_most_sig_bit_different2 :: Test
298 test_most_sig_bit_different2 =
299 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))
302 ipv4address_tests :: [Test]
303 ipv4address_tests = [ test_most_sig_bit_different1,
304 test_most_sig_bit_different2 ]