/src/openssl/crypto/packet.c

Source (jump to first uncovered line)
/*
 * Copyright 2015-2023 The OpenSSL Project Authors. All Rights Reserved.
 *
 * Licensed under the Apache License 2.0 (the "License").  You may not use
 * this file except in compliance with the License.  You can obtain a copy
 * in the file LICENSE in the source distribution or at
 * https://www.openssl.org/source/license.html
 */

#include "internal/cryptlib.h"
#include "internal/packet.h"
#if !defined OPENSSL_NO_QUIC && !defined FIPS_MODULE
# include "internal/packet_quic.h"
#endif
#include <openssl/err.h>

#define DEFAULT_BUF_SIZE    256

int WPACKET_allocate_bytes(WPACKET *pkt, size_t len, unsigned char **allocbytes)
{
    if (!WPACKET_reserve_bytes(pkt, len, allocbytes))
        return 0;

    pkt->written += len;
    pkt->curr += len;
    return 1;
}

int WPACKET_sub_allocate_bytes__(WPACKET *pkt, size_t len,
                                 unsigned char **allocbytes, size_t lenbytes)
{
    if (!WPACKET_start_sub_packet_len__(pkt, lenbytes)
            || !WPACKET_allocate_bytes(pkt, len, allocbytes)
            || !WPACKET_close(pkt))
        return 0;

    return 1;
}

#define GETBUF(p)   (((p)->staticbuf != NULL) \
                     ? (p)->staticbuf \
                     : ((p)->buf != NULL \
                        ? (unsigned char *)(p)->buf->data \
                        : NULL))

int WPACKET_reserve_bytes(WPACKET *pkt, size_t len, unsigned char **allocbytes)
{
    /* Internal API, so should not fail */
    if (!ossl_assert(pkt->subs != NULL && len != 0))
        return 0;

    if (pkt->maxsize - pkt->written < len)
        return 0;

    if (pkt->buf != NULL && (pkt->buf->length - pkt->written < len)) {
        size_t newlen;
        size_t reflen;

        reflen = (len > pkt->buf->length) ? len : pkt->buf->length;

        if (reflen > SIZE_MAX / 2) {
            newlen = SIZE_MAX;
        } else {
            newlen = reflen * 2;
            if (newlen < DEFAULT_BUF_SIZE)
                newlen = DEFAULT_BUF_SIZE;
        }
        if (BUF_MEM_grow(pkt->buf, newlen) == 0)
            return 0;
    }
    if (allocbytes != NULL) {
        *allocbytes = WPACKET_get_curr(pkt);
        if (pkt->endfirst && *allocbytes != NULL)
            *allocbytes -= len;
    }

    return 1;
}

int WPACKET_sub_reserve_bytes__(WPACKET *pkt, size_t len,
                                unsigned char **allocbytes, size_t lenbytes)
{
    if (pkt->endfirst && lenbytes > 0)
        return 0;

    if (!WPACKET_reserve_bytes(pkt, lenbytes + len, allocbytes))
        return 0;

    if (*allocbytes != NULL)
        *allocbytes += lenbytes;

    return 1;
}

static size_t maxmaxsize(size_t lenbytes)
{
    if (lenbytes >= sizeof(size_t) || lenbytes == 0)
        return SIZE_MAX;

    return ((size_t)1 << (lenbytes * 8)) - 1 + lenbytes;
}

static int wpacket_intern_init_len(WPACKET *pkt, size_t lenbytes)
{
    unsigned char *lenchars;

    pkt->curr = 0;
    pkt->written = 0;

    if ((pkt->subs = OPENSSL_zalloc(sizeof(*pkt->subs))) == NULL)
        return 0;

    if (lenbytes == 0)
        return 1;

    pkt->subs->pwritten = lenbytes;
    pkt->subs->lenbytes = lenbytes;

    if (!WPACKET_allocate_bytes(pkt, lenbytes, &lenchars)) {
        OPENSSL_free(pkt->subs);
        pkt->subs = NULL;
        return 0;
    }
    pkt->subs->packet_len = 0;

    return 1;
}

int WPACKET_init_static_len(WPACKET *pkt, unsigned char *buf, size_t len,
                            size_t lenbytes)
{
    size_t max = maxmaxsize(lenbytes);

    /* Internal API, so should not fail */
    if (!ossl_assert(buf != NULL && len > 0))
        return 0;

    pkt->staticbuf = buf;
    pkt->buf = NULL;
    pkt->maxsize = (max < len) ? max : len;
    pkt->endfirst = 0;

    return wpacket_intern_init_len(pkt, lenbytes);
}

int WPACKET_init_der(WPACKET *pkt, unsigned char *buf, size_t len)
{
    /* Internal API, so should not fail */
    if (!ossl_assert(buf != NULL && len > 0))
        return 0;

    pkt->staticbuf = buf;
    pkt->buf = NULL;
    pkt->maxsize = len;
    pkt->endfirst = 1;

    return wpacket_intern_init_len(pkt, 0);
}

