mirror of
https://github.com/tim-beatham/smegmesh.git
synced 2024-12-15 11:00:41 +01:00
121 lines
2.3 KiB
Go
121 lines
2.3 KiB
Go
package ipc
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
ipcRpc "net/rpc"
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
|
"github.com/tim-beatham/wgmesh/pkg/ctrlserver/rpc"
|
|
"github.com/tim-beatham/wgmesh/pkg/ipc"
|
|
ipctypes "github.com/tim-beatham/wgmesh/pkg/ipc"
|
|
"github.com/tim-beatham/wgmesh/pkg/wg"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
)
|
|
|
|
const SockAddr = "/tmp/wgmesh_ipc.sock"
|
|
|
|
type Mesh struct {
|
|
Server *ctrlserver.MeshCtrlServer
|
|
}
|
|
|
|
/*
|
|
* Create a new WireGuard mesh network
|
|
*/
|
|
func (n Mesh) CreateNewMesh(name *string, reply *string) error {
|
|
wg.CreateInterface("wgmesh")
|
|
|
|
mesh, err := n.Server.CreateMesh()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
*reply = mesh.SharedKey.String()
|
|
return nil
|
|
}
|
|
|
|
func (n Mesh) ListMeshes(name *string, reply *map[string]ctrlserver.Mesh) error {
|
|
meshes := n.Server.Meshes
|
|
*reply = meshes
|
|
return nil
|
|
}
|
|
|
|
func (n Mesh) JoinMesh(args *ipctypes.JoinMeshArgs, reply *string) error {
|
|
conn, err := grpc.Dial(args.IpAdress+":8080", grpc.WithTransportCredentials(insecure.NewCredentials()))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer conn.Close()
|
|
|
|
c := rpc.NewMeshCtrlServerClient(conn)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
|
|
defer cancel()
|
|
|
|
dev := n.Server.GetDevice()
|
|
|
|
joinMeshReq := rpc.JoinMeshRequest{
|
|
MeshId: args.MeshId,
|
|
HostPort: 8080,
|
|
PublicKey: dev.PublicKey.String(),
|
|
WgPort: int32(dev.ListenPort),
|
|
}
|
|
|
|
r, err := c.JoinMesh(ctx, &joinMeshReq)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
*reply = strconv.FormatBool(r.GetSuccess())
|
|
return nil
|
|
}
|
|
|
|
func (n Mesh) GetMesh(meshId string, reply *ipc.GetMeshReply) error {
|
|
mesh, contains := n.Server.Meshes[meshId]
|
|
|
|
if contains {
|
|
nodes := make([]ctrlserver.MeshNode, len(mesh.Nodes))
|
|
|
|
i := 0
|
|
for _, n := range mesh.Nodes {
|
|
fmt.Println(n.PublicKey)
|
|
nodes[i] = n
|
|
i += 1
|
|
}
|
|
|
|
*reply = ipc.GetMeshReply{Nodes: nodes}
|
|
} else {
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func RunIpcHandler(server *ctrlserver.MeshCtrlServer) error {
|
|
if err := os.RemoveAll(SockAddr); err != nil {
|
|
return errors.New("Could not find to address")
|
|
}
|
|
|
|
newMeshIpc := new(Mesh)
|
|
newMeshIpc.Server = server
|
|
ipcRpc.Register(newMeshIpc)
|
|
ipcRpc.HandleHTTP()
|
|
|
|
l, e := net.Listen("unix", SockAddr)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
|
|
http.Serve(l, nil)
|
|
return nil
|
|
}
|