1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
{-# LANGUAGE CPP #-}
{-# LANGUAGE FlexibleInstances #-}
{-# OPTIONS_GHC -fno-warn-orphans #-}
module IPSpec where
#if __GLASGOW_HASKELL__ < 709
import Control.Applicative
#endif
import Data.IP
import Data.Maybe (isJust)
import Safe (readMay)
import Test.Hspec
import Test.Hspec.QuickCheck (prop)
import Test.QuickCheck
import RouteTableSpec ()
----------------------------------------------------------------
--
-- Arbitrary
--
data InvalidIPv4Str = Iv4 String deriving (Show)
instance Arbitrary InvalidIPv4Str where
arbitrary =
frequency
[ (8, arbitraryIIPv4Str arbitrary 32)
, -- an IPv4 address should not end with a trailing `.`
(1, Iv4 . (++ ".") . show <$> genIPv4)
, -- an IPv4 address with mask should not include a white space
( 1
, (\ip (NonNegative len) -> Iv4 (show ip ++ " /" ++ show (len :: Integer)))
<$> genIPv4
<*> arbitrary
)
]
where
genIPv4 :: Gen IPv4
genIPv4 = arbitrary
arbitraryIIPv4Str :: Gen IPv4 -> Int -> Gen InvalidIPv4Str
arbitraryIIPv4Str adrGen msklen = toIv4 <$> adrGen <*> lenGen
where
toIv4 adr len = Iv4 $ show adr ++ "/" ++ show len
lenGen = oneof [choose (minBound, -1), choose (msklen + 1, maxBound)]
data InvalidIPv6Str = Iv6 String deriving (Show)
instance Arbitrary InvalidIPv6Str where
arbitrary = arbitraryIIPv6Str arbitrary 128
arbitraryIIPv6Str :: Gen IPv6 -> Int -> Gen InvalidIPv6Str
arbitraryIIPv6Str adrGen msklen = toIv6 <$> adrGen <*> lenGen
where
toIv6 adr len = Iv6 $ show adr ++ "/" ++ show len
lenGen = oneof [choose (minBound, -1), choose (msklen + 1, maxBound)]
----------------------------------------------------------------
--
-- Spec
--
spec :: Spec
spec = do
describe "read" $ do
prop "IPv4" to_str_ipv4
prop "IPv6" to_str_ipv6
prop "IPv4 failure" ipv4_fail
prop "IPv6 failure" ipv6_fail
it "can read even if unnecessary spaces exist" $ do
(readMay " 127.0.0.1" :: Maybe IPv4) `shouldBe` readMay "127.0.0.1"
it "does not read overflow IPv4 octets" $ do
(readMay "127.0.0.18446744073709551617" :: Maybe IPv4) `shouldBe` Nothing
it "can read even if unnecessary spaces exist" $ do
(readMay " ::1" :: Maybe IPv4) `shouldBe` readMay "::1"
it "does not read overflow mask lengths" $ do
(readMay "192.168.0.1/18446744073709551648" :: Maybe (AddrRange IPv4))
`shouldBe` Nothing
it "can read embedded v4 in v6 range" $ do
(readMay "::ffff:192.0.2.0/120" :: Maybe (AddrRange IPv6))
`shouldSatisfy` isJust
to_str_ipv4 :: AddrRange IPv4 -> Bool
to_str_ipv4 a = readMay (show a) == Just a
to_str_ipv6 :: AddrRange IPv6 -> Bool
to_str_ipv6 a = readMay (show a) == Just a
ipv4_fail :: InvalidIPv4Str -> Bool
ipv4_fail (Iv4 a) = (readMay a :: Maybe (AddrRange IPv4)) == Nothing
ipv6_fail :: InvalidIPv6Str -> Bool
ipv6_fail (Iv6 a) = (readMay a :: Maybe (AddrRange IPv6)) == Nothing
|
