More ibuf cleanup. rtr_proto.c still uses ibuf_add() where it could use
the new functions.

Two bits I'm unsure about:
- I had to change some sizeof() to use native types (I especially dislike
  the sizeof(struct rtr_header).
- ibuf_add_nXX() can fail if the value is too large. Which should be
  impossible but still maybe it would be better to check for errors.

-- 
:wq Claudio

Index: rtr_proto.c
===================================================================
RCS file: /cvs/src/usr.sbin/bgpd/rtr_proto.c,v
retrieving revision 1.17
diff -u -p -r1.17 rtr_proto.c
--- rtr_proto.c 16 Aug 2023 08:26:35 -0000      1.17
+++ rtr_proto.c 19 Oct 2023 08:35:49 -0000
@@ -233,24 +233,21 @@ rtr_newmsg(struct rtr_session *rs, enum 
     uint16_t session_id)
 {
        struct ibuf *buf;
-       struct rtr_header rh;
 
        if (len > RTR_MAX_LEN) {
                errno = ERANGE;
                return NULL;
        }
-       len += sizeof(rh);
+       len += sizeof(struct rtr_header);
        if ((buf = ibuf_open(len)) == NULL)
                return NULL;
 
-       memset(&rh, 0, sizeof(rh));
-       rh.version = rs->version;
-       rh.type = type;
-       rh.session_id = htons(session_id);
-       rh.length = htonl(len);
-
        /* cannot fail with fixed buffers */
-       ibuf_add(buf, &rh, sizeof(rh));
+       ibuf_add_n8(buf, rs->version);
+       ibuf_add_n8(buf, type);
+       ibuf_add_n16(buf, session_id);
+       ibuf_add_n32(buf, len);
+
        return buf;
 }
 
@@ -264,7 +261,6 @@ rtr_send_error(struct rtr_session *rs, e
 {
        struct ibuf *buf;
        size_t mlen = 0;
-       uint32_t hdrlen;
 
        rs->last_sent_error = err;
        if (msg) {
@@ -273,7 +269,7 @@ rtr_send_error(struct rtr_session *rs, e
        } else
                memset(rs->last_sent_msg, 0, sizeof(rs->last_sent_msg));
 
-       buf = rtr_newmsg(rs, ERROR_REPORT, 2 * sizeof(hdrlen) + len + mlen,
+       buf = rtr_newmsg(rs, ERROR_REPORT, 2 * sizeof(uint32_t) + len + mlen,
            err);
        if (buf == NULL) {
                log_warn("rtr %s: send error report", log_rtr(rs));
@@ -281,11 +277,9 @@ rtr_send_error(struct rtr_session *rs, e
        }
 
        /* cannot fail with fixed buffers */
-       hdrlen = ntohl(len);
-       ibuf_add(buf, &hdrlen, sizeof(hdrlen));
+       ibuf_add_n32(buf, len);
        ibuf_add(buf, pdu, len);
-       hdrlen = ntohl(mlen);
-       ibuf_add(buf, &hdrlen, sizeof(hdrlen));
+       ibuf_add_n32(buf, mlen);
        ibuf_add(buf, msg, mlen);
        ibuf_close(&rs->w, buf);
 
@@ -313,9 +307,8 @@ static void
 rtr_send_serial_query(struct rtr_session *rs)
 {
        struct ibuf *buf;
-       uint32_t s;
 
-       buf = rtr_newmsg(rs, SERIAL_QUERY, sizeof(s), rs->session_id);
+       buf = rtr_newmsg(rs, SERIAL_QUERY, sizeof(uint32_t), rs->session_id);
        if (buf == NULL) {
                log_warn("rtr %s: send serial query", log_rtr(rs));
                rtr_send_error(rs, INTERNAL_ERROR, "out of memory", NULL, 0);
@@ -323,8 +316,7 @@ rtr_send_serial_query(struct rtr_session
        }
 
        /* cannot fail with fixed buffers */
-       s = htonl(rs->serial);
-       ibuf_add(buf, &s, sizeof(s));
+       ibuf_add_n32(buf, rs->serial);
        ibuf_close(&rs->w, buf);
 }
 

Reply via email to