1 /* Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
2 * Use of this source code is governed by a BSD-style license that can be
3 * found in the LICENSE file.
4 */
5
6 /* Routines for verifying a file's signature. Useful in testing the core
7 * RSA verification implementation.
8 */
9
10 #include <fcntl.h>
11 #include <stdio.h>
12 #include <stdlib.h>
13 #include <string.h>
14 #include <sys/stat.h>
15 #include <sys/types.h>
16 #include <unistd.h>
17
18 #include "cryptolib.h"
19 #include "file_keys.h"
20 #include "verify_data.h"
21
22 /* ANSI Color coding sequences. */
23 #define COL_GREEN "\e[1;32m"
24 #define COL_RED "\e[0;31m"
25 #define COL_STOP "\e[m"
26
read_signature(char * input_file,int len)27 uint8_t* read_signature(char* input_file, int len) {
28 int i, sigfd;
29 uint8_t* signature = NULL;
30 if ((sigfd = open(input_file, O_RDONLY)) == -1) {
31 fprintf(stderr, "Couldn't open signature file\n");
32 return NULL;
33 }
34
35 /* Read the signature into a buffer*/
36 signature = (uint8_t*) malloc(len);
37 if (!signature)
38 return NULL;
39
40 if( (i = read(sigfd, signature, len)) != len ) {
41 fprintf(stderr, "Wrong signature length - Expected = %d, Received = %d\n",
42 len, i);
43 close(sigfd);
44 return NULL;
45 }
46
47 close(sigfd);
48 return signature;
49 }
50
main(int argc,char * argv[])51 int main(int argc, char* argv[]) {
52 int i, algorithm, sig_len;
53 int return_code = 1; /* Default to error. */
54 uint8_t* digest = NULL;
55 uint8_t* signature = NULL;
56 RSAPublicKey* key = NULL;
57
58 if (argc!=5) {
59 fprintf(stderr, "Usage: %s <algorithm> <key file> <signature file>"
60 " <input file>\n\n", argv[0]);
61 fprintf(stderr, "where <algorithm> depends on the signature algorithm"
62 " used:\n");
63 for(i = 0; i<kNumAlgorithms; i++)
64 fprintf(stderr, "\t%d for %s\n", i, algo_strings[i]);
65 return -1;
66 }
67
68 algorithm = atoi(argv[1]);
69 if (algorithm >= kNumAlgorithms) {
70 fprintf(stderr, "Invalid Algorithm!\n");
71 return 0;
72 }
73 /* Length of the RSA Signature/RSA Key */
74 sig_len = siglen_map[algorithm];
75 if ((key = RSAPublicKeyFromFile(argv[2])) &&
76 (signature = read_signature(argv[3], sig_len)) &&
77 (digest = DigestFile(argv[4], algorithm))) {
78 if (RSAVerify(key, signature, sig_len, algorithm, digest)) {
79 return_code = 0;
80 fprintf(stderr, "Signature Verification "
81 COL_GREEN "SUCCEEDED" COL_STOP "\n");
82 } else {
83 fprintf(stderr, "Signature Verification "
84 COL_RED "FAILED" COL_STOP "\n");
85 }
86 }
87 else
88 return_code = -1;
89
90 free(key);
91 free(signature);
92 free(digest);
93
94 return return_code;
95 }
96