Annotation of src/usr.bin/ssh/dispatch.c, Revision 1.33
1.33 ! dtucker 1: /* $OpenBSD: dispatch.c,v 1.32 2019/01/19 21:33:13 djm Exp $ */
1.1 markus 2: /*
3: * Copyright (c) 2000 Markus Friedl. All rights reserved.
4: *
5: * Redistribution and use in source and binary forms, with or without
6: * modification, are permitted provided that the following conditions
7: * are met:
8: * 1. Redistributions of source code must retain the above copyright
9: * notice, this list of conditions and the following disclaimer.
10: * 2. Redistributions in binary form must reproduce the above copyright
11: * notice, this list of conditions and the following disclaimer in the
12: * documentation and/or other materials provided with the distribution.
13: *
14: * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
15: * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
16: * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
17: * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
18: * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
19: * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20: * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
21: * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22: * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23: * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24: */
1.21 deraadt 25:
26: #include <sys/types.h>
1.20 stevesk 27:
28: #include <signal.h>
1.21 deraadt 29: #include <stdarg.h>
1.7 markus 30:
1.8 markus 31: #include "ssh2.h"
1.7 markus 32: #include "log.h"
1.1 markus 33: #include "dispatch.h"
34: #include "packet.h"
1.23 markus 35: #include "ssherr.h"
1.1 markus 36:
1.23 markus 37: int
1.30 markus 38: dispatch_protocol_error(int type, u_int32_t seq, struct ssh *ssh)
1.23 markus 39: {
40: int r;
1.1 markus 41:
1.16 itojun 42: logit("dispatch_protocol_error: type %d seq %u", type, seq);
1.23 markus 43: if ((r = sshpkt_start(ssh, SSH2_MSG_UNIMPLEMENTED)) != 0 ||
44: (r = sshpkt_put_u32(ssh, seq)) != 0 ||
1.25 djm 45: (r = sshpkt_send(ssh)) != 0 ||
46: (r = ssh_packet_write_wait(ssh)) != 0)
1.32 djm 47: sshpkt_fatal(ssh, r, "%s", __func__);
1.23 markus 48: return 0;
1.15 markus 49: }
1.23 markus 50:
51: int
1.30 markus 52: dispatch_protocol_ignore(int type, u_int32_t seq, struct ssh *ssh)
1.15 markus 53: {
1.16 itojun 54: logit("dispatch_protocol_ignore: type %d seq %u", type, seq);
1.23 markus 55: return 0;
1.1 markus 56: }
1.23 markus 57:
1.2 markus 58: void
1.23 markus 59: ssh_dispatch_init(struct ssh *ssh, dispatch_fn *dflt)
1.1 markus 60: {
1.15 markus 61: u_int i;
1.1 markus 62: for (i = 0; i < DISPATCH_MAX; i++)
1.23 markus 63: ssh->dispatch[i] = dflt;
1.15 markus 64: }
1.23 markus 65:
1.15 markus 66: void
1.23 markus 67: ssh_dispatch_range(struct ssh *ssh, u_int from, u_int to, dispatch_fn *fn)
1.15 markus 68: {
69: u_int i;
70:
71: for (i = from; i <= to; i++) {
72: if (i >= DISPATCH_MAX)
73: break;
1.23 markus 74: ssh->dispatch[i] = fn;
1.15 markus 75: }
1.1 markus 76: }
1.23 markus 77:
1.1 markus 78: void
1.23 markus 79: ssh_dispatch_set(struct ssh *ssh, int type, dispatch_fn *fn)
1.1 markus 80: {
1.23 markus 81: ssh->dispatch[type] = fn;
1.1 markus 82: }
1.23 markus 83:
84: int
1.31 markus 85: ssh_dispatch_run(struct ssh *ssh, int mode, volatile sig_atomic_t *done)
1.23 markus 86: {
87: int r;
88: u_char type;
89: u_int32_t seqnr;
90:
1.1 markus 91: for (;;) {
92: if (mode == DISPATCH_BLOCK) {
1.23 markus 93: r = ssh_packet_read_seqnr(ssh, &type, &seqnr);
94: if (r != 0)
95: return r;
1.1 markus 96: } else {
1.23 markus 97: r = ssh_packet_read_poll_seqnr(ssh, &type, &seqnr);
98: if (r != 0)
99: return r;
1.1 markus 100: if (type == SSH_MSG_NONE)
1.23 markus 101: return 0;
102: }
103: if (type > 0 && type < DISPATCH_MAX &&
104: ssh->dispatch[type] != NULL) {
105: if (ssh->dispatch_skip_packets) {
106: debug2("skipped packet (type %u)", type);
107: ssh->dispatch_skip_packets--;
108: continue;
109: }
1.31 markus 110: r = (*ssh->dispatch[type])(type, seqnr, ssh);
1.23 markus 111: if (r != 0)
112: return r;
113: } else {
114: r = sshpkt_disconnect(ssh,
115: "protocol error: rcvd type %d", type);
116: if (r != 0)
117: return r;
118: return SSH_ERR_DISCONNECTED;
1.1 markus 119: }
120: if (done != NULL && *done)
1.23 markus 121: return 0;
1.1 markus 122: }
1.23 markus 123: }
124:
125: void
1.31 markus 126: ssh_dispatch_run_fatal(struct ssh *ssh, int mode, volatile sig_atomic_t *done)
1.23 markus 127: {
128: int r;
129:
1.31 markus 130: if ((r = ssh_dispatch_run(ssh, mode, done)) != 0)
1.32 djm 131: sshpkt_fatal(ssh, r, "%s", __func__);
1.1 markus 132: }