Add WFDnsServerTask and specify WFDnsServer::new_session().

src/factory/DnsTaskImpl.cc

@@ -18,9 +18,10 @@
 
 #include <string>
 #include <atomic>
+#include "DnsMessage.h"
 #include "WFTaskError.h"
 #include "WFTaskFactory.h"
-#include "DnsMessage.h"
+#include "WFServer.h"
 
 using namespace protocol;
 
@@ -184,3 +185,42 @@ WFDnsTask *WFTaskFactory::create_dns_task(const ParsedURI& uri,
 	return task;
 }
 
+
+/**********Server**********/
+
+class WFDnsServerTask : public WFServerTask<DnsRequest, DnsResponse>
+{
+public:
+	WFDnsServerTask(CommService *service,
+					std::function<void (WFDnsTask *)>& proc) :
+		WFServerTask(service, WFGlobal::get_scheduler(), proc)
+	{
+		auto *server = (WFServer<DnsRequest, DnsResponse> *)service;
+		this->type = server->get_params()->transport_type;
+	}
+
+protected:
+	virtual CommMessageIn *message_in()
+	{
+		this->get_req()->set_single_packet(this->type == TT_UDP);
+		return this->WFServerTask::message_in();
+	}
+
+	virtual CommMessageOut *message_out()
+	{
+		this->get_resp()->set_single_packet(this->type == TT_UDP);
+		return this->WFServerTask::message_out();
+	}
+
+protected:
+	enum TransportType type;
+};
+
+/**********Server Factory**********/
+
+WFDnsTask *WFServerTaskFactory::create_dns_task(CommService *service,
+						std::function<void (WFDnsTask *)>& proc)
+{
+	return new WFDnsServerTask(service, proc);
+}
+

src/factory/WFTaskFactory.inl

@@ -554,6 +554,9 @@ WFNetworkTaskFactory<REQ, RESP>::create_server_task(CommService *service,
 class WFServerTaskFactory
 {
 public:
+	static WFDnsTask *create_dns_task(CommService *service,
+					std::function<void (WFDnsTask *)>& proc);
+
 	static WFHttpTask *create_http_task(CommService *service,
 					std::function<void (WFHttpTask *)>& proc)
 	{

src/server/WFDnsServer.h

@@ -45,5 +45,18 @@ WFDnsServer::WFServer(dns_process_t proc) :
 {
 }
 
+template<> inline
+CommSession *WFDnsServer::new_session(long long seq, CommConnection *conn)
+{
+	WFDnsTask *task;
+
+	task = WFServerTaskFactory::create_dns_task(this, this->process);
+	task->set_keep_alive(this->params.keep_alive_timeout);
+	task->set_receive_timeout(this->params.receive_timeout);
+	task->get_req()->set_size_limit(this->params.request_size_limit);
+
+	return task;
+}
+
 #endif
 

src/server/WFServer.h

@@ -159,6 +159,8 @@ class WFServerBase : protected CommService
 		return -1;
 	}
 
+	const struct WFServerParams *get_params() const { return &this->params; }
+
 protected:
 	/* Override this function to create the initial SSL CTX of the server */
 	virtual SSL_CTX *new_ssl_ctx(const char *cert_file, const char *key_file);