mirror of
https://github.com/tim-beatham/smegmesh.git
synced 2025-07-08 08:16:47 +02:00
Compare commits
10 Commits
72-pull-ra
...
bugfix-nod
Author | SHA1 | Date | |
---|---|---|---|
1f0914e2df | |||
27e00196cd | |||
dea6f1a22d | |||
913de57568 | |||
ce829114b1 | |||
cd844ff46e | |||
d0b1913796 | |||
90cfe820d2 | |||
8a49809855 | |||
dbc18bddc6 |
@ -3,7 +3,7 @@ package main
|
||||
import (
|
||||
"log"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/api"
|
||||
"github.com/tim-beatham/smegmesh/pkg/api"
|
||||
)
|
||||
|
||||
func main() {
|
||||
|
@ -3,7 +3,7 @@ package main
|
||||
import (
|
||||
"log"
|
||||
|
||||
smegdns "github.com/tim-beatham/wgmesh/pkg/dns"
|
||||
smegdns "github.com/tim-beatham/smegmesh/pkg/dns"
|
||||
)
|
||||
|
||||
func main() {
|
||||
|
@ -6,10 +6,10 @@ import (
|
||||
"os"
|
||||
|
||||
"github.com/akamensky/argparse"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
graph "github.com/tim-beatham/wgmesh/pkg/dot"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ipc"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
graph "github.com/tim-beatham/smegmesh/pkg/dot"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ipc"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
)
|
||||
|
||||
const SockAddr = "/tmp/wgmesh_ipc.sock"
|
||||
@ -22,7 +22,7 @@ type CreateMeshParams struct {
|
||||
AdvertiseDefault bool
|
||||
}
|
||||
|
||||
func createMesh(client *ipc.ClientIpc, args *ipc.NewMeshArgs) {
|
||||
func createMesh(client *ipc.SmegmeshIpc, args *ipc.NewMeshArgs) {
|
||||
var reply string
|
||||
err := client.CreateMesh(args, &reply)
|
||||
|
||||
@ -34,7 +34,7 @@ func createMesh(client *ipc.ClientIpc, args *ipc.NewMeshArgs) {
|
||||
fmt.Println(reply)
|
||||
}
|
||||
|
||||
func listMeshes(client *ipc.ClientIpc) {
|
||||
func listMeshes(client *ipc.SmegmeshIpc) {
|
||||
reply := new(ipc.ListMeshReply)
|
||||
|
||||
err := client.ListMeshes(reply)
|
||||
@ -49,7 +49,7 @@ func listMeshes(client *ipc.ClientIpc) {
|
||||
}
|
||||
}
|
||||
|
||||
func joinMesh(client *ipc.ClientIpc, args ipc.JoinMeshArgs) {
|
||||
func joinMesh(client *ipc.SmegmeshIpc, args ipc.JoinMeshArgs) {
|
||||
var reply string
|
||||
|
||||
err := client.JoinMesh(args, &reply)
|
||||
@ -61,7 +61,7 @@ func joinMesh(client *ipc.ClientIpc, args ipc.JoinMeshArgs) {
|
||||
fmt.Println(reply)
|
||||
}
|
||||
|
||||
func leaveMesh(client *ipc.ClientIpc, meshId string) {
|
||||
func leaveMesh(client *ipc.SmegmeshIpc, meshId string) {
|
||||
var reply string
|
||||
|
||||
err := client.LeaveMesh(meshId, &reply)
|
||||
@ -74,7 +74,7 @@ func leaveMesh(client *ipc.ClientIpc, meshId string) {
|
||||
fmt.Println(reply)
|
||||
}
|
||||
|
||||
func getGraph(client *ipc.ClientIpc) {
|
||||
func getGraph(client *ipc.SmegmeshIpc) {
|
||||
listMeshesReply := new(ipc.ListMeshReply)
|
||||
|
||||
err := client.ListMeshes(listMeshesReply)
|
||||
@ -110,7 +110,7 @@ func getGraph(client *ipc.ClientIpc) {
|
||||
fmt.Println(dot)
|
||||
}
|
||||
|
||||
func queryMesh(client *ipc.ClientIpc, meshId, query string) {
|
||||
func queryMesh(client *ipc.SmegmeshIpc, meshId, query string) {
|
||||
var reply string
|
||||
|
||||
args := ipc.QueryMesh{
|
||||
@ -128,7 +128,7 @@ func queryMesh(client *ipc.ClientIpc, meshId, query string) {
|
||||
fmt.Println(reply)
|
||||
}
|
||||
|
||||
func putDescription(client *ipc.ClientIpc, meshId, description string) {
|
||||
func putDescription(client *ipc.SmegmeshIpc, meshId, description string) {
|
||||
var reply string
|
||||
|
||||
err := client.PutDescription(ipc.PutDescriptionArgs{
|
||||
@ -145,7 +145,7 @@ func putDescription(client *ipc.ClientIpc, meshId, description string) {
|
||||
}
|
||||
|
||||
// putAlias: puts an alias for the node
|
||||
func putAlias(client *ipc.ClientIpc, meshid, alias string) {
|
||||
func putAlias(client *ipc.SmegmeshIpc, meshid, alias string) {
|
||||
var reply string
|
||||
|
||||
err := client.PutAlias(ipc.PutAliasArgs{
|
||||
@ -161,7 +161,7 @@ func putAlias(client *ipc.ClientIpc, meshid, alias string) {
|
||||
fmt.Println(reply)
|
||||
}
|
||||
|
||||
func setService(client *ipc.ClientIpc, meshId, service, value string) {
|
||||
func setService(client *ipc.SmegmeshIpc, meshId, service, value string) {
|
||||
var reply string
|
||||
|
||||
err := client.PutService(ipc.PutServiceArgs{
|
||||
@ -178,7 +178,7 @@ func setService(client *ipc.ClientIpc, meshId, service, value string) {
|
||||
fmt.Println(reply)
|
||||
}
|
||||
|
||||
func deleteService(client *ipc.ClientIpc, meshId, service string) {
|
||||
func deleteService(client *ipc.SmegmeshIpc, meshId, service string) {
|
||||
var reply string
|
||||
|
||||
err := client.DeleteService(ipc.DeleteServiceArgs{
|
||||
@ -195,8 +195,8 @@ func deleteService(client *ipc.ClientIpc, meshId, service string) {
|
||||
}
|
||||
|
||||
func main() {
|
||||
parser := argparse.NewParser("wg-mesh",
|
||||
"wg-mesh Manipulate WireGuard mesh networks")
|
||||
parser := argparse.NewParser("smgctl",
|
||||
"smegctl Manipulate WireGuard mesh networks")
|
||||
|
||||
newMeshCmd := parser.NewCommand("new-mesh", "Create a new mesh")
|
||||
listMeshCmd := parser.NewCommand("list-meshes", "List meshes the node is connected to")
|
@ -6,14 +6,14 @@ import (
|
||||
"os"
|
||||
"os/signal"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
ctrlserver "github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ipc"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/wgmesh/pkg/robin"
|
||||
"github.com/tim-beatham/wgmesh/pkg/sync"
|
||||
timer "github.com/tim-beatham/wgmesh/pkg/timers"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
ctrlserver "github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ipc"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/robin"
|
||||
"github.com/tim-beatham/smegmesh/pkg/sync"
|
||||
timer "github.com/tim-beatham/smegmesh/pkg/timers"
|
||||
"golang.zx2c4.com/wireguard/wgctrl"
|
||||
)
|
||||
|
4
go.mod
4
go.mod
@ -1,4 +1,4 @@
|
||||
module github.com/tim-beatham/wgmesh
|
||||
module github.com/tim-beatham/smegmesh
|
||||
|
||||
go 1.21.3
|
||||
|
||||
@ -11,11 +11,11 @@ require (
|
||||
github.com/google/uuid v1.3.0
|
||||
github.com/jmespath/go-jmespath v0.4.0
|
||||
github.com/jsimonetti/rtnetlink v1.3.5
|
||||
github.com/lithammer/shortuuid v3.0.0+incompatible
|
||||
github.com/miekg/dns v1.1.57
|
||||
github.com/sirupsen/logrus v1.9.3
|
||||
golang.org/x/sys v0.14.0
|
||||
golang.zx2c4.com/wireguard/wgctrl v0.0.0-20230429144221-925a1e7659e6
|
||||
gonum.org/v1/gonum v0.14.0
|
||||
google.golang.org/grpc v1.58.1
|
||||
google.golang.org/protobuf v1.31.0
|
||||
gopkg.in/yaml.v3 v3.0.1
|
||||
|
6
go.sum
6
go.sum
@ -27,8 +27,6 @@ github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/o
|
||||
github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
|
||||
github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
|
||||
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
|
||||
github.com/go-playground/validator/v10 v10.14.0 h1:vgvQWe3XCz3gIeFDm/HnTIbj6UGmg/+t63MyGU2n5js=
|
||||
github.com/go-playground/validator/v10 v10.14.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
|
||||
github.com/go-playground/validator/v10 v10.16.0 h1:x+plE831WK4vaKHO/jpgUGsvLKIqRRkz6M78GuJAfGE=
|
||||
github.com/go-playground/validator/v10 v10.16.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
|
||||
github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
|
||||
@ -57,6 +55,8 @@ github.com/klauspost/cpuid/v2 v2.2.4 h1:acbojRNwl3o09bUq+yDCtZFc1aiwaAAxtcn8YkZX
|
||||
github.com/klauspost/cpuid/v2 v2.2.4/go.mod h1:RVVoqg1df56z8g3pUjL/3lE5UfnlrJX8tyFgg4nqhuY=
|
||||
github.com/leodido/go-urn v1.2.4 h1:XlAE/cm/ms7TE/VMVoduSpNBoyc2dOxHs5MZSwAN63Q=
|
||||
github.com/leodido/go-urn v1.2.4/go.mod h1:7ZrI8mTSeBSHl/UaRyKQW1qZeMgak41ANeCNaVckg+4=
|
||||
github.com/lithammer/shortuuid v3.0.0+incompatible h1:NcD0xWW/MZYXEHa6ITy6kaXN5nwm/V115vj2YXfhS0w=
|
||||
github.com/lithammer/shortuuid v3.0.0+incompatible/go.mod h1:FR74pbAuElzOUuenUHTK2Tciko1/vKuIKS9dSkDrA4w=
|
||||
github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
|
||||
github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/mdlayher/genetlink v1.3.2 h1:KdrNKe+CTu+IbZnm/GVUMXSqBBLqcGpRDa0xkQy56gw=
|
||||
@ -123,8 +123,6 @@ golang.zx2c4.com/wireguard v0.0.0-20230704135630-469159ecf7d1 h1:EY138uSo1JYlDq+
|
||||
golang.zx2c4.com/wireguard v0.0.0-20230704135630-469159ecf7d1/go.mod h1:tqur9LnfstdR9ep2LaJT4lFUl0EjlHtge+gAjmsHUG4=
|
||||
golang.zx2c4.com/wireguard/wgctrl v0.0.0-20230429144221-925a1e7659e6 h1:CawjfCvYQH2OU3/TnxLx97WDSUDRABfT18pCOYwc2GE=
|
||||
golang.zx2c4.com/wireguard/wgctrl v0.0.0-20230429144221-925a1e7659e6/go.mod h1:3rxYc4HtVcSG9gVaTs2GEBdehh+sYPOwKtyUWEOTb80=
|
||||
gonum.org/v1/gonum v0.14.0 h1:2NiG67LD1tEH0D7kM+ps2V+fXmsAnpUeec7n8tcr4S0=
|
||||
gonum.org/v1/gonum v0.14.0/go.mod h1:AoWeoz0becf9QMWtE8iWXNXc27fK4fNeHNf/oMejGfU=
|
||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20230711160842-782d3b101e98 h1:bVf09lpb+OJbByTj913DRJioFFAjf/ZGxEz7MajTp2U=
|
||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20230711160842-782d3b101e98/go.mod h1:TUfxEVdsvPg18p6AslUXFoLdpED4oBnGwyqk3dV1XzM=
|
||||
google.golang.org/grpc v1.58.1 h1:OL+Vz23DTtrrldqHK49FUOPHyY75rvFqJfXC84NYW58=
|
||||
|
@ -5,10 +5,10 @@ import (
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ipc"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/wgmesh/pkg/what8words"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ipc"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/what8words"
|
||||
)
|
||||
|
||||
type ApiServer interface {
|
||||
@ -18,7 +18,7 @@ type ApiServer interface {
|
||||
|
||||
type SmegServer struct {
|
||||
router *gin.Engine
|
||||
client *ipc.ClientIpc
|
||||
client *ipc.SmegmeshIpc
|
||||
words *what8words.What8Words
|
||||
}
|
||||
|
||||
@ -235,9 +235,19 @@ func NewSmegServer(conf ApiServerConf) (ApiServer, error) {
|
||||
words: words,
|
||||
}
|
||||
|
||||
router.GET("/meshes", smegServer.GetMeshes)
|
||||
router.GET("/mesh/:meshid", smegServer.GetMesh)
|
||||
router.POST("/mesh/create", smegServer.CreateMesh)
|
||||
router.POST("/mesh/join", smegServer.JoinMesh)
|
||||
v1 := router.Group("/api/v1")
|
||||
{
|
||||
meshes := v1.Group("/meshes")
|
||||
{
|
||||
meshes.GET("/", smegServer.GetMeshes)
|
||||
}
|
||||
mesh := v1.Group("/mesh")
|
||||
{
|
||||
mesh.GET("/:meshid", smegServer.GetMesh)
|
||||
mesh.POST("/create", smegServer.CreateMesh)
|
||||
mesh.POST("/join", smegServer.JoinMesh)
|
||||
}
|
||||
}
|
||||
|
||||
return smegServer, nil
|
||||
}
|
||||
|
@ -9,10 +9,10 @@ import (
|
||||
"time"
|
||||
|
||||
"github.com/automerge/automerge-go"
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"golang.zx2c4.com/wireguard/wgctrl"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
)
|
||||
|
@ -2,7 +2,7 @@ package automerge
|
||||
|
||||
import (
|
||||
"github.com/automerge/automerge-go"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
)
|
||||
|
||||
type AutomergeSync struct {
|
||||
|
@ -6,9 +6,9 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
)
|
||||
|
||||
|
@ -3,9 +3,9 @@ package automerge
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
)
|
||||
|
||||
type CrdtProviderFactory struct{}
|
||||
|
@ -23,9 +23,10 @@ func binarySearch(global []string, selfId string, groupSize int) (int, int) {
|
||||
|
||||
lower := 0
|
||||
higher := len(global) - 1
|
||||
mid := (lower + higher) / 2
|
||||
|
||||
for (higher+1)-lower > groupSize {
|
||||
mid := (lower + higher) / 2
|
||||
|
||||
if global[mid] < selfId {
|
||||
lower = mid + 1
|
||||
} else if global[mid] > selfId {
|
||||
@ -33,8 +34,6 @@ func binarySearch(global []string, selfId string, groupSize int) (int, int) {
|
||||
} else {
|
||||
break
|
||||
}
|
||||
|
||||
mid = (lower + higher) / 2
|
||||
}
|
||||
|
||||
return lower, int(math.Min(float64(lower+groupSize), float64(len(global))))
|
||||
|
@ -6,7 +6,7 @@ import (
|
||||
"crypto/tls"
|
||||
"errors"
|
||||
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/credentials"
|
||||
)
|
||||
|
@ -7,7 +7,7 @@ import (
|
||||
"os"
|
||||
"sync"
|
||||
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
)
|
||||
|
||||
// ConnectionManager defines an interface for maintaining peer connections
|
||||
|
@ -8,9 +8,9 @@ import (
|
||||
"net"
|
||||
"os"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/wgmesh/pkg/rpc"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/rpc"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/credentials"
|
||||
)
|
||||
|
@ -9,10 +9,10 @@ import (
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"golang.zx2c4.com/wireguard/wgctrl"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
)
|
||||
|
@ -6,9 +6,9 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
)
|
||||
|
||||
|
@ -4,9 +4,9 @@ import (
|
||||
"fmt"
|
||||
"hash/fnv"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
)
|
||||
|
||||
type TwoPhaseMapFactory struct {
|
||||
|
@ -7,7 +7,7 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
)
|
||||
|
||||
func NewGmap() *GMap[string, bool] {
|
||||
|
@ -3,7 +3,7 @@ package crdt
|
||||
import (
|
||||
"cmp"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
)
|
||||
|
||||
type TwoPhaseMap[K cmp.Ordered, D any] struct {
|
||||
|
@ -4,7 +4,7 @@ import (
|
||||
"bytes"
|
||||
"encoding/gob"
|
||||
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
)
|
||||
|
||||
type SyncState int
|
||||
|
@ -5,7 +5,7 @@ import (
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
)
|
||||
|
||||
type VectorBucket struct {
|
||||
|
@ -1,16 +1,16 @@
|
||||
package ctrlserver
|
||||
|
||||
import (
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/conn"
|
||||
"github.com/tim-beatham/wgmesh/pkg/crdt"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ip"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/wgmesh/pkg/query"
|
||||
"github.com/tim-beatham/wgmesh/pkg/rpc"
|
||||
"github.com/tim-beatham/wgmesh/pkg/wg"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conn"
|
||||
"github.com/tim-beatham/smegmesh/pkg/crdt"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ip"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/query"
|
||||
"github.com/tim-beatham/smegmesh/pkg/rpc"
|
||||
"github.com/tim-beatham/smegmesh/pkg/wg"
|
||||
"golang.zx2c4.com/wireguard/wgctrl"
|
||||
)
|
||||
|
||||
|
@ -4,11 +4,11 @@ import (
|
||||
"net"
|
||||
"time"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/conn"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/wgmesh/pkg/query"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conn"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/query"
|
||||
"golang.zx2c4.com/wireguard/wgctrl"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
)
|
||||
|
@ -1,10 +1,10 @@
|
||||
package ctrlserver
|
||||
|
||||
import (
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/conn"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/wgmesh/pkg/query"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conn"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/query"
|
||||
"golang.zx2c4.com/wireguard/wgctrl"
|
||||
)
|
||||
|
||||
|
@ -6,18 +6,16 @@ import (
|
||||
"net"
|
||||
|
||||
"github.com/miekg/dns"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ipc"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/wgmesh/pkg/query"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ipc"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/query"
|
||||
)
|
||||
|
||||
const SockAddr = "/tmp/wgmesh_ipc.sock"
|
||||
|
||||
const MeshRegularExpression = `(?P<meshId>.+)\.(?P<alias>.+)\.smeg\.`
|
||||
|
||||
type DNSHandler struct {
|
||||
client *ipc.ClientIpc
|
||||
client *ipc.SmegmeshIpc
|
||||
server *dns.Server
|
||||
}
|
||||
|
||||
|
@ -6,7 +6,7 @@ import (
|
||||
"hash/fnv"
|
||||
"strings"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
)
|
||||
|
||||
type GraphType string
|
||||
|
@ -4,7 +4,7 @@ import (
|
||||
"fmt"
|
||||
"slices"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
)
|
||||
|
||||
// MeshGraphConverter converts a mesh to a graph
|
||||
|
@ -34,7 +34,7 @@ type CgaParameters struct {
|
||||
flag byte
|
||||
}
|
||||
|
||||
func NewCga(key wgtypes.Key, subnetPrefix [2 * InterfaceIdLen]byte) (*CgaParameters, error) {
|
||||
func NewCga(key wgtypes.Key, collisionCount uint8, subnetPrefix [2 * InterfaceIdLen]byte) (*CgaParameters, error) {
|
||||
var params CgaParameters
|
||||
|
||||
_, err := rand.Read(params.Modifier[:])
|
||||
@ -45,6 +45,7 @@ func NewCga(key wgtypes.Key, subnetPrefix [2 * InterfaceIdLen]byte) (*CgaParamet
|
||||
|
||||
params.PublicKey = key
|
||||
params.SubnetPrefix = subnetPrefix
|
||||
params.CollisionCount = collisionCount
|
||||
return ¶ms, nil
|
||||
}
|
||||
|
||||
@ -78,7 +79,6 @@ func (c *CgaParameters) generateHash1() []byte {
|
||||
byteVal[hash1Length-1] = c.CollisionCount
|
||||
|
||||
hash := sha1.Sum(byteVal[:])
|
||||
|
||||
return hash[:Hash1Prefix]
|
||||
}
|
||||
|
||||
@ -90,9 +90,6 @@ func clearBit(num, pos int) byte {
|
||||
}
|
||||
|
||||
func (c *CgaParameters) generateInterface() []byte {
|
||||
// TODO: On duplicate address detection increment collision.
|
||||
// Also incorporate SEC
|
||||
|
||||
hash1 := c.generateHash1()
|
||||
|
||||
var interfaceId []byte = make([]byte, InterfaceIdLen)
|
||||
|
@ -7,5 +7,5 @@ import (
|
||||
)
|
||||
|
||||
type IPAllocator interface {
|
||||
GetIP(key wgtypes.Key, meshId string) (net.IP, error)
|
||||
GetIP(key wgtypes.Key, meshId string, collisionCount uint8) (net.IP, error)
|
||||
}
|
||||
|
@ -39,10 +39,10 @@ func (u *ULABuilder) GetIPNet(meshId string) (*net.IPNet, error) {
|
||||
return net, nil
|
||||
}
|
||||
|
||||
func (u *ULABuilder) GetIP(key wgtypes.Key, meshId string) (net.IP, error) {
|
||||
func (u *ULABuilder) GetIP(key wgtypes.Key, meshId string, collisionCount uint8) (net.IP, error) {
|
||||
ulaPrefix := getMeshPrefix(meshId)
|
||||
|
||||
c, err := NewCga(key, ulaPrefix)
|
||||
c, err := NewCga(key, collisionCount, ulaPrefix)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
|
@ -8,7 +8,7 @@ import (
|
||||
ipcRpc "net/rpc"
|
||||
"os"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
)
|
||||
|
||||
const SockAddr = "/tmp/wgmesh_sock"
|
||||
@ -94,63 +94,76 @@ type QueryMesh struct {
|
||||
Query string
|
||||
}
|
||||
|
||||
type ClientIpc struct {
|
||||
type ClientIpc interface {
|
||||
CreateMesh(args *NewMeshArgs, reply *string) error
|
||||
ListMeshes(args *ListMeshReply, reply *string) error
|
||||
JoinMesh(args JoinMeshArgs, reply *string) error
|
||||
LeaveMesh(meshId string, reply *string) error
|
||||
GetMesh(meshId string, reply *GetMeshReply) error
|
||||
Query(query QueryMesh, reply *string) error
|
||||
PutDescription(args PutDescriptionArgs, reply *string) error
|
||||
PutAlias(args PutAliasArgs, reply *string) error
|
||||
PutService(args PutServiceArgs, reply *string) error
|
||||
DeleteService(args DeleteServiceArgs, reply *string) error
|
||||
}
|
||||
|
||||
type SmegmeshIpc struct {
|
||||
client *ipcRpc.Client
|
||||
}
|
||||
|
||||
func NewClientIpc() (*ClientIpc, error) {
|
||||
func NewClientIpc() (*SmegmeshIpc, error) {
|
||||
client, err := ipcRpc.DialHTTP("unix", SockAddr)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &ClientIpc{
|
||||
return &SmegmeshIpc{
|
||||
client: client,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (c *ClientIpc) CreateMesh(args *NewMeshArgs, reply *string) error {
|
||||
func (c *SmegmeshIpc) CreateMesh(args *NewMeshArgs, reply *string) error {
|
||||
return c.client.Call("IpcHandler.CreateMesh", args, reply)
|
||||
}
|
||||
|
||||
func (c *ClientIpc) ListMeshes(reply *ListMeshReply) error {
|
||||
func (c *SmegmeshIpc) ListMeshes(reply *ListMeshReply) error {
|
||||
return c.client.Call("IpcHandler.ListMeshes", "", reply)
|
||||
}
|
||||
|
||||
func (c *ClientIpc) JoinMesh(args JoinMeshArgs, reply *string) error {
|
||||
func (c *SmegmeshIpc) JoinMesh(args JoinMeshArgs, reply *string) error {
|
||||
return c.client.Call("IpcHandler.JoinMesh", &args, reply)
|
||||
}
|
||||
|
||||
func (c *ClientIpc) LeaveMesh(meshId string, reply *string) error {
|
||||
func (c *SmegmeshIpc) LeaveMesh(meshId string, reply *string) error {
|
||||
return c.client.Call("IpcHandler.LeaveMesh", &meshId, reply)
|
||||
}
|
||||
|
||||
func (c *ClientIpc) GetMesh(meshId string, reply *GetMeshReply) error {
|
||||
func (c *SmegmeshIpc) GetMesh(meshId string, reply *GetMeshReply) error {
|
||||
return c.client.Call("IpcHandler.GetMesh", &meshId, reply)
|
||||
}
|
||||
|
||||
func (c *ClientIpc) Query(query QueryMesh, reply *string) error {
|
||||
func (c *SmegmeshIpc) Query(query QueryMesh, reply *string) error {
|
||||
return c.client.Call("IpcHandler.Query", &query, reply)
|
||||
}
|
||||
|
||||
func (c *ClientIpc) PutDescription(args PutDescriptionArgs, reply *string) error {
|
||||
func (c *SmegmeshIpc) PutDescription(args PutDescriptionArgs, reply *string) error {
|
||||
return c.client.Call("IpcHandler.PutDescription", &args, reply)
|
||||
}
|
||||
|
||||
func (c *ClientIpc) PutAlias(args PutAliasArgs, reply *string) error {
|
||||
func (c *SmegmeshIpc) PutAlias(args PutAliasArgs, reply *string) error {
|
||||
return c.client.Call("IpcHandler.PutAlias", &args, reply)
|
||||
}
|
||||
|
||||
func (c *ClientIpc) PutService(args PutServiceArgs, reply *string) error {
|
||||
func (c *SmegmeshIpc) PutService(args PutServiceArgs, reply *string) error {
|
||||
return c.client.Call("IpcHandler.PutService", &args, reply)
|
||||
}
|
||||
|
||||
func (c *ClientIpc) DeleteService(args DeleteServiceArgs, reply *string) error {
|
||||
func (c *SmegmeshIpc) DeleteService(args DeleteServiceArgs, reply *string) error {
|
||||
return c.client.Call("IpcHandler.DeleteService", &args, reply)
|
||||
}
|
||||
|
||||
func (c *ClientIpc) Close() error {
|
||||
func (c *SmegmeshIpc) Close() error {
|
||||
return c.Close()
|
||||
}
|
||||
|
||||
|
@ -6,7 +6,7 @@ import (
|
||||
"net"
|
||||
|
||||
"github.com/jsimonetti/rtnetlink"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"golang.org/x/sys/unix"
|
||||
)
|
||||
|
||||
|
@ -7,10 +7,10 @@ import (
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ip"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/wgmesh/pkg/route"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ip"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/route"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
)
|
||||
|
||||
|
@ -3,13 +3,15 @@ package mesh
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"net"
|
||||
"sync"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/cmd"
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ip"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/wgmesh/pkg/wg"
|
||||
"github.com/tim-beatham/smegmesh/pkg/cmd"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ip"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/wg"
|
||||
"golang.zx2c4.com/wireguard/wgctrl"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
)
|
||||
@ -285,10 +287,36 @@ func (s *MeshManagerImpl) AddSelf(params *AddSelfParams) error {
|
||||
|
||||
pubKey := s.HostParameters.PrivateKey.PublicKey()
|
||||
|
||||
nodeIP, err := s.ipAllocator.GetIP(pubKey, params.MeshId)
|
||||
collisionCount := uint8(0)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
var nodeIP net.IP
|
||||
|
||||
// Perform Duplicate Address Detection with the nodes
|
||||
// that are already in the network
|
||||
for {
|
||||
generatedIP, err := s.ipAllocator.GetIP(pubKey, params.MeshId, collisionCount)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
snapshot, err := mesh.GetMesh()
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
proposition := func(node MeshNode) bool {
|
||||
ipNet := node.GetWgHost()
|
||||
return ipNet.IP.Equal(nodeIP)
|
||||
}
|
||||
|
||||
if lib.Contains(lib.MapValues(snapshot.GetNodes()), proposition) {
|
||||
collisionCount++
|
||||
} else {
|
||||
nodeIP = generatedIP
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
node := s.nodeFactory.Build(&MeshNodeFactoryParams{
|
||||
@ -328,7 +356,7 @@ func (s *MeshManagerImpl) LeaveMesh(meshId string) error {
|
||||
err := mesh.RemoveNode(s.HostParameters.GetPublicKey())
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
logging.Log.WriteErrorf(err.Error())
|
||||
}
|
||||
|
||||
if s.OnDelete != nil {
|
||||
|
@ -3,10 +3,10 @@ package mesh
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ip"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/wgmesh/pkg/wg"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ip"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/wg"
|
||||
)
|
||||
|
||||
func getMeshConfiguration() *conf.DaemonConfiguration {
|
||||
|
@ -3,8 +3,8 @@ package mesh
|
||||
import (
|
||||
"net"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/ip"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ip"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
)
|
||||
|
||||
type RouteManager interface {
|
||||
|
@ -5,8 +5,8 @@ import (
|
||||
"net"
|
||||
"time"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
"golang.zx2c4.com/wireguard/wgctrl"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
)
|
||||
|
@ -6,7 +6,7 @@ import (
|
||||
"net"
|
||||
"slices"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"golang.zx2c4.com/wireguard/wgctrl"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
)
|
||||
|
@ -6,9 +6,9 @@ import (
|
||||
"strings"
|
||||
|
||||
"github.com/jmespath/go-jmespath"
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
)
|
||||
|
||||
// Querier queries a data store for the given data
|
||||
|
@ -7,11 +7,11 @@ import (
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ipc"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/wgmesh/pkg/rpc"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ipc"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/rpc"
|
||||
)
|
||||
|
||||
type IpcHandler struct {
|
||||
|
@ -3,10 +3,10 @@ package robin
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ipc"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ipc"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
)
|
||||
|
||||
func getRequester() *IpcHandler {
|
||||
|
@ -4,8 +4,8 @@ import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/wgmesh/pkg/rpc"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/smegmesh/pkg/rpc"
|
||||
)
|
||||
|
||||
type WgRpc struct {
|
||||
|
@ -1 +0,0 @@
|
||||
package robin
|
@ -1,7 +1,7 @@
|
||||
package route
|
||||
|
||||
import (
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
"golang.org/x/sys/unix"
|
||||
)
|
||||
|
||||
|
@ -7,11 +7,11 @@ import (
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/conf"
|
||||
"github.com/tim-beatham/wgmesh/pkg/conn"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conf"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conn"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
)
|
||||
|
||||
// Syncer: picks random nodes from the meshs
|
||||
@ -28,6 +28,7 @@ type SyncerImpl struct {
|
||||
cluster conn.ConnCluster
|
||||
conf *conf.DaemonConfiguration
|
||||
lastSync map[string]int64
|
||||
lock sync.RWMutex
|
||||
}
|
||||
|
||||
// Sync: Sync with random nodes
|
||||
@ -38,11 +39,10 @@ func (s *SyncerImpl) Sync(correspondingMesh mesh.MeshProvider) error {
|
||||
|
||||
// Self can be nil if the node is removed
|
||||
selfID := s.manager.GetPublicKey()
|
||||
self, _ := correspondingMesh.GetNode(selfID.String())
|
||||
self, err := correspondingMesh.GetNode(selfID.String())
|
||||
|
||||
// Mesh has been removed
|
||||
if self == nil {
|
||||
return fmt.Errorf("mesh %s does not exist", correspondingMesh.GetMeshId())
|
||||
if err != nil {
|
||||
logging.Log.WriteErrorf(err.Error())
|
||||
}
|
||||
|
||||
correspondingMesh.Prune()
|
||||
@ -51,7 +51,8 @@ func (s *SyncerImpl) Sync(correspondingMesh mesh.MeshProvider) error {
|
||||
logging.Log.WriteInfof("meshes %s has changes", correspondingMesh.GetMeshId())
|
||||
}
|
||||
|
||||
if self.GetType() == conf.PEER_ROLE && !correspondingMesh.HasChanges() && s.infectionCount == 0 {
|
||||
// If removed sync with other nodes to gossip the node is removed
|
||||
if self != nil && self.GetType() == conf.PEER_ROLE && !correspondingMesh.HasChanges() && s.infectionCount == 0 {
|
||||
logging.Log.WriteInfof("no changes for %s", correspondingMesh.GetMeshId())
|
||||
|
||||
// If not synchronised in certain time pull from random neighbour
|
||||
@ -63,16 +64,19 @@ func (s *SyncerImpl) Sync(correspondingMesh mesh.MeshProvider) error {
|
||||
}
|
||||
|
||||
before := time.Now()
|
||||
s.manager.GetRouteManager().UpdateRoutes()
|
||||
err = s.manager.GetRouteManager().UpdateRoutes()
|
||||
|
||||
if err != nil {
|
||||
logging.Log.WriteErrorf(err.Error())
|
||||
}
|
||||
|
||||
publicKey := s.manager.GetPublicKey()
|
||||
nodeNames := correspondingMesh.GetPeers()
|
||||
|
||||
if self != nil {
|
||||
nodeNames = lib.Filter(nodeNames, func(s string) bool {
|
||||
return s != mesh.NodeID(self)
|
||||
})
|
||||
}
|
||||
nodeNames = lib.Filter(nodeNames, func(s string) bool {
|
||||
// Filter our only public key out so we dont sync with ourself
|
||||
return s != publicKey.String()
|
||||
})
|
||||
|
||||
var gossipNodes []string
|
||||
|
||||
@ -101,14 +105,14 @@ func (s *SyncerImpl) Sync(correspondingMesh mesh.MeshProvider) error {
|
||||
|
||||
// Do this synchronously to conserve bandwidth
|
||||
for _, node := range gossipNodes {
|
||||
correspondingPeer := s.manager.GetNode(correspondingMesh.GetMeshId(), node)
|
||||
correspondingPeer, err := correspondingMesh.GetNode(node)
|
||||
|
||||
if correspondingPeer == nil {
|
||||
if correspondingPeer == nil || err != nil {
|
||||
logging.Log.WriteErrorf("node %s does not exist", node)
|
||||
continue
|
||||
}
|
||||
|
||||
err := s.requester.SyncMesh(correspondingMesh.GetMeshId(), correspondingPeer)
|
||||
err = s.requester.SyncMesh(correspondingMesh.GetMeshId(), correspondingPeer)
|
||||
|
||||
if err == nil || err == io.EOF {
|
||||
succeeded = true
|
||||
@ -131,7 +135,9 @@ func (s *SyncerImpl) Sync(correspondingMesh mesh.MeshProvider) error {
|
||||
|
||||
correspondingMesh.SaveChanges()
|
||||
|
||||
s.lock.Lock()
|
||||
s.lastSync[correspondingMesh.GetMeshId()] = time.Now().Unix()
|
||||
s.lock.Unlock()
|
||||
return nil
|
||||
}
|
||||
|
||||
@ -188,6 +194,7 @@ func (s *SyncerImpl) SyncMeshes() error {
|
||||
|
||||
go sync()
|
||||
}
|
||||
wg.Wait()
|
||||
|
||||
logging.Log.WriteInfof("updating the WireGuard configuration")
|
||||
err := s.manager.ApplyConfig()
|
||||
|
@ -1,9 +1,9 @@
|
||||
package sync
|
||||
|
||||
import (
|
||||
"github.com/tim-beatham/wgmesh/pkg/conn"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/conn"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"google.golang.org/grpc/codes"
|
||||
"google.golang.org/grpc/status"
|
||||
)
|
||||
|
@ -6,10 +6,10 @@ import (
|
||||
"io"
|
||||
"time"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/wgmesh/pkg/rpc"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/rpc"
|
||||
)
|
||||
|
||||
// SyncRequester: coordinates the syncing of meshes
|
||||
|
@ -1,8 +1,8 @@
|
||||
package sync
|
||||
|
||||
import (
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
)
|
||||
|
||||
// Run implements SyncScheduler.
|
||||
|
@ -6,9 +6,9 @@ import (
|
||||
"errors"
|
||||
"io"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/wgmesh/pkg/rpc"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/smegmesh/pkg/mesh"
|
||||
"github.com/tim-beatham/smegmesh/pkg/rpc"
|
||||
)
|
||||
|
||||
type SyncServiceImpl struct {
|
||||
|
@ -1,9 +1,9 @@
|
||||
package timer
|
||||
|
||||
import (
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
)
|
||||
|
||||
func NewTimestampScheduler(ctrlServer *ctrlserver.MeshCtrlServer) lib.Timer {
|
||||
|
@ -5,8 +5,8 @@ import (
|
||||
"crypto/rand"
|
||||
"fmt"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
||||
"github.com/tim-beatham/smegmesh/pkg/lib"
|
||||
logging "github.com/tim-beatham/smegmesh/pkg/log"
|
||||
"golang.zx2c4.com/wireguard/wgctrl"
|
||||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||||
)
|
||||
|
Reference in New Issue
Block a user