source: protocols/events_libevent.c @ 19ac9c5

Last change on this file since 19ac9c5 was 19ac9c5, checked in by Wilmer van der Gaast <wilmer@…>, at 2006-05-13T19:44:59Z

Timeouts are now persistent.

  • Property mode set to 100644
File size: 5.6 KB
Line 
1  /********************************************************************\
2  * BitlBee -- An IRC to other IM-networks gateway                     *
3  *                                                                    *
4  * Copyright 2002-2006 Wilmer van der Gaast and others                *
5  \********************************************************************/
6
7/*
8 * Event handling (using libevent)
9 */
10
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 2 of the License, or
15  (at your option) 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 with
23  the Debian GNU/Linux distribution in /usr/share/common-licenses/GPL;
24  if not, write to the Free Software Foundation, Inc., 59 Temple Place,
25  Suite 330, Boston, MA  02111-1307  USA
26*/
27
28#define BITLBEE_CORE
29#include <stdio.h>
30#include <stdlib.h>
31#include <string.h>
32#include <sys/types.h>
33#include "proxy.h"
34
35#include <sys/time.h>
36#include <event.h>
37
38static guint id_next;
39static GHashTable *id_hash;
40
41/* Since libevent doesn't handle two event handlers for one fd-condition
42   very well (which happens sometimes when BitlBee changes event handlers
43   for a combination), let's buid some indexes so we can delete them here
44   already, just in time. */
45static GHashTable *read_hash;
46static GHashTable *write_hash;
47
48struct b_event_data
49{
50        guint id;
51        struct event evinfo;
52        gint timeout;
53        b_event_handler function;
54        void *data;
55};
56
57void b_main_init()
58{
59        event_init();
60       
61        id_next = 1;
62        id_hash = g_hash_table_new( g_int_hash, g_int_equal );
63        read_hash = g_hash_table_new( g_int_hash, g_int_equal );
64        write_hash = g_hash_table_new( g_int_hash, g_int_equal );
65}
66
67void b_main_run()
68{
69        event_dispatch();
70}
71
72void b_main_quit()
73{
74        struct timeval tv;
75       
76        memset( &tv, 0, sizeof( struct timeval ) );
77        event_loopexit( &tv );
78}
79
80static void b_event_passthrough( int fd, short event, void *data )
81{
82        struct b_event_data *b_ev = data;
83        b_input_condition cond = 0;
84        int id;
85       
86        if( fd >= 0 )
87        {
88                if( event & EV_READ )
89                        cond |= GAIM_INPUT_READ;
90                if( event & EV_WRITE )
91                        cond |= GAIM_INPUT_WRITE;
92        }
93       
94        event_debug( "b_event_passthrough( %d, %d, 0x%x ) (%d)\n", fd, event, (int) data, b_ev->id );
95       
96        /* Since the called function might cancel this handler already
97           (which free()s b_ev, we have to remember the ID here. */
98        id = b_ev->id;
99       
100        if( !b_ev->function( b_ev->data, fd, cond ) )
101        {
102                event_debug( "Handler returned FALSE: " );
103                b_event_remove( id );
104        }
105        else if( fd == -1 )
106        {
107                struct timeval tv;
108               
109                tv.tv_sec = b_ev->timeout / 1000;
110                tv.tv_usec = ( b_ev->timeout % 1000 ) * 1000;
111               
112                evtimer_add( &b_ev->evinfo, &tv );
113        }
114}
115
116gint b_input_add( gint fd, b_input_condition condition, b_event_handler function, gpointer data )
117{
118        struct b_event_data *b_ev;
119       
120        event_debug( "b_input_add( %d, %d, 0x%x, 0x%x ) ", fd, condition, function, data );
121       
122        if( ( condition & GAIM_INPUT_READ  && ( b_ev = g_hash_table_lookup( read_hash,  &fd ) ) ) ||
123            ( condition & GAIM_INPUT_WRITE && ( b_ev = g_hash_table_lookup( write_hash, &fd ) ) ) )
124        {
125                /* We'll stick with this libevent entry, but give it a new BitlBee id. */
126                g_hash_table_remove( id_hash, &b_ev->id );
127               
128                event_debug( "(replacing old handler (id = %d)) = %d\n", b_ev->id, id_next );
129               
130                b_ev->id = id_next++;
131                b_ev->function = function;
132                b_ev->data = data;
133        }
134        else
135        {
136                GIOCondition out_cond;
137               
138                event_debug( "(new) = %d\n", id_next );
139               
140                b_ev = g_new0( struct b_event_data, 1 );
141                b_ev->id = id_next++;
142                b_ev->function = function;
143                b_ev->data = data;
144               
145                out_cond = EV_PERSIST;
146                if( condition & GAIM_INPUT_READ )
147                        out_cond |= EV_READ;
148                if( condition & GAIM_INPUT_WRITE )
149                        out_cond |= EV_WRITE;
150               
151                event_set( &b_ev->evinfo, fd, out_cond, b_event_passthrough, b_ev );
152                event_add( &b_ev->evinfo, NULL );
153               
154                if( out_cond & EV_READ )
155                        g_hash_table_insert( read_hash, &b_ev->evinfo.ev_fd, b_ev );
156                if( out_cond & EV_WRITE )
157                        g_hash_table_insert( write_hash, &b_ev->evinfo.ev_fd, b_ev );
158        }
159       
160        g_hash_table_insert( id_hash, &b_ev->id, b_ev );
161        return b_ev->id;
162}
163
164/* TODO: Persistence for timers! */
165gint b_timeout_add( gint timeout, b_event_handler function, gpointer data )
166{
167        struct b_event_data *b_ev = g_new0( struct b_event_data, 1 );
168        struct timeval tv;
169       
170        b_ev->id = id_next++;
171        b_ev->timeout = timeout;
172        b_ev->function = function;
173        b_ev->data = data;
174       
175        tv.tv_sec = timeout / 1000;
176        tv.tv_usec = ( timeout % 1000 ) * 1000;
177       
178        evtimer_set( &b_ev->evinfo, b_event_passthrough, b_ev );
179        evtimer_add( &b_ev->evinfo, &tv );
180       
181        event_debug( "b_timeout_add( %d, 0x%x, 0x%x ) = %d\n", timeout, function, data, b_ev->id );
182       
183        g_hash_table_insert( id_hash, &b_ev->id, b_ev );
184       
185        return b_ev->id;
186}
187
188void b_event_remove( gint id )
189{
190        struct b_event_data *b_ev = g_hash_table_lookup( id_hash, &id );
191       
192        event_debug( "b_event_remove( %d )\n", id );
193        if( b_ev )
194        {
195                g_hash_table_remove( id_hash, &b_ev->id );
196                if( b_ev->evinfo.ev_fd >= 0 )
197                {
198                        if( b_ev->evinfo.ev_events & EV_READ )
199                                g_hash_table_remove( read_hash, &b_ev->evinfo.ev_fd );
200                        if( b_ev->evinfo.ev_events & EV_WRITE )
201                                g_hash_table_remove( write_hash, &b_ev->evinfo.ev_fd );
202                }
203               
204                event_del( &b_ev->evinfo );
205                g_free( b_ev );
206        }
207        else
208        {
209                event_debug( "Already removed?\n" );
210        }
211}
212
213gboolean b_event_remove_by_data( gpointer data )
214{
215        /* FIXME! */
216        event_debug( "FALSE!\n" );
217        return FALSE;
218}
Note: See TracBrowser for help on using the repository browser.