~ubuntu-branches/ubuntu/feisty/tcpreen/feisty

« back to all changes in this revision

Viewing changes to src/log.cc

  • Committer: Bazaar Package Importer
  • Author(s): Oliver Kurth
  • Date: 2003-03-04 23:19:35 UTC
  • Revision ID: james.westby@ubuntu.com-20030304231935-rlvnd0mv4be1n8rn
Tags: upstream-1.2.2
Import upstream version 1.2.2

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
/*
 
2
 * log.cc - log files handling functions
 
3
 * $Id: log.cc,v 1.9 2003/01/26 09:09:08 rdenisc Exp $
 
4
 */
 
5
 
 
6
/***********************************************************************
 
7
 *  Copyright (C) 2002-2003 R�mi Denis-Courmont.                       *
 
8
 *  This program is free software; you can redistribute and/or modify  *
 
9
 *  it under the terms of the GNU General Public License as published  *
 
10
 *  by the Free Software Foundation; version 2 of the license.         *
 
11
 *                                                                     *
 
12
 *  This program is distributed in the hope that it will be useful,    *
 
13
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of     *
 
14
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.               *
 
15
 *  See the GNU General Public License for more details.               *
 
16
 *                                                                     *
 
17
 *  You should have received a copy of the GNU General Pulic License   *
 
18
 *  along with this program; if not, you can get it from:              *
 
19
 *  http://www.gnu.org/copyleft/gpl.html                               *
 
20
 ***********************************************************************/
 
21
 
 
22
#ifdef HAVE_CONFIG_H
 
23
# include <config.h>
 
24
#endif
 
25
 
 
26
#include <stdio.h>
 
27
#include "log.h"
 
28
#include "safopen.h"
 
29
#include <string.h>
 
30
#include <time.h>
 
31
 
 
32
 
 
33
/*** DataLog class implementation ***/
 
34
DataLog::~DataLog (void)
 
35
{
 
36
        if ((out != NULL) && close_out)
 
37
                fclose (out);
 
38
}
 
39
 
 
40
 
 
41
void
 
42
DataLog::Connect (const char *, const char *)
 
