Source for gnu.inet.http.SimpleCookieManager

   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: }