Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(757)

Side by Side Diff: src/platform/vboot_reference/crypto/sha_utility.c

Issue 1585007: combined patch for: (Closed)
Patch Set: Created 10 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 /* Copyright (c) 2010 The Chromium OS Authors. All rights reserved. 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 2 * Use of this source code is governed by a BSD-style license that can be
3 * found in the LICENSE file. 3 * found in the LICENSE file.
4 * 4 *
5 * Utility functions for message digest functions. 5 * Utility functions for message digest functions.
6 */ 6 */
7 7
8 #include "sha_utility.h" 8 #include "cryptolib.h"
9
10 #include <fcntl.h>
11 #include <unistd.h>
12 #include <stdio.h>
13 #include <stdlib.h>
14 #include <string.h>
15 #include <sys/types.h>
16 #include <sys/stat.h>
17
18 #include "sha.h"
19 #include "utility.h" 9 #include "utility.h"
20 10
21 int digest_type_map[] = {
22 SHA1_DIGEST_ALGORITHM, /* RSA 1024 */
23 SHA256_DIGEST_ALGORITHM,
24 SHA512_DIGEST_ALGORITHM,
25 SHA1_DIGEST_ALGORITHM, /* RSA 2048 */
26 SHA256_DIGEST_ALGORITHM,
27 SHA512_DIGEST_ALGORITHM,
28 SHA1_DIGEST_ALGORITHM, /* RSA 4096 */
29 SHA256_DIGEST_ALGORITHM,
30 SHA512_DIGEST_ALGORITHM,
31 SHA1_DIGEST_ALGORITHM, /* RSA 8192 */
32 SHA256_DIGEST_ALGORITHM,
33 SHA512_DIGEST_ALGORITHM,
34 };
35
36 void DigestInit(DigestContext* ctx, int sig_algorithm) { 11 void DigestInit(DigestContext* ctx, int sig_algorithm) {
37 ctx->algorithm = digest_type_map[sig_algorithm]; 12 ctx->algorithm = hash_type_map[sig_algorithm];
38 switch(ctx->algorithm) { 13 switch(ctx->algorithm) {
39 case SHA1_DIGEST_ALGORITHM: 14 case SHA1_DIGEST_ALGORITHM:
40 ctx->sha1_ctx = (SHA1_CTX*) Malloc(sizeof(SHA1_CTX)); 15 ctx->sha1_ctx = (SHA1_CTX*) Malloc(sizeof(SHA1_CTX));
41 SHA1_init(ctx->sha1_ctx); 16 SHA1_init(ctx->sha1_ctx);
42 break; 17 break;
43 case SHA256_DIGEST_ALGORITHM: 18 case SHA256_DIGEST_ALGORITHM:
44 ctx->sha256_ctx = (SHA256_CTX*) Malloc(sizeof(SHA256_CTX)); 19 ctx->sha256_ctx = (SHA256_CTX*) Malloc(sizeof(SHA256_CTX));
45 SHA256_init(ctx->sha256_ctx); 20 SHA256_init(ctx->sha256_ctx);
46 break; 21 break;
47 case SHA512_DIGEST_ALGORITHM: 22 case SHA512_DIGEST_ALGORITHM:
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after
80 break; 55 break;
81 case SHA512_DIGEST_ALGORITHM: 56 case SHA512_DIGEST_ALGORITHM:
82 digest = (uint8_t*) Malloc(SHA512_DIGEST_SIZE); 57 digest = (uint8_t*) Malloc(SHA512_DIGEST_SIZE);
83 Memcpy(digest, SHA512_final(ctx->sha512_ctx), SHA512_DIGEST_SIZE); 58 Memcpy(digest, SHA512_final(ctx->sha512_ctx), SHA512_DIGEST_SIZE);
84 Free(ctx->sha512_ctx); 59 Free(ctx->sha512_ctx);
85 break; 60 break;
86 }; 61 };
87 return digest; 62 return digest;
88 } 63 }
89 64
90 uint8_t* DigestFile(char* input_file, int sig_algorithm) {
91 int input_fd, len;
92 uint8_t data[SHA1_BLOCK_SIZE];
93 uint8_t* digest = NULL;
94 DigestContext ctx;
95
96 if( (input_fd = open(input_file, O_RDONLY)) == -1 ) {
97 fprintf(stderr, "Couldn't open input file.\n");
98 return NULL;
99 }
100 DigestInit(&ctx, sig_algorithm);
101 while ( (len = read(input_fd, data, SHA1_BLOCK_SIZE)) ==
102 SHA1_BLOCK_SIZE)
103 DigestUpdate(&ctx, data, len);
104 if (len != -1)
105 DigestUpdate(&ctx, data, len);
106 digest = DigestFinal(&ctx);
107 close(input_fd);
108 return digest;
109 }
110
111 uint8_t* DigestBuf(const uint8_t* buf, uint64_t len, int sig_algorithm) { 65 uint8_t* DigestBuf(const uint8_t* buf, uint64_t len, int sig_algorithm) {
112 uint8_t* digest = (uint8_t*) Malloc(SHA512_DIGEST_SIZE); /* Use the max. */ 66 uint8_t* digest = (uint8_t*) Malloc(SHA512_DIGEST_SIZE); /* Use the max. */
113 /* Define an array mapping [sig_algorithm] to function pointers to the 67 /* Define an array mapping [sig_algorithm] to function pointers to the
114 * SHA{1|256|512} functions. 68 * SHA{1|256|512} functions.
115 */ 69 */
116 typedef uint8_t* (*Hash_ptr) (const uint8_t*, uint64_t, uint8_t*); 70 typedef uint8_t* (*Hash_ptr) (const uint8_t*, uint64_t, uint8_t*);
117 Hash_ptr hash[] = { 71 Hash_ptr hash[] = {
118 SHA1, /* RSA 1024 */ 72 SHA1, /* RSA 1024 */
119 SHA256, 73 SHA256,
120 SHA512, 74 SHA512,
121 SHA1, /* RSA 2048 */ 75 SHA1, /* RSA 2048 */
122 SHA256, 76 SHA256,
123 SHA512, 77 SHA512,
124 SHA1, /* RSA 4096 */ 78 SHA1, /* RSA 4096 */
125 SHA256, 79 SHA256,
126 SHA512, 80 SHA512,
127 SHA1, /* RSA 8192 */ 81 SHA1, /* RSA 8192 */
128 SHA256, 82 SHA256,
129 SHA512, 83 SHA512,
130 }; 84 };
131 /* Call the appropriate hash function. */ 85 /* Call the appropriate hash function. */
132 return hash[sig_algorithm](buf, len, digest); 86 return hash[sig_algorithm](buf, len, digest);
133 } 87 }
OLDNEW
« no previous file with comments | « src/platform/vboot_reference/crypto/sha2.c ('k') | src/platform/vboot_reference/include/cryptolib.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698