Frames | No Frames |
1: /* 2: * CookieManager.java 3: * Copyright (C) 2004 The Free Software Foundation 4: * 5: * This file is part of GNU inetlib, a library. 6: * 7: * GNU inetlib is free software; you can redistribute it and/or modify 8: * it under the terms of the GNU General Public License as published by 9: * the Free Software Foundation; either version 2 of the License, or 10: * (at your option) any later version. 11: * 12: * GNU inetlib is distributed in the hope that it will be useful, 13: * but WITHOUT ANY WARRANTY; without even the implied warranty of 14: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 15: * GNU General Public License for more details. 16: * 17: * You should have received a copy of the GNU General Public License 18: * along with this library; if not, write to the Free Software 19: * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 20: * 21: * Linking this library statically or dynamically with other modules is 22: * making a combined work based on this library. Thus, the terms and 23: * conditions of the GNU General Public License cover the whole 24: * combination. 25: * 26: * As a special exception, the copyright holders of this library give you 27: * permission to link this library with independent modules to produce an 28: * executable, regardless of the license terms of these independent 29: * modules, and to copy and distribute the resulting executable under 30: * terms of your choice, provided that you also meet, for each linked 31: * independent module, the terms and conditions of the license of that 32: * module. An independent module is a module which is not derived from 33: * or based on this library. If you modify this library, you may extend 34: * this exception to your version of the library, but you are not 35: * obliged to do so. If you do not wish to do so, delete this 36: * exception statement from your version. 37: */ 38: 39: package gnu.inet.http; 40: 41: import java.util.ArrayList; 42: import java.util.Date; 43: import java.util.HashMap; 44: import java.util.Iterator; 45: import java.util.List; 46: import java.util.Map; 47: 48: /** 49: * A simple non-persistent cookie manager. This class can be extended to 50: * provide cookie persistence. 51: * 52: * @author <a href='mailto:dog@gnu.org'>Chris Burdess</a> 53: */ 54: public class SimpleCookieManager 55: implements CookieManager 56: { 57: 58: /** 59: * The cookie cache. 60: * This is a dictionary mapping domains to maps of cookies by name. 61: */ 62: protected Map cookies; 63: 64: /** 65: * Constructor. 66: */ 67: public SimpleCookieManager() 68: { 69: cookies = new HashMap(); 70: } 71: 72: public void setCookie(Cookie cookie) 73: { 74: String domain = cookie.getDomain(); 75: Map map =(Map) cookies.get(domain); 76: if (map == null) 77: { 78: map = new HashMap(); 79: cookies.put(domain, map); 80: } 81: String name = cookie.getName(); 82: map.put(name, cookie); // will replace a cookie of the same name 83: } 84: 85: public Cookie[] getCookies(String host, boolean secure, String path) 86: { 87: List matches = new ArrayList(); 88: Date now = new Date(); 89: if (Character.isLetter(host.charAt(0))) 90: { 91: int di = host.indexOf('.'); 92: while (di != -1) 93: { 94: addCookies(matches, host, secure, path, now); 95: host = host.substring(di); 96: di = host.indexOf('.', 1); 97: } 98: } 99: addCookies(matches, host, secure, path, now); 100: Cookie[] ret = new Cookie[matches.size()]; 101: matches.toArray(ret); 102: return ret; 103: } 104: 105: private void addCookies(List matches, String domain, boolean secure, 106: String path, Date now) 107: { 108: Map map = (Map) cookies.get(domain); 109: if (map != null) 110: { 111: List expired = new ArrayList(); 112: for (Iterator i = map.entrySet().iterator(); i.hasNext(); ) 113: { 114: Map.Entry entry = (Map.Entry) i.next(); 115: Cookie cookie = (Cookie) entry.getValue(); 116: Date expires = cookie.getExpiryDate(); 117: if (expires != null && expires.before(now)) 118: { 119: expired.add(entry.getKey()); 120: continue; 121: } 122: if (secure && !cookie.isSecure()) 123: { 124: continue; 125: } 126: if (path.startsWith(cookie.getPath())) 127: { 128: matches.add(cookie); 129: } 130: } 131: // Good housekeeping 132: for (Iterator i = expired.iterator(); i.hasNext(); ) 133: { 134: map.remove(i.next()); 135: } 136: } 137: } 138: 139: }