crypto/evp/evp_enc.c

   1 /*
   2  * Copyright 1995-2018 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the Apache License 2.0 (the "License").  You may not use
   5  * this file except in compliance with the License.  You can obtain a copy
   6  * in the file LICENSE in the source distribution or at
   7  * https://www.openssl.org/source/license.html
   8  */
   9
  10 #include <stdio.h>
  11 #include <assert.h>
  12 #include "internal/cryptlib.h"
  13 #include <openssl/evp.h>
  14 #include <openssl/err.h>
  15 #include <openssl/rand.h>
  16 #include <openssl/rand_drbg.h>
  17 #include <openssl/engine.h>
  18 #include <openssl/params.h>
  19 #include <openssl/core_names.h>
  20 #include "internal/evp_int.h"
  21 #include "internal/provider.h"
  22 #include "evp_locl.h"
  23
  24 int EVP_CIPHER_CTX_reset(EVP_CIPHER_CTX *ctx)
  25 {
  26     if (ctx == NULL)
  27         return 1;
  28
  29     if (ctx->cipher == NULL || ctx->cipher->prov == NULL)
  30         goto legacy;
  31
  32     if (ctx->provctx != NULL) {
  33         if (ctx->cipher->freectx != NULL)
  34             ctx->cipher->freectx(ctx->provctx);
  35         ctx->provctx = NULL;
  36     }
  37     if (ctx->fetched_cipher != NULL)
  38         EVP_CIPHER_meth_free(ctx->fetched_cipher);
  39     memset(ctx, 0, sizeof(*ctx));
  40
  41     return 1;
  42
  43     /* TODO(3.0): Remove legacy code below */
  44  legacy:
  45
  46     if (ctx->cipher != NULL) {
  47         if (ctx->cipher->cleanup && !ctx->cipher->cleanup(ctx))
  48             return 0;
  49         /* Cleanse cipher context data */
  50         if (ctx->cipher_data && ctx->cipher->ctx_size)
  51             OPENSSL_cleanse(ctx->cipher_data, ctx->cipher->ctx_size);
  52     }
  53     OPENSSL_free(ctx->cipher_data);
  54 #ifndef OPENSSL_NO_ENGINE
  55     ENGINE_finish(ctx->engine);
  56 #endif
  57     memset(ctx, 0, sizeof(*ctx));
  58     return 1;
  59 }
  60
  61 EVP_CIPHER_CTX *EVP_CIPHER_CTX_new(void)
  62 {
  63     return OPENSSL_zalloc(sizeof(EVP_CIPHER_CTX));
  64 }
  65
  66 void EVP_CIPHER_CTX_free(EVP_CIPHER_CTX *ctx)
  67 {
  68     EVP_CIPHER_CTX_reset(ctx);
  69     OPENSSL_free(ctx);
  70 }
  71
  72 int EVP_CipherInit(EVP_CIPHER_CTX *ctx, const EVP_CIPHER *cipher,
  73                    const unsigned char *key, const unsigned char *iv, int enc)
  74 {
  75     if (cipher != NULL)
  76         EVP_CIPHER_CTX_reset(ctx);
  77     return EVP_CipherInit_ex(ctx, cipher, NULL, key, iv, enc);
  78 }
  79
  80 int EVP_CipherInit_ex(EVP_CIPHER_CTX *ctx, const EVP_CIPHER *cipher,
  81                       ENGINE *impl, const unsigned char *key,
  82                       const unsigned char *iv, int enc)
  83 {
  84     EVP_CIPHER *provciph = NULL;
  85     ENGINE *tmpimpl = NULL;
  86     const EVP_CIPHER *tmpcipher;
  87
  88     /*
  89      * enc == 1 means we are encrypting.
  90      * enc == 0 means we are decrypting.
  91      * enc == -1 means, use the previously initialised value for encrypt/decrypt
  92      */
  93     if (enc == -1) {
  94         enc = ctx->encrypt;
  95     } else {
  96         if (enc)
  97             enc = 1;
  98         ctx->encrypt = enc;
  99     }
 100
 101     if (cipher == NULL && ctx->cipher == NULL) {
 102         EVPerr(EVP_F_EVP_CIPHERINIT_EX, EVP_R_NO_CIPHER_SET);
 103         return 0;
 104     }
 105
 106     /* TODO(3.0): Legacy work around code below. Remove this */
 107
 108 #ifndef OPENSSL_NO_ENGINE
 109     /*
 110      * Whether it's nice or not, "Inits" can be used on "Final"'d contexts so
 111      * this context may already have an ENGINE! Try to avoid releasing the
 112      * previous handle, re-querying for an ENGINE, and having a
 113      * reinitialisation, when it may all be unnecessary.
 114      */
 115     if (ctx->engine && ctx->cipher
 116         && (cipher == NULL || cipher->nid == ctx->cipher->nid))
 117         goto skip_to_init;
 118
 119     if (cipher != NULL && impl == NULL) {
 120          /* Ask if an ENGINE is reserved for this job */
 121         tmpimpl = ENGINE_get_cipher_engine(cipher->nid);
 122     }
 123 #endif
 124
 125     /*
 126      * If there are engines involved then we should use legacy handling for now.
 127      */
 128     if (ctx->engine != NULL
 129             || impl != NULL
 130             || tmpimpl != NULL) {
 131         if (ctx->cipher == ctx->fetched_cipher)
 132             ctx->cipher = NULL;
 133         EVP_CIPHER_meth_free(ctx->fetched_cipher);
 134         ctx->fetched_cipher = NULL;
 135         goto legacy;
 136     }
 137
 138     tmpcipher = (cipher == NULL) ? ctx->cipher : cipher;
 139
 140     if (tmpcipher->prov == NULL) {
 141         switch(tmpcipher->nid) {
 142         case NID_aes_256_ecb:
 143         case NID_aes_192_ecb:
 144         case NID_aes_128_ecb:
 145         case NID_aes_256_cbc:
 146         case NID_aes_192_cbc:
 147         case NID_aes_128_cbc:
 148         case NID_aes_256_ofb128:
 149         case NID_aes_192_ofb128:
 150         case NID_aes_128_ofb128:
 151         case NID_aes_256_cfb128:
 152         case NID_aes_192_cfb128:
 153         case NID_aes_128_cfb128:
 154         case NID_aes_256_cfb1:
 155         case NID_aes_192_cfb1:
 156         case NID_aes_128_cfb1:
 157         case NID_aes_256_cfb8:
 158         case NID_aes_192_cfb8:
 159         case NID_aes_128_cfb8:
 160         case NID_aes_256_ctr:
 161         case NID_aes_192_ctr:
 162         case NID_aes_128_ctr:
 163             break;
 164         default:
 165             goto legacy;
 166         }
 167     }
 168
 169     /*
 170      * Ensure a context left lying around from last time is cleared
 171      * (legacy code)
 172      */
 173     if (cipher != NULL && ctx->cipher != NULL) {
 174         OPENSSL_clear_free(ctx->cipher_data, ctx->cipher->ctx_size);
 175         ctx->cipher_data = NULL;
 176     }
 177
 178
 179     /* TODO(3.0): Start of non-legacy code below */
 180
 181     /* Ensure a context left lying around from last time is cleared */
 182     if (cipher != NULL && ctx->cipher != NULL) {
 183         unsigned long flags = ctx->flags;
 184
 185         EVP_CIPHER_CTX_reset(ctx);
 186         /* Restore encrypt and flags */
 187         ctx->encrypt = enc;
 188         ctx->flags = flags;
 189     }
 190
 191     if (cipher != NULL)
 192         ctx->cipher = cipher;
 193     else
 194         cipher = ctx->cipher;
 195
 196     if (cipher->prov == NULL) {
 197         provciph = EVP_CIPHER_fetch(NULL, OBJ_nid2sn(cipher->nid), "");
 198         if (provciph == NULL) {
 199             EVPerr(EVP_F_EVP_CIPHERINIT_EX, EVP_R_INITIALIZATION_ERROR);
 200             return 0;
 201         }
 202         cipher = provciph;
 203         EVP_CIPHER_meth_free(ctx->fetched_cipher);
 204         ctx->fetched_cipher = provciph;
 205     }
 206
 207     ctx->cipher = cipher;
 208     if (ctx->provctx == NULL) {
 209         ctx->provctx = ctx->cipher->newctx(ossl_provider_ctx(cipher->prov));
 210         if (ctx->provctx == NULL) {
 211             EVPerr(EVP_F_EVP_CIPHERINIT_EX, EVP_R_INITIALIZATION_ERROR);
 212             return 0;
 213         }
 214     }
 215
 216     if ((ctx->flags & EVP_CIPH_NO_PADDING) != 0) {
 217         /*
 218          * If this ctx was already set up for no padding then we need to tell
 219          * the new cipher about it.
 220          */
 221         if (!EVP_CIPHER_CTX_set_padding(ctx, 0))
 222             return 0;
 223     }
 224
 225     switch (EVP_CIPHER_mode(ctx->cipher)) {
 226     case EVP_CIPH_CFB_MODE:
 227     case EVP_CIPH_OFB_MODE:
 228     case EVP_CIPH_CBC_MODE:
 229         /* For these modes we remember the original IV for later use */
 230         if (!ossl_assert(EVP_CIPHER_CTX_iv_length(ctx) <= (int)sizeof(ctx->oiv))) {
 231             EVPerr(EVP_F_EVP_CIPHERINIT_EX, EVP_R_INITIALIZATION_ERROR);
 232             return 0;
 233         }
 234         if (iv != NULL)
 235             memcpy(ctx->oiv, iv, EVP_CIPHER_CTX_iv_length(ctx));
 236     }
 237
 238     if (enc) {
 239         if (ctx->cipher->einit == NULL) {
 240             EVPerr(EVP_F_EVP_CIPHERINIT_EX, EVP_R_INITIALIZATION_ERROR);
 241             return 0;
 242         }
 243
 244         return ctx->cipher->einit(ctx->provctx,
 245                                   key,
 246                                   key == NULL ? 0
 247                                               : EVP_CIPHER_CTX_key_length(ctx),
 248                                   iv,
 249                                   iv == NULL ? 0
 250                                              : EVP_CIPHER_CTX_iv_length(ctx));
 251     }
 252
 253     if (ctx->cipher->dinit == NULL) {
 254         EVPerr(EVP_F_EVP_CIPHERINIT_EX, EVP_R_INITIALIZATION_ERROR);
 255         return 0;
 256     }
 257
 258     return ctx->cipher->dinit(ctx->provctx,
 259                               key,
 260                               key == NULL ? 0
 261                                           : EVP_CIPHER_CTX_key_length(ctx),
 262                               iv,
 263                               iv == NULL ? 0
 264                                          : EVP_CIPHER_CTX_iv_length(ctx));
 265
 266     /* TODO(3.0): Remove legacy code below */
 267  legacy:
 268
 269     if (cipher != NULL) {
 270         /*
 271          * Ensure a context left lying around from last time is cleared (we
 272          * previously attempted to avoid this if the same ENGINE and
 273          * EVP_CIPHER could be used).
 274          */
 275         if (ctx->cipher) {
 276             unsigned long flags = ctx->flags;
 277             EVP_CIPHER_CTX_reset(ctx);
 278             /* Restore encrypt and flags */
 279             ctx->encrypt = enc;
 280             ctx->flags = flags;
 281         }
 282 #ifndef OPENSSL_NO_ENGINE
 283         if (impl != NULL) {
 284             if (!ENGINE_init(impl)) {
 285                 EVPerr(EVP_F_EVP_CIPHERINIT_EX, EVP_R_INITIALIZATION_ERROR);
 286                 return 0;
 287             }
 288         } else {
 289             impl = tmpimpl;
 290         }
 291         if (impl != NULL) {
 292             /* There's an ENGINE for this job ... (apparently) */
 293             const EVP_CIPHER *c = ENGINE_get_cipher(impl, cipher->nid);
 294
 295             if (c == NULL) {
 296                 /*
 297                  * One positive side-effect of US's export control history,
 298                  * is that we should at least be able to avoid using US
 299                  * misspellings of "initialisation"?
 300                  */
 301                 EVPerr(EVP_F_EVP_CIPHERINIT_EX, EVP_R_INITIALIZATION_ERROR);
 302                 return 0;
 303             }
 304             /* We'll use the ENGINE's private cipher definition */
 305             cipher = c;
 306             /*
 307              * Store the ENGINE functional reference so we know 'cipher' came
 308              * from an ENGINE and we need to release it when done.
 309              */
 310             ctx->engine = impl;
 311         } else {
 312             ctx->engine = NULL;
 313         }
 314 #endif
 315
 316         ctx->cipher = cipher;
 317         if (ctx->cipher->ctx_size) {
 318             ctx->cipher_data = OPENSSL_zalloc(ctx->cipher->ctx_size);
 319             if (ctx->cipher_data == NULL) {
 320                 ctx->cipher = NULL;
 321                 EVPerr(EVP_F_EVP_CIPHERINIT_EX, ERR_R_MALLOC_FAILURE);
 322                 return 0;
 323             }
 324         } else {
 325             ctx->cipher_data = NULL;
 326         }
 327         ctx->key_len = cipher->key_len;
 328         /* Preserve wrap enable flag, zero everything else */
 329         ctx->flags &= EVP_CIPHER_CTX_FLAG_WRAP_ALLOW;
 330         if (ctx->cipher->flags & EVP_CIPH_CTRL_INIT) {
 331             if (!EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_INIT, 0, NULL)) {
 332                 ctx->cipher = NULL;
 333                 EVPerr(EVP_F_EVP_CIPHERINIT_EX, EVP_R_INITIALIZATION_ERROR);
 334                 return 0;
 335             }
 336         }
 337     }
 338 #ifndef OPENSSL_NO_ENGINE
 339  skip_to_init:
 340 #endif
 341     /* we assume block size is a power of 2 in *cryptUpdate */
 342     OPENSSL_assert(ctx->cipher->block_size == 1
 343                    || ctx->cipher->block_size == 8
 344                    || ctx->cipher->block_size == 16);
 345
 346     if (!(ctx->flags & EVP_CIPHER_CTX_FLAG_WRAP_ALLOW)
 347         && EVP_CIPHER_CTX_mode(ctx) == EVP_CIPH_WRAP_MODE) {
 348         EVPerr(EVP_F_EVP_CIPHERINIT_EX, EVP_R_WRAP_MODE_NOT_ALLOWED);
 349         return 0;
 350     }
 351
 352     if (!(EVP_CIPHER_flags(EVP_CIPHER_CTX_cipher(ctx)) & EVP_CIPH_CUSTOM_IV)) {
 353         switch (EVP_CIPHER_CTX_mode(ctx)) {
 354
 355         case EVP_CIPH_STREAM_CIPHER:
 356         case EVP_CIPH_ECB_MODE:
 357             break;
 358
 359         case EVP_CIPH_CFB_MODE:
 360         case EVP_CIPH_OFB_MODE:
 361
 362             ctx->num = 0;
 363             /* fall-through */
 364
 365         case EVP_CIPH_CBC_MODE:
 366
 367             OPENSSL_assert(EVP_CIPHER_CTX_iv_length(ctx) <=
 368                            (int)sizeof(ctx->iv));
 369             if (iv)
 370                 memcpy(ctx->oiv, iv, EVP_CIPHER_CTX_iv_length(ctx));
 371             memcpy(ctx->iv, ctx->oiv, EVP_CIPHER_CTX_iv_length(ctx));
 372             break;
 373
 374         case EVP_CIPH_CTR_MODE:
 375             ctx->num = 0;
 376             /* Don't reuse IV for CTR mode */
 377             if (iv)
 378                 memcpy(ctx->iv, iv, EVP_CIPHER_CTX_iv_length(ctx));
 379             break;
 380
 381         default:
 382             return 0;
 383         }
 384     }
 385
 386     if (key || (ctx->cipher->flags & EVP_CIPH_ALWAYS_CALL_INIT)) {
 387         if (!ctx->cipher->init(ctx, key, iv, enc))
 388             return 0;
 389     }
 390     ctx->buf_len = 0;
 391     ctx->final_used = 0;
 392     ctx->block_mask = ctx->cipher->block_size - 1;
 393     return 1;
 394 }
 395
 396 int EVP_CipherUpdate(EVP_CIPHER_CTX *ctx, unsigned char *out, int *outl,
 397                      const unsigned char *in, int inl)
 398 {
 399     if (ctx->encrypt)
 400         return EVP_EncryptUpdate(ctx, out, outl, in, inl);
 401     else
 402         return EVP_DecryptUpdate(ctx, out, outl, in, inl);
 403 }
 404
 405 int EVP_CipherFinal_ex(EVP_CIPHER_CTX *ctx, unsigned char *out, int *outl)
 406 {
 407     if (ctx->encrypt)
 408         return EVP_EncryptFinal_ex(ctx, out, outl);
 409     else
 410         return EVP_DecryptFinal_ex(ctx, out, outl);
 411 }
 412
 413 int EVP_CipherFinal(EVP_CIPHER_CTX *ctx, unsigned char *out, int *outl)
 414 {
 415     if (ctx->encrypt)
 416         return EVP_EncryptFinal(ctx, out, outl);
 417     else
 418         return EVP_DecryptFinal(ctx, out, outl);
 419 }
 420
 421 int EVP_EncryptInit(EVP_CIPHER_CTX *ctx, const EVP_CIPHER *cipher,
 422                     const unsigned char *key, const unsigned char *iv)
 423 {
 424     return EVP_CipherInit(ctx, cipher, key, iv, 1);
 425 }
 426
 427 int EVP_EncryptInit_ex(EVP_CIPHER_CTX *ctx, const EVP_CIPHER *cipher,
 428                        ENGINE *impl, const unsigned char *key,
 429                        const unsigned char *iv)
 430 {
 431     return EVP_CipherInit_ex(ctx, cipher, impl, key, iv, 1);
 432 }
 433
 434 int EVP_DecryptInit(EVP_CIPHER_CTX *ctx, const EVP_CIPHER *cipher,
 435                     const unsigned char *key, const unsigned char *iv)
 436 {
 437     return EVP_CipherInit(ctx, cipher, key, iv, 0);
 438 }
 439
 440 int EVP_DecryptInit_ex(EVP_CIPHER_CTX *ctx, const EVP_CIPHER *cipher,
 441                        ENGINE *impl, const unsigned char *key,
 442                        const unsigned char *iv)
 443 {
 444     return EVP_CipherInit_ex(ctx, cipher, impl, key, iv, 0);
 445 }
 446
 447 /*
 448  * According to the letter of standard difference between pointers
 449  * is specified to be valid only within same object. This makes
 450  * it formally challenging to determine if input and output buffers
 451  * are not partially overlapping with standard pointer arithmetic.
 452  */
 453 #ifdef PTRDIFF_T
 454 # undef PTRDIFF_T
 455 #endif
 456 #if defined(OPENSSL_SYS_VMS) && __INITIAL_POINTER_SIZE==64
 457 /*
 458  * Then we have VMS that distinguishes itself by adhering to
 459  * sizeof(size_t)==4 even in 64-bit builds, which means that
 460  * difference between two pointers might be truncated to 32 bits.
 461  * In the context one can even wonder how comparison for
 462  * equality is implemented. To be on the safe side we adhere to
 463  * PTRDIFF_T even for comparison for equality.
 464  */
 465 # define PTRDIFF_T uint64_t
 466 #else
 467 # define PTRDIFF_T size_t
 468 #endif
 469
 470 int is_partially_overlapping(const void *ptr1, const void *ptr2, int len)
 471 {
 472     PTRDIFF_T diff = (PTRDIFF_T)ptr1-(PTRDIFF_T)ptr2;
 473     /*
 474      * Check for partially overlapping buffers. [Binary logical
 475      * operations are used instead of boolean to minimize number
 476      * of conditional branches.]
 477      */
 478     int overlapped = (len > 0) & (diff != 0) & ((diff < (PTRDIFF_T)len) |
 479                                                 (diff > (0 - (PTRDIFF_T)len)));
 480
 481     return overlapped;
 482 }
 483
 484 static int evp_EncryptDecryptUpdate(EVP_CIPHER_CTX *ctx,
 485                                     unsigned char *out, int *outl,
 486                                     const unsigned char *in, int inl)
 487 {
 488     int i, j, bl, cmpl = inl;
 489
 490     if (EVP_CIPHER_CTX_test_flags(ctx, EVP_CIPH_FLAG_LENGTH_BITS))
 491         cmpl = (cmpl + 7) / 8;
 492
 493     bl = ctx->cipher->block_size;
 494
 495     if (inl <= 0) {
 496         *outl = 0;
 497         return inl == 0;
 498     }
 499
 500     if (ctx->cipher->flags & EVP_CIPH_FLAG_CUSTOM_CIPHER) {
 501         /* If block size > 1 then the cipher will have to do this check */
 502         if (bl == 1 && is_partially_overlapping(out, in, cmpl)) {
 503             EVPerr(EVP_F_EVP_ENCRYPTDECRYPTUPDATE, EVP_R_PARTIALLY_OVERLAPPING);
 504             return 0;
 505         }
 506
 507         i = ctx->cipher->do_cipher(ctx, out, in, inl);
 508         if (i < 0)
 509             return 0;
 510         else
 511             *outl = i;
 512         return 1;
 513     }
 514
 515     if (is_partially_overlapping(out + ctx->buf_len, in, cmpl)) {
 516         EVPerr(EVP_F_EVP_ENCRYPTDECRYPTUPDATE, EVP_R_PARTIALLY_OVERLAPPING);
 517         return 0;
 518     }
 519
 520     if (ctx->buf_len == 0 && (inl & (ctx->block_mask)) == 0) {
 521         if (ctx->cipher->do_cipher(ctx, out, in, inl)) {
 522             *outl = inl;
 523             return 1;
 524         } else {
 525             *outl = 0;
 526             return 0;
 527         }
 528     }
 529     i = ctx->buf_len;
 530     OPENSSL_assert(bl <= (int)sizeof(ctx->buf));
 531     if (i != 0) {
 532         if (bl - i > inl) {
 533             memcpy(&(ctx->buf[i]), in, inl);
 534             ctx->buf_len += inl;
 535             *outl = 0;
 536             return 1;
 537         } else {
 538             j = bl - i;
 539             memcpy(&(ctx->buf[i]), in, j);
 540             inl -= j;
 541             in += j;
 542             if (!ctx->cipher->do_cipher(ctx, out, ctx->buf, bl))
 543                 return 0;
 544             out += bl;
 545             *outl = bl;
 546         }
 547     } else
 548         *outl = 0;
 549     i = inl & (bl - 1);
 550     inl -= i;
 551     if (inl > 0) {
 552         if (!ctx->cipher->do_cipher(ctx, out, in, inl))
 553             return 0;
 554         *outl += inl;
 555     }
 556
 557     if (i != 0)
 558         memcpy(ctx->buf, &(in[inl]), i);
 559     ctx->buf_len = i;
 560     return 1;
 561 }
 562
 563
 564 int EVP_EncryptUpdate(EVP_CIPHER_CTX *ctx, unsigned char *out, int *outl,
 565                       const unsigned char *in, int inl)
 566 {
 567     int ret;
 568     size_t soutl;
 569     int blocksize;
 570
 571     /* Prevent accidental use of decryption context when encrypting */
 572     if (!ctx->encrypt) {
 573         EVPerr(EVP_F_EVP_ENCRYPTUPDATE, EVP_R_INVALID_OPERATION);
 574         return 0;
 575     }
 576
 577     if (ctx->cipher == NULL || ctx->cipher->prov == NULL)
 578         goto legacy;
 579
 580     blocksize = EVP_CIPHER_CTX_block_size(ctx);
 581
 582     if (ctx->cipher->cupdate == NULL  || blocksize < 1) {
 583         EVPerr(EVP_F_EVP_ENCRYPTUPDATE, EVP_R_UPDATE_ERROR);
 584         return 0;
 585     }
 586     ret = ctx->cipher->cupdate(ctx->provctx, out, &soutl,
 587                                inl + (blocksize == 1 ? 0 : blocksize), in,
 588                                (size_t)inl);
 589
 590     if (soutl > INT_MAX) {
 591         EVPerr(EVP_F_EVP_ENCRYPTUPDATE, EVP_R_UPDATE_ERROR);
 592         return 0;
 593     }
 594     *outl = soutl;
 595     return ret;
 596
 597     /* TODO(3.0): Remove legacy code below */
 598  legacy:
 599
 600     return evp_EncryptDecryptUpdate(ctx, out, outl, in, inl);
 601 }
 602
 603 int EVP_EncryptFinal(EVP_CIPHER_CTX *ctx, unsigned char *out, int *outl)
 604 {
 605     int ret;
 606     ret = EVP_EncryptFinal_ex(ctx, out, outl);
 607     return ret;
 608 }
 609
 610 int EVP_EncryptFinal_ex(EVP_CIPHER_CTX *ctx, unsigned char *out, int *outl)
 611 {
 612     int n, ret;
 613     unsigned int i, b, bl;
 614     size_t soutl;
 615     int blocksize;
 616
 617     /* Prevent accidental use of decryption context when encrypting */
 618     if (!ctx->encrypt) {
 619         EVPerr(EVP_F_EVP_ENCRYPTFINAL_EX, EVP_R_INVALID_OPERATION);
 620         return 0;
 621     }
 622
 623     if (ctx->cipher == NULL || ctx->cipher->prov == NULL)
 624         goto legacy;
 625
 626     blocksize = EVP_CIPHER_CTX_block_size(ctx);
 627
 628     if (blocksize < 1 || ctx->cipher->cfinal == NULL) {
 629         EVPerr(EVP_F_EVP_ENCRYPTFINAL_EX, EVP_R_FINAL_ERROR);
 630         return 0;
 631     }
 632
 633     ret = ctx->cipher->cfinal(ctx->provctx, out, &soutl,
 634                               blocksize == 1 ? 0 : blocksize);
 635
 636     if (soutl > INT_MAX) {
 637         EVPerr(EVP_F_EVP_ENCRYPTFINAL_EX, EVP_R_FINAL_ERROR);
 638         return 0;
 639     }
 640     *outl = soutl;
 641
 642     return ret;
 643
 644     /* TODO(3.0): Remove legacy code below */
 645  legacy:
 646
 647     if (ctx->cipher->flags & EVP_CIPH_FLAG_CUSTOM_CIPHER) {
 648         ret = ctx->cipher->do_cipher(ctx, out, NULL, 0);
 649         if (ret < 0)
 650             return 0;
 651         else
 652             *outl = ret;
 653         return 1;
 654     }
 655
 656     b = ctx->cipher->block_size;
 657     OPENSSL_assert(b <= sizeof(ctx->buf));
 658     if (b == 1) {
 659         *outl = 0;
 660         return 1;
 661     }
 662     bl = ctx->buf_len;
 663     if (ctx->flags & EVP_CIPH_NO_PADDING) {
 664         if (bl) {
 665             EVPerr(EVP_F_EVP_ENCRYPTFINAL_EX,
 666                    EVP_R_DATA_NOT_MULTIPLE_OF_BLOCK_LENGTH);
 667             return 0;
 668         }
 669         *outl = 0;
 670         return 1;
 671     }
 672
 673     n = b - bl;
 674     for (i = bl; i < b; i++)
 675         ctx->buf[i] = n;
 676     ret = ctx->cipher->do_cipher(ctx, out, ctx->buf, b);
 677
 678     if (ret)
 679         *outl = b;
 680
 681     return ret;
 682 }
 683
 684 int EVP_DecryptUpdate(EVP_CIPHER_CTX *ctx, unsigned char *out, int *outl,
 685                       const unsigned char *in, int inl)
 686 {
 687     int fix_len, cmpl = inl, ret;
 688     unsigned int b;
 689     size_t soutl;
 690     int blocksize;
 691
 692     /* Prevent accidental use of encryption context when decrypting */
 693     if (ctx->encrypt) {
 694         EVPerr(EVP_F_EVP_DECRYPTUPDATE, EVP_R_INVALID_OPERATION);
 695         return 0;
 696     }
 697
 698     if (ctx->cipher == NULL || ctx->cipher->prov == NULL)
 699         goto legacy;
 700
 701     blocksize = EVP_CIPHER_CTX_block_size(ctx);
 702
 703     if (ctx->cipher->cupdate == NULL || blocksize < 1) {
 704         EVPerr(EVP_F_EVP_DECRYPTUPDATE, EVP_R_UPDATE_ERROR);
 705         return 0;
 706     }
 707     ret = ctx->cipher->cupdate(ctx->provctx, out, &soutl,
 708                                inl + (blocksize == 1 ? 0 : blocksize), in,
 709                                (size_t)inl);
 710
 711     if (ret) {
 712         if (soutl > INT_MAX) {
 713             EVPerr(EVP_F_EVP_DECRYPTUPDATE, EVP_R_UPDATE_ERROR);
 714             return 0;
 715         }
 716         *outl = soutl;
 717     }
 718
 719     return ret;
 720
 721     /* TODO(3.0): Remove legacy code below */
 722  legacy:
 723
 724     b = ctx->cipher->block_size;
 725
 726     if (EVP_CIPHER_CTX_test_flags(ctx, EVP_CIPH_FLAG_LENGTH_BITS))
 727         cmpl = (cmpl + 7) / 8;
 728
 729     if (inl <= 0) {
 730         *outl = 0;
 731         return inl == 0;
 732     }
 733
 734     if (ctx->cipher->flags & EVP_CIPH_FLAG_CUSTOM_CIPHER) {
 735         if (b == 1 && is_partially_overlapping(out, in, cmpl)) {
 736             EVPerr(EVP_F_EVP_DECRYPTUPDATE, EVP_R_PARTIALLY_OVERLAPPING);
 737             return 0;
 738         }
 739
 740         fix_len = ctx->cipher->do_cipher(ctx, out, in, inl);
 741         if (fix_len < 0) {
 742             *outl = 0;
 743             return 0;
 744         } else
 745             *outl = fix_len;
 746         return 1;
 747     }
 748
 749     if (ctx->flags & EVP_CIPH_NO_PADDING)
 750         return evp_EncryptDecryptUpdate(ctx, out, outl, in, inl);
 751
 752     OPENSSL_assert(b <= sizeof(ctx->final));
 753
 754     if (ctx->final_used) {
 755         /* see comment about PTRDIFF_T comparison above */
 756         if (((PTRDIFF_T)out == (PTRDIFF_T)in)
 757             || is_partially_overlapping(out, in, b)) {
 758             EVPerr(EVP_F_EVP_DECRYPTUPDATE, EVP_R_PARTIALLY_OVERLAPPING);
 759             return 0;
 760         }
 761         memcpy(out, ctx->final, b);
 762         out += b;
 763         fix_len = 1;
 764     } else
 765         fix_len = 0;
 766
 767     if (!evp_EncryptDecryptUpdate(ctx, out, outl, in, inl))
 768         return 0;
 769
 770     /*
 771      * if we have 'decrypted' a multiple of block size, make sure we have a
 772      * copy of this last block
 773      */
 774     if (b > 1 && !ctx->buf_len) {
 775         *outl -= b;
 776         ctx->final_used = 1;
 777         memcpy(ctx->final, &out[*outl], b);
 778     } else
 779         ctx->final_used = 0;
 780
 781     if (fix_len)
 782         *outl += b;
 783
 784     return 1;
 785 }
 786
 787 int EVP_DecryptFinal(EVP_CIPHER_CTX *ctx, unsigned char *out, int *outl)
 788 {
 789     int ret;
 790     ret = EVP_DecryptFinal_ex(ctx, out, outl);
 791     return ret;
 792 }
 793
 794 int EVP_DecryptFinal_ex(EVP_CIPHER_CTX *ctx, unsigned char *out, int *outl)
 795 {
 796     int i, n;
 797     unsigned int b;
 798     size_t soutl;
 799     int ret;
 800     int blocksize;
 801
 802     /* Prevent accidental use of encryption context when decrypting */
 803     if (ctx->encrypt) {
 804         EVPerr(EVP_F_EVP_DECRYPTFINAL_EX, EVP_R_INVALID_OPERATION);
 805         return 0;
 806     }
 807
 808     if (ctx->cipher == NULL || ctx->cipher->prov == NULL)
 809         goto legacy;
 810
 811     blocksize = EVP_CIPHER_CTX_block_size(ctx);
 812
 813     if (blocksize < 1 || ctx->cipher->cfinal == NULL) {
 814         EVPerr(EVP_F_EVP_DECRYPTFINAL_EX, EVP_R_FINAL_ERROR);
 815         return 0;
 816     }
 817
 818     ret = ctx->cipher->cfinal(ctx->provctx, out, &soutl,
 819                               blocksize == 1 ? 0 : blocksize);
 820
 821     if (ret) {
 822         if (soutl > INT_MAX) {
 823             EVPerr(EVP_F_EVP_DECRYPTFINAL_EX, EVP_R_FINAL_ERROR);
 824             return 0;
 825         }
 826         *outl = soutl;
 827     }
 828
 829     return ret;
 830
 831     /* TODO(3.0): Remove legacy code below */
 832  legacy:
 833
 834     *outl = 0;
 835
 836     if (ctx->cipher->flags & EVP_CIPH_FLAG_CUSTOM_CIPHER) {
 837         i = ctx->cipher->do_cipher(ctx, out, NULL, 0);
 838         if (i < 0)
 839             return 0;
 840         else
 841             *outl = i;
 842         return 1;
 843     }
 844
 845     b = ctx->cipher->block_size;
 846     if (ctx->flags & EVP_CIPH_NO_PADDING) {
 847         if (ctx->buf_len) {
 848             EVPerr(EVP_F_EVP_DECRYPTFINAL_EX,
 849                    EVP_R_DATA_NOT_MULTIPLE_OF_BLOCK_LENGTH);
 850             return 0;
 851         }
 852         *outl = 0;
 853         return 1;
 854     }
 855     if (b > 1) {
 856         if (ctx->buf_len || !ctx->final_used) {
 857             EVPerr(EVP_F_EVP_DECRYPTFINAL_EX, EVP_R_WRONG_FINAL_BLOCK_LENGTH);
 858             return 0;
 859         }
 860         OPENSSL_assert(b <= sizeof(ctx->final));
 861
 862         /*
 863          * The following assumes that the ciphertext has been authenticated.
 864          * Otherwise it provides a padding oracle.
 865          */
 866         n = ctx->final[b - 1];
 867         if (n == 0 || n > (int)b) {
 868             EVPerr(EVP_F_EVP_DECRYPTFINAL_EX, EVP_R_BAD_DECRYPT);
 869             return 0;
 870         }
 871         for (i = 0; i < n; i++) {
 872             if (ctx->final[--b] != n) {
 873                 EVPerr(EVP_F_EVP_DECRYPTFINAL_EX, EVP_R_BAD_DECRYPT);
 874                 return 0;
 875             }
 876         }
 877         n = ctx->cipher->block_size - n;
 878         for (i = 0; i < n; i++)
 879             out[i] = ctx->final[i];
 880         *outl = n;
 881     } else
 882         *outl = 0;
 883     return 1;
 884 }
 885
 886 int EVP_CIPHER_CTX_set_key_length(EVP_CIPHER_CTX *c, int keylen)
 887 {
 888     if (c->cipher->flags & EVP_CIPH_CUSTOM_KEY_LENGTH)
 889         return EVP_CIPHER_CTX_ctrl(c, EVP_CTRL_SET_KEY_LENGTH, keylen, NULL);
 890     if (EVP_CIPHER_CTX_key_length(c) == keylen)
 891         return 1;
 892     if ((keylen > 0) && (c->cipher->flags & EVP_CIPH_VARIABLE_LENGTH)) {
 893         c->key_len = keylen;
 894         return 1;
 895     }
 896     EVPerr(EVP_F_EVP_CIPHER_CTX_SET_KEY_LENGTH, EVP_R_INVALID_KEY_LENGTH);
 897     return 0;
 898 }
 899
 900 int EVP_CIPHER_CTX_set_padding(EVP_CIPHER_CTX *ctx, int pad)
 901 {
 902     if (pad)
 903         ctx->flags &= ~EVP_CIPH_NO_PADDING;
 904     else
 905         ctx->flags |= EVP_CIPH_NO_PADDING;
 906
 907     if (ctx->cipher != NULL && ctx->cipher->prov != NULL) {
 908         OSSL_PARAM params[] = {
 909             OSSL_PARAM_int(OSSL_CIPHER_PARAM_PADDING, NULL),
 910             OSSL_PARAM_END
 911         };
 912
 913         params[0].data = &pad;
 914
 915         if (ctx->cipher->ctx_set_params == NULL) {
 916             EVPerr(EVP_F_EVP_CIPHER_CTX_SET_PADDING, EVP_R_CTRL_NOT_IMPLEMENTED);
 917             return 0;
 918         }
 919
 920         if (!ctx->cipher->ctx_set_params(ctx->provctx, params))
 921             return 0;
 922     }
 923
 924     return 1;
 925 }
 926
 927 int EVP_CIPHER_CTX_ctrl(EVP_CIPHER_CTX *ctx, int type, int arg, void *ptr)
 928 {
 929     int ret;
 930
 931     if (!ctx->cipher) {
 932         EVPerr(EVP_F_EVP_CIPHER_CTX_CTRL, EVP_R_NO_CIPHER_SET);
 933         return 0;
 934     }
 935
 936     if (!ctx->cipher->ctrl) {
 937         EVPerr(EVP_F_EVP_CIPHER_CTX_CTRL, EVP_R_CTRL_NOT_IMPLEMENTED);
 938         return 0;
 939     }
 940
 941     ret = ctx->cipher->ctrl(ctx, type, arg, ptr);
 942     if (ret == -1) {
 943         EVPerr(EVP_F_EVP_CIPHER_CTX_CTRL,
 944                EVP_R_CTRL_OPERATION_NOT_IMPLEMENTED);
 945         return 0;
 946     }
 947     return ret;
 948 }
 949
 950 int EVP_CIPHER_CTX_rand_key(EVP_CIPHER_CTX *ctx, unsigned char *key)
 951 {
 952     if (ctx->cipher->flags & EVP_CIPH_RAND_KEY)
 953         return EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_RAND_KEY, 0, key);
 954     if (RAND_priv_bytes(key, EVP_CIPHER_CTX_key_length(ctx)) <= 0)
 955         return 0;
 956     return 1;
 957 }
 958
 959 int EVP_CIPHER_CTX_copy(EVP_CIPHER_CTX *out, const EVP_CIPHER_CTX *in)
 960 {
 961     if ((in == NULL) || (in->cipher == NULL)) {
 962         EVPerr(EVP_F_EVP_CIPHER_CTX_COPY, EVP_R_INPUT_NOT_INITIALIZED);
 963         return 0;
 964     }
 965
 966     if (in->cipher->prov == NULL)
 967         goto legacy;
 968
 969     if (in->cipher->dupctx == NULL) {
 970         EVPerr(EVP_F_EVP_CIPHER_CTX_COPY, EVP_R_NOT_ABLE_TO_COPY_CTX);
 971         return 0;
 972     }
 973
 974     EVP_CIPHER_CTX_reset(out);
 975
 976     *out = *in;
 977     out->provctx = NULL;
 978
 979     if (in->fetched_cipher != NULL && !EVP_CIPHER_upref(in->fetched_cipher)) {
 980         out->fetched_cipher = NULL;
 981         return 0;
 982     }
 983
 984     out->provctx = in->cipher->dupctx(in->provctx);
 985     if (out->provctx == NULL) {
 986         EVPerr(EVP_F_EVP_CIPHER_CTX_COPY, EVP_R_NOT_ABLE_TO_COPY_CTX);
 987         return 0;
 988     }
 989
 990     return 1;
 991
 992     /* TODO(3.0): Remove legacy code below */
 993  legacy:
 994
 995 #ifndef OPENSSL_NO_ENGINE
 996     /* Make sure it's safe to copy a cipher context using an ENGINE */
 997     if (in->engine && !ENGINE_init(in->engine)) {
 998         EVPerr(EVP_F_EVP_CIPHER_CTX_COPY, ERR_R_ENGINE_LIB);
 999         return 0;
1000     }
1001 #endif
1002
1003     EVP_CIPHER_CTX_reset(out);
1004     memcpy(out, in, sizeof(*out));
1005
1006     if (in->cipher_data && in->cipher->ctx_size) {
1007         out->cipher_data = OPENSSL_malloc(in->cipher->ctx_size);
1008         if (out->cipher_data == NULL) {
1009             out->cipher = NULL;
1010             EVPerr(EVP_F_EVP_CIPHER_CTX_COPY, ERR_R_MALLOC_FAILURE);
1011             return 0;
1012         }
1013         memcpy(out->cipher_data, in->cipher_data, in->cipher->ctx_size);
1014     }
1015
1016     if (in->cipher->flags & EVP_CIPH_CUSTOM_COPY)
1017         if (!in->cipher->ctrl((EVP_CIPHER_CTX *)in, EVP_CTRL_COPY, 0, out)) {
1018             out->cipher = NULL;
1019             EVPerr(EVP_F_EVP_CIPHER_CTX_COPY, EVP_R_INITIALIZATION_ERROR);
1020             return 0;
1021         }
1022     return 1;
1023 }
1024
1025 static void *evp_cipher_from_dispatch(int nid, const OSSL_DISPATCH *fns,
1026                                       OSSL_PROVIDER *prov)
1027 {
1028     EVP_CIPHER *cipher = NULL;
1029     int fnciphcnt = 0, fnctxcnt = 0;
1030
1031     if ((cipher = EVP_CIPHER_meth_new(nid, 0, 0)) == NULL)
1032         return NULL;
1033
1034     for (; fns->function_id != 0; fns++) {
1035         switch (fns->function_id) {
1036         case OSSL_FUNC_CIPHER_NEWCTX:
1037             if (cipher->newctx != NULL)
1038                 break;
1039             cipher->newctx = OSSL_get_OP_cipher_newctx(fns);
1040             fnctxcnt++;
1041             break;
1042         case OSSL_FUNC_CIPHER_ENCRYPT_INIT:
1043             if (cipher->einit != NULL)
1044                 break;
1045             cipher->einit = OSSL_get_OP_cipher_encrypt_init(fns);
1046             fnciphcnt++;
1047             break;
1048         case OSSL_FUNC_CIPHER_DECRYPT_INIT:
1049             if (cipher->dinit != NULL)
1050                 break;
1051             cipher->dinit = OSSL_get_OP_cipher_decrypt_init(fns);
1052             fnciphcnt++;
1053             break;
1054         case OSSL_FUNC_CIPHER_UPDATE:
1055             if (cipher->cupdate != NULL)
1056                 break;
1057             cipher->cupdate = OSSL_get_OP_cipher_update(fns);
1058             fnciphcnt++;
1059             break;
1060         case OSSL_FUNC_CIPHER_FINAL:
1061             if (cipher->cfinal != NULL)
1062                 break;
1063             cipher->cfinal = OSSL_get_OP_cipher_final(fns);
1064             fnciphcnt++;
1065             break;
1066         case OSSL_FUNC_CIPHER_CIPHER:
1067             if (cipher->ccipher != NULL)
1068                 break;
1069             cipher->ccipher = OSSL_get_OP_cipher_cipher(fns);
1070             break;
1071         case OSSL_FUNC_CIPHER_FREECTX:
1072             if (cipher->freectx != NULL)
1073                 break;
1074             cipher->freectx = OSSL_get_OP_cipher_freectx(fns);
1075             fnctxcnt++;
1076             break;
1077         case OSSL_FUNC_CIPHER_DUPCTX:
1078             if (cipher->dupctx != NULL)
1079                 break;
1080             cipher->dupctx = OSSL_get_OP_cipher_dupctx(fns);
1081             break;
1082         case OSSL_FUNC_CIPHER_KEY_LENGTH:
1083             if (cipher->key_length != NULL)
1084                 break;
1085             cipher->key_length = OSSL_get_OP_cipher_key_length(fns);
1086             break;
1087         case OSSL_FUNC_CIPHER_IV_LENGTH:
1088             if (cipher->iv_length != NULL)
1089                 break;
1090             cipher->iv_length = OSSL_get_OP_cipher_iv_length(fns);
1091             break;
1092         case OSSL_FUNC_CIPHER_BLOCK_SIZE:
1093             if (cipher->blocksize != NULL)
1094                 break;
1095             cipher->blocksize = OSSL_get_OP_cipher_block_size(fns);
1096             break;
1097         case OSSL_FUNC_CIPHER_GET_PARAMS:
1098             if (cipher->get_params != NULL)
1099                 break;
1100             cipher->get_params = OSSL_get_OP_cipher_get_params(fns);
1101             break;
1102         case OSSL_FUNC_CIPHER_CTX_GET_PARAMS:
1103             if (cipher->ctx_get_params != NULL)
1104                 break;
1105             cipher->ctx_get_params = OSSL_get_OP_cipher_ctx_get_params(fns);
1106             break;
1107         case OSSL_FUNC_CIPHER_CTX_SET_PARAMS:
1108             if (cipher->ctx_set_params != NULL)
1109                 break;
1110             cipher->ctx_set_params = OSSL_get_OP_cipher_ctx_set_params(fns);
1111             break;
1112         }
1113     }
1114     if ((fnciphcnt != 0 && fnciphcnt != 3 && fnciphcnt != 4)
1115             || (fnciphcnt == 0 && cipher->ccipher == NULL)
1116             || fnctxcnt != 2
1117             || cipher->blocksize == NULL
1118             || cipher->iv_length == NULL
1119             || cipher->key_length == NULL) {
1120         /*
1121          * In order to be a consistent set of functions we must have at least
1122          * a complete set of "encrypt" functions, or a complete set of "decrypt"
1123          * functions, or a single "cipher" function. In all cases we need a
1124          * complete set of context management functions, as well as the
1125          * blocksize, iv_length and key_length functions.
1126          */
1127         EVP_CIPHER_meth_free(cipher);
1128         EVPerr(EVP_F_EVP_CIPHER_FROM_DISPATCH, EVP_R_INVALID_PROVIDER_FUNCTIONS);
1129         return NULL;
1130     }
1131     cipher->prov = prov;
1132     if (prov != NULL)
1133         ossl_provider_upref(prov);
1134
1135     return cipher;
1136 }
1137
1138 static int evp_cipher_upref(void *cipher)
1139 {
1140     return EVP_CIPHER_upref(cipher);
1141 }
1142
1143 static void evp_cipher_free(void *cipher)
1144 {
1145     EVP_CIPHER_meth_free(cipher);
1146 }
1147
1148 static int evp_cipher_nid(void *vcipher)
1149 {
1150     EVP_CIPHER *cipher = vcipher;
1151
1152     return cipher->nid;
1153 }
1154
1155 EVP_CIPHER *EVP_CIPHER_fetch(OPENSSL_CTX *ctx, const char *algorithm,
1156                              const char *properties)
1157 {
1158     return evp_generic_fetch(ctx, OSSL_OP_CIPHER, algorithm, properties,
1159                              evp_cipher_from_dispatch, evp_cipher_upref,
1160                              evp_cipher_free, evp_cipher_nid);
1161 }