connections.c 29.7 KB
Newer Older
Linus Tolke Y's avatar
Linus Tolke Y committed
1
/*
2
 * Copyright (C) 1991-2003  Lysator Academic Computer Association.
Linus Tolke Y's avatar
Linus Tolke Y committed
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
 *
 * This file is part of the LysKOM server.
 * 
 * LysKOM 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 1, or (at your option) 
 * any later version.
 * 
 * LysKOM 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
 * along with LysKOM; see the file COPYING.  If not, write to
 * Lysator, c/o ISY, Linkoping University, S-581 83 Linkoping, SWEDEN,
 * or the Free Software Foundation, Inc., 675 Mass Ave, Cambridge, 
 * MA 02139, USA.
 *
Per Cederqvist's avatar
Per Cederqvist committed
22
 * Please report bugs at http://bugzilla.lysator.liu.se/. 
Linus Tolke Y's avatar
Linus Tolke Y committed
23
 */
Per Cederqvist's avatar
Per Cederqvist committed
24
25
26
27
28
29
30
31
/*
 * connections.c
 *
 * Denna fil inneh}ller niv}n ovanf|r isc.
 *
 * Created by Willf|r 31/3-90. Mostly written by ceder.
 */

David Byers's avatar
David Byers committed
32
33
34
35
36

#ifdef HAVE_CONFIG_H
#  include <config.h>
#endif

Per Cederqvist's avatar
Per Cederqvist committed
37
#include <errno.h>
Per Cederqvist's avatar
Per Cederqvist committed
38
#include <stdio.h>
Per Cederqvist's avatar
Per Cederqvist committed
39
#include <setjmp.h>
Per Cederqvist's avatar
Per Cederqvist committed
40
41
42
#ifdef HAVE_STRING_H
#  include <string.h>
#endif
Per Cederqvist's avatar
Per Cederqvist committed
43
#include <sys/types.h>
Per Cederqvist's avatar
Per Cederqvist committed
44
#include <sys/stat.h>
45
#include "timewrap.h"
46
#include <sys/socket.h>
47
#include <signal.h>
48
#include <assert.h>
49
#include <netdb.h>
Per Cederqvist's avatar
Per Cederqvist committed
50
#include <stdlib.h>
Per Cederqvist's avatar
Per Cederqvist committed
51

52
#include "adns.h"
53
54
55
#include "oop.h"

#include "unused.h"
56
#include "ldifftime.h"
Per Cederqvist's avatar
Per Cederqvist committed
57
#include "misc-types.h"
58
#include "s-string.h"
Per Cederqvist's avatar
Per Cederqvist committed
59
#include "kom-types.h"
60
#include "kom-memory.h"
Per Cederqvist's avatar
Per Cederqvist committed
61
#include "debug.h"
62
#include "isc-interface.h"
Per Cederqvist's avatar
Per Cederqvist committed
63
#include "com.h"
64
#include "async.h"
Per Cederqvist's avatar
Per Cederqvist committed
65
#include "connections.h"
Per Cederqvist's avatar
Per Cederqvist committed
66
#include "internal-connections.h"
Per Cederqvist's avatar
Per Cederqvist committed
67
#include "prot-a-parse-arg.h"
Per Cederqvist's avatar
Per Cederqvist committed
68
69
70
71
72
#include "log.h"
#include "lyskomd.h"
#include "services.h"
#include "isc-parse.h"
#include "prot-a.h"
73
#include "prot-a-parse.h"
Per Cederqvist's avatar
Per Cederqvist committed
74
75
76
77
#include "server/smalloc.h"
#include "end-of-atomic.h"
#include "send-async.h"
#include "cache.h"
78
#include "rfc931.h"
79
#include "param.h"
David Byers's avatar
David Byers committed
80
#include "kom-config.h"
81
#include "kom-errno.h"
Per Cederqvist's avatar
Per Cederqvist committed
82
#include "sigflags.h"
Per Cederqvist's avatar
Per Cederqvist committed
83
#include "server-time.h"
84
#include "aux-items.h"
85
#include "eintr.h"
86
#include "text-garb.h"
87
#include "timeval-util.h"
88
#include "stats.h"
89
#include "string-malloc.h"
90
#include "manipulate.h"
Per Cederqvist's avatar
Per Cederqvist committed
91

92
93
oop_source_sys * kom_server_oop_src = NULL;
struct isc_mcb * kom_server_mcb    = NULL;
Per Cederqvist's avatar
Per Cederqvist committed
94
95
Connection     * active_connection = NULL;

Per Cederqvist's avatar
Per Cederqvist committed
96
97
98
/*
 * This is set TRUE when the server should be closed. It is checked
 * each time around the main loop. It is set if someone with enough
99
 * privileges issues a `shutdown', or of lyskomd receives a SIGTERM.
Per Cederqvist's avatar
Per Cederqvist committed
100
101
 * This not an abort: all data is saved before we exit.
 */
102
Bool go_and_die = FALSE;
Per Cederqvist's avatar
Per Cederqvist committed
103
104

/*
105
106
107
108
109
110
111
112
113
 * The number of times that the session penalties has been averaged.
 * Beware: this number will wrap around.
 */
static unsigned int penalty_generation = 0;

