srsRAN_4G/lib/test/common/byte_buffer_queue_test.cc

/**
 *
 * \section COPYRIGHT
 *
 * Copyright 2013-2020 Software Radio Systems Limited
 *
 * By using this file, you agree to the terms and conditions set
 * forth in the LICENSE file which can be found at the top level of
 * the distribution.
 *
 */

#define NMSGS 1000000

#include "srslte/common/buffer_pool.h"
#include "srslte/upper/byte_buffer_queue.h"
#include <stdio.h>

using namespace srslte;

typedef struct {
  byte_buffer_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;
  byte_buffer_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);
  }
}