1 line
8.4 KiB
Plaintext
1 line
8.4 KiB
Plaintext
|
{"version":3,"file":"index.js","sources":["../src/index.ts"],"sourcesContent":["import { forwarded } from '@tinyhttp/forwarded'\nimport type { IncomingMessage } from 'node:http'\nimport ipaddr, { IPv6, IPv4 } from 'ipaddr.js'\n\ntype Req = Pick<IncomingMessage, 'headers' | 'socket'>\n\ntype Trust = ((addr: string, i: number) => boolean) | number[] | string[] | string\n\nconst DIGIT_REGEXP = /^[0-9]+$/\nconst isip = ipaddr.isValid\nconst parseip = ipaddr.parse\n/**\n * Pre-defined IP ranges.\n */\nconst IP_RANGES = {\n linklocal: ['169.254.0.0/16', 'fe80::/10'],\n loopback: ['127.0.0.1/8', '::1/128'],\n uniquelocal: ['10.0.0.0/8', '172.16.0.0/12', '192.168.0.0/16', 'fc00::/7']\n}\n\n/**\n * Static trust function to trust nothing.\n */\nconst trustNone = () => false\n\n/**\n * Get all addresses in the request, optionally stopping\n * at the first untrusted.\n *\n * @param request\n * @param trust\n */\nfunction alladdrs(req: Req, trust: Trust): string[] {\n // get addresses\n\n const addrs = forwarded(req)\n\n if (!trust) return addrs\n\n if (typeof trust !== 'function') trust = compile(trust)\n\n for (let i = 0; i < addrs.length - 1; i++) {\n if (trust(addrs[i], i)) continue\n addrs.length = i + 1\n }\n return addrs\n}\n/**\n * Compile argument into trust function.\n *\n * @param val\n */\nfunction compile(val: string | string[] | number[]): (addr: string) => boolean {\n let trust: string[]\n if (typeof val === 'string') trust = [val]\n else if (Array.isArray(val)) trust = val.slice() as string[]\n else throw new TypeError('unsupported trust argument')\n\n for (let i = 0; i < trust.length; i++) {\n val = trust[i]\n if (!Object.prototype.hasOwnProperty.call(IP_RANGES, val)) continue\n\n // Splice in pre-defined range\n val = IP_RANGES[val as string]\n // eslint-disable-next-line prefer-spread\n trust.splice.apply(trust, [i, 1].concat(val as number[]))\n i += val.length - 1\n }\n return compileTrust(compileRangeSubnets(trust))\n}\n/**\n * Compile `arr` elements into range subnets.\n */\nfunction compileRangeSubnets(arr: string[]) {\n const rangeSubnets = new Array(arr.length)\n for (let i = 0; i < arr.length; i++) rangeSubnets[i] = parseIPNotation(arr[i])\n\n return rangeSubnets\n}\n/**\n * Compile range subnet array into trust function.\n *\n * @param rangeSubnets\n */\nfunction compileTrust(rangeSubnets: (IPv4 | IPv6)[]) {\n // Return optimized function based on length\n const len = rangeSubnets.length\n return len === 0 ? trustNone : len === 1 ? trustSingle(rangeSubnets[0]) : trustMulti(rangeSubnets)\n}\n/**\n * Parse IP notation string into range subnet.\n *\n * @param {String} note\n * @private\n */\nexport function parseIPNotation(note: string): [IPv4 | IPv6, string | number] {\n const pos = note.lastIndexOf('/')\n const str = pos !== -1 ? note.substring(0, pos) : note\n\n if (!isip(str)) throw new TypeError('invalid IP address: ' + str)\n\n let ip = parseip(str)\n\n if (pos === -1 && ip.kind() === 'ipv6') {\n ip = ip as IPv6\n\n if (ip.isIPv4MappedAddress()) ip = ip.toIPv4Address()\n }\n\n const max = ip.kind() === 'ipv6' ? 128 : 32\n\n let range: string | number = pos !== -1 ? note.substring(pos + 1, note.length) : null\n\n if (range === null) range = max\n else if (DIGIT_REGEXP.test(range)) range = parseInt(range, 10)\n else if (ip.kind() === 'ipv4' && isip(range)) range = parseNetmask(range)\n else range = null\n\n if (typeof range === 'number' && (range <= 0 || range > max)) throw new TypeError('invalid range on address: ' + note)\n\n return [ip, range]\n}\n/**\n * Parse netmask string into CIDR range.\n *\n * @param netmask\n * @private\n */\nfunction parseNetmask(netmask: string) {\n const ip = parseip(netmask)\n return ip.kind() === 'ipv4' ? ip.prefixLengthFromSubnetMask() : null\n}\n/**\n * Determine address of proxied request.\n *\n * @param request\n * @param trust\n * @public\n */\nexport function proxyaddr(req: Req, trust: Trust): string {\n const addrs = alladdrs(req, trust)\n\n return addrs[addrs.length - 1]\n}\n\n/**\n * Compil
|