Fix tests to work with all crypto backends.
[platform/upstream/cryptsetup.git] / tests / api-test.c
1 /*
2  * cryptsetup library API check functions
3  *
4  * Copyright (C) 2009-2010 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18  */
19
20 #include <stdio.h>
21 #include <stdlib.h>
22 #include <string.h>
23 #include <unistd.h>
24 #include <fcntl.h>
25 #include <linux/fs.h>
26 #include <errno.h>
27 #include <assert.h>
28 #include <sys/stat.h>
29 #include <sys/ioctl.h>
30
31 #include "libcryptsetup.h"
32
33 #define DMDIR "/dev/mapper/"
34
35 #define DEVICE_1 "/dev/loop5"
36 #define DEVICE_1_UUID "28632274-8c8a-493f-835b-da802e1c576b"
37 #define DEVICE_2 "/dev/loop6"
38 #define DEVICE_EMPTY_name "crypt_zero"
39 #define DEVICE_EMPTY DMDIR DEVICE_EMPTY_name
40 #define DEVICE_ERROR_name "crypt_error"
41 #define DEVICE_ERROR DMDIR DEVICE_ERROR_name
42
43 #define CDEVICE_1 "ctest1"
44 #define CDEVICE_2 "ctest2"
45 #define CDEVICE_WRONG "O_o"
46
47 #define IMAGE1 "compatimage.img"
48 #define IMAGE_EMPTY "empty.img"
49
50 #define KEYFILE1 "key1.file"
51 #define KEY1 "compatkey"
52
53 #define KEYFILE2 "key2.file"
54 #define KEY2 "0123456789abcdef"
55
56 #define PASSPHRASE "blabla"
57
58 #define DEVICE_TEST_UUID "12345678-1234-1234-1234-123456789abc"
59
60 static int _debug   = 0;
61 static int _verbose = 1;
62
63 static char global_log[4096];
64 static int global_lines = 0;
65
66 // Helpers
67 static int _prepare_keyfile(const char *name, const char *passphrase)
68 {
69         int fd, r;
70
71         fd = open(name, O_RDWR | O_CREAT | O_TRUNC, S_IRUSR);
72         if (fd != -1) {
73                 r = write(fd, passphrase, strlen(passphrase));
74                 close(fd);
75         } else
76                 r = 0;
77
78         return r == strlen(passphrase) ? 0 : 1;
79 }
80
81 static void _remove_keyfiles(void)
82 {
83         remove(KEYFILE1);
84         remove(KEYFILE2);
85 }
86
87 // Decode key from its hex representation
88 static int crypt_decode_key(char *key, char *hex, unsigned int size)
89 {
90         char buffer[3];
91         char *endp;
92         unsigned int i;
93
94         buffer[2] = '\0';
95
96         for (i = 0; i < size; i++) {
97                 buffer[0] = *hex++;
98                 buffer[1] = *hex++;
99
100                 key[i] = (unsigned char)strtoul(buffer, &endp, 16);
101
102                 if (endp != &buffer[2])
103                         return -1;
104         }
105
106         if (*hex != '\0')
107                 return -1;
108
109         return 0;
110 }
111
112 static int yesDialog(char *msg)
113 {
114         return 1;
115 }
116
117 static void cmdLineLog(int level, char *msg)
118 {
119         strncat(global_log, msg, sizeof(global_log) - strlen(global_log));
120         global_lines++;
121 }
122
123 static void new_log(int level, const char *msg, void *usrptr)
124 {
125         cmdLineLog(level, (char*)msg);
126 }
127
128
129 static void reset_log()
130 {
131         memset(global_log, 0, sizeof(global_log));
132         global_lines = 0;
133 }
134
135 static void _system(const char *command, int warn)
136 {
137         if (system(command) < 0 && warn)
138                 printf("System command failed: %s", command);
139 }
140
141 static struct interface_callbacks cmd_icb = {
142         .yesDialog = yesDialog,
143         .log = cmdLineLog,
144 };
145
146 static void _cleanup(void)
147 {
148         struct stat st;
149
150         //_system("udevadm settle", 0);
151
152         if (!stat(DMDIR CDEVICE_1, &st))
153                 _system("dmsetup remove " CDEVICE_1, 0);
154
155         if (!stat(DMDIR CDEVICE_2, &st))
156                 _system("dmsetup remove " CDEVICE_2, 0);
157
158         if (!stat(DEVICE_EMPTY, &st))
159                 _system("dmsetup remove " DEVICE_EMPTY_name, 0);
160
161         if (!stat(DEVICE_ERROR, &st))
162                 _system("dmsetup remove " DEVICE_ERROR_name, 0);
163
164         if (!strncmp("/dev/loop", DEVICE_1, 9))
165                 _system("losetup -d " DEVICE_1, 0);
166
167         if (!strncmp("/dev/loop", DEVICE_2, 9))
168                 _system("losetup -d " DEVICE_2, 0);
169
170         _system("rm -f " IMAGE_EMPTY, 0);
171         _remove_keyfiles();
172 }
173
174 static void _setup(void)
175 {
176         _system("dmsetup create " DEVICE_EMPTY_name " --table \"0 10000 zero\"", 1);
177         _system("dmsetup create " DEVICE_ERROR_name " --table \"0 10000 error\"", 1);
178         if (!strncmp("/dev/loop", DEVICE_1, 9)) {
179                 _system(" [ ! -e " IMAGE1 " ] && bzip2 -dk " IMAGE1 ".bz2", 1);
180                 _system("losetup " DEVICE_1 " " IMAGE1, 1);
181         }
182         if (!strncmp("/dev/loop", DEVICE_2, 9)) {
183                 _system("dd if=/dev/zero of=" IMAGE_EMPTY " bs=1M count=4", 1);
184                 _system("losetup " DEVICE_2 " " IMAGE_EMPTY, 1);
185         }
186
187 }
188
189 void check_ok(int status, int line, const char *func)
190 {
191         char buf[256];
192
193         if (status) {
194                 crypt_get_error(buf, sizeof(buf));
195                 printf("FAIL line %d [%s]: code %d, %s\n", line, func, status, buf);
196                 _cleanup();
197                 exit(-1);
198         }
199 }
200
201 void check_ko(int status, int line, const char *func)
202 {
203         char buf[256];
204
205         memset(buf, 0, sizeof(buf));
206         crypt_get_error(buf, sizeof(buf));
207         if (status >= 0) {
208                 printf("FAIL line %d [%s]: code %d, %s\n", line, func, status, buf);
209                 _cleanup();
210                 exit(-1);
211         } else if (_verbose)
212                 printf("   => errno %d, errmsg: %s\n", status, buf);
213 }
214
215 void check_equal(int line, const char *func)
216 {
217         printf("FAIL line %d [%s]: expected equal values differs.\n", line, func);
218         _cleanup();
219         exit(-1);
220 }
221
222 void xlog(const char *msg, const char *tst, const char *func, int line, const char *txt)
223 {
224         if (_verbose) {
225                 if (txt)
226                         printf(" [%s,%s:%d] %s [%s]\n", msg, func, line, tst, txt);
227                 else
228                         printf(" [%s,%s:%d] %s\n", msg, func, line, tst);
229         }
230 }
231 #define OK_(x)          do { xlog("(success)", #x, __FUNCTION__, __LINE__, NULL); \
232                              check_ok((x), __LINE__, __FUNCTION__); \
233                         } while(0)
234 #define FAIL_(x, y)     do { xlog("(fail)   ", #x, __FUNCTION__, __LINE__, y); \
235                              check_ko((x), __LINE__, __FUNCTION__); \
236                         } while(0)
237 #define EQ_(x, y)       do { xlog("(equal)  ", #x " == " #y, __FUNCTION__, __LINE__, NULL); \
238                              if ((x) != (y)) check_equal(__LINE__, __FUNCTION__); \
239                         } while(0)
240
241 #define RUN_(x, y)              do { printf("%s: %s\n", #x, (y)); x(); } while (0)
242
243 // OLD API TESTS
244 static void LuksUUID(void)
245 {
246         struct crypt_options co = { .icb = &cmd_icb };
247
248         co.device = DEVICE_EMPTY;
249         EQ_(crypt_luksUUID(&co), -EINVAL);
250
251         co.device = DEVICE_ERROR;
252         EQ_(crypt_luksUUID(&co), -EINVAL);
253
254         reset_log();
255         co.device = DEVICE_1;
256         OK_(crypt_luksUUID(&co));
257         EQ_(strlen(global_log), 37); /* UUID + "\n" */
258         EQ_(strncmp(global_log, DEVICE_1_UUID, strlen(DEVICE_1_UUID)), 0);
259
260 }
261
262 static void IsLuks(void)
263 {
264         struct crypt_options co = {  .icb = &cmd_icb };
265
266         co.device = DEVICE_EMPTY;
267         EQ_(crypt_isLuks(&co), -EINVAL);
268
269         co.device = DEVICE_ERROR;
270         EQ_(crypt_isLuks(&co), -EINVAL);
271
272         co.device = DEVICE_1;
273         OK_(crypt_isLuks(&co));
274 }
275
276 static void LuksOpen(void)
277 {
278         struct crypt_options co = {
279                 .name = CDEVICE_1,
280                 //.passphrase = "blabla",
281                 .icb = &cmd_icb,
282         };
283
284         OK_(_prepare_keyfile(KEYFILE1, KEY1));
285         co.key_file = KEYFILE1;
286
287         co.device = DEVICE_EMPTY;
288         EQ_(crypt_luksOpen(&co), -EINVAL);
289
290         co.device = DEVICE_ERROR;
291         EQ_(crypt_luksOpen(&co), -EINVAL);
292
293         co.device = DEVICE_1;
294         OK_(crypt_luksOpen(&co));
295         FAIL_(crypt_luksOpen(&co), "already open");
296
297         _remove_keyfiles();
298 }
299
300 static void query_device(void)
301 {
302         struct crypt_options co = {.icb = &cmd_icb };
303
304         co.name = CDEVICE_WRONG;
305         EQ_(crypt_query_device(&co), 0);
306
307         co.name = CDEVICE_1;
308         EQ_(crypt_query_device(&co), 1);
309
310         OK_(strncmp(crypt_get_dir(), DMDIR, 11));
311         OK_(strcmp(co.cipher, "aes-cbc-essiv:sha256"));
312         EQ_(co.key_size, 16);
313         EQ_(co.offset, 1032);
314         EQ_(co.flags & CRYPT_FLAG_READONLY, 0);
315         EQ_(co.skip, 0);
316         crypt_put_options(&co);
317 }
318
319 static void remove_device(void)
320 {
321         int fd;
322         struct crypt_options co = {.icb = &cmd_icb };
323
324         co.name = CDEVICE_WRONG;
325         EQ_(crypt_remove_device(&co), -ENODEV);
326
327         fd = open(DMDIR CDEVICE_1, O_RDONLY);
328         co.name = CDEVICE_1;
329         FAIL_(crypt_remove_device(&co), "device busy");
330         close(fd);
331
332         OK_(crypt_remove_device(&co));
333 }
334
335 static void LuksFormat(void)
336 {
337         struct crypt_options co = {
338                 .device = DEVICE_2,
339                 .key_size = 256 / 8,
340                 .key_slot = -1,
341                 .cipher = "aes-cbc-essiv:sha256",
342                 .hash = "sha1",
343                 .flags = 0,
344                 .iteration_time = 10,
345                 .align_payload = 0,
346                 .icb = &cmd_icb,
347         };
348
349         OK_(_prepare_keyfile(KEYFILE1, KEY1));
350
351         co.new_key_file = KEYFILE1;
352         co.device = DEVICE_ERROR;
353         FAIL_(crypt_luksFormat(&co), "error device");
354
355         co.device = DEVICE_2;
356         OK_(crypt_luksFormat(&co));
357
358         co.new_key_file = NULL;
359         co.key_file = KEYFILE1;
360         co.name = CDEVICE_2;
361         OK_(crypt_luksOpen(&co));
362         OK_(crypt_remove_device(&co));
363         _remove_keyfiles();
364 }
365
366 static void LuksKeyGame(void)
367 {
368         int i;
369         struct crypt_options co = {
370                 .device = DEVICE_2,
371                 .key_size = 256 / 8,
372                 .key_slot = -1,
373                 .cipher = "aes-cbc-essiv:sha256",
374                 .hash = "sha1",
375                 .flags = 0,
376                 .iteration_time = 10,
377                 .align_payload = 0,
378                 .icb = &cmd_icb,
379         };
380
381         OK_(_prepare_keyfile(KEYFILE1, KEY1));
382         OK_(_prepare_keyfile(KEYFILE2, KEY2));
383
384         co.new_key_file = KEYFILE1;
385         co.device = DEVICE_2;
386         co.key_slot = 8;
387         FAIL_(crypt_luksFormat(&co), "wrong slot #");
388
389         co.key_slot = 7; // last slot
390         OK_(crypt_luksFormat(&co));
391
392         co.new_key_file = KEYFILE1;
393         co.key_file = KEYFILE1;
394         co.key_slot = 8;
395         FAIL_(crypt_luksAddKey(&co), "wrong slot #");
396         co.key_slot = 7;
397         FAIL_(crypt_luksAddKey(&co), "slot already used");
398
399         co.key_slot = 6;
400         OK_(crypt_luksAddKey(&co));
401
402         co.key_file = KEYFILE2 "blah";
403         co.key_slot = 5;
404         FAIL_(crypt_luksAddKey(&co), "keyfile not found");
405
406         co.new_key_file = KEYFILE2; // key to add
407         co.key_file = KEYFILE1;
408         co.key_slot = -1;
409         for (i = 0; i < 6; i++)
410                 OK_(crypt_luksAddKey(&co)); //FIXME: EQ_(i)?
411
412         FAIL_(crypt_luksAddKey(&co), "all slots full");
413
414         // REMOVE KEY
415         co.new_key_file = KEYFILE1; // key to remove
416         co.key_file = NULL;
417         co.key_slot = 8; // should be ignored
418          // only 2 slots should use KEYFILE1
419         OK_(crypt_luksRemoveKey(&co));
420         OK_(crypt_luksRemoveKey(&co));
421         FAIL_(crypt_luksRemoveKey(&co), "no slot with this passphrase");
422
423         co.new_key_file = KEYFILE2 "blah";
424         co.key_file = NULL;
425         FAIL_(crypt_luksRemoveKey(&co), "keyfile not found");
426
427         // KILL SLOT
428         co.new_key_file = NULL;
429         co.key_file = NULL;
430         co.key_slot = 8;
431         FAIL_(crypt_luksKillSlot(&co), "wrong slot #");
432         co.key_slot = 7;
433         FAIL_(crypt_luksKillSlot(&co), "slot already wiped");
434
435         co.key_slot = 5;
436         OK_(crypt_luksKillSlot(&co));
437
438         _remove_keyfiles();
439 }
440
441 size_t _get_device_size(const char *device)
442 {
443         unsigned long size = 0;
444         int fd;
445
446         fd = open(device, O_RDONLY);
447         if (fd == -1)
448                 return 0;
449         (void)ioctl(fd, BLKGETSIZE, &size);
450         close(fd);
451
452         return size;
453 }
454
455 void DeviceResizeGame(void)
456 {
457         size_t orig_size;
458         struct crypt_options co = {
459                 .name = CDEVICE_2,
460                 .device = DEVICE_2,
461                 .key_size = 128 / 8,
462                 .cipher = "aes-cbc-plain",
463                 .hash = "sha1",
464                 .offset = 333,
465                 .skip = 0,
466                 .icb = &cmd_icb,
467         };
468
469         orig_size = _get_device_size(DEVICE_2);
470
471         OK_(_prepare_keyfile(KEYFILE2, KEY2));
472
473         co.key_file = KEYFILE2;
474         co.size = 1000;
475         OK_(crypt_create_device(&co));
476         EQ_(_get_device_size(DMDIR CDEVICE_2), 1000);
477
478         co.size = 2000;
479         OK_(crypt_resize_device(&co));
480         EQ_(_get_device_size(DMDIR CDEVICE_2), 2000);
481
482         co.size = 0;
483         OK_(crypt_resize_device(&co));
484         EQ_(_get_device_size(DMDIR CDEVICE_2), (orig_size - 333));
485         co.size = 0;
486         co.offset = 444;
487         co.skip = 555;
488         co.cipher = "aes-cbc-essiv:sha256";
489         OK_(crypt_update_device(&co));
490         EQ_(_get_device_size(DMDIR CDEVICE_2), (orig_size - 444));
491
492         memset(&co, 0, sizeof(co));
493         co.icb = &cmd_icb,
494         co.name = CDEVICE_2;
495         EQ_(crypt_query_device(&co), 1);
496         EQ_(strcmp(co.cipher, "aes-cbc-essiv:sha256"), 0);
497         EQ_(co.key_size, 128 / 8);
498         EQ_(co.offset, 444);
499         EQ_(co.skip, 555);
500         crypt_put_options(&co);
501
502         // dangerous switch device still works
503         memset(&co, 0, sizeof(co));
504         co.name = CDEVICE_2,
505         co.device = DEVICE_1;
506         co.key_file = KEYFILE2;
507         co.key_size = 128 / 8;
508         co.cipher = "aes-cbc-plain";
509         co.hash = "sha1";
510         co.icb = &cmd_icb;
511         OK_(crypt_update_device(&co));
512
513         memset(&co, 0, sizeof(co));
514         co.icb = &cmd_icb,
515         co.name = CDEVICE_2;
516         EQ_(crypt_query_device(&co), 1);
517         EQ_(strcmp(co.cipher, "aes-cbc-plain"), 0);
518         EQ_(co.key_size, 128 / 8);
519         EQ_(co.offset, 0);
520         EQ_(co.skip, 0);
521         // This expect lookup returns prefered /dev/loopX
522         EQ_(strcmp(co.device, DEVICE_1), 0);
523         crypt_put_options(&co);
524
525         memset(&co, 0, sizeof(co));
526         co.icb = &cmd_icb,
527         co.name = CDEVICE_2;
528         OK_(crypt_remove_device(&co));
529
530         _remove_keyfiles();
531 }
532
533 // NEW API tests
534
535 static void AddDevicePlain(void)
536 {
537         struct crypt_device *cd;
538         struct crypt_params_plain params = {
539                 .hash = "sha1",
540                 .skip = 0,
541                 .offset = 0,
542         };
543         int fd;
544         char key[128], key2[128], path[128];
545
546         char *passphrase = PASSPHRASE;
547         char *mk_hex = "bb21158c733229347bd4e681891e213d94c685be6a5b84818afe7a78a6de7a1a";
548         size_t key_size = strlen(mk_hex) / 2;
549         char *cipher = "aes";
550         char *cipher_mode = "cbc-essiv:sha256";
551
552         crypt_decode_key(key, mk_hex, key_size);
553
554         FAIL_(crypt_init(&cd, ""), "empty device string");
555
556         // default is "plain" hash - no password hash
557         OK_(crypt_init(&cd, DEVICE_1));
558         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, NULL));
559         FAIL_(crypt_activate_by_volume_key(cd, NULL, key, key_size, 0), "cannot verify key with plain");
560         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
561         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
562         // FIXME: this should get key from active device?
563         //OK_(crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key2, &key_size, passphrase, strlen(passphrase)));
564         //OK_(memcmp(key, key2, key_size));
565         OK_(crypt_deactivate(cd, CDEVICE_1));
566         crypt_free(cd);
567
568         // Now use hashed password
569         OK_(crypt_init(&cd, DEVICE_1));
570         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
571         FAIL_(crypt_activate_by_passphrase(cd, NULL, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0),
572               "cannot verify passphrase with plain" );
573         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0));
574
575         // device status check
576         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
577         snprintf(path, sizeof(path), "%s/%s", crypt_get_dir(), CDEVICE_1);
578         fd = open(path, O_RDONLY);
579         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_BUSY);
580         FAIL_(crypt_deactivate(cd, CDEVICE_1), "Device is busy");
581         close(fd);
582         OK_(crypt_deactivate(cd, CDEVICE_1));
583         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_INACTIVE);
584
585         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
586         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
587
588         // retrieve volume key check
589         memset(key2, 0, key_size);
590         key_size--;
591         // small buffer
592         FAIL_(crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key2, &key_size, passphrase, strlen(passphrase)), "small buffer");
593         key_size++;
594         OK_(crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key2, &key_size, passphrase, strlen(passphrase)));
595
596         OK_(memcmp(key, key2, key_size));
597         OK_(strcmp(cipher, crypt_get_cipher(cd)));
598         OK_(strcmp(cipher_mode, crypt_get_cipher_mode(cd)));
599         EQ_(key_size, crypt_get_volume_key_size(cd));
600         EQ_(0, crypt_get_data_offset(cd));
601         OK_(crypt_deactivate(cd, CDEVICE_1));
602
603         // now with keyfile
604         OK_(_prepare_keyfile(KEYFILE1, KEY1));
605         FAIL_(crypt_activate_by_keyfile(cd, NULL, CRYPT_ANY_SLOT, KEYFILE1, 0, 0), "cannot verify key with plain");
606         EQ_(0, crypt_activate_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 0, 0));
607         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
608         OK_(crypt_deactivate(cd, CDEVICE_1));
609         _remove_keyfiles();
610
611         crypt_free(cd);
612 }
613
614 #define CALLBACK_ERROR "calback_error xyz"
615 static int pass_callback_err(const char *msg, char *buf, size_t length, void *usrptr)
616 {
617         struct crypt_device *cd = usrptr;
618
619         assert(cd);
620         assert(length);
621         assert(msg);
622
623         crypt_log(cd, CRYPT_LOG_ERROR, CALLBACK_ERROR);
624         return -EINVAL;
625 }
626
627 static int pass_callback_ok(const char *msg, char *buf, size_t length, void *usrptr)
628 {
629         assert(length);
630         assert(msg);
631         strcpy(buf, PASSPHRASE);
632         return strlen(buf);
633 }
634
635 static void CallbacksTest(void)
636 {
637         struct crypt_device *cd;
638         struct crypt_params_plain params = {
639                 .hash = "sha1",
640                 .skip = 0,
641                 .offset = 0,
642         };
643
644         size_t key_size = 256 / 8;
645         char *cipher = "aes";
646         char *cipher_mode = "cbc-essiv:sha256";
647         char *passphrase = PASSPHRASE;
648
649         OK_(crypt_init(&cd, DEVICE_1));
650         crypt_set_log_callback(cd, &new_log, NULL);
651         //crypt_set_log_callback(cd, NULL, NULL);
652
653         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
654
655         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0));
656         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
657         OK_(crypt_deactivate(cd, CDEVICE_1));
658
659         reset_log();
660         crypt_set_password_callback(cd, pass_callback_err, cd);
661         FAIL_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, NULL, 0, 0), "callback fails");
662         EQ_(strncmp(global_log, CALLBACK_ERROR, strlen(CALLBACK_ERROR)), 0);
663
664         crypt_set_password_callback(cd, pass_callback_ok, NULL);
665         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, NULL, 0, 0));
666         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
667         OK_(crypt_deactivate(cd, CDEVICE_1));
668
669         crypt_free(cd);
670 }
671
672 static void UseLuksDevice(void)
673 {
674         struct crypt_device *cd;
675         char key[128];
676         size_t key_size;
677
678         OK_(crypt_init(&cd, DEVICE_1));
679         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
680         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_INACTIVE);
681         OK_(crypt_activate_by_passphrase(cd, NULL, CRYPT_ANY_SLOT, KEY1, strlen(KEY1), 0));
682         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1), 0));
683         FAIL_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1), 0), "already open");
684         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
685         OK_(crypt_deactivate(cd, CDEVICE_1));
686         FAIL_(crypt_deactivate(cd, CDEVICE_1), "no such device");
687
688         key_size = 16;
689         OK_(strcmp("aes", crypt_get_cipher(cd)));
690         OK_(strcmp("cbc-essiv:sha256", crypt_get_cipher_mode(cd)));
691         OK_(strcmp(DEVICE_1_UUID, crypt_get_uuid(cd)));
692         EQ_(key_size, crypt_get_volume_key_size(cd));
693         EQ_(1032, crypt_get_data_offset(cd));
694
695         EQ_(0, crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key, &key_size, KEY1, strlen(KEY1)));
696         OK_(crypt_volume_key_verify(cd, key, key_size));
697         OK_(crypt_activate_by_volume_key(cd, NULL, key, key_size, 0));
698         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
699         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
700         OK_(crypt_deactivate(cd, CDEVICE_1));
701
702         key[1] = ~key[1];
703         FAIL_(crypt_volume_key_verify(cd, key, key_size), "key mismatch");
704         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0), "key mismatch");
705         crypt_free(cd);
706 }
707
708 static void SuspendDevice(void)
709 {
710         int suspend_status;
711         struct crypt_device *cd;
712
713         OK_(crypt_init(&cd, DEVICE_1));
714         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
715         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1), 0));
716
717         suspend_status = crypt_suspend(cd, CDEVICE_1);
718         if (suspend_status == -ENOTSUP) {
719                 printf("WARNING: Suspend/Resume not supported, skipping test.\n");
720                 goto out;
721         }
722         OK_(suspend_status);
723         FAIL_(crypt_suspend(cd, CDEVICE_1), "already suspended");
724
725         FAIL_(crypt_resume_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1)-1), "wrong key");
726         OK_(crypt_resume_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1)));
727         FAIL_(crypt_resume_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1)), "not suspended");
728
729         OK_(_prepare_keyfile(KEYFILE1, KEY1));
730         OK_(crypt_suspend(cd, CDEVICE_1));
731         FAIL_(crypt_resume_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1 "blah", 0), "wrong keyfile");
732         OK_(crypt_resume_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 0));
733         FAIL_(crypt_resume_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 0), "not suspended");
734         _remove_keyfiles();
735 out:
736         OK_(crypt_deactivate(cd, CDEVICE_1));
737         crypt_free(cd);
738 }
739
740 static void AddDeviceLuks(void)
741 {
742         struct crypt_device *cd;
743         struct crypt_params_luks1 params = {
744                 .hash = "sha512",
745                 .data_alignment = 2048, // 4M, data offset will be 4096
746         };
747         char key[128], key2[128];
748
749         char *passphrase = "blabla";
750         char *mk_hex = "bb21158c733229347bd4e681891e213d94c685be6a5b84818afe7a78a6de7a1a";
751         size_t key_size = strlen(mk_hex) / 2;
752         char *cipher = "aes";
753         char *cipher_mode = "cbc-essiv:sha256";
754
755         crypt_decode_key(key, mk_hex, key_size);
756
757         OK_(crypt_init(&cd, DEVICE_2));
758         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
759
760         // even with no keyslots defined it can be activated by volume key
761         OK_(crypt_volume_key_verify(cd, key, key_size));
762         OK_(crypt_activate_by_volume_key(cd, CDEVICE_2, key, key_size, 0));
763         EQ_(crypt_status(cd, CDEVICE_2), CRYPT_ACTIVE);
764         OK_(crypt_deactivate(cd, CDEVICE_2));
765
766         // now with keyslot
767         EQ_(7, crypt_keyslot_add_by_volume_key(cd, 7, key, key_size, passphrase, strlen(passphrase)));
768         EQ_(CRYPT_SLOT_ACTIVE_LAST, crypt_keyslot_status(cd, 7));
769         EQ_(7, crypt_activate_by_passphrase(cd, CDEVICE_2, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0));
770         EQ_(crypt_status(cd, CDEVICE_2), CRYPT_ACTIVE);
771         OK_(crypt_deactivate(cd, CDEVICE_2));
772
773         EQ_(1, crypt_keyslot_add_by_volume_key(cd, 1, key, key_size, KEY1, strlen(KEY1)));
774         OK_(_prepare_keyfile(KEYFILE1, KEY1));
775         OK_(_prepare_keyfile(KEYFILE2, KEY2));
776         EQ_(2, crypt_keyslot_add_by_keyfile(cd, 2, KEYFILE1, 0, KEYFILE2, 0));
777         FAIL_(crypt_activate_by_keyfile(cd, CDEVICE_2, CRYPT_ANY_SLOT, KEYFILE2, strlen(KEY2)-1, 0), "key mismatch");
778         EQ_(2, crypt_activate_by_keyfile(cd, NULL, CRYPT_ANY_SLOT, KEYFILE2, 0, 0));
779         EQ_(2, crypt_activate_by_keyfile(cd, CDEVICE_2, CRYPT_ANY_SLOT, KEYFILE2, 0, 0));
780         OK_(crypt_keyslot_destroy(cd, 1));
781         OK_(crypt_keyslot_destroy(cd, 2));
782         OK_(crypt_deactivate(cd, CDEVICE_2));
783         _remove_keyfiles();
784
785         FAIL_(crypt_keyslot_add_by_volume_key(cd, 7, key, key_size, passphrase, strlen(passphrase)), "slot used");
786         key[1] = ~key[1];
787         FAIL_(crypt_keyslot_add_by_volume_key(cd, 6, key, key_size, passphrase, strlen(passphrase)), "key mismatch");
788         key[1] = ~key[1];
789         EQ_(6, crypt_keyslot_add_by_volume_key(cd, 6, key, key_size, passphrase, strlen(passphrase)));
790         EQ_(CRYPT_SLOT_ACTIVE, crypt_keyslot_status(cd, 6));
791
792         FAIL_(crypt_keyslot_destroy(cd, 8), "invalid keyslot");
793         FAIL_(crypt_keyslot_destroy(cd, CRYPT_ANY_SLOT), "invalid keyslot");
794         FAIL_(crypt_keyslot_destroy(cd, 0), "keyslot not used");
795         OK_(crypt_keyslot_destroy(cd, 7));
796         EQ_(CRYPT_SLOT_INACTIVE, crypt_keyslot_status(cd, 7));
797         EQ_(CRYPT_SLOT_ACTIVE_LAST, crypt_keyslot_status(cd, 6));
798
799         EQ_(6, crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key2, &key_size, passphrase, strlen(passphrase)));
800         OK_(crypt_volume_key_verify(cd, key2, key_size));
801
802         OK_(memcmp(key, key2, key_size));
803         OK_(strcmp(cipher, crypt_get_cipher(cd)));
804         OK_(strcmp(cipher_mode, crypt_get_cipher_mode(cd)));
805         EQ_(key_size, crypt_get_volume_key_size(cd));
806         EQ_(4096, crypt_get_data_offset(cd));
807         OK_(strcmp(DEVICE_2, crypt_get_device_name(cd)));
808
809         reset_log();
810         crypt_set_log_callback(cd, &new_log, NULL);
811         OK_(crypt_dump(cd));
812         OK_(!(global_lines != 0));
813         crypt_set_log_callback(cd, NULL, NULL);
814         reset_log();
815
816         FAIL_(crypt_set_uuid(cd, "blah"), "wrong UUID format");
817         OK_(crypt_set_uuid(cd, DEVICE_TEST_UUID));
818         OK_(strcmp(DEVICE_TEST_UUID, crypt_get_uuid(cd)));
819
820         FAIL_(crypt_deactivate(cd, CDEVICE_2), "not active");
821         crypt_free(cd);
822 }
823
824 static void UseTempVolumes(void)
825 {
826         struct crypt_device *cd;
827
828         // Tepmporary device without keyslot but with on-disk LUKS header
829         OK_(crypt_init(&cd, DEVICE_2));
830         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_2, NULL, 0, 0), "not yet formatted");
831         OK_(crypt_format(cd, CRYPT_LUKS1, "aes", "cbc-essiv:sha256", NULL, NULL, 16, NULL));
832         OK_(crypt_activate_by_volume_key(cd, CDEVICE_2, NULL, 0, 0));
833         EQ_(crypt_status(cd, CDEVICE_2), CRYPT_ACTIVE);
834         crypt_free(cd);
835
836         // Volume key is properly initialised from active device
837         OK_(crypt_init_by_name(&cd, CDEVICE_2));
838         OK_(crypt_deactivate(cd, CDEVICE_2));
839         OK_(crypt_activate_by_volume_key(cd, CDEVICE_2, NULL, 0, 0));
840         OK_(crypt_deactivate(cd, CDEVICE_2));
841         crypt_free(cd);
842
843         // Dirty checks: device without UUID
844         // we should be able to remove it but not manuipulate with it
845         _system("dmsetup create " CDEVICE_2 " --table \""
846                 "0 100 crypt aes-cbc-essiv:sha256 deadbabedeadbabedeadbabedeadbabe 0 "
847                 DEVICE_2 " 2048\"", 1);
848         OK_(crypt_init_by_name(&cd, CDEVICE_2));
849         OK_(crypt_deactivate(cd, CDEVICE_2));
850         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_2, NULL, 0, 0), "No known device type");
851         crypt_free(cd);
852
853         // Dirty checks: device with UUID but LUKS header key fingerprint must fail)
854         _system("dmsetup create " CDEVICE_2 " --table \""
855                 "0 100 crypt aes-cbc-essiv:sha256 deadbabedeadbabedeadbabedeadbabe 0 "
856                 DEVICE_2 " 2048\" "
857                 "-u CRYPT-LUKS1-aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa-ctest1", 1);
858         OK_(crypt_init_by_name(&cd, CDEVICE_2));
859         OK_(crypt_deactivate(cd, CDEVICE_2));
860         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_2, NULL, 0, 0), "wrong volume key");
861         crypt_free(cd);
862
863         // No slots
864         OK_(crypt_init(&cd, DEVICE_2));
865         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
866         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_2, NULL, 0, 0), "volume key is lost");
867         crypt_free(cd);
868
869         // Plain device
870         OK_(crypt_init(&cd, DEVICE_2));
871         OK_(crypt_format(cd, CRYPT_PLAIN, "aes", "cbc-essiv:sha256", NULL, NULL, 16, NULL));
872         FAIL_(crypt_activate_by_volume_key(cd, NULL, "xxx", 3, 0), "cannot verify key with plain");
873         FAIL_(crypt_volume_key_verify(cd, "xxx", 3), "cannot verify key with plain");
874         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_2, "xxx", 3, 0), "wrong key lenght");
875         OK_(crypt_activate_by_volume_key(cd, CDEVICE_2, "volumekeyvolumek", 16, 0));
876         EQ_(crypt_status(cd, CDEVICE_2), CRYPT_ACTIVE);
877         OK_(crypt_deactivate(cd, CDEVICE_2));
878         crypt_free(cd);
879 }
880
881 // Check that gcrypt is properly initialised in format
882 static void NonFIPSAlg(void)
883 {
884         struct crypt_device *cd;
885         struct crypt_params_luks1 params = {0};
886         char key[128] = "";
887         size_t key_size = 128;
888         char *cipher = "aes";
889         char *cipher_mode = "cbc-essiv:sha256";
890         int ret;
891
892         OK_(crypt_init(&cd, DEVICE_2));
893         params.hash = "sha256";
894         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
895
896         params.hash = "whirlpool";
897         ret = crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params);
898         if (ret < 0) {
899                 printf("WARNING: whirlpool not supported, skipping test.\n");
900                 crypt_free(cd);
901                 return;
902         }
903
904         params.hash = "md5";
905         FAIL_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params),
906               "MD5 unsupported, too short");
907         crypt_free(cd);
908 }
909
910 int main (int argc, char *argv[])
911 {
912         int i;
913
914         if (getuid() != 0) {
915                 printf("You must be root to run this test.\n");
916                 exit(0);
917         }
918
919         for (i = 1; i < argc; i++) {
920                 if (!strcmp("-v", argv[i]) || !strcmp("--verbose", argv[i]))
921                         _verbose = 1;
922                 else if (!strcmp("--debug", argv[i]))
923                         _debug = _verbose = 1;
924         }
925
926         _cleanup();
927         _setup();
928
929         crypt_set_debug_level(_debug ? CRYPT_DEBUG_ALL : CRYPT_DEBUG_NONE);
930
931         RUN_(NonFIPSAlg, "Crypto is properly initialised in format"); //must be the first!
932
933         RUN_(LuksUUID, "luksUUID API call");
934         RUN_(IsLuks, "isLuks API call");
935         RUN_(LuksOpen, "luksOpen API call");
936         RUN_(query_device, "crypt_query_device API call");
937         RUN_(remove_device, "crypt_remove_device API call");
938         RUN_(LuksFormat, "luksFormat API call");
939         RUN_(LuksKeyGame, "luksAddKey, RemoveKey, KillSlot API calls");
940         RUN_(DeviceResizeGame, "regular crypto, resize calls");
941
942         RUN_(AddDevicePlain, "plain device API creation exercise");
943         RUN_(AddDeviceLuks, "Format and use LUKS device");
944         RUN_(UseLuksDevice, "Use pre-formated LUKS device");
945         RUN_(SuspendDevice, "Suspend/Resume test");
946         RUN_(UseTempVolumes, "Format and use temporary encrypted device");
947
948         RUN_(CallbacksTest, "API callbacks test");
949
950         _cleanup();
951         return 0;
952 }