/*
 * These state variables are used to find out if we are busy or not.
 * When a packet arrives, data_available_callback() will set work_done
 * to TRUE and is_idle to FALSE.
Per Cederqvist's avatar
Per Cederqvist committed
114
 */
115
116
static Bool work_done = FALSE;
static Bool is_idle = FALSE;
Per Cederqvist's avatar
Per Cederqvist committed
117
118
119
120
121
122
123
124

jmp_buf 	 parse_env;


const Fnc_descriptor fnc_defs[]={
#include "fnc-def-init.incl"
};

125
const int num_fnc_defs = sizeof (fnc_defs) / sizeof (Fnc_descriptor);
126

Per Cederqvist's avatar
Per Cederqvist committed
127
unsigned long service_statistics[sizeof (fnc_defs) / sizeof (Fnc_descriptor)];
Per Cederqvist's avatar
Per Cederqvist committed
128
129
130

BUGDECL;

131
132
133
static oop_call_fd data_available_callback;
static oop_call_time check_kill_flg;
static oop_call_time check_idle_callback;
134
135
136
static isc_write_error_cb write_err_cb;
static isc_stale_output_cb stale_cb;
static isc_stale_output_cb idle_cb;
137
138
139
140

static Connection *queue_first = NULL;
static Connection *queue_last = NULL;

141
142
143
static void busy(void);


144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
static void
queue_add(Connection *c)
{
    assert(c->on_queue == FALSE);
    c->on_queue = TRUE;

    c->queue_prev = queue_last;
    c->queue_next = NULL;

    if (queue_first == NULL)
	queue_first = c;
    else
	queue_last->queue_next = c;

    queue_last = c;
159
    update_stat(STAT_RUN_QUEUE, 1);
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
}

static void
queue_remove(Connection *c)
{
    assert(c->on_queue == TRUE);
    c->on_queue = FALSE;

    if (c->queue_next != NULL)
	c->queue_next->queue_prev = c->queue_prev;
    else
	queue_last = c->queue_prev;
    
    if (c->queue_prev != NULL)
	c->queue_prev->queue_next = c->queue_next;
    else
176
	queue_first = c->queue_next;
Per Cederqvist's avatar
Per Cederqvist committed
177

178
179
    c->queue_prev = NULL;
    c->queue_next = NULL;
180
    update_stat(STAT_RUN_QUEUE, -1);
181
182
}

Per Cederqvist's avatar
Per Cederqvist committed
183
void
184
185
set_time(void)
{
186
187
    struct timeval last_time;
    static int limiter = 0;
188
189

    last_time = current_time;
190
191
192
193
194
195
196
197
198
199
200
    if (gettimeofday(&current_time, NULL) < 0)
    {
	if (limiter < 50)
	{
	    kom_log("WARNING: gettimeofday failed: %s\n", strerror(errno));
	    if (++limiter == 50)
		kom_log("WARNING: will not log the above message again.\n");
	}
    }

    if (timeval_less(current_time, last_time))
201
    {
202
203
	kom_log("WARNING: Time moved backward at least %g seconds.",
		timeval_diff_d(last_time, current_time));
204
205
206
	/* FIXME (bug 62): Should we take more decisive action here? */
    }
}
Per Cederqvist's avatar
Per Cederqvist committed
207

Per Cederqvist's avatar
Per Cederqvist committed
208
static void
Per Cederqvist's avatar
Per Cederqvist committed
209
210
211
logout_client(Connection *cp)
{
    Connection *real_active_connection;
212
    int ret;
Per Cederqvist's avatar
Per Cederqvist committed
213
214
215

    if ( active_connection != NULL )
    {
David Byers's avatar
David Byers committed
216
	kom_log("BUGCHK: logout_client(%ld): connection %ld is active.\n",
Per Cederqvist's avatar
Per Cederqvist committed
217
218
219
220
221
	    cp->session_no, active_connection->session_no);
    }
    
    if ( cp->pers_no != 0 )
    {
222
223
224
225
226
227
228
229
230
231
232
233
	int ctr = 0;

	if (active_connection != NULL)
	{
	    kom_log("WNG: logout_client(): active_connection != NULL\n");
	    if (ctr < 100)
		ctr++;
	    else
		kom_log("WNG: won't log the above message more\n");
	}

	real_active_connection = active_connection;
Per Cederqvist's avatar
Per Cederqvist committed
234
235
236
237
	active_connection = cp;
	logout();
	active_connection = real_active_connection;
    }
238
239
    else
    {
240
#if 0
241
	/* FIXME (bug 908): send a new async here instead.  This causes the
242
243
	   elisp client to say that a secret (or unknown) person has
	   left the system.  */
244
        async_logout( 0, cp->session_no );
245
#endif
246
    }
Per Cederqvist's avatar
Per Cederqvist committed
247
248
249
250
251
252
253
254
255
256
257
258
259
260

    switch(cp->protocol)
    {
    case 0:			/* Hasn't yet allocated any protocol. */
	break;
	
    case 'A':
	prot_a_destruct(cp);
	break;
	
    default:
	restart_kom("logout_client(): Bad protocol.\n");
    }

261
262
    ret = isc_destroy(kom_server_mcb, cp->isc_session);
    if (ret < 0)
David Byers's avatar
David Byers committed
263
	kom_log("logout_client(): isc_destroyed returned %d\n", ret);
264
    cp->isc_session = NULL;
265
266
    if (cp->on_queue)
	queue_remove(cp);
Per Cederqvist's avatar
Per Cederqvist committed
267
268

    kill_client(cp);		/* Free the Connection */
269
    update_stat(STAT_CLIENTS, -1);
Per Cederqvist's avatar
Per Cederqvist committed
270
271
}

