feat(node): 拨测支持请求级 DNS 与强制解析,前后端与中间件全链路透传

背景:排查解析差异(污染、区域 DNS、预发布域名)时需要「临时换解析服务器」或
「把某域名指到指定 IP」再拨测,此前只能改节点全局配置或 hosts。

节点 src
- webtest 新增 ReqResolve:dns / dns4 / dns6 + resolve / resolve4 / resolve6 九参数,
  ParseReqResolve 统一解析;Servers(stack) 按「栈内覆盖 → dns 回落」取服务器,
  Override(host, wantV6) 取强制解析目标(map 键经 canonicalHost 归一化)
- 解析层补 ResolveXRecordWith(domain, servers) 与 ResolveByType,ctx 传参走
  WithDNSServers / DNSServersFromContext
- 拨号:initHTTPClients 的 DialContext 从 ctx 取参数——命中 Override 时改写 addr,
  IP 直连、域名当作新 host 再解析一跳;未命中则用请求级服务器解析
- tcping:ResolveHostWith 同样支持 Override 与自定义 DNS;字面 IP 也做私网拦截
- SSRF:解析器改为可注入(ssrf.SetResolveFunc + webtest.InstallSSRFResolver),
  预检与实际拨号同源,新增 src/webtest/ssrfresolver.go
- 缓存:ReqResolve.CacheKey() 把参数按固定字段顺序拼进键,替代「有参数即跳过共享
  缓存」;零值返回空串,旧缓存键逐字节不变,升级不失效
- ws:wsProbe 接收 query,与 HTTP 侧同口径

middleware
- middleware-go:query 透传从 tcping/speed 分支提升到所有分支(detail/ssl/dns/whois
  也带 dns 系参数),WS 侧以 map 传给 RequestProbe
- frontend-ssr 内置中间件路由([...slug].get.ts)同步透传 query

frontend-ssr
- 新增 AdvancedOptions 组件:dns 通用 + 按栈 dns4/dns6、resolve 三档;单栈页只渲染
  本栈一列,dns/whois 页隐藏 resolve
- 8 个工具页接线;utils/tools.ts 新增 withQuery(空对象原样返回,不填参数时 URL 不变)
- dev 下注入本地调试节点并排在首位,中间件指向 127.0.0.1:8091

docs
- node-api 补 resolve 参数表、IP / 域名两种目标的语义、缓存键片段与 TTL 修正

验证:本地链路(前端 3010 → 中间件 8091 → 节点 18081)tcping/ssl/detail 带
dns= / resolve= 实测生效;SSRF 开启时私网目标被拦、关闭时放行。
This commit is contained in:
nomdn committed 2026-10-17 21:01:38 +08:00
1 parent 2947ad2125
commit 8b094bcb36
26 files changed
+1206 -321

No files matched your search

