From 378c8d247ea5233b61ccabe2b9cf2dc60d6362b4 Mon Sep 17 00:00:00 2001
From: Pedro Alvarez <pedro.alvarez@softwareradiosystems.com>
Date: Wed, 10 Jul 2019 22:37:32 +0100
Subject: [PATCH] Starting to add the mechanisms to store the last PDU in the
 PDCP tests.

---
 lib/test/upper/pdcp_nr_test.cc | 58 ++++++++++++++++++++++++++++++++--
 1 file changed, 55 insertions(+), 3 deletions(-)

diff --git a/lib/test/upper/pdcp_nr_test.cc b/lib/test/upper/pdcp_nr_test.cc
index e63d68788..e3d495b6d 100644
--- a/lib/test/upper/pdcp_nr_test.cc
+++ b/lib/test/upper/pdcp_nr_test.cc
@@ -43,6 +43,10 @@ uint8_t k_enc[] = {0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0
 uint8_t  sdu1[]   = {0x18, 0xE2};
 uint32_t SDU1_LEN = 2;
 
+// Test PDUs for rx
+uint8_t  pdu1[]   = {0x80, 0x00, 0x8f, 0xe3, 0xe0, 0xdf, 0x82, 0x92};
+uint32_t PDU1_LEN = 8;
+
 // fake classes
 class rlc_dummy : public srsue::rlc_interface_pdcp
 {
@@ -52,10 +56,13 @@ public:
   void write_sdu(uint32_t lcid, srslte::unique_byte_buffer_t sdu, bool blocking = true)
   {
     log->info_hex(sdu->msg, sdu->N_bytes, "RLC SDU");
+    last_pdcp_pdu(std::move(sdu));
   }
 
 private:
   srslte::log* log;
+  srslte::unique_byte_buffer_t last_pdcp_pdu;
+  srslte::unique_byte_buffer_t get_last_pdcp_pdu() { return last_pdcp_pdu; }
 
   bool rb_is_um(uint32_t lcid) { return false; }
 };
@@ -94,7 +101,7 @@ private:
  * Configure PDCP entity with EIA2 and EEA2
  * TX_NEXT initially at 0.
  * Input: {0x18, 0xE2}
- * Output: PDCP Header {0x80,0x00}, Ciphered Text {}, MAC-I {}
+ * Output: PDCP Header {0x80,0x00}, Ciphered Text {0x8f, 0xe3}, MAC-I {0xe0, 0xdf, 0x82, 0x92}
  */
 bool test_tx_basic(srslte::byte_buffer_pool* pool, srslte::log* log)
 {
@@ -110,6 +117,51 @@ bool test_tx_basic(srslte::byte_buffer_pool* pool, srslte::log* log)
   pdcp.enable_integrity();
   pdcp.enable_encryption();
 
+  // Test SDU
+  srslte::unique_byte_buffer_t sdu = allocate_unique_buffer(*pool);
+  memcpy(msg->msg, sdu1, SDU1_LEN);
+  msg->N_bytes = SDU1_LEN;
+
+  // Expected PDCP PDU
+  srslte::unique_byte_buffer_t pdu_exp = allocate_unique_buffer(*pool);
+  memcpy(pdu_exp->msg, pdu_exp, PDU1_LEN);
+  msg->N_bytes = PDU1_LEN;
+
+  // Actual PDCP PDU
+  srslte::unique_byte_buffer_t pdu_act = allocate_unique_buffer(*pool);
+
+  // Run test
+  pdcp.write_sdu(std::move(sdu), true);
+  rlc.get_last_pdcp_pdu(pdu);
+
+  TESTASSERT(pdu->N_Bytes == PDU1_LEN);
+  for (int i = 0; i < PDU1_LEN; ++i) {
+    TESTASSERT(pdu->msg[i] == PDU1_LEN);
+  }
+  return true;
+}
+
+/*
+ * Test 2: PDCP Entity RX
+ * Configure PDCP entity with EIA2 and EEA2
+ * TX_NEXT initially at 0.
+ * Input: {0x80, 0x00, 0x8f, 0xe3, 0xe0, 0xdf 0x82, 0x92}
+ * Output: {0x18, 0xE2}
+ */
+bool test_rx_basic(srslte::byte_buffer_pool* pool, srslte::log* log)
+{
+  srslte::pdcp_entity_nr pdcp;
+  srslte::srslte_pdcp_config_t cfg = {1, srslte::PDCP_RB_IS_DRB, SECURITY_DIRECTION_UPLINK, srslte::PDCP_SN_LEN_12};
+
+  rlc_dummy rlc(log);
+  rrc_dummy rrc(log);
+  gw_dummy gw(log);
+
+  pdcp.init(&rlc, &rrc, &gw, log, 0, cfg);
+  pdcp.config_security(k_enc, k_int, k_enc, k_int, srslte::CIPHERING_ALGORITHM_ID_128_EEA2, srslte::INTEGRITY_ALGORITHM_ID_128_EIA2);
+  pdcp.enable_integrity();
+  pdcp.enable_encryption();
+
   uint8_t mac_exp[4];
   srslte::unique_byte_buffer_t msg = allocate_unique_buffer(*pool);
   srslte::unique_byte_buffer_t ct_exp  = allocate_unique_buffer(*pool);
@@ -123,10 +175,10 @@ bool test_tx_basic(srslte::byte_buffer_pool* pool, srslte::log* log)
   log->info_hex(mac_exp, 4, "MAC-I:");
   log->info_hex(ct_exp->msg, ct_exp->N_bytes, "Cipher text:");
 
-  pdcp.write_sdu(std::move(msg), true);
+  pdcp.write_pdu(std::move(msg), true);
+
   return true;
 }
-
 // Setup all tests
 int run_all_tests(srslte::byte_buffer_pool* pool)
 {