src/limit-connect.cpp

   1 /* This file is part of the yazpp toolkit.
   2  * Copyright (C) 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::Rep {
  41     int m_period;
  42     Peer *m_peers;
  43     Peer **lookup(const char *peername);
  44 };
  45
  46 struct LimitConnect::Peer {
  47     Peer(int sz, const char *peername);
  48     ~Peer();
  49     void add_connect();
  50
  51     char *m_peername;
  52     TimeStat m_bw;
  53     Peer *m_next;
  54 };
  55
  56 LimitConnect::LimitConnect()
  57 {
  58     m_p = new Rep;
  59     m_p->m_period = 60;
  60     m_p->m_peers = 0;
  61 }
  62
  63
  64 LimitConnect::~LimitConnect()
  65 {
  66     cleanup(true);
  67     delete m_p;
  68 }
  69
  70 void LimitConnect::set_period(int sec)
  71 {
  72     m_p->m_period = sec;
  73 }
  74
  75 LimitConnect::Peer::Peer(int sz, const char *peername) : m_bw(sz)
  76 {
  77     m_peername = xstrdup(peername);
  78     m_next = 0;
  79 }
  80
  81 LimitConnect::Peer::~Peer()
  82 {
  83     xfree(m_peername);
  84 }
  85
  86 void LimitConnect::Peer::add_connect()
  87 {
  88     m_bw.add_bytes(1);
  89 }
  90
  91 LimitConnect::Peer **LimitConnect::Rep::lookup(const char *peername)
  92 {
  93     Peer **p = &m_peers;
  94     while (*p)
  95     {
  96         if (!strcmp((*p)->m_peername, peername))
  97             break;
  98         p = &(*p)->m_next;
  99     }
 100     return p;
 101 }
 102
 103 void LimitConnect::add_connect(const char *peername)
 104 {
 105     Peer **p = m_p->lookup(peername);
 106     if (!*p)
 107         *p = new Peer(m_p->m_period, peername);
 108     (*p)->add_connect();
 109 }
 110
 111 int LimitConnect::get_total(const char *peername)
 112 {
 113     Peer **p = m_p->lookup(peername);
 114     if (!*p)
 115         return 0;
 116     return (*p)->m_bw.get_total();
 117 }
 118
 119 void LimitConnect::cleanup(bool all)
 120 {
 121     Peer **p = &m_p->m_peers;
 122     while (*p)
 123     {
 124         Peer *tp = *p;
 125         if (all || (tp->m_bw.get_total() == 0))
 126         {
 127             *p = tp->m_next;
 128             delete tp;
 129         }
 130         else
 131             p = &tp->m_next;
 132     }
 133 }
 134
 135 /*
 136  * Local variables:
 137  * c-basic-offset: 4
 138  * c-file-style: "Stroustrup"
 139  * indent-tabs-mode: nil
 140  * End:
 141  * vim: shiftwidth=4 tabstop=8 expandtab
 142  */
 143