int WPACKET_init_len(WPACKET *pkt, BUF_MEM *buf, size_t lenbytes)
{
    /* Internal API, so should not fail */
    if (!ossl_assert(buf != NULL))
        return 0;

    pkt->staticbuf = NULL;
    pkt->buf = buf;
    pkt->maxsize = maxmaxsize(lenbytes);
    pkt->endfirst = 0;

    return wpacket_intern_init_len(pkt, lenbytes);
}

int WPACKET_init(WPACKET *pkt, BUF_MEM *buf)
{
    return WPACKET_init_len(pkt, buf, 0);
}

int WPACKET_init_null(WPACKET *pkt, size_t lenbytes)
{
    pkt->staticbuf = NULL;
    pkt->buf = NULL;
    pkt->maxsize = maxmaxsize(lenbytes);
    pkt->endfirst = 0;

    return wpacket_intern_init_len(pkt, 0);
}

int WPACKET_init_null_der(WPACKET *pkt)
{
    pkt->staticbuf = NULL;
    pkt->buf = NULL;
    pkt->maxsize = SIZE_MAX;
    pkt->endfirst = 1;

    return wpacket_intern_init_len(pkt, 0);
}

int WPACKET_set_flags(WPACKET *pkt, unsigned int flags)
{
    /* Internal API, so should not fail */
    if (!ossl_assert(pkt->subs != NULL))
        return 0;

    pkt->subs->flags = flags;

    return 1;
}

/* Store the |value| of length |len| at location |data| */
static int put_value(unsigned char *data, uint64_t value, size_t len)
{
    if (data == NULL)
        return 1;

    for (data += len - 1; len > 0; len--) {
        *data = (unsigned char)(value & 0xff);
        data--;
        value >>= 8;
    }

    /* Check whether we could fit the value in the assigned number of bytes */
    if (value > 0)
        return 0;

    return 1;
}

#if !defined OPENSSL_NO_QUIC && !defined FIPS_MODULE
static int put_quic_value(unsigned char *data, size_t value, size_t len)
{
    if (data == NULL)
        return 1;

    /* Value too large for field. */
    if (ossl_quic_vlint_encode_len(value) > len)
        return 0;

    ossl_quic_vlint_encode_n(data, value, len);
    return 1;
}
#endif

/*
 * Internal helper function used by WPACKET_close(), WPACKET_finish() and
 * WPACKET_fill_lengths() to close a sub-packet and write out its length if
 * necessary. If |doclose| is 0 then it goes through the motions of closing
 * (i.e. it fills in all the lengths), but doesn't actually close anything.
 */
static int wpacket_intern_close(WPACKET *pkt, WPACKET_SUB *sub, int doclose)
{
    size_t packlen = pkt->written - sub->pwritten;

    if (packlen == 0
            && (sub->flags & WPACKET_FLAGS_NON_ZERO_LENGTH) != 0)
        return 0;

    if (packlen == 0
            && sub->flags & WPACKET_FLAGS_ABANDON_ON_ZERO_LENGTH) {
        /* We can't handle this case. Return an error */
        if (!doclose)
            return 0;

        /* Deallocate any bytes allocated for the length of the WPACKET */
        if ((pkt->curr - sub->lenbytes) == sub->packet_len) {
            pkt->written -= sub->lenbytes;
            pkt->curr -= sub->lenbytes;
        }

        /* Don't write out the packet length */
        sub->packet_len = 0;
        sub->lenbytes = 0;
    }

    /* Write out the WPACKET length if needed */
    if (sub->lenbytes > 0) {
        unsigned char *buf = GETBUF(pkt);

        if (buf != NULL) {
#if !defined OPENSSL_NO_QUIC && !defined FIPS_MODULE
            if ((sub->flags & WPACKET_FLAGS_QUIC_VLINT) == 0) {
                if (!put_value(&buf[sub->packet_len], packlen, sub->lenbytes))
                    return 0;
            } else {
                if (!put_quic_value(&buf[sub->packet_len], packlen, sub->lenbytes))
                    return 0;
            }
#else
            if (!put_value(&buf[sub->packet_len], packlen, sub->lenbytes))
                return 0;
#endif
        }
    } else if (pkt->endfirst && sub->parent != NULL
               && (packlen != 0
                   || (sub->flags
                       & WPACKET_FLAGS_ABANDON_ON_ZERO_LENGTH) == 0)) {
        size_t tmplen = packlen;
        size_t numlenbytes = 1;

        while ((tmplen = tmplen >> 8) > 0)
            numlenbytes++;
        if (!WPACKET_put_bytes__(pkt, packlen, numlenbytes))
            return 0;
        if (packlen > 0x7f) {
            numlenbytes |= 0x80;
            if (!WPACKET_put_bytes_u8(pkt, numlenbytes))
                return 0;
        }
    }

    if (doclose) {
        pkt->subs = sub->parent;
        OPENSSL_free(sub);
    }

    return 1;
}

