src/DelayUser.cc

   1 /*
   2  * Copyright (C) 1996-2014 The Squid Software Foundation and contributors
   3  *
   4  * Squid software is distributed under GPLv2+ license and includes
   5  * contributions from numerous individuals and organizations.
   6  * Please see the COPYING and CONTRIBUTORS files for details.
   7  */
   8
   9 /* DEBUG: section 77    Delay Pools */
  10
  11 #include "squid.h"
  12
  13 #if USE_DELAY_POOLS && USE_AUTH
  14 #include "auth/User.h"
  15 #include "auth/UserRequest.h"
  16 #include "comm/Connection.h"
  17 #include "DelayUser.h"
  18 #include "NullDelayId.h"
  19 #include "Store.h"
  20
  21 void *
  22 DelayUser::operator new(size_t size)
  23 {
  24     DelayPools::MemoryUsed += sizeof (DelayUser);
  25     return ::operator new (size);
  26 }
  27
  28 void
  29 DelayUser::operator delete (void *address)
  30 {
  31     DelayPools::MemoryUsed -= sizeof (DelayUser);
  32     ::operator delete (address);
  33 }
  34
  35 DelayUser::DelayUser()
  36 {
  37     DelayPools::registerForUpdates (this);
  38 }
  39
  40 static SplayNode<DelayUserBucket::Pointer>::SPLAYFREE DelayUserFree;
  41
  42 DelayUser::~DelayUser()
  43 {
  44     DelayPools::deregisterForUpdates (this);
  45     buckets.head->destroy (DelayUserFree);
  46 }
  47
  48 static SplayNode<DelayUserBucket::Pointer>::SPLAYCMP DelayUserCmp;
  49
  50 int
  51 DelayUserCmp(DelayUserBucket::Pointer const &left, DelayUserBucket::Pointer const &right)
  52 {
  53     /* Verify for re-currance of Bug 2127. either of these missing will crash strcasecmp() */
  54     assert(left->authUser->username() != NULL);
  55     assert(right->authUser->username() != NULL);
  56
  57     /* for rate limiting, case insensitive */
  58     return strcasecmp(left->authUser->username(), right->authUser->username());
  59 }
  60
  61 void
  62 DelayUserFree(DelayUserBucket::Pointer &)
  63 {}
  64
  65 void
  66 DelayUserStatsWalkee(DelayUserBucket::Pointer const &current, void *state)
  67 {
  68     current->stats ((StoreEntry *)state);
  69 }
  70
  71 void
  72 DelayUser::stats(StoreEntry * sentry)
  73 {
  74     spec.stats (sentry, "Per User");
  75
  76     if (spec.restore_bps == -1)
  77         return;
  78
  79     storeAppendPrintf(sentry, "\t\tCurrent: ");
  80
  81     if (!buckets.head) {
  82         storeAppendPrintf (sentry, "Not used yet.\n\n");
  83         return;
  84     }
  85
  86     buckets.head->walk(DelayUserStatsWalkee, sentry);
  87     storeAppendPrintf(sentry, "\n\n");
  88 }
  89
  90 void
  91 DelayUser::dump(StoreEntry *entry) const
  92 {
  93     spec.dump(entry);
  94 }
  95
  96 struct DelayUserUpdater {
  97     DelayUserUpdater (DelaySpec &_spec, int _incr):spec(_spec),incr(_incr) {};
  98
  99     DelaySpec spec;
 100     int incr;
 101 };
 102
 103 void
 104 DelayUserUpdateWalkee(DelayUserBucket::Pointer const &current, void *state)
 105 {
 106     DelayUserUpdater *t = (DelayUserUpdater *)state;
 107     /* This doesn't change the value of the DelayUserBucket, so is safe */
 108     const_cast<DelayUserBucket *>(current.getRaw())->theBucket.update(t->spec, t->incr);
 109 }
 110
 111 void
 112 DelayUser::update(int incr)
 113 {
 114     DelayUserUpdater updater(spec, incr);
 115     buckets.head->walk (DelayUserUpdateWalkee, &updater);
 116 }
 117
 118 void
 119 DelayUser::parse()
 120 {
 121     spec.parse();
 122 }
 123
 124 DelayIdComposite::Pointer
 125 DelayUser::id(CompositePoolNode::CompositeSelectionDetails &details)
 126 {
 127     if (!details.user || !details.user->user() || !details.user->user()->username())
 128         return new NullDelayId;
 129
 130     debugs(77, 3, HERE << "Adding a slow-down for User '" << details.user->user()->username() << "'");
 131     return new Id(this, details.user->user());
 132 }
 133
 134 void *
 135 DelayUser::Id::operator new(size_t size)
 136 {
 137     DelayPools::MemoryUsed += sizeof (Id);
 138     return ::operator new (size);
 139 }
 140
 141 void
 142 DelayUser::Id::operator delete (void *address)
 143 {
 144     DelayPools::MemoryUsed -= sizeof (Id);
 145     ::operator delete (address);
 146 }
 147
 148 void *
 149 DelayUserBucket::operator new(size_t size)
 150 {
 151     DelayPools::MemoryUsed += sizeof (DelayUserBucket);
 152     return ::operator new (size);
 153 }
 154
 155 void
 156 DelayUserBucket::operator delete(void *address)
 157 {
 158     DelayPools::MemoryUsed -= sizeof(DelayUserBucket);
 159     ::operator delete(address);
 160 }
 161
 162 DelayUserBucket::DelayUserBucket(Auth::User::Pointer aUser) : authUser(aUser)
 163 {
 164     debugs(77, 3, "DelayUserBucket::DelayUserBucket");
 165 }
 166
 167 DelayUserBucket::~DelayUserBucket()
 168 {
 169     authUser = NULL;
 170     debugs(77, 3, "DelayUserBucket::~DelayUserBucket");
 171 }
 172
 173 void
 174 DelayUserBucket::stats (StoreEntry *entry) const
 175 {
 176     storeAppendPrintf(entry, " %s:", authUser->username());
 177     theBucket.stats(entry);
 178 }
 179
 180 DelayUser::Id::Id(DelayUser::Pointer aDelayUser, Auth::User::Pointer aUser) : theUser(aDelayUser)
 181 {
 182     theBucket = new DelayUserBucket(aUser);
 183     DelayUserBucket::Pointer const *existing = theUser->buckets.find(theBucket, DelayUserCmp);
 184
 185     if (existing) {
 186         theBucket = *existing;
 187         return;
 188     }
 189
 190     theBucket->theBucket.init(theUser->spec);
 191     theUser->buckets.head = theUser->buckets.head->insert (theBucket, DelayUserCmp);
 192 }
 193
 194 DelayUser::Id::~Id()
 195 {
 196     debugs(77, 3, "DelayUser::Id::~Id");
 197 }
 198
 199 int
 200 DelayUser::Id::bytesWanted (int min, int max) const
 201 {
 202     return theBucket->theBucket.bytesWanted(min,max);
 203 }
 204
 205 void
 206 DelayUser::Id::bytesIn(int qty)
 207 {
 208     theBucket->theBucket.bytesIn(qty);
 209 }
 210
 211 #endif /* USE_DELAY_POOLS && USE_AUTH */
 212