Compare commits

..
8 Commits
Author SHA1 Message Date
lbl8603 a76280c53b 调整代码结构 2024-05-29 22:47:48 +08:00
lbl8603 90ff03e27c 计算密码hash 2024-05-28 23:21:43 +08:00
lbl8603 5eed05989f 增加ChaCah20和signal-hook 2024-05-28 22:49:06 +08:00
lbl8603 ac939746bd 增加ChaCah20 2024-05-28 22:48:46 +08:00
lbl8603 b7532e89c2 简化条件编译 2024-05-28 22:48:09 +08:00
lbl8603 74f44d6961 增加ChaCah20 2024-05-28 22:47:50 +08:00
lbl8603 86fc27c233 使用anyhow调整错误处理 2024-05-28 22:45:15 +08:00
lbl8603 4b6bb0e5f7 调整加密 2024-05-28 22:42:58 +08:00
43 changed files with 1148 additions and 539 deletions
Generated
+56
View File
@@ -212,6 +212,36 @@ version = "1.0.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd" checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
[[package]]
name = "cfg_aliases"
version = "0.2.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "613afe47fcd5fac7ccf1db93babcb082c5994d996f20b8b159f2ad1658eb5724"
[[package]]
name = "chacha20"
version = "0.9.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c3613f74bd2eac03dad61bd53dbe620703d4371614fe0bc3b9f04dd36fe4e818"
dependencies = [
"cfg-if",
"cipher",
"cpufeatures",
]
[[package]]
name = "chacha20poly1305"
version = "0.10.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "10cd79432192d1c0f4e1a0fef9527696cc039165d729fb41b3f4f4f354c2dc35"
dependencies = [
"aead",
"chacha20",
"cipher",
"poly1305",
"zeroize",
]
[[package]] [[package]]
name = "chrono" name = "chrono"
version = "0.4.38" version = "0.4.38"
@@ -234,6 +264,7 @@ checksum = "773f3b9af64447d2ce9850330c473515014aa235e6a783b02db81ff39e4a3dad"
dependencies = [ dependencies = [
"crypto-common", "crypto-common",
"inout", "inout",
"zeroize",
] ]
[[package]] [[package]]
@@ -967,6 +998,17 @@ version = "0.3.30"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d231b230927b5e4ad203db57bbcbee2802f6bce620b1e4a9024a07d94e2907ec" checksum = "d231b230927b5e4ad203db57bbcbee2802f6bce620b1e4a9024a07d94e2907ec"
[[package]]
name = "poly1305"
version = "0.8.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8159bd90725d2df49889a078b54f4f79e87f1f8a8444194cdca81d38f5393abf"
dependencies = [
"cpufeatures",
"opaque-debug",
"universal-hash",
]
[[package]] [[package]]
name = "polyval" name = "polyval"
version = "0.6.2" version = "0.6.2"
@@ -1324,6 +1366,16 @@ dependencies = [
"digest", "digest",
] ]
[[package]]
name = "signal-hook"
version = "0.3.17"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8621587d4798caf8eb44879d42e56b9a93ea5dcd315a6487c357130095b62801"
dependencies = [
"libc",
"signal-hook-registry",
]
[[package]] [[package]]
name = "signal-hook-registry" name = "signal-hook-registry"
version = "1.4.2" version = "1.4.2"
@@ -1612,6 +1664,9 @@ dependencies = [
"anyhow", "anyhow",
"bytes", "bytes",
"cbc", "cbc",
"cfg_aliases",
"chacha20",
"chacha20poly1305",
"crossbeam-epoch", "crossbeam-epoch",
"crossbeam-queue", "crossbeam-queue",
"crossbeam-utils", "crossbeam-utils",
@@ -1658,6 +1713,7 @@ dependencies = [
"rand", "rand",
"serde", "serde",
"serde_yaml", "serde_yaml",
"signal-hook",
"sudo", "sudo",
"uuid", "uuid",
"vnt", "vnt",
+18 -17
View File
@@ -73,23 +73,24 @@ cargo build -p vnt-cli --no-default-features
features说明 features说明
| feature | 说明 | 是否默认 | | feature | 说明 | 是否默认 |
|------------------|----------------------|------| |-------------------|--------------------------------|------|
| openssl | 使用openssl中的aes_ecb算法 | 否 | | openssl | 使用openssl中的加密算法 | 否 |
| openssl-vendored | 从源码编译openssl | 否 | | openssl-vendored | 从源码编译openssl | 否 |
| ring-cipher | 使用ring中的aes_gcm算法 | 否 | | ring-cipher | 使用ring中的加密算法 | 否 |
| aes_cbc | 支持aes_cbc加密 | 是 | | aes_cbc | 支持aes_cbc加密 | 是 |
| aes_ecb | 支持aes_ecb加密 | 是 | | aes_ecb | 支持aes_ecb加密 | 是 |
| aes_gcm | 支持aes_gcm加密 | 是 | | aes_gcm | 支持aes_gcm加密 | 是 |
| sm4_cbc | 支持sm4_cbc加密 | 是 | | sm4_cbc | 支持sm4_cbc加密 | 是 |
| server_encrypt | 支持服务端加密 | 是 | | chacha20_poly1305 | 支持chacha20和chacha20_poly1305加密 | 是 |
| ip_proxy | 内置ip代理 | 是 | | server_encrypt | 支持服务端加密 | 是 |
| port_mapping | 端口映射 | 是 | | ip_proxy | 内置ip代理 | 是 |
| log | 日志 | 是 | | port_mapping | 端口映射 | 是 |
| command | list、route等命令 | 是 | | log | 日志 | 是 |
| file_config | yaml配置文件 | 是 | | command | list、route等命令 | 是 |
| lz4 | lz4压缩 | 是 | | file_config | yaml配置文件 | 是 |
| zstd | zstd压缩 | | | lz4 | lz4压缩 | |
| zstd | zstd压缩 | 否 |
### ip转发/代理 ### ip转发/代理
+3 -1
View File
@@ -24,12 +24,13 @@ features = [
[target.'cfg(any(target_os = "linux",target_os = "macos"))'.dependencies] [target.'cfg(any(target_os = "linux",target_os = "macos"))'.dependencies]
sudo = "0.6.0" sudo = "0.6.0"
signal-hook = "0.3.17"
[target.'cfg(target_os = "windows")'.dependencies] [target.'cfg(target_os = "windows")'.dependencies]
winapi = { version = "0.3.9", features = ["handleapi", "processthreadsapi", "winnt", "securitybaseapi", "impl-default"] } winapi = { version = "0.3.9", features = ["handleapi", "processthreadsapi", "winnt", "securitybaseapi", "impl-default"] }
[features] [features]
default = ["server_encrypt", "aes_gcm", "aes_cbc", "aes_ecb", "sm4_cbc", "ip_proxy", "port_mapping", "log", "command", "file_config", "lz4"] default = ["server_encrypt", "aes_gcm", "aes_cbc", "aes_ecb", "sm4_cbc", "chacha20_poly1305", "ip_proxy", "port_mapping", "log", "command", "file_config", "lz4"]
openssl = ["vnt/openssl"] openssl = ["vnt/openssl"]
openssl-vendored = ["vnt/openssl-vendored"] openssl-vendored = ["vnt/openssl-vendored"]
ring-cipher = ["vnt/ring-cipher"] ring-cipher = ["vnt/ring-cipher"]
@@ -37,6 +38,7 @@ aes_cbc = ["vnt/aes_cbc"]
aes_ecb = ["vnt/aes_ecb"] aes_ecb = ["vnt/aes_ecb"]
sm4_cbc = ["vnt/sm4_cbc"] sm4_cbc = ["vnt/sm4_cbc"]
aes_gcm = ["vnt/aes_gcm"] aes_gcm = ["vnt/aes_gcm"]
chacha20_poly1305 = ["vnt/chacha20_poly1305"]
server_encrypt = ["vnt/server_encrypt"] server_encrypt = ["vnt/server_encrypt"]
ip_proxy = ["vnt/ip_proxy"] ip_proxy = ["vnt/ip_proxy"]
port_mapping = ["vnt/port_mapping"] port_mapping = ["vnt/port_mapping"]
+86 -17
View File
@@ -1,33 +1,54 @@
## 模块介绍 ## 模块介绍
体积小,可以在服务器、路由器等环境使用 体积小,可以在服务器、路由器等环境使用
## 详细参数说明 ## 详细参数说明
### -k `<token>` ### -k `<token>`
一个虚拟局域网的标识,在同一服务器下,相同token的设备会组建一个局域网 一个虚拟局域网的标识,在同一服务器下,相同token的设备会组建一个局域网
### -n `<name>` ### -n `<name>`
设备名称,方便区分不同设备 设备名称,方便区分不同设备
### -d `<id>` ### -d `<id>`
设备id,每台设备的唯一标识,注意不要重复 设备id,每台设备的唯一标识,注意不要重复
### -c ### -c
关闭控制台交互式命令,后台运行时可以加此参数 关闭控制台交互式命令,后台运行时可以加此参数
### -s `<server>` ### -s `<server>`
注册和中继服务器地址,注册和转发数据,以'TXT:'开头表示解析TXT记录,TXT记录内容必须是'host:port'形式的服务器地址 注册和中继服务器地址,注册和转发数据,以'TXT:'开头表示解析TXT记录,TXT记录内容必须是'host:port'形式的服务器地址
### -e `<stun-server>` ### -e `<stun-server>`
使用stun服务探测客户端NAT类型,不同类型有不同的打洞策略 使用stun服务探测客户端NAT类型,不同类型有不同的打洞策略
### -a ### -a
加了此参数表示使用tap网卡,默认使用tun网卡,tun网卡效率更高 加了此参数表示使用tap网卡,默认使用tun网卡,tun网卡效率更高
注意:仅在windows上支持使用tap,用于兼容低版本windows系统(低版本windows不支持wintun 注意:仅在windows上支持使用tap,用于兼容低版本windows系统(低版本windows不支持wintun
### --nic `<tun0>` ### --nic `<tun0>`
指定虚拟网卡名称,默认tun模式使用vnt-tuntap模式使用vnt-tap 指定虚拟网卡名称,默认tun模式使用vnt-tuntap模式使用vnt-tap
### -i `<in-ip>`、-o `<out-ip>` ### -i `<in-ip>`、-o `<out-ip>`
配置点对网(IP代理)时使用,例如A(虚拟ip:10.26.0.2)通过B(虚拟ip:10.26.0.3,本地出口ip:192.168.0.10)访问C(目标网段192.168.0.0/24) 配置点对网(IP代理)时使用,例如A(虚拟ip:10.26.0.2)通过B(虚拟ip:10.26.0.3,本地出口ip:192.168.0.10)访问C(
目标网段192.168.0.0/24)
则在A配置 **'-i 192.168.0.0/24,10.26.0.3'** ,表示将192.168.0.0/24网段的数据都转发到10.26.0.3节点 则在A配置 **'-i 192.168.0.0/24,10.26.0.3'** ,表示将192.168.0.0/24网段的数据都转发到10.26.0.3节点
在B配置 **'-o 192.168.0.0/24'** ,表示允许将数据转发到 192.168.0.0/24 ,允许转发所有网段可以使用 **'-o 0.0.0.0/0'** 在B配置 **'-o 192.168.0.0/24'** ,表示允许将数据转发到 192.168.0.0/24 ,允许转发所有网段可以使用 **'-o 0.0.0.0/0'**
-i和-o参数均可使用多次,来指定不同网段,例如 **'-o 192.168.1.0/24 -o 192.168.2.0/24'** 表示允许转发目标为192.168.1.0/24或192.168.2.0/24这两个网段的数据 -i和-o参数均可使用多次,来指定不同网段,例如 **'-o 192.168.1.0/24 -o 192.168.2.0/24'**
表示允许转发目标为192.168.1.0/24或192.168.2.0/24这两个网段的数据
### -w `<password>` ### -w `<password>`
@@ -39,9 +60,11 @@
| 大于等于8 | AES256-GCM | | 大于等于8 | AES256-GCM |
### -W ### -W
开启和服务端通信的数据加密,采用rsa+aes256gcm加密客户端和服务端之间通信的数据,可以避免token泄漏、中间人攻击 开启和服务端通信的数据加密,采用rsa+aes256gcm加密客户端和服务端之间通信的数据,可以避免token泄漏、中间人攻击
注意: 注意:
1. -w `<password>`是用于客户端-客户端之间的加密,password不会传递到服务端,只添加这个参数不会加密客户端-服务端通信的数据 1. -w `<password>`是用于客户端-客户端之间的加密,password不会传递到服务端,只添加这个参数不会加密客户端-服务端通信的数据
2. -W 用于开启客户端-服务端之间的加密 2. -W 用于开启客户端-服务端之间的加密
@@ -49,54 +72,82 @@
设置虚拟网卡的mtu值,大多数情况下使用默认值效率会更高,也可根据实际情况微调这个值,不加密默认为1450,加密默认为1410 设置虚拟网卡的mtu值,大多数情况下使用默认值效率会更高,也可根据实际情况微调这个值,不加密默认为1450,加密默认为1410
### --tcp ### --tcp
和服务端使用tcp通信。有些网络提供商对UDP限制比较大,这个时候可以选择使用TCP模式,提高稳定性。一般来说udp延迟和消耗更低 和服务端使用tcp通信。有些网络提供商对UDP限制比较大,这个时候可以选择使用TCP模式,提高稳定性。一般来说udp延迟和消耗更低
### --ip `<IP>` ### --ip `<IP>`
指定虚拟ip,指定的ip不能和其他设备重复,必须有效并且在服务端所属网段下,默认情况由服务端分配 指定虚拟ip,指定的ip不能和其他设备重复,必须有效并且在服务端所属网段下,默认情况由服务端分配
### --par `<parallel>` ### --par `<parallel>`
任务并行度(必须为正整数),默认值为1,该值表示处理网卡读写的任务数,组网设备数较多、处理延迟较大时可适当调大此值 任务并行度(必须为正整数),默认值为1,该值表示处理网卡读写的任务数,组网设备数较多、处理延迟较大时可适当调大此值
### --model `<model>` ### --model `<model>`
加密模式,可选值 aes_gcm/aes_cbc/aes_ecb/sm4_cbc,默认使用aes_gcm,通常情况aes_gcm安全性高、aes_ecb性能更好,但是在低性能设备上sm4_cbc也许速度会更快;
加密模式,可选值
aes_gcm/aes_cbc/aes_ecb/sm4_cbc/chacha20_poly1305/chacha20/xor,默认使用aes_gcm,通常情况aes_gcm和chacha20_poly1305安全性高。
各种加密模式的安全性和速度都不相同,请按需选取
| 密码位数 | model | 加密算法 | 特别说明:xor只是对数据进行简单异或,仅仅避免了明文传输,安全性很差,同时对性能影响也极小;
|-------|---------|------------|
| 1~8位 | aes_gcm | AES128-GCM | | 密码位数 | model | 加密算法 |
| `>=`8 | aes_gcm | AES256-GCM | |-------|-------------------|-------------------|
| 1~8位 | aes_cbc | AES128-CBC | | 1~8位 | aes_gcm | AES128-GCM |
| `>=`8 | aes_cbc | AES256-CBC | | `>=`8 | aes_gcm | AES256-GCM |
| 1~8位 | aes_ecb | AES128-ECB | | 1~8位 | aes_cbc | AES128-CBC |
| `>=`8 | aes_ecb | AES256-ECB | | `>=`8 | aes_cbc | AES256-CBC |
| `>0` | sm4_cbc | SM4-CBC | | 1~8位 | aes_ecb | AES128-ECB |
### --finger | `>=`8 | aes_ecb | AES256-ECB |
| `>0` | sm4_cbc | SM4-CBC |
| `>0` | chacha20_poly1305 | ChaCha20-Poly1305 |
| `>0` | chacha20 | ChaCha20 |
| `>0` | xor | 简单异或混淆 |
### --finger
开启数据指纹校验,可增加安全性,如果服务端开启指纹校验,则客户端也必须开启,开启会损耗一部分性能 开启数据指纹校验,可增加安全性,如果服务端开启指纹校验,则客户端也必须开启,开启会损耗一部分性能
注意:默认情况下服务端不会对中转的数据做校验,如果要对中转的数据做校验,则需要客户端、服务端都开启此参数 注意:默认情况下服务端不会对中转的数据做校验,如果要对中转的数据做校验,则需要客户端、服务端都开启此参数
### --punch `<punch>` ### --punch `<punch>`
取值ipv4/ipv6,选择只使用ipv4打洞或者只使用ipv6打洞,默认两者都会使用 取值ipv4/ipv6,选择只使用ipv4打洞或者只使用ipv6打洞,默认两者都会使用
### --ports `<port1,port2>` ### --ports `<port1,port2>`
指定本地监听的端口组,多个端口使用逗号分隔,多个端口可以分摊流量,增加并发、减缓流量限制,tcp会监听端口组的第一个端口,用于tcp直连 指定本地监听的端口组,多个端口使用逗号分隔,多个端口可以分摊流量,增加并发、减缓流量限制,tcp会监听端口组的第一个端口,用于tcp直连
- 例1:‘--ports 12345,12346,12347 表示udp监听12345、12346、12347这三个端口,tcp监听12345端口 - 例1:‘--ports 12345,12346,12347 表示udp监听12345、12346、12347这三个端口,tcp监听12345端口
- 例2:‘--ports 0,0’ 表示udp监听两个未使用的端口,tcp监听一个未使用的端口 - 例2:‘--ports 0,0’ 表示udp监听两个未使用的端口,tcp监听一个未使用的端口
### --cmd ### --cmd
开启交互式命令,开启后可以直接在窗口下输入命令,如需后台运行请勿开启 开启交互式命令,开启后可以直接在窗口下输入命令,如需后台运行请勿开启
### --first_latency ### --first_latency
优先使用低延迟通道,默认情况下优先使用p2p通道,某些情况下可能p2p比客户端中继延迟更高,可使用此参数进行优化传输 优先使用低延迟通道,默认情况下优先使用p2p通道,某些情况下可能p2p比客户端中继延迟更高,可使用此参数进行优化传输
### --no-proxy ### --no-proxy
关闭内置的ip代理,内置的代理较为简单,而且一般来说直接使用网卡NAT转发性能会更高, 关闭内置的ip代理,内置的代理较为简单,而且一般来说直接使用网卡NAT转发性能会更高,
有需要可以自行配置NAT转发,[可参考‘编译’小节中的NAT配置](https://github.com/lbl8603/vnt#%E7%BC%96%E8%AF%91) 有需要可以自行配置NAT转发,[可参考‘编译’小节中的NAT配置](https://github.com/lbl8603/vnt#%E7%BC%96%E8%AF%91)
### --dns `<223.5.5.5>` ### --dns `<223.5.5.5>`
设置域名解析服务器地址,可以设置多个。如果使用TXT记录的域名,则dns默认使用223.5.5.5和114.114.114.114,端口省略值为53 设置域名解析服务器地址,可以设置多个。如果使用TXT记录的域名,则dns默认使用223.5.5.5和114.114.114.114,端口省略值为53
当地址解析失败时,会依次尝试后面的dns,直到有A记录、AAAA记录(或TXT记录)的解析结果 当地址解析失败时,会依次尝试后面的dns,直到有A记录、AAAA记录(或TXT记录)的解析结果
### --mapping `<udp:0.0.0.0:80->10.26.0.10:80>` ### --mapping `<udp:0.0.0.0:80->10.26.0.10:80>`
端口映射,可以设置多个映射地址,例如 '--mapping udp:0.0.0.0:80->10.26.0.10:80 --mapping tcp:0.0.0.0:80->10.26.0.11:81' 端口映射,可以设置多个映射地址,例如 '--mapping udp:0.0.0.0:80->10.26.0.10:80 --mapping tcp:0.0.0.0:80->10.26.0.11:81'
表示将本地udp 80端口的数据转发到10.26.0.10:80,将本地tcp 80端口的数据转发到10.26.0.11:81,转发的目的地址可以使用域名+端口 表示将本地udp 80端口的数据转发到10.26.0.10:80,将本地tcp 80端口的数据转发到10.26.0.11:81,转发的目的地址可以使用域名+端口
### --compressor `<lz4>` ### --compressor `<lz4>`
启用压缩,默认仅支持lz4压缩,开启压缩后,如果数据包长度大于等于128,则会使用压缩,否则还是会按原数据发送 启用压缩,默认仅支持lz4压缩,开启压缩后,如果数据包长度大于等于128,则会使用压缩,否则还是会按原数据发送
也支持开启zstd压缩,但是需要自行编译,编译时加入参数--features zstd 也支持开启zstd压缩,但是需要自行编译,编译时加入参数--features zstd
@@ -104,8 +155,10 @@
如果宽度速度比较慢,可以考虑使用高级别的压缩 如果宽度速度比较慢,可以考虑使用高级别的压缩
### -f `<conf>` ### -f `<conf>`
指定配置文件 指定配置文件
配置文件采用yaml格式,可参考: 配置文件采用yaml格式,可参考:
```yaml ```yaml
# 全部参数 # 全部参数
tap: false #是否使用tap 仅在windows上支持使用tap tap: false #是否使用tap 仅在windows上支持使用tap
@@ -113,7 +166,7 @@ token: xxx #组网token
device_id: xxx #当前设备id device_id: xxx #当前设备id
name: windows 11 #当前设备名称 name: windows 11 #当前设备名称
server_address: ip:port #注册和中继服务器 server_address: ip:port #注册和中继服务器
stun_server: #stun服务器 stun_server: #stun服务器
- stun1.l.google.com:19302 - stun1.l.google.com:19302
- stun2.l.google.com:19302 - stun2.l.google.com:19302
in_ips: #代理ip入站 in_ips: #代理ip入站
@@ -130,7 +183,7 @@ parallel: 1 #任务并行度
cipher_model: aes_gcm #客户端加密算法 cipher_model: aes_gcm #客户端加密算法
finger: false #关闭数据指纹 finger: false #关闭数据指纹
punch_model: ipv4 #打洞模式,表示只使用ipv4地址打洞,默认会同时使用v6和v4 punch_model: ipv4 #打洞模式,表示只使用ipv4地址打洞,默认会同时使用v6和v4
ports: ports:
- 0 #使用随机端口,tcp监听此端口 - 0 #使用随机端口,tcp监听此端口
- 0 - 0
cmd: false #关闭控制台输入 cmd: false #关闭控制台输入
@@ -149,25 +202,41 @@ mapping:
``` ```
或者需要哪个配置就加哪个,当然token是必须的 或者需要哪个配置就加哪个,当然token是必须的
```yaml ```yaml
# 部分参数 # 部分参数
token: xxx #组网token token: xxx #组网token
``` ```
### --use-channel `<relay/p2p>` ### --use-channel `<relay/p2p>`
- relay:仅中继模式,会禁止打洞/p2p直连,只使用服务器转发 - relay:仅中继模式,会禁止打洞/p2p直连,只使用服务器转发
- p2p:仅直连模式,会禁止网络数据从服务器/客户端转发,只会使用服务器转发控制包 - p2p:仅直连模式,会禁止网络数据从服务器/客户端转发,只会使用服务器转发控制包
### --packet-loss `<0>` ### --packet-loss `<0>`
模拟丢包,取值0~1之间的小数,程序会按设定的概率主动丢包。在模拟弱网环境时会有帮助。 模拟丢包,取值0~1之间的小数,程序会按设定的概率主动丢包。在模拟弱网环境时会有帮助。
### --packet-delay `<0>` ### --packet-delay `<0>`
模拟延迟,整数,单位毫秒(ms),程序会按设定的值延迟发包,可用于模拟弱网 模拟延迟,整数,单位毫秒(ms),程序会按设定的值延迟发包,可用于模拟弱网
### --list ### --list
在后台运行时,查看其他设备列表 在后台运行时,查看其他设备列表
### --all ### --all
在后台运行时,查看其他设备完整信息 在后台运行时,查看其他设备完整信息
### --info ### --info
在后台运行时,查看当前设备信息 在后台运行时,查看当前设备信息
### --route
### --route
在后台运行时,查看数据转发路径 在后台运行时,查看数据转发路径
### --stop ### --stop
停止后台运行 停止后台运行
-10
View File
@@ -123,16 +123,6 @@ pub fn read_config(file_path: &str) -> anyhow::Result<(Config, bool)> {
if file_conf.password.is_some() && file_conf.cipher_model.is_none() { if file_conf.password.is_some() && file_conf.cipher_model.is_none() {
Err(anyhow!("cipher_model undefined"))? Err(anyhow!("cipher_model undefined"))?
} }
#[cfg(not(any(
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
)))]
{
CipherModel::None
}
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
CipherModel::AesGcm CipherModel::AesGcm
}; };
+51 -49
View File
@@ -233,19 +233,6 @@ fn main() {
let cipher_model = match matches.opt_get::<CipherModel>("model") { let cipher_model = match matches.opt_get::<CipherModel>("model") {
Ok(model) => { Ok(model) => {
#[cfg(not(any(
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
)))]
{
if password.is_some() && model.is_none() {
println!("Encryption not supported");
return;
}
}
#[cfg(not(any(feature = "aes_gcm", feature = "server_encrypt")))] #[cfg(not(any(feature = "aes_gcm", feature = "server_encrypt")))]
{ {
if password.is_some() && model.is_none() { if password.is_some() && model.is_none() {
@@ -367,6 +354,29 @@ fn main0(config: Config, _show_cmd: bool) {
} }
} }
let vnt_util = Vnt::new(config, callback::VntHandler {}).unwrap(); let vnt_util = Vnt::new(config, callback::VntHandler {}).unwrap();
#[cfg(any(target_os = "linux", target_os = "macos"))]
{
let vnt_c = vnt_util.clone();
let mut signals = signal_hook::iterator::Signals::new(&[
signal_hook::consts::SIGINT,
signal_hook::consts::SIGTERM,
])
.unwrap();
let handle = signals.handle();
std::thread::spawn(move || {
for sig in signals.forever() {
match sig {
signal_hook::consts::SIGINT | signal_hook::consts::SIGTERM => {
println!("Received SIGINT, {}", sig);
vnt_c.stop();
handle.close();
break;
}
_ => {}
}
}
});
}
#[cfg(feature = "command")] #[cfg(feature = "command")]
{ {
let vnt_c = vnt_util.clone(); let vnt_c = vnt_util.clone();
@@ -400,6 +410,7 @@ fn main0(config: Config, _show_cmd: bool) {
vnt_util.wait() vnt_util.wait()
} }
#[cfg(feature = "command")] #[cfg(feature = "command")]
fn command(cmd: &str, vnt: &Vnt) -> bool { fn command(cmd: &str, vnt: &Vnt) -> bool {
if cmd.is_empty() { if cmd.is_empty() {
@@ -452,33 +463,8 @@ fn print_usage(program: &str, _opts: Options) {
println!(" -i <in-ip> 配置点对网(IP代理)时使用,-i 192.168.0.0/24,10.26.0.3表示允许接收网段192.168.0.0/24的数据"); println!(" -i <in-ip> 配置点对网(IP代理)时使用,-i 192.168.0.0/24,10.26.0.3表示允许接收网段192.168.0.0/24的数据");
println!(" 并转发到10.26.0.3,可指定多个网段"); println!(" 并转发到10.26.0.3,可指定多个网段");
println!(" -o <out-ip> 配置点对网时使用,-o 192.168.0.0/24表示允许将数据转发到192.168.0.0/24,可指定多个网段"); println!(" -o <out-ip> 配置点对网时使用,-o 192.168.0.0/24表示允许将数据转发到192.168.0.0/24,可指定多个网段");
#[cfg(not(any(
feature = "aes_gcm", println!(" -w <password> 使用该密码生成的密钥对客户端数据进行加密,并且服务端无法解密,使用相同密码的客户端才能通信");
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
)))]
let enums = String::new();
#[cfg(any(
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
let mut enums = String::new();
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
enums.push_str("/aes_gcm");
#[cfg(feature = "aes_cbc")]
enums.push_str("/aes_cbc");
#[cfg(feature = "aes_ecb")]
enums.push_str("/aes_ecb");
#[cfg(feature = "sm4_cbc")]
enums.push_str("/sm4_cbc");
if !enums.is_empty() {
println!(" -w <password> 使用该密码生成的密钥对客户端数据进行加密,并且服务端无法解密,使用相同密码的客户端才能通信");
}
#[cfg(feature = "server_encrypt")] #[cfg(feature = "server_encrypt")]
println!(" -W 加密当前客户端和服务端通信的数据,请留意服务端指纹是否正确"); println!(" -W 加密当前客户端和服务端通信的数据,请留意服务端指纹是否正确");
println!(" -u <mtu> 自定义mtu(不加密默认为1450,加密默认为1410)"); println!(" -u <mtu> 自定义mtu(不加密默认为1450,加密默认为1410)");
@@ -488,15 +474,31 @@ fn print_usage(program: &str, _opts: Options) {
println!(" --tcp 和服务端使用tcp通信,默认使用udp,遇到udp qos时可指定使用tcp"); println!(" --tcp 和服务端使用tcp通信,默认使用udp,遇到udp qos时可指定使用tcp");
println!(" --ip <ip> 指定虚拟ip,指定的ip不能和其他设备重复,必须有效并且在服务端所属网段下,默认情况由服务端分配"); println!(" --ip <ip> 指定虚拟ip,指定的ip不能和其他设备重复,必须有效并且在服务端所属网段下,默认情况由服务端分配");
println!(" --par <parallel> 任务并行度(必须为正整数),默认值为1"); println!(" --par <parallel> 任务并行度(必须为正整数),默认值为1");
if !enums.is_empty() { let mut enums = String::new();
println!( #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
" --model <model> 加密模式(默认aes_gcm),可选值{}", enums.push_str("/aes_gcm");
&enums[1..] #[cfg(feature = "chacha20_poly1305")]
); enums.push_str("/chacha20_poly1305/chacha20");
} #[cfg(feature = "aes_cbc")]
if !enums.is_empty() { enums.push_str("/aes_cbc");
println!(" --finger 增加数据指纹校验,可增加安全性,如果服务端开启指纹校验,则客户端也必须开启"); #[cfg(feature = "aes_ecb")]
} enums.push_str("/aes_ecb");
#[cfg(feature = "sm4_cbc")]
enums.push_str("/sm4_cbc");
enums.push_str("/xor");
println!(
" --model <model> 加密模式(默认aes_gcm),可选值{}",
&enums[1..]
);
#[cfg(any(
feature = "aes_gcm",
feature = "chacha20_poly1305",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
println!(" --finger 增加数据指纹校验,可增加安全性,如果服务端开启指纹校验,则客户端也必须开启");
println!(" --punch <punch> 取值ipv4/ipv6/all,ipv4表示仅使用ipv4打洞"); println!(" --punch <punch> 取值ipv4/ipv6/all,ipv4表示仅使用ipv4打洞");
println!(" --ports <port,port> 取值0~65535,指定本地监听的一组端口,默认监听两个随机端口,使用过多端口会增加网络负担"); println!(" --ports <port,port> 取值0~65535,指定本地监听的一组端口,默认监听两个随机端口,使用过多端口会增加网络负担");
#[cfg(feature = "command")] #[cfg(feature = "command")]
+6 -1
View File
@@ -23,6 +23,8 @@ aes-gcm = { version = "0.10.2", optional = true }
ring = { version = "0.17.0", optional = true } ring = { version = "0.17.0", optional = true }
cbc = { version = "0.1.2", optional = true } cbc = { version = "0.1.2", optional = true }
ecb = { version = "0.1.2", optional = true } ecb = { version = "0.1.2", optional = true }
chacha20poly1305 = { version = "0.10.1", optional = true }
chacha20 = { version = "0.9.1", optional = true }
aes = "0.8.3" aes = "0.8.3"
stun-format = { version = "1.0.1", features = ["fmt", "rfc3489"] } stun-format = { version = "1.0.1", features = ["fmt", "rfc3489"] }
rsa = { version = "0.9.2", features = [], optional = true } rsa = { version = "0.9.2", features = [], optional = true }
@@ -39,6 +41,7 @@ tokio = { version = "1.37.0", features = ["full"], optional = true }
lz4_flex = { version = "0.11", default-features = false, optional = true } lz4_flex = { version = "0.11", default-features = false, optional = true }
zstd = { version = "0.13.1", optional = true } zstd = { version = "0.13.1", optional = true }
[target.'cfg(target_os = "windows")'.dependencies] [target.'cfg(target_os = "windows")'.dependencies]
libloading = "0.8.0" libloading = "0.8.0"
@@ -46,9 +49,10 @@ libloading = "0.8.0"
[build-dependencies] [build-dependencies]
protobuf-codegen = "3.2.0" protobuf-codegen = "3.2.0"
protoc-bin-vendored = "3.0.0" protoc-bin-vendored = "3.0.0"
cfg_aliases = "0.2.1"
[features] [features]
default = ["server_encrypt", "aes_gcm", "aes_cbc", "aes_ecb", "sm4_cbc", "ip_proxy", "port_mapping", "lz4_compress","zstd_compress"] default = ["server_encrypt", "aes_gcm", "aes_cbc", "aes_ecb", "sm4_cbc", "chacha20_poly1305", "ip_proxy", "port_mapping", "lz4_compress", "zstd_compress"]
openssl = ["openssl-sys"] openssl = ["openssl-sys"]
# 从源码编译 # 从源码编译
openssl-vendored = ["openssl-sys/vendored"] openssl-vendored = ["openssl-sys/vendored"]
@@ -57,6 +61,7 @@ aes_cbc = ["cbc"]
aes_ecb = ["ecb"] aes_ecb = ["ecb"]
sm4_cbc = ["libsm"] sm4_cbc = ["libsm"]
aes_gcm = ["aes-gcm"] aes_gcm = ["aes-gcm"]
chacha20_poly1305 = ["chacha20poly1305", "chacha20"]
server_encrypt = ["aes-gcm", "rsa", "spki"] server_encrypt = ["aes-gcm", "rsa", "spki"]
ip_proxy = ["tokio"] ip_proxy = ["tokio"]
port_mapping = ["tokio"] port_mapping = ["tokio"]
+13
View File
@@ -1,4 +1,17 @@
use cfg_aliases::cfg_aliases;
fn main() { fn main() {
cfg_aliases! {
cipher: {
any(feature = "aes_gcm",
feature = "chacha20_poly1305",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
)},
}
std::fs::create_dir_all("src/proto").unwrap(); std::fs::create_dir_all("src/proto").unwrap();
protobuf_codegen::Codegen::new() protobuf_codegen::Codegen::new()
.pure() .pure()
+4 -4
View File
@@ -31,7 +31,7 @@ pub fn tcp_listen<H>(
stop_manager: StopManager, stop_manager: StopManager,
recv_handler: H, recv_handler: H,
context: ChannelContext, context: ChannelContext,
) -> io::Result<AcceptSocketSender<(TcpStream, SocketAddr, Option<Vec<u8>>)>> ) -> anyhow::Result<AcceptSocketSender<(TcpStream, SocketAddr, Option<Vec<u8>>)>>
where where
H: RecvChannelHandler, H: RecvChannelHandler,
{ {
@@ -75,7 +75,7 @@ fn tcp_listen0<H>(
accept_tcp_receiver: Receiver<(TcpStream, SocketAddr, Option<Vec<u8>>)>, accept_tcp_receiver: Receiver<(TcpStream, SocketAddr, Option<Vec<u8>>)>,
mut recv_handler: H, mut recv_handler: H,
context: ChannelContext, context: ChannelContext,
) -> io::Result<()> ) -> anyhow::Result<()>
where where
H: RecvChannelHandler, H: RecvChannelHandler,
{ {
@@ -109,7 +109,7 @@ where
if e.kind() == io::ErrorKind::WouldBlock { if e.kind() == io::ErrorKind::WouldBlock {
break; break;
} }
return Err(e); return Err(e)?;
} }
} }
}, },
@@ -164,7 +164,7 @@ fn init_writable_handler(
receiver: Receiver<(TcpStream, Token, SocketAddr, Option<Vec<u8>>)>, receiver: Receiver<(TcpStream, Token, SocketAddr, Option<Vec<u8>>)>,
stop_manager: StopManager, stop_manager: StopManager,
context: ChannelContext, context: ChannelContext,
) -> io::Result<WritableNotify> { ) -> anyhow::Result<WritableNotify> {
let poll = Poll::new()?; let poll = Poll::new()?;
let writable_notify = WritableNotify::new(Waker::new(poll.registry(), NOTIFY)?); let writable_notify = WritableNotify::new(Waker::new(poll.registry(), NOTIFY)?);
let worker = { let worker = {
+3 -3
View File
@@ -17,7 +17,7 @@ pub fn udp_listen<H>(
stop_manager: StopManager, stop_manager: StopManager,
recv_handler: H, recv_handler: H,
context: ChannelContext, context: ChannelContext,
) -> io::Result<AcceptSocketSender<Option<Vec<UdpSocket>>>> ) -> anyhow::Result<AcceptSocketSender<Option<Vec<UdpSocket>>>>
where where
H: RecvChannelHandler, H: RecvChannelHandler,
{ {
@@ -31,7 +31,7 @@ fn sub_udp_listen<H>(
stop_manager: StopManager, stop_manager: StopManager,
recv_handler: H, recv_handler: H,
context: ChannelContext, context: ChannelContext,
) -> io::Result<AcceptSocketSender<Option<Vec<UdpSocket>>>> ) -> anyhow::Result<AcceptSocketSender<Option<Vec<UdpSocket>>>>
where where
H: RecvChannelHandler, H: RecvChannelHandler,
{ {
@@ -208,7 +208,7 @@ fn main_udp_listen<H>(
stop_manager: StopManager, stop_manager: StopManager,
recv_handler: H, recv_handler: H,
context: ChannelContext, context: ChannelContext,
) -> io::Result<()> ) -> anyhow::Result<()>
where where
H: RecvChannelHandler, H: RecvChannelHandler,
{ {
+2
View File
@@ -0,0 +1,2 @@
mod rs_aes_cbc;
pub use rs_aes_cbc::*;
@@ -1,6 +1,5 @@
use std::io;
use aes::cipher::{block_padding::Pkcs7, BlockDecryptMut, BlockEncryptMut, KeyIvInit}; use aes::cipher::{block_padding::Pkcs7, BlockDecryptMut, BlockEncryptMut, KeyIvInit};
use anyhow::anyhow;
use rand::RngCore; use rand::RngCore;
use crate::cipher::Finger; use crate::cipher::Finger;
@@ -50,14 +49,14 @@ impl AesCbcCipher {
pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
if !net_packet.is_encrypt() { if !net_packet.is_encrypt() {
//未加密的数据直接丢弃 //未加密的数据直接丢弃
return Err(io::Error::new(io::ErrorKind::Other, "not encrypt")); return Err(anyhow!("not encrypt"));
} }
if net_packet.payload().len() < 16 { if net_packet.payload().len() < 16 {
log::error!("数据异常,长度{}小于{}", net_packet.payload().len(), 16); log::error!("数据异常,长度{}小于{}", net_packet.payload().len(), 16);
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("aes_cbc data err"));
} }
let mut iv = [0; 16]; let mut iv = [0; 16];
iv[0..4].copy_from_slice(&net_packet.source().octets()); iv[0..4].copy_from_slice(&net_packet.source().octets());
@@ -75,7 +74,7 @@ impl AesCbcCipher {
if let Some(finger) = &self.finger { if let Some(finger) = &self.finger {
let finger = finger.calculate_finger(&iv[..12], secret_body.en_body()); let finger = finger.calculate_finger(&iv[..12], secret_body.en_body());
if &finger != secret_body.finger() { if &finger != secret_body.finger() {
return Err(io::Error::new(io::ErrorKind::Other, "finger err")); return Err(anyhow!("aes_cbc finger err"));
} }
} }
let rs = match &self.cipher { let rs = match &self.cipher {
@@ -92,10 +91,7 @@ impl AesCbcCipher {
net_packet.set_data_len(HEAD_LEN + len - 4)?; net_packet.set_data_len(HEAD_LEN + len - 4)?;
Ok(()) Ok(())
} }
Err(e) => Err(io::Error::new( Err(e) => Err(anyhow!("aes_cbc 解密失败:{}", e)),
io::ErrorKind::Other,
format!("解密失败:{}", e),
)),
} }
} }
/// net_packet 必须预留足够长度 /// net_packet 必须预留足够长度
@@ -103,7 +99,7 @@ impl AesCbcCipher {
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let data_len = net_packet.data_len(); let data_len = net_packet.data_len();
let mut iv = [0; 16]; let mut iv = [0; 16];
iv[0..4].copy_from_slice(&net_packet.source().octets()); iv[0..4].copy_from_slice(&net_packet.source().octets());
@@ -146,10 +142,7 @@ impl AesCbcCipher {
net_packet.set_encrypt_flag(true); net_packet.set_encrypt_flag(true);
Ok(()) Ok(())
} }
Err(e) => Err(io::Error::new( Err(e) => Err(anyhow!("aes_cbc 加密失败:{}", e)),
io::ErrorKind::Other,
format!("加密失败:{}", e),
)),
}; };
} }
} }
+9
View File
@@ -0,0 +1,9 @@
#[cfg(not(any(feature = "openssl-vendored", feature = "openssl")))]
mod rs_aes_ecb;
#[cfg(not(any(feature = "openssl-vendored", feature = "openssl")))]
pub use rs_aes_ecb::*;
#[cfg(any(feature = "openssl-vendored", feature = "openssl"))]
mod openssl_aes_ecb;
#[cfg(any(feature = "openssl-vendored", feature = "openssl"))]
pub use openssl_aes_ecb::*;
@@ -1,8 +1,11 @@
use crate::cipher::Finger; use std::ptr;
use crate::protocol::{NetPacket, HEAD_LEN};
use anyhow::anyhow;
use libc::c_int; use libc::c_int;
use openssl_sys::EVP_CIPHER_CTX; use openssl_sys::EVP_CIPHER_CTX;
use std::{io, ptr};
use crate::cipher::Finger;
use crate::protocol::{NetPacket, HEAD_LEN};
pub struct AesEcbCipher { pub struct AesEcbCipher {
key: Vec<u8>, key: Vec<u8>,
@@ -100,10 +103,10 @@ impl AesEcbCipher {
pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
if !net_packet.is_encrypt() { if !net_packet.is_encrypt() {
//未加密的数据直接丢弃 //未加密的数据直接丢弃
return Err(io::Error::new(io::ErrorKind::Other, "not encrypt")); return Err(anyhow!("not encrypt"));
} }
if let Some(finger) = &self.finger { if let Some(finger) = &self.finger {
@@ -116,18 +119,18 @@ impl AesEcbCipher {
nonce_raw[11] = net_packet.source_ttl(); nonce_raw[11] = net_packet.source_ttl();
let len = net_packet.payload().len(); let len = net_packet.payload().len();
if len < 12 { if len < 12 {
return Err(io::Error::new(io::ErrorKind::Other, "data len err")); return Err(anyhow!("data len err"));
} }
let secret_body = &net_packet.payload()[..len - 12]; let secret_body = &net_packet.payload()[..len - 12];
let finger = finger.calculate_finger(&nonce_raw, secret_body); let finger = finger.calculate_finger(&nonce_raw, secret_body);
if &finger != &net_packet.payload()[len - 12..] { if &finger != &net_packet.payload()[len - 12..] {
return Err(io::Error::new(io::ErrorKind::Other, "finger err")); return Err(anyhow!("finger err"));
} }
net_packet.set_data_len(net_packet.data_len() - finger.len())?; net_packet.set_data_len(net_packet.data_len() - finger.len())?;
} }
if net_packet.payload().len() < 16 { if net_packet.payload().len() < 16 {
log::error!("数据异常,长度{}小于{}", net_packet.payload().len(), 16); log::error!("数据异常,长度{}小于{}", net_packet.payload().len(), 16);
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
let input = net_packet.payload(); let input = net_packet.payload();
let mut out = [0u8; 1024 * 5]; let mut out = [0u8; 1024 * 5];
@@ -147,22 +150,22 @@ impl AesEcbCipher {
//校验头部 //校验头部
let src_net_packet = NetPacket::new(text)?; let src_net_packet = NetPacket::new(text)?;
if src_net_packet.source() != net_packet.source() { if src_net_packet.source() != net_packet.source() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.destination() != net_packet.destination() { if src_net_packet.destination() != net_packet.destination() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.protocol() != net_packet.protocol() { if src_net_packet.protocol() != net_packet.protocol() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.transport_protocol() != net_packet.transport_protocol() { if src_net_packet.transport_protocol() != net_packet.transport_protocol() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.is_gateway() != net_packet.is_gateway() { if src_net_packet.is_gateway() != net_packet.is_gateway() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.source_ttl() != net_packet.source_ttl() { if src_net_packet.source_ttl() != net_packet.source_ttl() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
} }
net_packet.set_encrypt_flag(false); net_packet.set_encrypt_flag(false);
@@ -175,7 +178,7 @@ impl AesEcbCipher {
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let input = net_packet.buffer(); let input = net_packet.buffer();
let mut out = [0u8; 1024 * 5]; let mut out = [0u8; 1024 * 5];
let mut out_len = 0; let mut out_len = 0;
@@ -191,7 +194,7 @@ impl AesEcbCipher {
} }
let out_len = out_len as usize; let out_len = out_len as usize;
if out_len == 0 { if out_len == 0 {
return Err(io::Error::new(io::ErrorKind::Other, "ciphertext len err")); return Err(anyhow!("ciphertext len err"));
} }
//密文 //密文
let ciphertext = &out[..out_len]; let ciphertext = &out[..out_len];
@@ -1,7 +1,8 @@
use aes::cipher::{block_padding::Pkcs7, BlockDecryptMut, BlockEncryptMut, KeyInit};
use anyhow::anyhow;
use crate::cipher::Finger; use crate::cipher::Finger;
use crate::protocol::{NetPacket, HEAD_LEN}; use crate::protocol::{NetPacket, HEAD_LEN};
use aes::cipher::{block_padding::Pkcs7, BlockDecryptMut, BlockEncryptMut, KeyInit};
use std::io;
type Aes128EcbEnc = ecb::Encryptor<aes::Aes128>; type Aes128EcbEnc = ecb::Encryptor<aes::Aes128>;
type Aes128EcbDec = ecb::Decryptor<aes::Aes128>; type Aes128EcbDec = ecb::Decryptor<aes::Aes128>;
@@ -46,10 +47,10 @@ impl AesEcbCipher {
pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
if !net_packet.is_encrypt() { if !net_packet.is_encrypt() {
//未加密的数据直接丢弃 //未加密的数据直接丢弃
return Err(io::Error::new(io::ErrorKind::Other, "not encrypt")); return Err(anyhow!("not encrypt"));
} }
if let Some(finger) = &self.finger { if let Some(finger) = &self.finger {
@@ -62,18 +63,18 @@ impl AesEcbCipher {
nonce_raw[11] = net_packet.source_ttl(); nonce_raw[11] = net_packet.source_ttl();
let len = net_packet.payload().len(); let len = net_packet.payload().len();
if len < 12 { if len < 12 {
return Err(io::Error::new(io::ErrorKind::Other, "payload len <12")); return Err(anyhow!("payload len <12"));
} }
let secret_body = &net_packet.payload()[..len - 12]; let secret_body = &net_packet.payload()[..len - 12];
let finger = finger.calculate_finger(&nonce_raw, secret_body); let finger = finger.calculate_finger(&nonce_raw, secret_body);
if &finger != &net_packet.payload()[len - 12..] { if &finger != &net_packet.payload()[len - 12..] {
return Err(io::Error::new(io::ErrorKind::Other, "finger err")); return Err(anyhow!("finger err"));
} }
net_packet.set_data_len(net_packet.data_len() - finger.len())?; net_packet.set_data_len(net_packet.data_len() - finger.len())?;
} }
if net_packet.payload().len() < 16 { if net_packet.payload().len() < 16 {
log::error!("数据异常,长度{}小于{}", net_packet.payload().len(), 16); log::error!("数据异常,长度{}小于{}", net_packet.payload().len(), 16);
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
let mut out = [0u8; 1024 * 5]; let mut out = [0u8; 1024 * 5];
let rs = match self.key { let rs = match self.key {
@@ -87,32 +88,29 @@ impl AesEcbCipher {
//校验头部 //校验头部
let src_net_packet = NetPacket::new(buf)?; let src_net_packet = NetPacket::new(buf)?;
if src_net_packet.source() != net_packet.source() { if src_net_packet.source() != net_packet.source() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.destination() != net_packet.destination() { if src_net_packet.destination() != net_packet.destination() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.protocol() != net_packet.protocol() { if src_net_packet.protocol() != net_packet.protocol() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.transport_protocol() != net_packet.transport_protocol() { if src_net_packet.transport_protocol() != net_packet.transport_protocol() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.is_gateway() != net_packet.is_gateway() { if src_net_packet.is_gateway() != net_packet.is_gateway() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.source_ttl() != net_packet.source_ttl() { if src_net_packet.source_ttl() != net_packet.source_ttl() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
net_packet.set_data_len(buf.len())?; net_packet.set_data_len(buf.len())?;
net_packet.set_payload(src_net_packet.payload())?; net_packet.set_payload(src_net_packet.payload())?;
net_packet.set_encrypt_flag(false); net_packet.set_encrypt_flag(false);
Ok(()) Ok(())
} }
Err(e) => Err(io::Error::new( Err(e) => Err(anyhow!("aes_ecb解密失败:{}", e)),
io::ErrorKind::Other,
format!("aes_ecb解密失败:{}", e),
)),
} }
} }
/// net_packet 必须预留足够长度 /// net_packet 必须预留足够长度
@@ -120,7 +118,7 @@ impl AesEcbCipher {
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let mut out = [0u8; 1024 * 5]; let mut out = [0u8; 1024 * 5];
let rs = match self.key { let rs = match self.key {
AesEcbEnum::AES128ECB(key) => Aes128EcbEnc::new(&key.into()) AesEcbEnum::AES128ECB(key) => Aes128EcbEnc::new(&key.into())
@@ -152,10 +150,7 @@ impl AesEcbCipher {
} }
Ok(()) Ok(())
} }
Err(e) => Err(io::Error::new( Err(e) => Err(anyhow!("aes_ecb加密失败:{}", e)),
io::ErrorKind::Other,
format!("aes_ecb加密失败:{}", e),
)),
}; };
} }
} }
@@ -1,8 +1,7 @@
use std::io;
use aes_gcm::aead::consts::{U12, U16}; use aes_gcm::aead::consts::{U12, U16};
use aes_gcm::aead::generic_array::GenericArray; use aes_gcm::aead::generic_array::GenericArray;
use aes_gcm::{AeadInPlace, Aes128Gcm, Aes256Gcm, Key, KeyInit, Nonce, Tag}; use aes_gcm::{AeadInPlace, Aes128Gcm, Aes256Gcm, Key, KeyInit, Nonce, Tag};
use anyhow::anyhow;
use rand::RngCore; use rand::RngCore;
use crate::cipher::finger::Finger; use crate::cipher::finger::Finger;
@@ -39,14 +38,14 @@ impl AesGcmCipher {
pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
if !net_packet.is_encrypt() { if !net_packet.is_encrypt() {
//未加密的数据直接丢弃 //未加密的数据直接丢弃
return Err(io::Error::new(io::ErrorKind::Other, "not encrypt")); return Err(anyhow!("not encrypt"));
} }
if net_packet.payload().len() < AES_GCM_ENCRYPTION_RESERVED { if net_packet.payload().len() < AES_GCM_ENCRYPTION_RESERVED {
log::error!("数据异常,长度小于{}", AES_GCM_ENCRYPTION_RESERVED); log::error!("数据异常,长度小于{}", AES_GCM_ENCRYPTION_RESERVED);
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
let mut nonce_raw = [0; 12]; let mut nonce_raw = [0; 12];
nonce_raw[0..4].copy_from_slice(&net_packet.source().octets()); nonce_raw[0..4].copy_from_slice(&net_packet.source().octets());
@@ -62,7 +61,7 @@ impl AesGcmCipher {
if let Some(finger) = &self.finger { if let Some(finger) = &self.finger {
let finger = finger.calculate_finger(&nonce_raw, secret_body.en_body()); let finger = finger.calculate_finger(&nonce_raw, secret_body.en_body());
if &finger != secret_body.finger() { if &finger != secret_body.finger() {
return Err(io::Error::new(io::ErrorKind::Other, "finger err")); return Err(anyhow!("finger err"));
} }
} }
let tag: GenericArray<u8, U16> = Tag::clone_from_slice(tag); let tag: GenericArray<u8, U16> = Tag::clone_from_slice(tag);
@@ -75,10 +74,7 @@ impl AesGcmCipher {
} }
}; };
if let Err(e) = rs { if let Err(e) = rs {
return Err(io::Error::new( return Err(anyhow!("解密失败:{}", e));
io::ErrorKind::Other,
format!("解密失败:{}", e),
));
} }
net_packet.set_encrypt_flag(false); net_packet.set_encrypt_flag(false);
net_packet.set_data_len(net_packet.data_len() - AES_GCM_ENCRYPTION_RESERVED)?; net_packet.set_data_len(net_packet.data_len() - AES_GCM_ENCRYPTION_RESERVED)?;
@@ -89,9 +85,9 @@ impl AesGcmCipher {
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
if net_packet.reserve() < AES_GCM_ENCRYPTION_RESERVED { if net_packet.reserve() < AES_GCM_ENCRYPTION_RESERVED {
return Err(io::Error::new(io::ErrorKind::Other, "too short")); return Err(anyhow!("too short"));
} }
let mut nonce_raw = [0; 12]; let mut nonce_raw = [0; 12];
nonce_raw[0..4].copy_from_slice(&net_packet.source().octets()); nonce_raw[0..4].copy_from_slice(&net_packet.source().octets());
@@ -123,10 +119,7 @@ impl AesGcmCipher {
net_packet.set_encrypt_flag(true); net_packet.set_encrypt_flag(true);
Ok(()) Ok(())
} }
Err(e) => Err(io::Error::new( Err(e) => Err(anyhow!("加密失败:{}", e)),
io::ErrorKind::Other,
format!("加密失败:{}", e),
)),
}; };
} }
} }
+9
View File
@@ -0,0 +1,9 @@
#[cfg(feature = "ring-cipher")]
mod ring_aes_gcm_cipher;
#[cfg(feature = "ring-cipher")]
pub use ring_aes_gcm_cipher::*;
#[cfg(not(feature = "ring-cipher"))]
mod aes_gcm_cipher;
#[cfg(not(feature = "ring-cipher"))]
pub use aes_gcm_cipher::*;
@@ -1,9 +1,9 @@
use crate::cipher::Finger; use anyhow::anyhow;
use rand::RngCore; use rand::RngCore;
use ring::aead; use ring::aead;
use ring::aead::{LessSafeKey, UnboundKey}; use ring::aead::{LessSafeKey, UnboundKey};
use std::io;
use crate::cipher::Finger;
use crate::protocol::body::{SecretBody, AES_GCM_ENCRYPTION_RESERVED}; use crate::protocol::body::{SecretBody, AES_GCM_ENCRYPTION_RESERVED};
use crate::protocol::NetPacket; use crate::protocol::NetPacket;
@@ -53,14 +53,14 @@ impl AesGcmCipher {
pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
if !net_packet.is_encrypt() { if !net_packet.is_encrypt() {
//未加密的数据直接丢弃 //未加密的数据直接丢弃
return Err(io::Error::new(io::ErrorKind::Other, "not encrypt")); return Err(anyhow!("not encrypt"));
} }
if net_packet.payload().len() < AES_GCM_ENCRYPTION_RESERVED { if net_packet.payload().len() < AES_GCM_ENCRYPTION_RESERVED {
log::error!("数据异常,长度小于{}", AES_GCM_ENCRYPTION_RESERVED); log::error!("数据异常,长度小于{}", AES_GCM_ENCRYPTION_RESERVED);
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
let mut nonce_raw = [0; 12]; let mut nonce_raw = [0; 12];
nonce_raw[0..4].copy_from_slice(&net_packet.source().octets()); nonce_raw[0..4].copy_from_slice(&net_packet.source().octets());
@@ -74,7 +74,7 @@ impl AesGcmCipher {
if let Some(finger) = &self.finger { if let Some(finger) = &self.finger {
let finger = finger.calculate_finger(&nonce_raw, secret_body.en_body()); let finger = finger.calculate_finger(&nonce_raw, secret_body.en_body());
if &finger != secret_body.finger() { if &finger != secret_body.finger() {
return Err(io::Error::new(io::ErrorKind::Other, "ring aes finger err")); return Err(anyhow!("ring aes finger err"));
} }
} }
@@ -87,10 +87,7 @@ impl AesGcmCipher {
} }
}; };
if let Err(e) = rs { if let Err(e) = rs {
return Err(io::Error::new( return Err(anyhow!("解密失败:{}", e));
io::ErrorKind::Other,
format!("解密失败:{}", e),
));
} }
net_packet.set_encrypt_flag(false); net_packet.set_encrypt_flag(false);
net_packet.set_data_len(net_packet.data_len() - AES_GCM_ENCRYPTION_RESERVED)?; net_packet.set_data_len(net_packet.data_len() - AES_GCM_ENCRYPTION_RESERVED)?;
@@ -102,7 +99,7 @@ impl AesGcmCipher {
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let mut nonce_raw = [0; 12]; let mut nonce_raw = [0; 12];
nonce_raw[0..4].copy_from_slice(&net_packet.source().octets()); nonce_raw[0..4].copy_from_slice(&net_packet.source().octets());
nonce_raw[4..8].copy_from_slice(&net_packet.destination().octets()); nonce_raw[4..8].copy_from_slice(&net_packet.destination().octets());
@@ -128,10 +125,7 @@ impl AesGcmCipher {
Ok(tag) => { Ok(tag) => {
let tag = tag.as_ref(); let tag = tag.as_ref();
if tag.len() != 16 { if tag.len() != 16 {
return Err(io::Error::new( return Err(anyhow!("加密tag长度错误:{}", tag.len()));
io::ErrorKind::Other,
format!("加密tag长度错误:{}", tag.len()),
));
} }
secret_body.set_tag(tag)?; secret_body.set_tag(tag)?;
if let Some(finger) = &self.finger { if let Some(finger) = &self.finger {
@@ -141,10 +135,7 @@ impl AesGcmCipher {
net_packet.set_encrypt_flag(true); net_packet.set_encrypt_flag(true);
Ok(()) Ok(())
} }
Err(e) => Err(io::Error::new( Err(e) => Err(anyhow!("加密失败:{}", e)),
io::ErrorKind::Other,
format!("加密失败:{}", e),
)),
}; };
} }
} }
+2
View File
@@ -0,0 +1,2 @@
mod rs_chacha20;
pub use rs_chacha20::*;
+114
View File
@@ -0,0 +1,114 @@
use aes::cipher::Iv;
use anyhow::anyhow;
use chacha20::cipher::{Key, KeyIvInit, StreamCipher};
use chacha20::ChaCha20;
use crate::cipher::Finger;
use crate::protocol::body::ChaCah20SecretBody;
use crate::protocol::NetPacket;
#[derive(Clone)]
pub struct ChaCha20Cipher {
key: [u8; 32],
pub(crate) finger: Option<Finger>,
}
impl ChaCha20Cipher {
pub fn new_256(key: [u8; 32], finger: Option<Finger>) -> Self {
Self { key, finger }
}
}
impl ChaCha20Cipher {
pub fn key(&self) -> &[u8] {
&self.key
}
}
impl ChaCha20Cipher {
pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self,
net_packet: &mut NetPacket<B>,
) -> anyhow::Result<()> {
if !net_packet.is_encrypt() {
//未加密的数据直接丢弃
return Err(anyhow!("not encrypt"));
}
let mut iv = [0; 12];
iv[0..4].copy_from_slice(&net_packet.source().octets());
iv[4..8].copy_from_slice(&net_packet.destination().octets());
iv[8] = net_packet.protocol().into();
iv[9] = net_packet.transport_protocol();
iv[10] = net_packet.is_gateway() as u8;
iv[11] = net_packet.source_ttl();
let mut secret_body =
ChaCah20SecretBody::new(net_packet.payload_mut(), self.finger.is_some())?;
if let Some(finger) = &self.finger {
let finger = finger.calculate_finger(&iv[..12], secret_body.en_body());
if &finger != secret_body.finger() {
return Err(anyhow!("ChaCha20 finger err"));
}
}
ChaCha20::new(
Key::<ChaCha20>::from_slice(&self.key),
Iv::<ChaCha20>::from_slice(&iv),
)
.apply_keystream(secret_body.en_body_mut());
let len = secret_body.en_body().len();
net_packet.set_encrypt_flag(false);
net_packet.set_payload_len(len)?;
Ok(())
}
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self,
net_packet: &mut NetPacket<B>,
) -> anyhow::Result<()> {
let data_len = net_packet.data_len();
let mut iv = [0; 12];
iv[0..4].copy_from_slice(&net_packet.source().octets());
iv[4..8].copy_from_slice(&net_packet.destination().octets());
iv[8] = net_packet.protocol().into();
iv[9] = net_packet.transport_protocol();
iv[10] = net_packet.is_gateway() as u8;
iv[11] = net_packet.source_ttl();
if let Some(_) = &self.finger {
net_packet.set_data_len(data_len + 12)?;
}
let mut secret_body =
ChaCah20SecretBody::new(net_packet.payload_mut(), self.finger.is_some())?;
ChaCha20::new(
Key::<ChaCha20>::from_slice(&self.key),
Iv::<ChaCha20>::from_slice(&iv),
)
.apply_keystream(secret_body.en_body_mut());
if let Some(finger) = &self.finger {
let finger = finger.calculate_finger(&iv[..12], secret_body.en_body_mut());
let mut secret_body = ChaCah20SecretBody::new(net_packet.payload_mut(), true)?;
secret_body.set_finger(&finger)?;
}
net_packet.set_encrypt_flag(true);
Ok(())
}
}
#[test]
fn test_chacha20() {
let d = ChaCha20Cipher::new_256([0; 32], Some(Finger::new("123")));
let mut p =
NetPacket::new_encrypt([0; 13 + crate::protocol::body::ENCRYPTION_RESERVED]).unwrap();
let src = p.buffer().to_vec();
d.encrypt_ipv4(&mut p).unwrap();
d.decrypt_ipv4(&mut p).unwrap();
assert_eq!(p.buffer(), &src);
let d = ChaCha20Cipher::new_256([0; 32], None);
let mut p =
NetPacket::new_encrypt([0; 13 + crate::protocol::body::ENCRYPTION_RESERVED]).unwrap();
let src = p.buffer().to_vec();
d.encrypt_ipv4(&mut p).unwrap();
d.decrypt_ipv4(&mut p).unwrap();
assert_eq!(p.buffer(), &src);
}
+9
View File
@@ -0,0 +1,9 @@
#[cfg(feature = "ring-cipher")]
mod ring_chacha20_poly1305;
#[cfg(feature = "ring-cipher")]
pub use ring_chacha20_poly1305::*;
#[cfg(not(feature = "ring-cipher"))]
mod rs_chacha20_poly1305;
#[cfg(not(feature = "ring-cipher"))]
pub use rs_chacha20_poly1305::*;
@@ -0,0 +1,129 @@
use anyhow::anyhow;
use ring::aead;
use ring::aead::{LessSafeKey, UnboundKey};
use crate::cipher::Finger;
use crate::protocol::body::{SecretBody, AES_GCM_ENCRYPTION_RESERVED};
use crate::protocol::NetPacket;
#[derive(Clone)]
pub struct ChaCha20Poly1305Cipher {
key: Vec<u8>,
pub(crate) cipher: LessSafeKey,
pub(crate) finger: Option<Finger>,
}
impl ChaCha20Poly1305Cipher {
pub fn new_256(key: [u8; 32], finger: Option<Finger>) -> Self {
let cipher = LessSafeKey::new(UnboundKey::new(&aead::CHACHA20_POLY1305, &key).unwrap());
Self {
key: key.to_vec(),
cipher,
finger,
}
}
}
impl ChaCha20Poly1305Cipher {
pub fn key(&self) -> &[u8] {
&self.key
}
}
impl ChaCha20Poly1305Cipher {
pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self,
net_packet: &mut NetPacket<B>,
) -> anyhow::Result<()> {
if !net_packet.is_encrypt() {
//未加密的数据直接丢弃
return Err(anyhow!("not encrypt"));
}
if net_packet.payload().len() < AES_GCM_ENCRYPTION_RESERVED {
log::error!("数据异常,长度小于{}", AES_GCM_ENCRYPTION_RESERVED);
return Err(anyhow!("data err"));
}
let mut nonce_raw = [0; 12];
nonce_raw[0..4].copy_from_slice(&net_packet.source().octets());
nonce_raw[4..8].copy_from_slice(&net_packet.destination().octets());
nonce_raw[8] = net_packet.protocol().into();
nonce_raw[9] = net_packet.transport_protocol();
nonce_raw[10] = net_packet.is_gateway() as u8;
nonce_raw[11] = net_packet.source_ttl();
let nonce = aead::Nonce::assume_unique_for_key(nonce_raw);
let mut secret_body = SecretBody::new(net_packet.payload_mut(), self.finger.is_some())?;
if let Some(finger) = &self.finger {
let finger = finger.calculate_finger(&nonce_raw, secret_body.en_body());
if &finger != secret_body.finger() {
return Err(anyhow!("ring CHACHA20_POLY1305 finger err"));
}
}
let rs = self
.cipher
.open_in_place(nonce, aead::Aad::empty(), secret_body.en_body_mut());
if let Err(e) = rs {
return Err(anyhow!("ring CHACHA20_POLY1305 解密失败:{}", e));
}
net_packet.set_encrypt_flag(false);
net_packet.set_data_len(net_packet.data_len() - AES_GCM_ENCRYPTION_RESERVED)?;
return Ok(());
}
/// net_packet 必须预留足够长度
/// data_len是有效载荷的长度
/// 返回加密后载荷的长度
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self,
net_packet: &mut NetPacket<B>,
) -> anyhow::Result<()> {
let mut nonce_raw = [0; 12];
nonce_raw[0..4].copy_from_slice(&net_packet.source().octets());
nonce_raw[4..8].copy_from_slice(&net_packet.destination().octets());
nonce_raw[8] = net_packet.protocol().into();
nonce_raw[9] = net_packet.transport_protocol();
nonce_raw[10] = net_packet.is_gateway() as u8;
nonce_raw[11] = net_packet.source_ttl();
let nonce = aead::Nonce::assume_unique_for_key(nonce_raw);
let data_len = net_packet.data_len() + AES_GCM_ENCRYPTION_RESERVED;
net_packet.set_data_len(data_len)?;
let mut secret_body = SecretBody::new(net_packet.payload_mut(), self.finger.is_some())?;
let rs = self.cipher.seal_in_place_separate_tag(
nonce,
aead::Aad::empty(),
secret_body.body_mut(),
);
return match rs {
Ok(tag) => {
let tag = tag.as_ref();
if tag.len() != 16 {
return Err(anyhow!("加密tag长度错误:{}", tag.len()));
}
secret_body.set_tag(tag)?;
if let Some(finger) = &self.finger {
let finger = finger.calculate_finger(&nonce_raw, secret_body.en_body());
secret_body.set_finger(&finger)?;
}
net_packet.set_encrypt_flag(true);
Ok(())
}
Err(e) => Err(anyhow!("ring CHACHA20_POLY1305 加密失败:{}", e)),
};
}
}
#[test]
fn test_ring_chacha20_poly1305() {
let d = ChaCha20Poly1305Cipher::new_256([0; 32], Some(Finger::new("123")));
let mut p = NetPacket::new_encrypt([0; 73]).unwrap();
let src = p.buffer().to_vec();
d.encrypt_ipv4(&mut p).unwrap();
d.decrypt_ipv4(&mut p).unwrap();
assert_eq!(p.buffer(), &src);
let d = ChaCha20Poly1305Cipher::new_256([0; 32], None);
let mut p = NetPacket::new_encrypt([0; 73]).unwrap();
let src = p.buffer().to_vec();
d.encrypt_ipv4(&mut p).unwrap();
d.decrypt_ipv4(&mut p).unwrap();
assert_eq!(p.buffer(), &src);
}
@@ -0,0 +1,121 @@
use anyhow::anyhow;
use chacha20poly1305::aead::{Nonce, Tag};
use chacha20poly1305::{AeadInPlace, ChaCha20Poly1305, Key, KeyInit};
use crate::cipher::Finger;
use crate::protocol::body::{SecretBody, AES_GCM_ENCRYPTION_RESERVED};
use crate::protocol::NetPacket;
#[derive(Clone)]
pub struct ChaCha20Poly1305Cipher {
key: Vec<u8>,
pub(crate) cipher: ChaCha20Poly1305,
pub(crate) finger: Option<Finger>,
}
impl ChaCha20Poly1305Cipher {
pub fn new_256(key: [u8; 32], finger: Option<Finger>) -> Self {
let key: &Key = &key.into();
let cipher = ChaCha20Poly1305::new(key);
Self {
key: key.to_vec(),
cipher,
finger,
}
}
}
impl ChaCha20Poly1305Cipher {
pub fn key(&self) -> &[u8] {
&self.key
}
}
impl ChaCha20Poly1305Cipher {
pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self,
net_packet: &mut NetPacket<B>,
) -> anyhow::Result<()> {
if !net_packet.is_encrypt() {
//未加密的数据直接丢弃
return Err(anyhow!("not encrypt"));
}
if net_packet.payload().len() < AES_GCM_ENCRYPTION_RESERVED {
log::error!("数据异常,长度小于{}", AES_GCM_ENCRYPTION_RESERVED);
return Err(anyhow!("data err"));
}
let mut nonce_raw = [0; 12];
nonce_raw[0..4].copy_from_slice(&net_packet.source().octets());
nonce_raw[4..8].copy_from_slice(&net_packet.destination().octets());
nonce_raw[8] = net_packet.protocol().into();
nonce_raw[9] = net_packet.transport_protocol();
nonce_raw[10] = net_packet.is_gateway() as u8;
nonce_raw[11] = net_packet.source_ttl();
let mut secret_body = SecretBody::new(net_packet.payload_mut(), self.finger.is_some())?;
if let Some(finger) = &self.finger {
let finger = finger.calculate_finger(&nonce_raw, secret_body.en_body());
if &finger != secret_body.finger() {
return Err(anyhow!("rs CHACHA20_POLY1305 finger err"));
}
}
let nonce: Nonce<ChaCha20Poly1305> = nonce_raw.into();
let tag: Tag<ChaCha20Poly1305> =
Tag::<ChaCha20Poly1305>::from_slice(secret_body.tag()).clone();
if let Err(e) =
self.cipher
.decrypt_in_place_detached(&nonce, &[], secret_body.body_mut(), &tag)
{
return Err(anyhow!("rs CHACHA20_POLY1305 decrypt_ipv4 {:?}", e));
}
net_packet.set_encrypt_flag(false);
net_packet.set_data_len(net_packet.data_len() - AES_GCM_ENCRYPTION_RESERVED)?;
Ok(())
}
/// net_packet 必须预留足够长度
/// data_len是有效载荷的长度
/// 返回加密后载荷的长度
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self,
net_packet: &mut NetPacket<B>,
) -> anyhow::Result<()> {
let mut nonce_raw = [0; 12];
nonce_raw[0..4].copy_from_slice(&net_packet.source().octets());
nonce_raw[4..8].copy_from_slice(&net_packet.destination().octets());
nonce_raw[8] = net_packet.protocol().into();
nonce_raw[9] = net_packet.transport_protocol();
nonce_raw[10] = net_packet.is_gateway() as u8;
nonce_raw[11] = net_packet.source_ttl();
let nonce = nonce_raw.into();
let data_len = net_packet.data_len() + AES_GCM_ENCRYPTION_RESERVED;
net_packet.set_data_len(data_len)?;
let mut secret_body = SecretBody::new(net_packet.payload_mut(), self.finger.is_some())?;
let rs = self
.cipher
.encrypt_in_place_detached(&nonce, &[], secret_body.body_mut());
return match rs {
Ok(tag) => {
let tag: &[u8] = tag.as_ref();
if tag.len() != 16 {
return Err(anyhow!("加密tag长度错误:{}", tag.len(),));
}
secret_body.set_tag(tag)?;
if let Some(finger) = &self.finger {
let finger = finger.calculate_finger(&nonce_raw, secret_body.en_body());
secret_body.set_finger(&finger)?;
}
net_packet.set_encrypt_flag(true);
Ok(())
}
Err(e) => Err(anyhow!("rs CHACHA20_POLY1305 加密失败:{}", e)),
};
}
}
#[test]
fn test_rs_chacha20_poly1305() {
let d = ChaCha20Poly1305Cipher::new_256([0; 32], Some(Finger::new("123")));
let mut p = NetPacket::new_encrypt([0; 73]).unwrap();
let src = p.buffer().to_vec();
d.encrypt_ipv4(&mut p).unwrap();
d.decrypt_ipv4(&mut p).unwrap();
assert_eq!(p.buffer(), &src);
}
+108 -132
View File
@@ -1,51 +1,42 @@
#[cfg(feature = "aes_ecb")]
#[cfg(not(any(feature = "openssl-vendored", feature = "openssl")))]
use crate::cipher::aes_ecb::AesEcbCipher;
use std::fmt::Display; use std::fmt::Display;
use std::str::FromStr;
use anyhow::anyhow;
#[cfg(cipher)]
use sha2::Digest;
#[cfg(feature = "aes_cbc")] #[cfg(feature = "aes_cbc")]
use crate::cipher::aes_cbc::AesCbcCipher; use crate::cipher::aes_cbc::AesCbcCipher;
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
#[cfg(not(feature = "ring-cipher"))]
use crate::cipher::aes_gcm_cipher::AesGcmCipher;
#[cfg(feature = "aes_ecb")] #[cfg(feature = "aes_ecb")]
#[cfg(any(feature = "openssl-vendored", feature = "openssl"))] use crate::cipher::aes_ecb::AesEcbCipher;
use crate::cipher::openssl_aes_ecb::AesEcbCipher;
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
#[cfg(feature = "ring-cipher")] use crate::cipher::aes_gcm::AesGcmCipher;
use crate::cipher::ring_aes_gcm_cipher::AesGcmCipher; #[cfg(feature = "chacha20_poly1305")]
use crate::cipher::chacha20::ChaCha20Cipher;
#[cfg(feature = "chacha20_poly1305")]
use crate::cipher::chacha20_poly1305::ChaCha20Poly1305Cipher;
#[cfg(feature = "sm4_cbc")] #[cfg(feature = "sm4_cbc")]
use crate::cipher::sm4_cbc::Sm4CbcCipher; use crate::cipher::sm4_cbc::Sm4CbcCipher;
#[cfg(any( use crate::cipher::xor::XORCipher;
feature = "aes_gcm", #[cfg(cipher)]
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
use crate::cipher::Finger; use crate::cipher::Finger;
use crate::protocol::NetPacket; use crate::protocol::NetPacket;
#[cfg(any(
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
use sha2::Digest;
use std::io;
use std::str::FromStr;
#[derive(Copy, Clone, Eq, PartialEq, Debug)] #[derive(Copy, Clone, Eq, PartialEq, Debug)]
pub enum CipherModel { pub enum CipherModel {
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
AesGcm, AesGcm,
#[cfg(feature = "chacha20_poly1305")]
Chacha20Poly1305,
#[cfg(feature = "chacha20_poly1305")]
Chacha20,
#[cfg(feature = "aes_cbc")] #[cfg(feature = "aes_cbc")]
AesCbc, AesCbc,
#[cfg(feature = "aes_ecb")] #[cfg(feature = "aes_ecb")]
AesEcb, AesEcb,
#[cfg(feature = "sm4_cbc")] #[cfg(feature = "sm4_cbc")]
Sm4Cbc, Sm4Cbc,
Xor,
None, None,
} }
@@ -54,61 +45,55 @@ impl Display for CipherModel {
let str = match self { let str = match self {
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
CipherModel::AesGcm => "aes_gcm".to_string(), CipherModel::AesGcm => "aes_gcm".to_string(),
#[cfg(feature = "chacha20_poly1305")]
CipherModel::Chacha20Poly1305 => "chacha20_poly1305".to_string(),
#[cfg(feature = "chacha20_poly1305")]
CipherModel::Chacha20 => "chacha20".to_string(),
#[cfg(feature = "aes_cbc")] #[cfg(feature = "aes_cbc")]
CipherModel::AesCbc => "aes_cbc".to_string(), CipherModel::AesCbc => "aes_cbc".to_string(),
#[cfg(feature = "aes_ecb")] #[cfg(feature = "aes_ecb")]
CipherModel::AesEcb => "aes_ecb".to_string(), CipherModel::AesEcb => "aes_ecb".to_string(),
#[cfg(feature = "sm4_cbc")] #[cfg(feature = "sm4_cbc")]
CipherModel::Sm4Cbc => "sm4_cbc".to_string(), CipherModel::Sm4Cbc => "sm4_cbc".to_string(),
CipherModel::Xor => "xor".to_string(),
CipherModel::None => "none".to_string(), CipherModel::None => "none".to_string(),
}; };
write!(f, "{}", str) write!(f, "{}", str)
} }
} }
impl FromStr for CipherModel { impl FromStr for CipherModel {
type Err = String; type Err = String;
fn from_str(s: &str) -> Result<Self, Self::Err> { fn from_str(s: &str) -> Result<Self, Self::Err> {
#[cfg(not(any(
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
)))]
return Err(format!("not match '{}', no encrypt", s));
#[cfg(any(
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
match s.to_lowercase().trim() { match s.to_lowercase().trim() {
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
"aes_gcm" => Ok(CipherModel::AesGcm), "aes_gcm" => Ok(CipherModel::AesGcm),
#[cfg(feature = "chacha20_poly1305")]
"chacha20_poly1305" => Ok(CipherModel::Chacha20Poly1305),
#[cfg(feature = "chacha20_poly1305")]
"chacha20" => Ok(CipherModel::Chacha20),
#[cfg(feature = "aes_cbc")] #[cfg(feature = "aes_cbc")]
"aes_cbc" => Ok(CipherModel::AesCbc), "aes_cbc" => Ok(CipherModel::AesCbc),
#[cfg(feature = "aes_ecb")] #[cfg(feature = "aes_ecb")]
"aes_ecb" => Ok(CipherModel::AesEcb), "aes_ecb" => Ok(CipherModel::AesEcb),
#[cfg(feature = "sm4_cbc")] #[cfg(feature = "sm4_cbc")]
"sm4_cbc" => Ok(CipherModel::Sm4Cbc), "sm4_cbc" => Ok(CipherModel::Sm4Cbc),
"xor" => Ok(CipherModel::Xor),
_ => { _ => {
let mut enums = String::new(); let mut enums = String::new();
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
enums.push_str("/aes_gcm"); enums.push_str("/aes_gcm");
#[cfg(feature = "chacha20_poly1305")]
enums.push_str("/chacha20_poly1305/chacha20");
#[cfg(feature = "aes_cbc")] #[cfg(feature = "aes_cbc")]
enums.push_str("/aes_cbc"); enums.push_str("/aes_cbc");
#[cfg(feature = "aes_ecb")] #[cfg(feature = "aes_ecb")]
enums.push_str("/aes_ecb"); enums.push_str("/aes_ecb");
#[cfg(feature = "sm4_cbc")] #[cfg(feature = "sm4_cbc")]
enums.push_str("/sm4_cbc"); enums.push_str("/sm4_cbc");
let str = if enums.is_empty() { enums.push_str("/xor");
"no encrypt" Err(format!("not match '{}', enum:{}", s, &enums[1..]))
} else {
&enums[1..]
};
Err(format!("not match '{}', enum:{}", s, str))
} }
} }
} }
@@ -118,49 +103,37 @@ impl FromStr for CipherModel {
pub enum Cipher { pub enum Cipher {
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
AesGcm((AesGcmCipher, Vec<u8>)), AesGcm((AesGcmCipher, Vec<u8>)),
#[cfg(feature = "chacha20_poly1305")]
Chacha20Poly1305(ChaCha20Poly1305Cipher),
#[cfg(feature = "chacha20_poly1305")]
Chacha20(ChaCha20Cipher),
#[cfg(feature = "aes_cbc")] #[cfg(feature = "aes_cbc")]
AesCbc(AesCbcCipher), AesCbc(AesCbcCipher),
#[cfg(feature = "aes_ecb")] #[cfg(feature = "aes_ecb")]
AesEcb(AesEcbCipher), AesEcb(AesEcbCipher),
#[cfg(feature = "sm4_cbc")] #[cfg(feature = "sm4_cbc")]
Sm4Cbc(Sm4CbcCipher), Sm4Cbc(Sm4CbcCipher),
Xor(XORCipher),
None, None,
} }
impl Cipher { impl Cipher {
#[cfg(not(any(
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
)))]
pub fn new_password(
_model: CipherModel,
_password: Option<String>,
_token: Option<String>,
) -> Self {
Cipher::None
}
#[cfg(any(
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
pub fn new_password( pub fn new_password(
model: CipherModel, model: CipherModel,
password: Option<String>, password: Option<String>,
token: Option<String>, token: Option<String>,
) -> Self { ) -> Self {
let finger = token.map(|token| Finger::new(&token));
if let Some(password) = password { if let Some(password) = password {
let mut hasher = sha2::Sha256::new(); #[cfg(cipher)]
hasher.update(password.as_bytes()); let key: [u8; 32] = {
let key: [u8; 32] = hasher.finalize().into(); let mut hasher = sha2::Sha256::new();
hasher.update(password.as_bytes());
hasher.finalize().into()
};
match model { match model {
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
CipherModel::AesGcm => { CipherModel::AesGcm => {
let finger = token.map(|token| Finger::new(&token));
if password.len() < 8 { if password.len() < 8 {
let aes = AesGcmCipher::new_128(key[..16].try_into().unwrap(), finger); let aes = AesGcmCipher::new_128(key[..16].try_into().unwrap(), finger);
Cipher::AesGcm((aes, key[..16].to_vec())) Cipher::AesGcm((aes, key[..16].to_vec()))
@@ -169,8 +142,21 @@ impl Cipher {
Cipher::AesGcm((aes, key.to_vec())) Cipher::AesGcm((aes, key.to_vec()))
} }
} }
#[cfg(feature = "chacha20_poly1305")]
CipherModel::Chacha20Poly1305 => {
let finger = token.map(|token| Finger::new(&token));
let chacha = ChaCha20Poly1305Cipher::new_256(key, finger);
Cipher::Chacha20Poly1305(chacha)
}
#[cfg(feature = "chacha20_poly1305")]
CipherModel::Chacha20 => {
let finger = token.map(|token| Finger::new(&token));
let chacha = ChaCha20Cipher::new_256(key, finger);
Cipher::Chacha20(chacha)
}
#[cfg(feature = "aes_cbc")] #[cfg(feature = "aes_cbc")]
CipherModel::AesCbc => { CipherModel::AesCbc => {
let finger = token.map(|token| Finger::new(&token));
if password.len() < 8 { if password.len() < 8 {
let aes = AesCbcCipher::new_128(key[..16].try_into().unwrap(), finger); let aes = AesCbcCipher::new_128(key[..16].try_into().unwrap(), finger);
Cipher::AesCbc(aes) Cipher::AesCbc(aes)
@@ -181,6 +167,7 @@ impl Cipher {
} }
#[cfg(feature = "aes_ecb")] #[cfg(feature = "aes_ecb")]
CipherModel::AesEcb => { CipherModel::AesEcb => {
let finger = token.map(|token| Finger::new(&token));
if password.len() < 8 { if password.len() < 8 {
let aes = AesEcbCipher::new_128(key[..16].try_into().unwrap(), finger); let aes = AesEcbCipher::new_128(key[..16].try_into().unwrap(), finger);
Cipher::AesEcb(aes) Cipher::AesEcb(aes)
@@ -191,126 +178,97 @@ impl Cipher {
} }
#[cfg(feature = "sm4_cbc")] #[cfg(feature = "sm4_cbc")]
CipherModel::Sm4Cbc => { CipherModel::Sm4Cbc => {
let finger = token.map(|token| Finger::new(&token));
let aes = Sm4CbcCipher::new_128(key[..16].try_into().unwrap(), finger); let aes = Sm4CbcCipher::new_128(key[..16].try_into().unwrap(), finger);
Cipher::Sm4Cbc(aes) Cipher::Sm4Cbc(aes)
} }
CipherModel::Xor => {
let _token = token;
Cipher::Xor(XORCipher::new_256(crate::cipher::xor::simple_hash(
&password,
)))
}
CipherModel::None => Cipher::None, CipherModel::None => Cipher::None,
} }
} else { } else {
Cipher::None Cipher::None
} }
} }
#[cfg(not(any( #[cfg(not(any(feature = "aes_gcm", feature = "server_encrypt")))]
feature = "aes_gcm", pub fn new_key(_key: [u8; 32], _token: String) -> anyhow::Result<Self> {
feature = "server_encrypt", Err(anyhow!("key error"))
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
)))]
pub fn new_key(_key: [u8; 32], _token: String) -> io::Result<Self> {
Err(io::Error::new(io::ErrorKind::Other, "key error"))
} }
#[cfg(any( #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
feature = "aes_gcm", pub fn new_key(key: [u8; 32], token: String) -> anyhow::Result<Self> {
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
pub fn new_key(key: [u8; 32], token: String) -> io::Result<Self> {
let finger = Some(Finger::new(&token)); let finger = Some(Finger::new(&token));
match key.len() { match key.len() {
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
16 => { 16 => {
let aes = AesGcmCipher::new_128(key[..16].try_into().unwrap(), finger); let aes = AesGcmCipher::new_128(key[..16].try_into().unwrap(), finger);
Ok(Cipher::AesGcm((aes, key[..16].to_vec()))) Ok(Cipher::AesGcm((aes, key[..16].to_vec())))
} }
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
32 => { 32 => {
let aes = AesGcmCipher::new_256(key, finger); let aes = AesGcmCipher::new_256(key, finger);
Ok(Cipher::AesGcm((aes, key.to_vec()))) Ok(Cipher::AesGcm((aes, key.to_vec())))
} }
_ => Err(io::Error::new(io::ErrorKind::Other, "key error")), _ => Err(anyhow!("key error")),
} }
} }
pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
match self { match self {
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
Cipher::AesGcm((aes_gcm, _)) => aes_gcm.decrypt_ipv4(net_packet), Cipher::AesGcm((aes_gcm, _)) => aes_gcm.decrypt_ipv4(net_packet),
#[cfg(feature = "aes_cbc")] #[cfg(feature = "aes_cbc")]
Cipher::AesCbc(aes_cbc) => aes_cbc.decrypt_ipv4(net_packet), Cipher::AesCbc(aes_cbc) => aes_cbc.decrypt_ipv4(net_packet),
#[cfg(feature = "chacha20_poly1305")]
Cipher::Chacha20Poly1305(chacha20poly1305) => chacha20poly1305.decrypt_ipv4(net_packet),
#[cfg(feature = "chacha20_poly1305")]
Cipher::Chacha20(chacha20) => chacha20.decrypt_ipv4(net_packet),
#[cfg(feature = "aes_ecb")] #[cfg(feature = "aes_ecb")]
Cipher::AesEcb(aes_ecb) => aes_ecb.decrypt_ipv4(net_packet), Cipher::AesEcb(aes_ecb) => aes_ecb.decrypt_ipv4(net_packet),
#[cfg(feature = "sm4_cbc")] #[cfg(feature = "sm4_cbc")]
Cipher::Sm4Cbc(sm4_cbc) => sm4_cbc.decrypt_ipv4(net_packet), Cipher::Sm4Cbc(sm4_cbc) => sm4_cbc.decrypt_ipv4(net_packet),
Cipher::Xor(xor) => xor.decrypt_ipv4(net_packet),
Cipher::None => { Cipher::None => {
if net_packet.is_encrypt() { if net_packet.is_encrypt() {
return Err(io::Error::new(io::ErrorKind::Other, "not key")); return Err(anyhow!("not key"));
} }
Ok(()) Ok(())
} }
} }
} }
#[cfg(not(any(
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
)))]
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self,
_net_packet: &mut NetPacket<B>,
) -> io::Result<()> {
Ok(())
}
#[cfg(any(
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
match self { match self {
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
Cipher::AesGcm((aes_gcm, _)) => aes_gcm.encrypt_ipv4(net_packet), Cipher::AesGcm((aes_gcm, _)) => aes_gcm.encrypt_ipv4(net_packet),
#[cfg(feature = "chacha20_poly1305")]
Cipher::Chacha20Poly1305(chacha20poly1305) => chacha20poly1305.encrypt_ipv4(net_packet),
#[cfg(feature = "chacha20_poly1305")]
Cipher::Chacha20(chacha20) => chacha20.encrypt_ipv4(net_packet),
#[cfg(feature = "aes_cbc")] #[cfg(feature = "aes_cbc")]
Cipher::AesCbc(aes_cbc) => aes_cbc.encrypt_ipv4(net_packet), Cipher::AesCbc(aes_cbc) => aes_cbc.encrypt_ipv4(net_packet),
#[cfg(feature = "aes_ecb")] #[cfg(feature = "aes_ecb")]
Cipher::AesEcb(aes_ecb) => aes_ecb.encrypt_ipv4(net_packet), Cipher::AesEcb(aes_ecb) => aes_ecb.encrypt_ipv4(net_packet),
#[cfg(feature = "sm4_cbc")] #[cfg(feature = "sm4_cbc")]
Cipher::Sm4Cbc(sm4_cbc) => sm4_cbc.encrypt_ipv4(net_packet), Cipher::Sm4Cbc(sm4_cbc) => sm4_cbc.encrypt_ipv4(net_packet),
Cipher::Xor(xor) => xor.encrypt_ipv4(net_packet),
Cipher::None => Ok(()), Cipher::None => Ok(()),
} }
} }
#[cfg(not(any( #[cfg(not(cipher))]
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
)))]
pub fn check_finger<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn check_finger<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
_net_packet: &NetPacket<B>, _net_packet: &NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
Ok(()) Ok(())
} }
#[cfg(any( #[cfg(cipher)]
feature = "aes_gcm", pub fn check_finger<B: AsRef<[u8]>>(&self, net_packet: &NetPacket<B>) -> anyhow::Result<()> {
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
pub fn check_finger<B: AsRef<[u8]>>(&self, net_packet: &NetPacket<B>) -> io::Result<()> {
match self { match self {
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
Cipher::AesGcm((aes_gcm, _)) => aes_gcm Cipher::AesGcm((aes_gcm, _)) => aes_gcm
@@ -318,6 +276,18 @@ impl Cipher {
.as_ref() .as_ref()
.map(|f| f.check_finger(net_packet)) .map(|f| f.check_finger(net_packet))
.unwrap_or(Ok(())), .unwrap_or(Ok(())),
#[cfg(feature = "chacha20_poly1305")]
Cipher::Chacha20Poly1305(chacha20poly1305) => chacha20poly1305
.finger
.as_ref()
.map(|f| f.check_finger(net_packet))
.unwrap_or(Ok(())),
#[cfg(feature = "chacha20_poly1305")]
Cipher::Chacha20(chacha20) => chacha20
.finger
.as_ref()
.map(|f| f.check_finger(net_packet))
.unwrap_or(Ok(())),
#[cfg(feature = "aes_cbc")] #[cfg(feature = "aes_cbc")]
Cipher::AesCbc(aes_cbc) => aes_cbc Cipher::AesCbc(aes_cbc) => aes_cbc
.finger .finger
@@ -336,6 +306,7 @@ impl Cipher {
.as_ref() .as_ref()
.map(|f| f.check_finger(net_packet)) .map(|f| f.check_finger(net_packet))
.unwrap_or(Ok(())), .unwrap_or(Ok(())),
Cipher::Xor(_) => Ok(()),
Cipher::None => Ok(()), Cipher::None => Ok(()),
} }
} }
@@ -343,12 +314,17 @@ impl Cipher {
match self { match self {
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))] #[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
Cipher::AesGcm((_, key)) => Some(key), Cipher::AesGcm((_, key)) => Some(key),
#[cfg(feature = "chacha20_poly1305")]
Cipher::Chacha20Poly1305(chacha20poly1305) => Some(chacha20poly1305.key()),
#[cfg(feature = "chacha20_poly1305")]
Cipher::Chacha20(chacha20) => Some(chacha20.key()),
#[cfg(feature = "aes_cbc")] #[cfg(feature = "aes_cbc")]
Cipher::AesCbc(aes_cbc) => Some(aes_cbc.key()), Cipher::AesCbc(aes_cbc) => Some(aes_cbc.key()),
#[cfg(feature = "aes_ecb")] #[cfg(feature = "aes_ecb")]
Cipher::AesEcb(aes_ecb) => Some(aes_ecb.key()), Cipher::AesEcb(aes_ecb) => Some(aes_ecb.key()),
#[cfg(feature = "sm4_cbc")] #[cfg(feature = "sm4_cbc")]
Cipher::Sm4Cbc(sm4_cbc) => Some(sm4_cbc.key()), Cipher::Sm4Cbc(sm4_cbc) => Some(sm4_cbc.key()),
Cipher::Xor(xor) => Some(xor.key()),
Cipher::None => None, Cipher::None => None,
} }
} }
+5 -5
View File
@@ -1,4 +1,4 @@
use std::io; use anyhow::anyhow;
use sha2::Digest; use sha2::Digest;
@@ -16,15 +16,15 @@ impl Finger {
let hash: [u8; 32] = hasher.finalize().into(); let hash: [u8; 32] = hasher.finalize().into();
Finger { hash } Finger { hash }
} }
pub fn check_finger<B: AsRef<[u8]>>(&self, net_packet: &NetPacket<B>) -> io::Result<()> { pub fn check_finger<B: AsRef<[u8]>>(&self, net_packet: &NetPacket<B>) -> anyhow::Result<()> {
if !net_packet.is_encrypt() { if !net_packet.is_encrypt() {
//未加密的数据直接丢弃 //未加密的数据直接丢弃
return Err(io::Error::new(io::ErrorKind::Other, "not encrypt")); return Err(anyhow!("not encrypt"));
} }
let payload_len = net_packet.payload().len(); let payload_len = net_packet.payload().len();
if payload_len < 12 { if payload_len < 12 {
log::error!("数据异常,长度小于{}", 12); log::error!("数据异常,长度小于{}", 12);
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
let mut nonce_raw = [0; 12]; let mut nonce_raw = [0; 12];
nonce_raw[0..4].copy_from_slice(&net_packet.source().octets()); nonce_raw[0..4].copy_from_slice(&net_packet.source().octets());
@@ -36,7 +36,7 @@ impl Finger {
let payload = net_packet.payload(); let payload = net_packet.payload();
let finger = self.calculate_finger(&nonce_raw, &payload[..payload_len - 12]); let finger = self.calculate_finger(&nonce_raw, &payload[..payload_len - 12]);
if &finger[..] != &payload[payload_len - 12..] { if &finger[..] != &payload[payload_len - 12..] {
return Err(io::Error::new(io::ErrorKind::Other, "finger err")); return Err(anyhow!("finger err"));
} }
Ok(()) Ok(())
} }
+22 -30
View File
@@ -1,40 +1,32 @@
#[cfg(feature = "aes_cbc")]
mod aes_cbc;
#[cfg(feature = "aes_ecb")]
#[cfg(not(any(feature = "openssl-vendored", feature = "openssl")))]
mod aes_ecb;
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
#[cfg(not(feature = "ring-cipher"))]
mod aes_gcm_cipher;
mod cipher; mod cipher;
#[cfg(any( #[cfg(cipher)]
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
mod finger; mod finger;
#[cfg(feature = "aes_ecb")]
#[cfg(any(feature = "openssl-vendored", feature = "openssl"))]
mod openssl_aes_ecb;
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
#[cfg(feature = "ring-cipher")]
mod ring_aes_gcm_cipher;
#[cfg(feature = "sm4_cbc")]
mod sm4_cbc;
pub use cipher::Cipher; pub use cipher::Cipher;
pub use cipher::CipherModel; pub use cipher::CipherModel;
#[cfg(any( #[cfg(cipher)]
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
pub use finger::Finger; pub use finger::Finger;
#[cfg(feature = "server_encrypt")] #[cfg(feature = "server_encrypt")]
mod rsa_cipher; mod rsa_cipher;
#[cfg(feature = "server_encrypt")] #[cfg(feature = "server_encrypt")]
pub use rsa_cipher::RsaCipher; pub use rsa_cipher::RsaCipher;
#[cfg(any(feature = "aes_gcm", feature = "server_encrypt"))]
mod aes_gcm;
#[cfg(feature = "chacha20_poly1305")]
mod chacha20;
#[cfg(feature = "chacha20_poly1305")]
mod chacha20_poly1305;
#[cfg(feature = "aes_ecb")]
mod aes_ecb;
#[cfg(feature = "aes_cbc")]
mod aes_cbc;
#[cfg(feature = "sm4_cbc")]
mod sm4_cbc;
mod xor;
pub use xor::simple_hash;
+2
View File
@@ -0,0 +1,2 @@
mod rs_sm4_cbc;
pub use rs_sm4_cbc::*;
@@ -1,9 +1,9 @@
use crate::cipher::Finger; use crate::cipher::Finger;
use crate::protocol::{NetPacket, HEAD_LEN}; use crate::protocol::{NetPacket, HEAD_LEN};
use anyhow::anyhow;
use libsm::sm4::cipher_mode::CipherMode; use libsm::sm4::cipher_mode::CipherMode;
use libsm::sm4::Sm4CipherMode; use libsm::sm4::Sm4CipherMode;
use rand::RngCore; use rand::RngCore;
use std::io;
pub struct Sm4CbcCipher { pub struct Sm4CbcCipher {
key: [u8; 16], key: [u8; 16],
@@ -41,10 +41,10 @@ impl Sm4CbcCipher {
pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
if !net_packet.is_encrypt() { if !net_packet.is_encrypt() {
//未加密的数据直接丢弃 //未加密的数据直接丢弃
return Err(io::Error::new(io::ErrorKind::Other, "not encrypt")); return Err(anyhow!("not encrypt"));
} }
if let Some(finger) = &self.finger { if let Some(finger) = &self.finger {
@@ -57,12 +57,12 @@ impl Sm4CbcCipher {
nonce_raw[11] = net_packet.source_ttl(); nonce_raw[11] = net_packet.source_ttl();
let len = net_packet.payload().len(); let len = net_packet.payload().len();
if len < 12 { if len < 12 {
return Err(io::Error::new(io::ErrorKind::Other, "payload len <12")); return Err(anyhow!("payload len <12"));
} }
let secret_body = &net_packet.payload()[..len - 12]; let secret_body = &net_packet.payload()[..len - 12];
let finger = finger.calculate_finger(&nonce_raw, secret_body); let finger = finger.calculate_finger(&nonce_raw, secret_body);
if &finger != &net_packet.payload()[len - 12..] { if &finger != &net_packet.payload()[len - 12..] {
return Err(io::Error::new(io::ErrorKind::Other, "finger err")); return Err(anyhow!("finger err"));
} }
net_packet.set_data_len(net_packet.data_len() - finger.len())?; net_packet.set_data_len(net_packet.data_len() - finger.len())?;
} }
@@ -70,7 +70,7 @@ impl Sm4CbcCipher {
let len = payload.len(); let len = payload.len();
if len < 16 || len > 1024 * 4 { if len < 16 || len > 1024 * 4 {
log::error!("数据异常,长度{}小于16或大于4096", len); log::error!("数据异常,长度{}小于16或大于4096", len);
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
let mut out = [0u8; 1024 * 4]; let mut out = [0u8; 1024 * 4];
let data = &payload[..len - 16]; let data = &payload[..len - 16];
@@ -79,32 +79,29 @@ impl Sm4CbcCipher {
Ok(len) => { Ok(len) => {
let src_net_packet = NetPacket::new(&out[..len])?; let src_net_packet = NetPacket::new(&out[..len])?;
if src_net_packet.source() != net_packet.source() { if src_net_packet.source() != net_packet.source() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.destination() != net_packet.destination() { if src_net_packet.destination() != net_packet.destination() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.protocol() != net_packet.protocol() { if src_net_packet.protocol() != net_packet.protocol() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.transport_protocol() != net_packet.transport_protocol() { if src_net_packet.transport_protocol() != net_packet.transport_protocol() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.is_gateway() != net_packet.is_gateway() { if src_net_packet.is_gateway() != net_packet.is_gateway() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
if src_net_packet.source_ttl() != net_packet.source_ttl() { if src_net_packet.source_ttl() != net_packet.source_ttl() {
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
net_packet.set_data_len(len)?; net_packet.set_data_len(len)?;
net_packet.set_payload(src_net_packet.payload())?; net_packet.set_payload(src_net_packet.payload())?;
net_packet.set_encrypt_flag(false); net_packet.set_encrypt_flag(false);
Ok(()) Ok(())
} }
Err(e) => Err(io::Error::new( Err(e) => Err(anyhow!("sm4_cbc解密失败:{}", e)),
io::ErrorKind::Other,
format!("sm4_cbc解密失败:{}", e),
)),
} }
} }
/// net_packet 必须预留足够长度 /// net_packet 必须预留足够长度
@@ -112,7 +109,7 @@ impl Sm4CbcCipher {
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>( pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self, &self,
net_packet: &mut NetPacket<B>, net_packet: &mut NetPacket<B>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let mut out = [0u8; 1024 * 4]; let mut out = [0u8; 1024 * 4];
let mut iv = [0u8; 16]; let mut iv = [0u8; 16];
rand::thread_rng().fill_bytes(&mut iv); rand::thread_rng().fill_bytes(&mut iv);
@@ -121,7 +118,7 @@ impl Sm4CbcCipher {
"数据异常,长度{}大于1024 * 4 - 32", "数据异常,长度{}大于1024 * 4 - 32",
net_packet.buffer().len() net_packet.buffer().len()
); );
return Err(io::Error::new(io::ErrorKind::Other, "data err")); return Err(anyhow!("data err"));
} }
match self.cipher.encrypt(net_packet.buffer(), &iv, &mut out) { match self.cipher.encrypt(net_packet.buffer(), &iv, &mut out) {
Ok(len) => { Ok(len) => {
@@ -146,10 +143,7 @@ impl Sm4CbcCipher {
net_packet.set_encrypt_flag(true); net_packet.set_encrypt_flag(true);
Ok(()) Ok(())
} }
Err(e) => Err(io::Error::new( Err(e) => Err(anyhow!("sm4_cbc加密失败:{}", e)),
io::ErrorKind::Other,
format!("sm4_cbc加密失败:{}", e),
)),
} }
} }
} }
+2
View File
@@ -0,0 +1,2 @@
mod xor;
pub use xor::*;
+84
View File
@@ -0,0 +1,84 @@
use anyhow::anyhow;
use crate::protocol::NetPacket;
pub fn simple_hash(input: &str) -> [u8; 32] {
let mut result = [0u8; 32];
let bytes = input.as_bytes();
for (index, v) in result.iter_mut().enumerate() {
*v = bytes[index % bytes.len()];
}
let mut state = 0u8;
for (i, &byte) in bytes.iter().enumerate() {
let combined = byte.wrapping_add(state).rotate_left((i % 8) as u32);
result[i % 32] ^= combined;
state = state.wrapping_add(byte).rotate_left(3);
}
for i in 0..32 {
result[i] = result[i]
.rotate_left((result[(i + 1) % 32] % 8) as u32)
.wrapping_add(state);
state = state.wrapping_add(result[i]).rotate_left(3);
}
result
}
#[derive(Clone)]
pub struct XORCipher {
key: [u8; 32],
}
impl XORCipher {
pub fn new_256(key: [u8; 32]) -> Self {
Self { key }
}
}
impl XORCipher {
pub fn key(&self) -> &[u8] {
&self.key
}
}
impl XORCipher {
pub fn decrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self,
net_packet: &mut NetPacket<B>,
) -> anyhow::Result<()> {
if !net_packet.is_encrypt() {
//未加密的数据直接丢弃
return Err(anyhow!("not encrypt"));
}
let key = &self.key;
for (i, byte) in net_packet.payload_mut().iter_mut().enumerate() {
*byte ^= key[i & 31];
}
net_packet.set_encrypt_flag(false);
Ok(())
}
pub fn encrypt_ipv4<B: AsRef<[u8]> + AsMut<[u8]>>(
&self,
net_packet: &mut NetPacket<B>,
) -> anyhow::Result<()> {
net_packet.set_encrypt_flag(true);
let key = &self.key;
for (i, byte) in net_packet.payload_mut().iter_mut().enumerate() {
*byte ^= key[i & 31];
}
Ok(())
}
}
#[test]
fn test_xor() {
let d = XORCipher::new_256(simple_hash("password"));
let mut p = NetPacket::new_encrypt([0; 1000]).unwrap();
let src = p.buffer().to_vec();
d.encrypt_ipv4(&mut p).unwrap();
d.decrypt_ipv4(&mut p).unwrap();
assert_eq!(p.buffer(), &src)
}
+22 -26
View File
@@ -147,33 +147,29 @@ impl Config {
}) })
} }
} }
impl Config { impl Config {
#[cfg(any(
feature = "aes_gcm",
feature = "server_encrypt",
feature = "aes_cbc",
feature = "aes_ecb",
feature = "sm4_cbc"
))]
pub fn password_hash(&self) -> Option<[u8; 16]> { pub fn password_hash(&self) -> Option<[u8; 16]> {
self.password.as_ref().map(|v| { if let Some(p) = self.password.as_ref() {
use sha2::Digest; match self.cipher_model {
let mut hasher = sha2::Sha256::new(); CipherModel::Xor => {
hasher.update(self.cipher_model.to_string().as_bytes()); let key = crate::cipher::simple_hash(&format!("Xor{}{}", p, self.token));
hasher.update(v.as_bytes()); Some(key[16..].try_into().unwrap())
hasher.update(self.token.as_bytes()); }
let key: [u8; 32] = hasher.finalize().into(); CipherModel::None => None,
key[16..].try_into().unwrap() #[cfg(cipher)]
}) _ => {
} use sha2::Digest;
#[cfg(not(any( let mut hasher = sha2::Sha256::new();
feature = "aes_gcm", hasher.update(self.cipher_model.to_string().as_bytes());
feature = "server_encrypt", hasher.update(p.as_bytes());
feature = "aes_cbc", hasher.update(self.token.as_bytes());
feature = "aes_ecb", let key: [u8; 32] = hasher.finalize().into();
feature = "sm4_cbc" Some(key[16..].try_into().unwrap())
)))] }
pub fn password_hash(&self) -> Option<[u8; 16]> { }
None } else {
None
}
} }
} }
+4 -5
View File
@@ -1,4 +1,3 @@
use std::io;
use std::net::Ipv4Addr; use std::net::Ipv4Addr;
use std::sync::Arc; use std::sync::Arc;
use std::time::Duration; use std::time::Duration;
@@ -167,7 +166,7 @@ fn client_relay0(
current_device: &CurrentDeviceInfo, current_device: &CurrentDeviceInfo,
device_list: &Mutex<(u16, Vec<PeerDeviceInfo>)>, device_list: &Mutex<(u16, Vec<PeerDeviceInfo>)>,
client_cipher: &Cipher, client_cipher: &Cipher,
) -> io::Result<()> { ) -> anyhow::Result<()> {
// 离线了不再探测 // 离线了不再探测
if current_device.status.offline() { if current_device.status.offline() {
return Ok(()); return Ok(());
@@ -211,7 +210,7 @@ fn client_relay0(
fn heartbeat_packet( fn heartbeat_packet(
src: Ipv4Addr, src: Ipv4Addr,
dest: Ipv4Addr, dest: Ipv4Addr,
) -> io::Result<NetPacket<[u8; 12 + 4 + ENCRYPTION_RESERVED]>> { ) -> anyhow::Result<NetPacket<[u8; 12 + 4 + ENCRYPTION_RESERVED]>> {
let mut net_packet = NetPacket::new_encrypt([0u8; 12 + 4 + ENCRYPTION_RESERVED])?; let mut net_packet = NetPacket::new_encrypt([0u8; 12 + 4 + ENCRYPTION_RESERVED])?;
net_packet.set_default_version(); net_packet.set_default_version();
net_packet.set_protocol(Protocol::Control); net_packet.set_protocol(Protocol::Control);
@@ -228,7 +227,7 @@ fn heartbeat_packet_client(
client_cipher: &Cipher, client_cipher: &Cipher,
src: Ipv4Addr, src: Ipv4Addr,
dest: Ipv4Addr, dest: Ipv4Addr,
) -> io::Result<NetPacket<[u8; 12 + 4 + ENCRYPTION_RESERVED]>> { ) -> anyhow::Result<NetPacket<[u8; 12 + 4 + ENCRYPTION_RESERVED]>> {
let mut net_packet = heartbeat_packet(src, dest)?; let mut net_packet = heartbeat_packet(src, dest)?;
client_cipher.encrypt_ipv4(&mut net_packet)?; client_cipher.encrypt_ipv4(&mut net_packet)?;
Ok(net_packet) Ok(net_packet)
@@ -239,7 +238,7 @@ fn heartbeat_packet_server(
server_cipher: &Cipher, server_cipher: &Cipher,
src: Ipv4Addr, src: Ipv4Addr,
dest: Ipv4Addr, dest: Ipv4Addr,
) -> io::Result<NetPacket<[u8; 12 + 4 + ENCRYPTION_RESERVED]>> { ) -> anyhow::Result<NetPacket<[u8; 12 + 4 + ENCRYPTION_RESERVED]>> {
let mut net_packet = heartbeat_packet(src, dest)?; let mut net_packet = heartbeat_packet(src, dest)?;
let mut ping = PingPacket::new(net_packet.payload_mut())?; let mut ping = PingPacket::new(net_packet.payload_mut())?;
ping.set_epoch(device_list.lock().0); ping.set_epoch(device_list.lock().0);
+5 -4
View File
@@ -2,9 +2,10 @@ use std::collections::HashMap;
use std::net::Ipv4Addr; use std::net::Ipv4Addr;
use std::sync::mpsc::{sync_channel, Receiver, SyncSender}; use std::sync::mpsc::{sync_channel, Receiver, SyncSender};
use std::sync::Arc; use std::sync::Arc;
use std::thread;
use std::time::Duration; use std::time::Duration;
use std::{io, thread};
use anyhow::anyhow;
use crossbeam_utils::atomic::AtomicCell; use crossbeam_utils::atomic::AtomicCell;
use parking_lot::Mutex; use parking_lot::Mutex;
use protobuf::Message; use protobuf::Message;
@@ -222,7 +223,7 @@ fn punch0(
punch_record: &Mutex<HashMap<Ipv4Addr, usize>>, punch_record: &Mutex<HashMap<Ipv4Addr, usize>>,
last_punch_record: &mut HashMap<Ipv4Addr, usize>, last_punch_record: &mut HashMap<Ipv4Addr, usize>,
total_count: usize, total_count: usize,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let nat_info = nat_test.nat_info(); let nat_info = nat_test.nat_info();
if total_count < 10 if total_count < 10
&& (nat_info.public_ips.is_empty() && (nat_info.public_ips.is_empty()
@@ -297,7 +298,7 @@ fn punch_packet(
virtual_ip: Ipv4Addr, virtual_ip: Ipv4Addr,
nat_info: &NatInfo, nat_info: &NatInfo,
dest: Ipv4Addr, dest: Ipv4Addr,
) -> io::Result<NetPacket<Vec<u8>>> { ) -> anyhow::Result<NetPacket<Vec<u8>>> {
let mut punch_reply = PunchInfo::new(); let mut punch_reply = PunchInfo::new();
punch_reply.reply = false; punch_reply.reply = false;
punch_reply.public_ip_list = nat_info punch_reply.public_ip_list = nat_info
@@ -320,7 +321,7 @@ fn punch_packet(
log::info!("请求打洞={:?}", punch_reply); log::info!("请求打洞={:?}", punch_reply);
let bytes = punch_reply let bytes = punch_reply
.write_to_bytes() .write_to_bytes()
.map_err(|e| io::Error::new(io::ErrorKind::Other, format!("punch_packet {:?}", e)))?; .map_err(|e| anyhow!("punch_packet {:?}", e))?;
let mut net_packet = NetPacket::new_encrypt(vec![0u8; 12 + bytes.len() + ENCRYPTION_RESERVED])?; let mut net_packet = NetPacket::new_encrypt(vec![0u8; 12 + bytes.len() + ENCRYPTION_RESERVED])?;
net_packet.set_default_version(); net_packet.set_default_version();
net_packet.set_protocol(Protocol::OtherTurn); net_packet.set_protocol(Protocol::OtherTurn);
+14 -15
View File
@@ -1,13 +1,16 @@
use parking_lot::RwLock; use anyhow::anyhow;
use protobuf::Message;
use std::collections::HashMap; use std::collections::HashMap;
use std::io;
use std::net::{Ipv4Addr, Ipv6Addr}; use std::net::{Ipv4Addr, Ipv6Addr};
use std::sync::Arc; use std::sync::Arc;
use parking_lot::RwLock;
use protobuf::Message;
use packet::icmp::{icmp, Kind}; use packet::icmp::{icmp, Kind};
use packet::ip::ipv4; use packet::ip::ipv4;
use packet::ip::ipv4::packet::IpV4Packet; use packet::ip::ipv4::packet::IpV4Packet;
#[cfg(any(target_os = "windows", target_os = "linux", target_os = "macos"))]
use tun::device::IFace;
use crate::channel::context::ChannelContext; use crate::channel::context::ChannelContext;
use crate::channel::punch::NatInfo; use crate::channel::punch::NatInfo;
@@ -28,8 +31,6 @@ use crate::protocol::{
control_packet, ip_turn_packet, other_turn_packet, NetPacket, Protocol, MAX_TTL, control_packet, ip_turn_packet, other_turn_packet, NetPacket, Protocol, MAX_TTL,
}; };
use crate::tun_tap_device::tun_create_helper::DeviceAdapter; use crate::tun_tap_device::tun_create_helper::DeviceAdapter;
#[cfg(any(target_os = "windows", target_os = "linux", target_os = "macos"))]
use tun::device::IFace;
/// 处理来源于客户端的包 /// 处理来源于客户端的包
#[derive(Clone)] #[derive(Clone)]
@@ -116,7 +117,7 @@ impl ClientPacketHandler {
context: &ChannelContext, context: &ChannelContext,
current_device: &CurrentDeviceInfo, current_device: &CurrentDeviceInfo,
route_key: RouteKey, route_key: RouteKey,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let destination = net_packet.destination(); let destination = net_packet.destination();
let source = net_packet.source(); let source = net_packet.source();
match ip_turn_packet::Protocol::from(net_packet.transport_protocol()) { match ip_turn_packet::Protocol::from(net_packet.transport_protocol()) {
@@ -203,7 +204,7 @@ impl ClientPacketHandler {
current_device: &CurrentDeviceInfo, current_device: &CurrentDeviceInfo,
mut net_packet: NetPacket<&mut [u8]>, mut net_packet: NetPacket<&mut [u8]>,
route_key: RouteKey, route_key: RouteKey,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let metric = net_packet.source_ttl() - net_packet.ttl() + 1; let metric = net_packet.source_ttl() - net_packet.ttl() + 1;
let source = net_packet.source(); let source = net_packet.source();
match ControlPacket::new(net_packet.transport_protocol(), net_packet.payload())? { match ControlPacket::new(net_packet.transport_protocol(), net_packet.payload())? {
@@ -291,17 +292,15 @@ impl ClientPacketHandler {
current_device: &CurrentDeviceInfo, current_device: &CurrentDeviceInfo,
net_packet: NetPacket<&mut [u8]>, net_packet: NetPacket<&mut [u8]>,
route_key: RouteKey, route_key: RouteKey,
) -> io::Result<()> { ) -> anyhow::Result<()> {
if context.use_channel_type().is_only_relay() { if context.use_channel_type().is_only_relay() {
return Ok(()); return Ok(());
} }
let source = net_packet.source(); let source = net_packet.source();
match other_turn_packet::Protocol::from(net_packet.transport_protocol()) { match other_turn_packet::Protocol::from(net_packet.transport_protocol()) {
other_turn_packet::Protocol::Punch => { other_turn_packet::Protocol::Punch => {
let mut punch_info = let mut punch_info = PunchInfo::parse_from_bytes(net_packet.payload())
PunchInfo::parse_from_bytes(net_packet.payload()).map_err(|e| { .map_err(|e| anyhow!("PunchInfo {:?}", e))?;
io::Error::new(io::ErrorKind::Other, format!("PunchInfo {:?}", e))
})?;
let public_ips = punch_info let public_ips = punch_info
.public_ip_list .public_ip_list
.iter() .iter()
@@ -361,9 +360,9 @@ impl ClientPacketHandler {
punch_reply.ipv6 = ipv6.octets().to_vec(); punch_reply.ipv6 = ipv6.octets().to_vec();
punch_reply.ipv6_port = nat_info.udp_ports[0] as u32; punch_reply.ipv6_port = nat_info.udp_ports[0] as u32;
} }
let bytes = punch_reply.write_to_bytes().map_err(|e| { let bytes = punch_reply
io::Error::new(io::ErrorKind::Other, format!("punch_reply {:?}", e)) .write_to_bytes()
})?; .map_err(|e| anyhow!("punch_reply {:?}", e))?;
let mut punch_packet = let mut punch_packet =
NetPacket::new_encrypt(vec![0u8; 12 + bytes.len() + ENCRYPTION_RESERVED])?; NetPacket::new_encrypt(vec![0u8; 12 + bytes.len() + ENCRYPTION_RESERVED])?;
punch_packet.set_default_version(); punch_packet.set_default_version();
+6 -1
View File
@@ -62,7 +62,12 @@ impl<Call: VntCallback> RecvChannelHandler for RecvDataHandler<Call> {
} }
} }
if let Err(e) = self.handle0(buf, extend, route_key, context) { if let Err(e) = self.handle0(buf, extend, route_key, context) {
log::error!("[{}]-{:?}", thread::current().name().unwrap_or(""), e); log::error!(
"[{}]-{:?}-{:?}",
thread::current().name().unwrap_or(""),
route_key.addr,
e
);
} }
} }
} }
+10 -10
View File
@@ -1,3 +1,4 @@
use anyhow::anyhow;
use std::io; use std::io;
use std::net::Ipv4Addr; use std::net::Ipv4Addr;
use std::sync::Arc; use std::sync::Arc;
@@ -11,6 +12,8 @@ use protobuf::Message;
use packet::icmp::{icmp, Kind}; use packet::icmp::{icmp, Kind};
use packet::ip::ipv4; use packet::ip::ipv4;
use packet::ip::ipv4::packet::IpV4Packet; use packet::ip::ipv4::packet::IpV4Packet;
#[cfg(any(target_os = "windows", target_os = "linux", target_os = "macos"))]
use tun::device::IFace;
use crate::channel::context::ChannelContext; use crate::channel::context::ChannelContext;
use crate::channel::{Route, RouteKey}; use crate::channel::{Route, RouteKey};
@@ -34,8 +37,6 @@ use crate::protocol::error_packet::InErrorPacket;
use crate::protocol::{ip_turn_packet, service_packet, NetPacket, Protocol, MAX_TTL}; use crate::protocol::{ip_turn_packet, service_packet, NetPacket, Protocol, MAX_TTL};
use crate::tun_tap_device::tun_create_helper::DeviceAdapter; use crate::tun_tap_device::tun_create_helper::DeviceAdapter;
use crate::{proto, PeerClientInfo}; use crate::{proto, PeerClientInfo};
#[cfg(any(target_os = "windows", target_os = "linux", target_os = "macos"))]
use tun::device::IFace;
/// 处理来源于服务端的包 /// 处理来源于服务端的包
#[derive(Clone)] #[derive(Clone)]
@@ -136,10 +137,8 @@ impl<Call: VntCallback> PacketHandler for ServerPacketHandler<Call> {
} else if net_packet.protocol() == Protocol::Service } else if net_packet.protocol() == Protocol::Service
&& net_packet.transport_protocol() == service_packet::Protocol::HandshakeResponse.into() && net_packet.transport_protocol() == service_packet::Protocol::HandshakeResponse.into()
{ {
let response = let response = HandshakeResponse::parse_from_bytes(net_packet.payload())
HandshakeResponse::parse_from_bytes(net_packet.payload()).map_err(|e| { .map_err(|e| anyhow!("HandshakeResponse {:?}", e))?;
io::Error::new(io::ErrorKind::Other, format!("HandshakeResponse {:?}", e))
})?;
log::info!("握手响应:{:?},{}", route_key, response); log::info!("握手响应:{:?},{}", route_key, response);
//如果开启了加密,则发送加密握手请求 //如果开启了加密,则发送加密握手请求
#[cfg(feature = "server_encrypt")] #[cfg(feature = "server_encrypt")]
@@ -254,7 +253,7 @@ impl<Call: VntCallback> ServerPacketHandler<Call> {
current_device: &CurrentDeviceInfo, current_device: &CurrentDeviceInfo,
net_packet: NetPacket<&mut [u8]>, net_packet: NetPacket<&mut [u8]>,
route_key: RouteKey, route_key: RouteKey,
) -> io::Result<()> { ) -> anyhow::Result<()> {
match service_packet::Protocol::from(net_packet.transport_protocol()) { match service_packet::Protocol::from(net_packet.transport_protocol()) {
service_packet::Protocol::RegistrationResponse => { service_packet::Protocol::RegistrationResponse => {
let response = RegistrationResponse::parse_from_bytes(net_packet.payload()) let response = RegistrationResponse::parse_from_bytes(net_packet.payload())
@@ -440,7 +439,7 @@ impl<Call: VntCallback> ServerPacketHandler<Call> {
&self, &self,
current_device: &CurrentDeviceInfo, current_device: &CurrentDeviceInfo,
context: &ChannelContext, context: &ChannelContext,
) -> io::Result<()> { ) -> anyhow::Result<()> {
if current_device.status.online() { if current_device.status.online() {
log::info!("已连接的不需要注册,{:?}", self.config_info); log::info!("已连接的不需要注册,{:?}", self.config_info);
return Ok(()); return Ok(());
@@ -469,7 +468,8 @@ impl<Call: VntCallback> ServerPacketHandler<Call> {
)?; )?;
log::info!("发送注册请求,{:?}", self.config_info); log::info!("发送注册请求,{:?}", self.config_info);
//注册请求只发送到默认通道 //注册请求只发送到默认通道
context.send_default(response.buffer(), current_device.connect_server) context.send_default(response.buffer(), current_device.connect_server)?;
Ok(())
} }
fn error( fn error(
&self, &self,
@@ -527,7 +527,7 @@ impl<Call: VntCallback> ServerPacketHandler<Call> {
current_device: &CurrentDeviceInfo, current_device: &CurrentDeviceInfo,
net_packet: NetPacket<&mut [u8]>, net_packet: NetPacket<&mut [u8]>,
route_key: RouteKey, route_key: RouteKey,
) -> io::Result<()> { ) -> anyhow::Result<()> {
match ControlPacket::new(net_packet.transport_protocol(), net_packet.payload())? { match ControlPacket::new(net_packet.transport_protocol(), net_packet.payload())? {
ControlPacket::PongPacket(pong_packet) => { ControlPacket::PongPacket(pong_packet) => {
let current_time = crate::handle::now_time() as u16; let current_time = crate::handle::now_time() as u16;
+5 -5
View File
@@ -1,4 +1,4 @@
use std::io; use anyhow::anyhow;
use std::net::Ipv4Addr; use std::net::Ipv4Addr;
use protobuf::Message; use protobuf::Message;
@@ -19,7 +19,7 @@ pub fn registration_request_packet(
is_fast: bool, is_fast: bool,
allow_ip_change: bool, allow_ip_change: bool,
client_secret_hash: Option<&[u8]>, client_secret_hash: Option<&[u8]>,
) -> io::Result<NetPacket<Vec<u8>>> { ) -> anyhow::Result<NetPacket<Vec<u8>>> {
let mut request = RegistrationRequest::new(); let mut request = RegistrationRequest::new();
request.token = token; request.token = token;
request.device_id = device_id; request.device_id = device_id;
@@ -36,9 +36,9 @@ pub fn registration_request_packet(
.client_secret_hash .client_secret_hash
.extend_from_slice(client_secret_hash); .extend_from_slice(client_secret_hash);
} }
let bytes = request.write_to_bytes().map_err(|e| { let bytes = request
io::Error::new(io::ErrorKind::Other, format!("RegistrationRequest {:?}", e)) .write_to_bytes()
})?; .map_err(|e| anyhow!("RegistrationRequest {:?}", e))?;
let buf = vec![0u8; 12 + bytes.len() + ENCRYPTION_RESERVED]; let buf = vec![0u8; 12 + bytes.len() + ENCRYPTION_RESERVED];
let mut net_packet = NetPacket::new_encrypt(buf)?; let mut net_packet = NetPacket::new_encrypt(buf)?;
net_packet.set_destination(GATEWAY_IP); net_packet.set_destination(GATEWAY_IP);
+50 -74
View File
@@ -5,7 +5,6 @@ use std::{io, thread};
use crossbeam_utils::atomic::AtomicCell; use crossbeam_utils::atomic::AtomicCell;
use parking_lot::Mutex; use parking_lot::Mutex;
use crate::channel::BUFFER_SIZE;
use packet::icmp::icmp::IcmpPacket; use packet::icmp::icmp::IcmpPacket;
use packet::icmp::Kind; use packet::icmp::Kind;
use packet::ip::ipv4::packet::IpV4Packet; use packet::ip::ipv4::packet::IpV4Packet;
@@ -14,6 +13,7 @@ use tun::device::IFace;
use tun::Device; use tun::Device;
use crate::channel::context::ChannelContext; use crate::channel::context::ChannelContext;
use crate::channel::BUFFER_SIZE;
use crate::cipher::Cipher; use crate::cipher::Cipher;
use crate::compression::Compressor; use crate::compression::Compressor;
use crate::external_route::ExternalRoute; use crate::external_route::ExternalRoute;
@@ -45,47 +45,6 @@ fn icmp(device_writer: &Device, mut ipv4_packet: IpV4Packet<&mut [u8]>) -> anyho
Ok(()) Ok(())
} }
/// 接收tun数据,并且转发到udp上
pub(crate) fn handle(
context: &ChannelContext,
data: &mut [u8],
len: usize,
extend: &mut [u8],
device_writer: &Device,
current_device: CurrentDeviceInfo,
ip_route: &ExternalRoute,
#[cfg(feature = "ip_proxy")] proxy_map: &Option<IpProxyMap>,
client_cipher: &Cipher,
server_cipher: &Cipher,
device_list: &Mutex<(u16, Vec<PeerDeviceInfo>)>,
compressor: &Compressor,
) -> anyhow::Result<()> {
//忽略掉结构不对的情况(ipv6数据、win tap会读到空数据),不然日志打印太多了
let ipv4_packet = match IpV4Packet::new(&mut data[12..len]) {
Ok(packet) => packet,
Err(_) => return Ok(()),
};
let src_ip = ipv4_packet.source_ip();
let dest_ip = ipv4_packet.destination_ip();
if src_ip == dest_ip {
return icmp(&device_writer, ipv4_packet);
}
return base_handle(
context,
data,
len,
extend,
current_device,
ip_route,
#[cfg(feature = "ip_proxy")]
proxy_map,
client_cipher,
server_cipher,
device_list,
compressor,
);
}
pub fn start( pub fn start(
stop_manager: StopManager, stop_manager: StopManager,
context: ChannelContext, context: ChannelContext,
@@ -187,7 +146,7 @@ fn broadcast(
net_packet: &mut NetPacket<&mut [u8]>, net_packet: &mut NetPacket<&mut [u8]>,
current_device: &CurrentDeviceInfo, current_device: &CurrentDeviceInfo,
device_list: &Mutex<(u16, Vec<PeerDeviceInfo>)>, device_list: &Mutex<(u16, Vec<PeerDeviceInfo>)>,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let list: Vec<Ipv4Addr> = device_list let list: Vec<Ipv4Addr> = device_list
.lock() .lock()
.1 .1
@@ -261,18 +220,20 @@ fn broadcast(
broadcast.set_address(&p2p_ips)?; broadcast.set_address(&p2p_ips)?;
broadcast.set_data(net_packet.buffer())?; broadcast.set_data(net_packet.buffer())?;
server_cipher.encrypt_ipv4(&mut server_packet)?; server_cipher.encrypt_ipv4(&mut server_packet)?;
sender.send_default(server_packet.buffer(), current_device.connect_server) sender.send_default(server_packet.buffer(), current_device.connect_server)?;
Ok(())
} }
/// 接收tun数据,并且转发到udp上
/// 实现一个原地发送,必须保证是如下结构 /// 实现一个原地发送,必须保证是如下结构
/// |12字节开头|ip报文|至少1024字节结尾| /// |12字节开头|ip报文|至少1024字节结尾|
/// ///
#[inline] pub(crate) fn handle(
fn base_handle(
context: &ChannelContext, context: &ChannelContext,
buf: &mut [u8], buf: &mut [u8],
data_len: usize, //数据总长度=12+ip包长度 data_len: usize, //数据总长度=12+ip包长度
extend: &mut [u8], extend: &mut [u8],
device_writer: &Device,
current_device: CurrentDeviceInfo, current_device: CurrentDeviceInfo,
ip_route: &ExternalRoute, ip_route: &ExternalRoute,
#[cfg(feature = "ip_proxy")] proxy_map: &Option<IpProxyMap>, #[cfg(feature = "ip_proxy")] proxy_map: &Option<IpProxyMap>,
@@ -281,7 +242,16 @@ fn base_handle(
device_list: &Mutex<(u16, Vec<PeerDeviceInfo>)>, device_list: &Mutex<(u16, Vec<PeerDeviceInfo>)>,
compressor: &Compressor, compressor: &Compressor,
) -> anyhow::Result<()> { ) -> anyhow::Result<()> {
let ipv4_packet = IpV4Packet::new(&buf[12..data_len])?; //忽略掉结构不对的情况(ipv6数据、win tap会读到空数据),不然日志打印太多了
let ipv4_packet = match IpV4Packet::new(&mut buf[12..data_len]) {
Ok(packet) => packet,
Err(_) => return Ok(()),
};
let src_ip = ipv4_packet.source_ip();
let dest_ip = ipv4_packet.destination_ip();
if src_ip == dest_ip {
return icmp(&device_writer, ipv4_packet);
}
let protocol = ipv4_packet.protocol(); let protocol = ipv4_packet.protocol();
let src_ip = ipv4_packet.source_ip(); let src_ip = ipv4_packet.source_ip();
let mut dest_ip = ipv4_packet.destination_ip(); let mut dest_ip = ipv4_packet.destination_ip();
@@ -302,6 +272,38 @@ fn base_handle(
} }
return Ok(()); return Ok(());
} }
if !dest_ip.is_multicast() && !dest_ip.is_broadcast() && current_device.broadcast_ip != dest_ip
{
if !check_dest(
dest_ip,
current_device.virtual_netmask,
current_device.virtual_network,
) {
if let Some(r_dest_ip) = ip_route.route(&dest_ip) {
//路由的目标不能是自己
if r_dest_ip == src_ip {
return Ok(());
}
//需要修改目的地址
dest_ip = r_dest_ip;
net_packet.set_destination(r_dest_ip);
} else {
return Ok(());
}
}
#[cfg(feature = "ip_proxy")]
if let Some(proxy_map) = proxy_map {
let mut ipv4_packet = IpV4Packet::new(net_packet.payload_mut())?;
proxy_map.send_handle(&mut ipv4_packet)?;
}
}
if dest_ip.is_multicast() {
//当作广播处理
dest_ip = Ipv4Addr::BROADCAST;
net_packet.set_destination(Ipv4Addr::BROADCAST);
}
let mut net_packet = if compressor.compress(&net_packet, &mut out)? { let mut net_packet = if compressor.compress(&net_packet, &mut out)? {
out.set_default_version(); out.set_default_version();
out.set_protocol(protocol::Protocol::IpTurn); out.set_protocol(protocol::Protocol::IpTurn);
@@ -313,11 +315,6 @@ fn base_handle(
} else { } else {
net_packet net_packet
}; };
if dest_ip.is_multicast() {
//当作广播处理
dest_ip = Ipv4Addr::BROADCAST;
net_packet.set_destination(Ipv4Addr::BROADCAST);
}
if dest_ip.is_broadcast() || current_device.broadcast_ip == dest_ip { if dest_ip.is_broadcast() || current_device.broadcast_ip == dest_ip {
// 广播 发送到直连目标 // 广播 发送到直连目标
client_cipher.encrypt_ipv4(&mut net_packet)?; client_cipher.encrypt_ipv4(&mut net_packet)?;
@@ -330,28 +327,7 @@ fn base_handle(
)?; )?;
return Ok(()); return Ok(());
} }
if !check_dest(
dest_ip,
current_device.virtual_netmask,
current_device.virtual_network,
) {
if let Some(r_dest_ip) = ip_route.route(&dest_ip) {
//路由的目标不能是自己
if r_dest_ip == src_ip {
return Ok(());
}
//需要修改目的地址
dest_ip = r_dest_ip;
net_packet.set_destination(r_dest_ip);
} else {
return Ok(());
}
}
#[cfg(feature = "ip_proxy")]
if let Some(proxy_map) = proxy_map {
let mut ipv4_packet = IpV4Packet::new(net_packet.payload_mut())?;
proxy_map.send_handle(&mut ipv4_packet)?;
}
client_cipher.encrypt_ipv4(&mut net_packet)?; client_cipher.encrypt_ipv4(&mut net_packet)?;
context.send_ipv4_by_id( context.send_ipv4_by_id(
net_packet.buffer(), net_packet.buffer(),
+4 -4
View File
@@ -33,7 +33,7 @@ pub(crate) fn start_simple(
up_counter: &mut SingleU64Adder, up_counter: &mut SingleU64Adder,
device_list: Arc<Mutex<(u16, Vec<PeerDeviceInfo>)>>, device_list: Arc<Mutex<(u16, Vec<PeerDeviceInfo>)>>,
compressor: Compressor, compressor: Compressor,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let poll = Poll::new()?; let poll = Poll::new()?;
let waker = Arc::new(Waker::new(poll.registry(), STOP)?); let waker = Arc::new(Waker::new(poll.registry(), STOP)?);
let _waker = waker.clone(); let _waker = waker.clone();
@@ -73,7 +73,7 @@ fn start_simple0(
up_counter: &mut SingleU64Adder, up_counter: &mut SingleU64Adder,
device_list: Arc<Mutex<(u16, Vec<PeerDeviceInfo>)>>, device_list: Arc<Mutex<(u16, Vec<PeerDeviceInfo>)>>,
compressor: Compressor, compressor: Compressor,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let mut buf = [0; BUFFER_SIZE]; let mut buf = [0; BUFFER_SIZE];
let mut extend = [0; BUFFER_SIZE]; let mut extend = [0; BUFFER_SIZE];
let fd = device.as_tun_fd(); let fd = device.as_tun_fd();
@@ -134,7 +134,7 @@ pub(crate) fn start_multi(
device: Arc<Device>, device: Arc<Device>,
group_sync_sender: GroupSyncSender<(Vec<u8>, usize)>, group_sync_sender: GroupSyncSender<(Vec<u8>, usize)>,
up_counter: &mut SingleU64Adder, up_counter: &mut SingleU64Adder,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let poll = Poll::new()?; let poll = Poll::new()?;
let waker = Arc::new(Waker::new(poll.registry(), STOP)?); let waker = Arc::new(Waker::new(poll.registry(), STOP)?);
let _waker = waker.clone(); let _waker = waker.clone();
@@ -154,7 +154,7 @@ fn start_multi0(
device: Arc<Device>, device: Arc<Device>,
mut group_sync_sender: GroupSyncSender<(Vec<u8>, usize)>, mut group_sync_sender: GroupSyncSender<(Vec<u8>, usize)>,
up_counter: &mut SingleU64Adder, up_counter: &mut SingleU64Adder,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let fd = device.as_tun_fd(); let fd = device.as_tun_fd();
fd.set_nonblock()?; fd.set_nonblock()?;
SourceFd(&fd.as_raw_fd()).register(poll.registry(), FD, Interest::READABLE)?; SourceFd(&fd.as_raw_fd()).register(poll.registry(), FD, Interest::READABLE)?;
+4 -5
View File
@@ -10,7 +10,6 @@ use crate::ip_proxy::IpProxyMap;
use crate::util::{SingleU64Adder, StopManager}; use crate::util::{SingleU64Adder, StopManager};
use crossbeam_utils::atomic::AtomicCell; use crossbeam_utils::atomic::AtomicCell;
use parking_lot::Mutex; use parking_lot::Mutex;
use std::io;
use std::sync::Arc; use std::sync::Arc;
use tun::device::IFace; use tun::device::IFace;
use tun::Device; use tun::Device;
@@ -27,7 +26,7 @@ pub(crate) fn start_simple(
up_counter: &mut SingleU64Adder, up_counter: &mut SingleU64Adder,
device_list: Arc<Mutex<(u16, Vec<PeerDeviceInfo>)>>, device_list: Arc<Mutex<(u16, Vec<PeerDeviceInfo>)>>,
compressor: Compressor, compressor: Compressor,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let worker = { let worker = {
let device = device.clone(); let device = device.clone();
stop_manager.add_listener("tun_device".into(), move || { stop_manager.add_listener("tun_device".into(), move || {
@@ -65,7 +64,7 @@ fn start_simple0(
up_counter: &mut SingleU64Adder, up_counter: &mut SingleU64Adder,
device_list: Arc<Mutex<(u16, Vec<PeerDeviceInfo>)>>, device_list: Arc<Mutex<(u16, Vec<PeerDeviceInfo>)>>,
compressor: Compressor, compressor: Compressor,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let mut buf = [0; BUFFER_SIZE]; let mut buf = [0; BUFFER_SIZE];
let mut extend = [0; BUFFER_SIZE]; let mut extend = [0; BUFFER_SIZE];
loop { loop {
@@ -101,7 +100,7 @@ pub(crate) fn start_multi(
device: Arc<Device>, device: Arc<Device>,
group_sync_sender: GroupSyncSender<(Vec<u8>, usize)>, group_sync_sender: GroupSyncSender<(Vec<u8>, usize)>,
up_counter: &mut SingleU64Adder, up_counter: &mut SingleU64Adder,
) -> io::Result<()> { ) -> anyhow::Result<()> {
let worker = { let worker = {
let device = device.clone(); let device = device.clone();
stop_manager.add_listener("tun_device_multi".into(), move || { stop_manager.add_listener("tun_device_multi".into(), move || {
@@ -120,7 +119,7 @@ fn start_multi0(
device: Arc<Device>, device: Arc<Device>,
mut group_sync_sender: GroupSyncSender<(Vec<u8>, usize)>, mut group_sync_sender: GroupSyncSender<(Vec<u8>, usize)>,
up_counter: &mut SingleU64Adder, up_counter: &mut SingleU64Adder,
) -> io::Result<()> { ) -> anyhow::Result<()> {
loop { loop {
let mut buf = vec![0; 1024 * 16]; let mut buf = vec![0; 1024 * 16];
let len = device.read(&mut buf[12..])? + 12; let len = device.read(&mut buf[12..])? + 12;
+79
View File
@@ -253,6 +253,85 @@ impl<B: AsRef<[u8]> + AsMut<[u8]>> AesCbcSecretBody<B> {
&mut self.buffer.as_mut()[..end] &mut self.buffer.as_mut()[..end]
} }
} }
/* ChaCah20加密数据体
0 15 31
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| 数据体 |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| finger(32) |
| finger(32) |
| finger(32) |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
注:finger用于快速校验数据是否被修改,上层可使用token、协议头参与计算finger
确保服务端和客户端都能感知修改(服务端不能解密也能校验指纹)
*/
pub struct ChaCah20SecretBody<B> {
buffer: B,
exist_finger: bool,
}
impl<B: AsRef<[u8]>> ChaCah20SecretBody<B> {
pub fn new(buffer: B, exist_finger: bool) -> io::Result<ChaCah20SecretBody<B>> {
let len = buffer.as_ref().len();
let min_len = if exist_finger { 12 } else { 0 };
// 不能大于udp最大载荷长度
if len < min_len || len > 65535 - 20 - 8 - 12 {
return Err(io::Error::new(
io::ErrorKind::InvalidData,
"ChaCah20SecretBody length overflow",
));
}
Ok(ChaCah20SecretBody {
buffer,
exist_finger,
})
}
pub fn en_body(&self) -> &[u8] {
let mut end = self.buffer.as_ref().len();
if self.exist_finger {
end -= 12;
}
&self.buffer.as_ref()[..end]
}
pub fn finger(&self) -> &[u8] {
if self.exist_finger {
let end = self.buffer.as_ref().len();
&self.buffer.as_ref()[end - 12..end]
} else {
&[]
}
}
}
impl<B: AsRef<[u8]> + AsMut<[u8]>> ChaCah20SecretBody<B> {
pub fn set_finger(&mut self, finger: &[u8]) -> io::Result<()> {
if self.exist_finger {
if finger.len() != 12 {
return Err(io::Error::new(
io::ErrorKind::InvalidData,
"finger.len != 12",
));
}
let end = self.buffer.as_ref().len();
self.buffer.as_mut()[end - 12..end].copy_from_slice(finger);
Ok(())
} else {
Err(io::Error::new(
io::ErrorKind::InvalidData,
"cbc not exist finger",
))
}
}
pub fn en_body_mut(&mut self) -> &mut [u8] {
let mut end = self.buffer.as_ref().len();
if self.exist_finger {
end -= 12;
}
&mut self.buffer.as_mut()[..end]
}
}
/* rsa加密数据体 /* rsa加密数据体
0 15 31 0 15 31
+7 -9
View File
@@ -1,9 +1,10 @@
use std::sync::atomic::{AtomicBool, AtomicUsize, Ordering}; use std::sync::atomic::{AtomicBool, AtomicUsize, Ordering};
use std::sync::Arc; use std::sync::Arc;
use std::thread;
use std::thread::Thread; use std::thread::Thread;
use std::time::Duration; use std::time::Duration;
use std::{io, thread};
use anyhow::anyhow;
use parking_lot::Mutex; use parking_lot::Mutex;
#[derive(Clone)] #[derive(Clone)]
@@ -20,7 +21,7 @@ impl StopManager {
inner: Arc::new(StopManagerInner::new(f)), inner: Arc::new(StopManagerInner::new(f)),
} }
} }
pub fn add_listener<F>(&self, name: String, f: F) -> io::Result<Worker> pub fn add_listener<F>(&self, name: String, f: F) -> anyhow::Result<Worker>
where where
F: FnOnce() + Send + 'static, F: FnOnce() + Send + 'static,
{ {
@@ -61,23 +62,20 @@ impl StopManagerInner {
stop_call: Mutex::new(Some(Box::new(f))), stop_call: Mutex::new(Some(Box::new(f))),
} }
} }
fn add_listener<F>(self: &Arc<Self>, name: String, f: F) -> io::Result<Worker> fn add_listener<F>(self: &Arc<Self>, name: String, f: F) -> anyhow::Result<Worker>
where where
F: FnOnce() + Send + 'static, F: FnOnce() + Send + 'static,
{ {
if name.is_empty() { if name.is_empty() {
return Err(io::Error::new(io::ErrorKind::Other, "name cannot be empty")); return Err(anyhow!("name cannot be empty"));
} }
let mut guard = self.listeners.lock(); let mut guard = self.listeners.lock();
if guard.0 { if guard.0 {
return Err(io::Error::new(io::ErrorKind::Other, "stopped")); return Err(anyhow!("stopped"));
} }
for (n, _) in &guard.1 { for (n, _) in &guard.1 {
if &name == n { if &name == n {
return Err(io::Error::new( return Err(anyhow!("stop add_listener {:?} name already exists", name));
io::ErrorKind::Other,
format!("stop add_listener {:?} name already exists", name),
));
} }
} }
guard.1.push((name.clone(), Box::new(f))); guard.1.push((name.clone(), Box::new(f)));
+1 -2
View File
@@ -2,7 +2,6 @@ use crate::util::StopManager;
use std::collections::BinaryHeap; use std::collections::BinaryHeap;
use std::{ use std::{
cmp::Ordering, cmp::Ordering,
io,
sync::mpsc::{sync_channel, Receiver, SyncSender}, sync::mpsc::{sync_channel, Receiver, SyncSender},
time::{Duration, Instant}, time::{Duration, Instant},
}; };
@@ -36,7 +35,7 @@ pub struct Scheduler {
sender: SyncSender<Op>, sender: SyncSender<Op>,
} }
impl Scheduler { impl Scheduler {
pub fn new(stop_manager: StopManager) -> io::Result<Self> { pub fn new(stop_manager: StopManager) -> anyhow::Result<Self> {
let (sender, receiver) = sync_channel::<Op>(32); let (sender, receiver) = sync_channel::<Op>(32);
let s = Self { sender }; let s = Self { sender };
let s_inner = s.clone(); let s_inner = s.clone();