2019-06-30 17:07:58 -05:00
|
|
|
// Copyright 2015 Matthew Holt and The Caddy Authors
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
//
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
|
2019-05-04 14:21:20 -05:00
|
|
|
package caddyhttp
|
|
|
|
|
|
|
|
import (
|
2019-05-07 10:56:13 -05:00
|
|
|
"net"
|
2019-05-04 14:21:20 -05:00
|
|
|
"net/http"
|
2019-07-16 13:27:11 -05:00
|
|
|
"net/textproto"
|
2019-05-20 11:59:20 -05:00
|
|
|
"path"
|
2019-07-16 13:27:11 -05:00
|
|
|
"strconv"
|
2019-05-04 14:21:20 -05:00
|
|
|
"strings"
|
2019-05-14 15:14:05 -05:00
|
|
|
|
2019-07-02 13:37:06 -05:00
|
|
|
"github.com/caddyserver/caddy/v2"
|
2019-05-04 14:21:20 -05:00
|
|
|
)
|
|
|
|
|
2019-06-14 12:58:28 -05:00
|
|
|
func addHTTPVarsToReplacer(repl caddy.Replacer, req *http.Request, w http.ResponseWriter) {
|
2019-07-16 13:27:11 -05:00
|
|
|
httpVars := func(key string) (string, bool) {
|
2019-05-20 11:59:20 -05:00
|
|
|
if req != nil {
|
2019-07-16 13:27:11 -05:00
|
|
|
if strings.HasPrefix(key, queryReplPrefix) {
|
|
|
|
vals := req.URL.Query()[key[len(queryReplPrefix):]]
|
|
|
|
// always return true, since the query param might
|
|
|
|
// be present only in some requests
|
|
|
|
return strings.Join(vals, ","), true
|
|
|
|
}
|
|
|
|
|
|
|
|
if strings.HasPrefix(key, reqHeaderReplPrefix) {
|
|
|
|
field := key[len(reqHeaderReplPrefix):]
|
|
|
|
vals := req.Header[textproto.CanonicalMIMEHeaderKey(field)]
|
|
|
|
// always return true, since the header field might
|
|
|
|
// be present only in some requests
|
|
|
|
return strings.Join(vals, ","), true
|
|
|
|
}
|
|
|
|
|
|
|
|
if strings.HasPrefix(key, cookieReplPrefix) {
|
|
|
|
name := key[len(cookieReplPrefix):]
|
|
|
|
for _, cookie := range req.Cookies() {
|
|
|
|
if strings.EqualFold(name, cookie.Name) {
|
|
|
|
// always return true, since the cookie might
|
|
|
|
// be present only in some requests
|
|
|
|
return cookie.Value, true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
switch key {
|
|
|
|
case "http.request.host":
|
2019-05-20 11:59:20 -05:00
|
|
|
host, _, err := net.SplitHostPort(req.Host)
|
|
|
|
if err != nil {
|
2019-07-16 13:27:11 -05:00
|
|
|
return req.Host, true // OK; there probably was no port
|
2019-05-20 11:59:20 -05:00
|
|
|
}
|
2019-07-16 13:27:11 -05:00
|
|
|
return host, true
|
|
|
|
case "http.request.hostport":
|
|
|
|
return req.Host, true
|
|
|
|
case "http.request.method":
|
|
|
|
return req.Method, true
|
|
|
|
case "http.request.port":
|
2019-05-20 11:59:20 -05:00
|
|
|
_, port, _ := net.SplitHostPort(req.Host)
|
2019-07-16 13:27:11 -05:00
|
|
|
return port, true
|
|
|
|
case "http.request.scheme":
|
2019-05-20 11:59:20 -05:00
|
|
|
if req.TLS != nil {
|
2019-07-16 13:27:11 -05:00
|
|
|
return "https", true
|
2019-05-20 11:59:20 -05:00
|
|
|
}
|
2019-07-16 13:27:11 -05:00
|
|
|
return "http", true
|
|
|
|
case "http.request.uri":
|
|
|
|
return req.URL.RequestURI(), true
|
|
|
|
case "http.request.uri.path":
|
|
|
|
return req.URL.Path, true
|
|
|
|
case "http.request.uri.path.file":
|
2019-05-20 11:59:20 -05:00
|
|
|
_, file := path.Split(req.URL.Path)
|
2019-07-16 13:27:11 -05:00
|
|
|
return file, true
|
|
|
|
case "http.request.uri.path.dir":
|
2019-05-20 11:59:20 -05:00
|
|
|
dir, _ := path.Split(req.URL.Path)
|
2019-07-16 13:27:11 -05:00
|
|
|
return dir, true
|
|
|
|
case "http.request.uri.query":
|
|
|
|
return req.URL.RawQuery, true
|
2019-05-21 14:10:14 -05:00
|
|
|
}
|
2019-07-16 13:27:11 -05:00
|
|
|
|
|
|
|
if strings.HasPrefix(key, respHeaderReplPrefix) {
|
|
|
|
field := key[len(respHeaderReplPrefix):]
|
|
|
|
vals := w.Header()[textproto.CanonicalMIMEHeaderKey(field)]
|
|
|
|
// always return true, since the header field might
|
|
|
|
// be present only in some requests
|
|
|
|
return strings.Join(vals, ","), true
|
2019-05-20 11:59:20 -05:00
|
|
|
}
|
2019-05-04 14:21:20 -05:00
|
|
|
|
2019-07-16 13:27:11 -05:00
|
|
|
if strings.HasPrefix(key, hostLabelReplPrefix) {
|
|
|
|
idxStr := key[len(hostLabelReplPrefix):]
|
|
|
|
idx, err := strconv.Atoi(idxStr)
|
|
|
|
if err != nil {
|
|
|
|
return "", false
|
|
|
|
}
|
|
|
|
hostLabels := strings.Split(req.Host, ".")
|
|
|
|
if idx < 0 {
|
|
|
|
return "", false
|
|
|
|
}
|
|
|
|
if idx >= len(hostLabels) {
|
|
|
|
return "", true
|
|
|
|
}
|
|
|
|
return hostLabels[idx], true
|
2019-05-20 11:59:20 -05:00
|
|
|
}
|
2019-07-08 17:46:55 -05:00
|
|
|
|
2019-07-16 13:27:11 -05:00
|
|
|
if strings.HasPrefix(key, pathPartsReplPrefix) {
|
|
|
|
idxStr := key[len(pathPartsReplPrefix):]
|
|
|
|
idx, err := strconv.Atoi(idxStr)
|
|
|
|
if err != nil {
|
|
|
|
return "", false
|
|
|
|
}
|
|
|
|
pathParts := strings.Split(req.URL.Path, "/")
|
|
|
|
if len(pathParts) > 0 && pathParts[0] == "" {
|
|
|
|
pathParts = pathParts[1:]
|
|
|
|
}
|
|
|
|
if idx < 0 {
|
|
|
|
return "", false
|
|
|
|
}
|
|
|
|
if idx >= len(pathParts) {
|
|
|
|
return "", true
|
|
|
|
}
|
|
|
|
return pathParts[idx], true
|
2019-07-08 17:46:55 -05:00
|
|
|
}
|
2019-05-16 12:46:17 -05:00
|
|
|
}
|
|
|
|
|
2019-05-20 11:59:20 -05:00
|
|
|
if w != nil {
|
2019-07-16 13:27:11 -05:00
|
|
|
if strings.HasPrefix(key, respHeaderReplPrefix) {
|
|
|
|
field := key[len(respHeaderReplPrefix):]
|
|
|
|
vals := w.Header()[textproto.CanonicalMIMEHeaderKey(field)]
|
|
|
|
// always return true, since the header field might
|
|
|
|
// be present only in some responses
|
|
|
|
return strings.Join(vals, ","), true
|
2019-05-20 11:59:20 -05:00
|
|
|
}
|
2019-05-14 15:14:05 -05:00
|
|
|
}
|
2019-05-04 14:21:20 -05:00
|
|
|
|
2019-07-16 13:27:11 -05:00
|
|
|
return "", false
|
2019-05-20 11:59:20 -05:00
|
|
|
}
|
2019-05-04 14:21:20 -05:00
|
|
|
|
2019-05-20 11:59:20 -05:00
|
|
|
repl.Map(httpVars)
|
|
|
|
}
|
2019-07-16 13:27:11 -05:00
|
|
|
|
|
|
|
const (
|
|
|
|
queryReplPrefix = "http.request.uri.query."
|
|
|
|
reqHeaderReplPrefix = "http.request.header."
|
|
|
|
cookieReplPrefix = "http.request.cookie."
|
|
|
|
hostLabelReplPrefix = "http.request.host.labels."
|
|
|
|
pathPartsReplPrefix = "http.request.uri.path."
|
|
|
|
respHeaderReplPrefix = "http.response.header."
|
|
|
|
)
|