src/limit-connect.cpp

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