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 #ifndef _OASYS_RATE_LIMITED_SOCKET_H_ 00039 #define _OASYS_RATE_LIMITED_SOCKET_H_ 00040 00041 // XXX/demmer for now this only works for IPSocket, but once Bowei 00042 // finishes the IO porting, there will be a new base class that this 00043 // class should use 00044 #include "IPSocket.h" 00045 #include "../debug/Log.h" 00046 #include "../util/TokenBucket.h" 00047 00048 namespace oasys { 00049 00057 class RateLimitedSocket : public Logger { 00058 public: 00062 RateLimitedSocket(const char* logpath, 00063 u_int32_t rate, 00064 IPSocket* socket = NULL); 00065 00074 int send(const char* bp, size_t len, int flags); 00075 00084 int sendto(char* bp, size_t len, int flags, 00085 in_addr_t addr, u_int16_t port); 00086 00087 00089 TokenBucket* bucket() { return &bucket_; } 00090 IPSocket* socket() { return socket_; } 00092 00094 void set_socket(IPSocket* sock) { socket_ = sock; } 00096 00097 protected: 00098 TokenBucket bucket_; 00099 IPSocket* socket_; 00100 }; 00101 00102 } // namespace oasys 00103 00104 #endif /* _OASYS_RATE_LIMITED_SOCKET_H_ */