http: Extract IP from RemoteAddr correctly

This commit is contained in:
Jonathan Sokolowski 2015-05-20 15:23:41 +10:00
parent 513a8edc67
commit be2538aca3
1 changed files with 9 additions and 1 deletions

View File

@ -2,6 +2,7 @@ package http
import (
"io"
"net"
"net/http"
"strings"
"time"
@ -53,6 +54,13 @@ func handleLogical(core *vault.Core) http.Handler {
}
}
// http.Server will set RemoteAddr to an "IP:port" string
var remoteAddr string
remoteAddr, _, err := net.SplitHostPort(r.RemoteAddr)
if err != nil {
remoteAddr = ""
}
// Make the internal request. We attach the connection info
// as well in case this is an authentication request that requires
// it. Vault core handles stripping this if we need to.
@ -61,7 +69,7 @@ func handleLogical(core *vault.Core) http.Handler {
Path: path,
Data: req,
Connection: &logical.Connection{
RemoteAddr: r.RemoteAddr,
RemoteAddr: remoteAddr,
ConnState: r.TLS,
},
}))