aboutsummaryrefslogtreecommitdiffstats
path: root/unix.c
blob: 75ffcf9589caae36dfef450de898ff1f58c5c53b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
  /********************************************************************\
  * BitlBee -- An IRC to other IM-networks gateway                     *
  *                                                                    *
  * Copyright 2002-2004 Wilmer van der Gaast and others                *
  \********************************************************************/

/* Main file (Unix specific part)                                       */

/*
  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License with
  the Debian GNU/Linux distribution in /usr/share/common-licenses/GPL;
  if not, write to the Free Software Foundation, Inc., 59 Temple Place,
  Suite 330, Boston, MA  02111-1307  USA
*/

#include "bitlbee.h"
#include "commands.h"
#include "crypting.h"
#include "protocols/nogaim.h"
#include "help.h"
#include "ipc.h"
#include <signal.h>
#include <unistd.h>
#include <sys/time.h>
#include <sys/wait.h>
#include <pwd.h>

global_t global;	/* Against global namespace pollution */

static void sighandler( int signal );

int main( int argc, char *argv[], char **envp )
{
	int i = 0;
	char *old_cwd = NULL;
	struct sigaction sig, old;
	
	log_init();
	CONF_FILE = g_strdup( CONF_FILE_DEF );
	global.conf = conf_load( argc, argv );
	if( global.conf == NULL )
		return( 1 );
	
	b_main_init();
	nogaim_init();
	
	srand( time( NULL ) ^ getpid() );
	global.helpfile = g_strdup( HELP_FILE );
	
	if( global.conf->runmode == RUNMODE_INETD )
	{
		i = bitlbee_inetd_init();
		log_message( LOGLVL_INFO, "Bitlbee %s starting in inetd mode.", BITLBEE_VERSION );

	}
	else if( global.conf->runmode == RUNMODE_DAEMON )
	{
		i = bitlbee_daemon_init();
		log_message( LOGLVL_INFO, "Bitlbee %s starting in daemon mode.", BITLBEE_VERSION );
	}
	else if( global.conf->runmode == RUNMODE_FORKDAEMON )
	{
		/* In case the operator requests a restart, we need this. */
		old_cwd = g_malloc( 256 );
		if( getcwd( old_cwd, 255 ) == NULL )
		{
			log_message( LOGLVL_WARNING, "Could not save current directory: %s", strerror( errno ) );
			g_free( old_cwd );
			old_cwd = NULL;
		}
		
		i = bitlbee_daemon_init();
		log_message( LOGLVL_INFO, "Bitlbee %s starting in forking daemon mode.", BITLBEE_VERSION );
	}
	if( i != 0 )
		return( i );
	
	if( ( global.conf->user && *global.conf->user ) &&
	    ( global.conf->runmode == RUNMODE_DAEMON || 
	      global.conf->runmode == RUNMODE_FORKDAEMON ) &&
	    ( !getuid() || !geteuid() ) )
	{
		struct passwd *pw = NULL;
		pw = getpwnam( global.conf->user );
		if( pw )
		{
			setgid( pw->pw_gid );
			setuid( pw->pw_uid );
		}
	}

	global.storage = storage_init( global.conf->primary_storage, global.conf->migrate_storage );
	if( global.storage == NULL )
	{
		log_message( LOGLVL_ERROR, "Unable to load storage backend '%s'", global.conf->primary_storage );
		return( 1 );
	}
 	
	/* Catch some signals to tell the user what's happening before quitting */
	memset( &sig, 0, sizeof( sig ) );
	sig.sa_handler = sighandler;
	sigaction( SIGCHLD, &sig, &old );
	sigaction( SIGPIPE, &sig, &old );
	sig.sa_flags = SA_RESETHAND;
	sigaction( SIGINT,  &sig, &old );
	sigaction( SIGILL,  &sig, &old );
	sigaction( SIGBUS,  &sig, &old );
	sigaction( SIGFPE,  &sig, &old );
	sigaction( SIGSEGV, &sig, &old );
	sigaction( SIGTERM, &sig, &old );
	sigaction( SIGQUIT, &sig, &old );
	sigaction( SIGXCPU, &sig, &old );
	
	if( !getuid() || !geteuid() )
		log_message( LOGLVL_WARNING, "BitlBee is running with root privileges. Why?" );
	if( help_init( &(global.help), global.helpfile ) == NULL )
		log_message( LOGLVL_WARNING, "Error opening helpfile %s.", HELP_FILE );
	
	b_main_run();
	
	if( global.restart )
	{
		char *fn = ipc_master_save_state();
		char **args;
		int n, i;
		
		chdir( old_cwd );
		
		n = 0;
		args = g_new0( char *, argc + 3 );
		args[n++] = argv[0];
		if( fn )
		{
			args[n++] = "-R";
			args[n++] = fn;
		}
		for( i = 1; argv[i] && i < argc; i ++ )
		{
			if( strcmp( argv[i], "-R" ) == 0 )
				i += 2;
			
			args[n++] = argv[i];
		}
		
		close( global.listen_socket );
		
		execve( args[0], args, envp );
	}
	
	return( 0 );
}

