75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
|
// waste-daemon: the local peer process.
|
||
|
|
// Run one of these on each friend's machine.
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"net"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/waste-go/internal/crypto"
|
||
|
|
"github.com/waste-go/internal/ipc"
|
||
|
|
"github.com/waste-go/internal/mesh"
|
||
|
|
"github.com/waste-go/internal/nat"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
dataDir := flag.String("data-dir", "~/.waste", "path to identity/config directory")
|
||
|
|
alias := flag.String("alias", "anon", "display name shown to peers (advisory only)")
|
||
|
|
peerPort := flag.Int("peer-port", 17338, "port to accept incoming peer connections")
|
||
|
|
ipcPort := flag.Int("ipc-port", 17337, "port for local IPC (UI connects here)")
|
||
|
|
relay := flag.String("relay", "", "optional relay server address (host:port)")
|
||
|
|
flag.Parse()
|
||
|
|
|
||
|
|
// Load or generate identity keypair
|
||
|
|
id, err := crypto.LoadOrCreate(expandHome(*dataDir), *alias)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("identity: %v", err)
|
||
|
|
}
|
||
|
|
log.Printf("daemon: local peer id: %s alias: %s", id.PeerID().Short(), id.Alias)
|
||
|
|
|
||
|
|
// Shared mesh state
|
||
|
|
m := mesh.New(id)
|
||
|
|
|
||
|
|
// Peer listener (inbound connections from other nodes)
|
||
|
|
go func() {
|
||
|
|
addr := fmt.Sprintf("0.0.0.0:%d", *peerPort)
|
||
|
|
ln, err := net.Listen("tcp", addr)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("peer listener: %v", err)
|
||
|
|
}
|
||
|
|
log.Printf("daemon: listening for peers on %s", addr)
|
||
|
|
for {
|
||
|
|
conn, err := ln.Accept()
|
||
|
|
if err != nil {
|
||
|
|
log.Printf("peer accept error: %v", err)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
go mesh.HandleConn(conn, m, false)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
// NAT traversal / relay client
|
||
|
|
go func() {
|
||
|
|
if err := nat.Run(m, *relay); err != nil {
|
||
|
|
log.Printf("nat: %v", err)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
// IPC server — blocks until error
|
||
|
|
if err := ipc.Run(m, *ipcPort); err != nil {
|
||
|
|
log.Fatalf("ipc: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func expandHome(path string) string {
|
||
|
|
if len(path) >= 2 && path[:2] == "~/" {
|
||
|
|
if home, err := os.UserHomeDir(); err == nil {
|
||
|
|
return home + path[1:]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return path
|
||
|
|
}
|