kenmirrors
/
ifconfig.io
mirror of https://github.com/georgyo/ifconfig.io.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add ability to specify country code header

This commit is contained in:
Lionel Nicolas 2023-01-30 20:09:51 -05:00
parent c2d3e43021
commit 096e8511cf
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
main.go
View File

@ -20,6 +20,7 @@ type Configuration struct {
port string // HTTP Port port string // HTTP Port
proxy_listener string // Proxy Protocol Listener proxy_listener string // Proxy Protocol Listener
ipheader string // Header to overwrite the remote IP ipheader string // Header to overwrite the remote IP
countryheader string // Header to find country code associated to remote IP
tls bool // TLS enabled tls bool // TLS enabled
tlscert string // TLS Cert Path tlscert string // TLS Cert Path
tlskey string // TLS Cert Key Path tlskey string // TLS Cert Key Path
@ -38,6 +39,8 @@ func init() {
// Most common alternative would be X-Forwarded-For // Most common alternative would be X-Forwarded-For
ipheader := getEnvWithDefault("FORWARD_IP_HEADER", "CF-Connecting-IP") ipheader := getEnvWithDefault("FORWARD_IP_HEADER", "CF-Connecting-IP")
countryheader := getEnvWithDefault("COUNTRY_CODE_HEADER", "CF-IPCountry")
tlsenabled := getEnvWithDefault("TLS", "0") tlsenabled := getEnvWithDefault("TLS", "0")
tlsport := getEnvWithDefault("TLSPORT", "8443") tlsport := getEnvWithDefault("TLSPORT", "8443")
tlscert := getEnvWithDefault("TLSCERT", "/opt/ifconfig/.cf/ifconfig.io.crt") tlscert := getEnvWithDefault("TLSCERT", "/opt/ifconfig/.cf/ifconfig.io.crt")
@ -49,6 +52,7 @@ func init() {
port: port, port: port,
proxy_listener: proxy_listener, proxy_listener: proxy_listener,
ipheader: ipheader, ipheader: ipheader,
countryheader: countryheader,
tls: tlsenabled == "1", tls: tlsenabled == "1",
tlscert: tlscert, tlscert: tlscert,
tlskey: tlskey, tlskey: tlskey,
@ -118,7 +122,7 @@ func mainHandler(c *gin.Context) {
c.Set("mime", c.Request.Header.Get("Accept")) c.Set("mime", c.Request.Header.Get("Accept"))
c.Set("referer", c.Request.Header.Get("Referer")) c.Set("referer", c.Request.Header.Get("Referer"))
c.Set("forwarded", c.Request.Header.Get("X-Forwarded-For")) c.Set("forwarded", c.Request.Header.Get("X-Forwarded-For"))
c.Set("country_code", c.Request.Header.Get("CF-IPCountry")) c.Set("country_code", c.Request.Header.Get(configuration.countryheader))
c.Set("host", ip.IP.String()) c.Set("host", ip.IP.String())
// Only lookup hostname if the results are going to need it. // Only lookup hostname if the results are going to need it.