sodium_utils.c 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177
  1. #define TEST_NAME "sodium_utils"
  2. #include "cmptest.h"
  3. int
  4. main(void)
  5. {
  6. unsigned char buf_add[1000];
  7. unsigned char buf1[1000];
  8. unsigned char buf2[1000];
  9. unsigned char buf1_rev[1000];
  10. unsigned char buf2_rev[1000];
  11. char buf3[33];
  12. unsigned char nonce[24];
  13. char nonce_hex[49];
  14. unsigned char *bin_padded;
  15. size_t bin_len, bin_len2;
  16. size_t bin_padded_len;
  17. size_t bin_padded_maxlen;
  18. size_t blocksize;
  19. unsigned int i;
  20. unsigned int j;
  21. randombytes_buf(buf1, sizeof buf1);
  22. memcpy(buf2, buf1, sizeof buf2);
  23. printf("%d\n", sodium_memcmp(buf1, buf2, sizeof buf1));
  24. sodium_memzero(buf1, 0U);
  25. printf("%d\n", sodium_memcmp(buf1, buf2, sizeof buf1));
  26. sodium_memzero(buf1, sizeof buf1 / 2);
  27. printf("%d\n", sodium_memcmp(buf1, buf2, sizeof buf1));
  28. printf("%d\n", sodium_memcmp(buf1, buf2, 0U));
  29. sodium_memzero(buf2, sizeof buf2 / 2);
  30. printf("%d\n", sodium_memcmp(buf1, buf2, sizeof buf1));
  31. memset(nonce, 0, sizeof nonce);
  32. sodium_increment(nonce, sizeof nonce);
  33. printf("%s\n",
  34. sodium_bin2hex(nonce_hex, sizeof nonce_hex, nonce, sizeof nonce));
  35. memset(nonce, 255, sizeof nonce);
  36. sodium_increment(nonce, sizeof nonce);
  37. printf("%s\n",
  38. sodium_bin2hex(nonce_hex, sizeof nonce_hex, nonce, sizeof nonce));
  39. nonce[1] = 1U;
  40. sodium_increment(nonce, sizeof nonce);
  41. printf("%s\n",
  42. sodium_bin2hex(nonce_hex, sizeof nonce_hex, nonce, sizeof nonce));
  43. nonce[1] = 0U;
  44. sodium_increment(nonce, sizeof nonce);
  45. printf("%s\n",
  46. sodium_bin2hex(nonce_hex, sizeof nonce_hex, nonce, sizeof nonce));
  47. nonce[0] = 255U;
  48. nonce[2] = 255U;
  49. sodium_increment(nonce, sizeof nonce);
  50. printf("%s\n",
  51. sodium_bin2hex(nonce_hex, sizeof nonce_hex, nonce, sizeof nonce));
  52. for (i = 0U; i < 1000U; i++) {
  53. bin_len = (size_t) randombytes_uniform(sizeof buf1);
  54. randombytes_buf(buf1, bin_len);
  55. randombytes_buf(buf2, bin_len);
  56. for (j = 0U; j < bin_len; j++) {
  57. buf1_rev[bin_len - 1 - j] = buf1[j];
  58. buf2_rev[bin_len - 1 - j] = buf2[j];
  59. }
  60. if (memcmp(buf1_rev, buf2_rev, bin_len) *
  61. sodium_compare(buf1, buf2, bin_len) <
  62. 0) {
  63. printf("sodium_compare() failure with length=%u\n",
  64. (unsigned int) bin_len);
  65. }
  66. memcpy(buf1, buf2, bin_len);
  67. if (sodium_compare(buf1, buf2, bin_len)) {
  68. printf("sodium_compare() equality failure with length=%u\n",
  69. (unsigned int) bin_len);
  70. }
  71. }
  72. memset(buf1, 0, sizeof buf1);
  73. if (sodium_is_zero(buf1, sizeof buf1) != 1) {
  74. printf("sodium_is_zero() failed\n");
  75. }
  76. for (i = 0U; i < sizeof buf1; i++) {
  77. buf1[i]++;
  78. if (sodium_is_zero(buf1, sizeof buf1) != 0) {
  79. printf("sodium_is_zero() failed\n");
  80. }
  81. buf1[i]--;
  82. }
  83. bin_len = randombytes_uniform(sizeof buf1);
  84. randombytes_buf(buf1, bin_len);
  85. memcpy(buf2, buf1, bin_len);
  86. memset(buf_add, 0, bin_len);
  87. j = randombytes_uniform(10000);
  88. for (i = 0U; i < j; i++) {
  89. sodium_increment(buf1, bin_len);
  90. sodium_increment(buf_add, bin_len);
  91. }
  92. sodium_add(buf2, buf_add, bin_len);
  93. if (sodium_compare(buf1, buf2, bin_len) != 0) {
  94. printf("sodium_add() failed\n");
  95. }
  96. bin_len = randombytes_uniform(sizeof buf1);
  97. randombytes_buf(buf1, bin_len);
  98. memcpy(buf2, buf1, bin_len);
  99. memset(buf_add, 0xff, bin_len);
  100. sodium_increment(buf2, bin_len);
  101. sodium_increment(buf2, 0U);
  102. sodium_add(buf2, buf_add, bin_len);
  103. sodium_add(buf2, buf_add, 0U);
  104. if (sodium_compare(buf1, buf2, bin_len) != 0) {
  105. printf("sodium_add() failed\n");
  106. }
  107. assert(sizeof nonce >= 24U);
  108. memset(nonce, 0xfe, 24U);
  109. memset(nonce, 0xff, 6U);
  110. sodium_increment(nonce, 8U);
  111. printf("%s\n",
  112. sodium_bin2hex(nonce_hex, sizeof nonce_hex, nonce, sizeof nonce));
  113. memset(nonce, 0xfe, 24U);
  114. memset(nonce, 0xff, 10U);
  115. sodium_increment(nonce, 12U);
  116. printf("%s\n",
  117. sodium_bin2hex(nonce_hex, sizeof nonce_hex, nonce, sizeof nonce));
  118. memset(nonce, 0xff, 22U);
  119. sodium_increment(nonce, 24U);
  120. printf("%s\n",
  121. sodium_bin2hex(nonce_hex, sizeof nonce_hex, nonce, sizeof nonce));
  122. assert(sizeof nonce >= 24U);
  123. memset(nonce, 0xfe, 24U);
  124. memset(nonce, 0xff, 6U);
  125. sodium_add(nonce, nonce, 7U);
  126. sodium_add(nonce, nonce, 8U);
  127. printf("%s\n",
  128. sodium_bin2hex(nonce_hex, sizeof nonce_hex, nonce, sizeof nonce));
  129. memset(nonce, 0xfe, 24U);
  130. memset(nonce, 0xff, 10U);
  131. sodium_add(nonce, nonce, 11U);
  132. sodium_add(nonce, nonce, 12U);
  133. printf("%s\n",
  134. sodium_bin2hex(nonce_hex, sizeof nonce_hex, nonce, sizeof nonce));
  135. memset(nonce, 0xff, 22U);
  136. sodium_add(nonce, nonce, 23U);
  137. sodium_add(nonce, nonce, 24U);
  138. printf("%s\n",
  139. sodium_bin2hex(nonce_hex, sizeof nonce_hex, nonce, sizeof nonce));
  140. for (i = 0; i < 2000U; i++) {
  141. bin_len = randombytes_uniform(200U);
  142. blocksize = 1U + randombytes_uniform(100U);
  143. bin_padded_maxlen = bin_len + (blocksize - bin_len % blocksize);
  144. bin_padded = (unsigned char *) sodium_malloc(bin_padded_maxlen);
  145. randombytes_buf(bin_padded, bin_padded_maxlen);
  146. assert(sodium_pad(&bin_padded_len, bin_padded, bin_len,
  147. blocksize, bin_padded_maxlen - 1U) == -1);
  148. assert(sodium_pad(NULL, bin_padded, bin_len,
  149. blocksize, bin_padded_maxlen + 1U) == 0);
  150. assert(sodium_pad(&bin_padded_len, bin_padded, bin_len,
  151. blocksize, bin_padded_maxlen + 1U) == 0);
  152. assert(sodium_pad(&bin_padded_len, bin_padded, bin_len,
  153. 0U, bin_padded_maxlen) == -1);
  154. assert(sodium_pad(&bin_padded_len, bin_padded, bin_len,
  155. blocksize, bin_padded_maxlen) == 0);
  156. assert(bin_padded_len == bin_padded_maxlen);
  157. assert(sodium_unpad(&bin_len2, bin_padded, bin_padded_len,
  158. bin_padded_len + 1U) == -1);
  159. assert(sodium_unpad(&bin_len2, bin_padded, bin_padded_len,
  160. 0U) == -1);
  161. assert(sodium_unpad(&bin_len2, bin_padded, bin_padded_len,
  162. blocksize) == 0);
  163. assert(bin_len2 == bin_len);
  164. sodium_free(bin_padded);
  165. }
  166. return 0;
  167. }