connections.c 30.6 KB
Newer Older
Linus Tolke Y's avatar
Linus Tolke Y committed
1
/*
2
 * Copyright (C) 1991-2005  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 "conf-file.h"
80
#include "param.h"
David Byers's avatar
David Byers committed
81
#include "kom-config.h"
82
#include "kom-errno.h"
Per Cederqvist's avatar
Per Cederqvist committed
83
#include "sigflags.h"
Per Cederqvist's avatar
Per Cederqvist committed
84
#include "server-time.h"
85
#include "aux-items.h"
86
#include "eintr.h"
87
#include "text-garb.h"
88
#include "timeval-util.h"
89
#include "stats.h"
90
#include "string-malloc.h"
91
#include "manipulate.h"
Per Cederqvist's avatar
Per Cederqvist committed
92

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

Per Cederqvist's avatar
Per Cederqvist committed
97 98 99
/*
 * 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
100
 * privileges issues a `shutdown', or of lyskomd receives a SIGTERM.
Per Cederqvist's avatar
Per Cederqvist committed
101 102
 * This not an abort: all data is saved before we exit.
 */
103
Bool go_and_die = FALSE;
Per Cederqvist's avatar
Per Cederqvist committed
104 105

/*
106 107 108 109 110 111 112 113 114
 * 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
115
 */
116 117
static Bool work_done = FALSE;
static Bool is_idle = FALSE;
Per Cederqvist's avatar
Per Cederqvist committed
118 119 120 121 122 123 124 125

jmp_buf 	 parse_env;


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

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

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

BUGDECL;

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

Per Cederqvist's avatar
Per Cederqvist committed
139 140
/* head and tail pointers for the wait queue that uses the queue_prev,
   queue_next and on_queue fields of Connection. */
141 142 143
static Connection *queue_first = NULL;
static Connection *queue_last = NULL;

144 145 146
static void busy(void);


Per Cederqvist's avatar
Per Cederqvist committed
147
/* Add c to the end of the wait queue. */
148 149 150 151 152 153 154 155 156 157 158 159 160 161 162
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;
163
    update_stat(STAT_RUN_QUEUE, 1);
164 165
}

Per Cederqvist's avatar
Per Cederqvist committed
166
/* Remove c from the wait queue. Any element on the queue can be removed. */
167 168 169 170 171 172 173 174 175 176 177 178 179 180
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
181
	queue_first = c->queue_next;
Per Cederqvist's avatar
Per Cederqvist committed
182

183 184
    c->queue_prev = NULL;
    c->queue_next = NULL;
185
    update_stat(STAT_RUN_QUEUE, -1);
186 187
}

Per Cederqvist's avatar
Per Cederqvist committed
188
static void
Per Cederqvist's avatar
Per Cederqvist committed
189 190 191
logout_client(Connection *cp)
{
    Connection *real_active_connection;
192
    int ret;
Per Cederqvist's avatar
Per Cederqvist committed
193 194 195

    if ( active_connection != NULL )
    {
David Byers's avatar
David Byers committed
196
	kom_log("BUGCHK: logout_client(%ld): connection %ld is active.\n",
Per Cederqvist's avatar
Per Cederqvist committed
197 198 199 200 201
	    cp->session_no, active_connection->session_no);
    }
    
    if ( cp->pers_no != 0 )
    {
202 203 204 205 206 207 208 209 210 211 212 213
	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
214 215 216 217
	active_connection = cp;
	logout();
	active_connection = real_active_connection;
    }
218 219
    else
    {
220
#if 0
221
	/* FIXME (bug 908): send a new async here instead.  This causes the
222 223
	   elisp client to say that a secret (or unknown) person has
	   left the system.  */
224
        async_logout( 0, cp->session_no );
225
#endif
226
    }
Per Cederqvist's avatar
Per Cederqvist committed
227 228 229 230 231 232 233 234 235 236 237 238 239 240

    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");
    }

241 242
    ret = isc_destroy(kom_server_mcb, cp->isc_session);
    if (ret < 0)
David Byers's avatar
David Byers committed
243
	kom_log("logout_client(): isc_destroyed returned %d\n", ret);
244
    cp->isc_session = NULL;
245 246
    if (cp->on_queue)
	queue_remove(cp);
Per Cederqvist's avatar
Per Cederqvist committed
247 248

    kill_client(cp);		/* Free the Connection */
249
    update_stat(STAT_CLIENTS, -1);
Per Cederqvist's avatar
Per Cederqvist committed
250 251
}

