Added buffer so we can buffer entire lines in a later patch.
[cascardo/rnetproxy.git] / pop.c
1 /*
2 ** Copyright (C) 2006 Thadeu Lima de Souza Cascardo <cascardo@minaslivre.org>
3 ** Copyright (C) 2009 Thadeu Lima de Souza Cascardo <cascardo@holoscopio.com>
4 **  
5 ** This program is free software; you can redistribute it and/or modify
6 ** it under the terms of the GNU General Public License as published by
7 ** the Free Software Foundation; either version 2 of the License, or
8 ** (at your option) any later version.
9 **  
10 ** This program is distributed in the hope that it will be useful,
11 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
12 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 ** GNU General Public License for more details.
14 **  
15 ** You should have received a copy of the GNU General Public License
16 ** along with this program; if not, write to the Free Software
17 ** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
18 **  
19 */
20
21 #include <gnet.h>
22 #include <glib.h>
23 #include <string.h>
24 #include "nethook.h"
25 #include "pop.h"
26
27 typedef struct
28 {
29   net_read orig_read;
30   gpointer orig_data;
31   GString *buffer;
32 } pop_t;
33
34 static void
35 pop_read (net_hook_t *hook, gchar *buffer, size_t len)
36 {
37   pop_t *pop = hook->data;
38   hook->data = pop->orig_data;
39   pop->orig_read (hook, buffer, len);
40   hook->data = pop;
41 }
42
43 net_hook_t *
44 pop_hook_new (net_hook_t *layer)
45 {
46   pop_t *pop;
47   pop = g_slice_new (pop_t);
48   pop->orig_read = layer->read;
49   pop->orig_data = layer->data;
50   pop->buffer = g_string_sized_new (4096);
51   layer->read = pop_read;
52   layer->data = pop;
53   return layer;
54 }
55
56 void
57 pop_destroy (net_hook_t *hook)
58 {
59   pop_t *pop = hook->data;
60   g_string_free (pop->buffer, TRUE);
61   hook->read = pop->orig_read;
62   hook->data = pop->orig_data;
63   g_slice_free (net_hook_t, (gpointer) pop);
64 }