Per Cederqvist's avatar
Per Cederqvist committed
272
273
274
275
276
277
278
279
280
281
282
283
284
285
/*
 * This function is part of the shutdown tidy-up sequence.
 */
void
logout_all_clients(void)
{
    Session_no sess = 0;
    Connection *conn;
    
    while ( (sess = traverse_connections (sess)) != 0)
    {
	conn = get_conn_by_number (sess);

	if ( conn == NULL )
286
	    restart_kom("logout_all_clients(): cant get session %ld.\n",
Per Cederqvist's avatar
Per Cederqvist committed
287
288
289
290
291
292
			sess);
	else
	    logout_client (conn);
    }

    if ( traverse_connections (0) != 0)
293
	restart_kom("logout_all_clients(): traverse_connections(0) == %ld.\n",
Per Cederqvist's avatar
Per Cederqvist committed
294
295
296
297
		    traverse_connections(0));
}

    
Per Cederqvist's avatar
Per Cederqvist committed
298
299
300
301
302
303
304
305
    

    
/*
 * Call a function in services.c. A pointer to the result is returned.
 * The pointer points to static data which is overwritten on each call.
 */
static Success
306
307
call_function(Connection *client,
	      union result_holder *res)
Per Cederqvist's avatar
Per Cederqvist committed
308
309
310
311
312
{
    Success	status=FAILURE;	/* OK if the call was successful. */

    if ( active_connection != NULL )
    {
313
	kom_log("call_function(%ld): active_connection = %ld\n",
Per Cederqvist's avatar
Per Cederqvist committed
314
315
	    client->session_no, active_connection->session_no);
    }
316
317
318

    if (client->function == illegal_fnc) 
    {
319
        err_stat = 0;
320
321
322
323
	kom_errno = KOM_NOT_IMPL;
	return FAILURE;
    }

Per Cederqvist's avatar
Per Cederqvist committed
324
325
    active_connection = client;

326
    service_statistics[client->function_index]++;
Per Cederqvist's avatar
Per Cederqvist committed
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349

#include "call-switch.incl"

    active_connection = NULL;

    return status;
}


static void
parse_packet(Connection *client)
{
    if ( client->protocol == '\0' ) /* Not known yet. */
    {
	client->protocol = parse_char(client);
	switch(client->protocol)
	{
	case 'A':
	    prot_a_init(client);
	    break;

	default:
	    client->protocol = '\0';
350
351
	    isc_puts("%%LysKOM unsupported protocol.\n", client->isc_session);
	    isc_flush(client->isc_session);
Per Cederqvist's avatar
Per Cederqvist committed
352
	    BUG(("%%%%Unsupported protocol.\n"));
353
	    longjmp(parse_env, KOM_LOGOUT);
Per Cederqvist's avatar
Per Cederqvist committed
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
	}
    }

    switch(client->protocol)
    {
    case 'A':
	prot_a_parse_packet(client);
	break;

    default:
	restart_kom("parse_packet(): Bad protocol.\n");
	break;
    }
}

/*
 * Free all parsed areas which are no longer needed. Re-initialize all
 * parse_pos fields so that the parse will expect a new function.
 *
 * This function is called
 *	when a parse error occurs
 *	when a parse is complete and the function has executed.
 */
static void
free_parsed(Connection *client)
{
    s_clear(&client->c_string0);
    s_clear(&client->c_string1);
    client->string0 = EMPTY_STRING; /* So that no one frees it. */
David Byers's avatar
David Byers committed
383
384
385
    sfree(client->misc_info_list.misc);
    client->misc_info_list.misc = 0;
    client->misc_info_list.no_of_misc = 0;
386
    s_clear(&client->aux_item.data);
387
    s_clear(&client->dummy_aux_item.data);
388
389
390
    sfree(client->read_range_list.ranges);
    client->read_range_list.ranges = NULL;
    client->read_range_list.length = 0;
Per Cederqvist's avatar
Per Cederqvist committed
391
392
    client->parse_pos = 0;
    client->fnc_parse_pos = 0;
393
    client->array_parse_index = 0;
David Byers's avatar
David Byers committed
394
    client->array_parse_parsed_length = 0;
Per Cederqvist's avatar
Per Cederqvist committed
395
396
397
    client->array_parse_pos = 0;
    client->struct_parse_pos = 0;
    client->string_parse_pos = 0;
David Byers's avatar
David Byers committed
398
399
400
    client->hunt_parse_pos = 0;
    client->array_hunt_num = 0;
    client->array_hunt_depth = 0;
401
402
403
    sfree(client->num_list.data);
    client->num_list.data = NULL;
    client->num_list.length = 0;
404
405
    free_aux_item_list(&client->aux_item_list);
    client->info.highest_aux_no = 0;
Per Cederqvist's avatar
Per Cederqvist committed
406
407
408
409
410
411
412
}

/*
 * Send a reply to a call.
 */