43
{
 
44
 
45
 
 
46
 
 
47
int
 
48
DataLog::WriteServerData (const void *, int length, int)
 
49
{
 
50
        return length;
 
51
}
 
52
 
 
53
 
 
54
int
 
55
DataLog::WriteClientData (const void *, int length, int)
 
56
{
 
57
        return length;
 
58
}
 
59
 
 
60
 
 
61
void
 
62
DataLog::ShutdownServer (void)
 
63
{
 
64
}
 
65
 
 
66
 
 
67
void
 
68
DataLog::ShutdownClient (void)
 
69
{
 
70
}
 
71
 
 
72
 
 
73
/*** DataLogList implementation ***/
 
74
int
 
75
DataLogList::AddLog (DataLog *log)
 
76
{
 
77
        if (log == NULL)
 
78
                return -1;
 
79
        
 
80
        struct listnode *node;
 
81
 
 
82
        try
 
83
        {
 
84
                node = new listnode;
 
85
        }
 
86
        catch (...)
 
87
        {
 
88
                return -1;
 
89
        }
 
90
 
 
91
        node->log = log;
 
92
        node->next = head;
 
93
        head = node;
 
94
        return 0;
 
95
}
 
96
 
 
97
 
 
98
DataLogList::~DataLogList (void)
 
99
{
 
100
        for (struct listnode *node = head; node != NULL; node = head)
 
101
        {
 
102
                head = node->next;
 
103
                delete node->log;
 
104
                delete node;
 
105
        }
 
106
}
 
107
 
 
108
 
 
109
void
 
110
DataLogList::WriteServerData (const void *data, int length, int oob)
 
111
{
 
112
        for (struct listnode *node = head; node != NULL; node = node->next)
 
113
                /* FIXME: errors not handled */
 
114
                node->log->WriteServerData (data, length, oob);
 
115
}
 
116
 
 
117
 
 
118
void
 
119
DataLogList::WriteClientData (const void *data, int length, int oob)
 
120
{
 
121
        for (struct listnode *node = head; node != NULL; node = node->next)
 
122
                /* FIXME: errors not handled */
 
123
                node->log->WriteClientData (data, length, oob);
 
124
}
 
125
 
 
126
 
 
127
void
 
128
DataLogList::ShutdownServer (void)
 
129
{
 
130
        for (struct listnode *node = head; node != NULL; node = node->next)
 
131
                node->log->ShutdownServer ();
 
132
}
 
133
 
 
134
 
 
135
void
 
136
DataLogList::ShutdownClient (void)
 
137
{
 
138
        for (struct listnode *node = head; node != NULL; node = node->next)
 
139
                node->log->ShutdownClient ();
 
140
}
 
141
 
 
142
 
 
143
void
 
144
DataLogList::Connect (const char *server, const char *client)
 
145
{
 
146
        for (struct listnode *node = head; node != NULL; node = node->next)
 
147
                node->log->Connect (server, client);
 
148
}
 
149
 
 
150
 
 
151
/*** DataLogListMaker implementation ***/
 
152
int
 
153
DataLogListMaker::AddLogMaker (DataLog *(*maker) (void), FILE *stream)
 
154
{
 
155
        struct listnode *node;
 
156
 
 
157
        try
 
158
        {
 
159
                node = new struct listnode;
 
160
        }
 
161
        catch (...)
 
162
        {
 
163
                return -1;
 
164
        }
 
165
 
 
166
        node->maker = maker;
 
167
        node->next = head;
 
168
        node->basename = NULL;
 
169
        node->stream = stream;
 
170
        head = node;
 
171
        return 0;
 
172
}
 
173
 
 
174
 
 
175
int
 
176
DataLogListMaker::AddLogMaker (DataLog *(*maker) (void), const char *prefix)
 
177
{
 
178
        struct listnode *node;
 
179
 
 
180
        try
 
181
        {
 
182
                node = new struct listnode;
 
183
        }
 
184
        catch (...)
 
185
        {
 
186
                return -1;
 
187
        }
 
188
 
 
189
        node->maker = maker;
 
190
        node->next = head;
 
191
 
 
192
        /* FIXME: could be improved */
 
193
        if (prefix != NULL)
 
194
        {
 
195
                int len = strlen (prefix) + 1;
 
196
                char *buf;
 
197
                try
 
198
                {
 
199
                        buf = new char[len];
 
200
                }
 
201
                catch (...)
 
202
                {
 
203
                        delete node;
 
204
                        return -1;
 
205
                }
 
206
                memcpy (buf, prefix, len);
 
207
                
 
208
                node->basename = buf;
 
209
        }
 
210
        
 
211
        node->stream = NULL;
 
212
        head = node;
 
213
        return 0;
 
214
}
 
215
 
 
216
 
 
217
DataLogListMaker::~DataLogListMaker (void)
 
218
{
 
219
        for (struct listnode *node = head; node != NULL; node = head)
 
220
        {
 
221
                head = node->next;
 
222
                if (node->basename != NULL)
 
223
                        delete node->basename;
 
224
                delete node;
 
225
        }
 
226
}
 
227
 
 
228
 
 
229
static
 
230
char *valid_chars (char *str)
 
231
{
 
232
        char c;
 
233
        
 
234
        for (char *ptr = str; (c = *ptr) != 0; ptr++)
 
235
                switch (c)
 
236
                {
 
237
                        case '[':
 
238
                        case ']':
 
239
                        case ':':
 
240
                                *ptr = '-';
 
241
                }
 
242
        return str;
 
243
}
 
244
 
 
245
 
 
246
DataLogList *
 
247
DataLogListMaker::MakeLogList (const char *suffix)
 
248
{
 
249
        DataLogList *list;
 
250
 
 
251
        try
 
252
        {
 
253
                list = new DataLogList;
 
254
        }
 
255
        catch (...)
 
256
        {
 
257
                return NULL;
 
258
        }
 
259
 
 
260
        // Creates a log for each maker in the list
 
261
        for (struct listnode *node = head; node != NULL; node = node->next)
 
262
        {
 
263
                DataLog *log;
 
264
                
 
265
                try
 
266
                {
 
267
                        log = node->maker ();
 
268
                }
 
269
                catch (...)
 
270
                {
 
271
                        delete list;
 
272
                        return NULL;
 
273
                }
 
274
                
 
275
                if (log != NULL)
 
276
                {
 
277
                        // Adds a log in the list
 
278
                        if (list->AddLog (log))
 
279
                        {
 
280
                                delete log;
 
281
                                delete list;
 
282
                                return NULL;
 
283
                        }
 
284
 
 
285
                        // Attaches a stream to the new log
 
286
                        if (node->stream != NULL)
 
287
                                log->SetStream (node->stream, 0);
 
288
                        else
 
289
                        {       // Do we need more abstraction?
 
290
                                const char *prefix = node->basename;
 
291
                                if (prefix == NULL)
 
292
                                        prefix = "";
 
293
                                if (suffix == NULL)
 
294
                                        suffix = "";
 
295
                                
 
296
                                char *filename;
 
297
                                
 
298
                                int len = strlen (prefix) + strlen (suffix)+1;
 
299
 
 
300
                                try
 
301
                                {
 
302
                                        filename = new char[len];
 
303
                                }
 
304
                                catch (...)
 
305
                                {
 
306
                                        delete list;
 
307
                                        return NULL;
 
308
                                }
 
309
                                sprintf (filename, "%s%s", prefix, suffix);
 
310
                                valid_chars (filename);
 
311
                                
 
312
                                FILE *stream = secure_append_fopen (filename);
 
313
                                delete filename;
 
314
 
 
315
                                if (stream == NULL)
 
316
                                {
 
317
                                        delete list;
 
318
                                        return NULL;
 
319
                                }
 
320
                                log->SetStream (stream);
 
321
                        }       
 
322
                }
 
323
        }
 
324
 
 
325
        return list;
 
326
}
 
327