Per Cederqvist's avatar
Per Cederqvist committed
252 253 254 255 256 257 258 259 260 261 262 263 264 265
/*
 * 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 )
266
	    restart_kom("logout_all_clients(): cant get session %ld.\n",
Per Cederqvist's avatar
Per Cederqvist committed
267 268 269 270 271 272
			sess);
	else
	    logout_client (conn);
    }

    if ( traverse_connections (0) != 0)
273
	restart_kom("logout_all_clients(): traverse_connections(0) == %ld.\n",
Per Cederqvist's avatar
Per Cederqvist committed
274 275 276 277
		    traverse_connections(0));
}

    
Per Cederqvist's avatar
Per Cederqvist committed
278 279 280 281 282 283 284 285
    

    
/*
 * 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
286 287
call_function(Connection *client,
	      union result_holder *res)
Per Cederqvist's avatar
Per Cederqvist committed
288 289 290 291 292
{
    Success	status=FAILURE;	/* OK if the call was successful. */

    if ( active_connection != NULL )
    {
293
	kom_log("call_function(%ld): active_connection = %ld\n",
Per Cederqvist's avatar
Per Cederqvist committed
294 295
	    client->session_no, active_connection->session_no);
    }
296 297 298

    if (client->function == illegal_fnc) 
    {
299
        err_stat = 0;
300 301 302 303
	kom_errno = KOM_NOT_IMPL;
	return FAILURE;
    }

Per Cederqvist's avatar
Per Cederqvist committed
304 305
    active_connection = client;

306
    service_statistics[client->function_index]++;
Per Cederqvist's avatar
Per Cederqvist committed
307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329

#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';
330 331
	    isc_puts("%%LysKOM unsupported protocol.\n", client->isc_session);
	    isc_flush(client->isc_session);
Per Cederqvist's avatar
Per Cederqvist committed
332
	    BUG(("%%%%Unsupported protocol.\n"));
333
	    longjmp(parse_env, KOM_LOGOUT);
Per Cederqvist's avatar
Per Cederqvist committed
334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362
	}
    }

    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
363 364 365
    sfree(client->misc_info_list.misc);
    client->misc_info_list.misc = 0;
    client->misc_info_list.no_of_misc = 0;
366
    s_clear(&client->aux_item.data);
367
    s_clear(&client->dummy_aux_item.data);
368 369 370
    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
371 372
    client->parse_pos = 0;
    client->fnc_parse_pos = 0;
373
    client->array_parse_index = 0;
David Byers's avatar
David Byers committed
374
    client->array_parse_parsed_length = 0;
Per Cederqvist's avatar
Per Cederqvist committed
375 376 377
    client->array_parse_pos = 0;
    client->struct_parse_pos = 0;
    client->string_parse_pos = 0;
David Byers's avatar
David Byers committed
378 379 380
    client->hunt_parse_pos = 0;
    client->array_hunt_num = 0;
    client->array_hunt_depth = 0;
381 382 383
    sfree(client->num_list.data);
    client->num_list.data = NULL;
    client->num_list.length = 0;
384 385
    free_aux_item_list(&client->aux_item_list);
    client->info.highest_aux_no = 0;
Per Cederqvist's avatar
Per Cederqvist committed
386 387 388 389 390 391 392
}

/*
 * Send a reply to a call.
 */
static void
reply(Connection *client,
393 394
      Success status,
      union result_holder *result)
