mirror of
https://github.com/byReqz/probehost2.git
synced 2024-11-14 19:23:14 +00:00
add environment values
This commit is contained in:
parent
62564fab41
commit
ce95696ef0
@ -58,6 +58,8 @@ The app currently has 4 runtime flags:
|
|||||||
- `-x / --disable-x-forwarded-for` -- disables checking for the X-Forwarded-For header
|
- `-x / --disable-x-forwarded-for` -- disables checking for the X-Forwarded-For header
|
||||||
- `-l / --allow-private` -- allows lookups of private IP ranges
|
- `-l / --allow-private` -- allows lookups of private IP ranges
|
||||||
|
|
||||||
|
All of the Flags also have an accompanying environment value: `PROBEHOST_LOGPATH`, `PROBEHOST_ALLOW_PRIVATE`, `PROBEHOST_LISTEN_PORT` and `PROBEHOST_DISABLE_X_FORWARDED_FOR` but the options given via commandline have priority.
|
||||||
|
|
||||||
The app will log every request including the IP thats querying and show failed requests on stdout.
|
The app will log every request including the IP thats querying and show failed requests on stdout.
|
||||||
|
|
||||||
Requests that contain an X-Forwarded-For header (implying the app is behind a reverse proxy) will automatically log that address instead of the requesting IP (the proxy itself), this can be turned off with -x.
|
Requests that contain an X-Forwarded-For header (implying the app is behind a reverse proxy) will automatically log that address instead of the requesting IP (the proxy itself), this can be turned off with -x.
|
||||||
|
@ -5,6 +5,11 @@ services:
|
|||||||
container_name: probehost2
|
container_name: probehost2
|
||||||
image: byreqz/probehost2:latest
|
image: byreqz/probehost2:latest
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
|
environment:
|
||||||
|
- PROBEHOST_LOGPATH=/probehost2.log
|
||||||
|
- PROBEHOST_ALLOW_PRIVATE=false
|
||||||
|
- PROBEHOST_DISABLE_X_FORWARDED_FOR=false
|
||||||
|
- PROBEHOST_LISTEN_PORT=8000
|
||||||
ports:
|
ports:
|
||||||
- 1234:8000
|
- 1234:8000
|
||||||
volumes:
|
volumes:
|
||||||
|
39
main.go
39
main.go
@ -6,6 +6,7 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net"
|
"net"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
flag "github.com/spf13/pflag"
|
flag "github.com/spf13/pflag"
|
||||||
@ -20,17 +21,41 @@ var disablexforwardedfor bool
|
|||||||
var allowprivate bool
|
var allowprivate bool
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
var logfilepath string
|
|
||||||
flag.StringVarP(&logfilepath, "logfilepath", "o","probehost2.log", "sets the output file for the log")
|
|
||||||
flag.IntVarP(&listenport, "port", "p", 8000, "sets the port to listen on")
|
|
||||||
flag.BoolVarP(&disablexforwardedfor, "disable-x-forwarded-for", "x", false, "whether to show x-forwarded-for or the requesting IP")
|
|
||||||
flag.BoolVarP(&allowprivate, "allow-private", "l", false, "whether to show lookups of private IP ranges")
|
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
logstdout.SetFormatter(&log.TextFormatter{
|
logstdout.SetFormatter(&log.TextFormatter{
|
||||||
FullTimestamp: true})
|
FullTimestamp: true})
|
||||||
logstdout.SetOutput(os.Stdout)
|
logstdout.SetOutput(os.Stdout)
|
||||||
logstdout.SetLevel(log.InfoLevel)
|
logstdout.SetLevel(log.InfoLevel)
|
||||||
|
var logfilepath string
|
||||||
|
|
||||||
|
if _, exists := os.LookupEnv("PROBEHOST_LOGPATH"); exists == true {
|
||||||
|
logfilepath, _ = os.LookupEnv("PROBEHOST_LOGPATH")
|
||||||
|
} else {
|
||||||
|
logfilepath = "probehost2.log"
|
||||||
|
}
|
||||||
|
if exists, _ := os.LookupEnv("PROBEHOST_ALLOW_PRIVATE"); exists == "true" {
|
||||||
|
allowprivate = true
|
||||||
|
} else {
|
||||||
|
allowprivate = false
|
||||||
|
}
|
||||||
|
if envvalue, exists := os.LookupEnv("PROBEHOST_LISTEN_PORT"); exists == true {
|
||||||
|
var err error
|
||||||
|
listenport, err = strconv.Atoi(envvalue)
|
||||||
|
if err != nil {
|
||||||
|
logstdout.Fatal("Failed to read PROBEHOST_LISTEN_PORT: ", err.Error())
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
listenport = 8000
|
||||||
|
}
|
||||||
|
if exists, _ := os.LookupEnv("PROBEHOST_DISABLE_X_FORWARDED_FOR"); exists == "true" {
|
||||||
|
disablexforwardedfor = true
|
||||||
|
} else {
|
||||||
|
disablexforwardedfor = false
|
||||||
|
}
|
||||||
|
flag.StringVarP(&logfilepath, "logfilepath", "o", logfilepath, "sets the output file for the log")
|
||||||
|
flag.IntVarP(&listenport, "port", "p", listenport, "sets the port to listen on")
|
||||||
|
flag.BoolVarP(&disablexforwardedfor, "disable-x-forwarded-for", "x", disablexforwardedfor, "whether to show x-forwarded-for or the requesting IP")
|
||||||
|
flag.BoolVarP(&allowprivate, "allow-private", "l", allowprivate, "whether to show lookups of private IP ranges")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
logpath, err := os.OpenFile(logfilepath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0660)
|
logpath, err := os.OpenFile(logfilepath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0660)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
Loading…
Reference in New Issue
Block a user