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