Allow /stats to accept long names for the displayed statistics.
[ircu2.10.12-pk.git] / ircd / userload.c
1 /*
2  * Userload module by Michael L. VanLoon (mlv) <michaelv@iastate.edu>
3  * Written 2/93.  Originally grafted into irc2.7.2g 4/93.
4  * 
5  * Rewritten 9/97 by Carlo Wood (Run) <carlo@runaway.xs4all.nl>
6  * because previous version used ridiculous amounts of memory
7  * (stored all loads of the passed three days ~ 8 megs).
8  *
9  * IRC - Internet Relay Chat, ircd/userload.c
10  * Copyright (C) 1990 University of Oulu, Computing Center
11  *
12  * This program is free software; you can redistribute it and/or modify
13  * it under the terms of the GNU General Public License as published by
14  * the Free Software Foundation; either version 1, or (at your option)
15  * any later version.
16  *
17  * This program is distributed in the hope that it will be useful,
18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20  * GNU General Public License for more details.
21  *
22  * You should have received a copy of the GNU General Public License
23  * along with this program; if not, write to the Free Software
24  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
25  *
26  * $Id$
27  */
28 #include "config.h"
29
30 #include "userload.h"
31 #include "client.h"
32 #include "ircd.h"
33 #include "msg.h"
34 #include "numnicks.h"
35 #include "querycmds.h"
36 #include "s_misc.h"
37 #include "s_stats.h"
38 #include "send.h"
39 #include "struct.h"
40 #include "sys.h"
41
42 #include <stdio.h>
43 #include <string.h>
44 #include <time.h>
45
46 struct current_load_st current_load;    /* The current load */
47
48 static struct current_load_st cspm_sum; /* Number of connections times number
49                                            of seconds per minute. */
50 static struct current_load_st csph_sum; /* Number of connections times number
51                                            of seconds per hour. */
52 static struct current_load_st cspm[60]; /* Last 60 minutes */
53 static struct current_load_st csph[72]; /* Last 72 hours */
54
55 static int m_index, h_index;    /* Array indexes */
56
57 /*
58  * update_load
59  *
60  * A new connection was added or removed.
61  */
62 void update_load(void)
63 {
64   static struct tm tm_now;      /* Current time. */
65   static time_t last_sec;       /* Seconds of last time that
66                                    update_load() called. */
67   static time_t last_min;
68   static time_t last;           /* Last time that update_load() was called. */
69   static struct current_load_st last_load;      /* The load last time that
70                                                    update_load() was called. */
71   static int initialized;       /* Boolean, set when initialized. */
72   int diff_time;        /* Temp. variable used to hold time intervals
73                                    in seconds, minutes or hours. */
74
75   /* Update `current_load' */
76   current_load.client_count = UserStats.local_clients;
77   current_load.conn_count = UserStats.local_clients + UserStats.local_servers;
78
79   /* Nothing needed when still in the same second */
80   if (!(diff_time = CurrentTime - last))
81   {
82     last_load = current_load;   /* Update last_load to be the load last
83                                    time that update_load() was called. */
84     return;
85   }
86
87   /* If we get here we entered a new second */
88
89   /*
90    * Make sure we keep the accurate time in 'tm_now'
91    */
92   if ((tm_now.tm_sec += diff_time) > 59)
93   {
94     /* This is done once every minute */
95     diff_time = tm_now.tm_sec / 60;
96     tm_now.tm_sec -= 60 * diff_time;
97     if ((tm_now.tm_min += diff_time) > 59)
98     {
99       /* This is done once every hour */
100       diff_time = tm_now.tm_min / 60;
101       tm_now.tm_min -= 60 * diff_time;
102       if ((tm_now.tm_hour += diff_time) > 23)
103       {
104         tm_now = *localtime(&CurrentTime);      /* Only called once a day */
105         if (!initialized)
106         {
107           initialized = 1;
108           last_sec = 60;
109           last_min = tm_now.tm_min;
110         }
111       }
112     }
113
114     /* If we get here we entered a new minute */
115
116     /* Finish the calculation of cspm of the last minute first: */
117     diff_time = 60 - last_sec;
118     cspm_sum.conn_count += last_load.conn_count * diff_time;
119     cspm_sum.client_count += last_load.client_count * diff_time;
120     cspm_sum.local_count += last_load.local_count * diff_time;
121
122     /* Add the completed minute to the Connections*Seconds/Hour sum */
123     csph_sum.conn_count += cspm_sum.conn_count - cspm[m_index].conn_count;
124     csph_sum.client_count += cspm_sum.client_count - cspm[m_index].client_count;
125     csph_sum.local_count += cspm_sum.local_count - cspm[m_index].local_count;
126
127     /* Store the completed minute in an array */
128     cspm[m_index] = cspm_sum;
129
130     /* How long did last_cspm last ? */
131     diff_time = tm_now.tm_min - last_min;
132     last_min = tm_now.tm_min;
133
134     if (diff_time < 0)
135       diff_time += 60;          /* update_load() must be called at
136                                    _least_ once an hour */
137
138     if (diff_time > 1)          /* Did more then one minute pass ? */
139     {
140       /* Calculate the constant load during those extra minutes */
141       cspm_sum.conn_count = last_load.conn_count * 60;
142       cspm_sum.client_count = last_load.client_count * 60;
143       cspm_sum.local_count = last_load.local_count * 60;
144     }
145
146     for (;;)
147     {
148       /* Increase minute index */
149       if (++m_index == 60)
150       {
151         m_index = 0;
152         /* Keep a list of the last 72 hours */
153         csph[h_index] = csph_sum;
154         if (++h_index == 72)
155           h_index = 0;
156       }
157
158       if (--diff_time <= 0)     /* '<' to prevent endless loop if update_load()
159                                    was not called once an hour :/ */
160         break;
161
162       /* Add extra minutes to the Connections*Seconds/Hour sum */
163       csph_sum.conn_count += cspm_sum.conn_count - cspm[m_index].conn_count;
164       csph_sum.client_count +=
165           cspm_sum.client_count - cspm[m_index].client_count;
166       csph_sum.local_count += cspm_sum.local_count - cspm[m_index].local_count;
167
168       /* Store extra minutes in the array */
169       cspm[m_index] = cspm_sum;
170     }
171
172     /* Now start the calculation of the new minute: */
173     last_sec = tm_now.tm_sec;
174     cspm_sum.conn_count = last_load.conn_count * last_sec;
175     cspm_sum.client_count = last_load.client_count * last_sec;
176     cspm_sum.local_count = last_load.local_count * last_sec;
177   }
178   else
179   {
180     /* A new second, but the same minute as last time */
181     /* How long did last_load last ? */
182     diff_time = tm_now.tm_sec - last_sec;
183     last_sec = tm_now.tm_sec;
184     if (diff_time == 1)         /* Just one second ? */
185     {
186       cspm_sum.conn_count += last_load.conn_count;
187       cspm_sum.client_count += last_load.client_count;
188       cspm_sum.local_count += last_load.local_count;
189     }
190     else
191     {
192       /* More then one second */
193       /* At most 3 integer multiplication per second */
194       cspm_sum.conn_count += last_load.conn_count * diff_time;
195       cspm_sum.client_count += last_load.client_count * diff_time;
196       cspm_sum.local_count += last_load.local_count * diff_time;
197     }
198   }
199   last_load = current_load;     /* Update last_load to be the load last
200                                    time that update_load() was called. */
201   last = CurrentTime;
202 }
203
204 void
205 calc_load(struct Client *sptr, const struct StatDesc *sd, char *param)
206 {
207   /* *INDENT-OFF* */
208   static const char *header =
209   /*   ----.-  ----.-  ----  ----  ----   ------------ */
210       "Minute  Hour    Day   Yest. YYest. Userload for:";
211   /* *INDENT-ON* */
212   static const char *what[3] = {
213     "local clients",
214     "total clients",
215     "total connections"
216   };
217   int i, j, times[5][3];        /* [min,hour,day,Yest,YYest]
218                                    [local,client,conn] */
219   int last_m_index = m_index, last_h_index = h_index;
220
221   update_load();                /* We want stats accurate as of *now* */
222
223   if (--last_m_index < 0)
224     last_m_index = 59;
225   times[0][0] = (cspm[last_m_index].local_count + 3) / 6;
226   times[0][1] = (cspm[last_m_index].client_count + 3) / 6;
227   times[0][2] = (cspm[last_m_index].conn_count + 3) / 6;
228
229   times[1][0] = (csph_sum.local_count + 180) / 360;
230   times[1][1] = (csph_sum.client_count + 180) / 360;
231   times[1][2] = (csph_sum.conn_count + 180) / 360;
232
233   for (i = 2; i < 5; ++i)
234   {
235     times[i][0] = 43200;
236     times[i][1] = 43200;
237     times[i][2] = 43200;
238     for (j = 0; j < 24; ++j)
239     {
240       if (--last_h_index < 0)
241         last_h_index = 71;
242       times[i][0] += csph[last_h_index].local_count;
243       times[i][1] += csph[last_h_index].client_count;
244       times[i][2] += csph[last_h_index].conn_count;
245     }
246     times[i][0] /= 86400;
247     times[i][1] /= 86400;
248     times[i][2] /= 86400;
249   }
250
251   sendcmdto_one(&me, CMD_NOTICE, sptr, "%C :%s", sptr, header);
252   for (i = 0; i < 3; ++i)
253     sendcmdto_one(&me, CMD_NOTICE, sptr,
254                   "%C :%4d.%1d  %4d.%1d  %4d  %4d  %4d   %s", sptr,
255                   times[0][i] / 10, times[0][i] % 10,
256                   times[1][i] / 10, times[1][i] % 10,
257                   times[2][i], times[3][i], times[4][i], what[i]);
258 }
259
260 void initload(void)
261 {
262   memset(&current_load, 0, sizeof(current_load));
263   update_load();                /* Initialize the load list */
264 }