99 lines
1.9 KiB
Go
99 lines
1.9 KiB
Go
//go:build ignore
|
|
|
|
package peer
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"vppn/m"
|
|
)
|
|
|
|
type LocalConfig struct {
|
|
LocalPeerIP byte
|
|
Network []byte
|
|
WGPrivKey string
|
|
}
|
|
|
|
func configDir(netName string) string {
|
|
d, err := os.UserHomeDir()
|
|
if err != nil {
|
|
log.Fatalf("Failed to get user home directory: %v", err)
|
|
}
|
|
return filepath.Join(d, ".vppn", netName)
|
|
}
|
|
|
|
func lockFilePath(netName string) string {
|
|
return filepath.Join(configDir(netName), "__lock__")
|
|
}
|
|
|
|
func peerConfigPath(netName string) string {
|
|
return filepath.Join(configDir(netName), "config.json")
|
|
}
|
|
|
|
func peerStatePath(netName string) string {
|
|
return filepath.Join(configDir(netName), "state.json")
|
|
}
|
|
|
|
func statusSocketPath(netName string) string {
|
|
return filepath.Join(configDir(netName), "status.sock")
|
|
}
|
|
|
|
func storeJson(x any, outPath string) error {
|
|
outDir := filepath.Dir(outPath)
|
|
_ = os.MkdirAll(outDir, 0700)
|
|
|
|
tmpPath := outPath + ".tmp"
|
|
buf, err := json.Marshal(x)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
f, err := os.OpenFile(tmpPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0600)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := f.Write(buf); err != nil {
|
|
f.Close()
|
|
return err
|
|
}
|
|
|
|
if err := f.Sync(); err != nil {
|
|
f.Close()
|
|
return err
|
|
}
|
|
|
|
if err := f.Close(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return os.Rename(tmpPath, outPath)
|
|
}
|
|
|
|
func storePeerConfig(netName string, pc LocalConfig) error {
|
|
return storeJson(pc, peerConfigPath(netName))
|
|
}
|
|
|
|
func storeNetworkState(netName string, ps m.NetworkState) error {
|
|
return storeJson(ps, peerStatePath(netName))
|
|
}
|
|
|
|
func loadJson(dataPath string, ptr any) error {
|
|
data, err := os.ReadFile(dataPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return json.Unmarshal(data, ptr)
|
|
}
|
|
|
|
func loadPeerConfig(netName string) (pc LocalConfig, err error) {
|
|
return pc, loadJson(peerConfigPath(netName), &pc)
|
|
}
|
|
|
|
func loadNetworkState(netName string) (ps m.NetworkState, err error) {
|
|
return ps, loadJson(peerStatePath(netName), &ps)
|
|
}
|