Merge branch 'reenc' of https://code.google.com/p/cryptsetup into reenc
[platform/upstream/cryptsetup.git] / lib / verity / verity_hash.c
1 /*
2  * dm-verity volume handling
3  *
4  * Copyright (C) 2012, Red Hat, Inc. All rights reserved.
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * version 2 as published by the Free Software Foundation.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18  */
19
20 #include <errno.h>
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <stdint.h>
25
26 #include "verity.h"
27 #include "internal.h"
28
29 #define VERITY_MAX_LEVELS       63
30
31 static unsigned get_bits_up(size_t u)
32 {
33         unsigned i = 0;
34         while ((1U << i) < u)
35                 i++;
36         return i;
37 }
38
39 static unsigned get_bits_down(size_t u)
40 {
41         unsigned i = 0;
42         while ((u >> i) > 1U)
43                 i++;
44         return i;
45 }
46
47 static int verify_zero(struct crypt_device *cd, FILE *wr, size_t bytes)
48 {
49         char block[bytes];
50         size_t i;
51
52         if (fread(block, bytes, 1, wr) != 1) {
53                 log_dbg("EIO while reading spare area.");
54                 return -EIO;
55         }
56         for (i = 0; i < bytes; i++)
57                 if (block[i]) {
58                         log_err(cd, _("Spare area is not zeroed at position %" PRIu64 ".\n"),
59                                 ftello(wr) - bytes);
60                         return -EPERM;
61                 }
62         return 0;
63 }
64
65 static int verify_hash_block(const char *hash_name, int version,
66                               char *hash, size_t hash_size,
67                               const char *data, size_t data_size,
68                               const char *salt, size_t salt_size)
69 {
70         struct crypt_hash *ctx = NULL;
71         int r;
72
73         if (crypt_hash_init(&ctx, hash_name))
74                 return -EINVAL;
75
76         if (version == 1 && (r = crypt_hash_write(ctx, salt, salt_size)))
77                 goto out;
78
79         if ((r = crypt_hash_write(ctx, data, data_size)))
80                 goto out;
81
82         if (version == 0 && (r = crypt_hash_write(ctx, salt, salt_size)))
83                 goto out;
84
85         r = crypt_hash_final(ctx, hash, hash_size);
86 out:
87         crypt_hash_destroy(ctx);
88         return r;
89 }
90
91 static int mult_overflow(off_t *u, off_t b, size_t size)
92 {
93         *u = (uint64_t)b * size;
94         if ((off_t)(*u / size) != b || (off_t)*u < 0 || (off_t)*u != *u)
95                 return 1;
96         return 0;
97 }
98
99 static int create_or_verify(struct crypt_device *cd, FILE *rd, FILE *wr,
100                                    off_t data_block, size_t data_block_size,
101                                    off_t hash_block, size_t hash_block_size,
102                                    off_t blocks, int version,
103                                    const char *hash_name, int verify,
104                                    char *calculated_digest, size_t digest_size,
105                                    const char *salt, size_t salt_size)
106 {
107         char left_block[hash_block_size];
108         char data_buffer[data_block_size];
109         char read_digest[digest_size];
110         size_t hash_per_block = 1 << get_bits_down(hash_block_size / digest_size);
111         size_t digest_size_full = 1 << get_bits_up(digest_size);
112         off_t blocks_to_write = (blocks + hash_per_block - 1) / hash_per_block;
113         off_t seek_rd, seek_wr;
114         size_t left_bytes;
115         unsigned i;
116         int r;
117
118         if (mult_overflow(&seek_rd, data_block, data_block_size) ||
119             mult_overflow(&seek_wr, hash_block, hash_block_size)) {
120                 log_err(cd, _("Device offset overflow.\n"));
121                 return -EINVAL;
122         }
123
124         if (fseeko(rd, seek_rd, SEEK_SET)) {
125                 log_dbg("Cannot seek to requested position in data device.");
126                 return -EIO;
127         }
128
129         if (wr && fseeko(wr, seek_wr, SEEK_SET)) {
130                 log_dbg("Cannot seek to requested position in hash device.");
131                 return -EIO;
132         }
133
134         memset(left_block, 0, hash_block_size);
135         while (blocks_to_write--) {
136                 left_bytes = hash_block_size;
137                 for (i = 0; i < hash_per_block; i++) {
138                         if (!blocks)
139                                 break;
140                         blocks--;
141                         if (fread(data_buffer, data_block_size, 1, rd) != 1) {
142                                 log_dbg("Cannot read data device block.");
143                                 return -EIO;
144                         }
145
146                         if (verify_hash_block(hash_name, version,
147                                         calculated_digest, digest_size,
148                                         data_buffer, data_block_size,
149                                         salt, salt_size))
150                                 return -EINVAL;
151
152                         if (!wr)
153                                 break;
154                         if (verify) {
155                                 if (fread(read_digest, digest_size, 1, wr) != 1) {
156                                         log_dbg("Cannot read digest form hash device.");
157                                         return -EIO;
158                                 }
159                                 if (memcmp(read_digest, calculated_digest, digest_size)) {
160                                         log_err(cd, _("Verification failed at position %" PRIu64 ".\n"),
161                                                 ftello(rd) - data_block_size);
162                                         return -EPERM;
163                                 }
164                         } else {
165                                 if (fwrite(calculated_digest, digest_size, 1, wr) != 1) {
166                                         log_dbg("Cannot write digest to hash device.");
167                                         return -EIO;
168                                 }
169                         }
170                         if (version == 0) {
171                                 left_bytes -= digest_size;
172                         } else {
173                                 if (digest_size_full - digest_size) {
174                                         if (verify) {
175                                                 r = verify_zero(cd, wr, digest_size_full - digest_size);
176                                                 if (r)
177                                                         return r;
178                                         } else if (fwrite(left_block, digest_size_full - digest_size, 1, wr) != 1) {
179                                                 log_dbg("Cannot write spare area to hash device.");
180                                                 return -EIO;
181                                         }
182                                 }
183                                 left_bytes -= digest_size_full;
184                         }
185                 }
186                 if (wr && left_bytes) {
187                         if (verify) {
188                                 r = verify_zero(cd , wr, left_bytes);
189                                 if (r)
190                                         return r;
191                         } else if (fwrite(left_block, left_bytes, 1, wr) != 1) {
192                                 log_dbg("Cannot write remaining spare area to hash device.");
193                                 return -EIO;
194                         }
195                 }
196         }
197
198         return 0;
199 }
200
201 static int VERITY_create_or_verify_hash(struct crypt_device *cd,
202         int verify,
203         int version,
204         const char *hash_name,
205         const char *hash_device,
206         const char *data_device,
207         size_t hash_block_size,
208         size_t data_block_size,
209         off_t data_blocks,
210         off_t hash_position,
211         char *root_hash,
212         size_t digest_size,
213         const char *salt,
214         size_t salt_size)
215 {
216         char calculated_digest[digest_size];
217         FILE *data_file = NULL;
218         FILE *hash_file = NULL, *hash_file_2;
219         off_t hash_level_block[VERITY_MAX_LEVELS];
220         off_t hash_level_size[VERITY_MAX_LEVELS];
221         off_t data_file_blocks, s;
222         size_t hash_per_block, hash_per_block_bits;
223         off_t data_device_size = 0, hash_device_size = 0;
224         uint64_t dev_size;
225         int levels, i, r;
226
227         log_dbg("Hash %s %s, data device %s, data blocks %" PRIu64
228                 ", hash_device %s, offset %" PRIu64 ".",
229                 verify ? "verification" : "creation", hash_name,
230                 data_device, data_blocks, hash_device, hash_position);
231
232         if (data_blocks < 0 || hash_position < 0) {
233                 log_err(cd, _("Invalid size parameters for verity device.\n"));
234                 return -EINVAL;
235         }
236
237         if (!data_blocks) {
238                 r = device_size(data_device, &dev_size);
239                 if (r < 0)
240                         return r;
241
242                 data_file_blocks = dev_size / data_block_size;
243         } else
244                 data_file_blocks = data_blocks;
245
246         if (mult_overflow(&data_device_size, data_blocks, data_block_size)) {
247                 log_err(cd, _("Device offset overflow.\n"));
248                 return -EINVAL;
249         }
250
251         hash_per_block_bits = get_bits_down(hash_block_size / digest_size);
252         hash_per_block = 1 << hash_per_block_bits;
253         if (!hash_per_block_bits)
254                 return -EINVAL;
255
256         levels = 0;
257         if (data_file_blocks) {
258                 while (hash_per_block_bits * levels < 64 &&
259                        (data_file_blocks - 1) >> (hash_per_block_bits * levels))
260                         levels++;
261         }
262         log_dbg("Using %d hash levels.", levels);
263
264         if (levels > VERITY_MAX_LEVELS) {
265                 log_err(cd, _("Too many tree levels for verity volume.\n"));
266                 return -EINVAL;
267         }
268
269         for (i = levels - 1; i >= 0; i--) {
270                 hash_level_block[i] = hash_position;
271                 // verity position of block data_file_blocks at level i
272                 s = data_file_blocks >> (i * hash_per_block_bits);
273                 s = (s + hash_per_block - 1) / hash_per_block;
274                 hash_level_size[i] = s;
275                 if (hash_position + s < hash_position ||
276                     (hash_position + s) < 0 ||
277                     (hash_position + s) != hash_position + s) {
278                         log_err(cd, _("Device offset overflow.\n"));
279                         return -EINVAL;
280                 }
281                 hash_position += s;
282         }
283
284         if (mult_overflow(&hash_device_size, hash_position, hash_block_size)) {
285                 log_err(cd, _("Device offset overflow.\n"));
286                 return -EINVAL;
287         }
288
289         log_dbg("Data device size required: %" PRIu64 " bytes.",
290                 data_device_size);
291         data_file = fopen(data_device, "r");
292         if (!data_file) {
293                 log_err(cd, _("Cannot open device %s.\n"), data_device);
294                 r = -EIO;
295                 goto out;
296         }
297
298         log_dbg("Hash device size required: %" PRIu64 " bytes.",
299                 hash_device_size);
300         hash_file = fopen(hash_device, verify ? "r" : "r+");
301         if (!hash_file) {
302                 log_err(cd, _("Cannot open device %s.\n"), hash_device);
303                 r = -EIO;
304                 goto out;
305         }
306
307         memset(calculated_digest, 0, digest_size);
308
309         for (i = 0; i < levels; i++) {
310                 if (!i) {
311                         r = create_or_verify(cd, data_file, hash_file,
312                                                     0, data_block_size,
313                                                     hash_level_block[i], hash_block_size,
314                                                     data_file_blocks, version, hash_name, verify,
315                                                     calculated_digest, digest_size, salt, salt_size);
316                         if (r)
317                                 goto out;
318                 } else {
319                         hash_file_2 = fopen(hash_device, "r");
320                         if (!hash_file_2) {
321                                 log_err(cd, _("Cannot open device %s.\n"), hash_device);
322                                 r = -EIO;
323                                 goto out;
324                         }
325                         r = create_or_verify(cd, hash_file_2, hash_file,
326                                                     hash_level_block[i - 1], hash_block_size,
327                                                     hash_level_block[i], hash_block_size,
328                                                     hash_level_size[i - 1], version, hash_name, verify,
329                                                     calculated_digest, digest_size, salt, salt_size);
330                         fclose(hash_file_2);
331                         if (r)
332                                 goto out;
333                 }
334         }
335
336         if (levels)
337                 r = create_or_verify(cd, hash_file, NULL,
338                                             hash_level_block[levels - 1], hash_block_size,
339                                             0, hash_block_size,
340                                             1, version, hash_name, verify,
341                                             calculated_digest, digest_size, salt, salt_size);
342         else
343                 r = create_or_verify(cd, data_file, NULL,
344                                             0, data_block_size,
345                                             0, hash_block_size,
346                                             data_file_blocks, version, hash_name, verify,
347                                             calculated_digest, digest_size, salt, salt_size);
348 out:
349         if (verify) {
350                 if (r)
351                         log_err(cd, _("Verification of data area failed.\n"));
352                 else {
353                         log_dbg("Verification of data area succeeded.");
354                         r = memcmp(root_hash, calculated_digest, digest_size) ? -EPERM : 0;
355                         if (r)
356                                 log_err(cd, _("Verification of root hash failed.\n"));
357                         else
358                                 log_dbg("Verification of root hash succeeded.");
359                 }
360         } else {
361                 if (r == -EIO)
362                         log_err(cd, _("Input/output error while creating hash area.\n"));
363                 else if (r)
364                         log_err(cd, _("Creation of hash area failed.\n"));
365                 else {
366                         fsync(fileno(hash_file));
367                         memcpy(root_hash, calculated_digest, digest_size);
368                 }
369         }
370
371         if (data_file)
372                 fclose(data_file);
373         if (hash_file)
374                 fclose(hash_file);
375         return r;
376 }
377
378 /* Verify verity device using userspace crypto backend */
379 int VERITY_verify(struct crypt_device *cd,
380                   struct crypt_params_verity *verity_hdr,
381                   const char *data_device,
382                   const char *hash_device,
383                   const char *root_hash,
384                   size_t root_hash_size)
385 {
386         return VERITY_create_or_verify_hash(cd, 1,
387                 verity_hdr->hash_type,
388                 verity_hdr->hash_name,
389                 hash_device,
390                 data_device,
391                 verity_hdr->hash_block_size,
392                 verity_hdr->data_block_size,
393                 verity_hdr->data_size,
394                 VERITY_hash_offset_block(verity_hdr),
395                 CONST_CAST(char*)root_hash,
396                 root_hash_size,
397                 verity_hdr->salt,
398                 verity_hdr->salt_size);
399 }
400
401 /* Create verity hash */
402 int VERITY_create(struct crypt_device *cd,
403                   struct crypt_params_verity *verity_hdr,
404                   const char *data_device,
405                   const char *hash_device,
406                   char *root_hash,
407                   size_t root_hash_size)
408 {
409         unsigned pgsize = crypt_getpagesize();
410
411         if (verity_hdr->salt_size > 256)
412                 return -EINVAL;
413
414         if (verity_hdr->data_block_size > pgsize)
415                 log_err(cd, _("WARNING: Kernel cannot activate device if data "
416                               "block size exceeds page size (%u).\n"), pgsize);
417
418         return VERITY_create_or_verify_hash(cd, 0,
419                 verity_hdr->hash_type,
420                 verity_hdr->hash_name,
421                 hash_device,
422                 data_device,
423                 verity_hdr->hash_block_size,
424                 verity_hdr->data_block_size,
425                 verity_hdr->data_size,
426                 VERITY_hash_offset_block(verity_hdr),
427                 root_hash,
428                 root_hash_size,
429                 verity_hdr->salt,
430                 verity_hdr->salt_size);
431 }