Home | History | Annotate | Download | only in common
      1 /*
      2  * Copyright (c) 2000 Markus Friedl.  All rights reserved.
      3  *
      4  * Redistribution and use in source and binary forms, with or without
      5  * modification, are permitted provided that the following conditions
      6  * are met:
      7  * 1. Redistributions of source code must retain the above copyright
      8  *    notice, this list of conditions and the following disclaimer.
      9  * 2. Redistributions in binary form must reproduce the above copyright
     10  *    notice, this list of conditions and the following disclaimer in the
     11  *    documentation and/or other materials provided with the distribution.
     12  *
     13  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     14  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     15  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     16  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     17  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     18  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     19  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     20  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     21  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     22  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     23  */
     24 #include "includes.h"
     25 RCSID("$OpenBSD: dispatch.c,v 1.15 2002/01/11 13:39:36 markus Exp $");
     26 
     27 #pragma ident	"%Z%%M%	%I%	%E% SMI"
     28 
     29 #include "ssh1.h"
     30 #include "ssh2.h"
     31 #include "log.h"
     32 #include "dispatch.h"
     33 #include "packet.h"
     34 #include "compat.h"
     35 
     36 #define DISPATCH_MIN	0
     37 #define DISPATCH_MAX	255
     38 
     39 dispatch_fn *dispatch[DISPATCH_MAX];
     40 
     41 void
     42 dispatch_protocol_error(int type, u_int32_t seq, void *ctxt)
     43 {
     44 	log("dispatch_protocol_error: type %d seq %u", type, seq);
     45 	if (!compat20)
     46 		fatal("protocol error");
     47 	packet_start(SSH2_MSG_UNIMPLEMENTED);
     48 	packet_put_int(seq);
     49 	packet_send();
     50 	packet_write_wait();
     51 }
     52 void
     53 dispatch_protocol_ignore(int type, u_int32_t seq, void *ctxt)
     54 {
     55 	log("dispatch_protocol_ignore: type %d seq %u", type, seq);
     56 }
     57 void
     58 dispatch_init(dispatch_fn *dflt)
     59 {
     60 	u_int i;
     61 	for (i = 0; i < DISPATCH_MAX; i++)
     62 		dispatch[i] = dflt;
     63 }
     64 void
     65 dispatch_range(u_int from, u_int to, dispatch_fn *fn)
     66 {
     67 	u_int i;
     68 
     69 	for (i = from; i <= to; i++) {
     70 		if (i >= DISPATCH_MAX)
     71 			break;
     72 		dispatch[i] = fn;
     73 	}
     74 }
     75 void
     76 dispatch_set(int type, dispatch_fn *fn)
     77 {
     78 	dispatch[type] = fn;
     79 }
     80 void
     81 dispatch_run(int mode, int *done, void *ctxt)
     82 {
     83 	for (;;) {
     84 		int type;
     85 		u_int32_t seqnr;
     86 
     87 		if (mode == DISPATCH_BLOCK) {
     88 			type = packet_read_seqnr(&seqnr);
     89 		} else {
     90 			type = packet_read_poll_seqnr(&seqnr);
     91 			if (type == SSH_MSG_NONE)
     92 				return;
     93 		}
     94 		if (type > 0 && type < DISPATCH_MAX && dispatch[type] != NULL)
     95 			(*dispatch[type])(type, seqnr, ctxt);
     96 		else
     97 			packet_disconnect("protocol error: rcvd type %d", type);
     98 		if (done != NULL && *done)
     99 			return;
    100 	}
    101 }
    102