Per Cederqvist's avatar
Per Cederqvist committed
395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410
{
    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.
411 412
 * 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
413
 */
414
static Bool
Per Cederqvist's avatar
Per Cederqvist committed
415 416 417
parse_unparsed(Connection *client)
{
    Success       status;
418
    union result_holder result;
Per Cederqvist's avatar
Per Cederqvist committed
419 420 421 422 423 424
        
    switch ( setjmp(parse_env) )
    {
    case 0 :
	/* Parse message. If message is complete call function and reply. */
	parse_packet(client);
425 426
	if (client->blocked_by_dns)
	    return TRUE;
427 428 429 430
#ifdef DEBUG_CALLS
	if (client->blocked_by_disable_client)
	    return TRUE;
#endif
431
	update_stat(STAT_REQUESTS, 1);
Per Cederqvist's avatar
Per Cederqvist committed
432
	status = call_function(client, &result);
433
	update_stat(STAT_REQUESTS, -1);
Per Cederqvist's avatar
Per Cederqvist committed
434
	reply(client, status, &result);
435
	client->penalty += param.penalty_per_call;
Per Cederqvist's avatar
Per Cederqvist committed
436
	free_parsed(client);
437
	end_of_atomic();
438
	return TRUE;
Per Cederqvist's avatar
Per Cederqvist committed
439

440
    case KOM_PROTOCOL_ERR:
Per Cederqvist's avatar
Per Cederqvist committed
441 442
	s_clear(&client->string0);
	free_parsed(client);
443
	isc_puts("%% LysKOM protocol error.\n", client->isc_session);
Per Cederqvist's avatar
Per Cederqvist committed
444
	BUG(("%%%% Protocol error.\n"));
445
	client->penalty += param.max_penalty;
Per Cederqvist's avatar
Per Cederqvist committed
446 447 448
	s_clear(&client->unparsed);
	client->first_to_parse = 0;
	client->more_to_parse = FALSE;
449
	end_of_atomic();
450
	return TRUE;
Per Cederqvist's avatar
Per Cederqvist committed
451

452
    case KOM_MSG_INCOMPLETE:
Per Cederqvist's avatar
Per Cederqvist committed
453
	client->more_to_parse = FALSE;
454
	return FALSE;
Per Cederqvist's avatar
Per Cederqvist committed
455

456
    case KOM_LOGOUT:
457
	add_to_kill_list(client);
458
	client->more_to_parse = FALSE;
459 460 461
	return TRUE;
    default:
	restart_kom("Bad longjmp return value.\n");
Per Cederqvist's avatar
Per Cederqvist committed
462
    }
463
    /*NOTREACHED*/
Per Cederqvist's avatar
Per Cederqvist committed
464 465 466 467 468
}


/* Return 1 if the named file exists, 0 otherwise */
static int
Per Cederqvist's avatar
Per Cederqvist committed
469
fexists(const char *filename)
Per Cederqvist's avatar
Per Cederqvist committed
470 471 472 473 474 475 476 477 478 479 480 481 482 483
{
    struct stat buf;
    int code;
  
    code = !stat(filename, &buf);
    errno = 0;

    return code;
}


void
dump_statistics(void)
{
484
    static struct timeval last_dump = {0, 0};
Per Cederqvist's avatar
Per Cederqvist committed
485 486 487
    int i;
    FILE *fp;

488
    if ((fp = i_fopen(param.statistic_name, "a")) == NULL)
Per Cederqvist's avatar
Per Cederqvist committed
489
    {
490 491
	kom_log("dump_statistics(): can't open file %s\n",
		param.statistic_name);
Per Cederqvist's avatar
Per Cederqvist committed
492 493 494
	return;
    }

495
    if (timeval_zero(last_dump))
Per Cederqvist's avatar
Per Cederqvist committed
496 497
    {
	fprintf(fp, "RESTART\n");
Per Cederqvist's avatar
Per Cederqvist committed
498
	last_dump = current_time;
Per Cederqvist's avatar
Per Cederqvist committed
499 500
    }
    
501 502
    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
503
    fprintf(fp, "STATISTICS:");
504 505 506 507 508

    /* 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
509
    {
510
	fprintf(fp, " %d:%lu", fnc_defs[i].function, service_statistics[i]);
Per Cederqvist's avatar
Per Cederqvist committed
511 512 513 514
	service_statistics[i]=0;
    }

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

Per Cederqvist's avatar
Per Cederqvist committed
517
    last_dump = current_time;
Per Cederqvist's avatar
Per Cederqvist committed
518 519
}

520 521
/* List of connections to kill. */

Per Cederqvist's avatar
Per Cederqvist committed
522 523
static Session_no *kill_list = NULL;
static int kill_list_size = 0;
524
static int kill_pending = 0;
525 526 527 528 529

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

533
    switch (conn->kill_status)
534
    {
535 536 537
    case ks_pending:
	/* A kill is already pending.  Do nothing--but check that the
	   the client really is present on the kill_list.  */
538 539 540
	for (i = 0; i < kill_list_size; i++)
	    if (kill_list[i] == conn->session_no)
		return;
541

542
	restart_kom("add_to_kill_list(): kill_pending set but not on list.\n");
543 544 545 546 547 548 549 550 551 552
	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;
553 554
    }

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

560 561 562 563 564 565 566 567 568 569 570 571 572 573 574
    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;
575
    conn->kill_status = ks_pending;
576 577 578

    if (!kill_pending)
    {
579
	source = isc_getoopsource(conn->isc_session);
580 581 582
	source->on_time(source, OOP_TIME_NOW, check_kill_flg, NULL);
	kill_pending = 1;
    }
583
}
Per Cederqvist's avatar
Per Cederqvist committed
584

585

586
static void
587 588 589 590 591 592
dump_connections(void)
{
    Session_no s;
    Connection *conn;
    FILE *fp;

593
    if ((fp = i_fopen(param.connection_status_file_tmp, "w")) == NULL)
594
    {
595 596
	kom_log("dump_connections(): can't open file %s: %s\n",
		param.connection_status_file_tmp, strerror(errno));
597 598 599 600 601 602
	return;
    }

    for (s = 0; (s = traverse_connections(s)) != 0;)
    {
	conn = get_conn_by_number(s);
603 604
	fprintf(fp, "%d %lu %d %s\n", conn->isc_session->fd, conn->session_no,
		handshake_ok(conn, 0),
605 606 607
		conn->peer);
    }

608 609 610 611 612 613
    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");

614
    if (i_fclose(fp) < 0)
615 616 617 618 619
    {
	kom_log("dump_connections(): fclose failed: %s (ignored)\n",
		strerror(errno));
    }

620 621 622
    errno = 0;
    if (i_rename(param.connection_status_file_tmp,
		 param.connection_status_file) < 0)
623
    {
624 625 626 627
	kom_log("dump_connections(): can't rename %s to %s: %s\n",
		param.connection_status_file_tmp,
		param.connection_status_file,
		strerror(errno));
628 629 630 631
    }
}


Per Cederqvist's avatar
Per Cederqvist committed
632 633 634
/*
 * check_kill_flg must NEVER be called inside an atomic call!
 */
635 636 637 638
static void *
check_kill_flg(oop_source *UNUSED(source),
	       struct timeval UNUSED(tv),
	       void *UNUSED(user))
Per Cederqvist's avatar
Per Cederqvist committed
639 640
{
    Connection *conn;
641
    Bool changed = FALSE;
Per Cederqvist's avatar
Per Cederqvist committed
642

643 644
    kill_pending = 0;

Per Cederqvist's avatar
Per Cederqvist committed
645 646
    if ( active_connection != NULL )
    {
647
	restart_kom("check_kill_flg: active_connection == %ld",
Per Cederqvist's avatar
Per Cederqvist committed
648 649 650
		    active_connection->session_no);
    }

651
    while (kill_list_size > 0)
Per Cederqvist's avatar
Per Cederqvist committed
652
    {
653 654 655
	--kill_list_size;
	conn = get_conn_by_number (kill_list[kill_list_size]);
	if (conn == NULL)
Per Cederqvist's avatar
Per Cederqvist committed
656
	{
David Byers's avatar
David Byers committed
657
	    kom_log("check_kill_flg(): Connection %ld doesn't exist.\n",
658 659 660 661
		kill_list[kill_list_size]);
	}
	else
	{
662 663
	    assert(conn->kill_status == ks_pending);
	    conn->kill_status = ks_dying;
664 665
	    logout_client(conn);
	    end_of_atomic();
666
	    changed = TRUE;
Per Cederqvist's avatar
Per Cederqvist committed
667 668
	}
    }
669

670 671 672 673 674
    if (kill_list != NULL)
    {
	sfree (kill_list);
	kill_list = NULL;
    }
675 676 677

    if (changed == TRUE)
	dump_connections();
678 679

    return OOP_CONTINUE;
Per Cederqvist's avatar
Per Cederqvist committed
680 681
}

682 683 684 685
static void *
dns_resolution(struct isc_scb *scb,
	       enum isc_resolve_status res,
	       long errcode)
686 687 688
{
    struct timeval after;
    double diff = -1.0;
689 690
    Connection *conn;
    char *hostname = NULL;
691

692
    update_stat(STAT_DNS_QUEUE, -1);
693

694
    conn = scb->udg;
695

696 697 698 699
    conn->dns_done = TRUE;
    if (conn->blocked_by_dns)
    {
	conn->blocked_by_dns = FALSE;
700 701
	if (!conn->on_queue)
	    queue_add(conn);
702 703
    }

704 705 706 707 708 709 710
    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);
711

712 713 714
    busy();

    switch (res)
715
    {
716
    case isc_resolve_h_errno:
717
	if (hostname == NULL)
718 719 720 721 722 723 724 725 726 727
	    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);
728
	else
729 730
	    kom_log("Unknown resolver error %ld looking up %s.\n",
		    errcode, hostname);
731
	break;
732

733
    case isc_resolve_adns_error:
734 735 736
	if (hostname == NULL)
	    hostname = s_crea_c_str(scb->remote);

737 738 739 740 741 742
	/* 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));
743 744 745 746 747 748 749
	break;

    case isc_resolve_aborted:
	abort();

    case isc_resolve_ok:
	break;
750 751
    }

752 753 754 755 756 757 758 759 760 761 762 763 764 765 766
    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); 

767 768 769
    if (handshake_ok(conn, 0))
	dump_connections();

770
    return OOP_CONTINUE;
771 772
}

773
    
774
static void
775 776
write_err_cb(struct isc_scb *cb_session,
	     int saved_errno)
777
{
778
    Connection  * cp = cb_session->udg;
779
    if (saved_errno != ECONNRESET && saved_errno != EPIPE)
780 781
	kom_log("Failed to write to client %lu from %s: %s\n",
		cp->session_no, cp->peer, strerror(saved_errno));
782
    cp->penalty += param.max_penalty;
783 784 785 786
    add_to_kill_list(cp);
}


787
static void
788
stale_cb(struct isc_scb *cb_session)
789
{
790
    Connection *cp = cb_session->udg;
791 792 793 794 795 796
    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);
}
    
797

798
static void
799
idle_cb(struct isc_scb *cb_session)
800
{
801
    Connection *cp = cb_session->udg;
802 803 804 805 806 807 808
    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
809
static void
810
login_request(struct isc_scb *session)
Per Cederqvist's avatar
Per Cederqvist committed
811 812
{
    Connection  * cp;
813
    const char *realuser;
814 815
    char portbuf[1+2+3*sizeof(long)];
    size_t portlen;
816
    char *remote_ip = NULL;
817
    Bool dns_submitted = FALSE;
Per Cederqvist's avatar
Per Cederqvist committed
818 819

    /* Supress logins if /etc/nologin exists */
David Byers's avatar
David Byers committed
820
    if (fexists(param.nologin_file))
Per Cederqvist's avatar
Per Cederqvist committed
821
    {
822 823 824
	isc_puts("%% No logins allowed.\n", session);
	isc_flush(session);
 	isc_destroy(kom_server_mcb, session);
Per Cederqvist's avatar
Per Cederqvist committed
825 826 827
	return;
    }

828 829
    /* Create a Connection, and link the Connection and the
       isc_session together. */
830

831
    cp = new_client();
832 833
    cp->isc_session = session;
    session->udg = cp;
834
    update_stat(STAT_CLIENTS, 1);
835

836 837 838
    /* 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);
839

840
    /* Initiate DNS lookup. */
841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857
    if (param.use_dns)
    {
	int rv = isc_resolve_remote(session, dns_resolution);
	if (rv == 0)
	{
	    dns_submitted = TRUE;
	    update_stat(STAT_DNS_QUEUE, 1);
	}
	else
	{
	    kom_log("WNG: isc_resolve_remote of %s failed: %s\n",
		    remote_ip, strerror(rv));
	}
    }

    if (!dns_submitted)
    {
858
	s_strcpy(&session->remote, cp->remote_ip);
859 860 861
	cp->blocked_by_dns = FALSE;
	cp->dns_done = TRUE;
    }
862

863 864 865 866 867 868
    /* 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);
869

870
    dump_connections();
871

872 873 874 875
    /* 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;
876

877 878 879
    cp->schedule.priority = param.default_priority;
    cp->schedule.weight = param.default_weight;

880 881 882
    /* 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)
883
    {
884 885 886 887 888 889 890 891 892
	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;
893 894
    }

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

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

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

static void
906
adjust_penalty(Connection *conn)
Per Cederqvist's avatar
Per Cederqvist committed
907
{
908 909
    unsigned int gens = penalty_generation - conn->penalty_generation;
    if (gens > 0)
910
    {
911 912 913 914 915 916
	/* 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;
917
	else
918 919 920 921 922 923 924 925 926
	{
	    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;
927
    }
Per Cederqvist's avatar
Per Cederqvist committed
928 929 930
}


931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959
static Bool
may_read_more(Connection *conn)
{
    if (go_and_die)
	return FALSE;
    if (conn->penalty >= param.max_penalty)
	return FALSE;
    if (conn->kill_status != ks_none)
	return FALSE;
    if (conn->blocked_by_dns)
	return FALSE;
#ifdef DEBUG_CALLS
    if (conn->blocked_by_disable_client == 1)
	return FALSE;
#endif
    return TRUE;
}

static Bool
should_be_disabled(Connection *conn)
{
#ifdef DEBUG_CALLS
    if (conn->blocked_by_disable_client == 1)
	return TRUE;
#endif

    return conn->penalty >= param.max_penalty && !conn->blocked_by_dns;
}

Per Cederqvist's avatar
Per Cederqvist committed
960
static void
961
read_from_connection(Connection *conn)