module PowerDns ( RRType(..)
+ , rrType
, PdnsRequest(..)
, pdnsParse
, pdnsReport
- , pdnsOut
+ , pdnsOutQ
, pdnsOutXfr
) where
import Data.Text.Lazy (splitOn, pack)
+import Data.Map.Lazy (foldrWithKey)
import NmcDom
data RRType = RRTypeSRV | RRTypeA | RRTypeAAAA | RRTypeCNAME
| RRTypeDNAME | RRTypeSOA | RRTypeRP | RRTypeLOC
| RRTypeNS | RRTypeDS | RRTypeMX
- | RRTypeANY | RRTypeError String
+ | RRTypeANY | RRTypeError String
instance Show RRType where
show RRTypeSRV = "SRV"
show RRTypeA = "A"
show RRTypeAAAA = "AAAA"
show RRTypeCNAME = "CNAME"
- show RRTypeDNAME = "DNAME"
+ show RRTypeDNAME = "DNAME"
show RRTypeSOA = "SOA"
show RRTypeRP = "RP"
show RRTypeLOC = "LOC"
- show RRTypeNS = "NS"
+ show RRTypeNS = "NS"
show RRTypeDS = "DS"
show RRTypeMX = "MX"
show RRTypeANY = "ANY"
- show (RRTypeError s) = "RR type error: " ++ (show s)
+ show (RRTypeError s) = "Unknown RR type: " ++ (show s)
+
+rrType qt = case qt of
+ "SRV" -> RRTypeSRV
+ "A" -> RRTypeA
+ "AAAA" -> RRTypeAAAA
+ "CNAME" -> RRTypeCNAME
+ "DNAME" -> RRTypeDNAME
+ "SOA" -> RRTypeSOA
+ "RP" -> RRTypeRP
+ "LOC" -> RRTypeLOC
+ "NS" -> RRTypeNS
+ "DS" -> RRTypeDS
+ "MX" -> RRTypeMX
+ "ANY" -> RRTypeANY
+ _ -> RRTypeError qt
data PdnsRequest = PdnsRequestQ
{ qName :: String
let
getInt s = case reads s :: [(Int, String)] of
[(x, _)] -> x
- _ -> -1
- getQt qt = case qt of
- "SRV" -> RRTypeSRV
- "A" -> RRTypeA
- "AAAA" -> RRTypeAAAA
- "CNAME" -> RRTypeCNAME
- "DNAME" -> RRTypeDNAME
- "SOA" -> RRTypeSOA
- "RP" -> RRTypeRP
- "LOC" -> RRTypeLOC
- "NS" -> RRTypeNS
- "DS" -> RRTypeDS
- "MX" -> RRTypeMX
- "ANY" -> RRTypeANY
- _ -> RRTypeError qt
+ _ -> (-1)
getLIp ver xs
| ver >= 2 = case xs of
x:_ -> Just x
case words s of
"PING":[] -> Right PdnsRequestPing
"AXFR":x:[] -> Right (PdnsRequestAXFR (getInt x))
- "Q":qn:"IN":qt:id:rip:xs -> Right (PdnsRequestQ
+ "Q":qn:"IN":qt:id:rip:xs -> case rrType qt of
+ RRTypeError e ->
+ Left $ "Unrecognized RR type: " ++ e
+ rt ->
+ Right (PdnsRequestQ
{ qName = qn
- , qType = getQt qt
+ , qType = rrType qt
, iD = getInt id
, remoteIpAddress = rip
, localIpAddress = getLIp ver xs
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"
+pdnsOutQ :: Int -> Int -> String -> RRType -> Either String NmcDom -> String
+pdnsOutQ ver id name rrt edom =
+ let
+ rrl = case rrt of
+ RRTypeANY -> [ RRTypeSRV, RRTypeA, RRTypeAAAA, RRTypeCNAME
+ , RRTypeDNAME, RRTypeRP, RRTypeLOC, RRTypeNS
+ , RRTypeDS, RRTypeMX -- SOA not included
+ ]
+ x -> [x]
+ in
+ case edom of
+ Left err ->
+ pdnsReport $ err ++ " in the " ++ (show rrt) ++ " query for " ++ name
+ Right dom ->
+ formatDom ver id rrl name dom "END\n"
-- | Produce answer to the AXFR request
pdnsOutXfr :: Int -> Int -> String -> Either String NmcDom -> String
-pdnsOutXfr ver id name edom = "" -- FIXME
+pdnsOutXfr ver id name edom =
+ let
+ allrrs = [ RRTypeSRV, RRTypeA, RRTypeAAAA, RRTypeCNAME
+ , RRTypeDNAME, RRTypeRP, RRTypeLOC, RRTypeNS
+ , RRTypeDS, RRTypeMX, RRTypeSOA
+ ]
+ walkDom f acc name dom =
+ f name dom $ case domMap dom of
+ Nothing -> acc
+ Just dm ->
+ foldrWithKey (\n d a -> walkDom f a (n ++ "." ++ name) d) acc dm
+ in
+ case edom of
+ Left err ->
+ pdnsReport $ err ++ " in the AXFR request for " ++ name
+ Right dom ->
+ walkDom (formatDom ver id allrrs) "END\n" name dom
+
+formatDom ver id rrl name dom acc =
+ foldr (\x a -> (formatRR ver id name dom x) ++ a) acc rrl
+
+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
justl accessor _ dom = case accessor dom of
Nothing -> []
Nothing -> []
Just x -> [x]
-dotmail addr =
+dotmail addr =
let (aname, adom) = break (== '@') addr
in case adom of
"" -> aname ++ "."
-- Alternative would be to carry "top-ness" as a parameter through
-- all the calls from the very top where we split the fqdn.
case splitOn (pack ".") (pack name) of
- [_,_] -> [ns ++ " " ++ email ++ " 99999 10800 3600 604800 86400"]
+ [_,_] -> [ns ++ " " ++ email ++ " 0 10800 3600 604800 86400"]
_ -> []
dataRR RRTypeRP = \ _ dom ->
case domEmail dom of
++ (show (dsAlgo x)) ++ " "
++ (show (dsHashType x)) ++ " "
++ (dsHashValue x)
-
-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
+-- This only comes into play when data arrived _not_ from a PDNS request:
+dataRR (RRTypeError e) = \ _ _ ->
+ ["; No data for bad request type " ++ e]
projects / pdns-pipe-nmc.git / blobdiff