lib/setup.c

   1 #include <string.h>
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <stdarg.h>
   5 #include <fcntl.h>
   6 #include <errno.h>
   7
   8 #include "libcryptsetup.h"
   9 #include "luks.h"
  10 #include "internal.h"
  11
  12 struct crypt_device {
  13         char *type;
  14
  15         char *device;
  16         struct luks_masterkey *volume_key;
  17         uint64_t timeout;
  18         uint64_t iteration_time;
  19         int tries;
  20         int password_verify;
  21
  22         /* used in CRYPT_LUKS1 */
  23         struct luks_phdr hdr;
  24         uint64_t PBKDF2_per_sec;
  25
  26         /* used in CRYPT_PLAIN */
  27         struct crypt_params_plain plain_hdr;
  28         char *plain_cipher;
  29         char *plain_cipher_mode;
  30         char *plain_uuid;
  31
  32         /* callbacks definitions */
  33         void (*log)(int class, const char *msg, void *usrptr);
  34         void *log_usrptr;
  35         int (*confirm)(const char *msg, void *usrptr);
  36         void *confirm_usrptr;
  37         int (*password)(const char *msg, char *buf, size_t length, void *usrptr);
  38         void *password_usrptr;
  39 };
  40
  41 /* Log helper */
  42 static void (*_default_log)(int class, const char *msg, void *usrptr) = NULL;
  43 static int _debug_level = 0;
  44
  45 void crypt_set_debug_level(int level)
  46 {
  47         _debug_level = level;
  48 }
  49
  50 int crypt_get_debug_level()
  51 {
  52         return _debug_level;
  53 }
  54
  55 void crypt_log(struct crypt_device *cd, int class, const char *msg)
  56 {
  57         if (cd && cd->log)
  58                 cd->log(class, msg, cd->log_usrptr);
  59         else if (_default_log)
  60                 _default_log(class, msg, NULL);
  61 }
  62
  63 void logger(struct crypt_device *cd, int class, const char *file,
  64             int line, const char *format, ...)
  65 {
  66         va_list argp;
  67         char *target = NULL;
  68
  69         va_start(argp, format);
  70
  71         if (vasprintf(&target, format, argp) > 0) {
  72                 if (class >= 0) {
  73                         crypt_log(cd, class, target);
  74 #ifdef CRYPT_DEBUG
  75                 } else if (_debug_level)
  76                         printf("# %s:%d %s\n", file ?: "?", line, target);
  77 #else
  78                 } else if (_debug_level)
  79                         printf("# %s\n", target);
  80 #endif
  81         }
  82
  83         va_end(argp);
  84         free(target);
  85 }
  86
  87 /*
  88  * Password processing behaviour matrix of process_key
  89  *
  90  * from binary file: check if there is sufficently large key material
  91  * interactive & from fd: hash if requested, otherwise crop or pad with '0'
  92  */
  93 static char *process_key(struct crypt_device *cd, const char *hash_name,
  94                          const char *key_file, size_t key_size,
  95                          const char *pass, size_t passLen)
  96 {
  97         char *key = safe_alloc(key_size);
  98         memset(key, 0, key_size);
  99
 100         /* key is coming from binary file */
 101         if (key_file && strcmp(key_file, "-")) {
 102                 if(passLen < key_size) {
 103                         log_err(cd, _("Cannot not read %d bytes from key file %s.\n"),
 104                                 key_size, key_file);
 105                         safe_free(key);
 106                         return NULL;
 107                 }
 108                 memcpy(key, pass, key_size);
 109                 return key;
 110         }
 111
 112         /* key is coming from tty, fd or binary stdin */
 113         if (hash_name) {
 114                 if (hash(NULL, hash_name, key, key_size, pass, passLen) < 0) {
 115                         log_err(cd, _("Key processing error.\n"));
 116                         safe_free(key);
 117                         return NULL;
 118                 }
 119         } else if (passLen > key_size) {
 120                 memcpy(key, pass, key_size);
 121         } else {
 122                 memcpy(key, pass, passLen);
 123         }
 124
 125         return key;
 126 }
 127
 128 int parse_into_name_and_mode(const char *nameAndMode, char *name, char *mode)
 129 {
 130 /* Token content stringification, see info cpp/stringification */
 131 #define str(s) #s
 132 #define xstr(s) str(s)
 133 #define scanpattern1 "%" xstr(LUKS_CIPHERNAME_L) "[^-]-%" xstr(LUKS_CIPHERMODE_L)  "s"
 134 #define scanpattern2 "%" xstr(LUKS_CIPHERNAME_L) "[^-]"
 135
 136         int r;
 137
 138         if(sscanf(nameAndMode,scanpattern1, name, mode) != 2) {
 139                 if((r = sscanf(nameAndMode,scanpattern2,name)) == 1)
 140                         strncpy(mode,"cbc-plain",10);
 141                 else
 142                         return -EINVAL;
 143         }
 144
 145         return 0;
 146
 147 #undef scanpattern1
 148 #undef scanpattern2
 149 #undef str
 150 #undef xstr
 151 }
 152
 153 static int isPLAIN(const char *type)
 154 {
 155         return (type && !strcmp(CRYPT_PLAIN, type));
 156 }
 157
 158 static int isLUKS(const char *type)
 159 {
 160         return (type && !strcmp(CRYPT_LUKS1, type));
 161 }
 162
 163 /* keyslot helpers */
 164 static int keyslot_verify_or_find_empty(struct crypt_device *cd, int *keyslot)
 165 {
 166         if (*keyslot == CRYPT_ANY_SLOT) {
 167                 *keyslot = LUKS_keyslot_find_empty(&cd->hdr);
 168                 if (*keyslot < 0) {
 169                         log_err(cd, _("All key slots full.\n"));
 170                         return -EINVAL;
 171                 }
 172         }
 173
 174         switch (LUKS_keyslot_info(&cd->hdr, *keyslot)) {
 175                 case CRYPT_SLOT_INVALID:
 176                         log_err(cd, _("Key slot %d is invalid, please select between 0 and %d.\n"),
 177                                 *keyslot, LUKS_NUMKEYS - 1);
 178                         return -EINVAL;
 179                 case CRYPT_SLOT_INACTIVE:
 180                         break;
 181                 default:
 182                         log_err(cd, _("Key slot %d is full, please select another one.\n"),
 183                                 *keyslot);
 184                         return -EINVAL;
 185         }
 186
 187         return 0;
 188 }
 189
 190 static int verify_other_keyslot(struct crypt_device *cd,
 191                                 const char *key_file,
 192                                 unsigned int flags,
 193                                 int keyIndex)
 194 {
 195         struct luks_masterkey *mk;
 196         crypt_keyslot_info ki;
 197         int openedIndex;
 198         char *password = NULL;
 199         unsigned int passwordLen;
 200
 201         get_key(_("Enter any remaining LUKS passphrase: "), &password,
 202                 &passwordLen, 0, key_file, cd->timeout, flags, cd);
 203         if(!password)
 204                 return -EINVAL;
 205
 206         ki = crypt_keyslot_status(cd, keyIndex);
 207         if (ki == CRYPT_SLOT_ACTIVE) /* Not last slot */
 208                 LUKS_keyslot_set(&cd->hdr, keyIndex, 0);
 209
 210         openedIndex = LUKS_open_key_with_hdr(cd->device, CRYPT_ANY_SLOT,
 211                                              password, passwordLen,
 212                                              &cd->hdr, &mk, cd);
 213
 214         if (ki == CRYPT_SLOT_ACTIVE)
 215                 LUKS_keyslot_set(&cd->hdr, keyIndex, 1);
 216         LUKS_dealloc_masterkey(mk);
 217         safe_free(password);
 218
 219         if (openedIndex < 0)
 220                 return -EPERM;
 221
 222         log_std(cd, _("Key slot %d verified.\n"), openedIndex);
 223         return 0;
 224 }
 225
 226 static int find_keyslot_by_passphrase(struct crypt_device *cd,
 227                                       const char *key_file,
 228                                       unsigned int flags,
 229                                       char *message)
 230 {
 231         struct luks_masterkey *mk;
 232         char *password = NULL;
 233         unsigned int passwordLen;
 234         int keyIndex;
 235
 236         get_key(message,&password,&passwordLen, 0, key_file,
 237                 cd->timeout, flags, cd);
 238         if(!password)
 239                 return -EINVAL;
 240
 241         keyIndex = LUKS_open_key_with_hdr(cd->device, CRYPT_ANY_SLOT, password,
 242                                           passwordLen, &cd->hdr, &mk, cd);
 243         LUKS_dealloc_masterkey(mk);
 244         safe_free(password);
 245
 246         return keyIndex;
 247 }
 248
 249 static int device_check_and_adjust(struct crypt_device *cd,
 250                                    const char *device,
 251                                    uint64_t *size, uint64_t *offset,
 252                                    int *read_only)
 253 {
 254         struct device_infos infos;
 255
 256         if (get_device_infos(device, &infos, cd) < 0) {
 257                 log_err(cd, _("Cannot get info about device %s.\n"), device);
 258                 return -ENOTBLK;
 259         }
 260
 261         if (!*size) {
 262                 *size = infos.size;
 263                 if (!*size) {
 264                         log_err(cd, _("Device %s has zero size.\n"), device);
 265                         return -ENOTBLK;
 266                 }
 267                 if (*size < *offset) {
 268                         log_err(cd, _("Device %s is too small.\n"), device);
 269                         return -EINVAL;
 270                 }
 271                 *size -= *offset;
 272         }
 273
 274         if (infos.readonly)
 275                 *read_only = 1;
 276
 277         log_dbg("Calculated device size is %" PRIu64 " sectors (%s), offset %" PRIu64 ".",
 278                 *size, *read_only ? "RO" : "RW", *offset);
 279         return 0;
 280 }
 281
 282 static int luks_remove_helper(struct crypt_device *cd,
 283                               int key_slot,
 284                               const char *other_key_file,
 285                               const char *key_file,
 286                               int verify)
 287 {
 288         crypt_keyslot_info ki;
 289         int r = -EINVAL;
 290
 291         if (key_slot == CRYPT_ANY_SLOT) {
 292                 key_slot = find_keyslot_by_passphrase(cd, key_file, 0,
 293                                 _("Enter LUKS passphrase to be deleted: "));
 294                 if(key_slot < 0) {
 295                         r = -EPERM;
 296                         goto out;
 297                 }
 298
 299                 log_std(cd, _("key slot %d selected for deletion.\n"), key_slot);
 300         }
 301
 302         ki = crypt_keyslot_status(cd, key_slot);
 303         if (ki == CRYPT_SLOT_INVALID) {
 304                 log_err(cd, _("Key slot %d is invalid, please select between 0 and %d.\n"),
 305                         key_slot, LUKS_NUMKEYS - 1);
 306                 r = -EINVAL;
 307                 goto out;
 308         }
 309         if (ki <= CRYPT_SLOT_INACTIVE) {
 310                 log_err(cd, _("Key %d not active. Can't wipe.\n"), key_slot);
 311                 r = -EINVAL;
 312                 goto out;
 313         }
 314
 315         if (ki == CRYPT_SLOT_ACTIVE_LAST && cd->confirm &&
 316             !(cd->confirm(_("This is the last keyslot."
 317                             " Device will become unusable after purging this key."),
 318                          cd->confirm_usrptr))) {
 319                 r = -EINVAL;
 320                 goto out;
 321         }
 322
 323         if(verify)
 324                 r = verify_other_keyslot(cd, other_key_file, 0, key_slot);
 325         else
 326                 r = 0;
 327
 328         if (!r)
 329                 r = crypt_keyslot_destroy(cd, key_slot);
 330 out:
 331         return (r < 0) ? r : 0;
 332 }
 333
 334 static int create_device_helper(struct crypt_device *cd,
 335                                 const char *name,
 336                                 const char *hash,
 337                                 const char *cipher,
 338                                 const char *cipher_mode,
 339                                 const char *key_file,
 340                                 const char *key,
 341                                 unsigned int keyLen,
 342                                 int key_size,
 343                                 uint64_t size,
 344                                 uint64_t skip,
 345                                 uint64_t offset,
 346                                 const char *uuid,
 347                                 int read_only,
 348                                 unsigned int flags,
 349                                 int reload)
 350 {
 351         crypt_status_info ci;
 352         char *dm_cipher = NULL;
 353         char *processed_key = NULL;
 354         int r;
 355
 356         ci = crypt_status(cd, name);
 357         if (ci == CRYPT_INVALID)
 358                 return -EINVAL;
 359
 360         if (reload && ci < CRYPT_ACTIVE)
 361                 return -EINVAL;
 362
 363         if (!reload && ci >= CRYPT_ACTIVE) {
 364                 log_err(cd, _("Device %s already exists.\n"), name);
 365                 return -EEXIST;
 366         }
 367
 368         if (key_size < 0 || key_size > 1024) {
 369                 log_err(cd, _("Invalid key size %d.\n"), key_size);
 370                 return -EINVAL;
 371         }
 372
 373         r = device_check_and_adjust(cd, cd->device, &size, &offset, &read_only);
 374         if (r)
 375                 return r;
 376
 377         if (cipher_mode && asprintf(&dm_cipher, "%s-%s", cipher, cipher_mode) < 0)
 378                 return -ENOMEM;
 379
 380         processed_key = process_key(cd, hash, key_file, key_size, key, keyLen);
 381         if (!processed_key)
 382                 return -ENOENT;
 383
 384         r = dm_create_device(name, cd->device, dm_cipher ?: cipher, cd->type, uuid, size, skip, offset,
 385                              key_size, processed_key, read_only, reload);
 386
 387         free(dm_cipher);
 388         safe_free(processed_key);
 389         return r;
 390 }
 391
 392 static int open_from_hdr_and_mk(struct crypt_device *cd,
 393                                 struct luks_masterkey *mk,
 394                                 const char *name,
 395                                 uint32_t flags)
 396 {
 397         uint64_t size, offset;
 398         char *cipher;
 399         int read_only, no_uuid, r;
 400
 401         size = 0;
 402         offset = crypt_get_data_offset(cd);
 403         read_only = flags & CRYPT_ACTIVATE_READONLY;
 404         no_uuid = flags & CRYPT_ACTIVATE_NO_UUID;
 405
 406         r = device_check_and_adjust(cd, cd->device, &size, &offset, &read_only);
 407         if (r)
 408                 return r;
 409
 410         if (asprintf(&cipher, "%s-%s", crypt_get_cipher(cd),
 411                      crypt_get_cipher_mode(cd)) < 0)
 412                 r = -ENOMEM;
 413         else
 414                 r = dm_create_device(name, cd->device, cipher, cd->type,
 415                                      no_uuid ? NULL : crypt_get_uuid(cd),
 416                                      size, 0, offset, mk->keyLength, mk->key,
 417                                      read_only, 0);
 418         free(cipher);
 419         return r;
 420 }
 421
 422 static void log_wrapper(int class, const char *msg, void *usrptr)
 423 {
 424         void (*xlog)(int class, char *msg) = usrptr;
 425         xlog(class, (char *)msg);
 426 }
 427
 428 static int yesDialog_wrapper(const char *msg, void *usrptr)
 429 {
 430         int (*xyesDialog)(char *msg) = usrptr;
 431         return xyesDialog((char*)msg);
 432 }
 433
 434 int crypt_confirm(struct crypt_device *cd, const char *msg)
 435 {
 436         if (!cd || !cd->confirm)
 437                 return 1;
 438         else
 439                 return cd->confirm(msg, cd->confirm_usrptr);
 440 }
 441
 442 static void key_from_terminal(struct crypt_device *cd, char *msg, char **key,
 443                               unsigned int *key_len, int force_verify)
 444 {
 445         int r, flags = 0;
 446
 447         if (cd->password) {
 448                 *key = safe_alloc(MAX_TTY_PASSWORD_LEN);
 449                 if (*key)
 450                         return;
 451                 r = cd->password(msg, *key, (size_t)key_len, cd->password_usrptr);
 452                 if (r < 0) {
 453                         safe_free(*key);
 454                         *key = NULL;
 455                 } else
 456                         *key_len = r;
 457         } else {
 458                 if (force_verify || cd->password_verify)
 459                         flags |= CRYPT_FLAG_VERIFY_IF_POSSIBLE;
 460                 get_key(msg, key, key_len, 0, NULL, cd->timeout, flags, cd);
 461         }
 462 }
 463
 464 static int volume_key_by_terminal_passphrase(struct crypt_device *cd, int keyslot,
 465                                              struct luks_masterkey **mk)
 466 {
 467         char *prompt = NULL, *passphrase_read = NULL;
 468         unsigned int passphrase_size_read;
 469         int r = -EINVAL, tries = cd->tries;
 470
 471         if(asprintf(&prompt, _("Enter passphrase for %s: "), cd->device) < 0)
 472                 return -ENOMEM;
 473
 474         *mk = NULL;
 475         do {
 476                 if (*mk)
 477                         LUKS_dealloc_masterkey(*mk);
 478                 *mk = NULL;
 479
 480                 key_from_terminal(cd, prompt, &passphrase_read,
 481                                   &passphrase_size_read, 0);
 482                 if(!passphrase_read) {
 483                         r = -EINVAL;
 484                         break;
 485                 }
 486
 487                 r = LUKS_open_key_with_hdr(cd->device, keyslot, passphrase_read,
 488                                            passphrase_size_read, &cd->hdr, mk, cd);
 489                 safe_free(passphrase_read);
 490                 passphrase_read = NULL;
 491         } while (r == -EPERM && (--tries > 0));
 492
 493         if (r < 0 && *mk) {
 494                 LUKS_dealloc_masterkey(*mk);
 495                 *mk = NULL;
 496         }
 497         free(prompt);
 498
 499         return r;
 500
 501 }
 502
 503 static void key_from_file(struct crypt_device *cd, char *msg,
 504                           char **key, unsigned int *key_len,
 505                           const char *key_file, size_t key_size)
 506 {
 507         get_key(msg, key, key_len, key_size, key_file, cd->timeout, 0, cd);
 508 }
 509
 510 static int _crypt_init(struct crypt_device **cd,
 511                        const char *type,
 512                        struct crypt_options *options,
 513                        int load, int need_dm)
 514 {
 515         int init_by_name, r;
 516
 517         /* if it is plain device and mapping table is being reloaded
 518         initialize it by name*/
 519         init_by_name = (type && !strcmp(type, CRYPT_PLAIN) && load);
 520
 521         /* Some of old API calls do not require DM in kernel,
 522            fake initialisation by initialise it with kernel_check disabled */
 523         if (!need_dm)
 524                 (void)dm_init(NULL, 0);
 525         if (init_by_name)
 526                 r = crypt_init_by_name(cd, options->name);
 527         else
 528                 r = crypt_init(cd, options->device);
 529         if (!need_dm)
 530                 dm_exit();
 531
 532         if (r)
 533                 return -EINVAL;
 534
 535         crypt_set_log_callback(*cd, log_wrapper, options->icb->log);
 536         crypt_set_confirm_callback(*cd, yesDialog_wrapper, options->icb->yesDialog);
 537
 538         crypt_set_timeout(*cd, options->timeout);
 539         crypt_set_password_retry(*cd, options->tries);
 540         crypt_set_iterarion_time(*cd, options->iteration_time ?: 1000);
 541         crypt_set_password_verify(*cd, options->flags & CRYPT_FLAG_VERIFY);
 542
 543         if (load && !init_by_name)
 544                 r = crypt_load(*cd, type, NULL);
 545
 546         if (!r && type && !(*cd)->type) {
 547                 (*cd)->type = strdup(type);
 548                 if (!(*cd)->type)
 549                         r = -ENOMEM;
 550         }
 551
 552         if (r)
 553                 crypt_free(*cd);
 554
 555         return r;
 556 }
 557
 558 void crypt_set_log_callback(struct crypt_device *cd,
 559         void (*log)(int class, const char *msg, void *usrptr),
 560         void *usrptr)
 561 {
 562         if (!cd)
 563                 _default_log = log;
 564         else {
 565                 cd->log = log;
 566                 cd->log_usrptr = usrptr;
 567         }
 568 }
 569
 570 void crypt_set_confirm_callback(struct crypt_device *cd,
 571         int (*confirm)(const char *msg, void *usrptr),
 572         void *usrptr)
 573 {
 574         cd->confirm = confirm;
 575         cd->confirm_usrptr = usrptr;
 576 }
 577
 578 void crypt_set_password_callback(struct crypt_device *cd,
 579         int (*password)(const char *msg, char *buf, size_t length, void *usrptr),
 580         void *usrptr)
 581 {
 582         cd->password = password;
 583         cd->password_usrptr = usrptr;
 584 }
 585
 586 /* OPTIONS: name, cipher, device, hash, key_file, key_size, key_slot,
 587  *          offset, size, skip, timeout, tries, passphrase_fd (ignored),
 588  *          flags, icb */
 589 int crypt_create_device(struct crypt_options *options)
 590 {
 591         struct crypt_device *cd = NULL;
 592         char *key = NULL;
 593         unsigned int keyLen;
 594         int r;
 595
 596         r = _crypt_init(&cd, CRYPT_PLAIN, options, 0, 1);
 597         if (r)
 598                 return r;
 599
 600         get_key(_("Enter passphrase: "), &key, &keyLen, options->key_size,
 601                 options->key_file, cd->timeout, options->flags, cd);
 602         if (!key)
 603                 r = -ENOENT;
 604         else
 605                 r = create_device_helper(cd, options->name, options->hash,
 606                         options->cipher, NULL, options->key_file, key, keyLen,
 607                         options->key_size, options->size, options->skip,
 608                         options->offset, NULL, options->flags & CRYPT_FLAG_READONLY,
 609                         options->flags, 0);
 610
 611         safe_free(key);
 612         crypt_free(cd);
 613         return r;
 614 }
 615
 616 /* OPTIONS: same as create above */
 617 int crypt_update_device(struct crypt_options *options)
 618 {
 619         struct crypt_device *cd = NULL;
 620         char *key = NULL;
 621         unsigned int keyLen;
 622         int r;
 623
 624         r = _crypt_init(&cd, CRYPT_PLAIN, options, 1, 1);
 625         if (r)
 626                 return r;
 627
 628         get_key(_("Enter passphrase: "), &key, &keyLen, options->key_size,
 629                 options->key_file, cd->timeout, options->flags, cd);
 630         if (!key)
 631                 r = -ENOENT;
 632         else
 633                 r = create_device_helper(cd, options->name, options->hash,
 634                         options->cipher, NULL, options->key_file, key, keyLen,
 635                         options->key_size, options->size, options->skip,
 636                         options->offset, NULL, options->flags & CRYPT_FLAG_READONLY,
 637                         options->flags, 1);
 638
 639         safe_free(key);
 640         crypt_free(cd);
 641         return r;
 642 }
 643
 644 /* OPTIONS: name, size, icb */
 645 int crypt_resize_device(struct crypt_options *options)
 646 {
 647         struct crypt_device *cd = NULL;
 648         char *device = NULL, *cipher = NULL, *uuid = NULL, *key = NULL;
 649         char *type = NULL;
 650         uint64_t size, skip, offset;
 651         int key_size, read_only, r;
 652
 653         log_dbg("Resizing device %s to %" PRIu64 " sectors.", options->name, options->size);
 654
 655         if (dm_init(NULL, 1) < 0)
 656                 return -ENOSYS;
 657
 658         r = dm_query_device(options->name, &device, &size, &skip, &offset,
 659                             &cipher, &key_size, &key, &read_only, NULL, &uuid);
 660         if (r < 0) {
 661                 log_err(NULL, _("Device %s is not active.\n"), options->name);
 662                 goto out;
 663         }
 664
 665         /* Try to determine type of device from UUID */
 666         type = CRYPT_PLAIN;
 667         if (uuid) {
 668                 if (!strncmp(uuid, CRYPT_PLAIN, strlen(CRYPT_PLAIN))) {
 669                         type = CRYPT_PLAIN;
 670                         free (uuid);
 671                         uuid = NULL;
 672                 } else if (!strncmp(uuid, CRYPT_LUKS1, strlen(CRYPT_LUKS1)))
 673                         type = CRYPT_LUKS1;
 674         }
 675
 676         if (!options->device)
 677                 options->device = device;
 678
 679         r = _crypt_init(&cd, type, options, 1, 1);
 680         if (r)
 681                 goto out;
 682
 683         size = options->size;
 684         r = device_check_and_adjust(cd, device, &size, &offset, &read_only);
 685         if (r)
 686                 goto out;
 687
 688         r = dm_create_device(options->name, device, cipher, type,
 689                              crypt_get_uuid(cd), size, skip, offset,
 690                              key_size, key, read_only, 1);
 691 out:
 692         safe_free(key);
 693         free(cipher);
 694         if (options->device == device)
 695                 options->device = NULL;
 696         free(device);
 697         free(uuid);
 698         crypt_free(cd);
 699         dm_exit();
 700         return r;
 701 }
 702
 703 /* OPTIONS: name, icb */
 704 int crypt_query_device(struct crypt_options *options)
 705 {
 706         int read_only, r;
 707
 708         log_dbg("Query device %s.", options->name);
 709
 710         if (dm_init(NULL, 1) < 0)
 711                 return -ENOSYS;
 712
 713         r = dm_status_device(options->name);
 714         if (r == -ENODEV) {
 715                 dm_exit();
 716                 return 0;
 717         }
 718
 719         r = dm_query_device(options->name, (char **)&options->device, &options->size,
 720                             &options->skip, &options->offset, (char **)&options->cipher,
 721                             &options->key_size, NULL, &read_only, NULL, NULL);
 722
 723         dm_exit();
 724         if (r < 0)
 725                 return r;
 726
 727         if (read_only)
 728                 options->flags |= CRYPT_FLAG_READONLY;
 729
 730         options->flags |= CRYPT_FLAG_FREE_DEVICE;
 731         options->flags |= CRYPT_FLAG_FREE_CIPHER;
 732
 733         return 1;
 734 }
 735
 736 /* OPTIONS: name, icb */
 737 int crypt_remove_device(struct crypt_options *options)
 738 {
 739         struct crypt_device *cd = NULL;
 740         int r;
 741
 742         r = crypt_init_by_name(&cd, options->name);
 743         if (r)
 744                 return r;
 745
 746         r = crypt_deactivate(cd, options->name);
 747
 748         crypt_free(cd);
 749         return r;
 750
 751 }
 752
 753 /* OPTIONS: device, cipher, hash, align_payload, key_size (master key), key_slot
 754  *          new_key_file, iteration_time, timeout, flags, icb */
 755 int crypt_luksFormat(struct crypt_options *options)
 756 {
 757         char cipherName[LUKS_CIPHERNAME_L];
 758         char cipherMode[LUKS_CIPHERMODE_L];
 759         char *password=NULL;
 760         unsigned int passwordLen;
 761         struct crypt_device *cd;
 762         struct crypt_params_luks1 cp = {
 763                 .hash = options->hash,
 764                 .data_alignment = options->align_payload
 765         };
 766         int r;
 767
 768         r = parse_into_name_and_mode(options->cipher, cipherName, cipherMode);
 769         if(r < 0) {
 770                 log_err(cd, _("No known cipher specification pattern detected.\n"));
 771                 return r;
 772         }
 773
 774         if ((r = _crypt_init(&cd, CRYPT_LUKS1, options, 0, 1)))
 775                 return r;
 776
 777         if (options->key_slot >= LUKS_NUMKEYS && options->key_slot != CRYPT_ANY_SLOT) {
 778                 log_err(cd, _("Key slot %d is invalid, please select between 0 and %d.\n"),
 779                         options->key_slot, LUKS_NUMKEYS - 1);
 780                 r = -EINVAL;
 781                 goto out;
 782         }
 783
 784         get_key(_("Enter LUKS passphrase: "), &password, &passwordLen, 0,
 785                 options->new_key_file, options->timeout, options->flags, cd);
 786
 787         if(!password) {
 788                 r = -EINVAL;
 789                 goto out;
 790         }
 791
 792         r = crypt_format(cd, CRYPT_LUKS1, cipherName, cipherMode,
 793                          NULL, NULL, options->key_size, &cp);
 794         if (r < 0)
 795                 goto out;
 796
 797         /* Add keyslot using internally stored volume key generated during format */
 798         r = crypt_keyslot_add_by_volume_key(cd, options->key_slot, NULL, 0,
 799                                             password, passwordLen);
 800 out:
 801         crypt_free(cd);
 802         safe_free(password);
 803         return (r < 0) ? r : 0;
 804 }
 805
 806 /* OPTIONS: name, device, key_size, key_file, timeout, tries, flags, icb */
 807 int crypt_luksOpen(struct crypt_options *options)
 808 {
 809         struct crypt_device *cd = NULL;
 810         uint32_t flags = 0;
 811         int r;
 812
 813         if (!options->name)
 814                 return -EINVAL;
 815
 816         r = _crypt_init(&cd, CRYPT_LUKS1, options, 1, 1);
 817         if (r)
 818                 return r;
 819
 820         if (options->flags & CRYPT_FLAG_READONLY)
 821                 flags |= CRYPT_ACTIVATE_READONLY;
 822
 823         if (options->flags & CRYPT_FLAG_NON_EXCLUSIVE_ACCESS)
 824                 flags |= CRYPT_ACTIVATE_NO_UUID;
 825
 826         if (options->key_file)
 827                 r = crypt_activate_by_keyfile(cd, options->name,
 828                         CRYPT_ANY_SLOT, options->key_file, options->key_size,
 829                         flags);
 830         else
 831                 r = crypt_activate_by_passphrase(cd, options->name,
 832                         CRYPT_ANY_SLOT, options->passphrase,
 833                         options->passphrase ? strlen(options->passphrase) : 0,
 834                         flags);
 835
 836         crypt_free(cd);
 837         return (r < 0) ? r : 0;
 838 }
 839
 840 /* OPTIONS: device, keys_slot, key_file, timeout, flags, icb */
 841 int crypt_luksKillSlot(struct crypt_options *options)
 842 {
 843         struct crypt_device *cd = NULL;
 844         int r;
 845
 846         r = _crypt_init(&cd, CRYPT_LUKS1, options, 1, 1);
 847         if (r)
 848                 return r;
 849
 850         r = luks_remove_helper(cd, options->key_slot, options->key_file, NULL,
 851                                options->flags & CRYPT_FLAG_VERIFY_ON_DELKEY);
 852
 853         crypt_free(cd);
 854         return (r < 0) ? r : 0;
 855 }
 856
 857 /* OPTIONS: device, new_key_file, key_file, timeout, flags, icb */
 858 int crypt_luksRemoveKey(struct crypt_options *options)
 859 {
 860         struct crypt_device *cd = NULL;
 861         int r;
 862
 863         r = _crypt_init(&cd, CRYPT_LUKS1, options, 1, 1);
 864         if (r)
 865                 return r;
 866
 867         r = luks_remove_helper(cd, CRYPT_ANY_SLOT, options->key_file, options->new_key_file,
 868                                options->flags & CRYPT_FLAG_VERIFY_ON_DELKEY);
 869
 870         crypt_free(cd);
 871         return (r < 0) ? r : 0;
 872 }
 873
 874
 875 /* OPTIONS: device, new_key_file, key_file, key_slot, flags,
 876             iteration_time, timeout, icb */
 877 int crypt_luksAddKey(struct crypt_options *options)
 878 {
 879         struct crypt_device *cd = NULL;
 880         int r = -EINVAL;
 881
 882         r = _crypt_init(&cd, CRYPT_LUKS1, options, 1, 1);
 883         if (r)
 884                 return r;
 885
 886         if (options->key_file || options->new_key_file)
 887                 r = crypt_keyslot_add_by_keyfile(cd, options->key_slot,
 888                                                  options->key_file, 0,
 889                                                  options->new_key_file, 0);
 890         else
 891                 r = crypt_keyslot_add_by_passphrase(cd, options->key_slot,
 892                                                     NULL, 0, NULL, 0);
 893
 894         crypt_free(cd);
 895         return (r < 0) ? r : 0;
 896 }
 897
 898 /* OPTIONS: device, icb */
 899 int crypt_luksUUID(struct crypt_options *options)
 900 {
 901         struct crypt_device *cd = NULL;
 902         char *uuid;
 903         int r;
 904
 905         r = _crypt_init(&cd, CRYPT_LUKS1, options, 1, 0);
 906         if (r)
 907                 return r;
 908
 909         uuid = (char *)crypt_get_uuid(cd);
 910         log_std(cd, uuid ?: "");
 911         log_std(cd, "\n");
 912         crypt_free(cd);
 913         return 0;
 914 }
 915
 916 /* OPTIONS: device, icb */
 917 int crypt_isLuks(struct crypt_options *options)
 918 {
 919         struct crypt_device *cd = NULL;
 920         int r;
 921
 922         log_dbg("Check device %s for LUKS header.", options->device);
 923
 924         r = crypt_init(&cd, options->device);
 925         if (r < 0)
 926                 return -EINVAL;
 927
 928         /* Do print fail here, no need to crypt_load() */
 929         r = LUKS_read_phdr(cd->device, &cd->hdr, 0, cd) ? -EINVAL : 0;
 930
 931         crypt_free(cd);
 932         return r;
 933 }
 934
 935 /* OPTIONS: device, icb */
 936 int crypt_luksDump(struct crypt_options *options)
 937 {
 938         struct crypt_device *cd = NULL;
 939         int r;
 940
 941         r = _crypt_init(&cd, CRYPT_LUKS1, options, 1, 0);
 942         if(r < 0)
 943                 return r;
 944
 945         r = crypt_dump(cd);
 946
 947         crypt_free(cd);
 948         return 0;
 949 }
 950
 951 void crypt_get_error(char *buf, size_t size)
 952 {
 953         const char *error = get_error();
 954
 955         if (!buf || size < 1)
 956                 set_error(NULL);
 957         else if (error) {
 958                 strncpy(buf, error, size - 1);
 959                 buf[size - 1] = '\0';
 960                 set_error(NULL);
 961         } else
 962                 buf[0] = '\0';
 963 }
 964
 965 void crypt_put_options(struct crypt_options *options)
 966 {
 967         if (options->flags & CRYPT_FLAG_FREE_DEVICE) {
 968                 free((char *)options->device);
 969                 options->device = NULL;
 970                 options->flags &= ~CRYPT_FLAG_FREE_DEVICE;
 971         }
 972         if (options->flags & CRYPT_FLAG_FREE_CIPHER) {
 973                 free((char *)options->cipher);
 974                 options->cipher = NULL;
 975                 options->flags &= ~CRYPT_FLAG_FREE_CIPHER;
 976         }
 977 }
 978
 979 const char *crypt_get_dir(void)
 980 {
 981         return dm_get_dir();
 982 }
 983
 984 /////////////////////////////////
 985 //
 986 // New API
 987 //
 988
 989 int crypt_init(struct crypt_device **cd, const char *device)
 990 {
 991         struct crypt_device *h = NULL;
 992
 993         if (!cd)
 994                 return -EINVAL;
 995
 996         log_dbg("Allocating crypt device %s context.", device);
 997
 998         if (device && !device_ready(NULL, device, O_RDONLY))
 999                 return -ENOTBLK;
1000
1001         if (!(h = malloc(sizeof(struct crypt_device))))
1002                 return -ENOMEM;
1003
1004         memset(h, 0, sizeof(*h));
1005
1006         if (device) {
1007                 h->device = strdup(device);
1008                 if (!h->device) {
1009                         free(h);
1010                         return -ENOMEM;
1011                 }
1012         } else
1013                 h->device = NULL;
1014
1015         if (dm_init(h, 1) < 0) {
1016                 free(h);
1017                 return -ENOSYS;
1018         }
1019
1020         h->iteration_time = 1000;
1021         h->password_verify = 0;
1022         h->tries = 3;
1023         *cd = h;
1024         return 0;
1025 }
1026
1027 int crypt_init_by_name(struct crypt_device **cd, const char *name)
1028 {
1029         crypt_status_info ci;
1030         char *device = NULL;
1031         int r;
1032
1033         log_dbg("Allocating crypt device context by device %s.", name);
1034
1035         ci = crypt_status(NULL, name);
1036         if (ci == CRYPT_INVALID)
1037                 return -ENODEV;
1038
1039         if (ci < CRYPT_ACTIVE) {
1040                 log_err(NULL, _("Device %s is not active.\n"), name);
1041                 return -ENODEV;
1042         }
1043
1044         r = dm_query_device(name, &device, NULL, NULL, NULL,
1045                             NULL, NULL, NULL, NULL, NULL, NULL);
1046         if (r >= 0)
1047                 r = crypt_init(cd, device);
1048
1049         free(device);
1050         return r;
1051 }
1052
1053 static int _crypt_format_plain(struct crypt_device *cd,
1054                                const char *cipher,
1055                                const char *cipher_mode,
1056                                const char *uuid,
1057                                struct crypt_params_plain *params)
1058 {
1059         if (!cipher || !cipher_mode || !params || !params->hash) {
1060                 log_err(cd, _("Invalid plain crypt parameters.\n"));
1061                 return -EINVAL;
1062         }
1063
1064         if (cd->volume_key->keyLength > 1024) {
1065                 log_err(cd, _("Invalid key size.\n"));
1066                 return -EINVAL;
1067         }
1068
1069         cd->plain_cipher = strdup(cipher);
1070         cd->plain_cipher_mode = strdup(cipher_mode);
1071
1072         if (uuid)
1073                 cd->plain_uuid = strdup(uuid);
1074
1075         if (params->hash)
1076                 cd->plain_hdr.hash = strdup(params->hash);
1077
1078         cd->plain_hdr.offset = params->offset;
1079         cd->plain_hdr.skip = params->skip;
1080
1081         if ((params->hash && !cd->plain_hdr.hash) ||
1082             !cd->plain_cipher || !cd->plain_cipher_mode)
1083                 return -ENOMEM;
1084
1085         return 0;
1086 }
1087
1088 static int _crypt_format_luks1(struct crypt_device *cd,
1089                                const char *cipher,
1090                                const char *cipher_mode,
1091                                const char *uuid,
1092                                struct crypt_params_luks1 *params)
1093 {
1094         int r;
1095
1096         if (!cd->device) {
1097                 log_err(cd, _("Can't format LUKS without device.\n"));
1098                 return -EINVAL;
1099         }
1100
1101         r = LUKS_generate_phdr(&cd->hdr, cd->volume_key, cipher, cipher_mode,
1102                                (params && params->hash) ? params->hash : "sha1",
1103                                uuid, LUKS_STRIPES,
1104                                params ? params->data_alignment: DEFAULT_ALIGNMENT,
1105                                cd->iteration_time, &cd->PBKDF2_per_sec, cd);
1106         if(r < 0)
1107                 return r;
1108
1109         /* Wipe first 8 sectors - fs magic numbers etc. */
1110         r = wipe_device_header(cd->device, 8);
1111         if(r < 0) {
1112                 log_err(cd, _("Can't wipe header on device %s.\n"), cd->device);
1113                 return r;
1114         }
1115
1116         r = LUKS_write_phdr(cd->device, &cd->hdr, cd);
1117
1118         return r;
1119 }
1120
1121 int crypt_format(struct crypt_device *cd,
1122         const char *type,
1123         const char *cipher,
1124         const char *cipher_mode,
1125         const char *uuid,
1126         const char *volume_key,
1127         size_t volume_key_size,
1128         void *params)
1129 {
1130         int r;
1131
1132         log_dbg("Formatting device %s as type %s.", cd->device ?: "(none)", cd->type ?: "(none)");
1133
1134         if (!type)
1135                 return -EINVAL;
1136
1137         if (volume_key)
1138                 cd->volume_key = LUKS_alloc_masterkey(volume_key_size,
1139                                                       volume_key);
1140         else
1141                 cd->volume_key = LUKS_generate_masterkey(volume_key_size);
1142
1143         if(!cd->volume_key)
1144                 return -ENOMEM;
1145
1146         if (isPLAIN(type))
1147                 r = _crypt_format_plain(cd, cipher, cipher_mode,
1148                                         uuid, params);
1149         else if (isLUKS(type))
1150                 r = _crypt_format_luks1(cd, cipher, cipher_mode,
1151                                         uuid, params);
1152         else {
1153                 /* FIXME: allow plugins here? */
1154                 log_err(cd, _("Unknown crypt device type %s requested.\n"), type);
1155                 r = -EINVAL;
1156         }
1157
1158         if (!r && !(cd->type = strdup(type)))
1159                 r = -ENOMEM;
1160
1161         if (r < 0) {
1162                 LUKS_dealloc_masterkey(cd->volume_key);
1163                 cd->volume_key = NULL;
1164         }
1165
1166         return r;
1167 }
1168
1169 int crypt_load(struct crypt_device *cd,
1170                const char *requested_type,
1171                void *params)
1172 {
1173         struct luks_phdr hdr;
1174         int r;
1175
1176         log_dbg("Trying to load %s crypt type from device %s.",
1177                 requested_type ?: "any", cd->device ?: "(none)");
1178
1179         if (!cd->device)
1180                 return -EINVAL;
1181
1182         if (requested_type && !isLUKS(requested_type))
1183                 return -EINVAL;
1184
1185         /* Some hash functions need initialized gcrypt library */
1186         if (init_crypto()) {
1187                 log_err(cd, _("Cannot initialize crypto backend.\n"));
1188                 return -ENOSYS;
1189         }
1190
1191         r = LUKS_read_phdr(cd->device, &hdr, 1, cd);
1192
1193         if (!r) {
1194                 memcpy(&cd->hdr, &hdr, sizeof(hdr));
1195                 cd->type = strdup(requested_type);
1196                 if (!cd->type)
1197                         r = -ENOMEM;
1198         }
1199
1200         return r;
1201 }
1202
1203 int crypt_header_backup(struct crypt_device *cd,
1204                         const char *requested_type,
1205                         const char *backup_file)
1206 {
1207         if ((requested_type && !isLUKS(requested_type)) || !backup_file)
1208                 return -EINVAL;
1209
1210         log_dbg("Requested header backup of device %s (%s) to "
1211                 "file %s.", cd->device, requested_type, backup_file);
1212
1213         return LUKS_hdr_backup(backup_file, cd->device, &cd->hdr, cd);
1214 }
1215
1216 int crypt_header_restore(struct crypt_device *cd,
1217                          const char *requested_type,
1218                          const char *backup_file)
1219 {
1220         if (requested_type && !isLUKS(requested_type))
1221                 return -EINVAL;
1222
1223         log_dbg("Requested header restore to device %s (%s) from "
1224                 "file %s.", cd->device, requested_type, backup_file);
1225
1226         return LUKS_hdr_restore(backup_file, cd->device, &cd->hdr, cd);
1227 }
1228
1229 void crypt_free(struct crypt_device *cd)
1230 {
1231         if (cd) {
1232                 log_dbg("Releasing crypt device %s context.", cd->device);
1233
1234                 dm_exit();
1235                 if (cd->volume_key)
1236                         LUKS_dealloc_masterkey(cd->volume_key);
1237
1238                 free(cd->device);
1239                 free(cd->type);
1240
1241                 /* used in plain device only */
1242                 free((char*)cd->plain_hdr.hash);
1243                 free(cd->plain_cipher);
1244                 free(cd->plain_cipher_mode);
1245                 free(cd->plain_uuid);
1246
1247                 free(cd);
1248         }
1249 }
1250
1251 int crypt_suspend(struct crypt_device *cd,
1252                   const char *name)
1253 {
1254         crypt_status_info ci;
1255         int r, suspended = 0;
1256
1257         log_dbg("Suspending volume %s.", name);
1258
1259         ci = crypt_status(NULL, name);
1260         if (ci < CRYPT_ACTIVE) {
1261                 log_err(cd, _("Volume %s is not active.\n"), name);
1262                 return -EINVAL;
1263         }
1264
1265         if (!cd && dm_init(NULL, 1) < 0)
1266                 return -ENOSYS;
1267
1268         r = dm_query_device(name, NULL, NULL, NULL, NULL,
1269                             NULL, NULL, NULL, NULL, &suspended, NULL);
1270         if (r < 0)
1271                 goto out;
1272
1273         if (suspended) {
1274                 log_err(cd, _("Volume %s is already suspended.\n"), name);
1275                 r = -EINVAL;
1276                 goto out;
1277         }
1278
1279         r = dm_suspend_and_wipe_key(name);
1280         if (r)
1281                 log_err(cd, "Error during suspending device %s.\n", name);
1282 out:
1283         if (!cd)
1284                 dm_exit();
1285         return r;
1286 }
1287
1288 int crypt_resume_by_passphrase(struct crypt_device *cd,
1289                                const char *name,
1290                                int keyslot,
1291                                const char *passphrase,
1292                                size_t passphrase_size)
1293 {
1294         struct luks_masterkey *mk = NULL;
1295         int r, suspended = 0;
1296
1297         log_dbg("Resuming volume %s.", name);
1298
1299         if (!isLUKS(cd->type)) {
1300                 log_err(cd, _("This operation is supported only for LUKS device.\n"));
1301                 r = -EINVAL;
1302                 goto out;
1303         }
1304
1305         r = dm_query_device(name, NULL, NULL, NULL, NULL,
1306                             NULL, NULL, NULL, NULL, &suspended, NULL);
1307         if (r < 0)
1308                 return r;
1309
1310         if (!suspended) {
1311                 log_err(cd, _("Volume %s is not suspended.\n"), name);
1312                 return -EINVAL;
1313         }
1314
1315         if (passphrase) {
1316                 r = LUKS_open_key_with_hdr(cd->device, keyslot, passphrase,
1317                                            passphrase_size, &cd->hdr, &mk, cd);
1318         } else
1319                 r = volume_key_by_terminal_passphrase(cd, keyslot, &mk);
1320
1321         if (r >= 0) {
1322                 keyslot = r;
1323                 r = dm_resume_and_reinstate_key(name, mk->keyLength, mk->key);
1324                 if (r)
1325                         log_err(cd, "Error during resuming device %s.\n", name);
1326         } else
1327                 r = keyslot;
1328 out:
1329         LUKS_dealloc_masterkey(mk);
1330         return r < 0 ? r : keyslot;
1331 }
1332
1333 int crypt_resume_by_keyfile(struct crypt_device *cd,
1334                             const char *name,
1335                             int keyslot,
1336                             const char *keyfile,
1337                             size_t keyfile_size)
1338 {
1339         struct luks_masterkey *mk = NULL;
1340         char *passphrase_read = NULL;
1341         unsigned int passphrase_size_read;
1342         int r, suspended = 0;
1343
1344         log_dbg("Resuming volume %s.", name);
1345
1346         if (!isLUKS(cd->type)) {
1347                 log_err(cd, _("This operation is supported only for LUKS device.\n"));
1348                 r = -EINVAL;
1349                 goto out;
1350         }
1351
1352         r = dm_query_device(name, NULL, NULL, NULL, NULL,
1353                             NULL, NULL, NULL, NULL, &suspended, NULL);
1354         if (r < 0)
1355                 return r;
1356
1357         if (!suspended) {
1358                 log_err(cd, _("Volume %s is not suspended.\n"), name);
1359                 return -EINVAL;
1360         }
1361
1362         if (!keyfile)
1363                 return -EINVAL;
1364
1365         key_from_file(cd, _("Enter passphrase: "), &passphrase_read,
1366                       &passphrase_size_read, keyfile, keyfile_size);
1367
1368         if(!passphrase_read)
1369                 r = -EINVAL;
1370         else {
1371                 r = LUKS_open_key_with_hdr(cd->device, keyslot, passphrase_read,
1372                                            passphrase_size_read, &cd->hdr, &mk, cd);
1373                 safe_free(passphrase_read);
1374         }
1375
1376         if (r >= 0) {
1377                 keyslot = r;
1378                 r = dm_resume_and_reinstate_key(name, mk->keyLength, mk->key);
1379                 if (r)
1380                         log_err(cd, "Error during resuming device %s.\n", name);
1381         } else
1382                 r = keyslot;
1383 out:
1384         LUKS_dealloc_masterkey(mk);
1385         return r < 0 ? r : keyslot;
1386 }
1387
1388 // slot manipulation
1389 int crypt_keyslot_add_by_passphrase(struct crypt_device *cd,
1390         int keyslot, // -1 any
1391         const char *passphrase, // NULL -> terminal
1392         size_t passphrase_size,
1393         const char *new_passphrase, // NULL -> terminal
1394         size_t new_passphrase_size)
1395 {
1396         struct luks_masterkey *mk = NULL;
1397         char *password = NULL, *new_password = NULL;
1398         unsigned int passwordLen, new_passwordLen;
1399         int r;
1400
1401         log_dbg("Adding new keyslot, existing passphrase %sprovided,"
1402                 "new passphrase %sprovided.",
1403                 passphrase ? "" : "not ", new_passphrase  ? "" : "not ");
1404
1405         if (!isLUKS(cd->type)) {
1406                 log_err(cd, _("This operation is supported only for LUKS device.\n"));
1407                 return -EINVAL;
1408         }
1409
1410         r = keyslot_verify_or_find_empty(cd, &keyslot);
1411         if (r)
1412                 return r;
1413
1414         if (!LUKS_keyslot_active_count(&cd->hdr)) {
1415                 /* No slots used, try to use pre-generated key in header */
1416                 if (cd->volume_key) {
1417                         mk = LUKS_alloc_masterkey(cd->volume_key->keyLength, cd->volume_key->key);
1418                         r = mk ? 0 : -ENOMEM;
1419                 } else {
1420                         log_err(cd, _("Cannot add key slot, all slots disabled and no volume key provided.\n"));
1421                         return -EINVAL;
1422                 }
1423         } else if (passphrase) {
1424                 /* Passphrase provided, use it to unlock existing keyslot */
1425                 r = LUKS_open_key_with_hdr(cd->device, CRYPT_ANY_SLOT, passphrase,
1426                                            passphrase_size, &cd->hdr, &mk, cd);
1427         } else {
1428                 /* Passphrase not provided, ask first and use it to unlock existing keyslot */
1429                 key_from_terminal(cd, _("Enter any passphrase: "),
1430                                   &password, &passwordLen, 0);
1431                 if (!password) {
1432                         r = -EINVAL;
1433                         goto out;
1434                 }
1435
1436                 r = LUKS_open_key_with_hdr(cd->device, CRYPT_ANY_SLOT, password,
1437                                            passwordLen, &cd->hdr, &mk, cd);
1438                 safe_free(password);
1439         }
1440
1441         if(r < 0)
1442                 goto out;
1443
1444         if (new_passphrase) {
1445                 new_password = (char *)new_passphrase;
1446                 new_passwordLen = new_passphrase_size;
1447         } else {
1448                 key_from_terminal(cd, _("Enter new passphrase for key slot: "),
1449                                   &new_password, &new_passwordLen, 1);
1450                 if(!new_password) {
1451                         r = -EINVAL;
1452                         goto out;
1453                 }
1454         }
1455
1456         r = LUKS_set_key(cd->device, keyslot, new_password, new_passwordLen,
1457                          &cd->hdr, mk, cd->iteration_time, &cd->PBKDF2_per_sec, cd);
1458         if(r < 0) goto out;
1459
1460         r = 0;
1461 out:
1462         if (!new_passphrase)
1463                 safe_free(new_password);
1464         LUKS_dealloc_masterkey(mk);
1465         return r ?: keyslot;
1466 }
1467
1468 int crypt_keyslot_add_by_keyfile(struct crypt_device *cd,
1469         int keyslot,
1470         const char *keyfile,
1471         size_t keyfile_size,
1472         const char *new_keyfile,
1473         size_t new_keyfile_size)
1474 {
1475         struct luks_masterkey *mk=NULL;
1476         char *password=NULL; unsigned int passwordLen;
1477         char *new_password = NULL; unsigned int new_passwordLen;
1478         int r;
1479
1480         log_dbg("Adding new keyslot, existing keyfile %s, new keyfile %s.",
1481                 keyfile ?: "[none]", new_keyfile ?: "[none]");
1482
1483         if (!isLUKS(cd->type)) {
1484                 log_err(cd, _("This operation is supported only for LUKS device.\n"));
1485                 return -EINVAL;
1486         }
1487
1488         r = keyslot_verify_or_find_empty(cd, &keyslot);
1489         if (r)
1490                 return r;
1491
1492         if (!LUKS_keyslot_active_count(&cd->hdr)) {
1493                 /* No slots used, try to use pre-generated key in header */
1494                 if (cd->volume_key) {
1495                         mk = LUKS_alloc_masterkey(cd->volume_key->keyLength, cd->volume_key->key);
1496                         r = mk ? 0 : -ENOMEM;
1497                 } else {
1498                         log_err(cd, _("Cannot add key slot, all slots disabled and no volume key provided.\n"));
1499                         return -EINVAL;
1500                 }
1501         } else {
1502                 /* Read password from file of (if NULL) from terminal */
1503                 if (keyfile)
1504                         key_from_file(cd, _("Enter any passphrase: "), &password, &passwordLen,
1505                                       keyfile, keyfile_size);
1506                 else
1507                         key_from_terminal(cd, _("Enter any passphrase: "),
1508                                         &password, &passwordLen, 1);
1509
1510                 if (!password)
1511                         return -EINVAL;
1512
1513                 r = LUKS_open_key_with_hdr(cd->device, CRYPT_ANY_SLOT, password, passwordLen,
1514                                            &cd->hdr, &mk, cd);
1515                 safe_free(password);
1516         }
1517
1518         if(r < 0)
1519                 goto out;
1520
1521         if (new_keyfile)
1522                 key_from_file(cd, _("Enter new passphrase for key slot: "),
1523                               &new_password, &new_passwordLen, new_keyfile,
1524                               new_keyfile_size);
1525         else
1526                 key_from_terminal(cd, _("Enter new passphrase for key slot: "),
1527                                   &new_password, &new_passwordLen, 1);
1528
1529         if(!new_password) {
1530                 r = -EINVAL;
1531                 goto out;
1532         }
1533
1534         r = LUKS_set_key(cd->device, keyslot, new_password, new_passwordLen,
1535                          &cd->hdr, mk, cd->iteration_time, &cd->PBKDF2_per_sec, cd);
1536 out:
1537         safe_free(new_password);
1538         LUKS_dealloc_masterkey(mk);
1539         return r < 0 ? r : keyslot;
1540 }
1541
1542 int crypt_keyslot_add_by_volume_key(struct crypt_device *cd,
1543         int keyslot,
1544         const char *volume_key,
1545         size_t volume_key_size,
1546         const char *passphrase,
1547         size_t passphrase_size)
1548 {
1549         struct luks_masterkey *mk = NULL;
1550         int r = -EINVAL;
1551         char *new_password = NULL; unsigned int new_passwordLen;
1552
1553         log_dbg("Adding new keyslot %d using volume key.", keyslot);
1554
1555         if (!isLUKS(cd->type)) {
1556                 log_err(cd, _("This operation is supported only for LUKS device.\n"));
1557                 return -EINVAL;
1558         }
1559
1560         if (volume_key)
1561                 mk = LUKS_alloc_masterkey(volume_key_size, volume_key);
1562         else if (cd->volume_key)
1563                 mk = LUKS_alloc_masterkey(cd->volume_key->keyLength, cd->volume_key->key);
1564
1565         if (!mk)
1566                 return -ENOMEM;
1567
1568         r = LUKS_verify_master_key(&cd->hdr, mk);
1569         if (r < 0) {
1570                 log_err(cd, _("Volume key does not match the volume.\n"));
1571                 goto out;
1572         }
1573
1574         r = keyslot_verify_or_find_empty(cd, &keyslot);
1575         if (r)
1576                 goto out;
1577
1578         if (!passphrase) {
1579                 key_from_terminal(cd, _("Enter new passphrase for key slot: "),
1580                                   &new_password, &new_passwordLen, 1);
1581                 passphrase = new_password;
1582                 passphrase_size = new_passwordLen;
1583         }
1584
1585         r = LUKS_set_key(cd->device, keyslot, passphrase, passphrase_size,
1586                          &cd->hdr, mk, cd->iteration_time, &cd->PBKDF2_per_sec, cd);
1587 out:
1588         if (new_password)
1589                 safe_free(new_password);
1590         LUKS_dealloc_masterkey(mk);
1591         return r ?: keyslot;
1592 }
1593
1594 int crypt_keyslot_destroy(struct crypt_device *cd, int keyslot)
1595 {
1596         crypt_keyslot_info ki;
1597
1598         log_dbg("Destroying keyslot %d.", keyslot);
1599
1600         if (!isLUKS(cd->type)) {
1601                 log_err(cd, _("This operation is supported only for LUKS device.\n"));
1602                 return -EINVAL;
1603         }
1604
1605         ki = crypt_keyslot_status(cd, keyslot);
1606         if (ki == CRYPT_SLOT_INVALID) {
1607                 log_err(cd, _("Key slot %d is invalid.\n"), keyslot);
1608                 return -EINVAL;
1609         }
1610
1611         if (ki == CRYPT_SLOT_INACTIVE) {
1612                 log_err(cd, _("Key slot %d is not used.\n"), keyslot);
1613                 return -EINVAL;
1614         }
1615
1616         return LUKS_del_key(cd->device, keyslot, &cd->hdr, cd);
1617 }
1618
1619 // activation/deactivation of device mapping
1620 int crypt_activate_by_passphrase(struct crypt_device *cd,
1621         const char *name,
1622         int keyslot,
1623         const char *passphrase,
1624         size_t passphrase_size,
1625         uint32_t flags)
1626 {
1627         crypt_status_info ci;
1628         struct luks_masterkey *mk = NULL;
1629         char *prompt = NULL;
1630         int r;
1631
1632         log_dbg("%s volume %s [keyslot %d] using %spassphrase.",
1633                 name ? "Activating" : "Checking", name ?: "",
1634                 keyslot, passphrase ? "" : "[none] ");
1635
1636         if (!name)
1637                 return -EINVAL;
1638
1639         /* plain, use hashed passphrase */
1640         if (isPLAIN(cd->type))
1641                 return create_device_helper(cd, name, cd->plain_hdr.hash,
1642                         cd->plain_cipher, cd->plain_cipher_mode, NULL, passphrase, passphrase_size,
1643                         cd->volume_key->keyLength, 0, cd->plain_hdr.skip,
1644                         cd->plain_hdr.offset, cd->plain_uuid, flags & CRYPT_ACTIVATE_READONLY, 0, 0);
1645
1646         if (name) {
1647                 ci = crypt_status(NULL, name);
1648                 if (ci == CRYPT_INVALID)
1649                         return -EINVAL;
1650                 else if (ci >= CRYPT_ACTIVE) {
1651                         log_err(cd, _("Device %s already exists.\n"), name);
1652                         return -EEXIST;
1653                 }
1654         }
1655
1656         if(asprintf(&prompt, _("Enter passphrase for %s: "), cd->device) < 0)
1657                 return -ENOMEM;
1658
1659         /* provided passphrase, do not retry */
1660         if (passphrase) {
1661                 r = LUKS_open_key_with_hdr(cd->device, keyslot, passphrase,
1662                                            passphrase_size, &cd->hdr, &mk, cd);
1663         } else
1664                 r = volume_key_by_terminal_passphrase(cd, keyslot, &mk);
1665
1666         if (r >= 0) {
1667                 keyslot = r;
1668                 if (name)
1669                         r = open_from_hdr_and_mk(cd, mk, name, flags);
1670         }
1671
1672         LUKS_dealloc_masterkey(mk);
1673         free(prompt);
1674
1675         return r < 0  ? r : keyslot;
1676 }
1677
1678 int crypt_activate_by_keyfile(struct crypt_device *cd,
1679         const char *name,
1680         int keyslot,
1681         const char *keyfile,
1682         size_t keyfile_size,
1683         uint32_t flags)
1684 {
1685         crypt_status_info ci;
1686         struct luks_masterkey *mk = NULL;
1687         char *passphrase_read = NULL;
1688         unsigned int passphrase_size_read;
1689         int r;
1690
1691         log_dbg("Activating volume %s [keyslot %d] using keyfile %s.",
1692                 name, keyslot, keyfile ?: "[none]");
1693
1694         if (!isLUKS(cd->type)) {
1695                 log_err(cd, _("This operation is supported only for LUKS device.\n"));
1696                 return -EINVAL;
1697         }
1698
1699         if (name) {
1700                 ci = crypt_status(NULL, name);
1701                 if (ci == CRYPT_INVALID)
1702                         return -EINVAL;
1703                 else if (ci >= CRYPT_ACTIVE) {
1704                         log_err(cd, _("Device %s already exists.\n"), name);
1705                         return -EEXIST;
1706                 }
1707         }
1708
1709         if (!keyfile)
1710                 return -EINVAL;
1711
1712         key_from_file(cd, _("Enter passphrase: "), &passphrase_read,
1713                       &passphrase_size_read, keyfile, keyfile_size);
1714         if(!passphrase_read)
1715                 r = -EINVAL;
1716         else {
1717                 r = LUKS_open_key_with_hdr(cd->device, keyslot, passphrase_read,
1718                                            passphrase_size_read, &cd->hdr, &mk, cd);
1719                 safe_free(passphrase_read);
1720         }
1721
1722         if (r >= 0) {
1723                 keyslot = r;
1724                 r = open_from_hdr_and_mk(cd, mk, name, flags);
1725         }
1726
1727         LUKS_dealloc_masterkey(mk);
1728
1729         return r < 0 ? r : keyslot;
1730 }
1731
1732 int crypt_activate_by_volume_key(struct crypt_device *cd,
1733         const char *name,
1734         const char *volume_key,
1735         size_t volume_key_size,
1736         uint32_t flags)
1737 {
1738         crypt_status_info ci;
1739         struct luks_masterkey *mk;
1740         int r;
1741
1742         log_dbg("Activating volume %s by volume key.", name);
1743
1744         /* use key directly, no hash */
1745         if (isPLAIN(cd->type))
1746                 return create_device_helper(cd, name, NULL,
1747                         cd->plain_cipher, cd->plain_cipher_mode, NULL, volume_key, volume_key_size,
1748                         cd->volume_key->keyLength, 0, cd->plain_hdr.skip,
1749                         cd->plain_hdr.offset, cd->plain_uuid, flags & CRYPT_ACTIVATE_READONLY, 0, 0);
1750
1751         if (!isLUKS(cd->type)) {
1752                 log_err(cd, _("This operation is supported only for LUKS device.\n"));
1753                 return -EINVAL;
1754         }
1755
1756         if (name) {
1757                 ci = crypt_status(NULL, name);
1758                 if (ci == CRYPT_INVALID)
1759                         return -EINVAL;
1760                 else if (ci >= CRYPT_ACTIVE) {
1761                         log_err(cd, _("Device %s already exists.\n"), name);
1762                         return -EEXIST;
1763                 }
1764         }
1765
1766         mk = LUKS_alloc_masterkey(volume_key_size, volume_key);
1767         if (!mk)
1768                 return -ENOMEM;
1769         r = LUKS_verify_master_key(&cd->hdr, mk);
1770
1771         if (r == -EPERM)
1772                 log_err(cd, _("Volume key does not match the volume.\n"));
1773
1774         if (!r && name)
1775                 r = open_from_hdr_and_mk(cd, mk, name, flags);
1776
1777         LUKS_dealloc_masterkey(mk);
1778
1779         return r;
1780 }
1781
1782 int crypt_deactivate(struct crypt_device *cd, const char *name)
1783 {
1784         int r;
1785
1786         if (!name)
1787                 return -EINVAL;
1788
1789         log_dbg("Deactivating volume %s.", name);
1790
1791         if (!cd && dm_init(NULL, 1) < 0)
1792                 return -ENOSYS;
1793
1794         switch (crypt_status(cd, name)) {
1795                 case CRYPT_ACTIVE:
1796                         r = dm_remove_device(name, 0, 0);
1797                         break;
1798                 case CRYPT_BUSY:
1799                         log_err(cd, _("Device %s is busy.\n"), name);
1800                         r = -EBUSY;
1801                         break;
1802                 case CRYPT_INACTIVE:
1803                         log_err(cd, _("Device %s is not active.\n"), name);
1804                         r = -ENODEV;
1805                         break;
1806                 default:
1807                         log_err(cd, _("Invalid device %s.\n"), name);
1808                         r = -EINVAL;
1809         }
1810
1811         if (!cd)
1812                 dm_exit();
1813
1814         return r;
1815 }
1816
1817 // misc helper functions
1818 int crypt_volume_key_get(struct crypt_device *cd,
1819         int keyslot,
1820         char *volume_key,
1821         size_t *volume_key_size,
1822         const char *passphrase,
1823         size_t passphrase_size)
1824 {
1825         struct luks_masterkey *mk;
1826         char *processed_key = NULL;
1827         int r, key_len;
1828
1829         key_len = crypt_get_volume_key_size(cd);
1830         if (key_len > *volume_key_size) {
1831                 log_err(cd, _("Volume key buffer too small.\n"));
1832                 return -ENOMEM;
1833         }
1834
1835         if (isPLAIN(cd->type)) {
1836                 processed_key = process_key(cd, cd->plain_hdr.hash, NULL, key_len,
1837                                             passphrase, passphrase_size);
1838                 if (!processed_key) {
1839                         log_err(cd, _("Cannot retrieve volume key for plain device.\n"));
1840                         return -EINVAL;
1841                 }
1842                 memcpy(volume_key, processed_key, key_len);
1843                 *volume_key_size = key_len;
1844                 safe_free(processed_key);
1845                 return 0;
1846         }
1847
1848         if (isLUKS(cd->type)) {
1849                 r = LUKS_open_key_with_hdr(cd->device, keyslot, passphrase,
1850                                         passphrase_size, &cd->hdr, &mk, cd);
1851
1852                 if (r >= 0) {
1853                         memcpy(volume_key, mk->key, mk->keyLength);
1854                         *volume_key_size = mk->keyLength;
1855                 }
1856
1857                 LUKS_dealloc_masterkey(mk);
1858                 return r;
1859         }
1860
1861         log_err(cd, _("This operation is not supported for %s crypt device.\n"), cd->type ?: "(none)");
1862         return -EINVAL;
1863 }
1864
1865 int crypt_volume_key_verify(struct crypt_device *cd,
1866         const char *volume_key,
1867         size_t volume_key_size)
1868 {
1869         struct luks_masterkey *mk;
1870         int r;
1871
1872         if (!isLUKS(cd->type)) {
1873                 log_err(cd, _("This operation is supported only for LUKS device.\n"));
1874                 return -EINVAL;
1875         }
1876
1877         mk = LUKS_alloc_masterkey(volume_key_size, volume_key);
1878         if (!mk)
1879                 return -ENOMEM;
1880
1881         r = LUKS_verify_master_key(&cd->hdr, mk);
1882
1883         if (r == -EPERM)
1884                 log_err(cd, _("Volume key does not match the volume.\n"));
1885
1886         LUKS_dealloc_masterkey(mk);
1887
1888         return r;
1889 }
1890
1891 void crypt_set_timeout(struct crypt_device *cd, uint64_t timeout_sec)
1892 {
1893         log_dbg("Timeout set to %" PRIu64 " miliseconds.", timeout_sec);
1894         cd->timeout = timeout_sec;
1895 }
1896
1897 void crypt_set_password_retry(struct crypt_device *cd, int tries)
1898 {
1899         log_dbg("Password retry count set to %d.", tries);
1900         cd->tries = tries;
1901 }
1902
1903 void crypt_set_iterarion_time(struct crypt_device *cd, uint64_t iteration_time_ms)
1904 {
1905         log_dbg("Iteration time set to %" PRIu64 " miliseconds.", iteration_time_ms);
1906         cd->iteration_time = iteration_time_ms;
1907 }
1908
1909 void crypt_set_password_verify(struct crypt_device *cd, int password_verify)
1910 {
1911         log_dbg("Password verification %s.", password_verify ? "enabled" : "disabled");
1912         cd->password_verify = password_verify ? 1 : 0;
1913 }
1914
1915 int crypt_memory_lock(struct crypt_device *cd, int lock)
1916 {
1917         return lock ? crypt_memlock_inc(cd) : crypt_memlock_dec(cd);
1918 }
1919
1920 // reporting
1921 crypt_status_info crypt_status(struct crypt_device *cd, const char *name)
1922 {
1923         int r;
1924
1925         if (!cd && dm_init(NULL, 1) < 0)
1926                 return CRYPT_INVALID;
1927
1928         r = dm_status_device(name);
1929
1930         if (!cd)
1931                 dm_exit();
1932
1933         if (r < 0 && r != -ENODEV)
1934                 return CRYPT_INVALID;
1935
1936         if (r == 0)
1937                 return CRYPT_ACTIVE;
1938
1939         if (r > 0)
1940                 return CRYPT_BUSY;
1941
1942         return CRYPT_INACTIVE;
1943 }
1944
1945 static void hexprintICB(struct crypt_device *cd, char *d, int n)
1946 {
1947         int i;
1948         for(i = 0; i < n; i++)
1949                 log_std(cd, "%02hhx ", (char)d[i]);
1950 }
1951
1952 int crypt_dump(struct crypt_device *cd)
1953 {
1954         int i;
1955         if (!isLUKS(cd->type)) { //FIXME
1956                 log_err(cd, _("This operation is supported only for LUKS device.\n"));
1957                 return -EINVAL;
1958         }
1959
1960         log_std(cd, "LUKS header information for %s\n\n", cd->device);
1961         log_std(cd, "Version:       \t%d\n", cd->hdr.version);
1962         log_std(cd, "Cipher name:   \t%s\n", cd->hdr.cipherName);
1963         log_std(cd, "Cipher mode:   \t%s\n", cd->hdr.cipherMode);
1964         log_std(cd, "Hash spec:     \t%s\n", cd->hdr.hashSpec);
1965         log_std(cd, "Payload offset:\t%d\n", cd->hdr.payloadOffset);
1966         log_std(cd, "MK bits:       \t%d\n", cd->hdr.keyBytes * 8);
1967         log_std(cd, "MK digest:     \t");
1968         hexprintICB(cd, cd->hdr.mkDigest, LUKS_DIGESTSIZE);
1969         log_std(cd, "\n");
1970         log_std(cd, "MK salt:       \t");
1971         hexprintICB(cd, cd->hdr.mkDigestSalt, LUKS_SALTSIZE/2);
1972         log_std(cd, "\n               \t");
1973         hexprintICB(cd, cd->hdr.mkDigestSalt+LUKS_SALTSIZE/2, LUKS_SALTSIZE/2);
1974         log_std(cd, "\n");
1975         log_std(cd, "MK iterations: \t%d\n", cd->hdr.mkDigestIterations);
1976         log_std(cd, "UUID:          \t%s\n\n", cd->hdr.uuid);
1977         for(i = 0; i < LUKS_NUMKEYS; i++) {
1978                 if(cd->hdr.keyblock[i].active == LUKS_KEY_ENABLED) {
1979                         log_std(cd, "Key Slot %d: ENABLED\n",i);
1980                         log_std(cd, "\tIterations:         \t%d\n",
1981                                 cd->hdr.keyblock[i].passwordIterations);
1982                         log_std(cd, "\tSalt:               \t");
1983                         hexprintICB(cd, cd->hdr.keyblock[i].passwordSalt,
1984                                     LUKS_SALTSIZE/2);
1985                         log_std(cd, "\n\t                      \t");
1986                         hexprintICB(cd, cd->hdr.keyblock[i].passwordSalt +
1987                                     LUKS_SALTSIZE/2, LUKS_SALTSIZE/2);
1988                         log_std(cd, "\n");
1989
1990                         log_std(cd, "\tKey material offset:\t%d\n",
1991                                 cd->hdr.keyblock[i].keyMaterialOffset);
1992                         log_std(cd, "\tAF stripes:            \t%d\n",
1993                                 cd->hdr.keyblock[i].stripes);
1994                 }
1995                 else
1996                         log_std(cd, "Key Slot %d: DISABLED\n", i);
1997         }
1998         return 0;
1999 }
2000
2001 const char *crypt_get_cipher(struct crypt_device *cd)
2002 {
2003         if (isPLAIN(cd->type))
2004                 return cd->plain_cipher;
2005
2006         if (isLUKS(cd->type))
2007                 return cd->hdr.cipherName;
2008
2009         return NULL;
2010 }
2011
2012 const char *crypt_get_cipher_mode(struct crypt_device *cd)
2013 {
2014         if (isPLAIN(cd->type))
2015                 return cd->plain_cipher_mode;
2016
2017         if (isLUKS(cd->type))
2018                 return cd->hdr.cipherMode;
2019
2020         return NULL;
2021 }
2022
2023 const char *crypt_get_uuid(struct crypt_device *cd)
2024 {
2025         if (isLUKS(cd->type))
2026                 return cd->hdr.uuid;
2027
2028         return NULL;
2029 }
2030
2031 int crypt_get_volume_key_size(struct crypt_device *cd)
2032 {
2033         if (isPLAIN(cd->type))
2034                 return cd->volume_key->keyLength;
2035
2036         if (isLUKS(cd->type))
2037                 return cd->hdr.keyBytes;
2038
2039         return 0;
2040 }
2041
2042 uint64_t crypt_get_data_offset(struct crypt_device *cd)
2043 {
2044         if (isPLAIN(cd->type))
2045                 return cd->plain_hdr.offset;
2046
2047         if (isLUKS(cd->type))
2048                 return cd->hdr.payloadOffset;
2049
2050         return 0;
2051 }
2052
2053 crypt_keyslot_info crypt_keyslot_status(struct crypt_device *cd, int keyslot)
2054 {
2055         if (!isLUKS(cd->type)) {
2056                 log_err(cd, _("This operation is supported only for LUKS device.\n"));
2057                 return CRYPT_SLOT_INVALID;
2058         }
2059
2060         return LUKS_keyslot_info(&cd->hdr, keyslot);
2061 }