From 11c77a39b7fb24dcc62011a7e5d7d45fb973af79 Mon Sep 17 00:00:00 2001
From: Willy Tarreau <w@1wt.eu>
Date: Sun, 10 Jan 2010 00:42:19 +0100
Subject: [PATCH] [MINOR] http redirect: add the ability to append a '/' to the
 URL

Sometimes it can be desired to return a location which is the same
as the request with a slash appended when there was not one in the
request. A typical use of this is for sending a 301 so that people
don't reference links without the trailing slash. The name of the
new option is "append-slash" and it can be used on "redirect"
statements in prefix mode.

(cherry picked from commit 81e3b4f48d168da55c7902d51bb71212f2a817d6)
---
 doc/configuration.txt      | 10 ++++++++++
 include/types/proto_http.h |  1 +
 src/cfgparse.c             |  5 ++++-
 src/proto_http.c           | 10 ++++++++++
 4 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/doc/configuration.txt b/doc/configuration.txt
index 7ccc09c84..40907cad0 100644
--- a/doc/configuration.txt
+++ b/doc/configuration.txt
@@ -3004,6 +3004,12 @@ redirect prefix   <to> [code <code>] <option> {if | unless} <condition>
         for directing users to a non-secure page for instance. It has no effect
         with a location-type redirect.
 
+      - "append-slash"
+        This keyword may be used in conjunction with "drop-query" to redirect
+        users who use a URL not ending with a '/' to the same one with the '/'.
+        It can be useful to ensure that search engines will only see one URL.
+        For this, a return code 301 is preferred.
+
       - "set-cookie NAME[=value]"
         A "Set-Cookie" header will be added with NAME (and optionally "=value")
         to the response. This is sometimes used to indicate that a user has
@@ -3034,6 +3040,10 @@ redirect prefix   <to> [code <code>] <option> {if | unless} <condition>
         redirect location http://mysite.com/           if !login_page secure
         redirect location / clear-cookie USERID=       if logout
 
+  Example: send redirects for request for articles without a '/'.
+        acl missing_slash path_reg ^/article/[^/]*$
+        redirect code 301 prefix / drop-query append-slash if missing_slash
+
   See section 7 about ACL usage.
 
 
diff --git a/include/types/proto_http.h b/include/types/proto_http.h
index c0350d245..6ecfe1ee8 100644
--- a/include/types/proto_http.h
+++ b/include/types/proto_http.h
@@ -169,6 +169,7 @@ enum {
 enum {
 	REDIRECT_FLAG_NONE = 0,
 	REDIRECT_FLAG_DROP_QS = 1,	/* drop query string */
+	REDIRECT_FLAG_APPEND_SLASH = 2,	/* append a slash if missing at the end */
 };
 
 /* Redirect types (location, prefix, extended ) */
diff --git a/src/cfgparse.c b/src/cfgparse.c
index 3b99bfb13..534e1d164 100644
--- a/src/cfgparse.c
+++ b/src/cfgparse.c
@@ -1596,6 +1596,9 @@ int cfg_parse_listen(const char *file, int linenum, char **args, int inv)
 			else if (!strcmp(args[cur_arg],"drop-query")) {
 				flags |= REDIRECT_FLAG_DROP_QS;
 			}
+			else if (!strcmp(args[cur_arg],"append-slash")) {
+				flags |= REDIRECT_FLAG_APPEND_SLASH;
+			}
 			else if (!strcmp(args[cur_arg], "if")) {
 				pol = ACL_COND_IF;
 				cur_arg++;
@@ -1607,7 +1610,7 @@ int cfg_parse_listen(const char *file, int linenum, char **args, int inv)
 				break;
 			}
 			else {
-				Alert("parsing [%s:%d] : '%s' expects 'code', 'prefix', 'location', 'set-cookie', 'clear-cookie' or 'drop-query' (was '%s').\n",
+				Alert("parsing [%s:%d] : '%s' expects 'code', 'prefix', 'location', 'set-cookie', 'clear-cookie', 'drop-query' or 'append-slash' (was '%s').\n",
 				      file, linenum, args[0], args[cur_arg]);
 				err_code |= ERR_ALERT | ERR_FATAL;
 				goto out;
diff --git a/src/proto_http.c b/src/proto_http.c
index e656a29f9..2364bea30 100644
--- a/src/proto_http.c
+++ b/src/proto_http.c
@@ -2005,6 +2005,16 @@ int http_process_request(struct session *s, struct buffer *req)
 					/* add path */
 					memcpy(rdr.str + rdr.len, path, pathlen);
 					rdr.len += pathlen;
+
+					/* append a slash at the end of the location is needed and missing */
+					if (rdr.len && rdr.str[rdr.len - 1] != '/' &&
+					    (rule->flags & REDIRECT_FLAG_APPEND_SLASH)) {
+						if (rdr.len > sizeof(trash) - 5)
+							goto return_bad_req;
+						rdr.str[rdr.len] = '/';
+						rdr.len++;
+					}
+
 					break;
 				}
 				case REDIRECT_TYPE_LOCATION: