src/limit-connect.cpp

   1 /* This file is part of the yazpp toolkit.
   2  * Copyright (C) 1998-2011 Index Data and Mike Taylor
   3  * All rights reserved.
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are met:
   6  *
   7  *     * Redistributions of source code must retain the above copyright
   8  *       notice, this list of conditions and the following disclaimer.
   9  *     * Redistributions in binary form must reproduce the above copyright
  10  *       notice, this list of conditions and the following disclaimer in the
  11  *       documentation and/or other materials provided with the distribution.
  12  *     * Neither the name of Index Data nor the names of its contributors
  13  *       may be used to endorse or promote products derived from this
  14  *       software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
  17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  19  * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
  20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #if HAVE_CONFIG_H
  29 #include <config.h>
  30 #endif
  31 #include <yazpp/limit-connect.h>
  32
  33 #include <time.h>
  34 #include <string.h>
  35 #include <yaz/xmalloc.h>
  36 #include <yazpp/timestat.h>
  37
  38 using namespace yazpp_1;
  39
  40 struct LimitConnect::Peer {
  41     friend class LimitConnect;
  42
  43     Peer(int sz, const char *peername);
  44     ~Peer();
  45     void add_connect();
  46
  47     char *m_peername;
  48     TimeStat m_bw;
  49     Peer *m_next;
  50 };
  51
  52 LimitConnect::LimitConnect()
  53 {
  54     m_period = 60;
  55     m_peers = 0;
  56 }
  57
  58
  59 LimitConnect::~LimitConnect()
  60 {
  61     cleanup(true);
  62 }
  63
  64 void LimitConnect::set_period(int sec)
  65 {
  66     m_period = sec;
  67 }
  68
  69 LimitConnect::Peer::Peer(int sz, const char *peername) : m_bw(sz)
  70 {
  71     m_peername = xstrdup(peername);
  72     m_next = 0;
  73 }
  74
  75 LimitConnect::Peer::~Peer()
  76 {
  77     xfree(m_peername);
  78 }
  79
  80 void LimitConnect::Peer::add_connect()
  81 {
  82     m_bw.add_bytes(1);
  83 }
  84
  85 LimitConnect::Peer **LimitConnect::lookup(const char *peername)
  86 {
  87     Peer **p = &m_peers;
  88     while (*p)
  89     {
  90         if (!strcmp((*p)->m_peername, peername))
  91             break;
  92         p = &(*p)->m_next;
  93     }
  94     return p;
  95 }
  96
  97 void LimitConnect::add_connect(const char *peername)
  98 {
  99     Peer **p = lookup(peername);
 100     if (!*p)
 101         *p = new Peer(m_period, peername);
 102     (*p)->add_connect();
 103 }
 104
 105 int LimitConnect::get_total(const char *peername)
 106 {
 107     Peer **p = lookup(peername);
 108     if (!*p)
 109         return 0;
 110     return (*p)->m_bw.get_total();
 111 }
 112
 113 void LimitConnect::cleanup(bool all)
 114 {
 115     Peer **p = &m_peers;
 116     while (*p)
 117     {
 118         Peer *tp = *p;
 119         if (all || (tp->m_bw.get_total() == 0))
 120         {
 121             *p = tp->m_next;
 122             delete tp;
 123         }
 124         else
 125             p = &tp->m_next;
 126     }
 127 }
 128
 129 /*
 130  * Local variables:
 131  * c-basic-offset: 4
 132  * c-file-style: "Stroustrup"
 133  * indent-tabs-mode: nil
 134  * End:
 135  * vim: shiftwidth=4 tabstop=8 expandtab
 136  */
 137