Get rid of that ugly depencendy on COMM_ERR_CLOSING for freeing

[thirdparty/squid.git] / src / DelayId.cc

/*
 * $Id: DelayId.cc,v 1.4 2003/02/21 22:50:05 robertc Exp $
 *
 * DEBUG: section 77    Delay Pools
 * AUTHOR: Robert Collins <robertc@squid-cache.org>
 * Based upon original delay pools code by
 *   David Luyer <david@luyer.net>
 *
 * SQUID Web Proxy Cache          http://www.squid-cache.org/
 * ----------------------------------------------------------
 *
 *  Squid is the result of efforts by numerous individuals from
 *  the Internet community; see the CONTRIBUTORS file for full
 *  details.   Many organizations have provided support for Squid's
 *  development; see the SPONSORS file for full details.  Squid is
 *  Copyrighted (C) 2001 by the Regents of the University of
 *  California; see the COPYRIGHT file for full details.  Squid
 *  incorporates software developed and/or copyrighted by other
 *  sources; see the CREDITS file for full details.
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *  
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *  
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
 *
 *
 * Copyright (c) 2003, Robert Collins <robertc@squid-cache.org>
 */

#include "config.h"

#if DELAY_POOLS
#include "squid.h"
#include "DelayId.h"
#include "client_side_request.h"
#include "ACLChecklist.h"
#include "DelayPools.h"
#include "DelayPool.h"
#include "HttpRequest.h"
/*
#include "DelaySpec.h"
#include "StoreClient.h"
#include "Store.h"
#include "MemObject.h"
#include "ACL.h"
#include "Config.h"
#include "DelayId.h"
#include "Array.h"
#include "String.h"
#include "CommonPool.h"
#include "CompositePoolNode.h"
#include "DelayVector.h"
#include "NullDelayId.h"
#include "DelayBucket.h"
*/

DelayId::DelayId () : pool_ (0), compositeId(NULL)
{}

DelayId::DelayId (unsigned short aPool) :
        pool_ (aPool), compositeId (NULL)
{
    debug (77,3)("DelayId::DelayId: Pool %du\n", aPool);
}

DelayId::~DelayId ()
{}

void
DelayId::compositePosition(DelayIdComposite::Pointer newPosition)
{
    compositeId = newPosition;
}

unsigned short
DelayId::pool() const
{
    return pool_;
}

bool
DelayId::operator == (DelayId const &rhs) const
{
    /* Doesn't compare composites properly....
     * only use to test against default ID's 
     */
    return pool_ == rhs.pool_ && compositeId == rhs.compositeId;
}

DelayId::operator bool() const
{
    return pool_ || compositeId.getRaw();
}

DelayId
DelayId::DelayClient(clientHttpRequest * http)
{
    request_t *r;
    unsigned short pool;
    assert(http);
    r = http->request;

    if (r->client_addr.s_addr == INADDR_BROADCAST) {
        debug(77, 2) ("delayClient: WARNING: Called with 'allones' address, ignoring\n");
        return DelayId();
    }

    ACLChecklist ch;
    ch.src_addr = r->client_addr;
    ch.my_addr = r->my_addr;
    ch.my_port = r->my_port;

    if (http->conn)
        ch.conn(cbdataReference(http->conn));

    ch.request = requestLink(r);

    for (pool = 0; pool < DelayPools::pools(); pool++)
        if (DelayPools::delay_data[pool].theComposite().getRaw() &&
                aclCheckFast(DelayPools::delay_data[pool].access, &ch)) {
            DelayId result (pool + 1);
            result.compositePosition(DelayPools::delay_data[pool].theComposite()->id(ch.src_addr, r->auth_user_request));
            return result;
        }

    return DelayId();
}

/*
 * this returns the number of bytes the client is permitted. it does not take
 * into account bytes already buffered - that is up to the caller.
 */
int
DelayId::bytesWanted(int min, int max) const
{
    /* unlimited */

    if (! (*this))
        return XMAX(min, max);

    /* limited */
    int nbytes = XMAX(min, max);

    if (compositeId.getRaw())
        nbytes = compositeId->bytesWanted(min, nbytes);

    return nbytes;
}

/*
 * this records actual bytes recieved.  always recorded, even if the
 * class is disabled - it's more efficient to just do it than to do all
 * the checks.
 */
void
DelayId::bytesIn(int qty)
{
    if (! (*this))
        return;

    unsigned short tempPool = pool() - 1;

    assert (tempPool != 0xFFFF);

    if (compositeId.getRaw())
        compositeId->bytesIn(qty);
}

#endif