remove remote port from remoteaddr

cmd/reseed.go

@@ -0,0 +1,159 @@
+package cmd
+
+import (
+	"fmt"
+	"log"
+	"net"
+	"time"
+
+	"github.com/MDrollette/i2p-tools/reseed"
+	"github.com/codegangsta/cli"
+)
+
+func NewReseedCommand() cli.Command {
+	return cli.Command{
+		Name:   "reseed",
+		Usage:  "Start a reseed server",
+		Action: reseedAction,
+		Flags: []cli.Flag{
+			cli.StringFlag{
+				Name:  "signer",
+				Usage: "Your su3 signing ID (ex. something@mail.i2p)",
+			},
+			cli.StringFlag{
+				Name:  "tlsHost",
+				Usage: "The public hostname used on your TLS certificate",
+			},
+			cli.StringFlag{
+				Name:  "key",
+				Usage: "Path to your su3 signing private key",
+			},
+			cli.StringFlag{
+				Name:  "netdb",
+				Usage: "Path to NetDB directory containing routerInfos",
+			},
+			cli.StringFlag{
+				Name:  "tlsCert",
+				Usage: "Path to a TLS certificate",
+			},
+			cli.StringFlag{
+				Name:  "tlsKey",
+				Usage: "Path to a TLS private key",
+			},
+			cli.StringFlag{
+				Name:  "ip",
+				Value: "0.0.0.0",
+				Usage: "IP address to listen on",
+			},
+			cli.StringFlag{
+				Name:  "port",
+				Value: "8080",
+				Usage: "Port to listen on",
+			},
+			cli.IntFlag{
+				Name:  "numRi",
+				Value: 75,
+				Usage: "Number of routerInfos to include in each su3 file",
+			},
+			cli.IntFlag{
+				Name:  "numSu3",
+				Value: 0,
+				Usage: "Number of su3 files to build (0 = automatic based on size of netdb)",
+			},
+			cli.StringFlag{
+				Name:  "interval",
+				Value: "12h",
+				Usage: "Duration between SU3 cache rebuilds (ex. 12h, 15m)",
+			},
+			cli.StringFlag{
+				Name:  "prefix",
+				Value: "",
+				Usage: "Prefix path for the HTTP(S) server. (ex. /netdb)",
+			},
+			cli.BoolFlag{
+				Name:  "trustProxy",
+				Usage: "If provided, we will trust the 'X-Forwarded-For' header in requests (ex. behind cloudflare)",
+			},
+		},
+	}
+}
+
+func reseedAction(c *cli.Context) {
+	// validate flags
+	netdbDir := c.String("netdb")
+	if netdbDir == "" {
+		fmt.Println("--netdb is required")
+		return
+	}
+
+	signerId := c.String("signer")
+	if signerId == "" {
+		fmt.Println("--signer is required")
+		return
+	}
+
+	var tlsCert, tlsKey string
+	tlsHost := c.String("tlsHost")
+	if tlsHost != "" {
+		tlsKey = c.String("tlsKey")
+		// if no key is specified, default to the host.pem in the current dir
+		if tlsKey == "" {
+			tlsKey = tlsHost + ".pem"
+		}
+
+		tlsCert = c.String("tlsCert")
+		// if no certificate is specified, default to the host.crt in the current dir
+		if tlsCert == "" {
+			tlsCert = tlsHost + ".crt"
+		}
+
+		// prompt to create tls keys if they don't exist?
+		err := checkOrNewTLSCert(tlsHost, &tlsCert, &tlsKey)
+		if nil != err {
+			log.Fatalln(err)
+		}
+	}
+
+	reloadIntvl, err := time.ParseDuration(c.String("interval"))
+	if nil != err {
+		fmt.Printf("'%s' is not a valid time interval.\n", reloadIntvl)
+		return
+	}
+
+	signerKey := c.String("key")
+	// if no key is specified, default to the signerId.pem in the current dir
+	if signerKey == "" {
+		signerKey = signerFile(signerId) + ".pem"
+	}
+
+	// load our signing privKey
+	privKey, err := getOrNewSigningCert(&signerKey, signerId)
+	if nil != err {
+		log.Fatalln(err)
+	}
+
+	// create a local file netdb provider
+	netdb := reseed.NewLocalNetDb(netdbDir)
+
+	// create a reseeder
+	reseeder := reseed.NewReseeder(netdb)
+	reseeder.SigningKey = privKey
+	reseeder.SignerId = []byte(signerId)
+	reseeder.NumRi = c.Int("numRi")
+	reseeder.NumSu3 = c.Int("numSu3")
+	reseeder.RebuildInterval = reloadIntvl
+	reseeder.Start()
+
+	// create a server
+	server := reseed.NewServer(c.String("prefix"), c.Bool("trustProxy"))
+	server.Reseeder = reseeder
+	server.Addr = net.JoinHostPort(c.String("ip"), c.String("port"))
+
+	if tlsHost != "" && tlsCert != "" && tlsKey != "" {
+		log.Printf("HTTPS server started on %s\n", server.Addr)
+		log.Fatalln(server.ListenAndServeTLS(tlsCert, tlsKey))
+	} else {
+		log.Printf("HTTP server started on %s\n", server.Addr)
+		log.Fatalln(server.ListenAndServe())
+	}
+}