You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

hmac_test.cc 6.3 KiB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  1. /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
  2. * All rights reserved.
  3. *
  4. * This package is an SSL implementation written
  5. * by Eric Young (eay@cryptsoft.com).
  6. * The implementation was written so as to conform with Netscapes SSL.
  7. *
  8. * This library is free for commercial and non-commercial use as long as
  9. * the following conditions are aheared to. The following conditions
  10. * apply to all code found in this distribution, be it the RC4, RSA,
  11. * lhash, DES, etc., code; not just the SSL code. The SSL documentation
  12. * included with this distribution is covered by the same copyright terms
  13. * except that the holder is Tim Hudson (tjh@cryptsoft.com).
  14. *
  15. * Copyright remains Eric Young's, and as such any Copyright notices in
  16. * the code are not to be removed.
  17. * If this package is used in a product, Eric Young should be given attribution
  18. * as the author of the parts of the library used.
  19. * This can be in the form of a textual message at program startup or
  20. * in documentation (online or textual) provided with the package.
  21. *
  22. * Redistribution and use in source and binary forms, with or without
  23. * modification, are permitted provided that the following conditions
  24. * are met:
  25. * 1. Redistributions of source code must retain the copyright
  26. * notice, this list of conditions and the following disclaimer.
  27. * 2. Redistributions in binary form must reproduce the above copyright
  28. * notice, this list of conditions and the following disclaimer in the
  29. * documentation and/or other materials provided with the distribution.
  30. * 3. All advertising materials mentioning features or use of this software
  31. * must display the following acknowledgement:
  32. * "This product includes cryptographic software written by
  33. * Eric Young (eay@cryptsoft.com)"
  34. * The word 'cryptographic' can be left out if the rouines from the library
  35. * being used are not cryptographic related :-).
  36. * 4. If you include any Windows specific code (or a derivative thereof) from
  37. * the apps directory (application code) you must include an acknowledgement:
  38. * "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
  39. *
  40. * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
  41. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  42. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  43. * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  44. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  45. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  46. * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  47. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  48. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  49. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  50. * SUCH DAMAGE.
  51. *
  52. * The licence and distribution terms for any publically available version or
  53. * derivative of this code cannot be changed. i.e. this code cannot simply be
  54. * copied and put under another distribution licence
  55. * [including the GNU Public Licence.] */
  56. #include <stdio.h>
  57. #include <string.h>
  58. #include <string>
  59. #include <vector>
  60. #include <openssl/crypto.h>
  61. #include <openssl/digest.h>
  62. #include <openssl/hmac.h>
  63. #include "../test/file_test.h"
  64. #include "../test/scoped_types.h"
  65. #include "../test/stl_compat.h"
  66. static const EVP_MD *GetDigest(const std::string &name) {
  67. if (name == "MD5") {
  68. return EVP_md5();
  69. } else if (name == "SHA1") {
  70. return EVP_sha1();
  71. } else if (name == "SHA224") {
  72. return EVP_sha224();
  73. } else if (name == "SHA256") {
  74. return EVP_sha256();
  75. } else if (name == "SHA384") {
  76. return EVP_sha384();
  77. } else if (name == "SHA512") {
  78. return EVP_sha512();
  79. }
  80. return nullptr;
  81. }
  82. static bool TestHMAC(FileTest *t, void *arg) {
  83. std::string digest_str;
  84. if (!t->GetAttribute(&digest_str, "HMAC")) {
  85. return false;
  86. }
  87. const EVP_MD *digest = GetDigest(digest_str);
  88. if (digest == nullptr) {
  89. t->PrintLine("Unknown digest '%s'", digest_str.c_str());
  90. return false;
  91. }
  92. std::vector<uint8_t> key, input, output;
  93. if (!t->GetBytes(&key, "Key") ||
  94. !t->GetBytes(&input, "Input") ||
  95. !t->GetBytes(&output, "Output")) {
  96. return false;
  97. }
  98. // Test using the one-shot API.
  99. uint8_t mac[EVP_MAX_MD_SIZE];
  100. unsigned mac_len;
  101. if (nullptr == HMAC(digest, bssl::vector_data(&key), key.size(),
  102. bssl::vector_data(&input), input.size(), mac,
  103. &mac_len) ||
  104. !t->ExpectBytesEqual(bssl::vector_data(&output), output.size(), mac,
  105. mac_len)) {
  106. t->PrintLine("One-shot API failed.");
  107. return false;
  108. }
  109. // Test using HMAC_CTX.
  110. ScopedHMAC_CTX ctx;
  111. if (!HMAC_Init_ex(ctx.get(), bssl::vector_data(&key), key.size(), digest,
  112. nullptr) ||
  113. !HMAC_Update(ctx.get(), bssl::vector_data(&input), input.size()) ||
  114. !HMAC_Final(ctx.get(), mac, &mac_len) ||
  115. !t->ExpectBytesEqual(bssl::vector_data(&output), output.size(), mac,
  116. mac_len)) {
  117. t->PrintLine("HMAC_CTX failed.");
  118. return false;
  119. }
  120. // Test that an HMAC_CTX may be reset with the same key.
  121. if (!HMAC_Init_ex(ctx.get(), nullptr, 0, digest, nullptr) ||
  122. !HMAC_Update(ctx.get(), bssl::vector_data(&input), input.size()) ||
  123. !HMAC_Final(ctx.get(), mac, &mac_len) ||
  124. !t->ExpectBytesEqual(bssl::vector_data(&output), output.size(), mac,
  125. mac_len)) {
  126. t->PrintLine("HMAC_CTX with reset failed.");
  127. return false;
  128. }
  129. // Test feeding the input in byte by byte.
  130. if (!HMAC_Init_ex(ctx.get(), nullptr, 0, nullptr, nullptr)) {
  131. t->PrintLine("HMAC_CTX streaming failed.");
  132. return false;
  133. }
  134. for (size_t i = 0; i < input.size(); i++) {
  135. if (!HMAC_Update(ctx.get(), &input[i], 1)) {
  136. t->PrintLine("HMAC_CTX streaming failed.");
  137. return false;
  138. }
  139. }
  140. if (!HMAC_Final(ctx.get(), mac, &mac_len) ||
  141. !t->ExpectBytesEqual(bssl::vector_data(&output), output.size(), mac,
  142. mac_len)) {
  143. t->PrintLine("HMAC_CTX streaming failed.");
  144. return false;
  145. }
  146. return true;
  147. }
  148. int main(int argc, char *argv[]) {
  149. CRYPTO_library_init();
  150. if (argc != 2) {
  151. fprintf(stderr, "%s <test file.txt>\n", argv[0]);
  152. return 1;
  153. }
  154. return FileTestMain(TestHMAC, nullptr, argv[1]);
  155. }