+74 -14
View File
@@ -42,7 +42,7 @@ GET /info → {"version":"v1.2.3","capabilities":["probe","report","config","ot
| GET | `/v1/detail/*url` | 目标 URL(路径传入,自动补全协议) | 网站探测明细:DNS/TCP/HTTP 各阶段耗时、状态码、页面体积、下载速度、可达性;双栈分别返回 |
| GET | `/v1/ssl/*url` | 目标 URL | SSL 证书检查:剩余有效天数、颁发/主体、TLS 版本、是否过期;双栈分别返回 |
| GET | `/v1/speed/:version/*url` | `version` = `v4` / `v6`,目标 URL | 下载测速 |
| GET | `/v1/tcping/:ip` | `port`(默认 80)、`count`(默认 4,上限 20) | TCP 连通性与 RTT 统计 |
| GET | `/v1/tcping/:ip` | `port`(默认 80)、`count`(默认 4,上限 20)、`version`(见下文「请求级参数」) | TCP 连通性与 RTT 统计 |
| GET | `/v1/dns/:type/*domain` | `type` ∈ `a`/`aaaa`/`cname`/`mx`/`ns`/`ptr`/`srv`/`txt`/`caa` | DNS 记录查询(`dns-server` 支持主从 failover 与 DoH) |
| GET | `/v1/dnssec/:domain` | 域名 | DNSSEC 链式验证(`dnssec-server`,留空沿用 `dns-server`) |
| GET | `/v1/whois/:domain` | 域名 | Whois 注册信息 |
@@ -54,7 +54,36 @@ GET /info → {"version":"v1.2.3","capabilities":["probe","report","config","ot
**单栈模式**:`single-stack` 配为 `ipv4` / `ipv6` 时只跑对应栈,另一栈直接返回 `Skipped due to SINGLE_STACK=...`,不发起无意义的探测与超时。
**SSRF 防护**:`block-private-ips` 开启时,`detail` / `ssl` 对解析到内网/私有地址的目标直接返回伪造结果,出站客户端同时拒绝连接私有 IP、禁止跨跳重定向。
**SSRF 防护**:`block-private-ips` 开启时,`detail` / `ssl` 对解析到内网/私有地址的目标直接返回伪造结果,出站客户端同时拒绝连接私有 IP、禁止跨跳重定向。`tcping` 是裸 TCP、不经过出站客户端,故在 `webtest.ResolveHostWith` 内单独把关:**字面 IP**、**DNS 解析结果**、**`resolve*` 改写后的目标**三者只要落在私网/回环/链路本地/组播网段一律拒绝,该栈返回 `Error: connection to private/internal address <IP> is not allowed`(`sent` 仍为请求次数、`success=0`、`loss_rate=100`);关闭开关时三者一并放行。
### 请求级参数
除路径参数与常规 query 外,`/v1/tcping/:ip` 与所有走 DNS 的探测接口还接受一组**请求级参数**,用于按次覆盖解析行为。参数统一由 `webtest.ParseReqResolve` 解析,HTTP 拨测经 context、`tcping` 经形参透传到解析层;HTTP 与 WS 两条通道语义一致。
| 参数 | 取值 | 作用接口 | 说明 |
|------|------|----------|------|
| `dns` | 逗号分隔的 `ip:port` 或 `http(s)://` DoH URL | `detail` / `ssl` / `speed` / `tcping` / `whois` / `dns` | 本次探测改用指定 DNS 服务器,取代配置的 `dns-server` |
| `dns4` | 同上 | 同上 | 只覆盖 IPv4 族解析,**优先于 `dns`** |
| `dns6` | 同上 | 同上 | 只覆盖 IPv6 族解析,**优先于 `dns`** |
| `resolve` | 逗号分隔的 `域名:目标`,目标为 IP 或域名 | `detail` / `ssl` / `speed` / `tcping` | 强制改写某域名的解析结果(两栈通用) |
| `resolve4` | 同上 | 同上 | 只影响 IPv4 侧;非空时 IPv4 侧**只用这张表**(与 `dns4` 同口径) |
| `resolve6` | 同上 | 同上 | 只影响 IPv6 侧;非空时 IPv6 侧**只用这张表** |
| `version` | `ipv4` / `ipv6`(兼容 `v4` / `v6`,大小写与首尾空格不敏感) | `tcping` | 请求级指定单栈 |
`dns*` 与 `resolve*` 是两类正交的覆盖:前者「换服务器去问」,后者「直接改答案」,可同时使用。
**`resolve*` 的目标可以是 IP,也可以是域名**:
- 目标是 **IP**(`resolve4=example.com:1.2.3.4`,IPv6 可写 `[::1]` 或裸地址)→ 拨号阶段不再查 DNS,直接对该 IP 做私网校验后连接;
- 目标是 **域名**(`resolve=www.example.com:origin.internal`)→ 把它当作新的 host 继续走解析,解析服务器仍取 `dns*`/配置,且**不再查映射表**(因此不会绕成环)。
行为要点:
- **取值与配置侧同一套校验**(`NormalizeServers`):合法项为 `ip:port`(UDP)或 DoH URL,非法项被丢弃;**参数存在但所有项都非法时返回 `400`**,不静默忽略——避免「看起来指定了、实际按默认跑」。`resolve*` 同理:每组必须形如 `域名:目标`,目标为 IP 或形如域名(不含端口、空格),非法项丢弃、全非法返回 `400`。
- **`version` 不越权单栈配置**:节点 `single-stack` 非空时一律以配置为准(单栈机指定另一栈也跑不起来);仅当配置为空(双栈机)时 `version` 才生效。跳过文案借此区分来源:配置造成记 `SINGLE_STACK`,参数造成记 `version`。
- **作用止于「域名 → IP」这一步**:`dns*` 只更换解析服务器、`resolve*` 只改写解析结果,都不改变探测目标与端口;`dnssec` / `location` / `asn` 三类接口不接受上述参数,`whois` / `dns` 只取 `dns*`、忽略 `resolve*`(`dns` 拨测问的就是真实解析,改写会自相矛盾)。
- **SSRF 预检同源**:开启 `block-private-ips` 时,预检校验的是**拨号真正会连的地址**(即 `resolve*` 改写后的目标),不会拿原域名的解析结果去判定。`tcping` 没有独立的预检步骤,同一判断落在 `ResolveHostWith` 进入拨号之前,口径与上述一致。
- **带覆盖时按独立缓存键缓存**:规范化后的参数会拼进缓存键(见「七、缓存与并发」),不同服务器/不同映射的结果各占一条,既不串味、也不牺牲命中率。
---
@@ -198,16 +227,47 @@ POST /v1/ota
业务探测结果在节点侧缓存,避免同 key 洪打把上游打崩:
| 缓存 | 键 | TTL |
|------|-----|-----|
| `detail` | 归一化 URL | 5 分钟 |
| `ssl` | URL | 5 分钟 |
| `speed` | `url:version` | 5 分钟 |
| `tcping` | `host:port:count` | 5 分钟 |
| `whois` | 域名 | 5 分钟 |
| `asn whois` | `AS<asn>` | 5 分钟 |
| Bilibili 归属地(在线 API) | IP | 24 小时 |
| 缓存 | 键(末尾拼请求级解析片段) | TTL(成功 / 失败) |
|------|-----|--------------------|
| `detail` | 归一化 URL | 30s / 5s |
| `ssl` | URL | 30s / 5s |
| `speed` | `url:version` | 30s / 5s |
| `tcping` | `host:port:count:stack` | 30s / 5s |
| `whois` | 域名 | 30s / 5s |
| `asn whois` | `AS<asn>` | 30s / 5s |
| Bilibili 归属地(在线 API) | IP | 24 小时(不分成败) |
- **失败结果 30s 后删除**:探测失败(如目标不可达)不缓存整 5 分钟,避免故障恢复后长时间返回旧结论。
- **singleflight 合并**:`detail` / `ssl` / `tcping` / `speed` 的并发同 key 请求合并为一次真实探测。
- **定时清扫**:每 10 分钟清理过期缓存条目与 Bilibili 缓存,防止只靠"同 key 重访才淘汰"时内存无限增长。
- **TTL 按结果分档**(`cacheTTL`):成功 30s、失败 5s。失败结果不缓存太久,避免故障恢复后长时间返回旧结论;TTL 记在条目里、读时判断,无需后台协程唤醒。
- **缓存键含生效栈**:`tcping` 的键为 `host:port:count:stack`,`version=ipv6` 的结果与默认双栈不会互相污染。
- **singleflight 合并**:`detail` / `ssl` / `tcping` / `speed` 的并发同 key 请求合并为一次真实探测(合并键同缓存键)。
- **请求级解析参数进键**:`dns*` 与 `resolve*` 一并拼入,详见下一节。
- **定时清扫**:每 1 分钟清理过期缓存条目,防止只靠「同 key 重访才淘汰」时内存无限增长。
### 7.1 请求级解析参数如何进入缓存键
缓存键的基础部分只含**目标**维度(URL / host),但探测结果依赖**怎么解析**——同一域名经不同服务器、或经 `resolve*` 改写后可能指向完全不同的 IP(企业内网、CDN 调度、分线路权威、上线前的强制指向)。若把两者混在一条键里,会双向串味且无从察觉:
- 带 `dns=1.1.1.1:53` 的结果写进 `example.com` 这条 → 随后的普通请求(用配置的 `dns-server`)读出来,用户以为是自己默认 DNS 的结果;
- 反过来,普通请求的结果也会被带 `dns=` 的请求读走——用户明明指定了服务器,却拿到别人用另一台服务器查的结果;
- `resolve*` 更直接:强制指向 A 与强制指向 B 是两次不同探测,共用一条键等于互相覆盖。
因此把**规范化后的参数片段拼在键末尾**(`ReqResolve.CacheKey()`),两件事同时成立:不同解析配置各占一条,相同配置仍能命中。
```
https://example.com ← 无请求级参数
https://example.com;dns=1.1.1.1:53 ← dns=
https://example.com:v4;dns=8.8.8.8:53;dns6=[2400::1]:53 ← speed,按栈覆盖
qq.com:80:4:ipv4;dns=1.1.1.1:53 ← tcping
https://example.com;resolve4=example.com:127.0.0.1 ← resolve4,强制解析
```
片段规则:
- **字段顺序固定**为 `dns` → `dns4` → `dns6` → `resolve` → `resolve4` → `resolve6`。所以参数在 URL 里谁先谁后不影响键(`?dns6=…&dns=…` 与 `?dns=…&dns6=…` 是同一个键)。
- **`dns*` 内部按输入顺序,`resolve*` 内部按域名排序**:映射表本质是 map,迭代顺序随机;不排序的话同一请求每次算出的键都不同,缓存永远命不中。
- **无参数时片段为空串** ⇒ 键与未引入该特性时**逐字节一致**,升级后旧缓存条目照常命中。
- **按规范化结果拼**:非法项在校验阶段已被丢弃,故 `dns=1.1.1.1:53,garbage` 与 `dns=1.1.1.1:53` 落在同一个键上;`resolve4=Example.COM.:1.2.3.4` 与 `resolve4=example.com:1.2.3.4` 同理(域名统一小写、去尾点)。
- **键空间随组合增长**:每条仍受 30s / 5s 的 TTL 约束,由每分钟的清扫兜住,不会无限堆积。
- `asn` / `location` / `dnssec` 不接这些参数,键与以前相同。
> 已知残留(与本次改动无关):`detail` 与 `ssl` 的 singleflight 合并键都是「URL + 片段」,两种探测**并发**打同一 URL 时会撞在一起,后到者按另一类型断言会 panic(被 `recover` 接住 → 500)。修法是给合并键加探测类型前缀,暂未处理。
@@ -0,0 +1,207 @@
<script setup lang="ts">
import { ref } from 'vue'
/**
* 高级选项:请求级解析参数(与 ipw-ak 的 AdvancedOptions 同源)
*
* - 展开后按地址族分列:IPv4 → dns4 / resolve4,IPv6 → dns6 / resolve6
* - stack 非空时只渲染对应栈那一列 —— tcping / speed 的前端两个栈是分开的页面,
* 单栈页只可能发出该栈的请求,另一列的参数永远吃不到,给了就是死 UI。
* - dns / resolve 是两栈通用的兜底,dns4/dns6、resolve4/resolve6 非空时覆盖它们。
*
* 用法:
* <AdvancedOptions ref="adv" /> 双栈页(dns4/dns6 + resolve4/resolve6)
* <AdvancedOptions ref="adv" stack="ipv4" /> 单栈页,只给 IPv4 一列
* <AdvancedOptions ref="adv" :show-resolve="false" /> 仅 dns4/dns6(dns、whois 页不支持强制解析)
*
* 页面里取参数:
* withQuery(baseUrl, adv.value?.query() ?? {})
*/
const props = withDefaults(
defineProps<{
/** 是否显示 resolve / resolve4 / resolve6(dns、whois 页不支持强制解析) */
showResolve?: boolean
/** 只显示某一栈的配置项;单栈页面(tcping / speed 的 v4、v6 页)传,双栈页面不传 */
stack?: 'ipv4' | 'ipv6' | ''
}>(),
{ showResolve: true, stack: '' },
)
const expanded = ref(false)
/** 两栈通用的 DNS 服务器:dns4/dns6 非空时覆盖它 */
const dns = ref('')
const dns4 = ref('')
const dns6 = ref('')
/** 两栈通用的强制解析:resolve4/resolve6 非空时覆盖它 */
const resolve = ref('')
const resolve4 = ref('')
const resolve6 = ref('')
/** 组装请求级解析参数(空值不下发) */
function query(): Record<string, string> {
const q: Record<string, string> = {}
if (dns.value.trim()) q.dns = dns.value.trim()
if (dns4.value.trim()) q.dns4 = dns4.value.trim()
if (dns6.value.trim()) q.dns6 = dns6.value.trim()
if (props.showResolve) {
if (resolve.value.trim()) q.resolve = resolve.value.trim()
if (resolve4.value.trim()) q.resolve4 = resolve4.value.trim()
if (resolve6.value.trim()) q.resolve6 = resolve6.value.trim()
}
return q
}
defineExpose({ query })
</script>
<template>
<div class="adv">
<!-- 用原生 button:页面的全局 .el-button{width:165px} 会把 el-button 撑成实心大按钮 -->
<button type="button" class="adv__toggle" :aria-expanded="expanded" @click="expanded = !expanded">
<span class="adv__caret" :class="{ 'is-open': expanded }">▸</span>
<span>高级选项</span>
</button>
<div v-if="expanded" class="adv__body">
<!-- 两栈通用的 DNS 服务器(单栈页填这一项即可;dns4/dns6 非空时覆盖它) -->
<div class="adv__field adv__field--wide">
<span class="adv__label">DNS 服务器(dns,两栈通用)</span>
<el-input v-model="dns" clearable placeholder="如 1.1.1.1:53,逗号分隔" />
</div>
<!-- 两栈通用的强制解析(单栈页填这一项即可;resolve4/6 非空时覆盖它) -->
<div v-if="showResolve" class="adv__field adv__field--wide">
<span class="adv__label">强制解析(resolve,两栈通用)</span>
<el-input v-model="resolve" clearable placeholder="如 example.com:1.2.3.4,逗号分隔" />
</div>
<div class="adv__grid" :class="{ 'is-single': !!stack }">
<section v-if="!stack || stack === 'ipv4'" class="adv__col">
<h4 class="adv__col-title">IPv4</h4>
<div class="adv__field">
<span class="adv__label">DNS 服务器(dns4)</span>
<el-input v-model="dns4" clearable placeholder="如 1.1.1.1:53,逗号分隔" />
</div>
<div v-if="showResolve" class="adv__field">
<span class="adv__label">强制解析(resolve4)</span>
<el-input v-model="resolve4" clearable placeholder="如 example.com:1.2.3.4" />
</div>
</section>
<section v-if="!stack || stack === 'ipv6'" class="adv__col">
<h4 class="adv__col-title">IPv6</h4>
<div class="adv__field">
<span class="adv__label">DNS 服务器(dns6)</span>
<el-input v-model="dns6" clearable placeholder="如 [2400:3200::1]:53,逗号分隔" />
</div>
<div v-if="showResolve" class="adv__field">
<span class="adv__label">强制解析(resolve6)</span>
<el-input v-model="resolve6" clearable placeholder="如 example.com:[::1]" />
</div>
</section>
</div>
<p class="adv__hint">
DNS 服务器可填 <code>ip:port</code> 或 DoH 地址,逗号分隔多个;强制解析格式
<code>域名:IP</code> 或 <code>域名:CNAME</code>,IPv6 目标写 <code>[::1]</code>。{{
stack ? 'dns4 / dns6、resolve4 / resolve6 非空时覆盖 dns / resolve,仅对本栈生效。'
: 'dns4 / dns6、resolve4 / resolve6 非空时覆盖 dns / resolve,分别只作用于对应栈。'
}}
</p>
</div>
</div>
</template>
<style scoped>
.adv {
margin-top: 0.25rem;
}
.adv__toggle {
display: inline-flex;
align-items: center;
gap: 0.3rem;
width: auto;
height: auto;
margin: 0;
padding: 0;
border: 0;
background: none;
color: var(--el-color-primary, #299764);
font: inherit;
font-size: 0.85rem;
letter-spacing: 0.03em;
cursor: pointer;
}
.adv__caret {
display: inline-block;
font-size: 0.7rem;
transition: transform 150ms ease;
}
.adv__caret.is-open {
transform: rotate(90deg);
}
.adv__body {
margin-top: 0.5rem;
padding: 0.75rem 0.9rem;
border: 1px solid var(--el-border-color-lighter, #ebeef5);
border-radius: 6px;
background: var(--el-fill-color-lighter, #fafafa);
}
.adv__grid {
display: grid;
grid-template-columns: 1fr 1fr;
gap: 0 1rem;
}
/* 单栈页只渲染一列:撑满,别再留半幅空白 */
.adv__grid.is-single {
grid-template-columns: 1fr;
}
.adv__col-title {
margin: 0 0 0.4rem;
font-size: 0.8rem;
font-weight: 700;
letter-spacing: 0.08em;
color: var(--el-text-color-secondary, #909399);
}
.adv__field {
margin-bottom: 0.5rem;
}
.adv__field--wide {
margin-bottom: 0.75rem;
}
.adv__label {
display: block;
margin-bottom: 0.25rem;
font-size: 0.78rem;
color: var(--el-text-color-secondary, #909399);
}
.adv__hint {
margin: 0.35rem 0 0;
font-size: 0.75rem;
line-height: 1.6;
color: var(--el-text-color-secondary, #909399);
}
.adv__hint code {
padding: 0 0.2rem;
border-radius: 2px;
background: var(--el-fill-color, #f0f2f5);
}
@media (max-width: 768px) {
.adv__grid {
grid-template-columns: 1fr;
}
}
</style>
+12 -3
View File
@@ -1,10 +1,11 @@
<script setup lang="ts">
import { ref, onMounted, computed, nextTick } from 'vue'
import { useRoute } from 'vue-router'
import { config } from '../../config/index';
import { config, devNodes } from '../../config/index';
import { useMiddlewareFetch } from '../../utils/middleware';
import { isIPv6 } from 'is-ip';
import { formatTime } from '../../utils/tools';
import { formatTime, withQuery } from '../../utils/tools';
import AdvancedOptions from '../components/AdvancedOptions.vue';
const route = useRoute()
useHead({
@@ -74,12 +75,17 @@ const recordTypes = [
{ value: 'ptr', label: 'PTR 记录' }
]
// 高级选项:DNS 拨测问的就是真实解析结果,只取 DNS 服务器(dns4/dns6),不支持强制解析
const adv = ref<InstanceType<typeof AdvancedOptions> | null>(null)
// 调试节点放最前(仅 dev 注入,生产为空数组)
const dnsServerFetches = [
...devNodes,
...config.APIBaseURL.DualStack,
...config.APIBaseURL.IPv4,
...config.APIBaseURL.IPv6
].map((server) => {
const url = computed(() => "/middleware/" + server.id + "/dns/" + recordType.value + "/" + encodeURIComponent(domain.value));
const url = computed(() => withQuery("/middleware/" + server.id + "/dns/" + recordType.value + "/" + encodeURIComponent(domain.value), adv.value?.query() ?? {}));
const { data, error, execute } = useMiddlewareFetch(url, {
immediate: false,
watch: false,
@@ -219,6 +225,9 @@ const doc = page.value;
查询
</el-button>
</div>
<AdvancedOptions ref="adv" :show-resolve="false" />
<div class="result-section">
<table class="result-table" v-if="results.length > 0">
<thead>
+3 -1
View File
@@ -2,7 +2,7 @@
import { useMiddlewareFetch } from '../../utils/middleware';
import { ref, onMounted, computed, nextTick } from 'vue'
import { useRoute } from 'vue-router'
import { config } from '../../config/index'
import { config, devNodes } from '../../config/index'
import { isIPv6 } from 'is-ip'
@@ -71,7 +71,9 @@ const userIP = ref('')
// 查询序号:连续点击时旧查询的结果不允许写入新查询的结果表
let queryId = 0
// 调试节点放最前(仅 dev 注入,生产为空数组)
const apiList = [
...devNodes,
...config.APIBaseURL.DualStack,
...config.APIBaseURL.IPv4,
...config.APIBaseURL.IPv6
+8 -3
View File
@@ -3,7 +3,8 @@ import { ref, onMounted, computed, nextTick } from 'vue'
import { useRoute } from 'vue-router'
import { config } from '../../config/index';
import { useMiddlewareFetch } from '../../utils/middleware';
import { extractHost, formatTime, formatSpeed, formatSize, getStatusCodeClass } from '../../utils/tools';
import { extractHost, formatTime, formatSpeed, formatSize, getStatusCodeClass, withQuery } from '../../utils/tools';
import AdvancedOptions from '../components/AdvancedOptions.vue';
const route = useRoute()
@@ -48,6 +49,8 @@ const loading = ref(false)
const result = ref<any[]>([])
const error = ref('')
const domain = ref('')
// 高级选项(dns6 / resolve / resolve6):用户填了才拼进请求
const adv = ref<InstanceType<typeof AdvancedOptions> | null>(null)
/** 服务器结果接口 */
interface ServerResult {
@@ -72,7 +75,7 @@ const serverResults = ref<ServerResult[]>([])
/** 双栈服务器:同时支持 IPv4/IPv6 的节点 */
const dualStackFetches = config.APIBaseURL.DualStack.map((server) => {
const url = computed(() => "/middleware/" + server.id + "/speed/v6/" + encodeURIComponent(domain.value));
const url = computed(() => withQuery("/middleware/" + server.id + "/speed/v6/" + encodeURIComponent(domain.value), adv.value?.query() ?? {}));
const { data, error: fetchError, execute } = useMiddlewareFetch(url, {
immediate: false,
watch: false,
@@ -82,7 +85,7 @@ const dualStackFetches = config.APIBaseURL.DualStack.map((server) => {
/** IPv6 专用服务器:仅支持 IPv6 的节点 */
const ipv6Fetches = config.APIBaseURL.IPv6.map((server) => {
const url = computed(() => "/middleware/" + server.id + "/speed/v6/" + encodeURIComponent(domain.value));
const url = computed(() => withQuery("/middleware/" + server.id + "/speed/v6/" + encodeURIComponent(domain.value), adv.value?.query() ?? {}));
const { data, error: fetchError, execute } = useMiddlewareFetch(url, {
immediate: false,
watch: false,
@@ -191,6 +194,8 @@ onMounted(() => {
</el-button>
</div>
<AdvancedOptions ref="adv" stack="ipv6" />
<div v-if="error" class="error-message">
{{ error }}
</div>
+7 -2
View File
@@ -4,7 +4,8 @@ import { useRoute } from 'vue-router'
import { config } from '../../config/index';
import { useMiddlewareFetch } from '../../utils/middleware';
import { isIPv6 } from 'is-ip';
import { extractHost } from '../../utils/tools';
import { extractHost, withQuery } from '../../utils/tools';
import AdvancedOptions from '../components/AdvancedOptions.vue';
const route = useRoute()
useHead({
@@ -77,6 +78,8 @@ const port = ref('80')
const userIP = ref('')
const loading = ref(false)
const serverResults = ref<ServerResult[]>([])
// 高级选项(dns6 / resolve / resolve6):用户填了才拼进请求
const adv = ref<InstanceType<typeof AdvancedOptions> | null>(null)
// 查询序号:连续点击时旧查询的结果不允许写入新查询的结果表
let queryId = 0
@@ -86,7 +89,7 @@ const allServers = [
];
const tcpingFetches = allServers.map((server) => {
const url = computed(() => "/middleware/" + server.id + "/tcping/" + encodeURIComponent(extractHost(tmpDomain.value)) + "?port=" + encodeURIComponent(port.value));
const url = computed(() => withQuery("/middleware/" + server.id + "/tcping/" + encodeURIComponent(extractHost(tmpDomain.value)) + "?port=" + encodeURIComponent(port.value) + "&version=ipv6", adv.value?.query() ?? {}));
const { data, error: fetchError, execute } = useMiddlewareFetch<TCPingResponse>(url, {
immediate: false,
watch: false,
@@ -211,6 +214,8 @@ onMounted(() => {
</el-button>
</div>
<AdvancedOptions ref="adv" stack="ipv6" />
<div class="result-section">
<table class="result-table">
<thead>
+13 -4
View File
@@ -1,10 +1,11 @@
<script setup lang="ts">
import { ref, onMounted, computed, nextTick, watch } from 'vue'
import { useRoute } from 'vue-router'
import { config } from '../../config/index';
import { config, devNodes } from '../../config/index';
import { useMiddlewareFetch } from '../../utils/middleware';
import { CircleCheckFilled, CircleCloseFilled,InfoFilled,Position } from '@element-plus/icons-vue';
import { extractHost, getStatusCodeClass, formatTime, formatSize, formatSpeed } from '../../utils/tools';
import { extractHost, getStatusCodeClass, formatTime, formatSize, formatSpeed, withQuery } from '../../utils/tools';
import AdvancedOptions from '../components/AdvancedOptions.vue';
const route = useRoute()
@@ -63,7 +64,11 @@ interface PerformanceCheckItem {
download_speed: number
is_reachable: boolean
}
const apiList = config.APIBaseURL.DualStack
// 调试节点放最前(仅 dev 注入,生产为空数组)
const apiList = [
...devNodes,
...config.APIBaseURL.DualStack,
]
const currentApiIndex = ref(0)
const backendID = computed(() => apiList[currentApiIndex.value]?.id || '')
const tmpDomain = ref('https://www.zakoflare.com')
@@ -71,7 +76,9 @@ const testDomain = ref('')
const loading = ref(false)
const error = ref('')
const result = ref<PerformanceCheckResponse | null>(null)
const checkUrl = computed(() => "/middleware/" + backendID.value + "/detail/" + testDomain.value);
// 高级选项(dns4/dns6 + resolve/resolve4/resolve6):本页一次探测两栈
const adv = ref<InstanceType<typeof AdvancedOptions> | null>(null)
const checkUrl = computed(() => withQuery("/middleware/" + backendID.value + "/detail/" + testDomain.value, adv.value?.query() ?? {}));
// testDomain 是完整 URL(含 https:// 与路径斜杠),服务端 [...slug].get.ts 依赖
// 分段重组协议部分,这里刻意不做 encodeURIComponent
@@ -147,6 +154,8 @@ onMounted(() => {
</el-button>
</div>
<AdvancedOptions ref="adv" />
<div v-if="error" class="error-message">
{{ error }}
</div>
+20 -4
View File
@@ -1,9 +1,10 @@
<script setup lang="ts">
import { ref, onMounted, computed, nextTick } from 'vue'
import { useRoute } from 'vue-router'
import { config } from '../../config/index';
import { config, devNodes } from '../../config/index';
import { useMiddlewareFetch } from '../../utils/middleware';
import { extractHost, formatTime, formatSpeed, formatSize, getStatusCodeClass } from '../../utils/tools';
import { extractHost, formatTime, formatSpeed, formatSize, getStatusCodeClass, withQuery } from '../../utils/tools';
import AdvancedOptions from '../components/AdvancedOptions.vue';
const route = useRoute()
@@ -48,6 +49,8 @@ const loading = ref(false)
const result = ref<any[]>([])
const error = ref('')
const domain = ref('')
// 高级选项(dns4 / resolve / resolve4):用户填了才拼进请求
const adv = ref<InstanceType<typeof AdvancedOptions> | null>(null)
/** 服务器结果接口 */
interface ServerResult {
@@ -70,9 +73,19 @@ const serverResults = ref<ServerResult[]>([])
// 避免页面加载时自动发请求浪费资源。
// =============================================
/** 调试节点:仅 dev 注入,排最前(本页固定走 IPv4) */
const devFetches = devNodes.map((server) => {
const url = computed(() => withQuery("/middleware/" + server.id + "/speed/v4/" + encodeURIComponent(domain.value), adv.value?.query() ?? {}));
const { data, error: fetchError, execute } = useMiddlewareFetch(url, {
immediate: false,
watch: false,
});
return { label: server.label, data, error: fetchError, execute };
});
/** 双栈服务器:同时支持 IPv4/IPv6 的节点 */
const dualStackFetches = config.APIBaseURL.DualStack.map((server) => {
const url = computed(() => "/middleware/" + server.id + "/speed/v4/" + encodeURIComponent(domain.value));
const url = computed(() => withQuery("/middleware/" + server.id + "/speed/v4/" + encodeURIComponent(domain.value), adv.value?.query() ?? {}));
const { data, error: fetchError, execute } = useMiddlewareFetch(url, {
immediate: false,
watch: false,
@@ -82,7 +95,7 @@ const dualStackFetches = config.APIBaseURL.DualStack.map((server) => {
/** IPv4 专用服务器:仅支持 IPv4 的节点 */
const ipv4Fetches = config.APIBaseURL.IPv4.map((server) => {
const url = computed(() => "/middleware/" + server.id + "/speed/v4/" + encodeURIComponent(domain.value));
const url = computed(() => withQuery("/middleware/" + server.id + "/speed/v4/" + encodeURIComponent(domain.value), adv.value?.query() ?? {}));
const { data, error: fetchError, execute } = useMiddlewareFetch(url, {
immediate: false,
watch: false,
@@ -92,6 +105,7 @@ const ipv4Fetches = config.APIBaseURL.IPv4.map((server) => {
/** 合并所有服务器 fetch 列表 */
const allFetches = [
...devFetches.map(fetch => ({ ...fetch, type: 'IPv4' })),
...dualStackFetches.map(fetch => ({ ...fetch, type: 'DualStack' })),
...ipv4Fetches.map(fetch => ({ ...fetch, type: 'IPv4' }))
];
@@ -191,6 +205,8 @@ onMounted(() => {
</el-button>
</div>
<AdvancedOptions ref="adv" stack="ipv4" />
<div v-if="error" class="error-message">
{{ error }}
</div>
+13 -4
View File
@@ -2,9 +2,10 @@
import { ref, onMounted, computed, nextTick, watch } from 'vue'
import { useRoute } from 'vue-router'
import { CircleCheckFilled, CircleCloseFilled,InfoFilled } from '@element-plus/icons-vue';
import { config } from '../../config/index';
import { config, devNodes } from '../../config/index';
import { useMiddlewareFetch } from '../../utils/middleware';
import { extractHost, getStatusCodeClass, formatTime, formatSpeed } from '../../utils/tools';
import { extractHost, getStatusCodeClass, formatTime, formatSpeed, withQuery } from '../../utils/tools';
import AdvancedOptions from '../components/AdvancedOptions.vue';
const route = useRoute()
const siteUrlWithSlash = computed(() => config.siteUrl.replace(/\/$/, '') + '/')
@@ -65,7 +66,11 @@ interface SSLCheckItem {
is_expired: boolean
is_reachable: boolean
}
const apiList = config.APIBaseURL.DualStack
// 调试节点放最前(仅 dev 注入,生产为空数组)
const apiList = [
...devNodes,
...config.APIBaseURL.DualStack,
]
const currentApiIndex = ref(0)
const backendID = computed(() => apiList[currentApiIndex.value]?.id || '')
const tmpDomain = ref('https://www.zakoflare.com')
@@ -73,7 +78,9 @@ const testDomain = ref('')
const loading = ref(false)
const error = ref('')
const result = ref<SSLCheckResponse | null>(null)
const sslUrl = computed(() => "/middleware/" + backendID.value + "/ssl/" + encodeURIComponent(testDomain.value));
// 高级选项(dns4/dns6 + resolve/resolve4/resolve6):本页一次探测两栈
const adv = ref<InstanceType<typeof AdvancedOptions> | null>(null)
const sslUrl = computed(() => withQuery("/middleware/" + backendID.value + "/ssl/" + encodeURIComponent(testDomain.value), adv.value?.query() ?? {}));
const { data: sslData, error: sslError, execute: executeSSL } = useMiddlewareFetch<SSLCheckResponse>(sslUrl, {
immediate: false,
@@ -157,6 +164,8 @@ onMounted(() => {
</el-button>
</div>
<AdvancedOptions ref="adv" />
<div v-if="error" class="error-message">
{{ error }}
</div>
+20 -4
View File
@@ -1,9 +1,10 @@
<script setup lang="ts">
import { ref, onMounted, computed } from 'vue'
import { useRoute } from 'vue-router'
import { config } from '../../config/index';
import { config, devNodes } from '../../config/index';
import { useMiddlewareFetch } from '../../utils/middleware';
import { extractHost } from '../../utils/tools';
import { extractHost, withQuery } from '../../utils/tools';
import AdvancedOptions from '../components/AdvancedOptions.vue';
const route = useRoute()
@@ -76,16 +77,20 @@ const tmpDomain = ref('www.zakoflare.com')
const port = ref('80')
const loading = ref(false)
const serverResults = ref<ServerResult[]>([])
// 高级选项(dns4 / resolve / resolve4):用户填了才拼进请求
const adv = ref<InstanceType<typeof AdvancedOptions> | null>(null)
// 查询序号:连续点击时旧查询的结果不允许写入新查询的结果表
let queryId = 0
const allServers = [
// 调试节点放最前(仅 dev 注入,生产为空数组);本页固定 version=ipv4,故按 IPv4 组对待
...devNodes.map((s: ServerConfig) => ({ ...s, type: 'IPv4' })),
...config.APIBaseURL.DualStack.map((s: ServerConfig) => ({ ...s, type: 'DualStack' })),
...config.APIBaseURL.IPv4.map((s: ServerConfig) => ({ ...s, type: 'IPv4' }))
];
const tcpingFetches = allServers.map((server) => {
const url = computed(() => "/middleware/" + server.id + "/tcping/" + encodeURIComponent(extractHost(tmpDomain.value)) + "?port=" + encodeURIComponent(port.value));
const url = computed(() => withQuery("/middleware/" + server.id + "/tcping/" + encodeURIComponent(extractHost(tmpDomain.value)) + "?port=" + encodeURIComponent(port.value) + "&version=ipv4", adv.value?.query() ?? {}));
const { data, error: fetchError, execute } = useMiddlewareFetch<TCPingResponse>(url, {
immediate: false,
watch: false,
@@ -95,7 +100,16 @@ const tcpingFetches = allServers.map((server) => {
function initServerResults() {
const results: ServerResult[] = []
// 与 allServers 顺序保持一致:调试节点 → DualStack → IPv4
devNodes.forEach((server) => {
results.push({
label: server.label,
loading: false,
error: ''
})
})
config.APIBaseURL.DualStack.forEach((server: ServerConfig) => {
results.push({
label: server.label,
@@ -202,6 +216,8 @@ onMounted(() => {
</el-button>
</div>
<AdvancedOptions ref="adv" stack="ipv4" />
<div class="result-section">
<table class="result-table">
<thead>
+13 -3
View File
@@ -2,7 +2,9 @@
import { useMiddlewareFetch } from '../../utils/middleware';
import { ref, onMounted, computed, watch, nextTick } from 'vue'
import { useRoute } from 'vue-router'
import { config } from '../../config/index'
import { config, devNodes } from '../../config/index'
import { withQuery } from '../../utils/tools'
import AdvancedOptions from '../components/AdvancedOptions.vue'
import { isIPv6 } from 'is-ip'
const route = useRoute()
@@ -64,10 +66,16 @@ const result = ref<any>(null)
const error = ref('')
const userIP = ref('')
const apiList = config.APIBaseURL.DualStack
// 调试节点放最前(仅 dev 注入,生产为空数组)
const apiList = [
...devNodes,
...config.APIBaseURL.DualStack,
]
const currentApiIndex = ref(0)
const backendID = computed(() => apiList[currentApiIndex.value]?.id || '')
const whoisUrl = computed(() => '/middleware/' + backendID.value + '/whois/' + encodeURIComponent(domain.value))
// 高级选项:whois 只取 DNS 服务器(dns4/dns6),不支持强制解析
const adv = ref<InstanceType<typeof AdvancedOptions> | null>(null)
const whoisUrl = computed(() => withQuery('/middleware/' + backendID.value + '/whois/' + encodeURIComponent(domain.value), adv.value?.query() ?? {}))
const { data: whoisData, error: whoisError, execute: executeWhois } = useMiddlewareFetch<any>(whoisUrl, {
immediate: false,
@@ -178,6 +186,8 @@ function getStatusClass(status: string): string {
</el-button>
</div>
<AdvancedOptions ref="adv" :show-resolve="false" />
<div v-if="error" class="error-message">
{{ error }}
</div>
+20 -1
View File
@@ -2,6 +2,23 @@
/*
前端的一系列配置
*/
// 是否开发环境。
// 本文件同时被 nuxt.config 与 nitro server 引用,那两处 import.meta.dev 尚未注入(undefined),
// 因此只有严格为 true 才算 dev;应用代码里由 Vite 的 define 替换成真实布尔值。
const __DEV__ = (import.meta as any).dev === true
/**
* 本机调试节点:仅开发环境注入,对应 middleware-go/setting.json 里的 local
* (url 指向本机节点,默认 127.0.0.1:18081)。生产构建为空数组。
*
* 位置约定:调试节点放**第一优先级**——各页把 `...devNodes` 拼在节点列表最前面;
* 本机只有 IPv4,所以 IPv6 页(ipv6tcping / ipv6speedtest)不拼它。
*/
export const devNodes: { label: string; id: string; url: string }[] = __DEV__
? [{ label: "本机 127.0.0.1:18081(调试)", id: "local", url: "" }]
: []
const config = {
siteUrl: "https://nipw.cn/",
// 站点名称:用于页面标题 / 描述 / 页脚品牌展示
@@ -18,8 +35,10 @@ const config = {
// 前端自带中间件始终放在候选列表最后一位兜底。
// 未配置或为空数组时,直接使用前端自带中间件。
Middleware: <string[]>[
// dev:本机 middleware-go(8091)排最前 —— 调试节点 id "local" 只有它认识,
// 远端中间层拿到会直接返回 Invalid backend ID(400,不在重试码里)。
...(__DEV__ ? ["http://127.0.0.1:8091/"] : []),
"https://middleware-1.api-ipw.wsmdn.top/"
// "http://127.0.0.1:8092/",
// "https://middleware-2.wsmdn.top/",
],
// Umami 统计
@@ -133,7 +133,11 @@ export default defineEventHandler(async (event) => {
if (apiBaseUrl.slice(-1) != '/') {
apiBaseUrl = `${apiBaseUrl}/`
}
data = await $fetch(`${apiBaseUrl}v1/${apiType}/${raw}`, {
const queryString = new URLSearchParams(
Object.entries(query).filter(([_, v]) => v !== undefined) as [string, string][]
).toString()
data = await $fetch(`${apiBaseUrl}v1/${apiType}/${raw}${queryString ? '?' + queryString : ''}`, {
method: 'GET',
headers: authHeaders,
timeout: 15_000,
+14
View File
@@ -40,3 +40,17 @@ export function isIPv4(ip: string): boolean {
const ipRegex = /^(\d{1,3}\.){3}\d{1,3}$/;
return ipRegex.test(ip);
}
/**
* 把参数对象拼到 URL 上(空值跳过,自动判断用 ? 还是 &)。
* 用于高级选项:dns4/dns6/resolve/resolve4/resolve6 这些都是「请求级解析参数」,
* 只有用户填了才下发,没填时 URL 与加参数之前逐字节一致(缓存键也不变)。
*/
export function withQuery(url: string, params: Record<string, string>): string {
const pairs: string[] = []
for (const [key, value] of Object.entries(params)) {
if (value) pairs.push(encodeURIComponent(key) + '=' + encodeURIComponent(value))
}
if (pairs.length === 0) return url
return url + (url.includes('?') ? '&' : '?') + pairs.join('&')
}
+14 -12
View File
@@ -348,6 +348,18 @@ func middlewareHandler(c fiber.Ctx) error {
query := c.Queries()
// 请求级 query 透传给节点(过滤空值,等价 TS 版 URLSearchParams 过滤):
// 两个分支都要用 —— tcping/udping/speed 的 port/count/version,以及 detail/ssl/dns/whois 的 dns/dns4/dns6。
// HTTP 侧用 url.Values 重新编码后拼接(原样透传会被下一跳的 query 解析器重新切分),WS 侧以 map 形式进 RequestProbe。
queryString := url.Values{}
queryMap := make(map[string]string)
for k, v := range query {
if v != "" {
queryString.Set(k, v)
queryMap[k] = v
}
}
// 从 API_KEYS 查找 token,未配置时回退到环境变量 API_KEYS (JSON 字符串)
apiKey := lookupAPIKey(backendID)
@@ -378,7 +390,7 @@ func middlewareHandler(c fiber.Ctx) error {
// 不再静默落回空 URL 的 HTTP 转发(必然 502 且掩盖真实配置错误)
return c.Status(fiber.StatusBadGateway).SendString("Node " + backendID + " is WS-only but WS channel is disabled (ws-port=0 or bind failed)")
}
status, body, err := wsSrv.RequestProbe(backendID, apiType, raw, nil, wsProbeTimeout())
status, body, err := wsSrv.RequestProbe(backendID, apiType, raw, queryMap, wsProbeTimeout())
if err != nil {
return c.Status(fiber.StatusBadGateway).SendString("WS probe failed: " + err.Error())
}
@@ -391,22 +403,12 @@ func middlewareHandler(c fiber.Ctx) error {
}
// 上游错误原样透传(状态码 + body),网络错误返回 502
return forwardUpstream(c, apiBaseUrl, apiBaseUrl+"v1/"+apiType+"/"+raw, authHeaders, nil)
return forwardUpstream(c, apiBaseUrl, apiBaseUrl+"v1/"+apiType+"/"+raw, authHeaders, queryString)
case "tcping", "udping", "speed":
// tcping/udping/speed 统一走 APIBaseURL 节点池
apiBaseUrls := flattenStack(API_BASE_URLS)
// 转发 query 参数(过滤空值,等价于 TS 中 URLSearchParams 过滤 undefined)
queryString := url.Values{}
queryMap := make(map[string]string)
for k, v := range query {
if v != "" {
queryString.Set(k, v)
queryMap[k] = v
}
}
node := findNode(apiBaseUrls, backendID)
if node == nil || (!node.UseWS() && node.URL == "") {
return badRequest(c, "Invalid backend ID")
+161 -124
View File
@@ -38,26 +38,39 @@ func initHTTPClients() {
dialer := &net.Dialer{Timeout: 10 * time.Second, KeepAlive: 30 * time.Second}
return &http.Transport{
DialContext: func(ctx context.Context, _, addr string) (net.Conn, error) {
if ssrf.Enabled() {
host, port, err := net.SplitHostPort(addr)
// DNS 接管与 SSRF 解耦:域名一律先经 webtest 解析成 IP 再拨号。
// 服务器取请求级覆盖(dns=/dns4=/dns6=,经 ctx 传来),无覆盖则用配置 dns-server;
// 不这样时关掉 SSRF 会绕过配置/系统 DNS,detail/ssl/speed 与 tcping/dns/whois 各问各的。
// SSRF 只保留「私网校验」职责,不再是 DNS 接管的开关。
// 同源:SSRF 预检(ValidateOutboundTarget)经 ssrfresolver 读到的是同一份 ctx 配置。
req := webtest.ReqResolveFromContext(ctx)
host, port, err := net.SplitHostPort(addr)
if err != nil {
return nil, err
}
// 强制解析(resolve=/resolve4=/resolve6=):命中就把目标填回 host。
// 目标是 IP → 走下面的字面 IP 分支(私网检测后直连,不再查 DNS);
// 目标是域名 → 当作新 host 交给下面的解析分支。
if target := req.Override(host, network == "tcp6"); target != "" {
host = strings.Trim(target, "[]")
addr = net.JoinHostPort(host, port)
}
if ip := net.ParseIP(host); ip != nil {
if ssrf.Enabled() && ssrf.IsPrivateIP(ip) {
slog.Warn("Blocked connection to private IP", "host", host)
return nil, fmt.Errorf("request to private/internal address is not allowed")
}
} else {
var dnsResult webtest.DNSResult
if network == "tcp4" {
dnsResult, err = webtest.ResolveARecord(host, req.Servers("ipv4"))
} else {
dnsResult, err = webtest.ResolveAAAARecord(host, req.Servers("ipv6"))
}
if err != nil {
return nil, err
}
if ip := net.ParseIP(host); ip != nil {
if ssrf.IsPrivateIP(ip) {
slog.Warn("Blocked connection to private IP", "host", host)
return nil, fmt.Errorf("request to private/internal address is not allowed")
}
} else {
var dnsResult webtest.DNSResult
if network == "tcp4" {
dnsResult, err = webtest.ResolveARecord(host)
} else {
dnsResult, err = webtest.ResolveAAAARecord(host)
}
if err != nil {
return nil, err
}
if ssrf.Enabled() {
for _, ipStr := range dnsResult.Record {
ip := net.ParseIP(ipStr)
if ip != nil && ssrf.IsPrivateIP(ip) {
@@ -65,9 +78,9 @@ func initHTTPClients() {
return nil, fmt.Errorf("request to private/internal address is not allowed")
}
}
if len(dnsResult.Record) > 0 {
addr = net.JoinHostPort(dnsResult.Record[0], port)
}
}
if len(dnsResult.Record) > 0 {
addr = net.JoinHostPort(dnsResult.Record[0], port)
}
}
return dialer.DialContext(ctx, network, addr)
@@ -390,19 +403,27 @@ func checkWebsiteHandler(c fiber.Ctx) error {
})
}
if cached, ok := websiteCache.Load(testUrl); ok {
req, err := webtest.ParseReqResolve(func(k string) string { return c.Query(k) })
if err != nil {
return c.Status(http.StatusBadRequest).JSON(fiber.Map{"error": err.Error()})
}
// 缓存键带上请求级 DNS 覆盖:结果取决于所用服务器,参数不同必须各占一条,
// 否则互相串味;无覆盖时 req.CacheKey() 为空串,键与旧版完全一致。
cacheKey := testUrl + req.CacheKey()
if cached, ok := websiteCache.Load(cacheKey); ok {
entry := cached.(websiteCacheEntry)
if time.Since(entry.timestamp) < cacheTTL(entry.failed) {
return c.Status(200).JSON(entry.result)
}
websiteCache.Delete(testUrl)
websiteCache.Delete(cacheKey)
}
rawResult, _, _ := sfGroup.Do(testUrl, func() (interface{}, error) {
rawResult, _, _ := sfGroup.Do(cacheKey, func() (interface{}, error) {
result := &WebsiteCheckResult{}
switch SINGLE_STACK {
case "ipv4":
ipv4, errV4 := webtest.CheckWebsite(testUrl, "v4")
ipv4, errV4 := webtest.CheckWebsite(testUrl, "v4", req)
if errV4 != nil {
ipv4 = &webtest.WebsiteCheckDetail{
HostRecord: "Error: " + errV4.Error(),
@@ -415,7 +436,7 @@ func checkWebsiteHandler(c fiber.Ctx) error {
IsReachable: false,
}
case "ipv6":
ipv6, errV6 := webtest.CheckWebsite(testUrl, "v6")
ipv6, errV6 := webtest.CheckWebsite(testUrl, "v6", req)
if errV6 != nil {
ipv6 = &webtest.WebsiteCheckDetail{
HostRecord: "Error: " + errV6.Error(),
@@ -433,7 +454,7 @@ func checkWebsiteHandler(c fiber.Ctx) error {
go func() {
defer wg.Done()
ipv6, errV6 := webtest.CheckWebsite(testUrl, "v6")
ipv6, errV6 := webtest.CheckWebsite(testUrl, "v6", req)
if errV6 != nil {
ipv6 = &webtest.WebsiteCheckDetail{
HostRecord: "Error: " + errV6.Error(),
@@ -445,7 +466,7 @@ func checkWebsiteHandler(c fiber.Ctx) error {
go func() {
defer wg.Done()
ipv4, errV4 := webtest.CheckWebsite(testUrl, "v4")
ipv4, errV4 := webtest.CheckWebsite(testUrl, "v4", req)
if errV4 != nil {
ipv4 = &webtest.WebsiteCheckDetail{
HostRecord: "Error: " + errV4.Error(),
@@ -461,7 +482,7 @@ func checkWebsiteHandler(c fiber.Ctx) error {
failed := dualAnyFailed(result.IPv4, result.IPv6, func(d *webtest.WebsiteCheckDetail) bool {
return d != nil && !d.IsReachable
})
websiteCache.Store(testUrl, websiteCacheEntry{result: result, timestamp: time.Now(), failed: failed})
websiteCache.Store(cacheKey, websiteCacheEntry{result: result, timestamp: time.Now(), failed: failed})
return result, nil
})
@@ -478,6 +499,11 @@ func websiteSpeedTestHandler(c fiber.Ctx) error {
}
url := normalizeURL(testUrl)
req, err := webtest.ParseReqResolve(func(k string) string { return c.Query(k) })
if err != nil {
return c.Status(http.StatusBadRequest).JSON(fiber.Map{"error": err.Error()})
}
// 检查请求版本是否与 SINGLE_STACK 配置匹配
switch SINGLE_STACK {
case "ipv4":
@@ -496,10 +522,9 @@ func websiteSpeedTestHandler(c fiber.Ctx) error {
}
}
// 缓存键:URL + 版本
cacheKey := fmt.Sprintf("%s:%s", url, version)
// 缓存键:URL + 版本 + 请求级 DNS 覆盖(结果取决于所用服务器,参数不同不能共用)
cacheKey := fmt.Sprintf("%s:%s%s", url, version, req.CacheKey())
// 检查缓存
if cached, ok := speedCache.Load(cacheKey); ok {
entry := cached.(speedCacheEntry)
if time.Since(entry.timestamp) < cacheTTL(entry.failed) {
@@ -513,7 +538,7 @@ func websiteSpeedTestHandler(c fiber.Ctx) error {
switch version {
case "v6", "v4":
rawResult, _, _ := sfGroup.Do(cacheKey, func() (interface{}, error) {
r, e := webtest.SpeedTest(url, version)
r, e := webtest.SpeedTest(url, version, req)
if e != nil {
errorResult := &webtest.WebsiteSpeedTestResult{
HostRecord: "Error: " + e.Error(),
@@ -552,19 +577,26 @@ func sslCheckHandler(c fiber.Ctx) error {
})
}
if cached, ok := sslCache.Load(testUrl); ok {
req, err := webtest.ParseReqResolve(func(k string) string { return c.Query(k) })
if err != nil {
return c.Status(http.StatusBadRequest).JSON(fiber.Map{"error": err.Error()})
}
// 缓存键带上请求级 DNS 覆盖(结果取决于所用服务器,参数不同不能共用)
cacheKey := testUrl + req.CacheKey()
if cached, ok := sslCache.Load(cacheKey); ok {
entry := cached.(sslCacheEntry)
if time.Since(entry.timestamp) < cacheTTL(entry.failed) {
return c.Status(200).JSON(entry.result)
}
sslCache.Delete(testUrl)
sslCache.Delete(cacheKey)
}
rawResult, _, _ := sfGroup.Do(testUrl, func() (interface{}, error) {
rawResult, _, _ := sfGroup.Do(cacheKey, func() (interface{}, error) {
result := &SSLCheckResult{}
switch SINGLE_STACK {
case "ipv4":
ipv4, errV4 := webtest.CheckSSL(testUrl, "v4")
ipv4, errV4 := webtest.CheckSSL(testUrl, "v4", req)
if errV4 != nil {
ipv4 = &webtest.SSLCheckDetail{
HostRecord: "Error: " + errV4.Error(),
@@ -579,7 +611,7 @@ func sslCheckHandler(c fiber.Ctx) error {
IsReachable: false,
}
case "ipv6":
ipv6, errV6 := webtest.CheckSSL(testUrl, "v6")
ipv6, errV6 := webtest.CheckSSL(testUrl, "v6", req)
if errV6 != nil {
ipv6 = &webtest.SSLCheckDetail{
HostRecord: "Error: " + errV6.Error(),
@@ -599,7 +631,7 @@ func sslCheckHandler(c fiber.Ctx) error {
go func() {
defer wg.Done()
ipv6, errV6 := webtest.CheckSSL(testUrl, "v6")
ipv6, errV6 := webtest.CheckSSL(testUrl, "v6", req)
if errV6 != nil {
ipv6 = &webtest.SSLCheckDetail{
HostRecord: "Error: " + errV6.Error(),
@@ -612,7 +644,7 @@ func sslCheckHandler(c fiber.Ctx) error {
go func() {
defer wg.Done()
ipv4, errV4 := webtest.CheckSSL(testUrl, "v4")
ipv4, errV4 := webtest.CheckSSL(testUrl, "v4", req)
if errV4 != nil {
ipv4 = &webtest.SSLCheckDetail{
HostRecord: "Error: " + errV4.Error(),
@@ -629,7 +661,7 @@ func sslCheckHandler(c fiber.Ctx) error {
failed := dualAnyFailed(result.IPv4, result.IPv6, func(d *webtest.SSLCheckDetail) bool {
return d != nil && !d.IsReachable
})
sslCache.Store(testUrl, sslCacheEntry{result: result, timestamp: time.Now(), failed: failed})
sslCache.Store(cacheKey, sslCacheEntry{result: result, timestamp: time.Now(), failed: failed})
return result, nil
})
@@ -786,22 +818,29 @@ func whoisHandler(c fiber.Ctx) error {
})
}
if cached, ok := whoisCache.Load(domain); ok {
req, err := webtest.ParseReqResolve(func(k string) string { return c.Query(k) })
if err != nil {
return c.Status(http.StatusBadRequest).JSON(fiber.Map{"error": err.Error()})
}
// 缓存键带上请求级 DNS 覆盖(结果取决于所用服务器,参数不同不能共用)
cacheKey := domain + req.CacheKey()
if cached, ok := whoisCache.Load(cacheKey); ok {
entry := cached.(whoisCacheEntry)
if time.Since(entry.timestamp) < cacheTTL(entry.failed) {
return c.Status(200).JSON(entry.result)
}
whoisCache.Delete(domain)
whoisCache.Delete(cacheKey)
}
result, err := webtest.QueryWhois(domain)
result, err := webtest.QueryWhois(domain, req.Servers("ipv4"), req.Servers("ipv6"))
if err != nil {
return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
"error": err.Error(),
})
}
whoisCache.Store(domain, whoisCacheEntry{result: result, timestamp: time.Now()})
whoisCache.Store(cacheKey, whoisCacheEntry{result: result, timestamp: time.Now()})
return c.Status(http.StatusOK).JSON(result)
}
@@ -835,84 +874,62 @@ func dnsQueryHandler(c fiber.Ctx) error {
domain = parsedURL.Host
recodeType := c.Params("type")
switch recodeType {
case "a":
result, err := webtest.ResolveARecord(domain)
if err != nil {
return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
"error": err.Error(),
})
}
return c.Status(http.StatusOK).JSON(result)
case "aaaa":
result, err := webtest.ResolveAAAARecord(domain)
if err != nil {
return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
"error": err.Error(),
})
}
return c.Status(http.StatusOK).JSON(result)
case "cname":
result, err := webtest.ResolveCNAMERecord(domain)
if err != nil {
return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
"error": err.Error(),
})
}
return c.Status(http.StatusOK).JSON(result)
case "mx":
result, err := webtest.ResolveMXRecord(domain)
if err != nil {
return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
"error": err.Error(),
})
}
return c.Status(http.StatusOK).JSON(result)
case "ns":
result, err := webtest.ResolveNSRecord(domain)
if err != nil {
return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
"error": err.Error(),
})
}
return c.Status(http.StatusOK).JSON(result)
case "ptr":
result, err := webtest.ResolvePTRRecord(domain)
if err != nil {
return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
"error": err.Error(),
})
}
return c.Status(http.StatusOK).JSON(result)
case "srv":
result, err := webtest.ResolveSRVRecord(domain)
if err != nil {
return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
"error": err.Error(),
})
}
return c.Status(http.StatusOK).JSON(result)
case "txt":
result, err := webtest.ResolveTXTRecord(domain)
if err != nil {
return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
"error": err.Error(),
})
}
return c.Status(http.StatusOK).JSON(result)
case "caa":
result, err := webtest.ResolveCAARecord(domain)
if err != nil {
return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
"error": err.Error(),
})
}
return c.Status(http.StatusOK).JSON(result)
case "a", "aaaa", "cname", "mx", "ns", "ptr", "srv", "txt", "caa":
default:
return c.Status(http.StatusBadRequest).JSON(fiber.Map{
"error": "Invalid record type",
})
}
req, err := webtest.ParseReqResolve(func(k string) string { return c.Query(k) })
if err != nil {
return c.Status(http.StatusBadRequest).JSON(fiber.Map{"error": err.Error()})
}
// 按 qtype 选栈:a→dns4|dns,aaaa→dns6|dns,其余→dns
stack := ""
if recodeType == "a" {
stack = "ipv4"
} else if recodeType == "aaaa" {
stack = "ipv6"
}
result, err := webtest.ResolveByType(domain, recodeType, req.Servers(stack))
if err != nil {
return c.Status(http.StatusInternalServerError).JSON(fiber.Map{
"error": err.Error(),
})
}
return c.Status(http.StatusOK).JSON(result)
}
// normalizeStack 把请求里的栈标识归一化为 "ipv4" / "ipv6"。
// 同时接受 v4/v6 与 ipv4/ipv6 两种写法;无法识别时返回空串。
func normalizeStack(version string) string {
switch strings.ToLower(strings.TrimSpace(version)) {
case "v4", "ipv4":
return "ipv4"
case "v6", "ipv6":
return "ipv6"
}
return ""
}
// probeStackFor 计算一次拨测实际生效的单栈,并给出跳过文案的来源标签。
//
// stack == "" 双栈并行
// stack == "ipv4"/"ipv6" 只跑该栈,另一栈标记跳过
//
// 规则:SINGLE_STACK(节点配置)优先于请求参数 version —— 节点本身是单栈机时,
// 指定另一栈也跑不起来,统一按配置处理;只有配置为空(双栈机)时 version 才生效,
// 此时跳过文案记为 version=xxx,与配置造成的跳过区分开。
func probeStackFor(version string) (stack, skipLabel string) {
want := normalizeStack(version)
if want == "" || SINGLE_STACK != "" {
return SINGLE_STACK, "SINGLE_STACK"
}
return want, "version"
}
func pingHandler(c fiber.Ctx) error {
host := c.Params("ip")
port := c.Query("port")
@@ -944,7 +961,24 @@ func pingHandler(c fiber.Ctx) error {
count = n
}
cacheKey := fmt.Sprintf("%s:%s:%d", host, port, count)
// version=ipv4/ipv6:请求级指定单栈,行为等价于临时把 SINGLE_STACK 设成该值。
// 节点本身是单栈机(SINGLE_STACK 非空)时配置优先,请求不越权到另一栈。
version := c.Query("version")
if version != "" && normalizeStack(version) == "" {
return c.Status(http.StatusBadRequest).JSON(fiber.Map{
"error": "version must be ipv4 or ipv6",
})
}
stack, skipLabel := probeStackFor(version)
req, err := webtest.ParseReqResolve(func(k string) string { return c.Query(k) })
if err != nil {
return c.Status(http.StatusBadRequest).JSON(fiber.Map{"error": err.Error()})
}
// 缓存键带上生效栈与请求级 DNS 覆盖:version=ipv6 与默认双栈、
// 以及不同 dns= 的结果都不能互相污染。
cacheKey := fmt.Sprintf("%s:%s:%d:%s%s", host, port, count, stack, req.CacheKey())
if cached, ok := pingCache.Load(cacheKey); ok {
entry := cached.(pingCacheEntry)
if time.Since(entry.timestamp) < cacheTTL(entry.failed) {
@@ -956,9 +990,9 @@ func pingHandler(c fiber.Ctx) error {
rawResult, _, _ := sfGroup.Do(cacheKey, func() (interface{}, error) {
result := &TCPingResult{}
switch SINGLE_STACK {
switch stack {
case "ipv4":
ipv4, errV4 := webtest.TCPingRun(host, port, count, "v4", 10*time.Second, 100*time.Millisecond)
ipv4, errV4 := webtest.TCPingRunWith(host, port, count, "v4", 10*time.Second, 100*time.Millisecond, req)
if errV4 != nil {
ipv4 = &webtest.TCPingStats{
IP: "Error: " + errV4.Error(),
@@ -966,10 +1000,10 @@ func pingHandler(c fiber.Ctx) error {
}
result.IPv4 = ipv4
result.IPv6 = &webtest.TCPingStats{
IP: "Skipped due to SINGLE_STACK=ipv4",
IP: "Skipped due to " + skipLabel + "=ipv4",
}
case "ipv6":
ipv6, errV6 := webtest.TCPingRun(host, port, count, "v6", 10*time.Second, 100*time.Millisecond)
ipv6, errV6 := webtest.TCPingRunWith(host, port, count, "v6", 10*time.Second, 100*time.Millisecond, req)
if errV6 != nil {
ipv6 = &webtest.TCPingStats{
IP: "Error: " + errV6.Error(),
@@ -977,7 +1011,7 @@ func pingHandler(c fiber.Ctx) error {
}
result.IPv6 = ipv6
result.IPv4 = &webtest.TCPingStats{
IP: "Skipped due to SINGLE_STACK=ipv6",
IP: "Skipped due to " + skipLabel + "=ipv6",
}
default:
var wg sync.WaitGroup
@@ -985,7 +1019,7 @@ func pingHandler(c fiber.Ctx) error {
go func() {
defer wg.Done()
ipv6, errV6 := webtest.TCPingRun(host, port, count, "v6", 10*time.Second, 100*time.Millisecond)
ipv6, errV6 := webtest.TCPingRunWith(host, port, count, "v6", 10*time.Second, 100*time.Millisecond, req)
if errV6 != nil {
ipv6 = &webtest.TCPingStats{
IP: "Error: " + errV6.Error(),
@@ -996,7 +1030,7 @@ func pingHandler(c fiber.Ctx) error {
go func() {
defer wg.Done()
ipv4, errV4 := webtest.TCPingRun(host, port, count, "v4", 10*time.Second, 100*time.Millisecond)
ipv4, errV4 := webtest.TCPingRunWith(host, port, count, "v4", 10*time.Second, 100*time.Millisecond, req)
if errV4 != nil {
ipv4 = &webtest.TCPingStats{
IP: "Error: " + errV4.Error(),
@@ -1696,6 +1730,9 @@ func main() {
readConfig()
webtest.SetDNSServer(DNS_SERVER)
webtest.SetDNSSecServer(DNSSEC_DNS_SERVER)
// 让 ssrf 出站校验走 webtest 的自定义 DNS(含请求级 dns=/dns4=/dns6= 覆盖),
// 避免"预检用系统解析器、拨号用自定义 DNS"两套解析器。
webtest.InstallSSRFResolver()
initHTTPClients()
// 注入出站 HTTP 客户端到 webtest(探针函数内部按版本取用)
webtest.SetHTTPClient(V4Client, V6Client)
+20 -3
View File
@@ -37,6 +37,23 @@ func Enabled() bool {
return blockPrivateIPs
}
// ResolveFunc 把主机名解析为 IP 列表。默认 net.LookupIP(系统解析器)。
// 可由上层(webtest)通过 SetResolveFunc 注入,使出站校验与真正的拨号走同一个 DNS
// ——包括请求级 DNS 服务器覆盖。否则预检(系统解析器)与拨号(自定义 DNS)会用两套解析器,
// 既多一次查询,结果也可能不一致。
type ResolveFunc func(ctx context.Context, host string) ([]net.IP, error)
var resolveHostFn ResolveFunc = func(_ context.Context, host string) ([]net.IP, error) {
return net.LookupIP(host)
}
// SetResolveFunc 注入自定义解析器(传 nil 忽略,保持默认 net.LookupIP)。
func SetResolveFunc(f ResolveFunc) {
if f != nil {
resolveHostFn = f
}
}
// SetEnabled enables or disables SSRF protection globally.
// SetEnabled 全局启用或禁用 SSRF 防护。
// Typically called during configuration loading (readConfig).
@@ -99,7 +116,7 @@ func IsPrivateIP(ip net.IP) bool {
// Used in handlers to short-circuit requests to internal hosts before making outbound connections.
// 用于在发起出站连接前,在处理器中短路对内部主机的请求。
func HasLocalOrPrivateIP(host string) bool {
ips, err := net.LookupIP(host)
ips, err := resolveHostFn(context.Background(), host)
if err != nil {
return false
}
@@ -142,7 +159,7 @@ func ValidateOutboundTarget(ctx context.Context, targetURL string) (context.Cont
if parsed.Scheme != "http" && parsed.Scheme != "https" {
return ctx, fmt.Errorf("invalid scheme: %s", parsed.Scheme)
}
ips, err := net.LookupIP(host)
ips, err := resolveHostFn(ctx, host)
if err != nil {
return ctx, err
}
@@ -179,7 +196,7 @@ func SecureCheckRedirect(req *http.Request, via []*http.Request) error {
if len(via) >= 10 {
return fmt.Errorf("stopped after 10 redirects")
}
ips, err := net.LookupIP(host)
ips, err := resolveHostFn(req.Context(), host)
if err != nil {
return err
}
+294 -32
View File
@@ -2,12 +2,14 @@ package webtest
import (
"bytes"
"context"
"fmt"
"io"
"log/slog"
"net"
"net/http"
"net/url"
"sort"
"strconv"
"strings"
"sync"
@@ -102,8 +104,229 @@ const (
defaultDoHEndpoint = "https://doh.pub/dns-query"
)
// queryDNS 普通 DNS 查询:主从 failover——按配置顺序依次尝试每个服务器,
// 当前服务器查询失败(网络错误/超时)自动切换下一个;全部失败返回最后一个错误。
// NormalizeServers 解析「逗号分隔的 DNS 服务器」字符串并做格式校验,规则与配置侧一致:
// 每项为 "ip:port"(UDP/TCP)或 http(s):// DoH URL;非法项丢弃(记 Warn)。全非法时返回空。
// 供请求级参数(dns / dns4 / dns6)复用,避免两套校验漂移。
func NormalizeServers(s string) []string {
return validServers(splitServers(s))
}
// ---- 请求级解析配置:dns*/resolve* ----
//
// dns=1.1.1.1:53 两栈都用这个服务器(换服务器去问)
// dns4=1.1.1.1:53,dns6=[2400::1]:53 按栈覆盖(优先于 dns=)
// resolve=example.com:1.2.3.4 两栈都把 example.com 解析成 1.2.3.4(直接改答案)
// resolve4=example.com:1.2.3.4 只 IPv4 侧生效(优先于 resolve=)
// resolve6=example.com:[::1] 只 IPv6 侧生效(优先于 resolve=)
// resolve=www.example.com:origin.net 目标是域名时把它当作新的 host 再解析
//
// 只在「域名 → IP」这一步生效:detail/ssl/speed(经 DialContext)、tcping、SSRF 预检;
// whois 与 dns 拨测不接 resolve*(前者只做归属查询,后者问的就是真实解析)。
//
// 为什么走 context:HTTP 拨测的解析发生在 net/http 的 DialContext 里,那里只有 ctx、
// 没有直接入参;存进 ctx 后 DialContext 与 SSRF 预检便能同源读到同一份配置。
// ReqResolve 一次拨测的请求级解析配置。零值 = 没有任何请求级覆盖。
type ReqResolve struct {
dns []string // dns=
dns4 []string // dns4=
dns6 []string // dns6=
resolve map[string]string // resolve=:域名 → IP 或另一域名
resolve4 map[string]string // resolve4=
resolve6 map[string]string // resolve6=
}
// ParseReqResolve 从 query 解析全部 9 个参数(dns/dns4/dns6、resolve/resolve4/resolve6)。
// 参数存在但**全部项非法**时报错(handler 返 400),避免「看起来指定了、实际被静默忽略」;
// 缺省/为空则视为不覆盖。
func ParseReqResolve(get func(string) string) (ReqResolve, error) {
var out ReqResolve
// DNS 服务器:逗号分隔的 ip:port 或 DoH URL
for _, item := range []struct {
key string
dst *[]string
}{
{"dns", &out.dns}, {"dns4", &out.dns4}, {"dns6", &out.dns6},
} {
if raw := strings.TrimSpace(get(item.key)); raw != "" {
if list := NormalizeServers(raw); len(list) > 0 {
*item.dst = list
} else {
return ReqResolve{}, fmt.Errorf("%s must be a comma-separated list of ip:port or DoH URLs", item.key)
}
}
}
// 强制解析:逗号分隔的 域名:IP 或 域名:域名(后者按新 host 再解析)
for _, item := range []struct {
key string
dst *map[string]string
}{
{"resolve", &out.resolve}, {"resolve4", &out.resolve4}, {"resolve6", &out.resolve6},
} {
if raw := strings.TrimSpace(get(item.key)); raw != "" {
if m := parseResolveMap(raw); len(m) > 0 {
*item.dst = m
} else {
return ReqResolve{}, fmt.Errorf("%s must be a comma-separated list of host:ip or host:cname", item.key)
}
}
}
return out, nil
}
// parseResolveMap 把「域名:目标,域名:目标」切成映射表,非法项丢弃。
// 每组按**第一个冒号**切:域名不含冒号,IPv6 目标可带或不带方括号。
func parseResolveMap(s string) map[string]string {
out := map[string]string{}
for _, item := range strings.Split(s, ",") {
item = strings.TrimSpace(item)
idx := strings.Index(item, ":")
if idx <= 0 || idx == len(item)-1 {
continue
}
host, target := canonicalHost(item[:idx]), strings.TrimSpace(item[idx+1:])
if host == "" || strings.ContainsAny(host, " /\\") || !validResolveTarget(target) {
continue
}
out[host] = target
}
return out
}
// canonicalHost 归一化主机名:去首尾空白、转小写、去结尾的点,便于与查询域名匹配。
func canonicalHost(host string) string {
return strings.ToLower(strings.TrimSuffix(strings.TrimSpace(host), "."))
}
// validResolveTarget 目标必须是 IP 或形如域名的字符串(含点、无空格、无端口)。
func validResolveTarget(target string) bool {
t := strings.Trim(target, "[]")
if net.ParseIP(t) != nil {
return true
}
// 非 IP 却含冒号:视为「带端口」的误用,丢弃(映射不需要端口)
if strings.Contains(t, ":") || !strings.Contains(t, ".") || strings.ContainsAny(t, " /\\") {
return false
}
for _, r := range t {
if r >= 'a' && r <= 'z' || r >= 'A' && r <= 'Z' || r >= '0' && r <= '9' ||
r == '.' || r == '-' || r == '_' || r == '*' {
continue
}
return false
}
return true
}
// Empty 无任何请求级覆盖(此时才允许走共享缓存)。
func (r ReqResolve) Empty() bool {
return len(r.dns)+len(r.dns4)+len(r.dns6)+len(r.resolve)+len(r.resolve4)+len(r.resolve6) == 0
}
// Servers 按栈名取本次生效的 DNS 服务器:stack 为 "ipv4"/"ipv6" 时优先取对应覆盖,
// 其余(含空串)取 dns= 默认;最终为 nil 表示用配置 dns-server。
// 供 whois、dns 拨测等按记录类型选栈的调用方使用。
func (r ReqResolve) Servers(stack string) []string {
switch stack {
case "ipv4":
if len(r.dns4) > 0 {
return r.dns4
}
case "ipv6":
if len(r.dns6) > 0 {
return r.dns6
}
}
return r.dns
}
// Override 返回该域名在指定族下的强制解析目标;"" 表示未命中(照常走 DNS)。
// 优先级与 dns* 对称:resolve4/resolve6 非空时覆盖 resolve。
// 目标是 IP 还是域名由调用方判断——IP 直连、域名当作新 host 再解析。
func (r ReqResolve) Override(host string, wantV6 bool) string {
m := r.resolve
if wantV6 && len(r.resolve6) > 0 {
m = r.resolve6
} else if !wantV6 && len(r.resolve4) > 0 {
m = r.resolve4
}
return m[canonicalHost(host)]
}
// CacheKey 返回请求级配置的规范化片段,拼在拨测缓存键末尾。
// 零值返回空串(缓存键与「没有任何请求级参数」时完全一致);
// 否则形如 ";dns=1.1.1.1:53,8.8.8.8:53;resolve6=example.com:[::1]"。
//
// 逐字段拼接而不是 join 结构体:字段名与顺序由这里固定,避免参数书写顺序
// (dns= 在前还是 dns6= 在前)把同一个请求拆成两条缓存。
func (r ReqResolve) CacheKey() string {
var b strings.Builder
for _, f := range []struct {
name string
vals []string
}{
{"dns", r.dns}, {"dns4", r.dns4}, {"dns6", r.dns6},
} {
if len(f.vals) == 0 {
continue
}
b.WriteString(";")
b.WriteString(f.name)
b.WriteString("=")
b.WriteString(strings.Join(f.vals, ","))
}
for _, f := range []struct {
name string
vals map[string]string
}{
{"resolve", r.resolve}, {"resolve4", r.resolve4}, {"resolve6", r.resolve6},
} {
if len(f.vals) == 0 {
continue
}
// map 迭代顺序随机:必须按键排序,否则同一请求每次拼出的键都不同,缓存永远命不中
keys := make([]string, 0, len(f.vals))
for k := range f.vals {
keys = append(keys, k)
}
sort.Strings(keys)
pairs := make([]string, 0, len(keys))
for _, k := range keys {
pairs = append(pairs, k+":"+f.vals[k])
}
b.WriteString(";")
b.WriteString(f.name)
b.WriteString("=")
b.WriteString(strings.Join(pairs, ","))
}
return b.String()
}
// reqResolveCtxKey 请求级解析配置在 context 中的键。
type reqResolveCtxKey struct{}
// WithReqResolve 把请求级解析配置写进 context(空配置原样返回,零开销)。
func WithReqResolve(ctx context.Context, r ReqResolve) context.Context {
if r.Empty() {
return ctx
}
return context.WithValue(ctx, reqResolveCtxKey{}, r)
}
// ReqResolveFromContext 取出请求级解析配置(零值 = 无覆盖)。
func ReqResolveFromContext(ctx context.Context) ReqResolve {
if ctx == nil {
return ReqResolve{}
}
if v, ok := ctx.Value(reqResolveCtxKey{}).(ReqResolve); ok {
return v
}
return ReqResolve{}
}
// queryDNS 普通 DNS 查询:使用配置的 dns-server(主从 failover,见 queryWithServers)。
// resolveTyped 未收到请求级 servers 时走这里。
func queryDNS(msg *dns.Msg) (*dns.Msg, error) {
return queryWithServers(msg, dnsServers)
}
@@ -206,13 +429,21 @@ type DNSResult struct {
// 即视为一次有效查询,将 Rcode 透出并返回 200;仅在传输/网络层失败(超时、服务器不可达、
// DoH 异常)时才返回 error(上层据此返回 500)。避免把合法的负响应误判为节点故障。
// extract 从一个 RR 中抽取记录字符串(TXT 可能含多条),返回 (values, ttl, matched)。
func resolveTyped(domain, name string, qtype uint16, extract func(dns.RR) ([]string, uint32, bool)) (DNSResult, error) {
// servers 为请求级 DNS 覆盖:非空时用它(queryWithServers),为空则用配置 dns-server(queryDNS)。
func resolveTyped(domain, name string, qtype uint16, servers []string, extract func(dns.RR) ([]string, uint32, bool)) (DNSResult, error) {
start := time.Now()
msg := new(dns.Msg)
msg.SetQuestion(dns.Fqdn(name), qtype)
result := DNSResult{Domain: domain}
response, err := queryDNS(msg)
// 请求级 servers 优先;未传(nil/空)走配置 dns-server。
var response *dns.Msg
var err error
if len(servers) > 0 {
response, err = queryWithServers(msg, servers)
} else {
response, err = queryDNS(msg)
}
result.Duration = time.Since(start).Seconds() * 1000
if err != nil {
slog.Warn("Failed to query DNS", "domain", domain, "qtype", qtype, "error", err)
@@ -235,8 +466,12 @@ func resolveTyped(domain, name string, qtype uint16, extract func(dns.RR) ([]str
return result, nil
}
func ResolveARecord(domain string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeA, func(rr dns.RR) ([]string, uint32, bool) {
// ---- 记录解析:每类一个函数,第三个参数 servers 为「请求级 DNS 覆盖」 ----
//
// servers 为空 → 走 resolveTyped 的 queryDNS(配置 dns-server);非空 → queryWithServers(请求级覆盖)。
func ResolveARecord(domain string, servers []string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeA, servers, func(rr dns.RR) ([]string, uint32, bool) {
if a, ok := rr.(*dns.A); ok {
return []string{a.A.String()}, a.Header().Ttl, true
}
@@ -244,8 +479,8 @@ func ResolveARecord(domain string) (DNSResult, error) {
})
}
func ResolveAAAARecord(domain string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeAAAA, func(rr dns.RR) ([]string, uint32, bool) {
func ResolveAAAARecord(domain string, servers []string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeAAAA, servers, func(rr dns.RR) ([]string, uint32, bool) {
if a, ok := rr.(*dns.AAAA); ok {
return []string{a.AAAA.String()}, a.Header().Ttl, true
}
@@ -253,8 +488,8 @@ func ResolveAAAARecord(domain string) (DNSResult, error) {
})
}
func ResolveTXTRecord(domain string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeTXT, func(rr dns.RR) ([]string, uint32, bool) {
func ResolveTXTRecord(domain string, servers []string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeTXT, servers, func(rr dns.RR) ([]string, uint32, bool) {
if txt, ok := rr.(*dns.TXT); ok {
// TXT 一条 RR 可能含多段字符串,保持原有「逐段单独成项」的行为
return txt.Txt, txt.Header().Ttl, true
@@ -263,8 +498,8 @@ func ResolveTXTRecord(domain string) (DNSResult, error) {
})
}
func ResolveNSRecord(domain string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeNS, func(rr dns.RR) ([]string, uint32, bool) {
func ResolveNSRecord(domain string, servers []string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeNS, servers, func(rr dns.RR) ([]string, uint32, bool) {
if ns, ok := rr.(*dns.NS); ok {
return []string{ns.Ns}, ns.Header().Ttl, true
}
@@ -272,8 +507,8 @@ func ResolveNSRecord(domain string) (DNSResult, error) {
})
}
func ResolveCNAMERecord(domain string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeCNAME, func(rr dns.RR) ([]string, uint32, bool) {
func ResolveCNAMERecord(domain string, servers []string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeCNAME, servers, func(rr dns.RR) ([]string, uint32, bool) {
if cname, ok := rr.(*dns.CNAME); ok {
return []string{cname.Target}, cname.Header().Ttl, true
}
@@ -281,8 +516,8 @@ func ResolveCNAMERecord(domain string) (DNSResult, error) {
})
}
func ResolveMXRecord(domain string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeMX, func(rr dns.RR) ([]string, uint32, bool) {
func ResolveMXRecord(domain string, servers []string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeMX, servers, func(rr dns.RR) ([]string, uint32, bool) {
if mx, ok := rr.(*dns.MX); ok {
return []string{mx.Mx}, mx.Header().Ttl, true
}
@@ -290,8 +525,8 @@ func ResolveMXRecord(domain string) (DNSResult, error) {
})
}
func ResolveSRVRecord(domain string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeSRV, func(rr dns.RR) ([]string, uint32, bool) {
func ResolveSRVRecord(domain string, servers []string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeSRV, servers, func(rr dns.RR) ([]string, uint32, bool) {
if srv, ok := rr.(*dns.SRV); ok {
return []string{srv.Target}, srv.Header().Ttl, true
}
@@ -299,8 +534,8 @@ func ResolveSRVRecord(domain string) (DNSResult, error) {
})
}
func ResolveCAARecord(domain string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeCAA, func(rr dns.RR) ([]string, uint32, bool) {
func ResolveCAARecord(domain string, servers []string) (DNSResult, error) {
return resolveTyped(domain, domain, dns.TypeCAA, servers, func(rr dns.RR) ([]string, uint32, bool) {
if caa, ok := rr.(*dns.CAA); ok {
return []string{caa.Value}, caa.Header().Ttl, true
}
@@ -308,13 +543,13 @@ func ResolveCAARecord(domain string) (DNSResult, error) {
})
}
func ResolvePTRRecord(ip string) (DNSResult, error) {
func ResolvePTRRecord(ip string, servers []string) (DNSResult, error) {
ptrName, err := dns.ReverseAddr(ip)
if err != nil {
slog.Warn("Invalid IP address for PTR query", "ip", ip, "error", err)
return DNSResult{Domain: ip, Record: []string{}}, fmt.Errorf("invalid IP address: %v", err)
}
return resolveTyped(ip, ptrName, dns.TypePTR, func(rr dns.RR) ([]string, uint32, bool) {
return resolveTyped(ip, ptrName, dns.TypePTR, servers, func(rr dns.RR) ([]string, uint32, bool) {
if ptr, ok := rr.(*dns.PTR); ok {
return []string{ptr.Ptr}, ptr.Header().Ttl, true
}
@@ -322,6 +557,32 @@ func ResolvePTRRecord(ip string) (DNSResult, error) {
})
}
// ResolveByType 按记录类型名(a/aaaa/cname/mx/ns/txt/srv/ptr/caa)解析,servers 为请求级覆盖。
// DNS 拨测的 HTTP 路由与 WS 报文统一走这里,避免两处各维护一套 switch。
func ResolveByType(domain, recodeType string, servers []string) (DNSResult, error) {
switch recodeType {
case "a":
return ResolveARecord(domain, servers)
case "aaaa":
return ResolveAAAARecord(domain, servers)
case "cname":
return ResolveCNAMERecord(domain, servers)
case "mx":
return ResolveMXRecord(domain, servers)
case "ns":
return ResolveNSRecord(domain, servers)
case "srv":
return ResolveSRVRecord(domain, servers)
case "txt":
return ResolveTXTRecord(domain, servers)
case "ptr":
return ResolvePTRRecord(domain, servers)
case "caa":
return ResolveCAARecord(domain, servers)
}
return DNSResult{}, fmt.Errorf("unsupported record type: %s", recodeType)
}
// DNSFullResult 完整的DNS查询结果(不包含PTR,PTR需单独查询)
type DNSFullResult struct {
Domain string `json:"domain"`
@@ -335,8 +596,9 @@ type DNSFullResult struct {
CAA DNSResult `json:"caa"`
}
// ResolveARecordllDNSRecords 并行查询所有主流DNS记录类型(不包含PTR)
func ResolveARecordllDNSRecords(domain string) DNSFullResult {
// ResolveARecordllDNSRecords 并行查询所有主流DNS记录类型(不包含PTR)。
// servers 为请求级 DNS 覆盖(空则用配置 dns-server)。
func ResolveARecordllDNSRecords(domain string, servers []string) DNSFullResult {
result := DNSFullResult{Domain: domain}
var wg sync.WaitGroup
@@ -344,42 +606,42 @@ func ResolveARecordllDNSRecords(domain string) DNSFullResult {
go func() {
defer wg.Done()
result.A, _ = ResolveARecord(domain)
result.A, _ = ResolveARecord(domain, servers)
}()
go func() {
defer wg.Done()
result.AAAA, _ = ResolveAAAARecord(domain)
result.AAAA, _ = ResolveAAAARecord(domain, servers)
}()
go func() {
defer wg.Done()
result.CNAME, _ = ResolveCNAMERecord(domain)
result.CNAME, _ = ResolveCNAMERecord(domain, servers)
}()
go func() {
defer wg.Done()
result.MX, _ = ResolveMXRecord(domain)
result.MX, _ = ResolveMXRecord(domain, servers)
}()
go func() {
defer wg.Done()
result.NS, _ = ResolveNSRecord(domain)
result.NS, _ = ResolveNSRecord(domain, servers)
}()
go func() {
defer wg.Done()
result.TXT, _ = ResolveTXTRecord(domain)
result.TXT, _ = ResolveTXTRecord(domain, servers)
}()
go func() {
defer wg.Done()
result.SRV, _ = ResolveSRVRecord(domain)
result.SRV, _ = ResolveSRVRecord(domain, servers)
}()
go func() {
defer wg.Done()
result.CAA, _ = ResolveCAARecord(domain)
result.CAA, _ = ResolveCAARecord(domain, servers)
}()
wg.Wait()
+13 -4
View File
@@ -1,6 +1,7 @@
package webtest
import (
"net"
"net/netip"
"net/url"
"strings"
@@ -21,8 +22,8 @@ func CleanHostRecord(addr string) string {
return addr
}
// measureDNSTime 当 trace 未提供 DNS 耗时时的兜底测量
func measureDNSTime(urlStr string, version string) float64 {
// measureDNSTime 当 trace 未提供 DNS 耗时时的兜底测量。req 为请求级解析配置(dns 服务器)。
func measureDNSTime(urlStr string, version string, req ReqResolve) float64 {
parsed, err := url.Parse(urlStr)
if err != nil {
return 0
@@ -31,12 +32,20 @@ func measureDNSTime(urlStr string, version string) float64 {
if host == "" {
return 0
}
// 强制解析:目标是 IP 时原域名根本不查 DNS,不能报一个凭空的耗时;
// 目标是域名则改按新 host 计时(真实查询的是它)。
if target := req.Override(host, version == "v6"); target != "" {
host = strings.Trim(target, "[]")
if net.ParseIP(host) != nil {
return 0
}
}
start := time.Now()
var dnsErr error
if version == "v6" {
_, dnsErr = ResolveAAAARecord(host)
_, dnsErr = ResolveAAAARecord(host, req.Servers("ipv6"))
} else {
_, dnsErr = ResolveARecord(host)
_, dnsErr = ResolveARecord(host, req.Servers("ipv4"))
}
if dnsErr != nil {
return 0
+4 -3
View File
@@ -10,8 +10,9 @@ import (
)
// SpeedTest 网站测速(speed)。客户端由 main.go 经 SetHTTPClient 注入。
func SpeedTest(url string, version string) (*WebsiteSpeedTestResult, error) {
ctx := context.Background()
// req 为请求级解析配置(dns/dns4/dns6),经 ctx 传到 DialContext。
func SpeedTest(url string, version string, req ReqResolve) (*WebsiteSpeedTestResult, error) {
ctx := WithReqResolve(context.Background(), req)
var err error
ctx, err = lemon_ssrf.ValidateOutboundTarget(ctx, url)
if err != nil {
@@ -52,7 +53,7 @@ func SpeedTest(url string, version string) (*WebsiteSpeedTestResult, error) {
dnsLookupTime := trace.DNSLookup.Seconds() * 1000
if dnsLookupTime == 0 {
dnsLookupTime = measureDNSTime(url, version)
dnsLookupTime = measureDNSTime(url, version, req)
}
tcpConnectTime := trace.TCPConnTime.Seconds() * 1000
httpConnectTime := trace.ConnTime.Seconds() * 1000
+3 -2
View File
@@ -10,8 +10,9 @@ import (
)
// CheckSSL SSL 检查(ssl)。客户端由 main.go 经 SetHTTPClient 注入。
func CheckSSL(url string, version string) (*SSLCheckDetail, error) {
ctx := context.Background()
// req 为请求级解析配置(dns/dns4/dns6),经 ctx 传到 DialContext。
func CheckSSL(url string, version string, req ReqResolve) (*SSLCheckDetail, error) {
ctx := WithReqResolve(context.Background(), req)
var err error
ctx, err = lemon_ssrf.ValidateOutboundTarget(ctx, url)
if err != nil {
+65
View File
@@ -0,0 +1,65 @@
package webtest
import (
"context"
"net"
"strings"
"lemon-ipw/ssrf"
)
// InstallSSRFResolver 让 SSRF 出站校验走本包的自定义 DNS(含请求级覆盖)。
// 必须在启动阶段调用一次(main.go)。
//
// 动机:不注入时 ssrf 用 net.LookupIP(系统解析器)做预检,而真正拨号走自定义 DNS —— 两套解析器,
// 既多一次查询,结果也可能不一致;更关键的是,当调用方用请求级 dns= 指定了服务器、且目标域名只能
// 靠该服务器解析时,系统解析器预检会先行失败,把请求直接毙掉,覆盖形同失效。
//
// 校验只需判定「是否私网」,故 A / AAAA 都查(任一命中私网即拦,口径与原先 net.LookupIP 一致)。
// 请求级配置(DNS 服务器)取自 context,与真正拨号同源。
func InstallSSRFResolver() {
ssrf.SetResolveFunc(func(ctx context.Context, host string) ([]net.IP, error) {
// 字面 IP 直接返回,无需解析
if ip := net.ParseIP(host); ip != nil {
return []net.IP{ip}, nil
}
req := ReqResolveFromContext(ctx)
var ips []net.IP
var firstErr error
for _, wantV6 := range []bool{false, true} {
// 强制解析命中时,预检要验「拨号真正会连的地址」,否则会拿原域名的
// 真实解析去判定,预检与实际连接两个口径对不上(resolve= 也就白设了)。
name := host
if target := req.Override(name, wantV6); target != "" {
name = strings.Trim(target, "[]")
}
if ip := net.ParseIP(name); ip != nil {
ips = append(ips, ip)
continue
}
var res DNSResult
var err error
if wantV6 {
res, err = ResolveAAAARecord(name, req.Servers("ipv6"))
} else {
res, err = ResolveARecord(name, req.Servers("ipv4"))
}
if err != nil && firstErr == nil {
firstErr = err
}
for _, s := range res.Record {
if ip := net.ParseIP(s); ip != nil {
ips = append(ips, ip)
}
}
}
// 两栈都解析失败才报错;只要有一栈拿到 IP 就交给上层做私网判定
if len(ips) == 0 && firstErr != nil {
return nil, firstErr
}
return ips, nil
})
}
+56 -40
View File
@@ -34,53 +34,60 @@ type TCPingStats struct {
Results []TCPingResult `json:"results"` // 每次测试的详细结果
}
// resolveHost 将主机名解析为指定协议版本的 IP 地址
// version 参数支持 "v4"(IPv4)和 "v6"(IPv6)
// 如果 host 已经是 IP 地址(可带 []),直接返回,跳过 DNS 解析
// ResolveHost 将主机名解析为指定协议版本的 IP 地址。
// version 支持 "v4" / "v6";host 已是 IP(可带 [])时直接返回,跳过 DNS 解析。
func ResolveHost(host string, version string) (string, error) {
return ResolveHostWith(host, version, ReqResolve{})
}
// ResolveHostWith 同 ResolveHost,但带请求级解析配置(req 零值 = 用配置 DNS)。
// 字面 IP 不查 DNS,但仍要过私网校验(与 DNS 解析结果同口径)。
func ResolveHostWith(host string, version string, req ReqResolve) (string, error) {
// 强制解析:命中就把目标当作新的 host——IP 交给下面的字面 IP 分支,域名继续走 DNS。
if target := req.Override(host, version == "v6"); target != "" {
host = strings.Trim(target, "[]")
}
cleanHost := strings.Trim(host, "[]")
if ip := net.ParseIP(cleanHost); ip != nil {
if version == "v4" && ip.To4() != nil {
return ip.String(), nil
// SSRF:字面 IP 与 DNS 解析结果同口径,私网地址一律拒绝(tcping 是裸 TCP,
// 不经过 DialContext,这里不做就完全没人拦)。放在族校验之前:安全判定
// 优先于参数语义,也与 DialContext 里「先查私网、再拨号」的顺序一致。
if ssrf.Enabled() && ssrf.IsPrivateIP(ip) {
return "", fmt.Errorf("connection to private/internal address %s is not allowed", cleanHost)
}
if version == "v6" && ip.To4() == nil {
return ip.String(), nil
switch version {
case "v4":
if ip.To4() != nil {
return ip.String(), nil
}
case "v6":
if ip.To4() == nil {
return ip.String(), nil
}
}
return "", fmt.Errorf("host %s is not a %s address", host, version)
}
var ipStr string
var result DNSResult
var err error
if version == "v4" {
var result DNSResult
result, err = ResolveARecord(host)
if err != nil {
return "", fmt.Errorf("DNS lookup failed: %w", err)
}
if len(result.Record) == 0 {
return "", fmt.Errorf("no %s address found for %s", version, host)
}
ipStr = result.Record[0]
if version == "v6" {
result, err = ResolveAAAARecord(host, req.Servers("ipv6"))
} else {
var result DNSResult
result, err = ResolveAAAARecord(host)
if err != nil {
return "", fmt.Errorf("DNS lookup failed: %w", err)
}
if len(result.Record) == 0 {
return "", fmt.Errorf("no %s address found for %s", version, host)
}
ipStr = result.Record[0]
result, err = ResolveARecord(host, req.Servers("ipv4"))
}
if err != nil {
return "", fmt.Errorf("DNS lookup failed: %w", err)
}
if len(result.Record) == 0 {
return "", fmt.Errorf("no %s address found for %s", version, host)
}
ipStr := result.Record[0]
if ssrf.Enabled() {
ip := net.ParseIP(ipStr)
if ip != nil && ssrf.IsPrivateIP(ip) {
if ip := net.ParseIP(ipStr); ip != nil && ssrf.IsPrivateIP(ip) {
return "", fmt.Errorf("connection to private/internal address %s is not allowed", ipStr)
}
}
return ipStr, nil
}
@@ -97,13 +104,19 @@ func TCPing(host string, port string, version string, timeout time.Duration) (*T
if err != nil {
return nil, err
}
return tcpingDial(ip, port, version, timeout), nil
}
// tcpingDial 对已解析好的 IP 做一次 TCP 拨号,不再做 DNS 解析与 SSRF 校验。
// 拆出来是为了让 TCPingRun 的多次探测复用同一个解析结果:原实现每轮都走 TCPing→
// ResolveHost,既拖慢,又会因多 A 记录轮询导致 results 里每次连到不同 IP(与顶层 ip 不符)。
func tcpingDial(ip string, port string, version string, timeout time.Duration) *TCPingResult {
addr := ""
switch version {
case "v4":
addr = fmt.Sprintf("%s:%s", ip, port)
case "v6":
addr = fmt.Sprintf("[%s]:%s", ip, port)
default:
addr = fmt.Sprintf("%s:%s", ip, port)
}
start := time.Now()
@@ -127,7 +140,7 @@ func TCPing(host string, port string, version string, timeout time.Duration) (*T
result.RTT = float64(rtt.Microseconds()) / 1000.0
}
return result, nil
return result
}
// TCPingRun 执行多次 TCP 连接测试并返回统计结果
@@ -141,7 +154,12 @@ func TCPing(host string, port string, version string, timeout time.Duration) (*T
//
// 返回 TCPingStats 包含统计信息
func TCPingRun(host string, port string, count int, version string, timeout time.Duration, interval time.Duration) (*TCPingStats, error) {
ip, err := ResolveHost(host, version)
return TCPingRunWith(host, port, count, version, timeout, interval, ReqResolve{})
}
// TCPingRunWith 同 TCPingRun,但带请求级解析配置(req 零值 = 用配置 DNS)。
func TCPingRunWith(host string, port string, count int, version string, timeout time.Duration, interval time.Duration, req ReqResolve) (*TCPingStats, error) {
ip, err := ResolveHostWith(host, version, req)
if err != nil {
return &TCPingStats{
IP: "Error: " + err.Error(),
@@ -168,10 +186,8 @@ func TCPingRun(host string, port string, count int, version string, timeout time
successCount := 0
for i := 0; i < count; i++ {
result, err := TCPing(host, port, version, timeout)
if err != nil {
return nil, err
}
// 复用开头解析出的 ip,不再每轮重新 ResolveHost
result := tcpingDial(ip, port, version, timeout)
stats.Results = append(stats.Results, *result)
+4 -3
View File
@@ -9,8 +9,9 @@ import (
)
// CheckWebsite 网站检测(detail)。客户端由 main.go 经 SetHTTPClient 注入。
func CheckWebsite(url string, version string) (*WebsiteCheckDetail, error) {
ctx := context.Background()
// req 为请求级解析配置(dns/dns4/dns6),经 ctx 传到 DialContext。
func CheckWebsite(url string, version string, req ReqResolve) (*WebsiteCheckDetail, error) {
ctx := WithReqResolve(context.Background(), req)
var err error
ctx, err = lemon_ssrf.ValidateOutboundTarget(ctx, url)
if err != nil {
@@ -50,7 +51,7 @@ func CheckWebsite(url string, version string) (*WebsiteCheckDetail, error) {
dnsLookupTime := trace.DNSLookup.Seconds() * 1000
if dnsLookupTime == 0 {
dnsLookupTime = measureDNSTime(url, version)
dnsLookupTime = measureDNSTime(url, version, req)
}
tcpConnectTime := trace.TCPConnTime.Seconds() * 1000
httpConnectTime := trace.ConnTime.Seconds() * 1000
+62 -15
View File
@@ -104,20 +104,20 @@ const whoisReferralTimeout = 3 * time.Second
// 这样即便转介服务器不可达也只在 ~1s 内拿到注册局记录,不会干等满库的 30s 超时。
// 兜底:主路径拿不到内容时才 fallback(内置 TLD 映射 → IANA 查服务器 → 自定义 DNS 解析 IP →
// Happy Eyeballs 双栈竞争拨号)。
func QueryWhois(domain string) (*WhoisResult, error) {
raw, err := queryWhoisRegistry(domain)
func QueryWhois(domain string, v4Servers, v6Servers []string) (*WhoisResult, error) {
raw, err := queryWhoisRegistry(domain, v4Servers, v6Servers)
// 库在「连接成功但读失败」等场景会同时返回部分数据与错误,此时数据不能丢;
// 因此只在「内容为空」或「带错误」时让 fallback 再兜一次,并取内容更长的一方。
if err != nil || raw == "" {
if fb, fbErr := whoisRetryWithFallback(domain, err); len(fb) > len(raw) {
if fb, fbErr := whoisRetryWithFallback(domain, err, v4Servers, v6Servers); len(fb) > len(raw) {
raw, err = fb, fbErr
}
}
// 仅当注册局响应干净(err == nil)时才补查注册商转介
if err == nil && raw != "" {
if extra := chaseReferral(domain, raw); extra != "" {
if extra := chaseReferral(domain, raw, v4Servers, v6Servers); extra != "" {
raw += extra
}
}
@@ -129,16 +129,63 @@ func QueryWhois(domain string) (*WhoisResult, error) {
// queryWhoisRegistry 走 IANA → 注册局两跳取注册局记录,关闭库的转介链
// (转介改由 chaseReferral 限时补查,避免被不可达的转介服务器拖满 30s)
func queryWhoisRegistry(domain string) (string, error) {
return whois.NewClient().
func queryWhoisRegistry(domain string, v4Servers, v6Servers []string) (string, error) {
client := whois.NewClient().
SetDisableReferral(true).
SetTimeout(whoisLibTimeout).
Whois(domain)
SetTimeout(whoisLibTimeout)
// 请求级 DNS 覆盖:库默认走系统解析器,注入自定义拨号器改成走我们的 DNS
if len(v4Servers) > 0 || len(v6Servers) > 0 {
client.SetDialer(whoisResolveDialer{v4Servers: v4Servers, v6Servers: v6Servers})
}
return client.Whois(domain)
}
// whoisResolveDialer 实现 x/net/proxy.Dialer(结构体天然满足接口):
// likexian/whois 内部需要拨 whois 服务器时,先按请求级 DNS 解析成 IP 再拨,
// 避免库退回系统解析器、绕过 dns=/dns4=/dns6= 覆盖。
type whoisResolveDialer struct {
v4Servers, v6Servers []string
}
// Dial 优先 A(v4),失败再 AAAA(v6)——whois 服务器绝大多数只有 v4。
func (d whoisResolveDialer) Dial(network, addr string) (net.Conn, error) {
host, port, err := net.SplitHostPort(addr)
if err != nil {
return nil, err
}
dialer := &net.Dialer{Timeout: whoisConnectTimeout}
if ip := net.ParseIP(host); ip != nil {
return dialer.Dial(network, addr)
}
var lastErr error
if aRes, aErr := ResolveARecord(host, d.v4Servers); aErr == nil && len(aRes.Record) > 0 {
if conn, err := dialer.Dial("tcp", net.JoinHostPort(aRes.Record[0], port)); err == nil {
return conn, nil
} else {
lastErr = err
}
} else if aErr != nil {
lastErr = aErr
}
if aaaaRes, aaaaErr := ResolveAAAARecord(host, d.v6Servers); aaaaErr == nil && len(aaaaRes.Record) > 0 {
if conn, err := dialer.Dial("tcp", net.JoinHostPort(aaaaRes.Record[0], port)); err == nil {
return conn, nil
} else if lastErr == nil {
lastErr = err
}
} else if aaaaErr != nil && lastErr == nil {
lastErr = aaaaErr
}
if lastErr == nil {
lastErr = fmt.Errorf("no IP found for whois server %s", host)
}
return nil, lastErr
}
// chaseReferral 从注册局响应里取出注册商转介服务器,限时补查一次
// 任何失败(无转介 / 不可达 / 超时)都返回空串,绝不影响已到手的注册局数据
func chaseReferral(domain, raw string) string {
func chaseReferral(domain, raw string, v4Servers, v6Servers []string) string {
server, port := extractReferralServer(raw)
if server == "" {
return ""
@@ -148,7 +195,7 @@ func chaseReferral(domain, raw string) string {
return ""
}
v4IPs, v6IPs, err := resolveWhoisServerIPs(server)
v4IPs, v6IPs, err := resolveWhoisServerIPs(server, v4Servers, v6Servers)
if err != nil {
return ""
}
@@ -268,7 +315,7 @@ func filterPublicIPs(ips []string) []string {
// resolveWhoisServerIPs 使用 webtest/dns 的统一解析接口
// 并行查询 A + AAAA(走自定义 DNS 服务器,不受系统 resolver 策略影响)
func resolveWhoisServerIPs(server string) (v4IPs, v6IPs []string, err error) {
func resolveWhoisServerIPs(server string, v4Servers, v6Servers []string) (v4IPs, v6IPs []string, err error) {
var (
aResult DNSResult
aaaaResult DNSResult
@@ -279,11 +326,11 @@ func resolveWhoisServerIPs(server string) (v4IPs, v6IPs []string, err error) {
wg.Add(2)
go func() {
defer wg.Done()
aResult, aErr = ResolveARecord(server)
aResult, aErr = ResolveARecord(server, v4Servers)
}()
go func() {
defer wg.Done()
aaaaResult, aaaaErr = ResolveAAAARecord(server)
aaaaResult, aaaaErr = ResolveAAAARecord(server, v6Servers)
}()
wg.Wait()
@@ -372,7 +419,7 @@ func happyEyeballsWhoisQuery(parent context.Context, domain string, v4IPs, v6IPs
}
// whoisRetryWithFallback 在 whois.Whois 失败后,手动解析 WHOIS 服务器所有 IP 逐个尝试
func whoisRetryWithFallback(domain string, firstErr error) (string, error) {
func whoisRetryWithFallback(domain string, firstErr error, v4Servers, v6Servers []string) (string, error) {
ext := getExtension(domain)
// 解析 WHOIS 服务器地址(内置映射 + IANA fallback)
@@ -382,7 +429,7 @@ func whoisRetryWithFallback(domain string, firstErr error) (string, error) {
}
// 使用 ResolveA/AAAA(自定义 DNS)并行解析 v4/v6
v4IPs, v6IPs, err := resolveWhoisServerIPs(server)
v4IPs, v6IPs, err := resolveWhoisServerIPs(server, v4Servers, v6Servers)
if err != nil {
return "", firstErr
}
+81 -39
View File
@@ -35,15 +35,23 @@ func wsProbe(apiType, raw string, query map[string]string) (int, any) {
switch apiType {
case "detail":
testURL := normalizeURL(raw)
if cached, ok := websiteCache.Load(testURL); ok {
req, err := webtest.ParseReqResolve(func(k string) string { return query[k] })
if err != nil {
return 400, map[string]string{"error": err.Error()}
}
// 缓存键带上请求级 DNS 覆盖(结果取决于所用服务器,参数不同不能共用)
cacheKey := testURL + req.CacheKey()
if cached, ok := websiteCache.Load(cacheKey); ok {
entry := cached.(websiteCacheEntry)
if time.Since(entry.timestamp) < cacheTTL(entry.failed) {
return 200, entry.result
}
websiteCache.Delete(testURL)
websiteCache.Delete(cacheKey)
}
ipv4, ipv6 := wsDual[webtest.WebsiteCheckDetail](
func(v string) (*webtest.WebsiteCheckDetail, error) { return webtest.CheckWebsite(testURL, v) },
func(v string) (*webtest.WebsiteCheckDetail, error) {
return webtest.CheckWebsite(testURL, v, req)
},
func(err error) *webtest.WebsiteCheckDetail {
return &webtest.WebsiteCheckDetail{HostRecord: "Error: " + err.Error(), IsReachable: false}
},
@@ -55,20 +63,28 @@ func wsProbe(apiType, raw string, query map[string]string) (int, any) {
failed := dualAnyFailed(result.IPv4, result.IPv6, func(d *webtest.WebsiteCheckDetail) bool {
return d != nil && !d.IsReachable
})
websiteCache.Store(testURL, websiteCacheEntry{result: result, timestamp: time.Now(), failed: failed})
websiteCache.Store(cacheKey, websiteCacheEntry{result: result, timestamp: time.Now(), failed: failed})
return 200, result
case "ssl":
testURL := normalizeURL(raw)
if cached, ok := sslCache.Load(testURL); ok {
req, err := webtest.ParseReqResolve(func(k string) string { return query[k] })
if err != nil {
return 400, map[string]string{"error": err.Error()}
}
// 缓存键带上请求级 DNS 覆盖(结果取决于所用服务器,参数不同不能共用)
cacheKey := testURL + req.CacheKey()
if cached, ok := sslCache.Load(cacheKey); ok {
entry := cached.(sslCacheEntry)
if time.Since(entry.timestamp) < cacheTTL(entry.failed) {
return 200, entry.result
}
sslCache.Delete(testURL)
sslCache.Delete(cacheKey)
}
ipv4, ipv6 := wsDual[webtest.SSLCheckDetail](
func(v string) (*webtest.SSLCheckDetail, error) { return webtest.CheckSSL(testURL, v) },
func(v string) (*webtest.SSLCheckDetail, error) {
return webtest.CheckSSL(testURL, v, req)
},
func(err error) *webtest.SSLCheckDetail {
return &webtest.SSLCheckDetail{HostRecord: "Error: " + err.Error(), IsExpired: true, IsReachable: false}
},
@@ -80,7 +96,7 @@ func wsProbe(apiType, raw string, query map[string]string) (int, any) {
failed := dualAnyFailed(result.IPv4, result.IPv6, func(d *webtest.SSLCheckDetail) bool {
return d != nil && !d.IsReachable
})
sslCache.Store(testURL, sslCacheEntry{result: result, timestamp: time.Now(), failed: failed})
sslCache.Store(cacheKey, sslCacheEntry{result: result, timestamp: time.Now(), failed: failed})
return 200, result
case "speed":
@@ -90,7 +106,12 @@ func wsProbe(apiType, raw string, query map[string]string) (int, any) {
version = "v4"
}
testURL = normalizeURL(testURL)
cacheKey := fmt.Sprintf("%s:%s", testURL, version)
req, err := webtest.ParseReqResolve(func(k string) string { return query[k] })
if err != nil {
return 400, map[string]string{"error": err.Error()}
}
// 缓存键:URL + 版本 + 请求级 DNS 覆盖(结果取决于所用服务器,参数不同不能共用)
cacheKey := fmt.Sprintf("%s:%s%s", testURL, version, req.CacheKey())
if cached, ok := speedCache.Load(cacheKey); ok {
entry := cached.(speedCacheEntry)
if time.Since(entry.timestamp) < cacheTTL(entry.failed) {
@@ -98,7 +119,7 @@ func wsProbe(apiType, raw string, query map[string]string) (int, any) {
}
speedCache.Delete(cacheKey)
}
r, e := webtest.SpeedTest(testURL, version)
r, e := webtest.SpeedTest(testURL, version, req)
if e != nil {
errorResult := &webtest.WebsiteSpeedTestResult{HostRecord: "Error: " + e.Error()}
speedCache.Store(cacheKey, speedCacheEntry{result: errorResult, timestamp: time.Now(), failed: true})
@@ -126,7 +147,20 @@ func wsProbe(apiType, raw string, query map[string]string) (int, any) {
n = c
}
}
cacheKey := fmt.Sprintf("%s:%s:%d", host, port, n)
// version=ipv4/ipv6:与 HTTP handler 同义,请求级指定单栈(配置优先)
version := query["version"]
if version != "" && normalizeStack(version) == "" {
return 400, map[string]string{"error": "version must be ipv4 or ipv6"}
}
stack, skipLabel := probeStackFor(version)
// 请求级解析参数(dns=/dns4=/dns6=):与 HTTP handler 同解析;
// 带覆盖时旁路共享缓存(结果取决于所用服务器)
req, err := webtest.ParseReqResolve(func(k string) string { return query[k] })
if err != nil {
return 400, map[string]string{"error": err.Error()}
}
// 缓存键与 HTTP handler 保持一致(含生效栈 + 请求级 DNS 覆盖),两边共用 pingCache
cacheKey := fmt.Sprintf("%s:%s:%d:%s%s", host, port, n, stack, req.CacheKey())
if cached, ok := pingCache.Load(cacheKey); ok {
entry := cached.(pingCacheEntry)
if time.Since(entry.timestamp) < cacheTTL(entry.failed) {
@@ -134,13 +168,13 @@ func wsProbe(apiType, raw string, query map[string]string) (int, any) {
}
pingCache.Delete(cacheKey)
}
ipv4, ipv6 := wsDual[webtest.TCPingStats](
ipv4, ipv6 := wsDualStack[webtest.TCPingStats](stack,
func(v string) (*webtest.TCPingStats, error) {
return webtest.TCPingRun(host, port, n, v, 10*time.Second, 100*time.Millisecond)
return webtest.TCPingRunWith(host, port, n, v, 10*time.Second, 100*time.Millisecond, req)
},
func(err error) *webtest.TCPingStats { return &webtest.TCPingStats{IP: "Error: " + err.Error()} },
func(v string) *webtest.TCPingStats {
return &webtest.TCPingStats{IP: "Skipped due to SINGLE_STACK=" + v}
func(skipStack string) *webtest.TCPingStats {
return &webtest.TCPingStats{IP: "Skipped due to " + skipLabel + "=" + skipStack}
},
)
result := &TCPingResult{IPv4: ipv4, IPv6: ipv6}
@@ -157,29 +191,25 @@ func wsProbe(apiType, raw string, query map[string]string) (int, any) {
return 400, map[string]string{"error": "Invalid domain"}
}
domain = parsedURL.Host
var result any
switch recodeType {
case "a":
result, err = webtest.ResolveARecord(domain)
case "aaaa":
result, err = webtest.ResolveAAAARecord(domain)
case "cname":
result, err = webtest.ResolveCNAMERecord(domain)
case "mx":
result, err = webtest.ResolveMXRecord(domain)
case "ns":
result, err = webtest.ResolveNSRecord(domain)
case "txt":
result, err = webtest.ResolveTXTRecord(domain)
case "srv":
result, err = webtest.ResolveSRVRecord(domain)
case "ptr":
result, err = webtest.ResolvePTRRecord(domain)
case "caa":
result, err = webtest.ResolveCAARecord(domain)
case "a", "aaaa", "cname", "mx", "ns", "ptr", "srv", "txt", "caa":
default:
return 400, map[string]string{"error": "Unsupported record type: " + recodeType}
}
req, err := webtest.ParseReqResolve(func(k string) string { return query[k] })
if err != nil {
return 400, map[string]string{"error": err.Error()}
}
// 按 qtype 选栈:a→dns4|dns,aaaa→dns6|dns,其余→dns
stack := ""
if recodeType == "a" {
stack = "ipv4"
} else if recodeType == "aaaa" {
stack = "ipv6"
}
result, err := webtest.ResolveByType(domain, recodeType, req.Servers(stack))
if err != nil {
return 500, map[string]string{"error": err.Error()}
}
@@ -196,18 +226,24 @@ func wsProbe(apiType, raw string, query map[string]string) (int, any) {
if raw == "" {
return 400, map[string]string{"error": "Domain parameter is required"}
}
if cached, ok := whoisCache.Load(raw); ok {
req, err := webtest.ParseReqResolve(func(k string) string { return query[k] })
if err != nil {
return 400, map[string]string{"error": err.Error()}
}
// 缓存键带上请求级 DNS 覆盖(结果取决于所用服务器,参数不同不能共用)
cacheKey := raw + req.CacheKey()
if cached, ok := whoisCache.Load(cacheKey); ok {
entry := cached.(whoisCacheEntry)
if time.Since(entry.timestamp) < cacheTTL(entry.failed) {
return 200, entry.result
}
whoisCache.Delete(raw)
whoisCache.Delete(cacheKey)
}
result, err := webtest.QueryWhois(raw)
result, err := webtest.QueryWhois(raw, req.Servers("ipv4"), req.Servers("ipv6"))
if err != nil {
return 500, map[string]string{"error": err.Error()}
}
whoisCache.Store(raw, whoisCacheEntry{result: result, timestamp: time.Now()})
whoisCache.Store(cacheKey, whoisCacheEntry{result: result, timestamp: time.Now()})
return 200, result
case "location":
@@ -258,7 +294,13 @@ func splitRaw(raw string) (string, string) {
// wsDual 双栈编排(ws.go 内部,学 main.go handler 的 SINGLE_STACK 处理)
func wsDual[T any](run func(version string) (*T, error), mkErr func(err error) *T, mkSkip func(version string) *T) (ipv4, ipv6 *T) {
switch SINGLE_STACK {
return wsDualStack(SINGLE_STACK, run, mkErr, mkSkip)
}
// wsDualStack 是 wsDual 的可指定栈版本:stack 为 "" 双栈并行,"ipv4"/"ipv6" 只跑该栈。
// tcping 借此支持请求级 version 参数(probeStackFor 已把配置优先规则算好)。
func wsDualStack[T any](stack string, run func(version string) (*T, error), mkErr func(err error) *T, mkSkip func(stack string) *T) (ipv4, ipv6 *T) {
switch stack {
case "ipv4":
v4, err := run("v4")
if err != nil {