int WPACKET_fill_lengths(WPACKET *pkt)
{
    WPACKET_SUB *sub;

    if (!ossl_assert(pkt->subs != NULL))
        return 0;

    for (sub = pkt->subs; sub != NULL; sub = sub->parent) {
        if (!wpacket_intern_close(pkt, sub, 0))
            return 0;
    }

    return 1;
}

int WPACKET_close(WPACKET *pkt)
{
    /*
     * Internal API, so should not fail - but we do negative testing of this
     * so no assert (otherwise the tests fail)
     */
    if (pkt->subs == NULL || pkt->subs->parent == NULL)
        return 0;

    return wpacket_intern_close(pkt, pkt->subs, 1);
}

int WPACKET_finish(WPACKET *pkt)
{
    int ret;

    /*
     * Internal API, so should not fail - but we do negative testing of this
     * so no assert (otherwise the tests fail)
     */
    if (pkt->subs == NULL || pkt->subs->parent != NULL)
        return 0;

    ret = wpacket_intern_close(pkt, pkt->subs, 1);
    if (ret) {
        OPENSSL_free(pkt->subs);
        pkt->subs = NULL;
    }

    return ret;
}

int WPACKET_start_sub_packet_len__(WPACKET *pkt, size_t lenbytes)
{
    WPACKET_SUB *sub;
    unsigned char *lenchars;

    /* Internal API, so should not fail */
    if (!ossl_assert(pkt->subs != NULL))
        return 0;

    /* We don't support lenbytes greater than 0 when doing endfirst writing */
    if (lenbytes > 0 && pkt->endfirst)
        return 0;

    if ((sub = OPENSSL_zalloc(sizeof(*sub))) == NULL)
        return 0;

    sub->parent = pkt->subs;
    pkt->subs = sub;
    sub->pwritten = pkt->written + lenbytes;
    sub->lenbytes = lenbytes;

    if (lenbytes == 0) {
        sub->packet_len = 0;
        return 1;
    }

    sub->packet_len = pkt->written;

    if (!WPACKET_allocate_bytes(pkt, lenbytes, &lenchars))
        return 0;

    return 1;
}

int WPACKET_start_sub_packet(WPACKET *pkt)
{
    return WPACKET_start_sub_packet_len__(pkt, 0);
}

int WPACKET_put_bytes__(WPACKET *pkt, uint64_t val, size_t size)
{
    unsigned char *data;

    /* Internal API, so should not fail */
    if (!ossl_assert(size <= sizeof(uint64_t))
            || !WPACKET_allocate_bytes(pkt, size, &data)
            || !put_value(data, val, size))
        return 0;

    return 1;
}

int WPACKET_set_max_size(WPACKET *pkt, size_t maxsize)
{
    WPACKET_SUB *sub;
    size_t lenbytes;

    /* Internal API, so should not fail */
    if (!ossl_assert(pkt->subs != NULL))
        return 0;

    /* Find the WPACKET_SUB for the top level */
    for (sub = pkt->subs; sub->parent != NULL; sub = sub->parent)
        continue;

    lenbytes = sub->lenbytes;
    if (lenbytes == 0)
        lenbytes = sizeof(pkt->maxsize);

    if (maxmaxsize(lenbytes) < maxsize || maxsize < pkt->written)
        return 0;

    pkt->maxsize = maxsize;

    return 1;
}

int WPACKET_memset(WPACKET *pkt, int ch, size_t len)
{
    unsigned char *dest;

    if (len == 0)
        return 1;

    if (!WPACKET_allocate_bytes(pkt, len, &dest))
        return 0;

    if (dest != NULL)
        memset(dest, ch, len);

    return 1;
}

int WPACKET_memcpy(WPACKET *pkt, const void *src, size_t len)
{
    unsigned char *dest;

    if (len == 0)
        return 1;

    if (!WPACKET_allocate_bytes(pkt, len, &dest))
        return 0;

    if (dest != NULL)
        memcpy(dest, src, len);

    return 1;
}

int WPACKET_sub_memcpy__(WPACKET *pkt, const void *src, size_t len,
                         size_t lenbytes)
{
    if (!WPACKET_start_sub_packet_len__(pkt, lenbytes)
            || !WPACKET_memcpy(pkt, src, len)
            || !WPACKET_close(pkt))
        return 0;

    return 1;
}

int WPACKET_get_total_written(WPACKET *pkt, size_t *written)
{
    /* Internal API, so should not fail */
    if (!ossl_assert(written != NULL))
        return 0;

    *written = pkt->written;

    return 1;
}

int WPACKET_get_length(WPACKET *pkt, size_t *len)
{
    /* Internal API, so should not fail */
    if (!ossl_assert(pkt->subs != NULL && len != NULL))
        return 0;

    *len = pkt->written - pkt->subs->pwritten;

    return 1;
}

