jostreff/wireguard-ui
1
0
Fork 0
mirror of https://github.com/ngoduykhanh/wireguard-ui.git synced 2025-06-07 00:46:58 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

raise an error if duplicate key is present

Browse source
This commit is contained in:
Maxim Kochurov 2022-03-13 10:25:54 +00:00
parent 5a6a19c661
commit 94aaef70a7
1 changed files with 15 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

17
handler/routes.go
View file

@ -177,14 +177,27 @@ func NewClient(db store.IStore) echo.HandlerFunc {
log.Error("Cannot generate wireguard key pair: ", err) log.Error("Cannot generate wireguard key pair: ", err)
return c.JSON(http.StatusInternalServerError, jsonHTTPResponse{false, "Cannot generate Wireguard key pair"}) return c.JSON(http.StatusInternalServerError, jsonHTTPResponse{false, "Cannot generate Wireguard key pair"})
} }
client.PrivateKey = key.String() client.PrivateKey = key.String()
client.PublicKey = key.PublicKey().String() client.PublicKey = key.PublicKey().String()
} else { } else {
_, err := wgtypes.ParseKey(client.PublicKey) _, err := wgtypes.ParseKey(client.PublicKey)
if err != nil { if err != nil {
log.Error("Cannot verify wireguard public key: ", err) log.Error("Cannot verify wireguard public key: ", err)
return c.JSON(http.StatusInternalServerError, jsonHTTPResponse{false, "Cannot verify Wireguard public key"}) return c.JSON(http.StatusInternalServerError, jsonHTTPResponse{false, "Cannot verify Wireguard public key"})
} }
// check for duplicates
clients, err := db.GetClients(false)
if err != nil {
log.Error("Cannot get clients for duplicate check")
return c.JSON(http.StatusInternalServerError, jsonHTTPResponse{false, "Cannot get clients for duplicate check"})
}
for _, other := range clients {
if other.Client.PublicKey == client.PublicKey {
log.Error("Duplicate Public Key")
return c.JSON(http.StatusInternalServerError, jsonHTTPResponse{false, "Duplicate Public Key"})
}
}
} }
if client.PresharedKey == "" { if client.PresharedKey == "" {