first commit
This commit is contained in:
yann
280
core/client.go
Normal file
280
core/client.go
Normal file
@@ -0,0 +1,280 @@
|
||||
package core
|
||||
|
||||
import (
|
||||
"errors"
|
||||
uuid "github.com/satori/go.uuid"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"github.com/skip2/go-qrcode"
|
||||
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/model"
|
||||
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/storage"
|
||||
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/template"
|
||||
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/util"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
"gopkg.in/gomail.v2"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sort"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
// CreateClient client with all necessary data
|
||||
func CreateClient(client *model.Client) (*model.Client, error) {
|
||||
// check if client is valid
|
||||
errs := client.IsValid()
|
||||
if len(errs) != 0 {
|
||||
for _, err := range errs {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Error("client validation error")
|
||||
}
|
||||
return nil, errors.New("failed to validate client")
|
||||
}
|
||||
|
||||
u := uuid.NewV4()
|
||||
client.Id = u.String()
|
||||
|
||||
key, err := wgtypes.GeneratePrivateKey()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
client.PrivateKey = key.String()
|
||||
client.PublicKey = key.PublicKey().String()
|
||||
|
||||
presharedKey, err := wgtypes.GenerateKey()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
client.PresharedKey = presharedKey.String()
|
||||
|
||||
reserverIps, err := GetAllReservedIps()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
ips := make([]string, 0)
|
||||
for _, network := range client.Address {
|
||||
ip, err := util.GetAvailableIp(network, reserverIps)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if util.IsIPv6(ip) {
|
||||
ip = ip + "/128"
|
||||
} else {
|
||||
ip = ip + "/32"
|
||||
}
|
||||
ips = append(ips, ip)
|
||||
}
|
||||
client.Address = ips
|
||||
client.Created = time.Now().UTC()
|
||||
client.Updated = client.Created
|
||||
|
||||
err = storage.Serialize(client.Id, client)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
v, err := storage.Deserialize(client.Id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
client = v.(*model.Client)
|
||||
|
||||
// data modified, dump new config
|
||||
return client, UpdateServerConfigWg()
|
||||
}
|
||||
|
||||
// ReadClient client by id
|
||||
func ReadClient(id string) (*model.Client, error) {
|
||||
v, err := storage.Deserialize(id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
client := v.(*model.Client)
|
||||
|
||||
return client, nil
|
||||
}
|
||||
|
||||
// UpdateClient preserve keys
|
||||
func UpdateClient(Id string, client *model.Client) (*model.Client, error) {
|
||||
v, err := storage.Deserialize(Id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
current := v.(*model.Client)
|
||||
|
||||
if current.Id != client.Id {
|
||||
return nil, errors.New("records Id mismatch")
|
||||
}
|
||||
|
||||
// check if client is valid
|
||||
errs := client.IsValid()
|
||||
if len(errs) != 0 {
|
||||
for _, err := range errs {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Error("client validation error")
|
||||
}
|
||||
return nil, errors.New("failed to validate client")
|
||||
}
|
||||
|
||||
// keep keys
|
||||
client.PrivateKey = current.PrivateKey
|
||||
client.PublicKey = current.PublicKey
|
||||
client.Updated = time.Now().UTC()
|
||||
|
||||
err = storage.Serialize(client.Id, client)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
v, err = storage.Deserialize(Id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
client = v.(*model.Client)
|
||||
|
||||
// data modified, dump new config
|
||||
return client, UpdateServerConfigWg()
|
||||
}
|
||||
|
||||
// DeleteClient from disk
|
||||
func DeleteClient(id string) error {
|
||||
path := filepath.Join(os.Getenv("WG_CONF_DIR"), id)
|
||||
err := os.Remove(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// data modified, dump new config
|
||||
return UpdateServerConfigWg()
|
||||
}
|
||||
|
||||
// ReadClients all clients
|
||||
func ReadClients() ([]*model.Client, error) {
|
||||
clients := make([]*model.Client, 0)
|
||||
|
||||
files, err := ioutil.ReadDir(filepath.Join(os.Getenv("WG_CONF_DIR")))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for _, f := range files {
|
||||
// clients file name is an uuid
|
||||
_, err := uuid.FromString(f.Name())
|
||||
if err == nil {
|
||||
c, err := storage.Deserialize(f.Name())
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
"path": f.Name(),
|
||||
}).Error("failed to deserialize client")
|
||||
} else {
|
||||
clients = append(clients, c.(*model.Client))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
sort.Slice(clients, func(i, j int) bool {
|
||||
return clients[i].Created.After(clients[j].Created)
|
||||
})
|
||||
|
||||
return clients, nil
|
||||
}
|
||||
|
||||
// ReadClientConfig in wg format
|
||||
func ReadClientConfig(id string) ([]byte, error) {
|
||||
client, err := ReadClient(id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
server, err := ReadServer()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
configDataWg, err := template.DumpClientWg(client, server)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return configDataWg, nil
|
||||
}
|
||||
|
||||
// SendEmail to client
|
||||
func EmailClient(id string) error {
|
||||
client, err := ReadClient(id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
configData, err := ReadClientConfig(id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// conf as .conf file
|
||||
tmpfileCfg, err := ioutil.TempFile("", "wireguard-vpn-*.conf")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if _, err := tmpfileCfg.Write(configData); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := tmpfileCfg.Close(); err != nil {
|
||||
return err
|
||||
}
|
||||
defer os.Remove(tmpfileCfg.Name()) // clean up
|
||||
|
||||
// conf as png image
|
||||
png, err := qrcode.Encode(string(configData), qrcode.Medium, 280)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
tmpfilePng, err := ioutil.TempFile("", "qrcode-*.png")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if _, err := tmpfilePng.Write(png); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := tmpfilePng.Close(); err != nil {
|
||||
return err
|
||||
}
|
||||
defer os.Remove(tmpfilePng.Name()) // clean up
|
||||
|
||||
// get email body
|
||||
emailBody, err := template.DumpEmail(client, filepath.Base(tmpfilePng.Name()))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// port to int
|
||||
port, err := strconv.Atoi(os.Getenv("SMTP_PORT"))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
d := gomail.NewDialer(os.Getenv("SMTP_HOST"), port, os.Getenv("SMTP_USERNAME"), os.Getenv("SMTP_PASSWORD"))
|
||||
s, err := d.Dial()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
m := gomail.NewMessage()
|
||||
|
||||
m.SetHeader("From", os.Getenv("SMTP_FROM"))
|
||||
m.SetAddressHeader("To", client.Email, client.Name)
|
||||
m.SetHeader("Subject", "WireGuard VPN Configuration")
|
||||
m.SetBody("text/html", string(emailBody))
|
||||
m.Attach(tmpfileCfg.Name())
|
||||
m.Embed(tmpfilePng.Name())
|
||||
|
||||
err = gomail.Send(s, m)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
309
core/migrate.go
Normal file
309
core/migrate.go
Normal file
@@ -0,0 +1,309 @@
|
||||
package core
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
uuid "github.com/satori/go.uuid"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/model"
|
||||
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/storage"
|
||||
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/util"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// Migrate all changes, current struct fields change
|
||||
func MigrateInitialStructChange() error {
|
||||
clients, err := readClients()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
s, err := deserialize("server.json")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, client := range clients {
|
||||
switch v := client["allowedIPs"].(type) {
|
||||
case []interface{}:
|
||||
log.Infof("client %s has been already migrated", client["id"])
|
||||
continue
|
||||
default:
|
||||
log.Infof("unexpected type %T, mus be migrated", v)
|
||||
}
|
||||
|
||||
c := &model.Client{}
|
||||
c.Id = client["id"].(string)
|
||||
c.Name = client["name"].(string)
|
||||
c.Email = client["email"].(string)
|
||||
c.Enable = client["enable"].(bool)
|
||||
c.AllowedIPs = make([]string, 0)
|
||||
for _, address := range strings.Split(client["allowedIPs"].(string), ",") {
|
||||
if util.IsValidCidr(strings.TrimSpace(address)) {
|
||||
c.AllowedIPs = append(c.AllowedIPs, strings.TrimSpace(address))
|
||||
}
|
||||
}
|
||||
c.Address = make([]string, 0)
|
||||
for _, address := range strings.Split(client["address"].(string), ",") {
|
||||
if util.IsValidCidr(strings.TrimSpace(address)) {
|
||||
c.Address = append(c.Address, strings.TrimSpace(address))
|
||||
}
|
||||
}
|
||||
c.PrivateKey = client["privateKey"].(string)
|
||||
c.PublicKey = client["publicKey"].(string)
|
||||
created, err := time.Parse(time.RFC3339, client["created"].(string))
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to parse time")
|
||||
continue
|
||||
}
|
||||
c.Created = created
|
||||
updated, err := time.Parse(time.RFC3339, client["updated"].(string))
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to parse time")
|
||||
continue
|
||||
}
|
||||
c.Updated = updated
|
||||
|
||||
err = storage.Serialize(c.Id, c)
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to Serialize client")
|
||||
}
|
||||
}
|
||||
|
||||
switch v := s["address"].(type) {
|
||||
case []interface{}:
|
||||
log.Info("server has been already migrated")
|
||||
return nil
|
||||
default:
|
||||
log.Infof("unexpected type %T, mus be migrated", v)
|
||||
}
|
||||
|
||||
server := &model.Server{}
|
||||
|
||||
server.Address = make([]string, 0)
|
||||
for _, address := range strings.Split(s["address"].(string), ",") {
|
||||
if util.IsValidCidr(strings.TrimSpace(address)) {
|
||||
server.Address = append(server.Address, strings.TrimSpace(address))
|
||||
}
|
||||
}
|
||||
server.ListenPort = int(s["listenPort"].(float64))
|
||||
server.PrivateKey = s["privateKey"].(string)
|
||||
server.PublicKey = s["publicKey"].(string)
|
||||
//server.PresharedKey = s["presharedKey"].(string)
|
||||
server.Endpoint = s["endpoint"].(string)
|
||||
server.PersistentKeepalive = int(s["persistentKeepalive"].(float64))
|
||||
server.Dns = make([]string, 0)
|
||||
for _, address := range strings.Split(s["dns"].(string), ",") {
|
||||
if util.IsValidIp(strings.TrimSpace(address)) {
|
||||
server.Dns = append(server.Dns, strings.TrimSpace(address))
|
||||
}
|
||||
}
|
||||
if val, ok := s["preUp"]; ok {
|
||||
server.PreUp = val.(string)
|
||||
}
|
||||
if val, ok := s["postUp"]; ok {
|
||||
server.PostUp = val.(string)
|
||||
}
|
||||
if val, ok := s["preDown"]; ok {
|
||||
server.PreDown = val.(string)
|
||||
}
|
||||
if val, ok := s["postDown"]; ok {
|
||||
server.PostDown = val.(string)
|
||||
}
|
||||
created, err := time.Parse(time.RFC3339, s["created"].(string))
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to parse time")
|
||||
}
|
||||
server.Created = created
|
||||
updated, err := time.Parse(time.RFC3339, s["updated"].(string))
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to parse time")
|
||||
}
|
||||
server.Updated = updated
|
||||
|
||||
err = storage.Serialize("server.json", server)
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to Serialize server")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Migrate presharedKey issue #23
|
||||
func MigratePresharedKey() error {
|
||||
clients, err := readClients()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
s, err := deserialize("server.json")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, client := range clients {
|
||||
if _, ok := client["presharedKey"]; ok {
|
||||
log.Infof("client %s has been already migrated for preshared key", client["id"])
|
||||
continue
|
||||
}
|
||||
|
||||
c := &model.Client{}
|
||||
c.Id = client["id"].(string)
|
||||
c.Name = client["name"].(string)
|
||||
c.Email = client["email"].(string)
|
||||
c.Enable = client["enable"].(bool)
|
||||
if val, ok := client["ignorePersistentKeepalive"]; ok {
|
||||
c.IgnorePersistentKeepalive = val.(bool)
|
||||
} else {
|
||||
c.IgnorePersistentKeepalive = false
|
||||
}
|
||||
c.PresharedKey = s["presharedKey"].(string)
|
||||
c.AllowedIPs = make([]string, 0)
|
||||
for _, address := range client["allowedIPs"].([]interface{}) {
|
||||
c.AllowedIPs = append(c.AllowedIPs, address.(string))
|
||||
}
|
||||
c.Address = make([]string, 0)
|
||||
for _, address := range client["address"].([]interface{}) {
|
||||
c.Address = append(c.Address, address.(string))
|
||||
}
|
||||
c.PrivateKey = client["privateKey"].(string)
|
||||
c.PublicKey = client["publicKey"].(string)
|
||||
created, err := time.Parse(time.RFC3339, client["created"].(string))
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to parse time")
|
||||
continue
|
||||
}
|
||||
c.Created = created
|
||||
updated, err := time.Parse(time.RFC3339, client["updated"].(string))
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to parse time")
|
||||
continue
|
||||
}
|
||||
c.Updated = updated
|
||||
|
||||
err = storage.Serialize(c.Id, c)
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to Serialize client")
|
||||
}
|
||||
}
|
||||
|
||||
if _, ok := s["presharedKey"]; ok {
|
||||
server := &model.Server{}
|
||||
|
||||
server.Address = make([]string, 0)
|
||||
server.Address = make([]string, 0)
|
||||
for _, address := range s["address"].([]interface{}) {
|
||||
server.Address = append(server.Address, address.(string))
|
||||
}
|
||||
server.ListenPort = int(s["listenPort"].(float64))
|
||||
server.PrivateKey = s["privateKey"].(string)
|
||||
server.PublicKey = s["publicKey"].(string)
|
||||
server.Endpoint = s["endpoint"].(string)
|
||||
server.PersistentKeepalive = int(s["persistentKeepalive"].(float64))
|
||||
server.Dns = make([]string, 0)
|
||||
for _, address := range s["dns"].([]interface{}) {
|
||||
server.Dns = append(server.Dns, address.(string))
|
||||
}
|
||||
if val, ok := s["preUp"]; ok {
|
||||
server.PreUp = val.(string)
|
||||
}
|
||||
if val, ok := s["postUp"]; ok {
|
||||
server.PostUp = val.(string)
|
||||
}
|
||||
if val, ok := s["preDown"]; ok {
|
||||
server.PreDown = val.(string)
|
||||
}
|
||||
if val, ok := s["postDown"]; ok {
|
||||
server.PostDown = val.(string)
|
||||
}
|
||||
created, err := time.Parse(time.RFC3339, s["created"].(string))
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to parse time")
|
||||
}
|
||||
server.Created = created
|
||||
updated, err := time.Parse(time.RFC3339, s["updated"].(string))
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to parse time")
|
||||
}
|
||||
server.Updated = updated
|
||||
|
||||
err = storage.Serialize("server.json", server)
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Errorf("failed to Serialize server")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func readClients() ([]map[string]interface{}, error) {
|
||||
clients := make([]map[string]interface{}, 0)
|
||||
|
||||
files, err := ioutil.ReadDir(filepath.Join(os.Getenv("WG_CONF_DIR")))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for _, f := range files {
|
||||
// clients file name is an uuid
|
||||
_, err := uuid.FromString(f.Name())
|
||||
if err == nil {
|
||||
c, err := deserialize(f.Name())
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
"path": f.Name(),
|
||||
}).Error("failed to deserialize client")
|
||||
} else {
|
||||
clients = append(clients, c)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return clients, nil
|
||||
}
|
||||
|
||||
func deserialize(id string) (map[string]interface{}, error) {
|
||||
path := filepath.Join(os.Getenv("WG_CONF_DIR"), id)
|
||||
|
||||
data, err := util.ReadFile(path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var d map[string]interface{}
|
||||
err = json.Unmarshal(data, &d)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return d, nil
|
||||
}
|
||||
166
core/server.go
Normal file
166
core/server.go
Normal file
@@ -0,0 +1,166 @@
|
||||
package core
|
||||
|
||||
import (
|
||||
"errors"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/model"
|
||||
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/storage"
|
||||
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/template"
|
||||
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/util"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"time"
|
||||
)
|
||||
|
||||
// ReadServer object, create default one
|
||||
func ReadServer() (*model.Server, error) {
|
||||
if !util.FileExists(filepath.Join(os.Getenv("WG_CONF_DIR"), "server.json")) {
|
||||
server := &model.Server{}
|
||||
|
||||
key, err := wgtypes.GeneratePrivateKey()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
server.PrivateKey = key.String()
|
||||
server.PublicKey = key.PublicKey().String()
|
||||
|
||||
server.Endpoint = "wireguard.example.com:123"
|
||||
server.ListenPort = 51820
|
||||
|
||||
server.Address = make([]string, 0)
|
||||
server.Address = append(server.Address, "fd9f:6666::10:6:6:1/64")
|
||||
server.Address = append(server.Address, "10.6.6.1/24")
|
||||
|
||||
server.Dns = make([]string, 0)
|
||||
server.Dns = append(server.Dns, "fd9f::10:0:0:2")
|
||||
server.Dns = append(server.Dns, "10.0.0.2")
|
||||
|
||||
server.PersistentKeepalive = 16
|
||||
server.Mtu = 0
|
||||
server.PreUp = "echo WireGuard PreUp"
|
||||
server.PostUp = "echo WireGuard PostUp"
|
||||
server.PreDown = "echo WireGuard PreDown"
|
||||
server.PostDown = "echo WireGuard PostDown"
|
||||
server.Created = time.Now().UTC()
|
||||
server.Updated = server.Created
|
||||
|
||||
err = storage.Serialize("server.json", server)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// server.json was missing, dump wg config after creation
|
||||
err = UpdateServerConfigWg()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
c, err := storage.Deserialize("server.json")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return c.(*model.Server), nil
|
||||
}
|
||||
|
||||
// UpdateServer keep private values from existing one
|
||||
func UpdateServer(server *model.Server) (*model.Server, error) {
|
||||
current, err := storage.Deserialize("server.json")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// check if server is valid
|
||||
errs := server.IsValid()
|
||||
if len(errs) != 0 {
|
||||
for _, err := range errs {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
}).Error("server validation error")
|
||||
}
|
||||
return nil, errors.New("failed to validate server")
|
||||
}
|
||||
|
||||
server.PrivateKey = current.(*model.Server).PrivateKey
|
||||
server.PublicKey = current.(*model.Server).PublicKey
|
||||
//server.PresharedKey = current.(*model.Server).PresharedKey
|
||||
server.Updated = time.Now().UTC()
|
||||
|
||||
err = storage.Serialize("server.json", server)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
v, err := storage.Deserialize("server.json")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
server = v.(*model.Server)
|
||||
|
||||
return server, UpdateServerConfigWg()
|
||||
}
|
||||
|
||||
// UpdateServerConfigWg in wg format
|
||||
func UpdateServerConfigWg() error {
|
||||
clients, err := ReadClients()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
server, err := ReadServer()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = template.DumpServerWg(clients, server)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetAllReservedIps the list of all reserved IPs, client and server
|
||||
func GetAllReservedIps() ([]string, error) {
|
||||
clients, err := ReadClients()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
server, err := ReadServer()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
reserverIps := make([]string, 0)
|
||||
|
||||
for _, client := range clients {
|
||||
for _, cidr := range client.Address {
|
||||
ip, err := util.GetIpFromCidr(cidr)
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
"cidr": cidr,
|
||||
}).Error("failed to ip from cidr")
|
||||
} else {
|
||||
reserverIps = append(reserverIps, ip)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for _, cidr := range server.Address {
|
||||
ip, err := util.GetIpFromCidr(cidr)
|
||||
if err != nil {
|
||||
log.WithFields(log.Fields{
|
||||
"err": err,
|
||||
"cidr": err,
|
||||
}).Error("failed to ip from cidr")
|
||||
} else {
|
||||
reserverIps = append(reserverIps, ip)
|
||||
}
|
||||
}
|
||||
|
||||
return reserverIps, nil
|
||||
}
|
||||
Reference in New Issue
Block a user