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