1 /* Copyright (c) (2010,2011,2014-2016,2018,2019,2021) Apple Inc. All rights reserved.
2 *
3 * corecrypto is licensed under Apple Inc.’s Internal Use License Agreement (which
4 * is contained in the License.txt file distributed with corecrypto) and only to
5 * people who accept that license. IMPORTANT: Any license rights granted to you by
6 * Apple Inc. (if any) are limited to internal use within your organization only on
7 * devices and computers you own or control, for the sole purpose of verifying the
8 * security characteristics and correct functioning of the Apple Software. You may
9 * not, directly or indirectly, redistribute the Apple Software or any portions thereof.
10 *
11 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
12 *
13 * This file contains Original Code and/or Modifications of Original Code
14 * as defined in and that are subject to the Apple Public Source License
15 * Version 2.0 (the 'License'). You may not use this file except in
16 * compliance with the License. The rights granted to you under the License
17 * may not be used to create, or enable the creation or redistribution of,
18 * unlawful or unlicensed copies of an Apple operating system, or to
19 * circumvent, violate, or enable the circumvention or violation of, any
20 * terms of an Apple operating system software license agreement.
21 *
22 * Please obtain a copy of the License at
23 * http://www.opensource.apple.com/apsl/ and read it before using this file.
24 *
25 * The Original Code and all software distributed under the License are
26 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
27 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
28 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
29 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
30 * Please see the License for the specific language governing rights and
31 * limitations under the License.
32 *
33 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
34 */
35
36 #include "cc_internal.h"
37 #include <corecrypto/ccdigest.h>
38 #include <corecrypto/cc_priv.h>
39
40 void
ccdigest_update(const struct ccdigest_info * di,ccdigest_ctx_t ctx,size_t len,const void * data)41 ccdigest_update(const struct ccdigest_info *di, ccdigest_ctx_t ctx, size_t len, const void *data)
42 {
43 CC_ENSURE_DIT_ENABLED
44
45 const char * data_ptr = data;
46 size_t nblocks, nbytes;
47
48 // Sanity check to recover from ctx corruptions.
49 if (ccdigest_num(di, ctx) >= di->block_size) {
50 ccdigest_num(di, ctx) = 0;
51 }
52
53 while (len > 0) {
54 if (ccdigest_num(di, ctx) == 0 && len > di->block_size) {
55 if (di->block_size == 1 << 6) { // md5 & sha1 & sha256
56 nblocks = len >> 6;
57 nbytes = nblocks << 6;
58 } else if (di->block_size == 1 << 7) { // sha384 & sha512
59 nblocks = len >> 7;
60 nbytes = nblocks << 7;
61 } else {
62 nblocks = len / di->block_size;
63 nbytes = nblocks * di->block_size;
64 }
65
66 di->compress(ccdigest_state(di, ctx), nblocks, data_ptr);
67 len -= nbytes;
68 data_ptr += nbytes;
69 ccdigest_nbits(di, ctx) += (uint64_t) (nbytes) * 8;
70 } else {
71 size_t n = CC_MIN(di->block_size - ccdigest_num(di, ctx), len);
72 cc_memcpy(ccdigest_data(di, ctx) + ccdigest_num(di, ctx), data_ptr, n);
73 /* typecast: less than block size, will always fit into an int */
74 ccdigest_num(di, ctx) += (unsigned int)n;
75 len -= n;
76 data_ptr += n;
77 if (ccdigest_num(di, ctx) == di->block_size) {
78 di->compress(ccdigest_state(di, ctx), 1, ccdigest_data(di, ctx));
79 ccdigest_nbits(di, ctx) += ccdigest_num(di, ctx) * 8;
80 ccdigest_num(di, ctx) = 0;
81 }
82 }
83 }
84 }
85