00001 /* 00002 * IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING. By 00003 * downloading, copying, installing or using the software you agree to 00004 * this license. If you do not agree to this license, do not download, 00005 * install, copy or use the software. 00006 * 00007 * Intel Open Source License 00008 * 00009 * Copyright (c) 2006 Intel Corporation. All rights reserved. 00010 * 00011 * Redistribution and use in source and binary forms, with or without 00012 * modification, are permitted provided that the following conditions are 00013 * met: 00014 * 00015 * Redistributions of source code must retain the above copyright 00016 * notice, this list of conditions and the following disclaimer. 00017 * 00018 * Redistributions in binary form must reproduce the above copyright 00019 * notice, this list of conditions and the following disclaimer in the 00020 * documentation and/or other materials provided with the distribution. 00021 * 00022 * Neither the name of the Intel Corporation nor the names of its 00023 * contributors may be used to endorse or promote products derived from 00024 * this software without specific prior written permission. 00025 * 00026 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 00027 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 00028 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 00029 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE INTEL OR 00030 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 00031 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 00032 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 00033 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 00034 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 00035 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 00036 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00037 */ 00038 00039 #include "RateLimitedSocket.h" 00040 00041 namespace oasys { 00042 00043 //---------------------------------------------------------------------- 00044 RateLimitedSocket::RateLimitedSocket(const char* logpath, 00045 u_int32_t rate, 00046 IPSocket* socket) 00047 : Logger("RateLimitedSocket", logpath), 00048 bucket_(logpath, rate, 65535 * 8 /* max udp packet */), 00049 socket_(socket) 00050 { 00051 } 00052 00053 //---------------------------------------------------------------------- 00054 int 00055 RateLimitedSocket::send(const char* bp, size_t len, int flags) 00056 { 00057 ASSERT(socket_ != NULL); 00058 00059 if (bucket_.rate() != 0) { 00060 bool can_send = bucket_.drain(len * 8); 00061 if (!can_send) { 00062 log_debug("can't send %zu byte packet since only %u tokens in bucket", 00063 len, bucket_.tokens()); 00064 return IORATELIMIT; 00065 } 00066 00067 log_debug("%u tokens sufficient for %zu byte packet", 00068 bucket_.tokens(), len); 00069 } 00070 00071 return socket_->send(bp, len, flags); 00072 } 00073 00074 //---------------------------------------------------------------------- 00075 int 00076 RateLimitedSocket::sendto(char* bp, size_t len, int flags, 00077 in_addr_t addr, u_int16_t port) 00078 { 00079 ASSERT(socket_ != NULL); 00080 00081 if (bucket_.rate() != 0) { 00082 bool can_send = bucket_.drain(len * 8); 00083 if (!can_send) { 00084 log_debug("can't send %zu byte packet since only %u tokens in bucket", 00085 len, bucket_.tokens()); 00086 return IORATELIMIT; 00087 } 00088 00089 log_debug("%u tokens sufficient for %zu byte packet", 00090 bucket_.tokens(), len); 00091 } 00092 00093 return socket_->sendto(bp, len, flags, addr, port); 00094 } 00095 00096 } // namespace oasys