unsigned char *WPACKET_get_curr(WPACKET *pkt)
{
    unsigned char *buf = GETBUF(pkt);

    if (buf == NULL)
        return NULL;

    if (pkt->endfirst)
        return buf + pkt->maxsize - pkt->curr;

    return buf + pkt->curr;
}

int WPACKET_is_null_buf(WPACKET *pkt)
{
    return pkt->buf == NULL && pkt->staticbuf == NULL;
}

void WPACKET_cleanup(WPACKET *pkt)
{
    WPACKET_SUB *sub, *parent;

    for (sub = pkt->subs; sub != NULL; sub = parent) {
        parent = sub->parent;
        OPENSSL_free(sub);
    }
    pkt->subs = NULL;
}

#if !defined OPENSSL_NO_QUIC && !defined FIPS_MODULE

int WPACKET_start_quic_sub_packet_bound(WPACKET *pkt, size_t max_len)
{
    size_t enclen = ossl_quic_vlint_encode_len(max_len);

    if (enclen == 0)
        return 0;

    if (WPACKET_start_sub_packet_len__(pkt, enclen) == 0)
        return 0;

    pkt->subs->flags |= WPACKET_FLAGS_QUIC_VLINT;
    return 1;
}

int WPACKET_start_quic_sub_packet(WPACKET *pkt)
{
    /*
     * Assume no (sub)packet will exceed 4GiB, thus the 8-byte encoding need not
     * be used.
     */
    return WPACKET_start_quic_sub_packet_bound(pkt, OSSL_QUIC_VLINT_4B_MIN);
}

int WPACKET_quic_sub_allocate_bytes(WPACKET *pkt, size_t len, unsigned char **allocbytes)
{
    if (!WPACKET_start_quic_sub_packet_bound(pkt, len)
            || !WPACKET_allocate_bytes(pkt, len, allocbytes)
            || !WPACKET_close(pkt))
        return 0;

    return 1;
}

/*
 * Write a QUIC variable-length integer to the packet.
 */
int WPACKET_quic_write_vlint(WPACKET *pkt, uint64_t v)
{
    unsigned char *b = NULL;
    size_t enclen = ossl_quic_vlint_encode_len(v);

    if (enclen == 0)
        return 0;

    if (WPACKET_allocate_bytes(pkt, enclen, &b) == 0)
        return 0;

    ossl_quic_vlint_encode(b, v);
    return 1;
}

#endif

Coverage Report

Created: 2025-06-13 06:57