static void sighandler( int signal )
{
	/* FIXME: Calling log_message() here is not a very good idea! */
	
	if( signal == SIGTERM )
	{
		static int first = 1;
		
		if( first )
		{
			/* We don't know what we were doing when this signal came in. It's not safe to touch
			   the user data now (not to mention writing them to disk), so add a timer. */
			
			log_message( LOGLVL_ERROR, "SIGTERM received, cleaning up process." );
			b_timeout_add( 1, (b_event_handler) bitlbee_shutdown, NULL );
			
			first = 0;
		}
		else
		{
			/* Well, actually, for now we'll never need this part because this signal handler
			   will never be called more than once in a session for a non-SIGPIPE signal...
			   But just in case we decide to change that: */
			
			log_message( LOGLVL_ERROR, "SIGTERM received twice, so long for a clean shutdown." );
			raise( signal );
		}
	}
	else if( signal == SIGCHLD )
	{
		pid_t pid;
		int st;
		
		while( ( pid = waitpid( 0, &st, WNOHANG ) ) > 0 )
		{
			if( WIFSIGNALED( st ) )
				log_message( LOGLVL_INFO, "Client %d terminated normally. (status = %d)", pid, WEXITSTATUS( st ) );
			else if( WIFEXITED( st ) )
				log_message( LOGLVL_INFO, "Client %d killed by signal %d.", pid, WTERMSIG( st ) );
		}
	}
	else if( signal != SIGPIPE )
	{
		log_message( LOGLVL_ERROR, "Fatal signal received: %d. That's probably a bug.", signal );
		raise( signal );
	}
}

