src/DelayId.cc

   1
   2 /*
   3  * $Id: DelayId.cc,v 1.3 2003/02/13 10:28:01 robertc Exp $
   4  *
   5  * DEBUG: section 77    Delay Pools
   6  * AUTHOR: Robert Collins <robertc@squid-cache.org>
   7  * Based upon original delay pools code by
   8  *   David Luyer <david@luyer.net>
   9  *
  10  * SQUID Web Proxy Cache          http://www.squid-cache.org/
  11  * ----------------------------------------------------------
  12  *
  13  *  Squid is the result of efforts by numerous individuals from
  14  *  the Internet community; see the CONTRIBUTORS file for full
  15  *  details.   Many organizations have provided support for Squid's
  16  *  development; see the SPONSORS file for full details.  Squid is
  17  *  Copyrighted (C) 2001 by the Regents of the University of
  18  *  California; see the COPYRIGHT file for full details.  Squid
  19  *  incorporates software developed and/or copyrighted by other
  20  *  sources; see the CREDITS file for full details.
  21  *
  22  *  This program is free software; you can redistribute it and/or modify
  23  *  it under the terms of the GNU General Public License as published by
  24  *  the Free Software Foundation; either version 2 of the License, or
  25  *  (at your option) any later version.
  26  *
  27  *  This program is distributed in the hope that it will be useful,
  28  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  29  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  30  *  GNU General Public License for more details.
  31  *
  32  *  You should have received a copy of the GNU General Public License
  33  *  along with this program; if not, write to the Free Software
  34  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
  35  *
  36  *
  37  * Copyright (c) 2003, Robert Collins <robertc@squid-cache.org>
  38  */
  39
  40 #include "config.h"
  41
  42 #if DELAY_POOLS
  43 #include "squid.h"
  44 #include "DelayId.h"
  45 #include "client_side_request.h"
  46 #include "ACLChecklist.h"
  47 #include "DelayPools.h"
  48 #include "DelayPool.h"
  49 #include "HttpRequest.h"
  50 /*
  51 #include "DelaySpec.h"
  52 #include "StoreClient.h"
  53 #include "Store.h"
  54 #include "MemObject.h"
  55 #include "ACL.h"
  56 #include "Config.h"
  57 #include "DelayId.h"
  58 #include "Array.h"
  59 #include "String.h"
  60 #include "CommonPool.h"
  61 #include "CompositePoolNode.h"
  62 #include "DelayVector.h"
  63 #include "NullDelayId.h"
  64 #include "DelayBucket.h"
  65 */
  66
  67 DelayId::DelayId () : pool_ (0), compositeId(NULL)
  68 {
  69 }
  70
  71 DelayId::DelayId (unsigned short aPool) :
  72 pool_ (aPool), compositeId (NULL)
  73 {
  74     debug (77,3)("DelayId::DelayId: Pool %du\n", aPool);
  75 }
  76
  77 DelayId::~DelayId ()
  78 {
  79 }
  80
  81 void
  82 DelayId::compositePosition(DelayIdComposite::Pointer newPosition)
  83 {
  84     compositeId = newPosition;
  85 }
  86
  87 unsigned short
  88 DelayId::pool() const
  89 {
  90     return pool_;
  91 }
  92
  93 bool
  94 DelayId::operator == (DelayId const &rhs) const
  95 {
  96     /* Doesn't compare composites properly....
  97      * only use to test against default ID's
  98      */
  99     return pool_ == rhs.pool_ && compositeId == rhs.compositeId;
 100 }
 101
 102 DelayId::operator bool() const
 103 {
 104     return pool_ || compositeId.getRaw();
 105 }
 106
 107 DelayId
 108 DelayId::DelayClient(clientHttpRequest * http)
 109 {
 110     request_t *r;
 111     unsigned short pool;
 112     assert(http);
 113     r = http->request;
 114
 115     if (r->client_addr.s_addr == INADDR_BROADCAST) {
 116         debug(77, 2) ("delayClient: WARNING: Called with 'allones' address, ignoring\n");
 117         return DelayId();
 118     }
 119
 120     ACLChecklist ch;
 121     ch.src_addr = r->client_addr;
 122     ch.my_addr = r->my_addr;
 123     ch.my_port = r->my_port;
 124     if (http->conn)
 125         ch.conn(cbdataReference(http->conn));
 126     ch.request = requestLink(r);
 127     for (pool = 0; pool < DelayPools::pools(); pool++)
 128         if (DelayPools::delay_data[pool].theComposite().getRaw() &&
 129             aclCheckFast(DelayPools::delay_data[pool].access, &ch)) {
 130             DelayId result (pool + 1);
 131             result.compositePosition(DelayPools::delay_data[pool].theComposite()->id(ch.src_addr, r->auth_user_request));
 132             return result;
 133         }
 134     return DelayId();
 135 }
 136
 137 /*
 138  * this returns the number of bytes the client is permitted. it does not take
 139  * into account bytes already buffered - that is up to the caller.
 140  */
 141 int
 142 DelayId::bytesWanted(int min, int max) const
 143 {
 144     /* unlimited */
 145     if (! (*this))
 146         return XMAX(min, max);
 147
 148     /* limited */
 149     int nbytes = XMAX(min, max);
 150     if (compositeId.getRaw())
 151         nbytes = compositeId->bytesWanted(min, nbytes);
 152     return nbytes;
 153 }
 154
 155 /*
 156  * this records actual bytes recieved.  always recorded, even if the
 157  * class is disabled - it's more efficient to just do it than to do all
 158  * the checks.
 159  */
 160 void
 161 DelayId::bytesIn(int qty)
 162 {
 163     if (! (*this))
 164         return;
 165
 166     unsigned short tempPool = pool() - 1;
 167
 168     assert (tempPool != 0xFFFF);
 169
 170     if (compositeId.getRaw())
 171         compositeId->bytesIn(qty);
 172 }
 173 #endif
 174