DNS नीति मॉड्यूल (v2)

यह Linux C ABI v2 उदाहरण DNS क्वेरी नियंत्रित करना और प्रतिक्रिया बदलना दिखाता है। libresolv से संदेश पार्स करता है और बदली प्रतिक्रिया मॉड्यूल के अपने बफ़र में रखता है।

उदाहरण का व्यवहार

क्वेरी नाम	परिणाम
allow.test	क्वेरी हुक श्रृंखला समाप्त करके सामान्य DNS खोज जारी।
block.test	अपस्ट्रीम खोज के बिना REFUSED।
drop.test	अपस्ट्रीम खोज और प्रतिक्रिया के बिना क्वेरी त्याग।
redirect.test	A के लिए 192.0.2.42, AAAA के लिए 2001:db8::42, TTL 60 सेकंड।
rewrite.test	बिना हस्ताक्षर वाली प्रतिक्रिया के A पते को 203.0.113.7 और TTL को 60 सेकंड करें।
अन्य नाम	अगला मॉड्यूल और सामान्य प्रक्रिया जारी।

परीक्षण DNS सर्वर में allow.test और rewrite.test की प्रतिक्रियाएँ कॉन्फ़िगर करें। उदाहरण अपस्ट्रीम रिकॉर्ड नहीं बनाता। rewrite.test प्रतिक्रिया में संबंधित A रिकॉर्ड न हो या हस्ताक्षर रिकॉर्ड हो तो बिना बदलाव लौटाता है।

ALLOW केवल वर्तमान हुक श्रृंखला समाप्त करता है। डोमेन या ट्रांसपोर्ट जाँच नहीं छोड़ता और प्रतिक्रिया हुक परिणाम बदल सकता है। DNS क्रिया नियम देखें।

डाउनलोड और बिल्ड

policy.c और mudfish_dns_module.h एक निर्देशिका में डाउनलोड करें। हेडर मार्गदर्शिका में पूरा हेडर पढ़ सकते हैं।

Linux पर C कंपाइलर, मानक C लाइब्रेरी और libresolv विकास हेडर इंस्टॉल करें। डाउनलोड निर्देशिका में चलाएँ:

cc -std=c11 -shared -fPIC -Wall -Wextra -Werror -I. \
  policy.c -lresolv -o policy.so

मॉड्यूल लोड करें के अनुसार policy.so इंस्टॉल करके सेवा पुनः शुरू करें। क्वेरी परीक्षण से पहले ऐप में DNS शुरू करें।

डिफ़ॉल्ट स्थानीय DNS पते पर रोक और पता प्रतिक्रिया जाँचने के लिए, dig इंस्टॉल करके चलाएँ:

dig @127.0.0.1 block.test A
dig @127.0.0.1 redirect.test A
dig @127.0.0.1 redirect.test AAAA

पहली क्वेरी REFUSED, बाकी दोनों तालिका वाले पते लौटाएँगी। सुनने का पता/पोर्ट बदला हो तो कमांड भी बदलें। drop.test जानबूझकर प्रतिक्रिया नहीं देता, इसलिए timeout होगा।

काम करने का तरीका

क्वेरी और प्रतिक्रिया कॉलबैक सफलता पर 0 लौटाते हैं और result->action सेट करते हैं। क्रिया स्थिरांक सीधे नहीं लौटाते।

आरंभीकरण में मॉड्यूल संदर्भ का प्रतिक्रिया बफ़र आवंटित होता है। प्रतिक्रिया हुक उधार प्रतिक्रिया को इसमें कॉपी करके बदलता है; destroy() संदर्भ मुक्त करता है। बफ़र स्वामित्व नियम के अनुसार कॉलबैक लौटने के बाद भी वैकल्पिक बफ़र वैध रहता है।

हुक हस्ताक्षरित संदेश नहीं बदलता, DNSSEC सत्यापन या पुनः हस्ताक्षर नहीं करता। बनाई या बदली प्रतिक्रियाएँ होस्ट कैश में नहीं रहतीं।

पूरा स्रोत

policy.c

#include "mudfish_dns_module.h"

#include <arpa/inet.h>
#include <arpa/nameser.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>