double gettime()
{
	struct timeval time[1];

	gettimeofday( time, 0 );
	return( (double) time->tv_sec + (double) time->tv_usec / 1000000 );
}
pan class="p">("message", "groupchat", jc->name, node); if (!jabber_write_packet(ic, node)) { xt_free_node(node); return 0; } xt_free_node(node); return 1; } int jabber_chat_topic(struct groupchat *c, char *topic) { struct im_connection *ic = c->ic; struct jabber_chat *jc = c->data; struct xt_node *node; node = xt_new_node("subject", topic, NULL); node = jabber_make_packet("message", "groupchat", jc->name, node); if (!jabber_write_packet(ic, node)) { xt_free_node(node); return 0; } xt_free_node(node); return 1; } int jabber_chat_leave(struct groupchat *c, const char *reason) { struct im_connection *ic = c->ic; struct jabber_chat *jc = c->data; struct xt_node *node; node = xt_new_node("x", NULL, NULL); xt_add_attr(node, "xmlns", XMLNS_MUC); node = jabber_make_packet("presence", "unavailable", jc->my_full_jid, node); if (!jabber_write_packet(ic, node)) { xt_free_node(node); return 0; } xt_free_node(node); return 1; } void jabber_chat_invite(struct groupchat *c, char *who, char *message) { struct xt_node *node; struct im_connection *ic = c->ic; struct jabber_chat *jc = c->data; node = xt_new_node("reason", message, NULL); node = xt_new_node("invite", NULL, node); xt_add_attr(node, "to", who); node = xt_new_node("x", NULL, node); xt_add_attr(node, "xmlns", XMLNS_MUC_USER); node = jabber_make_packet("message", NULL, jc->name, node); jabber_write_packet(ic, node); xt_free_node(node); } /* Not really the same syntax as the normal pkt_ functions, but this isn't called by the xmltree parser directly and this way I can add some extra parameters so we won't have to repeat too many things done by the caller already. */ void jabber_chat_pkt_presence(struct im_connection *ic, struct jabber_buddy *bud, struct xt_node *node) { struct groupchat *chat; struct xt_node *c; char *type = xt_find_attr(node, "type"); struct jabber_data *jd = ic->proto_data; struct jabber_chat *jc; char *s; if ((chat = jabber_chat_by_jid(ic, bud->bare_jid)) == NULL) { /* How could this happen?? We could do kill( self, 11 ) now or just wait for the OS to do it. :-) */ return; } jc = chat->data; if (type == NULL && !(bud->flags & JBFLAG_IS_CHATROOM)) { bud->flags |= JBFLAG_IS_CHATROOM; /* If this one wasn't set yet, this buddy just joined the chat. Slightly hackish way of finding out eh? ;-) */ /* This is pretty messy... Here it sets ext_jid to the real JID of the participant. Works for non-anonymized channels. Might break if someone joins a chat twice, though. */ for (c = node->children; (c = xt_find_node(c, "x")); c = c->next) { if ((s = xt_find_attr(c, "xmlns")) && (strcmp(s, XMLNS_MUC_USER) == 0)) { struct xt_node *item; item = xt_find_node(c->children, "item"); if ((s = xt_find_attr(item, "jid"))) { /* Yay, found what we need. :-) */ bud->ext_jid = jabber_normalize(s); break; } } } /* Make up some other handle, if necessary. */ if (bud->ext_jid == NULL) { if (bud == jc->me) { bud->ext_jid = g_strdup(jd->me); } else { int i; /* Don't want the nick to be at the end, so let's think of some slightly different notation to use for anonymous groupchat participants in BitlBee. */ bud->ext_jid = g_strdup_printf("%s=%s", bud->resource, bud->bare_jid); /* And strip any unwanted characters. */ for (i = 0; bud->resource[i]; i++) { if (bud->ext_jid[i] == '=' || bud->ext_jid[i] == '@') { bud->ext_jid[i] = '_'; } } /* Some program-specific restrictions. */ imcb_clean_handle(ic, bud->ext_jid); } bud->flags |= JBFLAG_IS_ANONYMOUS; } if (bud != jc->me && bud->flags & JBFLAG_IS_ANONYMOUS) { /* If JIDs are anonymized, add them to the local list for the duration of this chat. */ imcb_add_buddy(ic, bud->ext_jid, NULL); imcb_buddy_nick_hint(ic, bud->ext_jid, bud->resource); } if (bud == jc->me && jc->invite != NULL) { char *msg = g_strdup_printf("Please join me in room %s", jc->name); jabber_chat_invite(chat, jc->invite, msg); g_free(jc->invite); g_free(msg); jc->invite = NULL; } s = strchr(bud->ext_jid, '/'); if (s) { *s = 0; /* Should NEVER be NULL, but who knows... */ } imcb_chat_add_buddy(chat, bud->ext_jid); if (s) { *s = '/'; } } else if (type) { /* type can only be NULL or "unavailable" in this function */ if ((bud->flags & JBFLAG_IS_CHATROOM) && bud->ext_jid) { s = strchr(bud->ext_jid, '/'); if (s) { *s = 0; } imcb_chat_remove_buddy(chat, bud->ext_jid, NULL); if (bud != jc->me && bud->flags & JBFLAG_IS_ANONYMOUS) { imcb_remove_buddy(ic, bud->ext_jid, NULL); } if (s) { *s = '/'; } } if (bud == jc->me) { jabber_chat_free(chat); } } } void jabber_chat_pkt_message(struct im_connection *ic, struct jabber_buddy *bud, struct xt_node *node) { struct xt_node *subject = xt_find_node(node->children, "subject"); struct xt_node *body = xt_find_node(node->children, "body"); struct groupchat *chat = NULL; struct jabber_chat *jc = NULL; char *from = NULL; char *nick = NULL; char *final_from = NULL; char *bare_jid = NULL; from = (bud) ? bud->full_jid : xt_find_attr(node, "from"); if (from) { nick = strchr(from, '/'); if (nick) { *nick = 0; } chat = jabber_chat_by_jid(ic, from); if (nick) { *nick = '/'; nick++; } } jc = (chat) ? chat->data : NULL; if (!bud) { struct xt_node *c; char *s; /* Try some clever stuff to find out the real JID here */ c = xt_find_node_by_attr(node->children, "delay", "xmlns", XMLNS_DELAY); if (c && ((s = xt_find_attr(c, "from")) || (s = xt_find_attr(c, "from_jid")))) { /* This won't be useful if it's the MUC JID */ if (!(jc && jabber_compare_jid(s, jc->name))) { /* Hopefully this one makes more sense! */ bud = jabber_buddy_by_jid(ic, s, GET_BUDDY_FIRST | GET_BUDDY_CREAT); } } } if (subject && chat) { char *subject_text = subject->text_len > 0 ? subject->text : NULL; if (g_strcmp0(chat->topic, subject_text) != 0) { bare_jid = (bud) ? jabber_get_bare_jid(bud->ext_jid) : NULL; imcb_chat_topic(chat, bare_jid, subject_text, jabber_get_timestamp(node)); g_free(bare_jid); } } if (body == NULL || body->text_len == 0) { /* Meh. Empty messages aren't very interesting, no matter how much some servers love to send them. */ return; } if (chat == NULL) { if (nick == NULL) { imcb_log(ic, "System message from unknown groupchat %s: %s", from, body->text); } else { imcb_log(ic, "Groupchat message from unknown JID %s: %s", from, body->text); } return; } else if (chat != NULL && bud == NULL && nick == NULL) { imcb_chat_log(chat, "From conference server: %s", body->text); return; } else if (jc && jc->flags & JCFLAG_MESSAGE_SENT && bud == jc->me) { /* exclude self-messages since they would get filtered out * but not the ones in the backlog */ return; } if (bud && jc && bud != jc->me) { bare_jid = jabber_get_bare_jid(bud->ext_jid ? bud->ext_jid : bud->full_jid); final_from = bare_jid; } else { final_from = nick; } imcb_chat_msg(chat, final_from, body->text, 0, jabber_get_timestamp(node)); g_free(bare_jid); }