利用Java实现轻松解析DNS报文

<dependency>  
    <groupId>io.pkts</groupId>  
    <artifactId>pkts-streams</artifactId>  
    <version>3.0.10</version>  
</dependency>  
<dependency>  
    <groupId>io.pkts</groupId>  
    <artifactId>pkts-core</artifactId>  
    <version>3.0.10</version>  
</dependency>  

@RestController
public class FileUpload {

    @PostMapping("/upload/udp")
    public void uploadUdp(MultipartFile uploadFile) throws IOException {
        try (InputStream inputStream = uploadFile.getInputStream()) {
            GZIPInputStream gzipInputStream = new GZIPInputStream(inputStream);
            Pcap pcap = Pcap.openStream(gzipInputStream);
            pcap.loop(packet -> {
                if (packet.hasProtocol(Protocol.UDP)) {
                    UDPPacket udpPacket = (UDPPacket) packet.getPacket(Protocol.UDP);
                    if (udpPacket.getSourcePort() == 53
                            || udpPacket.getDestinationPort() == 53) {
                        final byte[] payloadByteArray = udpPacket.getPayload().getArray();
                        // 在这里根据DNS报文格式解析数据
                        ......
                    }
                }
                return true;
            });
        }
    }

}

private String parseTransactionId(byte[] array) {
    // 前两字节是Transaction ID，表示会话标识
    // DNS请求报文和响应报文通过Transaction ID进行匹配
    return HexUtils.toHexString(Arrays.copyOfRange(array, 0, 2));
}

private int parseDnsPackageType(byte[] array) {
    // 第3和第4字节是Flags，表示标志
    // Flags的第0位代表请求或响应的类型
    // 0表示DNS请求报文，1表示DNS响应报文
    String binaryString = String.format("%08d", Integer.parseInt(Integer.toBinaryString(array[2] & 0xFF)));
    return Integer.parseInt(binaryString.substring(0, 1), 2);
}

private int parseDnsRcode(byte[] array) {
    // 第3和第4字节是Flags，表示标志
    // Flags的最后四位表示Reply Code
    String binaryString = String.format("%08d", Integer.parseInt(Integer.toBinaryString(array[3] & 0xFF)));
    return Integer.parseInt(binaryString.substring(4, 8), 2);
}

private int parseDnsQueryType(byte[] array) {
    // 第13字节开始，是域名，域名以0x00结尾
    // 域名结束后的两个字节就代表DNS查询类型
    int domainEndIndex = -1;
    for (int i = 12; i < array.length; i++) {
        if ((array[i] & 0xFF) == 0) {
            domainEndIndex = i;
            break;
        }
    }
    String s = HexUtils.toHexString(Arrays.copyOfRange(array, domainEndIndex + 1, domainEndIndex + 3));
    return Integer.parseInt(s, 16);
}

private String parseDnsQueryDomain(byte[] array) {
    // 从第13字节开始，遵循[域名长度][域名][域名长度][域名]...0x00的规律
    // 故按照上述规律，从第13字节开始，将域名的所有组成部分获取出来并拼接
    List<String> domainParts = new ArrayList<>();
    int lengthIndex = 12;
    do {
        int partLength = array[lengthIndex] & 0xFF;
        String s = new String(Arrays.copyOfRange(array, lengthIndex + 1, lengthIndex + partLength + 2)).trim();
        domainParts.add(s);
        lengthIndex = lengthIndex + partLength + 1;
    } while ((array[lengthIndex] & 0xFF) != 0);
    return String.join(".", domainParts);
}