static void
reply(Connection *client,
413
414
      Success status,
      union result_holder *result)
Per Cederqvist's avatar
Per Cederqvist committed
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
{
    switch(client->protocol)
    {
    case 'A':
	prot_a_reply(client, status, result);
	break;

    default:
	restart_kom("reply(): Bad protocol.\n");
	break;
    }
}


/*
 * Try to parse enough data from client->unparsed to call a function.
431
432
 * If more data is needed set client->more_to_parse to FALSE.  Returns
 * TRUE if anything was (or might have been) written to the client.
Per Cederqvist's avatar
Per Cederqvist committed
433
 */
434
static Bool
Per Cederqvist's avatar
Per Cederqvist committed
435
436
437
parse_unparsed(Connection *client)
{
    Success       status;
438
    union result_holder result;
Per Cederqvist's avatar
Per Cederqvist committed
439
440
441
442
443
444
        
    switch ( setjmp(parse_env) )
    {
    case 0 :
	/* Parse message. If message is complete call function and reply. */
	parse_packet(client);
445
446
	if (client->blocked_by_dns)
	    return TRUE;
447
	update_stat(STAT_REQUESTS, 1);
Per Cederqvist's avatar
Per Cederqvist committed
448
	status = call_function(client, &result);
449
	update_stat(STAT_REQUESTS, -1);
Per Cederqvist's avatar
Per Cederqvist committed
450
	reply(client, status, &result);
451
	client->penalty += param.penalty_per_call;
Per Cederqvist's avatar
Per Cederqvist committed
452
	free_parsed(client);
453
	end_of_atomic();
454
	return TRUE;
Per Cederqvist's avatar
Per Cederqvist committed
455

456
    case KOM_PROTOCOL_ERR:
Per Cederqvist's avatar
Per Cederqvist committed
457
458
	s_clear(&client->string0);
	free_parsed(client);
459
	isc_puts("%% LysKOM protocol error.\n", client->isc_session);
Per Cederqvist's avatar
Per Cederqvist committed
460
	BUG(("%%%% Protocol error.\n"));
461
	client->penalty += param.max_penalty;
Per Cederqvist's avatar
Per Cederqvist committed
462
463
464
	s_clear(&client->unparsed);
	client->first_to_parse = 0;
	client->more_to_parse = FALSE;
465
	end_of_atomic();
466
	return TRUE;
Per Cederqvist's avatar
Per Cederqvist committed
467

468
    case KOM_MSG_INCOMPLETE:
Per Cederqvist's avatar
Per Cederqvist committed
469
	client->more_to_parse = FALSE;
470
	return FALSE;
Per Cederqvist's avatar
Per Cederqvist committed
471

472
    case KOM_LOGOUT:
473
	add_to_kill_list(client);
474
	client->more_to_parse = FALSE;
475
476
477
	return TRUE;
    default:
	restart_kom("Bad longjmp return value.\n");
Per Cederqvist's avatar
Per Cederqvist committed
478
    }
479
    /*NOTREACHED*/
Per Cederqvist's avatar
Per Cederqvist committed
480
481
482
483
484
}


/* Return 1 if the named file exists, 0 otherwise */
static int
Per Cederqvist's avatar
Per Cederqvist committed
485
fexists(const char *filename)
Per Cederqvist's avatar
Per Cederqvist committed
486
487
488
489
490
491
492
493
494
495
496
497
498
499
{
    struct stat buf;
    int code;
  
    code = !stat(filename, &buf);
    errno = 0;

    return code;
}


void
dump_statistics(void)
{
500
    static struct timeval last_dump = {0, 0};
Per Cederqvist's avatar
Per Cederqvist committed
501
502
503
    int i;
    FILE *fp;

504
    if ((fp = i_fopen(param.statistic_name, "a")) == NULL)
Per Cederqvist's avatar
Per Cederqvist committed
505
    {
506
507
	kom_log("dump_statistics(): can't open file %s\n",
		param.statistic_name);
Per Cederqvist's avatar
Per Cederqvist committed
508
509
510
	return;
    }

511
    if (timeval_zero(last_dump))
Per Cederqvist's avatar
Per Cederqvist committed
512
513
    {
	fprintf(fp, "RESTART\n");
Per Cederqvist's avatar
Per Cederqvist committed
514
	last_dump = current_time;
Per Cederqvist's avatar
Per Cederqvist committed
515
516
    }
    
517
518
    fprintf(fp, "TIME: %s", ctime(&current_time.tv_sec));
    fprintf(fp, "SECONDS: %ld\n", timeval_diff_sec(current_time, last_dump));
Per Cederqvist's avatar
Per Cederqvist committed
519
    fprintf(fp, "STATISTICS:");
520
521
522
523
524

    /* The last entry corresponds to the dummy entry that is used to
       skip arguments to unimplemented requests.  Skip that, since it
       contains no useful statistics.  */
    for (i = 0; i < num_fnc_defs - 1; i++)
Per Cederqvist's avatar
Per Cederqvist committed
525
    {
526
	fprintf(fp, " %d:%lu", fnc_defs[i].function, service_statistics[i]);
Per Cederqvist's avatar
Per Cederqvist committed
527
528
529
530
	service_statistics[i]=0;
    }

    fprintf(fp, "\n");
531
    i_fclose(fp);
Per Cederqvist's avatar
Per Cederqvist committed
532

Per Cederqvist's avatar
Per Cederqvist committed
533
    last_dump = current_time;
Per Cederqvist's avatar
Per Cederqvist committed
534
535
}

