From 13417cf9c6eec555b51da49742e47939d2f5715b Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Fri, 19 Oct 2012 22:47:33 -0400 Subject: [PATCH] Exclude submodules from zip downloads (issue 151) --- src/com/gitblit/utils/HttpUtils.java | 59 ++++++++++++++++++++++++++++++++++++++++++++++++++++++----- 1 files changed, 54 insertions(+), 5 deletions(-) diff --git a/src/com/gitblit/utils/HttpUtils.java b/src/com/gitblit/utils/HttpUtils.java index 079d1a6..ad7d58c 100644 --- a/src/com/gitblit/utils/HttpUtils.java +++ b/src/com/gitblit/utils/HttpUtils.java @@ -32,15 +32,64 @@ * @return the host url */ public static String getGitblitURL(HttpServletRequest request) { + // default to the request scheme and port + String scheme = request.getScheme(); + int port = request.getServerPort(); + + // try to use reverse-proxy server's port + String forwardedPort = request.getHeader("X-Forwarded-Port"); + if (StringUtils.isEmpty(forwardedPort)) { + forwardedPort = request.getHeader("X_Forwarded_Port"); + } + if (!StringUtils.isEmpty(forwardedPort)) { + // reverse-proxy server has supplied the original port + try { + port = Integer.parseInt(forwardedPort); + } catch (Throwable t) { + } + } + + // try to use reverse-proxy server's scheme + String forwardedScheme = request.getHeader("X-Forwarded-Proto"); + if (StringUtils.isEmpty(forwardedScheme)) { + forwardedScheme = request.getHeader("X_Forwarded_Proto"); + } + if (!StringUtils.isEmpty(forwardedScheme)) { + // reverse-proxy server has supplied the original scheme + scheme = forwardedScheme; + + if ("https".equals(scheme) && port == 80) { + // proxy server is https, inside server is 80 + // this is likely because the proxy server has not supplied + // x-forwarded-port. since 80 is almost definitely wrong, + // make an educated guess that 443 is correct. + port = 443; + } + } + + String context = request.getContextPath(); + String forwardedContext = request.getHeader("X-Forwarded-Context"); + if (forwardedContext != null) { + forwardedContext = request.getHeader("X_Forwarded_Context"); + } + if (!StringUtils.isEmpty(forwardedContext)) { + context = forwardedContext; + } + + // trim any trailing slash + if (context.length() > 0 && context.charAt(context.length() - 1) == '/') { + context = context.substring(1); + } + StringBuilder sb = new StringBuilder(); - sb.append(request.getScheme()); + sb.append(scheme); sb.append("://"); sb.append(request.getServerName()); - if ((request.getScheme().equals("http") && request.getServerPort() != 80) - || (request.getScheme().equals("https") && request.getServerPort() != 443)) { - sb.append(":" + request.getServerPort()); + if (("http".equals(scheme) && port != 80) + || ("https".equals(scheme) && port != 443)) { + sb.append(":" + port); } - sb.append(request.getContextPath()); + sb.append(context); return sb.toString(); } } -- Gitblit v1.9.1