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,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, const TCAesKeySched_t sched)
39 {
40 
41     uint8_t buffer[TC_AES_BLOCK_SIZE];
42     uint8_t nonce[TC_AES_BLOCK_SIZE];
43     unsigned int block_num;
44     unsigned int i;
45 
46     /* input sanity check: */
47     if (out == (uint8_t *) 0 ||
48             in == (uint8_t *) 0 ||
49             ctr == (uint8_t *) 0 ||
50             sched == (TCAesKeySched_t) 0 ||
51             inlen == 0 ||
52             outlen == 0 ||
53             outlen != inlen) {
54         return TC_CRYPTO_FAIL;
55     }
56 
57     /* copy the ctr to the nonce */
58     (void)_copy(nonce, sizeof(nonce), ctr, sizeof(nonce));
59 
60     /* select the last 4 bytes of the nonce to be incremented */
61     block_num = (nonce[12] << 24) | (nonce[13] << 16) |
62                 (nonce[14] << 8) | (nonce[15]);
63     for (i = 0; i < inlen; ++i) {
64         if ((i % (TC_AES_BLOCK_SIZE)) == 0) {
65             /* encrypt data using the current nonce */
66             if (tc_aes_encrypt(buffer, nonce, sched)) {
67                 block_num++;
68                 nonce[12] = (uint8_t)(block_num >> 24);
69                 nonce[13] = (uint8_t)(block_num >> 16);
70                 nonce[14] = (uint8_t)(block_num >> 8);
71                 nonce[15] = (uint8_t)(block_num);
72             } else {
73                 return TC_CRYPTO_FAIL;
74             }
75         }
76         /* update the output */
77         *out++ = buffer[i % (TC_AES_BLOCK_SIZE)] ^ *in++;
78     }
79 
80     /* update the counter */
81     ctr[12] = nonce[12]; ctr[13] = nonce[13];
82     ctr[14] = nonce[14]; ctr[15] = nonce[15];
83 
84     return TC_CRYPTO_SUCCESS;
85 }
86