Test the size_t constant time functions

Reviewed-by: Rich Salz <rsalz@openssl.org>

2 files changed, 112 insertions, 7 deletions

diff --git a/ssl/bio_ssl.c b/ssl/bio_ssl.c
index 596df24..e48b90f 100644
--- a/ssl/bio_ssl.c
+++ b/ssl/bio_ssl.c
@@ -107,8 +107,6 @@ static int ssl_read(BIO *b, char *buf, size_t size, size_t *readbytes)
 
     switch (SSL_get_error(ssl, ret)) {
     case SSL_ERROR_NONE:
-        if (*readbytes == 0)
-            break;
         if (sb->renegotiate_count > 0) {
             sb->byte_count += *readbytes;
             if (sb->byte_count > sb->renegotiate_count) {
@@ -178,8 +176,6 @@ static int ssl_write(BIO *b, const char *buf, size_t size, size_t *written)
 
     switch (SSL_get_error(ssl, ret)) {
     case SSL_ERROR_NONE:
-        if (*written == 0)
-            break;
         if (bs->renegotiate_count > 0) {
             bs->byte_count += *written;
             if (bs->byte_count > bs->renegotiate_count) {
diff --git a/test/constant_time_test.c b/test/constant_time_test.c
index 3ee6a81..41587e2 100644
--- a/test/constant_time_test.c
+++ b/test/constant_time_test.c
@@ -9,6 +9,7 @@
 
 #include "internal/constant_time_locl.h"
 #include "e_os.h"
+#include <openssl/e_os2.h>
 
 #include <limits.h>
 #include <stdio.h>
@@ -18,6 +19,8 @@ static const unsigned int CONSTTIME_TRUE = (unsigned)(~0);
 static const unsigned int CONSTTIME_FALSE = 0;
 static const unsigned char CONSTTIME_TRUE_8 = 0xff;
 static const unsigned char CONSTTIME_FALSE_8 = 0;
+static const size_t CONSTTIME_TRUE_S = ~((size_t)0);
+static const size_t CONSTTIME_FALSE_S = 0;
 
 static int test_binary_op(unsigned int (*op) (unsigned int a, unsigned int b),
                           const char *op_name, unsigned int a, unsigned int b,
@@ -54,6 +57,25 @@ static int test_binary_op_8(unsigned
     return 0;
 }
 
+static int test_binary_op_s(size_t (*op) (size_t a, size_t b),
+                            const char *op_name, size_t a, size_t b,
+                            int is_true)
+{
+    size_t c = op(a, b);
+    if (is_true && c != CONSTTIME_TRUE_S) {
+        fprintf(stderr, "Test failed for %s(%"OSSLzu", %"OSSLzu
+        "): expected %"OSSLzu" (TRUE), got %"OSSLzu"\n",
+        op_name, a, b, CONSTTIME_TRUE_S, c);
+        return 1;
+    } else if (!is_true && c != CONSTTIME_FALSE_S) {
+        fprintf(stderr, "Test failed for  %s(%"OSSLzu", %"OSSLzu
+        "): expected %" OSSLzu " (FALSE), got %"OSSLzu"\n",
+        op_name, a, b, CONSTTIME_FALSE_S, c);
+        return 1;
+    }
+    return 0;
+}
+
 static int test_is_zero(unsigned int a)
 {
     unsigned int c = constant_time_is_zero(a);
@@ -84,6 +106,22 @@ static int test_is_zero_8(unsigned int a)
     return 0;
 }
 
+static int test_is_zero_s(size_t a)
+{
+    size_t c = constant_time_is_zero_s(a);
+    if (a == 0 && c != CONSTTIME_TRUE_S) {
+        fprintf(stderr, "Test failed for constant_time_is_zero_s(%"OSSLzu"): "
+                "expected %"OSSLzu" (TRUE), got %"OSSLzu"\n",
+                a, CONSTTIME_TRUE_S, c);
+        return 1;
+    } else if (a != 0 && c != CONSTTIME_FALSE) {
+        fprintf(stderr, "Test failed for constant_time_is_zero_s(%"OSSLzu"): "
+                "expected %"OSSLzu" (FALSE), got %"OSSLzu"\n",
+                a, CONSTTIME_FALSE_S, c);
+        return 1;
+    }
+    return 0;
+}
 static int test_select(unsigned int a, unsigned int b)
 {
     unsigned int selected = constant_time_select(CONSTTIME_TRUE, a, b);
@@ -141,6 +179,28 @@ static int test_select_int(int a, int b)
     return 0;
 }
 
+
+static int test_select_s(size_t a, size_t b)
+{
+    size_t selected = constant_time_select_s(CONSTTIME_TRUE_S, a, b);
+    if (selected != a) {
+        fprintf(stderr, "Test failed for constant_time_select_s(%"OSSLzu
+                ", %"OSSLzu",%"OSSLzu"): expected %"OSSLzu
+                "(first value), got %"OSSLzu"\n",
+                CONSTTIME_TRUE_S, a, b, a, selected);
+        return 1;
+    }
+    selected = constant_time_select_s(CONSTTIME_FALSE_S, a, b);
+    if (selected != b) {
+        fprintf(stderr, "Test failed for constant_time_select_s(%"OSSLzu
+                ", %"OSSLzu",%"OSSLzu"): expected %"OSSLzu
+                "(second value), got %"OSSLzu"\n",
+                CONSTTIME_FALSE_S, a, b, b, selected);
+        return 1;
+    }
+    return 0;
+}
+
 static int test_eq_int(int a, int b)
 {
     unsigned int equal = constant_time_eq_int(a, b);
@@ -173,6 +233,23 @@ static int test_eq_int_8(int a, int b)
     return 0;
 }
 
+static int test_eq_s(size_t a, size_t b)
+{
+    size_t equal = constant_time_eq_s(a, b);
+    if (a == b && equal != CONSTTIME_TRUE_S) {
+        fprintf(stderr, "Test failed for constant_time_eq_int(%"OSSLzu
+                ", %"OSSLzu"): expected %"OSSLzu"(TRUE), got %"OSSLzu"\n",
+                a, b, CONSTTIME_TRUE_S, equal);
+        return 1;
+    } else if (a != b && equal != CONSTTIME_FALSE_S) {
+        fprintf(stderr, "Test failed for constant_time_eq_int(%"OSSLzu", %"
+                OSSLzu"): expected %"OSSLzu"(FALSE), got %"OSSLzu"\n",
+                a, b, CONSTTIME_FALSE_S, equal);
+        return 1;
+    }
+    return 0;
+}
+
 static unsigned int test_values[] =
     { 0, 1, 1024, 12345, 32000, UINT_MAX / 2 - 1,
     UINT_MAX / 2, UINT_MAX / 2 + 1, UINT_MAX - 1,
@@ -187,34 +264,55 @@ static int signed_test_values[] = { 0, 1, -1, 1024, -1024, 12345, -12345,
     INT_MIN + 1
 };
 
+static size_t test_values_s[] =
+    { 0, 1, 1024, 12345, 32000, SIZE_MAX / 2 - 1,
+    SIZE_MAX / 2, SIZE_MAX / 2 + 1, SIZE_MAX - 1,
+    SIZE_MAX
+};
+
 int main(int argc, char *argv[])
 {
     unsigned int a, b, i, j;
     int c, d;
     unsigned char e, f;
+    size_t g, h;
     int num_failed = 0, num_all = 0;
     fprintf(stdout, "Testing constant time operations...\n");
 
+    if (OSSL_NELEM(test_values) != OSSL_NELEM(test_values_s)) {
+        fprintf(stdout, "Unexpected number of tests\n");
+        return EXIT_FAILURE;
+    }
+
     for (i = 0; i < OSSL_NELEM(test_values); ++i) {
         a = test_values[i];
+        g = test_values_s[i];
         num_failed += test_is_zero(a);
         num_failed += test_is_zero_8(a);
-        num_all += 2;
+        num_failed += test_is_zero_s(g);
+        num_all += 3;
         for (j = 0; j < OSSL_NELEM(test_values); ++j) {
             b = test_values[j];
+            h = test_values[j];
             num_failed += test_binary_op(&constant_time_lt,
                                          "constant_time_lt", a, b, a < b);
             num_failed += test_binary_op_8(&constant_time_lt_8,
                                            "constant_time_lt_8", a, b, a < b);
+            num_failed += test_binary_op_s(&constant_time_lt_s,
+                                           "constant_time_lt_s", g, h, g < h);
             num_failed += test_binary_op(&constant_time_lt,
-                                         "constant_time_lt_8", b, a, b < a);
+                                         "constant_time_lt", b, a, b < a);
             num_failed += test_binary_op_8(&constant_time_lt_8,
                                            "constant_time_lt_8", b, a, b < a);
+            num_failed += test_binary_op_s(&constant_time_lt_s,
+                                           "constant_time_lt_s", h, g, h < g);
             num_failed += test_binary_op(&constant_time_ge,
                                          "constant_time_ge", a, b, a >= b);
             num_failed += test_binary_op_8(&constant_time_ge_8,
                                            "constant_time_ge_8", a, b,
                                            a >= b);
+            num_failed += test_binary_op_s(&constant_time_ge_s,
+                                           "constant_time_ge_s", g, h, g >= h);
             num_failed +=
                 test_binary_op(&constant_time_ge, "constant_time_ge", b, a,
                                b >= a);
@@ -222,19 +320,30 @@ int main(int argc, char *argv[])
                 test_binary_op_8(&constant_time_ge_8, "constant_time_ge_8", b,
                                  a, b >= a);
             num_failed +=
+                test_binary_op_s(&constant_time_ge_s, "constant_time_ge_s", h, g,
+                                 h >= g);
+            num_failed +=
                 test_binary_op(&constant_time_eq, "constant_time_eq", a, b,
                                a == b);
             num_failed +=
                 test_binary_op_8(&constant_time_eq_8, "constant_time_eq_8", a,
                                  b, a == b);
             num_failed +=
+                test_binary_op_s(&constant_time_eq_s, "constant_time_eq_s", g, h,
+                                 g == h);
+            num_failed +=
                 test_binary_op(&constant_time_eq, "constant_time_eq", b, a,
                                b == a);
             num_failed +=
                 test_binary_op_8(&constant_time_eq_8, "constant_time_eq_8", b,
                                  a, b == a);
+            num_failed +=
+                test_binary_op_s(&constant_time_eq_s, "constant_time_eq_s", h, g,
+                                 h == g);
             num_failed += test_select(a, b);
-            num_all += 13;
+            num_failed += test_select_s(g, h);
+            num_failed += test_eq_s(g, h);
+            num_all += 21;
         }
     }