536
537
/* List of connections to kill. */

Per Cederqvist's avatar
Per Cederqvist committed
538
539
static Session_no *kill_list = NULL;
static int kill_list_size = 0;
540
static int kill_pending = 0;
541
542
543
544
545

/* Schedule this client for termination. */
void
add_to_kill_list(Connection *conn)
{
546
    oop_source *source;
Per Cederqvist's avatar
Per Cederqvist committed
547
548
    int i;

549
    switch (conn->kill_status)
550
    {
551
552
553
    case ks_pending:
	/* A kill is already pending.  Do nothing--but check that the
	   the client really is present on the kill_list.  */
554
555
556
	for (i = 0; i < kill_list_size; i++)
	    if (kill_list[i] == conn->session_no)
		return;
557

558
	restart_kom("add_to_kill_list(): kill_pending set but not on list.\n");
559
560
561
562
563
564
565
566
567
568
	return;

    case ks_dying:
	/* Don't add this client to the kill list while it is being
	   killed.  */
	return;

    case ks_none:
	/* The normal case.  Do all the work below. */
	break;
569
570
    }

571
    /* Check that the client isn't already present on the kill_list. */
Per Cederqvist's avatar
Per Cederqvist committed
572
    for (i = 0; i < kill_list_size; i++)
573
	if (kill_list[i] == conn->session_no)
574
	    restart_kom("add_to_kill_list(): on list but not kill_pending.\n");
Per Cederqvist's avatar
Per Cederqvist committed
575

576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
    if (kill_list == NULL)
    {
	if (kill_list_size != 0)
	    restart_kom("add_to_kill_list(): size = %d\n", kill_list_size);

	kill_list_size = 1;
	kill_list = smalloc(sizeof(Session_no));
    }
    else
    {
	kill_list_size++;
	kill_list = srealloc(kill_list, kill_list_size * sizeof(Session_no));
    }

    kill_list[kill_list_size-1] = conn->session_no;
591
    conn->kill_status = ks_pending;
592
593
594

    if (!kill_pending)
    {
595
	source = isc_getoopsource(conn->isc_session);
596
597
598
	source->on_time(source, OOP_TIME_NOW, check_kill_flg, NULL);
	kill_pending = 1;
    }
599
}
Per Cederqvist's avatar
Per Cederqvist committed
600

601

602
static void
603
604
605
606
607
608
dump_connections(void)
{
    Session_no s;
    Connection *conn;
    FILE *fp;

609
    if ((fp = i_fopen(param.connection_status_file_tmp, "w")) == NULL)
610
    {
611
612
	kom_log("dump_connections(): can't open file %s: %s\n",
		param.connection_status_file_tmp, strerror(errno));
613
614
615
616
617
618
	return;
    }

    for (s = 0; (s = traverse_connections(s)) != 0;)
    {
	conn = get_conn_by_number(s);
619
620
	fprintf(fp, "%d %lu %d %s\n", conn->isc_session->fd, conn->session_no,
		handshake_ok(conn, 0),
621
622
623
		conn->peer);
    }

624
625
626
627
628
629
    if (fflush(fp) != 0)
	kom_log("dump_connections(): fflush() says an error has occured.\n");
	
    if (ferror(fp))
	kom_log("dump_connections(): ferror() says an error has occured.\n");

630
    if (i_fclose(fp) < 0)
631
632
633
634
635
    {
	kom_log("dump_connections(): fclose failed: %s (ignored)\n",
		strerror(errno));
    }

636
637
638
    errno = 0;
    if (i_rename(param.connection_status_file_tmp,
		 param.connection_status_file) < 0)
639
    {
640
641
642
643
	kom_log("dump_connections(): can't rename %s to %s: %s\n",
		param.connection_status_file_tmp,
		param.connection_status_file,
		strerror(errno));
644
645
646
647
    }
}


Per Cederqvist's avatar
Per Cederqvist committed
648
649
650
/*
 * check_kill_flg must NEVER be called inside an atomic call!
 */
651
652
653
654
static void *
check_kill_flg(oop_source *UNUSED(source),
	       struct timeval UNUSED(tv),
	       void *UNUSED(user))
Per Cederqvist's avatar
Per Cederqvist committed
655
656
{
    Connection *conn;
657
    Bool changed = FALSE;
Per Cederqvist's avatar
Per Cederqvist committed
658

659
660
    kill_pending = 0;

Per Cederqvist's avatar
Per Cederqvist committed
661
662
    if ( active_connection != NULL )
    {
663
	restart_kom("check_kill_flg: active_connection == %ld",
Per Cederqvist's avatar
Per Cederqvist committed
664
665
666
		    active_connection->session_no);
    }

667
    while (kill_list_size > 0)
Per Cederqvist's avatar
Per Cederqvist committed
668
    {
669
670
671
	--kill_list_size;
	conn = get_conn_by_number (kill_list[kill_list_size]);
	if (conn == NULL)
Per Cederqvist's avatar
Per Cederqvist committed
672
	{
David Byers's avatar
David Byers committed
673
	    kom_log("check_kill_flg(): Connection %ld doesn't exist.\n",
674
675
676
677
		kill_list[kill_list_size]);
	}
	else
	{
678
679
	    assert(conn->kill_status == ks_pending);
	    conn->kill_status = ks_dying;
680
681
	    logout_client(conn);
	    end_of_atomic();
682
	    changed = TRUE;
Per Cederqvist's avatar
Per Cederqvist committed
683
684
	}
    }
685

686
687
688
689
690
    if (kill_list != NULL)
    {
	sfree (kill_list);
	kill_list = NULL;
    }
691
692
693

    if (changed == TRUE)
	dump_connections();
694
695

    return OOP_CONTINUE;
Per Cederqvist's avatar
Per Cederqvist committed
696
697
}

