-import Data.Char (digitToInt, intToDigit)
-import Data.List (concatMap, intercalate, intersperse, splitAt)
-import Numeric (readInt, showIntAtBase)
-import System.Exit (exitFailure)
-import Text.Regex.Posix
-
-import ListUtils
-
--- Takes an IP address in CIDR notation, and returns a list of its
--- octets (converted to Int).
-octets :: String -> [Int]
-octets cidr = map read (take 4 (splitWith (`elem` "./") cidr))
-
-
--- Returns the mask portion of a CIDR address. That is, everything
--- after the trailing slash.
-maskbits :: String -> Int
-maskbits cidr = read ((splitWith (`elem` "/") cidr) !! 1)
-
-
--- Takes an Int, and returns its base-two representation as a String.
-base_two :: Int -> String
-base_two n = showIntAtBase 2 intToDigit n ""
-
-
--- Takes a set of octets, and converts them to base-two
--- individually. The results are then zero-padded on the left to 8
--- characters, and concatenated together.
-octets_base_two :: [Int] -> String
-octets_base_two octet_list =
- concatMap ((pad_left_to 8 '0') .base_two) octet_list
-
-
--- Returns the minimum address (as a base-two string) satisfying the
--- given CIDR string.
-min_base_two_address :: String -> String
-min_base_two_address cidr =
- pad_right_to 32 '0' netpart
- where
- netpart = take (maskbits cidr) (octets_base_two (octets cidr))
-
-
--- Returns the maximum address (as a base-two string) satisfying the
--- given CIDR string.
-max_base_two_address :: String -> String
-max_base_two_address cidr =
- pad_right_to 32 '1' netpart
- where
- netpart = take (maskbits cidr) (octets_base_two (octets cidr))
-
-
--- The octet components of min_base_two_address, as a base-two String.
-min_base_two_octets :: String -> [String]
-min_base_two_octets cidr =
- [octet1, octet2, octet3, octet4]
- where
- addr = min_base_two_address cidr
- octet1 = fst (splitAt 8 addr)
- octet2 = fst (splitAt 8 (snd (splitAt 8 addr)))
- octet3 = fst (splitAt 8 (snd (splitAt 16 addr)))
- octet4 = snd (splitAt 24 addr)
-
-
--- The octet components of max_base_two_address, as a base-two String.
-max_base_two_octets :: String -> [String]
-max_base_two_octets cidr =
- [octet1, octet2, octet3, octet4]
- where
- addr = max_base_two_address cidr
- octet1 = fst (splitAt 8 addr)
- octet2 = fst (splitAt 8 (snd (splitAt 8 addr)))
- octet3 = fst (splitAt 8 (snd (splitAt 16 addr)))
- octet4 = snd (splitAt 24 addr)
-
-
--- The octet components of min_base_two_address, as Ints.
-min_octets :: String -> [Int]
-min_octets cidr =
- map base_two_to_base_ten (min_base_two_octets cidr)
-
-
--- The octet components of max_base_two_address, as Ints.
-max_octets :: String -> [Int]
-max_octets cidr =
- map base_two_to_base_ten (max_base_two_octets cidr)
-
-
--- The base_two_to_base_ten function requires a way to determine
--- whether or not the character it's currently parsing is valid. This
--- should do it.
-is_binary_digit :: Char -> Bool
-is_binary_digit c =
- if c `elem` ['0','1'] then
- True
- else
- False
-
-
--- Convert a base-two String to an Int.
-base_two_to_base_ten :: String -> Int
-base_two_to_base_ten s =
- if (length parsed) == 0 then
- 0
- else
- fst (parsed !! 0)
- where
- parsed = readInt 2 is_binary_digit digitToInt s
-
-
--- A regular expression that matches a non-address character.
+module Main
+where
+
+import Control.Concurrent.ParallelIO.Global ( stopGlobalPool )
+import Control.Monad (when)
+import qualified Data.ByteString.Char8 as BS (intercalate, pack, unpack)
+import Data.List ((\\), intercalate)
+import Data.Maybe (catMaybes, isNothing)
+import Data.String.Utils (splitWs)
+import System.Exit (ExitCode(..), exitWith)
+import System.IO (stderr, hPutStrLn)
+import Text.Read (readMaybe)
+
+import Cidr (
+ Cidr(..),
+ combine_all,
+ enumerate,
+ max_octet1,
+ max_octet2,
+ max_octet3,
+ max_octet4,
+ min_octet1,
+ min_octet2,
+ min_octet3,
+ min_octet4 )
+import CommandLine (Args(..), get_args)
+import DNS (Domain, PTRResult, lookup_ptrs)
+import ExitCodes ( exit_invalid_cidr )
+import Octet ()
+
+
+-- | A regular expression that matches a non-address character.