/
HttpAgent.ts
72 lines (65 loc) · 2.33 KB
/
HttpAgent.ts
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
import dns from 'dns';
import { LookupFunction, isIP } from 'net';
import {
Agent,
} from 'undici';
import type Dispatcher from 'undici/types/dispatcher';
import type buildConnector from 'undici/types/connector';
export type CheckAddressFunction = (ip: string, family: number | string) => boolean;
export type HttpAgentOptions = {
lookup?: LookupFunction;
checkAddress?: CheckAddressFunction;
connect?: buildConnector.BuildOptions,
};
class IllegalAddressError extends Error {
hostname: string;
ip: string;
family: number;
constructor(hostname: string, ip: string, family: number) {
const message = 'illegal address';
super(message);
this.name = this.constructor.name;
this.hostname = hostname;
this.ip = ip;
this.family = family;
Error.captureStackTrace(this, this.constructor);
}
}
export class HttpAgent extends Agent {
#checkAddress?: CheckAddressFunction;
constructor(options: HttpAgentOptions) {
/* eslint node/prefer-promises/dns: off*/
const _lookup = options.lookup ?? dns.lookup;
const lookup: LookupFunction = (hostname, dnsOptions, callback) => {
_lookup(hostname, dnsOptions, (err, address, family) => {
if (err) return callback(err, address, family);
if (options.checkAddress && !options.checkAddress(address, family)) {
err = new IllegalAddressError(hostname, address, family);
}
callback(err, address, family);
});
};
super({
connect: { ...options.connect, lookup },
});
this.#checkAddress = options.checkAddress;
}
dispatch(options: Agent.DispatchOptions, handler: Dispatcher.DispatchHandlers): boolean {
if (this.#checkAddress && options.origin) {
const originUrl = typeof options.origin === 'string' ? new URL(options.origin) : options.origin;
let hostname = originUrl.hostname;
// [2001:db8:2de::e13] => 2001:db8:2de::e13
if (hostname.startsWith('[') && hostname.endsWith(']')) {
hostname = hostname.substring(1, hostname.length - 1);
}
const family = isIP(hostname);
if (family === 4 || family === 6) {
// if request hostname is ip, custom lookup won't excute
if (!this.#checkAddress(hostname, family)) {
throw new IllegalAddressError(hostname, hostname, family);
}
}
}
return super.dispatch(options, handler);
}
}