mirror of https://github.com/mpolden/echoip
108 lines
2.8 KiB
Go
108 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"strings"
|
|
|
|
"os"
|
|
|
|
"github.com/mpolden/echoip/http"
|
|
"github.com/mpolden/echoip/iputil"
|
|
"github.com/mpolden/echoip/iputil/geo"
|
|
"github.com/mpolden/echoip/iputil/ipstack"
|
|
parser "github.com/mpolden/echoip/iputil/paser"
|
|
ipstackApi "github.com/qioalice/ipstack"
|
|
)
|
|
|
|
type multiValueFlag []string
|
|
|
|
func (f *multiValueFlag) String() string {
|
|
return strings.Join([]string(*f), ", ")
|
|
}
|
|
|
|
func (f *multiValueFlag) Set(v string) error {
|
|
*f = append(*f, v)
|
|
return nil
|
|
}
|
|
|
|
func init() {
|
|
log.SetPrefix("echoip: ")
|
|
log.SetFlags(log.Lshortfile)
|
|
}
|
|
|
|
func main() {
|
|
var ipstackApiKey string
|
|
service := flag.String("d", "geoip", "Which database to use, 'ipstack' or 'geoip'")
|
|
flag.StringVar(&ipstackApiKey, "S", "", "IP Stack API Key")
|
|
countryFile := flag.String("f", "", "Path to GeoIP country database")
|
|
cityFile := flag.String("c", "", "Path to GeoIP city database")
|
|
asnFile := flag.String("a", "", "Path to GeoIP ASN database")
|
|
listen := flag.String("l", ":8080", "Listening address")
|
|
reverseLookup := flag.Bool("r", false, "Perform reverse hostname lookups")
|
|
portLookup := flag.Bool("p", false, "Enable port lookup")
|
|
template := flag.String("t", "html", "Path to template dir")
|
|
cacheSize := flag.Int("C", 0, "Size of response cache. Set to 0 to disable")
|
|
profile := flag.Bool("P", false, "Enables profiling handlers")
|
|
sponsor := flag.Bool("s", false, "Show sponsor logo")
|
|
var headers multiValueFlag
|
|
flag.Var(&headers, "H", "Header to trust for remote IP, if present (e.g. X-Real-IP)")
|
|
flag.Parse()
|
|
|
|
if len(flag.Args()) != 0 {
|
|
flag.Usage()
|
|
return
|
|
}
|
|
|
|
var parser parser.Parser
|
|
if *service == "geoip" {
|
|
geo, err := geo.Open(*countryFile, *cityFile, *asnFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
parser = &geo
|
|
}
|
|
|
|
if *service == "ipstack" {
|
|
if err := ipstackApi.Init(ipstackApiKey); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
ips := ipstack.IPStack{}
|
|
parser = &ips
|
|
}
|
|
|
|
cache := http.NewCache(*cacheSize)
|
|
server := http.New(parser, cache, *profile)
|
|
server.IPHeaders = headers
|
|
if _, err := os.Stat(*template); err == nil {
|
|
server.Template = *template
|
|
} else {
|
|
log.Printf("Not configuring default handler: Template not found: %s", *template)
|
|
}
|
|
if *reverseLookup {
|
|
log.Println("Enabling reverse lookup")
|
|
server.LookupAddr = iputil.LookupAddr
|
|
}
|
|
if *portLookup {
|
|
log.Println("Enabling port lookup")
|
|
server.LookupPort = iputil.LookupPort
|
|
}
|
|
if *sponsor {
|
|
log.Println("Enabling sponsor logo")
|
|
server.Sponsor = *sponsor
|
|
}
|
|
if len(headers) > 0 {
|
|
log.Printf("Trusting remote IP from header(s): %s", headers.String())
|
|
}
|
|
if *cacheSize > 0 {
|
|
log.Printf("Cache capacity set to %d", *cacheSize)
|
|
}
|
|
if *profile {
|
|
log.Printf("Enabling profiling handlers")
|
|
}
|
|
log.Printf("Listening on http://%s", *listen)
|
|
if err := server.ListenAndServe(*listen); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|