03 基于 java udp 做一个dns服务器 和 一个dns代理服务器

前言

这个也是 来自于一个朋友的需求

最终的目的是实现一个 dns 代理服务器, 当然 这本质也是一个 dns 服务器

并且 dns 代理服务器是依赖于 一个 dns 服务器的, 因此 顺便给一个 dns 服务器的 demo

这里 主要是 基于 udp 的一个 dns 请求, 响应数据的交互

dns 服务器

一个基础的 dns 服务器 demo 如下

为了方便测试, 可以将 端口改为 53, 然后使用 System.out.println 之类的进行调试

复制代码
package com.hx.test15;

import org.xbill.DNS.*;

import java.io.ByteArrayOutputStream;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.nio.ByteBuffer;

/**
 * @author Jerry.X.He <970655147@qq.com>
 * @version 1.0
 * @date 2024-07-02 22:33
 */
public class Test25SimpleDnsServer {

    public static void main(String[] args) throws Exception {
        DatagramSocket socket = new DatagramSocket(8053);
        System.out.println("DNS Server listening on port 8053...");

        while (true) {
            byte[] receiveData = new byte[512];
            DatagramPacket requestPacket = new DatagramPacket(receiveData, receiveData.length);
            socket.receive(requestPacket);

            byte[] data = requestPacket.getData();
            int offset = requestPacket.getOffset();
            int length = requestPacket.getLength();

            Message request = new Message(ByteBuffer.wrap(data, offset, length));
            Message response = handleRequest(request);

            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            DNSOutput out = new DNSOutput();
            baos.write(out.toByteArray());
            baos.write(response.toWire());
            byte[] responseData = baos.toByteArray();

            DatagramPacket responsePacket = new DatagramPacket(responseData, responseData.length,
                    requestPacket.getAddress(), requestPacket.getPort());
            socket.send(responsePacket);
        }
    }

    private static Message handleRequest(Message request) throws Exception {
        Message response = new Message(request.getHeader().getID());
        response.getHeader().setFlag(Flags.QR);
        response.getHeader().setOpcode(Opcode.QUERY);

        Record question = request.getQuestion();
        if (question.getName().toString().startsWith("www.abc.com") && question.getType() == Type.A) {
            ARecord answer = new ARecord(question.getName(), DClass.IN, 300, InetAddress.getByName("172.0.0.2"));
            response.addRecord(answer, Section.ANSWER);
        } else {
            // apply default
        }

        return response;
    }


}

最终实现效果如下, dns 服务器中仅仅配置了 "www.abc.com" -> "172.0.0.2"

因此 www.abc.com 能够拿到查询结果, www.baidu.com 拿不到查询结果

dns 代理服务器

主需求是实现一个 dns 的代理服务器

实现如下, 无非就是 代理发送一次 dns 的请求 到目标 dns 服务器, 拿到响应之后 响应给客户端

复制代码
package com.hx.test15;

import org.xbill.DNS.Message;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.nio.ByteBuffer;

/**
 * Test25JavaDnsForwardServer
 *
 * @author Jerry.X.He <970655147@qq.com>
 * @version 1.0
 * @date 2024-07-02 22:10
 */
public class Test25JavaDnsForwardServer {

    public static void main(String[] args) throws IOException {
        DatagramSocket serverSocket = new DatagramSocket(53);
        System.out.println("DNS Proxy Server listening on port 53...");

        while (true) {
            byte[] receiveData = new byte[512];
            DatagramPacket requestPacket = new DatagramPacket(receiveData, receiveData.length);
            serverSocket.receive(requestPacket);

            Message request = readMessage(requestPacket);
            Message response = proxyRequest(request);

            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            baos.write(response.toWire());
            byte[] responseData = baos.toByteArray();

            DatagramPacket responsePacket = new DatagramPacket(responseData, responseData.length,
                    requestPacket.getAddress(), requestPacket.getPort());
            serverSocket.send(responsePacket);
        }
    }

    private static Message readMessage(DatagramPacket packet) throws IOException {
        byte[] data = packet.getData();
        int offset = packet.getOffset();
        int length = packet.getLength();
        return new Message(ByteBuffer.wrap(data, offset, length));
    }

    private static Message proxyRequest(Message request) throws IOException {
        DatagramSocket upstreamSocket = new DatagramSocket();
        InetAddress upstreamAddress = InetAddress.getByName("127.0.0.1");
        int upstreamPort = 8053;

        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        baos.write(request.toWire());
        byte[] requestData = baos.toByteArray();
        DatagramPacket upstreamRequest = new DatagramPacket(requestData, requestData.length, upstreamAddress, upstreamPort);
        upstreamSocket.send(upstreamRequest);

        byte[] upstreamResponseData = new byte[512];
        DatagramPacket upstreamResponse = new DatagramPacket(upstreamResponseData, upstreamResponseData.length);
        upstreamSocket.receive(upstreamResponse);

        Message response = new Message(ByteBuffer.wrap(upstreamResponseData, upstreamResponse.getOffset(), upstreamResponse.getLength()));
        upstreamSocket.close();
        return response;
    }


}

最终实现效果如下, dns 服务器中仅仅配置了 "www.abc.com" -> "172.0.0.2"

因此 www.abc.com 能够拿到查询结果, www.baidu.com 拿不到查询结果

如果 关闭 目标dns服务, 代理服务器会阻塞住

相关推荐
Jayyih41 分钟前
嵌入式系统学习Day30(udp)
网络协议·学习·udp
耐达讯通信技术2 小时前
耐达讯自动化RS485与Profinet双向奔赴,伺服驱动器连接“稳稳拿捏”
运维·人工智能·物联网·网络协议·自动化·信息与通信
耐达讯通信技术2 小时前
嘎嘎厉害!耐达讯自动化RS485转Profinet网关就是食品温控的“天选之子”
运维·服务器·网络·人工智能·网络协议·自动化·信息与通信
川石课堂软件测试3 小时前
Oracle 数据库:视图与索引
数据库·网络协议·nginx·http·oracle·grafana·prometheus
Craze_rd3 小时前
服务 HTTP 转 SRPC 技术方案
网络·网络协议·http·rpc·golang
卓码软件测评4 小时前
第三方web测评机构:【WEB安全测试中HTTP方法(GET/POST/PUT)的安全风险检测】
前端·网络协议·安全·web安全·http·xss
Whitess0074 小时前
Websocket链接如何配置nginx转发规则?
websocket·网络协议·nginx
zz-zjx7 小时前
进程与线程详解, IPC通信与RPC通信对比,Linux前台与后台作业
linux·网络协议·rpc
迈威通信12 小时前
迈威通信从送快递角度教你分清网络二层和三层
网络·网络协议·智能路由器·信息与通信
郝亚军12 小时前
根据Sec-WebSocket-Key计算Sec-WebSocket-Accept
网络·websocket·网络协议