initial commit
This commit is contained in:
commit
84abd9bcf1
5 changed files with 222 additions and 0 deletions
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
|
@ -0,0 +1 @@
|
||||||
|
bndd
|
41
client.go
Normal file
41
client.go
Normal file
|
@ -0,0 +1,41 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"log"
|
||||||
|
"net"
|
||||||
|
)
|
||||||
|
|
||||||
|
func client(conn net.Conn, instance string) error {
|
||||||
|
log.Println("Client connected")
|
||||||
|
buffer := make([]byte, 1024)
|
||||||
|
n, err := conn.Read(buffer)
|
||||||
|
message := string(buffer[:n])
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
log.Println("Message Received:", message)
|
||||||
|
var objmap map[string]string
|
||||||
|
err = json.Unmarshal([]byte(message), &objmap)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
val, has := objmap["command"]
|
||||||
|
if has {
|
||||||
|
switch val {
|
||||||
|
case "run":
|
||||||
|
_, err = runMap(conn, instance, objmap)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
err = errors.New("Unknown Command: " + val)
|
||||||
|
log.Println(err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
3
go.mod
Normal file
3
go.mod
Normal file
|
@ -0,0 +1,3 @@
|
||||||
|
module boundaries/bndd
|
||||||
|
|
||||||
|
go 1.23.4
|
82
main.go
Normal file
82
main.go
Normal file
|
@ -0,0 +1,82 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"io/fs"
|
||||||
|
"log"
|
||||||
|
"net"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"path/filepath"
|
||||||
|
"syscall"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
running := true
|
||||||
|
instance_path, err := filepath.Abs("/home/pauljako/boundaries")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
instance, err := filepath.EvalSymlinks(instance_path)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
socket_path, err := filepath.Abs(filepath.Join(instance, "/boundaries.sock"))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
ln, err := net.Listen("unix", socket_path)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
log.Println("Listening on", instance+"/boundaries.sock")
|
||||||
|
signalChannel := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(signalChannel, os.Interrupt, syscall.SIGTERM)
|
||||||
|
go func() {
|
||||||
|
sig := <-signalChannel
|
||||||
|
switch sig {
|
||||||
|
case os.Interrupt:
|
||||||
|
running = false
|
||||||
|
log.Println("Received interrupt")
|
||||||
|
ln.Close()
|
||||||
|
os.Exit(0)
|
||||||
|
case syscall.SIGTERM:
|
||||||
|
running = false
|
||||||
|
log.Println("Received SIGTERM")
|
||||||
|
ln.Close()
|
||||||
|
os.Exit(15)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
accept_clients(&ln, &instance, &running)
|
||||||
|
}
|
||||||
|
|
||||||
|
func exists(path string) (bool, error) {
|
||||||
|
_, err := os.Stat(path)
|
||||||
|
if err == nil {
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
if errors.Is(err, fs.ErrNotExist) {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func accept_clients(ln *net.Listener, instance *string, running *bool) {
|
||||||
|
for *running {
|
||||||
|
conn, err := (*ln).Accept()
|
||||||
|
if err != nil {
|
||||||
|
if !(*running) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
log.Println(err)
|
||||||
|
}
|
||||||
|
go func() {
|
||||||
|
err := client(conn, *instance)
|
||||||
|
if err != nil {
|
||||||
|
conn.Write([]byte(err.Error()))
|
||||||
|
}
|
||||||
|
conn.Close()
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
}
|
95
run.go
Normal file
95
run.go
Normal file
|
@ -0,0 +1,95 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"net"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
func runMap(conn net.Conn, instance string, objmap map[string]string) (int, error) {
|
||||||
|
program, has := objmap["package"]
|
||||||
|
if !has {
|
||||||
|
return 0, errors.New("a package is required")
|
||||||
|
}
|
||||||
|
arguments, has := objmap["arguments"]
|
||||||
|
if !has {
|
||||||
|
arguments = ""
|
||||||
|
}
|
||||||
|
target, has := objmap["target"]
|
||||||
|
if !has {
|
||||||
|
target = "run"
|
||||||
|
}
|
||||||
|
workdir, has := objmap["workdir"]
|
||||||
|
if !has {
|
||||||
|
return 0, errors.New("a workdir is required")
|
||||||
|
}
|
||||||
|
return run(conn, instance, program, arguments, target, workdir)
|
||||||
|
}
|
||||||
|
|
||||||
|
func run(conn net.Conn, instance string, program string, arguments string, target string, workdir string) (int, error) {
|
||||||
|
package_path, err := filepath.Abs(filepath.Join(instance, "apps", program))
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
info_path, err := filepath.Abs(filepath.Join(package_path, "boundaries.json"))
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
if exists, err := exists(info_path); !exists || err != nil {
|
||||||
|
return 0, errors.New("package " + program + " not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
working_dir, err := filepath.Abs(workdir)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
infofile_content, err := os.ReadFile(info_path)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var info map[string]interface{}
|
||||||
|
err = json.Unmarshal(infofile_content, &info)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
_, has := info["name"]
|
||||||
|
if !has {
|
||||||
|
return 0, errors.New("invalid boundaries.json file. name field not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
command_interface, has := info["command"]
|
||||||
|
if !has {
|
||||||
|
return 0, errors.New("invalid boundaries.json file. command field not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
commands := command_interface.(map[string]interface{})
|
||||||
|
|
||||||
|
command, has := commands[target]
|
||||||
|
if !has {
|
||||||
|
return 0, errors.New("target " + target + " not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
command, err = filepath.Abs(filepath.Join(package_path, command.(string)))
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
cmd := exec.Command("sh", "-c", command.(string)+" "+arguments)
|
||||||
|
cmd.Dir = working_dir
|
||||||
|
cmd.Stdout = io.MultiWriter(conn, os.Stdout)
|
||||||
|
cmd.Stderr = io.MultiWriter(conn, os.Stderr)
|
||||||
|
cmd.Stdin = conn
|
||||||
|
err = cmd.Run()
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0, nil
|
||||||
|
}
|
Loading…
Add table
Reference in a new issue