1 /* $NetBSD: msg.c,v 1.2 2009/06/07 22:38:46 christos Exp $ */ 2 /* $OpenBSD: msg.c,v 1.15 2006/08/03 03:34:42 deraadt Exp $ */ 3 /* 4 * Copyright (c) 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: msg.c,v 1.2 2009/06/07 22:38:46 christos Exp $"); 29 #include <sys/types.h> 30 #include <sys/uio.h> 31 32 #include <errno.h> 33 #include <stdio.h> 34 #include <string.h> 35 #include <unistd.h> 36 #include <stdarg.h> 37 #include <time.h> 38 39 #include "buffer.h" 40 #include "log.h" 41 #include "atomicio.h" 42 #include "msg.h" 43 #include "misc.h" 44 45 int 46 ssh_msg_send(int fd, u_char type, Buffer *m) 47 { 48 u_char buf[5]; 49 u_int mlen = buffer_len(m); 50 51 debug3("ssh_msg_send: type %u", (unsigned int)type & 0xff); 52 53 put_u32(buf, mlen + 1); 54 buf[4] = type; /* 1st byte of payload is mesg-type */ 55 if (atomicio(vwrite, fd, buf, sizeof(buf)) != sizeof(buf)) { 56 error("ssh_msg_send: write"); 57 return (-1); 58 } 59 if (atomicio(vwrite, fd, buffer_ptr(m), mlen) != mlen) { 60 error("ssh_msg_send: write"); 61 return (-1); 62 } 63 return (0); 64 } 65 66 int 67 ssh_msg_recv(int fd, Buffer *m) 68 { 69 u_char buf[4]; 70 u_int msg_len; 71 72 debug3("ssh_msg_recv entering"); 73 74 if (atomicio(read, fd, buf, sizeof(buf)) != sizeof(buf)) { 75 if (errno != EPIPE) 76 error("ssh_msg_recv: read: header"); 77 return (-1); 78 } 79 msg_len = get_u32(buf); 80 if (msg_len > 256 * 1024) { 81 error("ssh_msg_recv: read: bad msg_len %u", msg_len); 82 return (-1); 83 } 84 buffer_clear(m); 85 buffer_append_space(m, msg_len); 86 if (atomicio(read, fd, buffer_ptr(m), msg_len) != msg_len) { 87 error("ssh_msg_recv: read: %s", strerror(errno)); 88 return (-1); 89 } 90 return (0); 91 } 92