=================================================================== RCS file: /cvsrepo/anoncvs/cvs/src/usr.bin/ssh/msg.c,v retrieving revision 1.5.2.2 retrieving revision 1.6 diff -u -r1.5.2.2 -r1.6 --- src/usr.bin/ssh/msg.c 2004/03/04 18:18:16 1.5.2.2 +++ src/usr.bin/ssh/msg.c 2003/06/28 16:23:06 1.6 @@ -22,7 +22,7 @@ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include "includes.h" -RCSID("$OpenBSD: msg.c,v 1.5.2.2 2004/03/04 18:18:16 brad Exp $"); +RCSID("$OpenBSD: msg.c,v 1.6 2003/06/28 16:23:06 deraadt Exp $"); #include "buffer.h" #include "getput.h" @@ -30,7 +30,7 @@ #include "atomicio.h" #include "msg.h" -int +void ssh_msg_send(int fd, u_char type, Buffer *m) { u_char buf[5]; @@ -40,15 +40,10 @@ PUT_32BIT(buf, mlen + 1); buf[4] = type; /* 1st byte of payload is mesg-type */ - if (atomicio(vwrite, fd, buf, sizeof(buf)) != sizeof(buf)) { - error("ssh_msg_send: write"); - return (-1); - } - if (atomicio(vwrite, fd, buffer_ptr(m), mlen) != mlen) { - error("ssh_msg_send: write"); - return (-1); - } - return (0); + if (atomicio(vwrite, fd, buf, sizeof(buf)) != sizeof(buf)) + fatal("ssh_msg_send: write"); + if (atomicio(vwrite, fd, buffer_ptr(m), mlen) != mlen) + fatal("ssh_msg_send: write"); } int @@ -62,21 +57,17 @@ res = atomicio(read, fd, buf, sizeof(buf)); if (res != sizeof(buf)) { - if (res != 0) - error("ssh_msg_recv: read: header %ld", (long)res); - return (-1); + if (res == 0) + return -1; + fatal("ssh_msg_recv: read: header %ld", (long)res); } msg_len = GET_32BIT(buf); - if (msg_len > 256 * 1024) { - error("ssh_msg_recv: read: bad msg_len %u", msg_len); - return (-1); - } + if (msg_len > 256 * 1024) + fatal("ssh_msg_recv: read: bad msg_len %u", msg_len); buffer_clear(m); buffer_append_space(m, msg_len); res = atomicio(read, fd, buffer_ptr(m), msg_len); - if (res != msg_len) { - error("ssh_msg_recv: read: %ld != msg_len", (long)res); - return (-1); - } - return (0); + if (res != msg_len) + fatal("ssh_msg_recv: read: %ld != msg_len", (long)res); + return 0; }