Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
9e2bb4ceeb | ||
|
|
809e2aeb82 | ||
|
|
54981f7348 | ||
|
|
8df19f01e0 | ||
|
|
6c2fc33fff | ||
|
|
44f5b2b09c | ||
|
|
1576ce9038 | ||
|
|
decf48992e | ||
|
|
7c51ea3f86 | ||
|
|
3d64d6394b | ||
|
|
b903775fb7 | ||
|
|
9c5bbaf55d | ||
|
|
2f4cb008ae | ||
|
|
9bbd950b66 | ||
|
|
44b2313299 | ||
|
|
37861cda22 | ||
|
|
5f43698e2b | ||
|
|
ab6f1e1329 | ||
|
|
ff175174e4 | ||
|
|
ed0e240241 | ||
|
|
c6bfefe2fb | ||
|
|
fa990c2094 | ||
|
|
1797c50ea3 | ||
|
|
492a7f2002 | ||
|
|
5ddca88147 | ||
|
|
8117f6ed11 | ||
|
|
11cfd3f092 | ||
|
|
628878e055 | ||
|
|
7e0ec393a8 | ||
|
|
8a141ae86d | ||
|
|
2ff7976768 |
@@ -1,3 +1,4 @@
|
||||
/shell2.exe
|
||||
/BOT_PLAN.md
|
||||
/SECURITY_REVIEW.md
|
||||
/DragonCoreSSH-NewWEB.zip
|
||||
|
||||
@@ -15,6 +15,9 @@ DragonCoreSSH V40 é um painel/servidor em Go para SSH com HTTP Injection, paine
|
||||
- Endpoint XHTTP compartilhado no modo nativo: VLESS **ou** VMess em `/` e SSH em `/ssh`, usando o mesmo domínio/porta/TLS
|
||||
- Área compacta de infraestrutura com Servidores, Status, Monitoramento e Tráfego no mesmo seletor visual
|
||||
- Cartões de status ao vivo nos espaços SSH, Xray e Infraestrutura, com confirmações integradas ao painel
|
||||
- Listas de usuários SSH e Xray com botões de ordenação e filtros por status, conexão, uso, validade e cota, além de cabeçalhos clicáveis
|
||||
- Velocidade ao vivo (subida/descida) por conta nas listas SSH e Xray, somando todas as conexões do usuário, com ordenação por velocidade
|
||||
- Listas de usuários em formato de cartão no celular: cada linha vira um cartão com rótulos, sem rolagem lateral
|
||||
- Navegação interna consistente com o Bot: SSH/SlowDNS e Revendedores separam consulta de cadastro; Xray separa Usuários, Criar usuário, Configuração e Logs; Configurações separa Rede/SSH, SlowDNS, UDP, TLS e Xray
|
||||
- Contas de revendedor (reseller) com cota de usuários e escopo próprio
|
||||
- Gerenciamento multi-servidor (master/slave) direto pelo painel
|
||||
@@ -66,6 +69,18 @@ Para configurações XHTTP antigas, carregue a configuração visual e clique em
|
||||
|
||||
A confirmação dessa migração é exibida dentro do próprio painel. Se a gravação falhar, o inbound SSH temporário é removido do rascunho e o inbound antigo permanece intacto, permitindo tentar novamente após corrigir o erro exibido.
|
||||
|
||||
### Cota de tráfego e proteção de recursos
|
||||
|
||||
Contas SSH e clientes VLESS/VMess do modo nativo podem usar `data_quota_bytes` com ação `block` ou `throttle`. O botão **Reset/Zerar tráfego** limpa apenas os contadores; não renova validade, senha ou configuração da conta. Não existe reset periódico automático no servidor: qualquer período comercial mostrado no site é independente e o reset ocorre somente por ação explícita no painel/API. O valor `max_conns` é aplicado no momento em que o usuário VLESS/VMess é autenticado e vale em conjunto para TCP, UDP, WebSocket, XHTTP e conexões Mux (uma conexão Mux autenticada conta como uma conexão, independentemente dos streams filhos).
|
||||
|
||||
As antigas chaves globais de admissão continuam no JSON somente para compatibilidade, mas são sempre normalizadas para `-1` (ilimitado), inclusive quando um `config.json` antigo ainda contém `4096`, `8192`, `32768` ou outro valor positivo:
|
||||
|
||||
- `max_concurrent_connections`: sem limite global de conexões de transporte;
|
||||
- `max_concurrent_xhttp_requests`: sem limite global de requisições XHTTP;
|
||||
- `xhttp_max_sessions`: sem limite global de sessões XHTTP.
|
||||
|
||||
O painel não expõe mais esses três controles como limites ajustáveis. Xray XHTTP e XHTTP SSH usam o mesmo listener VPN sem teto por quantidade de requisições, streams HTTP/2, conexões de transporte ou sessões XHTTP. Rajadas de `packet-up` e a remontagem fora de ordem usam backpressure cancelável contabilizado em bytes; até pacotes vazios consomem um custo mínimo de memória contabilizada, portanto remover o limite por quantidade não cria uma fila de metadados sem limite. Não existem respostas `429` nem rejeições `503` por capacidade global. As políticas reais por usuário (`max_conns`, cota e banda) continuam ativas. Cada transporte Mux aceita no máximo 64 sessões filhas, com limite global padrão de 32768. Parar/reiniciar o Xray nativo fecha conexões e sessões existentes. Atualizações de tráfego e de conexões ativas são agregadas e persistidas em lote a cada cinco segundos, sem criar uma goroutine ou consulta PostgreSQL por conexão. Entradas pendentes de usuários removidos são descartadas para manter os mapas de retry limitados ao conjunto atual de contas.
|
||||
|
||||
### Requisitos
|
||||
|
||||
- Servidor Linux com `systemd`
|
||||
@@ -186,7 +201,7 @@ Também é possível editar diretamente o `config.json`:
|
||||
```json
|
||||
"dnstt": {
|
||||
"domain": "t.example.com",
|
||||
"udp_listen": "[::]:5300",
|
||||
"udp_listen": "0.0.0.0:5300",
|
||||
"privkey_file": "/opt/sshpanel/dnstt.key",
|
||||
"auto_restart_interval": "6h",
|
||||
"auto_restart_grace": "2s"
|
||||
@@ -206,7 +221,7 @@ Exemplo:
|
||||
"t.example.com",
|
||||
"t.local.lan"
|
||||
],
|
||||
"udp_listen": "[::]:5300",
|
||||
"udp_listen": "0.0.0.0:5300",
|
||||
"privkey_file": "/opt/sshpanel/dnstt.key"
|
||||
}
|
||||
```
|
||||
@@ -544,6 +559,8 @@ DragonCoreSSH V40 is a Go-based SSH HTTP Injection server with a web panel, Post
|
||||
- Native shared XHTTP endpoint: VLESS **or** VMess on `/` and SSH on `/ssh`, using the same domain/port/TLS
|
||||
- Compact infrastructure workspace with Servers, Status, Monitoring, and Traffic in one visual switcher
|
||||
- Live status cards across SSH, Xray, and Infrastructure, with panel-native confirmations
|
||||
- Live per-account up/down speed in the SSH and Xray user lists, summed across every connection the account has open, sortable by speed
|
||||
- User lists collapse into labelled cards on phones, so there is no sideways scrolling
|
||||
- Bot-style section navigation throughout the panel: SSH/SlowDNS and Resellers separate lists from creation; Xray separates Users, Create User, Configuration, and Logs; Settings separates Network/SSH, SlowDNS, UDP, TLS, and Xray
|
||||
- Reseller accounts with a user quota and self-scoped access
|
||||
- Multi-server (master/slave) management directly from the panel
|
||||
@@ -595,6 +612,18 @@ For older XHTTP configurations, load the visual configuration and click **Enable
|
||||
|
||||
The migration confirmation is rendered inside the panel. If saving fails, the temporary SSH inbound is removed from the draft and the old inbound remains intact, so the operation can be retried after fixing the displayed error.
|
||||
|
||||
### Traffic quotas and resource protection
|
||||
|
||||
SSH accounts and native-mode VLESS/VMess clients can use `data_quota_bytes` with either the `block` or `throttle` action. The **Reset traffic** action clears only usage counters; it does not renew expiry, change a password, or alter account settings. The server does not perform an automatic periodic reset: any commercial period shown on the website is independent, and counters reset only through an explicit panel/API action. `max_conns` is enforced when a native VLESS/VMess user is authenticated and is shared across TCP, UDP, WebSocket, XHTTP, and Mux transports (one authenticated Mux transport counts as one connection, regardless of its child streams).
|
||||
|
||||
The old global admission keys remain in JSON for compatibility, but they are always normalized to `-1` (unlimited), including when an old `config.json` still contains `4096`, `8192`, `32768`, or any other positive value:
|
||||
|
||||
- `max_concurrent_connections`: no global transport-connection count cap;
|
||||
- `max_concurrent_xhttp_requests`: no global XHTTP-request count cap;
|
||||
- `xhttp_max_sessions`: no global XHTTP-session count cap.
|
||||
|
||||
The panel no longer exposes those three controls as adjustable ceilings. Xray XHTTP and XHTTP SSH share the same VPN listener with no count ceiling for requests, HTTP/2 streams, transport connections, or XHTTP sessions. Packet-up bursts and out-of-order reassembly use cancelable byte-accounted backpressure; even empty packets are charged a minimum accounted-memory cost, so removing the request-count limit does not create an unbounded metadata queue. There are no `429` responses or global-capacity `503` rejections. Real per-user policies (`max_conns`, quota, and bandwidth) remain active. Each Mux transport accepts at most 64 child sessions, with a default global ceiling of 32768. Stopping/restarting native Xray closes existing transports and XHTTP sessions. Traffic and active-connection changes are aggregated and written in five-second batches rather than creating a PostgreSQL query or goroutine for every connection. Pending retry entries for deleted clients are removed so retry maps stay bounded by the current account set.
|
||||
|
||||
### Requirements
|
||||
|
||||
- Linux server with `systemd`
|
||||
@@ -713,7 +742,7 @@ You can also edit `config.json` directly:
|
||||
```json
|
||||
"dnstt": {
|
||||
"domain": "t.example.com",
|
||||
"udp_listen": "[::]:5300",
|
||||
"udp_listen": "0.0.0.0:5300",
|
||||
"privkey_file": "/opt/sshpanel/dnstt.key",
|
||||
"auto_restart_interval": "6h",
|
||||
"auto_restart_grace": "2s"
|
||||
@@ -733,7 +762,7 @@ Example:
|
||||
"t.example.com",
|
||||
"t.local.lan"
|
||||
],
|
||||
"udp_listen": "[::]:5300",
|
||||
"udp_listen": "0.0.0.0:5300",
|
||||
"privkey_file": "/opt/sshpanel/dnstt.key"
|
||||
}
|
||||
```
|
||||
@@ -1120,14 +1149,17 @@ curl -s "http://SERVER_IP:9090/api/users" -H "X-Session-Token: $TOKEN"
|
||||
|
||||
#### `GET /api/users` — session
|
||||
- Optional query: `server_id`. Resellers see only their own users; superadmins see all.
|
||||
- `200`: array of user objects: `username` (string), `active_conns` (int), `max_connections` (int), `expires_at` (string/null), `limit_mbps_up` (int), `limit_mbps_down` (int), `totp_secret` (string, omitempty), `totp_period` (int), `totp_window` (int), `totp_digits` (int), `allow_static_password` (bool), `totp_enabled` (bool), `owner_username` (string, omitempty), `server_id` (string, omitempty).
|
||||
- `200`: array of user objects: `username` (string), `active_conns` (int), `max_connections` (int), `expires_at` (string/null), `limit_mbps_up` (int), `limit_mbps_down` (int), `data_quota_bytes` (int64), `quota_action` (`block` or `throttle`), `quota_throttle_mbps` (int), `total_uplink_bytes`, `total_downlink_bytes`, `total_bytes`, `up_bytes_per_sec` (float, live account-wide upload speed), `down_bytes_per_sec` (float, live account-wide download speed), `quota_exceeded`, `totp_secret` (string, omitempty), `totp_period` (int), `totp_window` (int), `totp_digits` (int), `allow_static_password` (bool), `totp_enabled` (bool), `owner_username` (string, omitempty), `server_id` (string, omitempty).
|
||||
|
||||
#### `POST /api/users/create` — session
|
||||
Creates or updates (upsert) an SSH user.
|
||||
- Body: `username` (string, required); `password` (string, optional — empty keeps the existing password on an existing user; for a new user either `password` or `totp_secret` is required); `max_connections` (int); `expires_at` (string); `limit_mbps_up` (int); `limit_mbps_down` (int); `totp_secret` (string); `totp_period` (int); `totp_window` (int); `totp_digits` (int); `allow_static_password` (bool); `owner_username` (string, optional — honored only for superadmin; resellers are forced to themselves); `server_id` (string, optional).
|
||||
- Body: `username` (string, required); `password` (string, optional — empty keeps the existing password on an existing user; for a new user either `password` or `totp_secret` is required); `max_connections` (0–10000); `expires_at` (RFC3339 string); `limit_mbps_up` and `limit_mbps_down` (0–1000000); `data_quota_bytes` (non-negative int64); `quota_action` (`block` or `throttle`); `quota_throttle_mbps` (0–1000000; zero defaults to 1); `reset_usage` (bool); `totp_secret` (string); `totp_period` (int); `totp_window` (int); `totp_digits` (int); `allow_static_password` (bool); `owner_username` (string, optional — honored only for superadmin; resellers are forced to themselves); `server_id` (string, optional).
|
||||
- `201 Created` (empty body). A proxied create returns the remote node's status/body.
|
||||
- Errors: `400 username required`, `400 password or totp_secret required for new user`; `403 user limit reached (N)`; `403 SSH creation is disabled for this server`; `503 database not configured`.
|
||||
|
||||
#### `POST /api/users/reset-traffic` — session
|
||||
- Body: `username` (required), `server_id` (optional). Resets only byte counters. Resellers may reset only their own users.
|
||||
|
||||
#### `DELETE /api/users/delete` — session
|
||||
- Query: `username` (string, required); optional `server_id`. Resellers may delete only their own users.
|
||||
- `204 No Content`. Errors: `400 username required`; `403 forbidden`; `503 database not configured`.
|
||||
@@ -1237,16 +1269,19 @@ Read/write a managed server's `config.json`. Query: `server_id`. Local delegates
|
||||
#### `GET /api/xray/inbounds` — session
|
||||
- Optional `server_id`. Lists only inbounds that carry client lists (vless/vmess/trojan). Resellers see all inbounds but only their own clients. Clients are enriched with DB metadata and runtime stats.
|
||||
- `200`: array of `{ "tag": string, "protocol": string, "port": <raw>, "listen": string, "clients": [ XrayClientInfo ] }`.
|
||||
- **XrayClientInfo**: `id` (string, the UUID), `password` (string, omitempty), `email` (string), `level` (int), `online` (bool), `last_active` (string/null), `uplink_bytes` (int64), `downlink_bytes` (int64), `total_bytes` (int64), `active_connections` (int), `name` (string), `expires_at` (string/null), `expiration_days` (int; `-1` = no expiry, `0` = expired), `max_conns`, `owner_username`, `expired`.
|
||||
- **XrayClientInfo**: `id` (string, the UUID), `password` (string, omitempty), `email` (string), `level` (int), `online` (bool), `last_active` (string/null), `uplink_bytes` (int64), `downlink_bytes` (int64), `total_bytes` (int64), `up_bytes_per_sec` (float, live client-wide upload speed), `down_bytes_per_sec` (float, live client-wide download speed), `active_connections` (int), `name` (string), `expires_at` (string/null), `expiration_days` (int; `-1` = no expiry, `0` = expired), `max_conns`, `data_quota_bytes`, `quota_action`, `quota_throttle_mbps`, `quota_exceeded`, `owner_username`, `expired`.
|
||||
|
||||
#### `POST /api/xray/clients/add` — session
|
||||
- Body: `inbound_tag` (string, required), `uuid` (string, required), `email` (string, optional — defaults to name then uuid), `name` (string, optional), `expires_at` (string, optional), `max_connections` (int), `owner_username` (string, optional — superadmin only), `server_id` (string, optional).
|
||||
- Body: `inbound_tag` (string, required), `uuid` (valid UUID, required), `email` (string, optional — defaults to name then uuid), `name` (string, optional), `expires_at` (RFC3339, `YYYY-MM-DDThh:mm`, or `YYYY-MM-DD`), `max_connections` (0–10000), `data_quota_bytes`, `quota_action`, `quota_throttle_mbps`, `owner_username` (string, optional — superadmin only), `server_id` (string, optional).
|
||||
- `201 Created` (empty). Errors: `400 inbound_tag and uuid required` / `UUID already exists in database`; `403 reseller account suspended or expired` / `user limit reached (N)` / `Xray creation is disabled for this server`; `500`.
|
||||
|
||||
#### `POST /api/xray/clients/update` — session
|
||||
- Body: `uuid` (string, required), `name` (string), `email` (string), `expires_at` (string), `max_connections` (int), `server_id` (string, optional). Inbound tag and owner are preserved from existing metadata. Resellers may update only their own clients.
|
||||
- Body: `uuid` (valid UUID, required), `name` (string), `email` (string), `expires_at` (string), `max_connections` (0–10000), `data_quota_bytes`, `quota_action`, `quota_throttle_mbps`, `reset_usage` (bool), `server_id` (string, optional). Inbound tag and owner are preserved from existing metadata. Resellers may update only their own clients.
|
||||
- `200`. Errors: `400 uuid required`; `403 forbidden`; `404 client metadata not found`; `500`.
|
||||
|
||||
#### `POST /api/xray/clients/reset-traffic` — session
|
||||
- Body: `uuid` (required), `server_id` (optional). Resets only byte counters. Resellers may reset only their own clients.
|
||||
|
||||
#### `DELETE /api/xray/clients/remove` — session
|
||||
- Query: `inbound_tag` (string, required), `uuid` (string, required); optional `server_id`. Resellers may remove only their own clients.
|
||||
- `204 No Content`. Errors: `400 inbound_tag and uuid required`; `403 forbidden`; `500`.
|
||||
@@ -1255,7 +1290,7 @@ Read/write a managed server's `config.json`. Query: `server_id`. Local delegates
|
||||
|
||||
### TLS certificates (superadmin only)
|
||||
|
||||
All three accept `POST` only and support `server_id` proxying.
|
||||
All endpoints support `server_id` proxying, so a certificate can also be listed/updated on a managed slave node. The three issue/upload endpoints below accept `POST` only.
|
||||
|
||||
#### `POST /api/tls/generate-selfsigned`
|
||||
- Body: `domain` (string, required). Writes a self-signed ECDSA (P-256) cert (10-year validity) to `/opt/sshpanel/certs/<domain>/`.
|
||||
@@ -1269,6 +1304,19 @@ All three accept `POST` only and support `server_id` proxying.
|
||||
- Body: `name` (string, required), `cert` (string, required — PEM), `key` (string, required — PEM). Saves to `/opt/sshpanel/certs/<name>/`.
|
||||
- `200`: `{ "cert_file": string, "key_file": string }`. Errors: `400 name, cert, and key required` / `invalid name`; `500`.
|
||||
|
||||
#### `GET /api/tls/certs`
|
||||
Lists every certificate this node knows about: the ones stored under `/opt/sshpanel/certs/`, the ones referenced by `tls_forwarders`, and the ones referenced by Xray inbound `tlsSettings` (inbounds that enable TLS without naming a certificate are reported against the first TLS forwarder's material, which is what `buildInboundTLS` falls back to).
|
||||
- `200`: `{ "certs_dir": string, "certs": [ { "name", "cert_file", "key_file", "managed", "exists", "subject", "issuer", "domains": [string], "not_before", "not_after", "days_left", "expired", "expiring", "self_signed", "chain_length", "key_type", "key_ok", "modified", "error", "used_by": [ { "kind": "tls_forwarder"|"xray_inbound", "ref": string } ] } ] }`.
|
||||
|
||||
#### `POST /api/tls/certs/update`
|
||||
Replaces a certificate's `fullchain.pem` + `privkey.pem`. The panel's **Configuração → TLS → Certificados TLS** card uses this for renewals.
|
||||
- Body: `fullchain` (string, required — PEM; `cert` accepted as alias), `privkey` (string, required — PEM; `key` accepted as alias), plus **either** `cert_file` (+ optional `key_file`) to replace an existing certificate in place, **or** `name` to create/replace `/opt/sshpanel/certs/<name>/`. Optional `reload` (bool, default `true`) and `force` (bool, default `false`).
|
||||
- The pair is validated with `tls.X509KeyPair` before anything is written; the previous content is kept as `<file>.bak`; existing file modes are preserved; symlinked targets (certbot layout) are followed so the link structure survives.
|
||||
- `cert_file` must be inside `/opt/sshpanel/certs/` or already referenced by the running config / Xray config — this endpoint is not an arbitrary file-write primitive.
|
||||
- Because the paths do not change, no other configuration needs editing. With `reload` on, the TLS forwarders serving the certificate are rebound (established connections are untouched) and Xray is restarted if one of its inbounds uses it.
|
||||
- `200`: `{ "cert_file": string, "key_file": string, "cert": <same shape as the list entry>, "reloaded": { "tls_forwarders": [string], "xray_inbounds": [string], "xray_restarted": bool }, "warnings": [string] }`. Warnings cover a leaf-only PEM (no intermediates), a not-yet-valid certificate, a domain change versus the previous certificate, and certbot-managed paths.
|
||||
- Errors: `400` for a missing/mismatched pair, an expired certificate without `force=true`, or a path outside the allowed set; `413` for PEM over 1 MiB; `500` on write failure.
|
||||
|
||||
---
|
||||
|
||||
### Panel config
|
||||
|
||||
+73
-7
@@ -686,19 +686,18 @@ select:disabled {
|
||||
.tab-pane:not(#tab-bot)>.grid2,.tab-pane:not(#tab-bot)>#serversListView>.grid2{gap:16px}.tab-pane:not(#tab-bot) .card-hdr{padding-bottom:12px;border-bottom:1px solid rgba(148,163,184,.09)}.tab-pane:not(#tab-bot) .card-title{font-size:.96rem}.tab-pane:not(#tab-bot) .statusbar{margin-top:13px;padding-top:11px;border-top:1px solid rgba(148,163,184,.08)}
|
||||
|
||||
/* Xray visual configuration studio */
|
||||
.shared-endpoint-card{--hero-accent:139,92,246;position:relative;margin-bottom:18px;padding:20px;overflow:hidden;border:1px solid rgba(139,92,246,.22);border-radius:22px;background:radial-gradient(circle at 96% 0,rgba(139,92,246,.17),transparent 32%),rgba(8,12,20,.82)}
|
||||
.shared-endpoint-head{display:flex;align-items:flex-start;justify-content:space-between;gap:14px}.shared-endpoint-head h3,.visual-editor-heading h3{margin-top:5px;font-size:1.12rem;letter-spacing:-.02em}.shared-endpoint-head p{margin-top:5px;color:var(--muted);font-size:.75rem;line-height:1.45}.shared-endpoint-head code,.shared-route-preview code{font-family:ui-monospace,SFMono-Regular,Consolas,monospace;color:#c8bbff}
|
||||
.shared-route-preview{display:grid;grid-template-columns:1fr 48px 1fr;align-items:center;gap:8px;margin:17px 0;padding:10px;border:1px solid rgba(148,163,184,.11);border-radius:17px;background:rgba(255,255,255,.025)}.shared-route-preview span{display:flex;align-items:center;justify-content:space-between;gap:12px;padding:11px 12px;border:1px solid rgba(139,92,246,.16);border-radius:13px;background:rgba(139,92,246,.07)}.shared-route-preview strong{font-size:.78rem}.shared-route-preview code{font-size:.77rem;font-weight:900}.shared-route-preview i{height:1px;background:linear-gradient(90deg,rgba(139,92,246,.2),rgba(34,211,238,.7),rgba(139,92,246,.2));position:relative}.shared-route-preview i::after{content:"";position:absolute;right:0;top:-3px;width:7px;height:7px;border-radius:50%;background:var(--accent);box-shadow:0 0 12px var(--accent)}
|
||||
.shared-endpoint-grid{grid-template-columns:repeat(3,minmax(0,1fr));}.shared-endpoint-actions{display:flex;align-items:center;justify-content:space-between;gap:14px;margin-top:15px;padding-top:14px;border-top:1px solid rgba(148,163,184,.1)}.shared-endpoint-actions .hint{max-width:650px}
|
||||
.xray-inbound-launcher{--hero-accent:139,92,246;position:relative;display:grid;grid-template-columns:minmax(0,1fr) auto;align-items:center;gap:17px;margin-bottom:18px;padding:20px;overflow:hidden;border:1px solid rgba(139,92,246,.22);border-radius:22px;background:radial-gradient(circle at 96% 0,rgba(139,92,246,.17),transparent 32%),rgba(8,12,20,.82)}
|
||||
.xray-inbound-launcher-copy,.xray-inbound-launcher-actions,.azion-preset-summary,.xray-inbound-launcher>.hint{position:relative;z-index:1}.xray-inbound-launcher-copy h3,.visual-editor-heading h3{margin-top:5px;font-size:1.12rem;letter-spacing:-.02em}.xray-inbound-launcher-copy p{max-width:760px;margin-top:5px;color:var(--muted);font-size:.75rem;line-height:1.5}.xray-inbound-launcher-copy p strong{color:var(--text-2)}.xray-inbound-launcher-actions{display:flex;align-items:center;justify-content:flex-end;gap:8px;flex-wrap:wrap}.xray-inbound-launcher-actions .btn-soft{border-color:rgba(49,214,123,.28);background:linear-gradient(135deg,rgba(49,214,123,.16),rgba(34,211,238,.08));color:#8af0b5}
|
||||
.azion-preset-summary{grid-column:1/-1;display:grid;grid-template-columns:repeat(4,minmax(0,1fr));gap:8px;padding-top:15px;border-top:1px solid rgba(148,163,184,.1)}.azion-preset-summary span{display:flex;align-items:center;justify-content:space-between;gap:10px;padding:10px 11px;border:1px solid rgba(139,92,246,.14);border-radius:13px;background:rgba(139,92,246,.055)}.azion-preset-summary small{color:var(--muted);font-size:.65rem}.azion-preset-summary strong{color:var(--text-2);font-family:ui-monospace,SFMono-Regular,Consolas,monospace;font-size:.7rem}.xray-inbound-launcher>.hint{grid-column:1/-1}
|
||||
.legacy-xhttp-migration{display:flex;align-items:center;gap:13px;margin:-3px 0 15px;padding:13px 15px;border:1px solid rgba(49,214,123,.19);border-radius:18px;background:linear-gradient(135deg,rgba(49,214,123,.075),rgba(34,211,238,.035));color:var(--text-2)}.legacy-xhttp-icon{display:grid;place-items:center;flex:0 0 auto;width:42px;height:42px;border:1px solid rgba(49,214,123,.25);border-radius:14px;background:rgba(49,214,123,.11);color:#72e6a4;font-size:.68rem;font-weight:950;letter-spacing:.035em}.legacy-xhttp-migration strong{display:block;color:var(--text);font-size:.8rem}.legacy-xhttp-migration p{margin-top:3px;color:var(--muted);font-size:.71rem;line-height:1.5}
|
||||
.visual-config-toolbar{display:grid;grid-template-columns:180px minmax(0,1fr) auto;align-items:end;gap:14px;margin-bottom:13px;padding:13px 15px;border:1px solid rgba(148,163,184,.1);border-radius:18px;background:rgba(255,255,255,.025)}.visual-config-toolbar-copy{display:flex;flex-direction:column;gap:4px;padding-bottom:4px}.visual-config-toolbar-copy strong{font-size:.84rem}.visual-config-toolbar-copy span{color:var(--muted);font-size:.71rem}
|
||||
.visual-config-toolbar{display:grid;grid-template-columns:180px minmax(0,1fr);align-items:end;gap:14px;margin-bottom:13px;padding:13px 15px;border:1px solid rgba(148,163,184,.1);border-radius:18px;background:rgba(255,255,255,.025)}.visual-config-toolbar-copy{display:flex;flex-direction:column;gap:4px;padding-bottom:4px}.visual-config-toolbar-copy strong{font-size:.84rem}.visual-config-toolbar-copy span{color:var(--muted);font-size:.71rem}
|
||||
.visual-inbound-list{display:grid;grid-template-columns:repeat(2,minmax(0,1fr));gap:10px;margin-bottom:14px}.visual-inbound-card{position:relative;display:flex;flex-direction:column;gap:13px;min-width:0;padding:15px;border:1px solid rgba(148,163,184,.11);border-radius:18px;background:rgba(255,255,255,.027);transition:.15s ease}.visual-inbound-card:hover{border-color:rgba(139,92,246,.3);background:rgba(139,92,246,.045);transform:translateY(-1px)}.visual-inbound-card-head,.visual-inbound-meta,.visual-inbound-actions{display:flex;align-items:center;gap:8px}.visual-inbound-card-head{justify-content:space-between}.visual-inbound-name{min-width:0}.visual-inbound-name strong{display:block;overflow:hidden;text-overflow:ellipsis;color:var(--text);font-size:.84rem;white-space:nowrap}.visual-inbound-name small{display:block;margin-top:4px;color:var(--muted);font-family:ui-monospace,SFMono-Regular,Consolas,monospace;font-size:.67rem}.visual-inbound-meta{flex-wrap:wrap}.visual-inbound-meta span{padding:4px 7px;border-radius:8px;background:rgba(148,163,184,.07);color:var(--muted);font-size:.67rem}.visual-inbound-actions{justify-content:flex-end;margin-top:auto;padding-top:11px;border-top:1px solid rgba(148,163,184,.08)}
|
||||
.legacy-ssh-btn{margin-right:auto;border-color:rgba(49,214,123,.3)!important;background:linear-gradient(135deg,rgba(49,214,123,.18),rgba(34,211,238,.09))!important;color:#8af0b5!important;box-shadow:inset 0 1px 0 rgba(255,255,255,.04)}.legacy-ssh-btn:hover:not(:disabled){border-color:rgba(49,214,123,.52)!important;transform:translateY(-1px)}.legacy-ssh-btn.is-enabled:disabled{opacity:1;border-color:rgba(49,214,123,.16)!important;background:rgba(49,214,123,.07)!important;color:#72b98e!important;cursor:default}
|
||||
.visual-inbound-editor{margin:14px 0;padding:18px;border:1px solid rgba(34,211,238,.2);border-radius:22px;background:radial-gradient(circle at 100% 0,rgba(34,211,238,.09),transparent 28%),rgba(6,10,16,.86)}.visual-editor-heading{--hero-accent:34,211,238;display:flex;align-items:flex-start;justify-content:space-between;gap:12px;margin-bottom:15px;padding-bottom:13px;border-bottom:1px solid rgba(148,163,184,.1)}.visual-save-bar{position:sticky;bottom:14px;z-index:8;display:flex;align-items:center;justify-content:space-between;gap:14px;margin-top:16px;padding:13px 15px;border:1px solid rgba(139,92,246,.2);border-radius:19px;background:rgba(8,12,20,.9);box-shadow:0 18px 48px rgba(0,0,0,.35);backdrop-filter:blur(16px)}
|
||||
|
||||
@media(max-width:1180px){.workspace-overview-grid.five{grid-template-columns:repeat(3,minmax(0,1fr))}.infra-nav-shell,.workspace-nav-shell{top:78px}}
|
||||
@media(max-width:1100px){.workspace-overview-grid,.workspace-overview-grid.five{grid-template-columns:repeat(2,minmax(0,1fr))}#configSectionNav{grid-template-columns:repeat(3,minmax(0,1fr))}#xraySectionNav{grid-template-columns:repeat(2,minmax(0,1fr))}.shared-endpoint-grid{grid-template-columns:repeat(2,minmax(0,1fr));}.visual-inbound-list{grid-template-columns:1fr}}
|
||||
@media(max-width:760px){.page-hero{min-height:0;padding:20px;border-radius:22px;align-items:flex-start;flex-direction:column}.page-hero.status-hero{display:grid;grid-template-columns:1fr}.page-hero h2{font-size:1.55rem}.page-hero-pills{max-width:none;justify-content:flex-start}.page-hero-mark{width:50px;height:50px;border-radius:17px}.workspace-hero-actions{justify-content:flex-start;max-width:none;margin-top:14px}.workspace-live-status{max-width:100%}.workspace-overview-grid,.workspace-overview-grid.five{grid-template-columns:repeat(2,minmax(0,1fr));margin-top:18px}.workspace-hero-toolbar{align-items:stretch;flex-direction:column}.workspace-toolbar-actions{justify-content:flex-start}.workspace-toolbar-actions .input-sm{width:100%;max-width:none}.infra-section-nav,.workspace-section-nav{display:none}.infra-section-select,.workspace-section-select{display:block}.infra-nav-shell,.workspace-nav-shell{top:76px}.workspace-section-heading{align-items:flex-start;flex-direction:column}.workspace-section-heading>.btn,.workspace-section-heading>.card-actions{width:100%}.workspace-section-heading>.btn{justify-content:center}.settings-panel-grid{grid-template-columns:1fr}.settings-panel-grid>.settings-span-all{grid-column:auto}.shared-endpoint-card{padding:15px}.shared-endpoint-head,.shared-endpoint-actions,.visual-save-bar{align-items:flex-start;flex-direction:column}.shared-endpoint-actions .btn,.visual-save-bar .btn{width:100%}.shared-route-preview{grid-template-columns:1fr}.shared-route-preview i{width:1px;height:22px;justify-self:center}.shared-route-preview i::after{right:-3px;top:auto;bottom:0}.shared-endpoint-grid{grid-template-columns:1fr!important}.legacy-xhttp-migration{align-items:flex-start}.visual-config-toolbar{grid-template-columns:1fr;align-items:stretch}.visual-config-toolbar .btn{width:100%}.visual-inbound-actions{align-items:stretch;flex-wrap:wrap}.legacy-ssh-btn{flex:1 0 100%;margin-right:0}.panel-dialog{padding:14px}.panel-dialog-card{padding:19px}.panel-dialog-actions .btn{flex:1}.panel-toast-stack{right:14px;bottom:14px;width:calc(100vw - 28px)}}
|
||||
@media(max-width:1100px){.workspace-overview-grid,.workspace-overview-grid.five{grid-template-columns:repeat(2,minmax(0,1fr))}#configSectionNav{grid-template-columns:repeat(3,minmax(0,1fr))}#xraySectionNav{grid-template-columns:repeat(2,minmax(0,1fr))}.visual-inbound-list{grid-template-columns:1fr}}
|
||||
@media(max-width:760px){.page-hero{min-height:0;padding:20px;border-radius:22px;align-items:flex-start;flex-direction:column}.page-hero.status-hero{display:grid;grid-template-columns:1fr}.page-hero h2{font-size:1.55rem}.page-hero-pills{max-width:none;justify-content:flex-start}.page-hero-mark{width:50px;height:50px;border-radius:17px}.workspace-hero-actions{justify-content:flex-start;max-width:none;margin-top:14px}.workspace-live-status{max-width:100%}.workspace-overview-grid,.workspace-overview-grid.five{grid-template-columns:repeat(2,minmax(0,1fr));margin-top:18px}.workspace-hero-toolbar{align-items:stretch;flex-direction:column}.workspace-toolbar-actions{justify-content:flex-start}.workspace-toolbar-actions .input-sm{width:100%;max-width:none}.infra-section-nav,.workspace-section-nav{display:none}.infra-section-select,.workspace-section-select{display:block}.infra-nav-shell,.workspace-nav-shell{top:76px}.workspace-section-heading{align-items:flex-start;flex-direction:column}.workspace-section-heading>.btn,.workspace-section-heading>.card-actions{width:100%}.workspace-section-heading>.btn{justify-content:center}.settings-panel-grid{grid-template-columns:1fr}.settings-panel-grid>.settings-span-all{grid-column:auto}.xray-inbound-launcher{grid-template-columns:1fr;padding:15px}.xray-inbound-launcher-actions,.visual-save-bar{align-items:stretch;flex-direction:column}.xray-inbound-launcher-actions{justify-content:flex-start}.xray-inbound-launcher-actions .btn,.visual-save-bar .btn{width:100%}.azion-preset-summary{grid-column:auto;grid-template-columns:repeat(2,minmax(0,1fr))}.xray-inbound-launcher>.hint{grid-column:auto}.legacy-xhttp-migration{align-items:flex-start}.visual-config-toolbar{grid-template-columns:1fr;align-items:stretch}.visual-config-toolbar .btn{width:100%}.visual-inbound-actions{align-items:stretch;flex-wrap:wrap}.legacy-ssh-btn{flex:1 0 100%;margin-right:0}.panel-dialog{padding:14px}.panel-dialog-card{padding:19px}.panel-dialog-actions .btn{flex:1}.panel-toast-stack{right:14px;bottom:14px;width:calc(100vw - 28px)}}
|
||||
@media(max-width:460px){.workspace-overview-grid,.workspace-overview-grid.five{grid-template-columns:1fr}.workspace-hero-actions .btn{flex:1}.workspace-toolbar-actions .btn{flex:1}.workspace-overview-card{padding:11px 12px}}
|
||||
|
||||
/* --- Bot sales workspace --- */
|
||||
@@ -750,3 +749,70 @@ select:disabled {
|
||||
@media(max-width:1180px){.bot-overview-grid{grid-template-columns:repeat(2,minmax(0,1fr));}.bot-section-nav{grid-template-columns:repeat(3,minmax(0,1fr));}.bot-master-detail{grid-template-columns:1fr;}.bot-editor-card{position:static;}.bot-nav-shell{top:78px;}}
|
||||
@media(max-width:760px){.bot-hero{padding:20px;border-radius:22px;}.bot-hero h2{font-size:1.55rem;}.bot-hero-actions{position:relative;right:auto;top:auto;max-width:none;justify-content:flex-start;margin-top:16px;}.bot-overview-grid{grid-template-columns:1fr 1fr;margin-top:18px;}.bot-section-nav{display:none;}.bot-section-select{display:block;}.bot-nav-shell{top:76px;}.bot-config-grid,.bot-message-grid{grid-template-columns:1fr;}.bot-section-heading{align-items:flex-start;flex-direction:column;}.bot-section-heading>.card-actions{width:100%;justify-content:flex-start;}.bot-master-detail{display:block;}.bot-master-detail>.card+.card{margin-top:14px!important;}.bot-save-row{align-items:flex-start;flex-direction:column;}}
|
||||
@media(max-width:460px){.bot-overview-grid{grid-template-columns:1fr;}.bot-overview-card{padding:11px 12px;}.bot-hero-actions .btn{width:100%;}.bot-section-heading .btn{width:100%;}.bot-copy-row{align-items:stretch;flex-direction:column;}.bot-copy-row .btn{width:100%;}}
|
||||
|
||||
/* Sortable SSH user table headers */
|
||||
th[data-sort-key]{cursor:pointer;user-select:none;white-space:nowrap;transition:color .12s ease;}
|
||||
th[data-sort-key]:hover{color:var(--accent);}
|
||||
th[data-sort-key]::after{content:"";display:inline-block;width:.9em;font-size:.72em;opacity:.85;}
|
||||
th[data-sort-key].sort-asc::after{content:" \25B2";}
|
||||
th[data-sort-key].sort-desc::after{content:" \25BC";}
|
||||
|
||||
/* Shared SSH/Xray user list sorting and filtering */
|
||||
.user-list-controls{display:flex;align-items:flex-end;gap:12px;flex-wrap:wrap;margin:0 0 16px;padding:12px;border:1px solid rgba(var(--section-accent,139,92,246),.18);border-radius:16px;background:rgba(var(--section-accent,139,92,246),.055)}
|
||||
.user-list-control-group{display:flex;flex-direction:column;gap:6px;min-width:0}.user-list-control-label{color:var(--muted);font-size:.63rem;font-weight:900;letter-spacing:.09em;text-transform:uppercase}.user-list-buttons{display:flex;align-items:center;gap:5px;flex-wrap:wrap}.user-list-filter-btn{min-height:30px;padding:5px 9px;border:1px solid rgba(148,163,184,.16);border-radius:10px;background:rgba(255,255,255,.025);color:var(--muted);font-size:.68rem;font-weight:850;cursor:pointer;transition:.15s ease}.user-list-filter-btn:hover{color:var(--text);border-color:rgba(var(--section-accent,139,92,246),.38);background:rgba(var(--section-accent,139,92,246),.09)}.user-list-filter-btn.active{color:#fff;border-color:rgba(var(--section-accent,139,92,246),.44);background:linear-gradient(135deg,rgba(var(--section-accent,139,92,246),.3),rgba(34,211,238,.1));box-shadow:inset 0 1px 0 rgba(255,255,255,.06)}.user-list-filter-btn[data-direction]::after{margin-left:4px;font-size:.7em}.user-list-filter-btn[data-direction="asc"]::after{content:"\25B2"}.user-list-filter-btn[data-direction="desc"]::after{content:"\25BC"}.user-list-count{margin-left:auto;white-space:nowrap}
|
||||
@media(max-width:760px){.user-list-controls{align-items:stretch}.user-list-control-group{width:100%}.user-list-buttons{display:grid;grid-template-columns:repeat(3,minmax(0,1fr))}.user-list-filter-btn{width:100%}.user-list-count{margin-left:0;align-self:flex-start}}
|
||||
|
||||
/* Live per-account speed (whole account, all connections summed) */
|
||||
.speed-cell{display:inline-flex;flex-direction:column;gap:1px;line-height:1.25;font-variant-numeric:tabular-nums;font-weight:850;white-space:nowrap;}
|
||||
.speed-cell .speed-down{color:var(--accent-3);}
|
||||
.speed-cell .speed-up{color:var(--accent);}
|
||||
|
||||
/* Card tables: when the space left for these wide user lists is too small to
|
||||
show every column, the table stops scrolling sideways and each row becomes a
|
||||
labelled card. Labels come from each cell's data-label.
|
||||
The switch is driven by the card's own width instead of the viewport, so it
|
||||
also catches 1366/1440-class monitors, where the sidebar plus paddings leave
|
||||
the table ~900px and the last columns end up cut off. */
|
||||
.card:has(table.table-cards){container-type:inline-size;container-name:usertbl;}
|
||||
|
||||
@container usertbl (max-width:1120px){
|
||||
.tbl-wrap:has(table.table-cards){overflow:visible;border:0;border-radius:0;background:transparent;}
|
||||
table.table-cards{display:block;min-width:0;width:100%;font-size:.78rem;}
|
||||
table.table-cards thead{display:none;}
|
||||
table.table-cards tbody{display:flex;flex-direction:column;gap:10px;}
|
||||
table.table-cards tr{
|
||||
display:grid;grid-template-columns:repeat(4,minmax(0,1fr));gap:8px 12px;
|
||||
padding:13px 14px;border:1px solid rgba(148,163,184,.14);border-radius:18px;
|
||||
background:rgba(3,6,10,.55);
|
||||
}
|
||||
table.table-cards tbody tr:hover{background:rgba(34,211,238,.05);}
|
||||
table.table-cards td{
|
||||
display:flex;flex-direction:column;gap:3px;min-width:0;
|
||||
padding:0;border:0;font-size:.78rem!important;overflow-wrap:anywhere;
|
||||
}
|
||||
table.table-cards td::before{
|
||||
content:attr(data-label);color:var(--muted);font-size:.6rem;font-weight:900;
|
||||
letter-spacing:.1em;text-transform:uppercase;
|
||||
}
|
||||
table.table-cards td[colspan]{grid-column:1/-1;text-align:center;}
|
||||
table.table-cards td:not([data-label])::before{display:none;}
|
||||
table.table-cards td.cell-primary{grid-column:1/-1;font-size:.98rem!important;font-weight:900;color:var(--text);}
|
||||
table.table-cards td.cell-primary::before{display:none;}
|
||||
table.table-cards td.cell-wide{grid-column:span 2;}
|
||||
table.table-cards td.cell-actions{
|
||||
grid-column:1/-1;flex-direction:row;flex-wrap:wrap;gap:6px;
|
||||
padding-top:4px;white-space:normal!important;
|
||||
}
|
||||
table.table-cards td.cell-actions::before{display:none;}
|
||||
table.table-cards td.cell-actions .btn{margin:0!important;min-height:36px;}
|
||||
table.table-cards .table-meter{max-width:none;}
|
||||
table.table-cards .speed-cell{flex-direction:row;gap:12px;}
|
||||
}
|
||||
@container usertbl (max-width:860px){
|
||||
table.table-cards tr{grid-template-columns:repeat(3,minmax(0,1fr));}
|
||||
}
|
||||
@container usertbl (max-width:560px){
|
||||
table.table-cards tr{grid-template-columns:repeat(2,minmax(0,1fr));}
|
||||
table.table-cards td.cell-wide{grid-column:1/-1;}
|
||||
table.table-cards td.cell-actions .btn{flex:1 1 auto;}
|
||||
}
|
||||
|
||||
@@ -172,6 +172,24 @@ Object.assign(I18N_TEXT["pt-BR"], {
|
||||
"Reseller areas":"Áreas de revendedores","Reseller area":"Área de revendedores","Create reseller":"Criar revendedor","Edit reseller":"Editar revendedor","Registered resellers":"Revendedores cadastrados","Reseller list section copy":"Consulte cotas, consumo compartilhado, validade e situação de cada parceiro.","Reseller create section copy":"Defina login, limite compartilhado, validade e acesso em uma tela dedicada.","Reseller saved successfully.":"Revendedor salvo com sucesso.",
|
||||
"Configuration areas":"Áreas de configuração","Configuration area":"Área de configuração","Network and SSH":"Rede e SSH","SlowDNS / DNSTT":"SlowDNS / DNSTT","TLS forwarders":"Encaminhadores TLS","01 · Base":"01 · Base","02 · DNS tunnel":"02 · Túnel DNS","03 · UDP":"03 · UDP","04 · Security":"04 · Segurança","05 · Core":"05 · Core","Network and SSH section copy":"Configure listeners, limites padrão, tempo ocioso e o banner de conexão.","SlowDNS section copy":"Gerencie domínios, DNS local, capacidade, filas e reinício controlado.","UDP section copy":"Defina listener, capacidade, expiração de mapa e reinício do serviço.","TLS section copy":"Crie listeners TLS com certificado automático, colado ou armazenado em arquivo.","Xray core section copy":"Ative o core, escolha o runtime e aplique os ajustes nativos seguros."
|
||||
});
|
||||
Object.assign(I18N_TEXT["en-US"], {
|
||||
"Reset":"Reset","Reset traffic":"Reset traffic","Reset SSH traffic":"Reset SSH traffic","Reset Xray traffic":"Reset Xray traffic",
|
||||
"Reset traffic for user \"{name}\"?":"Reset traffic for user \"{name}\"?","Reset traffic for client {id}…?":"Reset traffic for client {id}…?",
|
||||
"Current uploaded and downloaded usage will return to zero. The account, password, expiry and quota remain unchanged.":"Current uploaded and downloaded usage will return to zero. The account, password, expiry and quota remain unchanged.",
|
||||
"Current uploaded and downloaded usage will return to zero. The account, expiry and quota remain unchanged.":"Current uploaded and downloaded usage will return to zero. The account, expiry and quota remain unchanged.",
|
||||
"Resetting traffic for {name}…":"Resetting traffic for {name}…","Resetting traffic for client {id}…":"Resetting traffic for client {id}…",
|
||||
"Traffic reset successfully.":"Traffic reset successfully.","Xray traffic reset successfully.":"Xray traffic reset successfully.",
|
||||
"Could not reset traffic: {error}":"Could not reset traffic: {error}","Reset traffic counter":"Reset traffic counter"
|
||||
});
|
||||
Object.assign(I18N_TEXT["pt-BR"], {
|
||||
"Reset":"Zerar","Reset traffic":"Zerar tráfego","Reset SSH traffic":"Zerar tráfego SSH","Reset Xray traffic":"Zerar tráfego Xray",
|
||||
"Reset traffic for user \"{name}\"?":"Zerar o tráfego do usuário \"{name}\"?","Reset traffic for client {id}…?":"Zerar o tráfego do cliente {id}…?",
|
||||
"Current uploaded and downloaded usage will return to zero. The account, password, expiry and quota remain unchanged.":"O consumo de upload e download voltará para zero. A conta, senha, validade e cota não serão alteradas.",
|
||||
"Current uploaded and downloaded usage will return to zero. The account, expiry and quota remain unchanged.":"O consumo de upload e download voltará para zero. A conta, validade e cota não serão alteradas.",
|
||||
"Resetting traffic for {name}…":"Zerando o tráfego de {name}…","Resetting traffic for client {id}…":"Zerando o tráfego do cliente {id}…",
|
||||
"Traffic reset successfully.":"Tráfego zerado com sucesso.","Xray traffic reset successfully.":"Tráfego Xray zerado com sucesso.",
|
||||
"Could not reset traffic: {error}":"Não foi possível zerar o tráfego: {error}","Reset traffic counter":"Zerar contador de tráfego"
|
||||
});
|
||||
const I18N_REVERSE = Object.fromEntries(SUPPORTED_LANGS.map(lang => [lang, Object.fromEntries(Object.entries(I18N_TEXT[lang] || {}).map(([k, v]) => [v, k]))]));
|
||||
let currentLang = detectInitialLanguage();
|
||||
let i18nTranslating = false;
|
||||
@@ -259,6 +277,8 @@ function applyLanguage(lang, options = {}) {
|
||||
if (languageSelect) languageSelect.value = currentLang;
|
||||
updatePageHeading();
|
||||
translateStatic(document.body);
|
||||
if (typeof window.renderSSHListControls === "function") window.renderSSHListControls();
|
||||
if (typeof window.renderXrayListControls === "function") window.renderXrayListControls();
|
||||
document.documentElement.classList.remove("i18n-pending");
|
||||
}
|
||||
|
||||
@@ -683,7 +703,40 @@ function clientTrafficHTML(c) {
|
||||
const up = Number(c.uplink_bytes || 0);
|
||||
const down = Number(c.downlink_bytes || 0);
|
||||
const total = Number(c.total_bytes || (up + down) || 0);
|
||||
return `${escapeHTML(formatBytes(total))}<div class="hint">↑ ${escapeHTML(formatBytes(up))} · ↓ ${escapeHTML(formatBytes(down))}</div>`;
|
||||
const quota = Number(c.data_quota_bytes || 0);
|
||||
const quotaLabel = quota > 0 ? formatBytes(quota) : "∞";
|
||||
const state = c.quota_exceeded
|
||||
? (c.quota_action === "throttle" ? ` · ${t("throttled")}` : ` · ${t("blocked")}`)
|
||||
: "";
|
||||
return `${escapeHTML(formatBytes(total))} / ${escapeHTML(quotaLabel)}${escapeHTML(state)}<div class="hint">↑ ${escapeHTML(formatBytes(up))} · ↓ ${escapeHTML(formatBytes(down))}</div>`;
|
||||
}
|
||||
|
||||
// ─── Live bandwidth ───────────────────────────────────────────────────────────
|
||||
// The API reports the account's current speed in bytes per second, summed over
|
||||
// every connection it has open. Speeds are shown in bits per second because
|
||||
// that is the unit the per-user limits use.
|
||||
function formatSpeed(bytesPerSec) {
|
||||
const bits = Number(bytesPerSec || 0) * 8;
|
||||
if (!Number.isFinite(bits) || bits < 1000) return "0";
|
||||
if (bits < 1e6) return `${Math.round(bits / 1e3)} kbps`;
|
||||
if (bits < 1e9) return `${(bits / 1e6).toFixed(bits < 1e7 ? 2 : 1)} Mbps`;
|
||||
return `${(bits / 1e9).toFixed(2)} Gbps`;
|
||||
}
|
||||
|
||||
function isIdleSpeed(upBytesPerSec, downBytesPerSec) {
|
||||
return Number(upBytesPerSec || 0) * 8 < 1000 && Number(downBytesPerSec || 0) * 8 < 1000;
|
||||
}
|
||||
|
||||
function speedHTML(upBytesPerSec, downBytesPerSec) {
|
||||
if (isIdleSpeed(upBytesPerSec, downBytesPerSec)) return `<span class="hint">${t("idle")}</span>`;
|
||||
return `<span class="speed-cell">`
|
||||
+ `<span class="speed-down">↓ ${escapeHTML(formatSpeed(downBytesPerSec))}</span>`
|
||||
+ `<span class="speed-up">↑ ${escapeHTML(formatSpeed(upBytesPerSec))}</span>`
|
||||
+ `</span>`;
|
||||
}
|
||||
|
||||
function speedTotalBytesPerSec(entry) {
|
||||
return Number(entry?.up_bytes_per_sec || 0) + Number(entry?.down_bytes_per_sec || 0);
|
||||
}
|
||||
|
||||
function updateCell(row, name, html) {
|
||||
@@ -721,9 +774,50 @@ function patchRenderedInbounds(inbounds) {
|
||||
updateCell(row, "expiry", escapeHTML(clientExpiryLabel(c)));
|
||||
updateCell(row, "status", clientStatusHTML(c));
|
||||
updateCell(row, "online", clientOnlineHTML(c));
|
||||
updateCell(row, "connections", escapeHTML(c.active_connections || 0));
|
||||
updateCell(row, "speed", speedHTML(c.up_bytes_per_sec, c.down_bytes_per_sec));
|
||||
updateCell(row, "traffic", clientTrafficHTML(c));
|
||||
updateCell(row, "max", escapeHTML(c.max_conns || "∞"));
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
// Native XHTTP/VPN tuning labels introduced by the high-traffic backpressure
|
||||
// update. Keep this block close to the UI code so both languages stay complete.
|
||||
Object.assign(I18N_TEXT["en-US"], {
|
||||
"Sort by":"Sort by","Show":"Show","All":"All","Offline":"Offline","Connections":"Connections","Usage":"Usage","Quota reached":"Quota reached","{visible} of {total} users":"{visible} of {total} users","No Xray users match this filter.":"No Xray users match this filter.","No SSH users match this filter.":"No SSH users match this filter.",
|
||||
"Native Xray scale tuning":"Native Xray scale tuning",
|
||||
"Go CPU threads (GOMAXPROCS)":"Go CPU threads (GOMAXPROCS)",
|
||||
"Global mux backend sessions":"Global mux backend sessions",
|
||||
"Transport and XHTTP admission":"Transport and XHTTP admission",
|
||||
"Unlimited for VPN traffic. There is no global HTTP request, HTTP/2 stream, transport-connection, or XHTTP-session count cap.":"Unlimited for VPN traffic. There is no global HTTP request, HTTP/2 stream, transport-connection, or XHTTP-session count cap.",
|
||||
"Trace every XHTTP/mux packet":"Trace every XHTTP/mux packet",
|
||||
"debug only, slows QUIC":"debug only, slows QUIC",
|
||||
"Apply high-traffic VPN defaults":"Apply high-traffic VPN defaults",
|
||||
"Apply safe defaults":"Apply safe defaults",
|
||||
"XHTTP is handled as VPN tunnel traffic: packet requests and reassembly are limited only by bounded byte backpressure, never by a request count. Existing saved web-style caps are ignored automatically after update. Per-user max_conns, quota, and bandwidth policies still work normally.":"XHTTP is handled as VPN tunnel traffic: packet requests and reassembly are limited only by bounded byte backpressure, never by a request count. Existing saved web-style caps are ignored automatically after update. Per-user max_conns, quota, and bandwidth policies still work normally."
|
||||
});
|
||||
Object.assign(I18N_TEXT["pt-BR"], {
|
||||
"Sort by":"Ordenar por","Show":"Mostrar","All":"Todos","Offline":"Offline","Connections":"Conexões","Usage":"Uso","Quota reached":"Cota atingida","{visible} of {total} users":"{visible} de {total} usuários","No Xray users match this filter.":"Nenhum usuário Xray corresponde a este filtro.","No SSH users match this filter.":"Nenhum usuário SSH corresponde a este filtro.",
|
||||
"Native Xray scale tuning":"Ajustes de escala do Xray nativo",
|
||||
"Go CPU threads (GOMAXPROCS)":"Threads de CPU do Go (GOMAXPROCS)",
|
||||
"Global mux backend sessions":"Sessões globais de backend Mux",
|
||||
"Transport and XHTTP admission":"Admissão de transporte e XHTTP",
|
||||
"Unlimited for VPN traffic. There is no global HTTP request, HTTP/2 stream, transport-connection, or XHTTP-session count cap.":"Ilimitado para tráfego VPN. Não existe limite global por quantidade de requisições HTTP, streams HTTP/2, conexões de transporte ou sessões XHTTP.",
|
||||
"Trace every XHTTP/mux packet":"Registrar cada pacote XHTTP/Mux",
|
||||
"debug only, slows QUIC":"somente debug, reduz a velocidade do QUIC",
|
||||
"Apply high-traffic VPN defaults":"Aplicar padrão VPN de alto tráfego",
|
||||
"Apply safe defaults":"Aplicar padrões seguros",
|
||||
"XHTTP is handled as VPN tunnel traffic: packet requests and reassembly are limited only by bounded byte backpressure, never by a request count. Existing saved web-style caps are ignored automatically after update. Per-user max_conns, quota, and bandwidth policies still work normally.":"O XHTTP é tratado como tráfego de túnel VPN: requisições de pacotes e remontagem usam somente backpressure com limite de bytes, nunca limite por quantidade de requisições. Limites web antigos já salvos são ignorados automaticamente após a atualização. As regras por usuário de max_conns, cota e banda continuam funcionando normalmente."
|
||||
});
|
||||
|
||||
// Live per-account bandwidth column, shared by the SSH and Xray user lists.
|
||||
Object.assign(I18N_TEXT["en-US"], {
|
||||
"Speed":"Speed", "Limit up":"Limit up", "Limit down":"Limit down",
|
||||
"Current up/down speed of the whole account, across all of its connections.":"Current up/down speed of the whole account, across all of its connections.",
|
||||
});
|
||||
Object.assign(I18N_TEXT["pt-BR"], {
|
||||
"Speed":"Velocidade", "Limit up":"Limite de envio", "Limit down":"Limite de download",
|
||||
"Current up/down speed of the whole account, across all of its connections.":"Velocidade atual de envio/recebimento da conta inteira, somando todas as conexões.",
|
||||
});
|
||||
|
||||
@@ -97,6 +97,7 @@ function setWorkspaceSection(workspace, section, options = {}) {
|
||||
if (!options.silent) {
|
||||
if (workspace === "xray" && section === "config" && currentRole === "superadmin" && typeof loadWizardFromConfig === "function") loadWizardFromConfig();
|
||||
if (workspace === "xray" && section === "logs" && currentRole === "superadmin" && typeof loadXrayLogs === "function") loadXrayLogs();
|
||||
if (workspace === "config" && section === "tls" && currentRole === "superadmin" && typeof loadTLSCertificates === "function") loadTLSCertificates();
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
+251
-22
@@ -9,6 +9,13 @@ cancelUserBtn.addEventListener("click", () => {
|
||||
function prepareNewSSHUser() {
|
||||
userForm.reset();
|
||||
fTotpPeriod.value = 60; fTotpWindow.value = 1; fTotpDigits.value = 6;
|
||||
// SSH and SSH-over-XHTTP plans normally remain connected after quota and
|
||||
// fall back to the configured post-quota speed. Existing users keep their
|
||||
// saved action when edited.
|
||||
fQuotaAction.value = "throttle";
|
||||
fQuotaThrottle.value = 1;
|
||||
fUsageDisplay.value = "0 B";
|
||||
fResetUsage.checked = false;
|
||||
const heading = document.getElementById("userFormHeading");
|
||||
const title = document.getElementById("userFormTitle");
|
||||
if (heading) heading.textContent = t("Create user");
|
||||
@@ -57,52 +64,232 @@ async function loadUsersSilent() {
|
||||
}
|
||||
}
|
||||
|
||||
// ---- Column sorting (click a header to sort) ----
|
||||
// Value extractor per sortable column. Numbers sort numerically, strings
|
||||
// alphabetically; online counts as 1 so "status" groups online users together.
|
||||
const USER_SORT_EXTRACT = {
|
||||
username: u => String(u.username || "").toLowerCase(),
|
||||
status: u => (u.active_conns || 0) > 0 ? 1 : 0,
|
||||
auth: u => u.use_pam ? "pam" : (u.totp_enabled ? (u.allow_static_password ? "totp+pw" : "totp") : "password"),
|
||||
conn: u => u.active_conns || 0,
|
||||
max: u => u.max_connections || 0,
|
||||
up: u => u.limit_mbps_up || 0,
|
||||
down: u => u.limit_mbps_down || 0,
|
||||
speed: u => speedTotalBytesPerSec(u),
|
||||
usage: u => Number(u.total_bytes || ((u.total_uplink_bytes || 0) + (u.total_downlink_bytes || 0)) || 0),
|
||||
expires: u => u.expires_at ? new Date(u.expires_at).getTime() : Infinity,
|
||||
owner: u => String(u.owner_username || "").toLowerCase(),
|
||||
};
|
||||
// Columns that default to descending on first click (most/online first).
|
||||
const USER_SORT_DEFAULT_DESC = new Set(["status", "conn", "max", "up", "down", "speed", "usage"]);
|
||||
const USER_SORT_OPTIONS = [
|
||||
["username", "User"], ["status", "Status"], ["auth", "Auth"], ["conn", "Connections"],
|
||||
["speed", "Speed"], ["usage", "Usage"], ["expires", "Expiry"], ["max", "Max"],
|
||||
["up", "Up"], ["down", "Dn"], ["owner", "Owner"],
|
||||
];
|
||||
const USER_FILTER_OPTIONS = [
|
||||
["all", "All"], ["online", "Online"], ["offline", "Offline"],
|
||||
["active", "Active"], ["expired", "Expired"], ["quota", "Quota reached"],
|
||||
];
|
||||
|
||||
let userSort = { key: "username", dir: "asc" };
|
||||
let userFilter = "all";
|
||||
let lastUsersData = [];
|
||||
|
||||
function userMatchesFilter(user) {
|
||||
const online = Number(user.active_conns || 0) > 0;
|
||||
const expired = isExpiredDate(user.expires_at);
|
||||
switch (userFilter) {
|
||||
case "online": return online;
|
||||
case "offline": return !online;
|
||||
case "active": return !expired;
|
||||
case "expired": return expired;
|
||||
case "quota": return !!user.quota_exceeded;
|
||||
default: return true;
|
||||
}
|
||||
}
|
||||
|
||||
function sortUsers(list) {
|
||||
const ext = USER_SORT_EXTRACT[userSort.key] || USER_SORT_EXTRACT.username;
|
||||
const dir = userSort.dir === "desc" ? -1 : 1;
|
||||
return list.slice().sort((a, b) => {
|
||||
const va = ext(a), vb = ext(b);
|
||||
let cmp;
|
||||
if (typeof va === "number" && typeof vb === "number") cmp = va - vb;
|
||||
else cmp = String(va).localeCompare(String(vb));
|
||||
// Stable tie-break by username so equal rows never shuffle between polls.
|
||||
if (cmp === 0) cmp = String(a.username || "").localeCompare(String(b.username || ""));
|
||||
return cmp * dir;
|
||||
});
|
||||
}
|
||||
|
||||
function updateSortIndicators() {
|
||||
const table = usersBody && usersBody.closest("table");
|
||||
if (!table) return;
|
||||
table.querySelectorAll("th[data-sort-key]").forEach(th => {
|
||||
th.classList.remove("sort-asc", "sort-desc");
|
||||
if (th.getAttribute("data-sort-key") === userSort.key) {
|
||||
th.classList.add(userSort.dir === "asc" ? "sort-asc" : "sort-desc");
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function renderSSHListControls() {
|
||||
const sortLabel = document.getElementById("sshSortLabel");
|
||||
const filterLabel = document.getElementById("sshFilterLabel");
|
||||
const sortButtons = document.getElementById("sshSortButtons");
|
||||
const filterButtons = document.getElementById("sshFilterButtons");
|
||||
const count = document.getElementById("sshListCount");
|
||||
if (sortLabel) sortLabel.textContent = t("Sort by");
|
||||
if (filterLabel) filterLabel.textContent = t("Show");
|
||||
if (sortButtons) {
|
||||
const options = USER_SORT_OPTIONS.filter(([key]) => key !== "owner" || currentRole === "superadmin");
|
||||
sortButtons.replaceChildren(...options.map(([key, label]) => {
|
||||
const button = document.createElement("button");
|
||||
button.type = "button";
|
||||
button.className = "user-list-filter-btn" + (userSort.key === key ? " active" : "");
|
||||
button.textContent = t(label);
|
||||
button.setAttribute("aria-pressed", userSort.key === key ? "true" : "false");
|
||||
if (userSort.key === key) button.dataset.direction = userSort.dir;
|
||||
button.addEventListener("click", () => setUserSort(key));
|
||||
return button;
|
||||
}));
|
||||
}
|
||||
if (filterButtons) {
|
||||
filterButtons.replaceChildren(...USER_FILTER_OPTIONS.map(([key, label]) => {
|
||||
const button = document.createElement("button");
|
||||
button.type = "button";
|
||||
button.className = "user-list-filter-btn" + (userFilter === key ? " active" : "");
|
||||
button.textContent = t(label);
|
||||
button.setAttribute("aria-pressed", userFilter === key ? "true" : "false");
|
||||
button.addEventListener("click", () => setUserFilter(key));
|
||||
return button;
|
||||
}));
|
||||
}
|
||||
if (count) {
|
||||
const visible = lastUsersData.filter(userMatchesFilter).length;
|
||||
count.textContent = t("{visible} of {total} users", {visible, total:lastUsersData.length});
|
||||
}
|
||||
}
|
||||
|
||||
function setUserSort(key) {
|
||||
if (!USER_SORT_EXTRACT[key]) return;
|
||||
if (userSort.key === key) {
|
||||
userSort.dir = userSort.dir === "asc" ? "desc" : "asc";
|
||||
} else {
|
||||
userSort.key = key;
|
||||
userSort.dir = USER_SORT_DEFAULT_DESC.has(key) ? "desc" : "asc";
|
||||
}
|
||||
renderUsers(lastUsersData);
|
||||
}
|
||||
|
||||
function setUserFilter(filter) {
|
||||
if (!USER_FILTER_OPTIONS.some(([key]) => key === filter)) return;
|
||||
userFilter = filter;
|
||||
renderUsers(lastUsersData);
|
||||
}
|
||||
|
||||
(function initUserSortHeaders() {
|
||||
const table = usersBody && usersBody.closest("table");
|
||||
if (!table) return;
|
||||
table.querySelectorAll("th[data-sort-key]").forEach(th => {
|
||||
th.addEventListener("click", () => setUserSort(th.getAttribute("data-sort-key")));
|
||||
});
|
||||
updateSortIndicators();
|
||||
})();
|
||||
|
||||
renderSSHListControls();
|
||||
|
||||
function sshTrafficHTML(u) {
|
||||
const up = Number(u.total_uplink_bytes || 0);
|
||||
const down = Number(u.total_downlink_bytes || 0);
|
||||
const total = Number(u.total_bytes || (up + down) || 0);
|
||||
const quota = Number(u.data_quota_bytes || 0);
|
||||
const quotaLabel = quota > 0 ? formatBytes(quota) : "∞";
|
||||
const state = u.quota_exceeded
|
||||
? (u.quota_action === "throttle" ? ` · ${t("throttled")}` : ` · ${t("blocked")}`)
|
||||
: "";
|
||||
return `${escapeHTML(formatBytes(total))} / ${escapeHTML(quotaLabel)}${escapeHTML(state)}<div class="hint">↑ ${escapeHTML(formatBytes(up))} · ↓ ${escapeHTML(formatBytes(down))}</div>`;
|
||||
}
|
||||
|
||||
function renderUsers(users) {
|
||||
updateDashboardFromUsers(users);
|
||||
// Cache the raw list so a header click can re-sort without refetching, and
|
||||
// order by the active column so rows don't shuffle on each live poll.
|
||||
lastUsersData = Array.isArray(users) ? users : [];
|
||||
users = sortUsers(lastUsersData.filter(userMatchesFilter));
|
||||
updateDashboardFromUsers(lastUsersData);
|
||||
renderSSHListControls();
|
||||
updateSortIndicators();
|
||||
const isSA = currentRole === "superadmin";
|
||||
userCountChip.textContent = users.length;
|
||||
if (isSA) ownerColHead.classList.remove("hidden");
|
||||
ownerColHead.classList.toggle("hidden", !isSA);
|
||||
usersBody.innerHTML = "";
|
||||
let online = 0;
|
||||
let expiredCount = 0;
|
||||
const online = lastUsersData.filter(u => Number(u.active_conns || 0) > 0).length;
|
||||
const expiredCount = lastUsersData.filter(u => isExpiredDate(u.expires_at)).length;
|
||||
users.forEach(u => {
|
||||
const on = (u.active_conns || 0) > 0;
|
||||
if (on) online++;
|
||||
if (isExpiredDate(u.expires_at)) expiredCount++;
|
||||
const tr = document.createElement("tr");
|
||||
// Every cell carries its column label so the table can collapse into
|
||||
// labelled cards on phones instead of scrolling sideways. "wide" cells span
|
||||
// the full card width there.
|
||||
const cells = [
|
||||
u.username,
|
||||
on ? `<span class="badge-on">${t("online")}</span>` : `<span class="badge-off">${t("idle")}</span>`,
|
||||
u.totp_enabled ? (u.allow_static_password ? "TOTP+pw" : "TOTP") : "Password",
|
||||
u.active_conns ?? 0,
|
||||
u.max_connections || 0,
|
||||
u.limit_mbps_up || 0,
|
||||
u.limit_mbps_down || 0,
|
||||
u.expires_at ? fmtDate(u.expires_at) : "—",
|
||||
{ label:"User", text:u.username, cls:"cell-primary" },
|
||||
{ label:"Status", html: on ? `<span class="badge-on">${t("online")}</span>` : `<span class="badge-off">${t("idle")}</span>` },
|
||||
{ label:"Auth", text: u.use_pam ? "PAM" : (u.totp_enabled ? (u.allow_static_password ? "TOTP+pw" : "TOTP") : "Password") },
|
||||
{ label:"Conn", text: String(u.active_conns ?? 0) },
|
||||
{ label:"Max", text: String(u.max_connections || 0) },
|
||||
// "Up"/"Dn" are speed limits, not current speed: spell that out on the
|
||||
// card layout where the label sits right next to the live speed.
|
||||
{ label:"Up", cardLabel:"Limit up", text: String(u.limit_mbps_up || 0) },
|
||||
{ label:"Dn", cardLabel:"Limit down", text: String(u.limit_mbps_down || 0) },
|
||||
{ label:"Speed", html: speedHTML(u.up_bytes_per_sec, u.down_bytes_per_sec), small:true, cls:"cell-wide" },
|
||||
{ label:"Traffic", html: sshTrafficHTML(u), small:true, cls:"cell-wide" },
|
||||
{ label:"Expires", text: u.expires_at ? fmtDate(u.expires_at) : "—" },
|
||||
];
|
||||
if (isSA) cells.push(u.owner_username || "—");
|
||||
cells.forEach((c, i) => {
|
||||
if (isSA) cells.push({ label:"Owner", text: u.owner_username || "—" });
|
||||
cells.forEach(cell => {
|
||||
const td = document.createElement("td");
|
||||
if (i === 1) td.innerHTML = c; else td.textContent = c;
|
||||
td.dataset.label = t(cell.cardLabel || cell.label);
|
||||
if (cell.cls) td.className = cell.cls;
|
||||
if (cell.html !== undefined) td.innerHTML = cell.html;
|
||||
else td.textContent = cell.text ?? "—";
|
||||
if (cell.small) td.style.fontSize = ".7rem";
|
||||
tr.appendChild(td);
|
||||
});
|
||||
const tdA = document.createElement("td");
|
||||
tdA.dataset.label = t("Actions");
|
||||
tdA.className = "cell-actions";
|
||||
const editBtn = Object.assign(document.createElement("button"), {
|
||||
className:"btn btn-ghost btn-sm", textContent:t("Edit"),
|
||||
onclick: () => fillUserForm(u),
|
||||
});
|
||||
const resetBtn = Object.assign(document.createElement("button"), {
|
||||
className:"btn btn-warn btn-sm", textContent:t("Reset"),
|
||||
style: "margin-left:4px;",
|
||||
title: t("Reset traffic"),
|
||||
onclick: () => resetUserTraffic(u.username, resetBtn),
|
||||
});
|
||||
const delBtn = Object.assign(document.createElement("button"), {
|
||||
className:"btn btn-danger btn-sm", textContent:t("Del"),
|
||||
style: "margin-left:4px;",
|
||||
onclick: () => deleteUser(u.username),
|
||||
});
|
||||
tdA.append(editBtn, delBtn);
|
||||
tdA.append(editBtn, resetBtn, delBtn);
|
||||
tr.appendChild(tdA);
|
||||
usersBody.appendChild(tr);
|
||||
});
|
||||
const activeCount = Math.max(0, users.length - expiredCount);
|
||||
userCountChip.textContent = t("{count} total · {active} active · {online} online", {count: users.length, active: activeCount, online});
|
||||
if (sshMetricTotal) sshMetricTotal.textContent = String(users.length);
|
||||
if (!users.length) {
|
||||
const row = document.createElement("tr");
|
||||
const cell = document.createElement("td");
|
||||
cell.colSpan = isSA ? 12 : 11;
|
||||
cell.className = "hint";
|
||||
cell.style.cssText = "padding:24px;text-align:center;";
|
||||
cell.textContent = t("No SSH users match this filter.");
|
||||
row.appendChild(cell);
|
||||
usersBody.appendChild(row);
|
||||
}
|
||||
const activeCount = Math.max(0, lastUsersData.length - expiredCount);
|
||||
userCountChip.textContent = t("{count} total · {active} active · {online} online", {count:lastUsersData.length, active:activeCount, online});
|
||||
if (sshMetricTotal) sshMetricTotal.textContent = String(lastUsersData.length);
|
||||
if (sshMetricActive) sshMetricActive.textContent = String(activeCount);
|
||||
if (sshMetricOnline) sshMetricOnline.textContent = String(online);
|
||||
if (sshMetricState) sshMetricState.textContent = t("Online");
|
||||
@@ -124,6 +311,12 @@ function fillUserForm(u) {
|
||||
fMaxConn.value = u.max_connections || "";
|
||||
fUp.value = u.limit_mbps_up || "";
|
||||
fDown.value = u.limit_mbps_down || "";
|
||||
fQuotaGB.value = u.data_quota_bytes ? (Number(u.data_quota_bytes) / (1024 ** 3)).toFixed(2).replace(/\.00$/, "") : "0";
|
||||
fQuotaAction.value = u.quota_action === "throttle" ? "throttle" : "block";
|
||||
fQuotaThrottle.value = u.quota_throttle_mbps || 1;
|
||||
const totalBytes = Number(u.total_bytes || ((u.total_uplink_bytes || 0) + (u.total_downlink_bytes || 0)) || 0);
|
||||
fUsageDisplay.value = `${formatBytes(totalBytes)} (↑ ${formatBytes(u.total_uplink_bytes || 0)} · ↓ ${formatBytes(u.total_downlink_bytes || 0)})`;
|
||||
fResetUsage.checked = false;
|
||||
fExpires.value = u.expires_at ? localFromISO(u.expires_at) : "";
|
||||
const heading = document.getElementById("userFormHeading");
|
||||
const title = document.getElementById("userFormTitle");
|
||||
@@ -148,6 +341,10 @@ userForm.addEventListener("submit", async e => {
|
||||
expires_at: isoFromLocal(fExpires.value),
|
||||
limit_mbps_up: parseInt(fUp.value||"0",10),
|
||||
limit_mbps_down: parseInt(fDown.value||"0",10),
|
||||
data_quota_bytes: Math.round((parseFloat(fQuotaGB.value || "0") || 0) * (1024 ** 3)),
|
||||
quota_action: fQuotaAction.value === "throttle" ? "throttle" : "block",
|
||||
quota_throttle_mbps: parseInt(fQuotaThrottle.value || "1", 10) || 1,
|
||||
reset_usage: !!fResetUsage.checked,
|
||||
server_id: selectedSSHServer(),
|
||||
};
|
||||
try {
|
||||
@@ -155,6 +352,7 @@ userForm.addEventListener("submit", async e => {
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
userStatus.textContent = t("Saved.");
|
||||
fPassword.value = "";
|
||||
fResetUsage.checked = false;
|
||||
loadUsers();
|
||||
if (currentRole === "reseller") loadMe();
|
||||
showPanelToast(t("SSH user saved successfully."), "success", t("SSH / SlowDNS"));
|
||||
@@ -167,6 +365,37 @@ userForm.addEventListener("submit", async e => {
|
||||
}
|
||||
});
|
||||
|
||||
async function resetUserTraffic(username, button) {
|
||||
const accepted = await panelConfirm({
|
||||
tone:"warning", icon:"↺", title:t("Reset SSH traffic"),
|
||||
message:t("Reset traffic for user \"{name}\"?", {name: username}),
|
||||
detail:t("Current uploaded and downloaded usage will return to zero. The account, password, expiry and quota remain unchanged."),
|
||||
confirmLabel:t("Reset traffic"),
|
||||
});
|
||||
if (!accepted) return;
|
||||
const previousDisabled = !!button?.disabled;
|
||||
if (button) button.disabled = true;
|
||||
userStatus.textContent = t("Resetting traffic for {name}…", {name: username});
|
||||
try {
|
||||
const res = await api("/api/users/reset-traffic", {
|
||||
method:"POST",
|
||||
body: JSON.stringify({ username, server_id:selectedSSHServer() }),
|
||||
});
|
||||
if (!res.ok) throw new Error((await res.text()) || "reset failed");
|
||||
userStatus.textContent = t("Traffic reset successfully.");
|
||||
showPanelToast(t("Traffic reset successfully."), "success", t("SSH / SlowDNS"));
|
||||
await loadUsers();
|
||||
} catch (e) {
|
||||
if (e.message === "auth") doAuthError();
|
||||
else {
|
||||
userStatus.textContent = t("Could not reset traffic: {error}", {error:e.message});
|
||||
showPanelToast(t("Could not reset traffic: {error}", {error:e.message}), "error", t("SSH / SlowDNS"));
|
||||
}
|
||||
} finally {
|
||||
if (button) button.disabled = previousDisabled;
|
||||
}
|
||||
}
|
||||
|
||||
async function deleteUser(username) {
|
||||
const accepted = await panelConfirm({
|
||||
tone:"danger", icon:"×", title:t("Delete SSH account"),
|
||||
|
||||
+213
-17
@@ -18,6 +18,134 @@ document.getElementById("xCreateUUIDBtn")?.addEventListener("click", () => {
|
||||
document.getElementById("xCreateInbound")?.addEventListener("change", updateXrayCreatorInboundLabel);
|
||||
document.getElementById("xCreateClientForm")?.addEventListener("submit", submitXrayClientCreator);
|
||||
|
||||
// Keep Xray list controls consistent with the sortable SSH user table while
|
||||
// preserving the inbound grouping. Sorting is applied inside each inbound;
|
||||
// filters apply across all inbound groups.
|
||||
const XRAY_CLIENT_SORT_EXTRACT = {
|
||||
name: c => String(c.name || c.email || c.id || "").toLowerCase(),
|
||||
status: c => c.expired ? 0 : 1,
|
||||
online: c => c.online ? 1 : 0,
|
||||
connections: c => Number(c.active_connections || 0),
|
||||
speed: c => speedTotalBytesPerSec(c),
|
||||
usage: c => Number(c.total_bytes || ((c.uplink_bytes || 0) + (c.downlink_bytes || 0)) || 0),
|
||||
expiry: c => c.expires_at ? new Date(c.expires_at).getTime() : Infinity,
|
||||
max: c => Number(c.max_conns || 0),
|
||||
};
|
||||
const XRAY_CLIENT_SORT_DEFAULT_DESC = new Set(["status", "online", "connections", "speed", "usage", "max"]);
|
||||
const XRAY_CLIENT_SORT_OPTIONS = [
|
||||
["name", "Name"], ["status", "Status"], ["online", "Online"],
|
||||
["connections", "Connections"], ["speed", "Speed"], ["usage", "Usage"], ["expiry", "Expiry"], ["max", "Max"],
|
||||
];
|
||||
const XRAY_CLIENT_FILTER_OPTIONS = [
|
||||
["all", "All"], ["online", "Online"], ["offline", "Offline"],
|
||||
["active", "Active"], ["expired", "Expired"], ["quota", "Quota reached"],
|
||||
];
|
||||
let xrayClientSort = { key: "name", dir: "asc" };
|
||||
let xrayClientFilter = "all";
|
||||
let lastXrayInboundsData = [];
|
||||
|
||||
function xrayClientMatchesFilter(client) {
|
||||
switch (xrayClientFilter) {
|
||||
case "online": return !!client.online;
|
||||
case "offline": return !client.online;
|
||||
case "active": return !client.expired;
|
||||
case "expired": return !!client.expired;
|
||||
case "quota": return !!client.quota_exceeded;
|
||||
default: return true;
|
||||
}
|
||||
}
|
||||
|
||||
function sortXrayClients(clients = []) {
|
||||
const extract = XRAY_CLIENT_SORT_EXTRACT[xrayClientSort.key] || XRAY_CLIENT_SORT_EXTRACT.name;
|
||||
const direction = xrayClientSort.dir === "desc" ? -1 : 1;
|
||||
return clients.slice().sort((a, b) => {
|
||||
const left = extract(a), right = extract(b);
|
||||
let comparison;
|
||||
if (typeof left === "number" && typeof right === "number") comparison = left - right;
|
||||
else comparison = String(left).localeCompare(String(right));
|
||||
if (comparison === 0) comparison = String(a.name || a.email || a.id || "").localeCompare(String(b.name || b.email || b.id || ""));
|
||||
return comparison * direction;
|
||||
});
|
||||
}
|
||||
|
||||
function prepareXrayInboundsForList(inbounds = []) {
|
||||
return (inbounds || []).map(inbound => ({
|
||||
...inbound,
|
||||
clients: sortXrayClients((inbound.clients || []).filter(xrayClientMatchesFilter)),
|
||||
})).filter(inbound => xrayClientFilter === "all" || inbound.clients.length > 0);
|
||||
}
|
||||
|
||||
function xrayClientListCounts(inbounds = []) {
|
||||
const clients = (inbounds || []).flatMap(inbound => inbound.clients || []);
|
||||
return { total: clients.length, visible: clients.filter(xrayClientMatchesFilter).length };
|
||||
}
|
||||
|
||||
function renderXrayListControls() {
|
||||
const sortLabel = document.getElementById("xraySortLabel");
|
||||
const filterLabel = document.getElementById("xrayFilterLabel");
|
||||
const sortButtons = document.getElementById("xraySortButtons");
|
||||
const filterButtons = document.getElementById("xrayFilterButtons");
|
||||
const count = document.getElementById("xrayListCount");
|
||||
if (sortLabel) sortLabel.textContent = t("Sort by");
|
||||
if (filterLabel) filterLabel.textContent = t("Show");
|
||||
if (sortButtons) {
|
||||
sortButtons.replaceChildren(...XRAY_CLIENT_SORT_OPTIONS.map(([key, label]) => {
|
||||
const button = document.createElement("button");
|
||||
button.type = "button";
|
||||
button.className = "user-list-filter-btn" + (xrayClientSort.key === key ? " active" : "");
|
||||
button.textContent = t(label);
|
||||
button.setAttribute("aria-pressed", xrayClientSort.key === key ? "true" : "false");
|
||||
if (xrayClientSort.key === key) button.dataset.direction = xrayClientSort.dir;
|
||||
button.addEventListener("click", () => setXrayClientSort(key));
|
||||
return button;
|
||||
}));
|
||||
}
|
||||
if (filterButtons) {
|
||||
filterButtons.replaceChildren(...XRAY_CLIENT_FILTER_OPTIONS.map(([key, label]) => {
|
||||
const button = document.createElement("button");
|
||||
button.type = "button";
|
||||
button.className = "user-list-filter-btn" + (xrayClientFilter === key ? " active" : "");
|
||||
button.textContent = t(label);
|
||||
button.setAttribute("aria-pressed", xrayClientFilter === key ? "true" : "false");
|
||||
button.addEventListener("click", () => setXrayClientFilter(key));
|
||||
return button;
|
||||
}));
|
||||
}
|
||||
if (count) {
|
||||
const counts = xrayClientListCounts(lastXrayInboundsData);
|
||||
count.textContent = t("{visible} of {total} users", counts);
|
||||
}
|
||||
}
|
||||
|
||||
function setXrayClientSort(key) {
|
||||
if (!XRAY_CLIENT_SORT_EXTRACT[key]) return;
|
||||
if (xrayClientSort.key === key) xrayClientSort.dir = xrayClientSort.dir === "asc" ? "desc" : "asc";
|
||||
else {
|
||||
xrayClientSort.key = key;
|
||||
xrayClientSort.dir = XRAY_CLIENT_SORT_DEFAULT_DESC.has(key) ? "desc" : "asc";
|
||||
}
|
||||
renderInbounds(lastXrayInboundsData, { force:true, fromControls:true });
|
||||
}
|
||||
|
||||
function setXrayClientFilter(filter) {
|
||||
if (!XRAY_CLIENT_FILTER_OPTIONS.some(([key]) => key === filter)) return;
|
||||
xrayClientFilter = filter;
|
||||
renderInbounds(lastXrayInboundsData, { force:true, fromControls:true });
|
||||
}
|
||||
|
||||
function bindXrayTableSortHeaders(table) {
|
||||
table.querySelectorAll("th[data-sort-key]").forEach(header => {
|
||||
const key = header.dataset.sortKey;
|
||||
const selected = key === xrayClientSort.key;
|
||||
header.classList.toggle("sort-asc", selected && xrayClientSort.dir === "asc");
|
||||
header.classList.toggle("sort-desc", selected && xrayClientSort.dir === "desc");
|
||||
header.setAttribute("aria-sort", selected ? (xrayClientSort.dir === "asc" ? "ascending" : "descending") : "none");
|
||||
header.addEventListener("click", () => setXrayClientSort(key));
|
||||
});
|
||||
}
|
||||
|
||||
renderXrayListControls();
|
||||
|
||||
|
||||
async function loadXrayStatus() {
|
||||
if (xrayChip) {
|
||||
@@ -167,25 +295,33 @@ async function copyText(text) {
|
||||
|
||||
function renderInbounds(inbounds, options = {}) {
|
||||
const { silent = false, force = false } = options || {};
|
||||
updateDashboardXray(inbounds);
|
||||
syncXrayCreatorInbounds(inbounds);
|
||||
const nextStructure = inboundStructure(inbounds);
|
||||
lastXrayInboundsData = Array.isArray(inbounds) ? inbounds : [];
|
||||
updateDashboardXray(lastXrayInboundsData);
|
||||
syncXrayCreatorInbounds(lastXrayInboundsData);
|
||||
const listInbounds = prepareXrayInboundsForList(lastXrayInboundsData);
|
||||
const nextStructure = inboundStructure(listInbounds);
|
||||
renderXrayListControls();
|
||||
|
||||
if (silent && !force && nextStructure === lastInboundsStructure && patchRenderedInbounds(inbounds)) return;
|
||||
if (silent && !force && nextStructure === lastInboundsStructure && patchRenderedInbounds(listInbounds)) return;
|
||||
if (silent && !force && isXrayClientEditorActive()) {
|
||||
patchRenderedInbounds(inbounds);
|
||||
patchRenderedInbounds(listInbounds);
|
||||
if (xStatus) xStatus.textContent = t("New client data is available; editing was preserved.");
|
||||
return;
|
||||
}
|
||||
|
||||
if (!inbounds.length) {
|
||||
if (!lastXrayInboundsData.length) {
|
||||
inboundsContainer.innerHTML = `<div class="hint" style="padding:8px 0;">${t("No VLESS/VMess/Trojan inbounds found.")}</div>`;
|
||||
lastInboundsStructure = nextStructure;
|
||||
return;
|
||||
}
|
||||
if (!listInbounds.length) {
|
||||
inboundsContainer.innerHTML = `<div class="hint" style="padding:8px 0;">${t("No Xray users match this filter.")}</div>`;
|
||||
lastInboundsStructure = nextStructure;
|
||||
return;
|
||||
}
|
||||
inboundsContainer.innerHTML = "";
|
||||
lastInboundsStructure = nextStructure;
|
||||
inbounds.forEach(ib => {
|
||||
listInbounds.forEach(ib => {
|
||||
const section = document.createElement("div");
|
||||
section.dataset.inboundTag = String(ib.tag || "");
|
||||
section.dataset.inboundProtocol = String(ib.protocol || "");
|
||||
@@ -219,21 +355,27 @@ function renderInbounds(inbounds, options = {}) {
|
||||
tblWrap.innerHTML = `<div class="hint" style="padding:4px 0;">${t("No clients.")}</div>`;
|
||||
} else {
|
||||
const tbl = document.createElement("table");
|
||||
tbl.innerHTML = `<thead><tr><th>${t("Name")}</th><th>UUID</th><th>${t("Email")}</th><th>${t("Expiry")}</th><th>${t("Status")}</th><th>${t("Online")}</th><th>${t("Traffic")}</th><th>${t("Max")}</th><th>${t("Actions")}</th></tr></thead>`;
|
||||
tbl.className = "table-cards";
|
||||
tbl.innerHTML = `<thead><tr><th data-sort-key="name">${t("Name")}</th><th>UUID</th><th>${t("Email")}</th><th data-sort-key="expiry">${t("Expiry")}</th><th data-sort-key="status">${t("Status")}</th><th data-sort-key="online">${t("Online")}</th><th data-sort-key="connections">${t("Conn")}</th><th data-sort-key="speed" title="${escapeHTML(t("Current up/down speed of the whole account, across all of its connections."))}">${t("Speed")}</th><th data-sort-key="usage">${t("Traffic")}</th><th data-sort-key="max">${t("Max")}</th><th>${t("Actions")}</th></tr></thead>`;
|
||||
const tbody = document.createElement("tbody");
|
||||
clients.forEach(c => {
|
||||
const tr = document.createElement("tr");
|
||||
tr.dataset.clientId = String(c.id || "");
|
||||
// data-label drives the labelled card layout used on narrow screens.
|
||||
tr.innerHTML = `
|
||||
<td data-cell="name">${escapeHTML(c.name || "—")}</td>
|
||||
<td data-cell="uuid" style="font-family:monospace;font-size:.65rem;">${escapeHTML(c.id || "—")}</td>
|
||||
<td data-cell="email">${escapeHTML(c.email || "—")}</td>
|
||||
<td data-cell="expiry" style="font-size:.7rem;">${escapeHTML(clientExpiryLabel(c))}</td>
|
||||
<td data-cell="status">${clientStatusHTML(c)}</td>
|
||||
<td data-cell="online">${clientOnlineHTML(c)}</td>
|
||||
<td data-cell="traffic" style="font-size:.7rem;">${clientTrafficHTML(c)}</td>
|
||||
<td data-cell="max" style="font-size:.7rem;">${escapeHTML(c.max_conns || "∞")}</td>`;
|
||||
<td data-cell="name" data-label="${escapeHTML(t("Name"))}" class="cell-primary">${escapeHTML(c.name || "—")}</td>
|
||||
<td data-cell="uuid" data-label="UUID" class="cell-wide" style="font-family:monospace;font-size:.65rem;word-break:break-all;">${escapeHTML(c.id || "—")}</td>
|
||||
<td data-cell="email" data-label="${escapeHTML(t("Email"))}" class="cell-wide">${escapeHTML(c.email || "—")}</td>
|
||||
<td data-cell="expiry" data-label="${escapeHTML(t("Expiry"))}" style="font-size:.7rem;">${escapeHTML(clientExpiryLabel(c))}</td>
|
||||
<td data-cell="status" data-label="${escapeHTML(t("Status"))}">${clientStatusHTML(c)}</td>
|
||||
<td data-cell="online" data-label="${escapeHTML(t("Online"))}">${clientOnlineHTML(c)}</td>
|
||||
<td data-cell="connections" data-label="${escapeHTML(t("Conn"))}" style="font-size:.7rem;">${escapeHTML(c.active_connections || 0)}</td>
|
||||
<td data-cell="speed" data-label="${escapeHTML(t("Speed"))}" class="cell-wide" style="font-size:.7rem;">${speedHTML(c.up_bytes_per_sec, c.down_bytes_per_sec)}</td>
|
||||
<td data-cell="traffic" data-label="${escapeHTML(t("Traffic"))}" class="cell-wide" style="font-size:.7rem;">${clientTrafficHTML(c)}</td>
|
||||
<td data-cell="max" data-label="${escapeHTML(t("Max"))}" style="font-size:.7rem;">${escapeHTML(c.max_conns || "∞")}</td>`;
|
||||
const actTd = document.createElement("td");
|
||||
actTd.dataset.label = t("Actions");
|
||||
actTd.className = "cell-actions";
|
||||
actTd.style.whiteSpace = "nowrap";
|
||||
const copyBtn = document.createElement("button");
|
||||
copyBtn.className = "btn btn-ghost btn-sm";
|
||||
@@ -244,16 +386,23 @@ function renderInbounds(inbounds, options = {}) {
|
||||
editBtn.style.marginLeft = "4px";
|
||||
editBtn.textContent = t("Edit");
|
||||
editBtn.onclick = () => openEditXrayClient(ib.tag, c);
|
||||
const resetBtn = document.createElement("button");
|
||||
resetBtn.className = "btn btn-warn btn-sm";
|
||||
resetBtn.style.marginLeft = "4px";
|
||||
resetBtn.textContent = t("Reset");
|
||||
resetBtn.title = t("Reset traffic");
|
||||
resetBtn.onclick = () => resetXrayClientTraffic(c.id, resetBtn);
|
||||
const delBtn = document.createElement("button");
|
||||
delBtn.className = "btn btn-danger btn-sm";
|
||||
delBtn.style.marginLeft = "4px";
|
||||
delBtn.textContent = t("Del");
|
||||
delBtn.onclick = () => removeClient(ib.tag, c.id);
|
||||
actTd.append(copyBtn, editBtn, delBtn);
|
||||
actTd.append(copyBtn, editBtn, resetBtn, delBtn);
|
||||
tr.appendChild(actTd);
|
||||
tbody.appendChild(tr);
|
||||
});
|
||||
tbl.appendChild(tbody);
|
||||
bindXrayTableSortHeaders(tbl);
|
||||
tblWrap.appendChild(tbl);
|
||||
}
|
||||
section.appendChild(tblWrap);
|
||||
@@ -325,6 +474,12 @@ function prepareXrayClientCreator(preferredTag = "") {
|
||||
if (uuid) uuid.value = genUUID();
|
||||
const maxConns = document.getElementById("xCreateMaxConns");
|
||||
if (maxConns) maxConns.value = "0";
|
||||
const quotaGB = document.getElementById("xCreateQuotaGB");
|
||||
if (quotaGB) quotaGB.value = "0";
|
||||
const quotaAction = document.getElementById("xCreateQuotaAction");
|
||||
if (quotaAction) quotaAction.value = "block";
|
||||
const quotaThrottle = document.getElementById("xCreateQuotaThrottle");
|
||||
if (quotaThrottle) quotaThrottle.value = "1";
|
||||
const status = document.getElementById("xCreateClientStatus");
|
||||
if (status) status.textContent = xrayCreatorInbounds.length ? t("Ready to create a new Xray client.") : t("Waiting for a compatible inbound.");
|
||||
updateXrayCreatorInboundLabel();
|
||||
@@ -351,6 +506,9 @@ async function submitXrayClientCreator(event) {
|
||||
name: (document.getElementById("xCreateName")?.value || "").trim(),
|
||||
expires_at: isoFromLocal(document.getElementById("xCreateExpiry")?.value || ""),
|
||||
max_connections: parseInt(document.getElementById("xCreateMaxConns")?.value || "0", 10) || 0,
|
||||
data_quota_bytes: Math.round((parseFloat(document.getElementById("xCreateQuotaGB")?.value || "0") || 0) * (1024 ** 3)),
|
||||
quota_action: document.getElementById("xCreateQuotaAction")?.value === "throttle" ? "throttle" : "block",
|
||||
quota_throttle_mbps: parseInt(document.getElementById("xCreateQuotaThrottle")?.value || "1", 10) || 1,
|
||||
server_id: selectedXrayServer(),
|
||||
};
|
||||
if (button) button.disabled = true;
|
||||
@@ -385,6 +543,44 @@ async function submitXrayClientCreator(event) {
|
||||
}
|
||||
}
|
||||
|
||||
async function resetXrayClientTraffic(uuid, button) {
|
||||
const shortID = String(uuid || "").slice(0, 8);
|
||||
const accepted = await panelConfirm({
|
||||
tone:"warning", icon:"↺", title:t("Reset Xray traffic"),
|
||||
message:t("Reset traffic for client {id}…?", {id:shortID}),
|
||||
detail:t("Current uploaded and downloaded usage will return to zero. The account, expiry and quota remain unchanged."),
|
||||
confirmLabel:t("Reset traffic"),
|
||||
});
|
||||
if (!accepted) return;
|
||||
const previousDisabled = !!button?.disabled;
|
||||
if (button) button.disabled = true;
|
||||
xStatus.textContent = t("Resetting traffic for client {id}…", {id:shortID});
|
||||
try {
|
||||
const res = await api("/api/xray/clients/reset-traffic", {
|
||||
method:"POST",
|
||||
body:JSON.stringify({ uuid, server_id:selectedXrayServer() }),
|
||||
});
|
||||
if (!res.ok) throw new Error((await res.text()) || "reset failed");
|
||||
xStatus.textContent = t("Xray traffic reset successfully.");
|
||||
showPanelToast(t("Xray traffic reset successfully."), "success", t("Xray user"));
|
||||
if (editingXrayClientId === uuid) {
|
||||
const usage = document.getElementById("editXrayUsage");
|
||||
if (usage) usage.value = "0 B (↑ 0 B · ↓ 0 B)";
|
||||
const resetUsage = document.getElementById("editXrayResetUsage");
|
||||
if (resetUsage) resetUsage.checked = false;
|
||||
}
|
||||
await loadInbounds({ force:true });
|
||||
} catch (e) {
|
||||
if (e.message === "auth") doAuthError();
|
||||
else {
|
||||
xStatus.textContent = t("Could not reset traffic: {error}", {error:e.message});
|
||||
showPanelToast(t("Could not reset traffic: {error}", {error:e.message}), "error", t("Xray user"));
|
||||
}
|
||||
} finally {
|
||||
if (button) button.disabled = previousDisabled;
|
||||
}
|
||||
}
|
||||
|
||||
async function removeClient(tag, uuid) {
|
||||
const accepted = await panelConfirm({
|
||||
tone:"danger", icon:"×", title:t("Remove Xray client"),
|
||||
|
||||
@@ -418,6 +418,7 @@ async function loadManagedServerConfig(id) {
|
||||
document.getElementById("managedCfgSSHIdleTimeout").value = c.ssh_idle_timeout || "0s";
|
||||
document.getElementById("managedCfgQuiet").checked = !!c.quiet;
|
||||
document.getElementById("managedCfgUserCount").checked = !!c.user_count;
|
||||
document.getElementById("managedCfgPamAuth").checked = !!c.pam_auth_enabled;
|
||||
document.getElementById("managedCfgBanner").value = c.banner || "";
|
||||
|
||||
const hasDnstt = !!c.dnstt;
|
||||
@@ -425,7 +426,7 @@ async function loadManagedServerConfig(id) {
|
||||
toggleManagedDnsttFields(hasDnstt);
|
||||
const d = c.dnstt || {};
|
||||
document.getElementById("managedCfgDnsttDomains").value = dnsttDomainsText(d);
|
||||
document.getElementById("managedCfgDnsttUDP").value = d.udp_listen || "";
|
||||
document.getElementById("managedCfgDnsttUDP").value = d.udp_listen || "0.0.0.0:5300";
|
||||
document.getElementById("managedCfgDnsttFakeEnabled").checked = !!d.fake_dns_enabled;
|
||||
document.getElementById("managedCfgDnsttFakeListen").value = d.fake_dns_listen || "";
|
||||
document.getElementById("managedCfgDnsttFakeDomain").value = d.fake_dns_domain || "t.local.lan";
|
||||
@@ -487,12 +488,13 @@ function managedConfigFromForm() {
|
||||
ssh_idle_timeout: document.getElementById("managedCfgSSHIdleTimeout").value.trim() || "0s",
|
||||
quiet: document.getElementById("managedCfgQuiet").checked,
|
||||
user_count: document.getElementById("managedCfgUserCount").checked,
|
||||
pam_auth_enabled: document.getElementById("managedCfgPamAuth").checked,
|
||||
banner: document.getElementById("managedCfgBanner").value,
|
||||
banner_file: "/opt/sshpanel/banner.txt",
|
||||
dnstt: document.getElementById("managedCfgDnsttEnabled").checked ? {
|
||||
domain: dnsttDomains[0] || "",
|
||||
domains: dnsttDomains,
|
||||
udp_listen: document.getElementById("managedCfgDnsttUDP").value.trim(),
|
||||
udp_listen: document.getElementById("managedCfgDnsttUDP").value.trim() || "0.0.0.0:5300",
|
||||
fake_dns_enabled: document.getElementById("managedCfgDnsttFakeEnabled").checked,
|
||||
fake_dns_listen: document.getElementById("managedCfgDnsttFakeListen").value.trim(),
|
||||
fake_dns_domain: document.getElementById("managedCfgDnsttFakeDomain").value.trim(),
|
||||
|
||||
@@ -29,18 +29,24 @@ function toggleUdpgwFields(on) {
|
||||
}
|
||||
|
||||
|
||||
// Only operator-safe knobs remain. Transport buffers (HTTP/2 flow control, XHTTP
|
||||
// reorder buffer, mux/UDP buffers) are fixed to xray-core defaults in the backend
|
||||
// and are no longer exposed here, so they cannot be misconfigured.
|
||||
// Only operator-safe knobs remain. Global transport/XHTTP count admission is
|
||||
// fixed to unlimited; byte backpressure and protocol buffers stay internal so a
|
||||
// panel value cannot turn normal VPN traffic into HTTP overload responses.
|
||||
const XRAY_NATIVE_TUNING_DEFAULTS = {
|
||||
safe: {
|
||||
runtime_gomaxprocs: 0,
|
||||
mux_global_sessions: 8192,
|
||||
mux_global_sessions: 32768,
|
||||
max_concurrent_connections: -1,
|
||||
max_concurrent_xhttp_requests: -1,
|
||||
xhttp_max_sessions: -1,
|
||||
trace_packets: false,
|
||||
},
|
||||
"2k": {
|
||||
"high": {
|
||||
runtime_gomaxprocs: 0,
|
||||
mux_global_sessions: 32768,
|
||||
mux_global_sessions: 65536,
|
||||
max_concurrent_connections: -1,
|
||||
max_concurrent_xhttp_requests: -1,
|
||||
xhttp_max_sessions: -1,
|
||||
trace_packets: false,
|
||||
},
|
||||
};
|
||||
@@ -50,7 +56,7 @@ const XRAY_NATIVE_TUNING_FIELDS = {
|
||||
mux_global_sessions: "cfgXrayMuxGlobalSessions",
|
||||
};
|
||||
|
||||
function setXrayNativeTuningDefaults(profile = "2k") {
|
||||
function setXrayNativeTuningDefaults(profile = "high") {
|
||||
const t = XRAY_NATIVE_TUNING_DEFAULTS[profile] || XRAY_NATIVE_TUNING_DEFAULTS.safe;
|
||||
writeXrayNativeTuning(t);
|
||||
}
|
||||
@@ -71,6 +77,12 @@ function readXrayNativeTuning() {
|
||||
const el = document.getElementById(id);
|
||||
out[key] = parseInt(el?.value || "0", 10) || 0;
|
||||
});
|
||||
// These legacy JSON keys are intentionally fixed at unlimited. Keeping them
|
||||
// in saved configs makes upgrades/downgrades explicit without exposing web
|
||||
// request ceilings that do not belong on a VPN transport.
|
||||
out.max_concurrent_connections = -1;
|
||||
out.max_concurrent_xhttp_requests = -1;
|
||||
out.xhttp_max_sessions = -1;
|
||||
out.trace_packets = !!document.getElementById("cfgXrayTracePackets")?.checked;
|
||||
return out;
|
||||
}
|
||||
@@ -99,6 +111,7 @@ async function loadServerConfig() {
|
||||
document.getElementById("cfgSSHIdleTimeout").value = c.ssh_idle_timeout || "0s";
|
||||
document.getElementById("cfgQuiet").checked = !!c.quiet;
|
||||
document.getElementById("cfgUserCount").checked = !!c.user_count;
|
||||
document.getElementById("cfgPamAuth").checked = !!c.pam_auth_enabled;
|
||||
|
||||
// Banner
|
||||
document.getElementById("cfgBanner").value = c.banner || "";
|
||||
@@ -109,7 +122,7 @@ async function loadServerConfig() {
|
||||
toggleDnsttFields(hasDnstt);
|
||||
const d = c.dnstt || {};
|
||||
document.getElementById("cfgDnsttDomains").value = dnsttDomainsText(d);
|
||||
document.getElementById("cfgDnsttUDP").value = d.udp_listen || "";
|
||||
document.getElementById("cfgDnsttUDP").value = d.udp_listen || "0.0.0.0:5300";
|
||||
document.getElementById("cfgDnsttFakeEnabled").checked = !!d.fake_dns_enabled;
|
||||
document.getElementById("cfgDnsttFakeListen").value = d.fake_dns_listen || "";
|
||||
document.getElementById("cfgDnsttFakeDomain").value = d.fake_dns_domain || "t.local.lan";
|
||||
@@ -145,6 +158,7 @@ async function loadServerConfig() {
|
||||
// TLS forwarders
|
||||
tlsForwardersState = c.tls_forwarders || [];
|
||||
renderTLSForwarders();
|
||||
loadTLSCertificates();
|
||||
|
||||
// Xray
|
||||
const x = c.xray || {};
|
||||
@@ -182,12 +196,13 @@ async function saveServerConfig() {
|
||||
ssh_idle_timeout: document.getElementById("cfgSSHIdleTimeout").value.trim() || "0s",
|
||||
quiet: document.getElementById("cfgQuiet").checked,
|
||||
user_count: document.getElementById("cfgUserCount").checked,
|
||||
pam_auth_enabled: document.getElementById("cfgPamAuth").checked,
|
||||
banner: document.getElementById("cfgBanner").value,
|
||||
banner_file: "/opt/sshpanel/banner.txt",
|
||||
dnstt: document.getElementById("cfgDnsttEnabled").checked ? {
|
||||
domain: dnsttDomains[0] || "",
|
||||
domains: dnsttDomains,
|
||||
udp_listen: document.getElementById("cfgDnsttUDP").value.trim(),
|
||||
udp_listen: document.getElementById("cfgDnsttUDP").value.trim() || "0.0.0.0:5300",
|
||||
fake_dns_enabled: document.getElementById("cfgDnsttFakeEnabled").checked,
|
||||
fake_dns_listen: document.getElementById("cfgDnsttFakeListen").value.trim(),
|
||||
fake_dns_domain: document.getElementById("cfgDnsttFakeDomain").value.trim(),
|
||||
@@ -274,6 +289,235 @@ function renderTLSForwarders() {
|
||||
});
|
||||
}
|
||||
|
||||
// ─── TLS Certificates (renew fullchain + privkey) ─────────────────────────────
|
||||
let tlsCertsState = [];
|
||||
|
||||
async function loadTLSCertificates() {
|
||||
const st = document.getElementById("tlsCertsStatus");
|
||||
const list = document.getElementById("tlsCertsList");
|
||||
if (!list) return;
|
||||
if (st) st.textContent = "Carregando certificados…";
|
||||
try {
|
||||
const res = await api("/api/tls/certs");
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const data = await res.json();
|
||||
tlsCertsState = data.certs || [];
|
||||
renderTLSCertificates();
|
||||
if (st) st.textContent = tlsCertsState.length
|
||||
? `${tlsCertsState.length} certificado(s). Pasta do painel: ${data.certs_dir || "/opt/sshpanel/certs"}`
|
||||
: "Nenhum certificado encontrado.";
|
||||
} catch (e) {
|
||||
if (e.message === "auth") doAuthError();
|
||||
else if (st) st.textContent = "Erro: " + e.message;
|
||||
}
|
||||
}
|
||||
|
||||
function certExpiryChip(c) {
|
||||
if (!c.exists) return '<span class="chip red">arquivo ausente</span>';
|
||||
if (c.error) return `<span class="chip red">${escapeHTML(c.error)}</span>`;
|
||||
if (c.expired) return '<span class="chip red">expirado</span>';
|
||||
if (c.expiring) return `<span class="chip warn">expira em ${c.days_left} dia(s)</span>`;
|
||||
return `<span class="chip green">válido por ${c.days_left} dia(s)</span>`;
|
||||
}
|
||||
|
||||
function renderTLSCertificates() {
|
||||
const list = document.getElementById("tlsCertsList");
|
||||
const chip = document.getElementById("tlsCertsCountChip");
|
||||
if (!list) return;
|
||||
if (chip) chip.textContent = tlsCertsState.length;
|
||||
if (!tlsCertsState.length) {
|
||||
list.innerHTML = '<div class="hint" style="padding:4px 0;">Nenhum certificado encontrado neste servidor.</div>';
|
||||
return;
|
||||
}
|
||||
list.innerHTML = "";
|
||||
tlsCertsState.forEach((c, i) => {
|
||||
const row = document.createElement("div");
|
||||
row.style = "padding:8px 0;border-bottom:1px solid var(--border);font-size:.73rem;";
|
||||
|
||||
const usedBy = (c.used_by || []).map(u => {
|
||||
const label = u.kind === "tls_forwarder" ? "TLS " + u.ref : "Xray " + u.ref;
|
||||
return `<span class="chip">${escapeHTML(label)}</span>`;
|
||||
}).join(" ") || '<span class="hint">não referenciado na configuração</span>';
|
||||
|
||||
const head = document.createElement("div");
|
||||
head.style = "display:flex;align-items:center;gap:8px;flex-wrap:wrap;";
|
||||
head.innerHTML = `<strong style="font-size:.78rem;">${escapeHTML(c.name || "cert")}</strong>
|
||||
${certExpiryChip(c)}
|
||||
${c.self_signed ? '<span class="chip warn">autoassinado</span>' : ""}
|
||||
${c.managed ? '<span class="chip">painel</span>' : ""}
|
||||
<span style="flex:1"></span>`;
|
||||
|
||||
const btn = document.createElement("button");
|
||||
btn.className = "btn btn-ghost btn-sm";
|
||||
btn.type = "button";
|
||||
btn.textContent = "Atualizar certificado";
|
||||
btn.onclick = () => toggleCertRenewForm(i);
|
||||
head.appendChild(btn);
|
||||
row.appendChild(head);
|
||||
|
||||
const meta = document.createElement("div");
|
||||
meta.className = "hint";
|
||||
meta.style = "margin-top:3px;font-family:monospace;word-break:break-all;";
|
||||
const domains = (c.domains || []).join(", ") || "sem SAN";
|
||||
meta.innerHTML = `${escapeHTML(domains)}<br/>${escapeHTML(c.cert_file || "")}<br/>${escapeHTML(c.key_file || "sem chave")}`;
|
||||
row.appendChild(meta);
|
||||
|
||||
const extra = document.createElement("div");
|
||||
extra.className = "hint";
|
||||
extra.style = "margin-top:3px;";
|
||||
const bits = [];
|
||||
if (c.issuer) bits.push("emissor: " + c.issuer);
|
||||
if (c.key_type) bits.push("chave: " + c.key_type);
|
||||
if (c.chain_length) bits.push("cadeia: " + c.chain_length + " cert(s)");
|
||||
if (c.not_after) bits.push("expira: " + c.not_after.replace("T", " ").replace("Z", " UTC"));
|
||||
extra.textContent = bits.join(" · ");
|
||||
row.appendChild(extra);
|
||||
|
||||
const usage = document.createElement("div");
|
||||
usage.style = "margin-top:5px;display:flex;gap:4px;flex-wrap:wrap;align-items:center;";
|
||||
usage.innerHTML = `<span class="hint">em uso por:</span> ${usedBy}`;
|
||||
row.appendChild(usage);
|
||||
|
||||
const panel = document.createElement("div");
|
||||
panel.id = "certRenewPanel-" + i;
|
||||
panel.className = "hidden";
|
||||
panel.style = "border:1px solid var(--border);border-radius:8px;padding:10px;margin-top:8px;";
|
||||
panel.innerHTML = `
|
||||
<div style="display:grid;grid-template-columns:1fr 1fr;gap:8px;">
|
||||
<div class="field"><label>fullchain.pem <span class="hint">(certificado + intermediários)</span></label>
|
||||
<textarea id="certRenewFullchain-${i}" rows="6" placeholder="-----BEGIN CERTIFICATE----- …" style="font-family:monospace;font-size:.7rem;width:100%;box-sizing:border-box;resize:vertical;background:var(--input-bg);border:1px solid var(--border);border-radius:4px;color:inherit;padding:4px;"></textarea></div>
|
||||
<div class="field"><label>privkey.pem <span class="hint">(chave privada)</span></label>
|
||||
<textarea id="certRenewPrivkey-${i}" rows="6" placeholder="-----BEGIN PRIVATE KEY----- …" style="font-family:monospace;font-size:.7rem;width:100%;box-sizing:border-box;resize:vertical;background:var(--input-bg);border:1px solid var(--border);border-radius:4px;color:inherit;padding:4px;"></textarea></div>
|
||||
</div>
|
||||
<div class="hint" style="margin-top:6px;">Grava em <code>${escapeHTML(c.cert_file || "")}</code> e <code>${escapeHTML(c.key_file || "")}</code>. O conteúdo anterior fica salvo como <code>.bak</code>.</div>
|
||||
<div class="form-actions" style="margin-top:8px;">
|
||||
<button class="btn btn-sm" type="button" onclick="submitCertRenew(${i})">Salvar e recarregar</button>
|
||||
<button class="btn btn-ghost btn-sm" type="button" onclick="toggleCertRenewForm(${i})">Cancelar</button>
|
||||
</div>
|
||||
<div id="certRenewStatus-${i}" class="hint" style="margin-top:4px;"></div>`;
|
||||
row.appendChild(panel);
|
||||
|
||||
list.appendChild(row);
|
||||
});
|
||||
}
|
||||
|
||||
function toggleCertRenewForm(i) {
|
||||
const panel = document.getElementById("certRenewPanel-" + i);
|
||||
if (!panel) return;
|
||||
panel.classList.toggle("hidden");
|
||||
if (!panel.classList.contains("hidden")) {
|
||||
document.getElementById("certRenewStatus-" + i).textContent = "";
|
||||
document.getElementById("certRenewFullchain-" + i).focus();
|
||||
}
|
||||
}
|
||||
|
||||
function reportCertUpdate(statusEl, data) {
|
||||
const r = data?.reloaded || {};
|
||||
const applied = [];
|
||||
if ((r.tls_forwarders || []).length) applied.push("TLS " + r.tls_forwarders.join(", "));
|
||||
if (r.xray_restarted) applied.push("Xray reiniciado (" + (r.xray_inbounds || []).join(", ") + ")");
|
||||
const warnings = data?.warnings || [];
|
||||
const cert = data?.cert || {};
|
||||
const parts = ["Certificado gravado."];
|
||||
if (cert.not_after) parts.push("Válido até " + cert.not_after.replace("T", " ").replace("Z", " UTC") + ".");
|
||||
if (applied.length) parts.push("Recarregado: " + applied.join(" | ") + ".");
|
||||
else parts.push("Nenhum listener em uso precisou recarregar.");
|
||||
if (warnings.length) parts.push("Avisos: " + warnings.join(" | "));
|
||||
statusEl.textContent = parts.join(" ");
|
||||
showPanelToast(
|
||||
warnings.length ? "Certificado atualizado com avisos." : "Certificado atualizado e aplicado.",
|
||||
warnings.length ? "warning" : "success",
|
||||
);
|
||||
}
|
||||
|
||||
async function submitCertRenew(i) {
|
||||
const c = tlsCertsState[i];
|
||||
const st = document.getElementById("certRenewStatus-" + i);
|
||||
if (!c || !st) return;
|
||||
const fullchain = document.getElementById("certRenewFullchain-" + i).value.trim();
|
||||
const privkey = document.getElementById("certRenewPrivkey-" + i).value.trim();
|
||||
if (!fullchain || !privkey) { st.textContent = "Cole o fullchain.pem e o privkey.pem."; return; }
|
||||
|
||||
const usedBy = (c.used_by || []).length;
|
||||
const ok = await panelConfirm({
|
||||
title: "Atualizar certificado",
|
||||
message: `Substituir o certificado de ${c.name || c.cert_file}?`,
|
||||
detail: usedBy
|
||||
? "Os listeners TLS que usam este certificado serão reabertos e o Xray será reiniciado se algum inbound usar o certificado. Conexões já estabelecidas não são encerradas."
|
||||
: "Os arquivos serão substituídos (backup .bak).",
|
||||
confirmLabel: "Atualizar",
|
||||
});
|
||||
if (!ok) return;
|
||||
|
||||
st.textContent = "Gravando e recarregando…";
|
||||
await postCertUpdate({ cert_file: c.cert_file, key_file: c.key_file, fullchain, privkey }, st, () => {
|
||||
document.getElementById("certRenewPanel-" + i)?.classList.add("hidden");
|
||||
loadTLSCertificates();
|
||||
});
|
||||
}
|
||||
|
||||
async function postCertUpdate(payload, st, onDone) {
|
||||
try {
|
||||
let res = await api("/api/tls/certs/update", { method: "POST", body: JSON.stringify(payload) });
|
||||
if (!res.ok) {
|
||||
const text = await res.text();
|
||||
if (res.status === 400 && text.includes("force=true")) {
|
||||
const force = await panelConfirm({
|
||||
title: "Certificado expirado",
|
||||
message: text.split(";")[0],
|
||||
detail: "Gravar mesmo assim? Clientes não conseguirão validar um certificado expirado.",
|
||||
confirmLabel: "Gravar mesmo assim",
|
||||
danger: true,
|
||||
});
|
||||
if (!force) { st.textContent = "Cancelado."; return; }
|
||||
res = await api("/api/tls/certs/update", { method: "POST", body: JSON.stringify({ ...payload, force: true }) });
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
} else {
|
||||
throw new Error(text);
|
||||
}
|
||||
}
|
||||
const data = await res.json();
|
||||
reportCertUpdate(st, data);
|
||||
onDone?.();
|
||||
} catch (e) {
|
||||
if (e.message === "auth") doAuthError();
|
||||
else st.textContent = "Erro: " + e.message;
|
||||
}
|
||||
}
|
||||
|
||||
function toggleNewCertForm() {
|
||||
const panel = document.getElementById("newCertPanel");
|
||||
if (!panel) return;
|
||||
panel.classList.toggle("hidden");
|
||||
if (!panel.classList.contains("hidden")) {
|
||||
document.getElementById("newCertStatus").textContent = "";
|
||||
document.getElementById("newCertName").value = "";
|
||||
document.getElementById("newCertFullchain").value = "";
|
||||
document.getElementById("newCertPrivkey").value = "";
|
||||
}
|
||||
}
|
||||
|
||||
async function saveNewCert() {
|
||||
const st = document.getElementById("newCertStatus");
|
||||
const name = document.getElementById("newCertName").value.trim();
|
||||
const fullchain = document.getElementById("newCertFullchain").value.trim();
|
||||
const privkey = document.getElementById("newCertPrivkey").value.trim();
|
||||
if (!name || !fullchain || !privkey) { st.textContent = "Nome, fullchain.pem e privkey.pem são obrigatórios."; return; }
|
||||
st.textContent = "Gravando…";
|
||||
await postCertUpdate({ name, fullchain, privkey }, st, () => {
|
||||
document.getElementById("newCertFullchain").value = "";
|
||||
document.getElementById("newCertPrivkey").value = "";
|
||||
loadTLSCertificates();
|
||||
});
|
||||
}
|
||||
|
||||
// Inline onclick handlers in index.html need these exposed explicitly.
|
||||
window.loadTLSCertificates = loadTLSCertificates;
|
||||
window.toggleCertRenewForm = toggleCertRenewForm;
|
||||
window.submitCertRenew = submitCertRenew;
|
||||
window.toggleNewCertForm = toggleNewCertForm;
|
||||
window.saveNewCert = saveNewCert;
|
||||
|
||||
function toggleAddTLSForm() {
|
||||
const panel = document.getElementById("addTLSPanel");
|
||||
panel.classList.toggle("hidden");
|
||||
@@ -383,7 +627,7 @@ async function wzSavePastedCert() {
|
||||
if (!name || !cert || !key) { st.textContent = "Name, cert, and key required."; return; }
|
||||
st.textContent = "Saving…";
|
||||
try {
|
||||
const res = await api("/api/tls/upload-pem", { method:"POST", body: JSON.stringify({ name, cert, key }) });
|
||||
const res = await api(withServerParam("/api/tls/upload-pem", selectedXrayServer()), { method:"POST", body: JSON.stringify({ name, cert, key }) });
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const data = await res.json();
|
||||
document.getElementById("wzTLSCert").value = data.cert_file;
|
||||
@@ -401,7 +645,7 @@ async function wzGenerateCert() {
|
||||
if (!domain) { st.textContent = "Domain required."; return; }
|
||||
st.textContent = "Generating…";
|
||||
try {
|
||||
const res = await api("/api/tls/generate-selfsigned", { method:"POST", body: JSON.stringify({ domain }) });
|
||||
const res = await api(withServerParam("/api/tls/generate-selfsigned", selectedXrayServer()), { method:"POST", body: JSON.stringify({ domain }) });
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const data = await res.json();
|
||||
document.getElementById("wzTLSCert").value = data.cert_file;
|
||||
|
||||
+174
-138
@@ -6,6 +6,11 @@ function openEditXrayClient(tag, client) {
|
||||
document.getElementById("editXrayEmail").value = client.email || "";
|
||||
document.getElementById("editXrayExpiry").value = client.expires_at ? localFromISO(client.expires_at) : "";
|
||||
document.getElementById("editXrayMaxConns").value = client.max_conns || 0;
|
||||
document.getElementById("editXrayQuotaGB").value = client.data_quota_bytes ? (Number(client.data_quota_bytes) / (1024 ** 3)).toFixed(2).replace(/\.00$/, "") : "0";
|
||||
document.getElementById("editXrayQuotaAction").value = client.quota_action === "throttle" ? "throttle" : "block";
|
||||
document.getElementById("editXrayQuotaThrottle").value = client.quota_throttle_mbps || 1;
|
||||
document.getElementById("editXrayUsage").value = `${formatBytes(client.total_bytes || 0)} (↑ ${formatBytes(client.uplink_bytes || 0)} · ↓ ${formatBytes(client.downlink_bytes || 0)})`;
|
||||
document.getElementById("editXrayResetUsage").checked = false;
|
||||
document.getElementById("editXrayClientStatus").textContent = "";
|
||||
document.getElementById("editXrayClientPanel").classList.remove("hidden");
|
||||
document.getElementById("editXrayClientPanel").scrollIntoView({ behavior:"smooth", block:"nearest" });
|
||||
@@ -26,6 +31,10 @@ async function saveEditXrayClient() {
|
||||
email: document.getElementById("editXrayEmail").value.trim(),
|
||||
expires_at: isoFromLocal(document.getElementById("editXrayExpiry").value),
|
||||
max_connections: parseInt(document.getElementById("editXrayMaxConns").value || "0", 10),
|
||||
data_quota_bytes: Math.round((parseFloat(document.getElementById("editXrayQuotaGB").value || "0") || 0) * (1024 ** 3)),
|
||||
quota_action: document.getElementById("editXrayQuotaAction").value === "throttle" ? "throttle" : "block",
|
||||
quota_throttle_mbps: parseInt(document.getElementById("editXrayQuotaThrottle").value || "1", 10) || 1,
|
||||
reset_usage: !!document.getElementById("editXrayResetUsage").checked,
|
||||
server_id: selectedXrayServer(),
|
||||
};
|
||||
try {
|
||||
@@ -83,9 +92,11 @@ function loadWizardFromConfig() {
|
||||
document.getElementById("wzLogLevel").value = cfg.log?.loglevel || "warning";
|
||||
wzInbounds = cloneJsonSafe((cfg.inbounds || []).filter(ib => ib && ib.tag !== "api")) || [];
|
||||
wzEditingIndex = -1;
|
||||
wzCancelInbound();
|
||||
renderWzInbounds();
|
||||
loadSharedEndpointForm();
|
||||
wzDirty = false;
|
||||
const presetStatus = document.getElementById("wzAzionDefaultStatus");
|
||||
if (presetStatus) presetStatus.textContent = "O padrão cria o certificado autoassinado, habilita TLS e salva/reinicia o Xray automaticamente.";
|
||||
if (st) st.textContent = `Config loaded from ${target}.`;
|
||||
}).catch(e => {
|
||||
wzLoadedServerID = null;
|
||||
@@ -93,8 +104,8 @@ function loadWizardFromConfig() {
|
||||
wzLoadedFullConfig = null;
|
||||
wzInbounds = [];
|
||||
wzEditingIndex = -1;
|
||||
wzCancelInbound();
|
||||
renderWzInbounds();
|
||||
loadSharedEndpointForm();
|
||||
if (e.message === "auth") doAuthError();
|
||||
else if (st) st.textContent = "Error: " + e.message;
|
||||
});
|
||||
@@ -107,7 +118,7 @@ function renderWzInbounds() {
|
||||
if (!wzInbounds.length) {
|
||||
const empty = document.createElement("div");
|
||||
empty.className = "hint visual-empty-state";
|
||||
empty.textContent = "Nenhum inbound configurado. Crie um endpoint compartilhado ou adicione um inbound.";
|
||||
empty.textContent = "Nenhum inbound configurado. Use “Adicionar inbound” ou “Criar padrão Azion XHTTP”.";
|
||||
list.appendChild(empty);
|
||||
return;
|
||||
}
|
||||
@@ -192,7 +203,6 @@ function renderWzInbounds() {
|
||||
else if (wzEditingIndex > i) wzEditingIndex--;
|
||||
wzDirty = true;
|
||||
renderWzInbounds();
|
||||
loadSharedEndpointForm();
|
||||
};
|
||||
actions.append(duplicateBtn, editBtn, delBtn);
|
||||
row.appendChild(actions);
|
||||
@@ -200,12 +210,29 @@ function renderWzInbounds() {
|
||||
});
|
||||
}
|
||||
|
||||
function wzToggleAddInbound() {
|
||||
function mountWzInboundEditor() {
|
||||
const form = document.getElementById("wzAddInboundForm");
|
||||
const anchor = document.getElementById("wzInboundEditorAnchor");
|
||||
if (form && anchor && form.previousElementSibling !== anchor) {
|
||||
anchor.insertAdjacentElement("afterend", form);
|
||||
}
|
||||
return form;
|
||||
}
|
||||
|
||||
function openWzInboundEditor(scrollBlock = "nearest") {
|
||||
const form = mountWzInboundEditor();
|
||||
if (!form) return null;
|
||||
form.classList.remove("hidden");
|
||||
requestAnimationFrame(() => form.scrollIntoView({ behavior:"smooth", block:scrollBlock }));
|
||||
return form;
|
||||
}
|
||||
|
||||
function wzToggleAddInbound() {
|
||||
const form = mountWzInboundEditor();
|
||||
if (!form) return;
|
||||
if (!form.classList.contains("hidden") && wzEditingIndex < 0) return wzCancelInbound();
|
||||
resetWzInboundForm();
|
||||
form.classList.remove("hidden");
|
||||
form.scrollIntoView({ behavior:"smooth", block:"nearest" });
|
||||
openWzInboundEditor("nearest");
|
||||
}
|
||||
|
||||
function setWzValue(id, value) {
|
||||
@@ -233,7 +260,8 @@ function resetWzInboundForm() {
|
||||
setWzValue("wzTLS", "none");
|
||||
["wzTLSCert", "wzTLSKey", "wzTLSCertPath", "wzTLSKeyPath", "wzRealityDest", "wzRealitySNI", "wzRealityPriv", "wzRealityShortID", "wzTrojanPass", "wzSSPass"].forEach(id => setWzValue(id, ""));
|
||||
setWzValue("wzSSMethod", "chacha20-ietf-poly1305");
|
||||
document.getElementById("wzInboundFormTitle").textContent = "Novo inbound";
|
||||
document.getElementById("wzInboundFormKicker").textContent = "Novo inbound";
|
||||
document.getElementById("wzInboundFormTitle").textContent = "Adicionar inbound";
|
||||
document.getElementById("wzSaveInboundBtn").textContent = "Adicionar inbound";
|
||||
document.getElementById("wzEditingBadge").classList.add("hidden");
|
||||
onWzProtoChange("vless");
|
||||
@@ -292,11 +320,10 @@ function editWzInbound(index) {
|
||||
setWzValue("wzSSMethod", ib.settings?.method || "chacha20-ietf-poly1305");
|
||||
|
||||
document.getElementById("wzInboundFormTitle").textContent = `Editar ${ib.tag || "inbound"}`;
|
||||
document.getElementById("wzInboundFormKicker").textContent = "Editar inbound existente";
|
||||
document.getElementById("wzSaveInboundBtn").textContent = "Salvar alterações";
|
||||
document.getElementById("wzEditingBadge").classList.remove("hidden");
|
||||
const form = document.getElementById("wzAddInboundForm");
|
||||
form.classList.remove("hidden");
|
||||
form.scrollIntoView({ behavior:"smooth", block:"start" });
|
||||
openWzInboundEditor("start");
|
||||
}
|
||||
|
||||
function duplicateWzInbound(index) {
|
||||
@@ -516,149 +543,159 @@ function validateVisualInbounds(inbounds) {
|
||||
}
|
||||
}
|
||||
|
||||
function findSharedEndpointPair() {
|
||||
const roots = wzInbounds.filter(ib => {
|
||||
const xh = visualXHTTPSettings(ib);
|
||||
return !!xh && ["vless", "vmess"].includes(String(ib?.protocol || "").toLowerCase()) && normalizeVisualPath(xh.path) === "/";
|
||||
const azionPresetProxyTags = new Set(["azion-vless-xhttp", "shared-proxy-xhttp"]);
|
||||
const azionPresetSSHTags = new Set(["azion-ssh-xhttp", "shared-ssh-xhttp"]);
|
||||
|
||||
function findAzionPresetInbound(tags) {
|
||||
return wzInbounds.find(ib => tags.has(String(ib?.tag || ""))) || null;
|
||||
}
|
||||
|
||||
function azionVLESSClients(existingProxy) {
|
||||
const clients = Array.isArray(existingProxy?.settings?.clients) ? cloneJsonSafe(existingProxy.settings.clients) : [];
|
||||
if (String(existingProxy?.protocol || "").toLowerCase() === "vless") return clients;
|
||||
return clients.map(client => {
|
||||
const converted = { id:client?.id };
|
||||
if (client?.email) converted.email = client.email;
|
||||
if (client?.flow) converted.flow = client.flow;
|
||||
return converted;
|
||||
}).filter(client => client.id);
|
||||
}
|
||||
|
||||
function buildAzionPresetInbounds(certFile, keyFile) {
|
||||
const existingProxy = findAzionPresetInbound(azionPresetProxyTags);
|
||||
const existingSSH = findAzionPresetInbound(azionPresetSSHTags);
|
||||
const managed = new Set([existingProxy, existingSSH].filter(Boolean));
|
||||
const others = wzInbounds.filter(ib => !managed.has(ib));
|
||||
const portConflict = others.find(ib => Number(ib?.port) === 443);
|
||||
if (portConflict) {
|
||||
throw new Error(`A porta 443 já é usada pelo inbound ${portConflict.tag || "sem tag"}. Edite ou remova esse inbound antes de criar o padrão Azion.`);
|
||||
}
|
||||
const stream = path => ({
|
||||
network:"xhttp",
|
||||
security:"tls",
|
||||
xhttpSettings:{ path, mode:"auto" },
|
||||
tlsSettings:{ certificates:[{ certificateFile:certFile, keyFile }] },
|
||||
});
|
||||
for (const proxy of roots) {
|
||||
const ssh = wzInbounds.find(ib => String(ib?.protocol || "").toLowerCase() === "ssh" &&
|
||||
String(ib.listen || "0.0.0.0") === String(proxy.listen || "0.0.0.0") && String(ib.port) === String(proxy.port) &&
|
||||
normalizeVisualPath(visualXHTTPSettings(ib)?.path) === "/ssh");
|
||||
if (ssh) return { proxy, ssh };
|
||||
}
|
||||
const proxy = wzInbounds.find(ib => ib?.tag === "shared-proxy-xhttp") || null;
|
||||
const ssh = wzInbounds.find(ib => ib?.tag === "shared-ssh-xhttp") || null;
|
||||
return proxy && ssh ? { proxy, ssh } : null;
|
||||
const proxyInbound = {
|
||||
tag:"azion-vless-xhttp",
|
||||
listen:"0.0.0.0",
|
||||
port:443,
|
||||
protocol:"vless",
|
||||
settings:{ clients:azionVLESSClients(existingProxy), decryption:"none" },
|
||||
streamSettings:stream("/"),
|
||||
};
|
||||
const sshInbound = {
|
||||
tag:"azion-ssh-xhttp",
|
||||
listen:"0.0.0.0",
|
||||
port:443,
|
||||
protocol:"ssh",
|
||||
settings:{},
|
||||
streamSettings:stream("/ssh"),
|
||||
};
|
||||
return [...others, proxyInbound, sshInbound];
|
||||
}
|
||||
|
||||
function loadSharedEndpointForm() {
|
||||
const status = document.getElementById("sharedXHTTPStatus");
|
||||
if (!status) return;
|
||||
const pair = findSharedEndpointPair();
|
||||
if (!pair) {
|
||||
status.textContent = wzLoadedConfigText ? "Nenhum endpoint compartilhado detectado. Preencha os campos para criar um." : "Carregue a configuração para detectar um endpoint existente.";
|
||||
return;
|
||||
}
|
||||
const xh = visualXHTTPSettings(pair.proxy) || {};
|
||||
const ss = pair.proxy.streamSettings || {};
|
||||
const cert = ss.tlsSettings?.certificates?.[0] || {};
|
||||
setWzValue("sharedXHTTPProtocol", pair.proxy.protocol || "vless");
|
||||
setWzValue("sharedXHTTPPort", pair.proxy.port || 443);
|
||||
setWzValue("sharedXHTTPListen", pair.proxy.listen || "0.0.0.0");
|
||||
setWzValue("sharedXHTTPHost", xh.host || "");
|
||||
setWzValue("sharedXHTTPMode", xh.mode || "auto");
|
||||
setWzValue("sharedXHTTPSecurity", ss.security === "tls" ? "tls" : "none");
|
||||
setWzValue("sharedXHTTPCert", cert.certificateFile || "");
|
||||
setWzValue("sharedXHTTPKey", cert.keyFile || "");
|
||||
updateSharedEndpointControls();
|
||||
status.textContent = `Endpoint detectado em ${pair.proxy.listen || "0.0.0.0"}:${pair.proxy.port} — ${String(pair.proxy.protocol).toUpperCase()} / e SSH /ssh.`;
|
||||
}
|
||||
|
||||
function updateSharedEndpointControls() {
|
||||
const protocol = document.getElementById("sharedXHTTPProtocol")?.value || "vless";
|
||||
const security = document.getElementById("sharedXHTTPSecurity")?.value || "none";
|
||||
document.getElementById("sharedProxyRouteLabel").textContent = protocol.toUpperCase();
|
||||
document.querySelectorAll(".shared-tls-field").forEach(el => el.classList.toggle("hidden", security !== "tls"));
|
||||
}
|
||||
|
||||
function applySharedXHTTPEndpoint() {
|
||||
const status = document.getElementById("sharedXHTTPStatus");
|
||||
async function createAzionDefaultXHTTP() {
|
||||
const status = document.getElementById("wzAzionDefaultStatus");
|
||||
const button = document.getElementById("wzAzionDefaultBtn");
|
||||
const selectedID = selectedXrayServer() || "local";
|
||||
const target = selectedXrayServerLabel();
|
||||
if (!wzLoadedConfigText || String(wzLoadedServerID || "") !== String(selectedID)) {
|
||||
status.textContent = "Carregue a configuração do servidor selecionado antes de editar.";
|
||||
if (status) status.textContent = `Carregue a configuração de ${target} antes de criar o padrão Azion.`;
|
||||
return;
|
||||
}
|
||||
if ((document.getElementById("xCoreMode")?.value || "native") !== "native") {
|
||||
status.textContent = "O endpoint compartilhado requer o modo Xray nativo.";
|
||||
return;
|
||||
}
|
||||
const protocol = document.getElementById("sharedXHTTPProtocol").value;
|
||||
const port = Number(document.getElementById("sharedXHTTPPort").value || 0);
|
||||
const listen = document.getElementById("sharedXHTTPListen").value.trim() || "0.0.0.0";
|
||||
const host = document.getElementById("sharedXHTTPHost").value.trim();
|
||||
const mode = document.getElementById("sharedXHTTPMode").value || "auto";
|
||||
const security = document.getElementById("sharedXHTTPSecurity").value;
|
||||
const cert = document.getElementById("sharedXHTTPCert").value.trim();
|
||||
const key = document.getElementById("sharedXHTTPKey").value.trim();
|
||||
if (!["vless", "vmess"].includes(protocol) || !Number.isInteger(port) || port < 1 || port > 65535) {
|
||||
status.textContent = "Escolha VLESS/VMess e uma porta válida.";
|
||||
return;
|
||||
}
|
||||
if (/[\u0000-\u001f\u007f]/.test(`${listen}${host}${cert}${key}`)) {
|
||||
status.textContent = "Os campos contêm caracteres de controle inválidos.";
|
||||
return;
|
||||
}
|
||||
if (security === "tls" && (!cert || !key)) {
|
||||
status.textContent = "Informe os arquivos do certificado e da chave para usar TLS.";
|
||||
if (status) status.textContent = "O padrão Azion com SSH requer o modo Xray nativo.";
|
||||
return;
|
||||
}
|
||||
|
||||
const pair = findSharedEndpointPair();
|
||||
const sameEndpoint = ib => String(ib?.listen || "0.0.0.0") === listen && String(ib?.port) === String(port);
|
||||
const existingProxy = pair?.proxy || wzInbounds.find(ib => ib?.tag === "shared-proxy-xhttp") || wzInbounds.find(ib => {
|
||||
const xh = visualXHTTPSettings(ib);
|
||||
return sameEndpoint(ib) && !!xh && ["vless", "vmess"].includes(String(ib?.protocol || "").toLowerCase()) && normalizeVisualPath(xh.path) === "/";
|
||||
}) || null;
|
||||
const existingSSH = pair?.ssh || wzInbounds.find(ib => ib?.tag === "shared-ssh-xhttp") || wzInbounds.find(ib => {
|
||||
const xh = visualXHTTPSettings(ib);
|
||||
return sameEndpoint(ib) && !!xh && String(ib?.protocol || "").toLowerCase() === "ssh" && normalizeVisualPath(xh.path) === "/ssh";
|
||||
}) || null;
|
||||
const removeSet = new Set([existingProxy, existingSSH].filter(Boolean));
|
||||
const others = wzInbounds.filter(ib => !removeSet.has(ib));
|
||||
const blocking = others.find(ib => String(ib.listen || "0.0.0.0") === listen && String(ib.port) === String(port) && !visualXHTTPSettings(ib));
|
||||
if (blocking) {
|
||||
status.textContent = `A porta já é usada pelo inbound não-XHTTP ${blocking.tag || "sem tag"}. Escolha outra porta.`;
|
||||
let nextInbounds;
|
||||
try {
|
||||
nextInbounds = buildAzionPresetInbounds(
|
||||
"/opt/sshpanel/certs/example.com/cert.pem",
|
||||
"/opt/sshpanel/certs/example.com/key.pem",
|
||||
);
|
||||
validateVisualInbounds(nextInbounds);
|
||||
} catch (error) {
|
||||
if (status) status.textContent = error.message;
|
||||
return;
|
||||
}
|
||||
|
||||
const buildSharedStream = (existing, path) => {
|
||||
const stream = cloneJsonSafe(existing?.streamSettings || {});
|
||||
stream.network = "xhttp";
|
||||
stream.xhttpSettings = Object.assign({}, stream.xhttpSettings || stream.splithttpSettings || {}, { path, mode });
|
||||
delete stream.splithttpSettings;
|
||||
if (host) stream.xhttpSettings.host = host;
|
||||
else delete stream.xhttpSettings.host;
|
||||
if (security === "tls") {
|
||||
stream.security = "tls";
|
||||
stream.tlsSettings = Object.assign({}, stream.tlsSettings || {}, { certificates:[{ certificateFile:cert, keyFile:key }] });
|
||||
} else {
|
||||
delete stream.security;
|
||||
delete stream.tlsSettings;
|
||||
const existingProxy = findAzionPresetInbound(azionPresetProxyTags);
|
||||
const accepted = await panelConfirm({
|
||||
tone:"success",
|
||||
icon:"AZ",
|
||||
eyebrow:"Azion XHTTP",
|
||||
title:existingProxy ? "Atualizar padrão Azion" : "Criar padrão Azion",
|
||||
message:existingProxy
|
||||
? "Atualizar o endpoint padrão e manter os clientes VLESS existentes?"
|
||||
: "Criar o endpoint padrão completo neste servidor?",
|
||||
detail:[
|
||||
`Servidor: ${target}`,
|
||||
"Listen: 0.0.0.0:443",
|
||||
"TLS autoassinado: example.com",
|
||||
"VLESS XHTTP: /",
|
||||
"SSH XHTTP: /ssh",
|
||||
"O Xray será salvo e reiniciado automaticamente.",
|
||||
].join("\n"),
|
||||
confirmLabel:existingProxy ? "Atualizar padrão" : "Criar padrão",
|
||||
});
|
||||
if (!accepted) return;
|
||||
|
||||
const previousInbounds = cloneJsonSafe(wzInbounds);
|
||||
const previousDirty = wzDirty;
|
||||
let presetApplied = false;
|
||||
let configSaved = false;
|
||||
if (button) {
|
||||
button.disabled = true;
|
||||
button.textContent = "Criando padrão…";
|
||||
}
|
||||
if (status) status.textContent = `Gerando certificado example.com em ${target}…`;
|
||||
try {
|
||||
const certResponse = await api(withServerParam("/api/tls/generate-selfsigned", selectedID), {
|
||||
method:"POST",
|
||||
body:JSON.stringify({ domain:"example.com" }),
|
||||
});
|
||||
if (!certResponse.ok) throw new Error(await certResponse.text());
|
||||
const cert = await certResponse.json();
|
||||
if (!cert?.cert_file || !cert?.key_file) throw new Error("o servidor não retornou os caminhos do certificado");
|
||||
|
||||
wzInbounds = buildAzionPresetInbounds(cert.cert_file, cert.key_file);
|
||||
validateVisualInbounds(wzInbounds);
|
||||
presetApplied = true;
|
||||
wzDirty = true;
|
||||
wzCancelInbound();
|
||||
renderWzInbounds();
|
||||
if (status) status.textContent = "Certificado criado. Salvando configuração e reiniciando o Xray…";
|
||||
const result = await applyWizardConfig();
|
||||
if (!result?.saved) throw new Error(result?.error || "não foi possível salvar a configuração");
|
||||
configSaved = true;
|
||||
if (status) status.textContent = result.restarted
|
||||
? "Padrão Azion ativo: TLS example.com, VLESS / e SSH /ssh em 0.0.0.0:443."
|
||||
: "O padrão foi salvo, mas o Xray não reiniciou. Verifique os logs e use Reiniciar.";
|
||||
if (typeof showPanelToast === "function") {
|
||||
showPanelToast(
|
||||
result.restarted ? "Padrão Azion XHTTP criado e ativo." : "Padrão Azion salvo; reinício pendente.",
|
||||
result.restarted ? "success" : "warning",
|
||||
"Azion XHTTP",
|
||||
);
|
||||
}
|
||||
delete stream.realitySettings;
|
||||
return stream;
|
||||
};
|
||||
const previousClients = Array.isArray(existingProxy?.settings?.clients) ? cloneJsonSafe(existingProxy.settings.clients) : [];
|
||||
const proxyInbound = cloneJsonSafe(existingProxy || {});
|
||||
proxyInbound.tag = existingProxy?.tag || "shared-proxy-xhttp";
|
||||
proxyInbound.listen = listen;
|
||||
proxyInbound.port = port;
|
||||
proxyInbound.protocol = protocol;
|
||||
proxyInbound.settings = existingProxy?.protocol === protocol ? cloneJsonSafe(existingProxy.settings || {}) : {};
|
||||
proxyInbound.settings.clients = previousClients;
|
||||
if (protocol === "vless") proxyInbound.settings.decryption = "none";
|
||||
else delete proxyInbound.settings.decryption;
|
||||
proxyInbound.streamSettings = buildSharedStream(existingProxy, "/");
|
||||
const sshInbound = cloneJsonSafe(existingSSH || {});
|
||||
sshInbound.tag = existingSSH?.tag || "shared-ssh-xhttp";
|
||||
sshInbound.listen = listen;
|
||||
sshInbound.port = port;
|
||||
sshInbound.protocol = "ssh";
|
||||
sshInbound.settings = {};
|
||||
sshInbound.streamSettings = buildSharedStream(existingSSH, "/ssh");
|
||||
wzInbounds = [...others, proxyInbound, sshInbound];
|
||||
wzDirty = true;
|
||||
renderWzInbounds();
|
||||
loadSharedEndpointForm();
|
||||
status.textContent = "Endpoint atualizado no rascunho. Clique em Salvar configuração e reiniciar para aplicar.";
|
||||
} catch (error) {
|
||||
if (presetApplied && !configSaved) {
|
||||
wzInbounds = previousInbounds;
|
||||
wzDirty = previousDirty;
|
||||
renderWzInbounds();
|
||||
}
|
||||
if (error.message === "auth") doAuthError();
|
||||
else if (status) status.textContent = "Erro ao criar padrão Azion: " + error.message;
|
||||
} finally {
|
||||
if (button) {
|
||||
button.disabled = false;
|
||||
button.textContent = "Criar padrão Azion XHTTP";
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
document.getElementById("sharedXHTTPProtocol")?.addEventListener("change", updateSharedEndpointControls);
|
||||
document.getElementById("sharedXHTTPSecurity")?.addEventListener("change", updateSharedEndpointControls);
|
||||
document.getElementById("sharedXHTTPApplyBtn")?.addEventListener("click", applySharedXHTTPEndpoint);
|
||||
updateSharedEndpointControls();
|
||||
|
||||
function onWzProtoChange(val) {
|
||||
const isSSH = val === "ssh";
|
||||
// SSH tunnels reuse the VLESS/VMess transport block to expose the XHTTP
|
||||
@@ -867,7 +904,6 @@ function wzSaveInbound() {
|
||||
else wzInbounds.push(ib);
|
||||
wzDirty = true;
|
||||
renderWzInbounds();
|
||||
loadSharedEndpointForm();
|
||||
st.textContent = original ? `Inbound ${tag} atualizado no rascunho.` : `Inbound ${tag} adicionado ao rascunho.`;
|
||||
wzCancelInbound();
|
||||
}
|
||||
|
||||
+108
-44
@@ -16,7 +16,7 @@
|
||||
setTimeout(function(){document.documentElement.classList.remove("i18n-pending");},2500);
|
||||
})();
|
||||
</script>
|
||||
<link rel="stylesheet" href="assets/app.css?v=20260713sections5"/>
|
||||
<link rel="stylesheet" href="assets/app.css?v=20260724xrayinboundsr3"/>
|
||||
</head>
|
||||
<body>
|
||||
<div class="app">
|
||||
@@ -269,12 +269,23 @@
|
||||
<button class="btn btn-ghost btn-sm" id="reloadUsersBtn">Reload</button>
|
||||
</div>
|
||||
</div>
|
||||
<div class="user-list-controls" id="sshListControls" aria-label="SSH user list controls">
|
||||
<div class="user-list-control-group">
|
||||
<span class="user-list-control-label" id="sshSortLabel"></span>
|
||||
<div class="user-list-buttons" id="sshSortButtons"></div>
|
||||
</div>
|
||||
<div class="user-list-control-group">
|
||||
<span class="user-list-control-label" id="sshFilterLabel"></span>
|
||||
<div class="user-list-buttons" id="sshFilterButtons"></div>
|
||||
</div>
|
||||
<span class="chip user-list-count" id="sshListCount"></span>
|
||||
</div>
|
||||
<div class="tbl-wrap">
|
||||
<table>
|
||||
<table class="table-cards">
|
||||
<thead><tr>
|
||||
<th>User</th><th>Status</th><th>Auth</th>
|
||||
<th>Conn</th><th>Max</th><th>Up</th><th>Dn</th><th>Expires</th>
|
||||
<th id="ownerColHead" class="superadmin-only hidden">Owner</th>
|
||||
<th data-sort-key="username">User</th><th data-sort-key="status">Status</th><th data-sort-key="auth">Auth</th>
|
||||
<th data-sort-key="conn">Conn</th><th data-sort-key="max">Max</th><th data-sort-key="up">Up</th><th data-sort-key="down">Dn</th><th data-sort-key="speed" title="Current up/down speed of the whole account, across all of its connections.">Speed</th><th data-sort-key="usage">Traffic</th><th data-sort-key="expires">Expires</th>
|
||||
<th id="ownerColHead" data-sort-key="owner" class="superadmin-only hidden">Owner</th>
|
||||
<th>Actions</th>
|
||||
</tr></thead>
|
||||
<tbody id="usersBody"></tbody>
|
||||
@@ -310,6 +321,11 @@
|
||||
<div class="field"><label>Expires at</label><input id="fExpires" type="datetime-local"/></div>
|
||||
<div class="field"><label>Max Upload (Mb/s)</label><input id="fUp" type="number" min="0" placeholder="0 = default"/></div>
|
||||
<div class="field"><label>Max Download (Mb/s)</label><input id="fDown" type="number" min="0" placeholder="0 = default"/></div>
|
||||
<div class="field"><label>Data quota (GB) <span class="hint">0 = unlimited · 1024 = 1 TB</span></label><input id="fQuotaGB" type="number" min="0" step="0.01" placeholder="0"/></div>
|
||||
<div class="field"><label>When quota is reached</label><select id="fQuotaAction"><option value="block">Block user</option><option value="throttle" selected>Reduce speed</option></select></div>
|
||||
<div class="field"><label>Post-quota speed (Mb/s)</label><input id="fQuotaThrottle" type="number" min="1" value="1"/></div>
|
||||
<div class="field"><label>Current usage</label><input id="fUsageDisplay" readonly value="0 B"/></div>
|
||||
<div class="field"><label>Reset traffic counter</label><input id="fResetUsage" type="checkbox" style="width:16px;height:16px;margin-top:10px;"/></div>
|
||||
</div>
|
||||
<div class="form-actions">
|
||||
<button class="btn" type="submit" id="saveUserBtn">Save user</button>
|
||||
@@ -372,6 +388,11 @@
|
||||
<div class="field"><label>Email / Label</label><input id="editXrayEmail" autocomplete="off"/></div>
|
||||
<div class="field"><label>Expiry Date</label><input type="datetime-local" id="editXrayExpiry" style="color-scheme:dark;"/></div>
|
||||
<div class="field"><label>Max Connections <span class="hint">(0 = unlimited)</span></label><input type="number" min="0" id="editXrayMaxConns"/></div>
|
||||
<div class="field"><label>Data quota (GB) <span class="hint">0 = unlimited · 1024 = 1 TB</span></label><input type="number" min="0" step="0.01" id="editXrayQuotaGB"/></div>
|
||||
<div class="field"><label>When quota is reached</label><select id="editXrayQuotaAction"><option value="block">Block user</option><option value="throttle">Reduce speed</option></select></div>
|
||||
<div class="field"><label>Post-quota speed (Mb/s)</label><input type="number" min="1" id="editXrayQuotaThrottle" value="1"/></div>
|
||||
<div class="field"><label>Current usage</label><input id="editXrayUsage" readonly value="0 B"/></div>
|
||||
<div class="field"><label>Reset traffic counter</label><input id="editXrayResetUsage" type="checkbox" style="width:16px;height:16px;margin-top:10px;"/></div>
|
||||
</div>
|
||||
<div class="form-actions" style="margin-top:8px;">
|
||||
<button class="btn btn-sm" onclick="saveEditXrayClient()">Save Changes</button>
|
||||
@@ -410,6 +431,17 @@
|
||||
<div class="card-title">Inbounds & Clients</div>
|
||||
<div class="card-actions"><button class="btn btn-ghost btn-sm" id="xLoadInboundsBtn">Reload</button></div>
|
||||
</div>
|
||||
<div class="user-list-controls" id="xrayListControls" aria-label="Xray user list controls">
|
||||
<div class="user-list-control-group">
|
||||
<span class="user-list-control-label" id="xraySortLabel"></span>
|
||||
<div class="user-list-buttons" id="xraySortButtons"></div>
|
||||
</div>
|
||||
<div class="user-list-control-group">
|
||||
<span class="user-list-control-label" id="xrayFilterLabel"></span>
|
||||
<div class="user-list-buttons" id="xrayFilterButtons"></div>
|
||||
</div>
|
||||
<span class="chip user-list-count" id="xrayListCount"></span>
|
||||
</div>
|
||||
<div id="inboundsContainer">
|
||||
<div class="hint" style="padding:8px 0;">Loading inbounds…</div>
|
||||
</div>
|
||||
@@ -428,6 +460,9 @@
|
||||
<div class="field"><label>Email / identificação</label><input id="xCreateEmail" autocomplete="off" placeholder="cliente@example"/></div>
|
||||
<div class="field"><label>Expira em</label><input id="xCreateExpiry" type="datetime-local"/></div>
|
||||
<div class="field"><label>Máximo de conexões <span class="hint">0 = ilimitado</span></label><input id="xCreateMaxConns" type="number" min="0" value="0"/></div>
|
||||
<div class="field"><label>Cota de dados (GB) <span class="hint">0 = ilimitado · 1024 = 1 TB</span></label><input id="xCreateQuotaGB" type="number" min="0" step="0.01" value="0"/></div>
|
||||
<div class="field"><label>Ao atingir a cota</label><select id="xCreateQuotaAction"><option value="block">Bloquear usuário</option><option value="throttle">Reduzir velocidade</option></select></div>
|
||||
<div class="field"><label>Velocidade após a cota (Mb/s)</label><input id="xCreateQuotaThrottle" type="number" min="1" value="1"/></div>
|
||||
</div>
|
||||
<div class="form-actions"><button class="btn" id="xCreateClientBtn" type="submit">Criar usuário</button><button class="btn btn-ghost" id="xCreateCancelBtn" type="button">Voltar aos usuários</button></div>
|
||||
<div class="statusbar"><span id="xCreateClientStatus">Preencha os dados do novo cliente.</span></div>
|
||||
@@ -448,29 +483,28 @@
|
||||
</div>
|
||||
<!-- Wizard pane -->
|
||||
<div id="xrayWizardPane">
|
||||
<section class="shared-endpoint-card">
|
||||
<div class="shared-endpoint-head">
|
||||
<div><span class="page-kicker">Shared XHTTP endpoint</span><h3>Um domínio e uma porta</h3><p>O protocolo selecionado usa <code>/</code>; SSH usa <code>/ssh</code>. Disponível no modo Xray nativo.</p></div>
|
||||
<span class="chip green">path routing</span>
|
||||
<section class="xray-inbound-launcher">
|
||||
<div class="xray-inbound-launcher-copy">
|
||||
<span class="page-kicker">Gerenciar inbounds</span>
|
||||
<h3>Adicionar um novo inbound</h3>
|
||||
<p>Crie um inbound em branco ou instale automaticamente o padrão usado com Azion XHTTP. Para alterar um inbound existente, use somente o botão <strong>Editar</strong> no cartão dele.</p>
|
||||
</div>
|
||||
<div class="shared-route-preview" aria-label="Shared endpoint route preview">
|
||||
<span><strong id="sharedProxyRouteLabel">VLESS</strong><code>/</code></span>
|
||||
<i></i>
|
||||
<span><strong>SSH</strong><code>/ssh</code></span>
|
||||
<div class="xray-inbound-launcher-actions">
|
||||
<button class="btn" id="wzAddInboundBtn" type="button" onclick="wzToggleAddInbound()">+ Adicionar inbound</button>
|
||||
<button class="btn btn-soft" id="wzAzionDefaultBtn" type="button" onclick="createAzionDefaultXHTTP()">Criar padrão Azion XHTTP</button>
|
||||
</div>
|
||||
<div class="form-grid shared-endpoint-grid">
|
||||
<div class="field"><label>Protocolo em /</label><select id="sharedXHTTPProtocol"><option value="vless">VLESS</option><option value="vmess">VMess</option></select></div>
|
||||
<div class="field"><label>Porta compartilhada</label><input id="sharedXHTTPPort" type="number" min="1" max="65535" value="443"/></div>
|
||||
<div class="field"><label>IP de listen</label><input id="sharedXHTTPListen" value="0.0.0.0" placeholder="0.0.0.0"/></div>
|
||||
<div class="field"><label>Host HTTP <span class="hint">opcional</span></label><input id="sharedXHTTPHost" placeholder="vpn.seudominio.com"/></div>
|
||||
<div class="field"><label>Modo XHTTP</label><select id="sharedXHTTPMode"><option value="auto">auto</option><option value="packet-up">packet-up</option><option value="stream-up">stream-up</option><option value="stream-down">stream-down</option><option value="stream-one">stream-one</option></select></div>
|
||||
<div class="field"><label>Segurança</label><select id="sharedXHTTPSecurity"><option value="none">Sem TLS</option><option value="tls">TLS</option></select></div>
|
||||
<div class="field shared-tls-field hidden"><label>Arquivo do certificado</label><input id="sharedXHTTPCert" placeholder="/opt/sshpanel/certs/domain/cert.pem"/></div>
|
||||
<div class="field shared-tls-field hidden"><label>Arquivo da chave</label><input id="sharedXHTTPKey" placeholder="/opt/sshpanel/certs/domain/key.pem"/></div>
|
||||
<div class="azion-preset-summary" aria-label="Configuração criada pelo padrão Azion XHTTP">
|
||||
<span><small>Listen</small><strong>0.0.0.0:443</strong></span>
|
||||
<span><small>TLS</small><strong>example.com</strong></span>
|
||||
<span><small>VLESS</small><strong>/</strong></span>
|
||||
<span><small>SSH</small><strong>/ssh</strong></span>
|
||||
</div>
|
||||
<div class="shared-endpoint-actions"><span id="sharedXHTTPStatus" class="hint">Carregue a configuração para detectar um endpoint existente.</span><button class="btn" id="sharedXHTTPApplyBtn" type="button">Criar / atualizar endpoint</button></div>
|
||||
<span id="wzAzionDefaultStatus" class="hint">O padrão cria o certificado autoassinado, habilita TLS e salva/reinicia o Xray automaticamente.</span>
|
||||
</section>
|
||||
|
||||
<!-- The shared add/edit editor is mounted here when opened. -->
|
||||
<div id="wzInboundEditorAnchor"></div>
|
||||
|
||||
<aside class="legacy-xhttp-migration" aria-label="Migração de configuração XHTTP antiga">
|
||||
<span class="legacy-xhttp-icon" aria-hidden="true">SSH+</span>
|
||||
<div>
|
||||
@@ -490,13 +524,12 @@
|
||||
<option value="debug">debug</option>
|
||||
</select>
|
||||
</div>
|
||||
<div class="visual-config-toolbar-copy"><strong>Inbounds configurados</strong><span>Edite qualquer cartão visualmente ou use JSON para campos avançados.</span></div>
|
||||
<button class="btn btn-ghost btn-sm" type="button" onclick="wzToggleAddInbound()">+ Novo inbound</button>
|
||||
<div class="visual-config-toolbar-copy"><strong>Inbounds configurados</strong><span>O editor permanece fechado. Clique em Editar somente no inbound que deseja alterar.</span></div>
|
||||
</div>
|
||||
<div id="wzInboundsList" class="visual-inbound-list"></div>
|
||||
<!-- Add inbound form -->
|
||||
<!-- Shared add/edit form: hidden until the user explicitly adds or edits. -->
|
||||
<div id="wzAddInboundForm" class="visual-inbound-editor hidden">
|
||||
<div class="visual-editor-heading"><div><span class="page-kicker">Visual editor</span><h3 id="wzInboundFormTitle">Novo inbound</h3></div><span id="wzEditingBadge" class="chip hidden">editing</span></div>
|
||||
<div class="visual-editor-heading"><div><span class="page-kicker" id="wzInboundFormKicker">Novo inbound</span><h3 id="wzInboundFormTitle">Adicionar inbound</h3></div><span id="wzEditingBadge" class="chip hidden">Editando</span></div>
|
||||
<div class="form-grid">
|
||||
<div class="field">
|
||||
<label>Protocol</label>
|
||||
@@ -807,6 +840,7 @@
|
||||
<div class="field"><label>SSH Idle Timeout <span class="hint">0s/off = disabled</span></label><input type="text" id="managedCfgSSHIdleTimeout" placeholder="0s" title="Keep disabled for VPN/XHTTP connections."/></div>
|
||||
<label style="font-size:.73rem;display:flex;align-items:center;gap:5px;cursor:pointer;"><input type="checkbox" id="managedCfgQuiet"/> Quiet Logs</label>
|
||||
<label style="font-size:.73rem;display:flex;align-items:center;gap:5px;cursor:pointer;"><input type="checkbox" id="managedCfgUserCount"/> User Count Display</label>
|
||||
<label style="font-size:.73rem;display:flex;align-items:center;gap:5px;cursor:pointer;" title="Verify SSH logins against the Linux system password (/etc/shadow); regular accounts (UID ≥ 1000) are auto-imported."><input type="checkbox" id="managedCfgPamAuth"/> Linux PAM Login (auto-import)</label>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
@@ -831,7 +865,7 @@
|
||||
</div>
|
||||
<div id="managedDnsttFields" class="form-grid" style="opacity:.4;pointer-events:none;">
|
||||
<div class="field" style="grid-column:1/-1"><label>NS / Root Domains <span class="hint">one per line</span></label><textarea id="managedCfgDnsttDomains" rows="3" placeholder="t.example.com t.local.lan"></textarea></div>
|
||||
<div class="field"><label>UDP Listen</label><input type="text" id="managedCfgDnsttUDP" placeholder="[::]:5300"/></div>
|
||||
<div class="field"><label>UDP Listen</label><input type="text" id="managedCfgDnsttUDP" placeholder="0.0.0.0:5300"/></div>
|
||||
<label style="font-size:.73rem;display:flex;align-items:center;gap:5px;cursor:pointer;grid-column:1/-1"><input type="checkbox" id="managedCfgDnsttFakeEnabled"/> Built-in Local DNS / Fake DNS</label>
|
||||
<div class="field"><label>Local DNS Listen <span class="hint">IPv6 ok</span></label><input type="text" id="managedCfgDnsttFakeListen" placeholder="[2001:db8::1234]:53"/></div>
|
||||
<div class="field"><label>Local DNS Domain</label><input type="text" id="managedCfgDnsttFakeDomain" placeholder="t.local.lan"/></div>
|
||||
@@ -1268,6 +1302,9 @@
|
||||
<label style="font-size:.73rem;display:flex;align-items:center;gap:5px;cursor:pointer;">
|
||||
<input type="checkbox" id="cfgUserCount"/> User Count Display
|
||||
</label>
|
||||
<label style="font-size:.73rem;display:flex;align-items:center;gap:5px;cursor:pointer;" title="Verify SSH logins against the Linux system password (/etc/shadow). Regular accounts (UID ≥ 1000) that log in successfully are auto-imported into the panel.">
|
||||
<input type="checkbox" id="cfgPamAuth"/> Linux PAM Login (auto-import)
|
||||
</label>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
@@ -1305,7 +1342,7 @@
|
||||
</div>
|
||||
<div class="field">
|
||||
<label>UDP Listen</label>
|
||||
<input type="text" id="cfgDnsttUDP" placeholder="[::]:5300"/>
|
||||
<input type="text" id="cfgDnsttUDP" placeholder="0.0.0.0:5300"/>
|
||||
</div>
|
||||
<label style="font-size:.73rem;display:flex;align-items:center;gap:5px;cursor:pointer;grid-column:1/-1">
|
||||
<input type="checkbox" id="cfgDnsttFakeEnabled"/> Built-in Local DNS / Fake DNS
|
||||
@@ -1422,7 +1459,33 @@
|
||||
</section>
|
||||
|
||||
<section class="workspace-section" data-workspace-panel="config" data-workspace-section-panel="tls">
|
||||
<div class="workspace-section-heading"><div><span>04 · Segurança</span><h3>Encaminhadores TLS</h3><p>Crie listeners TLS com certificado automático, colado ou armazenado em arquivo.</p></div></div>
|
||||
<div class="workspace-section-heading"><div><span>04 · Segurança</span><h3>Encaminhadores TLS</h3><p>Gerencie os certificados do servidor e crie listeners TLS com certificado automático, colado ou armazenado em arquivo.</p></div></div>
|
||||
|
||||
<!-- TLS Certificates -->
|
||||
<div class="card" style="margin-top:12px">
|
||||
<div class="card-hdr">
|
||||
<div class="card-title">Certificados TLS <span class="chip" id="tlsCertsCountChip">0</span></div>
|
||||
<span class="chip green">live</span>
|
||||
<button class="btn btn-ghost btn-sm" type="button" onclick="loadTLSCertificates()">Recarregar lista</button>
|
||||
<button class="btn btn-ghost btn-sm" type="button" onclick="toggleNewCertForm()">+ Novo</button>
|
||||
</div>
|
||||
<div class="hint" style="margin-top:2px;">Cole o <code>fullchain.pem</code> e o <code>privkey.pem</code> para renovar um certificado. Os arquivos são substituídos no mesmo caminho (com backup <code>.bak</code>), então nenhuma configuração precisa ser alterada, e os listeners TLS e inbounds Xray que usam o certificado são recarregados na hora.</div>
|
||||
<div id="tlsCertsList" style="margin-top:8px;"></div>
|
||||
<div id="newCertPanel" class="hidden" style="border:1px solid var(--border);border-radius:8px;padding:10px;margin-top:8px;">
|
||||
<div class="field"><label>Nome <span class="hint">(pasta de armazenamento, ex.: meu-dominio)</span></label><input type="text" id="newCertName" placeholder="meu-dominio"/></div>
|
||||
<div style="display:grid;grid-template-columns:1fr 1fr;gap:8px;margin-top:8px;">
|
||||
<div class="field"><label>fullchain.pem <span class="hint">(certificado + intermediários)</span></label><textarea id="newCertFullchain" rows="6" placeholder="-----BEGIN CERTIFICATE----- …" style="font-family:monospace;font-size:.7rem;width:100%;box-sizing:border-box;resize:vertical;background:var(--input-bg);border:1px solid var(--border);border-radius:4px;color:inherit;padding:4px;"></textarea></div>
|
||||
<div class="field"><label>privkey.pem <span class="hint">(chave privada)</span></label><textarea id="newCertPrivkey" rows="6" placeholder="-----BEGIN PRIVATE KEY----- …" style="font-family:monospace;font-size:.7rem;width:100%;box-sizing:border-box;resize:vertical;background:var(--input-bg);border:1px solid var(--border);border-radius:4px;color:inherit;padding:4px;"></textarea></div>
|
||||
</div>
|
||||
<div class="form-actions" style="margin-top:8px;">
|
||||
<button class="btn btn-sm" type="button" onclick="saveNewCert()">Salvar certificado</button>
|
||||
<button class="btn btn-ghost btn-sm" type="button" onclick="toggleNewCertForm()">Cancelar</button>
|
||||
</div>
|
||||
<div id="newCertStatus" class="hint" style="margin-top:4px;"></div>
|
||||
</div>
|
||||
<div id="tlsCertsStatus" class="hint" style="margin-top:6px;"></div>
|
||||
</div>
|
||||
|
||||
<!-- TLS Forwarders -->
|
||||
<div class="card" style="margin-top:12px">
|
||||
<div class="card-hdr">
|
||||
@@ -1493,12 +1556,13 @@
|
||||
<div class="grid2" style="margin-top:10px;gap:8px;">
|
||||
<div class="field"><label>Go CPU threads (GOMAXPROCS)</label><input type="number" min="0" id="cfgXrayRuntimeGomaxprocs" placeholder="0 = all CPU cores"/></div>
|
||||
<div class="field"><label>Global mux backend sessions</label><input type="number" min="1" id="cfgXrayMuxGlobalSessions" placeholder="32768"/></div>
|
||||
<div class="field" style="grid-column:1/-1;"><label>Transport and XHTTP admission</label><div class="hint">Unlimited for VPN traffic. There is no global HTTP request, HTTP/2 stream, transport-connection, or XHTTP-session count cap.</div></div>
|
||||
<label style="font-size:.73rem;display:flex;align-items:center;gap:5px;cursor:pointer;grid-column:1/-1"><input type="checkbox" id="cfgXrayTracePackets"/> Trace every XHTTP/mux packet <span class="hint">debug only, slows QUIC</span></label>
|
||||
<div class="card-actions" style="grid-column:1/-1;">
|
||||
<button class="btn btn-ghost btn-sm" type="button" onclick="setXrayNativeTuningDefaults('2k')">Apply 2K defaults</button>
|
||||
<button class="btn btn-ghost btn-sm" type="button" onclick="setXrayNativeTuningDefaults('high')">Apply high-traffic VPN defaults</button>
|
||||
<button class="btn btn-ghost btn-sm" type="button" onclick="setXrayNativeTuningDefaults('safe')">Apply safe defaults</button>
|
||||
</div>
|
||||
<div class="hint" style="grid-column:1/-1;margin-top:-4px;">Transport buffers (HTTP/2 flow control, XHTTP reorder buffer, mux/UDP buffers) are fixed to xray-core defaults and no longer tunable, so they can't be misconfigured. Go CPU threads = 0 means all detected cores. Saved in the panel config and applied live on restart/reload.</div>
|
||||
<div class="hint" style="grid-column:1/-1;margin-top:-4px;">XHTTP is handled as VPN tunnel traffic: packet requests and reassembly are limited only by bounded byte backpressure, never by a request count. Existing saved web-style caps are ignored automatically after update. Per-user max_conns, quota, and bandwidth policies still work normally.</div>
|
||||
</div>
|
||||
</details>
|
||||
</div>
|
||||
@@ -1538,17 +1602,17 @@
|
||||
<!-- app.js was split into ordered modules for maintainability. They are plain
|
||||
classic scripts sharing one global scope; `defer` preserves execution order,
|
||||
so behavior is identical to the old single file. Keep this load order. -->
|
||||
<script defer src="assets/js/01-core.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/02-shell.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/03-ssh-users.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/04-xray.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/05-resellers.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/06-servers.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/07-stats-logs.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/08-server-config.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/09-xray-wizard.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/11-update-status.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/12-bot.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/10-boot.js?v=20260713sections5"></script>
|
||||
<script defer src="assets/js/01-core.js?v=20260722xhttpunlimited2"></script>
|
||||
<script defer src="assets/js/02-shell.js?v=20260805certupdate1"></script>
|
||||
<script defer src="assets/js/03-ssh-users.js?v=20260720sshfilters1"></script>
|
||||
<script defer src="assets/js/04-xray.js?v=20260720sshfilters1"></script>
|
||||
<script defer src="assets/js/05-resellers.js?v=20260714pamfix1"></script>
|
||||
<script defer src="assets/js/06-servers.js?v=20260714pamfix1"></script>
|
||||
<script defer src="assets/js/07-stats-logs.js?v=20260714pamfix1"></script>
|
||||
<script defer src="assets/js/08-server-config.js?v=20260805certupdate1"></script>
|
||||
<script defer src="assets/js/09-xray-wizard.js?v=20260724xrayinboundsr4"></script>
|
||||
<script defer src="assets/js/11-update-status.js?v=20260714pamfix1"></script>
|
||||
<script defer src="assets/js/12-bot.js?v=20260714pamfix1"></script>
|
||||
<script defer src="assets/js/10-boot.js?v=20260714pamfix1"></script>
|
||||
</body>
|
||||
</html>
|
||||
|
||||
@@ -586,7 +586,7 @@ func startResellerExpiryChecker(store *Store) {
|
||||
u.IsActive = false
|
||||
adminUsers.set(u)
|
||||
disconnectOwnerUsers(u.Username)
|
||||
removeOwnerXrayClients(ctx, store, u.Username)
|
||||
suspendOwnerXrayClients(ctx, store, u.Username)
|
||||
}
|
||||
|
||||
// Reactivate resellers that have been renewed (inactive but expiry now in future/nil)
|
||||
@@ -602,6 +602,7 @@ func startResellerExpiryChecker(store *Store) {
|
||||
}
|
||||
u.IsActive = true
|
||||
adminUsers.set(u)
|
||||
restoreOwnerXrayClients(ctx, store, u.Username)
|
||||
}
|
||||
|
||||
sessions.cleanup()
|
||||
@@ -862,7 +863,9 @@ func handleCreateReseller(store *Store) http.HandlerFunc {
|
||||
if u.Role == RoleReseller {
|
||||
if !u.IsActive || (u.ExpiresAt != nil && time.Now().After(*u.ExpiresAt)) {
|
||||
disconnectOwnerUsers(u.Username)
|
||||
removeOwnerXrayClients(ctx, store, u.Username)
|
||||
suspendOwnerXrayClients(ctx, store, u.Username)
|
||||
} else {
|
||||
restoreOwnerXrayClients(ctx, store, u.Username)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,231 @@
|
||||
package main
|
||||
|
||||
// Live per-account bandwidth. The panel already keeps cumulative uploaded and
|
||||
// downloaded byte counters for every SSH user and Xray client; this file turns
|
||||
// those counters into a current speed so the UI can show "↑ 12 Mbps ↓ 40 Mbps"
|
||||
// for the whole account instead of only lifetime totals. Speeds are always the
|
||||
// sum of every connection the account has open, because the counters they are
|
||||
// derived from are per account, not per connection.
|
||||
|
||||
import (
|
||||
"math"
|
||||
"strings"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
)
|
||||
|
||||
// bandwidthRate is a smoothed instantaneous speed in bytes per second.
|
||||
type bandwidthRate struct {
|
||||
UpBytesPerSec float64
|
||||
DownBytesPerSec float64
|
||||
}
|
||||
|
||||
func (r bandwidthRate) isZero() bool {
|
||||
return r.UpBytesPerSec == 0 && r.DownBytesPerSec == 0
|
||||
}
|
||||
|
||||
type bandwidthSample struct {
|
||||
up int64
|
||||
down int64
|
||||
at time.Time
|
||||
rate bandwidthRate
|
||||
}
|
||||
|
||||
// bandwidthSampler converts monotonically increasing byte counters into a
|
||||
// speed. Deltas smaller than minSampleInterval are ignored so a double sample
|
||||
// cannot divide by an almost-zero interval, and a counter that moves backwards
|
||||
// (traffic reset, account recreated) re-baselines instead of reporting a
|
||||
// nonsensical negative or huge rate.
|
||||
type bandwidthSampler struct {
|
||||
mu sync.Mutex
|
||||
samples map[string]bandwidthSample
|
||||
|
||||
// tau is the exponential smoothing time constant. Larger values give a
|
||||
// calmer number; zero disables smoothing.
|
||||
tau time.Duration
|
||||
// staleAfter makes Rate report zero for accounts that stopped being
|
||||
// sampled (idle Xray clients dropped by the stats poller, for example),
|
||||
// instead of freezing the last speed on screen forever.
|
||||
staleAfter time.Duration
|
||||
}
|
||||
|
||||
const minBandwidthSampleInterval = 250 * time.Millisecond
|
||||
|
||||
func newBandwidthSampler(tau, staleAfter time.Duration) *bandwidthSampler {
|
||||
return &bandwidthSampler{
|
||||
samples: make(map[string]bandwidthSample),
|
||||
tau: tau,
|
||||
staleAfter: staleAfter,
|
||||
}
|
||||
}
|
||||
|
||||
// Observe records the current cumulative counters for key. The first
|
||||
// observation only establishes a baseline; the rate stays zero until a second
|
||||
// one arrives.
|
||||
func (s *bandwidthSampler) Observe(key string, up, down int64, now time.Time) {
|
||||
if s == nil {
|
||||
return
|
||||
}
|
||||
key = strings.TrimSpace(key)
|
||||
if key == "" {
|
||||
return
|
||||
}
|
||||
if up < 0 {
|
||||
up = 0
|
||||
}
|
||||
if down < 0 {
|
||||
down = 0
|
||||
}
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
prev, ok := s.samples[key]
|
||||
if !ok {
|
||||
s.samples[key] = bandwidthSample{up: up, down: down, at: now}
|
||||
return
|
||||
}
|
||||
// Counters went backwards: the account's traffic was reset or the entry was
|
||||
// recycled. Start over from this value.
|
||||
if up < prev.up || down < prev.down {
|
||||
s.samples[key] = bandwidthSample{up: up, down: down, at: now}
|
||||
return
|
||||
}
|
||||
dt := now.Sub(prev.at)
|
||||
if dt < minBandwidthSampleInterval {
|
||||
return
|
||||
}
|
||||
seconds := dt.Seconds()
|
||||
instant := bandwidthRate{
|
||||
UpBytesPerSec: float64(up-prev.up) / seconds,
|
||||
DownBytesPerSec: float64(down-prev.down) / seconds,
|
||||
}
|
||||
next := instant
|
||||
if s.tau > 0 && !prev.rate.isZero() {
|
||||
// alpha derived from the real interval so an irregular sampling
|
||||
// cadence still converges on the true average.
|
||||
alpha := 1 - math.Exp(-seconds/s.tau.Seconds())
|
||||
if alpha > 1 {
|
||||
alpha = 1
|
||||
}
|
||||
next = bandwidthRate{
|
||||
UpBytesPerSec: prev.rate.UpBytesPerSec + alpha*(instant.UpBytesPerSec-prev.rate.UpBytesPerSec),
|
||||
DownBytesPerSec: prev.rate.DownBytesPerSec + alpha*(instant.DownBytesPerSec-prev.rate.DownBytesPerSec),
|
||||
}
|
||||
}
|
||||
if next.UpBytesPerSec < 0 {
|
||||
next.UpBytesPerSec = 0
|
||||
}
|
||||
if next.DownBytesPerSec < 0 {
|
||||
next.DownBytesPerSec = 0
|
||||
}
|
||||
s.samples[key] = bandwidthSample{up: up, down: down, at: now, rate: next}
|
||||
}
|
||||
|
||||
// Rate returns the last known speed for key. Stale entries report zero.
|
||||
func (s *bandwidthSampler) Rate(key string) (bandwidthRate, bool) {
|
||||
if s == nil {
|
||||
return bandwidthRate{}, false
|
||||
}
|
||||
key = strings.TrimSpace(key)
|
||||
if key == "" {
|
||||
return bandwidthRate{}, false
|
||||
}
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
return s.rateLocked(key)
|
||||
}
|
||||
|
||||
// RateForKeys returns the first known speed among keys. Xray clients are
|
||||
// tracked under their UUID in native mode and under their email in external
|
||||
// mode, so callers pass every identifier the client may be stored under.
|
||||
func (s *bandwidthSampler) RateForKeys(keys ...string) (bandwidthRate, bool) {
|
||||
if s == nil {
|
||||
return bandwidthRate{}, false
|
||||
}
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
for _, key := range keys {
|
||||
key = strings.TrimSpace(key)
|
||||
if key == "" {
|
||||
continue
|
||||
}
|
||||
if rate, ok := s.rateLocked(key); ok {
|
||||
return rate, true
|
||||
}
|
||||
}
|
||||
return bandwidthRate{}, false
|
||||
}
|
||||
|
||||
func (s *bandwidthSampler) rateLocked(key string) (bandwidthRate, bool) {
|
||||
sample, ok := s.samples[key]
|
||||
if !ok {
|
||||
return bandwidthRate{}, false
|
||||
}
|
||||
if s.staleAfter > 0 && !sample.at.IsZero() && time.Since(sample.at) > s.staleAfter {
|
||||
return bandwidthRate{}, true
|
||||
}
|
||||
return sample.rate, true
|
||||
}
|
||||
|
||||
// Retain drops every tracked key that is not in keep, so the map cannot grow
|
||||
// forever as accounts are deleted or recreated.
|
||||
func (s *bandwidthSampler) Retain(keep map[string]struct{}) {
|
||||
if s == nil {
|
||||
return
|
||||
}
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
for key := range s.samples {
|
||||
if _, ok := keep[key]; !ok {
|
||||
delete(s.samples, key)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ---- SSH accounts ----
|
||||
|
||||
const sshBandwidthSampleInterval = 2 * time.Second
|
||||
|
||||
var sshBandwidth = newBandwidthSampler(5*time.Second, 20*time.Second)
|
||||
|
||||
func startSSHUserRateSampler() {
|
||||
go func() {
|
||||
ticker := time.NewTicker(sshBandwidthSampleInterval)
|
||||
defer ticker.Stop()
|
||||
for range ticker.C {
|
||||
sampleSSHUserRates(time.Now())
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
func sampleSSHUserRates(now time.Time) {
|
||||
if userMgr == nil {
|
||||
return
|
||||
}
|
||||
states := userMgr.List()
|
||||
active := make(map[string]struct{}, len(states))
|
||||
for _, u := range states {
|
||||
if u == nil {
|
||||
continue
|
||||
}
|
||||
u.mu.Lock()
|
||||
username := strings.TrimSpace(u.Cfg.Username)
|
||||
u.mu.Unlock()
|
||||
if username == "" {
|
||||
continue
|
||||
}
|
||||
sshBandwidth.Observe(
|
||||
username,
|
||||
atomic.LoadInt64(&u.TotalUplinkBytes),
|
||||
atomic.LoadInt64(&u.TotalDownlinkBytes),
|
||||
now,
|
||||
)
|
||||
active[username] = struct{}{}
|
||||
}
|
||||
sshBandwidth.Retain(active)
|
||||
}
|
||||
|
||||
func sshUserRate(username string) bandwidthRate {
|
||||
rate, _ := sshBandwidth.Rate(username)
|
||||
return rate
|
||||
}
|
||||
@@ -0,0 +1,121 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestBandwidthSamplerFirstObservationIsBaselineOnly(t *testing.T) {
|
||||
s := newBandwidthSampler(0, time.Minute)
|
||||
now := time.Now()
|
||||
s.Observe("bob", 10_000, 20_000, now)
|
||||
rate, ok := s.Rate("bob")
|
||||
if !ok {
|
||||
t.Fatal("expected the account to be tracked after the first observation")
|
||||
}
|
||||
if !rate.isZero() {
|
||||
t.Fatalf("first observation must not report a speed, got %+v", rate)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBandwidthSamplerComputesBytesPerSecond(t *testing.T) {
|
||||
s := newBandwidthSampler(0, time.Minute) // no smoothing: exact delta/dt
|
||||
now := time.Now()
|
||||
s.Observe("bob", 0, 0, now)
|
||||
// 2 MB up and 10 MB down over 2 seconds.
|
||||
s.Observe("bob", 2<<20, 10<<20, now.Add(2*time.Second))
|
||||
rate, _ := s.Rate("bob")
|
||||
if wantUp := float64(1 << 20); rate.UpBytesPerSec != wantUp {
|
||||
t.Fatalf("up = %v, want %v", rate.UpBytesPerSec, wantUp)
|
||||
}
|
||||
if wantDown := float64(5 << 20); rate.DownBytesPerSec != wantDown {
|
||||
t.Fatalf("down = %v, want %v", rate.DownBytesPerSec, wantDown)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBandwidthSamplerIgnoresSamplesTakenTooCloseTogether(t *testing.T) {
|
||||
s := newBandwidthSampler(0, time.Minute)
|
||||
now := time.Now()
|
||||
s.Observe("bob", 0, 0, now)
|
||||
s.Observe("bob", 5<<20, 5<<20, now.Add(10*time.Millisecond))
|
||||
rate, _ := s.Rate("bob")
|
||||
if !rate.isZero() {
|
||||
t.Fatalf("a 10ms interval must not produce a speed, got %+v", rate)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBandwidthSamplerRebaselinesAfterTrafficReset(t *testing.T) {
|
||||
s := newBandwidthSampler(0, time.Minute)
|
||||
now := time.Now()
|
||||
s.Observe("bob", 0, 0, now)
|
||||
s.Observe("bob", 4<<20, 4<<20, now.Add(2*time.Second))
|
||||
// Panel reset the account's traffic: counters go back to zero.
|
||||
s.Observe("bob", 0, 0, now.Add(4*time.Second))
|
||||
rate, _ := s.Rate("bob")
|
||||
if !rate.isZero() {
|
||||
t.Fatalf("counters moving backwards must reset the speed, got %+v", rate)
|
||||
}
|
||||
s.Observe("bob", 2<<20, 0, now.Add(6*time.Second))
|
||||
rate, _ = s.Rate("bob")
|
||||
if wantUp := float64(1 << 20); rate.UpBytesPerSec != wantUp {
|
||||
t.Fatalf("up after reset = %v, want %v", rate.UpBytesPerSec, wantUp)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBandwidthSamplerReportsZeroWhenIdle(t *testing.T) {
|
||||
s := newBandwidthSampler(0, time.Minute)
|
||||
now := time.Now()
|
||||
s.Observe("bob", 0, 0, now)
|
||||
s.Observe("bob", 4<<20, 4<<20, now.Add(2*time.Second))
|
||||
s.Observe("bob", 4<<20, 4<<20, now.Add(4*time.Second))
|
||||
rate, _ := s.Rate("bob")
|
||||
if !rate.isZero() {
|
||||
t.Fatalf("unchanged counters must report an idle account, got %+v", rate)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBandwidthSamplerDropsStaleSpeeds(t *testing.T) {
|
||||
s := newBandwidthSampler(0, time.Second)
|
||||
now := time.Now().Add(-time.Hour)
|
||||
s.Observe("bob", 0, 0, now)
|
||||
s.Observe("bob", 4<<20, 4<<20, now.Add(2*time.Second))
|
||||
rate, ok := s.Rate("bob")
|
||||
if !ok {
|
||||
t.Fatal("expected the account to still be tracked")
|
||||
}
|
||||
if !rate.isZero() {
|
||||
t.Fatalf("an hour-old sample must not still report a speed, got %+v", rate)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBandwidthSamplerSmoothsWithTimeConstant(t *testing.T) {
|
||||
s := newBandwidthSampler(5*time.Second, time.Minute)
|
||||
now := time.Now()
|
||||
s.Observe("bob", 0, 0, now)
|
||||
// First real sample has no previous rate to blend with, so it lands exactly.
|
||||
s.Observe("bob", 2<<20, 0, now.Add(2*time.Second))
|
||||
first, _ := s.Rate("bob")
|
||||
if first.UpBytesPerSec != float64(1<<20) {
|
||||
t.Fatalf("first speed = %v, want %v", first.UpBytesPerSec, float64(1<<20))
|
||||
}
|
||||
// Traffic stops: the smoothed value has to fall without jumping to zero.
|
||||
s.Observe("bob", 2<<20, 0, now.Add(4*time.Second))
|
||||
second, _ := s.Rate("bob")
|
||||
if second.UpBytesPerSec <= 0 || second.UpBytesPerSec >= first.UpBytesPerSec {
|
||||
t.Fatalf("smoothed speed = %v, want a value between 0 and %v", second.UpBytesPerSec, first.UpBytesPerSec)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBandwidthSamplerRateForKeysAndRetain(t *testing.T) {
|
||||
s := newBandwidthSampler(0, time.Minute)
|
||||
now := time.Now()
|
||||
s.Observe("uuid-1", 0, 0, now)
|
||||
s.Observe("uuid-1", 1<<20, 0, now.Add(1*time.Second))
|
||||
if _, ok := s.RateForKeys("", "unknown@example", "uuid-1"); !ok {
|
||||
t.Fatal("RateForKeys must find the client under any of its identifiers")
|
||||
}
|
||||
s.Retain(map[string]struct{}{"uuid-2": {}})
|
||||
if _, ok := s.Rate("uuid-1"); ok {
|
||||
t.Fatal("Retain must drop accounts that no longer exist")
|
||||
}
|
||||
}
|
||||
+28
-4
@@ -11,7 +11,7 @@ import (
|
||||
const (
|
||||
defaultMainListen = "0.0.0.0:80"
|
||||
defaultExtraListen = "0.0.0.0:8080"
|
||||
defaultDNSTTListen = "[::]:5300"
|
||||
defaultDNSTTListen = "0.0.0.0:5300"
|
||||
defaultUDPGWListen = "0.0.0.0:7400"
|
||||
)
|
||||
|
||||
@@ -109,9 +109,10 @@ func normalizeRuntimePorts(cfg *Config) []string {
|
||||
}
|
||||
}
|
||||
|
||||
cfg.DNSTT.UDPListen = strings.TrimSpace(cfg.DNSTT.UDPListen)
|
||||
if cfg.DNSTT.UDPListen == "" {
|
||||
cfg.DNSTT.UDPListen = defaultDNSTTListen
|
||||
var migratedLegacyDNSTTWildcard bool
|
||||
cfg.DNSTT.UDPListen, migratedLegacyDNSTTWildcard = normalizeDNSTTListenDefault(cfg.DNSTT.UDPListen)
|
||||
if migratedLegacyDNSTTWildcard {
|
||||
warn("DNSTT legacy default [::]:5300 is IPv6-only; using IPv4 default %s", cfg.DNSTT.UDPListen)
|
||||
}
|
||||
if err := udpAddrAvailableForDNSTT(cfg.DNSTT.UDPListen); err != nil {
|
||||
old := cfg.DNSTT.UDPListen
|
||||
@@ -295,6 +296,29 @@ func normalizeDNSTTDomainList(primary string, domains []string) []string {
|
||||
return out
|
||||
}
|
||||
|
||||
// normalizeDNSTTListenDefault keeps explicit IPv4 and concrete IPv6 listeners,
|
||||
// but migrates the old wildcard IPv6 default. listenDNSTTPacket deliberately
|
||||
// opens IPv6 addresses with udp6, so [::]:5300 never receives IPv4 queries.
|
||||
// Existing installations commonly inherited that value from the old default;
|
||||
// moving only that wildcard/default-port combination makes them work after an
|
||||
// update without changing intentionally selected IPv6 interface addresses.
|
||||
func normalizeDNSTTListenDefault(addr string) (string, bool) {
|
||||
addr = strings.TrimSpace(addr)
|
||||
if addr == "" {
|
||||
return defaultDNSTTListen, false
|
||||
}
|
||||
|
||||
host, port, err := net.SplitHostPort(addr)
|
||||
if err != nil || port != "5300" {
|
||||
return addr, false
|
||||
}
|
||||
ip := net.ParseIP(strings.Trim(host, "[]"))
|
||||
if ip != nil && ip.To4() == nil && ip.IsUnspecified() {
|
||||
return defaultDNSTTListen, true
|
||||
}
|
||||
return addr, false
|
||||
}
|
||||
|
||||
func udpAddrAvailableForDNSTT(addr string) error {
|
||||
if addr == "" {
|
||||
return nil
|
||||
|
||||
@@ -0,0 +1,30 @@
|
||||
package main
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestNormalizeDNSTTListenDefault(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
input string
|
||||
want string
|
||||
migrated bool
|
||||
}{
|
||||
{name: "empty uses IPv4 default", input: "", want: "0.0.0.0:5300"},
|
||||
{name: "whitespace uses IPv4 default", input: " ", want: "0.0.0.0:5300"},
|
||||
{name: "legacy IPv6 wildcard migrates", input: "[::]:5300", want: "0.0.0.0:5300", migrated: true},
|
||||
{name: "expanded legacy wildcard migrates", input: "[0:0:0:0:0:0:0:0]:5300", want: "0.0.0.0:5300", migrated: true},
|
||||
{name: "explicit IPv4 remains", input: "192.0.2.10:53", want: "192.0.2.10:53"},
|
||||
{name: "IPv4 wildcard remains", input: "0.0.0.0:5300", want: "0.0.0.0:5300"},
|
||||
{name: "concrete IPv6 remains", input: "[2001:db8::10]:53", want: "[2001:db8::10]:53"},
|
||||
{name: "IPv6 wildcard on custom port remains", input: "[::]:5301", want: "[::]:5301"},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
got, migrated := normalizeDNSTTListenDefault(tt.input)
|
||||
if got != tt.want || migrated != tt.migrated {
|
||||
t.Fatalf("normalizeDNSTTListenDefault(%q) = (%q, %v), want (%q, %v)", tt.input, got, migrated, tt.want, tt.migrated)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
@@ -19,8 +19,17 @@ func TestSSHIdleTimeoutExplicitValue(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestNativeXHTTPConnectedIdleSweepDisabled(t *testing.T) {
|
||||
if got := nativeXHTTPIdleTimeout(); got != 0 {
|
||||
t.Fatalf("native XHTTP idle timeout = %s, want disabled", got)
|
||||
// The connected-session sweeper is the backstop that reaps XHTTP->SSH sessions
|
||||
// whose stream-down GET context never fires (silent client drop behind a CDN).
|
||||
// Without it those sessions leak fds/goroutines until a process restart, which
|
||||
// is what produced the recurring reboot-only XHTTP 502s. It must stay enabled;
|
||||
// the window is generous so only zero-traffic (dead) sessions are reaped.
|
||||
func TestNativeXHTTPConnectedIdleSweepEnabled(t *testing.T) {
|
||||
got := nativeXHTTPIdleTimeout()
|
||||
if got <= 0 {
|
||||
t.Fatalf("native XHTTP idle timeout = %s, want a positive backstop window", got)
|
||||
}
|
||||
if got != 20*time.Minute {
|
||||
t.Fatalf("native XHTTP idle timeout = %s, want 20m backstop", got)
|
||||
}
|
||||
}
|
||||
|
||||
+232
@@ -0,0 +1,232 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Traditional DES-based crypt(3) — the 13-character, no-"$"-prefix hash used by
|
||||
// old Linux/UNIX systems (e.g. accounts created with perl's crypt() or legacy
|
||||
// SSH-account scripts). Pure Go; no dependency on libcrypt.
|
||||
//
|
||||
// Verified against the canonical vector crypt("rasmuslerdorf","rl") ==
|
||||
// "rl.3StKT.4T8M" (see descrypt_test.go).
|
||||
|
||||
const cryptAlphabet = "./0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
|
||||
|
||||
func crypt64Decode(c byte) int {
|
||||
return strings.IndexByte(cryptAlphabet, c)
|
||||
}
|
||||
|
||||
// ---- Standard DES permutation tables (1-indexed, MSB-first) ----
|
||||
|
||||
var ipTable = []int{
|
||||
58, 50, 42, 34, 26, 18, 10, 2, 60, 52, 44, 36, 28, 20, 12, 4,
|
||||
62, 54, 46, 38, 30, 22, 14, 6, 64, 56, 48, 40, 32, 24, 16, 8,
|
||||
57, 49, 41, 33, 25, 17, 9, 1, 59, 51, 43, 35, 27, 19, 11, 3,
|
||||
61, 53, 45, 37, 29, 21, 13, 5, 63, 55, 47, 39, 31, 23, 15, 7,
|
||||
}
|
||||
|
||||
var fpTable = []int{
|
||||
40, 8, 48, 16, 56, 24, 64, 32, 39, 7, 47, 15, 55, 23, 63, 31,
|
||||
38, 6, 46, 14, 54, 22, 62, 30, 37, 5, 45, 13, 53, 21, 61, 29,
|
||||
36, 4, 44, 12, 52, 20, 60, 28, 35, 3, 43, 11, 51, 19, 59, 27,
|
||||
34, 2, 42, 10, 50, 18, 58, 26, 33, 1, 41, 9, 49, 17, 57, 25,
|
||||
}
|
||||
|
||||
var eTable = []int{
|
||||
32, 1, 2, 3, 4, 5, 4, 5, 6, 7, 8, 9, 8, 9, 10, 11, 12, 13,
|
||||
12, 13, 14, 15, 16, 17, 16, 17, 18, 19, 20, 21, 20, 21, 22, 23, 24, 25,
|
||||
24, 25, 26, 27, 28, 29, 28, 29, 30, 31, 32, 1,
|
||||
}
|
||||
|
||||
var pTable = []int{
|
||||
16, 7, 20, 21, 29, 12, 28, 17, 1, 15, 23, 26, 5, 18, 31, 10,
|
||||
2, 8, 24, 14, 32, 27, 3, 9, 19, 13, 30, 6, 22, 11, 4, 25,
|
||||
}
|
||||
|
||||
var pc1Table = []int{
|
||||
57, 49, 41, 33, 25, 17, 9, 1, 58, 50, 42, 34, 26, 18,
|
||||
10, 2, 59, 51, 43, 35, 27, 19, 11, 3, 60, 52, 44, 36,
|
||||
63, 55, 47, 39, 31, 23, 15, 7, 62, 54, 46, 38, 30, 22,
|
||||
14, 6, 61, 53, 45, 37, 29, 21, 13, 5, 28, 20, 12, 4,
|
||||
}
|
||||
|
||||
var pc2Table = []int{
|
||||
14, 17, 11, 24, 1, 5, 3, 28, 15, 6, 21, 10,
|
||||
23, 19, 12, 4, 26, 8, 16, 7, 27, 20, 13, 2,
|
||||
41, 52, 31, 37, 47, 55, 30, 40, 51, 45, 33, 48,
|
||||
44, 49, 39, 56, 34, 53, 46, 42, 50, 36, 29, 32,
|
||||
}
|
||||
|
||||
var shiftTable = []int{1, 1, 2, 2, 2, 2, 2, 2, 1, 2, 2, 2, 2, 2, 2, 1}
|
||||
|
||||
var sBoxes = [8][64]int{
|
||||
{14, 4, 13, 1, 2, 15, 11, 8, 3, 10, 6, 12, 5, 9, 0, 7,
|
||||
0, 15, 7, 4, 14, 2, 13, 1, 10, 6, 12, 11, 9, 5, 3, 8,
|
||||
4, 1, 14, 8, 13, 6, 2, 11, 15, 12, 9, 7, 3, 10, 5, 0,
|
||||
15, 12, 8, 2, 4, 9, 1, 7, 5, 11, 3, 14, 10, 0, 6, 13},
|
||||
{15, 1, 8, 14, 6, 11, 3, 4, 9, 7, 2, 13, 12, 0, 5, 10,
|
||||
3, 13, 4, 7, 15, 2, 8, 14, 12, 0, 1, 10, 6, 9, 11, 5,
|
||||
0, 14, 7, 11, 10, 4, 13, 1, 5, 8, 12, 6, 9, 3, 2, 15,
|
||||
13, 8, 10, 1, 3, 15, 4, 2, 11, 6, 7, 12, 0, 5, 14, 9},
|
||||
{10, 0, 9, 14, 6, 3, 15, 5, 1, 13, 12, 7, 11, 4, 2, 8,
|
||||
13, 7, 0, 9, 3, 4, 6, 10, 2, 8, 5, 14, 12, 11, 15, 1,
|
||||
13, 6, 4, 9, 8, 15, 3, 0, 11, 1, 2, 12, 5, 10, 14, 7,
|
||||
1, 10, 13, 0, 6, 9, 8, 7, 4, 15, 14, 3, 11, 5, 2, 12},
|
||||
{7, 13, 14, 3, 0, 6, 9, 10, 1, 2, 8, 5, 11, 12, 4, 15,
|
||||
13, 8, 11, 5, 6, 15, 0, 3, 4, 7, 2, 12, 1, 10, 14, 9,
|
||||
10, 6, 9, 0, 12, 11, 7, 13, 15, 1, 3, 14, 5, 2, 8, 4,
|
||||
3, 15, 0, 6, 10, 1, 13, 8, 9, 4, 5, 11, 12, 7, 2, 14},
|
||||
{2, 12, 4, 1, 7, 10, 11, 6, 8, 5, 3, 15, 13, 0, 14, 9,
|
||||
14, 11, 2, 12, 4, 7, 13, 1, 5, 0, 15, 10, 3, 9, 8, 6,
|
||||
4, 2, 1, 11, 10, 13, 7, 8, 15, 9, 12, 5, 6, 3, 0, 14,
|
||||
11, 8, 12, 7, 1, 14, 2, 13, 6, 15, 0, 9, 10, 4, 5, 3},
|
||||
{12, 1, 10, 15, 9, 2, 6, 8, 0, 13, 3, 4, 14, 7, 5, 11,
|
||||
10, 15, 4, 2, 7, 12, 9, 5, 6, 1, 13, 14, 0, 11, 3, 8,
|
||||
9, 14, 15, 5, 2, 8, 12, 3, 7, 0, 4, 10, 1, 13, 11, 6,
|
||||
4, 3, 2, 12, 9, 5, 15, 10, 11, 14, 1, 7, 6, 0, 8, 13},
|
||||
{4, 11, 2, 14, 15, 0, 8, 13, 3, 12, 9, 7, 5, 10, 6, 1,
|
||||
13, 0, 11, 7, 4, 9, 1, 10, 14, 3, 5, 12, 2, 15, 8, 6,
|
||||
1, 4, 11, 13, 12, 3, 7, 14, 10, 15, 6, 8, 0, 5, 9, 2,
|
||||
6, 11, 13, 8, 1, 4, 10, 7, 9, 5, 0, 15, 14, 2, 3, 12},
|
||||
{13, 2, 8, 4, 6, 15, 11, 1, 10, 9, 3, 14, 5, 0, 12, 7,
|
||||
1, 15, 13, 8, 10, 3, 7, 4, 12, 5, 6, 11, 0, 14, 9, 2,
|
||||
7, 11, 4, 1, 9, 12, 14, 2, 0, 6, 10, 13, 15, 3, 5, 8,
|
||||
2, 1, 14, 7, 4, 10, 8, 13, 15, 12, 9, 0, 3, 5, 6, 11},
|
||||
}
|
||||
|
||||
// permute selects bits from in (each element 0/1, MSB-first) per a 1-indexed table.
|
||||
func permute(in []byte, table []int) []byte {
|
||||
out := make([]byte, len(table))
|
||||
for i, pos := range table {
|
||||
out[i] = in[pos-1]
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func keySchedule(key64 []byte) [][]byte {
|
||||
cd := permute(key64, pc1Table) // 56 bits
|
||||
c := cd[:28]
|
||||
d := cd[28:]
|
||||
subkeys := make([][]byte, 16)
|
||||
for i := 0; i < 16; i++ {
|
||||
c = rotl(c, shiftTable[i])
|
||||
d = rotl(d, shiftTable[i])
|
||||
combined := append(append([]byte{}, c...), d...)
|
||||
subkeys[i] = permute(combined, pc2Table) // 48 bits
|
||||
}
|
||||
return subkeys
|
||||
}
|
||||
|
||||
func rotl(b []byte, n int) []byte {
|
||||
out := make([]byte, len(b))
|
||||
for i := range b {
|
||||
out[i] = b[(i+n)%len(b)]
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
// feistel computes f(R, K) with the salt-perturbed E expansion.
|
||||
func feistel(r []byte, k []byte, saltMask [24]bool) []byte {
|
||||
e := permute(r, eTable) // 48 bits
|
||||
// Salt: for i in 0..23, if saltMask[i] swap E-output bits i and i+24.
|
||||
for i := 0; i < 24; i++ {
|
||||
if saltMask[i] {
|
||||
e[i], e[i+24] = e[i+24], e[i]
|
||||
}
|
||||
}
|
||||
x := make([]byte, 48)
|
||||
for i := range x {
|
||||
x[i] = e[i] ^ k[i]
|
||||
}
|
||||
out := make([]byte, 32)
|
||||
for box := 0; box < 8; box++ {
|
||||
off := box * 6
|
||||
row := int(x[off])<<1 | int(x[off+5])
|
||||
col := int(x[off+1])<<3 | int(x[off+2])<<2 | int(x[off+3])<<1 | int(x[off+4])
|
||||
val := sBoxes[box][row*16+col]
|
||||
for bit := 0; bit < 4; bit++ {
|
||||
out[box*4+bit] = byte((val >> (3 - bit)) & 1)
|
||||
}
|
||||
}
|
||||
return permute(out, pTable)
|
||||
}
|
||||
|
||||
func desEncryptBlock(block []byte, subkeys [][]byte, saltMask [24]bool) []byte {
|
||||
ip := permute(block, ipTable)
|
||||
l := ip[:32]
|
||||
r := ip[32:]
|
||||
for i := 0; i < 16; i++ {
|
||||
f := feistel(r, subkeys[i], saltMask)
|
||||
newR := make([]byte, 32)
|
||||
for j := 0; j < 32; j++ {
|
||||
newR[j] = l[j] ^ f[j]
|
||||
}
|
||||
l = r
|
||||
r = newR
|
||||
}
|
||||
pre := append(append([]byte{}, r...), l...) // R16 L16
|
||||
return permute(pre, fpTable)
|
||||
}
|
||||
|
||||
// desCrypt implements the traditional 13-char DES crypt. setting supplies the
|
||||
// 2-char salt (its first two characters).
|
||||
func desCrypt(password, setting string) (string, error) {
|
||||
if len(setting) < 2 {
|
||||
return "", errors.New("descrypt: salt too short")
|
||||
}
|
||||
s0 := crypt64Decode(setting[0])
|
||||
s1 := crypt64Decode(setting[1])
|
||||
if s0 < 0 || s1 < 0 {
|
||||
return "", errors.New("descrypt: bad salt characters")
|
||||
}
|
||||
salt := s0 | (s1 << 6)
|
||||
var saltMask [24]bool
|
||||
for i := 0; i < 24; i++ {
|
||||
if (salt>>i)&1 == 1 {
|
||||
saltMask[i] = true
|
||||
}
|
||||
}
|
||||
|
||||
// Key: first 8 bytes of the password, each char<<1 forms a key byte.
|
||||
key64 := make([]byte, 64)
|
||||
for i := 0; i < 8; i++ {
|
||||
var c byte
|
||||
if i < len(password) {
|
||||
c = password[i]
|
||||
}
|
||||
kb := c << 1
|
||||
for bit := 0; bit < 8; bit++ {
|
||||
key64[i*8+bit] = (kb >> (7 - bit)) & 1
|
||||
}
|
||||
}
|
||||
subkeys := keySchedule(key64)
|
||||
|
||||
block := make([]byte, 64) // all zeros
|
||||
for iter := 0; iter < 25; iter++ {
|
||||
block = desEncryptBlock(block, subkeys, saltMask)
|
||||
}
|
||||
|
||||
return string(setting[0]) + string(setting[1]) + encodeDESOutput(block), nil
|
||||
}
|
||||
|
||||
// encodeDESOutput packs the 64-bit result (MSB-first bit array) into 11
|
||||
// crypt-base64 characters: eleven 6-bit groups read most-significant-bit first,
|
||||
// the last group zero-padded to 6 bits.
|
||||
func encodeDESOutput(block []byte) string {
|
||||
out := make([]byte, 0, 11)
|
||||
for j := 0; j < 11; j++ {
|
||||
v := 0
|
||||
for k := 0; k < 6; k++ {
|
||||
idx := j*6 + k
|
||||
bit := 0
|
||||
if idx < len(block) {
|
||||
bit = int(block[idx])
|
||||
}
|
||||
v = (v << 1) | bit
|
||||
}
|
||||
out = append(out, cryptAlphabet[v])
|
||||
}
|
||||
return string(out)
|
||||
}
|
||||
@@ -0,0 +1,15 @@
|
||||
package main
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestDESCryptCanonical(t *testing.T) {
|
||||
got, err := desCrypt("rasmuslerdorf", "rl")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
want := "rl.3StKT.4T8M"
|
||||
t.Logf("got=%q want=%q", got, want)
|
||||
if got != want {
|
||||
t.Errorf("desCrypt mismatch: got %q want %q", got, want)
|
||||
}
|
||||
}
|
||||
@@ -595,11 +595,8 @@ func startDNSTTInstance(cfg *DNSTTConfig, sshConf *ssh.ServerConfig) error {
|
||||
return err
|
||||
}
|
||||
|
||||
udpListen := cfg.UDPListen
|
||||
if udpListen == "" {
|
||||
udpListen = defaultDNSTTListen
|
||||
cfg.UDPListen = udpListen
|
||||
}
|
||||
udpListen, _ := normalizeDNSTTListenDefault(cfg.UDPListen)
|
||||
cfg.UDPListen = udpListen
|
||||
|
||||
fakeDomains := domains
|
||||
if cfg.FakeDNSEnabled {
|
||||
|
||||
@@ -3,7 +3,9 @@ module shell2
|
||||
go 1.25.4
|
||||
|
||||
require (
|
||||
github.com/GehirnInc/crypt v0.0.0-20230320061759-8cc1b52080c5
|
||||
github.com/lib/pq v1.10.9
|
||||
github.com/openwall/yescrypt-go v1.0.0
|
||||
github.com/xtaci/kcp-go/v5 v5.6.61
|
||||
github.com/xtaci/smux v1.5.50
|
||||
golang.org/x/crypto v0.45.0
|
||||
|
||||
@@ -1,10 +1,12 @@
|
||||
cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=
|
||||
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
|
||||
github.com/GehirnInc/crypt v0.0.0-20230320061759-8cc1b52080c5 h1:IEjq88XO4PuBDcvmjQJcQGg+w+UaafSy8G5Kcb5tBhI=
|
||||
github.com/GehirnInc/crypt v0.0.0-20230320061759-8cc1b52080c5/go.mod h1:exZ0C/1emQJAw5tHOaUDyY1ycttqBAPcxuzf7QbY6ec=
|
||||
github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=
|
||||
github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw=
|
||||
github.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc=
|
||||
github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
|
||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
|
||||
github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=
|
||||
github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=
|
||||
@@ -36,13 +38,15 @@ github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
|
||||
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||
github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw=
|
||||
github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
|
||||
github.com/openwall/yescrypt-go v1.0.0 h1:jsGk48zkFvtUjGVOhYPGh+CS595JmTRcKnpggK2AON4=
|
||||
github.com/openwall/yescrypt-go v1.0.0/go.mod h1:e6CWtFizUEOUttaOjeVMiv1lJaJie3mfOtLJ9CCD6sA=
|
||||
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
||||
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=
|
||||
github.com/stretchr/testify v1.6.1 h1:hDPOHmpOpP40lSULcqw7IrRb/u7w6RpDC9399XyoNd0=
|
||||
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
|
||||
github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
|
||||
github.com/tjfoc/gmsm v1.4.1 h1:aMe1GlZb+0bLjn+cKTPEvvn9oUEBlJitaZiiBwsbgho=
|
||||
github.com/tjfoc/gmsm v1.4.1/go.mod h1:j4INPkHWMrhJb38G+J6W4Tw0AbuN8Thu3PbdVYhVcTE=
|
||||
github.com/xtaci/kcp-go/v5 v5.6.61 h1:ajm12pGuWO+GWQNusPyPESC7Rq0yTC2rEXVYkM8ExOg=
|
||||
|
||||
@@ -199,6 +199,32 @@ func (p *tlsListenerPool) Has(addr string) bool {
|
||||
return ok
|
||||
}
|
||||
|
||||
// Drop closes the listeners for the given addresses so a following Sync rebinds
|
||||
// them. Used after a certificate is replaced on disk: tls.Listen captures the
|
||||
// certificate when the listener is created, so the socket has to be recreated
|
||||
// for new material to be served. Accepted connections are not owned by the pool
|
||||
// and keep running.
|
||||
func (p *tlsListenerPool) Drop(addrs []string, reason string) {
|
||||
if p == nil {
|
||||
return
|
||||
}
|
||||
p.mu.Lock()
|
||||
defer p.mu.Unlock()
|
||||
for _, addr := range addrs {
|
||||
entry, ok := p.entries[addr]
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
_ = entry.Close()
|
||||
delete(p.entries, addr)
|
||||
if reason != "" {
|
||||
log.Printf("hotreload: dropped TLS %s (%s)", addr, reason)
|
||||
} else {
|
||||
log.Printf("hotreload: dropped TLS %s", addr)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (p *tlsListenerPool) StopAll(reason string) {
|
||||
if p == nil {
|
||||
return
|
||||
@@ -335,6 +361,7 @@ func applyFullConfigReload(newCfg *Config) ConfigReloadReport {
|
||||
setDefaultLimits(newCfg.DefaultLimitMbpsUp, newCfg.DefaultLimitMbpsDown)
|
||||
setSSHIdleTimeoutFromConfig(newCfg.SSHIdleTimeout)
|
||||
setMaxTotalConnsFromConfig(newCfg.MaxTotalConnections)
|
||||
setPAMAuthEnabled(newCfg.PAMAuthEnabled)
|
||||
|
||||
// Quiet logging / user count display
|
||||
if newCfg.Quiet {
|
||||
|
||||
+1
-1
@@ -674,7 +674,7 @@ ExecStart=${INSTALL_DIR}/sshpanel -config ${INSTALL_DIR}/config.json
|
||||
Restart=always
|
||||
RestartSec=5
|
||||
User=root
|
||||
LimitNOFILE=65536
|
||||
LimitNOFILE=1048576
|
||||
StandardOutput=journal
|
||||
StandardError=journal
|
||||
|
||||
|
||||
@@ -97,6 +97,13 @@ type Config struct {
|
||||
|
||||
UserCount bool `json:"user_count"`
|
||||
|
||||
// PAMAuthEnabled turns on Linux system-password login for this server. When
|
||||
// true, an SSH login with a username not present in the panel is verified
|
||||
// against /etc/shadow; on success the account (regular users, UID >= 1000)
|
||||
// is auto-imported into the panel. When false, only panel-managed accounts
|
||||
// can log in and previously-imported PAM accounts are refused.
|
||||
PAMAuthEnabled bool `json:"pam_auth_enabled"`
|
||||
|
||||
// SSHIdleTimeout controls how long an authenticated SSH connection may
|
||||
// remain with no bytes moving in either direction before it is closed and
|
||||
// released from the active user count. Empty, "0", or "0s" disables it.
|
||||
@@ -351,6 +358,12 @@ type UserConfig struct {
|
||||
// When false and totp_secret is set, only the TOTP code is accepted.
|
||||
AllowStaticPassword bool `json:"allow_static_password"`
|
||||
|
||||
// UsePAM is a legacy opt-in: when true, the supplied SSH password is
|
||||
// verified against the Linux PAM auth stack (auth phase only) for the
|
||||
// system account matching this username, instead of the panel-managed
|
||||
// Password/TOTP. New users leave this false and keep the script's own auth.
|
||||
UsePAM bool `json:"use_pam"`
|
||||
|
||||
MaxConnections int `json:"max_connections"`
|
||||
ExpiresAt string `json:"expires_at"` // RFC3339 or empty
|
||||
|
||||
@@ -358,6 +371,13 @@ type UserConfig struct {
|
||||
LimitMbpsUp int `json:"limit_mbps_up"` // Mbps upstream
|
||||
LimitMbpsDown int `json:"limit_mbps_down"` // Mbps downstream
|
||||
|
||||
// Persistent data quota. Zero means unlimited. When the total uploaded +
|
||||
// downloaded bytes reaches the quota, QuotaAction either blocks traffic or
|
||||
// throttles the account to QuotaThrottleMbps.
|
||||
DataQuotaBytes int64 `json:"data_quota_bytes"`
|
||||
QuotaAction string `json:"quota_action"`
|
||||
QuotaThrottleMbps int `json:"quota_throttle_mbps"`
|
||||
|
||||
// OwnerUsername is the reseller who created this SSH user. Empty = superadmin-owned.
|
||||
OwnerUsername string `json:"owner_username,omitempty"`
|
||||
}
|
||||
@@ -370,6 +390,18 @@ type UserState struct {
|
||||
mu sync.Mutex
|
||||
ActiveConns int
|
||||
conns map[*ssh.ServerConn]struct{} // active SSH connections for this user
|
||||
|
||||
// Persistent per-user tunnel traffic. totalBytes includes reservations made
|
||||
// by concurrent copy loops, while directional totals only include bytes that
|
||||
// were actually written. The pending counters are flushed to PostgreSQL.
|
||||
TotalUplinkBytes int64
|
||||
TotalDownlinkBytes int64
|
||||
totalBytes int64
|
||||
pendingUplinkBytes int64
|
||||
pendingDownlinkBytes int64
|
||||
trafficMu sync.RWMutex
|
||||
quotaLimiter *rate.Limiter
|
||||
quotaLimiterMbps int
|
||||
}
|
||||
|
||||
type UserManager struct {
|
||||
@@ -384,6 +416,19 @@ func (m *UserManager) Get(username string) (*UserState, bool) {
|
||||
return u, ok
|
||||
}
|
||||
|
||||
// AddIfAbsent inserts u only if no user with the same username exists yet, and
|
||||
// reports whether it was added. Used by PAM auto-import to register a freshly
|
||||
// authenticated system account without clobbering an existing runtime state.
|
||||
func (m *UserManager) AddIfAbsent(u *UserState) bool {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
if _, exists := m.users[u.Cfg.Username]; exists {
|
||||
return false
|
||||
}
|
||||
m.users[u.Cfg.Username] = u
|
||||
return true
|
||||
}
|
||||
|
||||
func (m *UserManager) List() []*UserState {
|
||||
m.mu.RLock()
|
||||
defer m.mu.RUnlock()
|
||||
@@ -533,12 +578,15 @@ var copyBufPool = sync.Pool{
|
||||
// io.Copy, which allocates a fresh 32 KiB buffer per direction per channel and
|
||||
// never pools it — at thousands of channels that churn dominated GC pressure.
|
||||
func copyWithRateLimit(dst io.Writer, src io.Reader, lim *rate.Limiter) (written int64, err error) {
|
||||
return copyWithRateLimitContext(context.Background(), dst, src, lim)
|
||||
}
|
||||
|
||||
func copyWithRateLimitContext(ctx context.Context, dst io.Writer, src io.Reader, lim *rate.Limiter) (written int64, err error) {
|
||||
bufp := copyBufPool.Get().(*[]byte)
|
||||
buf := *bufp
|
||||
defer copyBufPool.Put(bufp)
|
||||
|
||||
var ctx context.Context
|
||||
if lim != nil {
|
||||
if ctx == nil {
|
||||
ctx = context.Background()
|
||||
}
|
||||
|
||||
@@ -1347,17 +1395,29 @@ func (s *Store) EnsureUsersSchema(ctx context.Context) error {
|
||||
expires_at TEXT,
|
||||
limit_mbps_up INT NOT NULL DEFAULT 0,
|
||||
limit_mbps_down INT NOT NULL DEFAULT 0,
|
||||
data_quota_bytes BIGINT NOT NULL DEFAULT 0,
|
||||
quota_action TEXT NOT NULL DEFAULT 'block',
|
||||
quota_throttle_mbps INT NOT NULL DEFAULT 1,
|
||||
total_uplink_bytes BIGINT NOT NULL DEFAULT 0,
|
||||
total_downlink_bytes BIGINT NOT NULL DEFAULT 0,
|
||||
totp_secret TEXT NOT NULL DEFAULT '',
|
||||
totp_period INT NOT NULL DEFAULT 60,
|
||||
totp_window INT NOT NULL DEFAULT 1,
|
||||
totp_digits INT NOT NULL DEFAULT 6,
|
||||
allow_static_password BOOLEAN NOT NULL DEFAULT FALSE
|
||||
allow_static_password BOOLEAN NOT NULL DEFAULT FALSE,
|
||||
use_pam BOOLEAN NOT NULL DEFAULT FALSE
|
||||
)`,
|
||||
`ALTER TABLE ssh_users ADD COLUMN IF NOT EXISTS totp_secret TEXT NOT NULL DEFAULT ''`,
|
||||
`ALTER TABLE ssh_users ADD COLUMN IF NOT EXISTS totp_period INT NOT NULL DEFAULT 60`,
|
||||
`ALTER TABLE ssh_users ADD COLUMN IF NOT EXISTS totp_window INT NOT NULL DEFAULT 1`,
|
||||
`ALTER TABLE ssh_users ADD COLUMN IF NOT EXISTS totp_digits INT NOT NULL DEFAULT 6`,
|
||||
`ALTER TABLE ssh_users ADD COLUMN IF NOT EXISTS allow_static_password BOOLEAN NOT NULL DEFAULT FALSE`,
|
||||
`ALTER TABLE ssh_users ADD COLUMN IF NOT EXISTS use_pam BOOLEAN NOT NULL DEFAULT FALSE`,
|
||||
`ALTER TABLE ssh_users ADD COLUMN IF NOT EXISTS data_quota_bytes BIGINT NOT NULL DEFAULT 0`,
|
||||
`ALTER TABLE ssh_users ADD COLUMN IF NOT EXISTS quota_action TEXT NOT NULL DEFAULT 'block'`,
|
||||
`ALTER TABLE ssh_users ADD COLUMN IF NOT EXISTS quota_throttle_mbps INT NOT NULL DEFAULT 1`,
|
||||
`ALTER TABLE ssh_users ADD COLUMN IF NOT EXISTS total_uplink_bytes BIGINT NOT NULL DEFAULT 0`,
|
||||
`ALTER TABLE ssh_users ADD COLUMN IF NOT EXISTS total_downlink_bytes BIGINT NOT NULL DEFAULT 0`,
|
||||
`ALTER TABLE ssh_users ALTER COLUMN password SET DEFAULT ''`,
|
||||
}
|
||||
for _, stmt := range stmts {
|
||||
@@ -1405,9 +1465,11 @@ func (s *Store) migrateSSHPasswords(ctx context.Context) error {
|
||||
func (s *Store) LoadUsers(ctx context.Context) (map[string]*UserState, error) {
|
||||
rows, err := s.db.QueryContext(ctx, `
|
||||
SELECT username, password, max_connections, expires_at, limit_mbps_up, limit_mbps_down,
|
||||
COALESCE(data_quota_bytes, 0), COALESCE(quota_action, 'block'), COALESCE(quota_throttle_mbps, 1),
|
||||
COALESCE(total_uplink_bytes, 0), COALESCE(total_downlink_bytes, 0),
|
||||
COALESCE(totp_secret, ''), COALESCE(totp_period, 60), COALESCE(totp_window, 1),
|
||||
COALESCE(totp_digits, 6), COALESCE(allow_static_password, FALSE),
|
||||
COALESCE(owner_username, '')
|
||||
COALESCE(use_pam, FALSE), COALESCE(owner_username, '')
|
||||
FROM ssh_users`)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
@@ -1423,15 +1485,22 @@ func (s *Store) LoadUsers(ctx context.Context) (map[string]*UserState, error) {
|
||||
expiresAt sql.NullString
|
||||
limitUp int
|
||||
limitDown int
|
||||
dataQuotaBytes int64
|
||||
quotaAction string
|
||||
quotaThrottleMbps int
|
||||
totalUplinkBytes int64
|
||||
totalDownlinkBytes int64
|
||||
totpSecret string
|
||||
totpPeriod int
|
||||
totpWindow int
|
||||
totpDigits int
|
||||
allowStaticPassword bool
|
||||
usePAM bool
|
||||
ownerUsername string
|
||||
)
|
||||
if err := rows.Scan(&username, &password, &maxConnections, &expiresAt, &limitUp, &limitDown,
|
||||
&totpSecret, &totpPeriod, &totpWindow, &totpDigits, &allowStaticPassword, &ownerUsername); err != nil {
|
||||
&dataQuotaBytes, "aAction, "aThrottleMbps, &totalUplinkBytes, &totalDownlinkBytes,
|
||||
&totpSecret, &totpPeriod, &totpWindow, &totpDigits, &allowStaticPassword, &usePAM, &ownerUsername); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
password, err = openSSHPassword(password)
|
||||
@@ -1445,15 +1514,20 @@ func (s *Store) LoadUsers(ctx context.Context) (map[string]*UserState, error) {
|
||||
MaxConnections: maxConnections,
|
||||
LimitMbpsUp: limitUp,
|
||||
LimitMbpsDown: limitDown,
|
||||
DataQuotaBytes: dataQuotaBytes,
|
||||
QuotaAction: normalizeQuotaAction(quotaAction),
|
||||
QuotaThrottleMbps: quotaThrottleMbps,
|
||||
TOTPSecret: totpSecret,
|
||||
TOTPPeriod: totpPeriod,
|
||||
TOTPWindow: totpWindow,
|
||||
TOTPDigits: totpDigits,
|
||||
AllowStaticPassword: allowStaticPassword,
|
||||
UsePAM: usePAM,
|
||||
OwnerUsername: ownerUsername,
|
||||
}
|
||||
|
||||
st := &UserState{Cfg: cfg}
|
||||
initSSHRuntimeUsage(st, totalUplinkBytes, totalDownlinkBytes)
|
||||
if expiresAt.Valid && expiresAt.String != "" {
|
||||
t, err := time.Parse(time.RFC3339, expiresAt.String)
|
||||
if err != nil {
|
||||
@@ -1480,23 +1554,29 @@ func (s *Store) UpsertUser(ctx context.Context, u UserConfig) error {
|
||||
_, err = s.db.ExecContext(ctx, `
|
||||
INSERT INTO ssh_users (
|
||||
username, password, max_connections, expires_at, limit_mbps_up, limit_mbps_down,
|
||||
totp_secret, totp_period, totp_window, totp_digits, allow_static_password, owner_username
|
||||
data_quota_bytes, quota_action, quota_throttle_mbps,
|
||||
totp_secret, totp_period, totp_window, totp_digits, allow_static_password, use_pam, owner_username
|
||||
)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16)
|
||||
ON CONFLICT (username) DO UPDATE
|
||||
SET password = EXCLUDED.password,
|
||||
max_connections = EXCLUDED.max_connections,
|
||||
expires_at = EXCLUDED.expires_at,
|
||||
limit_mbps_up = EXCLUDED.limit_mbps_up,
|
||||
limit_mbps_down = EXCLUDED.limit_mbps_down,
|
||||
data_quota_bytes = EXCLUDED.data_quota_bytes,
|
||||
quota_action = EXCLUDED.quota_action,
|
||||
quota_throttle_mbps = EXCLUDED.quota_throttle_mbps,
|
||||
totp_secret = EXCLUDED.totp_secret,
|
||||
totp_period = EXCLUDED.totp_period,
|
||||
totp_window = EXCLUDED.totp_window,
|
||||
totp_digits = EXCLUDED.totp_digits,
|
||||
allow_static_password = EXCLUDED.allow_static_password`,
|
||||
allow_static_password = EXCLUDED.allow_static_password,
|
||||
use_pam = EXCLUDED.use_pam`,
|
||||
// owner_username is intentionally excluded from UPDATE — ownership is set at creation only.
|
||||
u.Username, storedPassword, u.MaxConnections, u.ExpiresAt, u.LimitMbpsUp, u.LimitMbpsDown,
|
||||
u.TOTPSecret, u.TOTPPeriod, u.TOTPWindow, u.TOTPDigits, u.AllowStaticPassword, u.OwnerUsername)
|
||||
u.DataQuotaBytes, normalizeQuotaAction(u.QuotaAction), quotaThrottleMbpsOrDefault(u.QuotaThrottleMbps),
|
||||
u.TOTPSecret, u.TOTPPeriod, u.TOTPWindow, u.TOTPDigits, u.AllowStaticPassword, u.UsePAM, u.OwnerUsername)
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -1612,6 +1692,7 @@ func startAdminAPI(store *Store, addr string, adminDir string) {
|
||||
// SSH user management (session required; role-filtered inside handlers)
|
||||
mux.Handle("/api/users", sessionMiddleware(http.HandlerFunc(handleListUsers)))
|
||||
mux.Handle("/api/users/create", sessionMiddleware(http.HandlerFunc(handleCreateUser(store))))
|
||||
mux.Handle("/api/users/reset-traffic", sessionMiddleware(http.HandlerFunc(handleResetUserTraffic(store))))
|
||||
mux.Handle("/api/users/delete", sessionMiddleware(http.HandlerFunc(handleDeleteUser(store))))
|
||||
|
||||
// Server stats: visible to authenticated sessions; reset remains superadmin-only.
|
||||
@@ -1648,12 +1729,15 @@ func startAdminAPI(store *Store, addr string, adminDir string) {
|
||||
mux.Handle("/api/xray/inbounds", sessionMiddleware(http.HandlerFunc(handleXrayInbounds)))
|
||||
mux.Handle("/api/xray/clients/add", sessionMiddleware(http.HandlerFunc(handleXrayClientAdd)))
|
||||
mux.Handle("/api/xray/clients/update", sessionMiddleware(http.HandlerFunc(handleXrayClientUpdate)))
|
||||
mux.Handle("/api/xray/clients/reset-traffic", sessionMiddleware(http.HandlerFunc(handleXrayClientResetTraffic)))
|
||||
mux.Handle("/api/xray/clients/remove", sessionMiddleware(http.HandlerFunc(handleXrayClientRemove)))
|
||||
|
||||
// Superadmin-only: TLS certificate generation
|
||||
mux.Handle("/api/tls/generate-selfsigned", saSession(handleManagedProxyOrLocal(store, handleTLSGenerateSelfSigned)))
|
||||
mux.Handle("/api/tls/letsencrypt", saSession(handleManagedProxyOrLocal(store, handleTLSLetsEncrypt)))
|
||||
mux.Handle("/api/tls/upload-pem", saSession(handleManagedProxyOrLocal(store, handleTLSUploadPEM)))
|
||||
mux.Handle("/api/tls/certs", saSession(handleManagedProxyOrLocal(store, handleTLSCertList)))
|
||||
mux.Handle("/api/tls/certs/update", saSession(handleManagedProxyOrLocal(store, handleTLSCertUpdate)))
|
||||
|
||||
// Superadmin-only: DNSTT key management
|
||||
mux.Handle("/api/dnstt/genkey", saSession(handleManagedProxyOrLocal(store, handleDnsttGenKey)))
|
||||
@@ -1702,20 +1786,32 @@ func startAdminAPI(store *Store, addr string, adminDir string) {
|
||||
|
||||
// UserDTO is returned by the admin API for listing.
|
||||
type UserDTO struct {
|
||||
Username string `json:"username"`
|
||||
ActiveConns int `json:"active_conns"`
|
||||
MaxConnections int `json:"max_connections"`
|
||||
ExpiresAt *time.Time `json:"expires_at,omitempty"`
|
||||
LimitUpMbps int `json:"limit_mbps_up"`
|
||||
LimitDownMbps int `json:"limit_mbps_down"`
|
||||
TOTPSecret string `json:"totp_secret,omitempty"`
|
||||
TOTPPeriod int `json:"totp_period"`
|
||||
TOTPWindow int `json:"totp_window"`
|
||||
TOTPDigits int `json:"totp_digits"`
|
||||
AllowStaticPassword bool `json:"allow_static_password"`
|
||||
TOTPEnabled bool `json:"totp_enabled"`
|
||||
OwnerUsername string `json:"owner_username,omitempty"`
|
||||
ServerID string `json:"server_id,omitempty"`
|
||||
Username string `json:"username"`
|
||||
ActiveConns int `json:"active_conns"`
|
||||
MaxConnections int `json:"max_connections"`
|
||||
ExpiresAt *time.Time `json:"expires_at,omitempty"`
|
||||
LimitUpMbps int `json:"limit_mbps_up"`
|
||||
LimitDownMbps int `json:"limit_mbps_down"`
|
||||
DataQuotaBytes int64 `json:"data_quota_bytes"`
|
||||
QuotaAction string `json:"quota_action"`
|
||||
QuotaThrottleMbps int `json:"quota_throttle_mbps"`
|
||||
TotalUplinkBytes int64 `json:"total_uplink_bytes"`
|
||||
TotalDownlinkBytes int64 `json:"total_downlink_bytes"`
|
||||
TotalBytes int64 `json:"total_bytes"`
|
||||
// Live account-wide speed in bytes per second, summed across every
|
||||
// connection the user has open.
|
||||
UpBytesPerSec float64 `json:"up_bytes_per_sec"`
|
||||
DownBytesPerSec float64 `json:"down_bytes_per_sec"`
|
||||
QuotaExceeded bool `json:"quota_exceeded"`
|
||||
TOTPSecret string `json:"totp_secret,omitempty"`
|
||||
TOTPPeriod int `json:"totp_period"`
|
||||
TOTPWindow int `json:"totp_window"`
|
||||
TOTPDigits int `json:"totp_digits"`
|
||||
AllowStaticPassword bool `json:"allow_static_password"`
|
||||
UsePAM bool `json:"use_pam"`
|
||||
TOTPEnabled bool `json:"totp_enabled"`
|
||||
OwnerUsername string `json:"owner_username,omitempty"`
|
||||
ServerID string `json:"server_id,omitempty"`
|
||||
}
|
||||
|
||||
func handleListUsers(w http.ResponseWriter, r *http.Request) {
|
||||
@@ -1741,12 +1837,17 @@ func handleListUsers(w http.ResponseWriter, r *http.Request) {
|
||||
cfg := u.Cfg
|
||||
expires := u.ExpiresAt
|
||||
u.mu.Unlock()
|
||||
totalUp := atomic.LoadInt64(&u.TotalUplinkBytes)
|
||||
totalDown := atomic.LoadInt64(&u.TotalDownlinkBytes)
|
||||
totalBytes := atomic.LoadInt64(&u.totalBytes)
|
||||
|
||||
// Resellers only see their own users
|
||||
if sess != nil && sess.Role == RoleReseller && cfg.OwnerUsername != sess.Username {
|
||||
continue
|
||||
}
|
||||
|
||||
rate := sshUserRate(cfg.Username)
|
||||
|
||||
out = append(out, UserDTO{
|
||||
Username: cfg.Username,
|
||||
ActiveConns: c,
|
||||
@@ -1754,11 +1855,21 @@ func handleListUsers(w http.ResponseWriter, r *http.Request) {
|
||||
ExpiresAt: expires,
|
||||
LimitUpMbps: cfg.LimitMbpsUp,
|
||||
LimitDownMbps: cfg.LimitMbpsDown,
|
||||
DataQuotaBytes: cfg.DataQuotaBytes,
|
||||
QuotaAction: normalizeQuotaAction(cfg.QuotaAction),
|
||||
QuotaThrottleMbps: quotaThrottleMbpsOrDefault(cfg.QuotaThrottleMbps),
|
||||
TotalUplinkBytes: totalUp,
|
||||
TotalDownlinkBytes: totalDown,
|
||||
TotalBytes: totalBytes,
|
||||
UpBytesPerSec: rate.UpBytesPerSec,
|
||||
DownBytesPerSec: rate.DownBytesPerSec,
|
||||
QuotaExceeded: cfg.DataQuotaBytes > 0 && totalBytes >= cfg.DataQuotaBytes,
|
||||
TOTPSecret: cfg.TOTPSecret,
|
||||
TOTPPeriod: cfg.TOTPPeriod,
|
||||
TOTPWindow: cfg.TOTPWindow,
|
||||
TOTPDigits: cfg.TOTPDigits,
|
||||
AllowStaticPassword: cfg.AllowStaticPassword,
|
||||
UsePAM: cfg.UsePAM,
|
||||
TOTPEnabled: strings.TrimSpace(cfg.TOTPSecret) != "",
|
||||
OwnerUsername: cfg.OwnerUsername,
|
||||
})
|
||||
@@ -1776,11 +1887,16 @@ type UserPayload struct {
|
||||
ExpiresAt string `json:"expires_at"`
|
||||
LimitUpMbps int `json:"limit_mbps_up"`
|
||||
LimitDownMbps int `json:"limit_mbps_down"`
|
||||
DataQuotaBytes int64 `json:"data_quota_bytes"`
|
||||
QuotaAction string `json:"quota_action"`
|
||||
QuotaThrottleMbps int `json:"quota_throttle_mbps"`
|
||||
ResetUsage bool `json:"reset_usage,omitempty"`
|
||||
TOTPSecret string `json:"totp_secret"`
|
||||
TOTPPeriod int `json:"totp_period"`
|
||||
TOTPWindow int `json:"totp_window"`
|
||||
TOTPDigits int `json:"totp_digits"`
|
||||
AllowStaticPassword bool `json:"allow_static_password"`
|
||||
UsePAM bool `json:"use_pam"`
|
||||
OwnerUsername string `json:"owner_username,omitempty"`
|
||||
ServerID string `json:"server_id,omitempty"`
|
||||
}
|
||||
@@ -1801,10 +1917,29 @@ func handleCreateUser(store *Store) http.HandlerFunc {
|
||||
http.Error(w, "invalid json", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
p.Username = strings.TrimSpace(p.Username)
|
||||
if p.Username == "" {
|
||||
http.Error(w, "username required", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if p.MaxConnections < 0 || p.MaxConnections > 10000 {
|
||||
http.Error(w, "max_connections must be between 0 and 10000", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if p.LimitUpMbps < 0 || p.LimitUpMbps > 1000000 || p.LimitDownMbps < 0 || p.LimitDownMbps > 1000000 {
|
||||
http.Error(w, "bandwidth limits must be between 0 and 1000000 Mbps", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if p.ExpiresAt != "" {
|
||||
if _, err := time.Parse(time.RFC3339, p.ExpiresAt); err != nil {
|
||||
http.Error(w, "invalid expires_at (RFC3339 required)", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
}
|
||||
if err := validateQuotaConfig(p.DataQuotaBytes, p.QuotaAction, p.QuotaThrottleMbps); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
ctx := r.Context()
|
||||
if ms, remote, err := managedServerFromID(ctx, store, p.ServerID); err != nil {
|
||||
@@ -1882,7 +2017,9 @@ func handleCreateUser(store *Store) http.HandlerFunc {
|
||||
).Scan(&existing)
|
||||
|
||||
if err == sql.ErrNoRows {
|
||||
if strings.TrimSpace(p.TOTPSecret) == "" {
|
||||
// PAM users authenticate against the system account, so they
|
||||
// need neither a panel password nor a TOTP secret.
|
||||
if strings.TrimSpace(p.TOTPSecret) == "" && !p.UsePAM {
|
||||
http.Error(w, "password or totp_secret required for new user", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
@@ -1928,11 +2065,15 @@ func handleCreateUser(store *Store) http.HandlerFunc {
|
||||
ExpiresAt: p.ExpiresAt,
|
||||
LimitMbpsUp: p.LimitUpMbps,
|
||||
LimitMbpsDown: p.LimitDownMbps,
|
||||
DataQuotaBytes: p.DataQuotaBytes,
|
||||
QuotaAction: normalizeQuotaAction(p.QuotaAction),
|
||||
QuotaThrottleMbps: quotaThrottleMbpsOrDefault(p.QuotaThrottleMbps),
|
||||
TOTPSecret: strings.TrimSpace(p.TOTPSecret),
|
||||
TOTPPeriod: p.TOTPPeriod,
|
||||
TOTPWindow: p.TOTPWindow,
|
||||
TOTPDigits: p.TOTPDigits,
|
||||
AllowStaticPassword: p.AllowStaticPassword,
|
||||
UsePAM: p.UsePAM,
|
||||
OwnerUsername: ownerUsername,
|
||||
}
|
||||
|
||||
@@ -1941,15 +2082,89 @@ func handleCreateUser(store *Store) http.HandlerFunc {
|
||||
http.Error(w, "db error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Force-disconnect all active sessions for this user so new config applies.
|
||||
userMgr.DisconnectUser(p.Username)
|
||||
if p.ResetUsage {
|
||||
if err := resetSSHUserTrafficAccounting(ctx, store, p.Username); err != nil {
|
||||
http.Error(w, "could not reset usage", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
reloadUsersFromDB(ctx, store)
|
||||
w.WriteHeader(http.StatusCreated)
|
||||
}
|
||||
}
|
||||
|
||||
func handleResetUserTraffic(store *Store) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
if store == nil {
|
||||
http.Error(w, "database not configured", http.StatusServiceUnavailable)
|
||||
return
|
||||
}
|
||||
|
||||
var req struct {
|
||||
Username string `json:"username"`
|
||||
ServerID string `json:"server_id,omitempty"`
|
||||
}
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
http.Error(w, "invalid json", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
req.Username = strings.TrimSpace(req.Username)
|
||||
if req.Username == "" {
|
||||
http.Error(w, "username required", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
ctx := r.Context()
|
||||
if ms, remote, err := managedServerFromID(ctx, store, req.ServerID); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
} else if remote {
|
||||
if sess := sessionFromCtx(ctx); sess != nil && sess.Role == RoleReseller && !remoteSSHUserOwned(ctx, ms, req.Username, sess.Username) {
|
||||
http.Error(w, "forbidden", http.StatusForbidden)
|
||||
return
|
||||
}
|
||||
req.ServerID = ""
|
||||
body, _ := json.Marshal(req)
|
||||
status, data, ct, err := proxyManagedServer(ctx, ms, http.MethodPost, "/api/users/reset-traffic", body, "application/json")
|
||||
if err != nil {
|
||||
http.Error(w, "remote server error: "+err.Error(), http.StatusBadGateway)
|
||||
return
|
||||
}
|
||||
writeProxyResponse(w, status, data, ct)
|
||||
return
|
||||
}
|
||||
|
||||
var owner string
|
||||
if err := store.db.QueryRowContext(ctx, `SELECT owner_username FROM ssh_users WHERE username=$1`, req.Username).Scan(&owner); err != nil {
|
||||
if err == sql.ErrNoRows {
|
||||
http.Error(w, "user not found", http.StatusNotFound)
|
||||
} else {
|
||||
http.Error(w, "db error", http.StatusInternalServerError)
|
||||
}
|
||||
return
|
||||
}
|
||||
if sess := sessionFromCtx(ctx); sess != nil && sess.Role == RoleReseller && strings.TrimSpace(owner) != sess.Username {
|
||||
http.Error(w, "forbidden", http.StatusForbidden)
|
||||
return
|
||||
}
|
||||
|
||||
if err := resetSSHUserTrafficAccounting(ctx, store, req.Username); err != nil {
|
||||
log.Printf("failed to reset SSH traffic for %s: %v", req.Username, err)
|
||||
http.Error(w, "could not reset traffic", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
_ = json.NewEncoder(w).Encode(map[string]interface{}{"ok": true, "username": req.Username})
|
||||
}
|
||||
}
|
||||
|
||||
func handleDeleteUser(store *Store) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodDelete {
|
||||
@@ -2143,19 +2358,52 @@ func matchTOTPPassword(u *UserState, supplied string, now time.Time) bool {
|
||||
// ---------- Auth callbacks ----------
|
||||
|
||||
func passwordCallback(meta ssh.ConnMetadata, pass []byte) (*ssh.Permissions, error) {
|
||||
supplied := string(pass)
|
||||
u, ok := userMgr.Get(meta.User())
|
||||
now := time.Now()
|
||||
|
||||
// Enforce panel policy (expiry / reseller owner) for known users up front,
|
||||
// so neither PAM nor the static password can bypass it.
|
||||
if ok {
|
||||
if u.ExpiresAt != nil && now.After(*u.ExpiresAt) {
|
||||
log.Printf("user %s tried to connect but account is expired", meta.User())
|
||||
return nil, fmt.Errorf("account expired")
|
||||
}
|
||||
if sshUserQuotaBlocked(u) {
|
||||
log.Printf("user %s tried to connect after reaching the data quota", meta.User())
|
||||
return nil, errDataQuotaExceeded
|
||||
}
|
||||
if err := ownerIsActive(u.Cfg.OwnerUsername); err != nil {
|
||||
return nil, fmt.Errorf("authentication failed: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
// System (PAM) login. When enabled server-wide, the Linux system password
|
||||
// (/etc/shadow) is accepted for any regular account (UID >= 1000) — whether
|
||||
// or not it is already a panel user. Unknown accounts are auto-imported on
|
||||
// success. Falls through to panel credentials if PAM does not accept.
|
||||
if isPAMAuthEnabled() {
|
||||
if isRegularLoginUser(meta.User()) {
|
||||
if err := authenticatePAM(meta.User(), supplied); err == nil {
|
||||
if !ok {
|
||||
importPAMUser(meta.User())
|
||||
}
|
||||
pamLogf("PAM: %q authenticated against /etc/shadow", meta.User())
|
||||
return nil, nil
|
||||
} else {
|
||||
pamLogf("PAM: %q rejected by /etc/shadow: %v", meta.User(), err)
|
||||
}
|
||||
} else if !ok {
|
||||
pamLogf("PAM: %q is not a regular login account (needs an /etc/passwd entry with UID >= %d)", meta.User(), minLoginUID)
|
||||
}
|
||||
}
|
||||
|
||||
if !ok {
|
||||
pamLogf("auth: user %q rejected (no panel account and PAM did not accept it)", meta.User())
|
||||
return nil, fmt.Errorf("authentication failed")
|
||||
}
|
||||
now := time.Now()
|
||||
if u.ExpiresAt != nil && now.After(*u.ExpiresAt) {
|
||||
log.Printf("user %s tried to connect but account is expired", meta.User())
|
||||
return nil, fmt.Errorf("account expired")
|
||||
}
|
||||
if err := ownerIsActive(u.Cfg.OwnerUsername); err != nil {
|
||||
return nil, fmt.Errorf("authentication failed: %w", err)
|
||||
}
|
||||
supplied := string(pass)
|
||||
|
||||
// Fall back to panel-managed credentials (TOTP and/or static password).
|
||||
if strings.TrimSpace(u.Cfg.TOTPSecret) != "" {
|
||||
if matchTOTPPassword(u, supplied, now) {
|
||||
return nil, nil
|
||||
@@ -2180,6 +2428,9 @@ func publicKeyCallback(meta ssh.ConnMetadata, key ssh.PublicKey) (*ssh.Permissio
|
||||
log.Printf("user %s tried to connect but account is expired", meta.User())
|
||||
return nil, fmt.Errorf("account expired")
|
||||
}
|
||||
if sshUserQuotaBlocked(u) {
|
||||
return nil, errDataQuotaExceeded
|
||||
}
|
||||
if err := ownerIsActive(u.Cfg.OwnerUsername); err != nil {
|
||||
return nil, fmt.Errorf("authentication failed: %w", err)
|
||||
}
|
||||
@@ -2311,6 +2562,10 @@ type directTCPIPReq struct {
|
||||
}
|
||||
|
||||
func handleDirectTCPIP(newChan ssh.NewChannel, u *UserState, upLimiter, downLimiter *rate.Limiter) {
|
||||
if sshUserQuotaBlocked(u) {
|
||||
newChan.Reject(ssh.Prohibited, "data quota exceeded")
|
||||
return
|
||||
}
|
||||
var req directTCPIPReq
|
||||
if err := ssh.Unmarshal(newChan.ExtraData(), &req); err != nil {
|
||||
newChan.Reject(ssh.Prohibited, "bad direct-tcpip request")
|
||||
@@ -2339,9 +2594,14 @@ func handleDirectTCPIP(newChan ssh.NewChannel, u *UserState, upLimiter, downLimi
|
||||
// half-close that never completes), both sides are force-closed so the
|
||||
// other direction unblocks. Close is idempotent, so calling it from both
|
||||
// directions is safe and no separate waiter goroutine is needed.
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
var closeOnce sync.Once
|
||||
closeAll := func() {
|
||||
_ = backend.Close()
|
||||
_ = ch.Close()
|
||||
closeOnce.Do(func() {
|
||||
cancel()
|
||||
_ = backend.Close()
|
||||
_ = ch.Close()
|
||||
})
|
||||
}
|
||||
|
||||
// Drain channel requests concurrently so the peer isn't left waiting.
|
||||
@@ -2355,7 +2615,7 @@ func handleDirectTCPIP(newChan ssh.NewChannel, u *UserState, upLimiter, downLimi
|
||||
|
||||
// upstream: SSH channel -> backend, in its own goroutine.
|
||||
go func() {
|
||||
_, _ = copyWithRateLimit(backend, ch, upLimiter)
|
||||
_, _ = copyWithRateLimitContext(ctx, sshQuotaWriter{w: backend, user: u, uplink: true, ctx: ctx}, ch, upLimiter)
|
||||
// Signal to the backend that we are done writing.
|
||||
if cw, ok := backend.(interface{ CloseWrite() error }); ok {
|
||||
_ = cw.CloseWrite()
|
||||
@@ -2366,7 +2626,7 @@ func handleDirectTCPIP(newChan ssh.NewChannel, u *UserState, upLimiter, downLimi
|
||||
// downstream: backend -> SSH channel, run in this goroutine.
|
||||
// handleDirectTCPIP already runs as its own goroutine (see handleConn),
|
||||
// so reusing it here avoids spawning a third goroutine per channel.
|
||||
_, _ = copyWithRateLimit(ch, backend, downLimiter)
|
||||
_, _ = copyWithRateLimitContext(ctx, sshQuotaWriter{w: ch, user: u, uplink: false, ctx: ctx}, backend, downLimiter)
|
||||
closeAll()
|
||||
}
|
||||
|
||||
@@ -3008,6 +3268,7 @@ func main() {
|
||||
// Optional: initialize interface totals persistence (best-effort).
|
||||
if store != nil {
|
||||
statsStore = store
|
||||
startSSHUserTrafficFlusher(store)
|
||||
ctx := context.Background()
|
||||
if err := store.EnsureXrayClientsSchema(ctx); err != nil {
|
||||
log.Printf("xray clients table: %v", err)
|
||||
@@ -3047,6 +3308,9 @@ func main() {
|
||||
primeCurrentStats()
|
||||
startStatsCollector()
|
||||
|
||||
// Turn the per-account byte counters into live up/down speeds for the panel.
|
||||
startSSHUserRateSampler()
|
||||
|
||||
adminAddr := os.Getenv("ADMIN_HTTP_ADDR")
|
||||
if adminAddr == "" {
|
||||
adminAddr = "0.0.0.0:9090"
|
||||
@@ -3205,6 +3469,7 @@ func main() {
|
||||
setDefaultLimits(cfg.DefaultLimitMbpsUp, cfg.DefaultLimitMbpsDown)
|
||||
setSSHIdleTimeoutFromConfig(cfg.SSHIdleTimeout)
|
||||
setMaxTotalConnsFromConfig(cfg.MaxTotalConnections)
|
||||
setPAMAuthEnabled(cfg.PAMAuthEnabled)
|
||||
|
||||
// Initialise listener pools (used for initial startup and hot-reload alike).
|
||||
publicPool = newListenerPool(serveHTTP80)
|
||||
|
||||
+227
@@ -0,0 +1,227 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/subtle"
|
||||
"errors"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"github.com/GehirnInc/crypt"
|
||||
_ "github.com/GehirnInc/crypt/apr1_crypt"
|
||||
_ "github.com/GehirnInc/crypt/md5_crypt"
|
||||
_ "github.com/GehirnInc/crypt/sha256_crypt"
|
||||
_ "github.com/GehirnInc/crypt/sha512_crypt"
|
||||
"github.com/openwall/yescrypt-go"
|
||||
"golang.org/x/crypto/bcrypt"
|
||||
)
|
||||
|
||||
const (
|
||||
shadowFile = "/etc/shadow"
|
||||
passwdFile = "/etc/passwd"
|
||||
|
||||
// minLoginUID / nobodyUID bound the accounts eligible for auto-import.
|
||||
// Regular human login accounts start at UID 1000 on Debian/Ubuntu; system
|
||||
// and service accounts (and "nobody") are excluded.
|
||||
minLoginUID = 1000
|
||||
nobodyUID = 65534
|
||||
)
|
||||
|
||||
var errNoSystemPassword = errors.New("account has no usable password")
|
||||
|
||||
// pamLogger writes PAM auth diagnostics straight to stderr (captured by
|
||||
// journald) so they remain visible even when "Quiet Logs" redirects the default
|
||||
// logger to io.Discard. Use pamLogf for anything an operator needs to see when
|
||||
// debugging why a system login was accepted or refused.
|
||||
var pamLogger = log.New(os.Stderr, "", log.LstdFlags)
|
||||
|
||||
func pamLogf(format string, args ...interface{}) { pamLogger.Printf(format, args...) }
|
||||
|
||||
// pamAuthEnabled mirrors Config.PAMAuthEnabled and is toggled live on config
|
||||
// reload. Guarded atomically so passwordCallback can read it lock-free.
|
||||
var pamAuthEnabled atomic.Bool
|
||||
|
||||
func setPAMAuthEnabled(v bool) {
|
||||
pamAuthEnabled.Store(v)
|
||||
state := "disabled"
|
||||
if v {
|
||||
state = "ENABLED"
|
||||
}
|
||||
pamLogf("PAM: system (Linux /etc/shadow) login is now %s", state)
|
||||
}
|
||||
func isPAMAuthEnabled() bool { return pamAuthEnabled.Load() }
|
||||
|
||||
// importPAMUser registers a freshly PAM-authenticated account in the running
|
||||
// user manager and persists it (marked use_pam) so it shows up in the panel and
|
||||
// later logins are re-verified against the system password. Idempotent: a
|
||||
// second concurrent/subsequent login for the same user is a no-op.
|
||||
func importPAMUser(username string) {
|
||||
cfg := UserConfig{Username: username, UsePAM: true}
|
||||
// Carry over the Linux account expiry (/etc/shadow field 8) so the panel's
|
||||
// "Vence em" shows the real expiration instead of "—".
|
||||
var expPtr *time.Time
|
||||
if exp := shadowAccountExpiry(username); exp != nil {
|
||||
cfg.ExpiresAt = exp.Format(time.RFC3339)
|
||||
expPtr = exp
|
||||
}
|
||||
st := &UserState{Cfg: cfg, ExpiresAt: expPtr}
|
||||
if !userMgr.AddIfAbsent(st) {
|
||||
return // already present in memory
|
||||
}
|
||||
pamLogf("PAM: auto-imported system user %s into the panel", username)
|
||||
if statsStore != nil {
|
||||
if err := statsStore.UpsertUser(context.Background(), cfg); err != nil {
|
||||
pamLogf("PAM: failed to persist auto-imported user %s: %v", username, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// isRegularLoginUser reports whether username is a regular human login account
|
||||
// (UID >= 1000 and not "nobody"), by parsing /etc/passwd. System/service
|
||||
// accounts and root are excluded from auto-import.
|
||||
func isRegularLoginUser(username string) bool {
|
||||
data, err := os.ReadFile(passwdFile)
|
||||
if err != nil {
|
||||
pamLogf("PAM: cannot read %s: %v", passwdFile, err)
|
||||
return false
|
||||
}
|
||||
for _, line := range strings.Split(string(data), "\n") {
|
||||
line = strings.TrimRight(line, "\r")
|
||||
fields := strings.Split(line, ":")
|
||||
if len(fields) < 3 || fields[0] != username {
|
||||
continue
|
||||
}
|
||||
uid, err := strconv.Atoi(fields[2])
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
return uid >= minLoginUID && uid != nobodyUID
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// shadowAccountExpiry returns the account expiration date from /etc/shadow
|
||||
// field 8 (days since 1970-01-01), or nil if the account never expires (empty
|
||||
// field) or the value is unusable. This is the `chage -E` / `useradd -e` date,
|
||||
// which maps to the panel's per-user expiry.
|
||||
func shadowAccountExpiry(username string) *time.Time {
|
||||
data, err := os.ReadFile(shadowFile)
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
for _, line := range strings.Split(string(data), "\n") {
|
||||
fields := strings.Split(strings.TrimRight(line, "\r"), ":")
|
||||
if len(fields) < 8 || fields[0] != username {
|
||||
continue
|
||||
}
|
||||
expStr := strings.TrimSpace(fields[7])
|
||||
if expStr == "" {
|
||||
return nil // no account expiry set
|
||||
}
|
||||
days, err := strconv.Atoi(expStr)
|
||||
if err != nil || days <= 0 {
|
||||
return nil
|
||||
}
|
||||
t := time.Unix(int64(days)*86400, 0).UTC()
|
||||
return &t
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// authenticatePAM verifies password against the Linux system account matching
|
||||
// username. It reads the account's hash from /etc/shadow (the panel runs as
|
||||
// root) and recomputes it with the same algorithm — this is the "just the auth"
|
||||
// behaviour: the supplied password is checked exactly as the system would,
|
||||
// with no account/session management and nothing to do with the SSH daemon.
|
||||
//
|
||||
// It is called "PAM" for continuity with the user-facing flag, but it does not
|
||||
// link libpam; it verifies the crypt(3) hash directly. Supported hash formats:
|
||||
// yescrypt ($y$), sha512-crypt ($6$), sha256-crypt ($5$), md5-crypt ($1$),
|
||||
// apr1 ($apr1$) and bcrypt ($2a$/$2b$/$2y$). Returns nil on success.
|
||||
func authenticatePAM(username, password string) error {
|
||||
if username == "" {
|
||||
return errors.New("shadow: empty username")
|
||||
}
|
||||
hash, err := lookupShadowHash(username)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return verifyCryptHash(hash, password)
|
||||
}
|
||||
|
||||
// lookupShadowHash returns the password hash field for username from /etc/shadow.
|
||||
func lookupShadowHash(username string) (string, error) {
|
||||
data, err := os.ReadFile(shadowFile)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("read %s: %w", shadowFile, err)
|
||||
}
|
||||
for _, line := range strings.Split(string(data), "\n") {
|
||||
line = strings.TrimRight(line, "\r")
|
||||
if line == "" {
|
||||
continue
|
||||
}
|
||||
fields := strings.Split(line, ":")
|
||||
if len(fields) < 2 || fields[0] != username {
|
||||
continue
|
||||
}
|
||||
hash := fields[1]
|
||||
// Empty, or locked/disabled accounts (! or * in the hash field) have no
|
||||
// password that any input can match — reject rather than risk a match.
|
||||
if hash == "" || strings.HasPrefix(hash, "!") || strings.HasPrefix(hash, "*") {
|
||||
return "", errNoSystemPassword
|
||||
}
|
||||
return hash, nil
|
||||
}
|
||||
return "", fmt.Errorf("shadow: user %q not found", username)
|
||||
}
|
||||
|
||||
// verifyCryptHash checks password against a crypt(3)-style hash string,
|
||||
// dispatching on the hash prefix. Returns nil only on an exact match.
|
||||
func verifyCryptHash(hash, password string) error {
|
||||
switch {
|
||||
case strings.HasPrefix(hash, "$y$"):
|
||||
computed, err := yescrypt.Hash([]byte(password), []byte(hash))
|
||||
if err != nil {
|
||||
return fmt.Errorf("yescrypt: %w", err)
|
||||
}
|
||||
if subtle.ConstantTimeCompare(computed, []byte(hash)) == 1 {
|
||||
return nil
|
||||
}
|
||||
return errors.New("password mismatch")
|
||||
case strings.HasPrefix(hash, "$2a$"), strings.HasPrefix(hash, "$2b$"), strings.HasPrefix(hash, "$2y$"):
|
||||
return bcrypt.CompareHashAndPassword([]byte(hash), []byte(password))
|
||||
case crypt.IsHashSupported(hash):
|
||||
return crypt.NewFromHash(hash).Verify(hash, []byte(password))
|
||||
case isTraditionalDES(hash):
|
||||
computed, err := desCrypt(password, hash)
|
||||
if err != nil {
|
||||
return fmt.Errorf("descrypt: %w", err)
|
||||
}
|
||||
if subtle.ConstantTimeCompare([]byte(computed), []byte(hash)) == 1 {
|
||||
return nil
|
||||
}
|
||||
return errors.New("password mismatch")
|
||||
default:
|
||||
return fmt.Errorf("shadow: unsupported hash format")
|
||||
}
|
||||
}
|
||||
|
||||
// isTraditionalDES reports whether hash looks like a classic 13-character
|
||||
// DES crypt(3) hash (2 salt chars + 11 hash chars, all from the crypt alphabet,
|
||||
// no "$" scheme prefix). Used by old Linux/UNIX accounts.
|
||||
func isTraditionalDES(hash string) bool {
|
||||
if len(hash) != 13 {
|
||||
return false
|
||||
}
|
||||
for i := 0; i < len(hash); i++ {
|
||||
if crypt64Decode(hash[i]) < 0 {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
@@ -0,0 +1,51 @@
|
||||
package main
|
||||
|
||||
import "testing"
|
||||
|
||||
// Known crypt(3) test vectors covering the formats found in /etc/shadow across
|
||||
// old and new Linux. verifyCryptHash must accept the right password and reject
|
||||
// the wrong one for each.
|
||||
func TestVerifyCryptHashVectors(t *testing.T) {
|
||||
cases := []struct {
|
||||
name string
|
||||
hash string
|
||||
pw string
|
||||
}{
|
||||
{
|
||||
name: "sha512crypt $6$ (glibc, older Linux)",
|
||||
// openssl passwd -6 -salt saltstring "Hello world!"
|
||||
hash: "$6$saltstring$svn8UoSVapNtMuq1ukKS4tPQd8iKwSMHWjl/O817G3uBnIFNjnQJuesI68u4OTLiBFdcbYEdFCoEOfaS35inz1",
|
||||
pw: "Hello world!",
|
||||
},
|
||||
{
|
||||
name: "yescrypt $y$ (Debian 11+/Ubuntu 22.04+)",
|
||||
hash: "$y$j9T$e8R9q85ZuzUkArEUurdtS.$esON.7y6H.u3UCPVCpbRFueRpAut2n2cMf1EhpjbuiC",
|
||||
pw: "pleaseletmein",
|
||||
},
|
||||
{
|
||||
// Real DES-crypt account from the production server's /etc/shadow.
|
||||
name: "traditional DES (old Linux) — testedragon",
|
||||
hash: "pae9A3UKpfaU6",
|
||||
pw: "testedragon",
|
||||
},
|
||||
{
|
||||
name: "traditional DES (old Linux) — ipv6dragon",
|
||||
hash: "pa9eao3LI6u.6",
|
||||
pw: "0tMGUL9chq8D",
|
||||
},
|
||||
}
|
||||
for _, c := range cases {
|
||||
t.Run(c.name, func(t *testing.T) {
|
||||
if err := verifyCryptHash(c.hash, c.pw); err != nil {
|
||||
t.Errorf("correct password REJECTED: %v", err)
|
||||
}
|
||||
// Build a wrong password that differs in the FIRST character, so the
|
||||
// check is meaningful even for traditional DES (which only considers
|
||||
// the first 8 bytes of the password).
|
||||
wrong := "Z" + c.pw
|
||||
if err := verifyCryptHash(c.hash, wrong); err == nil {
|
||||
t.Errorf("wrong password ACCEPTED")
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,367 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"strings"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"golang.org/x/time/rate"
|
||||
)
|
||||
|
||||
const (
|
||||
quotaActionBlock = "block"
|
||||
quotaActionThrottle = "throttle"
|
||||
)
|
||||
|
||||
var errDataQuotaExceeded = errors.New("data quota exceeded")
|
||||
|
||||
func normalizeQuotaAction(v string) string {
|
||||
if strings.EqualFold(strings.TrimSpace(v), quotaActionThrottle) {
|
||||
return quotaActionThrottle
|
||||
}
|
||||
return quotaActionBlock
|
||||
}
|
||||
|
||||
func quotaThrottleMbpsOrDefault(v int) int {
|
||||
if v <= 0 {
|
||||
return 1
|
||||
}
|
||||
return v
|
||||
}
|
||||
|
||||
type sshTrafficDelta struct {
|
||||
Uplink int64
|
||||
Downlink int64
|
||||
}
|
||||
|
||||
var (
|
||||
sshTrafficPersistenceMu sync.Mutex
|
||||
sshTrafficDirtyMu sync.Mutex
|
||||
sshTrafficDirty = make(map[string]*UserState)
|
||||
)
|
||||
|
||||
func markSSHUserTrafficDirty(u *UserState) {
|
||||
if u == nil || strings.TrimSpace(u.Cfg.Username) == "" {
|
||||
return
|
||||
}
|
||||
sshTrafficDirtyMu.Lock()
|
||||
sshTrafficDirty[u.Cfg.Username] = u
|
||||
sshTrafficDirtyMu.Unlock()
|
||||
}
|
||||
|
||||
func takeSSHUserTrafficDirty() map[string]*UserState {
|
||||
sshTrafficDirtyMu.Lock()
|
||||
dirty := sshTrafficDirty
|
||||
sshTrafficDirty = make(map[string]*UserState)
|
||||
sshTrafficDirtyMu.Unlock()
|
||||
return dirty
|
||||
}
|
||||
|
||||
func clearSSHUserTrafficDirty(username string, u *UserState) {
|
||||
sshTrafficDirtyMu.Lock()
|
||||
if current := sshTrafficDirty[username]; u == nil || current == u {
|
||||
delete(sshTrafficDirty, username)
|
||||
}
|
||||
sshTrafficDirtyMu.Unlock()
|
||||
}
|
||||
|
||||
func (s *Store) AddSSHUserTrafficBatch(ctx context.Context, deltas map[string]sshTrafficDelta) error {
|
||||
if s == nil || len(deltas) == 0 {
|
||||
return nil
|
||||
}
|
||||
tx, err := s.db.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
stmt, err := tx.PrepareContext(ctx, `
|
||||
UPDATE ssh_users SET
|
||||
total_uplink_bytes = GREATEST(total_uplink_bytes + GREATEST($2::BIGINT, 0), 0),
|
||||
total_downlink_bytes = GREATEST(total_downlink_bytes + GREATEST($3::BIGINT, 0), 0)
|
||||
WHERE username = $1`)
|
||||
if err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
defer stmt.Close()
|
||||
for username, d := range deltas {
|
||||
if strings.TrimSpace(username) == "" || (d.Uplink == 0 && d.Downlink == 0) {
|
||||
continue
|
||||
}
|
||||
if _, err := stmt.ExecContext(ctx, username, d.Uplink, d.Downlink); err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
}
|
||||
return tx.Commit()
|
||||
}
|
||||
|
||||
func (s *Store) ResetSSHUserTraffic(ctx context.Context, username string) error {
|
||||
if s == nil || strings.TrimSpace(username) == "" {
|
||||
return nil
|
||||
}
|
||||
_, err := s.db.ExecContext(ctx, `
|
||||
UPDATE ssh_users
|
||||
SET total_uplink_bytes = 0, total_downlink_bytes = 0
|
||||
WHERE username = $1`, username)
|
||||
return err
|
||||
}
|
||||
|
||||
func initSSHRuntimeUsage(u *UserState, uplink, downlink int64) {
|
||||
if u == nil {
|
||||
return
|
||||
}
|
||||
if uplink < 0 {
|
||||
uplink = 0
|
||||
}
|
||||
if downlink < 0 {
|
||||
downlink = 0
|
||||
}
|
||||
atomic.StoreInt64(&u.TotalUplinkBytes, uplink)
|
||||
atomic.StoreInt64(&u.TotalDownlinkBytes, downlink)
|
||||
atomic.StoreInt64(&u.totalBytes, uplink+downlink)
|
||||
atomic.StoreInt64(&u.pendingUplinkBytes, 0)
|
||||
atomic.StoreInt64(&u.pendingDownlinkBytes, 0)
|
||||
}
|
||||
|
||||
func resetSSHRuntimeUsageLocked(u *UserState) {
|
||||
if u == nil {
|
||||
return
|
||||
}
|
||||
initSSHRuntimeUsage(u, 0, 0)
|
||||
u.mu.Lock()
|
||||
u.quotaLimiter = nil
|
||||
u.quotaLimiterMbps = 0
|
||||
u.mu.Unlock()
|
||||
}
|
||||
|
||||
func resetSSHRuntimeUsage(username string) {
|
||||
u, ok := userMgr.Get(username)
|
||||
if !ok || u == nil {
|
||||
return
|
||||
}
|
||||
u.trafficMu.Lock()
|
||||
resetSSHRuntimeUsageLocked(u)
|
||||
clearSSHUserTrafficDirty(username, u)
|
||||
u.trafficMu.Unlock()
|
||||
}
|
||||
|
||||
func resetSSHUserTrafficAccounting(ctx context.Context, store *Store, username string) error {
|
||||
u, _ := userMgr.Get(username)
|
||||
if u != nil {
|
||||
u.trafficMu.Lock()
|
||||
defer u.trafficMu.Unlock()
|
||||
}
|
||||
sshTrafficPersistenceMu.Lock()
|
||||
defer sshTrafficPersistenceMu.Unlock()
|
||||
if err := store.ResetSSHUserTraffic(ctx, username); err != nil {
|
||||
return err
|
||||
}
|
||||
if u != nil {
|
||||
resetSSHRuntimeUsageLocked(u)
|
||||
clearSSHUserTrafficDirty(username, u)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func sshUserQuotaBlocked(u *UserState) bool {
|
||||
if u == nil {
|
||||
return false
|
||||
}
|
||||
u.mu.Lock()
|
||||
quota := u.Cfg.DataQuotaBytes
|
||||
action := normalizeQuotaAction(u.Cfg.QuotaAction)
|
||||
u.mu.Unlock()
|
||||
return quota > 0 && action == quotaActionBlock && atomic.LoadInt64(&u.totalBytes) >= quota
|
||||
}
|
||||
|
||||
func sshQuotaLimiter(u *UserState, mbps int) *rate.Limiter {
|
||||
mbps = quotaThrottleMbpsOrDefault(mbps)
|
||||
u.mu.Lock()
|
||||
defer u.mu.Unlock()
|
||||
if u.quotaLimiter == nil || u.quotaLimiterMbps != mbps {
|
||||
bps := mbpsToBytesPerSec(mbps)
|
||||
burst := int(bps)
|
||||
if burst < copyBufSize {
|
||||
burst = copyBufSize
|
||||
}
|
||||
u.quotaLimiter = rate.NewLimiter(rate.Limit(bps), burst)
|
||||
u.quotaLimiterMbps = mbps
|
||||
}
|
||||
return u.quotaLimiter
|
||||
}
|
||||
|
||||
func reserveSSHUserBytes(u *UserState, requested int) (allowed int, throttle *rate.Limiter, stopAfter bool) {
|
||||
if u == nil || requested <= 0 {
|
||||
return 0, nil, false
|
||||
}
|
||||
u.mu.Lock()
|
||||
quota := u.Cfg.DataQuotaBytes
|
||||
action := normalizeQuotaAction(u.Cfg.QuotaAction)
|
||||
throttleMbps := u.Cfg.QuotaThrottleMbps
|
||||
u.mu.Unlock()
|
||||
|
||||
n := int64(requested)
|
||||
if quota <= 0 {
|
||||
atomic.AddInt64(&u.totalBytes, n)
|
||||
return requested, nil, false
|
||||
}
|
||||
|
||||
if action == quotaActionThrottle {
|
||||
previous := atomic.AddInt64(&u.totalBytes, n) - n
|
||||
if previous+n > quota {
|
||||
return requested, sshQuotaLimiter(u, throttleMbps), false
|
||||
}
|
||||
return requested, nil, false
|
||||
}
|
||||
|
||||
for {
|
||||
used := atomic.LoadInt64(&u.totalBytes)
|
||||
remaining := quota - used
|
||||
if remaining <= 0 {
|
||||
return 0, nil, true
|
||||
}
|
||||
take := n
|
||||
if take > remaining {
|
||||
take = remaining
|
||||
}
|
||||
if atomic.CompareAndSwapInt64(&u.totalBytes, used, used+take) {
|
||||
return int(take), nil, take < n || used+take >= quota
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func finishSSHUserReservation(u *UserState, uplink bool, reserved, written int) {
|
||||
if u == nil || reserved <= 0 {
|
||||
return
|
||||
}
|
||||
if written < 0 {
|
||||
written = 0
|
||||
}
|
||||
if written > reserved {
|
||||
written = reserved
|
||||
}
|
||||
if written < reserved {
|
||||
atomic.AddInt64(&u.totalBytes, -int64(reserved-written))
|
||||
}
|
||||
if written == 0 {
|
||||
return
|
||||
}
|
||||
if uplink {
|
||||
atomic.AddInt64(&u.TotalUplinkBytes, int64(written))
|
||||
atomic.AddInt64(&u.pendingUplinkBytes, int64(written))
|
||||
} else {
|
||||
atomic.AddInt64(&u.TotalDownlinkBytes, int64(written))
|
||||
atomic.AddInt64(&u.pendingDownlinkBytes, int64(written))
|
||||
}
|
||||
markSSHUserTrafficDirty(u)
|
||||
}
|
||||
|
||||
type sshQuotaWriter struct {
|
||||
w io.Writer
|
||||
user *UserState
|
||||
uplink bool
|
||||
ctx context.Context
|
||||
}
|
||||
|
||||
func (qw sshQuotaWriter) Write(p []byte) (int, error) {
|
||||
if qw.user != nil {
|
||||
qw.user.trafficMu.RLock()
|
||||
defer qw.user.trafficMu.RUnlock()
|
||||
}
|
||||
allowed, quotaLimiter, stopAfter := reserveSSHUserBytes(qw.user, len(p))
|
||||
if allowed <= 0 {
|
||||
return 0, errDataQuotaExceeded
|
||||
}
|
||||
if quotaLimiter != nil {
|
||||
ctx := qw.ctx
|
||||
if ctx == nil {
|
||||
ctx = context.Background()
|
||||
}
|
||||
if err := quotaLimiter.WaitN(ctx, allowed); err != nil {
|
||||
finishSSHUserReservation(qw.user, qw.uplink, allowed, 0)
|
||||
return 0, err
|
||||
}
|
||||
}
|
||||
n, err := qw.w.Write(p[:allowed])
|
||||
finishSSHUserReservation(qw.user, qw.uplink, allowed, n)
|
||||
if err != nil {
|
||||
return n, err
|
||||
}
|
||||
if stopAfter || allowed < len(p) {
|
||||
return n, errDataQuotaExceeded
|
||||
}
|
||||
return n, nil
|
||||
}
|
||||
|
||||
func startSSHUserTrafficFlusher(store *Store) {
|
||||
if store == nil {
|
||||
return
|
||||
}
|
||||
go func() {
|
||||
ticker := time.NewTicker(5 * time.Second)
|
||||
defer ticker.Stop()
|
||||
for range ticker.C {
|
||||
flushSSHUserTraffic(store)
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
func flushSSHUserTraffic(store *Store) {
|
||||
if store == nil {
|
||||
return
|
||||
}
|
||||
sshTrafficPersistenceMu.Lock()
|
||||
defer sshTrafficPersistenceMu.Unlock()
|
||||
deltas := make(map[string]sshTrafficDelta)
|
||||
states := make(map[string]*UserState)
|
||||
for username, u := range takeSSHUserTrafficDirty() {
|
||||
if u == nil || strings.TrimSpace(username) == "" {
|
||||
continue
|
||||
}
|
||||
up := atomic.SwapInt64(&u.pendingUplinkBytes, 0)
|
||||
down := atomic.SwapInt64(&u.pendingDownlinkBytes, 0)
|
||||
if up == 0 && down == 0 {
|
||||
continue
|
||||
}
|
||||
deltas[username] = sshTrafficDelta{Uplink: up, Downlink: down}
|
||||
states[username] = u
|
||||
}
|
||||
if len(deltas) == 0 {
|
||||
return
|
||||
}
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
if err := store.AddSSHUserTrafficBatch(ctx, deltas); err != nil {
|
||||
log.Printf("ssh traffic flush failed: %v", err)
|
||||
for username, d := range deltas {
|
||||
if u := states[username]; u != nil {
|
||||
atomic.AddInt64(&u.pendingUplinkBytes, d.Uplink)
|
||||
atomic.AddInt64(&u.pendingDownlinkBytes, d.Downlink)
|
||||
markSSHUserTrafficDirty(u)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func validateQuotaConfig(quotaBytes int64, action string, throttleMbps int) error {
|
||||
if quotaBytes < 0 {
|
||||
return fmt.Errorf("data_quota_bytes must be non-negative")
|
||||
}
|
||||
rawAction := strings.ToLower(strings.TrimSpace(action))
|
||||
if rawAction != "" && rawAction != quotaActionBlock && rawAction != quotaActionThrottle {
|
||||
return fmt.Errorf("quota_action must be block or throttle")
|
||||
}
|
||||
if throttleMbps < 0 {
|
||||
return fmt.Errorf("quota_throttle_mbps must be non-negative")
|
||||
}
|
||||
if throttleMbps > 1000000 {
|
||||
return fmt.Errorf("quota_throttle_mbps must not exceed 1000000")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
@@ -0,0 +1,870 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"io"
|
||||
"net"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"os"
|
||||
"strings"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"golang.org/x/time/rate"
|
||||
)
|
||||
|
||||
func TestNativeClientMaxConnectionsAndBatchedActiveDelta(t *testing.T) {
|
||||
oldStore := statsStore
|
||||
statsStore = &Store{}
|
||||
defer func() { statsStore = oldStore }()
|
||||
|
||||
const uuid = "11111111-1111-1111-1111-111111111111"
|
||||
m := &XrayManager{
|
||||
nativeQuotaByUUID: map[string]*xrayNativeQuotaState{
|
||||
uuid: {maxConns: 1, generation: 1},
|
||||
},
|
||||
}
|
||||
state := m.nativeQuotaState(uuid)
|
||||
|
||||
release, acquiredState, ok := m.acquireNativeClientConnection(uuid, "user@example")
|
||||
if !ok || release == nil {
|
||||
t.Fatal("first native connection was rejected")
|
||||
}
|
||||
if acquiredState != state {
|
||||
t.Fatal("connection lease did not retain the authenticated policy state")
|
||||
}
|
||||
if _, _, ok := m.acquireNativeClientConnection(uuid, "user@example"); ok {
|
||||
t.Fatal("connection above max_conns was accepted")
|
||||
}
|
||||
|
||||
m.nativeDBMu.Lock()
|
||||
pending := m.nativeActivePending[uuid]
|
||||
m.nativeDBMu.Unlock()
|
||||
if pending.Delta != 1 || !pending.Connected || pending.State != state {
|
||||
t.Fatalf("connect was not queued for batch persistence: %+v", pending)
|
||||
}
|
||||
|
||||
release()
|
||||
release() // idempotent release must not underflow counters.
|
||||
|
||||
m.nativeDBMu.Lock()
|
||||
pending = m.nativeActivePending[uuid]
|
||||
m.nativeDBMu.Unlock()
|
||||
if pending.Delta != 0 || !pending.Connected {
|
||||
t.Fatalf("connect/disconnect batch should net to zero and retain last-active: %+v", pending)
|
||||
}
|
||||
|
||||
state.mu.Lock()
|
||||
active := state.activeConns
|
||||
state.mu.Unlock()
|
||||
if active != 0 {
|
||||
t.Fatalf("active connection count = %d, want 0", active)
|
||||
}
|
||||
|
||||
release2, _, ok := m.acquireNativeClientConnection(uuid, "user@example")
|
||||
if !ok {
|
||||
t.Fatal("slot was not reusable after release")
|
||||
}
|
||||
release2()
|
||||
}
|
||||
|
||||
func TestNativeOnlineUsersAreKeyedByUUID(t *testing.T) {
|
||||
oldStore := statsStore
|
||||
statsStore = nil
|
||||
defer func() { statsStore = oldStore }()
|
||||
|
||||
m := &XrayManager{}
|
||||
m.recordNativeConnect("aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa", "shared@example", nil)
|
||||
m.recordNativeConnect("bbbbbbbb-bbbb-bbbb-bbbb-bbbbbbbbbbbb", "shared@example", nil)
|
||||
if got := m.CountOnlineUsers(); got != 2 {
|
||||
t.Fatalf("online UUID count = %d, want 2 for two UUIDs sharing one email", got)
|
||||
}
|
||||
m.statsMu.RLock()
|
||||
_, first := m.statsByEmail["aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"]
|
||||
_, second := m.statsByEmail["bbbbbbbb-bbbb-bbbb-bbbb-bbbbbbbbbbbb"]
|
||||
m.statsMu.RUnlock()
|
||||
if !first || !second {
|
||||
t.Fatal("native runtime stats were not stored under canonical UUID keys")
|
||||
}
|
||||
}
|
||||
|
||||
func TestNativeExpiryRejectsAndDisconnectsClient(t *testing.T) {
|
||||
const uuid = "cccccccc-cccc-cccc-cccc-cccccccccccc"
|
||||
state := &xrayNativeQuotaState{hasExpiry: true, expiresAt: time.Now().Add(time.Hour), generation: 1}
|
||||
m := &XrayManager{nativeQuotaByUUID: map[string]*xrayNativeQuotaState{uuid: state}}
|
||||
closer := &closeTrackingReader{}
|
||||
release, _, ok := m.acquireNativeClientConnection(uuid, "expiry@example", closer)
|
||||
if !ok {
|
||||
t.Fatal("unexpired client was rejected")
|
||||
}
|
||||
m.disconnectNativeClient(uuid)
|
||||
if !closer.closed.Load() {
|
||||
t.Fatal("active native client was not closed during revocation")
|
||||
}
|
||||
release()
|
||||
|
||||
state.mu.Lock()
|
||||
state.expiresAt = time.Now().Add(-time.Second)
|
||||
state.mu.Unlock()
|
||||
if reason := m.nativeClientAccessDenied(uuid); reason != "expired" {
|
||||
t.Fatalf("expired client denial = %q, want expired", reason)
|
||||
}
|
||||
if _, _, ok := m.acquireNativeClientConnection(uuid, "expiry@example"); ok {
|
||||
t.Fatal("expired client acquired a new connection")
|
||||
}
|
||||
}
|
||||
|
||||
func TestQuotaAndExpiryValidationRejectsUnsafeInput(t *testing.T) {
|
||||
if err := validateQuotaConfig(1, "typo", 1); err == nil {
|
||||
t.Fatal("unknown quota action was accepted")
|
||||
}
|
||||
if err := validateQuotaConfig(1, quotaActionBlock, -1); err == nil {
|
||||
t.Fatal("negative throttle setting was accepted")
|
||||
}
|
||||
if _, err := parseOptionalXrayExpiry("not-a-date"); err == nil {
|
||||
t.Fatal("invalid Xray expiry was accepted")
|
||||
}
|
||||
if exp, err := parseOptionalXrayExpiry(""); err != nil || exp != nil {
|
||||
t.Fatalf("empty Xray expiry = (%v, %v), want nil, nil", exp, err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRemoveNativeQuotaPolicyPrunesPendingMaps(t *testing.T) {
|
||||
m := &XrayManager{
|
||||
nativeQuotaByUUID: map[string]*xrayNativeQuotaState{
|
||||
"gone": {generation: 1},
|
||||
},
|
||||
nativeTrafficPending: map[string]xrayPendingTraffic{
|
||||
"gone": {Uplink: 10},
|
||||
},
|
||||
nativeActivePending: map[string]xrayPendingActive{
|
||||
"gone": {Delta: 1},
|
||||
},
|
||||
}
|
||||
m.removeNativeQuotaPolicy("gone")
|
||||
if m.nativeQuotaState("gone") != nil {
|
||||
t.Fatal("quota policy was not removed")
|
||||
}
|
||||
m.nativeDBMu.Lock()
|
||||
_, trafficExists := m.nativeTrafficPending["gone"]
|
||||
_, activeExists := m.nativeActivePending["gone"]
|
||||
m.nativeDBMu.Unlock()
|
||||
if trafficExists || activeExists {
|
||||
t.Fatal("deleted UUID remained in a pending persistence map")
|
||||
}
|
||||
}
|
||||
|
||||
func TestNativeCounterIsBoundedAndReleaseIsIdempotent(t *testing.T) {
|
||||
var active atomicInt64ForTest
|
||||
release1, ok := acquireNativeCounter(&active.Int64, 2)
|
||||
if !ok {
|
||||
t.Fatal("first slot rejected")
|
||||
}
|
||||
release2, ok := acquireNativeCounter(&active.Int64, 2)
|
||||
if !ok {
|
||||
t.Fatal("second slot rejected")
|
||||
}
|
||||
if _, ok := acquireNativeCounter(&active.Int64, 2); ok {
|
||||
t.Fatal("slot above limit accepted")
|
||||
}
|
||||
release1()
|
||||
release1()
|
||||
if got := active.Load(); got != 1 {
|
||||
t.Fatalf("active after double release = %d, want 1", got)
|
||||
}
|
||||
release2()
|
||||
if got := active.Load(); got != 0 {
|
||||
t.Fatalf("active after releases = %d, want 0", got)
|
||||
}
|
||||
}
|
||||
|
||||
// Embedding keeps the test declaration readable while still passing the exact
|
||||
// atomic.Int64 type required by acquireNativeCounter.
|
||||
type atomicInt64ForTest struct{ Int64 atomic.Int64 }
|
||||
|
||||
func (a *atomicInt64ForTest) Load() int64 { return a.Int64.Load() }
|
||||
|
||||
func TestTrackedNativeConnectionsAreClosedOnShutdown(t *testing.T) {
|
||||
oldAccepting := nativeTransportAccepting.Load()
|
||||
defer nativeTransportAccepting.Store(oldAccepting)
|
||||
|
||||
beginNativeTransportAccepting()
|
||||
before := nativeTransportConnections.Load()
|
||||
serverSide, clientSide := net.Pipe()
|
||||
defer clientSide.Close()
|
||||
wrapped, ok := wrapTrackedNativeTransportConn(serverSide)
|
||||
if !ok {
|
||||
t.Fatal("tracked connection was rejected")
|
||||
}
|
||||
if got := nativeTransportConnections.Load(); got != before+1 {
|
||||
t.Fatalf("transport count = %d, want %d", got, before+1)
|
||||
}
|
||||
|
||||
stopNativeTransportAccepting()
|
||||
closeAllNativeTransportConnections()
|
||||
_ = clientSide.SetReadDeadline(time.Now().Add(time.Second))
|
||||
if _, err := clientSide.Read(make([]byte, 1)); err == nil {
|
||||
t.Fatal("peer remained open after native shutdown")
|
||||
}
|
||||
if err := wrapped.Close(); err != nil && !errors.Is(err, net.ErrClosed) {
|
||||
t.Fatalf("second close returned unexpected error: %v", err)
|
||||
}
|
||||
if got := nativeTransportConnections.Load(); got != before {
|
||||
t.Fatalf("transport count after shutdown = %d, want %d", got, before)
|
||||
}
|
||||
}
|
||||
|
||||
func TestXHTTPSessionsIgnoreLegacyGlobalCapAndReleaseCounters(t *testing.T) {
|
||||
before := nativeXHTTPSessions.Load()
|
||||
ib := &nativeInbound{xhttpMaxBufferedPosts: 2}
|
||||
for _, id := range []string{"one", "two"} {
|
||||
if sess := ib.upsertXHTTPSession(httptest.NewRecorder(), id); sess == nil {
|
||||
t.Fatalf("session %q was rejected", id)
|
||||
}
|
||||
}
|
||||
if got := nativeXHTTPSessions.Load(); got != before+2 {
|
||||
t.Fatalf("global XHTTP sessions = %d, want %d", got, before+2)
|
||||
}
|
||||
ib.closeAllXHTTPSessions()
|
||||
if got := nativeXHTTPSessions.Load(); got != before {
|
||||
t.Fatalf("global XHTTP sessions after close = %d, want %d", got, before)
|
||||
}
|
||||
ib.xhttpMu.Lock()
|
||||
remaining := len(ib.xhttpSessions)
|
||||
ib.xhttpMu.Unlock()
|
||||
if remaining != 0 {
|
||||
t.Fatalf("inbound retained %d XHTTP sessions", remaining)
|
||||
}
|
||||
}
|
||||
|
||||
func TestXHTTPSessionSafetyWindowIsAboveProductionScale(t *testing.T) {
|
||||
if got := (&nativeInbound{}).xhttpMaxActiveSessions(); got != fixedNativeMaxXHTTPSessions {
|
||||
t.Fatalf("XHTTP session safety window = %d, want %d", got, fixedNativeMaxXHTTPSessions)
|
||||
}
|
||||
if got := nativeXHTTPSessionLimit(); got < 8_000 {
|
||||
t.Fatalf("XHTTP session safety window = %d, want room for at least 8K users", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNativeHTTP2StreamsUseFiniteTransportBackpressure(t *testing.T) {
|
||||
if got := nativeHTTP2MaxConcurrentStreams(); got != fixedNativeHTTP2ConcurrentStreams {
|
||||
t.Fatalf("HTTP/2 stream setting = %d, want %d", got, fixedNativeHTTP2ConcurrentStreams)
|
||||
}
|
||||
if got := nativeHTTP2MaxConcurrentStreams(); got < 1024 {
|
||||
t.Fatalf("HTTP/2 stream setting = %d, too small for XHTTP packet bursts", got)
|
||||
}
|
||||
if got := nativeMuxMaxSessionLimit(); got != 64 {
|
||||
t.Fatalf("per-transport Mux session guard = %d, want 64", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestXHTTPHandlerDoesNotApplyWebRequestCeiling(t *testing.T) {
|
||||
ib := &nativeInbound{transport: "xhttp", path: "/"}
|
||||
req := httptest.NewRequest(http.MethodOptions, "/", nil)
|
||||
rec := httptest.NewRecorder()
|
||||
ib.ServeHTTP(rec, req)
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("XHTTP OPTIONS at legacy request ceiling = %d, want 200", rec.Code)
|
||||
}
|
||||
}
|
||||
|
||||
func TestXHTTPHandlerSafetySlotIsReleased(t *testing.T) {
|
||||
before := nativeXHTTPRequests.Load()
|
||||
ib := &nativeInbound{transport: "xhttp", path: "/"}
|
||||
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||
rec := httptest.NewRecorder()
|
||||
ib.ServeHTTP(rec, req)
|
||||
if rec.Code != http.StatusBadRequest {
|
||||
t.Fatalf("empty XHTTP request status = %d, want 400", rec.Code)
|
||||
}
|
||||
if got := nativeXHTTPRequests.Load(); got != before {
|
||||
t.Fatalf("XHTTP handler counter after return = %d, want %d", got, before)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPersistedXHTTPAdmissionTuningIsAlwaysUnlimited(t *testing.T) {
|
||||
got := normalizeNativeXrayTuning(&XrayNativeTuning{
|
||||
MuxGlobalSessions: 8192,
|
||||
MaxConcurrentConnections: 4096,
|
||||
MaxConcurrentXHTTPRequests: 8192,
|
||||
XHTTPMaxSessions: 4096,
|
||||
})
|
||||
if got.MuxGlobalSessions != 8192 ||
|
||||
got.MaxConcurrentConnections != -1 ||
|
||||
got.MaxConcurrentXHTTPRequests != defaultNativeMaxXHTTPRequests ||
|
||||
got.XHTTPMaxSessions != -1 {
|
||||
t.Fatalf("persisted admission limits were not removed: %+v", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestXHTTPMetadataLengthIsBoundedBeforeSessionAllocation(t *testing.T) {
|
||||
ib := &nativeInbound{transport: "xhttp", path: "/"}
|
||||
req := httptest.NewRequest("GET", "/"+strings.Repeat("a", nativeXHTTPMaxSessionIDBytes+1), nil)
|
||||
rec := httptest.NewRecorder()
|
||||
ib.ServeHTTP(rec, req)
|
||||
if rec.Code != 400 {
|
||||
t.Fatalf("oversized XHTTP session id status = %d, want 400", rec.Code)
|
||||
}
|
||||
ib.xhttpMu.Lock()
|
||||
sessions := len(ib.xhttpSessions)
|
||||
ib.xhttpMu.Unlock()
|
||||
if sessions != 0 {
|
||||
t.Fatalf("oversized metadata allocated %d sessions", sessions)
|
||||
}
|
||||
}
|
||||
|
||||
func TestXHTTPUploadMemoryIsReleasedOnReadAndClose(t *testing.T) {
|
||||
before := nativeXHTTPBufferedBytes.Load()
|
||||
q := newNativeXHTTPUploadQueue(4, 512)
|
||||
|
||||
accounted := nativeXHTTPAccountedPacketBytes(4)
|
||||
lease, ok := acquireNativeXHTTPMemory(accounted)
|
||||
if !ok {
|
||||
t.Fatal("failed to reserve XHTTP test memory")
|
||||
}
|
||||
if err := q.push(context.Background(), nativeXHTTPPacket{Payload: []byte("test"), Seq: 0}, lease); err != nil {
|
||||
lease.release()
|
||||
t.Fatalf("queue push failed: %v", err)
|
||||
}
|
||||
lease.release() // transferred leases are a no-op for the producer.
|
||||
if got := nativeXHTTPBufferedBytes.Load(); got != before+accounted {
|
||||
t.Fatalf("buffered bytes after push = %d, want %d", got, before+accounted)
|
||||
}
|
||||
|
||||
buf := make([]byte, 4)
|
||||
if n, err := q.Read(buf); err != nil || n != 4 || string(buf) != "test" {
|
||||
t.Fatalf("queue read = (%d, %v, %q), want (4, nil, test)", n, err, string(buf))
|
||||
}
|
||||
if got := nativeXHTTPBufferedBytes.Load(); got != before {
|
||||
t.Fatalf("buffered bytes after read = %d, want %d", got, before)
|
||||
}
|
||||
|
||||
lease, ok = acquireNativeXHTTPMemory(nativeXHTTPAccountedPacketBytes(3))
|
||||
if !ok {
|
||||
t.Fatal("failed to reserve second XHTTP test memory")
|
||||
}
|
||||
if err := q.push(context.Background(), nativeXHTTPPacket{Payload: []byte("xyz"), Seq: 2}, lease); err != nil {
|
||||
lease.release()
|
||||
t.Fatalf("second queue push failed: %v", err)
|
||||
}
|
||||
lease.release()
|
||||
q.close()
|
||||
if got := nativeXHTTPBufferedBytes.Load(); got != before {
|
||||
t.Fatalf("buffered bytes after close = %d, want %d", got, before)
|
||||
}
|
||||
}
|
||||
|
||||
func TestXHTTPUploadQueueEnforcesPerSessionByteBudget(t *testing.T) {
|
||||
before := nativeXHTTPBufferedBytes.Load()
|
||||
q := newNativeXHTTPUploadQueue(4, nativeXHTTPMinPacketAccountingBytes-1)
|
||||
defer q.close()
|
||||
|
||||
lease, ok := acquireNativeXHTTPMemory(nativeXHTTPAccountedPacketBytes(5))
|
||||
if !ok {
|
||||
t.Fatal("failed to reserve XHTTP test memory")
|
||||
}
|
||||
defer lease.release()
|
||||
err := q.push(context.Background(), nativeXHTTPPacket{Payload: make([]byte, 5)}, lease)
|
||||
if !errors.Is(err, errNativeXHTTPUploadBufferFull) {
|
||||
t.Fatalf("oversized queue push error = %v, want buffer limit", err)
|
||||
}
|
||||
lease.release()
|
||||
if got := nativeXHTTPBufferedBytes.Load(); got != before {
|
||||
t.Fatalf("rejected payload retained %d bytes, baseline %d", got, before)
|
||||
}
|
||||
}
|
||||
|
||||
func TestXHTTPUploadQueueBackpressuresInsteadOfRejectingBurst(t *testing.T) {
|
||||
before := nativeXHTTPBufferedBytes.Load()
|
||||
q := newNativeXHTTPUploadQueue(2, nativeXHTTPMinPacketAccountingBytes)
|
||||
defer q.close()
|
||||
|
||||
first, ok := acquireNativeXHTTPMemory(nativeXHTTPAccountedPacketBytes(4))
|
||||
if !ok {
|
||||
t.Fatal("failed to reserve first XHTTP payload")
|
||||
}
|
||||
if err := q.push(context.Background(), nativeXHTTPPacket{Payload: []byte("one!"), Seq: 0}, first); err != nil {
|
||||
first.release()
|
||||
t.Fatalf("first queue push failed: %v", err)
|
||||
}
|
||||
first.release()
|
||||
|
||||
second, ok := acquireNativeXHTTPMemory(nativeXHTTPAccountedPacketBytes(4))
|
||||
if !ok {
|
||||
t.Fatal("failed to reserve second XHTTP payload")
|
||||
}
|
||||
done := make(chan error, 1)
|
||||
go func() {
|
||||
done <- q.push(context.Background(), nativeXHTTPPacket{Payload: []byte("two!"), Seq: 1}, second)
|
||||
}()
|
||||
|
||||
select {
|
||||
case err := <-done:
|
||||
second.release()
|
||||
t.Fatalf("second burst packet did not backpressure: %v", err)
|
||||
case <-time.After(25 * time.Millisecond):
|
||||
}
|
||||
|
||||
buf := make([]byte, 4)
|
||||
if n, err := q.Read(buf); err != nil || n != 4 || string(buf) != "one!" {
|
||||
second.release()
|
||||
t.Fatalf("first queue read = (%d, %v, %q)", n, err, string(buf))
|
||||
}
|
||||
select {
|
||||
case err := <-done:
|
||||
if err != nil {
|
||||
second.release()
|
||||
t.Fatalf("backpressured packet failed after space released: %v", err)
|
||||
}
|
||||
second.release()
|
||||
case <-time.After(time.Second):
|
||||
second.release()
|
||||
t.Fatal("backpressured packet did not resume")
|
||||
}
|
||||
|
||||
q.close()
|
||||
if got := nativeXHTTPBufferedBytes.Load(); got != before {
|
||||
t.Fatalf("backpressure test leaked %d buffered bytes (baseline %d)", got, before)
|
||||
}
|
||||
}
|
||||
|
||||
func TestXHTTPGlobalMemoryBackpressureWakesWaitersFIFO(t *testing.T) {
|
||||
before := nativeXHTTPBufferedBytes.Load()
|
||||
fillBytes := nativeXHTTPMaxBufferedGlobalBytes - before
|
||||
filler, ok := acquireNativeXHTTPMemory(fillBytes)
|
||||
if !ok {
|
||||
t.Fatal("failed to fill XHTTP memory budget for waiter test")
|
||||
}
|
||||
defer filler.release()
|
||||
|
||||
type result struct {
|
||||
lease *nativeXHTTPMemoryLease
|
||||
err error
|
||||
}
|
||||
startWaiter := func() <-chan result {
|
||||
done := make(chan result, 1)
|
||||
go func() {
|
||||
lease, err := acquireNativeXHTTPMemoryContext(context.Background(), nativeXHTTPMinPacketAccountingBytes)
|
||||
done <- result{lease: lease, err: err}
|
||||
}()
|
||||
return done
|
||||
}
|
||||
waitForWaiters := func(want int) {
|
||||
t.Helper()
|
||||
deadline := time.Now().Add(time.Second)
|
||||
for {
|
||||
nativeXHTTPMemoryWait.Lock()
|
||||
got := nativeXHTTPMemoryWait.queued
|
||||
nativeXHTTPMemoryWait.Unlock()
|
||||
if got == want {
|
||||
return
|
||||
}
|
||||
if time.Now().After(deadline) {
|
||||
t.Fatalf("memory waiters = %d, want %d", got, want)
|
||||
}
|
||||
time.Sleep(time.Millisecond)
|
||||
}
|
||||
}
|
||||
|
||||
firstDone := startWaiter()
|
||||
waitForWaiters(1)
|
||||
secondDone := startWaiter()
|
||||
waitForWaiters(2)
|
||||
|
||||
filler.shrink(fillBytes - nativeXHTTPMinPacketAccountingBytes)
|
||||
first := <-firstDone
|
||||
if first.err != nil || first.lease == nil {
|
||||
t.Fatalf("first memory waiter = (%v, %v)", first.lease, first.err)
|
||||
}
|
||||
select {
|
||||
case second := <-secondDone:
|
||||
if second.lease != nil {
|
||||
second.lease.release()
|
||||
}
|
||||
t.Fatalf("second waiter woke before FIFO capacity was released: %v", second.err)
|
||||
case <-time.After(25 * time.Millisecond):
|
||||
}
|
||||
|
||||
first.lease.release()
|
||||
select {
|
||||
case second := <-secondDone:
|
||||
if second.err != nil || second.lease == nil {
|
||||
t.Fatalf("second memory waiter = (%v, %v)", second.lease, second.err)
|
||||
}
|
||||
second.lease.release()
|
||||
case <-time.After(time.Second):
|
||||
t.Fatal("second memory waiter did not wake after first released")
|
||||
}
|
||||
|
||||
filler.release()
|
||||
if got := nativeXHTTPBufferedBytes.Load(); got != before {
|
||||
t.Fatalf("FIFO waiter test leaked %d buffered bytes (baseline %d)", got, before)
|
||||
}
|
||||
}
|
||||
|
||||
func TestXHTTPReassemblyHasNoPacketRequestCountCeiling(t *testing.T) {
|
||||
before := nativeXHTTPBufferedBytes.Load()
|
||||
q := newNativeXHTTPUploadQueue(1, 4*nativeXHTTPMinPacketAccountingBytes)
|
||||
defer q.close()
|
||||
|
||||
done := make(chan error, 1)
|
||||
go func() {
|
||||
for _, seq := range []uint64{3, 2, 1, 0} {
|
||||
lease, ok := acquireNativeXHTTPMemory(nativeXHTTPAccountedPacketBytes(1))
|
||||
if !ok {
|
||||
done <- errors.New("could not reserve packet memory")
|
||||
return
|
||||
}
|
||||
err := q.push(context.Background(), nativeXHTTPPacket{Payload: []byte{byte('a' + seq)}, Seq: seq}, lease)
|
||||
lease.release()
|
||||
if err != nil {
|
||||
done <- err
|
||||
return
|
||||
}
|
||||
}
|
||||
done <- nil
|
||||
}()
|
||||
|
||||
buf := make([]byte, 1)
|
||||
for want := byte('a'); want <= byte('d'); want++ {
|
||||
n, err := q.Read(buf)
|
||||
if err != nil || n != 1 || buf[0] != want {
|
||||
t.Fatalf("reassembled packet = (%d, %v, %q), want %q", n, err, buf[:n], []byte{want})
|
||||
}
|
||||
}
|
||||
if err := <-done; err != nil {
|
||||
t.Fatalf("out-of-order burst was rejected: %v", err)
|
||||
}
|
||||
q.close()
|
||||
if got := nativeXHTTPBufferedBytes.Load(); got != before {
|
||||
t.Fatalf("reassembly test leaked %d buffered bytes (baseline %d)", got, before)
|
||||
}
|
||||
}
|
||||
|
||||
func TestXHTTPBodyReservationUsesActualContentLength(t *testing.T) {
|
||||
ib := &nativeInbound{xhttpMaxEachPostBytes: 1_000_000}
|
||||
req := httptest.NewRequest(http.MethodPost, "/session/0", strings.NewReader("small"))
|
||||
if got := ib.xhttpUploadReservationBytes(req); got != nativeXHTTPMinPacketAccountingBytes {
|
||||
t.Fatalf("body reservation = %d, want minimum accounted packet size", got)
|
||||
}
|
||||
req.ContentLength = -1
|
||||
if got := ib.xhttpUploadReservationBytes(req); got != 1_000_000 {
|
||||
t.Fatalf("chunked body reservation = %d, want configured maximum", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNativeQuotaResetWaitsForInFlightTraffic(t *testing.T) {
|
||||
const uuid = "22222222-2222-2222-2222-222222222222"
|
||||
state := &xrayNativeQuotaState{usedBytes: 123, generation: 1}
|
||||
m := &XrayManager{nativeQuotaByUUID: map[string]*xrayNativeQuotaState{uuid: state}}
|
||||
|
||||
state.trafficMu.RLock()
|
||||
done := make(chan struct{})
|
||||
go func() {
|
||||
m.resetNativeQuotaUsage(uuid)
|
||||
close(done)
|
||||
}()
|
||||
select {
|
||||
case <-done:
|
||||
state.trafficMu.RUnlock()
|
||||
t.Fatal("traffic reset crossed an in-flight writer boundary")
|
||||
case <-time.After(25 * time.Millisecond):
|
||||
}
|
||||
state.trafficMu.RUnlock()
|
||||
select {
|
||||
case <-done:
|
||||
case <-time.After(time.Second):
|
||||
t.Fatal("traffic reset did not complete after writer released")
|
||||
}
|
||||
state.mu.Lock()
|
||||
used, generation := state.usedBytes, state.generation
|
||||
state.mu.Unlock()
|
||||
if used != 0 || generation != 2 {
|
||||
t.Fatalf("reset state = used %d generation %d, want 0/2", used, generation)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNativeRateWaitCanBeCanceled(t *testing.T) {
|
||||
lim := rate.NewLimiter(1, 1)
|
||||
if !lim.AllowN(time.Now(), 1) {
|
||||
t.Fatal("failed to consume initial limiter token")
|
||||
}
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
cancel()
|
||||
if err := waitNativeRate(ctx, lim, 1); !errors.Is(err, context.Canceled) {
|
||||
t.Fatalf("waitNativeRate error = %v, want context.Canceled", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSSHDirtyQueueDoesNotScanInactiveUsers(t *testing.T) {
|
||||
sshTrafficDirtyMu.Lock()
|
||||
old := sshTrafficDirty
|
||||
sshTrafficDirty = make(map[string]*UserState)
|
||||
sshTrafficDirtyMu.Unlock()
|
||||
defer func() {
|
||||
sshTrafficDirtyMu.Lock()
|
||||
sshTrafficDirty = old
|
||||
sshTrafficDirtyMu.Unlock()
|
||||
}()
|
||||
|
||||
active := &UserState{Cfg: UserConfig{Username: "active"}}
|
||||
inactive := &UserState{Cfg: UserConfig{Username: "inactive"}}
|
||||
markSSHUserTrafficDirty(active)
|
||||
dirty := takeSSHUserTrafficDirty()
|
||||
if len(dirty) != 1 || dirty["active"] != active {
|
||||
t.Fatalf("dirty queue = %#v", dirty)
|
||||
}
|
||||
if _, found := dirty[inactive.Cfg.Username]; found {
|
||||
t.Fatal("inactive user appeared in dirty queue")
|
||||
}
|
||||
if next := takeSSHUserTrafficDirty(); len(next) != 0 {
|
||||
t.Fatalf("dirty queue was not drained: %#v", next)
|
||||
}
|
||||
}
|
||||
|
||||
func TestOldNativeConnectionCannotDecrementReplacementAccount(t *testing.T) {
|
||||
oldStore := statsStore
|
||||
statsStore = &Store{}
|
||||
defer func() { statsStore = oldStore }()
|
||||
|
||||
const uuid = "replacement-active-user"
|
||||
oldState := &xrayNativeQuotaState{maxConns: 1, generation: 1}
|
||||
m := &XrayManager{nativeQuotaByUUID: map[string]*xrayNativeQuotaState{uuid: oldState}}
|
||||
|
||||
release, acquiredState, ok := m.acquireNativeClientConnection(uuid, "old@example")
|
||||
if !ok || acquiredState != oldState {
|
||||
t.Fatal("failed to acquire old account connection")
|
||||
}
|
||||
// Discard the old account's successful connect delta so this assertion only
|
||||
// measures what happens when that old connection later disconnects.
|
||||
m.nativeDBMu.Lock()
|
||||
m.nativeActivePending = nil
|
||||
m.nativeDBMu.Unlock()
|
||||
|
||||
newState := &xrayNativeQuotaState{maxConns: 1, generation: 1}
|
||||
m.nativeQuotaMu.Lock()
|
||||
m.nativeQuotaByUUID[uuid] = newState
|
||||
m.nativeQuotaMu.Unlock()
|
||||
|
||||
release()
|
||||
|
||||
m.nativeDBMu.Lock()
|
||||
pending := m.nativeActivePending[uuid]
|
||||
m.nativeDBMu.Unlock()
|
||||
if pending.Delta != 0 || pending.State != nil {
|
||||
t.Fatalf("old disconnect was queued against replacement account: %+v", pending)
|
||||
}
|
||||
newState.mu.Lock()
|
||||
active := newState.activeConns
|
||||
newState.mu.Unlock()
|
||||
if active != 0 {
|
||||
t.Fatalf("replacement account active count changed to %d", active)
|
||||
}
|
||||
}
|
||||
|
||||
func TestOldNativeTrafficCannotAttachToReplacementAccount(t *testing.T) {
|
||||
oldStore := statsStore
|
||||
statsStore = &Store{}
|
||||
defer func() { statsStore = oldStore }()
|
||||
|
||||
const uuid = "replacement-traffic-user"
|
||||
oldState := &xrayNativeQuotaState{generation: 1}
|
||||
newState := &xrayNativeQuotaState{generation: 1}
|
||||
m := &XrayManager{nativeQuotaByUUID: map[string]*xrayNativeQuotaState{uuid: oldState}}
|
||||
|
||||
meter := newTrafficMeter(uuid, "old@example", true, oldState)
|
||||
meter.n = 1234
|
||||
|
||||
m.nativeQuotaMu.Lock()
|
||||
m.nativeQuotaByUUID[uuid] = newState
|
||||
m.nativeQuotaMu.Unlock()
|
||||
|
||||
oldMgr := xrayMgr
|
||||
xrayMgr = m
|
||||
defer func() { xrayMgr = oldMgr }()
|
||||
meter.flush()
|
||||
|
||||
m.nativeDBMu.Lock()
|
||||
pending := m.nativeTrafficPending[uuid]
|
||||
m.nativeDBMu.Unlock()
|
||||
if pending.Uplink != 0 || pending.Downlink != 0 || pending.State != nil {
|
||||
t.Fatalf("old traffic was queued against replacement account: %+v", pending)
|
||||
}
|
||||
m.statsMu.RLock()
|
||||
stat := m.statsByEmail["old@example"]
|
||||
m.statsMu.RUnlock()
|
||||
if stat.Uplink != 0 || stat.Downlink != 0 {
|
||||
t.Fatalf("old traffic resurfaced in runtime stats: %+v", stat)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNativeFlusherDropsMismatchedPolicyIdentity(t *testing.T) {
|
||||
oldStore := statsStore
|
||||
statsStore = &Store{}
|
||||
defer func() { statsStore = oldStore }()
|
||||
|
||||
const uuid = "identity-prune-user"
|
||||
oldState := &xrayNativeQuotaState{generation: 1}
|
||||
newState := &xrayNativeQuotaState{generation: 1}
|
||||
m := &XrayManager{
|
||||
nativeQuotaByUUID: map[string]*xrayNativeQuotaState{uuid: newState},
|
||||
nativeTrafficPending: map[string]xrayPendingTraffic{
|
||||
uuid: {Email: "old@example", Uplink: 99, State: oldState},
|
||||
},
|
||||
nativeActivePending: map[string]xrayPendingActive{
|
||||
uuid: {Email: "old@example", Delta: -1, State: oldState},
|
||||
},
|
||||
}
|
||||
m.flushNativeStatsToDB()
|
||||
|
||||
m.nativeDBMu.Lock()
|
||||
defer m.nativeDBMu.Unlock()
|
||||
if len(m.nativeTrafficPending) != 0 || len(m.nativeActivePending) != 0 {
|
||||
t.Fatalf("mismatched pending deltas survived prune: traffic=%v active=%v", m.nativeTrafficPending, m.nativeActivePending)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNativeMuxFinishRunsOnce(t *testing.T) {
|
||||
var calls atomic.Int32
|
||||
s := &nativeMuxSession{
|
||||
closed: make(chan struct{}),
|
||||
uplink: make(chan nativeMuxUplinkItem),
|
||||
onClose: func(*nativeMuxSession) {
|
||||
calls.Add(1)
|
||||
},
|
||||
}
|
||||
|
||||
var wg sync.WaitGroup
|
||||
for i := 0; i < 32; i++ {
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
s.finish()
|
||||
}()
|
||||
}
|
||||
wg.Wait()
|
||||
if got := calls.Load(); got != 1 {
|
||||
t.Fatalf("mux onClose called %d times, want 1", got)
|
||||
}
|
||||
}
|
||||
|
||||
type closeTrackingReader struct {
|
||||
closed atomic.Bool
|
||||
}
|
||||
|
||||
func (r *closeTrackingReader) Read([]byte) (int, error) { return 0, io.EOF }
|
||||
func (r *closeTrackingReader) Close() error {
|
||||
r.closed.Store(true)
|
||||
return nil
|
||||
}
|
||||
|
||||
func TestNativeXHTTPQueueCloseClosesQueuedStreamReader(t *testing.T) {
|
||||
q := newNativeXHTTPUploadQueue(1, 1024)
|
||||
r := &closeTrackingReader{}
|
||||
if err := q.push(context.Background(), nativeXHTTPPacket{Reader: r}, nil); err != nil {
|
||||
t.Fatalf("queue stream reader: %v", err)
|
||||
}
|
||||
q.close()
|
||||
if !r.closed.Load() {
|
||||
t.Fatal("queued stream reader was not closed during queue shutdown")
|
||||
}
|
||||
}
|
||||
|
||||
func TestNativeXHTTPQueueSkipsEmptyPacketsWithoutZeroProgressRead(t *testing.T) {
|
||||
before := nativeXHTTPBufferedBytes.Load()
|
||||
q := newNativeXHTTPUploadQueue(2, 2*nativeXHTTPMinPacketAccountingBytes)
|
||||
defer q.close()
|
||||
|
||||
for seq, payload := range [][]byte{nil, []byte("x")} {
|
||||
accounted := nativeXHTTPAccountedPacketBytes(int64(len(payload)))
|
||||
lease, ok := acquireNativeXHTTPMemory(accounted)
|
||||
if !ok {
|
||||
t.Fatal("failed to reserve packet memory")
|
||||
}
|
||||
if err := q.push(context.Background(), nativeXHTTPPacket{Payload: payload, Seq: uint64(seq)}, lease); err != nil {
|
||||
lease.release()
|
||||
t.Fatalf("queue packet %d: %v", seq, err)
|
||||
}
|
||||
lease.release()
|
||||
}
|
||||
|
||||
buf := make([]byte, 1)
|
||||
n, err := q.Read(buf)
|
||||
if err != nil || n != 1 || string(buf[:n]) != "x" {
|
||||
t.Fatalf("queue read after empty packet = (%d, %v, %q), want (1, nil, x)", n, err, buf[:n])
|
||||
}
|
||||
q.close()
|
||||
if got := nativeXHTTPBufferedBytes.Load(); got != before {
|
||||
t.Fatalf("empty-packet test leaked %d buffered bytes (baseline %d)", got, before)
|
||||
}
|
||||
}
|
||||
|
||||
type deadlineUnblockingResponseWriter struct {
|
||||
header http.Header
|
||||
writeStart chan struct{}
|
||||
unblock chan struct{}
|
||||
startOnce sync.Once
|
||||
unblockOnce sync.Once
|
||||
}
|
||||
|
||||
func newDeadlineUnblockingResponseWriter() *deadlineUnblockingResponseWriter {
|
||||
return &deadlineUnblockingResponseWriter{
|
||||
header: make(http.Header),
|
||||
writeStart: make(chan struct{}),
|
||||
unblock: make(chan struct{}),
|
||||
}
|
||||
}
|
||||
|
||||
func (w *deadlineUnblockingResponseWriter) Header() http.Header { return w.header }
|
||||
func (w *deadlineUnblockingResponseWriter) WriteHeader(int) {}
|
||||
func (w *deadlineUnblockingResponseWriter) Flush() {}
|
||||
func (w *deadlineUnblockingResponseWriter) Write([]byte) (int, error) {
|
||||
w.startOnce.Do(func() { close(w.writeStart) })
|
||||
<-w.unblock
|
||||
return 0, os.ErrDeadlineExceeded
|
||||
}
|
||||
func (w *deadlineUnblockingResponseWriter) SetWriteDeadline(deadline time.Time) error {
|
||||
if !deadline.IsZero() && !deadline.After(time.Now().Add(10*time.Millisecond)) {
|
||||
w.unblockOnce.Do(func() { close(w.unblock) })
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func TestNativeXHTTPResponseCloseInterruptsStalledWrite(t *testing.T) {
|
||||
underlying := newDeadlineUnblockingResponseWriter()
|
||||
writer := newNativeXHTTPResponseWriter(underlying)
|
||||
writeDone := make(chan error, 1)
|
||||
go func() {
|
||||
_, err := writer.Write([]byte("blocked"))
|
||||
writeDone <- err
|
||||
}()
|
||||
|
||||
select {
|
||||
case <-underlying.writeStart:
|
||||
case <-time.After(time.Second):
|
||||
t.Fatal("response write did not start")
|
||||
}
|
||||
|
||||
closeDone := make(chan struct{})
|
||||
go func() {
|
||||
writer.close()
|
||||
close(closeDone)
|
||||
}()
|
||||
select {
|
||||
case <-closeDone:
|
||||
case <-time.After(time.Second):
|
||||
t.Fatal("response close blocked behind stalled write")
|
||||
}
|
||||
select {
|
||||
case err := <-writeDone:
|
||||
if !errors.Is(err, os.ErrDeadlineExceeded) {
|
||||
t.Fatalf("stalled write error = %v, want deadline exceeded", err)
|
||||
}
|
||||
case <-time.After(time.Second):
|
||||
t.Fatal("stalled response write was not interrupted")
|
||||
}
|
||||
}
|
||||
+9
-2
@@ -20,8 +20,15 @@ func securePanelHandler(next http.Handler) http.Handler {
|
||||
w.Header().Set("Permissions-Policy", "camera=(), microphone=(), geolocation=(), payment=()")
|
||||
w.Header().Set("Cross-Origin-Opener-Policy", "same-origin")
|
||||
w.Header().Set("Content-Security-Policy", "default-src 'self'; base-uri 'none'; frame-ancestors 'none'; object-src 'none'; form-action 'self'; img-src 'self' data:; connect-src 'self'; style-src 'self' 'unsafe-inline'; script-src 'self' 'unsafe-inline'")
|
||||
if strings.HasPrefix(r.URL.Path, "/api/") || r.URL.Path == "/" || r.URL.Path == "/index.html" {
|
||||
w.Header().Set("Cache-Control", "no-store")
|
||||
// The panel is deployed in-place by update.sh. Do not let browsers or
|
||||
// reverse proxies keep an older JavaScript bundle after an update, because
|
||||
// stale form serializers can silently omit newly-added config fields.
|
||||
if strings.HasPrefix(r.URL.Path, "/api/") ||
|
||||
r.URL.Path == "/" || r.URL.Path == "/index.html" ||
|
||||
strings.HasPrefix(r.URL.Path, "/assets/") {
|
||||
w.Header().Set("Cache-Control", "no-store, no-cache, must-revalidate")
|
||||
w.Header().Set("Pragma", "no-cache")
|
||||
w.Header().Set("Expires", "0")
|
||||
}
|
||||
if r.Body != nil && r.Method != http.MethodGet && r.Method != http.MethodHead {
|
||||
r.Body = http.MaxBytesReader(w, r.Body, maxAdminRequestBody)
|
||||
|
||||
@@ -87,6 +87,17 @@ func serverConfigPost(w http.ResponseWriter, r *http.Request) {
|
||||
http.Error(w, "config exceeds 512 KiB", http.StatusRequestEntityTooLarge)
|
||||
return
|
||||
}
|
||||
// Keep track of optional field presence separately from its boolean value.
|
||||
// This protects a newly-added setting from being reset by a stale cached
|
||||
// panel bundle that does not know how to send the field yet.
|
||||
var fieldPresence struct {
|
||||
PAMAuthEnabled *bool `json:"pam_auth_enabled"`
|
||||
}
|
||||
if err := json.Unmarshal(body, &fieldPresence); err != nil {
|
||||
http.Error(w, "invalid JSON: "+err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
var newCfg Config
|
||||
if err := json.Unmarshal(body, &newCfg); err != nil {
|
||||
http.Error(w, "invalid JSON: "+err.Error(), http.StatusBadRequest)
|
||||
@@ -104,6 +115,9 @@ func serverConfigPost(w http.ResponseWriter, r *http.Request) {
|
||||
globalCfgMu.RLock()
|
||||
if globalCfg != nil {
|
||||
newCfg.Users = globalCfg.Users
|
||||
if fieldPresence.PAMAuthEnabled == nil {
|
||||
newCfg.PAMAuthEnabled = globalCfg.PAMAuthEnabled
|
||||
}
|
||||
}
|
||||
globalCfgMu.RUnlock()
|
||||
|
||||
|
||||
+3
-1
@@ -22,7 +22,9 @@ import (
|
||||
"time"
|
||||
)
|
||||
|
||||
const tlsCertsDir = "/opt/sshpanel/certs"
|
||||
// tlsCertsDir holds panel-managed certificates. It is a var so tests can point
|
||||
// it at a temporary directory.
|
||||
var tlsCertsDir = "/opt/sshpanel/certs"
|
||||
|
||||
var (
|
||||
tlsDNSNamePattern = regexp.MustCompile(`^(?:\*\.)?(?:[A-Za-z0-9](?:[A-Za-z0-9-]{0,61}[A-Za-z0-9])?\.)*[A-Za-z0-9](?:[A-Za-z0-9-]{0,61}[A-Za-z0-9])?$`)
|
||||
|
||||
@@ -0,0 +1,715 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"crypto/ecdsa"
|
||||
"crypto/ed25519"
|
||||
"crypto/rsa"
|
||||
"crypto/tls"
|
||||
"crypto/x509"
|
||||
"encoding/json"
|
||||
"encoding/pem"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sort"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// Certificate management for the panel: list the TLS material this node already
|
||||
// uses and replace it in place (fullchain + privkey) when an operator renews a
|
||||
// certificate. Replacing in place is what makes renewal painless — every place
|
||||
// that references the old paths (TLS forwarders, Xray inbounds) keeps working,
|
||||
// and only the listeners that actually serve the certificate are rebound.
|
||||
|
||||
const (
|
||||
tlsCertFileName = "cert.pem"
|
||||
tlsKeyFileName = "key.pem"
|
||||
// Two PEM blobs plus JSON overhead. Certificates are a few KB; RSA chains
|
||||
// with several intermediates still stay far below this.
|
||||
maxTLSCertRequestBody = 4 << 20
|
||||
maxTLSPEMBytes = 1 << 20
|
||||
// Certificates expiring inside this window are flagged in the panel.
|
||||
tlsCertExpiryWarnDays = 21
|
||||
)
|
||||
|
||||
// tlsCertUsage records one consumer of a certificate so the panel can show what
|
||||
// a replacement is going to affect.
|
||||
type tlsCertUsage struct {
|
||||
Kind string `json:"kind"` // tls_forwarder | xray_inbound
|
||||
Ref string `json:"ref"` // listen address or inbound tag
|
||||
}
|
||||
|
||||
type tlsCertInfo struct {
|
||||
Name string `json:"name"`
|
||||
CertFile string `json:"cert_file"`
|
||||
KeyFile string `json:"key_file"`
|
||||
Managed bool `json:"managed"` // stored under /opt/sshpanel/certs
|
||||
Exists bool `json:"exists"`
|
||||
Subject string `json:"subject,omitempty"`
|
||||
Issuer string `json:"issuer,omitempty"`
|
||||
Domains []string `json:"domains"`
|
||||
NotBefore string `json:"not_before,omitempty"`
|
||||
NotAfter string `json:"not_after,omitempty"`
|
||||
DaysLeft int `json:"days_left"`
|
||||
Expired bool `json:"expired"`
|
||||
Expiring bool `json:"expiring"`
|
||||
SelfSigned bool `json:"self_signed"`
|
||||
ChainLen int `json:"chain_length"`
|
||||
KeyType string `json:"key_type,omitempty"`
|
||||
KeyOK bool `json:"key_ok"`
|
||||
Modified string `json:"modified,omitempty"`
|
||||
Error string `json:"error,omitempty"`
|
||||
UsedBy []tlsCertUsage `json:"used_by"`
|
||||
}
|
||||
|
||||
type tlsCertRef struct {
|
||||
certFile string
|
||||
keyFile string
|
||||
managed bool
|
||||
usage []tlsCertUsage
|
||||
}
|
||||
|
||||
type tlsCertRefSet struct {
|
||||
byCert map[string]*tlsCertRef
|
||||
order []string
|
||||
}
|
||||
|
||||
func newTLSCertRefSet() *tlsCertRefSet {
|
||||
return &tlsCertRefSet{byCert: map[string]*tlsCertRef{}}
|
||||
}
|
||||
|
||||
func (s *tlsCertRefSet) add(certFile, keyFile string, usage ...tlsCertUsage) *tlsCertRef {
|
||||
certFile = strings.TrimSpace(certFile)
|
||||
if certFile == "" {
|
||||
return nil
|
||||
}
|
||||
certFile = filepath.Clean(certFile)
|
||||
ref, ok := s.byCert[certFile]
|
||||
if !ok {
|
||||
ref = &tlsCertRef{certFile: certFile, managed: isUnderTLSCertsDir(certFile)}
|
||||
s.byCert[certFile] = ref
|
||||
s.order = append(s.order, certFile)
|
||||
}
|
||||
if ref.keyFile == "" && strings.TrimSpace(keyFile) != "" {
|
||||
ref.keyFile = filepath.Clean(strings.TrimSpace(keyFile))
|
||||
}
|
||||
for _, u := range usage {
|
||||
if u.Kind == "" {
|
||||
continue
|
||||
}
|
||||
dup := false
|
||||
for _, have := range ref.usage {
|
||||
if have == u {
|
||||
dup = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !dup {
|
||||
ref.usage = append(ref.usage, u)
|
||||
}
|
||||
}
|
||||
return ref
|
||||
}
|
||||
|
||||
func (s *tlsCertRefSet) list() []*tlsCertRef {
|
||||
out := make([]*tlsCertRef, 0, len(s.order))
|
||||
for _, key := range s.order {
|
||||
out = append(out, s.byCert[key])
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func isUnderTLSCertsDir(path string) bool {
|
||||
rel, err := filepath.Rel(filepath.Clean(tlsCertsDir), filepath.Clean(path))
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
return rel != ".." && !strings.HasPrefix(rel, ".."+string(filepath.Separator))
|
||||
}
|
||||
|
||||
// samePathRef compares two file paths, following symlinks when both sides can be
|
||||
// resolved. /etc/letsencrypt/live/<domain>/fullchain.pem is a symlink, so a
|
||||
// plain string compare is not enough to match a config reference to a real file.
|
||||
func samePathRef(a, b string) bool {
|
||||
a, b = strings.TrimSpace(a), strings.TrimSpace(b)
|
||||
if a == "" || b == "" {
|
||||
return false
|
||||
}
|
||||
if filepath.Clean(a) == filepath.Clean(b) {
|
||||
return true
|
||||
}
|
||||
ra, errA := filepath.EvalSymlinks(a)
|
||||
rb, errB := filepath.EvalSymlinks(b)
|
||||
return errA == nil && errB == nil && ra == rb
|
||||
}
|
||||
|
||||
// collectTLSCertRefs gathers every certificate this node knows about: the ones
|
||||
// stored in the panel's cert directory plus the ones referenced by the running
|
||||
// config (TLS forwarders) and the Xray config (inbound tlsSettings).
|
||||
func collectTLSCertRefs() *tlsCertRefSet {
|
||||
set := newTLSCertRefSet()
|
||||
|
||||
gc := getGlobalCfg()
|
||||
var fallbackCert, fallbackKey string
|
||||
if gc != nil {
|
||||
for _, fwd := range gc.TLSForwarders {
|
||||
if strings.TrimSpace(fwd.CertFile) == "" {
|
||||
continue
|
||||
}
|
||||
if fallbackCert == "" {
|
||||
fallbackCert, fallbackKey = fwd.CertFile, fwd.KeyFile
|
||||
}
|
||||
listen := strings.TrimSpace(fwd.Listen)
|
||||
if listen == "" {
|
||||
listen = "(unbound)"
|
||||
}
|
||||
set.add(fwd.CertFile, fwd.KeyFile, tlsCertUsage{Kind: "tls_forwarder", Ref: listen})
|
||||
}
|
||||
}
|
||||
|
||||
for _, u := range xrayInboundCertUsage(fallbackCert, fallbackKey) {
|
||||
set.add(u.certFile, u.keyFile, tlsCertUsage{Kind: "xray_inbound", Ref: u.tag})
|
||||
}
|
||||
|
||||
// Panel-managed certificates (self-signed, pasted, or previously updated).
|
||||
entries, err := os.ReadDir(tlsCertsDir)
|
||||
if err == nil {
|
||||
names := make([]string, 0, len(entries))
|
||||
for _, e := range entries {
|
||||
if e.IsDir() {
|
||||
names = append(names, e.Name())
|
||||
}
|
||||
}
|
||||
sort.Strings(names)
|
||||
for _, name := range names {
|
||||
certFile := filepath.Join(tlsCertsDir, name, tlsCertFileName)
|
||||
if _, err := os.Stat(certFile); err != nil {
|
||||
continue
|
||||
}
|
||||
set.add(certFile, filepath.Join(tlsCertsDir, name, tlsKeyFileName))
|
||||
}
|
||||
}
|
||||
return set
|
||||
}
|
||||
|
||||
type xrayCertRef struct {
|
||||
tag string
|
||||
certFile string
|
||||
keyFile string
|
||||
}
|
||||
|
||||
// xrayInboundCertUsage returns the certificate each TLS-enabled Xray inbound
|
||||
// serves. Inbounds that enable TLS without naming a certificate inherit the
|
||||
// first TLS forwarder's material (see buildInboundTLS), so they are reported
|
||||
// against that path — replacing it does affect them.
|
||||
func xrayInboundCertUsage(fallbackCert, fallbackKey string) []xrayCertRef {
|
||||
if xrayMgr == nil {
|
||||
return nil
|
||||
}
|
||||
data, err := xrayMgr.GetConfig()
|
||||
if err != nil || len(data) == 0 {
|
||||
return nil
|
||||
}
|
||||
var cf struct {
|
||||
Inbounds []struct {
|
||||
Tag string `json:"tag"`
|
||||
StreamSettings struct {
|
||||
Security string `json:"security"`
|
||||
TLSSettings struct {
|
||||
Certificates []struct {
|
||||
CertificateFile string `json:"certificateFile"`
|
||||
KeyFile string `json:"keyFile"`
|
||||
} `json:"certificates"`
|
||||
} `json:"tlsSettings"`
|
||||
} `json:"streamSettings"`
|
||||
} `json:"inbounds"`
|
||||
}
|
||||
if err := json.Unmarshal(data, &cf); err != nil {
|
||||
return nil
|
||||
}
|
||||
var out []xrayCertRef
|
||||
for i, in := range cf.Inbounds {
|
||||
security := strings.ToLower(strings.TrimSpace(in.StreamSettings.Security))
|
||||
certs := in.StreamSettings.TLSSettings.Certificates
|
||||
if security != "tls" && len(certs) == 0 {
|
||||
continue
|
||||
}
|
||||
tag := strings.TrimSpace(in.Tag)
|
||||
if tag == "" {
|
||||
tag = fmt.Sprintf("inbound-%d", i+1)
|
||||
}
|
||||
if len(certs) > 0 && strings.TrimSpace(certs[0].CertificateFile) != "" {
|
||||
out = append(out, xrayCertRef{tag: tag, certFile: certs[0].CertificateFile, keyFile: certs[0].KeyFile})
|
||||
continue
|
||||
}
|
||||
if security == "tls" && strings.TrimSpace(fallbackCert) != "" {
|
||||
out = append(out, xrayCertRef{tag: tag + " (herda do TLS forwarder)", certFile: fallbackCert, keyFile: fallbackKey})
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func tlsCertDisplayName(certFile string) string {
|
||||
dir := filepath.Base(filepath.Dir(certFile))
|
||||
if dir == "" || dir == "." || dir == string(filepath.Separator) {
|
||||
return filepath.Base(certFile)
|
||||
}
|
||||
if dir == "live" || dir == "certs" {
|
||||
return filepath.Base(certFile)
|
||||
}
|
||||
return dir
|
||||
}
|
||||
|
||||
func tlsKeyTypeName(key interface{}) string {
|
||||
switch k := key.(type) {
|
||||
case *rsa.PrivateKey:
|
||||
return fmt.Sprintf("RSA %d", k.N.BitLen())
|
||||
case *ecdsa.PrivateKey:
|
||||
return "ECDSA " + k.Curve.Params().Name
|
||||
case ed25519.PrivateKey:
|
||||
return "Ed25519"
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func parsePEMCertChain(data []byte) ([]*x509.Certificate, error) {
|
||||
var chain []*x509.Certificate
|
||||
rest := data
|
||||
for {
|
||||
var block *pem.Block
|
||||
block, rest = pem.Decode(rest)
|
||||
if block == nil {
|
||||
break
|
||||
}
|
||||
if block.Type != "CERTIFICATE" {
|
||||
continue
|
||||
}
|
||||
crt, err := x509.ParseCertificate(block.Bytes)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
chain = append(chain, crt)
|
||||
}
|
||||
if len(chain) == 0 {
|
||||
return nil, fmt.Errorf("no CERTIFICATE block found")
|
||||
}
|
||||
return chain, nil
|
||||
}
|
||||
|
||||
func certDomains(leaf *x509.Certificate) []string {
|
||||
seen := map[string]bool{}
|
||||
out := make([]string, 0, len(leaf.DNSNames)+len(leaf.IPAddresses)+1)
|
||||
for _, d := range leaf.DNSNames {
|
||||
if d = strings.TrimSpace(d); d != "" && !seen[d] {
|
||||
seen[d] = true
|
||||
out = append(out, d)
|
||||
}
|
||||
}
|
||||
for _, ip := range leaf.IPAddresses {
|
||||
s := ip.String()
|
||||
if !seen[s] {
|
||||
seen[s] = true
|
||||
out = append(out, s)
|
||||
}
|
||||
}
|
||||
if len(out) == 0 && strings.TrimSpace(leaf.Subject.CommonName) != "" {
|
||||
out = append(out, strings.TrimSpace(leaf.Subject.CommonName))
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func describeTLSCert(ref *tlsCertRef) tlsCertInfo {
|
||||
info := tlsCertInfo{
|
||||
Name: tlsCertDisplayName(ref.certFile),
|
||||
CertFile: ref.certFile,
|
||||
KeyFile: ref.keyFile,
|
||||
Managed: ref.managed,
|
||||
Domains: []string{},
|
||||
UsedBy: ref.usage,
|
||||
}
|
||||
if info.UsedBy == nil {
|
||||
info.UsedBy = []tlsCertUsage{}
|
||||
}
|
||||
st, err := os.Stat(ref.certFile)
|
||||
if err != nil {
|
||||
info.Error = "arquivo não encontrado"
|
||||
return info
|
||||
}
|
||||
info.Exists = true
|
||||
info.Modified = st.ModTime().UTC().Format(time.RFC3339)
|
||||
|
||||
certPEM, err := os.ReadFile(ref.certFile)
|
||||
if err != nil {
|
||||
info.Error = "leitura do certificado: " + err.Error()
|
||||
return info
|
||||
}
|
||||
chain, err := parsePEMCertChain(certPEM)
|
||||
if err != nil {
|
||||
info.Error = "certificado inválido: " + err.Error()
|
||||
return info
|
||||
}
|
||||
leaf := chain[0]
|
||||
info.ChainLen = len(chain)
|
||||
info.Subject = leaf.Subject.CommonName
|
||||
info.Issuer = leaf.Issuer.CommonName
|
||||
if info.Issuer == "" && len(leaf.Issuer.Organization) > 0 {
|
||||
info.Issuer = leaf.Issuer.Organization[0]
|
||||
}
|
||||
info.Domains = certDomains(leaf)
|
||||
info.NotBefore = leaf.NotBefore.UTC().Format(time.RFC3339)
|
||||
info.NotAfter = leaf.NotAfter.UTC().Format(time.RFC3339)
|
||||
info.SelfSigned = string(leaf.RawIssuer) == string(leaf.RawSubject)
|
||||
now := time.Now()
|
||||
info.Expired = now.After(leaf.NotAfter)
|
||||
info.DaysLeft = int(leaf.NotAfter.Sub(now).Hours() / 24)
|
||||
info.Expiring = !info.Expired && info.DaysLeft <= tlsCertExpiryWarnDays
|
||||
|
||||
if ref.keyFile != "" {
|
||||
keyPEM, err := os.ReadFile(ref.keyFile)
|
||||
if err != nil {
|
||||
info.Error = "leitura da chave: " + err.Error()
|
||||
return info
|
||||
}
|
||||
pair, err := tls.X509KeyPair(certPEM, keyPEM)
|
||||
if err != nil {
|
||||
info.Error = "a chave privada não corresponde ao certificado"
|
||||
return info
|
||||
}
|
||||
info.KeyOK = true
|
||||
info.KeyType = tlsKeyTypeName(pair.PrivateKey)
|
||||
} else {
|
||||
info.Error = "nenhuma chave privada associada"
|
||||
}
|
||||
return info
|
||||
}
|
||||
|
||||
// handleTLSCertList returns every certificate this node uses, with expiry and
|
||||
// the listeners/inbounds that serve it.
|
||||
func handleTLSCertList(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
refs := collectTLSCertRefs().list()
|
||||
out := make([]tlsCertInfo, 0, len(refs))
|
||||
for _, ref := range refs {
|
||||
out = append(out, describeTLSCert(ref))
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
_ = json.NewEncoder(w).Encode(map[string]interface{}{
|
||||
"certs_dir": tlsCertsDir,
|
||||
"certs": out,
|
||||
})
|
||||
}
|
||||
|
||||
type tlsCertUpdateRequest struct {
|
||||
// Name creates or replaces a panel-managed certificate under
|
||||
// /opt/sshpanel/certs/<name>/. Ignored when CertFile is set.
|
||||
Name string `json:"name"`
|
||||
// CertFile/KeyFile target an existing certificate in place so every
|
||||
// reference to those paths keeps working after the renewal.
|
||||
CertFile string `json:"cert_file"`
|
||||
KeyFile string `json:"key_file"`
|
||||
// Fullchain/Privkey hold the PEM text. cert/key are accepted as aliases.
|
||||
Fullchain string `json:"fullchain"`
|
||||
Privkey string `json:"privkey"`
|
||||
Cert string `json:"cert"`
|
||||
Key string `json:"key"`
|
||||
Reload *bool `json:"reload"`
|
||||
Force bool `json:"force"`
|
||||
}
|
||||
|
||||
type tlsCertReloadResult struct {
|
||||
TLSForwarders []string `json:"tls_forwarders"`
|
||||
XrayInbounds []string `json:"xray_inbounds"`
|
||||
XrayRestarted bool `json:"xray_restarted"`
|
||||
}
|
||||
|
||||
func normalizeTLSFilePath(raw string) (string, error) {
|
||||
p := strings.TrimSpace(raw)
|
||||
if p == "" {
|
||||
return "", fmt.Errorf("caminho vazio")
|
||||
}
|
||||
if strings.ContainsAny(p, "\x00\r\n") {
|
||||
return "", fmt.Errorf("caminho inválido")
|
||||
}
|
||||
if !filepath.IsAbs(p) {
|
||||
return "", fmt.Errorf("o caminho precisa ser absoluto")
|
||||
}
|
||||
return filepath.Clean(p), nil
|
||||
}
|
||||
|
||||
func normalizePEMText(raw string) string {
|
||||
s := strings.ReplaceAll(strings.TrimSpace(raw), "\r\n", "\n")
|
||||
s = strings.ReplaceAll(s, "\r", "\n")
|
||||
if s == "" {
|
||||
return ""
|
||||
}
|
||||
return s + "\n"
|
||||
}
|
||||
|
||||
// resolveTLSCertTarget decides which files the new PEM material is written to
|
||||
// and rejects paths that are neither panel-managed nor already referenced by the
|
||||
// running configuration. Without that check this endpoint would be an arbitrary
|
||||
// root file-write primitive.
|
||||
func resolveTLSCertTarget(req tlsCertUpdateRequest) (certFile, keyFile string, warnings []string, err error) {
|
||||
if strings.TrimSpace(req.CertFile) != "" {
|
||||
certFile, err = normalizeTLSFilePath(req.CertFile)
|
||||
if err != nil {
|
||||
return "", "", nil, err
|
||||
}
|
||||
refs := collectTLSCertRefs()
|
||||
var known *tlsCertRef
|
||||
for _, ref := range refs.list() {
|
||||
if samePathRef(ref.certFile, certFile) {
|
||||
known = ref
|
||||
break
|
||||
}
|
||||
}
|
||||
if known == nil && !isUnderTLSCertsDir(certFile) {
|
||||
return "", "", nil, fmt.Errorf("caminho não gerenciado pelo painel: use um certificado já referenciado na configuração ou informe um nome para armazenar em %s", tlsCertsDir)
|
||||
}
|
||||
if strings.TrimSpace(req.KeyFile) != "" {
|
||||
keyFile, err = normalizeTLSFilePath(req.KeyFile)
|
||||
if err != nil {
|
||||
return "", "", nil, err
|
||||
}
|
||||
} else if known != nil && known.keyFile != "" {
|
||||
keyFile = known.keyFile
|
||||
} else {
|
||||
keyFile = filepath.Join(filepath.Dir(certFile), tlsKeyFileName)
|
||||
}
|
||||
if !isUnderTLSCertsDir(keyFile) {
|
||||
keyKnown := known != nil && samePathRef(known.keyFile, keyFile)
|
||||
if !keyKnown && filepath.Dir(keyFile) != filepath.Dir(certFile) {
|
||||
return "", "", nil, fmt.Errorf("a chave precisa estar na mesma pasta do certificado ou já estar referenciada na configuração")
|
||||
}
|
||||
}
|
||||
return certFile, keyFile, warnings, nil
|
||||
}
|
||||
|
||||
name, nameErr := normalizeTLSStoreName(req.Name)
|
||||
if nameErr != nil {
|
||||
return "", "", nil, fmt.Errorf("informe cert_file de um certificado existente ou um nome para armazenar: %v", nameErr)
|
||||
}
|
||||
dir := filepath.Join(tlsCertsDir, name)
|
||||
return filepath.Join(dir, tlsCertFileName), filepath.Join(dir, tlsKeyFileName), warnings, nil
|
||||
}
|
||||
|
||||
// writeTLSMaterial replaces path with data, keeping a .bak copy of the previous
|
||||
// content and preserving the existing file mode. Symlinked targets (certbot
|
||||
// layout) are followed so the link structure survives the update.
|
||||
func writeTLSMaterial(path string, data []byte, defaultMode os.FileMode) (string, []string, error) {
|
||||
var warnings []string
|
||||
target := path
|
||||
if lst, err := os.Lstat(path); err == nil && lst.Mode()&os.ModeSymlink != 0 {
|
||||
if resolved, err := filepath.EvalSymlinks(path); err == nil {
|
||||
target = resolved
|
||||
warnings = append(warnings, fmt.Sprintf("%s é um link para %s; o conteúdo real foi substituído", path, resolved))
|
||||
}
|
||||
}
|
||||
mode := defaultMode
|
||||
if st, err := os.Stat(target); err == nil {
|
||||
mode = st.Mode().Perm()
|
||||
if old, err := os.ReadFile(target); err == nil {
|
||||
if err := writeFileAtomic(target+".bak", old, mode); err != nil {
|
||||
warnings = append(warnings, "não foi possível gravar backup de "+filepath.Base(target)+": "+err.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
if err := os.MkdirAll(filepath.Dir(target), 0o700); err != nil {
|
||||
return target, warnings, err
|
||||
}
|
||||
if err := writeFileAtomic(target, data, mode); err != nil {
|
||||
return target, warnings, err
|
||||
}
|
||||
return target, warnings, nil
|
||||
}
|
||||
|
||||
// handleTLSCertUpdate replaces a certificate's fullchain + private key and
|
||||
// reloads whatever serves it, so a renewal takes effect without touching any
|
||||
// other configuration.
|
||||
func handleTLSCertUpdate(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
r.Body = http.MaxBytesReader(w, r.Body, maxTLSCertRequestBody)
|
||||
var req tlsCertUpdateRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
http.Error(w, "corpo inválido: "+err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if strings.TrimSpace(req.Fullchain) == "" {
|
||||
req.Fullchain = req.Cert
|
||||
}
|
||||
if strings.TrimSpace(req.Privkey) == "" {
|
||||
req.Privkey = req.Key
|
||||
}
|
||||
certPEM := normalizePEMText(req.Fullchain)
|
||||
keyPEM := normalizePEMText(req.Privkey)
|
||||
if certPEM == "" || keyPEM == "" {
|
||||
http.Error(w, "fullchain (certificado) e privkey (chave privada) são obrigatórios", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if len(certPEM) > maxTLSPEMBytes || len(keyPEM) > maxTLSPEMBytes {
|
||||
http.Error(w, "certificado ou chave muito grandes", http.StatusRequestEntityTooLarge)
|
||||
return
|
||||
}
|
||||
|
||||
pair, err := tls.X509KeyPair([]byte(certPEM), []byte(keyPEM))
|
||||
if err != nil || len(pair.Certificate) == 0 {
|
||||
http.Error(w, "certificado e chave privada inválidos ou não correspondentes", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
chain, err := parsePEMCertChain([]byte(certPEM))
|
||||
if err != nil {
|
||||
http.Error(w, "certificado inválido: "+err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
leaf := chain[0]
|
||||
now := time.Now()
|
||||
if now.After(leaf.NotAfter) && !req.Force {
|
||||
http.Error(w, fmt.Sprintf("este certificado expirou em %s; envie force=true para gravar mesmo assim",
|
||||
leaf.NotAfter.UTC().Format("2006-01-02")), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
certFile, keyFile, warnings, err := resolveTLSCertTarget(req)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
if len(chain) < 2 && !leaf.IsCA && string(leaf.RawIssuer) != string(leaf.RawSubject) {
|
||||
warnings = append(warnings, "o PEM enviado contém apenas o certificado final; cole o fullchain.pem completo para evitar erros de cadeia em alguns clientes")
|
||||
}
|
||||
if now.Before(leaf.NotBefore) {
|
||||
warnings = append(warnings, "o certificado só é válido a partir de "+leaf.NotBefore.UTC().Format("2006-01-02 15:04")+" UTC")
|
||||
}
|
||||
if now.After(leaf.NotAfter) {
|
||||
warnings = append(warnings, "certificado já expirado — gravado por causa de force=true")
|
||||
}
|
||||
// Domain mismatch is usually a wrong paste, but a domain change can be
|
||||
// intentional, so it is reported rather than blocked.
|
||||
if oldPEM, err := os.ReadFile(certFile); err == nil {
|
||||
if oldChain, err := parsePEMCertChain(oldPEM); err == nil {
|
||||
oldDomains, newDomains := certDomains(oldChain[0]), certDomains(leaf)
|
||||
if strings.Join(oldDomains, ",") != strings.Join(newDomains, ",") {
|
||||
warnings = append(warnings, fmt.Sprintf("os domínios mudaram: antes %s, agora %s",
|
||||
strings.Join(oldDomains, ", "), strings.Join(newDomains, ", ")))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
writtenCert, certWarn, err := writeTLSMaterial(certFile, []byte(certPEM), 0o600)
|
||||
warnings = append(warnings, certWarn...)
|
||||
if err != nil {
|
||||
http.Error(w, "gravar certificado: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
writtenKey, keyWarn, err := writeTLSMaterial(keyFile, []byte(keyPEM), 0o600)
|
||||
warnings = append(warnings, keyWarn...)
|
||||
if err != nil {
|
||||
http.Error(w, "gravar chave: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
log.Printf("tls: certificate updated cert=%s key=%s cn=%q not_after=%s",
|
||||
writtenCert, writtenKey, leaf.Subject.CommonName, leaf.NotAfter.UTC().Format(time.RFC3339))
|
||||
|
||||
reload := tlsCertReloadResult{TLSForwarders: []string{}, XrayInbounds: []string{}}
|
||||
if req.Reload == nil || *req.Reload {
|
||||
var reloadWarn []string
|
||||
reload, reloadWarn = reloadTLSCertConsumers(certFile, keyFile)
|
||||
warnings = append(warnings, reloadWarn...)
|
||||
}
|
||||
|
||||
info := describeTLSCert(&tlsCertRef{
|
||||
certFile: certFile,
|
||||
keyFile: keyFile,
|
||||
managed: isUnderTLSCertsDir(certFile),
|
||||
usage: certUsageFor(certFile, keyFile),
|
||||
})
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
_ = json.NewEncoder(w).Encode(map[string]interface{}{
|
||||
"cert_file": certFile,
|
||||
"key_file": keyFile,
|
||||
"cert": info,
|
||||
"reloaded": reload,
|
||||
"warnings": warnings,
|
||||
})
|
||||
}
|
||||
|
||||
func certUsageFor(certFile, keyFile string) []tlsCertUsage {
|
||||
for _, ref := range collectTLSCertRefs().list() {
|
||||
if samePathRef(ref.certFile, certFile) {
|
||||
return ref.usage
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// reloadTLSCertConsumers rebinds the TLS forwarders that serve the replaced
|
||||
// certificate and restarts Xray when one of its inbounds uses it. Certificates
|
||||
// are read once when a listener is created, so nothing short of rebinding picks
|
||||
// up new material. Established connections are not owned by the listeners and
|
||||
// keep running.
|
||||
func reloadTLSCertConsumers(certFile, keyFile string) (tlsCertReloadResult, []string) {
|
||||
result := tlsCertReloadResult{TLSForwarders: []string{}, XrayInbounds: []string{}}
|
||||
var warnings []string
|
||||
|
||||
gc := getGlobalCfg()
|
||||
var fallbackCert, fallbackKey string
|
||||
if gc != nil {
|
||||
for _, fwd := range gc.TLSForwarders {
|
||||
if strings.TrimSpace(fwd.CertFile) != "" {
|
||||
fallbackCert, fallbackKey = fwd.CertFile, fwd.KeyFile
|
||||
break
|
||||
}
|
||||
}
|
||||
var affected []string
|
||||
for _, fwd := range gc.TLSForwarders {
|
||||
if samePathRef(fwd.CertFile, certFile) || samePathRef(fwd.KeyFile, keyFile) {
|
||||
if listen := strings.TrimSpace(fwd.Listen); listen != "" {
|
||||
affected = append(affected, listen)
|
||||
}
|
||||
}
|
||||
}
|
||||
if len(affected) > 0 && tlsPool != nil {
|
||||
tlsPool.Drop(affected, "certificate updated")
|
||||
for _, e := range tlsPool.Sync(gc.TLSForwarders) {
|
||||
warnings = append(warnings, fmt.Sprintf("recarregar TLS forwarder: %v", e))
|
||||
}
|
||||
for _, addr := range affected {
|
||||
if tlsPool.Has(addr) {
|
||||
result.TLSForwarders = append(result.TLSForwarders, addr)
|
||||
} else {
|
||||
warnings = append(warnings, "o TLS forwarder "+addr+" não voltou a escutar; verifique os logs")
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for _, u := range xrayInboundCertUsage(fallbackCert, fallbackKey) {
|
||||
if samePathRef(u.certFile, certFile) || samePathRef(u.keyFile, keyFile) {
|
||||
result.XrayInbounds = append(result.XrayInbounds, u.tag)
|
||||
}
|
||||
}
|
||||
if len(result.XrayInbounds) > 0 && xrayMgr != nil {
|
||||
st := xrayMgr.Status()
|
||||
if st.Enabled && st.Running {
|
||||
if err := xrayMgr.Restart(); err != nil {
|
||||
warnings = append(warnings, fmt.Sprintf("reiniciar Xray: %v", err))
|
||||
} else {
|
||||
result.XrayRestarted = true
|
||||
}
|
||||
} else if st.Enabled {
|
||||
warnings = append(warnings, "o Xray usa este certificado mas não está em execução")
|
||||
}
|
||||
}
|
||||
return result, warnings
|
||||
}
|
||||
@@ -0,0 +1,306 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"crypto/ecdsa"
|
||||
"crypto/elliptic"
|
||||
"crypto/rand"
|
||||
"crypto/x509"
|
||||
"crypto/x509/pkix"
|
||||
"encoding/json"
|
||||
"encoding/pem"
|
||||
"math/big"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
// makeTestCertPair returns PEM cert/key material for the given domain.
|
||||
func makeTestCertPair(t *testing.T, domain string, notBefore, notAfter time.Time) (certPEM, keyPEM string) {
|
||||
t.Helper()
|
||||
priv, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
|
||||
if err != nil {
|
||||
t.Fatalf("keygen: %v", err)
|
||||
}
|
||||
tmpl := &x509.Certificate{
|
||||
SerialNumber: big.NewInt(time.Now().UnixNano()),
|
||||
Subject: pkix.Name{CommonName: domain},
|
||||
NotBefore: notBefore,
|
||||
NotAfter: notAfter,
|
||||
KeyUsage: x509.KeyUsageDigitalSignature,
|
||||
ExtKeyUsage: []x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth},
|
||||
DNSNames: []string{domain},
|
||||
}
|
||||
der, err := x509.CreateCertificate(rand.Reader, tmpl, tmpl, &priv.PublicKey, priv)
|
||||
if err != nil {
|
||||
t.Fatalf("certgen: %v", err)
|
||||
}
|
||||
keyDER, err := x509.MarshalECPrivateKey(priv)
|
||||
if err != nil {
|
||||
t.Fatalf("marshal key: %v", err)
|
||||
}
|
||||
certPEM = string(pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: der}))
|
||||
keyPEM = string(pem.EncodeToMemory(&pem.Block{Type: "EC PRIVATE KEY", Bytes: keyDER}))
|
||||
return certPEM, keyPEM
|
||||
}
|
||||
|
||||
func useTempCertsDir(t *testing.T) string {
|
||||
t.Helper()
|
||||
dir := t.TempDir()
|
||||
old := tlsCertsDir
|
||||
tlsCertsDir = dir
|
||||
t.Cleanup(func() { tlsCertsDir = old })
|
||||
oldCfg := getGlobalCfg()
|
||||
t.Cleanup(func() { setGlobalCfg(oldCfg) })
|
||||
return dir
|
||||
}
|
||||
|
||||
func postCertUpdate(t *testing.T, body map[string]interface{}) *httptest.ResponseRecorder {
|
||||
t.Helper()
|
||||
raw, err := json.Marshal(body)
|
||||
if err != nil {
|
||||
t.Fatalf("marshal: %v", err)
|
||||
}
|
||||
req := httptest.NewRequest(http.MethodPost, "/api/tls/certs/update", strings.NewReader(string(raw)))
|
||||
rec := httptest.NewRecorder()
|
||||
handleTLSCertUpdate(rec, req)
|
||||
return rec
|
||||
}
|
||||
|
||||
func TestCertUpdateStoresNamedCertAndReportsExpiry(t *testing.T) {
|
||||
dir := useTempCertsDir(t)
|
||||
certPEM, keyPEM := makeTestCertPair(t, "panel.example.com", time.Now().Add(-time.Hour), time.Now().Add(30*24*time.Hour))
|
||||
|
||||
rec := postCertUpdate(t, map[string]interface{}{
|
||||
"name": "panel-example",
|
||||
"fullchain": certPEM,
|
||||
"privkey": keyPEM,
|
||||
"reload": false,
|
||||
})
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("status %d: %s", rec.Code, rec.Body.String())
|
||||
}
|
||||
var resp struct {
|
||||
CertFile string `json:"cert_file"`
|
||||
KeyFile string `json:"key_file"`
|
||||
Cert tlsCertInfo `json:"cert"`
|
||||
}
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
|
||||
t.Fatalf("decode: %v", err)
|
||||
}
|
||||
wantCert := filepath.Join(dir, "panel-example", tlsCertFileName)
|
||||
if filepath.Clean(resp.CertFile) != wantCert {
|
||||
t.Fatalf("cert_file = %q, want %q", resp.CertFile, wantCert)
|
||||
}
|
||||
if !resp.Cert.KeyOK {
|
||||
t.Fatalf("expected key to match certificate: %+v", resp.Cert)
|
||||
}
|
||||
if resp.Cert.Expired || resp.Cert.DaysLeft < 25 {
|
||||
t.Fatalf("unexpected expiry data: %+v", resp.Cert)
|
||||
}
|
||||
if len(resp.Cert.Domains) != 1 || resp.Cert.Domains[0] != "panel.example.com" {
|
||||
t.Fatalf("domains = %v", resp.Cert.Domains)
|
||||
}
|
||||
data, err := os.ReadFile(wantCert)
|
||||
if err != nil || !strings.Contains(string(data), "BEGIN CERTIFICATE") {
|
||||
t.Fatalf("cert not written: %v", err)
|
||||
}
|
||||
if _, err := os.Stat(filepath.Join(dir, "panel-example", tlsKeyFileName)); err != nil {
|
||||
t.Fatalf("key not written: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCertUpdateReplacesInPlaceAndKeepsBackup(t *testing.T) {
|
||||
dir := useTempCertsDir(t)
|
||||
oldCert, oldKey := makeTestCertPair(t, "old.example.com", time.Now().Add(-time.Hour), time.Now().Add(24*time.Hour))
|
||||
if rec := postCertUpdate(t, map[string]interface{}{
|
||||
"name": "renew-me", "fullchain": oldCert, "privkey": oldKey, "reload": false,
|
||||
}); rec.Code != http.StatusOK {
|
||||
t.Fatalf("seed failed: %s", rec.Body.String())
|
||||
}
|
||||
certFile := filepath.Join(dir, "renew-me", tlsCertFileName)
|
||||
keyFile := filepath.Join(dir, "renew-me", tlsKeyFileName)
|
||||
|
||||
newCert, newKey := makeTestCertPair(t, "new.example.com", time.Now().Add(-time.Hour), time.Now().Add(90*24*time.Hour))
|
||||
rec := postCertUpdate(t, map[string]interface{}{
|
||||
"cert_file": certFile, "key_file": keyFile,
|
||||
"fullchain": newCert, "privkey": newKey, "reload": false,
|
||||
})
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("status %d: %s", rec.Code, rec.Body.String())
|
||||
}
|
||||
var resp struct {
|
||||
Cert tlsCertInfo `json:"cert"`
|
||||
Warnings []string `json:"warnings"`
|
||||
}
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
|
||||
t.Fatalf("decode: %v", err)
|
||||
}
|
||||
if resp.Cert.Domains[0] != "new.example.com" {
|
||||
t.Fatalf("cert was not replaced: %+v", resp.Cert)
|
||||
}
|
||||
backup, err := os.ReadFile(certFile + ".bak")
|
||||
if err != nil {
|
||||
t.Fatalf("no backup written: %v", err)
|
||||
}
|
||||
if strings.TrimSpace(string(backup)) != strings.TrimSpace(oldCert) {
|
||||
t.Fatal("backup does not hold the previous certificate")
|
||||
}
|
||||
if _, err := os.Stat(keyFile + ".bak"); err != nil {
|
||||
t.Fatalf("no key backup: %v", err)
|
||||
}
|
||||
joined := strings.Join(resp.Warnings, " | ")
|
||||
if !strings.Contains(joined, "domínios mudaram") {
|
||||
t.Fatalf("expected a domain-change warning, got %q", joined)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCertUpdateRejectsBadInput(t *testing.T) {
|
||||
dir := useTempCertsDir(t)
|
||||
certPEM, keyPEM := makeTestCertPair(t, "a.example.com", time.Now().Add(-time.Hour), time.Now().Add(24*time.Hour))
|
||||
_, otherKey := makeTestCertPair(t, "b.example.com", time.Now().Add(-time.Hour), time.Now().Add(24*time.Hour))
|
||||
expiredCert, expiredKey := makeTestCertPair(t, "old.example.com", time.Now().Add(-48*time.Hour), time.Now().Add(-time.Hour))
|
||||
// Absolute, but neither panel-managed nor referenced by the configuration.
|
||||
unmanaged := t.TempDir()
|
||||
|
||||
cases := []struct {
|
||||
name string
|
||||
body map[string]interface{}
|
||||
want string
|
||||
}{
|
||||
{"missing key", map[string]interface{}{"name": "x", "fullchain": certPEM}, "obrigatórios"},
|
||||
{"mismatched pair", map[string]interface{}{"name": "x", "fullchain": certPEM, "privkey": otherKey}, "não correspondentes"},
|
||||
{"expired without force", map[string]interface{}{"name": "x", "fullchain": expiredCert, "privkey": expiredKey}, "expirou"},
|
||||
{"unmanaged path", map[string]interface{}{
|
||||
"cert_file": filepath.Join(unmanaged, "cert.pem"),
|
||||
"key_file": filepath.Join(unmanaged, "key.pem"),
|
||||
"fullchain": certPEM, "privkey": keyPEM,
|
||||
}, "não gerenciado"},
|
||||
{"relative path", map[string]interface{}{"cert_file": "certs/cert.pem", "fullchain": certPEM, "privkey": keyPEM}, "absoluto"},
|
||||
{"bad name", map[string]interface{}{"name": "../escape", "fullchain": certPEM, "privkey": keyPEM}, "nome"},
|
||||
}
|
||||
for _, tc := range cases {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
rec := postCertUpdate(t, tc.body)
|
||||
if rec.Code != http.StatusBadRequest {
|
||||
t.Fatalf("status %d, want 400 (body %s)", rec.Code, rec.Body.String())
|
||||
}
|
||||
if !strings.Contains(rec.Body.String(), tc.want) {
|
||||
t.Fatalf("body %q does not mention %q", rec.Body.String(), tc.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
if entries, err := os.ReadDir(dir); err == nil && len(entries) != 0 {
|
||||
t.Fatalf("rejected requests wrote %d entries to the certs dir", len(entries))
|
||||
}
|
||||
}
|
||||
|
||||
func TestCertUpdateForceAcceptsExpiredCert(t *testing.T) {
|
||||
useTempCertsDir(t)
|
||||
expiredCert, expiredKey := makeTestCertPair(t, "old.example.com", time.Now().Add(-48*time.Hour), time.Now().Add(-time.Hour))
|
||||
rec := postCertUpdate(t, map[string]interface{}{
|
||||
"name": "forced", "fullchain": expiredCert, "privkey": expiredKey, "reload": false, "force": true,
|
||||
})
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("status %d: %s", rec.Code, rec.Body.String())
|
||||
}
|
||||
var resp struct {
|
||||
Cert tlsCertInfo `json:"cert"`
|
||||
Warnings []string `json:"warnings"`
|
||||
}
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
|
||||
t.Fatalf("decode: %v", err)
|
||||
}
|
||||
if !resp.Cert.Expired {
|
||||
t.Fatal("expected the stored certificate to be reported as expired")
|
||||
}
|
||||
if !strings.Contains(strings.Join(resp.Warnings, " | "), "expirado") {
|
||||
t.Fatalf("expected an expiry warning, got %v", resp.Warnings)
|
||||
}
|
||||
}
|
||||
|
||||
// A certificate referenced only by the running config (for example a certbot
|
||||
// path outside the panel directory) must still be updatable in place, because
|
||||
// that is what makes a renewal invisible to the rest of the configuration.
|
||||
func TestCertUpdateAllowsPathReferencedByConfig(t *testing.T) {
|
||||
useTempCertsDir(t)
|
||||
external := t.TempDir()
|
||||
certFile := filepath.Join(external, "fullchain.pem")
|
||||
keyFile := filepath.Join(external, "privkey.pem")
|
||||
oldCert, oldKey := makeTestCertPair(t, "tunnel.example.com", time.Now().Add(-time.Hour), time.Now().Add(24*time.Hour))
|
||||
if err := os.WriteFile(certFile, []byte(oldCert), 0o600); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.WriteFile(keyFile, []byte(oldKey), 0o600); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
setGlobalCfg(&Config{TLSForwarders: []TLSForwarderConfig{{
|
||||
Listen: "0.0.0.0:8443", CertFile: certFile, KeyFile: keyFile,
|
||||
}}})
|
||||
|
||||
newCert, newKey := makeTestCertPair(t, "tunnel.example.com", time.Now().Add(-time.Hour), time.Now().Add(60*24*time.Hour))
|
||||
rec := postCertUpdate(t, map[string]interface{}{
|
||||
"cert_file": certFile, "key_file": keyFile,
|
||||
"fullchain": newCert, "privkey": newKey, "reload": false,
|
||||
})
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("status %d: %s", rec.Code, rec.Body.String())
|
||||
}
|
||||
stored, err := os.ReadFile(certFile)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if strings.TrimSpace(string(stored)) != strings.TrimSpace(newCert) {
|
||||
t.Fatal("external certificate path was not updated")
|
||||
}
|
||||
var resp struct {
|
||||
Cert tlsCertInfo `json:"cert"`
|
||||
}
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
|
||||
t.Fatalf("decode: %v", err)
|
||||
}
|
||||
if len(resp.Cert.UsedBy) != 1 || resp.Cert.UsedBy[0].Ref != "0.0.0.0:8443" {
|
||||
t.Fatalf("expected the TLS forwarder to be reported as consumer, got %+v", resp.Cert.UsedBy)
|
||||
}
|
||||
}
|
||||
|
||||
func TestTLSCertListReportsConfiguredAndManagedCerts(t *testing.T) {
|
||||
dir := useTempCertsDir(t)
|
||||
certPEM, keyPEM := makeTestCertPair(t, "listed.example.com", time.Now().Add(-time.Hour), time.Now().Add(10*24*time.Hour))
|
||||
if err := os.MkdirAll(filepath.Join(dir, "listed"), 0o700); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.WriteFile(filepath.Join(dir, "listed", tlsCertFileName), []byte(certPEM), 0o600); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.WriteFile(filepath.Join(dir, "listed", tlsKeyFileName), []byte(keyPEM), 0o600); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
setGlobalCfg(&Config{})
|
||||
|
||||
req := httptest.NewRequest(http.MethodGet, "/api/tls/certs", nil)
|
||||
rec := httptest.NewRecorder()
|
||||
handleTLSCertList(rec, req)
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("status %d: %s", rec.Code, rec.Body.String())
|
||||
}
|
||||
var resp struct {
|
||||
Certs []tlsCertInfo `json:"certs"`
|
||||
}
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
|
||||
t.Fatalf("decode: %v", err)
|
||||
}
|
||||
if len(resp.Certs) != 1 {
|
||||
t.Fatalf("expected 1 cert, got %d (%+v)", len(resp.Certs), resp.Certs)
|
||||
}
|
||||
got := resp.Certs[0]
|
||||
if got.Name != "listed" || !got.Managed || !got.KeyOK || !got.SelfSigned {
|
||||
t.Fatalf("unexpected cert info: %+v", got)
|
||||
}
|
||||
if !got.Expiring || got.Expired {
|
||||
t.Fatalf("a cert expiring in 10 days should be flagged as expiring: %+v", got)
|
||||
}
|
||||
}
|
||||
@@ -480,6 +480,7 @@ write_sshpanel_systemd_override() {
|
||||
echo "[Service]"
|
||||
echo "Environment=PANEL_LOG_FILE=${INSTALL_DIR}/logs/panel.log"
|
||||
echo "Environment=PANEL_LOG_MAX_BYTES=${PANEL_LOG_MAX_BYTES}"
|
||||
echo "LimitNOFILE=1048576"
|
||||
echo "ExecStartPre="
|
||||
echo "ExecStartPre=${MKDIR_BIN} -p ${INSTALL_DIR}/logs"
|
||||
echo "ExecStartPre=${SH_BIN} -c '${MOUNTPOINT_BIN} -q ${INSTALL_DIR}/logs || ${MOUNT_BIN} -t tmpfs -o size=${LOG_TMPFS_SIZE},mode=0755 tmpfs ${INSTALL_DIR}/logs || true'"
|
||||
|
||||
+173
-47
@@ -18,6 +18,9 @@ type XrayClientMeta struct {
|
||||
OwnerUsername string
|
||||
ExpiresAt *time.Time
|
||||
MaxConns int
|
||||
DataQuotaBytes int64
|
||||
QuotaAction string
|
||||
QuotaThrottleMbps int
|
||||
CreatedAt time.Time
|
||||
TotalUplinkBytes int64
|
||||
TotalDownlinkBytes int64
|
||||
@@ -35,6 +38,9 @@ func (s *Store) EnsureXrayClientsSchema(ctx context.Context) error {
|
||||
owner_username TEXT NOT NULL DEFAULT '',
|
||||
expires_at TIMESTAMPTZ,
|
||||
max_conns INT NOT NULL DEFAULT 0,
|
||||
data_quota_bytes BIGINT NOT NULL DEFAULT 0,
|
||||
quota_action TEXT NOT NULL DEFAULT 'block',
|
||||
quota_throttle_mbps INT NOT NULL DEFAULT 1,
|
||||
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
||||
total_uplink_bytes BIGINT NOT NULL DEFAULT 0,
|
||||
total_downlink_bytes BIGINT NOT NULL DEFAULT 0,
|
||||
@@ -42,6 +48,9 @@ func (s *Store) EnsureXrayClientsSchema(ctx context.Context) error {
|
||||
active_connections INT NOT NULL DEFAULT 0
|
||||
)`,
|
||||
`ALTER TABLE xray_clients ADD COLUMN IF NOT EXISTS owner_username TEXT NOT NULL DEFAULT ''`,
|
||||
`ALTER TABLE xray_clients ADD COLUMN IF NOT EXISTS data_quota_bytes BIGINT NOT NULL DEFAULT 0`,
|
||||
`ALTER TABLE xray_clients ADD COLUMN IF NOT EXISTS quota_action TEXT NOT NULL DEFAULT 'block'`,
|
||||
`ALTER TABLE xray_clients ADD COLUMN IF NOT EXISTS quota_throttle_mbps INT NOT NULL DEFAULT 1`,
|
||||
`ALTER TABLE xray_clients ADD COLUMN IF NOT EXISTS total_uplink_bytes BIGINT NOT NULL DEFAULT 0`,
|
||||
`ALTER TABLE xray_clients ADD COLUMN IF NOT EXISTS total_downlink_bytes BIGINT NOT NULL DEFAULT 0`,
|
||||
`ALTER TABLE xray_clients ADD COLUMN IF NOT EXISTS last_active TIMESTAMPTZ`,
|
||||
@@ -61,16 +70,20 @@ func (s *Store) UpsertXrayClientMeta(ctx context.Context, m XrayClientMeta) erro
|
||||
expiresAt = *m.ExpiresAt
|
||||
}
|
||||
_, err := s.db.ExecContext(ctx, `
|
||||
INSERT INTO xray_clients (uuid, name, email, inbound_tag, owner_username, expires_at, max_conns)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
||||
INSERT INTO xray_clients (uuid, name, email, inbound_tag, owner_username, expires_at, max_conns, data_quota_bytes, quota_action, quota_throttle_mbps)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
|
||||
ON CONFLICT (uuid) DO UPDATE SET
|
||||
name = EXCLUDED.name,
|
||||
email = EXCLUDED.email,
|
||||
inbound_tag = CASE WHEN EXCLUDED.inbound_tag <> '' THEN EXCLUDED.inbound_tag ELSE xray_clients.inbound_tag END,
|
||||
owner_username = CASE WHEN EXCLUDED.owner_username <> '' THEN EXCLUDED.owner_username ELSE xray_clients.owner_username END,
|
||||
expires_at = EXCLUDED.expires_at,
|
||||
max_conns = EXCLUDED.max_conns`,
|
||||
m.UUID, m.Name, m.Email, m.InboundTag, m.OwnerUsername, expiresAt, m.MaxConns)
|
||||
expires_at = EXCLUDED.expires_at,
|
||||
max_conns = EXCLUDED.max_conns,
|
||||
data_quota_bytes = EXCLUDED.data_quota_bytes,
|
||||
quota_action = EXCLUDED.quota_action,
|
||||
quota_throttle_mbps = EXCLUDED.quota_throttle_mbps`,
|
||||
m.UUID, m.Name, m.Email, m.InboundTag, m.OwnerUsername, expiresAt, m.MaxConns,
|
||||
m.DataQuotaBytes, normalizeQuotaAction(m.QuotaAction), quotaThrottleMbpsOrDefault(m.QuotaThrottleMbps))
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -79,10 +92,13 @@ func (s *Store) GetXrayClientMeta(ctx context.Context, uuid string) (*XrayClient
|
||||
var expiresAt sql.NullTime
|
||||
var lastActive sql.NullTime
|
||||
err := s.db.QueryRowContext(ctx, `
|
||||
SELECT uuid, name, email, inbound_tag, COALESCE(owner_username, ''), expires_at, max_conns, created_at,
|
||||
SELECT uuid, name, email, inbound_tag, COALESCE(owner_username, ''), expires_at, max_conns,
|
||||
COALESCE(data_quota_bytes, 0), COALESCE(quota_action, 'block'), COALESCE(quota_throttle_mbps, 1), created_at,
|
||||
COALESCE(total_uplink_bytes, 0), COALESCE(total_downlink_bytes, 0), last_active, COALESCE(active_connections, 0)
|
||||
FROM xray_clients WHERE uuid = $1`, uuid).
|
||||
Scan(&m.UUID, &m.Name, &m.Email, &m.InboundTag, &m.OwnerUsername, &expiresAt, &m.MaxConns, &m.CreatedAt, &m.TotalUplinkBytes, &m.TotalDownlinkBytes, &lastActive, &m.ActiveConnections)
|
||||
Scan(&m.UUID, &m.Name, &m.Email, &m.InboundTag, &m.OwnerUsername, &expiresAt, &m.MaxConns,
|
||||
&m.DataQuotaBytes, &m.QuotaAction, &m.QuotaThrottleMbps, &m.CreatedAt,
|
||||
&m.TotalUplinkBytes, &m.TotalDownlinkBytes, &lastActive, &m.ActiveConnections)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@@ -96,13 +112,22 @@ func (s *Store) GetXrayClientMeta(ctx context.Context, uuid string) (*XrayClient
|
||||
}
|
||||
|
||||
func (s *Store) DeleteXrayClientMeta(ctx context.Context, uuid string) error {
|
||||
// Serialize deletion with the native stats flusher. Otherwise a batch that
|
||||
// was swapped out just before DELETE could finish afterward and, if the same
|
||||
// UUID is recreated quickly, apply stale traffic/active deltas to the new row.
|
||||
xrayMgr.nativeTrafficPersistMu.Lock()
|
||||
defer xrayMgr.nativeTrafficPersistMu.Unlock()
|
||||
_, err := s.db.ExecContext(ctx, `DELETE FROM xray_clients WHERE uuid = $1`, uuid)
|
||||
if err == nil {
|
||||
xrayMgr.removeNativeQuotaPolicy(uuid)
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *Store) ListAllXrayClients(ctx context.Context) ([]*XrayClientMeta, error) {
|
||||
rows, err := s.db.QueryContext(ctx, `
|
||||
SELECT uuid, name, email, inbound_tag, COALESCE(owner_username, ''), expires_at, max_conns, created_at,
|
||||
SELECT uuid, name, email, inbound_tag, COALESCE(owner_username, ''), expires_at, max_conns,
|
||||
COALESCE(data_quota_bytes, 0), COALESCE(quota_action, 'block'), COALESCE(quota_throttle_mbps, 1), created_at,
|
||||
COALESCE(total_uplink_bytes, 0), COALESCE(total_downlink_bytes, 0), last_active, COALESCE(active_connections, 0)
|
||||
FROM xray_clients ORDER BY created_at DESC`)
|
||||
if err != nil {
|
||||
@@ -114,7 +139,8 @@ func (s *Store) ListAllXrayClients(ctx context.Context) ([]*XrayClientMeta, erro
|
||||
|
||||
func (s *Store) ListXrayClientsByOwner(ctx context.Context, ownerUsername string) ([]*XrayClientMeta, error) {
|
||||
rows, err := s.db.QueryContext(ctx, `
|
||||
SELECT uuid, name, email, inbound_tag, COALESCE(owner_username, ''), expires_at, max_conns, created_at,
|
||||
SELECT uuid, name, email, inbound_tag, COALESCE(owner_username, ''), expires_at, max_conns,
|
||||
COALESCE(data_quota_bytes, 0), COALESCE(quota_action, 'block'), COALESCE(quota_throttle_mbps, 1), created_at,
|
||||
COALESCE(total_uplink_bytes, 0), COALESCE(total_downlink_bytes, 0), last_active, COALESCE(active_connections, 0)
|
||||
FROM xray_clients WHERE owner_username = $1 ORDER BY created_at DESC`, ownerUsername)
|
||||
if err != nil {
|
||||
@@ -132,7 +158,8 @@ func (s *Store) CountXrayClientsByOwner(ctx context.Context, ownerUsername strin
|
||||
|
||||
func (s *Store) ListExpiredXrayClients(ctx context.Context) ([]*XrayClientMeta, error) {
|
||||
rows, err := s.db.QueryContext(ctx, `
|
||||
SELECT uuid, name, email, inbound_tag, COALESCE(owner_username, ''), expires_at, max_conns, created_at,
|
||||
SELECT uuid, name, email, inbound_tag, COALESCE(owner_username, ''), expires_at, max_conns,
|
||||
COALESCE(data_quota_bytes, 0), COALESCE(quota_action, 'block'), COALESCE(quota_throttle_mbps, 1), created_at,
|
||||
COALESCE(total_uplink_bytes, 0), COALESCE(total_downlink_bytes, 0), last_active, COALESCE(active_connections, 0)
|
||||
FROM xray_clients WHERE expires_at IS NOT NULL AND expires_at <= NOW()`)
|
||||
if err != nil {
|
||||
@@ -148,7 +175,9 @@ func scanXrayClientMetaRows(rows *sql.Rows) ([]*XrayClientMeta, error) {
|
||||
m := &XrayClientMeta{}
|
||||
var expiresAt sql.NullTime
|
||||
var lastActive sql.NullTime
|
||||
if err := rows.Scan(&m.UUID, &m.Name, &m.Email, &m.InboundTag, &m.OwnerUsername, &expiresAt, &m.MaxConns, &m.CreatedAt, &m.TotalUplinkBytes, &m.TotalDownlinkBytes, &lastActive, &m.ActiveConnections); err != nil {
|
||||
if err := rows.Scan(&m.UUID, &m.Name, &m.Email, &m.InboundTag, &m.OwnerUsername, &expiresAt, &m.MaxConns,
|
||||
&m.DataQuotaBytes, &m.QuotaAction, &m.QuotaThrottleMbps, &m.CreatedAt,
|
||||
&m.TotalUplinkBytes, &m.TotalDownlinkBytes, &lastActive, &m.ActiveConnections); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if expiresAt.Valid {
|
||||
@@ -204,19 +233,38 @@ func (s *Store) AddXrayClientTrafficBatch(ctx context.Context, deltas map[string
|
||||
return tx.Commit()
|
||||
}
|
||||
|
||||
// UpdateXrayClientActive adjusts the native online connection counter.
|
||||
func (s *Store) UpdateXrayClientActive(ctx context.Context, uuid, email string, delta int) error {
|
||||
if uuid == "" || delta == 0 {
|
||||
// AddXrayClientActiveBatch persists native online-counter deltas without
|
||||
// launching a database goroutine/query for every connect and disconnect.
|
||||
func (s *Store) AddXrayClientActiveBatch(ctx context.Context, deltas map[string]xrayPendingActive) error {
|
||||
if len(deltas) == 0 {
|
||||
return nil
|
||||
}
|
||||
_, err := s.db.ExecContext(ctx, `
|
||||
tx, err := s.db.BeginTx(ctx, nil)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
stmt, err := tx.PrepareContext(ctx, `
|
||||
UPDATE xray_clients SET
|
||||
email = CASE WHEN email = '' AND $2 <> '' THEN $2 ELSE email END,
|
||||
name = CASE WHEN name = '' AND $2 <> '' THEN $2 ELSE name END,
|
||||
last_active = CASE WHEN $3::INT > 0 THEN NOW() ELSE last_active END,
|
||||
last_active = CASE WHEN $4::BOOLEAN THEN NOW() ELSE last_active END,
|
||||
active_connections = GREATEST(active_connections + $3::INT, 0)
|
||||
WHERE uuid = $1`, uuid, email, delta)
|
||||
return err
|
||||
WHERE uuid = $1`)
|
||||
if err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
defer stmt.Close()
|
||||
for uuid, d := range deltas {
|
||||
if uuid == "" || (d.Delta == 0 && !d.Connected) {
|
||||
continue
|
||||
}
|
||||
if _, err := stmt.ExecContext(ctx, uuid, d.Email, d.Delta, d.Connected); err != nil {
|
||||
_ = tx.Rollback()
|
||||
return err
|
||||
}
|
||||
}
|
||||
return tx.Commit()
|
||||
}
|
||||
|
||||
func countOwnedXrayClients(ctx context.Context, store *Store, ownerUsername string) int {
|
||||
@@ -262,6 +310,66 @@ func removeOwnerXrayClients(ctx context.Context, store *Store, ownerUsername str
|
||||
}
|
||||
}
|
||||
|
||||
// suspendOwnerXrayClients revokes transport access while preserving metadata,
|
||||
// expiry, traffic and quota. This makes reseller suspension/renewal reversible.
|
||||
func suspendOwnerXrayClients(ctx context.Context, store *Store, ownerUsername string) {
|
||||
if store == nil || ownerUsername == "" {
|
||||
return
|
||||
}
|
||||
clients, err := store.ListXrayClientsByOwner(ctx, ownerUsername)
|
||||
if err != nil {
|
||||
log.Printf("xray owner suspension: list %s: %v", ownerUsername, err)
|
||||
return
|
||||
}
|
||||
changed := false
|
||||
for _, m := range clients {
|
||||
xrayMgr.disconnectNativeClient(m.UUID)
|
||||
if m.InboundTag == "" {
|
||||
continue
|
||||
}
|
||||
if err := xrayMgr.RemoveXrayClient(m.InboundTag, m.UUID); err != nil {
|
||||
log.Printf("xray owner suspension: remove %s from %s: %v", m.UUID, m.InboundTag, err)
|
||||
continue
|
||||
}
|
||||
changed = true
|
||||
}
|
||||
if changed {
|
||||
xrayMgr.restartIfExternalRunning()
|
||||
}
|
||||
}
|
||||
|
||||
// restoreOwnerXrayClients reactivates non-expired clients after reseller renewal.
|
||||
func restoreOwnerXrayClients(ctx context.Context, store *Store, ownerUsername string) {
|
||||
if store == nil || ownerUsername == "" {
|
||||
return
|
||||
}
|
||||
clients, err := store.ListXrayClientsByOwner(ctx, ownerUsername)
|
||||
if err != nil {
|
||||
log.Printf("xray owner restore: list %s: %v", ownerUsername, err)
|
||||
return
|
||||
}
|
||||
now := time.Now()
|
||||
changed := false
|
||||
for _, m := range clients {
|
||||
if m.InboundTag == "" || (m.ExpiresAt != nil && !m.ExpiresAt.After(now)) {
|
||||
continue
|
||||
}
|
||||
email := m.Email
|
||||
if email == "" {
|
||||
email = m.UUID
|
||||
}
|
||||
if err := xrayMgr.EnsureXrayClient(m.InboundTag, m.UUID, email); err != nil {
|
||||
log.Printf("xray owner restore: add %s to %s: %v", m.UUID, m.InboundTag, err)
|
||||
continue
|
||||
}
|
||||
xrayMgr.setNativeQuotaPolicy(m)
|
||||
changed = true
|
||||
}
|
||||
if changed {
|
||||
xrayMgr.restartIfExternalRunning()
|
||||
}
|
||||
}
|
||||
|
||||
// startXrayClientExpiryChecker runs a background goroutine that removes expired
|
||||
// Xray clients from both the config file and the database every 5 minutes.
|
||||
func startXrayClientExpiryChecker(store *Store) {
|
||||
@@ -272,35 +380,53 @@ func startXrayClientExpiryChecker(store *Store) {
|
||||
ticker := time.NewTicker(5 * time.Minute)
|
||||
defer ticker.Stop()
|
||||
for range ticker.C {
|
||||
ctx := context.Background()
|
||||
expired, err := store.ListExpiredXrayClients(ctx)
|
||||
if err != nil {
|
||||
log.Printf("xray expiry checker: list error: %v", err)
|
||||
continue
|
||||
}
|
||||
if len(expired) == 0 {
|
||||
continue
|
||||
}
|
||||
needRestart := false
|
||||
for _, m := range expired {
|
||||
tag := m.InboundTag
|
||||
if tag == "" {
|
||||
_ = store.DeleteXrayClientMeta(ctx, m.UUID)
|
||||
continue
|
||||
}
|
||||
if err := xrayMgr.RemoveXrayClient(tag, m.UUID); err != nil {
|
||||
log.Printf("xray expiry: remove %s from %s: %v", m.UUID, tag, err)
|
||||
} else {
|
||||
needRestart = true
|
||||
}
|
||||
if err := store.DeleteXrayClientMeta(ctx, m.UUID); err != nil {
|
||||
log.Printf("xray expiry: delete meta %s: %v", m.UUID, err)
|
||||
}
|
||||
log.Printf("xray expiry: removed expired client %q (%s) from inbound %s", m.Name, m.UUID, tag)
|
||||
}
|
||||
if needRestart {
|
||||
xrayMgr.restartIfExternalRunning()
|
||||
}
|
||||
expireXrayClientsOnce(store)
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
func expireXrayClientsOnce(store *Store) {
|
||||
if store == nil {
|
||||
return
|
||||
}
|
||||
ctx := context.Background()
|
||||
expired, err := store.ListExpiredXrayClients(ctx)
|
||||
if err != nil {
|
||||
log.Printf("xray expiry checker: list error: %v", err)
|
||||
return
|
||||
}
|
||||
needRestart := false
|
||||
for _, m := range expired {
|
||||
tag := m.InboundTag
|
||||
xrayMgr.disconnectNativeClient(m.UUID)
|
||||
if tag != "" {
|
||||
if err := xrayMgr.RemoveXrayClient(tag, m.UUID); err != nil {
|
||||
log.Printf("xray expiry: remove %s from %s: %v", m.UUID, tag, err)
|
||||
} else {
|
||||
needRestart = true
|
||||
}
|
||||
}
|
||||
if err := store.DeleteXrayClientMeta(ctx, m.UUID); err != nil {
|
||||
log.Printf("xray expiry: delete meta %s: %v", m.UUID, err)
|
||||
}
|
||||
log.Printf("xray expiry: removed expired client %q (%s) from inbound %s", m.Name, m.UUID, tag)
|
||||
}
|
||||
if needRestart {
|
||||
xrayMgr.restartIfExternalRunning()
|
||||
}
|
||||
}
|
||||
|
||||
// ResetXrayClientTraffic clears a client's persistent usage without removing
|
||||
// the account or changing its expiry/quota policy.
|
||||
func (s *Store) ResetXrayClientTraffic(ctx context.Context, uuid string) error {
|
||||
if s == nil || uuid == "" {
|
||||
return nil
|
||||
}
|
||||
_, err := s.db.ExecContext(ctx, `
|
||||
UPDATE xray_clients SET
|
||||
total_uplink_bytes = 0,
|
||||
total_downlink_bytes = 0,
|
||||
last_active = NULL
|
||||
WHERE uuid = $1`, uuid)
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -64,7 +64,8 @@ func (s *Store) UpsertXrayConfig(ctx context.Context, configKey string, data []b
|
||||
|
||||
func (s *Store) ListXrayClientsByInbound(ctx context.Context, inboundTag string) ([]*XrayClientMeta, error) {
|
||||
rows, err := s.db.QueryContext(ctx, `
|
||||
SELECT uuid, name, email, inbound_tag, COALESCE(owner_username, ''), expires_at, max_conns, created_at,
|
||||
SELECT uuid, name, email, inbound_tag, COALESCE(owner_username, ''), expires_at, max_conns,
|
||||
COALESCE(data_quota_bytes, 0), COALESCE(quota_action, 'block'), COALESCE(quota_throttle_mbps, 1), created_at,
|
||||
COALESCE(total_uplink_bytes, 0), COALESCE(total_downlink_bytes, 0), last_active, COALESCE(active_connections, 0)
|
||||
FROM xray_clients WHERE inbound_tag = $1 ORDER BY created_at DESC`, inboundTag)
|
||||
if err != nil {
|
||||
|
||||
+502
-122
@@ -252,15 +252,21 @@ type XrayManager struct {
|
||||
startTime time.Time
|
||||
lastErr string
|
||||
|
||||
statsMu sync.RWMutex
|
||||
statsByEmail map[string]xrayRuntimeStat
|
||||
lastStatsErr string
|
||||
lastStatsPoll time.Time
|
||||
pollStarted bool
|
||||
statsMu sync.RWMutex
|
||||
statsByEmail map[string]xrayRuntimeStat
|
||||
lastStatsErr string
|
||||
lastStatsPoll time.Time
|
||||
pollStarted bool
|
||||
rateSamplerStarted bool
|
||||
|
||||
nativeDBMu sync.Mutex
|
||||
nativeTrafficPersistMu sync.Mutex
|
||||
nativeTrafficPending map[string]xrayPendingTraffic
|
||||
nativeActivePending map[string]xrayPendingActive
|
||||
nativeStatsFlushStarted bool
|
||||
|
||||
nativeQuotaMu sync.RWMutex
|
||||
nativeQuotaByUUID map[string]*xrayNativeQuotaState
|
||||
}
|
||||
|
||||
type xrayTrafficCounters struct {
|
||||
@@ -280,6 +286,14 @@ type xrayPendingTraffic struct {
|
||||
Email string
|
||||
Uplink int64
|
||||
Downlink int64
|
||||
State *xrayNativeQuotaState
|
||||
}
|
||||
|
||||
type xrayPendingActive struct {
|
||||
Email string
|
||||
Delta int
|
||||
Connected bool
|
||||
State *xrayNativeQuotaState
|
||||
}
|
||||
|
||||
var xrayMgr = &XrayManager{}
|
||||
@@ -296,10 +310,17 @@ func initXrayManager(cfg *XrayConfig) {
|
||||
}
|
||||
xrayMgr.mu.Unlock()
|
||||
|
||||
// Reconcile already-expired rows before the runtime loads DB-backed clients.
|
||||
// The periodic checker intentionally sleeps between passes, so doing one pass
|
||||
// here closes the startup window in which an expired UUID could reconnect.
|
||||
expireXrayClientsOnce(statsStore)
|
||||
xrayMgr.reloadNativeQuotaPolicies()
|
||||
|
||||
// In native mode the in-process emulator records traffic directly, so the
|
||||
// external `xray api statsquery` poller is not started (it would overwrite
|
||||
// the native counters with errors from a non-existent CLI endpoint).
|
||||
xrayMgr.startNativeStatsFlusher()
|
||||
xrayMgr.startRateSampler()
|
||||
if !cfg.UseNative() {
|
||||
xrayMgr.startStatsPoller()
|
||||
}
|
||||
@@ -449,7 +470,7 @@ func (m *XrayManager) Restart() error {
|
||||
// recordNativeConnect marks a native client stream as online immediately. This
|
||||
// is more accurate than external Xray's Stats API polling because it knows when
|
||||
// the decoded VMess/VLESS stream is authenticated and opened.
|
||||
func (m *XrayManager) recordNativeConnect(uuid, email string) {
|
||||
func (m *XrayManager) recordNativeConnect(uuid, email string, state *xrayNativeQuotaState) {
|
||||
uuid = strings.TrimSpace(uuid)
|
||||
email = strings.TrimSpace(email)
|
||||
if email == "" {
|
||||
@@ -463,25 +484,18 @@ func (m *XrayManager) recordNativeConnect(uuid, email string) {
|
||||
if m.statsByEmail == nil {
|
||||
m.statsByEmail = make(map[string]xrayRuntimeStat)
|
||||
}
|
||||
st := m.statsByEmail[email]
|
||||
key := firstNonEmpty(uuid, email)
|
||||
st := m.statsByEmail[key]
|
||||
st.Email = email
|
||||
st.LastActive = now
|
||||
st.ActiveConnections++
|
||||
m.statsByEmail[email] = st
|
||||
m.statsByEmail[key] = st
|
||||
m.statsMu.Unlock()
|
||||
|
||||
if statsStore != nil && uuid != "" {
|
||||
xrayGo("native xray stats active increment", func() {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||||
defer cancel()
|
||||
if err := statsStore.UpdateXrayClientActive(ctx, uuid, email, 1); err != nil {
|
||||
xrayLogf("xray native stats: active +1 for %s failed: %v", uuid, err)
|
||||
}
|
||||
})
|
||||
}
|
||||
m.queueNativeActiveDelta(uuid, email, 1, true, state)
|
||||
}
|
||||
|
||||
func (m *XrayManager) recordNativeDisconnect(uuid, email string) {
|
||||
func (m *XrayManager) recordNativeDisconnect(uuid, email string, state *xrayNativeQuotaState) {
|
||||
uuid = strings.TrimSpace(uuid)
|
||||
email = strings.TrimSpace(email)
|
||||
if email == "" {
|
||||
@@ -492,29 +506,53 @@ func (m *XrayManager) recordNativeDisconnect(uuid, email string) {
|
||||
}
|
||||
m.statsMu.Lock()
|
||||
if m.statsByEmail != nil {
|
||||
st := m.statsByEmail[email]
|
||||
key := firstNonEmpty(uuid, email)
|
||||
st := m.statsByEmail[key]
|
||||
if st.ActiveConnections > 0 {
|
||||
st.ActiveConnections--
|
||||
}
|
||||
m.statsByEmail[email] = st
|
||||
m.statsByEmail[key] = st
|
||||
}
|
||||
m.statsMu.Unlock()
|
||||
|
||||
if statsStore != nil && uuid != "" {
|
||||
xrayGo("native xray stats active decrement", func() {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||||
defer cancel()
|
||||
if err := statsStore.UpdateXrayClientActive(ctx, uuid, email, -1); err != nil {
|
||||
xrayLogf("xray native stats: active -1 for %s failed: %v", uuid, err)
|
||||
}
|
||||
})
|
||||
m.queueNativeActiveDelta(uuid, email, -1, false, state)
|
||||
}
|
||||
|
||||
func (m *XrayManager) queueNativeActiveDelta(uuid, email string, delta int, connected bool, state *xrayNativeQuotaState) {
|
||||
if statsStore == nil || uuid == "" || delta == 0 || state == nil {
|
||||
return
|
||||
}
|
||||
// Keep the policy identity stable until the delta is queued. A UUID can be
|
||||
// deleted and later recreated; an old connection must never decrement or add
|
||||
// traffic to the replacement account merely because the string key matches.
|
||||
m.nativeQuotaMu.RLock()
|
||||
if m.nativeQuotaByUUID[uuid] != state {
|
||||
m.nativeQuotaMu.RUnlock()
|
||||
return
|
||||
}
|
||||
m.nativeDBMu.Lock()
|
||||
if m.nativeActivePending == nil {
|
||||
m.nativeActivePending = make(map[string]xrayPendingActive)
|
||||
}
|
||||
p := m.nativeActivePending[uuid]
|
||||
if p.State != nil && p.State != state {
|
||||
p = xrayPendingActive{}
|
||||
}
|
||||
if p.Email == "" {
|
||||
p.Email = email
|
||||
}
|
||||
p.Delta += delta
|
||||
p.Connected = p.Connected || connected
|
||||
p.State = state
|
||||
m.nativeActivePending[uuid] = p
|
||||
m.nativeDBMu.Unlock()
|
||||
m.nativeQuotaMu.RUnlock()
|
||||
}
|
||||
|
||||
// recordNativeTraffic accumulates in-process byte counters for a client and
|
||||
// queues DB persistence. Used by the native emulator instead of external
|
||||
// `xray api statsquery` polling.
|
||||
func (m *XrayManager) recordNativeTraffic(uuid, email string, up, down int64) {
|
||||
func (m *XrayManager) recordNativeTraffic(uuid, email string, up, down int64, generation uint64, state *xrayNativeQuotaState) {
|
||||
uuid = strings.TrimSpace(uuid)
|
||||
email = strings.TrimSpace(email)
|
||||
if email == "" {
|
||||
@@ -523,31 +561,57 @@ func (m *XrayManager) recordNativeTraffic(uuid, email string, up, down int64) {
|
||||
if email == "" || (up == 0 && down == 0) {
|
||||
return
|
||||
}
|
||||
now := time.Now()
|
||||
m.statsMu.Lock()
|
||||
if m.statsByEmail == nil {
|
||||
m.statsByEmail = make(map[string]xrayRuntimeStat)
|
||||
if state != nil {
|
||||
// Keep generation validation and queuing in the same critical section as
|
||||
// resetNativeTrafficAccounting. Otherwise an old meter can validate just
|
||||
// before a reset and enqueue its bytes immediately after the DB was zeroed.
|
||||
state.mu.Lock()
|
||||
defer state.mu.Unlock()
|
||||
if generation != state.generation {
|
||||
return
|
||||
}
|
||||
}
|
||||
st := m.statsByEmail[email]
|
||||
st.Email = email
|
||||
st.Uplink += up
|
||||
st.Downlink += down
|
||||
st.LastActive = now
|
||||
m.statsByEmail[email] = st
|
||||
m.statsMu.Unlock()
|
||||
|
||||
if statsStore != nil && uuid != "" {
|
||||
// Only DB-backed clients have a native policy state. Config-only clients are
|
||||
// still shown in runtime stats, but queuing UPDATEs for rows that do not exist
|
||||
// can make the retry map grow during a database outage.
|
||||
if statsStore != nil && uuid != "" && state != nil {
|
||||
m.nativeQuotaMu.RLock()
|
||||
if m.nativeQuotaByUUID[uuid] != state {
|
||||
m.nativeQuotaMu.RUnlock()
|
||||
return
|
||||
}
|
||||
m.nativeDBMu.Lock()
|
||||
if m.nativeTrafficPending == nil {
|
||||
m.nativeTrafficPending = make(map[string]xrayPendingTraffic)
|
||||
}
|
||||
p := m.nativeTrafficPending[uuid]
|
||||
if p.State != nil && p.State != state {
|
||||
p = xrayPendingTraffic{}
|
||||
}
|
||||
p.Email = email
|
||||
p.Uplink += up
|
||||
p.Downlink += down
|
||||
p.State = state
|
||||
m.nativeTrafficPending[uuid] = p
|
||||
m.nativeDBMu.Unlock()
|
||||
m.nativeQuotaMu.RUnlock()
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
m.statsMu.Lock()
|
||||
if m.statsByEmail == nil {
|
||||
m.statsByEmail = make(map[string]xrayRuntimeStat)
|
||||
}
|
||||
key := firstNonEmpty(uuid, email)
|
||||
st := m.statsByEmail[key]
|
||||
st.Email = email
|
||||
st.Uplink += up
|
||||
st.Downlink += down
|
||||
st.LastActive = now
|
||||
m.statsByEmail[key] = st
|
||||
m.statsMu.Unlock()
|
||||
|
||||
}
|
||||
|
||||
func (m *XrayManager) startNativeStatsFlusher() {
|
||||
@@ -594,35 +658,108 @@ func (m *XrayManager) flushNativeStatsToDB() {
|
||||
if statsStore == nil {
|
||||
return
|
||||
}
|
||||
m.nativeTrafficPersistMu.Lock()
|
||||
defer m.nativeTrafficPersistMu.Unlock()
|
||||
persistent := m.nativePersistentStates()
|
||||
m.nativeDBMu.Lock()
|
||||
pending := m.nativeTrafficPending
|
||||
for uuid, pending := range m.nativeTrafficPending {
|
||||
if persistent[uuid] != pending.State {
|
||||
delete(m.nativeTrafficPending, uuid)
|
||||
}
|
||||
}
|
||||
for uuid, pending := range m.nativeActivePending {
|
||||
if persistent[uuid] != pending.State {
|
||||
delete(m.nativeActivePending, uuid)
|
||||
}
|
||||
}
|
||||
pendingTraffic := m.nativeTrafficPending
|
||||
pendingActive := m.nativeActivePending
|
||||
m.nativeTrafficPending = nil
|
||||
m.nativeActivePending = nil
|
||||
m.nativeDBMu.Unlock()
|
||||
if len(pending) == 0 {
|
||||
if len(pendingTraffic) == 0 && len(pendingActive) == 0 {
|
||||
return
|
||||
}
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
if err := statsStore.AddXrayClientTrafficBatch(ctx, pending); err != nil {
|
||||
xrayLogf("xray native stats: db traffic flush failed: %v", err)
|
||||
// Put deltas back so a transient DB failure does not lose accounting.
|
||||
|
||||
var trafficErr error
|
||||
if len(pendingTraffic) > 0 {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
trafficErr = statsStore.AddXrayClientTrafficBatch(ctx, pendingTraffic)
|
||||
cancel()
|
||||
}
|
||||
var activeErr error
|
||||
if len(pendingActive) > 0 {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
activeErr = statsStore.AddXrayClientActiveBatch(ctx, pendingActive)
|
||||
cancel()
|
||||
}
|
||||
|
||||
if trafficErr != nil {
|
||||
xrayLogf("xray native stats: db traffic flush failed: %v", trafficErr)
|
||||
}
|
||||
if activeErr != nil {
|
||||
xrayLogf("xray native stats: db active flush failed: %v", activeErr)
|
||||
}
|
||||
if trafficErr != nil || activeErr != nil {
|
||||
// Put only failed batches back so a successful write is never duplicated.
|
||||
persistent = m.nativePersistentStates()
|
||||
m.nativeDBMu.Lock()
|
||||
if m.nativeTrafficPending == nil {
|
||||
if trafficErr != nil && m.nativeTrafficPending == nil {
|
||||
m.nativeTrafficPending = make(map[string]xrayPendingTraffic)
|
||||
}
|
||||
for uuid, d := range pending {
|
||||
p := m.nativeTrafficPending[uuid]
|
||||
if p.Email == "" {
|
||||
p.Email = d.Email
|
||||
if trafficErr != nil {
|
||||
for uuid, d := range pendingTraffic {
|
||||
if persistent[uuid] != d.State {
|
||||
continue
|
||||
}
|
||||
p := m.nativeTrafficPending[uuid]
|
||||
if p.State != nil && p.State != d.State {
|
||||
p = xrayPendingTraffic{}
|
||||
}
|
||||
if p.Email == "" {
|
||||
p.Email = d.Email
|
||||
}
|
||||
p.Uplink += d.Uplink
|
||||
p.Downlink += d.Downlink
|
||||
p.State = d.State
|
||||
m.nativeTrafficPending[uuid] = p
|
||||
}
|
||||
}
|
||||
if activeErr != nil && m.nativeActivePending == nil {
|
||||
m.nativeActivePending = make(map[string]xrayPendingActive)
|
||||
}
|
||||
if activeErr != nil {
|
||||
for uuid, d := range pendingActive {
|
||||
if persistent[uuid] != d.State {
|
||||
continue
|
||||
}
|
||||
p := m.nativeActivePending[uuid]
|
||||
if p.State != nil && p.State != d.State {
|
||||
p = xrayPendingActive{}
|
||||
}
|
||||
if p.Email == "" {
|
||||
p.Email = d.Email
|
||||
}
|
||||
p.Delta += d.Delta
|
||||
p.Connected = p.Connected || d.Connected
|
||||
p.State = d.State
|
||||
m.nativeActivePending[uuid] = p
|
||||
}
|
||||
p.Uplink += d.Uplink
|
||||
p.Downlink += d.Downlink
|
||||
m.nativeTrafficPending[uuid] = p
|
||||
}
|
||||
m.nativeDBMu.Unlock()
|
||||
}
|
||||
}
|
||||
|
||||
func (m *XrayManager) nativePersistentStates() map[string]*xrayNativeQuotaState {
|
||||
m.nativeQuotaMu.RLock()
|
||||
out := make(map[string]*xrayNativeQuotaState, len(m.nativeQuotaByUUID))
|
||||
for uuid, state := range m.nativeQuotaByUUID {
|
||||
out[uuid] = state
|
||||
}
|
||||
m.nativeQuotaMu.RUnlock()
|
||||
return out
|
||||
}
|
||||
|
||||
// XrayStatusDTO is returned by /api/xray/status.
|
||||
type XrayStatusDTO struct {
|
||||
Enabled bool `json:"enabled"`
|
||||
@@ -751,6 +888,73 @@ func (m *XrayManager) startStatsPoller() {
|
||||
}()
|
||||
}
|
||||
|
||||
// Live per-client speed, derived from the same cumulative counters the panel
|
||||
// already reports as lifetime traffic.
|
||||
var xrayBandwidth = newBandwidthSampler(6*time.Second, 45*time.Second)
|
||||
|
||||
const xrayNativeRateSampleInterval = 2 * time.Second
|
||||
|
||||
// startRateSampler keeps xrayBandwidth fresh in native mode, where the
|
||||
// in-process runtime updates the counters continuously. In external mode the
|
||||
// counters only move once per stats poll (15s by default), so refreshRuntimeStats
|
||||
// feeds the sampler at its own cadence instead — sampling faster than the source
|
||||
// updates would show alternating spikes and zeros. The mode is re-checked on
|
||||
// every tick because a hot reload can switch it while running.
|
||||
func (m *XrayManager) startRateSampler() {
|
||||
m.mu.Lock()
|
||||
if m.rateSamplerStarted {
|
||||
m.mu.Unlock()
|
||||
return
|
||||
}
|
||||
m.rateSamplerStarted = true
|
||||
m.mu.Unlock()
|
||||
|
||||
go func() {
|
||||
ticker := time.NewTicker(xrayNativeRateSampleInterval)
|
||||
defer ticker.Stop()
|
||||
for range ticker.C {
|
||||
if !m.usesNativeSnapshot() {
|
||||
continue
|
||||
}
|
||||
m.sampleRuntimeRates(time.Now())
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
func (m *XrayManager) usesNativeSnapshot() bool {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
return m.cfg != nil && m.cfg.UseNative()
|
||||
}
|
||||
|
||||
func (m *XrayManager) sampleRuntimeRates(now time.Time) {
|
||||
type counterSnapshot struct {
|
||||
key string
|
||||
uplink int64
|
||||
downlink int64
|
||||
}
|
||||
m.statsMu.RLock()
|
||||
snapshots := make([]counterSnapshot, 0, len(m.statsByEmail))
|
||||
for key, st := range m.statsByEmail {
|
||||
snapshots = append(snapshots, counterSnapshot{key: key, uplink: st.Uplink, downlink: st.Downlink})
|
||||
}
|
||||
m.statsMu.RUnlock()
|
||||
|
||||
active := make(map[string]struct{}, len(snapshots))
|
||||
for _, snapshot := range snapshots {
|
||||
xrayBandwidth.Observe(snapshot.key, snapshot.uplink, snapshot.downlink, now)
|
||||
active[snapshot.key] = struct{}{}
|
||||
}
|
||||
xrayBandwidth.Retain(active)
|
||||
}
|
||||
|
||||
// RuntimeRateForKeys resolves a client's live speed. Clients are tracked under
|
||||
// their UUID in native mode and under their stats-API email in external mode,
|
||||
// so callers pass every identifier the client may be stored under.
|
||||
func (m *XrayManager) RuntimeRateForKeys(keys ...string) (bandwidthRate, bool) {
|
||||
return xrayBandwidth.RateForKeys(keys...)
|
||||
}
|
||||
|
||||
func (m *XrayManager) isRunningSnapshot() bool {
|
||||
m.mu.Lock()
|
||||
native := m.cfg != nil && m.cfg.UseNative()
|
||||
@@ -830,9 +1034,12 @@ func (m *XrayManager) refreshRuntimeStats() {
|
||||
if m.statsByEmail == nil {
|
||||
m.statsByEmail = make(map[string]xrayRuntimeStat, len(traffic))
|
||||
}
|
||||
seen := make(map[string]bool, len(traffic))
|
||||
// External mode: the counters only move once per poll, so this is also the
|
||||
// natural cadence for the live speed sampler.
|
||||
active := make(map[string]struct{}, len(traffic))
|
||||
for email, counters := range traffic {
|
||||
seen[email] = true
|
||||
active[email] = struct{}{}
|
||||
xrayBandwidth.Observe(email, counters.Uplink, counters.Downlink, now)
|
||||
prev := m.statsByEmail[email]
|
||||
st := xrayRuntimeStat{Email: email, Uplink: counters.Uplink, Downlink: counters.Downlink, LastActive: prev.LastActive, ActiveConnections: prev.ActiveConnections}
|
||||
changed := counters.Uplink != prev.Uplink || counters.Downlink != prev.Downlink
|
||||
@@ -844,9 +1051,10 @@ func (m *XrayManager) refreshRuntimeStats() {
|
||||
}
|
||||
m.statsByEmail[email] = st
|
||||
}
|
||||
// Keep old entries, but do not delete them immediately. Xray may omit zero
|
||||
// counters for users that have not moved traffic yet.
|
||||
_ = seen
|
||||
// Keep old stat entries, but do not delete them immediately: Xray may omit
|
||||
// zero counters for users that have not moved traffic yet. Speed samples are
|
||||
// dropped for absent users because a missing baseline only costs one poll.
|
||||
xrayBandwidth.Retain(active)
|
||||
}
|
||||
|
||||
func (m *XrayManager) refreshRuntimeStatsIfStale(maxAge time.Duration) {
|
||||
@@ -2026,13 +2234,21 @@ type XrayClientInfo struct {
|
||||
DownlinkBytes int64 `json:"downlink_bytes,omitempty"`
|
||||
TotalBytes int64 `json:"total_bytes,omitempty"`
|
||||
ActiveConnections int `json:"active_connections,omitempty"`
|
||||
// Live speed in bytes per second for the whole client, summed across every
|
||||
// connection it has open.
|
||||
UpBytesPerSec float64 `json:"up_bytes_per_sec"`
|
||||
DownBytesPerSec float64 `json:"down_bytes_per_sec"`
|
||||
// Metadata from PostgreSQL (enriched by handleXrayInbounds)
|
||||
Name string `json:"name,omitempty"`
|
||||
ExpiresAt *time.Time `json:"expires_at,omitempty"`
|
||||
ExpirationDays int `json:"expiration_days"`
|
||||
MaxConns int `json:"max_conns"`
|
||||
OwnerUsername string `json:"owner_username,omitempty"`
|
||||
Expired bool `json:"expired,omitempty"`
|
||||
Name string `json:"name,omitempty"`
|
||||
ExpiresAt *time.Time `json:"expires_at,omitempty"`
|
||||
ExpirationDays int `json:"expiration_days"`
|
||||
MaxConns int `json:"max_conns"`
|
||||
DataQuotaBytes int64 `json:"data_quota_bytes"`
|
||||
QuotaAction string `json:"quota_action"`
|
||||
QuotaThrottleMbps int `json:"quota_throttle_mbps"`
|
||||
QuotaExceeded bool `json:"quota_exceeded,omitempty"`
|
||||
OwnerUsername string `json:"owner_username,omitempty"`
|
||||
Expired bool `json:"expired,omitempty"`
|
||||
}
|
||||
|
||||
// XrayInboundInfo is returned by /api/xray/inbounds.
|
||||
@@ -2129,6 +2345,16 @@ func (m *XrayManager) modifyRawConfig(fn func(cfg map[string]interface{}) error)
|
||||
|
||||
// AddXrayClient adds a client to the named inbound and saves the config.
|
||||
func (m *XrayManager) AddXrayClient(inboundTag, uuid, email string) error {
|
||||
return m.addXrayClient(inboundTag, uuid, email, false)
|
||||
}
|
||||
|
||||
// EnsureXrayClient restores a previously suspended DB-backed client without
|
||||
// failing if it is already present in the active config.
|
||||
func (m *XrayManager) EnsureXrayClient(inboundTag, uuid, email string) error {
|
||||
return m.addXrayClient(inboundTag, uuid, email, true)
|
||||
}
|
||||
|
||||
func (m *XrayManager) addXrayClient(inboundTag, uuid, email string, allowExisting bool) error {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
err := m.modifyRawConfig(func(raw map[string]interface{}) error {
|
||||
@@ -2155,6 +2381,10 @@ func (m *XrayManager) AddXrayClient(inboundTag, uuid, email string) error {
|
||||
id, _ = cm["password"].(string)
|
||||
}
|
||||
if id == uuid {
|
||||
if allowExisting {
|
||||
cm["email"] = email
|
||||
return nil
|
||||
}
|
||||
return fmt.Errorf("UUID %s already exists in inbound %s", uuid, inboundTag)
|
||||
}
|
||||
}
|
||||
@@ -2338,6 +2568,10 @@ func handleXrayInbounds(w http.ResponseWriter, r *http.Request) {
|
||||
Name: m.Name,
|
||||
ExpiresAt: m.ExpiresAt,
|
||||
MaxConns: m.MaxConns,
|
||||
DataQuotaBytes: m.DataQuotaBytes,
|
||||
QuotaAction: normalizeQuotaAction(m.QuotaAction),
|
||||
QuotaThrottleMbps: quotaThrottleMbpsOrDefault(m.QuotaThrottleMbps),
|
||||
QuotaExceeded: m.DataQuotaBytes > 0 && m.TotalUplinkBytes+m.TotalDownlinkBytes >= m.DataQuotaBytes,
|
||||
OwnerUsername: m.OwnerUsername,
|
||||
UplinkBytes: m.TotalUplinkBytes,
|
||||
DownlinkBytes: m.TotalDownlinkBytes,
|
||||
@@ -2408,6 +2642,10 @@ func applyXrayRuntimeStats(c *XrayClientInfo) {
|
||||
if c == nil {
|
||||
return
|
||||
}
|
||||
if rate, ok := xrayMgr.RuntimeRateForKeys(c.Email, c.UUID, c.Name); ok {
|
||||
c.UpBytesPerSec = rate.UpBytesPerSec
|
||||
c.DownBytesPerSec = rate.DownBytesPerSec
|
||||
}
|
||||
st, ok := xrayMgr.RuntimeStatsForKeys(c.Email, c.UUID, c.Name)
|
||||
if !ok {
|
||||
return
|
||||
@@ -2421,6 +2659,7 @@ func applyXrayRuntimeStats(c *XrayClientInfo) {
|
||||
c.DownlinkBytes = st.Downlink
|
||||
}
|
||||
c.TotalBytes = c.UplinkBytes + c.DownlinkBytes
|
||||
c.QuotaExceeded = c.DataQuotaBytes > 0 && c.TotalBytes >= c.DataQuotaBytes
|
||||
if st.ActiveConnections > c.ActiveConnections {
|
||||
c.ActiveConnections = st.ActiveConnections
|
||||
}
|
||||
@@ -2437,14 +2676,17 @@ func handleXrayClientAdd(w http.ResponseWriter, r *http.Request) {
|
||||
return
|
||||
}
|
||||
var req struct {
|
||||
InboundTag string `json:"inbound_tag"`
|
||||
UUID string `json:"uuid"`
|
||||
Email string `json:"email"`
|
||||
Name string `json:"name"`
|
||||
ExpiresAt string `json:"expires_at"` // RFC3339 or YYYY-MM-DD or empty
|
||||
MaxConnections int `json:"max_connections"`
|
||||
OwnerUsername string `json:"owner_username,omitempty"`
|
||||
ServerID string `json:"server_id,omitempty"`
|
||||
InboundTag string `json:"inbound_tag"`
|
||||
UUID string `json:"uuid"`
|
||||
Email string `json:"email"`
|
||||
Name string `json:"name"`
|
||||
ExpiresAt string `json:"expires_at"` // RFC3339 or YYYY-MM-DD or empty
|
||||
MaxConnections int `json:"max_connections"`
|
||||
DataQuotaBytes int64 `json:"data_quota_bytes"`
|
||||
QuotaAction string `json:"quota_action"`
|
||||
QuotaThrottleMbps int `json:"quota_throttle_mbps"`
|
||||
OwnerUsername string `json:"owner_username,omitempty"`
|
||||
ServerID string `json:"server_id,omitempty"`
|
||||
}
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
http.Error(w, "invalid json", http.StatusBadRequest)
|
||||
@@ -2454,6 +2696,20 @@ func handleXrayClientAdd(w http.ResponseWriter, r *http.Request) {
|
||||
http.Error(w, "inbound_tag and uuid required", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
req.InboundTag = strings.TrimSpace(req.InboundTag)
|
||||
req.UUID = strings.TrimSpace(req.UUID)
|
||||
if _, err := parseUUID(req.UUID); err != nil {
|
||||
http.Error(w, "invalid uuid: "+err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if req.MaxConnections < 0 || req.MaxConnections > 10000 {
|
||||
http.Error(w, "max_connections must be between 0 and 10000", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if err := validateQuotaConfig(req.DataQuotaBytes, req.QuotaAction, req.QuotaThrottleMbps); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if ms, remote, err := managedServerFromID(r.Context(), statsStore, req.ServerID); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
@@ -2501,6 +2757,11 @@ func handleXrayClientAdd(w http.ResponseWriter, r *http.Request) {
|
||||
if req.Email == "" {
|
||||
req.Email = req.UUID
|
||||
}
|
||||
expiresAt, err := parseOptionalXrayExpiry(req.ExpiresAt)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
sess := sessionFromCtx(r.Context())
|
||||
ownerUsername := ""
|
||||
@@ -2537,35 +2798,35 @@ func handleXrayClientAdd(w http.ResponseWriter, r *http.Request) {
|
||||
return
|
||||
}
|
||||
}
|
||||
if err := xrayMgr.AddXrayClient(req.InboundTag, req.UUID, req.Email); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
var savedMeta *XrayClientMeta
|
||||
if statsStore != nil {
|
||||
meta := XrayClientMeta{
|
||||
UUID: req.UUID,
|
||||
Name: req.Name,
|
||||
Email: req.Email,
|
||||
InboundTag: req.InboundTag,
|
||||
OwnerUsername: ownerUsername,
|
||||
MaxConns: req.MaxConnections,
|
||||
}
|
||||
if req.ExpiresAt != "" {
|
||||
var t time.Time
|
||||
var err error
|
||||
for _, layout := range []string{time.RFC3339, "2006-01-02T15:04", "2006-01-02"} {
|
||||
t, err = time.Parse(layout, req.ExpiresAt)
|
||||
if err == nil {
|
||||
break
|
||||
}
|
||||
}
|
||||
if err == nil {
|
||||
meta.ExpiresAt = &t
|
||||
}
|
||||
UUID: req.UUID,
|
||||
Name: req.Name,
|
||||
Email: req.Email,
|
||||
InboundTag: req.InboundTag,
|
||||
OwnerUsername: ownerUsername,
|
||||
MaxConns: req.MaxConnections,
|
||||
DataQuotaBytes: req.DataQuotaBytes,
|
||||
QuotaAction: normalizeQuotaAction(req.QuotaAction),
|
||||
QuotaThrottleMbps: quotaThrottleMbpsOrDefault(req.QuotaThrottleMbps),
|
||||
}
|
||||
meta.ExpiresAt = expiresAt
|
||||
if err := statsStore.UpsertXrayClientMeta(r.Context(), meta); err != nil {
|
||||
xrayLogf("xray: save meta for %s: %v", req.UUID, err)
|
||||
http.Error(w, "save client metadata failed: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
xrayMgr.setNativeQuotaPolicy(&meta)
|
||||
savedMeta = &meta
|
||||
}
|
||||
// Publish the credential only after its quota/owner/expiry policy exists, so
|
||||
// a fast native client can never enter an unmetered window during creation.
|
||||
if err := xrayMgr.AddXrayClient(req.InboundTag, req.UUID, req.Email); err != nil {
|
||||
if savedMeta != nil {
|
||||
_ = statsStore.DeleteXrayClientMeta(r.Context(), req.UUID)
|
||||
}
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
xrayMgr.restartIfExternalRunning()
|
||||
w.WriteHeader(http.StatusCreated)
|
||||
@@ -2579,12 +2840,16 @@ func handleXrayClientUpdate(w http.ResponseWriter, r *http.Request) {
|
||||
return
|
||||
}
|
||||
var req struct {
|
||||
UUID string `json:"uuid"`
|
||||
Name string `json:"name"`
|
||||
Email string `json:"email"`
|
||||
ExpiresAt string `json:"expires_at"`
|
||||
MaxConnections int `json:"max_connections"`
|
||||
ServerID string `json:"server_id,omitempty"`
|
||||
UUID string `json:"uuid"`
|
||||
Name string `json:"name"`
|
||||
Email string `json:"email"`
|
||||
ExpiresAt string `json:"expires_at"`
|
||||
MaxConnections int `json:"max_connections"`
|
||||
DataQuotaBytes int64 `json:"data_quota_bytes"`
|
||||
QuotaAction string `json:"quota_action"`
|
||||
QuotaThrottleMbps int `json:"quota_throttle_mbps"`
|
||||
ResetUsage bool `json:"reset_usage,omitempty"`
|
||||
ServerID string `json:"server_id,omitempty"`
|
||||
}
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
http.Error(w, "invalid json", http.StatusBadRequest)
|
||||
@@ -2594,6 +2859,19 @@ func handleXrayClientUpdate(w http.ResponseWriter, r *http.Request) {
|
||||
http.Error(w, "uuid required", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
req.UUID = strings.TrimSpace(req.UUID)
|
||||
if _, err := parseUUID(req.UUID); err != nil {
|
||||
http.Error(w, "invalid uuid: "+err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if req.MaxConnections < 0 || req.MaxConnections > 10000 {
|
||||
http.Error(w, "max_connections must be between 0 and 10000", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if err := validateQuotaConfig(req.DataQuotaBytes, req.QuotaAction, req.QuotaThrottleMbps); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if ms, remote, err := managedServerFromID(r.Context(), statsStore, req.ServerID); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
@@ -2627,37 +2905,139 @@ func handleXrayClientUpdate(w http.ResponseWriter, r *http.Request) {
|
||||
http.Error(w, "forbidden", http.StatusForbidden)
|
||||
return
|
||||
}
|
||||
req.Email = strings.TrimSpace(req.Email)
|
||||
if req.Email == "" {
|
||||
req.Email = firstNonEmpty(strings.TrimSpace(req.Name), existing.Email, req.UUID)
|
||||
}
|
||||
expiresAt, err := parseOptionalXrayExpiry(req.ExpiresAt)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
meta := XrayClientMeta{
|
||||
UUID: req.UUID,
|
||||
Name: req.Name,
|
||||
Email: req.Email,
|
||||
InboundTag: existing.InboundTag,
|
||||
OwnerUsername: existing.OwnerUsername,
|
||||
MaxConns: req.MaxConnections,
|
||||
UUID: req.UUID,
|
||||
Name: req.Name,
|
||||
Email: req.Email,
|
||||
InboundTag: existing.InboundTag,
|
||||
OwnerUsername: existing.OwnerUsername,
|
||||
MaxConns: req.MaxConnections,
|
||||
DataQuotaBytes: req.DataQuotaBytes,
|
||||
QuotaAction: normalizeQuotaAction(req.QuotaAction),
|
||||
QuotaThrottleMbps: quotaThrottleMbpsOrDefault(req.QuotaThrottleMbps),
|
||||
TotalUplinkBytes: existing.TotalUplinkBytes,
|
||||
TotalDownlinkBytes: existing.TotalDownlinkBytes,
|
||||
}
|
||||
if req.ExpiresAt != "" {
|
||||
for _, layout := range []string{time.RFC3339, "2006-01-02T15:04", "2006-01-02"} {
|
||||
if t, err := time.Parse(layout, req.ExpiresAt); err == nil {
|
||||
meta.ExpiresAt = &t
|
||||
break
|
||||
}
|
||||
meta.ExpiresAt = expiresAt
|
||||
emailChanged := req.Email != existing.Email
|
||||
if emailChanged {
|
||||
if err := xrayMgr.UpdateXrayClientEmail(req.UUID, req.Email); err != nil {
|
||||
http.Error(w, "update config email failed: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
}
|
||||
if err := statsStore.UpsertXrayClientMeta(r.Context(), meta); err != nil {
|
||||
if emailChanged {
|
||||
_ = xrayMgr.UpdateXrayClientEmail(req.UUID, existing.Email)
|
||||
}
|
||||
http.Error(w, "update failed: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
if req.Email != "" {
|
||||
if err := xrayMgr.UpdateXrayClientEmail(req.UUID, req.Email); err != nil {
|
||||
xrayLogf("xray: update config email for %s: %v", req.UUID, err)
|
||||
} else {
|
||||
xrayMgr.restartIfExternalRunning()
|
||||
if req.ResetUsage {
|
||||
if err := xrayMgr.resetNativeTrafficAccounting(r.Context(), statsStore, req.UUID, existing.Email); err != nil {
|
||||
http.Error(w, "usage reset failed: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
meta.TotalUplinkBytes = 0
|
||||
meta.TotalDownlinkBytes = 0
|
||||
}
|
||||
xrayMgr.setNativeQuotaPolicy(&meta)
|
||||
if meta.ExpiresAt != nil && !meta.ExpiresAt.After(time.Now()) {
|
||||
xrayMgr.disconnectNativeClient(req.UUID)
|
||||
}
|
||||
if emailChanged {
|
||||
xrayMgr.restartIfExternalRunning()
|
||||
}
|
||||
w.WriteHeader(http.StatusOK)
|
||||
}
|
||||
|
||||
func parseOptionalXrayExpiry(raw string) (*time.Time, error) {
|
||||
raw = strings.TrimSpace(raw)
|
||||
if raw == "" {
|
||||
return nil, nil
|
||||
}
|
||||
for _, layout := range []string{time.RFC3339, "2006-01-02T15:04", "2006-01-02"} {
|
||||
if t, err := time.Parse(layout, raw); err == nil {
|
||||
return &t, nil
|
||||
}
|
||||
}
|
||||
return nil, fmt.Errorf("invalid expires_at (RFC3339, YYYY-MM-DDThh:mm, or YYYY-MM-DD required)")
|
||||
}
|
||||
|
||||
func handleXrayClientResetTraffic(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
var req struct {
|
||||
UUID string `json:"uuid"`
|
||||
ServerID string `json:"server_id,omitempty"`
|
||||
}
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
http.Error(w, "invalid json", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
req.UUID = strings.TrimSpace(req.UUID)
|
||||
if req.UUID == "" {
|
||||
http.Error(w, "uuid required", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
ctx := r.Context()
|
||||
if ms, remote, err := managedServerFromID(ctx, statsStore, req.ServerID); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
} else if remote {
|
||||
if sess := sessionFromCtx(ctx); sess != nil && sess.Role == RoleReseller && !remoteXrayClientOwned(ctx, ms, req.UUID, sess.Username) {
|
||||
http.Error(w, "forbidden", http.StatusForbidden)
|
||||
return
|
||||
}
|
||||
req.ServerID = ""
|
||||
body, _ := json.Marshal(req)
|
||||
status, data, ct, err := proxyManagedServer(ctx, ms, http.MethodPost, "/api/xray/clients/reset-traffic", body, "application/json")
|
||||
if err != nil {
|
||||
http.Error(w, "remote server error: "+err.Error(), http.StatusBadGateway)
|
||||
return
|
||||
}
|
||||
writeProxyResponse(w, status, data, ct)
|
||||
return
|
||||
}
|
||||
if statsStore == nil {
|
||||
http.Error(w, "storage not available", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
existing, err := statsStore.GetXrayClientMeta(ctx, req.UUID)
|
||||
if err != nil {
|
||||
if err == sql.ErrNoRows {
|
||||
http.Error(w, "client not found", http.StatusNotFound)
|
||||
} else {
|
||||
http.Error(w, "database error", http.StatusInternalServerError)
|
||||
}
|
||||
return
|
||||
}
|
||||
if sess := sessionFromCtx(ctx); sess != nil && sess.Role == RoleReseller && existing.OwnerUsername != sess.Username {
|
||||
http.Error(w, "forbidden", http.StatusForbidden)
|
||||
return
|
||||
}
|
||||
if err := xrayMgr.resetNativeTrafficAccounting(ctx, statsStore, req.UUID, existing.Email); err != nil {
|
||||
http.Error(w, "usage reset failed: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
_ = json.NewEncoder(w).Encode(map[string]interface{}{"ok": true, "uuid": req.UUID})
|
||||
}
|
||||
|
||||
func handleXrayClientRemove(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodDelete {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
|
||||
+97
-43
@@ -138,6 +138,14 @@ func (s *nativeXrayServer) start(configFile string) error {
|
||||
if s.running {
|
||||
return fmt.Errorf("native xray already running")
|
||||
}
|
||||
beginNativeTransportAccepting()
|
||||
started := false
|
||||
defer func() {
|
||||
if !started {
|
||||
stopNativeTransportAccepting()
|
||||
closeAllNativeTransportConnections()
|
||||
}
|
||||
}()
|
||||
if configFile == "" {
|
||||
return fmt.Errorf("native xray: no config file configured")
|
||||
}
|
||||
@@ -196,9 +204,11 @@ func (s *nativeXrayServer) start(configFile string) error {
|
||||
}
|
||||
return fmt.Errorf("native xray: listen %s (shared XHTTP): %w", addr, err)
|
||||
}
|
||||
serveLn := net.Listener(ln)
|
||||
// Track accepted sockets so stop/reload can close them. VPN transports are
|
||||
// not subject to a global website-style connection ceiling.
|
||||
serveLn := trackNativeListener(ln)
|
||||
if group.security == "tls" {
|
||||
serveLn = tls.NewListener(ln, group.tlsConfig)
|
||||
serveLn = tls.NewListener(serveLn, group.tlsConfig)
|
||||
}
|
||||
opened = append(opened, serveLn)
|
||||
xrayGo(fmt.Sprintf("native xray shared xhttp listener %s", addr), func() { group.serve(serveLn) })
|
||||
@@ -212,21 +222,34 @@ func (s *nativeXrayServer) start(configFile string) error {
|
||||
s.inboundsByTag = active
|
||||
s.running = true
|
||||
s.startTime = time.Now()
|
||||
started = true
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *nativeXrayServer) stop() {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if !s.running && len(s.listeners) == 0 {
|
||||
s.mu.Unlock()
|
||||
return
|
||||
}
|
||||
for _, l := range s.listeners {
|
||||
_ = l.Close()
|
||||
stopNativeTransportAccepting()
|
||||
listeners := append([]net.Listener(nil), s.listeners...)
|
||||
inbounds := make([]*nativeInbound, 0, len(s.inboundsByTag))
|
||||
for _, ib := range s.inboundsByTag {
|
||||
inbounds = append(inbounds, ib)
|
||||
}
|
||||
s.listeners = nil
|
||||
s.inboundsByTag = nil
|
||||
s.running = false
|
||||
s.mu.Unlock()
|
||||
|
||||
for _, l := range listeners {
|
||||
_ = l.Close()
|
||||
}
|
||||
closeAllNativeTransportConnections()
|
||||
for _, ib := range inbounds {
|
||||
ib.closeAllXHTTPSessions()
|
||||
}
|
||||
xrayLogf("native xray: stopped")
|
||||
}
|
||||
|
||||
@@ -241,6 +264,12 @@ func (ib *nativeInbound) acceptLoop(ln net.Listener) {
|
||||
xrayLogf("native xray: accept error on %s: %v", ln.Addr(), err)
|
||||
continue
|
||||
}
|
||||
counted, ok := waitWrapTrackedNativeTransportConn(c)
|
||||
if !ok {
|
||||
time.Sleep(nativeOverloadBackoff)
|
||||
continue
|
||||
}
|
||||
c = counted
|
||||
xrayGo(fmt.Sprintf("native xray connection remote=%s", c.RemoteAddr()), func() { ib.serve(c) })
|
||||
}
|
||||
}
|
||||
@@ -262,7 +291,7 @@ func (ib *nativeInbound) serve(raw net.Conn) {
|
||||
tconn := tls.Server(raw, ib.tlsConfig)
|
||||
_ = tconn.SetDeadline(time.Now().Add(tlsHandshakeTimeout))
|
||||
if err := tconn.Handshake(); err != nil {
|
||||
xrayLogf("native xray: tls handshake from %s failed: %v", raw.RemoteAddr(), err)
|
||||
logNativePreAuthRejection("native xray: tls handshake from %s failed: %v", raw.RemoteAddr(), err)
|
||||
return
|
||||
}
|
||||
_ = tconn.SetDeadline(time.Time{})
|
||||
@@ -275,11 +304,13 @@ func (ib *nativeInbound) serve(raw net.Conn) {
|
||||
case "tcp", "raw", "":
|
||||
// stream is already the protocol stream
|
||||
case "ws", "websocket":
|
||||
_ = conn.SetDeadline(time.Now().Add(tlsHandshakeTimeout))
|
||||
ws, err := wsServerHandshake(conn, ib.path)
|
||||
if err != nil {
|
||||
xrayLogf("native xray: ws handshake from %s failed: %v", raw.RemoteAddr(), err)
|
||||
logNativePreAuthRejection("native xray: ws handshake from %s failed: %v", raw.RemoteAddr(), err)
|
||||
return
|
||||
}
|
||||
_ = conn.SetDeadline(time.Time{})
|
||||
stream = ws
|
||||
case "xhttp", "splithttp":
|
||||
xrayLogf("native xray: inbound %q got raw connection for XHTTP; this transport is served by http.Server", ib.tag)
|
||||
@@ -331,11 +362,7 @@ const (
|
||||
|
||||
func (ib *nativeInbound) handleVLESS(stream net.Conn, remote net.Addr) {
|
||||
defer xrayRecover(fmt.Sprintf("native xray VLESS inbound=%q remote=%s", ib.tag, remote))
|
||||
if ib.isXHTTP() {
|
||||
xrayTracef("native xray: vless handshake start inbound=%q transport=%s remote=%s", ib.tag, ib.transport, remote)
|
||||
} else {
|
||||
xrayLogf("native xray: vless handshake start inbound=%q transport=%s remote=%s", ib.tag, ib.transport, remote)
|
||||
}
|
||||
xrayTracef("native xray: vless handshake start inbound=%q transport=%s remote=%s", ib.tag, ib.transport, remote)
|
||||
_ = stream.SetReadDeadline(time.Now().Add(30 * time.Second))
|
||||
|
||||
head := make([]byte, 1+16+1) // version + uuid + addonLen
|
||||
@@ -349,7 +376,11 @@ func (ib *nativeInbound) handleVLESS(stream net.Conn, remote net.Addr) {
|
||||
|
||||
client := ib.getNativeClient(id)
|
||||
if client == nil {
|
||||
xrayLogf("native xray: inbound %q rejected unknown VLESS uuid from %s", ib.tag, remote)
|
||||
logNativePreAuthRejection("native xray: inbound %q rejected unknown VLESS uuid from %s", ib.tag, remote)
|
||||
return
|
||||
}
|
||||
if reason := xrayMgr.nativeClientAccessDenied(client.uuid); reason != "" {
|
||||
xrayLogf("native xray: inbound %q rejected VLESS user %s: %s", ib.tag, client.email, reason)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -396,6 +427,18 @@ func (ib *nativeInbound) handleVLESS(stream net.Conn, remote net.Addr) {
|
||||
}
|
||||
_ = stream.SetReadDeadline(time.Time{})
|
||||
|
||||
switch cmd[0] {
|
||||
case vlessCmdTCP, vlessCmdUDP, vlessCmdMux:
|
||||
default:
|
||||
xrayLogf("native xray: inbound %q VLESS command %d not supported yet", ib.tag, cmd[0])
|
||||
return
|
||||
}
|
||||
releaseConnection, quotaState, ok := xrayMgr.acquireNativeClientConnection(client.uuid, client.email, stream)
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
defer releaseConnection()
|
||||
|
||||
// VLESS response header must be sent before relaying payload. CommandMux is
|
||||
// special: official Xray does not read a target from the VLESS header for it;
|
||||
// the following bytes are Mux.Cool/XUDP frames. Reading port/address here
|
||||
@@ -413,7 +456,7 @@ func (ib *nativeInbound) handleVLESS(stream net.Conn, remote net.Addr) {
|
||||
return
|
||||
}
|
||||
ib.nativeSuccessLogf("native xray: vless/tcp user=%s src=%s -> %s (inbound %q)", client.email, backend.LocalAddr(), target, ib.tag)
|
||||
nativeTunnel(stream, backend, client.uuid, client.email, ib.upLimiter(), ib.downLimiter())
|
||||
nativeTunnel(stream, backend, client.uuid, client.email, quotaState, ib.upLimiter(), ib.downLimiter())
|
||||
case vlessCmdUDP:
|
||||
backend, target, err := ib.nativeDialUDP(host, port)
|
||||
if err != nil {
|
||||
@@ -421,12 +464,10 @@ func (ib *nativeInbound) handleVLESS(stream net.Conn, remote net.Addr) {
|
||||
return
|
||||
}
|
||||
ib.nativeSuccessLogf("native xray: vless/udp user=%s src=%s -> %s (inbound %q)", client.email, backend.LocalAddr(), target, ib.tag)
|
||||
nativeVLESSUDPTunnel(stream, backend, client.uuid, client.email, ib.upLimiter(), ib.downLimiter())
|
||||
nativeVLESSUDPTunnel(stream, backend, client.uuid, client.email, quotaState, ib.upLimiter(), ib.downLimiter())
|
||||
case vlessCmdMux:
|
||||
ib.nativeSuccessLogf("native xray: vless/mux user=%s remote=%s (inbound %q)", client.email, remote, ib.tag)
|
||||
ib.nativeVLESSMuxTunnel(stream, client.uuid, client.email)
|
||||
default:
|
||||
xrayLogf("native xray: inbound %q VLESS command %d not supported yet", ib.tag, cmd[0])
|
||||
ib.nativeVLESSMuxTunnel(stream, client.uuid, client.email, quotaState)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -440,7 +481,7 @@ func (ib *nativeInbound) logVLESSReadFailure(stage string, remote net.Addr, emai
|
||||
return
|
||||
}
|
||||
if email == "" {
|
||||
xrayLogf("native xray: vless %s failed inbound=%q transport=%s remote=%s: %v", stage, ib.tag, ib.transport, remote, err)
|
||||
logNativePreAuthRejection("native xray: vless %s failed inbound=%q transport=%s remote=%s: %v", stage, ib.tag, ib.transport, remote, err)
|
||||
} else {
|
||||
xrayLogf("native xray: vless %s failed inbound=%q transport=%s user=%s remote=%s: %v", stage, ib.tag, ib.transport, email, remote, err)
|
||||
}
|
||||
@@ -657,18 +698,18 @@ func normalizeNativeTargetHost(raw string) string {
|
||||
// backend, applying per-direction rate limits and accounting traffic against
|
||||
// the client's email so the panel's online detection keeps working. It mirrors
|
||||
// handleDirectTCPIP in main.go.
|
||||
func nativeTunnel(client io.ReadWriteCloser, backend net.Conn, uuid, email string, up, down *rate.Limiter) {
|
||||
xrayMgr.recordNativeConnect(uuid, email)
|
||||
defer xrayMgr.recordNativeDisconnect(uuid, email)
|
||||
func nativeTunnel(client io.ReadWriteCloser, backend net.Conn, uuid, email string, quotaState *xrayNativeQuotaState, up, down *rate.Limiter) {
|
||||
defer xrayRecover(fmt.Sprintf("native xray TCP tunnel user=%s", email))
|
||||
|
||||
upMeter := &trafficMeter{uuid: uuid, email: email, uplink: true}
|
||||
downMeter := &trafficMeter{uuid: uuid, email: email, uplink: false}
|
||||
upMeter := newTrafficMeter(uuid, email, true, quotaState)
|
||||
downMeter := newTrafficMeter(uuid, email, false, quotaState)
|
||||
|
||||
var wg sync.WaitGroup
|
||||
var closeOnce sync.Once
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
closeAll := func() {
|
||||
closeOnce.Do(func() {
|
||||
cancel()
|
||||
_ = backend.Close()
|
||||
_ = client.Close()
|
||||
})
|
||||
@@ -678,7 +719,7 @@ func nativeTunnel(client io.ReadWriteCloser, backend net.Conn, uuid, email strin
|
||||
xrayGo("native xray TCP uplink", func() { // client -> backend
|
||||
defer wg.Done()
|
||||
defer closeAll()
|
||||
_, _ = copyWithRateLimit(meteredWriter{w: backend, meter: upMeter}, client, up)
|
||||
_, _ = copyWithRateLimitContext(ctx, xrayQuotaMeteredWriter{w: backend, meter: upMeter, ctx: ctx}, client, up)
|
||||
if cw, ok := backend.(interface{ CloseWrite() error }); ok {
|
||||
_ = cw.CloseWrite()
|
||||
}
|
||||
@@ -688,7 +729,7 @@ func nativeTunnel(client io.ReadWriteCloser, backend net.Conn, uuid, email strin
|
||||
xrayGo("native xray TCP downlink", func() { // backend -> client
|
||||
defer wg.Done()
|
||||
defer closeAll()
|
||||
_, _ = copyWithRateLimit(meteredWriter{w: client, meter: downMeter}, backend, down)
|
||||
_, _ = copyWithRateLimitContext(ctx, xrayQuotaMeteredWriter{w: client, meter: downMeter, ctx: ctx}, backend, down)
|
||||
})
|
||||
|
||||
wg.Wait()
|
||||
@@ -700,15 +741,24 @@ func nativeTunnel(client io.ReadWriteCloser, backend net.Conn, uuid, email strin
|
||||
// trafficMeter accumulates bytes for one direction and flushes them to the
|
||||
// stats manager in batches to avoid locking on every write.
|
||||
type trafficMeter struct {
|
||||
uuid string
|
||||
email string
|
||||
uplink bool
|
||||
n int64
|
||||
uuid string
|
||||
email string
|
||||
uplink bool
|
||||
n int64
|
||||
quotaGeneration uint64
|
||||
state *xrayNativeQuotaState
|
||||
}
|
||||
|
||||
const trafficFlushThreshold = 1024 * 1024
|
||||
|
||||
func newTrafficMeter(uuid, email string, uplink bool, state *xrayNativeQuotaState) *trafficMeter {
|
||||
t := &trafficMeter{uuid: uuid, email: email, uplink: uplink, state: state}
|
||||
t.syncQuotaGeneration()
|
||||
return t
|
||||
}
|
||||
|
||||
func (t *trafficMeter) add(n int) {
|
||||
t.syncQuotaGeneration()
|
||||
t.n += int64(n)
|
||||
if t.n >= trafficFlushThreshold {
|
||||
t.flush()
|
||||
@@ -716,29 +766,33 @@ func (t *trafficMeter) add(n int) {
|
||||
}
|
||||
|
||||
func (t *trafficMeter) flush() {
|
||||
t.syncQuotaGeneration()
|
||||
if t.n == 0 || t.email == "" {
|
||||
return
|
||||
}
|
||||
if t.uplink {
|
||||
xrayMgr.recordNativeTraffic(t.uuid, t.email, t.n, 0)
|
||||
xrayMgr.recordNativeTraffic(t.uuid, t.email, t.n, 0, t.quotaGeneration, t.state)
|
||||
} else {
|
||||
xrayMgr.recordNativeTraffic(t.uuid, t.email, 0, t.n)
|
||||
xrayMgr.recordNativeTraffic(t.uuid, t.email, 0, t.n, t.quotaGeneration, t.state)
|
||||
}
|
||||
t.n = 0
|
||||
}
|
||||
|
||||
// meteredWriter counts bytes as they are written through to the wrapped writer.
|
||||
type meteredWriter struct {
|
||||
w io.Writer
|
||||
meter *trafficMeter
|
||||
}
|
||||
|
||||
func (mw meteredWriter) Write(p []byte) (int, error) {
|
||||
n, err := mw.w.Write(p)
|
||||
if n > 0 {
|
||||
mw.meter.add(n)
|
||||
func (t *trafficMeter) syncQuotaGeneration() {
|
||||
var generation uint64
|
||||
if t.state != nil {
|
||||
t.state.mu.Lock()
|
||||
generation = t.state.generation
|
||||
t.state.mu.Unlock()
|
||||
}
|
||||
if t.quotaGeneration == 0 {
|
||||
t.quotaGeneration = generation
|
||||
return
|
||||
}
|
||||
if generation != t.quotaGeneration {
|
||||
t.n = 0
|
||||
t.quotaGeneration = generation
|
||||
}
|
||||
return n, err
|
||||
}
|
||||
|
||||
func (ib *nativeInbound) upLimiter() *rate.Limiter { return newByteLimiter(ib.upBytesPerSec) }
|
||||
|
||||
+184
-32
@@ -54,7 +54,11 @@ type nativeMuxUplinkItem struct {
|
||||
port uint16
|
||||
}
|
||||
|
||||
const nativeMuxUplinkQueue = 64
|
||||
const (
|
||||
nativeMuxUplinkQueue = 16
|
||||
nativeMuxMaxBufferedBytesPerSession = 1 * 1024 * 1024
|
||||
nativeMuxMaxBufferedBytesGlobal = 128 * 1024 * 1024
|
||||
)
|
||||
|
||||
var nativeMuxFramePool = sync.Pool{
|
||||
New: func() any {
|
||||
@@ -91,6 +95,11 @@ type nativeMuxSession struct {
|
||||
uplink chan nativeMuxUplinkItem
|
||||
closed chan struct{}
|
||||
closeOnce sync.Once
|
||||
finishOnce sync.Once
|
||||
enqueueMu sync.Mutex
|
||||
enqueueWG sync.WaitGroup
|
||||
enqueueDone bool
|
||||
buffered atomic.Int64
|
||||
ctx context.Context
|
||||
cancel context.CancelFunc
|
||||
onClose func(*nativeMuxSession)
|
||||
@@ -98,7 +107,11 @@ type nativeMuxSession struct {
|
||||
globalID [8]byte
|
||||
}
|
||||
|
||||
var nativeMuxGlobalActive atomic.Int64
|
||||
var (
|
||||
nativeMuxGlobalActive atomic.Int64
|
||||
nativeMuxBufferedBytes atomic.Int64
|
||||
nativeMuxBufferRejected atomic.Int64
|
||||
)
|
||||
|
||||
func acquireNativeMuxGlobalSlot() (func(), bool) {
|
||||
limit := int64(nativeMuxGlobalSessionLimit())
|
||||
@@ -117,15 +130,73 @@ func acquireNativeMuxGlobalSlot() (func(), bool) {
|
||||
}
|
||||
}
|
||||
|
||||
func reserveNativeMuxBufferedBytes(s *nativeMuxSession, n int64) bool {
|
||||
if s == nil || n <= 0 {
|
||||
return true
|
||||
}
|
||||
for {
|
||||
current := s.buffered.Load()
|
||||
if current > nativeMuxMaxBufferedBytesPerSession-n {
|
||||
logNativeLimitRejection("mux session buffered bytes", &nativeMuxBufferRejected, nativeMuxMaxBufferedBytesPerSession)
|
||||
return false
|
||||
}
|
||||
if s.buffered.CompareAndSwap(current, current+n) {
|
||||
break
|
||||
}
|
||||
}
|
||||
for {
|
||||
current := nativeMuxBufferedBytes.Load()
|
||||
if current > nativeMuxMaxBufferedBytesGlobal-n {
|
||||
s.buffered.Add(-n)
|
||||
logNativeLimitRejection("mux global buffered bytes", &nativeMuxBufferRejected, nativeMuxMaxBufferedBytesGlobal)
|
||||
return false
|
||||
}
|
||||
if nativeMuxBufferedBytes.CompareAndSwap(current, current+n) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func releaseNativeMuxBufferedBytes(s *nativeMuxSession, n int64) {
|
||||
if s == nil || n <= 0 {
|
||||
return
|
||||
}
|
||||
for {
|
||||
current := s.buffered.Load()
|
||||
release := n
|
||||
if release > current {
|
||||
release = current
|
||||
}
|
||||
if s.buffered.CompareAndSwap(current, current-release) {
|
||||
releaseNativeAtomicBytes(&nativeMuxBufferedBytes, release)
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func releaseNativeAtomicBytes(counter *atomic.Int64, n int64) {
|
||||
if counter == nil || n <= 0 {
|
||||
return
|
||||
}
|
||||
for {
|
||||
current := counter.Load()
|
||||
next := current - n
|
||||
if next < 0 {
|
||||
next = 0
|
||||
}
|
||||
if counter.CompareAndSwap(current, next) {
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// nativeVLESSMuxTunnel implements the server side of Xray's Mux.Cool framing
|
||||
// for VLESS CommandMux. CommandMux does not carry a VLESS target address; every
|
||||
// child TCP/UDP request is described by mux frame metadata. UDP is treated as a
|
||||
// packet protocol, not as a byte stream, and XUDP-style GlobalID/endpoint
|
||||
// metadata is accepted for full-cone friendly clients.
|
||||
func (ib *nativeInbound) nativeVLESSMuxTunnel(stream io.ReadWriteCloser, uuid, email string) {
|
||||
func (ib *nativeInbound) nativeVLESSMuxTunnel(stream io.ReadWriteCloser, uuid, email string, quotaState *xrayNativeQuotaState) {
|
||||
defer xrayRecover(fmt.Sprintf("native xray VLESS mux user=%s", email))
|
||||
xrayMgr.recordNativeConnect(uuid, email)
|
||||
defer xrayMgr.recordNativeDisconnect(uuid, email)
|
||||
|
||||
writeMu := &sync.Mutex{}
|
||||
sessions := make(map[uint16]*nativeMuxSession)
|
||||
@@ -256,7 +327,7 @@ func (ib *nativeInbound) nativeVLESSMuxTunnel(stream io.ReadWriteCloser, uuid, e
|
||||
}
|
||||
}
|
||||
|
||||
s, target, err := ib.newNativeMuxSession(meta.sessionID, meta.network, targetHost, targetPort, isXUDP, meta.globalID, stream, writeMu, uuid, email, removeSession)
|
||||
s, target, err := ib.newNativeMuxSession(meta.sessionID, meta.network, targetHost, targetPort, isXUDP, meta.globalID, stream, writeMu, uuid, email, quotaState, removeSession)
|
||||
if err != nil {
|
||||
xrayLogf("native xray: VLESS mux session %s setup failed: %v", target, err)
|
||||
writeMu.Lock()
|
||||
@@ -281,8 +352,11 @@ func (ib *nativeInbound) nativeVLESSMuxTunnel(stream io.ReadWriteCloser, uuid, e
|
||||
xrayTracef("native xray: vless/mux %s user=%s -> %s session=%d xudp=%v", nativeMuxNetworkName(meta.network), email, target, meta.sessionID, isXUDP)
|
||||
ib2, host2, port2 := ib, targetHost, targetPort
|
||||
xrayGo(fmt.Sprintf("native xray mux session=%d", s.id), func() { s.run(ib2, host2, port2) })
|
||||
if len(pkt.payload) > 0 {
|
||||
s.enqueueUplink(pkt.payload, pkt.host, pkt.port)
|
||||
if len(pkt.payload) > 0 && !s.enqueueUplink(pkt.payload, pkt.host, pkt.port) {
|
||||
closeSession(s.id)
|
||||
writeMu.Lock()
|
||||
_ = writeNativeMuxEnd(stream, meta.sessionID, true)
|
||||
writeMu.Unlock()
|
||||
}
|
||||
|
||||
case nativeMuxStatusKeep:
|
||||
@@ -319,8 +393,11 @@ func (ib *nativeInbound) nativeVLESSMuxTunnel(stream io.ReadWriteCloser, uuid, e
|
||||
pkt.host = meta.host
|
||||
pkt.port = meta.port
|
||||
}
|
||||
if len(pkt.payload) > 0 {
|
||||
s.enqueueUplink(pkt.payload, pkt.host, pkt.port)
|
||||
if len(pkt.payload) > 0 && !s.enqueueUplink(pkt.payload, pkt.host, pkt.port) {
|
||||
closeSession(s.id)
|
||||
writeMu.Lock()
|
||||
_ = writeNativeMuxEnd(stream, meta.sessionID, true)
|
||||
writeMu.Unlock()
|
||||
}
|
||||
|
||||
default:
|
||||
@@ -332,7 +409,7 @@ func (ib *nativeInbound) nativeVLESSMuxTunnel(stream io.ReadWriteCloser, uuid, e
|
||||
}
|
||||
}
|
||||
|
||||
func (ib *nativeInbound) newNativeMuxSession(id uint16, network byte, host string, port uint16, xudp bool, globalID [8]byte, client io.Writer, writeMu *sync.Mutex, uuid, email string, onClose func(*nativeMuxSession)) (*nativeMuxSession, string, error) {
|
||||
func (ib *nativeInbound) newNativeMuxSession(id uint16, network byte, host string, port uint16, xudp bool, globalID [8]byte, client io.Writer, writeMu *sync.Mutex, uuid, email string, quotaState *xrayNativeQuotaState, onClose func(*nativeMuxSession)) (*nativeMuxSession, string, error) {
|
||||
target := net.JoinHostPort(normalizeNativeTargetHost(host), strconv.Itoa(int(port)))
|
||||
if invalidNativeDestination(host, port) {
|
||||
return nil, target, fmt.Errorf("invalid destination")
|
||||
@@ -351,8 +428,8 @@ func (ib *nativeInbound) newNativeMuxSession(id uint16, network byte, host strin
|
||||
email: email,
|
||||
upLimiter: ib.upLimiter(),
|
||||
downLimiter: ib.downLimiter(),
|
||||
upMeter: &trafficMeter{uuid: uuid, email: email, uplink: true},
|
||||
downMeter: &trafficMeter{uuid: uuid, email: email, uplink: false},
|
||||
upMeter: newTrafficMeter(uuid, email, true, quotaState),
|
||||
downMeter: newTrafficMeter(uuid, email, false, quotaState),
|
||||
uplink: make(chan nativeMuxUplinkItem, nativeMuxUplinkQueue),
|
||||
closed: make(chan struct{}),
|
||||
onClose: onClose,
|
||||
@@ -365,6 +442,7 @@ func (ib *nativeInbound) newNativeMuxSession(id uint16, network byte, host strin
|
||||
|
||||
func (s *nativeMuxSession) run(ib *nativeInbound, host string, port uint16) {
|
||||
defer xrayRecover(fmt.Sprintf("native xray mux run session=%d", s.id))
|
||||
defer s.finish()
|
||||
|
||||
select {
|
||||
case <-s.closed:
|
||||
@@ -410,24 +488,61 @@ func (s *nativeMuxSession) failInit(notifyClient bool) {
|
||||
_ = writeNativeMuxEnd(s.client, s.id, true)
|
||||
s.writeMu.Unlock()
|
||||
}
|
||||
if s.onClose != nil {
|
||||
s.onClose(s)
|
||||
}
|
||||
s.closeBackend()
|
||||
s.finish()
|
||||
}
|
||||
|
||||
func (s *nativeMuxSession) enqueueUplink(payload []byte, host string, port uint16) {
|
||||
// finish is the single lifecycle exit for a mux child. The backend reader,
|
||||
// uplink loop, parent mux stream, and initialization path can all detect the
|
||||
// terminal condition concurrently, so both cleanup and map removal must be
|
||||
// exactly-once operations.
|
||||
func (s *nativeMuxSession) finish() {
|
||||
s.finishOnce.Do(func() {
|
||||
s.closeBackend()
|
||||
if s.onClose != nil {
|
||||
s.onClose(s)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func (s *nativeMuxSession) beginEnqueue() bool {
|
||||
s.enqueueMu.Lock()
|
||||
defer s.enqueueMu.Unlock()
|
||||
if s.enqueueDone {
|
||||
return false
|
||||
}
|
||||
s.enqueueWG.Add(1)
|
||||
return true
|
||||
}
|
||||
|
||||
func (s *nativeMuxSession) enqueueUplink(payload []byte, host string, port uint16) bool {
|
||||
if len(payload) == 0 {
|
||||
return
|
||||
return true
|
||||
}
|
||||
if !s.beginEnqueue() {
|
||||
return false
|
||||
}
|
||||
defer s.enqueueWG.Done()
|
||||
|
||||
bytes := int64(len(payload))
|
||||
if !reserveNativeMuxBufferedBytes(s, bytes) {
|
||||
return false
|
||||
}
|
||||
cp := make([]byte, len(payload))
|
||||
copy(cp, payload)
|
||||
select {
|
||||
case s.uplink <- nativeMuxUplinkItem{payload: cp, host: host, port: port}:
|
||||
return true
|
||||
case <-s.closed:
|
||||
releaseNativeMuxBufferedBytes(s, bytes)
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
func (s *nativeMuxSession) processUplinkItem(item nativeMuxUplinkItem) bool {
|
||||
defer releaseNativeMuxBufferedBytes(s, int64(len(item.payload)))
|
||||
return s.writeBackendItem(item)
|
||||
}
|
||||
|
||||
func (s *nativeMuxSession) uplinkLoop() {
|
||||
defer s.upMeter.flush()
|
||||
for {
|
||||
@@ -435,7 +550,7 @@ func (s *nativeMuxSession) uplinkLoop() {
|
||||
case <-s.closed:
|
||||
return
|
||||
case item := <-s.uplink:
|
||||
if !s.writeBackendItem(item) {
|
||||
if !s.processUplinkItem(item) {
|
||||
s.closeBackend()
|
||||
return
|
||||
}
|
||||
@@ -481,6 +596,13 @@ func (s *nativeMuxSession) writeBackendItem(item nativeMuxUplinkItem) bool {
|
||||
return false
|
||||
}
|
||||
}
|
||||
quotaReservation, quotaErr := reserveNativePacketQuota(s.upMeter, len(payload))
|
||||
if quotaErr != nil {
|
||||
return false
|
||||
}
|
||||
if err := quotaReservation.wait(s.ctx); err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
var n int
|
||||
var err error
|
||||
@@ -492,6 +614,7 @@ func (s *nativeMuxSession) writeBackendItem(item nativeMuxUplinkItem) bool {
|
||||
if isNativeDNSSinkTarget(item.host) || invalidNativeDestination(item.host, item.port) {
|
||||
// AdGuard/blocked endpoints must be ignored at the cheapest possible
|
||||
// point. Do not resolve, dial, log loudly, or keep the mux child busy.
|
||||
quotaReservation.finish(0)
|
||||
xrayTracef("native xray: VLESS mux UDP fast-ignored override sink session=%d target=%s:%d", s.id, item.host, item.port)
|
||||
return true
|
||||
}
|
||||
@@ -503,6 +626,7 @@ func (s *nativeMuxSession) writeBackendItem(item nativeMuxUplinkItem) bool {
|
||||
s.lastUDPPort = item.port
|
||||
s.lastUDPAddr = addr
|
||||
} else {
|
||||
quotaReservation.finish(0)
|
||||
xrayTracef("native xray: VLESS mux UDP override resolve failed session=%d target=%s:%d: %v", s.id, item.host, item.port, rerr)
|
||||
return true
|
||||
}
|
||||
@@ -512,9 +636,7 @@ func (s *nativeMuxSession) writeBackendItem(item nativeMuxUplinkItem) bool {
|
||||
if s.network == nativeMuxNetworkUDP && err == nil {
|
||||
_ = s.udp.SetReadDeadline(time.Now().Add(nativeMuxUDPIdleTimeout()))
|
||||
}
|
||||
if n > 0 {
|
||||
s.upMeter.add(n)
|
||||
}
|
||||
quotaReservation.finish(n)
|
||||
if err != nil {
|
||||
xrayLogf("native xray: VLESS mux backend write failed session=%d: %v", s.id, err)
|
||||
return false
|
||||
@@ -532,10 +654,7 @@ func (s *nativeMuxSession) readBackendLoop() {
|
||||
_ = writeNativeMuxEnd(s.client, s.id, false)
|
||||
s.writeMu.Unlock()
|
||||
}
|
||||
if s.onClose != nil {
|
||||
s.onClose(s)
|
||||
}
|
||||
s.closeBackend()
|
||||
s.finish()
|
||||
}()
|
||||
|
||||
if s.network == nativeMuxNetworkTCP {
|
||||
@@ -571,14 +690,22 @@ func (s *nativeMuxSession) readTCPBackendLoop() {
|
||||
if err := s.waitDownRate(n); err != nil {
|
||||
return
|
||||
}
|
||||
s.downMeter.add(n)
|
||||
quotaReservation, quotaErr := reserveNativePacketQuota(s.downMeter, n)
|
||||
if quotaErr != nil {
|
||||
return
|
||||
}
|
||||
if err := quotaReservation.wait(s.ctx); err != nil {
|
||||
return
|
||||
}
|
||||
s.writeMu.Lock()
|
||||
werr := writeNativeMuxData(s.client, s.id, nativeMuxStatusKeep, buf[:n])
|
||||
s.writeMu.Unlock()
|
||||
if werr != nil {
|
||||
quotaReservation.finish(0)
|
||||
xrayLogf("native xray: VLESS mux TCP client write failed session=%d: %v", s.id, werr)
|
||||
return
|
||||
}
|
||||
quotaReservation.finish(n)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -602,7 +729,13 @@ func (s *nativeMuxSession) readUDPBackendLoop() bool {
|
||||
if err := s.waitDownRate(n); err != nil {
|
||||
return true
|
||||
}
|
||||
s.downMeter.add(n)
|
||||
quotaReservation, quotaErr := reserveNativePacketQuota(s.downMeter, n)
|
||||
if quotaErr != nil {
|
||||
return true
|
||||
}
|
||||
if err := quotaReservation.wait(s.ctx); err != nil {
|
||||
return true
|
||||
}
|
||||
s.writeMu.Lock()
|
||||
// Include the UDP source endpoint on XUDP responses so clients that rely on
|
||||
// full-cone packet addressing can associate the datagram with the correct
|
||||
@@ -610,27 +743,46 @@ func (s *nativeMuxSession) readUDPBackendLoop() bool {
|
||||
werr := writeNativeMuxPacketData(s.client, s.id, nativeMuxStatusKeep, buf[:n], addr, s.xudp)
|
||||
s.writeMu.Unlock()
|
||||
if werr != nil {
|
||||
quotaReservation.finish(0)
|
||||
xrayLogf("native xray: VLESS mux UDP client write failed session=%d: %v", s.id, werr)
|
||||
return true
|
||||
}
|
||||
quotaReservation.finish(n)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *nativeMuxSession) closeBackend() {
|
||||
s.closeOnce.Do(func() {
|
||||
s.enqueueMu.Lock()
|
||||
s.enqueueDone = true
|
||||
close(s.closed)
|
||||
s.enqueueMu.Unlock()
|
||||
if s.cancel != nil {
|
||||
s.cancel()
|
||||
}
|
||||
if s.releaseSlot != nil {
|
||||
s.releaseSlot()
|
||||
}
|
||||
if s.tcp != nil {
|
||||
_ = s.tcp.Close()
|
||||
}
|
||||
if s.udp != nil {
|
||||
_ = s.udp.Close()
|
||||
}
|
||||
|
||||
// Wait for producers that passed beginEnqueue before the close flag, then
|
||||
// discard any payloads the consumer did not take. This returns every byte
|
||||
// reservation even when shutdown races a full queue.
|
||||
s.enqueueWG.Wait()
|
||||
for {
|
||||
select {
|
||||
case item := <-s.uplink:
|
||||
releaseNativeMuxBufferedBytes(s, int64(len(item.payload)))
|
||||
item.payload = nil
|
||||
default:
|
||||
if s.releaseSlot != nil {
|
||||
s.releaseSlot()
|
||||
}
|
||||
return
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
|
||||
+257
-1
@@ -1,6 +1,14 @@
|
||||
package main
|
||||
|
||||
import "runtime/debug"
|
||||
import (
|
||||
"net"
|
||||
"runtime/debug"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
)
|
||||
|
||||
const nativeOverloadBackoff = 10 * time.Millisecond
|
||||
|
||||
// xrayRecover prevents a bad client packet, closed HTTP stream, or mux/session
|
||||
// race from taking down the whole sshpanel process. A panic should only kill the
|
||||
@@ -18,3 +26,251 @@ func xrayGo(where string, fn func()) {
|
||||
fn()
|
||||
}()
|
||||
}
|
||||
|
||||
func init() {
|
||||
// Direct native-inbound tests and embedders may run an accept loop without
|
||||
// the singleton server start method. Production stop() flips this to false.
|
||||
nativeTransportAccepting.Store(true)
|
||||
}
|
||||
|
||||
var (
|
||||
nativeTransportConnections atomic.Int64
|
||||
nativeXHTTPSessions atomic.Int64
|
||||
nativeXHTTPRequests atomic.Int64
|
||||
nativeClientConnsRejected atomic.Int64
|
||||
nativePreAuthRejected atomic.Int64
|
||||
nativeXHTTPRejected atomic.Int64
|
||||
|
||||
nativeTransportAccepting atomic.Bool
|
||||
nativeTransportRegistry = struct {
|
||||
sync.Mutex
|
||||
conns map[*nativeCountedConn]struct{}
|
||||
}{conns: make(map[*nativeCountedConn]struct{})}
|
||||
)
|
||||
|
||||
// acquireNativeCounter tracks a counted resource and returns an exactly-once
|
||||
// release function. Limits here are simultaneous resource-safety windows, not
|
||||
// traffic-volume or request-rate ceilings.
|
||||
func acquireNativeCounter(active *atomic.Int64, limit int) (func(), bool) {
|
||||
for {
|
||||
current := active.Load()
|
||||
if limit > 0 && current >= int64(limit) {
|
||||
return nil, false
|
||||
}
|
||||
if active.CompareAndSwap(current, current+1) {
|
||||
var once sync.Once
|
||||
return func() {
|
||||
once.Do(func() {
|
||||
if active.Add(-1) < 0 {
|
||||
active.Store(0)
|
||||
}
|
||||
})
|
||||
}, true
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func shouldLogNativeSample(counter *atomic.Int64) (n int64, ok bool) {
|
||||
n = counter.Add(1)
|
||||
// Keep attacks visible without allowing logging itself to become a CPU/disk
|
||||
// amplifier. The first event and one event per 1024 repetitions are logged.
|
||||
return n, n == 1 || n%1024 == 0
|
||||
}
|
||||
|
||||
func logNativeLimitRejection(kind string, rejected *atomic.Int64, limit int) {
|
||||
n, ok := shouldLogNativeSample(rejected)
|
||||
if ok {
|
||||
xrayLogf("native xray: rejected %s at safety limit=%d (rejected=%d)", kind, limit, n)
|
||||
}
|
||||
}
|
||||
|
||||
func logNativeClientLimitRejection(email string, limit int) {
|
||||
n, ok := shouldLogNativeSample(&nativeClientConnsRejected)
|
||||
if ok {
|
||||
xrayLogf("native xray: rejected authenticated user %s at max_conns=%d (rejected=%d)", email, limit, n)
|
||||
}
|
||||
}
|
||||
|
||||
func logNativePreAuthRejection(format string, args ...interface{}) {
|
||||
if _, ok := shouldLogNativeSample(&nativePreAuthRejected); ok {
|
||||
xrayLogf(format, args...)
|
||||
}
|
||||
}
|
||||
|
||||
func acquireNativeTransportConnection() (func(), bool) {
|
||||
return acquireNativeCounter(&nativeTransportConnections, nativeTransportConnectionLimit())
|
||||
}
|
||||
|
||||
func acquireNativeXHTTPSession() (func(), bool) {
|
||||
return acquireNativeCounter(&nativeXHTTPSessions, nativeXHTTPSessionLimit())
|
||||
}
|
||||
|
||||
func acquireNativeXHTTPRequest() (func(), bool) {
|
||||
return acquireNativeCounter(&nativeXHTTPRequests, nativeXHTTPRequestLimit())
|
||||
}
|
||||
|
||||
func configureNativeTransportSocket(c net.Conn) {
|
||||
if tc, ok := c.(*net.TCPConn); ok {
|
||||
_ = tc.SetKeepAlive(true)
|
||||
_ = tc.SetKeepAlivePeriod(30 * time.Second)
|
||||
_ = tc.SetNoDelay(true)
|
||||
}
|
||||
}
|
||||
|
||||
// nativeCountedConn releases its global transport slot and unregisters itself
|
||||
// exactly once, even when several tunnel paths race to close the same socket.
|
||||
type nativeCountedConn struct {
|
||||
net.Conn
|
||||
release func()
|
||||
onClose func()
|
||||
closeOnce sync.Once
|
||||
closeErr error
|
||||
}
|
||||
|
||||
func (c *nativeCountedConn) Close() error {
|
||||
c.closeOnce.Do(func() {
|
||||
c.closeErr = c.Conn.Close()
|
||||
if c.release != nil {
|
||||
c.release()
|
||||
}
|
||||
if c.onClose != nil {
|
||||
c.onClose()
|
||||
}
|
||||
})
|
||||
return c.closeErr
|
||||
}
|
||||
|
||||
// wrapNativeTransportConn applies only the global counter. It is useful for
|
||||
// focused tests and for callers that own connection lifetime themselves.
|
||||
func wrapNativeTransportConn(c net.Conn) (net.Conn, bool) {
|
||||
if c == nil {
|
||||
return nil, false
|
||||
}
|
||||
configureNativeTransportSocket(c)
|
||||
release, ok := acquireNativeTransportConnection()
|
||||
if !ok {
|
||||
_ = c.Close()
|
||||
return nil, false
|
||||
}
|
||||
return &nativeCountedConn{Conn: c, release: release}, true
|
||||
}
|
||||
|
||||
// wrapTrackedNativeTransportConn additionally registers the accepted socket so
|
||||
// stopping/restarting native Xray closes established raw, WebSocket, TLS, HTTP/1
|
||||
// and HTTP/2 transports instead of leaving tunnel goroutines alive.
|
||||
func wrapTrackedNativeTransportConn(c net.Conn) (net.Conn, bool) {
|
||||
if c == nil {
|
||||
return nil, false
|
||||
}
|
||||
configureNativeTransportSocket(c)
|
||||
if !nativeTransportAccepting.Load() {
|
||||
_ = c.Close()
|
||||
return nil, false
|
||||
}
|
||||
release, ok := acquireNativeTransportConnection()
|
||||
if !ok {
|
||||
_ = c.Close()
|
||||
return nil, false
|
||||
}
|
||||
return registerTrackedNativeTransportConn(c, release)
|
||||
}
|
||||
|
||||
// registerTrackedNativeTransportConn finishes registration when the caller has
|
||||
// already reserved a transport slot. Keeping reservation and Accept separate is
|
||||
// what lets the production listener apply kernel/socket backpressure instead of
|
||||
// accepting and immediately resetting connections at capacity.
|
||||
func registerTrackedNativeTransportConn(c net.Conn, release func()) (net.Conn, bool) {
|
||||
counted := &nativeCountedConn{Conn: c, release: release}
|
||||
counted.onClose = func() {
|
||||
nativeTransportRegistry.Lock()
|
||||
delete(nativeTransportRegistry.conns, counted)
|
||||
nativeTransportRegistry.Unlock()
|
||||
}
|
||||
|
||||
nativeTransportRegistry.Lock()
|
||||
if !nativeTransportAccepting.Load() {
|
||||
nativeTransportRegistry.Unlock()
|
||||
_ = counted.Close()
|
||||
return nil, false
|
||||
}
|
||||
nativeTransportRegistry.conns[counted] = struct{}{}
|
||||
nativeTransportRegistry.Unlock()
|
||||
return counted, true
|
||||
}
|
||||
|
||||
// waitWrapTrackedNativeTransportConn is used by raw native accept loops. Waiting
|
||||
// here, before another connection is admitted to the protocol handler, applies
|
||||
// socket/kernel backpressure instead of creating an unbounded goroutine backlog.
|
||||
func waitWrapTrackedNativeTransportConn(c net.Conn) (net.Conn, bool) {
|
||||
if c == nil {
|
||||
return nil, false
|
||||
}
|
||||
configureNativeTransportSocket(c)
|
||||
for nativeTransportAccepting.Load() {
|
||||
release, ok := acquireNativeTransportConnection()
|
||||
if ok {
|
||||
return registerTrackedNativeTransportConn(c, release)
|
||||
}
|
||||
time.Sleep(nativeOverloadBackoff)
|
||||
}
|
||||
_ = c.Close()
|
||||
return nil, false
|
||||
}
|
||||
|
||||
func beginNativeTransportAccepting() {
|
||||
nativeTransportAccepting.Store(true)
|
||||
}
|
||||
|
||||
func stopNativeTransportAccepting() {
|
||||
nativeTransportAccepting.Store(false)
|
||||
}
|
||||
|
||||
func closeAllNativeTransportConnections() {
|
||||
nativeTransportRegistry.Lock()
|
||||
conns := make([]*nativeCountedConn, 0, len(nativeTransportRegistry.conns))
|
||||
for c := range nativeTransportRegistry.conns {
|
||||
conns = append(conns, c)
|
||||
}
|
||||
nativeTransportRegistry.Unlock()
|
||||
for _, c := range conns {
|
||||
_ = c.Close()
|
||||
}
|
||||
}
|
||||
|
||||
// nativeTrackingListener registers every accepted XHTTP socket so a live
|
||||
// stop/reload can close it. It reserves capacity before Accept so overload stays
|
||||
// in the kernel accept queue rather than allocating more Go handlers.
|
||||
type nativeTrackingListener struct {
|
||||
net.Listener
|
||||
}
|
||||
|
||||
func (l nativeTrackingListener) Accept() (net.Conn, error) {
|
||||
for {
|
||||
release, ok := acquireNativeTransportConnection()
|
||||
if !ok {
|
||||
// Unlimited admission can only fail if this implementation changes. Avoid
|
||||
// accepting and resetting a socket if that ever happens.
|
||||
time.Sleep(nativeOverloadBackoff)
|
||||
continue
|
||||
}
|
||||
c, err := l.Listener.Accept()
|
||||
if err != nil {
|
||||
release()
|
||||
return nil, err
|
||||
}
|
||||
configureNativeTransportSocket(c)
|
||||
if counted, ok := registerTrackedNativeTransportConn(c, release); ok {
|
||||
return counted, nil
|
||||
}
|
||||
// Shutdown may race Accept. Registration closes the socket and releases the
|
||||
// slot; the next Accept observes the listener close.
|
||||
time.Sleep(nativeOverloadBackoff)
|
||||
}
|
||||
}
|
||||
|
||||
func trackNativeListener(ln net.Listener) net.Listener {
|
||||
if ln == nil {
|
||||
return nil
|
||||
}
|
||||
return nativeTrackingListener{Listener: ln}
|
||||
}
|
||||
|
||||
+63
-13
@@ -7,29 +7,65 @@ import (
|
||||
)
|
||||
|
||||
type XrayNativeTuning struct {
|
||||
RuntimeGOMAXPROCS int `json:"runtime_gomaxprocs,omitempty"`
|
||||
MuxGlobalSessions int `json:"mux_global_sessions,omitempty"`
|
||||
TracePackets bool `json:"trace_packets,omitempty"`
|
||||
RuntimeGOMAXPROCS int `json:"runtime_gomaxprocs,omitempty"`
|
||||
MuxGlobalSessions int `json:"mux_global_sessions,omitempty"`
|
||||
MaxConcurrentConnections int `json:"max_concurrent_connections,omitempty"`
|
||||
MaxConcurrentXHTTPRequests int `json:"max_concurrent_xhttp_requests,omitempty"`
|
||||
XHTTPMaxSessions int `json:"xhttp_max_sessions,omitempty"`
|
||||
TracePackets bool `json:"trace_packets,omitempty"`
|
||||
}
|
||||
|
||||
const (
|
||||
defaultNativeRuntimeGOMAXPROCS = 0
|
||||
defaultNativeMuxGlobalSessions = 32768
|
||||
// Transport sockets, XHTTP requests, and XHTTP sessions are VPN traffic, not
|
||||
// website requests. Keep the legacy JSON fields for config compatibility, but
|
||||
// always normalize them to unlimited. Actual resource protection is provided by
|
||||
// socket/HTTP flow control and the bounded byte queues in xray_xhttp.go.
|
||||
defaultNativeMaxConnections = -1
|
||||
defaultNativeMaxXHTTPRequests = -1
|
||||
|
||||
fixedNativeMuxMaxSessions = 128
|
||||
fixedNativeMuxMaxSessions = 64
|
||||
fixedNativeMuxUDPIdleMS = 120000
|
||||
fixedNativeMuxUDPReadBuffer = 256 * 1024
|
||||
fixedNativeMuxUDPWriteBuffer = 256 * 1024
|
||||
|
||||
defaultNativeXHTTPMaxSessions = 16384
|
||||
defaultNativeXHTTPBufferedPosts = 512
|
||||
defaultNativeXHTTPMaxSessions = -1
|
||||
// Packet-up posts are also protected by byte budgets in xray_xhttp.go. Keep
|
||||
// the default reorder queue modest so thousands of unauthenticated sessions
|
||||
// cannot consume large amounts of memory merely by allocating empty channel
|
||||
// buffers. Operators may request more, up to the hard cap enforced there.
|
||||
defaultNativeXHTTPBufferedPosts = 64
|
||||
|
||||
// Do not impose an application-level lifetime on a connected XHTTP VPN
|
||||
// session. The official Xray server keeps a connected session for the
|
||||
// lifetime of its stream-down GET; request cancellation and I/O errors own
|
||||
// cleanup. A fixed five-minute sweeper incorrectly killed healthy but idle
|
||||
// VPNs. Zero disables the connected-session sweeper.
|
||||
fixedNativeXHTTPIdleMS = 0
|
||||
// These are simultaneous resource-safety windows, not request-rate limits.
|
||||
// They are deliberately far above the expected 6-8K connected-user load, but
|
||||
// finite so a reconnect storm, broken CDN, or hostile client cannot retain an
|
||||
// unbounded number of sockets, HTTP handlers, sessions, and goroutine stacks.
|
||||
// Transport Accept waits at capacity (kernel backpressure); XHTTP overloads
|
||||
// receive 503 rather than the web-rate-limit semantics of 429.
|
||||
fixedNativeMaxTransportConnections = 65536
|
||||
fixedNativeMaxXHTTPRequests = 65536
|
||||
fixedNativeMaxXHTTPSessions = 65536
|
||||
fixedNativeHTTP2ConcurrentStreams = 4096
|
||||
fixedNativeXHTTPWriteTimeoutMS = 60 * 1000
|
||||
|
||||
// Backstop reaper for connected XHTTP VPN sessions. The stream-down GET's
|
||||
// request context is the primary lifetime owner, but behind a CDN that context
|
||||
// frequently never fires when a client silently drops (mobile networks, CDN
|
||||
// connection pooling, half-open TCP). When it doesn't, an idle SSH backend
|
||||
// never errors either, so the session, its goroutines, socket/fd, and SSH
|
||||
// connection leak until the whole process restarts. That accumulation is what
|
||||
// drove the recurring XHTTP 502s that only a reboot cleared: the origin slowly
|
||||
// ran out of fds/memory and could no longer serve new stream-down GETs.
|
||||
//
|
||||
// This sweeper only ever reaps sessions with genuinely stale lastSeen. lastSeen
|
||||
// is refreshed on every successful read OR write via nativeXHTTPConn.onActivity,
|
||||
// so any tunnel still passing data or keepalives is never touched -- only a
|
||||
// session with zero bytes in BOTH directions for the full window (i.e. one that
|
||||
// looks dead) is closed. 20 minutes is generous enough not to disturb a
|
||||
// genuinely idle-but-alive tunnel while still bounding resource growth under
|
||||
// heavy 6-8K-user churn. Zero disables the connected-session sweeper.
|
||||
fixedNativeXHTTPIdleMS = 20 * 60 * 1000
|
||||
)
|
||||
|
||||
var (
|
||||
@@ -53,6 +89,12 @@ func normalizeNativeXrayTuning(t *XrayNativeTuning) XrayNativeTuning {
|
||||
if out.MuxGlobalSessions <= 0 {
|
||||
out.MuxGlobalSessions = defaultNativeMuxGlobalSessions
|
||||
}
|
||||
// Ignore every old positive/zero admission ceiling. This migration is
|
||||
// deliberately unconditional so upgrading an existing server immediately
|
||||
// removes the old 4K/8K/32K web-style caps without requiring a panel save.
|
||||
out.MaxConcurrentConnections = defaultNativeMaxConnections
|
||||
out.MaxConcurrentXHTTPRequests = defaultNativeMaxXHTTPRequests
|
||||
out.XHTTPMaxSessions = defaultNativeXHTTPMaxSessions
|
||||
return out
|
||||
}
|
||||
|
||||
@@ -79,8 +121,16 @@ func nativeTracePacketsEnabled() bool { return nativeTuneTracePackets.Load() }
|
||||
func nativeMuxMaxSessionLimit() int { return fixedNativeMuxMaxSessions }
|
||||
func nativeMuxUDPReadBufferSize() int { return fixedNativeMuxUDPReadBuffer }
|
||||
func nativeMuxUDPWriteBufferSize() int { return fixedNativeMuxUDPWriteBuffer }
|
||||
func nativeXHTTPMaxSessionLimit() int { return defaultNativeXHTTPMaxSessions }
|
||||
func nativeXHTTPBufferedPostLimit() int { return defaultNativeXHTTPBufferedPosts }
|
||||
func nativeHTTP2MaxConcurrentStreams() uint32 {
|
||||
return fixedNativeHTTP2ConcurrentStreams
|
||||
}
|
||||
func nativeTransportConnectionLimit() int { return fixedNativeMaxTransportConnections }
|
||||
func nativeXHTTPRequestLimit() int { return fixedNativeMaxXHTTPRequests }
|
||||
func nativeXHTTPSessionLimit() int { return fixedNativeMaxXHTTPSessions }
|
||||
func nativeXHTTPWriteTimeout() time.Duration {
|
||||
return fixedNativeXHTTPWriteTimeoutMS * time.Millisecond
|
||||
}
|
||||
func nativeMuxUDPIdleTimeout() time.Duration {
|
||||
return fixedNativeMuxUDPIdleMS * time.Millisecond
|
||||
}
|
||||
|
||||
+53
-24
@@ -25,18 +25,18 @@ const (
|
||||
// check and caused the server to block waiting for a fake second payload.
|
||||
// XUDP belongs to VLESS CommandMux and is handled separately when Mux support
|
||||
// is implemented.
|
||||
func nativeVLESSUDPTunnel(client io.ReadWriteCloser, backend net.Conn, uuid, email string, up, down *rate.Limiter) {
|
||||
xrayMgr.recordNativeConnect(uuid, email)
|
||||
defer xrayMgr.recordNativeDisconnect(uuid, email)
|
||||
func nativeVLESSUDPTunnel(client io.ReadWriteCloser, backend net.Conn, uuid, email string, quotaState *xrayNativeQuotaState, up, down *rate.Limiter) {
|
||||
defer xrayRecover(fmt.Sprintf("native xray VLESS UDP tunnel user=%s", email))
|
||||
|
||||
upMeter := &trafficMeter{uuid: uuid, email: email, uplink: true}
|
||||
downMeter := &trafficMeter{uuid: uuid, email: email, uplink: false}
|
||||
upMeter := newTrafficMeter(uuid, email, true, quotaState)
|
||||
downMeter := newTrafficMeter(uuid, email, false, quotaState)
|
||||
|
||||
var wg sync.WaitGroup
|
||||
var closeOnce sync.Once
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
closeAll := func() {
|
||||
closeOnce.Do(func() {
|
||||
cancel()
|
||||
_ = backend.Close()
|
||||
_ = client.Close()
|
||||
})
|
||||
@@ -57,13 +57,18 @@ func nativeVLESSUDPTunnel(client io.ReadWriteCloser, backend net.Conn, uuid, ema
|
||||
if len(payload) == 0 {
|
||||
continue
|
||||
}
|
||||
if err := waitNativeRate(up, len(payload)); err != nil {
|
||||
if err := waitNativeRate(ctx, up, len(payload)); err != nil {
|
||||
return
|
||||
}
|
||||
quotaReservation, err := reserveNativePacketQuota(upMeter, len(payload))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if err := quotaReservation.wait(ctx); err != nil {
|
||||
return
|
||||
}
|
||||
n, err := backend.Write(payload)
|
||||
if n > 0 {
|
||||
upMeter.add(n)
|
||||
}
|
||||
quotaReservation.finish(n)
|
||||
if err != nil {
|
||||
xrayLogf("native xray: VLESS UDP backend write failed: %v", err)
|
||||
return
|
||||
@@ -91,14 +96,22 @@ func nativeVLESSUDPTunnel(client io.ReadWriteCloser, backend net.Conn, uuid, ema
|
||||
if n <= 0 {
|
||||
continue
|
||||
}
|
||||
if err := waitNativeRate(down, n); err != nil {
|
||||
if err := waitNativeRate(ctx, down, n); err != nil {
|
||||
return
|
||||
}
|
||||
quotaReservation, err := reserveNativePacketQuota(downMeter, n)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if err := quotaReservation.wait(ctx); err != nil {
|
||||
return
|
||||
}
|
||||
if err := writeVLESSLengthPacket(client, buf[:n]); err != nil {
|
||||
quotaReservation.finish(0)
|
||||
xrayLogf("native xray: VLESS UDP client write failed: %v", err)
|
||||
return
|
||||
}
|
||||
downMeter.add(n)
|
||||
quotaReservation.finish(n)
|
||||
}
|
||||
})
|
||||
|
||||
@@ -301,18 +314,18 @@ func writeVLESSXUDPPacket(w io.Writer, payload []byte) error {
|
||||
// nativeVMessUDPTunnel maps one VMess body chunk to one UDP datagram. VMess AEAD
|
||||
// chunking already preserves packet boundaries, so no extra VLESS length prefix
|
||||
// is added inside the encrypted body.
|
||||
func nativeVMessUDPTunnel(client nativeVMessStream, backend net.Conn, uuid, email string, up, down *rate.Limiter) {
|
||||
xrayMgr.recordNativeConnect(uuid, email)
|
||||
defer xrayMgr.recordNativeDisconnect(uuid, email)
|
||||
func nativeVMessUDPTunnel(client nativeVMessStream, backend net.Conn, uuid, email string, quotaState *xrayNativeQuotaState, up, down *rate.Limiter) {
|
||||
defer xrayRecover(fmt.Sprintf("native xray VMess UDP tunnel user=%s", email))
|
||||
|
||||
upMeter := &trafficMeter{uuid: uuid, email: email, uplink: true}
|
||||
downMeter := &trafficMeter{uuid: uuid, email: email, uplink: false}
|
||||
upMeter := newTrafficMeter(uuid, email, true, quotaState)
|
||||
downMeter := newTrafficMeter(uuid, email, false, quotaState)
|
||||
|
||||
var wg sync.WaitGroup
|
||||
var closeOnce sync.Once
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
closeAll := func() {
|
||||
closeOnce.Do(func() {
|
||||
cancel()
|
||||
_ = backend.Close()
|
||||
_ = client.Close()
|
||||
})
|
||||
@@ -333,13 +346,18 @@ func nativeVMessUDPTunnel(client nativeVMessStream, backend net.Conn, uuid, emai
|
||||
if len(pkt) == 0 {
|
||||
continue
|
||||
}
|
||||
if err := waitNativeRate(up, len(pkt)); err != nil {
|
||||
if err := waitNativeRate(ctx, up, len(pkt)); err != nil {
|
||||
return
|
||||
}
|
||||
quotaReservation, err := reserveNativePacketQuota(upMeter, len(pkt))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if err := quotaReservation.wait(ctx); err != nil {
|
||||
return
|
||||
}
|
||||
n, err := backend.Write(pkt)
|
||||
if n > 0 {
|
||||
upMeter.add(n)
|
||||
}
|
||||
quotaReservation.finish(n)
|
||||
if err != nil {
|
||||
xrayLogf("native xray: VMess UDP backend write failed: %v", err)
|
||||
return
|
||||
@@ -367,14 +385,22 @@ func nativeVMessUDPTunnel(client nativeVMessStream, backend net.Conn, uuid, emai
|
||||
if n <= 0 {
|
||||
continue
|
||||
}
|
||||
if err := waitNativeRate(down, n); err != nil {
|
||||
if err := waitNativeRate(ctx, down, n); err != nil {
|
||||
return
|
||||
}
|
||||
quotaReservation, err := reserveNativePacketQuota(downMeter, n)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if err := quotaReservation.wait(ctx); err != nil {
|
||||
return
|
||||
}
|
||||
if err := client.WritePacket(buf[:n]); err != nil {
|
||||
quotaReservation.finish(0)
|
||||
xrayLogf("native xray: VMess UDP client write failed: %v", err)
|
||||
return
|
||||
}
|
||||
downMeter.add(n)
|
||||
quotaReservation.finish(n)
|
||||
}
|
||||
})
|
||||
|
||||
@@ -384,9 +410,12 @@ func nativeVMessUDPTunnel(client nativeVMessStream, backend net.Conn, uuid, emai
|
||||
closeAll()
|
||||
}
|
||||
|
||||
func waitNativeRate(lim *rate.Limiter, n int) error {
|
||||
func waitNativeRate(ctx context.Context, lim *rate.Limiter, n int) error {
|
||||
if lim == nil || n <= 0 {
|
||||
return nil
|
||||
}
|
||||
return lim.WaitN(context.Background(), n)
|
||||
if ctx == nil {
|
||||
ctx = context.Background()
|
||||
}
|
||||
return lim.WaitN(ctx, n)
|
||||
}
|
||||
|
||||
+513
@@ -0,0 +1,513 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"io"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"golang.org/x/time/rate"
|
||||
)
|
||||
|
||||
type xrayNativeQuotaState struct {
|
||||
// trafficMu establishes a clean reset boundary. Native stream and packet
|
||||
// writers hold a read lock from quota reservation through the actual write
|
||||
// and metering; traffic resets take the write lock. This prevents an
|
||||
// in-flight pre-reset reservation from being accounted in the new period or
|
||||
// subtracting from freshly reset usage.
|
||||
trafficMu sync.RWMutex
|
||||
mu sync.Mutex
|
||||
usedBytes int64
|
||||
quotaBytes int64
|
||||
action string
|
||||
throttleMbps int
|
||||
limiter *rate.Limiter
|
||||
generation uint64
|
||||
maxConns int
|
||||
activeConns int
|
||||
owner string
|
||||
expiresAt time.Time
|
||||
hasExpiry bool
|
||||
connections map[io.Closer]struct{}
|
||||
}
|
||||
|
||||
func (m *XrayManager) reloadNativeQuotaPolicies() {
|
||||
if statsStore == nil {
|
||||
return
|
||||
}
|
||||
metas, err := statsStore.ListAllXrayClients(context.Background())
|
||||
if err != nil {
|
||||
xrayLogf("xray native quota: load policies failed: %v", err)
|
||||
return
|
||||
}
|
||||
next := make(map[string]*xrayNativeQuotaState, len(metas))
|
||||
for _, meta := range metas {
|
||||
if meta == nil || strings.TrimSpace(meta.UUID) == "" {
|
||||
continue
|
||||
}
|
||||
next[meta.UUID] = newXrayNativeQuotaState(meta)
|
||||
}
|
||||
m.nativeQuotaMu.Lock()
|
||||
m.nativeQuotaByUUID = next
|
||||
m.nativeQuotaMu.Unlock()
|
||||
}
|
||||
|
||||
func newXrayNativeQuotaState(meta *XrayClientMeta) *xrayNativeQuotaState {
|
||||
used := meta.TotalUplinkBytes + meta.TotalDownlinkBytes
|
||||
if used < 0 {
|
||||
used = 0
|
||||
}
|
||||
return &xrayNativeQuotaState{
|
||||
usedBytes: used,
|
||||
quotaBytes: meta.DataQuotaBytes,
|
||||
action: normalizeQuotaAction(meta.QuotaAction),
|
||||
throttleMbps: quotaThrottleMbpsOrDefault(meta.QuotaThrottleMbps),
|
||||
generation: 1,
|
||||
maxConns: normalizeXrayMaxConns(meta.MaxConns),
|
||||
owner: strings.TrimSpace(meta.OwnerUsername),
|
||||
hasExpiry: meta.ExpiresAt != nil,
|
||||
expiresAt: xrayExpiryValue(meta.ExpiresAt),
|
||||
}
|
||||
}
|
||||
|
||||
func xrayExpiryValue(expiry *time.Time) time.Time {
|
||||
if expiry == nil {
|
||||
return time.Time{}
|
||||
}
|
||||
return *expiry
|
||||
}
|
||||
|
||||
func normalizeXrayMaxConns(v int) int {
|
||||
if v < 0 {
|
||||
return 0
|
||||
}
|
||||
return v
|
||||
}
|
||||
|
||||
func (m *XrayManager) setNativeQuotaPolicy(meta *XrayClientMeta) {
|
||||
if meta == nil || strings.TrimSpace(meta.UUID) == "" {
|
||||
return
|
||||
}
|
||||
uuid := strings.TrimSpace(meta.UUID)
|
||||
m.nativeQuotaMu.Lock()
|
||||
if m.nativeQuotaByUUID == nil {
|
||||
m.nativeQuotaByUUID = make(map[string]*xrayNativeQuotaState)
|
||||
}
|
||||
existing := m.nativeQuotaByUUID[uuid]
|
||||
if existing == nil {
|
||||
m.nativeQuotaByUUID[uuid] = newXrayNativeQuotaState(meta)
|
||||
m.nativeQuotaMu.Unlock()
|
||||
return
|
||||
}
|
||||
m.nativeQuotaMu.Unlock()
|
||||
|
||||
existing.mu.Lock()
|
||||
existing.quotaBytes = meta.DataQuotaBytes
|
||||
existing.action = normalizeQuotaAction(meta.QuotaAction)
|
||||
existing.throttleMbps = quotaThrottleMbpsOrDefault(meta.QuotaThrottleMbps)
|
||||
existing.maxConns = normalizeXrayMaxConns(meta.MaxConns)
|
||||
existing.owner = strings.TrimSpace(meta.OwnerUsername)
|
||||
existing.hasExpiry = meta.ExpiresAt != nil
|
||||
existing.expiresAt = xrayExpiryValue(meta.ExpiresAt)
|
||||
existing.limiter = nil
|
||||
existing.mu.Unlock()
|
||||
}
|
||||
|
||||
func (m *XrayManager) removeNativeQuotaPolicy(uuid string) {
|
||||
uuid = strings.TrimSpace(uuid)
|
||||
if uuid == "" {
|
||||
return
|
||||
}
|
||||
m.nativeQuotaMu.Lock()
|
||||
state := m.nativeQuotaByUUID[uuid]
|
||||
delete(m.nativeQuotaByUUID, uuid)
|
||||
m.nativeQuotaMu.Unlock()
|
||||
closeNativeClientConnections(state)
|
||||
|
||||
// Do not retain failed traffic/active deltas for a client that no longer
|
||||
// exists. This also bounds the pending maps during a prolonged DB outage.
|
||||
m.nativeDBMu.Lock()
|
||||
delete(m.nativeTrafficPending, uuid)
|
||||
delete(m.nativeActivePending, uuid)
|
||||
m.nativeDBMu.Unlock()
|
||||
}
|
||||
|
||||
func (m *XrayManager) resetNativeQuotaUsage(uuid string) {
|
||||
uuid = strings.TrimSpace(uuid)
|
||||
m.nativeQuotaMu.RLock()
|
||||
state := m.nativeQuotaByUUID[uuid]
|
||||
m.nativeQuotaMu.RUnlock()
|
||||
if state == nil {
|
||||
return
|
||||
}
|
||||
state.trafficMu.Lock()
|
||||
defer state.trafficMu.Unlock()
|
||||
state.mu.Lock()
|
||||
state.usedBytes = 0
|
||||
state.limiter = nil
|
||||
state.generation++
|
||||
if state.generation == 0 {
|
||||
state.generation = 1
|
||||
}
|
||||
state.mu.Unlock()
|
||||
}
|
||||
|
||||
func (m *XrayManager) resetNativeTrafficAccounting(ctx context.Context, store *Store, uuid, email string) error {
|
||||
state := m.nativeQuotaState(uuid)
|
||||
if state != nil {
|
||||
state.trafficMu.Lock()
|
||||
defer state.trafficMu.Unlock()
|
||||
state.mu.Lock()
|
||||
defer state.mu.Unlock()
|
||||
}
|
||||
|
||||
m.nativeTrafficPersistMu.Lock()
|
||||
defer m.nativeTrafficPersistMu.Unlock()
|
||||
|
||||
// Remove this client's queued pre-reset delta while holding only the short
|
||||
// map mutex. The database call may take seconds during an outage; keeping
|
||||
// nativeDBMu locked across it would stall traffic/accounting updates for
|
||||
// every other native user and could amplify a slow database into a goroutine
|
||||
// pile-up.
|
||||
m.nativeDBMu.Lock()
|
||||
key := strings.TrimSpace(uuid)
|
||||
var pending xrayPendingTraffic
|
||||
hadPending := false
|
||||
if m.nativeTrafficPending != nil {
|
||||
pending, hadPending = m.nativeTrafficPending[key]
|
||||
delete(m.nativeTrafficPending, key)
|
||||
}
|
||||
m.nativeDBMu.Unlock()
|
||||
|
||||
err := store.ResetXrayClientTraffic(ctx, uuid)
|
||||
if err != nil && hadPending && pending.State == state {
|
||||
m.nativeDBMu.Lock()
|
||||
if m.nativeTrafficPending == nil {
|
||||
m.nativeTrafficPending = make(map[string]xrayPendingTraffic)
|
||||
}
|
||||
current := m.nativeTrafficPending[key]
|
||||
if current.State != nil && current.State != state {
|
||||
current = xrayPendingTraffic{}
|
||||
}
|
||||
if current.Email == "" {
|
||||
current.Email = pending.Email
|
||||
}
|
||||
current.Uplink += pending.Uplink
|
||||
current.Downlink += pending.Downlink
|
||||
current.State = state
|
||||
m.nativeTrafficPending[key] = current
|
||||
m.nativeDBMu.Unlock()
|
||||
}
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if state != nil {
|
||||
state.usedBytes = 0
|
||||
state.limiter = nil
|
||||
state.generation++
|
||||
if state.generation == 0 {
|
||||
state.generation = 1
|
||||
}
|
||||
}
|
||||
|
||||
m.statsMu.Lock()
|
||||
for _, key := range []string{strings.TrimSpace(email), strings.TrimSpace(uuid)} {
|
||||
if key == "" {
|
||||
continue
|
||||
}
|
||||
if runtime, ok := m.statsByEmail[key]; ok {
|
||||
runtime.Uplink = 0
|
||||
runtime.Downlink = 0
|
||||
m.statsByEmail[key] = runtime
|
||||
}
|
||||
}
|
||||
m.statsMu.Unlock()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *XrayManager) nativeQuotaState(uuid string) *xrayNativeQuotaState {
|
||||
m.nativeQuotaMu.RLock()
|
||||
state := m.nativeQuotaByUUID[strings.TrimSpace(uuid)]
|
||||
m.nativeQuotaMu.RUnlock()
|
||||
return state
|
||||
}
|
||||
|
||||
// acquireNativeClientConnection enforces the DB-backed max_conns policy across
|
||||
// every native inbound and transport. The returned release function is safe to
|
||||
// call more than once and keeps runtime/DB online counters in sync.
|
||||
func (m *XrayManager) acquireNativeClientConnection(uuid, email string, closers ...io.Closer) (func(), *xrayNativeQuotaState, bool) {
|
||||
state := m.nativeQuotaState(uuid)
|
||||
var closer io.Closer
|
||||
if len(closers) > 0 {
|
||||
closer = closers[0]
|
||||
}
|
||||
if state != nil {
|
||||
state.mu.Lock()
|
||||
if reason := nativeClientAccessDeniedLocked(state); reason != "" {
|
||||
state.mu.Unlock()
|
||||
xrayLogf("native xray: rejected user %s: %s", email, reason)
|
||||
return nil, state, false
|
||||
}
|
||||
if state.maxConns > 0 && state.activeConns >= state.maxConns {
|
||||
limit := state.maxConns
|
||||
state.mu.Unlock()
|
||||
logNativeClientLimitRejection(email, limit)
|
||||
return nil, state, false
|
||||
}
|
||||
state.activeConns++
|
||||
if closer != nil {
|
||||
if state.connections == nil {
|
||||
state.connections = make(map[io.Closer]struct{})
|
||||
}
|
||||
state.connections[closer] = struct{}{}
|
||||
}
|
||||
state.mu.Unlock()
|
||||
}
|
||||
|
||||
m.recordNativeConnect(uuid, email, state)
|
||||
var once sync.Once
|
||||
return func() {
|
||||
once.Do(func() {
|
||||
if state != nil {
|
||||
state.mu.Lock()
|
||||
if closer != nil {
|
||||
delete(state.connections, closer)
|
||||
}
|
||||
if state.activeConns > 0 {
|
||||
state.activeConns--
|
||||
}
|
||||
state.mu.Unlock()
|
||||
}
|
||||
m.recordNativeDisconnect(uuid, email, state)
|
||||
})
|
||||
}, state, true
|
||||
}
|
||||
|
||||
func closeNativeClientConnections(state *xrayNativeQuotaState) {
|
||||
if state == nil {
|
||||
return
|
||||
}
|
||||
state.mu.Lock()
|
||||
closers := make([]io.Closer, 0, len(state.connections))
|
||||
for closer := range state.connections {
|
||||
closers = append(closers, closer)
|
||||
}
|
||||
state.mu.Unlock()
|
||||
for _, closer := range closers {
|
||||
_ = closer.Close()
|
||||
}
|
||||
}
|
||||
|
||||
func (m *XrayManager) disconnectNativeClient(uuid string) {
|
||||
closeNativeClientConnections(m.nativeQuotaState(uuid))
|
||||
}
|
||||
|
||||
func (m *XrayManager) nativeClientAccessDenied(uuid string) string {
|
||||
state := m.nativeQuotaState(uuid)
|
||||
if state == nil {
|
||||
return ""
|
||||
}
|
||||
state.mu.Lock()
|
||||
reason := nativeClientAccessDeniedLocked(state)
|
||||
state.mu.Unlock()
|
||||
return reason
|
||||
}
|
||||
|
||||
func nativeClientAccessDeniedLocked(state *xrayNativeQuotaState) string {
|
||||
if state.hasExpiry && !state.expiresAt.After(time.Now()) {
|
||||
return "expired"
|
||||
}
|
||||
if state.owner != "" {
|
||||
if err := ownerIsActive(state.owner); err != nil {
|
||||
return "owner suspended or expired"
|
||||
}
|
||||
}
|
||||
if state.quotaBytes > 0 && normalizeQuotaAction(state.action) == quotaActionBlock && state.usedBytes >= state.quotaBytes {
|
||||
return "data quota exceeded"
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (m *XrayManager) nativeQuotaBlocked(uuid string) bool {
|
||||
return nativeQuotaStateBlocked(m.nativeQuotaState(uuid))
|
||||
}
|
||||
|
||||
func nativeQuotaStateBlocked(state *xrayNativeQuotaState) bool {
|
||||
if state == nil {
|
||||
return false
|
||||
}
|
||||
state.mu.Lock()
|
||||
defer state.mu.Unlock()
|
||||
return state.quotaBytes > 0 && normalizeQuotaAction(state.action) == quotaActionBlock && state.usedBytes >= state.quotaBytes
|
||||
}
|
||||
|
||||
func (m *XrayManager) reserveNativeQuota(state *xrayNativeQuotaState, requested int) (allowed int, limiter *rate.Limiter, stopAfter bool) {
|
||||
if requested <= 0 {
|
||||
return 0, nil, false
|
||||
}
|
||||
if state == nil {
|
||||
return requested, nil, false
|
||||
}
|
||||
state.mu.Lock()
|
||||
defer state.mu.Unlock()
|
||||
|
||||
n := int64(requested)
|
||||
if state.quotaBytes <= 0 {
|
||||
state.usedBytes += n
|
||||
return requested, nil, false
|
||||
}
|
||||
if normalizeQuotaAction(state.action) == quotaActionThrottle {
|
||||
previous := state.usedBytes
|
||||
state.usedBytes += n
|
||||
if previous+n > state.quotaBytes {
|
||||
if state.limiter == nil {
|
||||
bps := mbpsToBytesPerSec(quotaThrottleMbpsOrDefault(state.throttleMbps))
|
||||
burst := int(bps)
|
||||
if burst < copyBufSize {
|
||||
burst = copyBufSize
|
||||
}
|
||||
state.limiter = rate.NewLimiter(rate.Limit(bps), burst)
|
||||
}
|
||||
return requested, state.limiter, false
|
||||
}
|
||||
return requested, nil, false
|
||||
}
|
||||
|
||||
remaining := state.quotaBytes - state.usedBytes
|
||||
if remaining <= 0 {
|
||||
return 0, nil, true
|
||||
}
|
||||
take := n
|
||||
if take > remaining {
|
||||
take = remaining
|
||||
}
|
||||
state.usedBytes += take
|
||||
return int(take), nil, take < n
|
||||
}
|
||||
|
||||
func (m *XrayManager) finishNativeQuotaReservation(state *xrayNativeQuotaState, reserved, written int) {
|
||||
if reserved <= 0 || written >= reserved {
|
||||
return
|
||||
}
|
||||
if written < 0 {
|
||||
written = 0
|
||||
}
|
||||
if state == nil {
|
||||
return
|
||||
}
|
||||
state.mu.Lock()
|
||||
state.usedBytes -= int64(reserved - written)
|
||||
if state.usedBytes < 0 {
|
||||
state.usedBytes = 0
|
||||
}
|
||||
state.mu.Unlock()
|
||||
}
|
||||
|
||||
type xrayQuotaMeteredWriter struct {
|
||||
w io.Writer
|
||||
meter *trafficMeter
|
||||
ctx context.Context
|
||||
}
|
||||
|
||||
func (mw xrayQuotaMeteredWriter) Write(p []byte) (int, error) {
|
||||
if mw.meter == nil {
|
||||
return mw.w.Write(p)
|
||||
}
|
||||
state := mw.meter.state
|
||||
if state != nil {
|
||||
state.trafficMu.RLock()
|
||||
defer state.trafficMu.RUnlock()
|
||||
}
|
||||
allowed, limiter, stopAfter := xrayMgr.reserveNativeQuota(state, len(p))
|
||||
if allowed <= 0 {
|
||||
return 0, errDataQuotaExceeded
|
||||
}
|
||||
if limiter != nil {
|
||||
ctx := mw.ctx
|
||||
if ctx == nil {
|
||||
ctx = context.Background()
|
||||
}
|
||||
if err := limiter.WaitN(ctx, allowed); err != nil {
|
||||
xrayMgr.finishNativeQuotaReservation(state, allowed, 0)
|
||||
return 0, err
|
||||
}
|
||||
}
|
||||
n, err := mw.w.Write(p[:allowed])
|
||||
xrayMgr.finishNativeQuotaReservation(state, allowed, n)
|
||||
if n > 0 {
|
||||
mw.meter.add(n)
|
||||
}
|
||||
if err != nil {
|
||||
return n, err
|
||||
}
|
||||
if stopAfter || allowed < len(p) || nativeQuotaStateBlocked(state) {
|
||||
return n, errDataQuotaExceeded
|
||||
}
|
||||
return n, nil
|
||||
}
|
||||
|
||||
type nativePacketQuotaReservation struct {
|
||||
meter *trafficMeter
|
||||
state *xrayNativeQuotaState
|
||||
limiter *rate.Limiter
|
||||
reserved int
|
||||
finished bool
|
||||
}
|
||||
|
||||
func reserveNativePacketQuota(meter *trafficMeter, n int) (nativePacketQuotaReservation, error) {
|
||||
if meter == nil || n <= 0 {
|
||||
return nativePacketQuotaReservation{}, nil
|
||||
}
|
||||
state := meter.state
|
||||
if state != nil {
|
||||
state.trafficMu.RLock()
|
||||
}
|
||||
allowed, limiter, stopAfter := xrayMgr.reserveNativeQuota(state, n)
|
||||
if allowed != n || stopAfter {
|
||||
if allowed > 0 {
|
||||
xrayMgr.finishNativeQuotaReservation(state, allowed, 0)
|
||||
}
|
||||
if state != nil {
|
||||
state.trafficMu.RUnlock()
|
||||
}
|
||||
return nativePacketQuotaReservation{}, errDataQuotaExceeded
|
||||
}
|
||||
return nativePacketQuotaReservation{
|
||||
meter: meter,
|
||||
state: state,
|
||||
limiter: limiter,
|
||||
reserved: n,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (r *nativePacketQuotaReservation) wait(ctx context.Context) error {
|
||||
if r == nil || r.finished || r.limiter == nil || r.reserved <= 0 {
|
||||
return nil
|
||||
}
|
||||
if err := waitNativeRate(ctx, r.limiter, r.reserved); err != nil {
|
||||
r.finish(0)
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *nativePacketQuotaReservation) finish(written int) {
|
||||
if r == nil || r.finished {
|
||||
return
|
||||
}
|
||||
r.finished = true
|
||||
if r.meter == nil || r.reserved <= 0 {
|
||||
if r.state != nil {
|
||||
r.state.trafficMu.RUnlock()
|
||||
}
|
||||
return
|
||||
}
|
||||
xrayMgr.finishNativeQuotaReservation(r.state, r.reserved, written)
|
||||
if written > 0 {
|
||||
r.meter.add(written)
|
||||
}
|
||||
if r.state != nil {
|
||||
r.state.trafficMu.RUnlock()
|
||||
}
|
||||
}
|
||||
+12
-3
@@ -670,7 +670,11 @@ func (ib *nativeInbound) handleVMess(stream net.Conn, remote net.Addr) {
|
||||
}
|
||||
client := ib.matchVMess(authid, time.Now().Unix())
|
||||
if client == nil {
|
||||
log.Printf("native xray: inbound %q rejected unknown/expired VMess auth id from %s", ib.tag, remote)
|
||||
logNativePreAuthRejection("native xray: inbound %q rejected unknown/expired VMess auth id from %s", ib.tag, remote)
|
||||
return
|
||||
}
|
||||
if reason := xrayMgr.nativeClientAccessDenied(client.uuid); reason != "" {
|
||||
log.Printf("native xray: inbound %q rejected VMess user %s: %s", ib.tag, client.email, reason)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -690,6 +694,11 @@ func (ib *nativeInbound) handleVMess(stream net.Conn, remote net.Addr) {
|
||||
log.Printf("native xray: inbound %q VMess command %d not supported yet", ib.tag, req.command)
|
||||
return
|
||||
}
|
||||
releaseConnection, quotaState, ok := xrayMgr.acquireNativeClientConnection(client.uuid, client.email, stream)
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
defer releaseConnection()
|
||||
|
||||
respBodyKey := sha256.Sum256(req.bodyKey[:])
|
||||
respBodyIV := sha256.Sum256(req.bodyIV[:])
|
||||
@@ -715,7 +724,7 @@ func (ib *nativeInbound) handleVMess(stream net.Conn, remote net.Addr) {
|
||||
return
|
||||
}
|
||||
log.Printf("native xray: vmess/tcp user=%s src=%s -> %s (inbound %q)", client.email, backend.LocalAddr(), target, ib.tag)
|
||||
nativeTunnel(vc, backend, client.uuid, client.email, ib.upLimiter(), ib.downLimiter())
|
||||
nativeTunnel(vc, backend, client.uuid, client.email, quotaState, ib.upLimiter(), ib.downLimiter())
|
||||
case vmessCmdUDP:
|
||||
backend, target, err := ib.nativeDialUDP(req.host, req.port)
|
||||
if err != nil {
|
||||
@@ -723,6 +732,6 @@ func (ib *nativeInbound) handleVMess(stream net.Conn, remote net.Addr) {
|
||||
return
|
||||
}
|
||||
log.Printf("native xray: vmess/udp user=%s src=%s -> %s (inbound %q)", client.email, backend.LocalAddr(), target, ib.tag)
|
||||
nativeVMessUDPTunnel(vc, backend, client.uuid, client.email, ib.upLimiter(), ib.downLimiter())
|
||||
nativeVMessUDPTunnel(vc, backend, client.uuid, client.email, quotaState, ib.upLimiter(), ib.downLimiter())
|
||||
}
|
||||
}
|
||||
|
||||
+571
-86
@@ -14,12 +14,45 @@ import (
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"golang.org/x/net/http2"
|
||||
"golang.org/x/net/http2/h2c"
|
||||
)
|
||||
|
||||
const (
|
||||
nativeXHTTPMaxSessionIDBytes = 256
|
||||
nativeXHTTPMaxSequenceBytes = 20
|
||||
nativeXHTTPHardMaxHeaderBytes = 256 * 1024
|
||||
nativeXHTTPHardMaxPostBytes int64 = 16 * 1024 * 1024
|
||||
nativeXHTTPMaxBufferedPosts = 512
|
||||
nativeXHTTPMaxBufferedSessionBytes = 16 * 1024 * 1024
|
||||
nativeXHTTPMaxBufferedGlobalBytes = 128 * 1024 * 1024
|
||||
// Tiny/empty packet-up requests still retain queue metadata. Charge a minimum
|
||||
// amount against the byte budgets so the reassembly queue can be count-unlimited
|
||||
// without allowing zero-byte packets to grow the heap without bound.
|
||||
nativeXHTTPMinPacketAccountingBytes int64 = 256
|
||||
)
|
||||
|
||||
var (
|
||||
nativeXHTTPBufferedBytes atomic.Int64
|
||||
nativeXHTTPBufferRejected atomic.Int64
|
||||
errNativeXHTTPUploadBufferFull = errors.New("xhttp upload buffer limit reached")
|
||||
nativeXHTTPMemoryWait = struct {
|
||||
sync.Mutex
|
||||
waiters []*nativeXHTTPMemoryWaiter
|
||||
head int
|
||||
queued int
|
||||
}{}
|
||||
)
|
||||
|
||||
type nativeXHTTPMemoryWaiter struct {
|
||||
bytes int64
|
||||
ready chan struct{}
|
||||
granted bool
|
||||
}
|
||||
|
||||
const (
|
||||
xhttpPlacementPath = "path"
|
||||
xhttpPlacementQuery = "query"
|
||||
@@ -157,7 +190,9 @@ func (ib *nativeInbound) serveXHTTPListener(ln net.Listener) {
|
||||
|
||||
func (g *nativeXHTTPListener) serve(ln net.Listener) {
|
||||
defer xrayRecover(fmt.Sprintf("native xray shared XHTTP listener addr=%s", ln.Addr()))
|
||||
h2s := &http2.Server{}
|
||||
h2s := &http2.Server{
|
||||
MaxConcurrentStreams: nativeHTTP2MaxConcurrentStreams(),
|
||||
}
|
||||
handler := http.Handler(g)
|
||||
// Official Xray accepts plaintext HTTP/1.1 and h2c on non-TLS XHTTP
|
||||
// listeners, and negotiates h2/http1 through ALPN on TLS listeners. Without
|
||||
@@ -258,6 +293,9 @@ func (ib *nativeInbound) reapStaleXHTTPSessions(idle time.Duration) {
|
||||
|
||||
func (ib *nativeInbound) xhttpServerMaxHeaderBytes() int {
|
||||
if ib.xhttpMaxHeaderBytes > 0 {
|
||||
if ib.xhttpMaxHeaderBytes > nativeXHTTPHardMaxHeaderBytes {
|
||||
return nativeXHTTPHardMaxHeaderBytes
|
||||
}
|
||||
return ib.xhttpMaxHeaderBytes
|
||||
}
|
||||
// Xray defaults to 8192. Keep a little room for custom headers/cookies used
|
||||
@@ -269,19 +307,25 @@ func (ib *nativeInbound) xhttpServerMaxHeaderBytes() int {
|
||||
// byte stream to the VLESS/VMess handlers as a net.Conn.
|
||||
func (ib *nativeInbound) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
defer xrayRecover(fmt.Sprintf("native xray XHTTP request inbound=%q method=%s path=%q remote=%s", ib.tag, r.Method, r.URL.RequestURI(), r.RemoteAddr))
|
||||
// XHTTP is a VPN transport, not a web API. A single connected user keeps a
|
||||
// long-lived download handler and can generate many short packet-up handlers.
|
||||
// Rejecting handlers at an application request ceiling turns normal tunnel
|
||||
// bursts into 429s and, through CDNs/reverse proxies, intermittent 502s.
|
||||
// HTTP/2 flow control plus the bounded, cancelable upload queues below provide
|
||||
// backpressure without applying website rate-limit semantics to tunnel traffic.
|
||||
if !ib.isXHTTP() {
|
||||
xrayLogf("native xray: xhttp reject inbound=%q reason=not-xhttp method=%s path=%q host=%q remote=%s", ib.tag, r.Method, r.URL.RequestURI(), r.Host, r.RemoteAddr)
|
||||
logNativePreAuthRejection("native xray: xhttp reject inbound=%q reason=not-xhttp method=%s path=%q host=%q remote=%s", ib.tag, r.Method, r.URL.RequestURI(), r.Host, r.RemoteAddr)
|
||||
xhttpBadRequest(w)
|
||||
return
|
||||
}
|
||||
if !ib.xhttpHostAllowed(r.Host) {
|
||||
xrayLogf("native xray: xhttp reject inbound=%q reason=host method=%s path=%q host=%q want=%q remote=%s", ib.tag, r.Method, r.URL.RequestURI(), r.Host, ib.xhttpHost, r.RemoteAddr)
|
||||
logNativePreAuthRejection("native xray: xhttp reject inbound=%q reason=host method=%s path=%q host=%q want=%q remote=%s", ib.tag, r.Method, r.URL.RequestURI(), r.Host, ib.xhttpHost, r.RemoteAddr)
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
return
|
||||
}
|
||||
base, ok := ib.matchXHTTPPath(r.URL.Path)
|
||||
if !ok {
|
||||
xrayLogf("native xray: xhttp reject inbound=%q reason=path method=%s path=%q want=%q host=%q remote=%s", ib.tag, r.Method, r.URL.RequestURI(), ib.path, r.Host, r.RemoteAddr)
|
||||
logNativePreAuthRejection("native xray: xhttp reject inbound=%q reason=path method=%s path=%q want=%q host=%q remote=%s", ib.tag, r.Method, r.URL.RequestURI(), ib.path, r.Host, r.RemoteAddr)
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
return
|
||||
}
|
||||
@@ -292,7 +336,21 @@ func (ib *nativeInbound) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
return
|
||||
}
|
||||
|
||||
releaseRequest, ok := acquireNativeXHTTPRequest()
|
||||
if !ok {
|
||||
logNativeLimitRejection("simultaneous XHTTP handlers", &nativeXHTTPRejected, nativeXHTTPRequestLimit())
|
||||
w.Header().Set("Retry-After", "1")
|
||||
http.Error(w, "xhttp transport temporarily busy", http.StatusServiceUnavailable)
|
||||
return
|
||||
}
|
||||
defer releaseRequest()
|
||||
|
||||
sessionID, seqStr := ib.extractXHTTPMeta(r, base)
|
||||
if len(sessionID) > nativeXHTTPMaxSessionIDBytes || len(seqStr) > nativeXHTTPMaxSequenceBytes {
|
||||
logNativePreAuthRejection("native xray: xhttp reject inbound=%q reason=metadata-size remote=%s", ib.tag, r.RemoteAddr)
|
||||
xhttpBadRequest(w)
|
||||
return
|
||||
}
|
||||
mode := ib.normalizedXHTTPMode()
|
||||
xrayTracef("native xray: xhttp request inbound=%q method=%s proto=%s path=%q host=%q session=%q seq=%q len=%d mode=%s remote=%s", ib.tag, r.Method, r.Proto, r.URL.RequestURI(), r.Host, sessionID, seqStr, r.ContentLength, mode, r.RemoteAddr)
|
||||
|
||||
@@ -519,18 +577,20 @@ func (ib *nativeInbound) upsertXHTTPSession(w http.ResponseWriter, id string) *n
|
||||
s.touch()
|
||||
return s
|
||||
}
|
||||
if max := ib.xhttpMaxActiveSessions(); max > 0 && len(ib.xhttpSessions) >= max {
|
||||
// XHTTP uses many HTTP requests/sessions by design. Returning HTTP 429
|
||||
// makes Xray clients tear down active tunnels, which is worse than allowing
|
||||
// a short soft-limit overflow and relying on stale-session cleanup.
|
||||
xrayTracef("native xray: xhttp session soft limit exceeded inbound=%q active=%d limit=%d", ib.tag, len(ib.xhttpSessions), max)
|
||||
releaseSlot, ok := acquireNativeXHTTPSession()
|
||||
if !ok {
|
||||
logNativeLimitRejection("simultaneous XHTTP sessions", &nativeXHTTPRejected, nativeXHTTPSessionLimit())
|
||||
w.Header().Set("Retry-After", "1")
|
||||
http.Error(w, "xhttp session capacity temporarily busy", http.StatusServiceUnavailable)
|
||||
return nil
|
||||
}
|
||||
s := &nativeXHTTPSession{
|
||||
id: id,
|
||||
queue: newNativeXHTTPUploadQueue(ib.xhttpMaxBufferedPosts),
|
||||
queue: newNativeXHTTPUploadQueue(ib.xhttpMaxBufferedPosts, nativeXHTTPMaxBufferedSessionBytes),
|
||||
done: make(chan struct{}),
|
||||
connectedCh: make(chan struct{}),
|
||||
lastSeen: time.Now(),
|
||||
releaseSlot: releaseSlot,
|
||||
}
|
||||
ib.xhttpSessions[id] = s
|
||||
xrayTracef("native xray: xhttp session created inbound=%q session=%q active=%d", ib.tag, id, len(ib.xhttpSessions))
|
||||
@@ -539,10 +599,7 @@ func (ib *nativeInbound) upsertXHTTPSession(w http.ResponseWriter, id string) *n
|
||||
}
|
||||
|
||||
func (ib *nativeInbound) xhttpMaxActiveSessions() int {
|
||||
if nativeXHTTPMaxSessionLimit() > 0 {
|
||||
return nativeXHTTPMaxSessionLimit()
|
||||
}
|
||||
return defaultNativeXHTTPMaxSessions
|
||||
return nativeXHTTPSessionLimit()
|
||||
}
|
||||
|
||||
func (ib *nativeInbound) reapUnconnectedXHTTPSession(id string, s *nativeXHTTPSession) {
|
||||
@@ -570,6 +627,19 @@ func (ib *nativeInbound) deleteXHTTPSession(id string, s *nativeXHTTPSession) {
|
||||
}
|
||||
}
|
||||
|
||||
func (ib *nativeInbound) closeAllXHTTPSessions() {
|
||||
ib.xhttpMu.Lock()
|
||||
sessions := make([]*nativeXHTTPSession, 0, len(ib.xhttpSessions))
|
||||
for id, session := range ib.xhttpSessions {
|
||||
delete(ib.xhttpSessions, id)
|
||||
sessions = append(sessions, session)
|
||||
}
|
||||
ib.xhttpMu.Unlock()
|
||||
for _, session := range sessions {
|
||||
session.close()
|
||||
}
|
||||
}
|
||||
|
||||
func (ib *nativeInbound) handleXHTTPStreamUpload(w http.ResponseWriter, r *http.Request, sess *nativeXHTTPSession) {
|
||||
sess.touch()
|
||||
xrayTracef("native xray: xhttp stream-up inbound=%q session=%q len=%d remote=%s", ib.tag, sess.id, r.ContentLength, r.RemoteAddr)
|
||||
@@ -577,7 +647,7 @@ func (ib *nativeInbound) handleXHTTPStreamUpload(w http.ResponseWriter, r *http.
|
||||
http.Error(w, "xhttp stream-up mode is not allowed", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if err := sess.queue.push(r.Context(), nativeXHTTPPacket{Reader: r.Body}); err != nil {
|
||||
if err := sess.queue.push(r.Context(), nativeXHTTPPacket{Reader: r.Body}, nil); err != nil {
|
||||
if errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) {
|
||||
return
|
||||
}
|
||||
@@ -605,16 +675,38 @@ func (ib *nativeInbound) handleXHTTPPacketUpload(w http.ResponseWriter, r *http.
|
||||
http.Error(w, "bad xhttp sequence", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
// Reserve the expected payload rather than the configured maximum. Normal
|
||||
// XHTTP body uploads have a Content-Length, so small packets no longer each
|
||||
// consume a full 1 MB reservation. Unknown/chunked or metadata-carried uploads
|
||||
// still reserve the maximum before decoding to preserve the hard memory bound.
|
||||
memory, err := acquireNativeXHTTPMemoryContext(r.Context(), ib.xhttpUploadReservationBytes(r))
|
||||
if err != nil {
|
||||
// If the client/CDN canceled while waiting for backpressure, there is no
|
||||
// useful HTTP error to send. Returning also releases every reservation.
|
||||
return
|
||||
}
|
||||
defer memory.release()
|
||||
payload, err := ib.readXHTTPPayload(r)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
memory.shrink(nativeXHTTPAccountedPacketBytes(int64(len(payload))))
|
||||
xrayTracef("native xray: xhttp packet-up inbound=%q session=%q seq=%d payload=%d remote=%s", ib.tag, sess.id, seq, len(payload), r.RemoteAddr)
|
||||
if err := sess.queue.push(r.Context(), nativeXHTTPPacket{Payload: payload, Seq: seq}); err != nil {
|
||||
if err := sess.queue.push(r.Context(), nativeXHTTPPacket{Payload: payload, Seq: seq}, memory); err != nil {
|
||||
if errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) {
|
||||
return
|
||||
}
|
||||
if errors.Is(err, io.ErrClosedPipe) {
|
||||
// A packet can race the stream-down request closing. Acknowledge the late
|
||||
// upload instead of leaking an origin 500/502 into the reconnect path.
|
||||
w.WriteHeader(http.StatusOK)
|
||||
return
|
||||
}
|
||||
if errors.Is(err, errNativeXHTTPUploadBufferFull) {
|
||||
http.Error(w, err.Error(), http.StatusRequestEntityTooLarge)
|
||||
return
|
||||
}
|
||||
xrayTracef("native xray: xhttp packet-up push failed inbound=%q session=%q seq=%d: %v", ib.tag, sess.id, seq, err)
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
@@ -719,11 +811,37 @@ func (ib *nativeInbound) readXHTTPBodyPayload(r *http.Request) ([]byte, error) {
|
||||
|
||||
func (ib *nativeInbound) xhttpMaxPostBytes() int64 {
|
||||
if ib.xhttpMaxEachPostBytes > 0 {
|
||||
if ib.xhttpMaxEachPostBytes > nativeXHTTPHardMaxPostBytes {
|
||||
return nativeXHTTPHardMaxPostBytes
|
||||
}
|
||||
return ib.xhttpMaxEachPostBytes
|
||||
}
|
||||
return 1_000_000
|
||||
}
|
||||
|
||||
// xhttpUploadReservationBytes returns a safe pre-read reservation. Body-mode
|
||||
// clients normally send Content-Length, which lets thousands of small packets
|
||||
// share the global budget. Header/cookie/auto and chunked bodies reserve the
|
||||
// configured maximum because their decoded size is not known until parsed.
|
||||
func (ib *nativeInbound) xhttpUploadReservationBytes(r *http.Request) int64 {
|
||||
maxBytes := ib.xhttpMaxPostBytes()
|
||||
placement := firstNonEmpty(ib.xhttpUplinkDataPlacement, xhttpPlacementBody)
|
||||
if placement == xhttpPlacementBody && r.ContentLength >= 0 {
|
||||
if r.ContentLength > maxBytes {
|
||||
return nativeXHTTPAccountedPacketBytes(maxBytes)
|
||||
}
|
||||
return nativeXHTTPAccountedPacketBytes(r.ContentLength)
|
||||
}
|
||||
return nativeXHTTPAccountedPacketBytes(maxBytes)
|
||||
}
|
||||
|
||||
func nativeXHTTPAccountedPacketBytes(payloadBytes int64) int64 {
|
||||
if payloadBytes < nativeXHTTPMinPacketAccountingBytes {
|
||||
return nativeXHTTPMinPacketAccountingBytes
|
||||
}
|
||||
return payloadBytes
|
||||
}
|
||||
|
||||
func (ib *nativeInbound) handleXHTTPStreamOne(w http.ResponseWriter, r *http.Request) {
|
||||
defer xrayRecover(fmt.Sprintf("native xray XHTTP stream-one inbound=%q remote=%s", ib.tag, r.RemoteAddr))
|
||||
xrayTracef("native xray: xhttp stream-one inbound=%q len=%d remote=%s", ib.tag, r.ContentLength, r.RemoteAddr)
|
||||
@@ -794,7 +912,7 @@ func (ib *nativeInbound) handleXHTTPDownload(w http.ResponseWriter, r *http.Requ
|
||||
// The stream-down HTTP request is the lifetime owner of an XHTTP
|
||||
// session. Log the actual transport cancellation so a CDN/proxy
|
||||
// timeout can be distinguished from a server idle policy.
|
||||
xrayLogf("native xray: xhttp stream-down ended inbound=%q session=%q remote=%s err=%v",
|
||||
xrayTracef("native xray: xhttp stream-down ended inbound=%q session=%q remote=%s err=%v",
|
||||
ib.tag, sessionID, r.RemoteAddr, r.Context().Err())
|
||||
_ = xc.Close()
|
||||
case <-sess.done:
|
||||
@@ -859,6 +977,7 @@ type nativeXHTTPSession struct {
|
||||
mu sync.Mutex
|
||||
connected bool
|
||||
lastSeen time.Time
|
||||
releaseSlot func()
|
||||
}
|
||||
|
||||
func (s *nativeXHTTPSession) touch() {
|
||||
@@ -879,6 +998,9 @@ func (s *nativeXHTTPSession) close() {
|
||||
s.closeOnce.Do(func() {
|
||||
close(s.done)
|
||||
s.queue.close()
|
||||
if s.releaseSlot != nil {
|
||||
s.releaseSlot()
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
@@ -949,12 +1071,19 @@ func (c *nativeXHTTPConn) SetReadDeadline(t time.Time) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *nativeXHTTPConn) SetWriteDeadline(time.Time) error { return nil }
|
||||
func (c *nativeXHTTPConn) SetWriteDeadline(t time.Time) error {
|
||||
if dw, ok := c.writer.(interface{ SetWriteDeadline(time.Time) error }); ok {
|
||||
return dw.SetWriteDeadline(t)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type nativeXHTTPResponseWriter struct {
|
||||
mu sync.Mutex
|
||||
w http.ResponseWriter
|
||||
closed bool
|
||||
writeMu sync.Mutex
|
||||
stateMu sync.Mutex
|
||||
w http.ResponseWriter
|
||||
closed bool
|
||||
deadline time.Time
|
||||
}
|
||||
|
||||
func newNativeXHTTPResponseWriter(w http.ResponseWriter) *nativeXHTTPResponseWriter {
|
||||
@@ -962,71 +1091,373 @@ func newNativeXHTTPResponseWriter(w http.ResponseWriter) *nativeXHTTPResponseWri
|
||||
}
|
||||
|
||||
func (w *nativeXHTTPResponseWriter) Write(p []byte) (int, error) {
|
||||
w.mu.Lock()
|
||||
defer w.mu.Unlock()
|
||||
if w.closed {
|
||||
w.writeMu.Lock()
|
||||
defer w.writeMu.Unlock()
|
||||
|
||||
w.stateMu.Lock()
|
||||
closed := w.closed
|
||||
deadline := w.deadline
|
||||
w.stateMu.Unlock()
|
||||
if closed {
|
||||
return 0, io.ErrClosedPipe
|
||||
}
|
||||
|
||||
safetyDeadline := time.Now().Add(nativeXHTTPWriteTimeout())
|
||||
if deadline.IsZero() || deadline.After(safetyDeadline) {
|
||||
deadline = safetyDeadline
|
||||
}
|
||||
controller := http.NewResponseController(w.w)
|
||||
if err := controller.SetWriteDeadline(deadline); err != nil && !errors.Is(err, http.ErrNotSupported) {
|
||||
return 0, err
|
||||
}
|
||||
n, err := w.w.Write(p)
|
||||
if err == nil {
|
||||
flushHTTP(w.w)
|
||||
if flushErr := controller.Flush(); flushErr != nil && !errors.Is(flushErr, http.ErrNotSupported) {
|
||||
err = flushErr
|
||||
}
|
||||
}
|
||||
return n, err
|
||||
}
|
||||
|
||||
func (w *nativeXHTTPResponseWriter) close() {
|
||||
w.mu.Lock()
|
||||
// Do not wait for writeMu: Close is commonly called by the request-context
|
||||
// watcher specifically because a CDN write is stalled. Mark the writer closed
|
||||
// and force the active net/http write deadline to expire so Write returns.
|
||||
w.stateMu.Lock()
|
||||
w.closed = true
|
||||
w.mu.Unlock()
|
||||
w.stateMu.Unlock()
|
||||
_ = http.NewResponseController(w.w).SetWriteDeadline(time.Now())
|
||||
}
|
||||
|
||||
func (w *nativeXHTTPResponseWriter) SetWriteDeadline(t time.Time) error {
|
||||
w.stateMu.Lock()
|
||||
w.deadline = t
|
||||
w.stateMu.Unlock()
|
||||
err := http.NewResponseController(w.w).SetWriteDeadline(t)
|
||||
if errors.Is(err, http.ErrNotSupported) {
|
||||
return nil
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
// nativeXHTTPMemoryLease reserves from a process-wide byte budget before a
|
||||
// packet-up handler allocates its payload. The same lease is transferred to the
|
||||
// session queue, so active request bodies and queued reassembly data share one
|
||||
// hard ceiling instead of each having an independent amplification window.
|
||||
type nativeXHTTPMemoryLease struct {
|
||||
bytes int64
|
||||
}
|
||||
|
||||
func acquireNativeXHTTPMemory(n int64) (*nativeXHTTPMemoryLease, bool) {
|
||||
if n <= 0 {
|
||||
return &nativeXHTTPMemoryLease{}, true
|
||||
}
|
||||
nativeXHTTPMemoryWait.Lock()
|
||||
defer nativeXHTTPMemoryWait.Unlock()
|
||||
current := nativeXHTTPBufferedBytes.Load()
|
||||
if nativeXHTTPMemoryWait.queued != 0 || current > nativeXHTTPMaxBufferedGlobalBytes-n {
|
||||
logNativeLimitRejection("XHTTP buffered upload bytes", &nativeXHTTPBufferRejected, nativeXHTTPMaxBufferedGlobalBytes)
|
||||
return nil, false
|
||||
}
|
||||
nativeXHTTPBufferedBytes.Store(current + n)
|
||||
return &nativeXHTTPMemoryLease{bytes: n}, true
|
||||
}
|
||||
|
||||
// acquireNativeXHTTPMemoryContext applies process-wide memory backpressure.
|
||||
// Unlike the old fail-fast admission path, a legitimate tunnel burst waits for
|
||||
// queued bytes to be consumed and remains cancelable if its HTTP request ends.
|
||||
func acquireNativeXHTTPMemoryContext(ctx context.Context, n int64) (*nativeXHTTPMemoryLease, error) {
|
||||
if n <= 0 {
|
||||
return &nativeXHTTPMemoryLease{}, nil
|
||||
}
|
||||
if n > nativeXHTTPMaxBufferedGlobalBytes {
|
||||
return nil, errNativeXHTTPUploadBufferFull
|
||||
}
|
||||
waiter := &nativeXHTTPMemoryWaiter{bytes: n, ready: make(chan struct{})}
|
||||
nativeXHTTPMemoryWait.Lock()
|
||||
current := nativeXHTTPBufferedBytes.Load()
|
||||
if nativeXHTTPMemoryWait.queued == 0 && current <= nativeXHTTPMaxBufferedGlobalBytes-n {
|
||||
nativeXHTTPBufferedBytes.Store(current + n)
|
||||
nativeXHTTPMemoryWait.Unlock()
|
||||
return &nativeXHTTPMemoryLease{bytes: n}, nil
|
||||
}
|
||||
nativeXHTTPMemoryWait.waiters = append(nativeXHTTPMemoryWait.waiters, waiter)
|
||||
nativeXHTTPMemoryWait.queued++
|
||||
nativeXHTTPMemoryWait.Unlock()
|
||||
|
||||
select {
|
||||
case <-waiter.ready:
|
||||
return &nativeXHTTPMemoryLease{bytes: n}, nil
|
||||
case <-ctx.Done():
|
||||
nativeXHTTPMemoryWait.Lock()
|
||||
if waiter.granted {
|
||||
current := nativeXHTTPBufferedBytes.Load() - n
|
||||
if current < 0 {
|
||||
current = 0
|
||||
}
|
||||
nativeXHTTPBufferedBytes.Store(current)
|
||||
} else {
|
||||
for i := nativeXHTTPMemoryWait.head; i < len(nativeXHTTPMemoryWait.waiters); i++ {
|
||||
candidate := nativeXHTTPMemoryWait.waiters[i]
|
||||
if candidate == waiter {
|
||||
nativeXHTTPMemoryWait.waiters[i] = nil
|
||||
nativeXHTTPMemoryWait.queued--
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
grantNativeXHTTPMemoryWaitersLocked()
|
||||
nativeXHTTPMemoryWait.Unlock()
|
||||
return nil, ctx.Err()
|
||||
}
|
||||
}
|
||||
|
||||
func releaseNativeXHTTPMemory(n int64) {
|
||||
if n <= 0 {
|
||||
return
|
||||
}
|
||||
nativeXHTTPMemoryWait.Lock()
|
||||
current := nativeXHTTPBufferedBytes.Load()
|
||||
next := current - n
|
||||
if next < 0 {
|
||||
next = 0
|
||||
}
|
||||
nativeXHTTPBufferedBytes.Store(next)
|
||||
grantNativeXHTTPMemoryWaitersLocked()
|
||||
nativeXHTTPMemoryWait.Unlock()
|
||||
}
|
||||
|
||||
// grantNativeXHTTPMemoryWaitersLocked wakes only the FIFO waiters whose exact
|
||||
// reservations now fit. The former broadcast channel woke every blocked HTTP
|
||||
// handler after every tiny release, creating a thundering herd and sustained
|
||||
// multi-core CPU usage while the 128 MB budget was full.
|
||||
func grantNativeXHTTPMemoryWaitersLocked() {
|
||||
for nativeXHTTPMemoryWait.queued > 0 {
|
||||
for nativeXHTTPMemoryWait.head < len(nativeXHTTPMemoryWait.waiters) &&
|
||||
nativeXHTTPMemoryWait.waiters[nativeXHTTPMemoryWait.head] == nil {
|
||||
nativeXHTTPMemoryWait.head++
|
||||
}
|
||||
if nativeXHTTPMemoryWait.head >= len(nativeXHTTPMemoryWait.waiters) {
|
||||
nativeXHTTPMemoryWait.waiters = nil
|
||||
nativeXHTTPMemoryWait.head = 0
|
||||
nativeXHTTPMemoryWait.queued = 0
|
||||
return
|
||||
}
|
||||
|
||||
waiter := nativeXHTTPMemoryWait.waiters[nativeXHTTPMemoryWait.head]
|
||||
current := nativeXHTTPBufferedBytes.Load()
|
||||
if current > nativeXHTTPMaxBufferedGlobalBytes-waiter.bytes {
|
||||
compactNativeXHTTPMemoryWaitersLocked()
|
||||
return
|
||||
}
|
||||
nativeXHTTPMemoryWait.waiters[nativeXHTTPMemoryWait.head] = nil
|
||||
nativeXHTTPMemoryWait.head++
|
||||
nativeXHTTPMemoryWait.queued--
|
||||
nativeXHTTPBufferedBytes.Store(current + waiter.bytes)
|
||||
waiter.granted = true
|
||||
close(waiter.ready)
|
||||
}
|
||||
compactNativeXHTTPMemoryWaitersLocked()
|
||||
}
|
||||
|
||||
func compactNativeXHTTPMemoryWaitersLocked() {
|
||||
head := nativeXHTTPMemoryWait.head
|
||||
if head == 0 {
|
||||
return
|
||||
}
|
||||
if nativeXHTTPMemoryWait.queued == 0 {
|
||||
nativeXHTTPMemoryWait.waiters = nil
|
||||
nativeXHTTPMemoryWait.head = 0
|
||||
return
|
||||
}
|
||||
if head < 1024 && head*2 < len(nativeXHTTPMemoryWait.waiters) {
|
||||
return
|
||||
}
|
||||
remaining := copy(nativeXHTTPMemoryWait.waiters, nativeXHTTPMemoryWait.waiters[head:])
|
||||
for i := remaining; i < len(nativeXHTTPMemoryWait.waiters); i++ {
|
||||
nativeXHTTPMemoryWait.waiters[i] = nil
|
||||
}
|
||||
nativeXHTTPMemoryWait.waiters = nativeXHTTPMemoryWait.waiters[:remaining]
|
||||
nativeXHTTPMemoryWait.head = 0
|
||||
}
|
||||
|
||||
func (l *nativeXHTTPMemoryLease) shrink(n int64) {
|
||||
if l == nil {
|
||||
return
|
||||
}
|
||||
if n < 0 {
|
||||
n = 0
|
||||
}
|
||||
if n >= l.bytes {
|
||||
return
|
||||
}
|
||||
release := l.bytes - n
|
||||
l.bytes = n
|
||||
releaseNativeXHTTPMemory(release)
|
||||
}
|
||||
|
||||
func (l *nativeXHTTPMemoryLease) release() {
|
||||
if l == nil || l.bytes <= 0 {
|
||||
return
|
||||
}
|
||||
n := l.bytes
|
||||
l.bytes = 0
|
||||
releaseNativeXHTTPMemory(n)
|
||||
}
|
||||
|
||||
type nativeXHTTPPacket struct {
|
||||
Reader io.ReadCloser
|
||||
Payload []byte
|
||||
Seq uint64
|
||||
Reader io.ReadCloser
|
||||
Payload []byte
|
||||
Seq uint64
|
||||
accountedBytes int64
|
||||
}
|
||||
|
||||
type nativeXHTTPUploadQueue struct {
|
||||
pushedPackets chan nativeXHTTPPacket
|
||||
maxPackets int
|
||||
maxBytes int64
|
||||
|
||||
mu sync.Mutex
|
||||
reader io.ReadCloser
|
||||
heap nativeXHTTPHeap
|
||||
nextSeq uint64
|
||||
readDeadline time.Time
|
||||
// readMu serializes the single decoded stream reader with close-time queue
|
||||
// cleanup. pushWG lets close wait until every producer that started before
|
||||
// closedFlag was set has either transferred or released its memory lease.
|
||||
readMu sync.Mutex
|
||||
pushWG sync.WaitGroup
|
||||
mu sync.Mutex
|
||||
reader io.ReadCloser
|
||||
readerQueued bool
|
||||
heap nativeXHTTPHeap
|
||||
nextSeq uint64
|
||||
readDeadline time.Time
|
||||
bufferedBytes int64
|
||||
closedFlag bool
|
||||
spaceChanged chan struct{}
|
||||
|
||||
closed chan struct{}
|
||||
closeOnce sync.Once
|
||||
}
|
||||
|
||||
func newNativeXHTTPUploadQueue(maxPackets int) *nativeXHTTPUploadQueue {
|
||||
func newNativeXHTTPUploadQueue(maxPackets int, maxBytes int64) *nativeXHTTPUploadQueue {
|
||||
if maxPackets <= 0 {
|
||||
maxPackets = defaultNativeXHTTPBufferedPosts
|
||||
}
|
||||
if maxPackets > nativeXHTTPMaxBufferedPosts {
|
||||
maxPackets = nativeXHTTPMaxBufferedPosts
|
||||
}
|
||||
if maxBytes <= 0 || maxBytes > nativeXHTTPMaxBufferedSessionBytes {
|
||||
maxBytes = nativeXHTTPMaxBufferedSessionBytes
|
||||
}
|
||||
return &nativeXHTTPUploadQueue{
|
||||
pushedPackets: make(chan nativeXHTTPPacket, maxPackets),
|
||||
maxPackets: maxPackets,
|
||||
maxBytes: maxBytes,
|
||||
closed: make(chan struct{}),
|
||||
spaceChanged: make(chan struct{}),
|
||||
}
|
||||
}
|
||||
|
||||
func (q *nativeXHTTPUploadQueue) push(ctx context.Context, p nativeXHTTPPacket) error {
|
||||
func (q *nativeXHTTPUploadQueue) beginPush() bool {
|
||||
q.mu.Lock()
|
||||
defer q.mu.Unlock()
|
||||
if q.closedFlag {
|
||||
return false
|
||||
}
|
||||
q.pushWG.Add(1)
|
||||
return true
|
||||
}
|
||||
|
||||
func (q *nativeXHTTPUploadQueue) adoptPayloadMemory(ctx context.Context, memory *nativeXHTTPMemoryLease, n int64) error {
|
||||
if n <= 0 {
|
||||
return nil
|
||||
}
|
||||
if memory == nil || memory.bytes != n {
|
||||
return errNativeXHTTPUploadBufferFull
|
||||
}
|
||||
if n > q.maxBytes {
|
||||
return errNativeXHTTPUploadBufferFull
|
||||
}
|
||||
for {
|
||||
q.mu.Lock()
|
||||
if q.closedFlag {
|
||||
q.mu.Unlock()
|
||||
return io.ErrClosedPipe
|
||||
}
|
||||
if q.bufferedBytes <= q.maxBytes-n {
|
||||
q.bufferedBytes += n
|
||||
memory.bytes = 0
|
||||
q.mu.Unlock()
|
||||
return nil
|
||||
}
|
||||
changed := q.spaceChanged
|
||||
q.mu.Unlock()
|
||||
select {
|
||||
case <-changed:
|
||||
case <-q.closed:
|
||||
return io.ErrClosedPipe
|
||||
case <-ctx.Done():
|
||||
return ctx.Err()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (q *nativeXHTTPUploadQueue) releasePayloadMemory(n int64) {
|
||||
if n <= 0 {
|
||||
return
|
||||
}
|
||||
q.mu.Lock()
|
||||
release := n
|
||||
if release > q.bufferedBytes {
|
||||
release = q.bufferedBytes
|
||||
}
|
||||
q.bufferedBytes -= release
|
||||
close(q.spaceChanged)
|
||||
q.spaceChanged = make(chan struct{})
|
||||
q.mu.Unlock()
|
||||
releaseNativeXHTTPMemory(release)
|
||||
}
|
||||
|
||||
func (q *nativeXHTTPUploadQueue) push(ctx context.Context, p nativeXHTTPPacket, memory *nativeXHTTPMemoryLease) error {
|
||||
if !q.beginPush() {
|
||||
return io.ErrClosedPipe
|
||||
}
|
||||
defer q.pushWG.Done()
|
||||
|
||||
readerReserved := false
|
||||
if p.Reader != nil {
|
||||
q.mu.Lock()
|
||||
if q.reader != nil {
|
||||
if q.reader != nil || q.readerQueued || q.closedFlag {
|
||||
q.mu.Unlock()
|
||||
return errors.New("xhttp upload reader already exists")
|
||||
}
|
||||
q.readerQueued = true
|
||||
readerReserved = true
|
||||
q.mu.Unlock()
|
||||
defer func() {
|
||||
if readerReserved {
|
||||
q.mu.Lock()
|
||||
q.readerQueued = false
|
||||
q.mu.Unlock()
|
||||
}
|
||||
}()
|
||||
}
|
||||
accountedBytes := int64(0)
|
||||
if p.Reader == nil {
|
||||
accountedBytes = nativeXHTTPAccountedPacketBytes(int64(len(p.Payload)))
|
||||
}
|
||||
if err := q.adoptPayloadMemory(ctx, memory, accountedBytes); err != nil {
|
||||
return err
|
||||
}
|
||||
p.accountedBytes = accountedBytes
|
||||
if accountedBytes > 0 {
|
||||
defer func() {
|
||||
if accountedBytes > 0 {
|
||||
q.releasePayloadMemory(accountedBytes)
|
||||
}
|
||||
}()
|
||||
}
|
||||
select {
|
||||
case q.pushedPackets <- p:
|
||||
select {
|
||||
case <-q.closed:
|
||||
return io.ErrClosedPipe
|
||||
default:
|
||||
}
|
||||
// Ownership has moved to the queue. close() waits for this producer and
|
||||
// then drains/releases anything not consumed by the stream reader.
|
||||
accountedBytes = 0
|
||||
readerReserved = false
|
||||
return nil
|
||||
case <-q.closed:
|
||||
return io.ErrClosedPipe
|
||||
@@ -1037,13 +1468,41 @@ func (q *nativeXHTTPUploadQueue) push(ctx context.Context, p nativeXHTTPPacket)
|
||||
|
||||
func (q *nativeXHTTPUploadQueue) close() {
|
||||
q.closeOnce.Do(func() {
|
||||
close(q.closed)
|
||||
q.mu.Lock()
|
||||
q.closedFlag = true
|
||||
reader := q.reader
|
||||
close(q.closed)
|
||||
q.mu.Unlock()
|
||||
if reader != nil {
|
||||
_ = reader.Close()
|
||||
}
|
||||
|
||||
q.pushWG.Wait()
|
||||
q.readMu.Lock()
|
||||
// No producers or readers can now change the queue. Drop references to
|
||||
// buffered payloads promptly and return their exact byte reservation.
|
||||
for {
|
||||
select {
|
||||
case p := <-q.pushedPackets:
|
||||
if p.Reader != nil {
|
||||
_ = p.Reader.Close()
|
||||
}
|
||||
p.Payload = nil
|
||||
default:
|
||||
goto drained
|
||||
}
|
||||
}
|
||||
drained:
|
||||
q.mu.Lock()
|
||||
remaining := q.bufferedBytes
|
||||
q.bufferedBytes = 0
|
||||
for i := range q.heap {
|
||||
q.heap[i].Payload = nil
|
||||
}
|
||||
q.heap = nil
|
||||
q.mu.Unlock()
|
||||
q.readMu.Unlock()
|
||||
releaseNativeXHTTPMemory(remaining)
|
||||
})
|
||||
}
|
||||
|
||||
@@ -1085,59 +1544,81 @@ func (q *nativeXHTTPUploadQueue) recv() (nativeXHTTPPacket, error) {
|
||||
}
|
||||
|
||||
func (q *nativeXHTTPUploadQueue) Read(b []byte) (int, error) {
|
||||
q.readMu.Lock()
|
||||
defer q.readMu.Unlock()
|
||||
|
||||
if reader := q.loadReader(); reader != nil {
|
||||
return reader.Read(b)
|
||||
}
|
||||
|
||||
select {
|
||||
case <-q.closed:
|
||||
return 0, io.EOF
|
||||
default:
|
||||
}
|
||||
|
||||
if len(q.heap) == 0 {
|
||||
p, err := q.recv()
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
if p.Reader != nil {
|
||||
q.setReader(p.Reader)
|
||||
return p.Reader.Read(b)
|
||||
}
|
||||
heap.Push(&q.heap, p)
|
||||
}
|
||||
|
||||
for len(q.heap) > 0 {
|
||||
packet := heap.Pop(&q.heap).(nativeXHTTPPacket)
|
||||
|
||||
if packet.Seq == q.nextSeq {
|
||||
n := copy(b, packet.Payload)
|
||||
if n < len(packet.Payload) {
|
||||
packet.Payload = packet.Payload[n:]
|
||||
heap.Push(&q.heap, packet)
|
||||
} else {
|
||||
q.nextSeq = packet.Seq + 1
|
||||
}
|
||||
return n, nil
|
||||
for {
|
||||
select {
|
||||
case <-q.closed:
|
||||
return 0, io.EOF
|
||||
default:
|
||||
}
|
||||
|
||||
if packet.Seq > q.nextSeq {
|
||||
if len(q.heap) > q.maxPackets {
|
||||
return 0, errors.New("xhttp upload reassembly buffer too large")
|
||||
}
|
||||
heap.Push(&q.heap, packet)
|
||||
if len(q.heap) == 0 {
|
||||
p, err := q.recv()
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
if p.Reader != nil {
|
||||
return 0, errors.New("xhttp mixed stream-up and packet-up upload")
|
||||
if !q.setReader(p.Reader) {
|
||||
_ = p.Reader.Close()
|
||||
return 0, io.EOF
|
||||
}
|
||||
return p.Reader.Read(b)
|
||||
}
|
||||
select {
|
||||
case <-q.closed:
|
||||
q.releasePayloadMemory(p.accountedBytes)
|
||||
return 0, io.EOF
|
||||
default:
|
||||
}
|
||||
heap.Push(&q.heap, p)
|
||||
}
|
||||
}
|
||||
|
||||
return 0, nil
|
||||
for len(q.heap) > 0 {
|
||||
packet := heap.Pop(&q.heap).(nativeXHTTPPacket)
|
||||
|
||||
if packet.Seq == q.nextSeq {
|
||||
if len(packet.Payload) == 0 {
|
||||
q.releasePayloadMemory(packet.accountedBytes)
|
||||
q.nextSeq = packet.Seq + 1
|
||||
continue
|
||||
}
|
||||
n := copy(b, packet.Payload)
|
||||
if n < len(packet.Payload) {
|
||||
q.releasePayloadMemory(int64(n))
|
||||
packet.accountedBytes -= int64(n)
|
||||
packet.Payload = packet.Payload[n:]
|
||||
heap.Push(&q.heap, packet)
|
||||
} else {
|
||||
q.releasePayloadMemory(packet.accountedBytes)
|
||||
q.nextSeq = packet.Seq + 1
|
||||
}
|
||||
return n, nil
|
||||
}
|
||||
|
||||
if packet.Seq > q.nextSeq {
|
||||
heap.Push(&q.heap, packet)
|
||||
p, err := q.recv()
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
if p.Reader != nil {
|
||||
_ = p.Reader.Close()
|
||||
return 0, errors.New("xhttp mixed stream-up and packet-up upload")
|
||||
}
|
||||
heap.Push(&q.heap, p)
|
||||
continue
|
||||
}
|
||||
|
||||
// A duplicate/late packet is discarded; release the bytes it owned.
|
||||
q.releasePayloadMemory(packet.accountedBytes)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (q *nativeXHTTPUploadQueue) loadReader() io.ReadCloser {
|
||||
@@ -1146,10 +1627,14 @@ func (q *nativeXHTTPUploadQueue) loadReader() io.ReadCloser {
|
||||
return q.reader
|
||||
}
|
||||
|
||||
func (q *nativeXHTTPUploadQueue) setReader(r io.ReadCloser) {
|
||||
func (q *nativeXHTTPUploadQueue) setReader(r io.ReadCloser) bool {
|
||||
q.mu.Lock()
|
||||
defer q.mu.Unlock()
|
||||
if q.closedFlag {
|
||||
return false
|
||||
}
|
||||
q.reader = r
|
||||
q.mu.Unlock()
|
||||
return true
|
||||
}
|
||||
|
||||
type nativeXHTTPHeap []nativeXHTTPPacket
|
||||
|
||||
Reference in New Issue
Block a user