698
699
700
701
static void *
dns_resolution(struct isc_scb *scb,
	       enum isc_resolve_status res,
	       long errcode)
702
703
704
{
    struct timeval after;
    double diff = -1.0;
705
706
    Connection *conn;
    char *hostname = NULL;
707

708
    update_stat(STAT_DNS_QUEUE, -1);
709

710
    conn = scb->udg;
711

712
713
714
715
    conn->dns_done = TRUE;
    if (conn->blocked_by_dns)
    {
	conn->blocked_by_dns = FALSE;
716
717
	if (!conn->on_queue)
	    queue_add(conn);
718
719
    }

720
721
722
723
724
725
726
    if (res == isc_resolve_aborted)
	return OOP_CONTINUE;

    if (gettimeofday(&after, NULL) < 0)
	kom_log("gettimeofday failed: %s\n", strerror(errno));

    diff = timeval_diff_d(after, conn->connect_time);
727

728
729
730
    busy();

    switch (res)
731
    {
732
    case isc_resolve_h_errno:
733
	if (hostname == NULL)
734
735
736
737
738
739
740
741
742
743
	    hostname = s_crea_c_str(scb->remote);

	if (errcode == HOST_NOT_FOUND)
	    kom_log("No hostname found for %s.\n", hostname);
	else if (errcode == TRY_AGAIN)
	    kom_log("Lookup of %s timed out.\n", hostname);
	else if (errcode == NO_RECOVERY)
	    kom_log("Non-recoverable error looking up %s.\n", hostname);
	else if (errcode == NO_ADDRESS)
	    kom_log("Got NO_ADDRESS error looking up %s.\n", hostname);
744
	else
745
746
	    kom_log("Unknown resolver error %ld looking up %s.\n",
		    errcode, hostname);
747
	break;
748

749
    case isc_resolve_adns_error:
750
751
752
	if (hostname == NULL)
	    hostname = s_crea_c_str(scb->remote);

753
754
755
756
757
758
	/* Misconfigurations of localhost are common and harmless.
	   Don't bother logging them, since that makes the test cases fail. */
	if ((errcode != adns_s_inconsistent && errcode != adns_s_nxdomain)
	    || strcmp(hostname, "127.0.0.1") != 0)
	    kom_log("Error looking up %s: %s\n",
		    hostname, adns_strerror(errcode));
759
760
761
762
763
764
765
	break;

    case isc_resolve_aborted:
	abort();

    case isc_resolve_ok:
	break;
766
767
    }

768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
    if (diff > param.dns_log_threshold)
    {
	if (hostname == NULL)
	    hostname = s_crea_c_str(scb->remote);

	if (res == isc_resolve_ok)
	    kom_log("Slow DNS: got %s after %f seconds\n", hostname, diff);
	else
	    kom_log("Slow bad DNS: %s failed after %f seconds\n",
		    hostname, diff);
   }

    if (hostname != NULL)
	string_free(hostname); 

783
784
785
    if (handshake_ok(conn, 0))
	dump_connections();

786
    return OOP_CONTINUE;
787
788
}

789
    
790
static void
791
792
write_err_cb(struct isc_scb *cb_session,
	     int saved_errno)
793
{
794
    Connection  * cp = cb_session->udg;
795
    if (saved_errno != ECONNRESET && saved_errno != EPIPE)
796
797
	kom_log("Failed to write to client %lu from %s: %s\n",
		cp->session_no, cp->peer, strerror(saved_errno));
798
    cp->penalty += param.max_penalty;
799
800
801
802
    add_to_kill_list(cp);
}


803
static void
804
stale_cb(struct isc_scb *cb_session)
805
{
806
    Connection *cp = cb_session->udg;
807
808
809
810
811
812
    kom_log("Client %lu from %s has stalled.  Killing it.\n",
	    cp->session_no, cp->peer);
    cp->penalty += param.max_penalty;
    add_to_kill_list(cp);
}
    
813

814
static void
815
idle_cb(struct isc_scb *cb_session)
816
{
817
    Connection *cp = cb_session->udg;
818
819
820
821
822
823
824
    kom_log("Client %lu from %s has been idle too long.  Killing it.\n",
	    cp->session_no, cp->peer);
    cp->penalty += param.max_penalty;
    add_to_kill_list(cp);
}


