/src/openssl30/crypto/packet.c

Source
/*
 * Copyright 2015-2022 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"
#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) {
        ERR_raise(ERR_LIB_CRYPTO, ERR_R_MALLOC_FAILURE);
        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;
}


/*
 * 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
                && !put_value(&buf[sub->packet_len], packlen,
                              sub->lenbytes))
            return 0;
    } 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) {
        ERR_raise(ERR_LIB_CRYPTO, ERR_R_MALLOC_FAILURE);
        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;
}

Coverage Report

Created: 2025-11-16 06:40

Line	Count	Source
1		/*
2		* Copyright 2015-2022 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		#include <openssl/err.h>
13
14	200k	#define DEFAULT_BUF_SIZE 256
15
16		int WPACKET_allocate_bytes(WPACKET pkt, size_t len, unsigned char *allocbytes)
17	141M	{
18	141M	if (!WPACKET_reserve_bytes(pkt, len, allocbytes))
19	130k	return 0;
20
21	141M	pkt->written += len;
22	141M	pkt->curr += len;
23	141M	return 1;
24	141M	}
25
26		int WPACKET_sub_allocate_bytes__(WPACKET *pkt, size_t len,
27		unsigned char **allocbytes, size_t lenbytes)
28	32.4k	{
29	32.4k	if (!WPACKET_start_sub_packet_len__(pkt, lenbytes)
30	32.4k	\|\| !WPACKET_allocate_bytes(pkt, len, allocbytes)
31	32.4k	\|\| !WPACKET_close(pkt))
32	0	return 0;
33
34	32.4k	return 1;
35	32.4k	}
36
37	145M	#define GETBUF(p) (((p)->staticbuf != NULL) \
38	145M	? (p)->staticbuf \
39	145M	: ((p)->buf != NULL \
40	21.6M	? (unsigned char *)(p)->buf->data \
41	21.6M	: NULL))
42
43		int WPACKET_reserve_bytes(WPACKET pkt, size_t len, unsigned char *allocbytes)
44	143M	{
45		/* Internal API, so should not fail */
46	143M	if (!ossl_assert(pkt->subs != NULL && len != 0))
47	0	return 0;
48
49	143M	if (pkt->maxsize - pkt->written < len)
50	130k	return 0;
51
52	143M	if (pkt->buf != NULL && (pkt->buf->length - pkt->written < len)) {
53	114k	size_t newlen;
54	114k	size_t reflen;
55
56	114k	reflen = (len > pkt->buf->length) ? len : pkt->buf->length;
57
58	114k	if (reflen > SIZE_MAX / 2) {
59	0	newlen = SIZE_MAX;
60	114k	} else {
61	114k	newlen = reflen * 2;
62	114k	if (newlen < DEFAULT_BUF_SIZE)
63	86.0k	newlen = DEFAULT_BUF_SIZE;
64	114k	}
65	114k	if (BUF_MEM_grow(pkt->buf, newlen) == 0)
66	0	return 0;
67	114k	}
68	143M	if (allocbytes != NULL) {
69	141M	*allocbytes = WPACKET_get_curr(pkt);
70	141M	if (pkt->endfirst && *allocbytes != NULL)
71	69.8k	*allocbytes -= len;
72	141M	}
73
74	143M	return 1;
75	143M	}
76
77		int WPACKET_sub_reserve_bytes__(WPACKET *pkt, size_t len,
78		unsigned char **allocbytes, size_t lenbytes)
79	8.69k	{
80	8.69k	if (pkt->endfirst && lenbytes > 0)
81	0	return 0;
82
83	8.69k	if (!WPACKET_reserve_bytes(pkt, lenbytes + len, allocbytes))
84	0	return 0;
85
86	8.69k	if (*allocbytes != NULL)
87	8.69k	*allocbytes += lenbytes;
88
89	8.69k	return 1;
90	8.69k	}
91
92		static size_t maxmaxsize(size_t lenbytes)
93	26.0M	{
94	26.0M	if (lenbytes >= sizeof(size_t) \|\| lenbytes == 0)
95	26.0M	return SIZE_MAX;
96
97	0	return ((size_t)1 << (lenbytes * 8)) - 1 + lenbytes;
98	26.0M	}
99
100		static int wpacket_intern_init_len(WPACKET *pkt, size_t lenbytes)
101	15.8M	{
102	15.8M	unsigned char *lenchars;
103
104	15.8M	pkt->curr = 0;
105	15.8M	pkt->written = 0;
106
107	15.8M	if ((pkt->subs = OPENSSL_zalloc(sizeof(*pkt->subs))) == NULL) {
108	0	ERR_raise(ERR_LIB_CRYPTO, ERR_R_MALLOC_FAILURE);
109	0	return 0;
110	0	}
111
112	15.8M	if (lenbytes == 0)
113	15.8M	return 1;
114
115	0	pkt->subs->pwritten = lenbytes;
116	0	pkt->subs->lenbytes = lenbytes;
117
118	0	if (!WPACKET_allocate_bytes(pkt, lenbytes, &lenchars)) {
119	0	OPENSSL_free(pkt->subs);
120	0	pkt->subs = NULL;
121	0	return 0;
122	0	}
123	0	pkt->subs->packet_len = 0;
124
125	0	return 1;
126	0	}
127
128		int WPACKET_init_static_len(WPACKET pkt, unsigned char buf, size_t len,
129		size_t lenbytes)
130	14.9M	{
131	14.9M	size_t max = maxmaxsize(lenbytes);
132
133		/* Internal API, so should not fail */
134	14.9M	if (!ossl_assert(buf != NULL && len > 0))
135	0	return 0;
136
137	14.9M	pkt->staticbuf = buf;
138	14.9M	pkt->buf = NULL;
139	14.9M	pkt->maxsize = (max < len) ? max : len;
140	14.9M	pkt->endfirst = 0;
141
142	14.9M	return wpacket_intern_init_len(pkt, lenbytes);
143	14.9M	}
144
145		int WPACKET_init_der(WPACKET pkt, unsigned char buf, size_t len)
146	23.2k	{
147		/* Internal API, so should not fail */
148	23.2k	if (!ossl_assert(buf != NULL && len > 0))
149	0	return 0;
150
151	23.2k	pkt->staticbuf = buf;
152	23.2k	pkt->buf = NULL;
153	23.2k	pkt->maxsize = len;
154	23.2k	pkt->endfirst = 1;
155
156	23.2k	return wpacket_intern_init_len(pkt, 0);
157	23.2k	}
158
159		int WPACKET_init_len(WPACKET pkt, BUF_MEM buf, size_t lenbytes)
160	324k	{
161		/* Internal API, so should not fail */
162	324k	if (!ossl_assert(buf != NULL))
163	0	return 0;
164
165	324k	pkt->staticbuf = NULL;
166	324k	pkt->buf = buf;
167	324k	pkt->maxsize = maxmaxsize(lenbytes);
168	324k	pkt->endfirst = 0;
169
170	324k	return wpacket_intern_init_len(pkt, lenbytes);
171	324k	}
172
173		int WPACKET_init(WPACKET pkt, BUF_MEM buf)
174	319k	{
175	319k	return WPACKET_init_len(pkt, buf, 0);
176	319k	}
177
178		int WPACKET_init_null(WPACKET *pkt, size_t lenbytes)
179	516k	{
180	516k	pkt->staticbuf = NULL;
181	516k	pkt->buf = NULL;
182	516k	pkt->maxsize = maxmaxsize(lenbytes);
183	516k	pkt->endfirst = 0;
184
185	516k	return wpacket_intern_init_len(pkt, 0);
186	516k	}
187
188		int WPACKET_init_null_der(WPACKET *pkt)
189	6	{
190	6	pkt->staticbuf = NULL;
191	6	pkt->buf = NULL;
192	6	pkt->maxsize = SIZE_MAX;
193	6	pkt->endfirst = 1;
194
195	6	return wpacket_intern_init_len(pkt, 0);
196	6	}
197
198		int WPACKET_set_flags(WPACKET *pkt, unsigned int flags)
199	245k	{
200		/* Internal API, so should not fail */
201	245k	if (!ossl_assert(pkt->subs != NULL))
202	0	return 0;
203
204	245k	pkt->subs->flags = flags;
205
206	245k	return 1;
207	245k	}
208
209		/* Store the \|value\| of length \|len\| at location \|data\| */
210		static int put_value(unsigned char *data, uint64_t value, size_t len)
211	30.9M	{
212	30.9M	if (data == NULL)
213	3.09M	return 1;
214
215	78.6M	for (data += len - 1; len > 0; len--) {
216	50.8M	*data = (unsigned char)(value & 0xff);
217	50.8M	data--;
218	50.8M	value >>= 8;
219	50.8M	}
220
221		/* Check whether we could fit the value in the assigned number of bytes */
222	27.8M	if (value > 0)
223	612	return 0;
224
225	27.8M	return 1;
226	27.8M	}
227
228
229		/*
230		* Internal helper function used by WPACKET_close(), WPACKET_finish() and
231		* WPACKET_fill_lengths() to close a sub-packet and write out its length if
232		* necessary. If \|doclose\| is 0 then it goes through the motions of closing
233		* (i.e. it fills in all the lengths), but doesn't actually close anything.
234		*/
235		static int wpacket_intern_close(WPACKET pkt, WPACKET_SUB sub, int doclose)
236	597k	{
237	597k	size_t packlen = pkt->written - sub->pwritten;
238
239	597k	if (packlen == 0
240	15.7k	&& (sub->flags & WPACKET_FLAGS_NON_ZERO_LENGTH) != 0)
241	0	return 0;
242
243	597k	if (packlen == 0
244	15.7k	&& sub->flags & WPACKET_FLAGS_ABANDON_ON_ZERO_LENGTH) {
245		/* We can't handle this case. Return an error */
246	1.50k	if (!doclose)
247	0	return 0;
248
249		/* Deallocate any bytes allocated for the length of the WPACKET */
250	1.50k	if ((pkt->curr - sub->lenbytes) == sub->packet_len) {
251	1.50k	pkt->written -= sub->lenbytes;
252	1.50k	pkt->curr -= sub->lenbytes;
253	1.50k	}
254
255		/* Don't write out the packet length */
256	1.50k	sub->packet_len = 0;
257	1.50k	sub->lenbytes = 0;
258	1.50k	}
259
260		/* Write out the WPACKET length if needed */
261	597k	if (sub->lenbytes > 0) {
262	206k	unsigned char *buf = GETBUF(pkt);
263
264	206k	if (buf != NULL
265	206k	&& !put_value(&buf[sub->packet_len], packlen,
266	206k	sub->lenbytes))
267	109	return 0;
268	391k	} else if (pkt->endfirst && sub->parent != NULL
269	3.31k	&& (packlen != 0
270	2	\|\| (sub->flags
271	3.31k	& WPACKET_FLAGS_ABANDON_ON_ZERO_LENGTH) == 0)) {
272	3.31k	size_t tmplen = packlen;
273	3.31k	size_t numlenbytes = 1;
274
275	3.31k	while ((tmplen = tmplen >> 8) > 0)
276	0	numlenbytes++;
277	3.31k	if (!WPACKET_put_bytes__(pkt, packlen, numlenbytes))
278	0	return 0;
279	3.31k	if (packlen > 0x7f) {
280	0	numlenbytes \|= 0x80;
281	0	if (!WPACKET_put_bytes_u8(pkt, numlenbytes))
282	0	return 0;
283	0	}
284	3.31k	}
285
286	597k	if (doclose) {
287	597k	pkt->subs = sub->parent;
288	597k	OPENSSL_free(sub);
289	597k	}
290
291	597k	return 1;
292	597k	}
293
294		int WPACKET_fill_lengths(WPACKET *pkt)
295	0	{
296	0	WPACKET_SUB *sub;
297
298	0	if (!ossl_assert(pkt->subs != NULL))
299	0	return 0;
300
301	0	for (sub = pkt->subs; sub != NULL; sub = sub->parent) {
302	0	if (!wpacket_intern_close(pkt, sub, 0))
303	0	return 0;
304	0	}
305
306	0	return 1;
307	0	}
308
309		int WPACKET_close(WPACKET *pkt)
310	7.14M	{
311		/*
312		* Internal API, so should not fail - but we do negative testing of this
313		* so no assert (otherwise the tests fail)
314		*/
315	7.14M	if (pkt->subs == NULL \|\| pkt->subs->parent == NULL)
316	0	return 0;
317
318	7.14M	return wpacket_intern_close(pkt, pkt->subs, 1);
319	7.14M	}
320
321		int WPACKET_finish(WPACKET *pkt)
322	15.4M	{
323	15.4M	int ret;
324
325		/*
326		* Internal API, so should not fail - but we do negative testing of this
327		* so no assert (otherwise the tests fail)
328		*/
329	15.4M	if (pkt->subs == NULL \|\| pkt->subs->parent != NULL)
330	0	return 0;
331
332	15.4M	ret = wpacket_intern_close(pkt, pkt->subs, 1);
333	15.4M	if (ret) {
334	15.4M	OPENSSL_free(pkt->subs);
335	15.4M	pkt->subs = NULL;
336	15.4M	}
337
338	15.4M	return ret;
339	15.4M	}
340
341		int WPACKET_start_sub_packet_len__(WPACKET *pkt, size_t lenbytes)
342	7.15M	{
343	7.15M	WPACKET_SUB *sub;
344	7.15M	unsigned char *lenchars;
345
346		/* Internal API, so should not fail */
347	7.15M	if (!ossl_assert(pkt->subs != NULL))
348	0	return 0;
349
350		/* We don't support lenbytes greater than 0 when doing endfirst writing */
351	7.15M	if (lenbytes > 0 && pkt->endfirst)
352	0	return 0;
353
354	7.15M	if ((sub = OPENSSL_zalloc(sizeof(*sub))) == NULL) {
355	0	ERR_raise(ERR_LIB_CRYPTO, ERR_R_MALLOC_FAILURE);
356	0	return 0;
357	0	}
358
359	7.15M	sub->parent = pkt->subs;
360	7.15M	pkt->subs = sub;
361	7.15M	sub->pwritten = pkt->written + lenbytes;
362	7.15M	sub->lenbytes = lenbytes;
363
364	7.15M	if (lenbytes == 0) {
365	1.65M	sub->packet_len = 0;
366	1.65M	return 1;
367	1.65M	}
368
369	5.49M	sub->packet_len = pkt->written;
370
371	5.49M	if (!WPACKET_allocate_bytes(pkt, lenbytes, &lenchars))
372	0	return 0;
373
374	5.49M	return 1;
375	5.49M	}
376
377		int WPACKET_start_sub_packet(WPACKET *pkt)
378	1.65M	{
379	1.65M	return WPACKET_start_sub_packet_len__(pkt, 0);
380	1.65M	}
381
382		int WPACKET_put_bytes__(WPACKET *pkt, uint64_t val, size_t size)
383	25.4M	{
384	25.4M	unsigned char *data;
385
386		/* Internal API, so should not fail */
387	25.4M	if (!ossl_assert(size <= sizeof(uint64_t))
388	25.4M	\|\| !WPACKET_allocate_bytes(pkt, size, &data)
389	25.4M	\|\| !put_value(data, val, size))
390	45.7k	return 0;
391
392	25.4M	return 1;
393	25.4M	}
394
395		int WPACKET_set_max_size(WPACKET *pkt, size_t maxsize)
396	10.2M	{
397	10.2M	WPACKET_SUB *sub;
398	10.2M	size_t lenbytes;
399
400		/* Internal API, so should not fail */
401	10.2M	if (!ossl_assert(pkt->subs != NULL))
402	0	return 0;
403
404		/* Find the WPACKET_SUB for the top level */
405	10.2M	for (sub = pkt->subs; sub->parent != NULL; sub = sub->parent)
406	0	continue;
407
408	10.2M	lenbytes = sub->lenbytes;
409	10.2M	if (lenbytes == 0)
410	10.2M	lenbytes = sizeof(pkt->maxsize);
411
412	10.2M	if (maxmaxsize(lenbytes) < maxsize \|\| maxsize < pkt->written)
413	0	return 0;
414
415	10.2M	pkt->maxsize = maxsize;
416
417	10.2M	return 1;
418	10.2M	}
419
420		int WPACKET_memset(WPACKET *pkt, int ch, size_t len)
421	1.08M	{
422	1.08M	unsigned char *dest;
423
424	1.08M	if (len == 0)
425	0	return 1;
426
427	1.08M	if (!WPACKET_allocate_bytes(pkt, len, &dest))
428	3	return 0;
429
430	1.08M	if (dest != NULL)
431	1.08M	memset(dest, ch, len);
432
433	1.08M	return 1;
434	1.08M	}
435
436		int WPACKET_memcpy(WPACKET pkt, const void src, size_t len)
437	12.4M	{
438	12.4M	unsigned char *dest;
439
440	12.4M	if (len == 0)
441	5.08M	return 1;
442
443	7.32M	if (!WPACKET_allocate_bytes(pkt, len, &dest))
444	83.7k	return 0;
445
446	7.24M	if (dest != NULL)
447	7.24M	memcpy(dest, src, len);
448
449	7.24M	return 1;
450	7.32M	}
451
452		int WPACKET_sub_memcpy__(WPACKET pkt, const void src, size_t len,
453		size_t lenbytes)
454	1.76M	{
455	1.76M	if (!WPACKET_start_sub_packet_len__(pkt, lenbytes)
456	1.76M	\|\| !WPACKET_memcpy(pkt, src, len)
457	1.76M	\|\| !WPACKET_close(pkt))
458	612	return 0;
459
460	1.76M	return 1;
461	1.76M	}
462
463		int WPACKET_get_total_written(WPACKET pkt, size_t written)
464	18.3M	{
465		/* Internal API, so should not fail */
466	18.3M	if (!ossl_assert(written != NULL))
467	0	return 0;
468
469	18.3M	*written = pkt->written;
470
471	18.3M	return 1;
472	18.3M	}
473
474		int WPACKET_get_length(WPACKET pkt, size_t len)
475	1.78M	{
476		/* Internal API, so should not fail */
477	1.78M	if (!ossl_assert(pkt->subs != NULL && len != NULL))
478	0	return 0;
479
480	1.78M	*len = pkt->written - pkt->subs->pwritten;
481
482	1.78M	return 1;
483	1.78M	}
484
485		unsigned char WPACKET_get_curr(WPACKET pkt)
486	145M	{
487	145M	unsigned char *buf = GETBUF(pkt);
488
489	145M	if (buf == NULL)
490	4.13M	return NULL;
491
492	141M	if (pkt->endfirst)
493	93.0k	return buf + pkt->maxsize - pkt->curr;
494
495	141M	return buf + pkt->curr;
496	141M	}
497
498		int WPACKET_is_null_buf(WPACKET *pkt)
499	516k	{
500	516k	return pkt->buf == NULL && pkt->staticbuf == NULL;
501	516k	}
502
503		void WPACKET_cleanup(WPACKET *pkt)
504	777k	{
505	777k	WPACKET_SUB sub, parent;
506
507	1.18M	for (sub = pkt->subs; sub != NULL; sub = parent) {
508	406k	parent = sub->parent;
509	406k	OPENSSL_free(sub);
510	406k	}
511		pkt->subs = NULL;
512	777k	}