1 /*
2  * AES-128/192/256 CTR
3  *
4  * Copyright (c) 2003-2007, Jouni Malinen <j@w1.fi>
5  *
6  * This software may be distributed under the terms of the BSD license.
7  * See README for more details.
8  */
9 
10 #include "includes.h"
11 
12 #include "common.h"
13 #include "aes.h"
14 #include "aes_wrap.h"
15 
16 /**
17  * aes_ctr_encrypt - AES-128/192/256 CTR mode encryption
18  * @key: Key for encryption (key_len bytes)
19  * @key_len: Length of the key (16, 24, or 32 bytes)
20  * @nonce: Nonce for counter mode (16 bytes)
21  * @data: Data to encrypt in-place
22  * @data_len: Length of data in bytes
23  * Returns: 0 on success, -1 on failure
24  */
25 int aes_ctr_encrypt(const u8 *key, size_t key_len, const u8 *nonce,
26 		    u8 *data, size_t data_len)
27 {
28 	void *ctx;
29 	size_t j, len, left = data_len;
30 	int i;
31 	u8 *pos = data;
32 	u8 counter[AES_BLOCK_SIZE], buf[AES_BLOCK_SIZE];
33 
34 	ctx = aes_encrypt_init(key, key_len);
35 	if (ctx == NULL)
36 		return -1;
37 	os_memcpy(counter, nonce, AES_BLOCK_SIZE);
38 
39 	while (left > 0) {
40 		aes_encrypt(ctx, counter, buf);
41 
42 		len = (left < AES_BLOCK_SIZE) ? left : AES_BLOCK_SIZE;
43 		for (j = 0; j < len; j++)
44 			pos[j] ^= buf[j];
45 		pos += len;
46 		left -= len;
47 
48 		for (i = AES_BLOCK_SIZE - 1; i >= 0; i--) {
49 			counter[i]++;
50 			if (counter[i])
51 				break;
52 		}
53 	}
54 	aes_encrypt_deinit(ctx);
55 	return 0;
56 }
57 
58 
59 /**
60  * aes_128_ctr_encrypt - AES-128 CTR mode encryption
61  * @key: Key for encryption (key_len bytes)
62  * @nonce: Nonce for counter mode (16 bytes)
63  * @data: Data to encrypt in-place
64  * @data_len: Length of data in bytes
65  * Returns: 0 on success, -1 on failure
66  */
67 int aes_128_ctr_encrypt(const u8 *key, const u8 *nonce,
68 			u8 *data, size_t data_len)
69 {
70 	return aes_ctr_encrypt(key, 16, nonce, data, data_len);
71 }
72