Per Cederqvist's avatar
Per Cederqvist committed
825
static void
826
login_request(struct isc_scb *session)
Per Cederqvist's avatar
Per Cederqvist committed
827
828
{
    Connection  * cp;
829
    const char *realuser;
830
831
    char portbuf[1+2+3*sizeof(long)];
    size_t portlen;
832
    char *remote_ip = NULL;
Per Cederqvist's avatar
Per Cederqvist committed
833
834

    /* Supress logins if /etc/nologin exists */
David Byers's avatar
David Byers committed
835
    if (fexists(param.nologin_file))
Per Cederqvist's avatar
Per Cederqvist committed
836
    {
837
838
839
	isc_puts("%% No logins allowed.\n", session);
	isc_flush(session);
 	isc_destroy(kom_server_mcb, session);
Per Cederqvist's avatar
Per Cederqvist committed
840
841
842
	return;
    }

843
844
    /* Create a Connection, and link the Connection and the
       isc_session together. */
845

846
    cp = new_client();
847
848
    cp->isc_session = session;
    session->udg = cp;
849
    update_stat(STAT_CLIENTS, 1);
850

851
852
853
    /* Store the IP address in readable form. */
    s_crea_str(&cp->remote_ip, isc_getipnum(session->raddr, NULL, 0));
    remote_ip = s_crea_c_str(cp->remote_ip);
854

855
    /* Initiate DNS lookup. */
856
    if (param.use_dns && isc_resolve_remote(session, dns_resolution) == 0)
857
	update_stat(STAT_DNS_QUEUE, 1);
858
    else
859
	s_strcpy(&session->remote, cp->remote_ip);
860

861
862
863
864
865
866
    /* Update the status file that contains all connection. */
    sprintf(portbuf, " %d", isc_getportnum(session->raddr));
    portlen = strlen(portbuf);
    cp->peer = smalloc(s_strlen(cp->remote_ip) + portlen + 1);
    strcpy(cp->peer, remote_ip);
    strcpy(cp->peer + s_strlen(cp->remote_ip), portbuf);
867

868
    dump_connections();
869

870
871
872
873
    /* Start with max penalty, so that it doesn't pay to make a lot of
       new connections.  */
    cp->penalty = param.max_penalty;
    cp->penalty_generation = penalty_generation;
874

875
876
877
    cp->schedule.priority = param.default_priority;
    cp->schedule.weight = param.default_weight;

878
879
880
    /* Get the real user name, as returned by the Ident protocol (rfc 931). */
    realuser = get_real_username(session, remote_ip);
    if (realuser == NULL && param.authentication_level == 2)
881
    {
882
883
884
885
886
887
888
889
890
	kom_log("Connection from %s rejected - no IDENT available.\n", 
		remote_ip);

	isc_puts("%% No IDENT server reachable at your site.\n",
		 session);
	isc_flush(session);
	logout_client(cp);
	string_free(remote_ip);
	return;
891
892
    }

893
894
895
896
    if (realuser != NULL)
	s_crea_str(&cp->ident_user, realuser);

    BUG(("\n[Client %lu from %s is connecting]\n", cp->session_no, remote_ip));
897

898
    isc_set_read_callback(session, data_available_callback, write_err_cb,
899
			  stale_cb, idle_cb);
900
    string_free(remote_ip);
Per Cederqvist's avatar
Per Cederqvist committed
901
902
903
}

static void
904
adjust_penalty(Connection *conn)
Per Cederqvist's avatar
Per Cederqvist committed
905
{
906
907
    unsigned int gens = penalty_generation - conn->penalty_generation;
    if (gens > 0)
908
    {
909
910
911
912
913
914
	/* The weight is in the range 1-0xffff (inclusive).
	   The penalty is in the range 0-0x10000 (inclusive).
	   This means that the multiplication can never overflow. */

	if (conn->penalty <= gens)
	    conn->penalty = 0;
915
	else
916
917
918
919
920
921
922
923
924
	{
	    unsigned int tmp = conn->penalty - gens * conn->schedule.weight;
	    if (tmp < conn->penalty)
		conn->penalty = tmp;
	    else
		conn->penalty = 0;
	}

	conn->penalty_generation = penalty_generation;
925
    }
Per Cederqvist's avatar
Per Cederqvist committed
926
927
928
929
}


