[Java][网络程序开发]《DNS服务简单实现》

酒诡

项目使用了Netty高性能网络通信框架。
使用了4.1.X版本
启动后，就可以使用该服务器作为DNS服务器。

1. public class Dns {
   
2. 
   
3.     public static void main(String[] args) {
   
4.         final NioEventLoopGroup group = new NioEventLoopGroup();
   
5.         try {
   
6.             Bootstrap bootstrap = new Bootstrap();
   
7.             bootstrap.group(group).channel(NioDatagramChannel.class)
   
8.                     .handler(new ChannelInitializer<NioDatagramChannel>() {
   
9.                         @Override
   
10.                         protected void initChannel(NioDatagramChannel nioDatagramChannel) throws Exception {
    
11.                             nioDatagramChannel.pipeline().addLast(new DatagramDnsQueryDecoder());
    
12.                             nioDatagramChannel.pipeline().addLast(new DatagramDnsResponseEncoder());
    
13.                             nioDatagramChannel.pipeline().addLast(new DnsHandler());
    
14.                         }
    
15.                     }).option(ChannelOption.SO_BROADCAST, true);
    
16. 
    
17.             ChannelFuture future = bootstrap.bind(53).sync();
    
18.             future.channel().closeFuture().sync();
    
19.         } catch (InterruptedException e) {
    
20.             e.printStackTrace();
    
21.         } finally {
    
22.             group.shutdownGracefully();
    
23.         }
    
24.     }
    
25. }
    
26. 
    
27. class DnsHandler extends SimpleChannelInboundHandler<DatagramDnsQuery> {
    
28.     @Override
    
29.     public void channelRead0(ChannelHandlerContext ctx, DatagramDnsQuery query) throws UnsupportedEncodingException {
    
30.         // 假数据，域名和ip的对应关系应该放到数据库中
    
31.         Map<String, byte[]> ipMap = new HashMap<>();
    
32.         ipMap.put("www.cnhongker.net.", new byte[] { 74, 121, (byte) 149, (byte)218 });
    
33.         DatagramDnsResponse response = new DatagramDnsResponse(query.recipient(), query.sender(), query.id());
    
34.         try {
    
35.             DefaultDnsQuestion dnsQuestion = query.recordAt(DnsSection.QUESTION);
    
36.             response.addRecord(DnsSection.QUESTION, dnsQuestion);
    
37.             System.out.println("查询的域名：" + dnsQuestion.name());
    
38. 
    
39.             ByteBuf buf = null;
    
40.             if (ipMap.containsKey(dnsQuestion.name())) {
    
41.                 buf = Unpooled.wrappedBuffer(ipMap.get(dnsQuestion.name()));
    
42.             } else {
    
43.                 // TODO  对于没有的域名采用迭代方式
    
44.                 // buf = Unpooled.wrappedBuffer(new byte[] { 127, 0, 0, 1});
    
45.             }
    
46.             // TTL设置为10s, 如果短时间内多次请求，客户端会使用本地缓存
    
47.             DefaultDnsRawRecord queryAnswer = new DefaultDnsRawRecord(dnsQuestion.name(), DnsRecordType.A, 10, buf);
    
48.             response.addRecord(DnsSection.ANSWER, queryAnswer);
    
49. 
    
50.         } catch (Exception e) {
    
51.             System.out.println("异常了：" + e);
    
52.         }finally {
    
53.             ctx.writeAndFlush(response);
    
54.         }
    
55.     }
    
56. 
    
57.     @Override
    
58.     public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
    
59.         cause.printStackTrace();
    
60.     }
    
61. }

复制代码