diff options
| -rw-r--r-- | sw/trng_extractor.c | 24 |
1 files changed, 17 insertions, 7 deletions
diff --git a/sw/trng_extractor.c b/sw/trng_extractor.c index 5c373f4..fe37044 100644 --- a/sw/trng_extractor.c +++ b/sw/trng_extractor.c @@ -42,6 +42,8 @@ #include "cryptech.h" +int verbose; + char *usage = "%s [-a|r|c] [-n #] [-o file]\n\ \n\ @@ -50,6 +52,7 @@ char *usage = -c csprng (default data source)\n\ -n number of 4-byte samples (scale with K, M, or G suffix)\n\ -o output file (defaults to stdout)\n\ +-v verbose operation "; /* check availability of avalanche entropy core by reading core name and version */ @@ -103,14 +106,8 @@ int main(int argc, char *argv[]) FILE *output = stdout; uint32_t data; - // Check that we have can talk to the trng. - if (avalanche_check() || rosc_check() || csprng_check()) { - fprintf(stderr, "Can't properly access the trng.\n"); - return EXIT_FAILURE; - } - /* parse command line */ - while ((opt = getopt(argc, argv, "h?arcn:o:")) != -1) { + while ((opt = getopt(argc, argv, "h?varcn:o:")) != -1) { switch (opt) { case 'h': case '?': @@ -128,6 +125,9 @@ int main(int argc, char *argv[]) status_addr = CSPRNG_ADDR_STATUS; data_addr = CSPRNG_ADDR_RANDOM; break; + case 'v': + verbose = 1; + break; case 'n': num_words = strtoul(optarg, &endptr, 10); switch (toupper(*endptr)) { @@ -172,6 +172,16 @@ int main(int argc, char *argv[]) goto errout; } + // Check that we have can talk to the trng. + if (verbose) + printf("Checking that we can access the TRNG...\n"); + if (avalanche_check() || rosc_check() || csprng_check()) { + fprintf(stderr, "Can't properly access the trng.\n"); + return EXIT_FAILURE; + } + if (verbose) + printf("TRNG access ok..\n"); + /* get the data */ for (i = 0; i < num_words; ++i) { if (extract(status_addr, data_addr, &data) != 0) |