tests/api-test.c

   1 /*
   2  * cryptsetup library API check functions
   3  *
   4  * Copyright (C) 2009-2013 Red Hat, Inc. All rights reserved.
   5  * Copyright (C) 2009-2014, 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  * as published by the Free Software Foundation; either version 2
  10  * of the License, or (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  20  */
  21
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <unistd.h>
  26 #include <fcntl.h>
  27 #include <linux/fs.h>
  28 #include <errno.h>
  29 #include <assert.h>
  30 #include <signal.h>
  31 #include <sys/stat.h>
  32 #include <sys/ioctl.h>
  33 #include <libdevmapper.h>
  34
  35 #include "luks.h"
  36 #include "libcryptsetup.h"
  37 #include "utils_loop.h"
  38
  39 #define DMDIR "/dev/mapper/"
  40
  41 #define DEVICE_1_UUID "28632274-8c8a-493f-835b-da802e1c576b"
  42 #define DEVICE_EMPTY_name "crypt_zero"
  43 #define DEVICE_EMPTY DMDIR DEVICE_EMPTY_name
  44 #define DEVICE_ERROR_name "crypt_error"
  45 #define DEVICE_ERROR DMDIR DEVICE_ERROR_name
  46
  47 #define CDEVICE_1 "ctest1"
  48 #define CDEVICE_2 "ctest2"
  49 #define CDEVICE_WRONG "O_o"
  50 #define H_DEVICE "head_ok"
  51 #define H_DEVICE_WRONG "head_wr"
  52 #define L_DEVICE_1S "luks_onesec"
  53 #define L_DEVICE_0S "luks_zerosec"
  54 #define L_DEVICE_WRONG "luks_wr"
  55 #define L_DEVICE_OK "luks_ok"
  56 #define EVL_HEADER_1 "evil_hdr-luks_hdr_damage"
  57 #define EVL_HEADER_2 "evil_hdr-payload_overwrite"
  58 #define EVL_HEADER_3 "evil_hdr-stripes_payload_dmg"
  59 #define EVL_HEADER_4 "evil_hdr-small_luks_device"
  60 #define VALID_HEADER "valid_header_file"
  61 #define BACKUP_FILE "csetup_backup_file"
  62 #define IMAGE1 "compatimage.img"
  63 #define IMAGE_EMPTY "empty.img"
  64
  65 #define KEYFILE1 "key1.file"
  66 #define KEY1 "compatkey"
  67
  68 #define KEYFILE2 "key2.file"
  69 #define KEY2 "0123456789abcdef"
  70
  71 #define PASSPHRASE "blabla"
  72
  73 #define DEVICE_TEST_UUID "12345678-1234-1234-1234-123456789abc"
  74
  75 #define DEVICE_WRONG "/dev/Ooo_"
  76 #define DEVICE_CHAR "/dev/zero"
  77 #define THE_LFILE_TEMPLATE "cryptsetup-tstlp.XXXXXX"
  78
  79 #define SECTOR_SHIFT 9L
  80 #define SECTOR_SIZE 512
  81 #define TST_LOOP_FILE_SIZE (((1<<20)*50)>>SECTOR_SHIFT)
  82 #define DIV_ROUND_UP(n,d) (((n) + (d) - 1) / (d))
  83 #define DIV_ROUND_UP_MODULO(n,d) (DIV_ROUND_UP(n,d)*(d))
  84 #define LUKS_PHDR_SIZE_B 1024
  85
  86 static int _debug   = 0;
  87 static int _verbose = 1;
  88 static int _fips_mode = 0;
  89
  90 static int _quit = 0;
  91
  92 static char global_log[4096];
  93 static int global_lines = 0;
  94
  95 static char *DEVICE_1 = NULL;
  96 static char *DEVICE_2 = NULL;
  97 static char *DEVICE_3 = NULL;
  98 static char *THE_LOOP_DEV = NULL;
  99
 100 static char *tmp_file_1 = NULL;
 101 static char *test_loop_file = NULL;
 102 static uint64_t t_dev_offset = 0;
 103
 104 static int _system(const char*, int);
 105
 106 // Helpers
 107
 108 static int device_size(const char *device, uint64_t *size)
 109 {
 110         int devfd, r = 0;
 111
 112         devfd = open(device, O_RDONLY);
 113         if(devfd == -1)
 114                 return -EINVAL;
 115
 116         if (ioctl(devfd, BLKGETSIZE64, size) < 0)
 117                 r = -EINVAL;
 118         close(devfd);
 119         return r;
 120 }
 121
 122 static int fips_mode(void)
 123 {
 124         int fd;
 125         char buf = 0;
 126
 127         if (access("/etc/system-fips", F_OK))
 128                 return 0;
 129
 130         fd = open("/proc/sys/crypto/fips_enabled", O_RDONLY);
 131
 132         if (fd < 0)
 133                 return 0;
 134
 135         if (read(fd, &buf, 1) != 1)
 136                 buf = '0';
 137
 138         close(fd);
 139
 140         return (buf == '1');
 141 }
 142
 143 static int get_luks_offsets(int metadata_device,
 144                             size_t keylength,
 145                             unsigned int alignpayload_sec,
 146                             unsigned int alignoffset_sec,
 147                             uint64_t *r_header_size,
 148                             uint64_t *r_payload_offset)
 149 {
 150         int i;
 151         uint64_t current_sector;
 152         uint32_t sectors_per_stripes_set;
 153
 154         if (!keylength)
 155                 return -1;
 156
 157         sectors_per_stripes_set = DIV_ROUND_UP(keylength*LUKS_STRIPES, SECTOR_SIZE);
 158         printf("sectors_per_stripes %" PRIu32 "\n", sectors_per_stripes_set);
 159         current_sector = DIV_ROUND_UP_MODULO(DIV_ROUND_UP(LUKS_PHDR_SIZE_B, SECTOR_SIZE),
 160                         LUKS_ALIGN_KEYSLOTS / SECTOR_SIZE);
 161         for(i=0;i < (LUKS_NUMKEYS - 1);i++)
 162                 current_sector = DIV_ROUND_UP_MODULO(current_sector + sectors_per_stripes_set,
 163                                 LUKS_ALIGN_KEYSLOTS / SECTOR_SIZE);
 164         if (r_header_size)
 165                 *r_header_size = current_sector + sectors_per_stripes_set;
 166
 167         current_sector = DIV_ROUND_UP_MODULO(current_sector + sectors_per_stripes_set,
 168                                 LUKS_ALIGN_KEYSLOTS / SECTOR_SIZE);
 169
 170         if (r_payload_offset) {
 171                 if (metadata_device)
 172                         *r_payload_offset = alignpayload_sec;
 173                 else
 174                         *r_payload_offset = DIV_ROUND_UP_MODULO(current_sector, alignpayload_sec)
 175                                 + alignoffset_sec;
 176         }
 177
 178         return 0;
 179 }
 180
 181 /*
 182  * Creates dm-linear target over the test loop device. Offset is held in
 183  * global variables so that size can be tested whether it fits into remaining
 184  * size of the loop device or not
 185  */
 186 static int create_dmdevice_over_loop(const char *dm_name, const uint64_t size)
 187 {
 188         char cmd[128];
 189         int r;
 190         uint64_t r_size;
 191
 192         if(device_size(THE_LOOP_DEV, &r_size) < 0 || r_size <= t_dev_offset || !size)
 193                 return -1;
 194         if ((r_size - t_dev_offset) < size) {
 195                 printf("No enough space on backing loop device\n.");
 196                 return -2;
 197         }
 198         snprintf(cmd, sizeof(cmd),
 199                  "dmsetup create %s --table \"0 %" PRIu64 " linear %s %" PRIu64 "\"",
 200                  dm_name, size, THE_LOOP_DEV, t_dev_offset);
 201         if  (!(r = _system(cmd, 1))) {
 202                 t_dev_offset += size;
 203         }
 204         return r;
 205 }
 206
 207 // TODO some utility to remove dmdevice over the loop file
 208
 209 // Get key from kernel dm mapping table using dm-ioctl
 210 static int _get_key_dm(const char *name, char *buffer, unsigned int buffer_size)
 211 {
 212         struct dm_task *dmt;
 213         struct dm_info dmi;
 214         uint64_t start, length;
 215         char *target_type, *key, *params;
 216         void *next = NULL;
 217         int r = -EINVAL;
 218
 219         if (!(dmt = dm_task_create(DM_DEVICE_TABLE)))
 220                 goto out;
 221         if (!dm_task_set_name(dmt, name))
 222                 goto out;
 223         if (!dm_task_run(dmt))
 224                 goto out;
 225         if (!dm_task_get_info(dmt, &dmi))
 226                 goto out;
 227         if (!dmi.exists)
 228                 goto out;
 229
 230         next = dm_get_next_target(dmt, next, &start, &length, &target_type, &params);
 231         if (!target_type || strcmp(target_type, "crypt") != 0)
 232                 goto out;
 233
 234         (void)strsep(&params, " "); /* rcipher */
 235         key = strsep(&params, " ");
 236
 237         if (buffer_size <= strlen(key))
 238                 goto out;
 239
 240         strncpy(buffer, key, buffer_size);
 241         r = 0;
 242 out:
 243         if (dmt)
 244                 dm_task_destroy(dmt);
 245
 246         return r;
 247 }
 248
 249 static int _prepare_keyfile(const char *name, const char *passphrase, int size)
 250 {
 251         int fd, r;
 252
 253         fd = open(name, O_RDWR | O_CREAT | O_TRUNC, S_IRUSR|S_IWUSR);
 254         if (fd != -1) {
 255                 r = write(fd, passphrase, size);
 256                 close(fd);
 257         } else
 258                 r = 0;
 259
 260         return r == size ? 0 : 1;
 261 }
 262
 263 static void _remove_keyfiles(void)
 264 {
 265         remove(KEYFILE1);
 266         remove(KEYFILE2);
 267 }
 268
 269 // Decode key from its hex representation
 270 static int crypt_decode_key(char *key, const char *hex, unsigned int size)
 271 {
 272         char buffer[3];
 273         char *endp;
 274         unsigned int i;
 275
 276         buffer[2] = '\0';
 277
 278         for (i = 0; i < size; i++) {
 279                 buffer[0] = *hex++;
 280                 buffer[1] = *hex++;
 281
 282                 key[i] = (unsigned char)strtoul(buffer, &endp, 16);
 283
 284                 if (endp != &buffer[2])
 285                         return -1;
 286         }
 287
 288         if (*hex != '\0')
 289                 return -1;
 290
 291         return 0;
 292 }
 293
 294 static void cmdLineLog(int level, const char *msg)
 295 {
 296         strncat(global_log, msg, sizeof(global_log) - strlen(global_log));
 297         global_lines++;
 298 }
 299
 300 static void new_log(int level, const char *msg, void *usrptr)
 301 {
 302         if (_debug)
 303                 printf("LOG: %s", msg);
 304         cmdLineLog(level, msg);
 305 }
 306
 307 static void reset_log(void)
 308 {
 309         memset(global_log, 0, sizeof(global_log));
 310         global_lines = 0;
 311 }
 312
 313 static int _system(const char *command, int warn)
 314 {
 315         int r;
 316         if (_debug)
 317                 printf("Running system: %s\n", command);
 318         if ((r=system(command)) < 0 && warn)
 319                 printf("System command failed: %s", command);
 320         return r;
 321 }
 322
 323 static void _cleanup_dmdevices(void)
 324 {
 325         struct stat st;
 326
 327         if (!stat(DMDIR H_DEVICE, &st)) {
 328                 _system("dmsetup remove " H_DEVICE, 0);
 329         }
 330         if (!stat(DMDIR H_DEVICE_WRONG, &st)) {
 331                 _system("dmsetup remove " H_DEVICE_WRONG, 0);
 332         }
 333         if (!stat(DMDIR L_DEVICE_0S, &st)) {
 334                 _system("dmsetup remove " L_DEVICE_0S, 0);
 335         }
 336         if (!stat(DMDIR L_DEVICE_1S, &st)) {
 337                 _system("dmsetup remove " L_DEVICE_1S, 0);
 338         }
 339         if (!stat(DMDIR L_DEVICE_WRONG, &st)) {
 340                 _system("dmsetup remove " L_DEVICE_WRONG, 0);
 341         }
 342         if (!stat(DMDIR L_DEVICE_OK, &st)) {
 343                 _system("dmsetup remove " L_DEVICE_OK, 0);
 344         }
 345
 346         t_dev_offset = 0;
 347 }
 348
 349 static void _cleanup(void)
 350 {
 351         struct stat st;
 352
 353         //_system("udevadm settle", 0);
 354
 355         if (!stat(DMDIR CDEVICE_1, &st))
 356                 _system("dmsetup remove " CDEVICE_1, 0);
 357
 358         if (!stat(DMDIR CDEVICE_2, &st))
 359                 _system("dmsetup remove " CDEVICE_2, 0);
 360
 361         if (!stat(DEVICE_EMPTY, &st))
 362                 _system("dmsetup remove " DEVICE_EMPTY_name, 0);
 363
 364         if (!stat(DEVICE_ERROR, &st))
 365                 _system("dmsetup remove " DEVICE_ERROR_name, 0);
 366
 367         _cleanup_dmdevices();
 368
 369         if (crypt_loop_device(THE_LOOP_DEV))
 370                 crypt_loop_detach(THE_LOOP_DEV);
 371
 372         if (crypt_loop_device(DEVICE_1))
 373                 crypt_loop_detach(DEVICE_1);
 374
 375         if (crypt_loop_device(DEVICE_2))
 376                 crypt_loop_detach(DEVICE_2);
 377
 378         if (crypt_loop_device(DEVICE_3))
 379                 crypt_loop_detach(DEVICE_3);
 380
 381         _system("rm -f " IMAGE_EMPTY, 0);
 382         _system("rm -f " IMAGE1, 0);
 383
 384         remove(test_loop_file);
 385         remove(tmp_file_1);
 386
 387         remove(EVL_HEADER_1);
 388         remove(EVL_HEADER_2);
 389         remove(EVL_HEADER_3);
 390         remove(EVL_HEADER_4);
 391         remove(VALID_HEADER);
 392         remove(BACKUP_FILE);
 393
 394         _remove_keyfiles();
 395
 396         free(tmp_file_1);
 397         free(test_loop_file);
 398         free(THE_LOOP_DEV);
 399         free(DEVICE_1);
 400         free(DEVICE_2);
 401         free(DEVICE_3);
 402 }
 403
 404 static int _setup(void)
 405 {
 406         int fd, ro = 0;
 407         char cmd[128];
 408
 409         test_loop_file = strdup(THE_LFILE_TEMPLATE);
 410         if ((fd=mkstemp(test_loop_file)) == -1) {
 411                 printf("cannot create temporary file with template %s\n", test_loop_file);
 412                 return 1;
 413         }
 414         close(fd);
 415         snprintf(cmd, sizeof(cmd), "dd if=/dev/zero of=%s bs=%d count=%d 2>/dev/null",
 416                  test_loop_file, SECTOR_SIZE, TST_LOOP_FILE_SIZE);
 417         if (_system(cmd, 1))
 418                 return 1;
 419
 420         if (!THE_LOOP_DEV)
 421                 THE_LOOP_DEV = crypt_loop_get_device();
 422         if (!THE_LOOP_DEV) {
 423                 printf("Cannot find free loop device.\n");
 424                 return 1;
 425         }
 426         if (crypt_loop_device(THE_LOOP_DEV)) {
 427                 fd = crypt_loop_attach(THE_LOOP_DEV, test_loop_file, 0, 0, &ro);
 428                 close(fd);
 429         }
 430
 431         tmp_file_1 = strdup(THE_LFILE_TEMPLATE);
 432         if ((fd=mkstemp(tmp_file_1)) == -1) {
 433                 printf("cannot create temporary file with template %s\n", tmp_file_1);
 434                 return 1;
 435         }
 436         close(fd);
 437         snprintf(cmd, sizeof(cmd), "dd if=/dev/zero of=%s bs=%d count=%d 2>/dev/null",
 438                  tmp_file_1, SECTOR_SIZE, 10);
 439         if (_system(cmd, 1))
 440                 return 1;
 441
 442         _system("dmsetup create " DEVICE_EMPTY_name " --table \"0 10000 zero\"", 1);
 443         _system("dmsetup create " DEVICE_ERROR_name " --table \"0 10000 error\"", 1);
 444         if (!DEVICE_1)
 445                 DEVICE_1 = crypt_loop_get_device();
 446         if (!DEVICE_1) {
 447                 printf("Cannot find free loop device.\n");
 448                 return 1;
 449         }
 450         if (crypt_loop_device(DEVICE_1)) {
 451                 _system(" [ ! -e " IMAGE1 " ] && bzip2 -dk " IMAGE1 ".bz2", 1);
 452                 fd = crypt_loop_attach(DEVICE_1, IMAGE1, 0, 0, &ro);
 453                 close(fd);
 454         }
 455         if (!DEVICE_2)
 456                 DEVICE_2 = crypt_loop_get_device();
 457         if (!DEVICE_2) {
 458                 printf("Cannot find free loop device.\n");
 459                 return 1;
 460         }
 461         if (crypt_loop_device(DEVICE_2)) {
 462                 _system("dd if=/dev/zero of=" IMAGE_EMPTY " bs=1M count=4 2>/dev/null", 1);
 463                 fd = crypt_loop_attach(DEVICE_2, IMAGE_EMPTY, 0, 0, &ro);
 464                 close(fd);
 465         }
 466         if (!DEVICE_3)
 467                 DEVICE_3 = crypt_loop_get_device();
 468         if (!DEVICE_3) {
 469                 printf("Cannot find free loop device.\n");
 470                 return 1;
 471         }
 472         /* Keymaterial offset is less than 8 sectors */
 473         _system(" [ ! -e " EVL_HEADER_1 " ] && bzip2 -dk " EVL_HEADER_1 ".bz2", 1);
 474         /* keymaterial offset aims into payload area */
 475         _system(" [ ! -e " EVL_HEADER_2 " ] && bzip2 -dk " EVL_HEADER_2 ".bz2", 1);
 476         /* keymaterial offset is valid, number of stripes causes payload area to be overwriten */
 477         _system(" [ ! -e " EVL_HEADER_3 " ] && bzip2 -dk " EVL_HEADER_3 ".bz2", 1);
 478         /* luks device header for data and header on same device. payloadOffset is greater than
 479          * device size (crypt_load() test) */
 480         _system(" [ ! -e " EVL_HEADER_4 " ] && bzip2 -dk " EVL_HEADER_4 ".bz2", 1);
 481         /* valid header: payloadOffset=4096, key_size=32,
 482          * volume_key = bb21158c733229347bd4e681891e213d94c685be6a5b84818afe7a78a6de7a1a */
 483         _system(" [ ! -e " VALID_HEADER " ] && bzip2 -dk " VALID_HEADER ".bz2", 1);
 484
 485         /* Prepare tcrypt images */
 486         _system(" [ ! -d tcrypt-images ] && tar xjf tcrypt-images.tar.bz2 2>/dev/null", 1);
 487
 488         _system("modprobe dm-crypt", 0);
 489         _system("modprobe dm-verity", 0);
 490
 491         _fips_mode = fips_mode();
 492         if (_debug)
 493                 printf("FIPS MODE: %d\n", _fips_mode);
 494
 495         return 0;
 496 }
 497
 498 static void check_ok(int status, int line, const char *func)
 499 {
 500         char buf[256];
 501
 502         if (status) {
 503                 crypt_get_error(buf, sizeof(buf));
 504                 printf("FAIL line %d [%s]: code %d, %s\n", line, func, status, buf);
 505                 _cleanup();
 506                 exit(-1);
 507         }
 508 }
 509
 510 static void check_ko(int status, int line, const char *func)
 511 {
 512         char buf[256];
 513
 514         memset(buf, 0, sizeof(buf));
 515         crypt_get_error(buf, sizeof(buf));
 516         if (status >= 0) {
 517                 printf("FAIL line %d [%s]: code %d, %s\n", line, func, status, buf);
 518                 _cleanup();
 519                 exit(-1);
 520         } else if (_verbose)
 521                 printf("   => errno %d, errmsg: %s\n", status, buf);
 522 }
 523
 524 static void check_equal(int line, const char *func, int64_t x, int64_t y)
 525 {
 526         printf("FAIL line %d [%s]: expected equal values differs: %"
 527                 PRIi64 " != %" PRIi64 "\n", line, func, x, y);
 528         _cleanup();
 529         exit(-1);
 530 }
 531
 532 static void xlog(const char *msg, const char *tst, const char *func, int line, const char *txt)
 533 {
 534         if (_verbose) {
 535                 if (txt)
 536                         printf(" [%s,%s:%d] %s [%s]\n", msg, func, line, tst, txt);
 537                 else
 538                         printf(" [%s,%s:%d] %s\n", msg, func, line, tst);
 539         }
 540         if (_quit) {
 541                 if (_verbose)
 542                         printf("Interrupted by a signal.\n");
 543                 _cleanup();
 544                 exit(-1);
 545         }
 546 }
 547
 548 /* crypt_device context must be "cd" to parse error properly here */
 549 #define OK_(x)          do { xlog("(success)", #x, __FUNCTION__, __LINE__, NULL); \
 550                              check_ok((x), __LINE__, __FUNCTION__); \
 551                         } while(0)
 552 #define FAIL_(x, y)     do { xlog("(fail)   ", #x, __FUNCTION__, __LINE__, y); \
 553                              check_ko((x), __LINE__, __FUNCTION__); \
 554                         } while(0)
 555 #define EQ_(x, y)       do { int64_t _x = (x), _y = (y); \
 556                              xlog("(equal)  ", #x " == " #y, __FUNCTION__, __LINE__, NULL); \
 557                              if (_x != _y) check_equal(__LINE__, __FUNCTION__, _x, _y); \
 558                         } while(0)
 559 #define RUN_(x, y)              do { printf("%s: %s\n", #x, (y)); x(); } while (0)
 560
 561 static void AddDevicePlain(void)
 562 {
 563         struct crypt_device *cd;
 564         struct crypt_params_plain params = {
 565                 .hash = "sha1",
 566                 .skip = 0,
 567                 .offset = 0,
 568                 .size = 0
 569         };
 570         int fd;
 571         char key[128], key2[128], path[128];
 572
 573         const char *passphrase = PASSPHRASE;
 574         // hashed hex version of PASSPHRASE
 575         const char *mk_hex = "bb21158c733229347bd4e681891e213d94c685be6a5b84818afe7a78a6de7a1a";
 576         size_t key_size = strlen(mk_hex) / 2;
 577         const char *cipher = "aes";
 578         const char *cipher_mode = "cbc-essiv:sha256";
 579
 580         uint64_t size, r_size;
 581
 582         crypt_decode_key(key, mk_hex, key_size);
 583         FAIL_(crypt_init(&cd, ""), "empty device string");
 584         FAIL_(crypt_init(&cd, DEVICE_WRONG), "nonexistent device name ");
 585         FAIL_(crypt_init(&cd, DEVICE_CHAR), "character device as backing device");
 586         OK_(crypt_init(&cd, tmp_file_1));
 587         crypt_free(cd);
 588
 589         // test crypt_format, crypt_get_cipher, crypt_get_cipher_mode, crypt_get_volume_key_size
 590         OK_(crypt_init(&cd,DEVICE_1));
 591         params.skip = 3;
 592         params.offset = 42;
 593         FAIL_(crypt_format(cd,CRYPT_PLAIN,NULL,cipher_mode,NULL,NULL,key_size,&params),"cipher param is null");
 594         FAIL_(crypt_format(cd,CRYPT_PLAIN,cipher,NULL,NULL,NULL,key_size,&params),"cipher_mode param is null");
 595         OK_(crypt_format(cd,CRYPT_PLAIN,cipher,cipher_mode,NULL,NULL,key_size,&params));
 596         OK_(strcmp(cipher_mode,crypt_get_cipher_mode(cd)));
 597         OK_(strcmp(cipher,crypt_get_cipher(cd)));
 598         EQ_((int)key_size, crypt_get_volume_key_size(cd));
 599         EQ_(params.skip, crypt_get_iv_offset(cd));
 600         EQ_(params.offset, crypt_get_data_offset(cd));
 601         params.skip = 0;
 602         params.offset = 0;
 603
 604         // crypt_set_uuid()
 605         FAIL_(crypt_set_uuid(cd,DEVICE_1_UUID),"can't set uuid to plain device");
 606
 607         crypt_free(cd);
 608
 609         // default is "plain" hash - no password hash
 610         OK_(crypt_init(&cd, DEVICE_1));
 611         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, NULL));
 612         FAIL_(crypt_activate_by_volume_key(cd, NULL, key, key_size, 0), "cannot verify key with plain");
 613         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
 614         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 615         OK_(crypt_deactivate(cd, CDEVICE_1));
 616         crypt_free(cd);
 617
 618         // test boundaries in offset parameter
 619         device_size(DEVICE_1,&size);
 620         params.hash = NULL;
 621         // zero sectors length
 622         params.offset = size >> SECTOR_SHIFT;
 623         OK_(crypt_init(&cd, DEVICE_1));
 624         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
 625         EQ_(crypt_get_data_offset(cd),params.offset);
 626         // device size is 0 sectors
 627         FAIL_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0), "invalid device size (0 blocks)");
 628         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_INACTIVE);
 629         // data part of crypt device is of 1 sector size
 630         params.offset = (size >> SECTOR_SHIFT) - 1;
 631         crypt_free(cd);
 632
 633         OK_(crypt_init(&cd, DEVICE_1));
 634         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
 635         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0));
 636         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 637         snprintf(path, sizeof(path), "%s/%s", crypt_get_dir(), CDEVICE_1);
 638         if (device_size(path, &r_size) >= 0)
 639                 EQ_(r_size>>SECTOR_SHIFT, 1);
 640         OK_(crypt_deactivate(cd, CDEVICE_1));
 641         crypt_free(cd);
 642
 643         // size > device_size
 644         params.offset = 0;
 645         params.size = (size >> SECTOR_SHIFT) + 1;
 646         crypt_init(&cd, DEVICE_1);
 647         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
 648         FAIL_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0),"Device too small");
 649         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_INACTIVE);
 650         crypt_free(cd);
 651
 652         // offset == device_size (autodetect size)
 653         params.offset = (size >> SECTOR_SHIFT);
 654         params.size = 0;
 655         crypt_init(&cd, DEVICE_1);
 656         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
 657         FAIL_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0),"Device too small");
 658         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_INACTIVE);
 659         crypt_free(cd);
 660
 661         // offset == device_size (user defined size)
 662         params.offset = (size >> SECTOR_SHIFT);
 663         params.size = 123;
 664         crypt_init(&cd, DEVICE_1);
 665         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
 666         FAIL_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0),"Device too small");
 667         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_INACTIVE);
 668         crypt_free(cd);
 669
 670         // offset+size > device_size
 671         params.offset = 42;
 672         params.size = (size >> SECTOR_SHIFT) - params.offset + 1;
 673         crypt_init(&cd, DEVICE_1);
 674         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
 675         FAIL_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0),"Offset and size are beyond device real size");
 676         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_INACTIVE);
 677         crypt_free(cd);
 678
 679         // offset+size == device_size
 680         params.offset = 42;
 681         params.size = (size >> SECTOR_SHIFT) - params.offset;
 682         crypt_init(&cd, DEVICE_1);
 683         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
 684         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0));
 685         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 686         if (!device_size(path, &r_size))
 687                 EQ_((r_size >> SECTOR_SHIFT),params.size);
 688         OK_(crypt_deactivate(cd,CDEVICE_1));
 689
 690         crypt_free(cd);
 691         params.hash = "sha1";
 692         params.offset = 0;
 693         params.size = 0;
 694         params.skip = 0;
 695
 696         // Now use hashed password
 697         OK_(crypt_init(&cd, DEVICE_1));
 698         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
 699         FAIL_(crypt_activate_by_passphrase(cd, NULL, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0),
 700               "cannot verify passphrase with plain" );
 701         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0));
 702
 703         // device status check
 704         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 705         snprintf(path, sizeof(path), "%s/%s", crypt_get_dir(), CDEVICE_1);
 706         fd = open(path, O_RDONLY);
 707         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_BUSY);
 708         FAIL_(crypt_deactivate(cd, CDEVICE_1), "Device is busy");
 709         close(fd);
 710         OK_(crypt_deactivate(cd, CDEVICE_1));
 711         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_INACTIVE);
 712         crypt_free(cd);
 713
 714         // crypt_init_by_name_and_header
 715         OK_(crypt_init(&cd,DEVICE_1));
 716         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
 717         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
 718         crypt_free(cd);
 719
 720         FAIL_(crypt_init_by_name_and_header(&cd, CDEVICE_1, H_DEVICE),"can't init plain device by header device");
 721         OK_(crypt_init_by_name(&cd, CDEVICE_1));
 722         OK_(strcmp(cipher_mode,crypt_get_cipher_mode(cd)));
 723         OK_(strcmp(cipher,crypt_get_cipher(cd)));
 724         EQ_((int)key_size, crypt_get_volume_key_size(cd));
 725         EQ_(params.skip, crypt_get_iv_offset(cd));
 726         EQ_(params.offset, crypt_get_data_offset(cd));
 727         OK_(crypt_deactivate(cd, CDEVICE_1));
 728         crypt_free(cd);
 729
 730         OK_(crypt_init(&cd,DEVICE_1));
 731         OK_(crypt_format(cd,CRYPT_PLAIN,cipher,cipher_mode,NULL,NULL,key_size,&params));
 732         params.size = 0;
 733         params.offset = 0;
 734
 735         // crypt_set_data_device
 736         FAIL_(crypt_set_data_device(cd,H_DEVICE),"can't set data device for plain device");
 737
 738         // crypt_get_type
 739         OK_(strcmp(crypt_get_type(cd),CRYPT_PLAIN));
 740
 741         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
 742         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 743
 744         // crypt_resize()
 745         OK_(crypt_resize(cd,CDEVICE_1,size>>SECTOR_SHIFT)); // same size
 746         if (!device_size(path,&r_size))
 747                 EQ_(r_size, size);
 748
 749         // size overlaps
 750         FAIL_(crypt_resize(cd, CDEVICE_1, (uint64_t)-1),"Backing device is too small");
 751         FAIL_(crypt_resize(cd, CDEVICE_1, (size>>SECTOR_SHIFT)+1),"crypt device overlaps backing device");
 752
 753         // resize ok
 754         OK_(crypt_resize(cd,CDEVICE_1, 123));
 755         if (!device_size(path,&r_size))
 756                 EQ_(r_size>>SECTOR_SHIFT, 123);
 757         OK_(crypt_resize(cd,CDEVICE_1,0)); // full size (autodetect)
 758         if (!device_size(path,&r_size))
 759                 EQ_(r_size, size);
 760         OK_(crypt_deactivate(cd,CDEVICE_1));
 761         EQ_(crypt_status(cd,CDEVICE_1),CRYPT_INACTIVE);
 762         crypt_free(cd);
 763
 764         // offset tests
 765         OK_(crypt_init(&cd,DEVICE_1));
 766         params.offset = 42;
 767         params.size = (size>>SECTOR_SHIFT) - params.offset - 10;
 768         OK_(crypt_format(cd,CRYPT_PLAIN,cipher,cipher_mode,NULL,NULL,key_size,&params));
 769         OK_(crypt_activate_by_volume_key(cd,CDEVICE_1,key,key_size,0));
 770         if (!device_size(path,&r_size))
 771                 EQ_(r_size>>SECTOR_SHIFT, params.size);
 772         // resize to fill remaining capacity
 773         OK_(crypt_resize(cd,CDEVICE_1,params.size + 10));
 774         if (!device_size(path,&r_size))
 775                 EQ_(r_size>>SECTOR_SHIFT, params.size + 10);
 776
 777         // 1 sector beyond real size
 778         FAIL_(crypt_resize(cd,CDEVICE_1,params.size + 11), "new device size overlaps backing device"); // with respect to offset
 779         if (!device_size(path,&r_size))
 780                 EQ_(r_size>>SECTOR_SHIFT, params.size + 10);
 781         EQ_(crypt_status(cd,CDEVICE_1),CRYPT_ACTIVE);
 782         fd = open(path, O_RDONLY);
 783         close(fd);
 784         OK_(fd < 0);
 785
 786         // resize to minimal size
 787         OK_(crypt_resize(cd,CDEVICE_1, 1)); // minimal device size
 788         if (!device_size(path,&r_size))
 789                 EQ_(r_size>>SECTOR_SHIFT, 1);
 790         // use size of backing device (autodetect with respect to offset)
 791         OK_(crypt_resize(cd,CDEVICE_1,0));
 792         if (!device_size(path,&r_size))
 793                 EQ_(r_size>>SECTOR_SHIFT, (size >> SECTOR_SHIFT)- 42);
 794         OK_(crypt_deactivate(cd,CDEVICE_1));
 795         crypt_free(cd);
 796
 797         params.size = 0;
 798         params.offset = 0;
 799         OK_(crypt_init(&cd,DEVICE_1));
 800         OK_(crypt_format(cd,CRYPT_PLAIN,cipher,cipher_mode,NULL,NULL,key_size,&params));
 801         OK_(crypt_activate_by_volume_key(cd,CDEVICE_1,key,key_size,0));
 802
 803         // suspend/resume tests
 804         FAIL_(crypt_suspend(cd,CDEVICE_1),"cannot suspend plain device");
 805         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 806         FAIL_(crypt_resume_by_passphrase(cd,CDEVICE_1,CRYPT_ANY_SLOT,passphrase, strlen(passphrase)),"cannot resume plain device");
 807         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 808
 809         // retrieve volume key check
 810         if (!_fips_mode) {
 811                 memset(key2, 0, key_size);
 812                 key_size--;
 813                 // small buffer
 814                 FAIL_(crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key2, &key_size, passphrase, strlen(passphrase)), "small buffer");
 815                 key_size++;
 816                 OK_(crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key2, &key_size, passphrase, strlen(passphrase)));
 817
 818                 OK_(memcmp(key, key2, key_size));
 819         }
 820         OK_(strcmp(cipher, crypt_get_cipher(cd)));
 821         OK_(strcmp(cipher_mode, crypt_get_cipher_mode(cd)));
 822         EQ_((int)key_size, crypt_get_volume_key_size(cd));
 823         EQ_(0, crypt_get_data_offset(cd));
 824         OK_(crypt_deactivate(cd, CDEVICE_1));
 825
 826         // now with keyfile
 827         OK_(_prepare_keyfile(KEYFILE1, KEY1, strlen(KEY1)));
 828         OK_(_prepare_keyfile(KEYFILE2, KEY2, strlen(KEY2)));
 829         FAIL_(crypt_activate_by_keyfile(cd, NULL, CRYPT_ANY_SLOT, KEYFILE1, 0, 0), "cannot verify key with plain");
 830         EQ_(0, crypt_activate_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 0, 0));
 831         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 832         OK_(crypt_deactivate(cd, CDEVICE_1));
 833         FAIL_(crypt_activate_by_keyfile_offset(cd, NULL, CRYPT_ANY_SLOT, KEYFILE1, 0, strlen(KEY1) + 1, 0), "cannot seek");
 834         EQ_(0, crypt_activate_by_keyfile_offset(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 0, 0, 0));
 835         OK_(crypt_deactivate(cd, CDEVICE_1));
 836         _remove_keyfiles();
 837         crypt_free(cd);
 838
 839         OK_(crypt_init(&cd,DEVICE_1));
 840         OK_(crypt_format(cd,CRYPT_PLAIN,cipher,cipher_mode,NULL,NULL,key_size,&params));
 841
 842         // crypt_keyslot_*()
 843         FAIL_(crypt_keyslot_add_by_passphrase(cd,CRYPT_ANY_SLOT,passphrase,strlen(passphrase),passphrase,strlen(passphrase)), "can't add keyslot to plain device");
 844         FAIL_(crypt_keyslot_add_by_volume_key(cd,CRYPT_ANY_SLOT ,key,key_size,passphrase,strlen(passphrase)),"can't add keyslot to plain device");
 845         FAIL_(crypt_keyslot_add_by_keyfile(cd,CRYPT_ANY_SLOT,KEYFILE1,strlen(KEY1),KEYFILE2,strlen(KEY2)),"can't add keyslot to plain device");
 846         FAIL_(crypt_keyslot_destroy(cd,1),"can't manipulate keyslots on plain device");
 847         EQ_(crypt_keyslot_status(cd, 0), CRYPT_SLOT_INVALID);
 848         _remove_keyfiles();
 849
 850         crypt_free(cd);
 851 }
 852
 853 #define CALLBACK_ERROR "calback_error xyz"
 854 static int pass_callback_err(const char *msg, char *buf, size_t length, void *usrptr)
 855 {
 856         struct crypt_device *cd = usrptr;
 857
 858         assert(cd);
 859         assert(length);
 860         assert(msg);
 861
 862         crypt_log(cd, CRYPT_LOG_ERROR, CALLBACK_ERROR);
 863         return -EINVAL;
 864 }
 865
 866 static int pass_callback_ok(const char *msg, char *buf, size_t length, void *usrptr)
 867 {
 868         assert(length);
 869         assert(msg);
 870         strcpy(buf, PASSPHRASE);
 871         return strlen(buf);
 872 }
 873
 874 static void CallbacksTest(void)
 875 {
 876         struct crypt_device *cd;
 877         struct crypt_params_plain params = {
 878                 .hash = "sha1",
 879                 .skip = 0,
 880                 .offset = 0,
 881         };
 882
 883         size_t key_size = 256 / 8;
 884         const char *cipher = "aes";
 885         const char *cipher_mode = "cbc-essiv:sha256";
 886         const char *passphrase = PASSPHRASE;
 887         char buf1[256] = {0}, buf2[256] = {0};
 888
 889         OK_(crypt_init(&cd, DEVICE_1));
 890         crypt_set_log_callback(cd, &new_log, NULL);
 891         //crypt_set_log_callback(cd, NULL, NULL);
 892
 893         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &params));
 894
 895         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0));
 896         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 897         OK_(crypt_deactivate(cd, CDEVICE_1));
 898
 899         reset_log();
 900         crypt_set_password_callback(cd, pass_callback_err, cd);
 901         FAIL_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, NULL, 0, 0), "callback fails");
 902         EQ_(strncmp(global_log, CALLBACK_ERROR, strlen(CALLBACK_ERROR)), 0);
 903
 904         crypt_set_password_callback(cd, pass_callback_ok, NULL);
 905         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, NULL, 0, 0));
 906         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 907         OK_(crypt_deactivate(cd, CDEVICE_1));
 908
 909         // Check error reporting.
 910         // This must fail and create error message
 911         crypt_deactivate(cd, CDEVICE_1);
 912
 913         // Here context must be the same
 914         crypt_get_error(buf1, sizeof(buf1));
 915         crypt_last_error(cd, buf2, sizeof(buf2));
 916         OK_(!*buf1);
 917         OK_(!*buf2);
 918         OK_(strcmp(buf1, buf2));
 919
 920         crypt_get_error(buf1, sizeof(buf1));
 921         crypt_last_error(cd, buf2, sizeof(buf2));
 922         OK_(*buf1);
 923         OK_(*buf2);
 924
 925         crypt_free(cd);
 926 }
 927
 928 static void UseLuksDevice(void)
 929 {
 930         struct crypt_device *cd;
 931         char key[128];
 932         size_t key_size;
 933
 934         OK_(crypt_init(&cd, DEVICE_1));
 935         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
 936         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_INACTIVE);
 937         OK_(crypt_activate_by_passphrase(cd, NULL, CRYPT_ANY_SLOT, KEY1, strlen(KEY1), 0));
 938         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1), 0));
 939         FAIL_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1), 0), "already open");
 940         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 941         OK_(crypt_deactivate(cd, CDEVICE_1));
 942         FAIL_(crypt_deactivate(cd, CDEVICE_1), "no such device");
 943
 944         key_size = 16;
 945         OK_(strcmp("aes", crypt_get_cipher(cd)));
 946         OK_(strcmp("cbc-essiv:sha256", crypt_get_cipher_mode(cd)));
 947         OK_(strcmp(DEVICE_1_UUID, crypt_get_uuid(cd)));
 948         EQ_((int)key_size, crypt_get_volume_key_size(cd));
 949         EQ_(1032, crypt_get_data_offset(cd));
 950
 951         if (!_fips_mode) {
 952                 EQ_(0, crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key, &key_size, KEY1, strlen(KEY1)));
 953                 OK_(crypt_volume_key_verify(cd, key, key_size));
 954                 OK_(crypt_activate_by_volume_key(cd, NULL, key, key_size, 0));
 955                 OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
 956                 EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
 957                 OK_(crypt_deactivate(cd, CDEVICE_1));
 958
 959                 key[1] = ~key[1];
 960                 FAIL_(crypt_volume_key_verify(cd, key, key_size), "key mismatch");
 961                 FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0), "key mismatch");
 962         }
 963         crypt_free(cd);
 964 }
 965
 966 static void SuspendDevice(void)
 967 {
 968         int suspend_status;
 969         struct crypt_device *cd;
 970
 971         OK_(crypt_init(&cd, DEVICE_1));
 972         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
 973         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1), 0));
 974
 975         suspend_status = crypt_suspend(cd, CDEVICE_1);
 976         if (suspend_status == -ENOTSUP) {
 977                 printf("WARNING: Suspend/Resume not supported, skipping test.\n");
 978                 OK_(crypt_deactivate(cd, CDEVICE_1));
 979                 crypt_free(cd);
 980                 return;
 981         }
 982
 983         OK_(suspend_status);
 984         FAIL_(crypt_suspend(cd, CDEVICE_1), "already suspended");
 985
 986         FAIL_(crypt_resume_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1)-1), "wrong key");
 987         OK_(crypt_resume_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1)));
 988         FAIL_(crypt_resume_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1)), "not suspended");
 989
 990         OK_(_prepare_keyfile(KEYFILE1, KEY1, strlen(KEY1)));
 991         OK_(crypt_suspend(cd, CDEVICE_1));
 992         FAIL_(crypt_resume_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1 "blah", 0), "wrong keyfile");
 993         FAIL_(crypt_resume_by_keyfile_offset(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 1, 0), "wrong key");
 994         OK_(crypt_resume_by_keyfile_offset(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 0, 0));
 995         FAIL_(crypt_resume_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 0), "not suspended");
 996         OK_(crypt_deactivate(cd, CDEVICE_1));
 997         crypt_free(cd);
 998
 999         /* create LUKS device with detached header */
