From: Pauli <pauli@openssl.org>
Date: Mon, 27 Jun 2022 01:02:51 +0000 (+1000)
Subject: Add unit test for event queue
X-Git-Tag: openssl-3.2.0-alpha1~2434
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=0eb27659435768d1a2370858d340a9c6793ed244;p=thirdparty%2Fopenssl.git

Add unit test for event queue

Reviewed-by: Tomas Mraz <tomas@openssl.org>
Reviewed-by: Matt Caswell <matt@openssl.org>
(Merged from https://github.com/openssl/openssl/pull/18345)
---

diff --git a/test/build.info b/test/build.info
index 2721945197a..13653f9efa5 100644
--- a/test/build.info
+++ b/test/build.info
@@ -70,7 +70,7 @@ IF[{- !$disabled{tests} -}]
   ENDIF
 
   IF[{- !$disabled{quic} -}]
-    PROGRAMS{noinst}=priority_queue_test
+    PROGRAMS{noinst}=priority_queue_test event_queue_test
   ENDIF
 
   SOURCE[confdump]=confdump.c
@@ -710,7 +710,11 @@ IF[{- !$disabled{tests} -}]
     IF[{- !$disabled{quic} -}]
       SOURCE[priority_queue_test]=priority_queue_test.c
       INCLUDE[priority_queue_test]=../include ../apps/include
-      DEPEND[priority_queue_test]=../libssl.a ../libcrypto.a libtestutil.a
+      DEPEND[priority_queue_test]=../libcrypto ../libssl.a libtestutil.a
+
+      SOURCE[event_queue_test]=event_queue_test.c
+      INCLUDE[event_queue_test]=../include ../apps/include
+      DEPEND[event_queue_test]=../libcrypto ../libssl.a libtestutil.a
     ENDIF
 
     SOURCE[dhtest]=dhtest.c
diff --git a/test/event_queue_test.c b/test/event_queue_test.c
new file mode 100644
index 00000000000..76765c5ba17
--- /dev/null
+++ b/test/event_queue_test.c
@@ -0,0 +1,101 @@
+/*
+ * 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/event_queue.h"
+#include "internal/nelem.h"
+#include "testutil.h"
+
+static OSSL_TIME cur_time = 100;
+
+OSSL_TIME ossl_time_now(void)
+{
+    return cur_time;
+}
+
+#define PAYLOAD(s)  s, strlen(s) + 1
+
+static int event_test(void)
+{
+    int res = 0;
+    size_t len = 0;
+    OSSL_EVENT *e1, *e2, e3, *e4 = NULL, *ep = NULL;
+    OSSL_EVENT_QUEUE *q = NULL;
+    void *p;
+    static char payload[] = "payload";
+
+    /* Create an event queue and add some events */
+    if (!TEST_ptr(q = ossl_event_queue_new())
+            || !TEST_ptr(e1 = ossl_event_queue_add_new(q, 1, 10, 1100, "ctx 1",
+                                                       PAYLOAD(payload)))
+            || !TEST_ptr(e2 = ossl_event_queue_add_new(q, 2, 5, 1100, "ctx 2",
+                                                       PAYLOAD("data")))
+            || !TEST_true(ossl_event_queue_add(q, &e3, 3, 20, 1200, "ctx 3",
+                                               PAYLOAD("more data")))
+            || !TEST_ptr(e4 = ossl_event_queue_add_new(q, 2, 5, 1150, "ctx 2",
+                                                       PAYLOAD("data")))
+
+            /* Verify some event details */
+            || !TEST_uint_eq(ossl_event_get_type(e1), 1)
+            || !TEST_uint_eq(ossl_event_get_priority(e1), 10)
+            || !TEST_uint64_t_eq(ossl_event_get_when(e1), 1100)
+            || !TEST_str_eq(ossl_event_get0_ctx(e1), "ctx 1")
+            || !TEST_ptr(p = ossl_event_get0_payload(e1, &len))
+            || !TEST_str_eq((char *)p, payload)
+            || !TEST_uint64_t_eq(ossl_event_time_until(&e3), 1100)
+            || !TEST_uint64_t_eq(ossl_event_queue_time_until_next(q), 1000)
+
+            /* Modify an event's time */
+            || !TEST_true(ossl_event_queue_postpone_until(q, e1, 1200))
+            || !TEST_uint64_t_eq(ossl_event_get_when(e1), 1200)
+            || !TEST_true(ossl_event_queue_remove(q, e4)))
+        goto err;
+    ossl_event_free(e4);
+
+    /* Execute the queue */
+    cur_time = 1000;
+    if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
+            || !TEST_ptr_null(ep))
+        goto err;
+    cur_time = 1100;
+    if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
+            || !TEST_ptr_eq(ep, e2))
+        goto err;
+    ossl_event_free(ep);
+    ep = e2 = NULL;
+    if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
+            || !TEST_ptr_null(ep))
+        goto err;
+
+    cur_time = 1250;
+    if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
+            || !TEST_ptr_eq(ep, &e3))
+        goto err;
+    ossl_event_free(ep);
+    ep = NULL;
+    if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
+            || !TEST_ptr_eq(ep, e1))
+        goto err;
+    ossl_event_free(ep);
+    ep = e1 = NULL;
+    if (!TEST_true(ossl_event_queue_get1_next_event(q, &ep))
+            || !TEST_ptr_null(ep))
+        goto err;
+
+    res = 1;
+ err:
+    ossl_event_free(ep);
+    ossl_event_queue_free(q);
+    return res;
+}
+
+int setup_tests(void)
+{
+    ADD_TEST(event_test);
+    return 1;
+}