X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FMain.hs;h=a3b1bfe4a811faad00f6e8503b58de540bd8703f;hb=6c671cd410b81257514e1be76ca3f252ea69ccdb;hp=c87520951088c46f095f334bd90e78587b98c53d;hpb=364735d8f7cbc78528517faeccac606813ef9998;p=hath.git diff --git a/src/Main.hs b/src/Main.hs index c875209..a3b1bfe 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -1,17 +1,27 @@ -import Data.List (intercalate, intersperse) -import System.Exit (exitFailure) -import Text.Regex.Posix - -import Cidr (Cidr, - from_string, - min_first_octet, - min_second_octet, - min_third_octet, - min_fourth_octet, - max_first_octet, - max_second_octet, - max_third_octet, - max_fourth_octet) +import Data.List ((\\), intercalate, intersperse) +import System.Exit (ExitCode(..), exitWith) +import System.IO (stderr, hPutStrLn) + +import Cidr (Cidr(..), + cidr_from_string, + combine_all) + +import CommandLine (help_set, + help_text, + input_function, + Mode(..), + parse_errors, + parse_mode) + +import IPv4Address +import Octet + +-- Some exit codes, used in the ExitFailure constructor. +exit_invalid_cidr :: Int +exit_invalid_cidr = 1 + +exit_args_parse_failed :: Int +exit_args_parse_failed = 2 -- A regular expression that matches a non-address character. @@ -44,21 +54,16 @@ cidr_to_regex cidr = range2 = numeric_range min2 max2 range3 = numeric_range min3 max3 range4 = numeric_range min4 max4 - min1 = min_first_octet cidr - min2 = min_second_octet cidr - min3 = min_third_octet cidr - min4 = min_fourth_octet cidr - max1 = max_first_octet cidr - max2 = max_second_octet cidr - max3 = max_third_octet cidr - max4 = max_fourth_octet cidr + min1 = octet_to_int (min_octet1 (ipv4address cidr) (maskbits cidr)) + min2 = octet_to_int (min_octet2 (ipv4address cidr) (maskbits cidr)) + min3 = octet_to_int (min_octet3 (ipv4address cidr) (maskbits cidr)) + min4 = octet_to_int (min_octet4 (ipv4address cidr) (maskbits cidr)) + max1 = octet_to_int (max_octet1 (ipv4address cidr) (maskbits cidr)) + max2 = octet_to_int (max_octet2 (ipv4address cidr) (maskbits cidr)) + max3 = octet_to_int (max_octet3 (ipv4address cidr) (maskbits cidr)) + max4 = octet_to_int (max_octet4 (ipv4address cidr) (maskbits cidr)) --- Will return True if the passed String is in CIDR notation, False --- otherwise. -is_valid_cidr :: String -> Bool -is_valid_cidr cidr = cidr =~ "([0-9]{1,3}\\.){3}[0-9]{1,3}/[0-9]{1,2}" - -- Take a list of Strings, and return a regular expression matching -- any of them. @@ -76,23 +81,52 @@ numeric_range x y = upper = maximum [x,y] --- Take a CIDR String, and exitFailure if it's invalid. -validate_or_die :: String -> IO () -validate_or_die cidr = do - if (is_valid_cidr cidr) +main :: IO () +main = do + -- First, check for any errors that occurred while parsing + -- the command line options. + errors <- CommandLine.parse_errors + if not (null errors) + then do + hPutStrLn stderr (concat errors) + putStrLn CommandLine.help_text + exitWith (ExitFailure exit_args_parse_failed) + else do -- Nothing + + -- Next, check to see if the 'help' option was passed to the + -- program. If it was, display the help, and exit successfully. + help_opt_set <- CommandLine.help_set + if help_opt_set then do - return () - else do - putStrLn "Error: not valid CIDR notation." - exitFailure + putStrLn CommandLine.help_text + exitWith ExitSuccess + else do -- Nothing + -- The input function we receive here should know what to read. + inputfunc <- (CommandLine.input_function) + input <- inputfunc -main :: IO () -main = do - input <- getContents let cidr_strings = lines input - mapM validate_or_die cidr_strings - let cidrs = map Cidr.from_string cidr_strings - let regexes = map cidr_to_regex cidrs - putStrLn $ alternate regexes + let cidrs = map Cidr.cidr_from_string cidr_strings + if (any (== Cidr.None) cidrs) + then do + putStrLn "Error: not valid CIDR notation." + exitWith (ExitFailure exit_invalid_cidr) + else do -- Nothing + + -- Get the mode of operation. + mode <- CommandLine.parse_mode + + case mode of + Regex -> do + let regexes = map cidr_to_regex cidrs + putStrLn $ alternate regexes + Reduce -> do + mapM (putStrLn . show) (combine_all cidrs) + return () + Dupe -> do + mapM (putStrLn . show) dupes + return () + where + dupes = cidrs \\ (combine_all cidrs)