mirror of
https://github.com/whyvl/wireproxy.git
synced 2025-04-29 10:51:42 +02:00
87 lines
1.8 KiB
Go
87 lines
1.8 KiB
Go
package wireproxy
|
|
|
|
import (
|
|
"github.com/go-ini/ini"
|
|
"testing"
|
|
)
|
|
|
|
func loadIniConfig(config string) (*ini.File, error) {
|
|
iniOpt := ini.LoadOptions{
|
|
Insensitive: true,
|
|
AllowShadows: true,
|
|
AllowNonUniqueSections: true,
|
|
}
|
|
|
|
return ini.LoadSources(iniOpt, []byte(config))
|
|
}
|
|
|
|
func TestWireguardConfWithoutSubnet(t *testing.T) {
|
|
const config = `
|
|
[Interface]
|
|
PrivateKey = LAr1aNSNF9d0MjwUgAVC4020T0N/E5NUtqVv5EnsSz0=
|
|
Address = 10.5.0.2
|
|
DNS = 1.1.1.1
|
|
|
|
[Peer]
|
|
PublicKey = e8LKAc+f9xEzq9Ar7+MfKRrs+gZ/4yzvpRJLRJ/VJ1w=
|
|
AllowedIPs = 0.0.0.0/0, ::/0
|
|
Endpoint = 94.140.11.15:51820
|
|
PersistentKeepalive = 25`
|
|
var cfg DeviceConfig
|
|
iniData, err := loadIniConfig(config)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = ParseInterface(iniData, &cfg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestWireguardConfWithSubnet(t *testing.T) {
|
|
const config = `
|
|
[Interface]
|
|
PrivateKey = LAr1aNSNF9d0MjwUgAVC4020T0N/E5NUtqVv5EnsSz0=
|
|
Address = 10.5.0.2/23
|
|
DNS = 1.1.1.1
|
|
|
|
[Peer]
|
|
PublicKey = e8LKAc+f9xEzq9Ar7+MfKRrs+gZ/4yzvpRJLRJ/VJ1w=
|
|
AllowedIPs = 0.0.0.0/0, ::/0
|
|
Endpoint = 94.140.11.15:51820
|
|
PersistentKeepalive = 25`
|
|
var cfg DeviceConfig
|
|
iniData, err := loadIniConfig(config)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = ParseInterface(iniData, &cfg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestWireguardConfWithManyAddress(t *testing.T) {
|
|
const config = `
|
|
[Interface]
|
|
PrivateKey = mBsVDahr1XIu9PPd17UmsDdB6E53nvmS47NbNqQCiFM=
|
|
Address = 100.96.0.190,2606:B300:FFFF:fe8a:2ac6:c7e8:b021:6f5f/128
|
|
DNS = 198.18.0.1,198.18.0.2
|
|
|
|
[Peer]
|
|
PublicKey = SHnh4C2aDXhp1gjIqceGhJrhOLSeNYcqWLKcYnzj00U=
|
|
AllowedIPs = 0.0.0.0/0,::/0
|
|
Endpoint = 192.200.144.22:51820`
|
|
var cfg DeviceConfig
|
|
iniData, err := loadIniConfig(config)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = ParseInterface(iniData, &cfg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|