Home | History | Annotate | Line # | Download | only in dist 
      1 /*	$NetBSD: msg.c,v 1.13 2026/04/08 18:58:41 christos Exp $	*/
      2 /* $OpenBSD: msg.c,v 1.22 2026/02/14 00:18:34 jsg Exp $ */
      3 
      4 /*
      5  * Copyright (c) 2002 Markus Friedl.  All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     26  */
     27 
     28 #include "includes.h"
     29 __RCSID("$NetBSD: msg.c,v 1.13 2026/04/08 18:58:41 christos Exp $");
     30 #include <sys/types.h>
     31 
     32 #include <errno.h>
     33 #include <string.h>
     34 #include <unistd.h>
     35 #include <stdarg.h>
     36 #include <time.h>
     37 
     38 #include "sshbuf.h"
     39 #include "log.h"
     40 #include "atomicio.h"
     41 #include "msg.h"
     42 #include "misc.h"
     43 
     44 int
     45 ssh_msg_send(int fd, u_char type, struct sshbuf *m)
     46 {
     47 	u_char buf[5];
     48 	u_int mlen = sshbuf_len(m);
     49 
     50 	debug3_f("type %u len %zu", (unsigned int)type & 0xff, sshbuf_len(m));
     51 
     52 	put_u32(buf, mlen + 1);
     53 	buf[4] = type;		/* 1st byte of payload is mesg-type */
     54 	if (atomicio(vwrite, fd, buf, sizeof(buf)) != sizeof(buf)) {
     55 		error_f("write: %s", strerror(errno));
     56 		return (-1);
     57 	}
     58 	if (atomicio(vwrite, fd, sshbuf_mutable_ptr(m), mlen) != mlen) {
     59 		error_f("write: %s", strerror(errno));
     60 		return (-1);
     61 	}
     62 	debug3_f("done");
     63 	return (0);
     64 }
     65 
     66 int
     67 ssh_msg_recv(int fd, struct sshbuf *m)
     68 {
     69 	u_char buf[4], *p;
     70 	u_int msg_len;
     71 	int r;
     72 
     73 	debug3("ssh_msg_recv entering");
     74 
     75 	if (atomicio(read, fd, buf, sizeof(buf)) != sizeof(buf)) {
     76 		if (errno != EPIPE)
     77 			error_f("read header: %s", strerror(errno));
     78 		return (-1);
     79 	}
     80 	msg_len = get_u32(buf);
     81 	if (msg_len > sshbuf_max_size(m)) {
     82 		error_f("read: bad msg_len %u", msg_len);
     83 		return (-1);
     84 	}
     85 	sshbuf_reset(m);
     86 	if ((r = sshbuf_reserve(m, msg_len, &p)) != 0) {
     87 		error_fr(r, "reserve");
     88 		return -1;
     89 	}
     90 	if (atomicio(read, fd, p, msg_len) != msg_len) {
     91 		error_f("read: %s", strerror(errno));
     92 		return (-1);
     93 	}
     94 	return (0);
     95 }
     96