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.4 2000/09/07 20:27:51 deraadt Exp $"); 26 #include "ssh.h" 27 #include "dispatch.h" 28 #include "packet.h" 29 30 #define DISPATCH_MIN 0 31 #define DISPATCH_MAX 255 32 33 dispatch_fn *dispatch[DISPATCH_MAX]; 34 35 void 36 dispatch_protocol_error(int type, int plen) 37 { 38 error("Hm, dispatch protocol error: type %d plen %d", type, plen); 39 } 40 void 41 dispatch_init(dispatch_fn *dflt) 42 { 43 int i; 44 for (i = 0; i < DISPATCH_MAX; i++) 45 dispatch[i] = dflt; 46 } 47 void 48 dispatch_set(int type, dispatch_fn *fn) 49 { 50 dispatch[type] = fn; 51 } 52 void 53 dispatch_run(int mode, int *done) 54 { 55 for (;;) { 56 int plen; 57 int type; 58 59 if (mode == DISPATCH_BLOCK) { 60 type = packet_read(&plen); 61 } else { 62 type = packet_read_poll(&plen); 63 if (type == SSH_MSG_NONE) 64 return; 65 } 66 if (type > 0 && type < DISPATCH_MAX && dispatch[type] != NULL) 67 (*dispatch[type])(type, plen); 68 else 69 packet_disconnect("protocol error: rcvd type %d", type); 70 if (done != NULL && *done) 71 return; 72 } 73 } 74