Line	Count	Source (jump to first uncovered line)
1		/*
2		* Copyright 2015-2023 The OpenSSL Project Authors. All Rights Reserved.
3		*
4		* Licensed under the Apache License 2.0 (the "License"). You may not use
5		* this file except in compliance with the License. You can obtain a copy
6		* in the file LICENSE in the source distribution or at
7		* https://www.openssl.org/source/license.html
8		*/
9
10		#include "internal/cryptlib.h"
11		#include "internal/packet.h"
12		#if !defined OPENSSL_NO_QUIC && !defined FIPS_MODULE
13		# include "internal/packet_quic.h"
14		#endif
15		#include <openssl/err.h>
16
17	0	#define DEFAULT_BUF_SIZE 256
18
19		int WPACKET_allocate_bytes(WPACKET pkt, size_t len, unsigned char *allocbytes)
20	21.0k	{
21	21.0k	if (!WPACKET_reserve_bytes(pkt, len, allocbytes))
22	0	return 0;
23
24	21.0k	pkt->written += len;
25	21.0k	pkt->curr += len;
26	21.0k	return 1;
27	21.0k	}
28
29		int WPACKET_sub_allocate_bytes__(WPACKET *pkt, size_t len,
30		unsigned char **allocbytes, size_t lenbytes)
31	0	{
32	0	if (!WPACKET_start_sub_packet_len__(pkt, lenbytes)
33	0	\|\| !WPACKET_allocate_bytes(pkt, len, allocbytes)
34	0	\|\| !WPACKET_close(pkt))
35	0	return 0;
36
37	0	return 1;
38	0	}
39
40	21.0k	#define GETBUF(p) (((p)->staticbuf != NULL) \
41	21.0k	? (p)->staticbuf \
42	21.0k	: ((p)->buf != NULL \
43	21.0k	? (unsigned char *)(p)->buf->data \
44	21.0k	: NULL))
45
46		int WPACKET_reserve_bytes(WPACKET pkt, size_t len, unsigned char *allocbytes)
47	21.0k	{
48		/* Internal API, so should not fail */
49	21.0k	if (!ossl_assert(pkt->subs != NULL && len != 0))
50	0	return 0;
51
52	21.0k	if (pkt->maxsize - pkt->written < len)
53	0	return 0;
54
55	21.0k	if (pkt->buf != NULL && (pkt->buf->length - pkt->written < len)) {
56	0	size_t newlen;
57	0	size_t reflen;
58
59	0	reflen = (len > pkt->buf->length) ? len : pkt->buf->length;
60
61	0	if (reflen > SIZE_MAX / 2) {
62	0	newlen = SIZE_MAX;
63	0	} else {
64	0	newlen = reflen * 2;
65	0	if (newlen < DEFAULT_BUF_SIZE)
66	0	newlen = DEFAULT_BUF_SIZE;
67	0	}
68	0	if (BUF_MEM_grow(pkt->buf, newlen) == 0)
69	0	return 0;
70	0	}
71	21.0k	if (allocbytes != NULL) {
72	21.0k	*allocbytes = WPACKET_get_curr(pkt);
73	21.0k	if (pkt->endfirst && *allocbytes != NULL)
74	0	*allocbytes -= len;
75	21.0k	}
76
77	21.0k	return 1;
78	21.0k	}
79
80		int WPACKET_sub_reserve_bytes__(WPACKET *pkt, size_t len,
81		unsigned char **allocbytes, size_t lenbytes)
82	0	{
83	0	if (pkt->endfirst && lenbytes > 0)
84	0	return 0;
85
86	0	if (!WPACKET_reserve_bytes(pkt, lenbytes + len, allocbytes))
87	0	return 0;
88
89	0	if (*allocbytes != NULL)
90	0	*allocbytes += lenbytes;
91
92	0	return 1;
93	0	}
94
95		static size_t maxmaxsize(size_t lenbytes)
96	2.59k	{
97	2.59k	if (lenbytes >= sizeof(size_t) \|\| lenbytes == 0)
98	2.59k	return SIZE_MAX;
99
100	0	return ((size_t)1 << (lenbytes * 8)) - 1 + lenbytes;
101	2.59k	}
102
103		static int wpacket_intern_init_len(WPACKET *pkt, size_t lenbytes)
104	2.59k	{
105	2.59k	unsigned char *lenchars;
106
107	2.59k	pkt->curr = 0;
108	2.59k	pkt->written = 0;
109
110	2.59k	if ((pkt->subs = OPENSSL_zalloc(sizeof(*pkt->subs))) == NULL)
111	0	return 0;
112
113	2.59k	if (lenbytes == 0)
114	2.59k	return 1;
115
116	0	pkt->subs->pwritten = lenbytes;
117	0	pkt->subs->lenbytes = lenbytes;
118
119	0	if (!WPACKET_allocate_bytes(pkt, lenbytes, &lenchars)) {
120	0	OPENSSL_free(pkt->subs);
121	0	pkt->subs = NULL;
122	0	return 0;
123	0	}
124	0	pkt->subs->packet_len = 0;
125
126	0	return 1;
127	0	}
128
129		int WPACKET_init_static_len(WPACKET pkt, unsigned char buf, size_t len,
130		size_t lenbytes)
131	0	{
132	0	size_t max = maxmaxsize(lenbytes);
133
134		/* Internal API, so should not fail */
135	0	if (!ossl_assert(buf != NULL && len > 0))
136	0	return 0;
137
138	0	pkt->staticbuf = buf;
139	0	pkt->buf = NULL;
140	0	pkt->maxsize = (max < len) ? max : len;
141	0	pkt->endfirst = 0;
142
143	0	return wpacket_intern_init_len(pkt, lenbytes);
144	0	}
145
146		int WPACKET_init_der(WPACKET pkt, unsigned char buf, size_t len)
147	0	{
148		/* Internal API, so should not fail */
149	0	if (!ossl_assert(buf != NULL && len > 0))
150	0	return 0;
151
152	0	pkt->staticbuf = buf;
153	0	pkt->buf = NULL;
154	0	pkt->maxsize = len;
155	0	pkt->endfirst = 1;
156
157	0	return wpacket_intern_init_len(pkt, 0);
158	0	}
159
160		int WPACKET_init_len(WPACKET pkt, BUF_MEM buf, size_t lenbytes)
161	0	{
162		/* Internal API, so should not fail */
163	0	if (!ossl_assert(buf != NULL))
164	0	return 0;
165
166	0	pkt->staticbuf = NULL;
167	0	pkt->buf = buf;
168	0	pkt->maxsize = maxmaxsize(lenbytes);
169	0	pkt->endfirst = 0;
170
171	0	return wpacket_intern_init_len(pkt, lenbytes);
172	0	}
173
174		int WPACKET_init(WPACKET pkt, BUF_MEM buf)
175	0	{
176	0	return WPACKET_init_len(pkt, buf, 0);
177	0	}
178
179		int WPACKET_init_null(WPACKET *pkt, size_t lenbytes)
180	2.59k	{
181	2.59k	pkt->staticbuf = NULL;
182	2.59k	pkt->buf = NULL;
183	2.59k	pkt->maxsize = maxmaxsize(lenbytes);
184	2.59k	pkt->endfirst = 0;
185
186	2.59k	return wpacket_intern_init_len(pkt, 0);
187	2.59k	}
188
189		int WPACKET_init_null_der(WPACKET *pkt)
190	0	{
191	0	pkt->staticbuf = NULL;
192	0	pkt->buf = NULL;
193	0	pkt->maxsize = SIZE_MAX;
194	0	pkt->endfirst = 1;
195
196	0	return wpacket_intern_init_len(pkt, 0);
197	0	}
198
199		int WPACKET_set_flags(WPACKET *pkt, unsigned int flags)
200	0	{
201		/* Internal API, so should not fail */
202	0	if (!ossl_assert(pkt->subs != NULL))
203	0	return 0;
204
205	0	pkt->subs->flags = flags;
206
207	0	return 1;
208	0	}
209
210		/* Store the \|value\| of length \|len\| at location \|data\| */
211		static int put_value(unsigned char *data, uint64_t value, size_t len)
212	15.8k	{
213	15.8k	if (data == NULL)
214	15.8k	return 1;
215
216	0	for (data += len - 1; len > 0; len--) {
217	0	*data = (unsigned char)(value & 0xff);
218	0	data--;
219	0	value >>= 8;
220	0	}
221
222		/* Check whether we could fit the value in the assigned number of bytes */
223	0	if (value > 0)
224	0	return 0;
225
226	0	return 1;
227	0	}
228
229		#if !defined OPENSSL_NO_QUIC && !defined FIPS_MODULE
230		static int put_quic_value(unsigned char *data, size_t value, size_t len)
231	0	{
232	0	if (data == NULL)
233	0	return 1;
234
235		/* Value too large for field. */
236	0	if (ossl_quic_vlint_encode_len(value) > len)
237	0	return 0;
238
239	0	ossl_quic_vlint_encode_n(data, value, len);
240	0	return 1;
241	0	}
242		#endif
243
244		/*
245		* Internal helper function used by WPACKET_close(), WPACKET_finish() and
246		* WPACKET_fill_lengths() to close a sub-packet and write out its length if
247		* necessary. If \|doclose\| is 0 then it goes through the motions of closing
248		* (i.e. it fills in all the lengths), but doesn't actually close anything.
249		*/
250		static int wpacket_intern_close(WPACKET pkt, WPACKET_SUB sub, int doclose)
251	10.3k	{
252	10.3k	size_t packlen = pkt->written - sub->pwritten;
253
254	10.3k	if (packlen == 0
255	10.3k	&& (sub->flags & WPACKET_FLAGS_NON_ZERO_LENGTH) != 0)
256	0	return 0;
257
258	10.3k	if (packlen == 0
259	10.3k	&& sub->flags & WPACKET_FLAGS_ABANDON_ON_ZERO_LENGTH) {
260		/* We can't handle this case. Return an error */
261	0	if (!doclose)
262	0	return 0;
263
264		/* Deallocate any bytes allocated for the length of the WPACKET */
265	0	if ((pkt->curr - sub->lenbytes) == sub->packet_len) {
266	0	pkt->written -= sub->lenbytes;
267	0	pkt->curr -= sub->lenbytes;
268	0	}
269
270		/* Don't write out the packet length */
271	0	sub->packet_len = 0;
272	0	sub->lenbytes = 0;
273	0	}
274
275		/* Write out the WPACKET length if needed */
276	10.3k	if (sub->lenbytes > 0) {
277	0	unsigned char *buf = GETBUF(pkt);
278
279	0	if (buf != NULL) {
280	0	#if !defined OPENSSL_NO_QUIC && !defined FIPS_MODULE
281	0	if ((sub->flags & WPACKET_FLAGS_QUIC_VLINT) == 0) {
282	0	if (!put_value(&buf[sub->packet_len], packlen, sub->lenbytes))
283	0	return 0;
284	0	} else {
285	0	if (!put_quic_value(&buf[sub->packet_len], packlen, sub->lenbytes))
286	0	return 0;
287	0	}
288		#else
289		if (!put_value(&buf[sub->packet_len], packlen, sub->lenbytes))
290		return 0;
291		#endif
292	0	}
293	10.3k	} else if (pkt->endfirst && sub->parent != NULL
294	10.3k	&& (packlen != 0
295	0	\|\| (sub->flags
296	0	& WPACKET_FLAGS_ABANDON_ON_ZERO_LENGTH) == 0)) {
297	0	size_t tmplen = packlen;
298	0	size_t numlenbytes = 1;
299
300	0	while ((tmplen = tmplen >> 8) > 0)
301	0	numlenbytes++;
302	0	if (!WPACKET_put_bytes__(pkt, packlen, numlenbytes))
303	0	return 0;
304	0	if (packlen > 0x7f) {
305	0	numlenbytes \|= 0x80;
306	0	if (!WPACKET_put_bytes_u8(pkt, numlenbytes))
307	0	return 0;
308	0	}
309	0	}
310
311	10.3k	if (doclose) {
312	10.3k	pkt->subs = sub->parent;
313	10.3k	OPENSSL_free(sub);
314	10.3k	}
315
316	10.3k	return 1;
317	10.3k	}
318
319		int WPACKET_fill_lengths(WPACKET *pkt)
320	0	{
321	0	WPACKET_SUB *sub;
322
323	0	if (!ossl_assert(pkt->subs != NULL))
324	0	return 0;
325
326	0	for (sub = pkt->subs; sub != NULL; sub = sub->parent) {
327	0	if (!wpacket_intern_close(pkt, sub, 0))
328	0	return 0;
329	0	}
330
331	0	return 1;
332	0	}
333
334		int WPACKET_close(WPACKET *pkt)
335	7.75k	{
336		/*
337		* Internal API, so should not fail - but we do negative testing of this
338		* so no assert (otherwise the tests fail)
339		*/
340	7.75k	if (pkt->subs == NULL \|\| pkt->subs->parent == NULL)
341	0	return 0;
342
343	7.75k	return wpacket_intern_close(pkt, pkt->subs, 1);
344	7.75k	}
345
346		int WPACKET_finish(WPACKET *pkt)
347	2.57k	{
348	2.57k	int ret;
349
350		/*
351		* Internal API, so should not fail - but we do negative testing of this
352		* so no assert (otherwise the tests fail)
353		*/
354	2.57k	if (pkt->subs == NULL \|\| pkt->subs->parent != NULL)
355	0	return 0;
356
357	2.57k	ret = wpacket_intern_close(pkt, pkt->subs, 1);
358	2.57k	if (ret) {
359	2.57k	OPENSSL_free(pkt->subs);
360	2.57k	pkt->subs = NULL;
361	2.57k	}
362
363	2.57k	return ret;
364	2.57k	}
365
366		int WPACKET_start_sub_packet_len__(WPACKET *pkt, size_t lenbytes)
367	7.76k	{
368	7.76k	WPACKET_SUB *sub;
369	7.76k	unsigned char *lenchars;
370
371		/* Internal API, so should not fail */
372	7.76k	if (!ossl_assert(pkt->subs != NULL))
373	0	return 0;
374
375		/* We don't support lenbytes greater than 0 when doing endfirst writing */
376	7.76k	if (lenbytes > 0 && pkt->endfirst)
377	0	return 0;
378
379	7.76k	if ((sub = OPENSSL_zalloc(sizeof(*sub))) == NULL)
380	0	return 0;
381
382	7.76k	sub->parent = pkt->subs;
383	7.76k	pkt->subs = sub;
384	7.76k	sub->pwritten = pkt->written + lenbytes;
385	7.76k	sub->lenbytes = lenbytes;
386
387	7.76k	if (lenbytes == 0) {
388	7.76k	sub->packet_len = 0;
389	7.76k	return 1;
390	7.76k	}
391
392	0	sub->packet_len = pkt->written;
393
394	0	if (!WPACKET_allocate_bytes(pkt, lenbytes, &lenchars))
395	0	return 0;
396
397	0	return 1;
398	0	}
399
400		int WPACKET_start_sub_packet(WPACKET *pkt)
401	7.76k	{
402	7.76k	return WPACKET_start_sub_packet_len__(pkt, 0);
403	7.76k	}
404
405		int WPACKET_put_bytes__(WPACKET *pkt, uint64_t val, size_t size)
406	15.8k	{
407	15.8k	unsigned char *data;
408
409		/* Internal API, so should not fail */
410	15.8k	if (!ossl_assert(size <= sizeof(uint64_t))
411	15.8k	\|\| !WPACKET_allocate_bytes(pkt, size, &data)
412	15.8k	\|\| !put_value(data, val, size))
413	0	return 0;
414
415	15.8k	return 1;
416	15.8k	}
417
418		int WPACKET_set_max_size(WPACKET *pkt, size_t maxsize)
419	0	{
420	0	WPACKET_SUB *sub;
421	0	size_t lenbytes;
422
423		/* Internal API, so should not fail */
424	0	if (!ossl_assert(pkt->subs != NULL))
425	0	return 0;
426
427		/* Find the WPACKET_SUB for the top level */
428	0	for (sub = pkt->subs; sub->parent != NULL; sub = sub->parent)
429	0	continue;
430
431	0	lenbytes = sub->lenbytes;
432	0	if (lenbytes == 0)
433	0	lenbytes = sizeof(pkt->maxsize);
434
435	0	if (maxmaxsize(lenbytes) < maxsize \|\| maxsize < pkt->written)
436	0	return 0;
437
438	0	pkt->maxsize = maxsize;
439
440	0	return 1;
441	0	}
442
443		int WPACKET_memset(WPACKET *pkt, int ch, size_t len)
444	0	{
445	0	unsigned char *dest;
446
447	0	if (len == 0)
448	0	return 1;
449
450	0	if (!WPACKET_allocate_bytes(pkt, len, &dest))
451	0	return 0;
452
453	0	if (dest != NULL)
454	0	memset(dest, ch, len);
455
456	0	return 1;
457	0	}
458
459		int WPACKET_memcpy(WPACKET pkt, const void src, size_t len)
460	0	{
461	0	unsigned char *dest;
462
463	0	if (len == 0)
464	0	return 1;
465
466	0	if (!WPACKET_allocate_bytes(pkt, len, &dest))
467	0	return 0;
468
469	0	if (dest != NULL)
470	0	memcpy(dest, src, len);
471
472	0	return 1;
473	0	}
474
475		int WPACKET_sub_memcpy__(WPACKET pkt, const void src, size_t len,
476		size_t lenbytes)
477	0	{
478	0	if (!WPACKET_start_sub_packet_len__(pkt, lenbytes)
479	0	\|\| !WPACKET_memcpy(pkt, src, len)
480	0	\|\| !WPACKET_close(pkt))
481	0	return 0;
482
483	0	return 1;
484	0	}
485
486		int WPACKET_get_total_written(WPACKET pkt, size_t written)
487	2.57k	{
488		/* Internal API, so should not fail */
489	2.57k	if (!ossl_assert(written != NULL))
490	0	return 0;
491
492	2.57k	*written = pkt->written;
493
494	2.57k	return 1;
495	2.57k	}
496
497		int WPACKET_get_length(WPACKET pkt, size_t len)
498	2.58k	{
499		/* Internal API, so should not fail */
500	2.58k	if (!ossl_assert(pkt->subs != NULL && len != NULL))
501	0	return 0;
502
503	2.58k	*len = pkt->written - pkt->subs->pwritten;
504
505	2.58k	return 1;
506	2.58k	}
507
508		unsigned char WPACKET_get_curr(WPACKET pkt)
509	21.0k	{
510	21.0k	unsigned char *buf = GETBUF(pkt);
511
512	21.0k	if (buf == NULL)
513	21.0k	return NULL;
514
515	0	if (pkt->endfirst)
516	0	return buf + pkt->maxsize - pkt->curr;
517
518	0	return buf + pkt->curr;
519	0	}
520
521		int WPACKET_is_null_buf(WPACKET *pkt)
522	2.59k	{
523	2.59k	return pkt->buf == NULL && pkt->staticbuf == NULL;
524	2.59k	}
525
526		void WPACKET_cleanup(WPACKET *pkt)
527	16	{
528	16	WPACKET_SUB sub, parent;
529
530	50	for (sub = pkt->subs; sub != NULL; sub = parent) {
531	34	parent = sub->parent;
532	34	OPENSSL_free(sub);
533	34	}
534	16	pkt->subs = NULL;
535	16	}
536
537		#if !defined OPENSSL_NO_QUIC && !defined FIPS_MODULE
538
539		int WPACKET_start_quic_sub_packet_bound(WPACKET *pkt, size_t max_len)
540	0	{
541	0	size_t enclen = ossl_quic_vlint_encode_len(max_len);
542
543	0	if (enclen == 0)
544	0	return 0;
545
546	0	if (WPACKET_start_sub_packet_len__(pkt, enclen) == 0)
547	0	return 0;
548
549	0	pkt->subs->flags \|= WPACKET_FLAGS_QUIC_VLINT;
550	0	return 1;
551	0	}
552
553		int WPACKET_start_quic_sub_packet(WPACKET *pkt)
554	0	{
555		/*
556		* Assume no (sub)packet will exceed 4GiB, thus the 8-byte encoding need not
557		* be used.
558		*/
559	0	return WPACKET_start_quic_sub_packet_bound(pkt, OSSL_QUIC_VLINT_4B_MIN);
560	0	}
561
562		int WPACKET_quic_sub_allocate_bytes(WPACKET pkt, size_t len, unsigned char *allocbytes)
563	0	{
564	0	if (!WPACKET_start_quic_sub_packet_bound(pkt, len)
565	0	\|\| !WPACKET_allocate_bytes(pkt, len, allocbytes)
566	0	\|\| !WPACKET_close(pkt))
567	0	return 0;
568
569	0	return 1;
570	0	}
571
572		/*
573		* Write a QUIC variable-length integer to the packet.
574		*/
575		int WPACKET_quic_write_vlint(WPACKET *pkt, uint64_t v)
576	0	{
577	0	unsigned char *b = NULL;
578	0	size_t enclen = ossl_quic_vlint_encode_len(v);
579
580	0	if (enclen == 0)
581	0	return 0;
582
583	0	if (WPACKET_allocate_bytes(pkt, enclen, &b) == 0)
584	0	return 0;
585
586	0	ossl_quic_vlint_encode(b, v);
587	0	return 1;
588	0	}
589
590		#endif