Compare commits
14 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 004c9ed252 | |||
| 90d7f0903b | |||
| 10f9c2e609 | |||
| 3c515c7d7f | |||
| 773eb6426e | |||
| c520220df2 | |||
| f8bdb991c8 | |||
| d4bad38908 | |||
| a293986d6d | |||
| 96a3159c5d | |||
| 3f40506246 | |||
| 180282ba86 | |||
| c49fcaf1ce | |||
| fdeba5eeb5 |
+13
-7
@@ -26,11 +26,17 @@
|
|||||||
]
|
]
|
||||||
},
|
},
|
||||||
"release": {
|
"release": {
|
||||||
"registries": [
|
"targets": {
|
||||||
"https://verdaccio.lossless.digital",
|
"npm": {
|
||||||
"https://registry.npmjs.org"
|
"registries": [
|
||||||
],
|
"https://verdaccio.lossless.digital",
|
||||||
"accessLevel": "public"
|
"https://registry.npmjs.org"
|
||||||
}
|
],
|
||||||
}
|
"accessLevel": "public"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"schemaVersion": 2
|
||||||
|
},
|
||||||
|
"@ship.zone/szci": {}
|
||||||
}
|
}
|
||||||
@@ -1,5 +1,67 @@
|
|||||||
# Changelog
|
# Changelog
|
||||||
|
|
||||||
|
## Pending
|
||||||
|
|
||||||
|
|
||||||
|
## 2026-05-24 - 1.20.0
|
||||||
|
|
||||||
|
### Features
|
||||||
|
|
||||||
|
- add PROXY v2 real-source forwarding with authenticated VPN metadata TLVs (userspace-nat)
|
||||||
|
- allows socket forwarding to emit the client remote IP instead of the tunnel IP when configured
|
||||||
|
- adds SmartVPN client metadata to outbound PROXY v2 headers for trusted downstream authorization
|
||||||
|
|
||||||
|
## 2026-05-12 - 1.19.4
|
||||||
|
|
||||||
|
### Fixes
|
||||||
|
|
||||||
|
- keep WireGuard peer registration and client state in sync (rust-wireguard)
|
||||||
|
- index WireGuard public keys in the client registry for duplicate detection and direct lookup
|
||||||
|
- skip inactive clients when loading or adding WireGuard peers and fail fast when peer registration cannot be completed
|
||||||
|
- make IP reservation idempotent for the same client and avoid releasing WireGuard-assigned IPs on disconnect
|
||||||
|
- roll back client registry and peer state when WireGuard peer creation or key rotation fails
|
||||||
|
- update hybrid routing entries when registered client networking flags change
|
||||||
|
|
||||||
|
## 2026-05-12 - 1.19.3
|
||||||
|
|
||||||
|
### Fixes
|
||||||
|
|
||||||
|
- update release config schema, bump dependencies, and refresh runtime documentation (release)
|
||||||
|
- migrates .smartconfig.json release settings to the targets-based schema with schemaVersion 2
|
||||||
|
- bumps runtime and development dependencies including smartnftables and smartrust
|
||||||
|
- clarifies README details for custom Rust binaries, transport behavior, runtime events, and protocol frame types
|
||||||
|
|
||||||
|
## 2026-04-06 - 1.19.2 - fix(server)
|
||||||
|
clean up bridge and hybrid shutdown handling
|
||||||
|
|
||||||
|
- persist bridge teardown metadata so stop() can restore host IP configuration and remove the bridge in bridge and hybrid modes
|
||||||
|
- use separate shutdown channels for hybrid socket and bridge engines to stop both forwarding paths correctly
|
||||||
|
- avoid IP pool leaks when client registration fails and ignore unspecified IPv4 addresses when selecting WireGuard peer addresses
|
||||||
|
- make daemon bridge stop await nftables cleanup and process exit, and cap effective tunnel MTU to the link MTU
|
||||||
|
|
||||||
|
## 2026-04-01 - 1.19.1 - fix(rust)
|
||||||
|
clean up unused Rust warnings in bridge, network, and server modules
|
||||||
|
|
||||||
|
- remove the unused error import from the bridge module
|
||||||
|
- mark IpPool.prefix_len as intentionally unused to suppress dead code warnings
|
||||||
|
- rename the unused socket shutdown sender binding in the server to an underscore-prefixed variable
|
||||||
|
|
||||||
|
## 2026-04-01 - 1.19.0 - feat(forwarding)
|
||||||
|
add hybrid forwarding mode with per-client bridge and VLAN settings
|
||||||
|
|
||||||
|
- introduces a new hybrid forwarding mode that routes each client through either userspace NAT or bridge mode based on per-client configuration
|
||||||
|
- adds per-client bridge options including useHostIp, DHCP, static LAN IP, and VLAN assignment fields to the server and TypeScript interfaces
|
||||||
|
- adds Linux bridge VLAN helper functions and updates documentation to cover hybrid mode and VLAN-capable bridge clients
|
||||||
|
|
||||||
|
## 2026-03-31 - 1.18.0 - feat(server)
|
||||||
|
add bridge forwarding mode and per-client destination policy overrides
|
||||||
|
|
||||||
|
- introduces Linux bridge-based forwarding so VPN clients can receive IPs from a LAN subnet via TAP/bridge integration
|
||||||
|
- adds bridge server configuration options for LAN subnet, physical interface, and client IP allocation range
|
||||||
|
- adds per-client destinationPolicy overrides in the client registry and applies them in the userspace NAT engine based on assigned tunnel IP
|
||||||
|
- extends IP pool allocation to support constrained address ranges needed for bridge mode
|
||||||
|
- updates TypeScript interfaces and documentation to cover bridge mode and per-client destination policy behavior
|
||||||
|
|
||||||
## 2026-03-31 - 1.17.1 - fix(readme)
|
## 2026-03-31 - 1.17.1 - fix(readme)
|
||||||
document per-transport metrics and handshake-driven WireGuard connection state
|
document per-transport metrics and handshake-driven WireGuard connection state
|
||||||
|
|
||||||
|
|||||||
+8
-8
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "@push.rocks/smartvpn",
|
"name": "@push.rocks/smartvpn",
|
||||||
"version": "1.17.1",
|
"version": "1.20.0",
|
||||||
"private": false,
|
"private": false,
|
||||||
"description": "A VPN solution with TypeScript control plane and Rust data plane daemon",
|
"description": "A VPN solution with TypeScript control plane and Rust data plane daemon",
|
||||||
"type": "module",
|
"type": "module",
|
||||||
@@ -29,16 +29,16 @@
|
|||||||
],
|
],
|
||||||
"license": "MIT",
|
"license": "MIT",
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@push.rocks/smartnftables": "1.1.0",
|
"@push.rocks/smartnftables": "1.2.0",
|
||||||
"@push.rocks/smartpath": "^6.0.0",
|
"@push.rocks/smartpath": "^6.0.0",
|
||||||
"@push.rocks/smartrust": "^1.3.2"
|
"@push.rocks/smartrust": "^1.4.0"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@git.zone/tsbuild": "^4.4.0",
|
"@git.zone/tsbuild": "^4.4.1",
|
||||||
"@git.zone/tsrun": "^2.0.2",
|
"@git.zone/tsrun": "^2.0.4",
|
||||||
"@git.zone/tsrust": "^1.3.2",
|
"@git.zone/tsrust": "^1.3.4",
|
||||||
"@git.zone/tstest": "^3.6.3",
|
"@git.zone/tstest": "^3.6.6",
|
||||||
"@types/node": "^25.5.0"
|
"@types/node": "^25.7.0"
|
||||||
},
|
},
|
||||||
"files": [
|
"files": [
|
||||||
"ts/**/*",
|
"ts/**/*",
|
||||||
|
|||||||
Generated
+2134
-2808
File diff suppressed because it is too large
Load Diff
@@ -2,16 +2,19 @@
|
|||||||
|
|
||||||
A high-performance VPN solution with a **TypeScript control plane** and a **Rust data plane daemon**. Enterprise-ready client authentication, triple transport support (WebSocket + QUIC + WireGuard), and a typed hub API for managing clients from code.
|
A high-performance VPN solution with a **TypeScript control plane** and a **Rust data plane daemon**. Enterprise-ready client authentication, triple transport support (WebSocket + QUIC + WireGuard), and a typed hub API for managing clients from code.
|
||||||
|
|
||||||
🔐 **Noise IK** mutual authentication — per-client X25519 keypairs, server-side registry
|
- 🔐 **Noise IK** mutual authentication — per-client X25519 keypairs, server-side registry
|
||||||
🚀 **Triple transport**: WebSocket (Cloudflare-friendly), raw **QUIC** (datagrams), and **WireGuard** (standard protocol)
|
- 🚀 **Triple transport**: WebSocket (Cloudflare-friendly), raw **QUIC** (datagrams), and **WireGuard** (standard protocol)
|
||||||
🛡️ **ACL engine** — deny-overrides-allow IP filtering, aligned with SmartProxy conventions
|
- 🛡️ **ACL engine** — deny-overrides-allow IP filtering, aligned with SmartProxy conventions
|
||||||
🔀 **PROXY protocol v2** — real client IPs behind reverse proxies (HAProxy, SmartProxy, Cloudflare Spectrum)
|
- 🔀 **PROXY protocol v2** — real client IPs behind reverse proxies (HAProxy, SmartProxy, Cloudflare Spectrum)
|
||||||
📊 **Per-transport metrics**: active clients and total connections broken down by websocket, QUIC, and WireGuard
|
- 📊 **Per-transport metrics**: active clients and total connections broken down by websocket, QUIC, and WireGuard
|
||||||
🔄 **Hub API**: one `createClient()` call generates keys, assigns IP, returns both SmartVPN + WireGuard configs
|
- 🔄 **Hub API**: one `createClient()` call generates keys, assigns IP, returns both SmartVPN + WireGuard configs
|
||||||
📡 **Real-time telemetry**: RTT, jitter, loss ratio, link health — all via typed APIs
|
- 📡 **Real-time telemetry**: RTT, jitter, loss ratio, link health — all via typed APIs
|
||||||
🌐 **Unified forwarding pipeline**: all transports share the same engine — TUN (kernel), userspace NAT (no root), or testing mode
|
- 🌐 **Unified forwarding pipeline**: all transports share the same engine — TUN (kernel), userspace NAT (no root), L2 bridge, hybrid, or testing mode
|
||||||
🎯 **Destination routing policy**: force-target, block, or allow traffic per destination with nftables integration
|
- 🏠 **Bridge mode**: VPN clients get IPs from your LAN subnet — seamlessly bridge remote clients onto a physical network
|
||||||
⚡ **Handshake-driven WireGuard state**: peers appear as "connected" only after a successful WireGuard handshake, and auto-disconnect on idle timeout
|
- 🔀 **Hybrid mode**: per-client routing — some clients bridge to the LAN, others use userspace NAT, all on the same server
|
||||||
|
- 🏷️ **VLAN support**: assign individual clients to 802.1Q VLANs on the bridge
|
||||||
|
- 🎯 **Destination routing policy**: force-target, block, or allow traffic per destination with nftables integration
|
||||||
|
- ⚡ **Handshake-driven WireGuard state**: peers appear as "connected" only after a successful WireGuard handshake, and auto-disconnect on idle timeout
|
||||||
|
|
||||||
## Issue Reporting and Security
|
## Issue Reporting and Security
|
||||||
|
|
||||||
@@ -21,11 +24,9 @@ For reporting bugs, issues, or security vulnerabilities, please visit [community
|
|||||||
|
|
||||||
```bash
|
```bash
|
||||||
pnpm install @push.rocks/smartvpn
|
pnpm install @push.rocks/smartvpn
|
||||||
# or
|
|
||||||
npm install @push.rocks/smartvpn
|
|
||||||
```
|
```
|
||||||
|
|
||||||
The package ships with pre-compiled Rust binaries for **linux/amd64** and **linux/arm64**. No Rust toolchain required at runtime.
|
The package ships with pre-compiled Rust binaries for **linux/amd64** and **linux/arm64**. No Rust toolchain is required at runtime. Set `SMARTVPN_RUST_BINARY` if you want the TypeScript bridge to use a custom daemon binary.
|
||||||
|
|
||||||
## Architecture 🏗️
|
## Architecture 🏗️
|
||||||
|
|
||||||
@@ -34,7 +35,7 @@ The package ships with pre-compiled Rust binaries for **linux/amd64** and **linu
|
|||||||
│ TypeScript Control Plane │ ◄─────────────────────► │ Rust Data Plane Daemon │
|
│ TypeScript Control Plane │ ◄─────────────────────► │ Rust Data Plane Daemon │
|
||||||
│ │ stdio or Unix sock │ │
|
│ │ stdio or Unix sock │ │
|
||||||
│ VpnServer / VpnClient │ │ Noise IK handshake │
|
│ VpnServer / VpnClient │ │ Noise IK handshake │
|
||||||
│ Typed IPC commands │ │ XChaCha20-Poly1305 │
|
│ Typed IPC commands │ │ Noise transport encryption │
|
||||||
│ Config validation │ │ WS + QUIC + WireGuard │
|
│ Config validation │ │ WS + QUIC + WireGuard │
|
||||||
│ Hub: client management │ │ TUN device, IP pool, NAT │
|
│ Hub: client management │ │ TUN device, IP pool, NAT │
|
||||||
│ WireGuard .conf generation │ │ Rate limiting, ACLs, QoS │
|
│ WireGuard .conf generation │ │ Rate limiting, ACLs, QoS │
|
||||||
@@ -42,7 +43,7 @@ The package ships with pre-compiled Rust binaries for **linux/amd64** and **linu
|
|||||||
└──────────────────────────────┘ └───────────────────────────────┘
|
└──────────────────────────────┘ └───────────────────────────────┘
|
||||||
```
|
```
|
||||||
|
|
||||||
**Split-plane design** — TypeScript handles orchestration, config, and DX; Rust handles every hot-path byte with zero-copy async I/O (tokio, mimalloc).
|
**Split-plane design** — TypeScript handles orchestration, config, and DX; Rust handles the hot path with async I/O, framed packet codecs, and the Noise transport state after authentication.
|
||||||
|
|
||||||
### IPC Transport Modes
|
### IPC Transport Modes
|
||||||
|
|
||||||
@@ -84,7 +85,7 @@ await server.start({
|
|||||||
publicKey: '<server-noise-public-key-base64>',
|
publicKey: '<server-noise-public-key-base64>',
|
||||||
subnet: '10.8.0.0/24',
|
subnet: '10.8.0.0/24',
|
||||||
transportMode: 'all', // WebSocket + QUIC + WireGuard simultaneously (default)
|
transportMode: 'all', // WebSocket + QUIC + WireGuard simultaneously (default)
|
||||||
forwardingMode: 'tun', // 'tun' (kernel), 'socket' (userspace NAT), or 'testing'
|
forwardingMode: 'tun', // 'tun' | 'socket' | 'bridge' | 'hybrid' | 'testing'
|
||||||
wgPrivateKey: '<server-wg-private-key-base64>', // required for WireGuard transport
|
wgPrivateKey: '<server-wg-private-key-base64>', // required for WireGuard transport
|
||||||
enableNat: true,
|
enableNat: true,
|
||||||
dns: ['1.1.1.1', '8.8.8.8'],
|
dns: ['1.1.1.1', '8.8.8.8'],
|
||||||
@@ -139,7 +140,7 @@ Every client authenticates with a **Noise IK handshake** (`Noise_IK_25519_ChaCha
|
|||||||
| **QUIC** | UDP (via quinn) | Low latency, datagram support for IP packets |
|
| **QUIC** | UDP (via quinn) | Low latency, datagram support for IP packets |
|
||||||
| **WireGuard** | UDP (via boringtun) | Standard WG clients (iOS, Android, wg-quick) |
|
| **WireGuard** | UDP (via boringtun) | Standard WG clients (iOS, Android, wg-quick) |
|
||||||
|
|
||||||
The server runs **all three simultaneously** by default with `transportMode: 'all'`. All transports share the same unified forwarding pipeline (`ForwardingEngine`), IP pool, client registry, and stats — so WireGuard peers get the same userspace NAT, rate limiting, and monitoring as WS/QUIC clients. Clients auto-negotiate with `transport: 'auto'` (tries QUIC first, falls back to WS).
|
The server runs with `transportMode: 'all'` by default: WebSocket and QUIC are enabled, and WireGuard joins the same server when `wgPrivateKey` is configured. All server transports share the same forwarding pipeline (`ForwardingEngine`), IP pool, client registry, and statistics, so WireGuard peers can use the same userspace NAT, bridge/hybrid routing, and monitoring model as WS/QUIC clients. Native SmartVPN clients auto-negotiate with `transport: 'auto'` (tries QUIC first, falls back to WS).
|
||||||
|
|
||||||
### 📊 Per-Transport Metrics
|
### 📊 Per-Transport Metrics
|
||||||
|
|
||||||
@@ -237,6 +238,21 @@ In **TUN mode**, destination policies are enforced via **nftables** rules (using
|
|||||||
|
|
||||||
In **socket mode**, the policy is evaluated in the userspace NAT engine before per-client ACLs.
|
In **socket mode**, the policy is evaluated in the userspace NAT engine before per-client ACLs.
|
||||||
|
|
||||||
|
**Per-client override** — individual clients can have their own destination policy that overrides the server-level default:
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
await server.createClient({
|
||||||
|
clientId: 'restricted-client',
|
||||||
|
security: {
|
||||||
|
destinationPolicy: {
|
||||||
|
default: 'block', // block everything by default
|
||||||
|
allowList: ['10.0.0.0/8'], // except internal network
|
||||||
|
},
|
||||||
|
// ... other security settings
|
||||||
|
},
|
||||||
|
});
|
||||||
|
```
|
||||||
|
|
||||||
### 🔗 Socket Forward Proxy Protocol
|
### 🔗 Socket Forward Proxy Protocol
|
||||||
|
|
||||||
When using `forwardingMode: 'socket'` (userspace NAT), you can prepend **PROXY protocol v2 headers** on outbound TCP connections. This conveys the VPN client's tunnel IP as the source address to downstream services (e.g., SmartProxy):
|
When using `forwardingMode: 'socket'` (userspace NAT), you can prepend **PROXY protocol v2 headers** on outbound TCP connections. This conveys the VPN client's tunnel IP as the source address to downstream services (e.g., SmartProxy):
|
||||||
@@ -251,12 +267,14 @@ await server.start({
|
|||||||
|
|
||||||
### 📦 Packet Forwarding Modes
|
### 📦 Packet Forwarding Modes
|
||||||
|
|
||||||
SmartVPN supports three forwarding modes, configurable per-server and per-client:
|
SmartVPN supports five forwarding modes, configurable per-server:
|
||||||
|
|
||||||
| Mode | Flag | Description | Root Required |
|
| Mode | Flag | Description | Root Required |
|
||||||
|------|------|-------------|---------------|
|
|------|------|-------------|---------------|
|
||||||
| **TUN** | `'tun'` | Kernel TUN device — real packet forwarding with system routing | ✅ Yes |
|
| **TUN** | `'tun'` | Kernel TUN device — real packet forwarding with system routing | ✅ Yes |
|
||||||
| **Userspace NAT** | `'socket'` | Userspace TCP/UDP proxy via `connect(2)` — no TUN, no root needed | ❌ No |
|
| **Userspace NAT** | `'socket'` | Userspace TCP/UDP proxy via `connect(2)` — no TUN, no root needed | ❌ No |
|
||||||
|
| **Bridge** | `'bridge'` | L2 bridge — VPN clients get IPs from a physical LAN subnet | ✅ Yes |
|
||||||
|
| **Hybrid** | `'hybrid'` | Per-client routing: some clients use socket NAT, others use bridge — both engines run simultaneously | ✅ Yes |
|
||||||
| **Testing** | `'testing'` | Monitoring only — packets are counted but not forwarded | ❌ No |
|
| **Testing** | `'testing'` | Monitoring only — packets are counted but not forwarded | ❌ No |
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
@@ -267,6 +285,23 @@ await server.start({
|
|||||||
enableNat: true,
|
enableNat: true,
|
||||||
});
|
});
|
||||||
|
|
||||||
|
// Server with bridge mode — VPN clients appear on the LAN
|
||||||
|
await server.start({
|
||||||
|
// ...
|
||||||
|
forwardingMode: 'bridge',
|
||||||
|
bridgeLanSubnet: '192.168.1.0/24', // LAN subnet to bridge into
|
||||||
|
bridgePhysicalInterface: 'eth0', // auto-detected if omitted
|
||||||
|
bridgeIpRangeStart: 200, // clients get .200–.250 (defaults)
|
||||||
|
bridgeIpRangeEnd: 250,
|
||||||
|
});
|
||||||
|
|
||||||
|
// Server with hybrid mode — per-client routing
|
||||||
|
await server.start({
|
||||||
|
// ...
|
||||||
|
forwardingMode: 'hybrid',
|
||||||
|
bridgePhysicalInterface: 'eth0', // for bridge clients
|
||||||
|
});
|
||||||
|
|
||||||
// Client with TUN device
|
// Client with TUN device
|
||||||
const { assignedIp } = await client.connect({
|
const { assignedIp } = await client.connect({
|
||||||
// ...
|
// ...
|
||||||
@@ -274,7 +309,55 @@ const { assignedIp } = await client.connect({
|
|||||||
});
|
});
|
||||||
```
|
```
|
||||||
|
|
||||||
The userspace NAT mode extracts destination IP/port from IP packets, opens a real socket to the destination, and relays data — supporting both TCP streams and UDP datagrams without requiring `CAP_NET_ADMIN` or root privileges.
|
The **userspace NAT** mode extracts destination IP/port from IP packets, opens a real socket to the destination, and relays data — supporting both TCP streams and UDP datagrams without requiring `CAP_NET_ADMIN` or root privileges.
|
||||||
|
|
||||||
|
The **bridge** mode assigns VPN clients IPs from a real LAN subnet instead of a virtual VPN subnet. Clients appear as if they're directly on the physical network — perfect for remote access to home labs, office networks, or IoT devices.
|
||||||
|
|
||||||
|
The **hybrid** mode runs both engines simultaneously with a **per-client routing table**. Each client's `useHostIp` flag determines whether its packets go through the bridge (L2, LAN IP) or socket NAT (userspace, VPN IP). This is ideal when most clients need internet NAT but some need direct LAN access.
|
||||||
|
|
||||||
|
### 🏠 Per-Client Bridge & VLAN Settings
|
||||||
|
|
||||||
|
When using `bridge` or `hybrid` mode, each client can be individually configured for LAN bridging, static IPs, DHCP, and 802.1Q VLAN assignment:
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
// Client that bridges to the LAN with a static IP
|
||||||
|
await server.createClient({
|
||||||
|
clientId: 'office-printer',
|
||||||
|
useHostIp: true, // bridge to LAN instead of VPN subnet
|
||||||
|
staticIp: '192.168.1.210', // fixed LAN IP
|
||||||
|
});
|
||||||
|
|
||||||
|
// Client that gets a LAN IP via DHCP
|
||||||
|
await server.createClient({
|
||||||
|
clientId: 'roaming-laptop',
|
||||||
|
useHostIp: true,
|
||||||
|
useDhcp: true, // obtain IP from LAN DHCP server
|
||||||
|
});
|
||||||
|
|
||||||
|
// Client on a specific VLAN
|
||||||
|
await server.createClient({
|
||||||
|
clientId: 'iot-sensor',
|
||||||
|
useHostIp: true,
|
||||||
|
forceVlan: true,
|
||||||
|
vlanId: 100, // 802.1Q VLAN ID (1-4094)
|
||||||
|
});
|
||||||
|
|
||||||
|
// Regular NAT client (default, no bridge)
|
||||||
|
await server.createClient({
|
||||||
|
clientId: 'remote-worker',
|
||||||
|
// useHostIp defaults to false → uses socket NAT
|
||||||
|
});
|
||||||
|
```
|
||||||
|
|
||||||
|
| Field | Type | Description |
|
||||||
|
|-------|------|-------------|
|
||||||
|
| `useHostIp` | `boolean` | `true` = bridge to LAN (host IP), `false` = VPN subnet via NAT (default) |
|
||||||
|
| `useDhcp` | `boolean` | When `useHostIp` is true, obtain IP via DHCP relay instead of static/auto-assign |
|
||||||
|
| `staticIp` | `string` | Fixed LAN IP when `useHostIp` is true and `useDhcp` is false |
|
||||||
|
| `forceVlan` | `boolean` | Assign this client to a specific 802.1Q VLAN on the bridge |
|
||||||
|
| `vlanId` | `number` | VLAN ID (1-4094), required when `forceVlan` is true |
|
||||||
|
|
||||||
|
VLAN support uses Linux bridge VLAN filtering — each client's TAP port gets tagged with the specified VLAN ID, isolating traffic at Layer 2.
|
||||||
|
|
||||||
### 📊 Telemetry & QoS
|
### 📊 Telemetry & QoS
|
||||||
|
|
||||||
@@ -396,37 +479,23 @@ const unit = VpnInstaller.generateServiceUnit({
|
|||||||
|
|
||||||
You can also call `generateSystemdUnit()` or `generateLaunchdPlist()` directly for platform-specific options like custom descriptions.
|
You can also call `generateSystemdUnit()` or `generateLaunchdPlist()` directly for platform-specific options like custom descriptions.
|
||||||
|
|
||||||
### 📢 Events
|
### 📢 Runtime Events
|
||||||
|
|
||||||
Both `VpnServer` and `VpnClient` extend `EventEmitter` and emit typed events:
|
`VpnServer` and `VpnClient` extend `EventEmitter`. The high-level wrappers currently forward bridge lifecycle events:
|
||||||
|
|
||||||
```typescript
|
```typescript
|
||||||
server.on('client-connected', (info: IVpnClientInfo) => {
|
|
||||||
console.log(`${info.registeredClientId} connected from ${info.remoteAddr} via ${info.transportType}`);
|
|
||||||
});
|
|
||||||
|
|
||||||
server.on('client-disconnected', ({ clientId, reason }) => {
|
|
||||||
console.log(`${clientId} disconnected: ${reason}`);
|
|
||||||
});
|
|
||||||
|
|
||||||
client.on('status', (status: IVpnStatus) => {
|
|
||||||
console.log(`State: ${status.state}, IP: ${status.assignedIp}`);
|
|
||||||
});
|
|
||||||
|
|
||||||
// Both server and client emit:
|
|
||||||
server.on('exit', ({ code, signal }) => { /* daemon process exited */ });
|
server.on('exit', ({ code, signal }) => { /* daemon process exited */ });
|
||||||
server.on('reconnected', () => { /* socket transport reconnected */ });
|
server.on('reconnected', () => { /* socket transport reconnected */ });
|
||||||
|
client.on('exit', ({ code, signal }) => { /* daemon process exited */ });
|
||||||
```
|
```
|
||||||
|
|
||||||
| Event | Emitted By | Payload |
|
| Event | Emitted By | Payload |
|
||||||
|-------|-----------|---------|
|
|-------|-----------|---------|
|
||||||
| `status` | Both | `IVpnStatus` — connection state changes |
|
|
||||||
| `error` | Both | `{ message, code? }` |
|
|
||||||
| `client-connected` | Server | `IVpnClientInfo` — full client info including transport type |
|
|
||||||
| `client-disconnected` | Server | `{ clientId, reason? }` |
|
|
||||||
| `exit` | Both | `{ code, signal }` — daemon process exited |
|
| `exit` | Both | `{ code, signal }` — daemon process exited |
|
||||||
| `reconnected` | Both | `void` — socket transport reconnected |
|
| `reconnected` | Both | `void` — socket transport reconnected |
|
||||||
|
|
||||||
|
For connection state and telemetry, use `getStatus()`, `getStatistics()`, `listClients()`, and `getClientTelemetry()`.
|
||||||
|
|
||||||
## API Reference 📖
|
## API Reference 📖
|
||||||
|
|
||||||
### Classes
|
### Classes
|
||||||
@@ -444,10 +513,10 @@ server.on('reconnected', () => { /* socket transport reconnected */ });
|
|||||||
|
|
||||||
| Interface | Purpose |
|
| Interface | Purpose |
|
||||||
|-----------|---------|
|
|-----------|---------|
|
||||||
| `IVpnServerConfig` | Server configuration (listen addr, keys, subnet, transport mode, forwarding mode, clients, proxy protocol, destination policy) |
|
| `IVpnServerConfig` | Server configuration (listen addr, keys, subnet, transport mode, forwarding mode incl. bridge/hybrid, clients, proxy protocol, destination policy) |
|
||||||
| `IVpnClientConfig` | Client configuration (server URL, keys, transport, forwarding mode, WG options, client-defined tags) |
|
| `IVpnClientConfig` | Client configuration (server URL, keys, transport, forwarding mode, WG options, client-defined tags) |
|
||||||
| `IClientEntry` | Server-side client definition (ID, keys, security, priority, server/client tags, expiry) |
|
| `IClientEntry` | Server-side client definition (ID, keys, security, priority, server/client tags, expiry, bridge/VLAN settings) |
|
||||||
| `IClientSecurity` | Per-client ACLs and rate limits (SmartProxy-aligned naming) |
|
| `IClientSecurity` | Per-client ACLs, rate limits, and destination policy override (SmartProxy-aligned naming) |
|
||||||
| `IClientRateLimit` | Rate limiting config (bytesPerSec, burstBytes) |
|
| `IClientRateLimit` | Rate limiting config (bytesPerSec, burstBytes) |
|
||||||
| `IClientConfigBundle` | Full config bundle returned by `createClient()` — includes SmartVPN config, WireGuard .conf, and secrets |
|
| `IClientConfigBundle` | Full config bundle returned by `createClient()` — includes SmartVPN config, WireGuard .conf, and secrets |
|
||||||
| `IVpnClientInfo` | Connected client info (IP, stats, authenticated key, remote addr, transport type) |
|
| `IVpnClientInfo` | Connected client info (IP, stats, authenticated key, remote addr, transport type) |
|
||||||
@@ -456,7 +525,7 @@ server.on('reconnected', () => { /* socket transport reconnected */ });
|
|||||||
| `IVpnMtuInfo` | TUN MTU, effective MTU, overhead bytes, oversized packet stats |
|
| `IVpnMtuInfo` | TUN MTU, effective MTU, overhead bytes, oversized packet stats |
|
||||||
| `IVpnKeypair` | Base64-encoded public/private key pair |
|
| `IVpnKeypair` | Base64-encoded public/private key pair |
|
||||||
| `IDestinationPolicy` | Destination routing policy (forceTarget / block / allow with allow/block lists) |
|
| `IDestinationPolicy` | Destination routing policy (forceTarget / block / allow with allow/block lists) |
|
||||||
| `IVpnEventMap` | Typed event map for server and client EventEmitter |
|
| `IVpnEventMap` | Exported event payload shapes for lifecycle and daemon event integrations |
|
||||||
|
|
||||||
### Server IPC Commands
|
### Server IPC Commands
|
||||||
|
|
||||||
@@ -515,8 +584,8 @@ All transport modes share the same `forwardingMode` — WireGuard peers can use
|
|||||||
// Explicit QUIC with certificate pinning
|
// Explicit QUIC with certificate pinning
|
||||||
{ transport: 'quic', serverUrl: '1.2.3.4:4433', serverCertHash: '<sha256-base64>' }
|
{ transport: 'quic', serverUrl: '1.2.3.4:4433', serverCertHash: '<sha256-base64>' }
|
||||||
|
|
||||||
// WireGuard
|
// WireGuard clients use the standard .conf returned by createClient()
|
||||||
{ transport: 'wireguard', wgPrivateKey: '...', wgEndpoint: 'vpn.example.com:51820', ... }
|
// or generated via WgConfigGenerator.
|
||||||
```
|
```
|
||||||
|
|
||||||
## Cryptography 🔑
|
## Cryptography 🔑
|
||||||
@@ -525,7 +594,7 @@ All transport modes share the same `forwardingMode` — WireGuard peers can use
|
|||||||
|-------|-----------|---------|
|
|-------|-----------|---------|
|
||||||
| **Handshake** | Noise IK (X25519 + ChaChaPoly + BLAKE2s) | Mutual authentication + key exchange |
|
| **Handshake** | Noise IK (X25519 + ChaChaPoly + BLAKE2s) | Mutual authentication + key exchange |
|
||||||
| **Transport** | Noise transport state (ChaChaPoly) | All post-handshake data encryption |
|
| **Transport** | Noise transport state (ChaChaPoly) | All post-handshake data encryption |
|
||||||
| **Additional** | XChaCha20-Poly1305 | Extended nonce space for data-at-rest |
|
| **Utility** | XChaCha20-Poly1305 helper | Nonce-safe symmetric encryption helper in the Rust crypto module |
|
||||||
| **WireGuard** | X25519 + ChaCha20-Poly1305 (via boringtun) | Standard WireGuard crypto |
|
| **WireGuard** | X25519 + ChaCha20-Poly1305 (via boringtun) | Standard WireGuard crypto |
|
||||||
|
|
||||||
## Binary Protocol 📡
|
## Binary Protocol 📡
|
||||||
@@ -539,6 +608,9 @@ All frames use `[type:1B][length:4B][payload:NB]` with a 64KB max payload:
|
|||||||
| IpPacket | `0x10` | Bidirectional | Encrypted tunnel data |
|
| IpPacket | `0x10` | Bidirectional | Encrypted tunnel data |
|
||||||
| Keepalive | `0x20` | Client → Server | App-level keepalive (not WS ping) |
|
| Keepalive | `0x20` | Client → Server | App-level keepalive (not WS ping) |
|
||||||
| KeepaliveAck | `0x21` | Server → Client | Keepalive response with RTT payload |
|
| KeepaliveAck | `0x21` | Server → Client | Keepalive response with RTT payload |
|
||||||
|
| SessionResume | `0x30` | Client → Server | Session resume attempt |
|
||||||
|
| SessionResumeOk | `0x31` | Server → Client | Session resume accepted |
|
||||||
|
| SessionResumeErr | `0x32` | Server → Client | Session resume rejected |
|
||||||
| Disconnect | `0x3F` | Bidirectional | Graceful disconnect |
|
| Disconnect | `0x3F` | Bidirectional | Graceful disconnect |
|
||||||
|
|
||||||
## Development 🛠️
|
## Development 🛠️
|
||||||
@@ -589,6 +661,7 @@ smartvpn/
|
|||||||
│ ├── transport_trait.rs # Transport abstraction (Sink/Stream)
|
│ ├── transport_trait.rs # Transport abstraction (Sink/Stream)
|
||||||
│ ├── quic_transport.rs # QUIC transport
|
│ ├── quic_transport.rs # QUIC transport
|
||||||
│ ├── wireguard.rs # WireGuard (boringtun)
|
│ ├── wireguard.rs # WireGuard (boringtun)
|
||||||
|
│ ├── bridge.rs # Linux bridge/TAP integration
|
||||||
│ ├── codec.rs # Binary frame protocol
|
│ ├── codec.rs # Binary frame protocol
|
||||||
│ ├── keepalive.rs # Adaptive keepalives
|
│ ├── keepalive.rs # Adaptive keepalives
|
||||||
│ ├── ratelimit.rs # Token bucket
|
│ ├── ratelimit.rs # Token bucket
|
||||||
@@ -606,7 +679,7 @@ smartvpn/
|
|||||||
|
|
||||||
## License and Legal Information
|
## License and Legal Information
|
||||||
|
|
||||||
This repository contains open-source code licensed under the MIT License. A copy of the license can be found in the [LICENSE](./LICENSE) file.
|
This repository contains open-source code licensed under the MIT License. A copy of the license can be found in the [license.md](./license.md) file.
|
||||||
|
|
||||||
**Please note:** The MIT License does not grant permission to use the trade names, trademarks, service marks, or product names of the project, except as required for reasonable and customary use in describing the origin of the work and reproducing the content of the NOTICE file.
|
**Please note:** The MIT License does not grant permission to use the trade names, trademarks, service marks, or product names of the project, except as required for reasonable and customary use in describing the origin of the work and reproducing the content of the NOTICE file.
|
||||||
|
|
||||||
|
|||||||
@@ -164,6 +164,7 @@ mod tests {
|
|||||||
destination_block_list: dst_block.map(|v| v.into_iter().map(String::from).collect()),
|
destination_block_list: dst_block.map(|v| v.into_iter().map(String::from).collect()),
|
||||||
max_connections: None,
|
max_connections: None,
|
||||||
rate_limit: None,
|
rate_limit: None,
|
||||||
|
destination_policy: None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,396 @@
|
|||||||
|
//! L2 Bridge forwarding engine.
|
||||||
|
//!
|
||||||
|
//! Provides server-side bridging: receives L3 IP packets from VPN clients,
|
||||||
|
//! wraps them in Ethernet frames, and injects them into a Linux bridge
|
||||||
|
//! connected to the host's physical network interface.
|
||||||
|
//!
|
||||||
|
//! Return traffic from the bridge is stripped of its Ethernet header and
|
||||||
|
//! routed back to VPN clients via `tun_routes`.
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use std::net::Ipv4Addr;
|
||||||
|
use std::sync::Arc;
|
||||||
|
use tokio::io::{AsyncReadExt, AsyncWriteExt};
|
||||||
|
use tokio::sync::mpsc;
|
||||||
|
use tracing::{debug, info, warn};
|
||||||
|
|
||||||
|
use crate::server::ServerState;
|
||||||
|
|
||||||
|
/// Configuration for the bridge forwarding engine.
|
||||||
|
pub struct BridgeConfig {
|
||||||
|
/// TAP device name (e.g., "svpn_tap0")
|
||||||
|
pub tap_name: String,
|
||||||
|
/// Linux bridge name (e.g., "svpn_br0")
|
||||||
|
pub bridge_name: String,
|
||||||
|
/// Physical interface to bridge (e.g., "eth0")
|
||||||
|
pub physical_interface: String,
|
||||||
|
/// Gateway IP on the bridge (host's LAN IP)
|
||||||
|
pub gateway_ip: Ipv4Addr,
|
||||||
|
/// Subnet prefix length (e.g., 24)
|
||||||
|
pub prefix_len: u8,
|
||||||
|
/// MTU for the TAP device
|
||||||
|
pub mtu: u16,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ethernet frame constants
|
||||||
|
const ETH_HEADER_LEN: usize = 14;
|
||||||
|
const ETH_TYPE_IPV4: [u8; 2] = [0x08, 0x00];
|
||||||
|
const ETH_TYPE_ARP: [u8; 2] = [0x08, 0x06];
|
||||||
|
const BROADCAST_MAC: [u8; 6] = [0xff; 6];
|
||||||
|
|
||||||
|
/// Generate a deterministic locally-administered MAC from an IPv4 address.
|
||||||
|
/// Uses prefix 02:53:56 (locally administered, "SVP" in hex-ish).
|
||||||
|
fn mac_from_ip(ip: Ipv4Addr) -> [u8; 6] {
|
||||||
|
let octets = ip.octets();
|
||||||
|
[0x02, 0x53, 0x56, octets[1], octets[2], octets[3]]
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Wrap an IP packet in an Ethernet frame.
|
||||||
|
fn wrap_in_ethernet(ip_packet: &[u8], src_mac: [u8; 6], dst_mac: [u8; 6]) -> Vec<u8> {
|
||||||
|
let mut frame = Vec::with_capacity(ETH_HEADER_LEN + ip_packet.len());
|
||||||
|
frame.extend_from_slice(&dst_mac);
|
||||||
|
frame.extend_from_slice(&src_mac);
|
||||||
|
frame.extend_from_slice(Ð_TYPE_IPV4);
|
||||||
|
frame.extend_from_slice(ip_packet);
|
||||||
|
frame
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Extract the EtherType and payload from an Ethernet frame.
|
||||||
|
fn unwrap_ethernet(frame: &[u8]) -> Option<([u8; 2], &[u8])> {
|
||||||
|
if frame.len() < ETH_HEADER_LEN {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let ether_type = [frame[12], frame[13]];
|
||||||
|
Some((ether_type, &frame[ETH_HEADER_LEN..]))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Extract destination IPv4 from a raw IP packet header.
|
||||||
|
fn dst_ip_from_packet(packet: &[u8]) -> Option<Ipv4Addr> {
|
||||||
|
if packet.len() < 20 {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
// Version must be 4
|
||||||
|
if (packet[0] >> 4) != 4 {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
Some(Ipv4Addr::new(packet[16], packet[17], packet[18], packet[19]))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Extract source IPv4 from a raw IP packet header.
|
||||||
|
fn src_ip_from_packet(packet: &[u8]) -> Option<Ipv4Addr> {
|
||||||
|
if packet.len() < 20 {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
if (packet[0] >> 4) != 4 {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
Some(Ipv4Addr::new(packet[12], packet[13], packet[14], packet[15]))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Build a gratuitous ARP announcement frame.
|
||||||
|
fn build_garp(ip: Ipv4Addr, mac: [u8; 6]) -> Vec<u8> {
|
||||||
|
let ip_bytes = ip.octets();
|
||||||
|
let mut frame = Vec::with_capacity(42); // 14 eth + 28 ARP
|
||||||
|
// Ethernet header
|
||||||
|
frame.extend_from_slice(&BROADCAST_MAC); // dst: broadcast
|
||||||
|
frame.extend_from_slice(&mac); // src: our MAC
|
||||||
|
frame.extend_from_slice(Ð_TYPE_ARP); // EtherType: ARP
|
||||||
|
// ARP payload
|
||||||
|
frame.extend_from_slice(&[0x00, 0x01]); // Hardware type: Ethernet
|
||||||
|
frame.extend_from_slice(&[0x08, 0x00]); // Protocol type: IPv4
|
||||||
|
frame.push(6); // Hardware addr len
|
||||||
|
frame.push(4); // Protocol addr len
|
||||||
|
frame.extend_from_slice(&[0x00, 0x01]); // Operation: ARP Request (GARP uses request)
|
||||||
|
frame.extend_from_slice(&mac); // Sender hardware addr
|
||||||
|
frame.extend_from_slice(&ip_bytes); // Sender protocol addr
|
||||||
|
frame.extend_from_slice(&[0x00; 6]); // Target hardware addr (ignored in GARP)
|
||||||
|
frame.extend_from_slice(&ip_bytes); // Target protocol addr (same as sender for GARP)
|
||||||
|
frame
|
||||||
|
}
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// Linux bridge management (ip commands)
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
async fn run_ip_cmd(args: &[&str]) -> Result<String> {
|
||||||
|
let output = tokio::process::Command::new("ip")
|
||||||
|
.args(args)
|
||||||
|
.output()
|
||||||
|
.await?;
|
||||||
|
if !output.status.success() {
|
||||||
|
let stderr = String::from_utf8_lossy(&output.stderr);
|
||||||
|
anyhow::bail!("ip {} failed: {}", args.join(" "), stderr.trim());
|
||||||
|
}
|
||||||
|
Ok(String::from_utf8_lossy(&output.stdout).to_string())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a Linux bridge interface.
|
||||||
|
pub async fn create_bridge(name: &str) -> Result<()> {
|
||||||
|
run_ip_cmd(&["link", "add", name, "type", "bridge"]).await?;
|
||||||
|
info!("Created bridge {}", name);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Add an interface to a bridge.
|
||||||
|
pub async fn bridge_add_interface(bridge: &str, iface: &str) -> Result<()> {
|
||||||
|
run_ip_cmd(&["link", "set", iface, "master", bridge]).await?;
|
||||||
|
info!("Added {} to bridge {}", iface, bridge);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Bring an interface up.
|
||||||
|
pub async fn set_interface_up(iface: &str) -> Result<()> {
|
||||||
|
run_ip_cmd(&["link", "set", iface, "up"]).await?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Remove a bridge interface.
|
||||||
|
pub async fn remove_bridge(name: &str) -> Result<()> {
|
||||||
|
// First bring it down, ignore errors
|
||||||
|
let _ = run_ip_cmd(&["link", "set", name, "down"]).await;
|
||||||
|
run_ip_cmd(&["link", "del", name]).await?;
|
||||||
|
info!("Removed bridge {}", name);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Detect the default network interface from the routing table.
|
||||||
|
pub async fn detect_default_interface() -> Result<String> {
|
||||||
|
let output = run_ip_cmd(&["route", "show", "default"]).await?;
|
||||||
|
// Format: "default via X.X.X.X dev IFACE ..."
|
||||||
|
let parts: Vec<&str> = output.split_whitespace().collect();
|
||||||
|
if let Some(idx) = parts.iter().position(|&s| s == "dev") {
|
||||||
|
if let Some(iface) = parts.get(idx + 1) {
|
||||||
|
return Ok(iface.to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
anyhow::bail!("Could not detect default network interface from route table");
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get the IP address and prefix length of a network interface.
|
||||||
|
pub async fn get_interface_ip(iface: &str) -> Result<(Ipv4Addr, u8)> {
|
||||||
|
let output = run_ip_cmd(&["-4", "addr", "show", "dev", iface]).await?;
|
||||||
|
// Parse "inet X.X.X.X/NN" from output
|
||||||
|
for line in output.lines() {
|
||||||
|
let trimmed = line.trim();
|
||||||
|
if let Some(rest) = trimmed.strip_prefix("inet ") {
|
||||||
|
let addr_cidr = rest.split_whitespace().next().unwrap_or("");
|
||||||
|
let parts: Vec<&str> = addr_cidr.split('/').collect();
|
||||||
|
if parts.len() == 2 {
|
||||||
|
let ip: Ipv4Addr = parts[0].parse()?;
|
||||||
|
let prefix: u8 = parts[1].parse()?;
|
||||||
|
return Ok((ip, prefix));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
anyhow::bail!("Could not find IPv4 address on interface {}", iface);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Migrate the host's IP from a physical interface to a bridge.
|
||||||
|
/// This is the most delicate operation — briefly interrupts connectivity.
|
||||||
|
pub async fn migrate_host_ip_to_bridge(
|
||||||
|
physical_iface: &str,
|
||||||
|
bridge: &str,
|
||||||
|
ip: Ipv4Addr,
|
||||||
|
prefix: u8,
|
||||||
|
) -> Result<()> {
|
||||||
|
let cidr = format!("{}/{}", ip, prefix);
|
||||||
|
// Remove IP from physical interface
|
||||||
|
let _ = run_ip_cmd(&["addr", "del", &cidr, "dev", physical_iface]).await;
|
||||||
|
// Add IP to bridge
|
||||||
|
run_ip_cmd(&["addr", "add", &cidr, "dev", bridge]).await?;
|
||||||
|
info!("Migrated IP {} from {} to {}", cidr, physical_iface, bridge);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Restore the host's IP from bridge back to the physical interface.
|
||||||
|
pub async fn restore_host_ip(
|
||||||
|
physical_iface: &str,
|
||||||
|
bridge: &str,
|
||||||
|
ip: Ipv4Addr,
|
||||||
|
prefix: u8,
|
||||||
|
) -> Result<()> {
|
||||||
|
let cidr = format!("{}/{}", ip, prefix);
|
||||||
|
let _ = run_ip_cmd(&["addr", "del", &cidr, "dev", bridge]).await;
|
||||||
|
run_ip_cmd(&["addr", "add", &cidr, "dev", physical_iface]).await?;
|
||||||
|
info!("Restored IP {} to {}", cidr, physical_iface);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Enable proxy ARP on an interface via sysctl.
|
||||||
|
pub async fn enable_proxy_arp(iface: &str) -> Result<()> {
|
||||||
|
let path = format!("/proc/sys/net/ipv4/conf/{}/proxy_arp", iface);
|
||||||
|
tokio::fs::write(&path, "1").await?;
|
||||||
|
info!("Enabled proxy_arp on {}", iface);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// VLAN support (802.1Q via Linux bridge VLAN filtering)
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
async fn run_bridge_cmd(args: &[&str]) -> Result<String> {
|
||||||
|
let output = tokio::process::Command::new("bridge")
|
||||||
|
.args(args)
|
||||||
|
.output()
|
||||||
|
.await?;
|
||||||
|
if !output.status.success() {
|
||||||
|
let stderr = String::from_utf8_lossy(&output.stderr);
|
||||||
|
anyhow::bail!("bridge {} failed: {}", args.join(" "), stderr.trim());
|
||||||
|
}
|
||||||
|
Ok(String::from_utf8_lossy(&output.stdout).to_string())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Enable VLAN filtering on a bridge.
|
||||||
|
pub async fn enable_vlan_filtering(bridge: &str) -> Result<()> {
|
||||||
|
run_ip_cmd(&["link", "set", bridge, "type", "bridge", "vlan_filtering", "1"]).await?;
|
||||||
|
info!("Enabled VLAN filtering on bridge {}", bridge);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Add a VLAN ID to a bridge port (TAP or physical interface).
|
||||||
|
/// `pvid` = set as port VLAN ID (untagged ingress), `untagged` = strip tag on egress.
|
||||||
|
pub async fn add_vlan_to_port(port: &str, vlan_id: u16, pvid: bool, untagged: bool) -> Result<()> {
|
||||||
|
let mut args = vec!["vlan", "add", "dev", port, "vid"];
|
||||||
|
let vid_str = vlan_id.to_string();
|
||||||
|
args.push(&vid_str);
|
||||||
|
if pvid { args.push("pvid"); }
|
||||||
|
if untagged { args.push("untagged"); }
|
||||||
|
run_bridge_cmd(&args).await?;
|
||||||
|
info!("Added VLAN {} to port {} (pvid={}, untagged={})", vlan_id, port, pvid, untagged);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Remove a VLAN ID from a bridge port.
|
||||||
|
pub async fn remove_vlan_from_port(port: &str, vlan_id: u16) -> Result<()> {
|
||||||
|
let vid_str = vlan_id.to_string();
|
||||||
|
run_bridge_cmd(&["vlan", "del", "dev", port, "vid", &vid_str]).await?;
|
||||||
|
info!("Removed VLAN {} from port {}", vlan_id, port);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a TAP device (L2) using the tun crate.
|
||||||
|
pub fn create_tap(name: &str, mtu: u16) -> Result<tun::AsyncDevice> {
|
||||||
|
let mut config = tun::Configuration::default();
|
||||||
|
config
|
||||||
|
.tun_name(name)
|
||||||
|
.layer(tun::Layer::L2)
|
||||||
|
.mtu(mtu)
|
||||||
|
.up();
|
||||||
|
|
||||||
|
#[cfg(target_os = "linux")]
|
||||||
|
config.platform_config(|p| {
|
||||||
|
p.ensure_root_privileges(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
let device = tun::create_as_async(&config)?;
|
||||||
|
info!("TAP device {} created (L2, mtu={})", name, mtu);
|
||||||
|
Ok(device)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// BridgeEngine — main event loop
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
|
/// The BridgeEngine wraps/unwraps Ethernet frames and bridges VPN traffic
|
||||||
|
/// to the host's physical LAN via a Linux bridge + TAP device.
|
||||||
|
pub struct BridgeEngine {
|
||||||
|
state: Arc<ServerState>,
|
||||||
|
/// Learned MAC addresses for LAN peers (dst IP → MAC).
|
||||||
|
/// Populated from ARP replies and Ethernet frame src MACs.
|
||||||
|
arp_cache: HashMap<Ipv4Addr, [u8; 6]>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl BridgeEngine {
|
||||||
|
pub fn new(state: Arc<ServerState>) -> Self {
|
||||||
|
Self {
|
||||||
|
state,
|
||||||
|
arp_cache: HashMap::new(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Run the bridge engine event loop.
|
||||||
|
/// Receives L3 IP packets from VPN clients, wraps in Ethernet, writes to TAP.
|
||||||
|
/// Reads Ethernet frames from TAP, strips header, routes back to VPN clients.
|
||||||
|
pub async fn run(
|
||||||
|
mut self,
|
||||||
|
mut tap_device: tun::AsyncDevice,
|
||||||
|
mut packet_rx: mpsc::Receiver<Vec<u8>>,
|
||||||
|
mut shutdown_rx: mpsc::Receiver<()>,
|
||||||
|
) -> Result<()> {
|
||||||
|
let mut buf = vec![0u8; 2048];
|
||||||
|
|
||||||
|
info!("BridgeEngine started");
|
||||||
|
|
||||||
|
loop {
|
||||||
|
tokio::select! {
|
||||||
|
// Packet from VPN client → wrap in Ethernet → write to TAP
|
||||||
|
Some(ip_packet) = packet_rx.recv() => {
|
||||||
|
if let Some(dst_ip) = dst_ip_from_packet(&ip_packet) {
|
||||||
|
let src_ip = src_ip_from_packet(&ip_packet).unwrap_or(Ipv4Addr::UNSPECIFIED);
|
||||||
|
let src_mac = mac_from_ip(src_ip);
|
||||||
|
let dst_mac = self.arp_cache.get(&dst_ip)
|
||||||
|
.copied()
|
||||||
|
.unwrap_or(BROADCAST_MAC);
|
||||||
|
let frame = wrap_in_ethernet(&ip_packet, src_mac, dst_mac);
|
||||||
|
if let Err(e) = tap_device.write_all(&frame).await {
|
||||||
|
warn!("TAP write error: {}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Frame from TAP (LAN) → strip Ethernet → route to VPN client
|
||||||
|
result = tap_device.read(&mut buf) => {
|
||||||
|
match result {
|
||||||
|
Ok(len) if len >= ETH_HEADER_LEN => {
|
||||||
|
let frame = &buf[..len];
|
||||||
|
|
||||||
|
// Learn src MAC from incoming frames
|
||||||
|
if let Some((ether_type, payload)) = unwrap_ethernet(frame) {
|
||||||
|
// Learn ARP cache from src MAC + src IP
|
||||||
|
let src_mac: [u8; 6] = frame[6..12].try_into().unwrap_or([0; 6]);
|
||||||
|
if ether_type == ETH_TYPE_IPV4 {
|
||||||
|
if let Some(src_ip) = src_ip_from_packet(payload) {
|
||||||
|
self.arp_cache.insert(src_ip, src_mac);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Only forward IPv4 packets to VPN clients
|
||||||
|
if ether_type == ETH_TYPE_IPV4 {
|
||||||
|
if let Some(dst_ip) = dst_ip_from_packet(payload) {
|
||||||
|
// Look up VPN client by dst IP in tun_routes
|
||||||
|
let routes = self.state.tun_routes.read().await;
|
||||||
|
if let Some(sender) = routes.get(&dst_ip) {
|
||||||
|
let _ = sender.try_send(payload.to_vec());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(_) => {} // Frame too short, ignore
|
||||||
|
Err(e) => {
|
||||||
|
warn!("TAP read error: {}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
_ = shutdown_rx.recv() => {
|
||||||
|
info!("BridgeEngine shutting down");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Send a gratuitous ARP for a VPN client IP.
|
||||||
|
pub async fn announce_client(tap: &mut tun::AsyncDevice, ip: Ipv4Addr) -> Result<()> {
|
||||||
|
let mac = mac_from_ip(ip);
|
||||||
|
let garp = build_garp(ip, mac);
|
||||||
|
tap.write_all(&garp).await?;
|
||||||
|
debug!("Sent GARP for {} (MAC {:02x}:{:02x}:{:02x}:{:02x}:{:02x}:{:02x})",
|
||||||
|
ip, mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
+120
-3
@@ -26,6 +26,9 @@ pub struct ClientSecurity {
|
|||||||
pub max_connections: Option<u32>,
|
pub max_connections: Option<u32>,
|
||||||
/// Per-client rate limiting.
|
/// Per-client rate limiting.
|
||||||
pub rate_limit: Option<ClientRateLimit>,
|
pub rate_limit: Option<ClientRateLimit>,
|
||||||
|
/// Per-client destination routing policy override.
|
||||||
|
/// When set, overrides the server-level DestinationPolicy for this client's traffic.
|
||||||
|
pub destination_policy: Option<crate::server::DestinationPolicyConfig>,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// A registered client entry — the server-side source of truth.
|
/// A registered client entry — the server-side source of truth.
|
||||||
@@ -57,6 +60,19 @@ pub struct ClientEntry {
|
|||||||
pub expires_at: Option<String>,
|
pub expires_at: Option<String>,
|
||||||
/// Assigned VPN IP address.
|
/// Assigned VPN IP address.
|
||||||
pub assigned_ip: Option<String>,
|
pub assigned_ip: Option<String>,
|
||||||
|
|
||||||
|
// Per-client bridge/host-IP settings
|
||||||
|
|
||||||
|
/// If true, client gets a host network IP via bridge mode.
|
||||||
|
pub use_host_ip: Option<bool>,
|
||||||
|
/// If true and use_host_ip is true, obtain IP via DHCP relay.
|
||||||
|
pub use_dhcp: Option<bool>,
|
||||||
|
/// Static LAN IP when use_host_ip is true and use_dhcp is false.
|
||||||
|
pub static_ip: Option<String>,
|
||||||
|
/// If true, assign this client to a specific 802.1Q VLAN.
|
||||||
|
pub force_vlan: Option<bool>,
|
||||||
|
/// 802.1Q VLAN ID (1-4094).
|
||||||
|
pub vlan_id: Option<u16>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ClientEntry {
|
impl ClientEntry {
|
||||||
@@ -76,12 +92,16 @@ impl ClientEntry {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// In-memory client registry with dual-key indexing.
|
/// In-memory client registry with triple-key indexing.
|
||||||
pub struct ClientRegistry {
|
pub struct ClientRegistry {
|
||||||
/// Primary index: clientId → ClientEntry
|
/// Primary index: clientId → ClientEntry
|
||||||
entries: HashMap<String, ClientEntry>,
|
entries: HashMap<String, ClientEntry>,
|
||||||
/// Secondary index: publicKey (base64) → clientId (fast lookup during handshake)
|
/// Secondary index: publicKey (base64) → clientId (fast lookup during handshake)
|
||||||
key_index: HashMap<String, String>,
|
key_index: HashMap<String, String>,
|
||||||
|
/// WireGuard public key → clientId (fast lookup during WG handshakes)
|
||||||
|
wg_key_index: HashMap<String, String>,
|
||||||
|
/// Tertiary index: assignedIp → clientId (fast lookup during NAT destination policy)
|
||||||
|
ip_index: HashMap<String, String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ClientRegistry {
|
impl ClientRegistry {
|
||||||
@@ -89,6 +109,8 @@ impl ClientRegistry {
|
|||||||
Self {
|
Self {
|
||||||
entries: HashMap::new(),
|
entries: HashMap::new(),
|
||||||
key_index: HashMap::new(),
|
key_index: HashMap::new(),
|
||||||
|
wg_key_index: HashMap::new(),
|
||||||
|
ip_index: HashMap::new(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -113,7 +135,16 @@ impl ClientRegistry {
|
|||||||
if self.key_index.contains_key(&entry.public_key) {
|
if self.key_index.contains_key(&entry.public_key) {
|
||||||
anyhow::bail!("Public key already registered to another client");
|
anyhow::bail!("Public key already registered to another client");
|
||||||
}
|
}
|
||||||
|
if let Some(ref wg_key) = entry.wg_public_key {
|
||||||
|
if self.wg_key_index.contains_key(wg_key) {
|
||||||
|
anyhow::bail!("WireGuard public key already registered to another client");
|
||||||
|
}
|
||||||
|
self.wg_key_index.insert(wg_key.clone(), entry.client_id.clone());
|
||||||
|
}
|
||||||
self.key_index.insert(entry.public_key.clone(), entry.client_id.clone());
|
self.key_index.insert(entry.public_key.clone(), entry.client_id.clone());
|
||||||
|
if let Some(ref ip) = entry.assigned_ip {
|
||||||
|
self.ip_index.insert(ip.clone(), entry.client_id.clone());
|
||||||
|
}
|
||||||
self.entries.insert(entry.client_id.clone(), entry);
|
self.entries.insert(entry.client_id.clone(), entry);
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
@@ -123,6 +154,12 @@ impl ClientRegistry {
|
|||||||
let entry = self.entries.remove(client_id)
|
let entry = self.entries.remove(client_id)
|
||||||
.ok_or_else(|| anyhow::anyhow!("Client '{}' not found", client_id))?;
|
.ok_or_else(|| anyhow::anyhow!("Client '{}' not found", client_id))?;
|
||||||
self.key_index.remove(&entry.public_key);
|
self.key_index.remove(&entry.public_key);
|
||||||
|
if let Some(ref wg_key) = entry.wg_public_key {
|
||||||
|
self.wg_key_index.remove(wg_key);
|
||||||
|
}
|
||||||
|
if let Some(ref ip) = entry.assigned_ip {
|
||||||
|
self.ip_index.remove(ip);
|
||||||
|
}
|
||||||
Ok(entry)
|
Ok(entry)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -137,6 +174,18 @@ impl ClientRegistry {
|
|||||||
self.entries.get(client_id)
|
self.entries.get(client_id)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Get a client by WireGuard public key.
|
||||||
|
pub fn get_by_wg_key(&self, public_key: &str) -> Option<&ClientEntry> {
|
||||||
|
let client_id = self.wg_key_index.get(public_key)?;
|
||||||
|
self.entries.get(client_id)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get a client by assigned IP (used for per-client destination policy in NAT engine).
|
||||||
|
pub fn get_by_assigned_ip(&self, ip: &str) -> Option<&ClientEntry> {
|
||||||
|
let client_id = self.ip_index.get(ip)?;
|
||||||
|
self.entries.get(client_id)
|
||||||
|
}
|
||||||
|
|
||||||
/// Check if a public key is authorized (exists, enabled, not expired).
|
/// Check if a public key is authorized (exists, enabled, not expired).
|
||||||
pub fn is_authorized(&self, public_key: &str) -> bool {
|
pub fn is_authorized(&self, public_key: &str) -> bool {
|
||||||
match self.get_by_key(public_key) {
|
match self.get_by_key(public_key) {
|
||||||
@@ -153,12 +202,32 @@ impl ClientRegistry {
|
|||||||
let entry = self.entries.get_mut(client_id)
|
let entry = self.entries.get_mut(client_id)
|
||||||
.ok_or_else(|| anyhow::anyhow!("Client '{}' not found", client_id))?;
|
.ok_or_else(|| anyhow::anyhow!("Client '{}' not found", client_id))?;
|
||||||
let old_key = entry.public_key.clone();
|
let old_key = entry.public_key.clone();
|
||||||
|
let old_wg_key = entry.wg_public_key.clone();
|
||||||
|
let old_ip = entry.assigned_ip.clone();
|
||||||
updater(entry);
|
updater(entry);
|
||||||
// If public key changed, update the index
|
// If public key changed, update the key index
|
||||||
if entry.public_key != old_key {
|
if entry.public_key != old_key {
|
||||||
self.key_index.remove(&old_key);
|
self.key_index.remove(&old_key);
|
||||||
self.key_index.insert(entry.public_key.clone(), client_id.to_string());
|
self.key_index.insert(entry.public_key.clone(), client_id.to_string());
|
||||||
}
|
}
|
||||||
|
// If WireGuard public key changed, update the WG key index.
|
||||||
|
if entry.wg_public_key != old_wg_key {
|
||||||
|
if let Some(ref old) = old_wg_key {
|
||||||
|
self.wg_key_index.remove(old);
|
||||||
|
}
|
||||||
|
if let Some(ref new_key) = entry.wg_public_key {
|
||||||
|
self.wg_key_index.insert(new_key.clone(), client_id.to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// If assigned IP changed, update the IP index
|
||||||
|
if entry.assigned_ip != old_ip {
|
||||||
|
if let Some(ref old) = old_ip {
|
||||||
|
self.ip_index.remove(old);
|
||||||
|
}
|
||||||
|
if let Some(ref new_ip) = entry.assigned_ip {
|
||||||
|
self.ip_index.insert(new_ip.clone(), client_id.to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -169,13 +238,32 @@ impl ClientRegistry {
|
|||||||
|
|
||||||
/// Rotate a client's keys. Returns the updated entry.
|
/// Rotate a client's keys. Returns the updated entry.
|
||||||
pub fn rotate_key(&mut self, client_id: &str, new_public_key: String, new_wg_public_key: Option<String>) -> Result<()> {
|
pub fn rotate_key(&mut self, client_id: &str, new_public_key: String, new_wg_public_key: Option<String>) -> Result<()> {
|
||||||
|
if let Some(existing_client_id) = self.key_index.get(&new_public_key) {
|
||||||
|
if existing_client_id != client_id {
|
||||||
|
anyhow::bail!("Public key already registered to another client");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if let Some(ref new_wg_key) = new_wg_public_key {
|
||||||
|
if let Some(existing_client_id) = self.wg_key_index.get(new_wg_key) {
|
||||||
|
if existing_client_id != client_id {
|
||||||
|
anyhow::bail!("WireGuard public key already registered to another client");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
let entry = self.entries.get_mut(client_id)
|
let entry = self.entries.get_mut(client_id)
|
||||||
.ok_or_else(|| anyhow::anyhow!("Client '{}' not found", client_id))?;
|
.ok_or_else(|| anyhow::anyhow!("Client '{}' not found", client_id))?;
|
||||||
// Update key index
|
// Update key index
|
||||||
self.key_index.remove(&entry.public_key);
|
self.key_index.remove(&entry.public_key);
|
||||||
|
if let Some(ref old_wg_key) = entry.wg_public_key {
|
||||||
|
self.wg_key_index.remove(old_wg_key);
|
||||||
|
}
|
||||||
entry.public_key = new_public_key.clone();
|
entry.public_key = new_public_key.clone();
|
||||||
entry.wg_public_key = new_wg_public_key;
|
entry.wg_public_key = new_wg_public_key;
|
||||||
self.key_index.insert(new_public_key, client_id.to_string());
|
self.key_index.insert(new_public_key, client_id.to_string());
|
||||||
|
if let Some(ref wg_key) = entry.wg_public_key {
|
||||||
|
self.wg_key_index.insert(wg_key.clone(), client_id.to_string());
|
||||||
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -208,6 +296,11 @@ mod tests {
|
|||||||
description: None,
|
description: None,
|
||||||
expires_at: None,
|
expires_at: None,
|
||||||
assigned_ip: None,
|
assigned_ip: None,
|
||||||
|
use_host_ip: None,
|
||||||
|
use_dhcp: None,
|
||||||
|
static_ip: None,
|
||||||
|
force_vlan: None,
|
||||||
|
vlan_id: None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -321,13 +414,36 @@ mod tests {
|
|||||||
let mut reg = ClientRegistry::new();
|
let mut reg = ClientRegistry::new();
|
||||||
reg.add(make_entry("alice", "old_key")).unwrap();
|
reg.add(make_entry("alice", "old_key")).unwrap();
|
||||||
|
|
||||||
reg.rotate_key("alice", "new_key".to_string(), None).unwrap();
|
reg.rotate_key("alice", "new_key".to_string(), Some("new_wg_key".to_string())).unwrap();
|
||||||
|
|
||||||
assert!(reg.get_by_key("old_key").is_none());
|
assert!(reg.get_by_key("old_key").is_none());
|
||||||
assert!(reg.get_by_key("new_key").is_some());
|
assert!(reg.get_by_key("new_key").is_some());
|
||||||
|
assert!(reg.get_by_wg_key("new_wg_key").is_some());
|
||||||
assert_eq!(reg.get_by_id("alice").unwrap().public_key, "new_key");
|
assert_eq!(reg.get_by_id("alice").unwrap().public_key, "new_key");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn lookup_by_wireguard_key() {
|
||||||
|
let mut reg = ClientRegistry::new();
|
||||||
|
let mut entry = make_entry("alice", "key_alice");
|
||||||
|
entry.wg_public_key = Some("wg_key_alice".to_string());
|
||||||
|
reg.add(entry).unwrap();
|
||||||
|
|
||||||
|
assert_eq!(reg.get_by_wg_key("wg_key_alice").unwrap().client_id, "alice");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn reject_duplicate_wireguard_key() {
|
||||||
|
let mut reg = ClientRegistry::new();
|
||||||
|
let mut alice = make_entry("alice", "key_alice");
|
||||||
|
alice.wg_public_key = Some("same_wg_key".to_string());
|
||||||
|
let mut bob = make_entry("bob", "key_bob");
|
||||||
|
bob.wg_public_key = Some("same_wg_key".to_string());
|
||||||
|
|
||||||
|
reg.add(alice).unwrap();
|
||||||
|
assert!(reg.add(bob).is_err());
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn from_entries() {
|
fn from_entries() {
|
||||||
let entries = vec![
|
let entries = vec![
|
||||||
@@ -362,6 +478,7 @@ mod tests {
|
|||||||
bytes_per_sec: 1_000_000,
|
bytes_per_sec: 1_000_000,
|
||||||
burst_bytes: 2_000_000,
|
burst_bytes: 2_000_000,
|
||||||
}),
|
}),
|
||||||
|
destination_policy: None,
|
||||||
});
|
});
|
||||||
let mut reg = ClientRegistry::new();
|
let mut reg = ClientRegistry::new();
|
||||||
reg.add(entry).unwrap();
|
reg.add(entry).unwrap();
|
||||||
|
|||||||
@@ -22,3 +22,4 @@ pub mod client_registry;
|
|||||||
pub mod acl;
|
pub mod acl;
|
||||||
pub mod proxy_protocol;
|
pub mod proxy_protocol;
|
||||||
pub mod userspace_nat;
|
pub mod userspace_nat;
|
||||||
|
pub mod bridge;
|
||||||
|
|||||||
+58
-9
@@ -8,11 +8,16 @@ pub struct IpPool {
|
|||||||
/// Network address (e.g., 10.8.0.0)
|
/// Network address (e.g., 10.8.0.0)
|
||||||
network: Ipv4Addr,
|
network: Ipv4Addr,
|
||||||
/// Prefix length (e.g., 24)
|
/// Prefix length (e.g., 24)
|
||||||
|
#[allow(dead_code)]
|
||||||
prefix_len: u8,
|
prefix_len: u8,
|
||||||
/// Allocated IPs: IP -> client_id
|
/// Allocated IPs: IP -> client_id
|
||||||
allocated: HashMap<Ipv4Addr, String>,
|
allocated: HashMap<Ipv4Addr, String>,
|
||||||
/// Next candidate offset (skipping .0 network and .1 gateway)
|
/// Next candidate offset (skipping .0 network and .1 gateway)
|
||||||
next_offset: u32,
|
next_offset: u32,
|
||||||
|
/// Minimum allocation offset (inclusive). Default: 2 (skip .0 network and .1 gateway).
|
||||||
|
min_offset: u32,
|
||||||
|
/// Maximum allocation offset (exclusive). Default: broadcast offset.
|
||||||
|
max_offset: u32,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl IpPool {
|
impl IpPool {
|
||||||
@@ -28,11 +33,47 @@ impl IpPool {
|
|||||||
anyhow::bail!("Prefix too long for VPN pool: /{}", prefix_len);
|
anyhow::bail!("Prefix too long for VPN pool: /{}", prefix_len);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
let host_bits = 32 - prefix_len as u32;
|
||||||
|
let max_offset = (1u32 << host_bits) - 1; // broadcast offset
|
||||||
|
|
||||||
Ok(Self {
|
Ok(Self {
|
||||||
network,
|
network,
|
||||||
prefix_len,
|
prefix_len,
|
||||||
allocated: HashMap::new(),
|
allocated: HashMap::new(),
|
||||||
next_offset: 2, // Skip .0 (network) and .1 (server/gateway)
|
next_offset: 2, // Skip .0 (network) and .1 (server/gateway)
|
||||||
|
min_offset: 2,
|
||||||
|
max_offset,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a new IP pool with a restricted allocation range within the subnet.
|
||||||
|
/// `range_start` and `range_end` are host offsets (e.g., 200 and 250 for .200-.250).
|
||||||
|
pub fn new_with_range(subnet: &str, range_start: u32, range_end: u32) -> Result<Self> {
|
||||||
|
let parts: Vec<&str> = subnet.split('/').collect();
|
||||||
|
if parts.len() != 2 {
|
||||||
|
anyhow::bail!("Invalid subnet format: {}", subnet);
|
||||||
|
}
|
||||||
|
let network: Ipv4Addr = parts[0].parse()?;
|
||||||
|
let prefix_len: u8 = parts[1].parse()?;
|
||||||
|
if prefix_len > 30 {
|
||||||
|
anyhow::bail!("Prefix too long for VPN pool: /{}", prefix_len);
|
||||||
|
}
|
||||||
|
if range_start >= range_end {
|
||||||
|
anyhow::bail!("Invalid IP range: start ({}) must be less than end ({})", range_start, range_end);
|
||||||
|
}
|
||||||
|
let host_bits = 32 - prefix_len as u32;
|
||||||
|
let broadcast_offset = (1u32 << host_bits) - 1;
|
||||||
|
if range_end > broadcast_offset {
|
||||||
|
anyhow::bail!("IP range end ({}) exceeds subnet broadcast ({})", range_end, broadcast_offset);
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
network,
|
||||||
|
prefix_len,
|
||||||
|
allocated: HashMap::new(),
|
||||||
|
next_offset: range_start,
|
||||||
|
min_offset: range_start,
|
||||||
|
max_offset: range_end + 1, // exclusive
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -44,22 +85,17 @@ impl IpPool {
|
|||||||
|
|
||||||
/// Total number of usable client addresses in the pool.
|
/// Total number of usable client addresses in the pool.
|
||||||
pub fn capacity(&self) -> u32 {
|
pub fn capacity(&self) -> u32 {
|
||||||
let host_bits = 32 - self.prefix_len as u32;
|
self.max_offset.saturating_sub(self.min_offset)
|
||||||
let total = 1u32 << host_bits;
|
|
||||||
total.saturating_sub(3) // minus network, gateway, broadcast
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Allocate an IP for a client. Returns the assigned IP.
|
/// Allocate an IP for a client. Returns the assigned IP.
|
||||||
pub fn allocate(&mut self, client_id: &str) -> Result<Ipv4Addr> {
|
pub fn allocate(&mut self, client_id: &str) -> Result<Ipv4Addr> {
|
||||||
let host_bits = 32 - self.prefix_len as u32;
|
|
||||||
let max_offset = (1u32 << host_bits) - 1; // broadcast offset
|
|
||||||
|
|
||||||
// Try to find a free IP starting from next_offset
|
// Try to find a free IP starting from next_offset
|
||||||
let start = self.next_offset;
|
let start = self.next_offset;
|
||||||
let mut offset = start;
|
let mut offset = start;
|
||||||
loop {
|
loop {
|
||||||
if offset >= max_offset {
|
if offset >= self.max_offset {
|
||||||
offset = 2; // wrap around
|
offset = self.min_offset; // wrap around
|
||||||
}
|
}
|
||||||
|
|
||||||
let ip = Ipv4Addr::from(u32::from(self.network) + offset);
|
let ip = Ipv4Addr::from(u32::from(self.network) + offset);
|
||||||
@@ -88,7 +124,10 @@ impl IpPool {
|
|||||||
|
|
||||||
/// Reserve a specific IP for a client (e.g., WireGuard static IP from allowed_ips).
|
/// Reserve a specific IP for a client (e.g., WireGuard static IP from allowed_ips).
|
||||||
pub fn reserve(&mut self, ip: Ipv4Addr, client_id: &str) -> Result<()> {
|
pub fn reserve(&mut self, ip: Ipv4Addr, client_id: &str) -> Result<()> {
|
||||||
if self.allocated.contains_key(&ip) {
|
if let Some(existing_client_id) = self.allocated.get(&ip) {
|
||||||
|
if existing_client_id == client_id {
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
anyhow::bail!("IP {} is already allocated", ip);
|
anyhow::bail!("IP {} is already allocated", ip);
|
||||||
}
|
}
|
||||||
self.allocated.insert(ip, client_id.to_string());
|
self.allocated.insert(ip, client_id.to_string());
|
||||||
@@ -197,6 +236,16 @@ mod tests {
|
|||||||
assert!(pool.allocate("client2").is_err());
|
assert!(pool.allocate("client2").is_err());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn ip_pool_reserve_is_idempotent_for_same_client() {
|
||||||
|
let mut pool = IpPool::new("10.9.0.0/24").unwrap();
|
||||||
|
let ip = pool.allocate("alice").unwrap();
|
||||||
|
|
||||||
|
pool.reserve(ip, "alice").unwrap();
|
||||||
|
assert_eq!(pool.allocated_count(), 1);
|
||||||
|
assert!(pool.reserve(ip, "bob").is_err());
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn ip_pool_invalid_subnet() {
|
fn ip_pool_invalid_subnet() {
|
||||||
assert!(IpPool::new("invalid").is_err());
|
assert!(IpPool::new("invalid").is_err());
|
||||||
|
|||||||
@@ -4,6 +4,7 @@
|
|||||||
//! Spec: <https://www.haproxy.org/download/2.9/doc/proxy-protocol.txt>
|
//! Spec: <https://www.haproxy.org/download/2.9/doc/proxy-protocol.txt>
|
||||||
|
|
||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
|
use serde::Serialize;
|
||||||
use std::net::{Ipv4Addr, Ipv6Addr, SocketAddr, SocketAddrV4, SocketAddrV6};
|
use std::net::{Ipv4Addr, Ipv6Addr, SocketAddr, SocketAddrV4, SocketAddrV6};
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
use tokio::io::AsyncReadExt;
|
use tokio::io::AsyncReadExt;
|
||||||
@@ -17,6 +18,20 @@ const PP_V2_SIGNATURE: [u8; 12] = [
|
|||||||
0x0D, 0x0A, 0x0D, 0x0A, 0x00, 0x0D, 0x0A, 0x51, 0x55, 0x49, 0x54, 0x0A,
|
0x0D, 0x0A, 0x0D, 0x0A, 0x00, 0x0D, 0x0A, 0x51, 0x55, 0x49, 0x54, 0x0A,
|
||||||
];
|
];
|
||||||
|
|
||||||
|
/// Custom PROXY v2 TLV type for authenticated SmartVPN connection metadata.
|
||||||
|
/// 0xEA is in the PP2_TYPE_MIN_CUSTOM range (0xE0-0xEF).
|
||||||
|
pub const PP2_TYPE_SMARTVPN_METADATA: u8 = 0xEA;
|
||||||
|
|
||||||
|
/// Authenticated VPN metadata sent to trusted downstream proxies.
|
||||||
|
#[derive(Debug, Clone, Serialize)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct VpnProxyMetadata {
|
||||||
|
pub client_id: String,
|
||||||
|
pub assigned_ip: String,
|
||||||
|
pub transport_type: String,
|
||||||
|
pub remote_addr: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
/// Parsed PROXY protocol v2 header.
|
/// Parsed PROXY protocol v2 header.
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct ProxyHeader {
|
pub struct ProxyHeader {
|
||||||
@@ -124,6 +139,26 @@ async fn read_proxy_header_inner(stream: &mut TcpStream) -> Result<ProxyHeader>
|
|||||||
|
|
||||||
/// Build a PROXY protocol v2 header (for testing / proxy implementations).
|
/// Build a PROXY protocol v2 header (for testing / proxy implementations).
|
||||||
pub fn build_pp_v2_header(src: SocketAddr, dst: SocketAddr) -> Vec<u8> {
|
pub fn build_pp_v2_header(src: SocketAddr, dst: SocketAddr) -> Vec<u8> {
|
||||||
|
build_pp_v2_header_with_vpn_metadata(src, dst, None)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Build a PROXY protocol v2 header with optional SmartVPN metadata TLV.
|
||||||
|
pub fn build_pp_v2_header_with_vpn_metadata(
|
||||||
|
src: SocketAddr,
|
||||||
|
dst: SocketAddr,
|
||||||
|
vpn_metadata: Option<&VpnProxyMetadata>,
|
||||||
|
) -> Vec<u8> {
|
||||||
|
let mut tlv_bytes = Vec::new();
|
||||||
|
if let Some(metadata) = vpn_metadata {
|
||||||
|
if let Ok(json) = serde_json::to_vec(metadata) {
|
||||||
|
if json.len() <= u16::MAX as usize {
|
||||||
|
tlv_bytes.push(PP2_TYPE_SMARTVPN_METADATA);
|
||||||
|
tlv_bytes.extend_from_slice(&(json.len() as u16).to_be_bytes());
|
||||||
|
tlv_bytes.extend_from_slice(&json);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
let mut buf = Vec::new();
|
let mut buf = Vec::new();
|
||||||
buf.extend_from_slice(&PP_V2_SIGNATURE);
|
buf.extend_from_slice(&PP_V2_SIGNATURE);
|
||||||
|
|
||||||
@@ -131,22 +166,38 @@ pub fn build_pp_v2_header(src: SocketAddr, dst: SocketAddr) -> Vec<u8> {
|
|||||||
(SocketAddr::V4(s), SocketAddr::V4(d)) => {
|
(SocketAddr::V4(s), SocketAddr::V4(d)) => {
|
||||||
buf.push(0x21); // version 2 | PROXY command
|
buf.push(0x21); // version 2 | PROXY command
|
||||||
buf.push(0x11); // AF_INET | STREAM
|
buf.push(0x11); // AF_INET | STREAM
|
||||||
buf.extend_from_slice(&12u16.to_be_bytes()); // addr length
|
buf.extend_from_slice(&((12 + tlv_bytes.len()) as u16).to_be_bytes()); // addr length
|
||||||
buf.extend_from_slice(&s.ip().octets());
|
buf.extend_from_slice(&s.ip().octets());
|
||||||
buf.extend_from_slice(&d.ip().octets());
|
buf.extend_from_slice(&d.ip().octets());
|
||||||
buf.extend_from_slice(&s.port().to_be_bytes());
|
buf.extend_from_slice(&s.port().to_be_bytes());
|
||||||
buf.extend_from_slice(&d.port().to_be_bytes());
|
buf.extend_from_slice(&d.port().to_be_bytes());
|
||||||
|
buf.extend_from_slice(&tlv_bytes);
|
||||||
}
|
}
|
||||||
(SocketAddr::V6(s), SocketAddr::V6(d)) => {
|
(SocketAddr::V6(s), SocketAddr::V6(d)) => {
|
||||||
buf.push(0x21); // version 2 | PROXY command
|
buf.push(0x21); // version 2 | PROXY command
|
||||||
buf.push(0x21); // AF_INET6 | STREAM
|
buf.push(0x21); // AF_INET6 | STREAM
|
||||||
buf.extend_from_slice(&36u16.to_be_bytes()); // addr length
|
buf.extend_from_slice(&((36 + tlv_bytes.len()) as u16).to_be_bytes()); // addr length
|
||||||
buf.extend_from_slice(&s.ip().octets());
|
buf.extend_from_slice(&s.ip().octets());
|
||||||
buf.extend_from_slice(&d.ip().octets());
|
buf.extend_from_slice(&d.ip().octets());
|
||||||
buf.extend_from_slice(&s.port().to_be_bytes());
|
buf.extend_from_slice(&s.port().to_be_bytes());
|
||||||
buf.extend_from_slice(&d.port().to_be_bytes());
|
buf.extend_from_slice(&d.port().to_be_bytes());
|
||||||
|
buf.extend_from_slice(&tlv_bytes);
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
let src_v6 = match src.ip() {
|
||||||
|
std::net::IpAddr::V4(v4) => v4.to_ipv6_mapped(),
|
||||||
|
std::net::IpAddr::V6(v6) => v6,
|
||||||
|
};
|
||||||
|
let dst_v6 = match dst.ip() {
|
||||||
|
std::net::IpAddr::V4(v4) => v4.to_ipv6_mapped(),
|
||||||
|
std::net::IpAddr::V6(v6) => v6,
|
||||||
|
};
|
||||||
|
return build_pp_v2_header_with_vpn_metadata(
|
||||||
|
SocketAddr::V6(SocketAddrV6::new(src_v6, src.port(), 0, 0)),
|
||||||
|
SocketAddr::V6(SocketAddrV6::new(dst_v6, dst.port(), 0, 0)),
|
||||||
|
vpn_metadata,
|
||||||
|
);
|
||||||
}
|
}
|
||||||
_ => panic!("Mismatched address families"),
|
|
||||||
}
|
}
|
||||||
buf
|
buf
|
||||||
}
|
}
|
||||||
@@ -197,6 +248,30 @@ mod tests {
|
|||||||
assert_eq!(parsed.dst_addr, dst);
|
assert_eq!(parsed.dst_addr, dst);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn build_ipv4_header_with_smartvpn_metadata_tlv() {
|
||||||
|
let src = "203.0.113.50:12345".parse::<SocketAddr>().unwrap();
|
||||||
|
let dst = "10.0.0.1:443".parse::<SocketAddr>().unwrap();
|
||||||
|
let metadata = VpnProxyMetadata {
|
||||||
|
client_id: "alice".to_string(),
|
||||||
|
assigned_ip: "10.8.0.2".to_string(),
|
||||||
|
transport_type: "wireguard".to_string(),
|
||||||
|
remote_addr: Some("203.0.113.50:51820".to_string()),
|
||||||
|
};
|
||||||
|
let header = build_pp_v2_header_with_vpn_metadata(src, dst, Some(&metadata));
|
||||||
|
|
||||||
|
let addr_len = u16::from_be_bytes([header[14], header[15]]) as usize;
|
||||||
|
assert!(addr_len > 12);
|
||||||
|
assert_eq!(header[28], PP2_TYPE_SMARTVPN_METADATA);
|
||||||
|
let tlv_len = u16::from_be_bytes([header[29], header[30]]) as usize;
|
||||||
|
let json = std::str::from_utf8(&header[31..31 + tlv_len]).unwrap();
|
||||||
|
assert!(json.contains("\"clientId\":\"alice\""));
|
||||||
|
|
||||||
|
let parsed = parse_header_from_bytes(&header).await.unwrap();
|
||||||
|
assert_eq!(parsed.src_addr, src);
|
||||||
|
assert_eq!(parsed.dst_addr, dst);
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn parse_valid_ipv6_header() {
|
async fn parse_valid_ipv6_header() {
|
||||||
let src = "[2001:db8::1]:54321".parse::<SocketAddr>().unwrap();
|
let src = "[2001:db8::1]:54321".parse::<SocketAddr>().unwrap();
|
||||||
|
|||||||
+423
-34
@@ -25,7 +25,7 @@ use crate::tunnel::{self, TunConfig};
|
|||||||
const DEAD_PEER_TIMEOUT: Duration = Duration::from_secs(180);
|
const DEAD_PEER_TIMEOUT: Duration = Duration::from_secs(180);
|
||||||
|
|
||||||
/// Destination routing policy for VPN client traffic.
|
/// Destination routing policy for VPN client traffic.
|
||||||
#[derive(Debug, Clone, Deserialize)]
|
#[derive(Debug, Clone, Deserialize, Serialize)]
|
||||||
#[serde(rename_all = "camelCase")]
|
#[serde(rename_all = "camelCase")]
|
||||||
pub struct DestinationPolicyConfig {
|
pub struct DestinationPolicyConfig {
|
||||||
/// Default action: "forceTarget", "block", or "allow".
|
/// Default action: "forceTarget", "block", or "allow".
|
||||||
@@ -73,9 +73,12 @@ pub struct ServerConfig {
|
|||||||
/// Server-level IP block list — applied at TCP accept, before Noise handshake.
|
/// Server-level IP block list — applied at TCP accept, before Noise handshake.
|
||||||
pub connection_ip_block_list: Option<Vec<String>>,
|
pub connection_ip_block_list: Option<Vec<String>>,
|
||||||
/// When true and forwarding_mode is "socket", the userspace NAT engine prepends
|
/// When true and forwarding_mode is "socket", the userspace NAT engine prepends
|
||||||
/// PROXY protocol v2 headers on outbound TCP connections, conveying the VPN client's
|
/// PROXY protocol v2 headers on outbound TCP connections.
|
||||||
/// tunnel IP as the source address.
|
|
||||||
pub socket_forward_proxy_protocol: Option<bool>,
|
pub socket_forward_proxy_protocol: Option<bool>,
|
||||||
|
/// Source address for outbound PROXY protocol headers: "tunnelIp" (legacy) or "remoteIp".
|
||||||
|
pub socket_forward_proxy_protocol_source: Option<String>,
|
||||||
|
/// Include authenticated SmartVPN metadata as a custom PROXY v2 TLV.
|
||||||
|
pub socket_forward_proxy_protocol_vpn_metadata: Option<bool>,
|
||||||
/// Destination routing policy for VPN client traffic (socket mode).
|
/// Destination routing policy for VPN client traffic (socket mode).
|
||||||
pub destination_policy: Option<DestinationPolicyConfig>,
|
pub destination_policy: Option<DestinationPolicyConfig>,
|
||||||
/// WireGuard: server X25519 private key (base64). Required when transport includes WG.
|
/// WireGuard: server X25519 private key (base64). Required when transport includes WG.
|
||||||
@@ -92,6 +95,17 @@ pub struct ServerConfig {
|
|||||||
/// Defaults to ["0.0.0.0/0"] (full tunnel).
|
/// Defaults to ["0.0.0.0/0"] (full tunnel).
|
||||||
#[serde(alias = "clientAllowedIPs")]
|
#[serde(alias = "clientAllowedIPs")]
|
||||||
pub client_allowed_ips: Option<Vec<String>>,
|
pub client_allowed_ips: Option<Vec<String>>,
|
||||||
|
|
||||||
|
// Bridge mode configuration (forwarding_mode: "bridge")
|
||||||
|
|
||||||
|
/// LAN subnet CIDR for bridge mode (e.g. "192.168.1.0/24").
|
||||||
|
pub bridge_lan_subnet: Option<String>,
|
||||||
|
/// Physical network interface to bridge (e.g. "eth0"). Auto-detected if omitted.
|
||||||
|
pub bridge_physical_interface: Option<String>,
|
||||||
|
/// Start of VPN client IP range within the LAN subnet (host offset, e.g. 200).
|
||||||
|
pub bridge_ip_range_start: Option<u32>,
|
||||||
|
/// End of VPN client IP range within the LAN subnet (host offset, e.g. 250).
|
||||||
|
pub bridge_ip_range_end: Option<u32>,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Information about a connected client.
|
/// Information about a connected client.
|
||||||
@@ -148,10 +162,28 @@ pub enum ForwardingEngine {
|
|||||||
Tun(tokio::io::WriteHalf<tun::AsyncDevice>),
|
Tun(tokio::io::WriteHalf<tun::AsyncDevice>),
|
||||||
/// Userspace NAT — packets sent to smoltcp-based NAT engine via channel.
|
/// Userspace NAT — packets sent to smoltcp-based NAT engine via channel.
|
||||||
Socket(mpsc::Sender<Vec<u8>>),
|
Socket(mpsc::Sender<Vec<u8>>),
|
||||||
|
/// L2 Bridge — packets sent to BridgeEngine via channel, bridged to host LAN.
|
||||||
|
Bridge(mpsc::Sender<Vec<u8>>),
|
||||||
|
/// Hybrid — both socket NAT and bridge engines running simultaneously.
|
||||||
|
/// Per-client routing: look up src_ip in routing_table to decide socket vs bridge.
|
||||||
|
Hybrid {
|
||||||
|
socket_tx: mpsc::Sender<Vec<u8>>,
|
||||||
|
bridge_tx: mpsc::Sender<Vec<u8>>,
|
||||||
|
/// Fast lookup: VPN IP → true if client uses bridge (host IP), false for socket.
|
||||||
|
routing_table: Arc<RwLock<HashMap<Ipv4Addr, bool>>>,
|
||||||
|
},
|
||||||
/// Testing/monitoring — packets are counted but not forwarded.
|
/// Testing/monitoring — packets are counted but not forwarded.
|
||||||
Testing,
|
Testing,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Info needed to tear down bridge infrastructure on stop().
|
||||||
|
pub struct BridgeCleanupInfo {
|
||||||
|
pub physical_iface: String,
|
||||||
|
pub bridge_name: String,
|
||||||
|
pub host_ip: Ipv4Addr,
|
||||||
|
pub host_prefix: u8,
|
||||||
|
}
|
||||||
|
|
||||||
/// Shared server state.
|
/// Shared server state.
|
||||||
pub struct ServerState {
|
pub struct ServerState {
|
||||||
pub config: ServerConfig,
|
pub config: ServerConfig,
|
||||||
@@ -168,6 +200,10 @@ pub struct ServerState {
|
|||||||
pub tun_routes: RwLock<HashMap<Ipv4Addr, mpsc::Sender<Vec<u8>>>>,
|
pub tun_routes: RwLock<HashMap<Ipv4Addr, mpsc::Sender<Vec<u8>>>>,
|
||||||
/// Shutdown signal for the forwarding background task (TUN reader or NAT engine).
|
/// Shutdown signal for the forwarding background task (TUN reader or NAT engine).
|
||||||
pub tun_shutdown: mpsc::Sender<()>,
|
pub tun_shutdown: mpsc::Sender<()>,
|
||||||
|
/// Shutdown signal for the bridge engine (bridge/hybrid modes only).
|
||||||
|
pub bridge_shutdown: Option<mpsc::Sender<()>>,
|
||||||
|
/// Bridge teardown info (bridge/hybrid modes only).
|
||||||
|
pub bridge_cleanup: Option<BridgeCleanupInfo>,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// The VPN server.
|
/// The VPN server.
|
||||||
@@ -191,7 +227,15 @@ impl VpnServer {
|
|||||||
anyhow::bail!("Server is already running");
|
anyhow::bail!("Server is already running");
|
||||||
}
|
}
|
||||||
|
|
||||||
let ip_pool = IpPool::new(&config.subnet)?;
|
let mode = config.forwarding_mode.as_deref().unwrap_or("testing");
|
||||||
|
let ip_pool = if mode == "bridge" {
|
||||||
|
let lan_subnet = config.bridge_lan_subnet.as_deref().unwrap_or(&config.subnet);
|
||||||
|
let range_start = config.bridge_ip_range_start.unwrap_or(200);
|
||||||
|
let range_end = config.bridge_ip_range_end.unwrap_or(250);
|
||||||
|
IpPool::new_with_range(lan_subnet, range_start, range_end)?
|
||||||
|
} else {
|
||||||
|
IpPool::new(&config.subnet)?
|
||||||
|
};
|
||||||
|
|
||||||
if config.enable_nat.unwrap_or(false) {
|
if config.enable_nat.unwrap_or(false) {
|
||||||
if let Err(e) = crate::network::enable_ip_forwarding() {
|
if let Err(e) = crate::network::enable_ip_forwarding() {
|
||||||
@@ -205,7 +249,6 @@ impl VpnServer {
|
|||||||
}
|
}
|
||||||
|
|
||||||
let link_mtu = config.mtu.unwrap_or(1420);
|
let link_mtu = config.mtu.unwrap_or(1420);
|
||||||
let mode = config.forwarding_mode.as_deref().unwrap_or("testing");
|
|
||||||
let gateway_ip = ip_pool.gateway_addr();
|
let gateway_ip = ip_pool.gateway_addr();
|
||||||
|
|
||||||
// Create forwarding engine based on mode
|
// Create forwarding engine based on mode
|
||||||
@@ -220,9 +263,28 @@ impl VpnServer {
|
|||||||
packet_rx: mpsc::Receiver<Vec<u8>>,
|
packet_rx: mpsc::Receiver<Vec<u8>>,
|
||||||
shutdown_rx: mpsc::Receiver<()>,
|
shutdown_rx: mpsc::Receiver<()>,
|
||||||
},
|
},
|
||||||
|
Bridge {
|
||||||
|
packet_tx: mpsc::Sender<Vec<u8>>,
|
||||||
|
packet_rx: mpsc::Receiver<Vec<u8>>,
|
||||||
|
tap_device: tun::AsyncDevice,
|
||||||
|
shutdown_rx: mpsc::Receiver<()>,
|
||||||
|
},
|
||||||
|
Hybrid {
|
||||||
|
socket_tx: mpsc::Sender<Vec<u8>>,
|
||||||
|
socket_rx: mpsc::Receiver<Vec<u8>>,
|
||||||
|
socket_shutdown_rx: mpsc::Receiver<()>,
|
||||||
|
bridge_tx: mpsc::Sender<Vec<u8>>,
|
||||||
|
bridge_rx: mpsc::Receiver<Vec<u8>>,
|
||||||
|
bridge_shutdown_rx: mpsc::Receiver<()>,
|
||||||
|
tap_device: tun::AsyncDevice,
|
||||||
|
routing_table: Arc<RwLock<HashMap<Ipv4Addr, bool>>>,
|
||||||
|
},
|
||||||
Testing,
|
Testing,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
let mut bridge_cleanup_info: Option<BridgeCleanupInfo> = None;
|
||||||
|
let mut bridge_shut_tx: Option<mpsc::Sender<()>> = None;
|
||||||
|
|
||||||
let (setup, fwd_shutdown_tx) = match mode {
|
let (setup, fwd_shutdown_tx) = match mode {
|
||||||
"tun" => {
|
"tun" => {
|
||||||
let tun_config = TunConfig {
|
let tun_config = TunConfig {
|
||||||
@@ -243,6 +305,88 @@ impl VpnServer {
|
|||||||
let (tx, rx) = mpsc::channel::<()>(1);
|
let (tx, rx) = mpsc::channel::<()>(1);
|
||||||
(ForwardingSetup::Socket { packet_tx, packet_rx, shutdown_rx: rx }, tx)
|
(ForwardingSetup::Socket { packet_tx, packet_rx, shutdown_rx: rx }, tx)
|
||||||
}
|
}
|
||||||
|
"bridge" => {
|
||||||
|
info!("Starting L2 bridge forwarding (requires CAP_NET_ADMIN)");
|
||||||
|
let phys_iface = match &config.bridge_physical_interface {
|
||||||
|
Some(i) => i.clone(),
|
||||||
|
None => crate::bridge::detect_default_interface().await?,
|
||||||
|
};
|
||||||
|
let (host_ip, host_prefix) = crate::bridge::get_interface_ip(&phys_iface).await?;
|
||||||
|
|
||||||
|
let bridge_name = "svpn_br0";
|
||||||
|
let tap_name = "svpn_tap0";
|
||||||
|
|
||||||
|
// Create TAP + bridge infrastructure
|
||||||
|
let tap_device = crate::bridge::create_tap(tap_name, link_mtu)?;
|
||||||
|
crate::bridge::create_bridge(bridge_name).await?;
|
||||||
|
crate::bridge::set_interface_up(bridge_name).await?;
|
||||||
|
crate::bridge::bridge_add_interface(bridge_name, tap_name).await?;
|
||||||
|
crate::bridge::set_interface_up(tap_name).await?;
|
||||||
|
crate::bridge::bridge_add_interface(bridge_name, &phys_iface).await?;
|
||||||
|
crate::bridge::migrate_host_ip_to_bridge(&phys_iface, bridge_name, host_ip, host_prefix).await?;
|
||||||
|
crate::bridge::enable_proxy_arp(bridge_name).await?;
|
||||||
|
|
||||||
|
info!("Bridge {} created: TAP={}, physical={}, IP={}/{}", bridge_name, tap_name, phys_iface, host_ip, host_prefix);
|
||||||
|
|
||||||
|
bridge_cleanup_info = Some(BridgeCleanupInfo {
|
||||||
|
physical_iface: phys_iface,
|
||||||
|
bridge_name: bridge_name.to_string(),
|
||||||
|
host_ip,
|
||||||
|
host_prefix,
|
||||||
|
});
|
||||||
|
|
||||||
|
let (packet_tx, packet_rx) = mpsc::channel::<Vec<u8>>(4096);
|
||||||
|
let (tx, rx) = mpsc::channel::<()>(1);
|
||||||
|
(ForwardingSetup::Bridge { packet_tx, packet_rx, tap_device, shutdown_rx: rx }, tx)
|
||||||
|
}
|
||||||
|
"hybrid" => {
|
||||||
|
info!("Starting hybrid forwarding (socket + bridge, per-client routing)");
|
||||||
|
|
||||||
|
// Socket engine setup
|
||||||
|
let (s_tx, s_rx) = mpsc::channel::<Vec<u8>>(4096);
|
||||||
|
let (s_shut_tx, s_shut_rx) = mpsc::channel::<()>(1);
|
||||||
|
|
||||||
|
// Bridge engine setup
|
||||||
|
let phys_iface = match &config.bridge_physical_interface {
|
||||||
|
Some(i) => i.clone(),
|
||||||
|
None => crate::bridge::detect_default_interface().await?,
|
||||||
|
};
|
||||||
|
let (host_ip, host_prefix) = crate::bridge::get_interface_ip(&phys_iface).await?;
|
||||||
|
let bridge_name = "svpn_br0";
|
||||||
|
let tap_name = "svpn_tap0";
|
||||||
|
|
||||||
|
let tap_device = crate::bridge::create_tap(tap_name, link_mtu)?;
|
||||||
|
crate::bridge::create_bridge(bridge_name).await?;
|
||||||
|
crate::bridge::set_interface_up(bridge_name).await?;
|
||||||
|
crate::bridge::bridge_add_interface(bridge_name, tap_name).await?;
|
||||||
|
crate::bridge::set_interface_up(tap_name).await?;
|
||||||
|
crate::bridge::bridge_add_interface(bridge_name, &phys_iface).await?;
|
||||||
|
crate::bridge::migrate_host_ip_to_bridge(&phys_iface, bridge_name, host_ip, host_prefix).await?;
|
||||||
|
crate::bridge::enable_proxy_arp(bridge_name).await?;
|
||||||
|
|
||||||
|
let (b_tx, b_rx) = mpsc::channel::<Vec<u8>>(4096);
|
||||||
|
let (b_shut_tx, b_shut_rx) = mpsc::channel::<()>(1);
|
||||||
|
|
||||||
|
// Build routing table from registered clients
|
||||||
|
let routing_table = Arc::new(RwLock::new(HashMap::<Ipv4Addr, bool>::new()));
|
||||||
|
|
||||||
|
info!("Hybrid mode: socket + bridge (TAP={}, physical={}, IP={}/{})", tap_name, phys_iface, host_ip, host_prefix);
|
||||||
|
|
||||||
|
bridge_cleanup_info = Some(BridgeCleanupInfo {
|
||||||
|
physical_iface: phys_iface,
|
||||||
|
bridge_name: bridge_name.to_string(),
|
||||||
|
host_ip,
|
||||||
|
host_prefix,
|
||||||
|
});
|
||||||
|
bridge_shut_tx = Some(b_shut_tx);
|
||||||
|
|
||||||
|
// Socket engine uses fwd_shutdown_tx (stored in state.tun_shutdown)
|
||||||
|
(ForwardingSetup::Hybrid {
|
||||||
|
socket_tx: s_tx, socket_rx: s_rx, socket_shutdown_rx: s_shut_rx,
|
||||||
|
bridge_tx: b_tx, bridge_rx: b_rx, bridge_shutdown_rx: b_shut_rx,
|
||||||
|
tap_device, routing_table,
|
||||||
|
}, s_shut_tx)
|
||||||
|
}
|
||||||
_ => {
|
_ => {
|
||||||
info!("Forwarding disabled (testing/monitoring mode)");
|
info!("Forwarding disabled (testing/monitoring mode)");
|
||||||
let (tx, _rx) = mpsc::channel::<()>(1);
|
let (tx, _rx) = mpsc::channel::<()>(1);
|
||||||
@@ -252,7 +396,7 @@ impl VpnServer {
|
|||||||
|
|
||||||
// Compute effective MTU from overhead
|
// Compute effective MTU from overhead
|
||||||
let overhead = TunnelOverhead::default_overhead();
|
let overhead = TunnelOverhead::default_overhead();
|
||||||
let mtu_config = MtuConfig::new(overhead.effective_tun_mtu(1500).max(link_mtu));
|
let mtu_config = MtuConfig::new(overhead.effective_tun_mtu(1500).min(link_mtu));
|
||||||
|
|
||||||
// Build client registry from config
|
// Build client registry from config
|
||||||
let registry = ClientRegistry::from_entries(
|
let registry = ClientRegistry::from_entries(
|
||||||
@@ -272,6 +416,8 @@ impl VpnServer {
|
|||||||
forwarding_engine: Mutex::new(ForwardingEngine::Testing),
|
forwarding_engine: Mutex::new(ForwardingEngine::Testing),
|
||||||
tun_routes: RwLock::new(HashMap::new()),
|
tun_routes: RwLock::new(HashMap::new()),
|
||||||
tun_shutdown: fwd_shutdown_tx,
|
tun_shutdown: fwd_shutdown_tx,
|
||||||
|
bridge_shutdown: bridge_shut_tx,
|
||||||
|
bridge_cleanup: bridge_cleanup_info,
|
||||||
});
|
});
|
||||||
|
|
||||||
// Spawn the forwarding background task and set the engine
|
// Spawn the forwarding background task and set the engine
|
||||||
@@ -288,11 +434,17 @@ impl VpnServer {
|
|||||||
ForwardingSetup::Socket { packet_tx, packet_rx, shutdown_rx } => {
|
ForwardingSetup::Socket { packet_tx, packet_rx, shutdown_rx } => {
|
||||||
*state.forwarding_engine.lock().await = ForwardingEngine::Socket(packet_tx);
|
*state.forwarding_engine.lock().await = ForwardingEngine::Socket(packet_tx);
|
||||||
let proxy_protocol = config.socket_forward_proxy_protocol.unwrap_or(false);
|
let proxy_protocol = config.socket_forward_proxy_protocol.unwrap_or(false);
|
||||||
|
let proxy_protocol_source = crate::userspace_nat::ProxyProtocolSource::from_config(
|
||||||
|
config.socket_forward_proxy_protocol_source.as_deref(),
|
||||||
|
);
|
||||||
|
let proxy_protocol_vpn_metadata = config.socket_forward_proxy_protocol_vpn_metadata.unwrap_or(false);
|
||||||
let nat_engine = crate::userspace_nat::NatEngine::new(
|
let nat_engine = crate::userspace_nat::NatEngine::new(
|
||||||
gateway_ip,
|
gateway_ip,
|
||||||
link_mtu as usize,
|
link_mtu as usize,
|
||||||
state.clone(),
|
state.clone(),
|
||||||
proxy_protocol,
|
proxy_protocol,
|
||||||
|
proxy_protocol_source,
|
||||||
|
proxy_protocol_vpn_metadata,
|
||||||
config.destination_policy.clone(),
|
config.destination_policy.clone(),
|
||||||
);
|
);
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
@@ -301,6 +453,66 @@ impl VpnServer {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
ForwardingSetup::Bridge { packet_tx, packet_rx, tap_device, shutdown_rx } => {
|
||||||
|
*state.forwarding_engine.lock().await = ForwardingEngine::Bridge(packet_tx);
|
||||||
|
let bridge_engine = crate::bridge::BridgeEngine::new(state.clone());
|
||||||
|
tokio::spawn(async move {
|
||||||
|
if let Err(e) = bridge_engine.run(tap_device, packet_rx, shutdown_rx).await {
|
||||||
|
error!("Bridge engine error: {}", e);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
ForwardingSetup::Hybrid {
|
||||||
|
socket_tx, socket_rx, socket_shutdown_rx,
|
||||||
|
bridge_tx, bridge_rx, bridge_shutdown_rx,
|
||||||
|
tap_device, routing_table,
|
||||||
|
} => {
|
||||||
|
// Populate routing table from registered clients
|
||||||
|
{
|
||||||
|
let registry = state.client_registry.read().await;
|
||||||
|
let mut rt = routing_table.write().await;
|
||||||
|
for entry in registry.list() {
|
||||||
|
if let Some(ref ip_str) = entry.assigned_ip {
|
||||||
|
if let Ok(ip) = ip_str.parse::<Ipv4Addr>() {
|
||||||
|
rt.insert(ip, entry.use_host_ip.unwrap_or(false));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start socket (NAT) engine
|
||||||
|
let proxy_protocol = config.socket_forward_proxy_protocol.unwrap_or(false);
|
||||||
|
let proxy_protocol_source = crate::userspace_nat::ProxyProtocolSource::from_config(
|
||||||
|
config.socket_forward_proxy_protocol_source.as_deref(),
|
||||||
|
);
|
||||||
|
let proxy_protocol_vpn_metadata = config.socket_forward_proxy_protocol_vpn_metadata.unwrap_or(false);
|
||||||
|
let nat_engine = crate::userspace_nat::NatEngine::new(
|
||||||
|
gateway_ip,
|
||||||
|
link_mtu as usize,
|
||||||
|
state.clone(),
|
||||||
|
proxy_protocol,
|
||||||
|
proxy_protocol_source,
|
||||||
|
proxy_protocol_vpn_metadata,
|
||||||
|
config.destination_policy.clone(),
|
||||||
|
);
|
||||||
|
tokio::spawn(async move {
|
||||||
|
if let Err(e) = nat_engine.run(socket_rx, socket_shutdown_rx).await {
|
||||||
|
error!("NAT engine error (hybrid): {}", e);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// Start bridge engine
|
||||||
|
let bridge_engine = crate::bridge::BridgeEngine::new(state.clone());
|
||||||
|
tokio::spawn(async move {
|
||||||
|
if let Err(e) = bridge_engine.run(tap_device, bridge_rx, bridge_shutdown_rx).await {
|
||||||
|
error!("Bridge engine error (hybrid): {}", e);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
*state.forwarding_engine.lock().await = ForwardingEngine::Hybrid {
|
||||||
|
socket_tx, bridge_tx, routing_table,
|
||||||
|
};
|
||||||
|
}
|
||||||
ForwardingSetup::Testing => {}
|
ForwardingSetup::Testing => {}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -387,6 +599,9 @@ impl VpnServer {
|
|||||||
if self.wg_command_tx.is_some() {
|
if self.wg_command_tx.is_some() {
|
||||||
let registry = state.client_registry.read().await;
|
let registry = state.client_registry.read().await;
|
||||||
for entry in registry.list() {
|
for entry in registry.list() {
|
||||||
|
if !entry.is_enabled() || entry.is_expired() {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
if let (Some(ref wg_key), Some(ref ip_str)) = (&entry.wg_public_key, &entry.assigned_ip) {
|
if let (Some(ref wg_key), Some(ref ip_str)) = (&entry.wg_public_key, &entry.assigned_ip) {
|
||||||
let peer_config = crate::wireguard::WgPeerConfig {
|
let peer_config = crate::wireguard::WgPeerConfig {
|
||||||
public_key: wg_key.clone(),
|
public_key: wg_key.clone(),
|
||||||
@@ -421,6 +636,43 @@ impl VpnServer {
|
|||||||
let _ = state.tun_shutdown.send(()).await;
|
let _ = state.tun_shutdown.send(()).await;
|
||||||
*state.forwarding_engine.lock().await = ForwardingEngine::Testing;
|
*state.forwarding_engine.lock().await = ForwardingEngine::Testing;
|
||||||
}
|
}
|
||||||
|
"bridge" => {
|
||||||
|
let _ = state.tun_shutdown.send(()).await;
|
||||||
|
*state.forwarding_engine.lock().await = ForwardingEngine::Testing;
|
||||||
|
// Restore host networking: move IP back and remove bridge
|
||||||
|
if let Some(ref cleanup) = state.bridge_cleanup {
|
||||||
|
if let Err(e) = crate::bridge::restore_host_ip(
|
||||||
|
&cleanup.physical_iface, &cleanup.bridge_name,
|
||||||
|
cleanup.host_ip, cleanup.host_prefix,
|
||||||
|
).await {
|
||||||
|
warn!("Failed to restore host IP: {}", e);
|
||||||
|
}
|
||||||
|
if let Err(e) = crate::bridge::remove_bridge(&cleanup.bridge_name).await {
|
||||||
|
warn!("Failed to remove bridge: {}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"hybrid" => {
|
||||||
|
// Shut down socket (NAT) engine
|
||||||
|
let _ = state.tun_shutdown.send(()).await;
|
||||||
|
// Shut down bridge engine
|
||||||
|
if let Some(ref bridge_shut) = state.bridge_shutdown {
|
||||||
|
let _ = bridge_shut.send(()).await;
|
||||||
|
}
|
||||||
|
*state.forwarding_engine.lock().await = ForwardingEngine::Testing;
|
||||||
|
// Restore host networking: move IP back and remove bridge
|
||||||
|
if let Some(ref cleanup) = state.bridge_cleanup {
|
||||||
|
if let Err(e) = crate::bridge::restore_host_ip(
|
||||||
|
&cleanup.physical_iface, &cleanup.bridge_name,
|
||||||
|
cleanup.host_ip, cleanup.host_prefix,
|
||||||
|
).await {
|
||||||
|
warn!("Failed to restore host IP: {}", e);
|
||||||
|
}
|
||||||
|
if let Err(e) = crate::bridge::remove_bridge(&cleanup.bridge_name).await {
|
||||||
|
warn!("Failed to remove bridge: {}", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -491,8 +743,10 @@ impl VpnServer {
|
|||||||
if let Some(ref state) = self.state {
|
if let Some(ref state) = self.state {
|
||||||
let mut clients = state.clients.write().await;
|
let mut clients = state.clients.write().await;
|
||||||
if let Some(client) = clients.remove(client_id) {
|
if let Some(client) = clients.remove(client_id) {
|
||||||
let ip: Ipv4Addr = client.assigned_ip.parse()?;
|
if client.transport_type != "wireguard" {
|
||||||
state.ip_pool.lock().await.release(&ip);
|
let ip: Ipv4Addr = client.assigned_ip.parse()?;
|
||||||
|
state.ip_pool.lock().await.release(&ip);
|
||||||
|
}
|
||||||
state.rate_limiters.lock().await.remove(client_id);
|
state.rate_limiters.lock().await.remove(client_id);
|
||||||
info!("Client {} disconnected", client_id);
|
info!("Client {} disconnected", client_id);
|
||||||
}
|
}
|
||||||
@@ -633,10 +887,18 @@ impl VpnServer {
|
|||||||
description: partial.get("description").and_then(|v| v.as_str()).map(String::from),
|
description: partial.get("description").and_then(|v| v.as_str()).map(String::from),
|
||||||
expires_at: partial.get("expiresAt").and_then(|v| v.as_str()).map(String::from),
|
expires_at: partial.get("expiresAt").and_then(|v| v.as_str()).map(String::from),
|
||||||
assigned_ip: Some(assigned_ip.to_string()),
|
assigned_ip: Some(assigned_ip.to_string()),
|
||||||
|
use_host_ip: partial.get("useHostIp").and_then(|v| v.as_bool()),
|
||||||
|
use_dhcp: partial.get("useDhcp").and_then(|v| v.as_bool()),
|
||||||
|
static_ip: partial.get("staticIp").and_then(|v| v.as_str()).map(String::from),
|
||||||
|
force_vlan: partial.get("forceVlan").and_then(|v| v.as_bool()),
|
||||||
|
vlan_id: partial.get("vlanId").and_then(|v| v.as_u64()).map(|v| v as u16),
|
||||||
};
|
};
|
||||||
|
|
||||||
// Add to registry
|
// Add to registry — release IP on failure to avoid pool leak
|
||||||
state.client_registry.write().await.add(entry.clone())?;
|
if let Err(e) = state.client_registry.write().await.add(entry.clone()) {
|
||||||
|
state.ip_pool.lock().await.release(&assigned_ip);
|
||||||
|
return Err(e);
|
||||||
|
}
|
||||||
|
|
||||||
// Register WG peer with the running WG listener (if active)
|
// Register WG peer with the running WG listener (if active)
|
||||||
if self.wg_command_tx.is_some() {
|
if self.wg_command_tx.is_some() {
|
||||||
@@ -648,7 +910,9 @@ impl VpnServer {
|
|||||||
persistent_keepalive: Some(25),
|
persistent_keepalive: Some(25),
|
||||||
};
|
};
|
||||||
if let Err(e) = self.add_wg_peer(wg_peer_config).await {
|
if let Err(e) = self.add_wg_peer(wg_peer_config).await {
|
||||||
warn!("Failed to register WG peer for client {}: {}", client_id, e);
|
let _ = state.client_registry.write().await.remove(&client_id);
|
||||||
|
state.ip_pool.lock().await.release(&assigned_ip);
|
||||||
|
return Err(anyhow::anyhow!("Failed to register WG peer for client {}: {}", client_id, e));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -706,15 +970,28 @@ impl VpnServer {
|
|||||||
pub async fn remove_registered_client(&self, client_id: &str) -> Result<()> {
|
pub async fn remove_registered_client(&self, client_id: &str) -> Result<()> {
|
||||||
let state = self.state.as_ref()
|
let state = self.state.as_ref()
|
||||||
.ok_or_else(|| anyhow::anyhow!("Server not running"))?;
|
.ok_or_else(|| anyhow::anyhow!("Server not running"))?;
|
||||||
let entry = state.client_registry.write().await.remove(client_id)?;
|
|
||||||
|
let entry = {
|
||||||
|
let registry = state.client_registry.read().await;
|
||||||
|
registry.get_by_id(client_id)
|
||||||
|
.cloned()
|
||||||
|
.ok_or_else(|| anyhow::anyhow!("Client '{}' not found", client_id))?
|
||||||
|
};
|
||||||
|
|
||||||
// Remove WG peer from running listener
|
// Remove WG peer from running listener
|
||||||
if self.wg_command_tx.is_some() {
|
if self.wg_command_tx.is_some() {
|
||||||
if let Some(ref wg_key) = entry.wg_public_key {
|
if let Some(ref wg_key) = entry.wg_public_key {
|
||||||
if let Err(e) = self.remove_wg_peer(wg_key).await {
|
if let Err(e) = self.remove_wg_peer(wg_key).await {
|
||||||
debug!("Failed to remove WG peer for client {}: {}", client_id, e);
|
if entry.is_enabled() && !entry.is_expired() {
|
||||||
|
return Err(anyhow::anyhow!("Failed to remove WG peer for client {}: {}", client_id, e));
|
||||||
|
}
|
||||||
|
debug!("Failed to remove inactive WG peer for client {}: {}", client_id, e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
state.client_registry.write().await.remove(client_id)?;
|
||||||
|
|
||||||
// Release the IP if assigned
|
// Release the IP if assigned
|
||||||
if let Some(ref ip_str) = entry.assigned_ip {
|
if let Some(ref ip_str) = entry.assigned_ip {
|
||||||
if let Ok(ip) = ip_str.parse::<Ipv4Addr>() {
|
if let Ok(ip) = ip_str.parse::<Ipv4Addr>() {
|
||||||
@@ -771,7 +1048,37 @@ impl VpnServer {
|
|||||||
if let Some(expires) = update.get("expiresAt").and_then(|v| v.as_str()) {
|
if let Some(expires) = update.get("expiresAt").and_then(|v| v.as_str()) {
|
||||||
entry.expires_at = Some(expires.to_string());
|
entry.expires_at = Some(expires.to_string());
|
||||||
}
|
}
|
||||||
|
if let Some(use_host_ip) = update.get("useHostIp").and_then(|v| v.as_bool()) {
|
||||||
|
entry.use_host_ip = Some(use_host_ip);
|
||||||
|
}
|
||||||
|
if let Some(use_dhcp) = update.get("useDhcp").and_then(|v| v.as_bool()) {
|
||||||
|
entry.use_dhcp = Some(use_dhcp);
|
||||||
|
}
|
||||||
|
if let Some(static_ip) = update.get("staticIp").and_then(|v| v.as_str()) {
|
||||||
|
entry.static_ip = Some(static_ip.to_string());
|
||||||
|
}
|
||||||
|
if let Some(force_vlan) = update.get("forceVlan").and_then(|v| v.as_bool()) {
|
||||||
|
entry.force_vlan = Some(force_vlan);
|
||||||
|
}
|
||||||
|
if let Some(vlan_id) = update.get("vlanId").and_then(|v| v.as_u64()) {
|
||||||
|
entry.vlan_id = Some(vlan_id as u16);
|
||||||
|
}
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
|
let updated_entry = {
|
||||||
|
let registry = state.client_registry.read().await;
|
||||||
|
registry.get_by_id(client_id).cloned()
|
||||||
|
};
|
||||||
|
if let Some(entry) = updated_entry {
|
||||||
|
if let Some(ref ip_str) = entry.assigned_ip {
|
||||||
|
if let Ok(ip) = ip_str.parse::<Ipv4Addr>() {
|
||||||
|
if let ForwardingEngine::Hybrid { routing_table, .. } = &*state.forwarding_engine.lock().await {
|
||||||
|
routing_table.write().await.insert(ip, entry.use_host_ip.unwrap_or(false));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -781,13 +1088,56 @@ impl VpnServer {
|
|||||||
.ok_or_else(|| anyhow::anyhow!("Server not running"))?;
|
.ok_or_else(|| anyhow::anyhow!("Server not running"))?;
|
||||||
state.client_registry.write().await.update(client_id, |entry| {
|
state.client_registry.write().await.update(client_id, |entry| {
|
||||||
entry.enabled = Some(true);
|
entry.enabled = Some(true);
|
||||||
})
|
})?;
|
||||||
|
|
||||||
|
let entry = {
|
||||||
|
let registry = state.client_registry.read().await;
|
||||||
|
registry.get_by_id(client_id)
|
||||||
|
.cloned()
|
||||||
|
.ok_or_else(|| anyhow::anyhow!("Client '{}' not found", client_id))?
|
||||||
|
};
|
||||||
|
|
||||||
|
if self.wg_command_tx.is_some() {
|
||||||
|
if let (Some(ref wg_key), Some(ref ip_str)) = (&entry.wg_public_key, &entry.assigned_ip) {
|
||||||
|
let peer_config = crate::wireguard::WgPeerConfig {
|
||||||
|
public_key: wg_key.clone(),
|
||||||
|
preshared_key: None,
|
||||||
|
allowed_ips: vec![format!("{}/32", ip_str)],
|
||||||
|
endpoint: None,
|
||||||
|
persistent_keepalive: Some(25),
|
||||||
|
};
|
||||||
|
if let Err(e) = self.add_wg_peer(peer_config).await {
|
||||||
|
let _ = state.client_registry.write().await.update(client_id, |entry| {
|
||||||
|
entry.enabled = Some(false);
|
||||||
|
});
|
||||||
|
return Err(anyhow::anyhow!("Failed to register WG peer for enabled client {}: {}", client_id, e));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Disable a registered client (also disconnects if connected).
|
/// Disable a registered client (also disconnects if connected).
|
||||||
pub async fn disable_client(&self, client_id: &str) -> Result<()> {
|
pub async fn disable_client(&self, client_id: &str) -> Result<()> {
|
||||||
let state = self.state.as_ref()
|
let state = self.state.as_ref()
|
||||||
.ok_or_else(|| anyhow::anyhow!("Server not running"))?;
|
.ok_or_else(|| anyhow::anyhow!("Server not running"))?;
|
||||||
|
|
||||||
|
let entry = {
|
||||||
|
let registry = state.client_registry.read().await;
|
||||||
|
registry.get_by_id(client_id).cloned()
|
||||||
|
};
|
||||||
|
|
||||||
|
if self.wg_command_tx.is_some() {
|
||||||
|
if let Some(ref entry) = entry {
|
||||||
|
if let Some(ref wg_key) = entry.wg_public_key {
|
||||||
|
if let Err(e) = self.remove_wg_peer(wg_key).await {
|
||||||
|
debug!("Failed to remove WG peer for disabled client {}: {}", client_id, e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
state.client_registry.write().await.update(client_id, |entry| {
|
state.client_registry.write().await.update(client_id, |entry| {
|
||||||
entry.enabled = Some(false);
|
entry.enabled = Some(false);
|
||||||
})?;
|
})?;
|
||||||
@@ -801,17 +1151,33 @@ impl VpnServer {
|
|||||||
let state = self.state.as_ref()
|
let state = self.state.as_ref()
|
||||||
.ok_or_else(|| anyhow::anyhow!("Server not running"))?;
|
.ok_or_else(|| anyhow::anyhow!("Server not running"))?;
|
||||||
|
|
||||||
// Capture old WG key before rotation (needed to remove from WG listener)
|
// Capture old keys before rotation so listener and registry can be rolled back together.
|
||||||
let old_wg_pub = {
|
let old_entry = {
|
||||||
let registry = state.client_registry.read().await;
|
let registry = state.client_registry.read().await;
|
||||||
let entry = registry.get_by_id(client_id)
|
registry.get_by_id(client_id)
|
||||||
.ok_or_else(|| anyhow::anyhow!("Client '{}' not found", client_id))?;
|
.cloned()
|
||||||
entry.wg_public_key.clone()
|
.ok_or_else(|| anyhow::anyhow!("Client '{}' not found", client_id))?
|
||||||
};
|
};
|
||||||
|
let old_noise_pub = old_entry.public_key.clone();
|
||||||
|
let old_wg_pub = old_entry.wg_public_key.clone();
|
||||||
|
let assigned_ip = old_entry.assigned_ip.clone().unwrap_or_else(|| "0.0.0.0".to_string());
|
||||||
|
let should_have_wg_peer = self.wg_command_tx.is_some()
|
||||||
|
&& old_entry.is_enabled()
|
||||||
|
&& !old_entry.is_expired()
|
||||||
|
&& old_wg_pub.is_some()
|
||||||
|
&& assigned_ip != "0.0.0.0";
|
||||||
|
|
||||||
let (noise_pub, noise_priv) = crypto::generate_keypair()?;
|
let (noise_pub, noise_priv) = crypto::generate_keypair()?;
|
||||||
let (wg_pub, wg_priv) = crate::wireguard::generate_wg_keypair();
|
let (wg_pub, wg_priv) = crate::wireguard::generate_wg_keypair();
|
||||||
|
|
||||||
|
if should_have_wg_peer {
|
||||||
|
if let Some(ref old_key) = old_wg_pub {
|
||||||
|
if let Err(e) = self.remove_wg_peer(old_key).await {
|
||||||
|
return Err(anyhow::anyhow!("Failed to remove old WG peer during rotation for {}: {}", client_id, e));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
state.client_registry.write().await.rotate_key(
|
state.client_registry.write().await.rotate_key(
|
||||||
client_id,
|
client_id,
|
||||||
noise_pub.clone(),
|
noise_pub.clone(),
|
||||||
@@ -821,31 +1187,40 @@ impl VpnServer {
|
|||||||
// Disconnect existing connection (old key is no longer valid)
|
// Disconnect existing connection (old key is no longer valid)
|
||||||
let _ = self.disconnect_client(client_id).await;
|
let _ = self.disconnect_client(client_id).await;
|
||||||
|
|
||||||
// Get updated entry for the config bundle
|
// Update WG listener with the new key. Roll back the registry if this fails.
|
||||||
let entry_json = self.get_registered_client(client_id).await?;
|
if should_have_wg_peer {
|
||||||
let assigned_ip = entry_json.get("assignedIp")
|
|
||||||
.and_then(|v| v.as_str())
|
|
||||||
.unwrap_or("0.0.0.0");
|
|
||||||
|
|
||||||
// Update WG listener: remove old peer, add new peer
|
|
||||||
if self.wg_command_tx.is_some() {
|
|
||||||
if let Some(ref old_key) = old_wg_pub {
|
|
||||||
if let Err(e) = self.remove_wg_peer(old_key).await {
|
|
||||||
debug!("Failed to remove old WG peer during rotation: {}", e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
let wg_peer_config = crate::wireguard::WgPeerConfig {
|
let wg_peer_config = crate::wireguard::WgPeerConfig {
|
||||||
public_key: wg_pub.clone(),
|
public_key: wg_pub.clone(),
|
||||||
preshared_key: None,
|
preshared_key: None,
|
||||||
allowed_ips: vec![format!("{}/32", assigned_ip)],
|
allowed_ips: vec![format!("{}/32", assigned_ip.as_str())],
|
||||||
endpoint: None,
|
endpoint: None,
|
||||||
persistent_keepalive: Some(25),
|
persistent_keepalive: Some(25),
|
||||||
};
|
};
|
||||||
if let Err(e) = self.add_wg_peer(wg_peer_config).await {
|
if let Err(e) = self.add_wg_peer(wg_peer_config).await {
|
||||||
warn!("Failed to register new WG peer during rotation: {}", e);
|
let _ = state.client_registry.write().await.rotate_key(
|
||||||
|
client_id,
|
||||||
|
old_noise_pub,
|
||||||
|
old_wg_pub.clone(),
|
||||||
|
);
|
||||||
|
if let Some(ref old_key) = old_wg_pub {
|
||||||
|
let rollback_peer_config = crate::wireguard::WgPeerConfig {
|
||||||
|
public_key: old_key.clone(),
|
||||||
|
preshared_key: None,
|
||||||
|
allowed_ips: vec![format!("{}/32", assigned_ip.as_str())],
|
||||||
|
endpoint: None,
|
||||||
|
persistent_keepalive: Some(25),
|
||||||
|
};
|
||||||
|
if let Err(rollback_err) = self.add_wg_peer(rollback_peer_config).await {
|
||||||
|
warn!("Failed to restore old WG peer after rotation failure for {}: {}", client_id, rollback_err);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return Err(anyhow::anyhow!("Failed to register new WG peer during rotation for {}: {}", client_id, e));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Get updated entry for the config bundle
|
||||||
|
let entry_json = self.get_registered_client(client_id).await?;
|
||||||
|
|
||||||
let smartvpn_server_url = format!("wss://{}",
|
let smartvpn_server_url = format!("wss://{}",
|
||||||
state.config.server_endpoint.as_deref()
|
state.config.server_endpoint.as_deref()
|
||||||
.unwrap_or(&state.config.listen_addr)
|
.unwrap_or(&state.config.listen_addr)
|
||||||
@@ -1430,6 +1805,20 @@ async fn handle_client_connection(
|
|||||||
ForwardingEngine::Socket(sender) => {
|
ForwardingEngine::Socket(sender) => {
|
||||||
let _ = sender.try_send(buf[..len].to_vec());
|
let _ = sender.try_send(buf[..len].to_vec());
|
||||||
}
|
}
|
||||||
|
ForwardingEngine::Bridge(sender) => {
|
||||||
|
let _ = sender.try_send(buf[..len].to_vec());
|
||||||
|
}
|
||||||
|
ForwardingEngine::Hybrid { socket_tx, bridge_tx, routing_table } => {
|
||||||
|
if len >= 20 {
|
||||||
|
let src_ip = Ipv4Addr::new(buf[12], buf[13], buf[14], buf[15]);
|
||||||
|
let use_bridge = routing_table.read().await.get(&src_ip).copied().unwrap_or(false);
|
||||||
|
if use_bridge {
|
||||||
|
let _ = bridge_tx.try_send(buf[..len].to_vec());
|
||||||
|
} else {
|
||||||
|
let _ = socket_tx.try_send(buf[..len].to_vec());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
ForwardingEngine::Testing => {}
|
ForwardingEngine::Testing => {}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+112
-10
@@ -21,6 +21,21 @@ use crate::tunnel;
|
|||||||
/// Sessions exceeding this are aborted — the client cannot keep up.
|
/// Sessions exceeding this are aborted — the client cannot keep up.
|
||||||
const TCP_PENDING_SEND_MAX: usize = 512 * 1024;
|
const TCP_PENDING_SEND_MAX: usize = 512 * 1024;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
||||||
|
pub enum ProxyProtocolSource {
|
||||||
|
TunnelIp,
|
||||||
|
RemoteIp,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ProxyProtocolSource {
|
||||||
|
pub fn from_config(value: Option<&str>) -> Self {
|
||||||
|
match value {
|
||||||
|
Some("remoteIp") => ProxyProtocolSource::RemoteIp,
|
||||||
|
_ => ProxyProtocolSource::TunnelIp,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// ============================================================================
|
// ============================================================================
|
||||||
// Virtual IP device for smoltcp
|
// Virtual IP device for smoltcp
|
||||||
// ============================================================================
|
// ============================================================================
|
||||||
@@ -208,8 +223,10 @@ pub struct NatEngine {
|
|||||||
bridge_tx: mpsc::Sender<BridgeMessage>,
|
bridge_tx: mpsc::Sender<BridgeMessage>,
|
||||||
start_time: std::time::Instant,
|
start_time: std::time::Instant,
|
||||||
/// When true, outbound TCP connections prepend PROXY protocol v2 headers
|
/// When true, outbound TCP connections prepend PROXY protocol v2 headers
|
||||||
/// with the VPN client's tunnel IP as source address.
|
/// with VPN source identity.
|
||||||
proxy_protocol: bool,
|
proxy_protocol: bool,
|
||||||
|
proxy_protocol_source: ProxyProtocolSource,
|
||||||
|
proxy_protocol_vpn_metadata: bool,
|
||||||
/// Destination routing policy: forceTarget, block, or allow.
|
/// Destination routing policy: forceTarget, block, or allow.
|
||||||
destination_policy: Option<DestinationPolicyConfig>,
|
destination_policy: Option<DestinationPolicyConfig>,
|
||||||
}
|
}
|
||||||
@@ -225,7 +242,15 @@ enum DestinationAction {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl NatEngine {
|
impl NatEngine {
|
||||||
pub fn new(gateway_ip: Ipv4Addr, mtu: usize, state: Arc<ServerState>, proxy_protocol: bool, destination_policy: Option<DestinationPolicyConfig>) -> Self {
|
pub fn new(
|
||||||
|
gateway_ip: Ipv4Addr,
|
||||||
|
mtu: usize,
|
||||||
|
state: Arc<ServerState>,
|
||||||
|
proxy_protocol: bool,
|
||||||
|
proxy_protocol_source: ProxyProtocolSource,
|
||||||
|
proxy_protocol_vpn_metadata: bool,
|
||||||
|
destination_policy: Option<DestinationPolicyConfig>,
|
||||||
|
) -> Self {
|
||||||
let mut device = VirtualIpDevice::new(mtu);
|
let mut device = VirtualIpDevice::new(mtu);
|
||||||
let config = Config::new(HardwareAddress::Ip);
|
let config = Config::new(HardwareAddress::Ip);
|
||||||
let now = smoltcp::time::Instant::from_millis(0);
|
let now = smoltcp::time::Instant::from_millis(0);
|
||||||
@@ -258,6 +283,8 @@ impl NatEngine {
|
|||||||
bridge_tx,
|
bridge_tx,
|
||||||
start_time: std::time::Instant::now(),
|
start_time: std::time::Instant::now(),
|
||||||
proxy_protocol,
|
proxy_protocol,
|
||||||
|
proxy_protocol_source,
|
||||||
|
proxy_protocol_vpn_metadata,
|
||||||
destination_policy,
|
destination_policy,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -267,8 +294,19 @@ impl NatEngine {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Evaluate destination policy for a packet's destination IP.
|
/// Evaluate destination policy for a packet's destination IP.
|
||||||
fn evaluate_destination(&self, dst_ip: Ipv4Addr, dst_port: u16) -> DestinationAction {
|
/// Checks per-client policy first (via src_ip → client registry lookup),
|
||||||
let policy = match &self.destination_policy {
|
/// falls back to server-wide policy.
|
||||||
|
fn evaluate_destination(&self, src_ip: Ipv4Addr, dst_ip: Ipv4Addr, dst_port: u16) -> DestinationAction {
|
||||||
|
// Try per-client destination policy (lookup by tunnel IP)
|
||||||
|
let client_policy = if let Ok(registry) = self.state.client_registry.try_read() {
|
||||||
|
registry.get_by_assigned_ip(&src_ip.to_string())
|
||||||
|
.and_then(|e| e.security.as_ref())
|
||||||
|
.and_then(|s| s.destination_policy.clone())
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
};
|
||||||
|
|
||||||
|
let policy = match client_policy.as_ref().or(self.destination_policy.as_ref()) {
|
||||||
Some(p) => p,
|
Some(p) => p,
|
||||||
None => return DestinationAction::PassThrough(SocketAddr::new(dst_ip.into(), dst_port)),
|
None => return DestinationAction::PassThrough(SocketAddr::new(dst_ip.into(), dst_port)),
|
||||||
};
|
};
|
||||||
@@ -326,7 +364,7 @@ impl NatEngine {
|
|||||||
// Skip if session exists (including closing sessions — let FIN complete)
|
// Skip if session exists (including closing sessions — let FIN complete)
|
||||||
let session_exists = self.tcp_sessions.contains_key(&key);
|
let session_exists = self.tcp_sessions.contains_key(&key);
|
||||||
if is_syn && !session_exists {
|
if is_syn && !session_exists {
|
||||||
match self.evaluate_destination(dst_ip, dst_port) {
|
match self.evaluate_destination(src_ip, dst_ip, dst_port) {
|
||||||
DestinationAction::Drop => {
|
DestinationAction::Drop => {
|
||||||
debug!("NAT: destination policy blocked TCP {}:{} -> {}:{}", src_ip, src_port, dst_ip, dst_port);
|
debug!("NAT: destination policy blocked TCP {}:{} -> {}:{}", src_ip, src_port, dst_ip, dst_port);
|
||||||
return;
|
return;
|
||||||
@@ -350,7 +388,7 @@ impl NatEngine {
|
|||||||
};
|
};
|
||||||
|
|
||||||
if !self.udp_sessions.contains_key(&key) {
|
if !self.udp_sessions.contains_key(&key) {
|
||||||
match self.evaluate_destination(dst_ip, dst_port) {
|
match self.evaluate_destination(src_ip, dst_ip, dst_port) {
|
||||||
DestinationAction::Drop => {
|
DestinationAction::Drop => {
|
||||||
debug!("NAT: destination policy blocked UDP {}:{} -> {}:{}", src_ip, src_port, dst_ip, dst_port);
|
debug!("NAT: destination policy blocked UDP {}:{} -> {}:{}", src_ip, src_port, dst_ip, dst_port);
|
||||||
return;
|
return;
|
||||||
@@ -470,6 +508,9 @@ impl NatEngine {
|
|||||||
// Start bridge tasks for sessions whose handshake just completed
|
// Start bridge tasks for sessions whose handshake just completed
|
||||||
let bridge_tx_clone = self.bridge_tx.clone();
|
let bridge_tx_clone = self.bridge_tx.clone();
|
||||||
let proxy_protocol = self.proxy_protocol;
|
let proxy_protocol = self.proxy_protocol;
|
||||||
|
let proxy_protocol_source = self.proxy_protocol_source;
|
||||||
|
let proxy_protocol_vpn_metadata = self.proxy_protocol_vpn_metadata;
|
||||||
|
let state = Arc::clone(&self.state);
|
||||||
for (key, session) in self.tcp_sessions.iter_mut() {
|
for (key, session) in self.tcp_sessions.iter_mut() {
|
||||||
if !session.bridge_started && !session.closing {
|
if !session.bridge_started && !session.closing {
|
||||||
let socket = self.sockets.get_mut::<tcp::Socket>(session.smoltcp_handle);
|
let socket = self.sockets.get_mut::<tcp::Socket>(session.smoltcp_handle);
|
||||||
@@ -481,8 +522,11 @@ impl NatEngine {
|
|||||||
let k = key.clone();
|
let k = key.clone();
|
||||||
let addr = session.connect_addr;
|
let addr = session.connect_addr;
|
||||||
let pp = proxy_protocol;
|
let pp = proxy_protocol;
|
||||||
|
let pp_source = proxy_protocol_source;
|
||||||
|
let pp_metadata = proxy_protocol_vpn_metadata;
|
||||||
|
let state = Arc::clone(&state);
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
tcp_bridge_task(k, data_rx, btx, pp, addr).await;
|
tcp_bridge_task(k, data_rx, btx, pp, pp_source, pp_metadata, state, addr).await;
|
||||||
});
|
});
|
||||||
debug!("NAT: TCP handshake complete, starting bridge for {}:{} -> {}:{}",
|
debug!("NAT: TCP handshake complete, starting bridge for {}:{} -> {}:{}",
|
||||||
key.src_ip, key.src_port, key.dst_ip, key.dst_port);
|
key.src_ip, key.src_port, key.dst_ip, key.dst_port);
|
||||||
@@ -737,6 +781,9 @@ async fn tcp_bridge_task(
|
|||||||
mut data_rx: mpsc::Receiver<Vec<u8>>,
|
mut data_rx: mpsc::Receiver<Vec<u8>>,
|
||||||
bridge_tx: mpsc::Sender<BridgeMessage>,
|
bridge_tx: mpsc::Sender<BridgeMessage>,
|
||||||
proxy_protocol: bool,
|
proxy_protocol: bool,
|
||||||
|
proxy_protocol_source: ProxyProtocolSource,
|
||||||
|
proxy_protocol_vpn_metadata: bool,
|
||||||
|
state: Arc<ServerState>,
|
||||||
connect_addr: SocketAddr,
|
connect_addr: SocketAddr,
|
||||||
) {
|
) {
|
||||||
// Connect to resolved destination (may differ from key.dst_ip if policy rewrote it)
|
// Connect to resolved destination (may differ from key.dst_ip if policy rewrote it)
|
||||||
@@ -757,11 +804,21 @@ async fn tcp_bridge_task(
|
|||||||
|
|
||||||
let (mut reader, mut writer) = stream.into_split();
|
let (mut reader, mut writer) = stream.into_split();
|
||||||
|
|
||||||
// Send PROXY protocol v2 header with VPN client's tunnel IP as source
|
// Send PROXY protocol v2 header with configured client source identity.
|
||||||
if proxy_protocol {
|
if proxy_protocol {
|
||||||
let src = SocketAddr::new(key.src_ip.into(), key.src_port);
|
let (src, metadata) = build_proxy_protocol_identity(
|
||||||
|
&state,
|
||||||
|
key.src_ip,
|
||||||
|
key.src_port,
|
||||||
|
proxy_protocol_source,
|
||||||
|
proxy_protocol_vpn_metadata,
|
||||||
|
).await;
|
||||||
let dst = SocketAddr::new(key.dst_ip.into(), key.dst_port);
|
let dst = SocketAddr::new(key.dst_ip.into(), key.dst_port);
|
||||||
let pp_header = crate::proxy_protocol::build_pp_v2_header(src, dst);
|
let pp_header = crate::proxy_protocol::build_pp_v2_header_with_vpn_metadata(
|
||||||
|
src,
|
||||||
|
dst,
|
||||||
|
metadata.as_ref(),
|
||||||
|
);
|
||||||
if let Err(e) = writer.write_all(&pp_header).await {
|
if let Err(e) = writer.write_all(&pp_header).await {
|
||||||
debug!("NAT: failed to send PP v2 header to {}: {}", connect_addr, e);
|
debug!("NAT: failed to send PP v2 header to {}: {}", connect_addr, e);
|
||||||
let _ = bridge_tx.send(BridgeMessage::TcpClosed { key }).await;
|
let _ = bridge_tx.send(BridgeMessage::TcpClosed { key }).await;
|
||||||
@@ -807,6 +864,51 @@ async fn tcp_bridge_task(
|
|||||||
read_task.abort();
|
read_task.abort();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn build_proxy_protocol_identity(
|
||||||
|
state: &Arc<ServerState>,
|
||||||
|
tunnel_ip: Ipv4Addr,
|
||||||
|
tunnel_port: u16,
|
||||||
|
proxy_protocol_source: ProxyProtocolSource,
|
||||||
|
include_metadata: bool,
|
||||||
|
) -> (SocketAddr, Option<crate::proxy_protocol::VpnProxyMetadata>) {
|
||||||
|
let tunnel_addr = SocketAddr::new(tunnel_ip.into(), tunnel_port);
|
||||||
|
|
||||||
|
let client_id = state
|
||||||
|
.client_registry
|
||||||
|
.read()
|
||||||
|
.await
|
||||||
|
.get_by_assigned_ip(&tunnel_ip.to_string())
|
||||||
|
.map(|entry| entry.client_id.clone());
|
||||||
|
|
||||||
|
let client_info = if let Some(ref client_id) = client_id {
|
||||||
|
state.clients.read().await.get(client_id).cloned()
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
};
|
||||||
|
|
||||||
|
let remote_addr = client_info
|
||||||
|
.as_ref()
|
||||||
|
.and_then(|info| info.remote_addr.as_ref())
|
||||||
|
.and_then(|addr| addr.parse::<SocketAddr>().ok());
|
||||||
|
let source_addr = match proxy_protocol_source {
|
||||||
|
ProxyProtocolSource::RemoteIp => remote_addr.unwrap_or(tunnel_addr),
|
||||||
|
ProxyProtocolSource::TunnelIp => tunnel_addr,
|
||||||
|
};
|
||||||
|
|
||||||
|
let metadata = if include_metadata {
|
||||||
|
client_info.map(|info| crate::proxy_protocol::VpnProxyMetadata {
|
||||||
|
client_id: info.registered_client_id,
|
||||||
|
assigned_ip: info.assigned_ip,
|
||||||
|
transport_type: info.transport_type,
|
||||||
|
remote_addr: info.remote_addr,
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
};
|
||||||
|
|
||||||
|
(source_addr, metadata)
|
||||||
|
}
|
||||||
|
|
||||||
async fn udp_bridge_task(
|
async fn udp_bridge_task(
|
||||||
key: SessionKey,
|
key: SessionKey,
|
||||||
mut data_rx: mpsc::Receiver<Vec<u8>>,
|
mut data_rx: mpsc::Receiver<Vec<u8>>,
|
||||||
|
|||||||
+108
-25
@@ -215,6 +215,8 @@ pub enum WgCommand {
|
|||||||
struct PeerState {
|
struct PeerState {
|
||||||
tunn: Tunn,
|
tunn: Tunn,
|
||||||
public_key_b64: String,
|
public_key_b64: String,
|
||||||
|
/// Registered SmartVPN client ID when this peer belongs to a registry entry.
|
||||||
|
client_id: Option<String>,
|
||||||
allowed_ips: Vec<AllowedIp>,
|
allowed_ips: Vec<AllowedIp>,
|
||||||
endpoint: Option<SocketAddr>,
|
endpoint: Option<SocketAddr>,
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
@@ -237,6 +239,28 @@ impl PeerState {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn synthetic_wg_client_id(pubkey: &str) -> String {
|
||||||
|
format!("wg-{}", &pubkey[..8.min(pubkey.len())])
|
||||||
|
}
|
||||||
|
|
||||||
|
fn peer_client_id(peer: &PeerState) -> String {
|
||||||
|
peer.client_id
|
||||||
|
.clone()
|
||||||
|
.unwrap_or_else(|| synthetic_wg_client_id(&peer.public_key_b64))
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn registered_peer_info(
|
||||||
|
state: &Arc<ServerState>,
|
||||||
|
pubkey: &str,
|
||||||
|
) -> Option<(String, bool, bool)> {
|
||||||
|
let registry = state.client_registry.read().await;
|
||||||
|
registry.get_by_wg_key(pubkey).map(|entry| (
|
||||||
|
entry.client_id.clone(),
|
||||||
|
entry.use_host_ip.unwrap_or(false),
|
||||||
|
entry.is_enabled() && !entry.is_expired(),
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
fn add_peer_to_loop(
|
fn add_peer_to_loop(
|
||||||
peers: &mut Vec<PeerState>,
|
peers: &mut Vec<PeerState>,
|
||||||
@@ -281,6 +305,7 @@ fn add_peer_to_loop(
|
|||||||
peers.push(PeerState {
|
peers.push(PeerState {
|
||||||
tunn,
|
tunn,
|
||||||
public_key_b64: config.public_key.clone(),
|
public_key_b64: config.public_key.clone(),
|
||||||
|
client_id: None,
|
||||||
allowed_ips,
|
allowed_ips,
|
||||||
endpoint,
|
endpoint,
|
||||||
persistent_keepalive: config.persistent_keepalive,
|
persistent_keepalive: config.persistent_keepalive,
|
||||||
@@ -319,10 +344,12 @@ fn extract_peer_vpn_ip(allowed_ips: &[AllowedIp]) -> Option<Ipv4Addr> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// Fallback: use the first IPv4 address from any prefix length
|
// Fallback: use the first non-unspecified IPv4 address from any prefix length
|
||||||
for aip in allowed_ips {
|
for aip in allowed_ips {
|
||||||
if let IpAddr::V4(v4) = aip.addr {
|
if let IpAddr::V4(v4) = aip.addr {
|
||||||
return Some(v4);
|
if !v4.is_unspecified() {
|
||||||
|
return Some(v4);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
None
|
None
|
||||||
@@ -342,7 +369,7 @@ fn wg_timestamp_now() -> String {
|
|||||||
/// Returns the VPN IP.
|
/// Returns the VPN IP.
|
||||||
async fn register_wg_peer(
|
async fn register_wg_peer(
|
||||||
state: &Arc<ServerState>,
|
state: &Arc<ServerState>,
|
||||||
peer: &PeerState,
|
peer: &mut PeerState,
|
||||||
wg_return_tx: &mpsc::Sender<(String, Vec<u8>)>,
|
wg_return_tx: &mpsc::Sender<(String, Vec<u8>)>,
|
||||||
) -> Result<Option<Ipv4Addr>> {
|
) -> Result<Option<Ipv4Addr>> {
|
||||||
let vpn_ip = match extract_peer_vpn_ip(&peer.allowed_ips) {
|
let vpn_ip = match extract_peer_vpn_ip(&peer.allowed_ips) {
|
||||||
@@ -354,12 +381,24 @@ async fn register_wg_peer(
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
let client_id = format!("wg-{}", &peer.public_key_b64[..8.min(peer.public_key_b64.len())]);
|
let (client_id, use_host_ip) = match registered_peer_info(state, &peer.public_key_b64).await {
|
||||||
|
Some((client_id, use_host_ip, true)) => (client_id, use_host_ip),
|
||||||
|
Some((client_id, _, false)) => {
|
||||||
|
warn!("WG peer {} maps to disabled or expired client {}, skipping registration", peer.public_key_b64, client_id);
|
||||||
|
return Ok(None);
|
||||||
|
}
|
||||||
|
None => (synthetic_wg_client_id(&peer.public_key_b64), false),
|
||||||
|
};
|
||||||
|
peer.client_id = Some(client_id.clone());
|
||||||
|
|
||||||
// Reserve IP in the pool
|
// Reserve IP in the pool
|
||||||
if let Err(e) = state.ip_pool.lock().await.reserve(vpn_ip, &client_id) {
|
if let Err(e) = state.ip_pool.lock().await.reserve(vpn_ip, &client_id) {
|
||||||
warn!("Failed to reserve IP {} for WG peer {}: {}", vpn_ip, client_id, e);
|
warn!("Failed to reserve IP {} for WG peer {}: {}", vpn_ip, client_id, e);
|
||||||
return Ok(None);
|
return Err(e);
|
||||||
|
}
|
||||||
|
|
||||||
|
if let ForwardingEngine::Hybrid { routing_table, .. } = &*state.forwarding_engine.lock().await {
|
||||||
|
routing_table.write().await.insert(vpn_ip, use_host_ip);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Create per-peer return channel and register in tun_routes
|
// Create per-peer return channel and register in tun_routes
|
||||||
@@ -391,7 +430,7 @@ async fn connect_wg_peer(
|
|||||||
peer: &PeerState,
|
peer: &PeerState,
|
||||||
vpn_ip: Ipv4Addr,
|
vpn_ip: Ipv4Addr,
|
||||||
) {
|
) {
|
||||||
let client_id = format!("wg-{}", &peer.public_key_b64[..8.min(peer.public_key_b64.len())]);
|
let client_id = peer_client_id(peer);
|
||||||
let client_info = ClientInfo {
|
let client_info = ClientInfo {
|
||||||
client_id: client_id.clone(),
|
client_id: client_id.clone(),
|
||||||
assigned_ip: vpn_ip.to_string(),
|
assigned_ip: vpn_ip.to_string(),
|
||||||
@@ -424,24 +463,27 @@ async fn connect_wg_peer(
|
|||||||
/// Remove a WG peer from state.clients (disconnect without unregistering).
|
/// Remove a WG peer from state.clients (disconnect without unregistering).
|
||||||
async fn disconnect_wg_peer(
|
async fn disconnect_wg_peer(
|
||||||
state: &Arc<ServerState>,
|
state: &Arc<ServerState>,
|
||||||
pubkey: &str,
|
peer: &PeerState,
|
||||||
) {
|
) {
|
||||||
let client_id = format!("wg-{}", &pubkey[..8.min(pubkey.len())]);
|
let client_id = peer_client_id(peer);
|
||||||
if state.clients.write().await.remove(&client_id).is_some() {
|
if state.clients.write().await.remove(&client_id).is_some() {
|
||||||
info!("WG peer {} disconnected (removed from active clients)", pubkey);
|
info!("WG peer {} disconnected (removed from active clients)", peer.public_key_b64);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Unregister a WG peer from ServerState.
|
/// Unregister a WG peer from ServerState.
|
||||||
async fn unregister_wg_peer(
|
async fn unregister_wg_peer(
|
||||||
state: &Arc<ServerState>,
|
state: &Arc<ServerState>,
|
||||||
pubkey: &str,
|
peer: &PeerState,
|
||||||
vpn_ip: Option<Ipv4Addr>,
|
vpn_ip: Option<Ipv4Addr>,
|
||||||
) {
|
) {
|
||||||
let client_id = format!("wg-{}", &pubkey[..8.min(pubkey.len())]);
|
let client_id = peer_client_id(peer);
|
||||||
|
|
||||||
if let Some(ip) = vpn_ip {
|
if let Some(ip) = vpn_ip {
|
||||||
state.tun_routes.write().await.remove(&ip);
|
state.tun_routes.write().await.remove(&ip);
|
||||||
|
if let ForwardingEngine::Hybrid { routing_table, .. } = &*state.forwarding_engine.lock().await {
|
||||||
|
routing_table.write().await.remove(&ip);
|
||||||
|
}
|
||||||
state.ip_pool.lock().await.release(&ip);
|
state.ip_pool.lock().await.release(&ip);
|
||||||
}
|
}
|
||||||
state.clients.write().await.remove(&client_id);
|
state.clients.write().await.remove(&client_id);
|
||||||
@@ -499,6 +541,7 @@ pub async fn run_wg_listener(
|
|||||||
peers.push(PeerState {
|
peers.push(PeerState {
|
||||||
tunn,
|
tunn,
|
||||||
public_key_b64: peer_config.public_key.clone(),
|
public_key_b64: peer_config.public_key.clone(),
|
||||||
|
client_id: None,
|
||||||
allowed_ips,
|
allowed_ips,
|
||||||
endpoint,
|
endpoint,
|
||||||
persistent_keepalive: peer_config.persistent_keepalive,
|
persistent_keepalive: peer_config.persistent_keepalive,
|
||||||
@@ -521,9 +564,13 @@ pub async fn run_wg_listener(
|
|||||||
// Register initial peers in ServerState (IP reservation + tun_routes only, NOT state.clients)
|
// Register initial peers in ServerState (IP reservation + tun_routes only, NOT state.clients)
|
||||||
let mut peer_vpn_ips: HashMap<String, Ipv4Addr> = HashMap::new();
|
let mut peer_vpn_ips: HashMap<String, Ipv4Addr> = HashMap::new();
|
||||||
for peer in peers.iter_mut() {
|
for peer in peers.iter_mut() {
|
||||||
if let Ok(Some(ip)) = register_wg_peer(&state, peer, &wg_return_tx).await {
|
match register_wg_peer(&state, peer, &wg_return_tx).await {
|
||||||
peer_vpn_ips.insert(peer.public_key_b64.clone(), ip);
|
Ok(Some(ip)) => {
|
||||||
peer.vpn_ip = Some(ip);
|
peer_vpn_ips.insert(peer.public_key_b64.clone(), ip);
|
||||||
|
peer.vpn_ip = Some(ip);
|
||||||
|
}
|
||||||
|
Ok(None) => {}
|
||||||
|
Err(e) => warn!("Failed to register initial WG peer {}: {}", peer.public_key_b64, e),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -576,6 +623,20 @@ pub async fn run_wg_listener(
|
|||||||
ForwardingEngine::Socket(sender) => {
|
ForwardingEngine::Socket(sender) => {
|
||||||
let _ = sender.try_send(packet.to_vec());
|
let _ = sender.try_send(packet.to_vec());
|
||||||
}
|
}
|
||||||
|
ForwardingEngine::Bridge(sender) => {
|
||||||
|
let _ = sender.try_send(packet.to_vec());
|
||||||
|
}
|
||||||
|
ForwardingEngine::Hybrid { socket_tx, bridge_tx, routing_table } => {
|
||||||
|
if packet.len() >= 20 {
|
||||||
|
let src_ip = Ipv4Addr::new(packet[12], packet[13], packet[14], packet[15]);
|
||||||
|
let use_bridge = routing_table.read().await.get(&src_ip).copied().unwrap_or(false);
|
||||||
|
if use_bridge {
|
||||||
|
let _ = bridge_tx.try_send(packet.to_vec());
|
||||||
|
} else {
|
||||||
|
let _ = socket_tx.try_send(packet.to_vec());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
ForwardingEngine::Testing => {}
|
ForwardingEngine::Testing => {}
|
||||||
}
|
}
|
||||||
peer.stats.bytes_received += pkt_len;
|
peer.stats.bytes_received += pkt_len;
|
||||||
@@ -608,6 +669,20 @@ pub async fn run_wg_listener(
|
|||||||
ForwardingEngine::Socket(sender) => {
|
ForwardingEngine::Socket(sender) => {
|
||||||
let _ = sender.try_send(packet.to_vec());
|
let _ = sender.try_send(packet.to_vec());
|
||||||
}
|
}
|
||||||
|
ForwardingEngine::Bridge(sender) => {
|
||||||
|
let _ = sender.try_send(packet.to_vec());
|
||||||
|
}
|
||||||
|
ForwardingEngine::Hybrid { socket_tx, bridge_tx, routing_table } => {
|
||||||
|
if packet.len() >= 20 {
|
||||||
|
let src_ip = Ipv4Addr::new(packet[12], packet[13], packet[14], packet[15]);
|
||||||
|
let use_bridge = routing_table.read().await.get(&src_ip).copied().unwrap_or(false);
|
||||||
|
if use_bridge {
|
||||||
|
let _ = bridge_tx.try_send(packet.to_vec());
|
||||||
|
} else {
|
||||||
|
let _ = socket_tx.try_send(packet.to_vec());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
ForwardingEngine::Testing => {}
|
ForwardingEngine::Testing => {}
|
||||||
}
|
}
|
||||||
peer.stats.bytes_received += pkt_len;
|
peer.stats.bytes_received += pkt_len;
|
||||||
@@ -675,7 +750,7 @@ pub async fn run_wg_listener(
|
|||||||
warn!("WG peer {} connection expired", peer.public_key_b64);
|
warn!("WG peer {} connection expired", peer.public_key_b64);
|
||||||
if peer.is_connected {
|
if peer.is_connected {
|
||||||
peer.is_connected = false;
|
peer.is_connected = false;
|
||||||
disconnect_wg_peer(&state, &peer.public_key_b64).await;
|
disconnect_wg_peer(&state, peer).await;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
TunnResult::Err(e) => {
|
TunnResult::Err(e) => {
|
||||||
@@ -703,7 +778,7 @@ pub async fn run_wg_listener(
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Only update ClientInfo if peer is connected (in state.clients)
|
// Only update ClientInfo if peer is connected (in state.clients)
|
||||||
let client_id = format!("wg-{}", &peer.public_key_b64[..8.min(peer.public_key_b64.len())]);
|
let client_id = peer_client_id(peer);
|
||||||
if let Some(info) = clients.get_mut(&client_id) {
|
if let Some(info) = clients.get_mut(&client_id) {
|
||||||
info.bytes_sent = peer.stats.bytes_sent;
|
info.bytes_sent = peer.stats.bytes_sent;
|
||||||
info.bytes_received = peer.stats.bytes_received;
|
info.bytes_received = peer.stats.bytes_received;
|
||||||
@@ -721,7 +796,7 @@ pub async fn run_wg_listener(
|
|||||||
if now.duration_since(last) > std::time::Duration::from_secs(180) {
|
if now.duration_since(last) > std::time::Duration::from_secs(180) {
|
||||||
info!("WG peer {} idle timeout (180s), disconnecting", peer.public_key_b64);
|
info!("WG peer {} idle timeout (180s), disconnecting", peer.public_key_b64);
|
||||||
peer.is_connected = false;
|
peer.is_connected = false;
|
||||||
disconnect_wg_peer(&state, &peer.public_key_b64).await;
|
disconnect_wg_peer(&state, peer).await;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -732,7 +807,12 @@ pub async fn run_wg_listener(
|
|||||||
cmd = command_rx.recv() => {
|
cmd = command_rx.recv() => {
|
||||||
match cmd {
|
match cmd {
|
||||||
Some(WgCommand::AddPeer(peer_config, resp_tx)) => {
|
Some(WgCommand::AddPeer(peer_config, resp_tx)) => {
|
||||||
let result = add_peer_to_loop(
|
if let Some((client_id, _, false)) = registered_peer_info(&state, &peer_config.public_key).await {
|
||||||
|
let _ = resp_tx.send(Err(anyhow!("WG peer maps to disabled or expired client: {}", client_id)));
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut result = add_peer_to_loop(
|
||||||
&mut peers,
|
&mut peers,
|
||||||
&peer_config,
|
&peer_config,
|
||||||
&peer_index,
|
&peer_index,
|
||||||
@@ -747,20 +827,23 @@ pub async fn run_wg_listener(
|
|||||||
peer_vpn_ips.insert(peer_config.public_key.clone(), ip);
|
peer_vpn_ips.insert(peer_config.public_key.clone(), ip);
|
||||||
peer.vpn_ip = Some(ip);
|
peer.vpn_ip = Some(ip);
|
||||||
}
|
}
|
||||||
Ok(None) => {}
|
Ok(None) => {
|
||||||
|
peers.retain(|p| p.public_key_b64 != peer_config.public_key);
|
||||||
|
result = Err(anyhow!("WG peer was not registered"));
|
||||||
|
}
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
warn!("Failed to register WG peer: {}", e);
|
peers.retain(|p| p.public_key_b64 != peer_config.public_key);
|
||||||
|
result = Err(e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
let _ = resp_tx.send(result);
|
let _ = resp_tx.send(result);
|
||||||
}
|
}
|
||||||
Some(WgCommand::RemovePeer(pubkey, resp_tx)) => {
|
Some(WgCommand::RemovePeer(pubkey, resp_tx)) => {
|
||||||
let prev_len = peers.len();
|
if let Some(index) = peers.iter().position(|p| p.public_key_b64 == pubkey) {
|
||||||
peers.retain(|p| p.public_key_b64 != pubkey);
|
let peer = peers.remove(index);
|
||||||
if peers.len() < prev_len {
|
|
||||||
let vpn_ip = peer_vpn_ips.remove(&pubkey);
|
let vpn_ip = peer_vpn_ips.remove(&pubkey);
|
||||||
unregister_wg_peer(&state, &pubkey, vpn_ip).await;
|
unregister_wg_peer(&state, &peer, vpn_ip).await;
|
||||||
let _ = resp_tx.send(Ok(()));
|
let _ = resp_tx.send(Ok(()));
|
||||||
} else {
|
} else {
|
||||||
let _ = resp_tx.send(Err(anyhow!("Peer not found: {}", pubkey)));
|
let _ = resp_tx.send(Err(anyhow!("Peer not found: {}", pubkey)));
|
||||||
@@ -784,7 +867,7 @@ pub async fn run_wg_listener(
|
|||||||
// Cleanup: unregister all peers from ServerState
|
// Cleanup: unregister all peers from ServerState
|
||||||
for peer in &peers {
|
for peer in &peers {
|
||||||
let vpn_ip = peer_vpn_ips.get(&peer.public_key_b64).copied();
|
let vpn_ip = peer_vpn_ips.get(&peer.public_key_b64).copied();
|
||||||
unregister_wg_peer(&state, &peer.public_key_b64, vpn_ip).await;
|
unregister_wg_peer(&state, peer, vpn_ip).await;
|
||||||
}
|
}
|
||||||
|
|
||||||
info!("WireGuard listener stopped");
|
info!("WireGuard listener stopped");
|
||||||
|
|||||||
@@ -3,6 +3,6 @@
|
|||||||
*/
|
*/
|
||||||
export const commitinfo = {
|
export const commitinfo = {
|
||||||
name: '@push.rocks/smartvpn',
|
name: '@push.rocks/smartvpn',
|
||||||
version: '1.17.1',
|
version: '1.20.0',
|
||||||
description: 'A VPN solution with TypeScript control plane and Rust data plane daemon'
|
description: 'A VPN solution with TypeScript control plane and Rust data plane daemon'
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -333,17 +333,35 @@ export class VpnServer extends plugins.events.EventEmitter {
|
|||||||
/**
|
/**
|
||||||
* Stop the daemon bridge.
|
* Stop the daemon bridge.
|
||||||
*/
|
*/
|
||||||
public stop(): void {
|
public async stop(): Promise<void> {
|
||||||
// Clean up nftables rules
|
// Clean up nftables rules
|
||||||
if (this.nftHealthInterval) {
|
if (this.nftHealthInterval) {
|
||||||
clearInterval(this.nftHealthInterval);
|
clearInterval(this.nftHealthInterval);
|
||||||
this.nftHealthInterval = undefined;
|
this.nftHealthInterval = undefined;
|
||||||
}
|
}
|
||||||
if (this.nft) {
|
if (this.nft) {
|
||||||
this.nft.cleanup().catch(() => {}); // best-effort cleanup
|
try {
|
||||||
|
await this.nft.cleanup();
|
||||||
|
} catch (e) {
|
||||||
|
console.warn(`[smartvpn] nftables cleanup failed: ${e}`);
|
||||||
|
}
|
||||||
this.nft = undefined;
|
this.nft = undefined;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Wait for bridge process to exit (with timeout)
|
||||||
|
const exitPromise = new Promise<void>((resolve) => {
|
||||||
|
if (!this.bridge.running) {
|
||||||
|
resolve();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const timeout = setTimeout(() => resolve(), 5000);
|
||||||
|
this.bridge.once('exit', () => {
|
||||||
|
clearTimeout(timeout);
|
||||||
|
resolve();
|
||||||
|
});
|
||||||
|
});
|
||||||
this.bridge.stop();
|
this.bridge.stop();
|
||||||
|
await exitPromise;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|||||||
@@ -92,8 +92,9 @@ export interface IVpnServerConfig {
|
|||||||
/** Enable NAT/masquerade for client traffic */
|
/** Enable NAT/masquerade for client traffic */
|
||||||
enableNat?: boolean;
|
enableNat?: boolean;
|
||||||
/** Forwarding mode: 'tun' (kernel TUN, requires root), 'socket' (userspace NAT),
|
/** Forwarding mode: 'tun' (kernel TUN, requires root), 'socket' (userspace NAT),
|
||||||
|
* 'bridge' (L2 bridge to host LAN), 'hybrid' (per-client socket+bridge),
|
||||||
* or 'testing' (monitoring only). Default: 'testing'. */
|
* or 'testing' (monitoring only). Default: 'testing'. */
|
||||||
forwardingMode?: 'tun' | 'socket' | 'testing';
|
forwardingMode?: 'tun' | 'socket' | 'bridge' | 'hybrid' | 'testing';
|
||||||
/** Default rate limit for new clients (bytes/sec). Omit for unlimited. */
|
/** Default rate limit for new clients (bytes/sec). Omit for unlimited. */
|
||||||
defaultRateLimitBytesPerSec?: number;
|
defaultRateLimitBytesPerSec?: number;
|
||||||
/** Default burst size for new clients (bytes). Omit for unlimited. */
|
/** Default burst size for new clients (bytes). Omit for unlimited. */
|
||||||
@@ -121,10 +122,15 @@ export interface IVpnServerConfig {
|
|||||||
* Supports exact IPs, CIDR, wildcards, ranges. */
|
* Supports exact IPs, CIDR, wildcards, ranges. */
|
||||||
connectionIpBlockList?: string[];
|
connectionIpBlockList?: string[];
|
||||||
/** When true and forwardingMode is 'socket', the userspace NAT engine prepends
|
/** When true and forwardingMode is 'socket', the userspace NAT engine prepends
|
||||||
* PROXY protocol v2 headers on outbound TCP connections, conveying the VPN client's
|
* PROXY protocol v2 headers on outbound TCP connections. */
|
||||||
* tunnel IP as the source address. This allows downstream services (e.g. SmartProxy)
|
|
||||||
* to see the real VPN client identity instead of 127.0.0.1. */
|
|
||||||
socketForwardProxyProtocol?: boolean;
|
socketForwardProxyProtocol?: boolean;
|
||||||
|
/** Source address to place into outbound PROXY v2 headers.
|
||||||
|
* 'tunnelIp' preserves legacy behavior. 'remoteIp' exposes the VPN client's
|
||||||
|
* real connecting IP when known, with tunnel IP fallback. */
|
||||||
|
socketForwardProxyProtocolSource?: 'tunnelIp' | 'remoteIp';
|
||||||
|
/** When true, outbound PROXY v2 headers include authenticated SmartVPN metadata
|
||||||
|
* in a vendor TLV: clientId, assignedIp, transportType, and remoteAddr. */
|
||||||
|
socketForwardProxyProtocolVpnMetadata?: boolean;
|
||||||
/** Destination routing policy for VPN client traffic (socket mode).
|
/** Destination routing policy for VPN client traffic (socket mode).
|
||||||
* Controls where decrypted traffic goes: allow through, block, or redirect to a target.
|
* Controls where decrypted traffic goes: allow through, block, or redirect to a target.
|
||||||
* Default: all traffic passes through (backward compatible). */
|
* Default: all traffic passes through (backward compatible). */
|
||||||
@@ -137,6 +143,22 @@ export interface IVpnServerConfig {
|
|||||||
* Controls what traffic the client routes through the VPN tunnel.
|
* Controls what traffic the client routes through the VPN tunnel.
|
||||||
* Defaults to ['0.0.0.0/0'] (full tunnel). Set to e.g. ['10.8.0.0/24'] for split tunnel. */
|
* Defaults to ['0.0.0.0/0'] (full tunnel). Set to e.g. ['10.8.0.0/24'] for split tunnel. */
|
||||||
clientAllowedIPs?: string[];
|
clientAllowedIPs?: string[];
|
||||||
|
|
||||||
|
// Bridge mode configuration (forwardingMode: 'bridge')
|
||||||
|
|
||||||
|
/** LAN subnet CIDR for bridge mode (e.g. '192.168.1.0/24').
|
||||||
|
* VPN clients get IPs from this subnet instead of the VPN subnet.
|
||||||
|
* Required when forwardingMode is 'bridge'. */
|
||||||
|
bridgeLanSubnet?: string;
|
||||||
|
/** Physical network interface to bridge (e.g. 'eth0').
|
||||||
|
* Auto-detected from the default route if omitted. */
|
||||||
|
bridgePhysicalInterface?: string;
|
||||||
|
/** Start of VPN client IP range within the LAN subnet (host offset, e.g. 200 for .200).
|
||||||
|
* Default: 200. */
|
||||||
|
bridgeIpRangeStart?: number;
|
||||||
|
/** End of VPN client IP range within the LAN subnet (host offset, e.g. 250 for .250).
|
||||||
|
* Default: 250. */
|
||||||
|
bridgeIpRangeEnd?: number;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -310,6 +332,10 @@ export interface IClientSecurity {
|
|||||||
maxConnections?: number;
|
maxConnections?: number;
|
||||||
/** Per-client rate limiting. */
|
/** Per-client rate limiting. */
|
||||||
rateLimit?: IClientRateLimit;
|
rateLimit?: IClientRateLimit;
|
||||||
|
/** Per-client destination routing policy override.
|
||||||
|
* When set, overrides the server-level destinationPolicy for this client's traffic.
|
||||||
|
* Supports the same options: forceTarget, block, allow with allow/block lists. */
|
||||||
|
destinationPolicy?: IDestinationPolicy;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -341,6 +367,21 @@ export interface IClientEntry {
|
|||||||
expiresAt?: string;
|
expiresAt?: string;
|
||||||
/** Assigned VPN IP address (set by server) */
|
/** Assigned VPN IP address (set by server) */
|
||||||
assignedIp?: string;
|
assignedIp?: string;
|
||||||
|
|
||||||
|
// Per-client bridge/host-IP settings
|
||||||
|
|
||||||
|
/** If true, client gets a host network IP via bridge mode (L2 to LAN).
|
||||||
|
* If false (default), client gets a VPN subnet IP via socket/NAT mode. */
|
||||||
|
useHostIp?: boolean;
|
||||||
|
/** If true and useHostIp is true, obtain IP via DHCP relay.
|
||||||
|
* If false or omitted, use staticIp or auto-assign from bridge IP range. */
|
||||||
|
useDhcp?: boolean;
|
||||||
|
/** Static LAN IP when useHostIp is true and useDhcp is false. */
|
||||||
|
staticIp?: string;
|
||||||
|
/** If true, assign this client to a specific 802.1Q VLAN on the bridge. */
|
||||||
|
forceVlan?: boolean;
|
||||||
|
/** 802.1Q VLAN ID (1-4094). Required when forceVlan is true. */
|
||||||
|
vlanId?: number;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|||||||
Reference in New Issue
Block a user