1000         OK_(crypt_init(&cd, DEVICE_1));
1001         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
1002         OK_(crypt_set_data_device(cd, DEVICE_2));
1003         OK_(crypt_activate_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1), 0));
1004         crypt_free(cd);
1005
1006         /* Should be able to suspend but not resume if not header specified */
1007         OK_(crypt_init_by_name(&cd, CDEVICE_1));
1008         OK_(crypt_suspend(cd, CDEVICE_1));
1009         FAIL_(crypt_suspend(cd, CDEVICE_1), "already suspended");
1010         FAIL_(crypt_resume_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1)-1), "no header");
1011         crypt_free(cd);
1012
1013         OK_(crypt_init_by_name_and_header(&cd, CDEVICE_1, DEVICE_1));
1014         OK_(crypt_resume_by_passphrase(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEY1, strlen(KEY1)));
1015
1016         OK_(crypt_deactivate(cd, CDEVICE_1));
1017         crypt_free(cd);
1018
1019         _remove_keyfiles();
1020 }
1021
1022 static void AddDeviceLuks(void)
1023 {
1024         struct crypt_device *cd;
1025         struct crypt_params_luks1 params = {
1026                 .hash = "sha512",
1027                 .data_alignment = 2048, // 4M, data offset will be 4096
1028                 .data_device = DEVICE_2
1029         };
1030         char key[128], key2[128];
1031
1032         const char *passphrase = "blabla", *passphrase2 = "nsdkFI&Y#.sd";
1033         const char *mk_hex = "bb21158c733229347bd4e681891e213d94c685be6a5b84818afe7a78a6de7a1a";
1034         size_t key_size = strlen(mk_hex) / 2;
1035         const char *cipher = "aes";
1036         const char *cipher_mode = "cbc-essiv:sha256";
1037         uint64_t r_payload_offset, r_header_size, r_size_1;
1038
1039         crypt_decode_key(key, mk_hex, key_size);
1040
1041         // init test devices
1042         OK_(get_luks_offsets(1, key_size, 0, 0, &r_header_size, &r_payload_offset));
1043         OK_(create_dmdevice_over_loop(H_DEVICE, r_header_size));
1044         OK_(create_dmdevice_over_loop(H_DEVICE_WRONG, r_header_size - 1));
1045
1046         // format
1047         OK_(crypt_init(&cd, DMDIR H_DEVICE_WRONG));
1048         params.data_alignment = 0;
1049         FAIL_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params), "Not enough space for keyslots material");
1050         crypt_free(cd);
1051
1052         // test payload_offset = 0 for encrypted device with external header device
1053         OK_(crypt_init(&cd, DMDIR H_DEVICE));
1054         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1055         EQ_(crypt_get_data_offset(cd), 0);
1056         crypt_free(cd);
1057
1058         params.data_alignment = 0;
1059         params.data_device = NULL;
1060
1061         // test payload_offset = 0. format() should look up alignment offset from device topology
1062         OK_(crypt_init(&cd, DEVICE_2));
1063         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1064         OK_(!(crypt_get_data_offset(cd) > 0));
1065         crypt_free(cd);
1066
1067         /*
1068          * test limit values for backing device size
1069          */
1070         params.data_alignment = 4096;
1071         OK_(get_luks_offsets(0, key_size, params.data_alignment, 0, NULL, &r_payload_offset));
1072         OK_(create_dmdevice_over_loop(L_DEVICE_0S, r_payload_offset));
1073         OK_(create_dmdevice_over_loop(L_DEVICE_1S, r_payload_offset + 1));
1074         //OK_(create_dmdevice_over_loop(L_DEVICE_WRONG, r_payload_offset - 1));
1075         OK_(create_dmdevice_over_loop(L_DEVICE_WRONG, 2050 - 1)); //FIXME last keyslot - 1 sector
1076
1077         // 1 sector less than required
1078         OK_(crypt_init(&cd, DMDIR L_DEVICE_WRONG));
1079         FAIL_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params), "Device too small");
1080         crypt_free(cd);
1081
1082         // 0 sectors for encrypted area
1083         OK_(crypt_init(&cd, DMDIR L_DEVICE_0S));
1084         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1085         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0), "Encrypted area too small");
1086         crypt_free(cd);
1087
1088         // 1 sector for encrypted area
1089         OK_(crypt_init(&cd, DMDIR L_DEVICE_1S));
1090         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1091         EQ_(crypt_get_data_offset(cd), params.data_alignment);
1092         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
1093         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1094         OK_(device_size(DMDIR CDEVICE_1, &r_size_1));
1095         EQ_(r_size_1, SECTOR_SIZE);
1096         OK_(crypt_deactivate(cd, CDEVICE_1));
1097         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_INACTIVE);
1098         // restrict format only to empty context
1099         FAIL_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params), "Context is already formated");
1100         FAIL_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, NULL), "Context is already formated");
1101         // change data device to wrong one
1102         OK_(crypt_set_data_device(cd, DMDIR L_DEVICE_0S));
1103         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0), "Device too small");
1104         OK_(crypt_set_data_device(cd, DMDIR L_DEVICE_1S));
1105         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
1106         OK_(crypt_deactivate(cd, CDEVICE_1));
1107         crypt_free(cd);
1108
1109         params.data_alignment = 0;
1110         params.data_device = DEVICE_2;
1111
1112         // generate keyslot material at the end of luks header
1113         OK_(crypt_init(&cd, DMDIR H_DEVICE));
1114         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1115         EQ_(crypt_keyslot_add_by_volume_key(cd, 7, key, key_size, passphrase, strlen(passphrase)), 7);
1116         EQ_(crypt_activate_by_passphrase(cd, CDEVICE_1, 7, passphrase, strlen(passphrase) ,0), 7);
1117         crypt_free(cd);
1118         OK_(crypt_init_by_name_and_header(&cd, CDEVICE_1, DMDIR H_DEVICE));
1119         FAIL_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params), "Context is already formated");
1120         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1121         crypt_free(cd);
1122         // check active status without header
1123         OK_(crypt_init_by_name_and_header(&cd, CDEVICE_1, NULL));
1124         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1125         OK_(!!crypt_get_type(cd));
1126         OK_(strcmp(cipher, crypt_get_cipher(cd)));
1127         OK_(strcmp(cipher_mode, crypt_get_cipher_mode(cd)));
1128         EQ_((int)key_size, crypt_get_volume_key_size(cd));
1129         OK_(crypt_deactivate(cd, CDEVICE_1));
1130         crypt_free(cd);
1131
1132         params.data_alignment = 2048;
1133         params.data_device = NULL;
1134
1135         // test uuid mismatch and _init_by_name_and_header
1136         OK_(crypt_init(&cd, DMDIR L_DEVICE_1S));
1137         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1138         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
1139         crypt_free(cd);
1140         params.data_alignment = 0;
1141         params.data_device = DEVICE_2;
1142         OK_(crypt_init(&cd, DMDIR H_DEVICE));
1143         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1144         crypt_free(cd);
1145         // there we've got uuid mismatch
1146         OK_(crypt_init_by_name_and_header(&cd, CDEVICE_1, DMDIR H_DEVICE));
1147         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1148         OK_(!!crypt_get_type(cd));
1149         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0), "Device is active");
1150         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_2, key, key_size, 0), "Device is active");
1151         EQ_(crypt_status(cd, CDEVICE_2), CRYPT_INACTIVE);
1152         OK_(crypt_deactivate(cd, CDEVICE_1));
1153         crypt_free(cd);
1154
1155         params.data_device = NULL;
1156
1157         OK_(crypt_init(&cd, DEVICE_2));
1158         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1159
1160         // even with no keyslots defined it can be activated by volume key
1161         OK_(crypt_volume_key_verify(cd, key, key_size));
1162         OK_(crypt_activate_by_volume_key(cd, CDEVICE_2, key, key_size, 0));
1163         EQ_(crypt_status(cd, CDEVICE_2), CRYPT_ACTIVE);
1164         OK_(crypt_deactivate(cd, CDEVICE_2));
1165
1166         // now with keyslot
1167         EQ_(7, crypt_keyslot_add_by_volume_key(cd, 7, key, key_size, passphrase, strlen(passphrase)));
1168         EQ_(CRYPT_SLOT_ACTIVE_LAST, crypt_keyslot_status(cd, 7));
1169         EQ_(7, crypt_activate_by_passphrase(cd, CDEVICE_2, CRYPT_ANY_SLOT, passphrase, strlen(passphrase), 0));
1170         EQ_(crypt_status(cd, CDEVICE_2), CRYPT_ACTIVE);
1171         OK_(crypt_deactivate(cd, CDEVICE_2));
1172
1173         crypt_set_iteration_time(cd, 1);
1174         EQ_(1, crypt_keyslot_add_by_volume_key(cd, 1, key, key_size, KEY1, strlen(KEY1)));
1175         OK_(_prepare_keyfile(KEYFILE1, KEY1, strlen(KEY1)));
1176         OK_(_prepare_keyfile(KEYFILE2, KEY2, strlen(KEY2)));
1177         EQ_(2, crypt_keyslot_add_by_keyfile(cd, 2, KEYFILE1, 0, KEYFILE2, 0));
1178         FAIL_(crypt_keyslot_add_by_keyfile_offset(cd, 3, KEYFILE1, 0, 1, KEYFILE2, 0, 1), "wrong key");
1179         EQ_(3, crypt_keyslot_add_by_keyfile_offset(cd, 3, KEYFILE1, 0, 0, KEYFILE2, 0, 1));
1180         EQ_(4, crypt_keyslot_add_by_keyfile_offset(cd, 4, KEYFILE2, 0, 1, KEYFILE1, 0, 1));
1181         FAIL_(crypt_activate_by_keyfile(cd, CDEVICE_2, CRYPT_ANY_SLOT, KEYFILE2, strlen(KEY2)-1, 0), "key mismatch");
1182         EQ_(2, crypt_activate_by_keyfile(cd, NULL, CRYPT_ANY_SLOT, KEYFILE2, 0, 0));
1183         EQ_(3, crypt_activate_by_keyfile_offset(cd, NULL, CRYPT_ANY_SLOT, KEYFILE2, 0, 1, 0));
1184         EQ_(4, crypt_activate_by_keyfile_offset(cd, NULL, CRYPT_ANY_SLOT, KEYFILE1, 0, 1, 0));
1185         FAIL_(crypt_activate_by_keyfile_offset(cd, CDEVICE_2, CRYPT_ANY_SLOT, KEYFILE2, strlen(KEY2), 2, 0), "not enough data");
1186         FAIL_(crypt_activate_by_keyfile_offset(cd, CDEVICE_2, CRYPT_ANY_SLOT, KEYFILE2, 0, strlen(KEY2) + 1, 0), "cannot seek");
1187         FAIL_(crypt_activate_by_keyfile_offset(cd, CDEVICE_2, CRYPT_ANY_SLOT, KEYFILE2, 0, 2, 0), "wrong key");
1188         EQ_(2, crypt_activate_by_keyfile(cd, CDEVICE_2, CRYPT_ANY_SLOT, KEYFILE2, 0, 0));
1189         OK_(crypt_keyslot_destroy(cd, 1));
1190         OK_(crypt_keyslot_destroy(cd, 2));
1191         OK_(crypt_keyslot_destroy(cd, 3));
1192         OK_(crypt_keyslot_destroy(cd, 4));
1193         OK_(crypt_deactivate(cd, CDEVICE_2));
1194         _remove_keyfiles();
1195
1196         FAIL_(crypt_keyslot_add_by_volume_key(cd, 7, key, key_size, passphrase, strlen(passphrase)), "slot used");
1197         key[1] = ~key[1];
1198         FAIL_(crypt_keyslot_add_by_volume_key(cd, 6, key, key_size, passphrase, strlen(passphrase)), "key mismatch");
1199         key[1] = ~key[1];
1200         EQ_(6, crypt_keyslot_add_by_volume_key(cd, 6, key, key_size, passphrase, strlen(passphrase)));
1201         EQ_(CRYPT_SLOT_ACTIVE, crypt_keyslot_status(cd, 6));
1202
1203         FAIL_(crypt_keyslot_destroy(cd, 8), "invalid keyslot");
1204         FAIL_(crypt_keyslot_destroy(cd, CRYPT_ANY_SLOT), "invalid keyslot");
1205         FAIL_(crypt_keyslot_destroy(cd, 0), "keyslot not used");
1206         OK_(crypt_keyslot_destroy(cd, 7));
1207         EQ_(CRYPT_SLOT_INACTIVE, crypt_keyslot_status(cd, 7));
1208         EQ_(CRYPT_SLOT_ACTIVE_LAST, crypt_keyslot_status(cd, 6));
1209
1210         EQ_(7, crypt_keyslot_change_by_passphrase(cd, 6, 7, passphrase, strlen(passphrase), passphrase2, strlen(passphrase2)));
1211         EQ_(CRYPT_SLOT_ACTIVE_LAST, crypt_keyslot_status(cd, 7));
1212         EQ_(7, crypt_activate_by_passphrase(cd, NULL, 7, passphrase2, strlen(passphrase2), 0));
1213         EQ_(6, crypt_keyslot_change_by_passphrase(cd, CRYPT_ANY_SLOT, 6, passphrase2, strlen(passphrase2), passphrase, strlen(passphrase)));
1214
1215         if (!_fips_mode) {
1216                 EQ_(6, crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key2, &key_size, passphrase, strlen(passphrase)));
1217                 OK_(crypt_volume_key_verify(cd, key2, key_size));
1218
1219                 OK_(memcmp(key, key2, key_size));
1220         }
1221         OK_(strcmp(cipher, crypt_get_cipher(cd)));
1222         OK_(strcmp(cipher_mode, crypt_get_cipher_mode(cd)));
1223         EQ_((int)key_size, crypt_get_volume_key_size(cd));
1224         EQ_(4096, crypt_get_data_offset(cd));
1225         OK_(strcmp(DEVICE_2, crypt_get_device_name(cd)));
1226
1227         reset_log();
1228         crypt_set_log_callback(cd, &new_log, NULL);
1229         OK_(crypt_dump(cd));
1230         OK_(!(global_lines != 0));
1231         crypt_set_log_callback(cd, NULL, NULL);
1232         reset_log();
1233
1234         FAIL_(crypt_set_uuid(cd, "blah"), "wrong UUID format");
1235         OK_(crypt_set_uuid(cd, DEVICE_TEST_UUID));
1236         OK_(strcmp(DEVICE_TEST_UUID, crypt_get_uuid(cd)));
1237
1238         FAIL_(crypt_deactivate(cd, CDEVICE_2), "not active");
1239         crypt_free(cd);
1240         _cleanup_dmdevices();
1241 }
1242
1243 static void UseTempVolumes(void)
1244 {
1245         struct crypt_device *cd;
1246         char tmp[256];
1247
1248         // Tepmporary device without keyslot but with on-disk LUKS header
1249         OK_(crypt_init(&cd, DEVICE_2));
1250         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_2, NULL, 0, 0), "not yet formatted");
1251         OK_(crypt_format(cd, CRYPT_LUKS1, "aes", "cbc-essiv:sha256", NULL, NULL, 16, NULL));
1252         OK_(crypt_activate_by_volume_key(cd, CDEVICE_2, NULL, 0, 0));
1253         EQ_(crypt_status(cd, CDEVICE_2), CRYPT_ACTIVE);
1254         crypt_free(cd);
1255
1256         OK_(crypt_init_by_name(&cd, CDEVICE_2));
1257         OK_(crypt_deactivate(cd, CDEVICE_2));
1258         crypt_free(cd);
1259
1260         // Dirty checks: device without UUID
1261         // we should be able to remove it but not manuipulate with it
1262         snprintf(tmp, sizeof(tmp), "dmsetup create %s --table \""
1263                 "0 100 crypt aes-cbc-essiv:sha256 deadbabedeadbabedeadbabedeadbabe 0 "
1264                 "%s 2048\"", CDEVICE_2, DEVICE_2);
1265         _system(tmp, 1);
1266         OK_(crypt_init_by_name(&cd, CDEVICE_2));
1267         OK_(crypt_deactivate(cd, CDEVICE_2));
1268         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_2, NULL, 0, 0), "No known device type");
1269         crypt_free(cd);
1270
1271         // Dirty checks: device with UUID but LUKS header key fingerprint must fail)
1272         snprintf(tmp, sizeof(tmp), "dmsetup create %s --table \""
1273                 "0 100 crypt aes-cbc-essiv:sha256 deadbabedeadbabedeadbabedeadbabe 0 "
1274                 "%s 2048\" -u CRYPT-LUKS1-aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa-ctest1",
1275                  CDEVICE_2, DEVICE_2);
1276         _system(tmp, 1);
1277         OK_(crypt_init_by_name(&cd, CDEVICE_2));
1278         OK_(crypt_deactivate(cd, CDEVICE_2));
1279         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_2, NULL, 0, 0), "wrong volume key");
1280         crypt_free(cd);
1281
1282         // No slots
1283         OK_(crypt_init(&cd, DEVICE_2));
1284         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
1285         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_2, NULL, 0, 0), "volume key is lost");
1286         crypt_free(cd);
1287
1288         // Plain device
1289         OK_(crypt_init(&cd, DEVICE_2));
1290         OK_(crypt_format(cd, CRYPT_PLAIN, "aes", "cbc-essiv:sha256", NULL, NULL, 16, NULL));
1291         FAIL_(crypt_activate_by_volume_key(cd, NULL, "xxx", 3, 0), "cannot verify key with plain");
1292         FAIL_(crypt_volume_key_verify(cd, "xxx", 3), "cannot verify key with plain");
1293         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_2, "xxx", 3, 0), "wrong key lenght");
1294         OK_(crypt_activate_by_volume_key(cd, CDEVICE_2, "volumekeyvolumek", 16, 0));
1295         EQ_(crypt_status(cd, CDEVICE_2), CRYPT_ACTIVE);
1296         OK_(crypt_deactivate(cd, CDEVICE_2));
1297         crypt_free(cd);
1298 }
1299
1300 static void LuksHeaderRestore(void)
1301 {
1302         struct crypt_device *cd;
1303         struct crypt_params_luks1 params = {
1304                 .hash = "sha512",
1305                 .data_alignment = 2048, // 4M, data offset will be 4096
1306         };
1307         struct crypt_params_plain pl_params = {
1308                 .hash = "sha1",
1309                 .skip = 0,
1310                 .offset = 0,
1311                 .size = 0
1312         };
1313         char key[128], key2[128], cmd[256];
1314
1315         const char *mk_hex = "bb21158c733229347bd4e681891e213d94c685be6a5b84818afe7a78a6de7a1a";
1316         size_t key_size = strlen(mk_hex) / 2;
1317         const char *cipher = "aes";
1318         const char *cipher_mode = "cbc-essiv:sha256";
1319         uint64_t r_payload_offset;
1320
1321         crypt_decode_key(key, mk_hex, key_size);
1322
1323         OK_(get_luks_offsets(0, key_size, params.data_alignment, 0, NULL, &r_payload_offset));
1324         OK_(create_dmdevice_over_loop(L_DEVICE_OK, r_payload_offset + 5000));
1325
1326         // do not restore header over plain device
1327         OK_(crypt_init(&cd, DMDIR L_DEVICE_OK));
1328         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, NULL, key_size, &pl_params));
1329         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
1330         FAIL_(crypt_header_restore(cd, CRYPT_PLAIN, VALID_HEADER), "Cannot restore header to PLAIN type device");
1331         FAIL_(crypt_header_restore(cd, CRYPT_LUKS1, VALID_HEADER), "Cannot restore header over PLAIN type device");
1332         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1333         OK_(crypt_deactivate(cd, CDEVICE_1));
1334         crypt_free(cd);
1335
1336         // invalid headers
1337         OK_(crypt_init(&cd, DMDIR L_DEVICE_OK));
1338         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1339         FAIL_(crypt_header_restore(cd, CRYPT_LUKS1, EVL_HEADER_1), "Header corrupted");
1340         FAIL_(crypt_header_restore(cd, CRYPT_LUKS1, EVL_HEADER_2), "Header corrupted");
1341         FAIL_(crypt_header_restore(cd, CRYPT_LUKS1, EVL_HEADER_3), "Header corrupted");
1342         FAIL_(crypt_header_restore(cd, CRYPT_LUKS1, EVL_HEADER_4), "Header too small");
1343         OK_(crypt_header_restore(cd, CRYPT_LUKS1, VALID_HEADER));
1344         // wipe valid luks header
1345         snprintf(cmd, sizeof(cmd), "dd if=/dev/zero of=" DMDIR L_DEVICE_OK " bs=512 count=%" PRIu64 " 2>/dev/null", r_payload_offset);
1346         OK_(_system(cmd, 1));
1347         FAIL_(crypt_header_restore(cd, CRYPT_LUKS1, EVL_HEADER_1), "Header corrupted");
1348         FAIL_(crypt_header_restore(cd, CRYPT_LUKS1, EVL_HEADER_2), "Header corrupted");
1349         FAIL_(crypt_header_restore(cd, CRYPT_LUKS1, EVL_HEADER_3), "Header corrupted");
1350         FAIL_(crypt_header_restore(cd, CRYPT_LUKS1, EVL_HEADER_4), "Header too small");
1351         OK_(crypt_header_restore(cd, CRYPT_LUKS1, VALID_HEADER));
1352         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
1353         OK_(crypt_deactivate(cd, CDEVICE_1));
1354         crypt_free(cd);
1355
1356         // volume key_size mismatch
1357         OK_(crypt_init(&cd, DMDIR L_DEVICE_OK));
1358         memcpy(key2, key, key_size / 2);
1359         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key2, key_size / 2, &params));
1360         FAIL_(crypt_header_restore(cd, CRYPT_LUKS1, VALID_HEADER), "Volume keysize mismatch");
1361         crypt_free(cd);
1362
1363         // payload offset mismatch
1364         params.data_alignment = 8192;
1365         OK_(crypt_init(&cd, DMDIR L_DEVICE_OK));
1366         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1367         FAIL_(crypt_header_restore(cd, CRYPT_LUKS1, VALID_HEADER), "Payload offset mismatch");
1368         //_system("dmsetup table;sleep 1",1);
1369         crypt_free(cd);
1370
1371         _cleanup_dmdevices();
1372 }
1373
1374 static void LuksHeaderLoad(void)
1375 {
1376         struct crypt_device *cd;
1377         struct crypt_params_luks1 params = {
1378                 .hash = "sha512",
1379                 .data_alignment = 2048,
1380         };
1381         struct crypt_params_plain pl_params = {
1382                 .hash = "sha1",
1383                 .skip = 0,
1384                 .offset = 0,
1385                 .size = 0
1386         };
1387         char key[128], cmd[256];
1388
1389         const char *mk_hex = "bb21158c733229347bd4e681891e213d94c685be6a5b84818afe7a78a6de7a1a";
1390         size_t key_size = strlen(mk_hex) / 2;
1391         const char *cipher = "aes";
1392         const char *cipher_mode = "cbc-essiv:sha256";
1393         uint64_t r_payload_offset, r_header_size;
1394
1395         crypt_decode_key(key, mk_hex, key_size);
1396
1397         // prepare test env
1398         OK_(get_luks_offsets(0, key_size, params.data_alignment, 0, &r_header_size, &r_payload_offset));
1399         // external header device
1400         OK_(create_dmdevice_over_loop(H_DEVICE, r_header_size));
1401         // prepared header on a device too small to contain header and payload
1402         //OK_(create_dmdevice_over_loop(H_DEVICE_WRONG, r_payload_offset - 1));
1403         OK_(create_dmdevice_over_loop(H_DEVICE_WRONG, 2050 - 1)); //FIXME
1404         //snprintf(cmd, sizeof(cmd), "dd if=" EVL_HEADER_4 " of=" DMDIR H_DEVICE_WRONG " bs=512 count=%" PRIu64, r_payload_offset - 1);
1405         snprintf(cmd, sizeof(cmd), "dd if=" EVL_HEADER_4 " of=" DMDIR H_DEVICE_WRONG " bs=512 count=%d 2>/dev/null", 2050 - 1);
1406         OK_(_system(cmd, 1));
1407         // some device
1408         OK_(create_dmdevice_over_loop(L_DEVICE_OK, r_payload_offset + 1000));
1409         // 1 sector device
1410         OK_(create_dmdevice_over_loop(L_DEVICE_1S, r_payload_offset + 1));
1411         // 0 sectors device for payload
1412         OK_(create_dmdevice_over_loop(L_DEVICE_0S, r_payload_offset));
1413
1414         // valid metadata and device size
1415         params.data_alignment = 0;
1416         params.data_device = DMDIR L_DEVICE_OK;
1417         OK_(crypt_init(&cd, DMDIR H_DEVICE));
1418         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1419         crypt_free(cd);
1420         OK_(crypt_init(&cd, DMDIR H_DEVICE));
1421         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
1422         OK_(crypt_set_data_device(cd, DMDIR L_DEVICE_OK));
1423         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
1424         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1425         OK_(crypt_deactivate(cd, CDEVICE_1));
1426         crypt_free(cd);
1427
1428         // bad header: device too small (payloadOffset > device_size)
1429         OK_(crypt_init(&cd, DMDIR H_DEVICE_WRONG));
1430         FAIL_(crypt_load(cd, CRYPT_LUKS1, NULL), "Device too small");
1431         OK_(!!crypt_get_type(cd));
1432         crypt_free(cd);
1433
1434         // 0 secs for encrypted data area
1435         params.data_alignment = 2048;
1436         params.data_device = NULL;
1437         OK_(crypt_init(&cd, DMDIR L_DEVICE_0S));
1438         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1439         crypt_free(cd);
1440         // load should be ok
1441         OK_(crypt_init(&cd, DMDIR L_DEVICE_0S));
1442         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
1443         FAIL_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0), "Device too small");
1444         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_INACTIVE);
1445         crypt_free(cd);
1446
1447         // damaged header
1448         OK_(_system("dd if=/dev/zero of=" DMDIR L_DEVICE_OK " bs=512 count=8 2>/dev/null", 1));
1449         OK_(crypt_init(&cd, DMDIR L_DEVICE_OK));
1450         FAIL_(crypt_load(cd, CRYPT_LUKS1, NULL), "Header not found");
1451         crypt_free(cd);
1452
1453         // plain device
1454         OK_(crypt_init(&cd, DMDIR H_DEVICE));
1455         FAIL_(crypt_load(cd, CRYPT_PLAIN, NULL), "Can't load nonLUKS device type");
1456         crypt_free(cd);
1457         OK_(crypt_init(&cd, DMDIR H_DEVICE));
1458         OK_(crypt_format(cd, CRYPT_PLAIN, cipher, cipher_mode, NULL, key, key_size, &pl_params));
1459         FAIL_(crypt_load(cd, CRYPT_LUKS1, NULL), "Can't load over nonLUKS device type");
1460         crypt_free(cd);
1461
1462         _cleanup_dmdevices();
1463 }
1464
1465 static void LuksHeaderBackup(void)
1466 {
1467         struct crypt_device *cd;
1468         struct crypt_params_luks1 params = {
1469                 .hash = "sha512",
1470                 .data_alignment = 2048,
1471         };
1472         char key[128];
1473         int fd, ro = O_RDONLY;
1474
1475         const char *mk_hex = "bb21158c733229347bd4e681891e213d94c685be6a5b84818afe7a78a6de7a1a";
1476         size_t key_size = strlen(mk_hex) / 2;
1477         const char *cipher = "aes";
1478         const char *cipher_mode = "cbc-essiv:sha256";
1479         uint64_t r_payload_offset;
1480
1481         const char *passphrase = PASSPHRASE;
1482
1483         crypt_decode_key(key, mk_hex, key_size);
1484
1485         OK_(get_luks_offsets(0, key_size, params.data_alignment, 0, NULL, &r_payload_offset));
1486         OK_(create_dmdevice_over_loop(L_DEVICE_OK, r_payload_offset + 1));
1487
1488         // create LUKS device and backup the header
1489         OK_(crypt_init(&cd, DMDIR L_DEVICE_OK));
1490         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1491         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
1492         EQ_(crypt_keyslot_add_by_volume_key(cd, 7, key, key_size, passphrase, strlen(passphrase)), 7);
1493         EQ_(crypt_keyslot_add_by_volume_key(cd, 0, key, key_size, passphrase, strlen(passphrase)), 0);
1494         OK_(crypt_header_backup(cd, CRYPT_LUKS1, BACKUP_FILE));
1495         OK_(crypt_deactivate(cd, CDEVICE_1));
1496         crypt_free(cd);
1497
1498         // restore header from backup
1499         OK_(crypt_init(&cd, DMDIR L_DEVICE_OK));
1500         OK_(crypt_header_restore(cd, CRYPT_LUKS1, BACKUP_FILE));
1501         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
1502         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
1503         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1504         OK_(crypt_deactivate(cd, CDEVICE_1));
1505         crypt_free(cd);
1506
1507         // exercise luksOpen using backup header in file
1508         OK_(crypt_init(&cd, BACKUP_FILE));
1509         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
1510         OK_(crypt_set_data_device(cd, DMDIR L_DEVICE_OK));
1511         EQ_(crypt_activate_by_passphrase(cd, CDEVICE_1, 0, passphrase, strlen(passphrase), 0), 0);
1512         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1513         OK_(crypt_deactivate(cd, CDEVICE_1));
1514         crypt_free(cd);
1515
1516         OK_(crypt_init(&cd, BACKUP_FILE));
1517         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
1518         OK_(crypt_set_data_device(cd, DMDIR L_DEVICE_OK));
1519         EQ_(crypt_activate_by_passphrase(cd, CDEVICE_1, 7, passphrase, strlen(passphrase), 0), 7);
1520         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1521         OK_(crypt_deactivate(cd, CDEVICE_1));
1522         crypt_free(cd);
1523
1524         // exercise luksOpen using backup header on block device
1525         fd = crypt_loop_attach(DEVICE_3, BACKUP_FILE, 0, 0, &ro);
1526         close(fd);
1527         OK_(fd < 0);
1528         OK_(crypt_init(&cd, DEVICE_3));
1529         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
1530         OK_(crypt_set_data_device(cd, DMDIR L_DEVICE_OK));
1531         EQ_(crypt_activate_by_passphrase(cd, CDEVICE_1, 0, passphrase, strlen(passphrase), 0), 0);
1532         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1533         OK_(crypt_deactivate(cd, CDEVICE_1));
1534         crypt_free(cd);
1535
1536         OK_(crypt_init(&cd, DEVICE_3));
1537         OK_(crypt_load(cd, CRYPT_LUKS1, NULL));
1538         OK_(crypt_set_data_device(cd, DMDIR L_DEVICE_OK));
1539         EQ_(crypt_activate_by_passphrase(cd, CDEVICE_1, 7, passphrase, strlen(passphrase), 0), 7);
1540         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1541         OK_(crypt_deactivate(cd, CDEVICE_1));
1542         crypt_free(cd);
1543
1544         _cleanup_dmdevices();
1545 }
1546
1547 static void ResizeDeviceLuks(void)
1548 {
1549         struct crypt_device *cd;
1550         struct crypt_params_luks1 params = {
1551                 .hash = "sha512",
1552                 .data_alignment = 2048,
1553         };
1554         char key[128];
1555
1556         const char *mk_hex = "bb21158c733229347bd4e681891e213d94c685be6a5b84818afe7a78a6de7a1a";
1557         size_t key_size = strlen(mk_hex) / 2;
1558         const char *cipher = "aes";
1559         const char *cipher_mode = "cbc-essiv:sha256";
1560         uint64_t r_payload_offset, r_header_size, r_size;
1561
1562         crypt_decode_key(key, mk_hex, key_size);
1563
1564         // prepare env
1565         OK_(get_luks_offsets(0, key_size, params.data_alignment, 0, NULL, &r_payload_offset));
1566         OK_(get_luks_offsets(1, key_size, 0, 0, &r_header_size, NULL));
1567         OK_(create_dmdevice_over_loop(H_DEVICE, r_header_size));
1568         OK_(create_dmdevice_over_loop(L_DEVICE_OK, r_payload_offset + 1000));
1569         OK_(create_dmdevice_over_loop(L_DEVICE_0S, 1000));
1570
1571         // test header and encrypted payload all in one device
1572         OK_(crypt_init(&cd, DMDIR L_DEVICE_OK));
1573         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1574         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
1575         OK_(crypt_resize(cd, CDEVICE_1, 42));
1576         if (!device_size(DMDIR CDEVICE_1, &r_size))
1577                 EQ_(42, r_size >> SECTOR_SHIFT);
1578         // autodetect encrypted device area size
1579         OK_(crypt_resize(cd, CDEVICE_1, 0));
1580         if (!device_size(DMDIR CDEVICE_1, &r_size))
1581                 EQ_(1000, r_size >> SECTOR_SHIFT);
1582         FAIL_(crypt_resize(cd, CDEVICE_1, 1001), "Device too small");
1583         if (!device_size(DMDIR CDEVICE_1, &r_size))
1584                 EQ_(1000, r_size >> SECTOR_SHIFT);
1585         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1586         OK_(crypt_deactivate(cd, CDEVICE_1));
1587         crypt_free(cd);
1588
1589         params.data_alignment = 0;
1590         params.data_device = DMDIR L_DEVICE_0S;
1591         // test case for external header
1592         OK_(crypt_init(&cd, DMDIR H_DEVICE));
1593         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1594         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, key, key_size, 0));
1595         OK_(crypt_resize(cd, CDEVICE_1, 666));
1596         if (!device_size(DMDIR CDEVICE_1, &r_size))
1597                 EQ_(666, r_size >> SECTOR_SHIFT);
1598         // autodetect encrypted device size
1599         OK_(crypt_resize(cd, CDEVICE_1, 0));
1600         if (!device_size(DMDIR CDEVICE_1, &r_size))
1601                 EQ_(1000, r_size >> SECTOR_SHIFT);
1602         FAIL_(crypt_resize(cd, CDEVICE_1, 1001), "Device too small");
1603         if (!device_size(DMDIR CDEVICE_1, &r_size))
1604                 EQ_(1000, r_size >> SECTOR_SHIFT);
1605         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1606         OK_(crypt_deactivate(cd, CDEVICE_1));
1607         crypt_free(cd);
1608
1609         _cleanup_dmdevices();
1610 }
1611
1612 static void HashDevicePlain(void)
1613 {
1614         struct crypt_device *cd;
1615         struct crypt_params_plain params = {
1616                 .hash = NULL,
1617                 .skip = 0,
1618                 .offset = 0,
1619         };
1620
1621         size_t key_size;
1622         const char *mk_hex, *keystr;
1623         char key[256];
1624
1625         OK_(crypt_init(&cd, DEVICE_1));
1626         OK_(crypt_format(cd, CRYPT_PLAIN, "aes", "cbc-essiv:sha256", NULL, NULL, 16, &params));
1627
1628         // hash PLAIN, short key
1629         OK_(_prepare_keyfile(KEYFILE1, "tooshort", 8));
1630         FAIL_(crypt_activate_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 16, 0), "not enough data in keyfile");
1631         _remove_keyfiles();
1632
1633         // hash PLAIN, exact key
1634         //         0 1 2 3 4 5 6 7 8 9 a b c d e f
1635         mk_hex = "caffeecaffeecaffeecaffeecaffee88";
1636         key_size = 16;
1637         crypt_decode_key(key, mk_hex, key_size);
1638         OK_(_prepare_keyfile(KEYFILE1, key, key_size));
1639         OK_(crypt_activate_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, key_size, 0));
1640         OK_(_get_key_dm(CDEVICE_1, key, sizeof(key)));
1641         OK_(strcmp(key, mk_hex));
1642         OK_(crypt_deactivate(cd, CDEVICE_1));
1643
1644         // Limit plain key
1645         mk_hex = "caffeecaffeecaffeecaffeeca000000";
1646         OK_(crypt_activate_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, key_size - 3, 0));
1647         OK_(_get_key_dm(CDEVICE_1, key, sizeof(key)));
1648         OK_(strcmp(key, mk_hex));
1649         OK_(crypt_deactivate(cd, CDEVICE_1));
1650
1651         _remove_keyfiles();
1652
1653         // hash PLAIN, long key
1654         //         0 1 2 3 4 5 6 7 8 9 a b c d e f
1655         mk_hex = "caffeecaffeecaffeecaffeecaffee88babebabe";
1656         key_size = 16;
1657         crypt_decode_key(key, mk_hex, key_size);
1658         OK_(_prepare_keyfile(KEYFILE1, key, strlen(mk_hex) / 2));
1659         OK_(crypt_activate_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, key_size, 0));
1660         OK_(_get_key_dm(CDEVICE_1, key, sizeof(key)));
1661         FAIL_(strcmp(key, mk_hex), "only key length used");
1662         OK_(strncmp(key, mk_hex, key_size));
1663         OK_(crypt_deactivate(cd, CDEVICE_1));
1664
1665
1666         // Now without explicit limit
1667         OK_(crypt_activate_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 0, 0));
1668         OK_(_get_key_dm(CDEVICE_1, key, sizeof(key)));
1669         FAIL_(strcmp(key, mk_hex), "only key length used");
1670         OK_(strncmp(key, mk_hex, key_size));
1671         OK_(crypt_deactivate(cd, CDEVICE_1));
1672         crypt_free(cd);
1673
1674         _remove_keyfiles();
1675
1676         // hash sha256
1677         params.hash = "sha256";
1678         OK_(crypt_init(&cd, DEVICE_1));
1679         OK_(crypt_format(cd, CRYPT_PLAIN, "aes", "cbc-essiv:sha256", NULL, NULL, 16, &params));
1680
1681         //         0 1 2 3 4 5 6 7 8 9 a b c d e f
1682         mk_hex = "c62e4615bd39e222572f3a1bf7c2132e";
1683         keystr = "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx";
1684         key_size = strlen(keystr); // 32
1685         OK_(_prepare_keyfile(KEYFILE1, keystr, strlen(keystr)));
1686         OK_(crypt_activate_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, key_size, 0));
1687         OK_(_get_key_dm(CDEVICE_1, key, sizeof(key)));
1688         OK_(strcmp(key, mk_hex));
1689         OK_(crypt_deactivate(cd, CDEVICE_1));
1690
1691         // Read full keyfile
1692         OK_(crypt_activate_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 0, 0));
1693         OK_(_get_key_dm(CDEVICE_1, key, sizeof(key)));
1694         OK_(strcmp(key, mk_hex));
1695         OK_(crypt_deactivate(cd, CDEVICE_1));
1696
1697         _remove_keyfiles();
1698
1699         // Limit keyfile read
1700         keystr = "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxAAAAAAAA";
1701         OK_(_prepare_keyfile(KEYFILE1, keystr, strlen(keystr)));
1702         OK_(crypt_activate_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, key_size, 0));
1703         OK_(_get_key_dm(CDEVICE_1, key, sizeof(key)));
1704         OK_(strcmp(key, mk_hex));
1705         OK_(crypt_deactivate(cd, CDEVICE_1));
1706
1707         // Full keyfile
1708         OK_(crypt_activate_by_keyfile(cd, CDEVICE_1, CRYPT_ANY_SLOT, KEYFILE1, 0, 0));
1709         OK_(_get_key_dm(CDEVICE_1, key, sizeof(key)));
1710         OK_(strcmp(key, "0e49cb34a1dee1df33f6505e4de44a66"));
1711         OK_(crypt_deactivate(cd, CDEVICE_1));
1712
1713         _remove_keyfiles();
1714
1715         // FIXME: add keyfile="-" tests somehow
1716
1717         crypt_free(cd);
1718 }
1719
1720 static void VerityTest(void)
1721 {
1722         struct crypt_device *cd;
1723         const char *salt_hex =  "20c28ffc129c12360ba6ceea2b6cf04e89c2b41cfe6b8439eb53c1897f50df7b";
1724         const char *root_hex =  "ab018b003a967fc782effb293b6dccb60b4f40c06bf80d16391acf686d28b5d6";
1725         char salt[256], root_hash[256];
1726         struct crypt_active_device cad;
1727         struct crypt_params_verity params = {
1728                 .data_device = DEVICE_EMPTY,
1729                 .salt = salt,
1730                 .data_size = 0, /* whole device */
1731                 .hash_area_offset = 0,
1732                 .flags = CRYPT_VERITY_CREATE_HASH,
1733         };
1734
1735         crypt_decode_key(salt, salt_hex, strlen(salt_hex) / 2);
1736         crypt_decode_key(root_hash, root_hex, strlen(root_hex) / 2);
1737
1738         /* Format */
1739         OK_(crypt_init(&cd, DEVICE_2));
1740
1741         /* block size */
1742         params.data_block_size = 333;
1743         FAIL_(crypt_format(cd, CRYPT_VERITY, NULL, NULL, NULL, NULL, 0, &params),
1744                 "Unsupppored block size.");
1745         params.data_block_size = 4096;
1746         params.hash_block_size = 333;
1747         FAIL_(crypt_format(cd, CRYPT_VERITY, NULL, NULL, NULL, NULL, 0, &params),
1748                 "Unsupppored block size.");
1749         params.hash_block_size = 4096;
1750
1751         /* salt size */
1752         params.salt_size = 257;
1753         FAIL_(crypt_format(cd, CRYPT_VERITY, NULL, NULL, NULL, NULL, 0, &params),
1754                 "Too large salt.");
1755         params.salt_size = 32;
1756
1757         /* hash_type */
1758         params.hash_type = 3;
1759         FAIL_(crypt_format(cd, CRYPT_VERITY, NULL, NULL, NULL, NULL, 0, &params),
1760                 "Unsupported hash type.");
1761         params.hash_type = 1;
1762         params.hash_name = "blah";
1763         FAIL_(crypt_format(cd, CRYPT_VERITY, NULL, NULL, NULL, NULL, 0, &params),
1764                 "Unsupported hash name.");
1765         params.hash_name = "sha256";
1766
1767         OK_(crypt_format(cd, CRYPT_VERITY, NULL, NULL, NULL, NULL, 0, &params));
1768         crypt_free(cd);
1769
1770         /* Verify */
1771         OK_(crypt_init(&cd, DEVICE_2));
1772         memset(&params, 0, sizeof(params));
1773         params.data_device = DEVICE_EMPTY;
1774         params.flags = CRYPT_VERITY_CHECK_HASH;
1775         OK_(crypt_load(cd, CRYPT_VERITY, &params));
1776
1777         /* check verity params */
1778         EQ_(crypt_get_volume_key_size(cd), 32);
1779         OK_(strcmp(CRYPT_VERITY, crypt_get_type(cd)));
1780         memset(&params, 0, sizeof(params));
1781         OK_(crypt_get_verity_info(cd, &params));
1782         OK_(strcmp("sha256", params.hash_name));
1783         EQ_(strlen(salt_hex) / 2, params.salt_size);
1784         OK_(memcmp(salt, params.salt, params.salt_size));
1785         EQ_(4096, params.data_block_size);
1786         EQ_(4096, params.hash_block_size);
1787         EQ_(1, params.hash_type);
1788         EQ_(crypt_get_volume_key_size(cd), 32);
1789
1790         OK_(crypt_activate_by_volume_key(cd, NULL, root_hash, 32, 0));
1791         OK_(crypt_set_data_device(cd, DEVICE_1));
1792         FAIL_(crypt_activate_by_volume_key(cd, NULL, root_hash, 32, 0), "Data corrupted");;
1793
1794         OK_(crypt_set_data_device(cd, DEVICE_EMPTY));
1795         if (crypt_activate_by_volume_key(cd, CDEVICE_1, root_hash, 32,
1796             CRYPT_ACTIVATE_READONLY) == -ENOTSUP) {
1797                 printf("WARNING: kernel dm-verity not supported, skipping test.\n");
1798                 crypt_free(cd);
1799                 return;
1800         }
1801         OK_(crypt_get_active_device(cd, CDEVICE_1, &cad));
1802         EQ_(CRYPT_ACTIVATE_READONLY, cad.flags);
1803         crypt_free(cd);
1804
1805         OK_(crypt_init_by_name(&cd, CDEVICE_1));
1806         OK_(crypt_deactivate(cd, CDEVICE_1));
1807
1808         /* hash fail */
1809         root_hash[1] = ~root_hash[1];
1810         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, root_hash, 32, CRYPT_ACTIVATE_READONLY));
1811         OK_(crypt_get_active_device(cd, CDEVICE_1, &cad));
1812         EQ_(CRYPT_ACTIVATE_READONLY|CRYPT_ACTIVATE_CORRUPTED, cad.flags);
1813         OK_(crypt_deactivate(cd, CDEVICE_1));
1814         root_hash[1] = ~root_hash[1];
1815
1816         /* data fail */
1817         OK_(crypt_set_data_device(cd, DEVICE_1));
1818         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, root_hash, 32, CRYPT_ACTIVATE_READONLY));
1819         OK_(crypt_get_active_device(cd, CDEVICE_1, &cad));
1820         EQ_(CRYPT_ACTIVATE_READONLY|CRYPT_ACTIVATE_CORRUPTED, cad.flags);
1821         OK_(crypt_deactivate(cd, CDEVICE_1));
1822
1823         crypt_free(cd);
1824 }
1825
1826 static void TcryptTest(void)
1827 {
1828         struct crypt_device *cd = NULL;
1829         struct crypt_active_device cad;
1830         const char *passphrase = "aaaaaaaaaaaa";
1831         const char *kf1 = "tcrypt-images/keyfile1";
1832         const char *kf2 = "tcrypt-images/keyfile2";
1833         const char *keyfiles[] = { kf1, kf2 };
1834         struct crypt_params_tcrypt params = {
1835                 .passphrase = passphrase,
1836                 .passphrase_size = strlen(passphrase),
1837                 .keyfiles = keyfiles,
1838                 .keyfiles_count = 2,
1839         };
1840         double enc_mbr = 0, dec_mbr = 0;
1841         const char *tcrypt_dev = "tcrypt-images/tck_5-sha512-xts-aes";
1842         const char *tcrypt_dev2 = "tcrypt-images/tc_5-sha512-xts-serpent-twofish-aes";
1843         size_t key_size = 64;
1844         char key[key_size], key_def[key_size];
1845         const char *key_hex =
1846                 "98dee64abe44bbf41d171c1f7b3e8eacda6d6b01f459097459a167f8c2872a96"
1847                 "3979531d1cdc18af62757cf22286f16f8583d848524f128d7594ac2082668c73";
1848         int r;
1849
1850         crypt_decode_key(key_def, key_hex, strlen(key_hex) / 2);
1851
1852         // First ensure we can use af_alg skcipher interface
1853         r = crypt_benchmark(NULL, "aes", "xts", 512, 16, 1024, &enc_mbr, &dec_mbr);
1854         if (r == -ENOTSUP || r == -ENOENT) {
1855                 printf("WARNING: algif_skcipher interface not present, skipping test.\n");
1856                 return;
1857         }
1858
1859         OK_(crypt_init(&cd, tcrypt_dev));
1860         params.passphrase_size--;
1861         FAIL_(crypt_load(cd, CRYPT_TCRYPT, &params), "Wrong passphrase");
1862         params.passphrase_size++;
1863         OK_(crypt_load(cd, CRYPT_TCRYPT, &params));
1864
1865         // check params after load
1866         OK_(strcmp("xts-plain64", crypt_get_cipher_mode(cd)));
1867         OK_(strcmp("aes", crypt_get_cipher(cd)));
1868         EQ_(key_size, crypt_get_volume_key_size(cd));
1869         EQ_(256, crypt_get_iv_offset(cd));
1870         EQ_(256, crypt_get_data_offset(cd));
1871
1872         memset(key, 0, key_size);
1873         if (!_fips_mode) {
1874                 key_size--;
1875                 // small buffer
1876                 FAIL_(crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key, &key_size, NULL, 0), "small buffer");
1877                 key_size++;
1878                 OK_(crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key, &key_size, NULL, 0));
1879                 OK_(memcmp(key, key_def, key_size));
1880         }
1881
1882         reset_log();
1883         crypt_set_log_callback(cd, &new_log, NULL);
1884         OK_(crypt_dump(cd));
1885         OK_(!(global_lines != 0));
1886         crypt_set_log_callback(cd, NULL, NULL);
1887         reset_log();
1888
1889         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, NULL, 0, CRYPT_ACTIVATE_READONLY));
1890         crypt_free(cd);
1891
1892         OK_(crypt_init_by_name_and_header(&cd, CDEVICE_1, NULL));
1893         EQ_(crypt_status(cd, CDEVICE_1), CRYPT_ACTIVE);
1894
1895         FAIL_(crypt_volume_key_get(cd, CRYPT_ANY_SLOT, key, &key_size, NULL, 0), "Need crypt_load");
1896
1897         // check params after init_by_name
1898         OK_(strcmp("xts-plain64", crypt_get_cipher_mode(cd)));
1899         OK_(strcmp("aes", crypt_get_cipher(cd)));
1900         EQ_(key_size, crypt_get_volume_key_size(cd));
1901         EQ_(256, crypt_get_iv_offset(cd));
1902         EQ_(256, crypt_get_data_offset(cd));
1903
1904         OK_(crypt_get_active_device(cd, CDEVICE_1, &cad));
1905         EQ_(CRYPT_ACTIVATE_READONLY, cad.flags);
1906         EQ_(256, cad.offset);
1907         EQ_(256, cad.iv_offset);
1908         EQ_(72, cad.size);
1909
1910         OK_(crypt_deactivate(cd, CDEVICE_1));
1911         crypt_free(cd);
1912
1913         OK_(crypt_init(&cd, tcrypt_dev2));
1914         params.keyfiles = NULL;
1915         params.keyfiles_count = 0;
1916         OK_(crypt_load(cd, CRYPT_TCRYPT, &params));
1917         OK_(crypt_activate_by_volume_key(cd, CDEVICE_1, NULL, 0, CRYPT_ACTIVATE_READONLY));
1918         crypt_free(cd);
1919
1920         // Deactivate the whole chain
1921         EQ_(crypt_status(NULL, CDEVICE_1 "_1"), CRYPT_BUSY);
1922         OK_(crypt_deactivate(NULL, CDEVICE_1));
1923         EQ_(crypt_status(NULL, CDEVICE_1 "_1"), CRYPT_INACTIVE);
1924 }
1925
1926 // Check that gcrypt is properly initialised in format
1927 static void NonFIPSAlg(void)
1928 {
1929         struct crypt_device *cd;
1930         struct crypt_params_luks1 params = {0};
1931         char key[128] = "";
1932         size_t key_size = 128 / 8;
1933         const char *cipher = "aes";
1934         const char *cipher_mode = "cbc-essiv:sha256";
1935         int ret;
1936
1937         OK_(crypt_init(&cd, DEVICE_2));
1938         params.hash = "sha256";
1939         OK_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params));
1940         FAIL_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params),
1941               "Already formatted.");
1942         crypt_free(cd);
1943
1944         params.hash = "whirlpool";
1945         OK_(crypt_init(&cd, DEVICE_2));
1946         ret = crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params);
1947         if (ret < 0) {
1948                 printf("WARNING: whirlpool not supported, skipping test.\n");
1949                 crypt_free(cd);
1950                 return;
1951         }
1952         crypt_free(cd);
1953
1954         params.hash = "md5";
1955         OK_(crypt_init(&cd, DEVICE_2));
1956         FAIL_(crypt_format(cd, CRYPT_LUKS1, cipher, cipher_mode, NULL, key, key_size, &params),
1957               "MD5 unsupported, too short");
1958         crypt_free(cd);
1959 }
1960
1961 static void int_handler(int sig __attribute__((__unused__)))
1962 {
1963         _quit++;
1964 }
1965
1966 int main(int argc, char *argv[])
1967 {
1968         struct sigaction sa = { .sa_handler = int_handler };
1969         int i;
1970
1971         if (getuid() != 0) {
1972                 printf("You must be root to run this test.\n");
1973                 exit(0);
1974         }
1975
1976         for (i = 1; i < argc; i++) {
1977                 if (!strcmp("-v", argv[i]) || !strcmp("--verbose", argv[i]))
1978                         _verbose = 1;
1979                 else if (!strcmp("--debug", argv[i]))
1980                         _debug = _verbose = 1;
1981         }
1982
1983         /* Handle interrupt properly */
1984         sigaction(SIGINT, &sa, NULL);
1985         sigaction(SIGTERM, &sa, NULL);
1986
1987         _cleanup();
1988         if (_setup())
1989                 goto out;
1990
1991         crypt_set_debug_level(_debug ? CRYPT_DEBUG_ALL : CRYPT_DEBUG_NONE);
1992
1993         RUN_(NonFIPSAlg, "Crypto is properly initialised in format"); //must be the first!
1994         RUN_(AddDevicePlain, "plain device API creation exercise");
1995         RUN_(HashDevicePlain, "plain device API hash test");
1996         RUN_(AddDeviceLuks, "Format and use LUKS device");
1997         RUN_(LuksHeaderLoad, "test header load");
1998         RUN_(LuksHeaderRestore, "test LUKS header restore");
1999         RUN_(LuksHeaderBackup, "test LUKS header backup");
2000         RUN_(ResizeDeviceLuks, "Luks device resize tests");
2001         RUN_(UseLuksDevice, "Use pre-formated LUKS device");
2002         RUN_(SuspendDevice, "Suspend/Resume test");
2003         RUN_(UseTempVolumes, "Format and use temporary encrypted device");
2004         RUN_(CallbacksTest, "API callbacks test");
2005         RUN_(VerityTest, "DM verity test");
2006         RUN_(TcryptTest, "Tcrypt API test");
2007 out:
2008         _cleanup();
2009         return 0;
2010 }