forked from pyrogram/tgcrypto
-
Notifications
You must be signed in to change notification settings - Fork 0
/
aes256.h
45 lines (36 loc) · 1.58 KB
/
aes256.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
/*
* Pyrogram - Telegram MTProto API Client Library for Python
* Copyright (C) 2017-2020 Dan <https://github.com/delivrance>
*
* This file is part of Pyrogram.
*
* Pyrogram is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published
* by the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Pyrogram is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
*/
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#ifndef AES256_H
#define AES256_H
#define AES_BLOCK_SIZE 16
#define EXPANDED_KEY_SIZE 60
#define LROTL(x) (((x) << 8) | ((x) >> 24))
#define LROTR(x) (((x) >> 8) | ((x) << 24))
#define SWAP(x) ((LROTL((x)) & 0x00ff00ff) | (LROTR((x)) & 0xff00ff00))
#define GET(p) SWAP(*((uint32_t *)(p)))
#define PUT(ct, st) {*((uint32_t *)(ct)) = SWAP((st));} (void)0
void aes256_set_encryption_key(const uint8_t key[32], uint32_t expandedKey[60]);
void aes256_set_decryption_key(const uint8_t key[32], uint32_t expandedKey[60]);
void aes256_encrypt(const uint8_t in[16], uint8_t out[16], const uint32_t expandedKey[60]);
void aes256_decrypt(const uint8_t in[16], uint8_t out[16], const uint32_t expandedKey[60]);
#endif // AES256_H