static void
930
read_from_connection(Connection *conn)
Per Cederqvist's avatar
Per Cederqvist committed
931
{
932
    Bool would_block = FALSE;
933
    Bool need_flush = FALSE;
934
    String_size pre;
Per Cederqvist's avatar
Per Cederqvist committed
935

936
937
    adjust_penalty(conn);

938
    while (!would_block && !go_and_die && conn->penalty < param.max_penalty
939
	   && conn->kill_status == ks_none && !conn->blocked_by_dns)
940
    {
941
	pre = s_strlen(conn->unparsed) - conn->first_to_parse;
942
	while (conn->more_to_parse
943
	       && !go_and_die && conn->penalty < param.max_penalty
944
	       && conn->kill_status == ks_none && !conn->blocked_by_dns)
945
	    need_flush |= parse_unparsed(conn);
946
947
	update_stat(STAT_RECV_QUEUE,
		    s_strlen(conn->unparsed) - conn->first_to_parse - pre);
948

949
	if (go_and_die || conn->penalty >= param.max_penalty
950
	    || conn->kill_status != ks_none || conn->blocked_by_dns)
951
952
953
954
	    break;

	if (!conn->more_to_parse)
	{
955
	    pre = s_strlen(conn->unparsed) - conn->first_to_parse;
956
957
958
959
960
	    switch (isc_read_data(conn->isc_session,
				  &conn->unparsed,
				  &conn->first_to_parse))
	    {
	    case ISC_READ_DATA:
961
962
963
		update_stat(STAT_RECV_QUEUE,
			    s_strlen(conn->unparsed) - conn->first_to_parse
			    - pre);
964
965
966
967
		conn->penalty += param.penalty_per_read;
		conn->more_to_parse = TRUE;
		break;
	    case ISC_READ_ERROR:
968
		if (errno != ECONNRESET && errno != ETIMEDOUT)
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
		    kom_log("Error reading from client: %s\n",
			    strerror(errno));
		/*FALLTHROUGH*/
	    case ISC_READ_LOGOUT:
		add_to_kill_list(conn);
		break;
	    case ISC_READ_WOULDBLOCK:
		would_block = TRUE;
		break;
	    case ISC_READ_NOMEM:
		restart_kom("isc_read_data() reports no memory\n");
	    }

	    if (!conn->more_to_parse)
		break;
	}
    }
986

987
    if (need_flush)
988
989
	isc_flush(conn->isc_session);

990
991
992
993
994
    /* Delete the parsed part of 'unparsed' */
    if (s_trim_left(&conn->unparsed, conn->first_to_parse) != OK)
	restart_kom("parse_unparsed: s_trim_left\n");
    conn->first_to_parse = 0;

995
    if (conn->penalty >= param.max_penalty && !conn->blocked_by_dns)
996
    {
997
998
999
1000
1001
	/* isc_disable() will fail if we have received EPIPE on
	   this socket.  In that case will soon close it, since
	   write_err_cb() has added it to the kill list. */
	if (isc_disable(conn->isc_session) == 0)
	    queue_add(conn);
1002
    }
Per Cederqvist's avatar
Per Cederqvist committed
1003
1004
}

1005
1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
1021

static void
enable_idle_check(void)
{
    oop_source *source = oop_sys_source(kom_server_oop_src);

    source->on_time(source, OOP_TIME_NOW, check_idle_callback, NULL);
}


static void *
check_idle_callback(oop_source *UNUSED(source),
		    struct timeval UNUSED(tv),
		    void *UNUSED(user))
{
    Connection *c;
    Connection *next;
1022
1023
    Connection *head = NULL;
    Connection *tail = NULL;
1024
1025
1026
1027
1028
1029
1030
1031
1032

    set_time();
    is_idle = !work_done;
    work_done = FALSE;

    if (is_idle && queue_first != NULL)
    {
	is_idle = FALSE;
	++penalty_generation;
1033

1034
	for (next = queue_first; next != NULL; )
Per Cederqvist's avatar
Per Cederqvist committed
1035
	{
1036
1037
1038
1039
	    c = next;
	    next = next->queue_next;
	    adjust_penalty(c);
	    if (c->penalty < param.low_penalty)
Per Cederqvist's avatar
Per Cederqvist committed
1040
	    {
1041
		queue_remove(c);
1042
1043
1044
1045
1046
1047

		if (head == NULL)
		    head = c;
		else
		    tail->queue_next = c;
		tail = c;
Per Cederqvist's avatar
Per Cederqvist committed
1048
	    }
1049
1050
1051
	}
    }

1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
1066
    for (next = head; next != NULL; )
    {
	c = next;
	next = next->queue_next;
	c->queue_next = NULL;
	if (isc_enable(c->isc_session) < 0)
	    restart_kom("failed to re-enable session\n");

	/* The call to read_from_connection() might add the
	   connection to the queue.  That's why we remove all entries
	   first, and use a private queue within this function.  */
	if (!go_and_die)
	    read_from_connection(c);
    }

1067
1068
1069
    if (!is_idle)
	enable_idle_check();

1070
1071
    /* Check if a client issued a shutdown command.  */
    return go_and_die ? OOP_HALT : OOP_CONTINUE;
1072
1073
1074
}

static void *
Per Cederqvist's avatar
Per Cederqvist committed
1075
saver_callback(oop_source *source,
1076
1077
1078
	       struct timeval UNUSED(tv),
	       void *user)
{
1079
    struct timeval timeout;
1080
    struct timeval *next_timer = user;
1081

1082
    timeout = end_of_atomic();
1083
1084
    if (setup_timer(next_timer, timeout) < 0)
	kom_log("gettimeofday failed: %s\n", strerror(errno));
Per Cederqvist's avatar
Per Cederqvist committed
1085

1086
1087
1088
    source->on_time(source, *next_timer, saver_callback, user);
    return OOP_CONTINUE;
}
Per Cederqvist's avatar
Per Cederqvist committed
1089
1090


1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
static void
busy(void)
{
    /* Something arrived, so we are busy.  */
    if (is_idle)
    {
	is_idle = FALSE;
	enable_idle_check();
    }
	
    work_done = TRUE;
}

1104
1105
1106
1107
1108
1109
static void *
data_available_callback(oop_source *source,
			int fd,
			oop_event event,
			void *user)
{
1110
    Connection *conn = ((struct isc_scb*)user)->udg;
1111
1112
1113

    assert(event == OOP_READ);
    assert(conn->isc_session->fd == fd);