1 /* 2 * libjingle 3 * Copyright 2004--2006, Google Inc. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright notice, 9 * this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright notice, 11 * this list of conditions and the following disclaimer in the documentation 12 * and/or other materials provided with the distribution. 13 * 3. The name of the author may not be used to endorse or promote products 14 * derived from this software without specific prior written permission. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO 19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 */ 27 28 #include <list> 29 #include <string> 30 31 #include "talk/xmpp/ratelimitmanager.h" 32 33 namespace buzz { 34 35 RateLimitManager::RateLimit* RateLimitManager::GetRateLimit( 36 const std::string event_name) { 37 RateLimitMap::iterator it = rate_limits_.find(event_name); 38 if (it != rate_limits_.end()) { 39 return it->second; 40 } 41 return NULL; 42 } 43 44 bool RateLimitManager::IsWithinRateLimit(const std::string event_name) { 45 RateLimit* current_rate = GetRateLimit(event_name); 46 if (current_rate) { 47 return current_rate->IsWithinRateLimit(); 48 } 49 return true; // If no rate limit is set, then you must be under the limit 50 } 51 52 void RateLimitManager::UpdateRateLimit(const std::string event_name, 53 int max_count, 54 int per_x_seconds) { 55 RateLimit* current_rate = GetRateLimit(event_name); 56 if (!current_rate) { 57 current_rate = new RateLimit(max_count, per_x_seconds); 58 rate_limits_[event_name] = current_rate; 59 } 60 current_rate->UpdateRateLimit(); 61 } 62 63 bool RateLimitManager::VerifyRateLimit(const std::string event_name, 64 int max_count, 65 int per_x_seconds) { 66 return VerifyRateLimit(event_name, max_count, per_x_seconds, false); 67 } 68 69 bool RateLimitManager::VerifyRateLimit(const std::string event_name, 70 int max_count, 71 int per_x_seconds, 72 bool always_update) { 73 bool within_rate_limit = IsWithinRateLimit(event_name); 74 if (within_rate_limit || always_update) { 75 UpdateRateLimit(event_name, max_count, per_x_seconds); 76 } 77 return within_rate_limit; 78 } 79 80 } 81
