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