sevices/friendbot: Allow friendbot to include cloudflare derived IP address in request logs

services/friendbot/main.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/go-chi/chi"
 	"github.com/spf13/cobra"
+
 	"github.com/stellar/go/services/friendbot/internal"
 	"github.com/stellar/go/support/app"
 	"github.com/stellar/go/support/config"
@@ -29,6 +30,7 @@ type Config struct {
 	BaseFee                int64       `toml:"base_fee" valid:"optional"`
 	MinionBatchSize        int         `toml:"minion_batch_size" valid:"optional"`
 	SubmitTxRetriesAllowed int         `toml:"submit_tx_retries_allowed" valid:"optional"`
+	UseCloudflareIP        bool        `toml:"use_cloudflare_ip" valid:"optional"`
 }
 
 func main() {
@@ -68,7 +70,7 @@ func run(cmd *cobra.Command, args []string) {
 		log.Error(err)
 		os.Exit(1)
 	}
-	router := initRouter(fb)
+	router := initRouter(cfg, fb)
 	registerProblems()
 
 	addr := fmt.Sprintf("0.0.0.0:%d", cfg.Port)
@@ -84,8 +86,12 @@ func run(cmd *cobra.Command, args []string) {
 	})
 }
 
-func initRouter(fb *internal.Bot) *chi.Mux {
-	mux := http.NewAPIMux(log.DefaultLogger)
+func initRouter(cfg Config, fb *internal.Bot) *chi.Mux {
+	mux := chi.NewRouter()
+	// first apply XFFMiddleware so we can have the real ip in the subsequent
+	// middlewares
+	mux.Use(http.XFFMiddleware(http.XFFMiddlewareConfig{BehindCloudflare: cfg.UseCloudflareIP}))
+	mux.Use(http.NewAPIMux(log.DefaultLogger).Middlewares()...)
 
 	handler := &internal.FriendbotHandler{Friendbot: fb}
 	mux.Get("/", handler.Handle)

support/http/logging_middleware.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/go-chi/chi"
 	"github.com/go-chi/chi/middleware"
+
 	"github.com/stellar/go/support/http/mutil"
 	"github.com/stellar/go/support/log"
 )
@@ -136,6 +137,7 @@ func logEndOfRequest(
 		"subsys":   "http",
 		"path":     r.URL.String(),
 		"method":   r.Method,
+		"ip":       r.RemoteAddr,
 		"status":   mw.Status(),
 		"bytes":    mw.BytesWritten(),
 		"duration": duration,

support/http/logging_middleware_test.go

@@ -6,9 +6,10 @@ import (
 
 	"github.com/go-chi/chi"
 	"github.com/go-chi/chi/middleware"
+	"github.com/stretchr/testify/assert"
+
 	"github.com/stellar/go/support/http/httptest"
 	"github.com/stellar/go/support/log"
-	"github.com/stretchr/testify/assert"
 )
 
 // setXFFMiddleware sets "X-Forwarded-For" header to test LoggingMiddlewareWithOptions.
@@ -143,7 +144,7 @@ func TestHTTPMiddlewareWithOptions(t *testing.T) {
 		assert.Equal(t, req1, logged[2].Data["req"])
 		assert.Equal(t, "/path/1234", logged[2].Data["path"])
 		assert.Equal(t, "/path/{value}", logged[2].Data["route"])
-		assert.Equal(t, 9, len(logged[2].Data))
+		assert.Equal(t, 10, len(logged[2].Data))
 
 		assert.Equal(t, "starting request", logged[3].Message)
 		assert.Equal(t, "http", logged[3].Data["subsys"])
@@ -162,7 +163,7 @@ func TestHTTPMiddlewareWithOptions(t *testing.T) {
 		assert.Equal(t, req2, logged[4].Data["req"])
 		assert.Equal(t, "/not_found", logged[4].Data["path"])
 		assert.Equal(t, "/not_found", logged[4].Data["route"])
-		assert.Equal(t, 9, len(logged[4].Data))
+		assert.Equal(t, 10, len(logged[4].Data))
 
 		assert.Equal(t, "starting request", logged[5].Message)
 		assert.Equal(t, "http", logged[5].Data["subsys"])
@@ -181,7 +182,7 @@ func TestHTTPMiddlewareWithOptions(t *testing.T) {
 		assert.Equal(t, req3, logged[6].Data["req"])
 		assert.Equal(t, "/really_not_found", logged[6].Data["path"])
 		assert.Equal(t, "", logged[6].Data["route"])
-		assert.Equal(t, 9, len(logged[6].Data))
+		assert.Equal(t, 10, len(logged[6].Data))
 	}
 }
 

support/http/mux.go

@@ -4,6 +4,7 @@ import (
 	"github.com/go-chi/chi"
 	"github.com/go-chi/chi/middleware"
 	"github.com/rs/cors"
+
 	"github.com/stellar/go/support/log"
 )