RateLimitedSocket.cc

Go to the documentation of this file.
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

Generated on Fri Dec 22 14:48:00 2006 for DTN Reference Implementation by  doxygen 1.5.1