2021-01-25 10:39:05 +00:00
|
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
|
2021-01-25 12:23:03 +00:00
|
|
|
// go-roomsrv hosts the database and p2p server for replication.
|
2021-01-25 10:39:05 +00:00
|
|
|
// It supplies various flags to contol options.
|
2021-01-25 12:23:03 +00:00
|
|
|
// See 'go-roomsrv -h' for a list and their usage.
|
2021-01-25 10:39:05 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"encoding/base64"
|
|
|
|
"flag"
|
|
|
|
"fmt"
|
2021-01-25 12:23:03 +00:00
|
|
|
"net"
|
2021-01-25 10:39:05 +00:00
|
|
|
"os"
|
|
|
|
"os/signal"
|
|
|
|
"os/user"
|
|
|
|
"path/filepath"
|
2021-01-25 12:23:03 +00:00
|
|
|
"strings"
|
2021-01-25 10:39:05 +00:00
|
|
|
"syscall"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
// debug
|
|
|
|
"net/http"
|
|
|
|
_ "net/http/pprof"
|
|
|
|
|
|
|
|
kitlog "github.com/go-kit/kit/log"
|
|
|
|
"github.com/go-kit/kit/log/level"
|
|
|
|
"github.com/pkg/errors"
|
2021-01-25 12:23:03 +00:00
|
|
|
"go.cryptoscope.co/muxrpc/v2/debug"
|
|
|
|
|
|
|
|
"go.mindeco.de/ssb-rooms/roomsrv"
|
|
|
|
mksrv "go.mindeco.de/ssb-rooms/roomsrv"
|
2021-01-25 10:39:05 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
// flags
|
|
|
|
flagDisableUNIXSock bool
|
|
|
|
|
|
|
|
listenAddr string
|
|
|
|
debugAddr string
|
|
|
|
logToFile string
|
|
|
|
repoDir string
|
|
|
|
|
|
|
|
// helper
|
|
|
|
log kitlog.Logger
|
|
|
|
|
|
|
|
// juicy bits
|
|
|
|
appKey string
|
|
|
|
)
|
|
|
|
|
|
|
|
// Version and Build are set by ldflags
|
|
|
|
var (
|
|
|
|
Version = "snapshot"
|
|
|
|
Build = ""
|
|
|
|
|
|
|
|
flagPrintVersion bool
|
|
|
|
)
|
|
|
|
|
|
|
|
func checkFatal(err error) {
|
|
|
|
checkAndLog(err)
|
|
|
|
if err != nil {
|
|
|
|
os.Exit(1)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func checkAndLog(err error) {
|
|
|
|
if err != nil {
|
|
|
|
level.Error(log).Log("event", "fatal error", "err", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func initFlags() {
|
|
|
|
u, err := user.Current()
|
|
|
|
checkFatal(err)
|
|
|
|
|
|
|
|
flag.StringVar(&appKey, "shscap", "1KHLiKZvAvjbY1ziZEHMXawbCEIM6qwjCDm3VYRan/s=", "secret-handshake app-key (or capability)")
|
|
|
|
|
|
|
|
flag.StringVar(&listenAddr, "l", ":8008", "address to listen on")
|
|
|
|
|
|
|
|
flag.BoolVar(&flagDisableUNIXSock, "nounixsock", false, "disable the UNIX socket RPC interface")
|
|
|
|
|
|
|
|
flag.StringVar(&repoDir, "repo", filepath.Join(u.HomeDir, ".ssb-go"), "where to put the log and indexes")
|
|
|
|
|
|
|
|
flag.StringVar(&debugAddr, "dbg", "localhost:6078", "listen addr for metrics and pprof HTTP server")
|
|
|
|
flag.StringVar(&logToFile, "path", "", "where to write debug output to (otherwise just stderr)")
|
|
|
|
|
|
|
|
flag.BoolVar(&flagPrintVersion, "version", false, "print version number and build date")
|
|
|
|
|
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
if logToFile != "" {
|
|
|
|
logDir := filepath.Join(repoDir, logToFile)
|
|
|
|
os.MkdirAll(logDir, 0700) // nearly everything is a log here so..
|
|
|
|
logFileName := fmt.Sprintf("%s-%s.log",
|
|
|
|
filepath.Base(os.Args[0]),
|
|
|
|
time.Now().Format("2006-01-02_15-04"))
|
|
|
|
logFile, err := os.Create(filepath.Join(logDir, logFileName))
|
|
|
|
if err != nil {
|
|
|
|
panic(err) // logging not ready yet...
|
|
|
|
}
|
2021-01-25 12:23:03 +00:00
|
|
|
log = kitlog.NewJSONLogger(kitlog.NewSyncWriter(logFile))
|
2021-01-25 10:39:05 +00:00
|
|
|
} else {
|
2021-01-25 12:23:03 +00:00
|
|
|
log = kitlog.NewLogfmtLogger(os.Stderr)
|
2021-01-25 10:39:05 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-01-25 12:23:03 +00:00
|
|
|
func runroomsrv() error {
|
2021-01-25 10:39:05 +00:00
|
|
|
// DEBUGGING
|
|
|
|
// runtime.SetMutexProfileFraction(1)
|
|
|
|
// runtime.SetBlockProfileRate(1)
|
|
|
|
// DEBUGGING
|
|
|
|
|
|
|
|
initFlags()
|
|
|
|
|
|
|
|
if flagPrintVersion {
|
|
|
|
log.Log("version", Version, "build", Build)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
|
|
|
|
|
|
ak, err := base64.StdEncoding.DecodeString(appKey)
|
|
|
|
if err != nil {
|
|
|
|
return errors.Wrap(err, "application key")
|
|
|
|
}
|
|
|
|
|
2021-01-25 12:23:03 +00:00
|
|
|
opts := []roomsrv.Option{
|
|
|
|
roomsrv.WithLogger(log),
|
|
|
|
roomsrv.WithAppKey(ak),
|
|
|
|
roomsrv.WithRepoPath(repoDir),
|
|
|
|
roomsrv.WithListenAddr(listenAddr),
|
2021-01-25 10:39:05 +00:00
|
|
|
}
|
|
|
|
|
2021-01-25 12:23:03 +00:00
|
|
|
if !flagDisableUNIXSock {
|
|
|
|
opts = append(opts, roomsrv.WithUNIXSocket())
|
|
|
|
}
|
2021-01-25 10:39:05 +00:00
|
|
|
|
2021-01-25 12:23:03 +00:00
|
|
|
if logToFile != "" {
|
|
|
|
opts = append(opts, roomsrv.WithPostSecureConnWrapper(func(conn net.Conn) (net.Conn, error) {
|
|
|
|
parts := strings.Split(conn.RemoteAddr().String(), "|")
|
|
|
|
|
|
|
|
if len(parts) != 2 {
|
|
|
|
return conn, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
muxrpcDumpDir := filepath.Join(
|
|
|
|
repoDir,
|
|
|
|
logToFile,
|
|
|
|
parts[1], // key first
|
|
|
|
parts[0],
|
|
|
|
)
|
|
|
|
|
|
|
|
return debug.WrapDump(muxrpcDumpDir, conn)
|
|
|
|
}))
|
|
|
|
}
|
2021-01-25 10:39:05 +00:00
|
|
|
|
|
|
|
if debugAddr != "" {
|
|
|
|
go func() {
|
|
|
|
// http.Handle("/metrics", promhttp.Handler())
|
|
|
|
log.Log("starting", "metrics", "addr", debugAddr)
|
|
|
|
err := http.ListenAndServe(debugAddr, nil)
|
|
|
|
checkAndLog(err)
|
|
|
|
}()
|
|
|
|
}
|
|
|
|
|
2021-01-25 12:23:03 +00:00
|
|
|
roomsrv, err := mksrv.New(opts...)
|
2021-01-25 10:39:05 +00:00
|
|
|
if err != nil {
|
|
|
|
return errors.Wrap(err, "failed to instantiate ssb server")
|
|
|
|
}
|
|
|
|
|
|
|
|
c := make(chan os.Signal)
|
|
|
|
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
|
|
|
|
go func() {
|
|
|
|
sig := <-c
|
|
|
|
level.Warn(log).Log("event", "killed", "msg", "received signal, shutting down", "signal", sig.String())
|
|
|
|
cancel()
|
2021-01-25 12:23:03 +00:00
|
|
|
roomsrv.Shutdown()
|
2021-01-25 10:39:05 +00:00
|
|
|
time.Sleep(2 * time.Second)
|
|
|
|
|
2021-01-25 12:23:03 +00:00
|
|
|
err := roomsrv.Close()
|
2021-01-25 10:39:05 +00:00
|
|
|
checkAndLog(err)
|
|
|
|
|
|
|
|
time.Sleep(2 * time.Second)
|
|
|
|
os.Exit(0)
|
|
|
|
}()
|
|
|
|
|
2021-01-25 12:23:03 +00:00
|
|
|
level.Info(log).Log("event", "serving", "ID", roomsrv.KeyPair.Feed.Ref(), "addr", listenAddr, "version", Version, "build", Build)
|
2021-01-25 10:39:05 +00:00
|
|
|
for {
|
|
|
|
// Note: This is where the serving starts ;)
|
2021-01-25 12:23:03 +00:00
|
|
|
err = roomsrv.Network.Serve(ctx)
|
2021-01-25 10:39:05 +00:00
|
|
|
if err != nil {
|
2021-01-25 12:23:03 +00:00
|
|
|
level.Warn(log).Log("event", "roomsrv node.Serve returned", "err", err)
|
2021-01-25 10:39:05 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
select {
|
|
|
|
case <-ctx.Done():
|
2021-01-25 12:23:03 +00:00
|
|
|
err := roomsrv.Close()
|
2021-01-25 10:39:05 +00:00
|
|
|
return err
|
|
|
|
default:
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func main() {
|
2021-01-25 12:23:03 +00:00
|
|
|
if err := runroomsrv(); err != nil {
|
2021-01-25 10:39:05 +00:00
|
|
|
fmt.Fprintf(os.Stderr, "go-ssb-tunnel: %s\n", err)
|
|
|
|
os.Exit(1)
|
|
|
|
}
|
|
|
|
}
|