Implement directBackendCmd in remotebackend

Habbie · Jun 30, 2015 · 0bcb7cc · 0bcb7cc
1 parent 78903c5
commit 0bcb7cc
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 0 deletions.
diff --git a/modules/remotebackend/httpconnector.cc b/modules/remotebackend/httpconnector.cc
@@ -240,6 +240,11 @@ void HTTPConnector::restful_requestbuilder(const std::string &method, const rapi
         req.POST()["serial"] = sparam;
         req.preparePost();
         verb = "PATCH";
+    } else if (method == "directBackendCmd") {
+        json2string(parameters["query"],sparam);
+        req.POST()["query"] = sparam;
+        req.preparePost();
+        verb = "POST";
     } else {
         // perform normal get
         verb = "GET";

diff --git a/modules/remotebackend/remotebackend.cc b/modules/remotebackend/remotebackend.cc
@@ -938,6 +938,22 @@ bool RemoteBackend::calculateSOASerial(const string& domain, const SOAData& sd,
    return true;
 }
 
+string RemoteBackend::directBackendCmd(const string& querystr) {
+   rapidjson::Document query,answer;
+   rapidjson::Value parameters;
+
+   query.SetObject();
+   JSON_ADD_MEMBER(query, "method", "directBackendCmd", query.GetAllocator());
+   parameters.SetObject();
+   JSON_ADD_MEMBER(parameters, "query", querystr.c_str(), query.GetAllocator());
+   query.AddMember("parameters", parameters, query.GetAllocator());
+
+   if (this->send(query) == false || this->recv(answer) == false)
+     return "backend command failed";
+
+   return getString(answer["result"]);
+}
+
 // some rapidjson helpers 
 bool RemoteBackend::getBool(rapidjson::Value &value) {
    if (value.IsNull()) return false;

diff --git a/modules/remotebackend/remotebackend.hh b/modules/remotebackend/remotebackend.hh
@@ -162,6 +162,7 @@ class RemoteBackend : public DNSBackend
   virtual bool setTSIGKey(const string& name, const string& algorithm, const string& content);
   virtual bool deleteTSIGKey(const string& name);
   virtual bool getTSIGKeys(std::vector< struct TSIGKey > &keys);
+  virtual string directBackendCmd(const string& querystr);
 
   static DNSBackend *maker();