mirror of https://github.com/openssl/openssl
68 lines
1.9 KiB
C
68 lines
1.9 KiB
C
/*
|
|
* Copyright 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/packet.h"
|
|
#include "internal/quic_txpim.h"
|
|
#include "testutil.h"
|
|
|
|
static int test_txpim(void)
|
|
{
|
|
int testresult = 0;
|
|
QUIC_TXPIM *txpim;
|
|
size_t i, j;
|
|
QUIC_TXPIM_PKT *pkts[10] = {NULL};
|
|
QUIC_TXPIM_CHUNK chunks[3];
|
|
const QUIC_TXPIM_CHUNK *rchunks;
|
|
|
|
if (!TEST_ptr(txpim = ossl_quic_txpim_new()))
|
|
goto err;
|
|
|
|
for (i = 0; i < OSSL_NELEM(pkts); ++i) {
|
|
if (!TEST_ptr(pkts[i] = ossl_quic_txpim_pkt_alloc(txpim)))
|
|
goto err;
|
|
|
|
if (!TEST_size_t_eq(ossl_quic_txpim_pkt_get_num_chunks(pkts[i]), 0))
|
|
goto err;
|
|
|
|
for (j = 0; j < OSSL_NELEM(chunks); ++j) {
|
|
chunks[j].stream_id = 100 - j;
|
|
chunks[j].start = 1000 * i + j * 10;
|
|
chunks[j].end = chunks[j].start + 5;
|
|
|
|
if (!TEST_true(ossl_quic_txpim_pkt_append_chunk(pkts[i], chunks + j)))
|
|
goto err;
|
|
}
|
|
|
|
if (!TEST_size_t_eq(ossl_quic_txpim_pkt_get_num_chunks(pkts[i]),
|
|
OSSL_NELEM(chunks)))
|
|
goto err;
|
|
|
|
rchunks = ossl_quic_txpim_pkt_get_chunks(pkts[i]);
|
|
if (!TEST_uint64_t_eq(rchunks[0].stream_id, 98)
|
|
|| !TEST_uint64_t_eq(rchunks[1].stream_id, 99)
|
|
|| !TEST_uint64_t_eq(rchunks[2].stream_id, 100))
|
|
goto err;
|
|
}
|
|
|
|
testresult = 1;
|
|
err:
|
|
for (i = 0; i < OSSL_NELEM(pkts); ++i)
|
|
if (txpim != NULL && pkts[i] != NULL)
|
|
ossl_quic_txpim_pkt_release(txpim, pkts[i]);
|
|
|
|
ossl_quic_txpim_free(txpim);
|
|
return testresult;
|
|
}
|
|
|
|
int setup_tests(void)
|
|
{
|
|
ADD_TEST(test_txpim);
|
|
return 1;
|
|
}
|