Annotation of src/usr.bin/ssh/dispatch.c, Revision 1.30
1.30 ! markus 1: /* $OpenBSD: dispatch.c,v 1.29 2017/04/30 23:28:42 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.8 markus 35: #include "compat.h"
1.23 markus 36: #include "ssherr.h"
1.1 markus 37:
1.23 markus 38: int
1.30 ! markus 39: dispatch_protocol_error(int type, u_int32_t seq, struct ssh *ssh)
1.23 markus 40: {
41: int r;
1.1 markus 42:
1.16 itojun 43: logit("dispatch_protocol_error: type %d seq %u", type, seq);
1.23 markus 44: if ((r = sshpkt_start(ssh, SSH2_MSG_UNIMPLEMENTED)) != 0 ||
45: (r = sshpkt_put_u32(ssh, seq)) != 0 ||
1.25 djm 46: (r = sshpkt_send(ssh)) != 0 ||
47: (r = ssh_packet_write_wait(ssh)) != 0)
48: sshpkt_fatal(ssh, __func__, r);
1.23 markus 49: return 0;
1.15 markus 50: }
1.23 markus 51:
52: int
1.30 ! markus 53: dispatch_protocol_ignore(int type, u_int32_t seq, struct ssh *ssh)
1.15 markus 54: {
1.16 itojun 55: logit("dispatch_protocol_ignore: type %d seq %u", type, seq);
1.23 markus 56: return 0;
1.1 markus 57: }
1.23 markus 58:
1.2 markus 59: void
1.23 markus 60: ssh_dispatch_init(struct ssh *ssh, dispatch_fn *dflt)
1.1 markus 61: {
1.15 markus 62: u_int i;
1.1 markus 63: for (i = 0; i < DISPATCH_MAX; i++)
1.23 markus 64: ssh->dispatch[i] = dflt;
1.15 markus 65: }
1.23 markus 66:
1.15 markus 67: void
1.23 markus 68: ssh_dispatch_range(struct ssh *ssh, u_int from, u_int to, dispatch_fn *fn)
1.15 markus 69: {
70: u_int i;
71:
72: for (i = from; i <= to; i++) {
73: if (i >= DISPATCH_MAX)
74: break;
1.23 markus 75: ssh->dispatch[i] = fn;
1.15 markus 76: }
1.1 markus 77: }
1.23 markus 78:
1.1 markus 79: void
1.23 markus 80: ssh_dispatch_set(struct ssh *ssh, int type, dispatch_fn *fn)
1.1 markus 81: {
1.23 markus 82: ssh->dispatch[type] = fn;
1.1 markus 83: }
1.23 markus 84:
85: int
86: ssh_dispatch_run(struct ssh *ssh, int mode, volatile sig_atomic_t *done,
87: void *ctxt)
88: {
89: int r;
90: u_char type;
91: u_int32_t seqnr;
92:
1.1 markus 93: for (;;) {
94: if (mode == DISPATCH_BLOCK) {
1.23 markus 95: r = ssh_packet_read_seqnr(ssh, &type, &seqnr);
96: if (r != 0)
97: return r;
1.1 markus 98: } else {
1.23 markus 99: r = ssh_packet_read_poll_seqnr(ssh, &type, &seqnr);
100: if (r != 0)
101: return r;
1.1 markus 102: if (type == SSH_MSG_NONE)
1.23 markus 103: return 0;
104: }
105: if (type > 0 && type < DISPATCH_MAX &&
106: ssh->dispatch[type] != NULL) {
107: if (ssh->dispatch_skip_packets) {
108: debug2("skipped packet (type %u)", type);
109: ssh->dispatch_skip_packets--;
110: continue;
111: }
112: /* XXX 'ssh' will replace 'ctxt' later */
113: r = (*ssh->dispatch[type])(type, seqnr, ctxt);
114: if (r != 0)
115: return r;
116: } else {
117: r = sshpkt_disconnect(ssh,
118: "protocol error: rcvd type %d", type);
119: if (r != 0)
120: return r;
121: return SSH_ERR_DISCONNECTED;
1.1 markus 122: }
123: if (done != NULL && *done)
1.23 markus 124: return 0;
1.1 markus 125: }
1.23 markus 126: }
127:
128: void
129: ssh_dispatch_run_fatal(struct ssh *ssh, int mode, volatile sig_atomic_t *done,
130: void *ctxt)
131: {
132: int r;
133:
1.27 djm 134: if ((r = ssh_dispatch_run(ssh, mode, done, ctxt)) != 0)
135: sshpkt_fatal(ssh, __func__, r);
1.1 markus 136: }