skysocks.go

/*
proxy server app for skywire visor
*/
package main

import (
	"flag"
	"os"
	"os/signal"

	"github.com/sirupsen/logrus"
	"github.com/skycoin/dmsg/buildinfo"

	"github.com/skycoin/skywire/internal/skysocks"
	"github.com/skycoin/skywire/pkg/app"
	"github.com/skycoin/skywire/pkg/app/appnet"
	"github.com/skycoin/skywire/pkg/routing"
)

const (
	netType              = appnet.TypeSkynet
	port    routing.Port = 3
)

var log = logrus.New()

func main() {
	appC := app.NewClient(nil)
	defer appC.Close()

	skysocks.Log = log

	if _, err := buildinfo.Get().WriteTo(log.Writer()); err != nil {
		log.Printf("Failed to output build info: %v", err)
	}

	var passcode = flag.String("passcode", "", "Authorize user against this passcode")
	flag.Parse()

	srv, err := skysocks.NewServer(*passcode, log)
	if err != nil {
		log.Fatal("Failed to create a new server: ", err)
	}

	l, err := appC.Listen(netType, port)
	if err != nil {
		log.Fatalf("Error listening network %v on port %d: %v\n", netType, port, err)
	}

	log.Infoln("Starting serving proxy server")

	termCh := make(chan os.Signal, 1)
	signal.Notify(termCh, os.Interrupt)

	go func() {
		<-termCh

		if err := srv.Close(); err != nil {
			log.Fatalf("Failed to close server: %v", err)
		}
	}()

	if err := srv.Serve(l); err != nil {
		log.Fatal(err)
	}
}