Handle kernel crypto api init failure better.
[platform/upstream/cryptsetup.git] / lib / tcrypt / tcrypt.c
1 /*
2  * TCRYPT compatible volume handling
3  *
4  * Copyright (C) 2012, Red Hat, Inc. All rights reserved.
5  * Copyright (C) 2012, Milan Broz
6  *
7  * This program is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License
9  * version 2 as published by the Free Software Foundation.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19  */
20
21 #include <errno.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <fcntl.h>
26 #include <assert.h>
27
28 #include "libcryptsetup.h"
29 #include "tcrypt.h"
30 #include "internal.h"
31
32 /* TCRYPT PBKDF variants */
33 static struct {
34         unsigned int legacy:1;
35         char *name;
36         char *hash;
37         unsigned int iterations;
38 } tcrypt_kdf[] = {
39         { 0, "pbkdf2", "ripemd160", 2000 },
40         { 0, "pbkdf2", "ripemd160", 1000 },
41         { 0, "pbkdf2", "sha512",    1000 },
42         { 0, "pbkdf2", "whirlpool", 1000 },
43         { 1, "pbkdf2", "sha1",      2000 },
44         { 0, NULL,     NULL,           0 }
45 };
46
47 struct tcrypt_alg {
48                 const char *name;
49                 unsigned int key_size;
50                 unsigned int iv_size;
51                 unsigned int key_offset;
52                 unsigned int iv_offset; /* or tweak key offset */
53 };
54
55 struct tcrypt_algs {
56         unsigned int legacy:1;
57         unsigned int chain_count;
58         unsigned int chain_key_size;
59         const char *long_name;
60         const char *mode;
61         struct tcrypt_alg cipher[3];
62 };
63
64 /* TCRYPT cipher variants */
65 static struct tcrypt_algs tcrypt_cipher[] = {
66 /* XTS mode */
67 {0,1,64,"aes","xts-plain64",
68         {{"aes",    64,16,0,32}}},
69 {0,1,64,"serpent","xts-plain64",
70         {{"serpent",64,16,0,32}}},
71 {0,1,64,"twofish","xts-plain64",
72         {{"twofish",64,16,0,32}}},
73 {0,2,128,"twofish-aes","xts-plain64",
74         {{"twofish",64,16, 0,64},
75          {"aes",    64,16,32,96}}},
76 {0,3,192,"serpent-twofish-aes","xts-plain64",
77         {{"serpent",64,16, 0, 96},
78          {"twofish",64,16,32,128},
79          {"aes",    64,16,64,160}}},
80 {0,2,128,"aes-serpent","xts-plain64",
81         {{"aes",    64,16, 0,64},
82          {"serpent",64,16,32,96}}},
83 {0,3,192,"aes-twofish-serpent","xts-plain64",
84         {{"aes",    64,16, 0, 96},
85          {"twofish",64,16,32,128},
86          {"serpent",64,16,64,160}}},
87 {0,2,128,"serpent-twofish","xts-plain64",
88         {{"serpent",64,16, 0,64},
89          {"twofish",64,16,32,96}}},
90 /* LRW mode */
91 {0,1,48,"aes","lrw-benbi",
92         {{"aes",    48,16,32,0}}},
93 {0,1,48,"serpent","lrw-benbi",
94         {{"serpent",48,16,32,0}}},
95 {0,1,48,"twofish","lrw-benbi",
96         {{"twofish",48,16,32,0}}},
97 {0,2,96,"twofish-aes","lrw-benbi",
98         {{"twofish",48,16,32,0},
99          {"aes",    48,16,64,0}}},
100 {0,3,144,"serpent-twofish-aes","lrw-benbi",
101         {{"serpent",48,16,32,0},
102          {"twofish",48,16,64,0},
103          {"aes",    48,16,96,0}}},
104 {0,2,96,"aes-serpent","lrw-benbi",
105         {{"aes",    48,16,32,0},
106          {"serpent",48,16,64,0}}},
107 {0,3,144,"aes-twofish-serpent","lrw-benbi",
108         {{"aes",    48,16,32,0},
109          {"twofish",48,16,64,0},
110          {"serpent",48,16,96,0}}},
111 {0,2,96,"serpent-twofish", "lrw-benbi",
112         {{"serpent",48,16,32,0},
113          {"twofish",48,16,64,0}}},
114 /* Kernel LRW block size is fixed to 16 bytes for GF(2^128)
115  * thus cannot be used with blowfish where block is 8 bytes.
116  * There also no GF(2^64) support.
117 {1,1,64,"blowfish_le","lrw-benbi",
118          {{"blowfish_le",64,8,32,0}}},
119 {1,2,112,"blowfish_le-aes","lrw-benbi",
120          {{"blowfish_le",64, 8,32,0},
121           {"aes",        48,16,88,0}}},
122 {1,3,160,"serpent-blowfish_le-aes","lrw-benbi",
123           {{"serpent",    48,16, 32,0},
124            {"blowfish_le",64, 8, 64,0},
125            {"aes",        48,16,120,0}}},*/
126 /* CBC + "outer" CBC (both with whitening) */
127 {1,1,32,"aes","cbc-tcrypt",
128         {{"aes",    32,16,32,0}}},
129 {1,1,32,"serpent","cbc-tcrypt",
130         {{"serpent",32,16,32,0}}},
131 {1,1,32,"twofish","cbc-tcrypt",
132         {{"twofish",32,16,32,0}}},
133 {1,2,64,"twofish-aes","cbci-tcrypt",
134         {{"twofish",32,16,32,0},
135          {"aes",    32,16,64,0}}},
136 {1,3,96,"serpent-twofish-aes","cbci-tcrypt",
137         {{"serpent",32,16,32,0},
138          {"twofish",32,16,64,0},
139          {"aes",    32,16,96,0}}},
140 {1,2,64,"aes-serpent","cbci-tcrypt",
141         {{"aes",    32,16,32,0},
142          {"serpent",32,16,64,0}}},
143 {1,3,96,"aes-twofish-serpent", "cbci-tcrypt",
144         {{"aes",    32,16,32,0},
145          {"twofish",32,16,64,0},
146          {"serpent",32,16,96,0}}},
147 {1,2,64,"serpent-twofish", "cbci-tcrypt",
148         {{"serpent",32,16,32,0},
149          {"twofish",32,16,64,0}}},
150 {1,1,16,"cast5","cbc-tcrypt",
151         {{"cast5",   16,8,32,0}}},
152 {1,1,24,"des3_ede","cbc-tcrypt",
153         {{"des3_ede",24,8,32,0}}},
154 {1,1,56,"blowfish_le","cbc-tcrypt",
155         {{"blowfish_le",56,8,32,0}}},
156 {1,2,88,"blowfish_le-aes","cbc-tcrypt",
157         {{"blowfish_le",56, 8,32,0},
158          {"aes",        32,16,88,0}}},
159 {1,3,120,"serpent-blowfish_le-aes","cbc-tcrypt",
160         {{"serpent",    32,16, 32,0},
161          {"blowfish_le",56, 8, 64,0},
162          {"aes",        32,16,120,0}}},
163 {}
164 };
165
166 static int hdr_from_disk(struct tcrypt_phdr *hdr,
167                          struct crypt_params_tcrypt *params,
168                          int kdf_index, int cipher_index)
169 {
170         uint32_t crc32;
171         size_t size;
172
173         /* Check CRC32 of header */
174         size = TCRYPT_HDR_LEN - sizeof(hdr->d.keys) - sizeof(hdr->d.header_crc32);
175         crc32 = crypt_crc32(~0, (unsigned char*)&hdr->d, size) ^ ~0;
176         if (be16_to_cpu(hdr->d.version) > 3 &&
177             crc32 != be32_to_cpu(hdr->d.header_crc32)) {
178                 log_dbg("TCRYPT header CRC32 mismatch.");
179                 return -EINVAL;
180         }
181
182         /* Check CRC32 of keys */
183         crc32 = crypt_crc32(~0, (unsigned char*)hdr->d.keys, sizeof(hdr->d.keys)) ^ ~0;
184         if (crc32 != be32_to_cpu(hdr->d.keys_crc32)) {
185                 log_dbg("TCRYPT keys CRC32 mismatch.");
186                 return -EINVAL;
187         }
188
189         /* Convert header to cpu format */
190         hdr->d.version  =  be16_to_cpu(hdr->d.version);
191         hdr->d.version_tc = le16_to_cpu(hdr->d.version_tc);
192
193         hdr->d.keys_crc32 = be32_to_cpu(hdr->d.keys_crc32);
194
195         hdr->d.hidden_volume_size = be64_to_cpu(hdr->d.hidden_volume_size);
196         hdr->d.volume_size        = be64_to_cpu(hdr->d.volume_size);
197
198         hdr->d.mk_offset = be64_to_cpu(hdr->d.mk_offset);
199         if (!hdr->d.mk_offset)
200                 hdr->d.mk_offset = 512;
201
202         hdr->d.mk_size = be64_to_cpu(hdr->d.mk_size);
203
204         hdr->d.flags = be32_to_cpu(hdr->d.flags);
205
206         hdr->d.sector_size = be32_to_cpu(hdr->d.sector_size);
207         if (!hdr->d.sector_size)
208                 hdr->d.sector_size = 512;
209
210         hdr->d.header_crc32 = be32_to_cpu(hdr->d.header_crc32);
211
212         /* Set params */
213         params->passphrase = NULL;
214         params->passphrase_size = 0;
215         params->hash_name  = tcrypt_kdf[kdf_index].hash;
216         params->key_size = tcrypt_cipher[cipher_index].chain_key_size;
217         params->cipher = tcrypt_cipher[cipher_index].long_name;
218         params->mode = tcrypt_cipher[cipher_index].mode;
219
220         return 0;
221 }
222
223 /*
224  * Kernel implements just big-endian version of blowfish, hack it here
225  */
226 static void blowfish_le(char *buf)
227 {
228         uint32_t *l = (uint32_t*)&buf[0];
229         uint32_t *r = (uint32_t*)&buf[4];
230         *l = swab32(*l);
231         *r = swab32(*r);
232 }
233
234 static int decrypt_blowfish_le_cbc(struct tcrypt_alg *alg,
235                                    const char *key, char *buf)
236 {
237         char iv[alg->iv_size], iv_old[alg->iv_size];
238         struct crypt_cipher *cipher = NULL;
239         int bs = alg->iv_size;
240         int i, j, r;
241
242         assert(bs == 2*sizeof(uint32_t));
243
244         r = crypt_cipher_init(&cipher, "blowfish", "ecb",
245                               &key[alg->key_offset], alg->key_size);
246         if (r < 0)
247                 return r;
248
249         memcpy(iv, &key[alg->iv_offset], alg->iv_size);
250         for (i = 0; i < TCRYPT_HDR_LEN; i += bs) {
251                 memcpy(iv_old, &buf[i], bs);
252                 blowfish_le(&buf[i]);
253                 r = crypt_cipher_decrypt(cipher, &buf[i], &buf[i],
254                                           bs, NULL, 0);
255                 blowfish_le(&buf[i]);
256                 if (r < 0)
257                         break;
258                 for (j = 0; j < bs; j++)
259                         buf[i + j] ^= iv[j];
260                 memcpy(iv, iv_old, bs);
261         }
262
263         crypt_cipher_destroy(cipher);
264         return r;
265 }
266
267 static void remove_whitening(char *buf, const char *key)
268 {
269         int j;
270
271         for (j = 0; j < TCRYPT_HDR_LEN; j++)
272                 buf[j] ^= key[j % 8];
273 }
274
275 static void copy_key(struct tcrypt_alg *alg, const char *mode,
276                      char *out_key, const char *key)
277 {
278         int ks2;
279         if (!strncmp(mode, "xts", 3)) {
280                 ks2 = alg->key_size / 2;
281                 memcpy(out_key, &key[alg->key_offset], ks2);
282                 memcpy(&out_key[ks2], &key[alg->iv_offset], ks2);
283         } else if (!strncmp(mode, "lrw", 3)) {
284                 ks2 = alg->key_size - TCRYPT_LRW_IKEY_LEN;
285                 memcpy(out_key, &key[alg->key_offset], ks2);
286                 memcpy(&out_key[ks2], key, TCRYPT_LRW_IKEY_LEN);
287         } else if (!strncmp(mode, "cbc", 3)) {
288                 memcpy(out_key, &key[alg->key_offset], alg->key_size);
289         }
290 }
291
292 static int decrypt_hdr_one(struct tcrypt_alg *alg, const char *mode,
293                            const char *key,struct tcrypt_phdr *hdr)
294 {
295         char backend_key[TCRYPT_HDR_KEY_LEN];
296         char iv[TCRYPT_HDR_IV_LEN] = {};
297         char mode_name[MAX_CIPHER_LEN];
298         struct crypt_cipher *cipher;
299         char *c, *buf = (char*)&hdr->e;
300         int r;
301
302         /* Remove IV if present */
303         strncpy(mode_name, mode, MAX_CIPHER_LEN);
304         c = strchr(mode_name, '-');
305         if (c)
306                 *c = '\0';
307
308         if (!strncmp(mode, "lrw", 3))
309                 iv[alg->iv_size - 1] = 1;
310         else if (!strncmp(mode, "cbc", 3)) {
311                 remove_whitening(buf, &key[8]);
312                 if (!strcmp(alg->name, "blowfish_le"))
313                         return decrypt_blowfish_le_cbc(alg, key, buf);
314                 memcpy(iv, &key[alg->iv_offset], alg->iv_size);
315         }
316
317         copy_key(alg, mode, backend_key, key);
318         r = crypt_cipher_init(&cipher, alg->name, mode_name,
319                               backend_key, alg->key_size);
320         memset(backend_key, 0, sizeof(backend_key));
321         if (r < 0)
322                 return r;
323
324         r = crypt_cipher_decrypt(cipher, buf, buf, TCRYPT_HDR_LEN, iv, alg->iv_size);
325         crypt_cipher_destroy(cipher);
326
327         return r;
328 }
329
330 /*
331  * For chanined ciphers and CBC mode we need "outer" decryption.
332  * Backend doesn't provide this, so implement it here directly using ECB.
333  */
334 static int decrypt_hdr_cbci(struct tcrypt_algs *ciphers,
335                              const char *key, struct tcrypt_phdr *hdr)
336 {
337         struct crypt_cipher *cipher[ciphers->chain_count];
338         int bs = ciphers->cipher[0].iv_size;
339         char *buf = (char*)&hdr->e, iv[bs], iv_old[bs];
340         int i, j, r = -EINVAL;
341
342         remove_whitening(buf, &key[8]);
343
344         memcpy(iv, &key[ciphers->cipher[0].iv_offset], bs);
345
346         /* Initialize all ciphers in chain in ECB mode */
347         for (j = 0; j < ciphers->chain_count; j++) {
348                 r = crypt_cipher_init(&cipher[j], ciphers->cipher[j].name, "ecb",
349                                       &key[ciphers->cipher[j].key_offset],
350                                       ciphers->cipher[j].key_size);
351                 if (r < 0)
352                         goto out;
353         }
354
355         /* Implements CBC with chained ciphers in loop inside */
356         for (i = 0; i < TCRYPT_HDR_LEN; i += bs) {
357                 memcpy(iv_old, &buf[i], bs);
358                 for (j = ciphers->chain_count - 1; j >= 0; j--) {
359                         r = crypt_cipher_decrypt(cipher[j], &buf[i], &buf[i],
360                                                   bs, NULL, 0);
361                         if (r < 0)
362                                 goto out;
363                 }
364                 for (j = 0; j < bs; j++)
365                         buf[i + j] ^= iv[j];
366                 memcpy(iv, iv_old, bs);
367         }
368 out:
369         for (j = 0; j < ciphers->chain_count; j++)
370                 if (cipher[j])
371                         crypt_cipher_destroy(cipher[j]);
372
373         return r;
374 }
375
376 static int decrypt_hdr(struct crypt_device *cd, struct tcrypt_phdr *hdr,
377                         const char *key, int legacy_modes)
378 {
379         struct tcrypt_phdr hdr2;
380         int i, j, r;
381
382         for (i = 0; tcrypt_cipher[i].chain_count; i++) {
383                 if (!legacy_modes && tcrypt_cipher[i].legacy)
384                         continue;
385                 log_dbg("TCRYPT:  trying cipher %s-%s",
386                         tcrypt_cipher[i].long_name, tcrypt_cipher[i].mode);
387
388                 memcpy(&hdr2.e, &hdr->e, TCRYPT_HDR_LEN);
389
390                 if (!strncmp(tcrypt_cipher[i].mode, "cbci", 4))
391                         r = decrypt_hdr_cbci(&tcrypt_cipher[i], key, &hdr2);
392                 else for (j = tcrypt_cipher[i].chain_count - 1; j >= 0 ; j--) {
393                         if (!tcrypt_cipher[i].cipher[j].name)
394                                 continue;
395                         r = decrypt_hdr_one(&tcrypt_cipher[i].cipher[j],
396                                             tcrypt_cipher[i].mode, key, &hdr2);
397                         if (r < 0)
398                                 break;
399                 }
400
401                 if (r == -ENOENT) {
402                         log_err(cd, _("Required kernel crypto interface is not available.\n"
403                                       "Ensure you have af_skcipher kernel module loaded.\n"));
404                         return -ENOTSUP;
405                 }
406                 if (r < 0) {
407                         log_dbg("TCRYPT:   returned error %d, skipped.", r);
408                         continue;
409                 }
410
411                 if (!strncmp(hdr2.d.magic, TCRYPT_HDR_MAGIC, TCRYPT_HDR_MAGIC_LEN)) {
412                         log_dbg("TCRYPT: Signature magic detected.");
413                         memcpy(&hdr->e, &hdr2.e, TCRYPT_HDR_LEN);
414                         memset(&hdr2.e, 0, TCRYPT_HDR_LEN);
415                         r = i;
416                         break;
417                 }
418                 r = -EPERM;
419         }
420
421         return r;
422 }
423
424 static int pool_keyfile(struct crypt_device *cd,
425                         unsigned char pool[TCRYPT_KEY_POOL_LEN],
426                         const char *keyfile)
427 {
428         unsigned char data[TCRYPT_KEYFILE_LEN];
429         int i, j, fd, data_size;
430         uint32_t crc;
431         unsigned char *crc_c = (unsigned char*)&crc;
432
433         log_dbg("TCRYPT: using keyfile %s.", keyfile);
434
435         fd = open(keyfile, O_RDONLY);
436         if (fd < 0) {
437                 log_err(cd, _("Failed to open key file.\n"));
438                 return -EIO;
439         }
440
441         /* FIXME: add while */
442         data_size = read(fd, data, TCRYPT_KEYFILE_LEN);
443         close(fd);
444         if (data_size < 0) {
445                 log_err(cd, _("Error reading keyfile %s.\n"), keyfile);
446                 return -EIO;
447         }
448
449         for (i = 0, j = 0, crc = ~0U; i < data_size; i++) {
450                 crc = crypt_crc32(crc, &data[i], 1);
451                 pool[j++] += crc_c[3];
452                 pool[j++] += crc_c[2];
453                 pool[j++] += crc_c[1];
454                 pool[j++] += crc_c[0];
455                 j %= TCRYPT_KEY_POOL_LEN;
456         }
457
458         crc = 0;
459         memset(data, 0, TCRYPT_KEYFILE_LEN);
460
461         return 0;
462 }
463
464 static int TCRYPT_init_hdr(struct crypt_device *cd,
465                            struct tcrypt_phdr *hdr,
466                            struct crypt_params_tcrypt *params)
467 {
468         unsigned char pwd[TCRYPT_KEY_POOL_LEN] = {};
469         size_t passphrase_size;
470         char *key;
471         int r = -EINVAL, i, legacy_modes;
472
473         if (posix_memalign((void*)&key, crypt_getpagesize(), TCRYPT_HDR_KEY_LEN))
474                 return -ENOMEM;
475
476         if (params->keyfiles_count)
477                 passphrase_size = TCRYPT_KEY_POOL_LEN;
478         else
479                 passphrase_size = params->passphrase_size;
480
481         /* Calculate pool content from keyfiles */
482         for (i = 0; i < params->keyfiles_count; i++) {
483                 r = pool_keyfile(cd, pwd, params->keyfiles[i]);
484                 if (r < 0)
485                         goto out;
486         }
487
488         /* If provided password, combine it with pool */
489         for (i = 0; i < params->passphrase_size; i++)
490                 pwd[i] += params->passphrase[i];
491
492         legacy_modes = params->flags & CRYPT_TCRYPT_LEGACY_MODES ? 1 : 0;
493         for (i = 0; tcrypt_kdf[i].name; i++) {
494                 if (!legacy_modes && tcrypt_kdf[i].legacy)
495                         continue;
496                 /* Derive header key */
497                 log_dbg("TCRYPT: trying KDF: %s-%s-%d.",
498                         tcrypt_kdf[i].name, tcrypt_kdf[i].hash, tcrypt_kdf[i].iterations);
499                 r = crypt_pbkdf(tcrypt_kdf[i].name, tcrypt_kdf[i].hash,
500                                 (char*)pwd, passphrase_size,
501                                 hdr->salt, TCRYPT_HDR_SALT_LEN,
502                                 key, TCRYPT_HDR_KEY_LEN,
503                                 tcrypt_kdf[i].iterations);
504                 if (r < 0)
505                         break;
506
507                 /* Decrypt header */
508                 r = decrypt_hdr(cd, hdr, key, legacy_modes);
509                 if (r != -EPERM)
510                         break;
511         }
512
513         if (r < 0)
514                 goto out;
515
516         r = hdr_from_disk(hdr, params, i, r);
517         if (!r) {
518                 log_dbg("TCRYPT: Header version: %d, req. %d, sector %d"
519                         ", mk_offset %" PRIu64 ", hidden_size %" PRIu64
520                         ", volume size %" PRIu64, (int)hdr->d.version,
521                         (int)hdr->d.version_tc, (int)hdr->d.sector_size,
522                         hdr->d.mk_offset, hdr->d.hidden_volume_size, hdr->d.volume_size);
523                 log_dbg("TCRYPT: Header cipher %s-%s, key size %d",
524                         params->cipher, params->mode, params->key_size);
525         }
526 out:
527         memset(pwd, 0, TCRYPT_KEY_POOL_LEN);
528         if (key)
529                 memset(key, 0, TCRYPT_HDR_KEY_LEN);
530         free(key);
531         return r;
532 }
533
534 int TCRYPT_read_phdr(struct crypt_device *cd,
535                      struct tcrypt_phdr *hdr,
536                      struct crypt_params_tcrypt *params)
537 {
538         struct device *device = crypt_metadata_device(cd);
539         ssize_t hdr_size = sizeof(struct tcrypt_phdr);
540         int devfd = 0, r, bs;
541
542         assert(sizeof(struct tcrypt_phdr) == 512);
543
544         log_dbg("Reading TCRYPT header of size %d bytes from device %s.",
545                 hdr_size, device_path(device));
546
547         bs = device_block_size(device);
548         if (bs < 0)
549                 return bs;
550
551         devfd = open(device_path(device), O_RDONLY | O_DIRECT);
552         if (devfd == -1) {
553                 log_err(cd, _("Cannot open device %s.\n"), device_path(device));
554                 return -EINVAL;
555         }
556
557         r = -EIO;
558         if (params->flags & CRYPT_TCRYPT_HIDDEN_HEADER) {
559                 if (params->flags & CRYPT_TCRYPT_BACKUP_HEADER) {
560                         if (lseek(devfd, TCRYPT_HDR_HIDDEN_OFFSET_BCK, SEEK_END) >= 0 &&
561                             read_blockwise(devfd, bs, hdr, hdr_size) == hdr_size)
562                                 r = TCRYPT_init_hdr(cd, hdr, params);
563                 } else {
564                         if (lseek(devfd, TCRYPT_HDR_HIDDEN_OFFSET, SEEK_SET) >= 0 &&
565                             read_blockwise(devfd, bs, hdr, hdr_size) == hdr_size)
566                                 r = TCRYPT_init_hdr(cd, hdr, params);
567                         if (r &&
568                             lseek(devfd, TCRYPT_HDR_HIDDEN_OFFSET_OLD, SEEK_END) >= 0 &&
569                             read_blockwise(devfd, bs, hdr, hdr_size) == hdr_size)
570                                 r = TCRYPT_init_hdr(cd, hdr, params);
571                 }
572         } else if (params->flags & CRYPT_TCRYPT_BACKUP_HEADER) {
573                 if (lseek(devfd, TCRYPT_HDR_OFFSET_BCK, SEEK_END) >= 0 &&
574                             read_blockwise(devfd, bs, hdr, hdr_size) == hdr_size)
575                         r = TCRYPT_init_hdr(cd, hdr, params);
576         } else if (read_blockwise(devfd, bs, hdr, hdr_size) == hdr_size)
577                 r = TCRYPT_init_hdr(cd, hdr, params);
578
579         close(devfd);
580         return r;
581 }
582
583 static struct tcrypt_algs *get_algs(struct crypt_params_tcrypt *params)
584 {
585         int i;
586
587         if (!params->cipher || !params->mode)
588                 return NULL;
589
590         for (i = 0; tcrypt_cipher[i].chain_count; i++)
591                 if (!strcmp(tcrypt_cipher[i].long_name, params->cipher) &&
592                     !strcmp(tcrypt_cipher[i].mode, params->mode))
593                     return &tcrypt_cipher[i];
594
595         return NULL;
596 }
597
598 int TCRYPT_activate(struct crypt_device *cd,
599                      const char *name,
600                      struct tcrypt_phdr *hdr,
601                      struct crypt_params_tcrypt *params,
602                      uint32_t flags)
603 {
604         char cipher[MAX_CIPHER_LEN], dm_name[PATH_MAX], dm_dev_name[PATH_MAX];
605         struct device *device = NULL;
606         int i, r;
607         struct tcrypt_algs *algs;
608         struct crypt_dm_active_device dmd = {
609                 .target = DM_CRYPT,
610                 .size   = 0,
611                 .data_device = crypt_data_device(cd),
612                 .u.crypt  = {
613                         .cipher = cipher,
614                         .offset = crypt_get_data_offset(cd),
615                         .iv_offset = crypt_get_iv_offset(cd),
616                 }
617         };
618
619         if (!hdr->d.version) {
620                 log_dbg("TCRYPT: this function is not supported without encrypted header load.");
621                 return -ENOTSUP;
622         }
623
624         if (hdr->d.sector_size && hdr->d.sector_size != SECTOR_SIZE) {
625                 log_err(cd, _("Activation is not supported for %d sector size.\n"),
626                         hdr->d.sector_size);
627                 return -ENOTSUP;
628         }
629
630         if (strstr(params->mode, "-tcrypt")) {
631                 log_err(cd, _("Kernel doesn't support activation for this TCRYPT legacy mode.\n"));
632                 return -ENOTSUP;
633         }
634
635         algs = get_algs(params);
636         if (!algs)
637                 return -EINVAL;
638
639         if (params->flags & CRYPT_TCRYPT_HIDDEN_HEADER)
640                 dmd.size = hdr->d.hidden_volume_size / hdr->d.sector_size;
641         else
642                 dmd.size = hdr->d.volume_size / hdr->d.sector_size;
643
644         r = device_block_adjust(cd, dmd.data_device, DEV_EXCL,
645                                 dmd.u.crypt.offset, &dmd.size, &dmd.flags);
646         if (r)
647                 return r;
648
649         /* Frome here, key size for every cipher must be the same */
650         dmd.u.crypt.vk = crypt_alloc_volume_key(algs->cipher[0].key_size, NULL);
651         if (!dmd.u.crypt.vk)
652                 return -ENOMEM;
653
654         for (i = algs->chain_count - 1; i >= 0; i--) {
655                 if (i == 0) {
656                         strncpy(dm_name, name, sizeof(dm_name));
657                         dmd.flags = flags;
658                 } else {
659                         snprintf(dm_name, sizeof(dm_name), "%s_%d", name, i);
660                         dmd.flags = flags | CRYPT_ACTIVATE_PRIVATE;
661                 }
662
663                 snprintf(cipher, sizeof(cipher), "%s-%s",
664                          algs->cipher[i].name, algs->mode);
665
666                 copy_key(&algs->cipher[i], algs->mode, dmd.u.crypt.vk->key, hdr->d.keys);
667
668                 if ((algs->chain_count - 1) != i) {
669                         snprintf(dm_dev_name, sizeof(dm_dev_name), "%s/%s_%d",
670                                  dm_get_dir(), name, i + 1);
671                         r = device_alloc(&device, dm_dev_name);
672                         if (r)
673                                 break;
674                         dmd.data_device = device;
675                         dmd.u.crypt.offset = 0;
676                 }
677
678                 log_dbg("Trying to activate TCRYPT device %s using cipher %s.",
679                         dm_name, dmd.u.crypt.cipher);
680                 r = dm_create_device(cd, dm_name, CRYPT_TCRYPT, &dmd, 0);
681
682                 device_free(device);
683                 device = NULL;
684
685                 if (r)
686                         break;
687         }
688
689         if (!r && !(dm_flags() & DM_PLAIN64_SUPPORTED)) {
690                 log_err(cd, _("Kernel doesn't support plain64 IV.\n"));
691                 r = -ENOTSUP;
692         }
693
694         crypt_free_volume_key(dmd.u.crypt.vk);
695         return r;
696 }
697
698 static int remove_one(struct crypt_device *cd, const char *name,
699                       const char *base_uuid, int index)
700 {
701         struct crypt_dm_active_device dmd = {};
702         char dm_name[PATH_MAX];
703         int r;
704
705         if (snprintf(dm_name, sizeof(dm_name), "%s_%d", name, index) < 0)
706                 return -ENOMEM;
707
708         r = dm_status_device(cd, dm_name);
709         if (r < 0)
710                 return r;
711
712         r = dm_query_device(cd, dm_name, DM_ACTIVE_UUID, &dmd);
713         if (!r && !strncmp(dmd.uuid, base_uuid, strlen(base_uuid)))
714                 r = dm_remove_device(cd, dm_name, 0, 0);
715
716         free(CONST_CAST(void*)dmd.uuid);
717         return r;
718 }
719
720 int TCRYPT_deactivate(struct crypt_device *cd, const char *name)
721 {
722         struct crypt_dm_active_device dmd = {};
723         int r;
724
725         r = dm_query_device(cd, name, DM_ACTIVE_UUID, &dmd);
726         if (r < 0)
727                 return r;
728         if (!dmd.uuid)
729                 return -EINVAL;
730
731         r = dm_remove_device(cd, name, 0, 0);
732         if (r < 0)
733                 goto out;
734
735         r = remove_one(cd, name, dmd.uuid, 1);
736         if (r < 0)
737                 goto out;
738
739         r = remove_one(cd, name, dmd.uuid, 2);
740         if (r < 0)
741                 goto out;
742 out:
743         free(CONST_CAST(void*)dmd.uuid);
744         return (r == -ENODEV) ? 0 : r;
745 }
746
747 static int status_one(struct crypt_device *cd, const char *name,
748                       const char *base_uuid, int index,
749                       size_t *key_size, char *cipher, uint64_t *data_offset,
750                       struct device **device)
751 {
752         struct crypt_dm_active_device dmd = {};
753         char dm_name[PATH_MAX], *c;
754         int r;
755
756         if (snprintf(dm_name, sizeof(dm_name), "%s_%d", name, index) < 0)
757                 return -ENOMEM;
758
759         r = dm_status_device(cd, dm_name);
760         if (r < 0)
761                 return r;
762
763         r = dm_query_device(cd, dm_name, DM_ACTIVE_DEVICE |
764                                           DM_ACTIVE_UUID |
765                                           DM_ACTIVE_CRYPT_CIPHER |
766                                           DM_ACTIVE_CRYPT_KEYSIZE, &dmd);
767         if (r > 0)
768                 r = 0;
769         if (!r && !strncmp(dmd.uuid, base_uuid, strlen(base_uuid))) {
770                 if ((c = strchr(dmd.u.crypt.cipher, '-')))
771                         *c = '\0';
772                 strcat(cipher, "-");
773                 strncat(cipher, dmd.u.crypt.cipher, MAX_CIPHER_LEN);
774                 *key_size += dmd.u.crypt.vk->keylength;
775                 *data_offset = dmd.u.crypt.offset * SECTOR_SIZE;
776                 device_free(*device);
777                 *device = dmd.data_device;
778         } else {
779                 device_free(dmd.data_device);
780                 r = -ENODEV;
781         }
782
783         free(CONST_CAST(void*)dmd.uuid);
784         free(CONST_CAST(void*)dmd.u.crypt.cipher);
785         crypt_free_volume_key(dmd.u.crypt.vk);
786         return r;
787 }
788
789 int TCRYPT_init_by_name(struct crypt_device *cd, const char *name,
790                         const struct crypt_dm_active_device *dmd,
791                         struct device **device,
792                         struct crypt_params_tcrypt *tcrypt_params,
793                         struct tcrypt_phdr *tcrypt_hdr)
794 {
795         char cipher[MAX_CIPHER_LEN * 4], *mode;
796
797         memset(tcrypt_params, 0, sizeof(*tcrypt_params));
798         memset(tcrypt_hdr, 0, sizeof(*tcrypt_hdr));
799         tcrypt_hdr->d.sector_size = SECTOR_SIZE;
800         tcrypt_hdr->d.mk_offset = dmd->u.crypt.offset * SECTOR_SIZE;
801
802         strncpy(cipher, dmd->u.crypt.cipher, MAX_CIPHER_LEN);
803
804         if ((mode = strchr(cipher, '-'))) {
805                 *mode = '\0';
806                 tcrypt_params->mode = strdup(++mode);
807         }
808         tcrypt_params->key_size = dmd->u.crypt.vk->keylength;
809
810         if (!status_one(cd, name, dmd->uuid, 1, &tcrypt_params->key_size,
811                         cipher, &tcrypt_hdr->d.mk_offset, device))
812                 status_one(cd, name, dmd->uuid, 2, &tcrypt_params->key_size,
813                            cipher, &tcrypt_hdr->d.mk_offset, device);
814
815         tcrypt_params->cipher = strdup(cipher);
816         return 0;
817 }
818
819 uint64_t TCRYPT_get_data_offset(struct crypt_device *cd,
820                                  struct tcrypt_phdr *hdr,
821                                  struct crypt_params_tcrypt *params)
822 {
823         uint64_t size;
824
825         if (params->mode && !strncmp(params->mode, "xts", 3)) {
826                 if (hdr->d.version < 3)
827                         return 1;
828
829                 if (params->flags & CRYPT_TCRYPT_HIDDEN_HEADER) {
830                         if (hdr->d.version > 3)
831                                 return (hdr->d.mk_offset / hdr->d.sector_size);
832                         if (device_size(crypt_metadata_device(cd), &size) < 0)
833                                 return 0;
834                         return (size - hdr->d.hidden_volume_size +
835                                 (TCRYPT_HDR_HIDDEN_OFFSET_OLD)) / hdr->d.sector_size;
836                 }
837                 return (hdr->d.mk_offset / hdr->d.sector_size);
838         }
839
840         if (params->flags & CRYPT_TCRYPT_HIDDEN_HEADER) {
841                 if (device_size(crypt_metadata_device(cd), &size) < 0)
842                         return 0;
843                 return (size - hdr->d.hidden_volume_size +
844                         (TCRYPT_HDR_HIDDEN_OFFSET_OLD)) / hdr->d.sector_size;
845         }
846
847         // FIXME: system vol.
848         return hdr->d.mk_offset / hdr->d.sector_size;
849 }
850
851 uint64_t TCRYPT_get_iv_offset(struct crypt_device *cd,
852                               struct tcrypt_phdr *hdr,
853                               struct crypt_params_tcrypt *params
854 )
855 {
856         if (params->mode && !strncmp(params->mode, "xts", 3))
857                 return TCRYPT_get_data_offset(cd, hdr, params);
858         else if (params->mode && !strncmp(params->mode, "lrw", 3))
859                 return 0;
860
861         return hdr->d.mk_offset / hdr->d.sector_size;
862 }
863
864 int TCRYPT_get_volume_key(struct crypt_device *cd,
865                           struct tcrypt_phdr *hdr,
866                           struct crypt_params_tcrypt *params,
867                           struct volume_key **vk)
868 {
869         struct tcrypt_algs *algs;
870         int i, key_index;
871
872         if (!hdr->d.version) {
873                 log_dbg("TCRYPT: this function is not supported without encrypted header load.");
874                 return -ENOTSUP;
875         }
876
877         algs = get_algs(params);
878         if (!algs)
879                 return -EINVAL;
880
881         *vk = crypt_alloc_volume_key(params->key_size, NULL);
882         if (!*vk)
883                 return -ENOMEM;
884
885         for (i = 0, key_index = 0; i < algs->chain_count; i++) {
886                 copy_key(&algs->cipher[i], algs->mode,
887                          &(*vk)->key[key_index], hdr->d.keys);
888                 key_index += algs->cipher[i].key_size;
889         }
890
891         return 0;
892 }
893
894 int TCRYPT_dump(struct crypt_device *cd,
895                 struct tcrypt_phdr *hdr,
896                 struct crypt_params_tcrypt *params)
897 {
898         log_std(cd, "TCRYPT header information for %s\n",
899                 device_path(crypt_metadata_device(cd)));
900         if (hdr->d.version) {
901                 log_std(cd, "Version:       \t%d\n", hdr->d.version);
902                 log_std(cd, "Driver req.:\t%d\n", hdr->d.version_tc);
903
904                 log_std(cd, "Sector size:\t%" PRIu32 "\n", hdr->d.sector_size);
905                 log_std(cd, "MK offset:\t%" PRIu64 "\n", hdr->d.mk_offset);
906                 log_std(cd, "PBKDF2 hash:\t%s\n", params->hash_name);
907         }
908         log_std(cd, "Cipher chain:\t%s\n", params->cipher);
909         log_std(cd, "Cipher mode:\t%s\n", params->mode);
910         log_std(cd, "MK bits:       \t%d\n", params->key_size * 8);
911         return 0;
912 }