struct policy {
	uint8_t response[65535];
};

static int
question(const struct mudfish_dns_event *event, ns_rr *rr)
{
	ns_msg message;

	if (event->query.len > 65535 ||
	    ns_initparse(event->query.data, (int)event->query.len, &message) < 0)
		return (-1);
	return (ns_parserr(&message, ns_s_qd, 0, rr));
}

static int32_t
dns_query(void *context, const struct mudfish_dns_event *event,
    struct mudfish_dns_result *result)
{
	ns_rr rr;

	(void)context;
	if (question(event, &rr) < 0)
		return (-1);
	if (strcasecmp(ns_rr_name(rr), "allow.test") == 0)
		result->action = MUDFISH_DNS_ALLOW;
	else if (strcasecmp(ns_rr_name(rr), "block.test") == 0)
		result->action = MUDFISH_DNS_BLOCK;
	else if (strcasecmp(ns_rr_name(rr), "drop.test") == 0)
		result->action = MUDFISH_DNS_DROP;
	else if (strcasecmp(ns_rr_name(rr), "redirect.test") == 0) {
		result->action = MUDFISH_DNS_ANSWER;
		result->ttl = 60;
		if (ns_rr_type(rr) == ns_t_aaaa) {
			result->address_family = MUDFISH_DNS_IPV6;
			inet_pton(AF_INET6, "2001:db8::42", result->address);
		} else {
			result->address_family = MUDFISH_DNS_IPV4;
			inet_pton(AF_INET, "192.0.2.42", result->address);
		}
	}
	return (0);
}

static int32_t
dns_response(void *context, const struct mudfish_dns_event *event,
    struct mudfish_dns_result *result)
{
	struct policy *policy = context;
	ns_msg message;
	ns_rr rr;
	uint8_t *rdata;
	int section, i, changed = 0;

	if (question(event, &rr) < 0)
		return (-1);
	if (strcasecmp(ns_rr_name(rr), "rewrite.test") != 0)
		return (0);
	if (event->response.len > sizeof(policy->response))
		return (-1);
	memcpy(policy->response, event->response.data, event->response.len);
	if (ns_initparse(policy->response, (int)event->response.len, &message) < 0)
		return (-1);
	/* This example only edits unsigned answers. A policy that edits signed
	 * data must also rebuild/remove the affected signatures.
	 */
	for (section = ns_s_an; section <= ns_s_ar; section++) {
		for (i = 0; i < ns_msg_count(message, section); i++) {
			if (ns_parserr(&message, (ns_sect)section, i, &rr) < 0)
				return (-1);
			if (ns_rr_type(rr) == ns_t_rrsig || ns_rr_type(rr) == ns_t_sig ||
			    ns_rr_type(rr) == ns_t_tsig)
				return (0);
		}
	}
	for (i = 0; i < ns_msg_count(message, ns_s_an); i++) {
		if (ns_parserr(&message, ns_s_an, i, &rr) < 0)
			return (-1);
		if (ns_rr_class(rr) != ns_c_in || ns_rr_type(rr) != ns_t_a ||
		    ns_rr_rdlen(rr) != 4)
			continue;
		/* rdata belongs to our copy. TTL precedes RDLENGTH (2 bytes). */
		rdata = policy->response + (ns_rr_rdata(rr) - policy->response);
		inet_pton(AF_INET, "203.0.113.7", rdata);
		ns_put32(60, rdata - 6);
		changed = 1;
	}
	if (changed) {
		result->action = MUDFISH_DNS_REPLACE;
		result->response.data = policy->response;
		result->response.len = event->response.len;
	}
	return (0);
}

static void
destroy(void *context)
{

	free(context);
}

int32_t
mudfish_dns_module_init_v2(uint32_t abi_version, size_t module_size,
    struct mudfish_dns_module_v2 *module)
{

	if (abi_version != MUDFISH_DNS_MODULE_ABI_V2 || module_size != sizeof(*module))
		return (-1);
	module->context = calloc(1, sizeof(struct policy));
	if (module->context == NULL)
		return (-1);
	module->on_dns_query = dns_query;
	module->on_dns_response = dns_response;
	module->destroy = destroy;
	return (0);
}