Mercurial Hosting > nabble
comparison src/fschmidt/util/servlet/ServletUtils.java @ 68:00520880ad02
add fschmidt source
| author | Franklin Schmidt <fschmidt@gmail.com> |
|---|---|
| date | Sun, 05 Oct 2025 17:24:15 -0600 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| 67:9d0fefce6985 | 68:00520880ad02 |
|---|---|
| 1 /* | |
| 2 Copyright (c) 2008 Franklin Schmidt <fschmidt@gmail.com> | |
| 3 | |
| 4 Permission is hereby granted, free of charge, to any person obtaining a copy | |
| 5 of this software and associated documentation files (the "Software"), to deal | |
| 6 in the Software without restriction, including without limitation the rights | |
| 7 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
| 8 copies of the Software, and to permit persons to whom the Software is | |
| 9 furnished to do so, subject to the following conditions: | |
| 10 | |
| 11 The above copyright notice and this permission notice shall be included in | |
| 12 all copies or substantial portions of the Software. | |
| 13 | |
| 14 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
| 15 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
| 16 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |
| 17 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
| 18 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
| 19 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | |
| 20 THE SOFTWARE. | |
| 21 */ | |
| 22 | |
| 23 package fschmidt.util.servlet; | |
| 24 | |
| 25 import fschmidt.util.java.Base64; | |
| 26 import fschmidt.util.java.HtmlUtils; | |
| 27 | |
| 28 import javax.servlet.http.Cookie; | |
| 29 import javax.servlet.http.HttpServletRequest; | |
| 30 import javax.servlet.http.HttpServletResponse; | |
| 31 import javax.servlet.http.HttpUtils; | |
| 32 import java.io.IOException; | |
| 33 import java.net.MalformedURLException; | |
| 34 import java.net.URL; | |
| 35 import java.util.Collections; | |
| 36 import java.util.Enumeration; | |
| 37 import java.util.Iterator; | |
| 38 import java.util.Map; | |
| 39 | |
| 40 | |
| 41 public final class ServletUtils { | |
| 42 | |
| 43 private ServletUtils() { throw new RuntimeException(); } | |
| 44 | |
| 45 public static String getQueryString(HttpServletRequest request) { | |
| 46 return getQueryString(request,0); | |
| 47 } | |
| 48 | |
| 49 public static String getQueryString(HttpServletRequest request,int maxValueLen) { | |
| 50 String method = request.getMethod(); | |
| 51 if( method.equals("GET") ) | |
| 52 return request.getQueryString(); | |
| 53 if( !method.equals("POST") && !method.equals("HEAD") ) | |
| 54 throw new RuntimeException(method); | |
| 55 Enumeration en = request.getParameterNames(); | |
| 56 StringBuilder queryBuf = new StringBuilder(); | |
| 57 if( !en.hasMoreElements() ) | |
| 58 return null; | |
| 59 do { | |
| 60 String param = (String)en.nextElement(); | |
| 61 String value = request.getParameter(param); | |
| 62 if( maxValueLen > 0 ) { | |
| 63 int len = value.length(); | |
| 64 if( len > maxValueLen ) | |
| 65 value = value.substring(0,maxValueLen) + "..." + (len-maxValueLen); | |
| 66 } | |
| 67 queryBuf.append(param); | |
| 68 queryBuf.append('='); | |
| 69 queryBuf.append(value); | |
| 70 queryBuf.append('&'); | |
| 71 } while( en.hasMoreElements() ); | |
| 72 queryBuf.deleteCharAt(queryBuf.length() - 1); | |
| 73 return queryBuf.toString(); | |
| 74 } | |
| 75 | |
| 76 public static String getCurrentURL(HttpServletRequest request) { | |
| 77 return getCurrentURL(request,0); | |
| 78 } | |
| 79 | |
| 80 public static String getCurrentURL(HttpServletRequest request,int maxValueLen) { | |
| 81 // StringBuffer buf = HttpUtils.getRequestURL(request); | |
| 82 StringBuffer buf = request.getRequestURL(); | |
| 83 String qStr = getQueryString(request,maxValueLen); | |
| 84 if(qStr != null && qStr.length() > 0) { | |
| 85 buf.append('?'); | |
| 86 buf.append(qStr); | |
| 87 } | |
| 88 return buf.toString(); | |
| 89 } | |
| 90 | |
| 91 public static String fullURL(HttpServletRequest request,String relativeURL) | |
| 92 throws MalformedURLException | |
| 93 { | |
| 94 return new URL(new URL(request.getRequestURL().toString()),relativeURL).toString(); | |
| 95 } | |
| 96 | |
| 97 public static String getHost(HttpServletRequest request) { | |
| 98 /* | |
| 99 String host = request.getHeader("host"); | |
| 100 if( host.endsWith(":80") ) | |
| 101 host = host.substring(0,host.length()-3); | |
| 102 return host; | |
| 103 */ | |
| 104 String host = request.getServerName(); | |
| 105 int port = request.getServerPort(); | |
| 106 if( port != 80 ) { | |
| 107 host += ":" + port; | |
| 108 } | |
| 109 return host; | |
| 110 } | |
| 111 | |
| 112 static String getContextUrl(String scheme,String host,String contextPath) { | |
| 113 StringBuilder buf = new StringBuilder(); | |
| 114 buf.append( scheme ); | |
| 115 buf.append( "://" ); | |
| 116 buf.append( host ); | |
| 117 buf.append( contextPath ); | |
| 118 return buf.toString(); | |
| 119 } | |
| 120 | |
| 121 public static String getContextURL(HttpServletRequest request) { | |
| 122 return getContextUrl( request.getScheme(), getHost(request), request.getContextPath() ); | |
| 123 } | |
| 124 | |
| 125 public static String getServletPath(HttpServletRequest request,String relativeURL) | |
| 126 throws MalformedURLException | |
| 127 { | |
| 128 int i = relativeURL.indexOf('?'); | |
| 129 if( i != -1 ) | |
| 130 relativeURL = relativeURL.substring(0,i); | |
| 131 String url = fullURL(request,relativeURL); | |
| 132 String context = getContextURL(request); | |
| 133 if( !url.startsWith(context) ) | |
| 134 throw new RuntimeException("context="+context+" url="+url); | |
| 135 return url.substring(context.length()); | |
| 136 } | |
| 137 | |
| 138 public static Map<String,String[]> getParameterMap(String url) { | |
| 139 int i = url.indexOf('?'); | |
| 140 if( i == -1 ) | |
| 141 return Collections.emptyMap(); | |
| 142 String query = url.substring(i+1); | |
| 143 return parseQueryString(query); | |
| 144 } | |
| 145 | |
| 146 @SuppressWarnings("unchecked") | |
| 147 private static Map<String,String[]> parseQueryString(String query) { | |
| 148 return HttpUtils.parseQueryString(query); | |
| 149 } | |
| 150 | |
| 151 public static String getQueryString(Map<String,String[]> params) { | |
| 152 StringBuilder buf = new StringBuilder(); | |
| 153 for( Iterator<Map.Entry<String,String[]>> iter=params.entrySet().iterator(); iter.hasNext(); ) { | |
| 154 Map.Entry<String,String[]> entry = iter.next(); | |
| 155 String name = entry.getKey(); | |
| 156 String[] values = entry.getValue(); | |
| 157 for( int i=0; i<values.length; i++ ) { | |
| 158 if( buf.length() > 0 ) | |
| 159 buf.append( '&' ); | |
| 160 buf.append( name ); | |
| 161 buf.append( '=' ); | |
| 162 buf.append( HtmlUtils.urlEncode(values[i]) ); | |
| 163 } | |
| 164 } | |
| 165 return buf.toString(); | |
| 166 } | |
| 167 | |
| 168 private static String escape(String value) { | |
| 169 return value.replaceAll(";", "%3B"); | |
| 170 } | |
| 171 | |
| 172 private static String unescape(String value) { | |
| 173 return value.replaceAll("%3B", ";"); | |
| 174 } | |
| 175 | |
| 176 private static Cookie getCookie(HttpServletRequest request,String name) { | |
| 177 Cookie[] cookies = request.getCookies(); | |
| 178 if( cookies == null ) | |
| 179 return null; | |
| 180 for (Cookie cookie : cookies) { | |
| 181 if (cookie.getName().equals(name)) | |
| 182 return cookie; | |
| 183 } | |
| 184 return null; | |
| 185 } | |
| 186 | |
| 187 public static String getCookieValue(HttpServletRequest request,String name) { | |
| 188 Cookie cookie = getCookie(request,name); | |
| 189 return cookie==null ? null : unescape(cookie.getValue()); | |
| 190 } | |
| 191 | |
| 192 public static void setCookie(HttpServletRequest request,HttpServletResponse response,String name,String value,boolean isPersistent, String domain) { | |
| 193 Cookie cookie = getCookie(request,name); | |
| 194 if( cookie==null || !cookie.getValue().equals(value) ) { | |
| 195 cookie = new Cookie(name, escape(value)); | |
| 196 cookie.setPath("/"); | |
| 197 if (domain != null && domain.length() > 0) | |
| 198 cookie.setDomain(domain); | |
| 199 if( isPersistent ) | |
| 200 cookie.setMaxAge(10000000); | |
| 201 response.addCookie(cookie); | |
| 202 } | |
| 203 } | |
| 204 | |
| 205 public static void removeCookie(HttpServletRequest request, | |
| 206 HttpServletResponse response, | |
| 207 String name, | |
| 208 String domain | |
| 209 | |
| 210 ) { | |
| 211 Cookie cookie = getCookie(request, name); | |
| 212 if(cookie != null) { | |
| 213 Cookie delCookie = new Cookie(name, "delete"); | |
| 214 delCookie.setPath("/"); | |
| 215 delCookie.setMaxAge(0); | |
| 216 if (domain != null && domain.length() > 0) | |
| 217 delCookie.setDomain(domain); | |
| 218 response.addCookie(delCookie); | |
| 219 } | |
| 220 } | |
| 221 | |
| 222 | |
| 223 public static String getRemoteAddr(HttpServletRequest request) { | |
| 224 String addr = request.getHeader("X-Forwarded-For"); | |
| 225 if( addr==null ) | |
| 226 addr = request.getRemoteAddr(); | |
| 227 return addr; | |
| 228 } | |
| 229 | |
| 230 /* | |
| 231 public static boolean authenticate(HttpServletRequest request,HttpServletResponse response,String authRealm,String authUsernameAndPassword) | |
| 232 throws IOException | |
| 233 { | |
| 234 String auth = request.getHeader("Authorization"); | |
| 235 if( auth==null ) { | |
| 236 response.setHeader("WWW-Authenticate","Basic realm=\""+authRealm+"\""); | |
| 237 response.sendError(HttpServletResponse.SC_UNAUTHORIZED); | |
| 238 return false; | |
| 239 } | |
| 240 String[] a = auth.split(" +"); | |
| 241 if( a.length != 2 ) | |
| 242 throw new RuntimeException("auth = "+auth); | |
| 243 if( !a[0].equals("Basic") ) | |
| 244 throw new RuntimeException("auth = "+auth); | |
| 245 if( !new String(Base64.decode(a[1])).equals(authUsernameAndPassword) ) { | |
| 246 response.setHeader("WWW-Authenticate","Basic realm=\""+authRealm+"\""); | |
| 247 response.sendError(HttpServletResponse.SC_UNAUTHORIZED); | |
| 248 return false; | |
| 249 } | |
| 250 return true; | |
| 251 } | |
| 252 */ | |
| 253 public static String getAuthorization(HttpServletRequest request) { | |
| 254 String auth = request.getHeader("Authorization"); | |
| 255 if( auth==null ) | |
| 256 return null; | |
| 257 String[] a = auth.split(" +"); | |
| 258 if( a.length != 2 ) | |
| 259 throw new RuntimeException("auth = "+auth); | |
| 260 if( !a[0].equals("Basic") ) | |
| 261 throw new RuntimeException("auth = "+auth); | |
| 262 return new String(Base64.decode(a[1])); | |
| 263 } | |
| 264 | |
| 265 public static void sendAuthenticate(HttpServletResponse response,String authRealm) | |
| 266 throws IOException | |
| 267 { | |
| 268 response.setHeader("WWW-Authenticate","Basic realm=\""+authRealm+"\""); | |
| 269 response.sendError(HttpServletResponse.SC_UNAUTHORIZED); | |
| 270 } | |
| 271 } |
