2019-09-10 02:23:27 +08: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.
package reverseproxy
import (
2023-05-16 02:14:50 +08:00
"fmt"
2019-09-10 02:23:27 +08:00
"net/http"
2020-02-28 11:56:24 +08:00
"reflect"
2019-09-10 02:23:27 +08:00
"strconv"
"strings"
"github.com/caddyserver/caddy/v2"
"github.com/caddyserver/caddy/v2/caddyconfig"
"github.com/caddyserver/caddy/v2/caddyconfig/caddyfile"
"github.com/caddyserver/caddy/v2/caddyconfig/httpcaddyfile"
"github.com/caddyserver/caddy/v2/modules/caddyhttp"
2019-09-21 03:13:49 +08:00
"github.com/caddyserver/caddy/v2/modules/caddyhttp/headers"
2022-05-06 22:50:26 +08:00
"github.com/caddyserver/caddy/v2/modules/caddyhttp/rewrite"
2019-09-10 02:23:27 +08:00
"github.com/dustin/go-humanize"
)
func init ( ) {
httpcaddyfile . RegisterHandlerDirective ( "reverse_proxy" , parseCaddyfile )
reverseproxy: copy_response and copy_response_headers for handle_response routes (#4391)
* reverseproxy: New `copy_response` handler for `handle_response` routes
Followup to #4298 and #4388.
This adds a new `copy_response` handler which may only be used in `reverse_proxy`'s `handle_response` routes, which can be used to actually copy the proxy response downstream.
Previously, if `handle_response` was used (with routes, not the status code mode), it was impossible to use the upstream's response body at all, because we would always close the body, expecting the routes to write a new body from scratch.
To implement this, I had to refactor `h.reverseProxy()` to move all the code that came after the `HandleResponse` loop into a new function. This new function `h.finalizeResponse()` takes care of preparing the response by removing extra headers, dealing with trailers, then copying the headers and body downstream.
Since basically what we want `copy_response` to do is invoke `h.finalizeResponse()` at a configurable point in time, we need to pass down the proxy handler, the response, and some other state via a new `req.WithContext(ctx)`. Wrapping a new context is pretty much the only way we have to jump a few layers in the HTTP middleware chain and let a handler pick up this information. Feels a bit dirty, but it works.
Also fixed a bug with the `http.reverse_proxy.upstream.duration` placeholder, it always had the same duration as `http.reverse_proxy.upstream.latency`, but the former was meant to be the time taken for the roundtrip _plus_ copying/writing the response.
* Delete the "Content-Length" header if we aren't copying
Fixes a bug where the Content-Length will mismatch the actual bytes written if we skipped copying the response, so we get a message like this when using curl:
```
curl: (18) transfer closed with 18 bytes remaining to read
```
To replicate:
```
{
admin off
debug
}
:8881 {
reverse_proxy 127.0.0.1:8882 {
@200 status 200
handle_response @200 {
header Foo bar
}
}
}
:8882 {
header Content-Type application/json
respond `{"hello": "world"}` 200
}
```
* Implement `copy_response_headers`, with include/exclude list support
* Apply suggestions from code review
Co-authored-by: Matt Holt <mholt@users.noreply.github.com>
2022-03-10 02:00:51 +08:00
httpcaddyfile . RegisterHandlerDirective ( "copy_response" , parseCopyResponseCaddyfile )
httpcaddyfile . RegisterHandlerDirective ( "copy_response_headers" , parseCopyResponseHeadersCaddyfile )
2019-09-10 02:23:27 +08:00
}
func parseCaddyfile ( h httpcaddyfile . Helper ) ( caddyhttp . MiddlewareHandler , error ) {
rp := new ( Handler )
err := rp . UnmarshalCaddyfile ( h . Dispenser )
if err != nil {
return nil , err
}
2021-05-03 02:39:06 +08:00
err = rp . FinalizeUnmarshalCaddyfile ( h )
if err != nil {
return nil , err
}
2019-09-10 02:23:27 +08:00
return rp , nil
}
// UnmarshalCaddyfile sets up the handler from Caddyfile tokens. Syntax:
//
2022-08-26 11:42:48 +08:00
// reverse_proxy [<matcher>] [<upstreams...>] {
// # backends
// to <upstreams...>
// dynamic <name> [...]
2019-09-10 02:23:27 +08:00
//
2022-08-26 11:42:48 +08:00
// # load balancing
// lb_policy <name> [<options...>]
// lb_retries <retries>
// lb_try_duration <duration>
// lb_try_interval <interval>
// lb_retry_match <request-matcher>
2019-09-10 02:23:27 +08:00
//
2022-08-26 11:42:48 +08:00
// # active health checking
// health_uri <uri>
// health_port <port>
// health_interval <interval>
// health_timeout <duration>
// health_status <status>
// health_body <regexp>
// health_headers {
// <field> [<values...>]
// }
2019-09-10 02:23:27 +08:00
//
2022-08-26 11:42:48 +08:00
// # passive health checking
// fail_duration <duration>
// max_fails <num>
// unhealthy_status <status>
// unhealthy_latency <duration>
// unhealthy_request_count <num>
2019-09-10 02:23:27 +08:00
//
2022-08-26 11:42:48 +08:00
// # streaming
2023-06-20 05:54:43 +08:00
// flush_interval <duration>
2022-08-26 11:42:48 +08:00
// buffer_requests
// buffer_responses
2023-06-20 05:54:43 +08:00
// max_buffer_size <size>
// stream_timeout <duration>
// stream_close_delay <duration>
2019-11-28 02:51:32 +08:00
//
2022-08-26 11:42:48 +08:00
// # request manipulation
// trusted_proxies [private_ranges] <ranges...>
// header_up [+|-]<field> [<value|regexp> [<replacement>]]
// header_down [+|-]<field> [<value|regexp> [<replacement>]]
// method <method>
// rewrite <to>
2019-09-21 03:13:49 +08:00
//
2022-08-26 11:42:48 +08:00
// # round trip
// transport <name> {
// ...
// }
2021-05-03 02:39:06 +08:00
//
2022-08-26 11:42:48 +08:00
// # optionally intercept responses from upstream
// @name {
// status <code...>
// header <field> [<value>]
// }
// replace_status [<matcher>] <status_code>
// handle_response [<matcher>] {
// <directives...>
2022-03-13 14:38:11 +08:00
//
2022-08-26 11:42:48 +08:00
// # special directives only available in handle_response
// copy_response [<matcher>] [<status>] {
// status <status>
// }
// copy_response_headers [<matcher>] {
// include <fields...>
// exclude <fields...>
// }
// }
// }
2019-09-10 02:23:27 +08:00
//
2020-02-28 11:56:24 +08:00
// Proxy upstream addresses should be network dial addresses such
// as `host:port`, or a URL such as `scheme://host:port`. Scheme
// and port may be inferred from other parts of the address/URL; if
// either are missing, defaults to HTTP.
2021-05-03 02:39:06 +08:00
//
// The FinalizeUnmarshalCaddyfile method should be called after this
// to finalize parsing of "handle_response" blocks, if possible.
2019-09-10 02:23:27 +08:00
func ( h * Handler ) UnmarshalCaddyfile ( d * caddyfile . Dispenser ) error {
2020-02-28 11:56:24 +08:00
// currently, all backends must use the same scheme/protocol (the
// underlying JSON does not yet support per-backend transports)
var commonScheme string
// we'll wait until the very end of parsing before
// validating and encoding the transport
var transport http . RoundTripper
var transportModuleName string
2021-05-03 02:39:06 +08:00
// collect the response matchers defined as subdirectives
// prefixed with "@" for use with "handle_response" blocks
h . responseMatchers = make ( map [ string ] caddyhttp . ResponseMatcher )
2020-03-25 00:53:53 +08:00
// appendUpstream creates an upstream for address and adds
2023-04-11 04:08:40 +08:00
// it to the list.
2020-03-25 00:53:53 +08:00
appendUpstream := func ( address string ) error {
2023-08-06 05:30:02 +08:00
pa , err := parseUpstreamDialAddress ( address )
2020-03-25 00:53:53 +08:00
if err != nil {
2022-03-06 07:34:19 +08:00
return d . WrapErr ( err )
}
// the underlying JSON does not yet support different
// transports (protocols or schemes) to each backend,
// so we remember the last one we see and compare them
2023-08-06 05:30:02 +08:00
if commonScheme != "" && pa . scheme != commonScheme {
2022-03-06 07:34:19 +08:00
return d . Errf ( "for now, all proxy upstreams must use the same scheme (transport protocol); expecting '%s://' but got '%s://'" ,
2023-08-06 05:30:02 +08:00
commonScheme , pa . scheme )
2020-03-25 00:53:53 +08:00
}
2023-08-06 05:30:02 +08:00
commonScheme = pa . scheme
2022-03-06 07:34:19 +08:00
2023-08-06 05:30:02 +08:00
// if the port of upstream address contains a placeholder, only wrap it with the `Upstream` struct,
// delaying actual resolution of the address until request time.
if pa . replaceablePort ( ) {
h . Upstreams = append ( h . Upstreams , & Upstream { Dial : pa . dialAddr ( ) } )
return nil
}
parsedAddr , err := caddy . ParseNetworkAddress ( pa . dialAddr ( ) )
2023-05-16 02:14:50 +08:00
if err != nil {
return d . WrapErr ( err )
}
2023-08-06 05:30:02 +08:00
if pa . isUnix ( ) || ! pa . rangedPort ( ) {
2023-05-16 02:14:50 +08:00
// unix networks don't have ports
h . Upstreams = append ( h . Upstreams , & Upstream {
2023-08-06 05:30:02 +08:00
Dial : pa . dialAddr ( ) ,
2023-05-16 02:14:50 +08:00
} )
} else {
// expand a port range into multiple upstreams
for i := parsedAddr . StartPort ; i <= parsedAddr . EndPort ; i ++ {
h . Upstreams = append ( h . Upstreams , & Upstream {
Dial : caddy . JoinNetworkAddress ( "" , parsedAddr . Host , fmt . Sprint ( i ) ) ,
} )
}
}
2020-03-25 00:53:53 +08:00
return nil
}
2023-04-21 02:43:51 +08:00
d . Next ( ) // consume the directive name
for _ , up := range d . RemainingArgs ( ) {
err := appendUpstream ( up )
if err != nil {
return err
}
}
for nesting := d . Nesting ( ) ; d . NextBlock ( nesting ) ; {
// if the subdirective has an "@" prefix then we
// parse it as a response matcher for use with "handle_response"
if strings . HasPrefix ( d . Val ( ) , matcherPrefix ) {
err := caddyhttp . ParseNamedResponseMatcher ( d . NewFromNextSegment ( ) , h . responseMatchers )
2020-02-28 11:56:24 +08:00
if err != nil {
return err
}
2023-04-21 02:43:51 +08:00
continue
2019-09-10 02:23:27 +08:00
}
2023-04-21 02:43:51 +08:00
switch d . Val ( ) {
case "to" :
args := d . RemainingArgs ( )
if len ( args ) == 0 {
return d . ArgErr ( )
}
for _ , up := range args {
err := appendUpstream ( up )
2021-05-03 02:39:06 +08:00
if err != nil {
return err
}
}
2023-04-21 02:43:51 +08:00
case "dynamic" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . DynamicUpstreams != nil {
return d . Err ( "dynamic upstreams already specified" )
}
dynModule := d . Val ( )
modID := "http.reverse_proxy.upstreams." + dynModule
unm , err := caddyfile . UnmarshalModule ( d , modID )
if err != nil {
return err
}
source , ok := unm . ( UpstreamSource )
if ! ok {
return d . Errf ( "module %s (%T) is not an UpstreamSource" , modID , unm )
}
h . DynamicUpstreamsRaw = caddyconfig . JSONModuleObject ( source , "source" , dynModule , nil )
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "lb_policy" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . LoadBalancing != nil && h . LoadBalancing . SelectionPolicyRaw != nil {
return d . Err ( "load balancing selection policy already specified" )
}
name := d . Val ( )
modID := "http.reverse_proxy.selection_policies." + name
unm , err := caddyfile . UnmarshalModule ( d , modID )
if err != nil {
return err
}
sel , ok := unm . ( Selector )
if ! ok {
return d . Errf ( "module %s (%T) is not a reverseproxy.Selector" , modID , unm )
}
if h . LoadBalancing == nil {
h . LoadBalancing = new ( LoadBalancing )
}
h . LoadBalancing . SelectionPolicyRaw = caddyconfig . JSONModuleObject ( sel , "policy" , name , nil )
2022-03-07 08:43:39 +08:00
2023-04-21 02:43:51 +08:00
case "lb_retries" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
tries , err := strconv . Atoi ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad lb_retries number '%s': %v" , d . Val ( ) , err )
}
if h . LoadBalancing == nil {
h . LoadBalancing = new ( LoadBalancing )
}
h . LoadBalancing . Retries = tries
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "lb_try_duration" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . LoadBalancing == nil {
h . LoadBalancing = new ( LoadBalancing )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad duration value %s: %v" , d . Val ( ) , err )
}
h . LoadBalancing . TryDuration = caddy . Duration ( dur )
2022-07-14 04:15:00 +08:00
2023-04-21 02:43:51 +08:00
case "lb_try_interval" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . LoadBalancing == nil {
h . LoadBalancing = new ( LoadBalancing )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad interval value '%s': %v" , d . Val ( ) , err )
}
h . LoadBalancing . TryInterval = caddy . Duration ( dur )
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "lb_retry_match" :
matcherSet , err := caddyhttp . ParseCaddyfileNestedMatcherSet ( d )
if err != nil {
return d . Errf ( "failed to parse lb_retry_match: %v" , err )
}
if h . LoadBalancing == nil {
h . LoadBalancing = new ( LoadBalancing )
}
h . LoadBalancing . RetryMatchRaw = append ( h . LoadBalancing . RetryMatchRaw , matcherSet )
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "health_uri" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Active == nil {
h . HealthChecks . Active = new ( ActiveHealthChecks )
}
h . HealthChecks . Active . URI = d . Val ( )
2022-07-14 04:15:00 +08:00
2023-04-21 02:43:51 +08:00
case "health_path" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Active == nil {
h . HealthChecks . Active = new ( ActiveHealthChecks )
}
h . HealthChecks . Active . Path = d . Val ( )
caddy . Log ( ) . Named ( "config.adapter.caddyfile" ) . Warn ( "the 'health_path' subdirective is deprecated, please use 'health_uri' instead!" )
2021-03-30 08:36:40 +08:00
2023-04-21 02:43:51 +08:00
case "health_port" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Active == nil {
h . HealthChecks . Active = new ( ActiveHealthChecks )
}
portNum , err := strconv . Atoi ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad port number '%s': %v" , d . Val ( ) , err )
}
h . HealthChecks . Active . Port = portNum
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "health_headers" :
healthHeaders := make ( http . Header )
for nesting := d . Nesting ( ) ; d . NextBlock ( nesting ) ; {
key := d . Val ( )
values := d . RemainingArgs ( )
if len ( values ) == 0 {
values = append ( values , "" )
2021-01-05 02:26:18 +08:00
}
2023-04-21 02:43:51 +08:00
healthHeaders [ key ] = values
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Active == nil {
h . HealthChecks . Active = new ( ActiveHealthChecks )
}
h . HealthChecks . Active . Headers = healthHeaders
2021-01-05 02:26:18 +08:00
2023-04-21 02:43:51 +08:00
case "health_interval" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Active == nil {
h . HealthChecks . Active = new ( ActiveHealthChecks )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad interval value %s: %v" , d . Val ( ) , err )
}
h . HealthChecks . Active . Interval = caddy . Duration ( dur )
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "health_timeout" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Active == nil {
h . HealthChecks . Active = new ( ActiveHealthChecks )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad timeout value %s: %v" , d . Val ( ) , err )
}
h . HealthChecks . Active . Timeout = caddy . Duration ( dur )
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "health_status" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Active == nil {
h . HealthChecks . Active = new ( ActiveHealthChecks )
}
val := d . Val ( )
if len ( val ) == 3 && strings . HasSuffix ( val , "xx" ) {
val = val [ : 1 ]
}
statusNum , err := strconv . Atoi ( val )
if err != nil {
return d . Errf ( "bad status value '%s': %v" , d . Val ( ) , err )
}
h . HealthChecks . Active . ExpectStatus = statusNum
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "health_body" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Active == nil {
h . HealthChecks . Active = new ( ActiveHealthChecks )
}
h . HealthChecks . Active . ExpectBody = d . Val ( )
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "max_fails" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Passive == nil {
h . HealthChecks . Passive = new ( PassiveHealthChecks )
}
maxFails , err := strconv . Atoi ( d . Val ( ) )
if err != nil {
return d . Errf ( "invalid maximum fail count '%s': %v" , d . Val ( ) , err )
}
h . HealthChecks . Passive . MaxFails = maxFails
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "fail_duration" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Passive == nil {
h . HealthChecks . Passive = new ( PassiveHealthChecks )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad duration value '%s': %v" , d . Val ( ) , err )
}
h . HealthChecks . Passive . FailDuration = caddy . Duration ( dur )
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "unhealthy_request_count" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Passive == nil {
h . HealthChecks . Passive = new ( PassiveHealthChecks )
}
maxConns , err := strconv . Atoi ( d . Val ( ) )
if err != nil {
return d . Errf ( "invalid maximum connection count '%s': %v" , d . Val ( ) , err )
}
h . HealthChecks . Passive . UnhealthyRequestCount = maxConns
case "unhealthy_status" :
args := d . RemainingArgs ( )
if len ( args ) == 0 {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Passive == nil {
h . HealthChecks . Passive = new ( PassiveHealthChecks )
}
for _ , arg := range args {
if len ( arg ) == 3 && strings . HasSuffix ( arg , "xx" ) {
arg = arg [ : 1 ]
2019-09-10 02:23:27 +08:00
}
2023-04-21 02:43:51 +08:00
statusNum , err := strconv . Atoi ( arg )
2019-09-10 02:23:27 +08:00
if err != nil {
2023-04-21 02:43:51 +08:00
return d . Errf ( "bad status value '%s': %v" , d . Val ( ) , err )
2019-09-10 02:23:27 +08:00
}
2023-04-21 02:43:51 +08:00
h . HealthChecks . Passive . UnhealthyStatus = append ( h . HealthChecks . Passive . UnhealthyStatus , statusNum )
}
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "unhealthy_latency" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . HealthChecks == nil {
h . HealthChecks = new ( HealthChecks )
}
if h . HealthChecks . Passive == nil {
h . HealthChecks . Passive = new ( PassiveHealthChecks )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad duration value '%s': %v" , d . Val ( ) , err )
}
h . HealthChecks . Passive . UnhealthyLatency = caddy . Duration ( dur )
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "flush_interval" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if fi , err := strconv . Atoi ( d . Val ( ) ) ; err == nil {
h . FlushInterval = caddy . Duration ( fi )
} else {
2020-05-12 06:41:11 +08:00
dur , err := caddy . ParseDuration ( d . Val ( ) )
2019-09-10 02:23:27 +08:00
if err != nil {
return d . Errf ( "bad duration value '%s': %v" , d . Val ( ) , err )
}
2023-04-21 02:43:51 +08:00
h . FlushInterval = caddy . Duration ( dur )
}
2019-09-10 02:23:27 +08:00
2023-04-21 02:43:51 +08:00
case "request_buffers" , "response_buffers" :
subdir := d . Val ( )
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
size , err := humanize . ParseBytes ( d . Val ( ) )
if err != nil {
return d . Errf ( "invalid byte size '%s': %v" , d . Val ( ) , err )
}
if d . NextArg ( ) {
return d . ArgErr ( )
}
if subdir == "request_buffers" {
h . RequestBuffers = int64 ( size )
} else if subdir == "response_buffers" {
h . ResponseBuffers = int64 ( size )
}
2020-09-09 00:37:46 +08:00
2023-04-21 02:43:51 +08:00
// TODO: These three properties are deprecated; remove them sometime after v2.6.4
case "buffer_requests" : // TODO: deprecated
if d . NextArg ( ) {
return d . ArgErr ( )
}
caddy . Log ( ) . Named ( "config.adapter.caddyfile" ) . Warn ( "DEPRECATED: buffer_requests: use request_buffers instead (with a maximum buffer size)" )
h . DeprecatedBufferRequests = true
case "buffer_responses" : // TODO: deprecated
if d . NextArg ( ) {
return d . ArgErr ( )
}
caddy . Log ( ) . Named ( "config.adapter.caddyfile" ) . Warn ( "DEPRECATED: buffer_responses: use response_buffers instead (with a maximum buffer size)" )
h . DeprecatedBufferResponses = true
case "max_buffer_size" : // TODO: deprecated
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
size , err := humanize . ParseBytes ( d . Val ( ) )
if err != nil {
return d . Errf ( "invalid byte size '%s': %v" , d . Val ( ) , err )
}
if d . NextArg ( ) {
return d . ArgErr ( )
}
caddy . Log ( ) . Named ( "config.adapter.caddyfile" ) . Warn ( "DEPRECATED: max_buffer_size: use request_buffers and/or response_buffers instead (with maximum buffer sizes)" )
h . DeprecatedMaxBufferSize = int64 ( size )
2023-02-12 08:25:29 +08:00
2023-06-20 05:54:43 +08:00
case "stream_timeout" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if fi , err := strconv . Atoi ( d . Val ( ) ) ; err == nil {
h . StreamTimeout = caddy . Duration ( fi )
} else {
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad duration value '%s': %v" , d . Val ( ) , err )
}
h . StreamTimeout = caddy . Duration ( dur )
}
case "stream_close_delay" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if fi , err := strconv . Atoi ( d . Val ( ) ) ; err == nil {
h . StreamCloseDelay = caddy . Duration ( fi )
} else {
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad duration value '%s': %v" , d . Val ( ) , err )
}
h . StreamCloseDelay = caddy . Duration ( dur )
}
2023-04-21 02:43:51 +08:00
case "trusted_proxies" :
for d . NextArg ( ) {
if d . Val ( ) == "private_ranges" {
h . TrustedProxies = append ( h . TrustedProxies , caddyhttp . PrivateRangesCIDR ( ) ... )
continue
2021-02-10 05:15:04 +08:00
}
2023-04-21 02:43:51 +08:00
h . TrustedProxies = append ( h . TrustedProxies , d . Val ( ) )
}
case "header_up" :
var err error
if h . Headers == nil {
h . Headers = new ( headers . Handler )
}
if h . Headers . Request == nil {
h . Headers . Request = new ( headers . HeaderOps )
}
args := d . RemainingArgs ( )
switch len ( args ) {
case 1 :
err = headers . CaddyfileHeaderOp ( h . Headers . Request , args [ 0 ] , "" , "" )
case 2 :
// some lint checks, I guess
if strings . EqualFold ( args [ 0 ] , "host" ) && ( args [ 1 ] == "{hostport}" || args [ 1 ] == "{http.request.hostport}" ) {
caddy . Log ( ) . Named ( "caddyfile" ) . Warn ( "Unnecessary header_up Host: the reverse proxy's default behavior is to pass headers to the upstream" )
2021-02-10 05:15:04 +08:00
}
2023-04-21 02:43:51 +08:00
if strings . EqualFold ( args [ 0 ] , "x-forwarded-for" ) && ( args [ 1 ] == "{remote}" || args [ 1 ] == "{http.request.remote}" || args [ 1 ] == "{remote_host}" || args [ 1 ] == "{http.request.remote.host}" ) {
caddy . Log ( ) . Named ( "caddyfile" ) . Warn ( "Unnecessary header_up X-Forwarded-For: the reverse proxy's default behavior is to pass headers to the upstream" )
2021-02-10 05:15:04 +08:00
}
2023-04-21 02:43:51 +08:00
if strings . EqualFold ( args [ 0 ] , "x-forwarded-proto" ) && ( args [ 1 ] == "{scheme}" || args [ 1 ] == "{http.request.scheme}" ) {
caddy . Log ( ) . Named ( "caddyfile" ) . Warn ( "Unnecessary header_up X-Forwarded-Proto: the reverse proxy's default behavior is to pass headers to the upstream" )
2021-02-10 05:15:04 +08:00
}
2023-04-21 02:43:51 +08:00
if strings . EqualFold ( args [ 0 ] , "x-forwarded-host" ) && ( args [ 1 ] == "{host}" || args [ 1 ] == "{http.request.host}" || args [ 1 ] == "{hostport}" || args [ 1 ] == "{http.request.hostport}" ) {
caddy . Log ( ) . Named ( "caddyfile" ) . Warn ( "Unnecessary header_up X-Forwarded-Host: the reverse proxy's default behavior is to pass headers to the upstream" )
2022-03-07 07:51:55 +08:00
}
2023-04-21 02:43:51 +08:00
err = headers . CaddyfileHeaderOp ( h . Headers . Request , args [ 0 ] , args [ 1 ] , "" )
case 3 :
err = headers . CaddyfileHeaderOp ( h . Headers . Request , args [ 0 ] , args [ 1 ] , args [ 2 ] )
default :
return d . ArgErr ( )
}
2022-03-07 07:51:55 +08:00
2023-04-21 02:43:51 +08:00
if err != nil {
return d . Err ( err . Error ( ) )
}
2020-11-21 03:38:16 +08:00
2023-04-21 02:43:51 +08:00
case "header_down" :
var err error
2019-09-21 03:13:49 +08:00
2023-04-21 02:43:51 +08:00
if h . Headers == nil {
h . Headers = new ( headers . Handler )
}
if h . Headers . Response == nil {
h . Headers . Response = & headers . RespHeaderOps {
HeaderOps : new ( headers . HeaderOps ) ,
2020-11-21 03:38:16 +08:00
}
2023-04-21 02:43:51 +08:00
}
args := d . RemainingArgs ( )
switch len ( args ) {
case 1 :
err = headers . CaddyfileHeaderOp ( h . Headers . Response . HeaderOps , args [ 0 ] , "" , "" )
case 2 :
err = headers . CaddyfileHeaderOp ( h . Headers . Response . HeaderOps , args [ 0 ] , args [ 1 ] , "" )
case 3 :
err = headers . CaddyfileHeaderOp ( h . Headers . Response . HeaderOps , args [ 0 ] , args [ 1 ] , args [ 2 ] )
default :
return d . ArgErr ( )
}
2020-11-21 03:38:16 +08:00
2023-04-21 02:43:51 +08:00
if err != nil {
return d . Err ( err . Error ( ) )
}
2020-11-21 03:38:16 +08:00
2023-04-21 02:43:51 +08:00
case "method" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . Rewrite == nil {
h . Rewrite = & rewrite . Rewrite { }
}
h . Rewrite . Method = d . Val ( )
if d . NextArg ( ) {
return d . ArgErr ( )
}
2019-09-21 03:13:49 +08:00
2023-04-21 02:43:51 +08:00
case "rewrite" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . Rewrite == nil {
h . Rewrite = & rewrite . Rewrite { }
}
h . Rewrite . URI = d . Val ( )
if d . NextArg ( ) {
return d . ArgErr ( )
}
2020-11-21 03:38:16 +08:00
2023-04-21 02:43:51 +08:00
case "transport" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . TransportRaw != nil {
return d . Err ( "transport already specified" )
}
transportModuleName = d . Val ( )
modID := "http.reverse_proxy.transport." + transportModuleName
unm , err := caddyfile . UnmarshalModule ( d , modID )
if err != nil {
return err
}
rt , ok := unm . ( http . RoundTripper )
if ! ok {
return d . Errf ( "module %s (%T) is not a RoundTripper" , modID , unm )
}
transport = rt
case "handle_response" :
// delegate the parsing of handle_response to the caller,
// since we need the httpcaddyfile.Helper to parse subroutes.
// See h.FinalizeUnmarshalCaddyfile
h . handleResponseSegments = append ( h . handleResponseSegments , d . NewFromNextSegment ( ) )
case "replace_status" :
args := d . RemainingArgs ( )
if len ( args ) != 1 && len ( args ) != 2 {
return d . Errf ( "must have one or two arguments: an optional response matcher, and a status code" )
}
2022-05-06 22:50:26 +08:00
2023-04-21 02:43:51 +08:00
responseHandler := caddyhttp . ResponseHandler { }
2022-05-06 22:50:26 +08:00
2023-04-21 02:43:51 +08:00
if len ( args ) == 2 {
if ! strings . HasPrefix ( args [ 0 ] , matcherPrefix ) {
return d . Errf ( "must use a named response matcher, starting with '@'" )
2019-09-10 02:23:27 +08:00
}
2023-04-21 02:43:51 +08:00
foundMatcher , ok := h . responseMatchers [ args [ 0 ] ]
2019-09-10 02:23:27 +08:00
if ! ok {
2023-04-21 02:43:51 +08:00
return d . Errf ( "no named response matcher defined with name '%s'" , args [ 0 ] [ 1 : ] )
2022-03-02 05:12:43 +08:00
}
2023-04-21 02:43:51 +08:00
responseHandler . Match = & foundMatcher
responseHandler . StatusCode = caddyhttp . WeakString ( args [ 1 ] )
} else if len ( args ) == 1 {
responseHandler . StatusCode = caddyhttp . WeakString ( args [ 0 ] )
}
2022-03-02 05:12:43 +08:00
2023-04-21 02:43:51 +08:00
// make sure there's no block, cause it doesn't make sense
if d . NextBlock ( 1 ) {
return d . Errf ( "cannot define routes for 'replace_status', use 'handle_response' instead." )
}
2022-03-02 05:12:43 +08:00
2023-04-21 02:43:51 +08:00
h . HandleResponse = append (
h . HandleResponse ,
responseHandler ,
)
2022-03-02 05:12:43 +08:00
2023-04-21 02:43:51 +08:00
default :
return d . Errf ( "unrecognized subdirective %s" , d . Val ( ) )
2019-09-10 02:23:27 +08:00
}
}
2020-02-28 11:56:24 +08:00
// if the scheme inferred from the backends' addresses is
2020-11-24 03:18:26 +08:00
// HTTPS, we will need a non-nil transport to enable TLS,
// or if H2C, to set the transport versions.
if ( commonScheme == "https" || commonScheme == "h2c" ) && transport == nil {
2020-02-28 11:56:24 +08:00
transport = new ( HTTPTransport )
transportModuleName = "http"
}
// verify transport configuration, and finally encode it
if transport != nil {
2020-04-07 22:31:52 +08:00
if te , ok := transport . ( TLSTransport ) ; ok {
if commonScheme == "https" && ! te . TLSEnabled ( ) {
err := te . EnableTLS ( new ( TLSConfig ) )
if err != nil {
return err
}
2020-02-28 11:56:24 +08:00
}
2020-04-07 22:31:52 +08:00
if commonScheme == "http" && te . TLSEnabled ( ) {
2020-02-28 11:56:24 +08:00
return d . Errf ( "upstream address scheme is HTTP but transport is configured for HTTP+TLS (HTTPS)" )
}
2020-11-24 03:18:26 +08:00
if te , ok := transport . ( * HTTPTransport ) ; ok && commonScheme == "h2c" {
te . Versions = [ ] string { "h2c" , "2" }
}
2020-04-07 22:31:52 +08:00
} else if commonScheme == "https" {
return d . Errf ( "upstreams are configured for HTTPS but transport module does not support TLS: %T" , transport )
2020-02-28 11:56:24 +08:00
}
2020-07-18 03:18:32 +08:00
// no need to encode empty default transport
if ! reflect . DeepEqual ( transport , new ( HTTPTransport ) ) {
2020-02-28 11:56:24 +08:00
h . TransportRaw = caddyconfig . JSONModuleObject ( transport , "protocol" , transportModuleName , nil )
}
}
2019-09-10 02:23:27 +08:00
return nil
}
2021-05-03 02:39:06 +08:00
// FinalizeUnmarshalCaddyfile finalizes the Caddyfile parsing which
// requires having an httpcaddyfile.Helper to function, to parse subroutes.
func ( h * Handler ) FinalizeUnmarshalCaddyfile ( helper httpcaddyfile . Helper ) error {
for _ , d := range h . handleResponseSegments {
// consume the "handle_response" token
d . Next ( )
args := d . RemainingArgs ( )
2022-03-02 05:12:43 +08:00
// TODO: Remove this check at some point in the future
if len ( args ) == 2 {
return d . Errf ( "configuring 'handle_response' for status code replacement is no longer supported. Use 'replace_status' instead." )
}
if len ( args ) > 1 {
2021-05-03 02:39:06 +08:00
return d . Errf ( "too many arguments for 'handle_response': %s" , args )
}
2022-03-02 05:12:43 +08:00
var matcher * caddyhttp . ResponseMatcher
if len ( args ) == 1 {
// the first arg should always be a matcher.
2021-05-03 02:39:06 +08:00
if ! strings . HasPrefix ( args [ 0 ] , matcherPrefix ) {
return d . Errf ( "must use a named response matcher, starting with '@'" )
}
foundMatcher , ok := h . responseMatchers [ args [ 0 ] ]
if ! ok {
return d . Errf ( "no named response matcher defined with name '%s'" , args [ 0 ] [ 1 : ] )
}
matcher = & foundMatcher
}
// parse the block as routes
handler , err := httpcaddyfile . ParseSegmentAsSubroute ( helper . WithDispenser ( d . NewFromNextSegment ( ) ) )
if err != nil {
return err
}
subroute , ok := handler . ( * caddyhttp . Subroute )
if ! ok {
return helper . Errf ( "segment was not parsed as a subroute" )
}
h . HandleResponse = append (
h . HandleResponse ,
caddyhttp . ResponseHandler {
Match : matcher ,
Routes : subroute . Routes ,
} ,
)
}
// move the handle_response entries without a matcher to the end.
// we can't use sort.SliceStable because it will reorder the rest of the
// entries which may be undesirable because we don't have a good
// heuristic to use for sorting.
withoutMatchers := [ ] caddyhttp . ResponseHandler { }
withMatchers := [ ] caddyhttp . ResponseHandler { }
for _ , hr := range h . HandleResponse {
if hr . Match == nil {
withoutMatchers = append ( withoutMatchers , hr )
} else {
withMatchers = append ( withMatchers , hr )
}
}
h . HandleResponse = append ( withMatchers , withoutMatchers ... )
// clean up the bits we only needed for adapting
h . handleResponseSegments = nil
h . responseMatchers = nil
return nil
}
2019-09-10 02:23:27 +08:00
// UnmarshalCaddyfile deserializes Caddyfile tokens into h.
//
2022-08-26 11:42:48 +08:00
// transport http {
// read_buffer <size>
// write_buffer <size>
// max_response_header <size>
// dial_timeout <duration>
// dial_fallback_delay <duration>
// response_header_timeout <duration>
// expect_continue_timeout <duration>
// resolvers <resolvers...>
// tls
// tls_client_auth <automate_name> | <cert_file> <key_file>
// tls_insecure_skip_verify
// tls_timeout <duration>
// tls_trusted_ca_certs <cert_files...>
// tls_server_name <sni>
// tls_renegotiation <level>
// tls_except_ports <ports...>
// keepalive [off|<duration>]
// keepalive_interval <interval>
// keepalive_idle_conns <max_count>
// keepalive_idle_conns_per_host <count>
// versions <versions...>
// compression off
// max_conns_per_host <count>
// max_idle_conns_per_host <count>
// }
2019-09-10 02:23:27 +08:00
func ( h * HTTPTransport ) UnmarshalCaddyfile ( d * caddyfile . Dispenser ) error {
2019-09-12 08:53:44 +08:00
for d . Next ( ) {
for d . NextBlock ( 0 ) {
switch d . Val ( ) {
case "read_buffer" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
size , err := humanize . ParseBytes ( d . Val ( ) )
if err != nil {
return d . Errf ( "invalid read buffer size '%s': %v" , d . Val ( ) , err )
}
h . ReadBufferSize = int ( size )
2019-09-10 02:23:27 +08:00
2019-09-12 08:53:44 +08:00
case "write_buffer" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
size , err := humanize . ParseBytes ( d . Val ( ) )
if err != nil {
return d . Errf ( "invalid write buffer size '%s': %v" , d . Val ( ) , err )
}
h . WriteBufferSize = int ( size )
2019-09-10 02:23:27 +08:00
2022-07-24 12:38:41 +08:00
case "read_timeout" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
timeout , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "invalid read timeout duration '%s': %v" , d . Val ( ) , err )
}
h . ReadTimeout = caddy . Duration ( timeout )
case "write_timeout" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
timeout , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "invalid write timeout duration '%s': %v" , d . Val ( ) , err )
}
h . WriteTimeout = caddy . Duration ( timeout )
2020-11-03 05:59:02 +08:00
case "max_response_header" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
size , err := humanize . ParseBytes ( d . Val ( ) )
if err != nil {
return d . Errf ( "invalid max response header size '%s': %v" , d . Val ( ) , err )
}
h . MaxResponseHeaderSize = int64 ( size )
2023-04-01 05:44:53 +08:00
case "proxy_protocol" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
switch proxyProtocol := d . Val ( ) ; proxyProtocol {
case "v1" , "v2" :
h . ProxyProtocol = proxyProtocol
default :
return d . Errf ( "invalid proxy protocol version '%s'" , proxyProtocol )
}
2019-09-12 08:53:44 +08:00
case "dial_timeout" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
2020-05-12 06:41:11 +08:00
dur , err := caddy . ParseDuration ( d . Val ( ) )
2019-09-12 08:53:44 +08:00
if err != nil {
return d . Errf ( "bad timeout value '%s': %v" , d . Val ( ) , err )
}
h . DialTimeout = caddy . Duration ( dur )
2019-09-10 02:23:27 +08:00
2020-11-03 05:59:02 +08:00
case "dial_fallback_delay" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad fallback delay value '%s': %v" , d . Val ( ) , err )
}
h . FallbackDelay = caddy . Duration ( dur )
case "response_header_timeout" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad timeout value '%s': %v" , d . Val ( ) , err )
}
h . ResponseHeaderTimeout = caddy . Duration ( dur )
case "expect_continue_timeout" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad timeout value '%s': %v" , d . Val ( ) , err )
}
h . ExpectContinueTimeout = caddy . Duration ( dur )
2022-03-07 08:43:39 +08:00
case "resolvers" :
if h . Resolver == nil {
h . Resolver = new ( UpstreamResolver )
}
h . Resolver . Addresses = d . RemainingArgs ( )
if len ( h . Resolver . Addresses ) == 0 {
return d . Errf ( "must specify at least one resolver address" )
}
2022-07-09 01:04:22 +08:00
case "tls" :
if h . TLS == nil {
h . TLS = new ( TLSConfig )
}
2019-09-12 08:53:44 +08:00
case "tls_client_auth" :
if h . TLS == nil {
h . TLS = new ( TLSConfig )
}
2020-06-09 00:30:26 +08:00
args := d . RemainingArgs ( )
switch len ( args ) {
case 1 :
h . TLS . ClientCertificateAutomate = args [ 0 ]
case 2 :
h . TLS . ClientCertificateFile = args [ 0 ]
h . TLS . ClientCertificateKeyFile = args [ 1 ]
default :
return d . ArgErr ( )
}
2019-09-10 02:23:27 +08:00
2019-09-12 08:53:44 +08:00
case "tls_insecure_skip_verify" :
if d . NextArg ( ) {
return d . ArgErr ( )
}
if h . TLS == nil {
h . TLS = new ( TLSConfig )
}
h . TLS . InsecureSkipVerify = true
2019-09-10 02:23:27 +08:00
2019-09-12 08:53:44 +08:00
case "tls_timeout" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
2020-05-12 06:41:11 +08:00
dur , err := caddy . ParseDuration ( d . Val ( ) )
2019-09-12 08:53:44 +08:00
if err != nil {
return d . Errf ( "bad timeout value '%s': %v" , d . Val ( ) , err )
}
if h . TLS == nil {
h . TLS = new ( TLSConfig )
}
h . TLS . HandshakeTimeout = caddy . Duration ( dur )
2019-09-10 02:23:27 +08:00
2020-01-08 03:07:42 +08:00
case "tls_trusted_ca_certs" :
args := d . RemainingArgs ( )
if len ( args ) == 0 {
return d . ArgErr ( )
}
if h . TLS == nil {
h . TLS = new ( TLSConfig )
}
2020-01-23 00:34:16 +08:00
h . TLS . RootCAPEMFiles = args
2020-01-08 03:07:42 +08:00
2020-05-06 02:39:39 +08:00
case "tls_server_name" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . TLS == nil {
h . TLS = new ( TLSConfig )
}
h . TLS . ServerName = d . Val ( )
2022-07-09 01:04:22 +08:00
case "tls_renegotiation" :
if h . TLS == nil {
h . TLS = new ( TLSConfig )
}
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
switch renegotiation := d . Val ( ) ; renegotiation {
case "never" , "once" , "freely" :
h . TLS . Renegotiation = renegotiation
default :
return d . ArgErr ( )
}
case "tls_except_ports" :
if h . TLS == nil {
h . TLS = new ( TLSConfig )
}
h . TLS . ExceptPorts = d . RemainingArgs ( )
if len ( h . TLS . ExceptPorts ) == 0 {
return d . ArgErr ( )
}
2019-09-12 08:53:44 +08:00
case "keepalive" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if h . KeepAlive == nil {
h . KeepAlive = new ( KeepAlive )
}
if d . Val ( ) == "off" {
var disable bool
h . KeepAlive . Enabled = & disable
2019-10-12 04:25:39 +08:00
break
2019-09-12 08:53:44 +08:00
}
2020-05-12 06:41:11 +08:00
dur , err := caddy . ParseDuration ( d . Val ( ) )
2019-09-12 08:53:44 +08:00
if err != nil {
return d . Errf ( "bad duration value '%s': %v" , d . Val ( ) , err )
}
h . KeepAlive . IdleConnTimeout = caddy . Duration ( dur )
2019-09-10 02:23:27 +08:00
2021-11-24 14:32:25 +08:00
case "keepalive_interval" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad interval value '%s': %v" , d . Val ( ) , err )
}
if h . KeepAlive == nil {
h . KeepAlive = new ( KeepAlive )
}
h . KeepAlive . ProbeInterval = caddy . Duration ( dur )
2019-09-12 08:53:44 +08:00
case "keepalive_idle_conns" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
num , err := strconv . Atoi ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad integer value '%s': %v" , d . Val ( ) , err )
}
if h . KeepAlive == nil {
h . KeepAlive = new ( KeepAlive )
}
h . KeepAlive . MaxIdleConns = num
2021-06-16 04:54:48 +08:00
case "keepalive_idle_conns_per_host" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
num , err := strconv . Atoi ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad integer value '%s': %v" , d . Val ( ) , err )
}
if h . KeepAlive == nil {
h . KeepAlive = new ( KeepAlive )
}
2019-10-12 04:25:39 +08:00
h . KeepAlive . MaxIdleConnsPerHost = num
2019-09-10 02:23:27 +08:00
2020-05-06 02:33:21 +08:00
case "versions" :
h . Versions = d . RemainingArgs ( )
if len ( h . Versions ) == 0 {
return d . ArgErr ( )
}
2020-08-01 01:30:20 +08:00
case "compression" :
if d . NextArg ( ) {
if d . Val ( ) == "off" {
var disable bool
h . Compression = & disable
}
}
2020-10-31 02:05:21 +08:00
case "max_conns_per_host" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
num , err := strconv . Atoi ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad integer value '%s': %v" , d . Val ( ) , err )
}
h . MaxConnsPerHost = num
2019-09-12 08:53:44 +08:00
default :
return d . Errf ( "unrecognized subdirective %s" , d . Val ( ) )
}
2019-09-10 02:23:27 +08:00
}
}
return nil
}
reverseproxy: copy_response and copy_response_headers for handle_response routes (#4391)
* reverseproxy: New `copy_response` handler for `handle_response` routes
Followup to #4298 and #4388.
This adds a new `copy_response` handler which may only be used in `reverse_proxy`'s `handle_response` routes, which can be used to actually copy the proxy response downstream.
Previously, if `handle_response` was used (with routes, not the status code mode), it was impossible to use the upstream's response body at all, because we would always close the body, expecting the routes to write a new body from scratch.
To implement this, I had to refactor `h.reverseProxy()` to move all the code that came after the `HandleResponse` loop into a new function. This new function `h.finalizeResponse()` takes care of preparing the response by removing extra headers, dealing with trailers, then copying the headers and body downstream.
Since basically what we want `copy_response` to do is invoke `h.finalizeResponse()` at a configurable point in time, we need to pass down the proxy handler, the response, and some other state via a new `req.WithContext(ctx)`. Wrapping a new context is pretty much the only way we have to jump a few layers in the HTTP middleware chain and let a handler pick up this information. Feels a bit dirty, but it works.
Also fixed a bug with the `http.reverse_proxy.upstream.duration` placeholder, it always had the same duration as `http.reverse_proxy.upstream.latency`, but the former was meant to be the time taken for the roundtrip _plus_ copying/writing the response.
* Delete the "Content-Length" header if we aren't copying
Fixes a bug where the Content-Length will mismatch the actual bytes written if we skipped copying the response, so we get a message like this when using curl:
```
curl: (18) transfer closed with 18 bytes remaining to read
```
To replicate:
```
{
admin off
debug
}
:8881 {
reverse_proxy 127.0.0.1:8882 {
@200 status 200
handle_response @200 {
header Foo bar
}
}
}
:8882 {
header Content-Type application/json
respond `{"hello": "world"}` 200
}
```
* Implement `copy_response_headers`, with include/exclude list support
* Apply suggestions from code review
Co-authored-by: Matt Holt <mholt@users.noreply.github.com>
2022-03-10 02:00:51 +08:00
func parseCopyResponseCaddyfile ( h httpcaddyfile . Helper ) ( caddyhttp . MiddlewareHandler , error ) {
crh := new ( CopyResponseHandler )
err := crh . UnmarshalCaddyfile ( h . Dispenser )
if err != nil {
return nil , err
}
return crh , nil
}
// UnmarshalCaddyfile sets up the handler from Caddyfile tokens. Syntax:
//
2022-08-26 11:42:48 +08:00
// copy_response [<matcher>] [<status>] {
// status <status>
// }
reverseproxy: copy_response and copy_response_headers for handle_response routes (#4391)
* reverseproxy: New `copy_response` handler for `handle_response` routes
Followup to #4298 and #4388.
This adds a new `copy_response` handler which may only be used in `reverse_proxy`'s `handle_response` routes, which can be used to actually copy the proxy response downstream.
Previously, if `handle_response` was used (with routes, not the status code mode), it was impossible to use the upstream's response body at all, because we would always close the body, expecting the routes to write a new body from scratch.
To implement this, I had to refactor `h.reverseProxy()` to move all the code that came after the `HandleResponse` loop into a new function. This new function `h.finalizeResponse()` takes care of preparing the response by removing extra headers, dealing with trailers, then copying the headers and body downstream.
Since basically what we want `copy_response` to do is invoke `h.finalizeResponse()` at a configurable point in time, we need to pass down the proxy handler, the response, and some other state via a new `req.WithContext(ctx)`. Wrapping a new context is pretty much the only way we have to jump a few layers in the HTTP middleware chain and let a handler pick up this information. Feels a bit dirty, but it works.
Also fixed a bug with the `http.reverse_proxy.upstream.duration` placeholder, it always had the same duration as `http.reverse_proxy.upstream.latency`, but the former was meant to be the time taken for the roundtrip _plus_ copying/writing the response.
* Delete the "Content-Length" header if we aren't copying
Fixes a bug where the Content-Length will mismatch the actual bytes written if we skipped copying the response, so we get a message like this when using curl:
```
curl: (18) transfer closed with 18 bytes remaining to read
```
To replicate:
```
{
admin off
debug
}
:8881 {
reverse_proxy 127.0.0.1:8882 {
@200 status 200
handle_response @200 {
header Foo bar
}
}
}
:8882 {
header Content-Type application/json
respond `{"hello": "world"}` 200
}
```
* Implement `copy_response_headers`, with include/exclude list support
* Apply suggestions from code review
Co-authored-by: Matt Holt <mholt@users.noreply.github.com>
2022-03-10 02:00:51 +08:00
func ( h * CopyResponseHandler ) UnmarshalCaddyfile ( d * caddyfile . Dispenser ) error {
for d . Next ( ) {
args := d . RemainingArgs ( )
if len ( args ) == 1 {
if num , err := strconv . Atoi ( args [ 0 ] ) ; err == nil && num > 0 {
h . StatusCode = caddyhttp . WeakString ( args [ 0 ] )
break
}
}
for d . NextBlock ( 0 ) {
switch d . Val ( ) {
case "status" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
h . StatusCode = caddyhttp . WeakString ( d . Val ( ) )
default :
return d . Errf ( "unrecognized subdirective '%s'" , d . Val ( ) )
}
}
}
return nil
}
func parseCopyResponseHeadersCaddyfile ( h httpcaddyfile . Helper ) ( caddyhttp . MiddlewareHandler , error ) {
crh := new ( CopyResponseHeadersHandler )
err := crh . UnmarshalCaddyfile ( h . Dispenser )
if err != nil {
return nil , err
}
return crh , nil
}
// UnmarshalCaddyfile sets up the handler from Caddyfile tokens. Syntax:
//
2022-08-26 11:42:48 +08:00
// copy_response_headers [<matcher>] {
// include <fields...>
// exclude <fields...>
// }
reverseproxy: copy_response and copy_response_headers for handle_response routes (#4391)
* reverseproxy: New `copy_response` handler for `handle_response` routes
Followup to #4298 and #4388.
This adds a new `copy_response` handler which may only be used in `reverse_proxy`'s `handle_response` routes, which can be used to actually copy the proxy response downstream.
Previously, if `handle_response` was used (with routes, not the status code mode), it was impossible to use the upstream's response body at all, because we would always close the body, expecting the routes to write a new body from scratch.
To implement this, I had to refactor `h.reverseProxy()` to move all the code that came after the `HandleResponse` loop into a new function. This new function `h.finalizeResponse()` takes care of preparing the response by removing extra headers, dealing with trailers, then copying the headers and body downstream.
Since basically what we want `copy_response` to do is invoke `h.finalizeResponse()` at a configurable point in time, we need to pass down the proxy handler, the response, and some other state via a new `req.WithContext(ctx)`. Wrapping a new context is pretty much the only way we have to jump a few layers in the HTTP middleware chain and let a handler pick up this information. Feels a bit dirty, but it works.
Also fixed a bug with the `http.reverse_proxy.upstream.duration` placeholder, it always had the same duration as `http.reverse_proxy.upstream.latency`, but the former was meant to be the time taken for the roundtrip _plus_ copying/writing the response.
* Delete the "Content-Length" header if we aren't copying
Fixes a bug where the Content-Length will mismatch the actual bytes written if we skipped copying the response, so we get a message like this when using curl:
```
curl: (18) transfer closed with 18 bytes remaining to read
```
To replicate:
```
{
admin off
debug
}
:8881 {
reverse_proxy 127.0.0.1:8882 {
@200 status 200
handle_response @200 {
header Foo bar
}
}
}
:8882 {
header Content-Type application/json
respond `{"hello": "world"}` 200
}
```
* Implement `copy_response_headers`, with include/exclude list support
* Apply suggestions from code review
Co-authored-by: Matt Holt <mholt@users.noreply.github.com>
2022-03-10 02:00:51 +08:00
func ( h * CopyResponseHeadersHandler ) UnmarshalCaddyfile ( d * caddyfile . Dispenser ) error {
for d . Next ( ) {
args := d . RemainingArgs ( )
if len ( args ) > 0 {
return d . ArgErr ( )
}
for d . NextBlock ( 0 ) {
switch d . Val ( ) {
case "include" :
h . Include = append ( h . Include , d . RemainingArgs ( ) ... )
case "exclude" :
h . Exclude = append ( h . Exclude , d . RemainingArgs ( ) ... )
default :
return d . Errf ( "unrecognized subdirective '%s'" , d . Val ( ) )
}
}
}
return nil
}
2022-03-07 08:43:39 +08:00
// UnmarshalCaddyfile deserializes Caddyfile tokens into h.
//
2022-08-26 11:42:48 +08:00
// dynamic srv [<name>] {
// service <service>
// proto <proto>
// name <name>
// refresh <interval>
// resolvers <resolvers...>
// dial_timeout <timeout>
// dial_fallback_delay <timeout>
// }
2022-03-07 08:43:39 +08:00
func ( u * SRVUpstreams ) UnmarshalCaddyfile ( d * caddyfile . Dispenser ) error {
for d . Next ( ) {
args := d . RemainingArgs ( )
if len ( args ) > 1 {
return d . ArgErr ( )
}
if len ( args ) > 0 {
u . Name = args [ 0 ]
}
for d . NextBlock ( 0 ) {
switch d . Val ( ) {
case "service" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if u . Service != "" {
return d . Errf ( "srv service has already been specified" )
}
u . Service = d . Val ( )
case "proto" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if u . Proto != "" {
return d . Errf ( "srv proto has already been specified" )
}
u . Proto = d . Val ( )
case "name" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if u . Name != "" {
return d . Errf ( "srv name has already been specified" )
}
u . Name = d . Val ( )
case "refresh" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "parsing refresh interval duration: %v" , err )
}
u . Refresh = caddy . Duration ( dur )
case "resolvers" :
if u . Resolver == nil {
u . Resolver = new ( UpstreamResolver )
}
u . Resolver . Addresses = d . RemainingArgs ( )
if len ( u . Resolver . Addresses ) == 0 {
return d . Errf ( "must specify at least one resolver address" )
}
case "dial_timeout" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad timeout value '%s': %v" , d . Val ( ) , err )
}
u . DialTimeout = caddy . Duration ( dur )
case "dial_fallback_delay" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad delay value '%s': %v" , d . Val ( ) , err )
}
u . FallbackDelay = caddy . Duration ( dur )
default :
return d . Errf ( "unrecognized srv option '%s'" , d . Val ( ) )
}
}
}
return nil
}
// UnmarshalCaddyfile deserializes Caddyfile tokens into h.
//
2022-08-26 11:42:48 +08:00
// dynamic a [<name> <port] {
// name <name>
// port <port>
// refresh <interval>
// resolvers <resolvers...>
// dial_timeout <timeout>
// dial_fallback_delay <timeout>
2023-02-28 01:23:09 +08:00
// versions ipv4|ipv6
2022-08-26 11:42:48 +08:00
// }
2022-03-07 08:43:39 +08:00
func ( u * AUpstreams ) UnmarshalCaddyfile ( d * caddyfile . Dispenser ) error {
for d . Next ( ) {
args := d . RemainingArgs ( )
if len ( args ) > 2 {
return d . ArgErr ( )
}
if len ( args ) > 0 {
u . Name = args [ 0 ]
2022-08-26 11:42:48 +08:00
if len ( args ) == 2 {
u . Port = args [ 1 ]
}
2022-03-07 08:43:39 +08:00
}
for d . NextBlock ( 0 ) {
switch d . Val ( ) {
case "name" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if u . Name != "" {
return d . Errf ( "a name has already been specified" )
}
u . Name = d . Val ( )
case "port" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
if u . Port != "" {
return d . Errf ( "a port has already been specified" )
}
u . Port = d . Val ( )
case "refresh" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "parsing refresh interval duration: %v" , err )
}
u . Refresh = caddy . Duration ( dur )
case "resolvers" :
if u . Resolver == nil {
u . Resolver = new ( UpstreamResolver )
}
u . Resolver . Addresses = d . RemainingArgs ( )
if len ( u . Resolver . Addresses ) == 0 {
return d . Errf ( "must specify at least one resolver address" )
}
case "dial_timeout" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad timeout value '%s': %v" , d . Val ( ) , err )
}
u . DialTimeout = caddy . Duration ( dur )
case "dial_fallback_delay" :
if ! d . NextArg ( ) {
return d . ArgErr ( )
}
dur , err := caddy . ParseDuration ( d . Val ( ) )
if err != nil {
return d . Errf ( "bad delay value '%s': %v" , d . Val ( ) , err )
}
u . FallbackDelay = caddy . Duration ( dur )
2023-02-28 01:23:09 +08:00
case "versions" :
args := d . RemainingArgs ( )
if len ( args ) == 0 {
return d . Errf ( "must specify at least one version" )
}
if u . Versions == nil {
2023-07-26 02:50:21 +08:00
u . Versions = & IPVersions { }
2023-02-28 01:23:09 +08:00
}
trueBool := true
for _ , arg := range args {
switch arg {
case "ipv4" :
u . Versions . IPv4 = & trueBool
case "ipv6" :
u . Versions . IPv6 = & trueBool
default :
return d . Errf ( "unsupported version: '%s'" , arg )
}
}
2022-03-07 08:43:39 +08:00
default :
2023-02-28 01:23:09 +08:00
return d . Errf ( "unrecognized a option '%s'" , d . Val ( ) )
2022-03-07 08:43:39 +08:00
}
}
}
return nil
}
2022-08-26 11:42:48 +08:00
// UnmarshalCaddyfile deserializes Caddyfile tokens into h.
//
// dynamic multi {
// <source> [...]
// }
func ( u * MultiUpstreams ) UnmarshalCaddyfile ( d * caddyfile . Dispenser ) error {
for d . Next ( ) {
if d . NextArg ( ) {
return d . ArgErr ( )
}
for nesting := d . Nesting ( ) ; d . NextBlock ( nesting ) ; {
dynModule := d . Val ( )
modID := "http.reverse_proxy.upstreams." + dynModule
unm , err := caddyfile . UnmarshalModule ( d , modID )
if err != nil {
return err
}
source , ok := unm . ( UpstreamSource )
if ! ok {
return d . Errf ( "module %s (%T) is not an UpstreamSource" , modID , unm )
}
u . SourcesRaw = append ( u . SourcesRaw , caddyconfig . JSONModuleObject ( source , "source" , dynModule , nil ) )
}
}
return nil
}
2021-05-03 02:39:06 +08:00
const matcherPrefix = "@"
2019-09-10 02:23:27 +08:00
// Interface guards
var (
_ caddyfile . Unmarshaler = ( * Handler ) ( nil )
_ caddyfile . Unmarshaler = ( * HTTPTransport ) ( nil )
2022-03-07 08:43:39 +08:00
_ caddyfile . Unmarshaler = ( * SRVUpstreams ) ( nil )
_ caddyfile . Unmarshaler = ( * AUpstreams ) ( nil )
2022-08-26 11:42:48 +08:00
_ caddyfile . Unmarshaler = ( * MultiUpstreams ) ( nil )
2019-09-10 02:23:27 +08:00
)