@dnslink/test/dns-server.js

A programming-language independent test harness for DNSLink implementations.

const dns = require('dns2')
const { Packet } = dns

module.exports = async function createServer (entries) {
  const lookup = {}
  for (const domain in entries) {
    lookup[toCharCodes(domain)] = entries[domain]
  }
  const server = dns.createServer({
    udp: true,
    tcp: true,
    doh: true,
    handle: (request, send) => {
      const response = Packet.createResponseFromRequest(request)
      // Note: common public DNS servers only support single requests
      const [question] = request.questions
      const { name, type, class: clazz } = question
      let answers = lookup[name] && lookup[name][type]
      if (answers) {
        if (!Array.isArray(answers)) {
          answers = [answers]
        }
        response.answers = answers.map((entry) => {
          return {
            name,
            type,
            class: clazz || Packet.CLASS.IN,
            ttl: entry.ttl,
            data: entry.data
          }
        })
      } else {
        const rcode = lookup[name] && lookup[name].DNS_RCODE && lookup[name].DNS_RCODE[0] && entries[name].DNS_RCODE[0].data
        // In case no answer is given and no error is provided, we use the rcode === 3 aka unknown domain.
        response.header.rcode = rcode || 3
      }
      send(response)
    }
  })
  await server.listen()
  return server
}

function toCharCodes (testDomain) {
  const buff = Buffer.from(testDomain)
  let string = ''
  for (const i of buff) {
    string += String.fromCharCode(i)
  }
  return string
}