mirror of https://github.com/mpolden/echoip
67 lines
2.2 KiB
Go
67 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
flags "github.com/jessevdk/go-flags"
|
|
|
|
"os"
|
|
|
|
"github.com/mpolden/ipd/http"
|
|
"github.com/mpolden/ipd/iputil"
|
|
"github.com/mpolden/ipd/iputil/database"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func main() {
|
|
var opts struct {
|
|
CountryDBPath string `short:"f" long:"country-db" description:"Path to GeoIP country database" value-name:"FILE" default:""`
|
|
CityDBPath string `short:"c" long:"city-db" description:"Path to GeoIP city database" value-name:"FILE" default:""`
|
|
Listen string `short:"l" long:"listen" description:"Listening address" value-name:"ADDR" default:":8080"`
|
|
ReverseLookup bool `short:"r" long:"reverse-lookup" description:"Perform reverse hostname lookups"`
|
|
PortLookup bool `short:"p" long:"port-lookup" description:"Enable port lookup"`
|
|
Template string `short:"t" long:"template" description:"Path to template" default:"index.html" value-name:"FILE"`
|
|
IPHeader string `short:"H" long:"trusted-header" description:"Header to trust for remote IP, if present (e.g. X-Real-IP)" value-name:"NAME"`
|
|
LogLevel string `short:"L" long:"log-level" description:"Log level to use" default:"info" choice:"debug" choice:"info" choice:"warn" choice:"error" choice:"fatal" choice:"panic"`
|
|
}
|
|
_, err := flags.ParseArgs(&opts, os.Args)
|
|
if err != nil {
|
|
os.Exit(1)
|
|
}
|
|
|
|
log := logrus.New()
|
|
level, err := logrus.ParseLevel(opts.LogLevel)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
log.Level = level
|
|
|
|
db := database.Empty()
|
|
if opts.CountryDBPath != "" || opts.CityDBPath != "" {
|
|
db, err = database.New(opts.CountryDBPath, opts.CityDBPath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
var lookupAddr http.LookupAddr
|
|
var lookupPort http.LookupPort
|
|
if opts.ReverseLookup {
|
|
log.Println("Enabling reverse lookup")
|
|
lookupAddr = iputil.LookupAddr
|
|
}
|
|
if opts.PortLookup {
|
|
log.Println("Enabling port lookup")
|
|
lookupPort = iputil.LookupPort
|
|
}
|
|
if opts.IPHeader != "" {
|
|
log.Printf("Trusting header %s to contain correct remote IP", opts.IPHeader)
|
|
}
|
|
|
|
server := http.New(db, lookupAddr, lookupPort, log)
|
|
server.Template = opts.Template
|
|
server.IPHeader = opts.IPHeader
|
|
|
|
log.Printf("Listening on http://%s", opts.Listen)
|
|
if err := server.ListenAndServe(opts.Listen); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|