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

媒介

这个也是 来自于一个朋侪的需求 
终极的目标是实现一个 dns 代理服务器, 当然 这本质也是一个 dns 服务器 
并且 dns 代理服务器是依赖于 一个 dns 服务器的, 因此 趁便给一个 dns 服务器的 demo 
这里 主要是 基于 udp 的一个 dns 哀求, 响应数据的交互 

 
 
dns 服务器 

一个基础的 dns 服务器 demo 如下 
为了方便测试, 可以将 端口改为 53, 然后使用 System.out.println 之类的举行调试 

1. package com.hx.test15;
3. import org.xbill.DNS.*;
5. import java.io.ByteArrayOutputStream;
6. import java.net.DatagramPacket;
7. import java.net.DatagramSocket;
8. import java.net.InetAddress;
9. import java.nio.ByteBuffer;
11. /**
12. * @author Jerry.X.He <970655147@qq.com>
13. * @version 1.0
14. * @date 2024-07-02 22:33
15. */
16. public class Test25SimpleDnsServer {
18.     public static void main(String[] args) throws Exception {
19.         DatagramSocket socket = new DatagramSocket(8053);
20.         System.out.println("DNS Server listening on port 8053...");
22.         while (true) {
23.             byte[] receiveData = new byte[512];
24.             DatagramPacket requestPacket = new DatagramPacket(receiveData, receiveData.length);
25.             socket.receive(requestPacket);
27.             byte[] data = requestPacket.getData();
28.             int offset = requestPacket.getOffset();
29.             int length = requestPacket.getLength();
31.             Message request = new Message(ByteBuffer.wrap(data, offset, length));
32.             Message response = handleRequest(request);
34.             ByteArrayOutputStream baos = new ByteArrayOutputStream();
35.             DNSOutput out = new DNSOutput();
36.             baos.write(out.toByteArray());
37.             baos.write(response.toWire());
38.             byte[] responseData = baos.toByteArray();
40.             DatagramPacket responsePacket = new DatagramPacket(responseData, responseData.length,
41.                     requestPacket.getAddress(), requestPacket.getPort());
42.             socket.send(responsePacket);
43.         }
44.     }
46.     private static Message handleRequest(Message request) throws Exception {
47.         Message response = new Message(request.getHeader().getID());
48.         response.getHeader().setFlag(Flags.QR);
49.         response.getHeader().setOpcode(Opcode.QUERY);
51.         Record question = request.getQuestion();
52.         if (question.getName().toString().startsWith("www.abc.com") && question.getType() == Type.A) {
53.             ARecord answer = new ARecord(question.getName(), DClass.IN, 300, InetAddress.getByName("172.0.0.2"));
54.             response.addRecord(answer, Section.ANSWER);
55.         } else {
56.             // apply default
57.         }
59.         return response;
60.     }
63. }

复制代码

 
 
终极实现效果如下, dns 服务器中仅仅配置了 "www.abc.com" -> "172.0.0.2" 
因此 www.abc.com 能够拿到查询效果, www.baidu.com 拿不到查询效果 

 
 
dns 代理服务器

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

1. package com.hx.test15;
3. import org.xbill.DNS.Message;
5. import java.io.ByteArrayOutputStream;
6. import java.io.IOException;
7. import java.net.DatagramPacket;
8. import java.net.DatagramSocket;
9. import java.net.InetAddress;
10. import java.nio.ByteBuffer;
12. /**
13. * Test25JavaDnsForwardServer
14. *
15. * @author Jerry.X.He <970655147@qq.com>
16. * @version 1.0
17. * @date 2024-07-02 22:10
18. */
19. public class Test25JavaDnsForwardServer {
21.     public static void main(String[] args) throws IOException {
22.         DatagramSocket serverSocket = new DatagramSocket(53);
23.         System.out.println("DNS Proxy Server listening on port 53...");
25.         while (true) {
26.             byte[] receiveData = new byte[512];
27.             DatagramPacket requestPacket = new DatagramPacket(receiveData, receiveData.length);
28.             serverSocket.receive(requestPacket);
30.             Message request = readMessage(requestPacket);
31.             Message response = proxyRequest(request);
33.             ByteArrayOutputStream baos = new ByteArrayOutputStream();
34.             baos.write(response.toWire());
35.             byte[] responseData = baos.toByteArray();
37.             DatagramPacket responsePacket = new DatagramPacket(responseData, responseData.length,
38.                     requestPacket.getAddress(), requestPacket.getPort());
39.             serverSocket.send(responsePacket);
40.         }
41.     }
43.     private static Message readMessage(DatagramPacket packet) throws IOException {
44.         byte[] data = packet.getData();
45.         int offset = packet.getOffset();
46.         int length = packet.getLength();
47.         return new Message(ByteBuffer.wrap(data, offset, length));
48.     }
50.     private static Message proxyRequest(Message request) throws IOException {
51.         DatagramSocket upstreamSocket = new DatagramSocket();
52.         InetAddress upstreamAddress = InetAddress.getByName("127.0.0.1");
53.         int upstreamPort = 8053;
55.         ByteArrayOutputStream baos = new ByteArrayOutputStream();
56.         baos.write(request.toWire());
57.         byte[] requestData = baos.toByteArray();
58.         DatagramPacket upstreamRequest = new DatagramPacket(requestData, requestData.length, upstreamAddress, upstreamPort);
59.         upstreamSocket.send(upstreamRequest);
61.         byte[] upstreamResponseData = new byte[512];
62.         DatagramPacket upstreamResponse = new DatagramPacket(upstreamResponseData, upstreamResponseData.length);
63.         upstreamSocket.receive(upstreamResponse);
65.         Message response = new Message(ByteBuffer.wrap(upstreamResponseData, upstreamResponse.getOffset(), upstreamResponse.getLength()));
66.         upstreamSocket.close();
67.         return response;
68.     }
71. }

复制代码

 
终极实现效果如下, dns 服务器中仅仅配置了 "www.abc.com" -> "172.0.0.2" 
因此 www.abc.com 能够拿到查询效果, www.baidu.com 拿不到查询效果 
假如 关闭 目标dns服务, 代理服务器会阻塞住 

 
 
完 
 
 
 
 
 

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。