Update configure to generate config.h
[yazpp-moved-to-github.git] / 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