1 {-# LANGUAGE OverloadedStrings #-}
7 import qualified Data.ByteString.Char8 as C (pack, unpack)
8 import qualified Data.ByteString.Lazy.Char8 as L (pack, unpack)
9 import Data.ByteString.Lazy as BS hiding (reverse, putStr, putStrLn)
10 import Data.List.Split
11 import Data.Aeson (encode, decode, Value(..))
12 import Network.HTTP.Types
14 import Network.HTTP.Conduit
22 confFile = "/etc/namecoin.conf"
24 -- HTTP/JsonRpc interface
26 qReq :: Config -> ByteString -> ByteString -> Request m
27 qReq cf q id = applyBasicAuth (C.pack (rpcuser cf)) (C.pack (rpcpassword cf))
28 $ def { host = (C.pack (rpchost cf))
31 , requestHeaders = [ (hAccept, "application/json")
32 , (hContentType, "application/json")
33 , (hConnection, "Keep-Alive")
35 , requestBody = RequestBodyLBS $ encode $
36 JsonRpcRequest JsonRpcV1
40 , checkStatus = \_ _ _ -> Nothing
43 qRsp :: Response ByteString -> Either String NmcDom
45 case parseJsonRpc (responseBody rsp) :: Either JsonRpcError NmcRes of
47 case (jrpcErrCode jerr) of
48 -4 -> Right emptyNmcDom
49 _ -> Left $ "JsonRpc error response: " ++ (show jerr)
52 "" -> Right emptyNmcDom
54 case decode vstr :: Maybe NmcDom of
55 Nothing -> Left $ "Unparseable value: " ++ (show vstr)
60 queryNmc :: Manager -> Config -> String -> String
61 -> IO (Either String NmcDom)
62 queryNmc mgr cfg fqdn qid = do
63 case reverse (splitOn "." fqdn) of
66 httpLbs (qReq cfg (L.pack ("d/" ++ dn)) (L.pack qid)) mgr
67 return $ case qRsp rsp of
69 Right dom -> Right $ descendNmc xs dom
71 return $ Left "Only \".bit\" domain is supported"
77 cfg <- readConfig confFile
79 hSetBuffering stdin LineBuffering
80 hSetBuffering stdout LineBuffering
83 loopErr e = forever $ do
84 putStrLn $ "FAIL\t" ++ e
89 ["HELO", "1"] -> return 1
90 ["HELO", "2"] -> return 2
91 ["HELO", "3"] -> return 3
92 ["HELO", x ] -> loopErr $ "unsupported ABI version " ++ (show x)
93 _ -> loopErr $ "bad HELO " ++ (show s)
95 putStrLn $ "OK\tDnsNmc ready to serve, protocol v." ++ (show ver)
100 case pdnsParse ver l of
101 Left e -> putStr $ pdnsReport e
104 PdnsRequestQ qname qtype id _ _ _ ->
105 queryNmc mgr cfg qname id >>= putStr . (pdnsOut ver id qname qtype)
106 PdnsRequestAXFR xfrreq ->
107 putStr $ pdnsReport ("No support for AXFR " ++ xfrreq)
108 PdnsRequestPing -> putStrLn "END"
113 cfg <- readConfig confFile
114 mgr <- newManager def
115 queryNmc mgr cfg str "askid" >>= putStr . (pdnsOut 1 "askid" str RRTypeANY)