1 /* $NetBSD: nchan.c,v 1.9 2017/10/07 19:39:19 christos Exp $ */ 2 /* $OpenBSD: nchan.c,v 1.67 2017/09/12 06:35:32 djm Exp $ */ 3 /* 4 * Copyright (c) 1999, 2000, 2001, 2002 Markus Friedl. All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25 */ 26 27 #include "includes.h" 28 __RCSID("$NetBSD: nchan.c,v 1.9 2017/10/07 19:39:19 christos Exp $"); 29 #include <sys/types.h> 30 #include <sys/socket.h> 31 #include <sys/queue.h> 32 33 #include <errno.h> 34 #include <string.h> 35 #include <stdarg.h> 36 37 #include "ssh2.h" 38 #include "sshbuf.h" 39 #include "ssherr.h" 40 #include "packet.h" 41 #include "channels.h" 42 #include "compat.h" 43 #include "log.h" 44 45 /* 46 * SSH Protocol 1.5 aka New Channel Protocol 47 * Thanks to Martina, Axel and everyone who left Erlangen, leaving me bored. 48 * Written by Markus Friedl in October 1999 49 * 50 * Protocol versions 1.3 and 1.5 differ in the handshake protocol used for the 51 * tear down of channels: 52 * 53 * 1.3: strict request-ack-protocol: 54 * CLOSE -> 55 * <- CLOSE_CONFIRM 56 * 57 * 1.5: uses variations of: 58 * IEOF -> 59 * <- OCLOSE 60 * <- IEOF 61 * OCLOSE -> 62 * i.e. both sides have to close the channel 63 * 64 * 2.0: the EOF messages are optional 65 * 66 * See the debugging output from 'ssh -v' and 'sshd -d' of 67 * ssh-1.2.27 as an example. 68 * 69 */ 70 71 /* functions manipulating channel states */ 72 /* 73 * EVENTS update channel input/output states execute ACTIONS 74 */ 75 /* 76 * ACTIONS: should never update the channel states 77 */ 78 static void chan_send_eof2(struct ssh *, Channel *); 79 static void chan_send_eow2(struct ssh *, Channel *); 80 81 /* helper */ 82 static void chan_shutdown_write(struct ssh *, Channel *); 83 static void chan_shutdown_read(struct ssh *, Channel *); 84 85 static const char *ostates[] = { "open", "drain", "wait_ieof", "closed" }; 86 static const char *istates[] = { "open", "drain", "wait_oclose", "closed" }; 87 88 static void 89 chan_set_istate(Channel *c, u_int next) 90 { 91 if (c->istate > CHAN_INPUT_CLOSED || next > CHAN_INPUT_CLOSED) 92 fatal("chan_set_istate: bad state %d -> %d", c->istate, next); 93 debug2("channel %d: input %s -> %s", c->self, istates[c->istate], 94 istates[next]); 95 c->istate = next; 96 } 97 98 static void 99 chan_set_ostate(Channel *c, u_int next) 100 { 101 if (c->ostate > CHAN_OUTPUT_CLOSED || next > CHAN_OUTPUT_CLOSED) 102 fatal("chan_set_ostate: bad state %d -> %d", c->ostate, next); 103 debug2("channel %d: output %s -> %s", c->self, ostates[c->ostate], 104 ostates[next]); 105 c->ostate = next; 106 } 107 108 void 109 chan_read_failed(struct ssh *ssh, Channel *c) 110 { 111 debug2("channel %d: read failed", c->self); 112 switch (c->istate) { 113 case CHAN_INPUT_OPEN: 114 chan_shutdown_read(ssh, c); 115 chan_set_istate(c, CHAN_INPUT_WAIT_DRAIN); 116 break; 117 default: 118 error("channel %d: chan_read_failed for istate %d", 119 c->self, c->istate); 120 break; 121 } 122 } 123 124 void 125 chan_ibuf_empty(struct ssh *ssh, Channel *c) 126 { 127 debug2("channel %d: ibuf empty", c->self); 128 if (sshbuf_len(c->input)) { 129 error("channel %d: chan_ibuf_empty for non empty buffer", 130 c->self); 131 return; 132 } 133 switch (c->istate) { 134 case CHAN_INPUT_WAIT_DRAIN: 135 if (!(c->flags & (CHAN_CLOSE_SENT|CHAN_LOCAL))) 136 chan_send_eof2(ssh, c); 137 chan_set_istate(c, CHAN_INPUT_CLOSED); 138 break; 139 default: 140 error("channel %d: chan_ibuf_empty for istate %d", 141 c->self, c->istate); 142 break; 143 } 144 } 145 146 void 147 chan_obuf_empty(struct ssh *ssh, Channel *c) 148 { 149 debug2("channel %d: obuf empty", c->self); 150 if (sshbuf_len(c->output)) { 151 error("channel %d: chan_obuf_empty for non empty buffer", 152 c->self); 153 return; 154 } 155 switch (c->ostate) { 156 case CHAN_OUTPUT_WAIT_DRAIN: 157 chan_shutdown_write(ssh, c); 158 chan_set_ostate(c, CHAN_OUTPUT_CLOSED); 159 break; 160 default: 161 error("channel %d: internal error: obuf_empty for ostate %d", 162 c->self, c->ostate); 163 break; 164 } 165 } 166 167 void 168 chan_rcvd_eow(struct ssh *ssh, Channel *c) 169 { 170 debug2("channel %d: rcvd eow", c->self); 171 switch (c->istate) { 172 case CHAN_INPUT_OPEN: 173 chan_shutdown_read(ssh, c); 174 chan_set_istate(c, CHAN_INPUT_CLOSED); 175 break; 176 } 177 } 178 179 static void 180 chan_send_eof2(struct ssh *ssh, Channel *c) 181 { 182 int r; 183 184 debug2("channel %d: send eof", c->self); 185 switch (c->istate) { 186 case CHAN_INPUT_WAIT_DRAIN: 187 if (!c->have_remote_id) 188 fatal("%s: channel %d: no remote_id", 189 __func__, c->self); 190 if ((r = sshpkt_start(ssh, SSH2_MSG_CHANNEL_EOF)) != 0 || 191 (r = sshpkt_put_u32(ssh, c->remote_id)) != 0 || 192 (r = sshpkt_send(ssh)) != 0) 193 fatal("%s: send CHANNEL_EOF: %s", __func__, ssh_err(r)); 194 c->flags |= CHAN_EOF_SENT; 195 break; 196 default: 197 error("channel %d: cannot send eof for istate %d", 198 c->self, c->istate); 199 break; 200 } 201 } 202 203 static void 204 chan_send_close2(struct ssh *ssh, Channel *c) 205 { 206 int r; 207 208 debug2("channel %d: send close", c->self); 209 if (c->ostate != CHAN_OUTPUT_CLOSED || 210 c->istate != CHAN_INPUT_CLOSED) { 211 error("channel %d: cannot send close for istate/ostate %d/%d", 212 c->self, c->istate, c->ostate); 213 } else if (c->flags & CHAN_CLOSE_SENT) { 214 error("channel %d: already sent close", c->self); 215 } else { 216 if (!c->have_remote_id) 217 fatal("%s: channel %d: no remote_id", 218 __func__, c->self); 219 if ((r = sshpkt_start(ssh, SSH2_MSG_CHANNEL_CLOSE)) != 0 || 220 (r = sshpkt_put_u32(ssh, c->remote_id)) != 0 || 221 (r = sshpkt_send(ssh)) != 0) 222 fatal("%s: send CHANNEL_EOF: %s", __func__, ssh_err(r)); 223 c->flags |= CHAN_CLOSE_SENT; 224 } 225 } 226 227 static void 228 chan_send_eow2(struct ssh *ssh, Channel *c) 229 { 230 int r; 231 232 debug2("channel %d: send eow", c->self); 233 if (c->ostate == CHAN_OUTPUT_CLOSED) { 234 error("channel %d: must not sent eow on closed output", 235 c->self); 236 return; 237 } 238 if (!(datafellows & SSH_NEW_OPENSSH)) 239 return; 240 if (!c->have_remote_id) 241 fatal("%s: channel %d: no remote_id", __func__, c->self); 242 if ((r = sshpkt_start(ssh, SSH2_MSG_CHANNEL_REQUEST)) != 0 || 243 (r = sshpkt_put_u32(ssh, c->remote_id)) != 0 || 244 (r = sshpkt_put_cstring(ssh, "eow@openssh.com")) != 0 || 245 (r = sshpkt_put_u8(ssh, 0)) != 0 || 246 (r = sshpkt_send(ssh)) != 0) 247 fatal("%s: send CHANNEL_EOF: %s", __func__, ssh_err(r)); 248 } 249 250 /* shared */ 251 252 void 253 chan_rcvd_ieof(struct ssh *ssh, Channel *c) 254 { 255 debug2("channel %d: rcvd eof", c->self); 256 c->flags |= CHAN_EOF_RCVD; 257 if (c->ostate == CHAN_OUTPUT_OPEN) 258 chan_set_ostate(c, CHAN_OUTPUT_WAIT_DRAIN); 259 if (c->ostate == CHAN_OUTPUT_WAIT_DRAIN && 260 sshbuf_len(c->output) == 0 && 261 !CHANNEL_EFD_OUTPUT_ACTIVE(c)) 262 chan_obuf_empty(ssh, c); 263 } 264 265 void 266 chan_rcvd_oclose(struct ssh *ssh, Channel *c) 267 { 268 debug2("channel %d: rcvd close", c->self); 269 if (!(c->flags & CHAN_LOCAL)) { 270 if (c->flags & CHAN_CLOSE_RCVD) 271 error("channel %d: protocol error: close rcvd twice", 272 c->self); 273 c->flags |= CHAN_CLOSE_RCVD; 274 } 275 if (c->type == SSH_CHANNEL_LARVAL) { 276 /* tear down larval channels immediately */ 277 chan_set_ostate(c, CHAN_OUTPUT_CLOSED); 278 chan_set_istate(c, CHAN_INPUT_CLOSED); 279 return; 280 } 281 switch (c->ostate) { 282 case CHAN_OUTPUT_OPEN: 283 /* 284 * wait until a data from the channel is consumed if a CLOSE 285 * is received 286 */ 287 chan_set_ostate(c, CHAN_OUTPUT_WAIT_DRAIN); 288 break; 289 } 290 switch (c->istate) { 291 case CHAN_INPUT_OPEN: 292 chan_shutdown_read(ssh, c); 293 chan_set_istate(c, CHAN_INPUT_CLOSED); 294 break; 295 case CHAN_INPUT_WAIT_DRAIN: 296 if (!(c->flags & CHAN_LOCAL)) 297 chan_send_eof2(ssh, c); 298 chan_set_istate(c, CHAN_INPUT_CLOSED); 299 break; 300 } 301 } 302 303 void 304 chan_write_failed(struct ssh *ssh, Channel *c) 305 { 306 debug2("channel %d: write failed", c->self); 307 switch (c->ostate) { 308 case CHAN_OUTPUT_OPEN: 309 case CHAN_OUTPUT_WAIT_DRAIN: 310 chan_shutdown_write(ssh, c); 311 if (strcmp(c->ctype, "session") == 0) 312 chan_send_eow2(ssh, c); 313 chan_set_ostate(c, CHAN_OUTPUT_CLOSED); 314 break; 315 default: 316 error("channel %d: chan_write_failed for ostate %d", 317 c->self, c->ostate); 318 break; 319 } 320 } 321 322 void 323 chan_mark_dead(struct ssh *ssh, Channel *c) 324 { 325 c->type = SSH_CHANNEL_ZOMBIE; 326 } 327 328 int 329 chan_is_dead(struct ssh *ssh, Channel *c, int do_send) 330 { 331 if (c->type == SSH_CHANNEL_ZOMBIE) { 332 debug2("channel %d: zombie", c->self); 333 return 1; 334 } 335 if (c->istate != CHAN_INPUT_CLOSED || c->ostate != CHAN_OUTPUT_CLOSED) 336 return 0; 337 if ((datafellows & SSH_BUG_EXTEOF) && 338 c->extended_usage == CHAN_EXTENDED_WRITE && 339 c->efd != -1 && 340 sshbuf_len(c->extended) > 0) { 341 debug2("channel %d: active efd: %d len %zu", 342 c->self, c->efd, sshbuf_len(c->extended)); 343 return 0; 344 } 345 if (c->flags & CHAN_LOCAL) { 346 debug2("channel %d: is dead (local)", c->self); 347 return 1; 348 } 349 if (!(c->flags & CHAN_CLOSE_SENT)) { 350 if (do_send) { 351 chan_send_close2(ssh, c); 352 } else { 353 /* channel would be dead if we sent a close */ 354 if (c->flags & CHAN_CLOSE_RCVD) { 355 debug2("channel %d: almost dead", 356 c->self); 357 return 1; 358 } 359 } 360 } 361 if ((c->flags & CHAN_CLOSE_SENT) && 362 (c->flags & CHAN_CLOSE_RCVD)) { 363 debug2("channel %d: is dead", c->self); 364 return 1; 365 } 366 return 0; 367 } 368 369 /* helper */ 370 static void 371 chan_shutdown_write(struct ssh *ssh, Channel *c) 372 { 373 sshbuf_reset(c->output); 374 if (c->type == SSH_CHANNEL_LARVAL) 375 return; 376 /* shutdown failure is allowed if write failed already */ 377 debug2("channel %d: close_write", c->self); 378 if (c->sock != -1) { 379 if (shutdown(c->sock, SHUT_WR) < 0) 380 debug2("channel %d: chan_shutdown_write: " 381 "shutdown() failed for fd %d: %.100s", 382 c->self, c->sock, strerror(errno)); 383 } else { 384 if (channel_close_fd(ssh, &c->wfd) < 0) 385 logit("channel %d: chan_shutdown_write: " 386 "close() failed for fd %d: %.100s", 387 c->self, c->wfd, strerror(errno)); 388 } 389 } 390 391 static void 392 chan_shutdown_read(struct ssh *ssh, Channel *c) 393 { 394 if (c->type == SSH_CHANNEL_LARVAL) 395 return; 396 debug2("channel %d: close_read", c->self); 397 if (c->sock != -1) { 398 if (shutdown(c->sock, SHUT_RD) < 0) 399 error("channel %d: chan_shutdown_read: " 400 "shutdown() failed for fd %d [i%d o%d]: %.100s", 401 c->self, c->sock, c->istate, c->ostate, 402 strerror(errno)); 403 } else { 404 if (channel_close_fd(ssh, &c->rfd) < 0) 405 logit("channel %d: chan_shutdown_read: " 406 "close() failed for fd %d: %.100s", 407 c->self, c->rfd, strerror(errno)); 408 } 409 } 410