DNS নীতি মডিউল (v2)

এই Linux C ABI v2 উদাহরণটি দেখায় কীভাবে DNS কুয়েরি নিয়ন্ত্রণ করতে হয় এবং DNS উত্তর প্রতিস্থাপন করতে হয়। এটি DNS বার্তা পার্স করার জন্য libresolv ব্যবহার করে এবং পরিবর্তিত উত্তরগুলো মডিউলের নিজস্ব মেমরি বাফারে সংরক্ষণ করে।

উদাহরণের আচরণ

কুয়েরির নাম	ফলাফল
allow.test	কুয়েরি হুক চেইন সমাপ্ত করে এবং স্বাভাবিক DNS রেজোলিউশন চালিয়ে যায়।
block.test	আপস্ট্রিম সার্ভারে না পাঠিয়ে সরাসরি REFUSED প্রদান করে।
drop.test	কোনো আপস্ট্রিম অনুসন্ধান বা DNS উত্তর ছাড়াই কুয়েরিটি ড্রপ করে দেয়।
redirect.test	A কুয়েরির জন্য 192.0.2.42 এবং AAAA কুয়েরির জন্য 2001:db8::42 প্রদান করে, যার TTL ৬০ সেকেন্ড।
rewrite.test	স্বাক্ষরবিহীন উত্তরের A রেকর্ডের ঠিকানা পরিবর্তন করে 203.0.113.7 করে এবং TTL ৬০ সেকেন্ড নির্ধারণ করে।
অন্যান্য নাম	পরবর্তী মডিউল এবং স্বাভাবিক প্রক্রিয়াকরণ চালিয়ে যায়।

allow.test এবং rewrite.test-এর জন্য একটি টেস্ট DNS সার্ভার কনফিগার করুন যাতে সেই নামের উত্তর পাওয়া যায়। উদাহরণটি নিজে কোনো আপস্ট্রিম রেকর্ড তৈরি করে না। rewrite.test-এর উত্তরে কোনো সংশ্লিষ্ট A রেকর্ড না থাকলে বা স্বাক্ষর রেকর্ড (DNSSEC) থাকলে উত্তরটি অপরিবর্তিত অবস্থায় ফেরত দেওয়া হয়।

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 কুয়েরিটি ইচ্ছাকৃতভাবেই কোনো উত্তর পাঠায় না, ফলে টাইমআউট ঘটবে।

কাজের পদ্ধতি

কুয়েরি এবং রেসপন্স কলব্যাকগুলো সাফল্যে 0 রিটার্ন করে এবং অ্যাকশনটি result->action-এ সেট করে। কোনো অ্যাকশন কনস্ট্যান্ট সরাসরি রিটার্ন করা হয় না।

ইনিশিয়ালাইজেশন প্রক্রিয়ায় মডিউল কনটেক্সটে একটি রেসপন্স বাফার বরাদ্দ করা হয়। রেসপন্স হুক মূল উত্তরটিকে এই বাফারে কপি করে পরিবর্তন করে, এবং destroy() ফাংশনে কনটেক্সট মুক্ত (free) করে দেয়। বাফার মালিকানার নিয়মাবলী অনুসারে কলব্যাক রিটার্ন করার পরেও প্রতিস্থাপিত বাফারটি বৈধ থাকে।

রেসপন্স হুক কোনো স্বাক্ষরিত বার্তা পরিবর্তন করে না। এটি 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);
}