PowerDns.hs

   1 module PowerDns ( RRType(..)
   2                 , PdnsRequest(..)
   3                 , pdnsParse
   4                 , pdnsReport
   5                 , pdnsOut
   6                 ) where
   7
   8 import NmcDom
   9
  10 data RRType = RRTypeSRV   | RRTypeA   | RRTypeAAAA | RRTypeCNAME
  11             | RRTypeDNAME | RRTypeSOA | RRTypeRP   | RRTypeLOC
  12             | RRTypeNS    | RRTypeDS  | RRTypeMX
  13             | RRTypeANY   | RRTypeError String
  14         deriving (Show)
  15
  16 data PdnsRequest = PdnsRequestQ
  17                    { qName              :: String
  18                    , qType              :: RRType
  19                    , iD                 :: String
  20                    , remoteIpAddress    :: String
  21                    , localIpAddress     :: Maybe String
  22                    , ednsSubnetAddress  :: Maybe String
  23                    }
  24                  | PdnsRequestAXFR String
  25                  | PdnsRequestPing
  26         deriving (Show)
  27
  28 pdnsParse ver s =
  29   let
  30     getQt qt = case qt of
  31       "SRV"     -> RRTypeSRV
  32       "A"       -> RRTypeA
  33       "AAAA"    -> RRTypeAAAA
  34       "CNAME"   -> RRTypeCNAME
  35       "DNAME"   -> RRTypeDNAME
  36       "SOA"     -> RRTypeSOA
  37       "RP"      -> RRTypeRP
  38       "LOC"     -> RRTypeLOC
  39       "NS"      -> RRTypeNS
  40       "DS"      -> RRTypeDS
  41       "MX"      -> RRTypeMX
  42       "ANY"     -> RRTypeANY
  43       _         -> RRTypeError qt
  44     getLIp ver xs
  45       | ver >= 2  = case xs of
  46                       x:_       -> Just x
  47                       _         -> Nothing
  48       | otherwise = Nothing
  49     getRIp ver xs
  50       | ver >= 3  = case xs of
  51                       _:x:_     -> Just x
  52                       _         -> Nothing
  53       | otherwise = Nothing
  54   in
  55     case words s of
  56       "PING":[]                 -> Right PdnsRequestPing
  57       "AXFR":x:[]               -> Right (PdnsRequestAXFR x)
  58       "Q":qn:"IN":qt:id:rip:xs  -> Right (PdnsRequestQ
  59                                             { qName = qn
  60                                             , qType = getQt qt
  61                                             , iD = id
  62                                             , remoteIpAddress = rip
  63                                             , localIpAddress = getLIp ver xs
  64                                             , ednsSubnetAddress = getRIp ver xs
  65                                             })
  66       _                         -> Left $ "Unparseable PDNS Request: " ++ s
  67
  68 pdnsReport :: String -> String
  69 pdnsReport err =
  70   "LOG\tError: " ++ err ++ "\nFAIL\n"
  71
  72 pdnsOut :: Int -> String -> String -> RRType -> Either String NmcDom -> String
  73 pdnsOut ver id name rrtype edom = case edom of
  74   Left  err -> pdnsReport $ err ++ " in a query for " ++ name
  75   Right dom -> foldr addLine "END\n" $ n2p rrtype
  76     where
  77       addLine (nm, ty, dt) accum =
  78         "DATA\t" ++ v3ext ++ nm ++ "\tIN\t" ++ ty ++ "\t" ++ ttl ++
  79             "\t" ++ id ++ "\t" ++ dt ++ "\n" ++ accum
  80       v3ext = case ver of
  81         3 -> "0\t1\t"
  82         _ -> ""
  83       ttl = show 3600
  84
  85       n2p RRTypeANY   =
  86         foldr (\r accum -> (n2p r) ++ accum) []
  87           [RRTypeSRV, RRTypeA, RRTypeAAAA, RRTypeCNAME, RRTypeDNAME,
  88            RRTypeSOA, RRTypeRP, RRTypeLOC, RRTypeNS, RRTypeDS, RRTypeMX]
  89       n2p RRTypeSRV   = makesrv  "SRV"   $ domService dom
  90       n2p RRTypeMX    = mapto    "MX"    $ domMx dom
  91       n2p RRTypeA     = mapto    "A"     $ domIp dom
  92       n2p RRTypeAAAA  = mapto    "AAAA"  $ domIp6 dom
  93       n2p RRTypeCNAME = takejust "CNAME" $ domAlias dom
  94       n2p RRTypeDNAME = takejust "DNAME" $ domTranslate dom
  95       n2p RRTypeSOA   = -- FIXME generate only for top domain
  96         if dom == emptyNmcDom then []
  97         else
  98           let
  99             email = case domEmail dom of
 100               Nothing   -> "hostmaster." ++ name
 101               Just addr ->
 102                 let (aname, adom) = break (== '@') addr
 103                 in case adom of
 104                   "" -> aname
 105                   _  -> aname ++ "." ++ (tail adom)
 106           in [(name, "SOA", email ++ " 99999999 10800 3600 604800 86400")]
 107       n2p RRTypeRP    = [] --FIXME
 108       n2p RRTypeLOC   = takejust "LOC"  $ domLoc dom
 109       n2p RRTypeNS    = mapto    "NS"   $ domNs dom
 110       n2p RRTypeDS    = [] --FIXME
 111
 112       mapto    rrstr maybel   = case maybel of
 113         Nothing  -> []
 114         Just l   -> map (\x -> (name, rrstr, x)) l
 115       takejust rrstr maybestr = case maybestr of
 116         Nothing  -> []
 117         Just str -> [(name, rrstr, str)]
 118       makesrv  rrstr mayberl  = case mayberl of
 119         Nothing  -> []
 120         Just srl  -> map (\x -> (name, rrstr, fmtsrv x)) srl
 121           where
 122             fmtsrv rl = (show (srvPrio rl)) ++ " "
 123                       ++ (show (srvWeight rl)) ++ " "
 124                       ++ (show (srvPort rl)) ++ " "
 125                       ++ (srvHost rl)