1 /* ctr_mode.c - TinyCrypt CTR mode implementation */
2 
3 /*
4  *  Copyright (C) 2017 by Intel Corporation, All Rights Reserved.
5  *
6  *  Redistribution and use in source and binary forms, with or without
7  *  modification, are permitted provided that the following conditions are met:
8  *
9  *    - Redistributions of source code must retain the above copyright notice,
10  *     this list of conditions and the following disclaimer.
11  *
12  *    - Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  *    - Neither the name of Intel Corporation nor the names of its contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
21  *  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  *  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  *  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
24  *  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25  *  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26  *  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27  *  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28  *  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29  *  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30  *  POSSIBILITY OF SUCH DAMAGE.
31  */
32 
33 #include <tinycrypt/constants.h>
34 #include <tinycrypt/ctr_mode.h>
35 #include <tinycrypt/utils.h>
36 
tc_ctr_mode(uint8_t * out,unsigned int outlen,const uint8_t * in,unsigned int inlen,uint8_t * ctr,uint32_t * blk_off,const TCAesKeySched_t sched)37 int tc_ctr_mode(uint8_t *out, unsigned int outlen, const uint8_t *in,
38 		unsigned int inlen, uint8_t *ctr, uint32_t *blk_off,
39 		const TCAesKeySched_t sched)
40 {
41 
42 	uint8_t buffer[TC_AES_BLOCK_SIZE];
43 	uint8_t nonce[TC_AES_BLOCK_SIZE];
44 	unsigned int block_num;
45 	unsigned int i;
46 	uint32_t n;
47 
48 	/* input sanity check: */
49 	if (out == (uint8_t *) 0 ||
50 	    in == (uint8_t *) 0 ||
51 	    ctr == (uint8_t *) 0 ||
52 	    blk_off == (uint32_t *) 0 ||
53 	    sched == (TCAesKeySched_t) 0 ||
54 	    inlen == 0 ||
55 	    outlen == 0 ||
56 	    outlen != inlen) {
57 		return TC_CRYPTO_FAIL;
58 	}
59 
60 	/* copy the ctr to the nonce */
61 	(void)_copy(nonce, sizeof(nonce), ctr, sizeof(nonce));
62 
63 	/* select the last 4 bytes of the nonce to be incremented */
64 	block_num = (nonce[12] << 24) | (nonce[13] << 16) |
65 		    (nonce[14] << 8) | (nonce[15]);
66 	n = *blk_off;
67 	for (i = 0; i < inlen; ++i) {
68 		if (n == 0) {
69 			/* encrypt data using the current nonce */
70 			if (tc_aes_encrypt(buffer, nonce, sched)) {
71 				block_num++;
72 				nonce[12] = (uint8_t)(block_num >> 24);
73 				nonce[13] = (uint8_t)(block_num >> 16);
74 				nonce[14] = (uint8_t)(block_num >> 8);
75 				nonce[15] = (uint8_t)(block_num);
76 			} else {
77 				return TC_CRYPTO_FAIL;
78 			}
79 		}
80 		/* update the output */
81 		*out++ = buffer[n] ^ *in++;
82 		n = (n + 1) % TC_AES_BLOCK_SIZE;
83 	}
84 	*blk_off = n;
85 
86 	/* update the counter */
87 	ctr[12] = nonce[12]; ctr[13] = nonce[13];
88 	ctr[14] = nonce[14]; ctr[15] = nonce[15];
89 
90 	return TC_CRYPTO_SUCCESS;
91 }
92