src/acl/MaxConnection.cc

   1 /*
   2  * $Id$
   3  *
   4  * DEBUG: section 28    Access Control
   5  * AUTHOR: Duane Wessels
   6  *
   7  * SQUID Web Proxy Cache          http://www.squid-cache.org/
   8  * ----------------------------------------------------------
   9  *
  10  *  Squid is the result of efforts by numerous individuals from
  11  *  the Internet community; see the CONTRIBUTORS file for full
  12  *  details.   Many organizations have provided support for Squid's
  13  *  development; see the SPONSORS file for full details.  Squid is
  14  *  Copyrighted (C) 2001 by the Regents of the University of
  15  *  California; see the COPYRIGHT file for full details.  Squid
  16  *  incorporates software developed and/or copyrighted by other
  17  *  sources; see the CREDITS file for full details.
  18  *
  19  *  This program is free software; you can redistribute it and/or modify
  20  *  it under the terms of the GNU General Public License as published by
  21  *  the Free Software Foundation; either version 2 of the License, or
  22  *  (at your option) any later version.
  23  *
  24  *  This program is distributed in the hope that it will be useful,
  25  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  26  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  27  *  GNU General Public License for more details.
  28  *
  29  *  You should have received a copy of the GNU General Public License
  30  *  along with this program; if not, write to the Free Software
  31  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
  32  *
  33  *
  34  * Copyright (c) 2003, Robert Collins <robertc@squid-cache.org>
  35  */
  36
  37 #include "squid.h"
  38 #include "acl/FilledChecklist.h"
  39 #include "acl/MaxConnection.h"
  40 #include "wordlist.h"
  41
  42 ACL::Prototype ACLMaxConnection::RegistryProtoype(&ACLMaxConnection::RegistryEntry_, "maxconn");
  43
  44 ACLMaxConnection ACLMaxConnection::RegistryEntry_("maxconn");
  45
  46 ACL *
  47 ACLMaxConnection::clone() const
  48 {
  49     return new ACLMaxConnection(*this);
  50 }
  51
  52 ACLMaxConnection::ACLMaxConnection (char const *theClass) : class_ (theClass), limit(-1)
  53 {}
  54
  55 ACLMaxConnection::ACLMaxConnection (ACLMaxConnection const & old) :class_ (old.class_), limit (old.limit)
  56 {}
  57
  58 ACLMaxConnection::~ACLMaxConnection()
  59 {}
  60
  61 char const *
  62 ACLMaxConnection::typeString() const
  63 {
  64     return class_;
  65 }
  66
  67 bool
  68 ACLMaxConnection::empty () const
  69 {
  70     return false;
  71 }
  72
  73 bool
  74 ACLMaxConnection::valid () const
  75 {
  76     return limit > 0;
  77 }
  78
  79 void
  80 ACLMaxConnection::parse()
  81 {
  82     char *t = strtokFile();
  83
  84     if (!t)
  85         return;
  86
  87     limit = (atoi (t));
  88
  89     /* suck out file contents */
  90
  91     while ((t = strtokFile())) {
  92         limit = 0;
  93     }
  94 }
  95
  96 int
  97 ACLMaxConnection::match(ACLChecklist *checklist)
  98 {
  99     return clientdbEstablished(Filled(checklist)->src_addr, 0) > limit ? 1 : 0;
 100 }
 101
 102 wordlist *
 103 ACLMaxConnection::dump() const
 104 {
 105     if (!limit)
 106         return NULL;
 107
 108     wordlist *W = NULL;
 109
 110     char buf[32];
 111
 112     snprintf(buf, sizeof(buf), "%d", limit);
 113
 114     wordlistAdd(&W, buf);
 115
 116     return W;
 117 }
 118
 119 void
 120 ACLMaxConnection::prepareForUse()
 121 {
 122     if (0 != Config.onoff.client_db)
 123         return;
 124
 125     debugs(22, 0, "WARNING: 'maxconn' ACL (" << name << ") won't work with client_db disabled");
 126 }