, pdnsParse
, pdnsReport
, pdnsOut
+ , pdnsOutXfr
) where
+import Data.Text.Lazy (splitOn, pack)
+
import NmcDom
data RRType = RRTypeSRV | RRTypeA | RRTypeAAAA | RRTypeCNAME
| RRTypeDNAME | RRTypeSOA | RRTypeRP | RRTypeLOC
- | RRTypeNS | RRTypeDS
+ | RRTypeNS | RRTypeDS | RRTypeMX
| RRTypeANY | RRTypeError String
- deriving (Show)
+
+instance Show RRType where
+ show RRTypeSRV = "SRV"
+ show RRTypeA = "A"
+ show RRTypeAAAA = "AAAA"
+ show RRTypeCNAME = "CNAME"
+ show RRTypeDNAME = "DNAME"
+ show RRTypeSOA = "SOA"
+ show RRTypeRP = "RP"
+ show RRTypeLOC = "LOC"
+ show RRTypeNS = "NS"
+ show RRTypeDS = "DS"
+ show RRTypeMX = "MX"
+ show RRTypeANY = "ANY"
+ show (RRTypeError s) = "RR type error: " ++ (show s)
data PdnsRequest = PdnsRequestQ
{ qName :: String
, qType :: RRType
- , iD :: String
+ , iD :: Int
, remoteIpAddress :: String
, localIpAddress :: Maybe String
, ednsSubnetAddress :: Maybe String
}
- | PdnsRequestAXFR String
+ | PdnsRequestAXFR Int
| PdnsRequestPing
deriving (Show)
+-- | Parse request string read from the core PowerDNS process
+pdnsParse :: Int -> String -> Either String PdnsRequest
pdnsParse ver s =
let
+ getInt s = case reads s :: [(Int, String)] of
+ [(x, _)] -> x
+ _ -> -1
getQt qt = case qt of
- "SRV" -> RRTypeSRV
+ "SRV" -> RRTypeSRV
"A" -> RRTypeA
"AAAA" -> RRTypeAAAA
"CNAME" -> RRTypeCNAME
"DNAME" -> RRTypeDNAME
- "SOA" -> RRTypeSOA
- "RP" -> RRTypeRP
+ "SOA" -> RRTypeSOA
+ "RP" -> RRTypeRP
"LOC" -> RRTypeLOC
- "NS" -> RRTypeNS
+ "NS" -> RRTypeNS
"DS" -> RRTypeDS
+ "MX" -> RRTypeMX
"ANY" -> RRTypeANY
_ -> RRTypeError qt
getLIp ver xs
in
case words s of
"PING":[] -> Right PdnsRequestPing
- "AXFR":x:[] -> Right (PdnsRequestAXFR x)
+ "AXFR":x:[] -> Right (PdnsRequestAXFR (getInt x))
"Q":qn:"IN":qt:id:rip:xs -> Right (PdnsRequestQ
{ qName = qn
, qType = getQt qt
- , iD = id
+ , iD = getInt id
, remoteIpAddress = rip
, localIpAddress = getLIp ver xs
, ednsSubnetAddress = getRIp ver xs
})
_ -> Left $ "Unparseable PDNS Request: " ++ s
+-- | Produce LOG entry followed by FAIL
pdnsReport :: String -> String
-pdnsReport err =
- "LOG\tError: " ++ err ++ "\nFAIL\n"
+pdnsReport err = "LOG\tError: " ++ err ++ "\nFAIL\n"
+
+-- | Produce answer to the Q request
+pdnsOut :: Int -> Int -> String -> RRType -> Either String NmcDom -> String
+pdnsOut ver id name rrtype edom = case edom of
+ Left err ->
+ pdnsReport $ err ++ " in a " ++ (show rrtype) ++ "query for " ++ name
+ Right dom ->
+ case rrtype of
+ RRTypeANY -> foldr (\x a -> (formatRR ver id name dom x) ++ a) "END\n"
+ [RRTypeSRV, RRTypeA, RRTypeAAAA, RRTypeCNAME, RRTypeDNAME,
+ RRTypeRP, RRTypeLOC, RRTypeNS, RRTypeDS, RRTypeMX]
+ _ -> (formatRR ver id name dom rrtype) ++ "END\n"
+
+-- | Produce answer to the AXFR request
+pdnsOutXfr :: Int -> Int -> String -> Either String NmcDom -> String
+pdnsOutXfr ver id name edom = "" -- FIXME
+
+justl accessor _ dom = case accessor dom of
+ Nothing -> []
+ Just xs -> xs
-pdnsOut :: Int -> String -> String -> RRType -> Either String NmcDom -> String
-pdnsOut ver id name rrtype edom =
- case edom of
- Left err -> pdnsReport $ err ++ " in a query for " ++ name
- Right dom -> foldr addLine "END\n" $ nmc2pdns name rrtype dom
- where
- addLine (nm, ty, dt) accum =
- "DATA\t" ++ v3ext ++ nm ++ "\tIN\t" ++ ty ++ "\t" ++ ttl ++
- "\t" ++ id ++ "\t" ++ dt ++ "\n" ++ accum
- v3ext = case ver of
- 3 -> "0\t1\t"
- _ -> ""
- ttl = show 3600
+justv accessor _ dom = case accessor dom of
+ Nothing -> []
+ Just x -> [x]
-nmc2pdns :: String -> RRType -> NmcDom -> [(String, String, String)]
-nmc2pdns name RRTypeANY dom =
- foldr (\r accum -> (nmc2pdns name r dom) ++ accum) []
- [RRTypeA, RRTypeAAAA, RRTypeCNAME, RRTypeDNAME, -- no SRV here!
- RRTypeSOA, RRTypeRP, RRTypeLOC, RRTypeNS, RRTypeDS]
-nmc2pdns name RRTypeSRV dom = [] -- FIXME
-nmc2pdns name RRTypeA dom = mapto name "A" $ domIp dom
-nmc2pdns name RRTypeAAAA dom = mapto name "AAAA" $ domIp6 dom
-nmc2pdns name RRTypeCNAME dom = takejust name "CNAME" $ domAlias dom
-nmc2pdns name RRTypeDNAME dom = takejust name "DNAME" $ domTranslate dom
-nmc2pdns name RRTypeSOA dom =
- if dom == emptyNmcDom then []
- else
- let
- email = case domEmail dom of
- Nothing -> "hostmaster." ++ name
- Just addr ->
- let (aname, adom) = break (== '@') addr
- in case adom of
- "" -> aname
- _ -> aname ++ "." ++ (tail adom)
- in [(name, "SOA", email ++ " 99999999 10800 3600 604800 86400")]
-nmc2pdns name RRTypeRP dom = [] --FIXME
-nmc2pdns name RRTypeLOC dom = takejust name "LOC" $ domLoc dom
-nmc2pdns name RRTypeNS dom = mapto name "NS" $ domNs dom
-nmc2pdns name RRTypeDS dom = [] --FIXME
+dotmail addr =
+ let (aname, adom) = break (== '@') addr
+ in case adom of
+ "" -> aname
+ _ -> aname ++ "." ++ (tail adom)
-mapto name rrstr maybel = case maybel of
- Nothing -> []
- Just l -> map (\x -> (name, rrstr, x)) l
+dataRR RRTypeSRV = justl domSrv
+dataRR RRTypeMX = justl domMx
+dataRR RRTypeA = justl domIp
+dataRR RRTypeAAAA = justl domIp6
+dataRR RRTypeCNAME = justv domAlias
+dataRR RRTypeDNAME = justv domTranslate
+dataRR RRTypeSOA = soa
+ where
+ soa name dom = -- FIXME generate only for top domain
+ -- FIXME make realistic version field
+ -- FIXME make realistic nameserver field
+ -- Follows a relatively ugly hack to figure if we are at the top
+ -- level domain ("something.bit"). Only in such case we provide
+ -- the synthetic SOA RR. Otherwise yield empty.
+ case splitOn (pack ".") (pack name) of
+ [_,_] ->
+ if dom == emptyNmcDom then []
+ else ["ns " ++ email ++ " 99999 10800 3600 604800 86400"]
+ where
+ email = case domEmail dom of
+ Nothing -> "hostmaster." ++ name
+ Just addr -> dotmail addr
+ _ -> []
+dataRR RRTypeRP = rp
+ where
+ rp _ dom = case domEmail dom of
+ Nothing -> []
+ Just addr -> [(dotmail addr) ++ " ."]
+dataRR RRTypeLOC = justv domLoc
+dataRR RRTypeNS = justl domNs
+dataRR RRTypeDS = ds
+ where
+ ds _ dom = case domDs dom of
+ Nothing -> []
+ Just dss -> map dsStr dss
+ where
+ dsStr x = (show (dsKeyTag x)) ++ " "
+ ++ (show (dsAlgo x)) ++ " "
+ ++ (show (dsHashType x)) ++ " "
+ ++ (dsHashValue x)
-takejust name rrstr maybestr = case maybestr of
- Nothing -> []
- Just str -> [(name, rrstr, str)]
+formatRR ver id name dom rrtype =
+ foldr (\x a -> "DATA\t" ++ v3ext ++ name ++ "\tIN\t" ++ (show rrtype)
+ ++ "\t" ++ ttl ++ "\t" ++ (show id) ++ "\t" ++ x ++ "\n" ++ a)
+ "" $ dataRR rrtype name dom
+ where
+ v3ext = case ver of
+ 3 -> "0\t1\t"
+ _ -> ""
+ ttl = show 3600