This repository has been archived by the owner on Jul 13, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathdns.js
83 lines (70 loc) · 2.02 KB
/
dns.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
const dns = require('native-dns');
const authority = { address: '8.8.8.8', port: 53, type: 'udp' };
const entries = [
{
domain: "ximalaya.com",
records: [
{ type: "A", address: "192.168.1.29", ttl: 1800 }
]
},
{
domain: "xmcdn.com",
records: [
{ type: "A", address: "192.168.1.29", ttl: 1800 }
]
}
];
// DNS Proxy
function dnsProxy (question, response, cb) {
console.log('dns proxying:', question.name);
var request = dns.Request({ // eslint-disable-line
question: question, // forwarding the question
server: authority, // this is the DNS server we are asking
timeout: 1000
});
// when we get answers, append them to the response
request.on('message', (err, msg) => {
msg.answer.forEach(a => {
let entry = entries.filter(r => new RegExp(r.domain, 'i').exec(question.name));
if (entry.length) {
entry[0].records.forEach(record => {
console.log('rewrite', question.name, record.address);
record.name = question.name;
record.ttl = record.ttl || 1800;
response.answer.push(dns[record.type](record));
});
} else {
response.answer.push(a);
}
});
});
request.on('end', cb);
request.send();
}
module.exports = function () {
const dnsServer = dns.createServer();
dnsServer.on('request', function handleRequest (request, response) {
console.log('request from', request.address.address, 'for', request.question[0].name);
const f = request.question.map(question =>
new Promise((resolve, reject) =>
dnsProxy(question, response, (err) => {
if (err) {
return reject(err);
}
resolve();
})
)
);
Promise.all(f).then(() => response.send());
});
dnsServer.on('close', () =>
console.log('server closed', dnsServer.address())
);
dnsServer.on('error', (err, buff, req, res) =>
console.error(err.stack)
);
dnsServer.on('socketError', (err, socket) =>
console.error(err)
);
return dnsServer;
};