srsLTE/lib/test/common/msg_queue_test.cc

82 lines
1.9 KiB
C++

/*
* Copyright 2013-2020 Software Radio Systems Limited
*
* This file is part of srsLTE.
*
* srsLTE is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of
* the License, or (at your option) any later version.
*
* srsLTE is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* A copy of the GNU Affero General Public License can be found in
* the LICENSE file in the top-level directory of this distribution
* and at http://www.gnu.org/licenses/.
*
*/
#define NMSGS 1000000
#include "srslte/common/buffer_pool.h"
#include "srslte/upper/rlc_tx_queue.h"
#include <stdio.h>
using namespace srslte;
typedef struct {
rlc_tx_queue* q;
} args_t;
void* write_thread(void* a)
{
args_t* args = (args_t*)a;
byte_buffer_pool* pool = byte_buffer_pool::get_instance();
for (uint32_t i = 0; i < NMSGS; i++) {
unique_byte_buffer_t b = srslte::allocate_unique_buffer(*pool, true);
memcpy(b->msg, &i, 4);
b->N_bytes = 4;
args->q->write(std::move(b));
}
return NULL;
}
int main(int argc, char** argv)
{
bool result;
rlc_tx_queue q;
unique_byte_buffer_t b;
pthread_t thread;
args_t args;
u_int32_t r;
result = true;
args.q = &q;
pthread_create(&thread, NULL, &write_thread, &args);
for (uint32_t i = 0; i < NMSGS; i++) {
b = q.read();
memcpy(&r, b->msg, 4);
if (r != i)
result = false;
}
pthread_join(thread, NULL);
if (q.size() != 0 || q.size_bytes() != 0) {
result = false;
}
if (result) {
printf("Passed\n");
exit(0);
} else {
printf("Failed\n;");
exit(1);
}
}