load_roots.c 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192
  1. /*
  2. * Root CA reading application
  3. *
  4. * Copyright The Mbed TLS Contributors
  5. * SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
  6. *
  7. * This file is provided under the Apache License 2.0, or the
  8. * GNU General Public License v2.0 or later.
  9. *
  10. * **********
  11. * Apache License 2.0:
  12. *
  13. * Licensed under the Apache License, Version 2.0 (the "License"); you may
  14. * not use this file except in compliance with the License.
  15. * You may obtain a copy of the License at
  16. *
  17. * http://www.apache.org/licenses/LICENSE-2.0
  18. *
  19. * Unless required by applicable law or agreed to in writing, software
  20. * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  21. * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  22. * See the License for the specific language governing permissions and
  23. * limitations under the License.
  24. *
  25. * **********
  26. *
  27. * **********
  28. * GNU General Public License v2.0 or later:
  29. *
  30. * This program is free software; you can redistribute it and/or modify
  31. * it under the terms of the GNU General Public License as published by
  32. * the Free Software Foundation; either version 2 of the License, or
  33. * (at your option) any later version.
  34. *
  35. * This program is distributed in the hope that it will be useful,
  36. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  37. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  38. * GNU General Public License for more details.
  39. *
  40. * You should have received a copy of the GNU General Public License along
  41. * with this program; if not, write to the Free Software Foundation, Inc.,
  42. * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  43. *
  44. * **********
  45. */
  46. #include "mbedtls/build_info.h"
  47. #include "mbedtls/platform.h"
  48. #if !defined(MBEDTLS_X509_CRT_PARSE_C) || !defined(MBEDTLS_FS_IO) || \
  49. !defined(MBEDTLS_TIMING_C)
  50. int main(void)
  51. {
  52. mbedtls_printf("MBEDTLS_X509_CRT_PARSE_C and/or MBEDTLS_FS_IO and/or "
  53. "MBEDTLS_TIMING_C not defined.\n");
  54. mbedtls_exit(0);
  55. }
  56. #else
  57. #include "mbedtls/error.h"
  58. #include "mbedtls/timing.h"
  59. #include "mbedtls/x509_crt.h"
  60. #include <stdio.h>
  61. #include <stdlib.h>
  62. #include <string.h>
  63. #define DFL_ITERATIONS 1
  64. #define DFL_PRIME_CACHE 1
  65. #define USAGE \
  66. "\n usage: load_roots param=<>... [--] FILE...\n" \
  67. "\n acceptable parameters:\n" \
  68. " iterations=%%d Iteration count (not including cache priming); default: 1\n" \
  69. " prime=%%d Prime the disk read cache? Default: 1 (yes)\n" \
  70. "\n"
  71. /*
  72. * global options
  73. */
  74. struct options {
  75. const char **filenames; /* NULL-terminated list of file names */
  76. unsigned iterations; /* Number of iterations to time */
  77. int prime_cache; /* Prime the disk read cache? */
  78. } opt;
  79. int read_certificates(const char *const *filenames)
  80. {
  81. mbedtls_x509_crt cas;
  82. int ret = 0;
  83. const char *const *cur;
  84. mbedtls_x509_crt_init(&cas);
  85. for (cur = filenames; *cur != NULL; cur++) {
  86. ret = mbedtls_x509_crt_parse_file(&cas, *cur);
  87. if (ret != 0) {
  88. #if defined(MBEDTLS_ERROR_C) || defined(MBEDTLS_ERROR_STRERROR_DUMMY)
  89. char error_message[200];
  90. mbedtls_strerror(ret, error_message, sizeof(error_message));
  91. printf("\n%s: -0x%04x (%s)\n",
  92. *cur, (unsigned) -ret, error_message);
  93. #else
  94. printf("\n%s: -0x%04x\n",
  95. *cur, (unsigned) -ret);
  96. #endif
  97. goto exit;
  98. }
  99. }
  100. exit:
  101. mbedtls_x509_crt_free(&cas);
  102. return ret == 0;
  103. }
  104. int main(int argc, char *argv[])
  105. {
  106. int exit_code = MBEDTLS_EXIT_FAILURE;
  107. unsigned i, j;
  108. struct mbedtls_timing_hr_time timer;
  109. unsigned long ms;
  110. if (argc <= 1) {
  111. mbedtls_printf(USAGE);
  112. goto exit;
  113. }
  114. opt.filenames = NULL;
  115. opt.iterations = DFL_ITERATIONS;
  116. opt.prime_cache = DFL_PRIME_CACHE;
  117. for (i = 1; i < (unsigned) argc; i++) {
  118. char *p = argv[i];
  119. char *q = NULL;
  120. if (strcmp(p, "--") == 0) {
  121. break;
  122. }
  123. if ((q = strchr(p, '=')) == NULL) {
  124. break;
  125. }
  126. *q++ = '\0';
  127. for (j = 0; p + j < q; j++) {
  128. if (argv[i][j] >= 'A' && argv[i][j] <= 'Z') {
  129. argv[i][j] |= 0x20;
  130. }
  131. }
  132. if (strcmp(p, "iterations") == 0) {
  133. opt.iterations = atoi(q);
  134. } else if (strcmp(p, "prime") == 0) {
  135. opt.iterations = atoi(q) != 0;
  136. } else {
  137. mbedtls_printf("Unknown option: %s\n", p);
  138. mbedtls_printf(USAGE);
  139. goto exit;
  140. }
  141. }
  142. opt.filenames = (const char **) argv + i;
  143. if (*opt.filenames == 0) {
  144. mbedtls_printf("Missing list of certificate files to parse\n");
  145. goto exit;
  146. }
  147. mbedtls_printf("Parsing %u certificates", argc - i);
  148. if (opt.prime_cache) {
  149. if (!read_certificates(opt.filenames)) {
  150. goto exit;
  151. }
  152. mbedtls_printf(" ");
  153. }
  154. (void) mbedtls_timing_get_timer(&timer, 1);
  155. for (i = 1; i <= opt.iterations; i++) {
  156. if (!read_certificates(opt.filenames)) {
  157. goto exit;
  158. }
  159. mbedtls_printf(".");
  160. }
  161. ms = mbedtls_timing_get_timer(&timer, 0);
  162. mbedtls_printf("\n%u iterations -> %lu ms\n", opt.iterations, ms);
  163. exit_code = MBEDTLS_EXIT_SUCCESS;
  164. exit:
  165. mbedtls_exit(exit_code);
  166. }
  167. #endif /* necessary configuration */