diff options
author | Toni Uhlig <matzeton@googlemail.com> | 2023-03-12 20:01:21 +0100 |
---|---|---|
committer | Toni Uhlig <matzeton@googlemail.com> | 2023-03-12 20:01:21 +0100 |
commit | ba6815ef8fb8ae472412b5af2837a7caba2799c2 (patch) | |
tree | bc8f40525e662d5f72ec40b0c8127676150ff8cf | |
parent | cc3dbf16d29bf8f74c284d94a6da10ed004cf9d3 (diff) |
Revert "AES256 support"
This reverts commit cc3dbf16d29bf8f74c284d94a6da10ed004cf9d3.
-rw-r--r-- | Makefile | 18 | ||||
-rw-r--r-- | aes/aes256.cpp | 567 | ||||
-rw-r--r-- | aes/aes256.hpp | 88 | ||||
-rw-r--r-- | examples/driver-aes.bat | 27 | ||||
-rw-r--r-- | examples/driver-aes.cpp | 63 |
5 files changed, 4 insertions, 759 deletions
@@ -16,10 +16,6 @@ DRIVER2_NAME = driver-protobuf-c-tcp DRIVER2_OBJECTS = examples/$(DRIVER2_NAME).o ksocket/ksocket.o ksocket/berkeley.o protobuf-c/protobuf-c.o examples/example.pb-c.o DRIVER2_TARGET = $(DRIVER2_NAME).sys -DRIVER3_NAME = driver-aes -DRIVER3_OBJECTS = examples/$(DRIVER3_NAME).o ksocket/ksocket.o ksocket/berkeley.o aes/aes256.o -DRIVER3_TARGET = $(DRIVER3_NAME).sys - USERSPACE0_NAME = userspace_client USERSPACE0_OBJECTS = examples/$(USERSPACE0_NAME).o USERSPACE0_TARGET = $(USERSPACE0_NAME).exe @@ -34,7 +30,7 @@ CUSTOM_CFLAGS = -I. -Wl,--exclude-all-symbols -DNDEBUG DRIVER_LIBS += -lnetio USER_LIBS += -lws2_32 -all: $(DRIVER0_TARGET) $(DRIVER1_TARGET) $(DRIVER2_TARGET) $(DRIVER3_TARGET) $(USERSPACE0_TARGET) $(USERSPACE1_TARGET) +all: $(DRIVER0_TARGET) $(DRIVER1_TARGET) $(DRIVER2_TARGET) $(USERSPACE0_TARGET) $(USERSPACE1_TARGET) %.o: %.cpp $(call BUILD_CPP_OBJECT,$<,$@) @@ -51,9 +47,6 @@ $(DRIVER1_TARGET): $(DRIVER1_OBJECTS) $(DRIVER2_TARGET): $(DRIVER2_OBJECTS) $(call LINK_CPP_KERNEL_TARGET,$(DRIVER2_OBJECTS),$@) -$(DRIVER3_TARGET): $(DRIVER3_OBJECTS) - $(call LINK_CPP_KERNEL_TARGET,$(DRIVER3_OBJECTS),$@) - $(USERSPACE0_TARGET): $(USERSPACE0_OBJECTS) $(call LINK_CPP_USER_TARGET,$(USERSPACE0_OBJECTS),$@) @@ -68,24 +61,21 @@ generate: @echo protoc-c --c_out=. examples/example.proto -install: $(DRIVER0_TARGET) $(DRIVER1_TARGET) $(DRIVER2_TARGET) $(DRIVER3_TARGET) $(USERSPACE0_TARGET) $(USERSPACE1_TARGET) +install: $(DRIVER0_TARGET) $(DRIVER1_TARGET) $(DRIVER2_TARGET) $(USERSPACE0_TARGET) $(USERSPACE1_TARGET) $(call INSTALL_EXEC_SIGN,$(DRIVER0_TARGET)) $(call INSTALL_EXEC_SIGN,$(DRIVER1_TARGET)) $(call INSTALL_EXEC_SIGN,$(DRIVER2_TARGET)) - $(call INSTALL_EXEC_SIGN,$(DRIVER3_TARGET)) $(call INSTALL_EXEC,$(USERSPACE0_TARGET)) $(call INSTALL_EXEC,$(USERSPACE1_TARGET)) $(INSTALL) 'examples/$(DRIVER0_NAME).bat' '$(DESTDIR)/' $(INSTALL) 'examples/$(DRIVER1_NAME).bat' '$(DESTDIR)/' $(INSTALL) 'examples/$(DRIVER2_NAME).bat' '$(DESTDIR)/' - $(INSTALL) 'examples/$(DRIVER3_NAME).bat' '$(DESTDIR)/' clean: - rm -f $(DRIVER0_OBJECTS) $(DRIVER1_OBJECTS) $(DRIVER2_OBJECTS) $(DRIVER3_OBJECTS) + rm -f $(DRIVER0_OBJECTS) $(DRIVER1_OBJECTS) $(DRIVER2_OBJECTS) rm -f $(DRIVER0_TARGET) $(DRIVER0_TARGET).map \ $(DRIVER1_TARGET) $(DRIVER1_TARGET).map \ - $(DRIVER2_TARGET) $(DRIVER2_TARGET).map \ - $(DRIVER3_TARGET) $(DRIVER3_TARGET).map + $(DRIVER2_TARGET) $(DRIVER2_TARGET).map rm -f $(USERSPACE0_OBJECTS) $(USERSPACE1_OBJECTS) rm -f $(USERSPACE0_TARGET) $(USERSPACE1_TARGET) diff --git a/aes/aes256.cpp b/aes/aes256.cpp deleted file mode 100644 index 0ac76e8..0000000 --- a/aes/aes256.cpp +++ /dev/null @@ -1,567 +0,0 @@ -/* - * aes256.cpp - * - * Copyright (c) 2014, Danilo Treffiletti <urban82@gmail.com> - * All rights reserved. - * - * This file is part of Aes256. - * - * Aes256 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 2.1 - * of the License, or (at your option) any later version. - * - * Aes256 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 Aes256. - * If not, see <http://www.gnu.org/licenses/>. - */ - -#include "aes256.hpp" - -#include <stdlib.h> - -#define FE(x) (((x) << 1) ^ ((((x)>>7) & 1) * 0x1b)) -#define FD(x) (((x) >> 1) ^ (((x) & 1) ? 0x8d : 0)) - -#define KEY_SIZE 32 -#define NUM_ROUNDS 14 - -unsigned char rj_xtime(unsigned char x); - -const unsigned char sbox[256] = { - 0x63, 0x7c, 0x77, 0x7b, 0xf2, 0x6b, 0x6f, 0xc5, - 0x30, 0x01, 0x67, 0x2b, 0xfe, 0xd7, 0xab, 0x76, - 0xca, 0x82, 0xc9, 0x7d, 0xfa, 0x59, 0x47, 0xf0, - 0xad, 0xd4, 0xa2, 0xaf, 0x9c, 0xa4, 0x72, 0xc0, - 0xb7, 0xfd, 0x93, 0x26, 0x36, 0x3f, 0xf7, 0xcc, - 0x34, 0xa5, 0xe5, 0xf1, 0x71, 0xd8, 0x31, 0x15, - 0x04, 0xc7, 0x23, 0xc3, 0x18, 0x96, 0x05, 0x9a, - 0x07, 0x12, 0x80, 0xe2, 0xeb, 0x27, 0xb2, 0x75, - 0x09, 0x83, 0x2c, 0x1a, 0x1b, 0x6e, 0x5a, 0xa0, - 0x52, 0x3b, 0xd6, 0xb3, 0x29, 0xe3, 0x2f, 0x84, - 0x53, 0xd1, 0x00, 0xed, 0x20, 0xfc, 0xb1, 0x5b, - 0x6a, 0xcb, 0xbe, 0x39, 0x4a, 0x4c, 0x58, 0xcf, - 0xd0, 0xef, 0xaa, 0xfb, 0x43, 0x4d, 0x33, 0x85, - 0x45, 0xf9, 0x02, 0x7f, 0x50, 0x3c, 0x9f, 0xa8, - 0x51, 0xa3, 0x40, 0x8f, 0x92, 0x9d, 0x38, 0xf5, - 0xbc, 0xb6, 0xda, 0x21, 0x10, 0xff, 0xf3, 0xd2, - 0xcd, 0x0c, 0x13, 0xec, 0x5f, 0x97, 0x44, 0x17, - 0xc4, 0xa7, 0x7e, 0x3d, 0x64, 0x5d, 0x19, 0x73, - 0x60, 0x81, 0x4f, 0xdc, 0x22, 0x2a, 0x90, 0x88, - 0x46, 0xee, 0xb8, 0x14, 0xde, 0x5e, 0x0b, 0xdb, - 0xe0, 0x32, 0x3a, 0x0a, 0x49, 0x06, 0x24, 0x5c, - 0xc2, 0xd3, 0xac, 0x62, 0x91, 0x95, 0xe4, 0x79, - 0xe7, 0xc8, 0x37, 0x6d, 0x8d, 0xd5, 0x4e, 0xa9, - 0x6c, 0x56, 0xf4, 0xea, 0x65, 0x7a, 0xae, 0x08, - 0xba, 0x78, 0x25, 0x2e, 0x1c, 0xa6, 0xb4, 0xc6, - 0xe8, 0xdd, 0x74, 0x1f, 0x4b, 0xbd, 0x8b, 0x8a, - 0x70, 0x3e, 0xb5, 0x66, 0x48, 0x03, 0xf6, 0x0e, - 0x61, 0x35, 0x57, 0xb9, 0x86, 0xc1, 0x1d, 0x9e, - 0xe1, 0xf8, 0x98, 0x11, 0x69, 0xd9, 0x8e, 0x94, - 0x9b, 0x1e, 0x87, 0xe9, 0xce, 0x55, 0x28, 0xdf, - 0x8c, 0xa1, 0x89, 0x0d, 0xbf, 0xe6, 0x42, 0x68, - 0x41, 0x99, 0x2d, 0x0f, 0xb0, 0x54, 0xbb, 0x16 -}; -const unsigned char sboxinv[256] = { - 0x52, 0x09, 0x6a, 0xd5, 0x30, 0x36, 0xa5, 0x38, - 0xbf, 0x40, 0xa3, 0x9e, 0x81, 0xf3, 0xd7, 0xfb, - 0x7c, 0xe3, 0x39, 0x82, 0x9b, 0x2f, 0xff, 0x87, - 0x34, 0x8e, 0x43, 0x44, 0xc4, 0xde, 0xe9, 0xcb, - 0x54, 0x7b, 0x94, 0x32, 0xa6, 0xc2, 0x23, 0x3d, - 0xee, 0x4c, 0x95, 0x0b, 0x42, 0xfa, 0xc3, 0x4e, - 0x08, 0x2e, 0xa1, 0x66, 0x28, 0xd9, 0x24, 0xb2, - 0x76, 0x5b, 0xa2, 0x49, 0x6d, 0x8b, 0xd1, 0x25, - 0x72, 0xf8, 0xf6, 0x64, 0x86, 0x68, 0x98, 0x16, - 0xd4, 0xa4, 0x5c, 0xcc, 0x5d, 0x65, 0xb6, 0x92, - 0x6c, 0x70, 0x48, 0x50, 0xfd, 0xed, 0xb9, 0xda, - 0x5e, 0x15, 0x46, 0x57, 0xa7, 0x8d, 0x9d, 0x84, - 0x90, 0xd8, 0xab, 0x00, 0x8c, 0xbc, 0xd3, 0x0a, - 0xf7, 0xe4, 0x58, 0x05, 0xb8, 0xb3, 0x45, 0x06, - 0xd0, 0x2c, 0x1e, 0x8f, 0xca, 0x3f, 0x0f, 0x02, - 0xc1, 0xaf, 0xbd, 0x03, 0x01, 0x13, 0x8a, 0x6b, - 0x3a, 0x91, 0x11, 0x41, 0x4f, 0x67, 0xdc, 0xea, - 0x97, 0xf2, 0xcf, 0xce, 0xf0, 0xb4, 0xe6, 0x73, - 0x96, 0xac, 0x74, 0x22, 0xe7, 0xad, 0x35, 0x85, - 0xe2, 0xf9, 0x37, 0xe8, 0x1c, 0x75, 0xdf, 0x6e, - 0x47, 0xf1, 0x1a, 0x71, 0x1d, 0x29, 0xc5, 0x89, - 0x6f, 0xb7, 0x62, 0x0e, 0xaa, 0x18, 0xbe, 0x1b, - 0xfc, 0x56, 0x3e, 0x4b, 0xc6, 0xd2, 0x79, 0x20, - 0x9a, 0xdb, 0xc0, 0xfe, 0x78, 0xcd, 0x5a, 0xf4, - 0x1f, 0xdd, 0xa8, 0x33, 0x88, 0x07, 0xc7, 0x31, - 0xb1, 0x12, 0x10, 0x59, 0x27, 0x80, 0xec, 0x5f, - 0x60, 0x51, 0x7f, 0xa9, 0x19, 0xb5, 0x4a, 0x0d, - 0x2d, 0xe5, 0x7a, 0x9f, 0x93, 0xc9, 0x9c, 0xef, - 0xa0, 0xe0, 0x3b, 0x4d, 0xae, 0x2a, 0xf5, 0xb0, - 0xc8, 0xeb, 0xbb, 0x3c, 0x83, 0x53, 0x99, 0x61, - 0x17, 0x2b, 0x04, 0x7e, 0xba, 0x77, 0xd6, 0x26, - 0xe1, 0x69, 0x14, 0x63, 0x55, 0x21, 0x0c, 0x7d -}; - -Aes256::Aes256(const ByteArray& key) - : m_key(ByteArray(key.size() > KEY_SIZE ? KEY_SIZE : key.size(), 0)) - , m_salt(ByteArray(KEY_SIZE - m_key.size(), 0)) - , m_rkey(ByteArray(KEY_SIZE, 0)) - , m_buffer_pos(0) - , m_remainingLength(0) - , m_decryptInitialized(false) -{ - for(ByteArray::size_type i = 0; i < m_key.size(); ++i) - m_key[i] = key[i]; -} - -Aes256::~Aes256() -{} - -ByteArray::size_type Aes256::encrypt(const ByteArray& key, const ByteArray& plain, ByteArray& encrypted) -{ - Aes256 aes(key); - - aes.encrypt_start(plain.size(), encrypted); - aes.encrypt_continue(plain, encrypted); - aes.encrypt_end(encrypted); - - return encrypted.size(); -} - -ByteArray::size_type Aes256::encrypt(const ByteArray& key, const unsigned char* plain, const ByteArray::size_type plain_length, ByteArray& encrypted) -{ - Aes256 aes(key); - - aes.encrypt_start(plain_length, encrypted); - aes.encrypt_continue(plain, plain_length, encrypted); - aes.encrypt_end(encrypted); - - return encrypted.size(); -} - -ByteArray::size_type Aes256::decrypt(const ByteArray& key, const ByteArray& encrypted, ByteArray& plain) -{ - Aes256 aes(key); - - aes.decrypt_start(encrypted.size()); - aes.decrypt_continue(encrypted, plain); - aes.decrypt_end(plain); - - return plain.size(); -} - -ByteArray::size_type Aes256::decrypt(const ByteArray& key, const unsigned char* encrypted, const ByteArray::size_type encrypted_length, ByteArray& plain) -{ - Aes256 aes(key); - - aes.decrypt_start(encrypted_length); - aes.decrypt_continue(encrypted, encrypted_length, plain); - aes.decrypt_end(plain); - - return plain.size(); -} - -ByteArray::size_type Aes256::encrypt_start(const ByteArray::size_type plain_length, ByteArray& encrypted) -{ - m_remainingLength = plain_length; - - // Generate salt - ByteArray::iterator it = m_salt.begin(), itEnd = m_salt.end(); - while (it != itEnd) - *(it++) = (rand() & 0xFF); - - // Calculate padding - ByteArray::size_type padding = 0; - if (m_remainingLength % BLOCK_SIZE != 0) - padding = (BLOCK_SIZE - (m_remainingLength % BLOCK_SIZE)); - m_remainingLength += padding; - - // Add salt - encrypted.insert(encrypted.end(), m_salt.begin(), m_salt.end()); - m_remainingLength += m_salt.size(); - - // Add 1 bytes for padding size - encrypted.push_back(padding & 0xFF); - ++m_remainingLength; - - // Reset buffer - m_buffer_pos = 0; - - return encrypted.size(); -} - -ByteArray::size_type Aes256::encrypt_continue(const ByteArray& plain, ByteArray& encrypted) -{ - ByteArray::const_iterator it = plain.begin(), itEnd = plain.end(); - - while(it != itEnd) { - m_buffer[m_buffer_pos++] = *(it++); - - check_and_encrypt_buffer(encrypted); - } - - return encrypted.size(); -} - -ByteArray::size_type Aes256::encrypt_continue(const unsigned char* plain, const ByteArray::size_type plain_length, ByteArray& encrypted) -{ - ByteArray::size_type i = 0; - - while(i < plain_length) { - m_buffer[m_buffer_pos++] = plain[i++]; - - check_and_encrypt_buffer(encrypted); - } - - return encrypted.size(); -} - -void Aes256::check_and_encrypt_buffer(ByteArray& encrypted) -{ - if (m_buffer_pos == BLOCK_SIZE) { - encrypt(m_buffer); - - for (m_buffer_pos = 0; m_buffer_pos < BLOCK_SIZE; ++m_buffer_pos) { - encrypted.push_back(m_buffer[m_buffer_pos]); - --m_remainingLength; - } - - m_buffer_pos = 0; - } -} - -ByteArray::size_type Aes256::encrypt_end(ByteArray& encrypted) -{ - if (m_buffer_pos > 0) { - while (m_buffer_pos < BLOCK_SIZE) - m_buffer[m_buffer_pos++] = 0; - - encrypt(m_buffer); - - for (m_buffer_pos = 0; m_buffer_pos < BLOCK_SIZE; ++m_buffer_pos) { - encrypted.push_back(m_buffer[m_buffer_pos]); - --m_remainingLength; - } - - m_buffer_pos = 0; - } - - return encrypted.size(); -} - -void Aes256::encrypt(unsigned char* buffer) -{ - unsigned char i, rcon; - - copy_key(); - add_round_key(buffer, 0); - for(i = 1, rcon = 1; i < NUM_ROUNDS; ++i) - { - sub_bytes(buffer); - shift_rows(buffer); - mix_columns(buffer); - if( !(i & 1) ) - expand_enc_key(&rcon); - add_round_key(buffer, i); - } - sub_bytes(buffer); - shift_rows(buffer); - expand_enc_key(&rcon); - add_round_key(buffer, i); -} - -ByteArray::size_type Aes256::decrypt_start(const ByteArray::size_type encrypted_length) -{ - unsigned char j; - - m_remainingLength = encrypted_length; - - // Reset salt - for(j = 0; j < m_salt.size(); ++j) - m_salt[j] = 0; - m_remainingLength -= m_salt.size(); - - // Reset buffer - m_buffer_pos = 0; - - m_decryptInitialized = false; - - return m_remainingLength; -} - -ByteArray::size_type Aes256::decrypt_continue(const ByteArray& encrypted, ByteArray& plain) -{ - ByteArray::const_iterator it = encrypted.begin(), itEnd = encrypted.end(); - - while(it != itEnd) { - m_buffer[m_buffer_pos++] = *(it++); - - check_and_decrypt_buffer(plain); - } - - return plain.size(); -} - -ByteArray::size_type Aes256::decrypt_continue(const unsigned char* encrypted, const ByteArray::size_type encrypted_length, ByteArray& plain) -{ - ByteArray::size_type i = 0; - - while(i < encrypted_length) { - m_buffer[m_buffer_pos++] = encrypted[i++]; - - check_and_decrypt_buffer(plain); - } - - return plain.size(); -} - -void Aes256::check_and_decrypt_buffer(ByteArray& plain) -{ - if (!m_decryptInitialized && m_buffer_pos == m_salt.size() + 1) { - unsigned char j; - ByteArray::size_type padding; - - // Get salt - for(j = 0; j < m_salt.size(); ++j) - m_salt[j] = m_buffer[j]; - - // Get padding - padding = (m_buffer[j] & 0xFF); - m_remainingLength -= padding + 1; - - // Start decrypting - m_buffer_pos = 0; - - m_decryptInitialized = true; - } - else if (m_decryptInitialized && m_buffer_pos == BLOCK_SIZE) { - decrypt(m_buffer); - - for (m_buffer_pos = 0; m_buffer_pos < BLOCK_SIZE; ++m_buffer_pos) - if (m_remainingLength > 0) { - plain.push_back(m_buffer[m_buffer_pos]); - --m_remainingLength; - } - - m_buffer_pos = 0; - } -} - -ByteArray::size_type Aes256::decrypt_end(ByteArray& plain) -{ - return plain.size(); -} - -void Aes256::decrypt(unsigned char* buffer) -{ - unsigned char i, rcon = 1; - - copy_key(); - for (i = NUM_ROUNDS / 2; i > 0; --i) - expand_enc_key(&rcon); - - add_round_key(buffer, NUM_ROUNDS); - shift_rows_inv(buffer); - sub_bytes_inv(buffer); - - for (i = NUM_ROUNDS, rcon = 0x80; --i;) - { - if( (i & 1) ) - expand_dec_key(&rcon); - add_round_key(buffer, i); - mix_columns_inv(buffer); - shift_rows_inv(buffer); - sub_bytes_inv(buffer); - } - add_round_key(buffer, i); -} - -void Aes256::expand_enc_key(unsigned char* rc) -{ - unsigned char i; - - m_rkey[0] = m_rkey[0] ^ sbox[m_rkey[29]] ^ (*rc); - m_rkey[1] = m_rkey[1] ^ sbox[m_rkey[30]]; - m_rkey[2] = m_rkey[2] ^ sbox[m_rkey[31]]; - m_rkey[3] = m_rkey[3] ^ sbox[m_rkey[28]]; - *rc = FE(*rc); - - for(i = 4; i < 16; i += 4) { - m_rkey[i] = m_rkey[i] ^ m_rkey[i-4]; - m_rkey[i+1] = m_rkey[i+1] ^ m_rkey[i-3]; - m_rkey[i+2] = m_rkey[i+2] ^ m_rkey[i-2]; - m_rkey[i+3] = m_rkey[i+3] ^ m_rkey[i-1]; - } - m_rkey[16] = m_rkey[16] ^ sbox[m_rkey[12]]; - m_rkey[17] = m_rkey[17] ^ sbox[m_rkey[13]]; - m_rkey[18] = m_rkey[18] ^ sbox[m_rkey[14]]; - m_rkey[19] = m_rkey[19] ^ sbox[m_rkey[15]]; - - for(i = 20; i < 32; i += 4) { - m_rkey[i] = m_rkey[i] ^ m_rkey[i-4]; - m_rkey[i+1] = m_rkey[i+1] ^ m_rkey[i-3]; - m_rkey[i+2] = m_rkey[i+2] ^ m_rkey[i-2]; - m_rkey[i+3] = m_rkey[i+3] ^ m_rkey[i-1]; - } -} - -void Aes256::expand_dec_key(unsigned char* rc) -{ - unsigned char i; - - for(i = 28; i > 16; i -= 4) { - m_rkey[i+0] = m_rkey[i+0] ^ m_rkey[i-4]; - m_rkey[i+1] = m_rkey[i+1] ^ m_rkey[i-3]; - m_rkey[i+2] = m_rkey[i+2] ^ m_rkey[i-2]; - m_rkey[i+3] = m_rkey[i+3] ^ m_rkey[i-1]; - } - - m_rkey[16] = m_rkey[16] ^ sbox[m_rkey[12]]; - m_rkey[17] = m_rkey[17] ^ sbox[m_rkey[13]]; - m_rkey[18] = m_rkey[18] ^ sbox[m_rkey[14]]; - m_rkey[19] = m_rkey[19] ^ sbox[m_rkey[15]]; - - for(i = 12; i > 0; i -= 4) { - m_rkey[i+0] = m_rkey[i+0] ^ m_rkey[i-4]; - m_rkey[i+1] = m_rkey[i+1] ^ m_rkey[i-3]; - m_rkey[i+2] = m_rkey[i+2] ^ m_rkey[i-2]; - m_rkey[i+3] = m_rkey[i+3] ^ m_rkey[i-1]; - } - - *rc = FD(*rc); - m_rkey[0] = m_rkey[0] ^ sbox[m_rkey[29]] ^ (*rc); - m_rkey[1] = m_rkey[1] ^ sbox[m_rkey[30]]; - m_rkey[2] = m_rkey[2] ^ sbox[m_rkey[31]]; - m_rkey[3] = m_rkey[3] ^ sbox[m_rkey[28]]; -} - -void Aes256::sub_bytes(unsigned char* buffer) -{ - unsigned char i = KEY_SIZE / 2; - - while (i--) - buffer[i] = sbox[buffer[i]]; -} - -void Aes256::sub_bytes_inv(unsigned char* buffer) -{ - unsigned char i = KEY_SIZE / 2; - - while (i--) - buffer[i] = sboxinv[buffer[i]]; -} - -void Aes256::copy_key() -{ - ByteArray::size_type i; - - for (i = 0; i < m_key.size(); ++i) - m_rkey[i] = m_key[i]; - for (i = 0; i < m_salt.size(); ++i) - m_rkey[i + m_key.size()] = m_salt[i]; -} - -void Aes256::add_round_key(unsigned char* buffer, const unsigned char round) -{ - unsigned char i = KEY_SIZE / 2; - - while (i--) - buffer[i] ^= m_rkey[ (round & 1) ? i + 16 : i ]; -} - -void Aes256::shift_rows(unsigned char* buffer) -{ - unsigned char i, j, k, l; /* to make it potentially parallelable :) */ - - i = buffer[1]; - buffer[1] = buffer[5]; - buffer[5] = buffer[9]; - buffer[9] = buffer[13]; - buffer[13] = i; - - j = buffer[10]; - buffer[10] = buffer[2]; - buffer[2] = j; - - k = buffer[3]; - buffer[3] = buffer[15]; - buffer[15] = buffer[11]; - buffer[11] = buffer[7]; - buffer[7] = k; - - l = buffer[14]; - buffer[14] = buffer[6]; - buffer[6] = l; -} - -void Aes256::shift_rows_inv(unsigned char* buffer) -{ - unsigned char i, j, k, l; /* same as above :) */ - - i = buffer[1]; - buffer[1] = buffer[13]; - buffer[13] = buffer[9]; - buffer[9] = buffer[5]; - buffer[5] = i; - - j = buffer[2]; - buffer[2] = buffer[10]; - buffer[10] = j; - - k = buffer[3]; - buffer[3] = buffer[7]; - buffer[7] = buffer[11]; - buffer[11] = buffer[15]; - buffer[15] = k; - - l = buffer[6]; - buffer[6] = buffer[14]; - buffer[14] = l; -} - -void Aes256::mix_columns(unsigned char* buffer) -{ - unsigned char i, a, b, c, d, e; - - for (i = 0; i < 16; i += 4) - { - a = buffer[i]; - b = buffer[i + 1]; - c = buffer[i + 2]; - d = buffer[i + 3]; - - e = a ^ b ^ c ^ d; - - buffer[i ] ^= e ^ rj_xtime(a^b); - buffer[i + 1] ^= e ^ rj_xtime(b^c); - buffer[i + 2] ^= e ^ rj_xtime(c^d); - buffer[i + 3] ^= e ^ rj_xtime(d^a); - } -} - -void Aes256::mix_columns_inv(unsigned char* buffer) -{ - unsigned char i, a, b, c, d, e, x, y, z; - - for (i = 0; i < 16; i += 4) - { - a = buffer[i]; - b = buffer[i + 1]; - c = buffer[i + 2]; - d = buffer[i + 3]; - - e = a ^ b ^ c ^ d; - z = rj_xtime(e); - x = e ^ rj_xtime(rj_xtime(z^a^c)); y = e ^ rj_xtime(rj_xtime(z^b^d)); - - buffer[i ] ^= x ^ rj_xtime(a^b); - buffer[i + 1] ^= y ^ rj_xtime(b^c); - buffer[i + 2] ^= x ^ rj_xtime(c^d); - buffer[i + 3] ^= y ^ rj_xtime(d^a); - } -} - -inline unsigned char rj_xtime(unsigned char x) -{ - return (x & 0x80) ? ((x << 1) ^ 0x1b) : (x << 1); -} diff --git a/aes/aes256.hpp b/aes/aes256.hpp deleted file mode 100644 index 90173a6..0000000 --- a/aes/aes256.hpp +++ /dev/null @@ -1,88 +0,0 @@ -/* - * aes256.hpp - * - * Copyright (c) 2014, Danilo Treffiletti <urban82@gmail.com> - * All rights reserved. - * - * This file is part of Aes256. - * - * Aes256 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 2.1 - * of the License, or (at your option) any later version. - * - * Aes256 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 Aes256. - * If not, see <http://www.gnu.org/licenses/>. - */ - -#ifndef AES256_HPP -#define AES256_HPP - -#include <EASTL/vector.h> - -typedef eastl::vector<unsigned char> ByteArray; - -#define BLOCK_SIZE 16 - -class Aes256 { - - public: - Aes256(const ByteArray& key); - ~Aes256(); - - static ByteArray::size_type encrypt(const ByteArray& key, const ByteArray& plain, ByteArray& encrypted); - static ByteArray::size_type encrypt(const ByteArray& key, const unsigned char* plain, const ByteArray::size_type plain_length, ByteArray& encrypted); - static ByteArray::size_type decrypt(const ByteArray& key, const ByteArray& encrypted, ByteArray& plain); - static ByteArray::size_type decrypt(const ByteArray& key, const unsigned char* encrypted, const ByteArray::size_type encrypted_length, ByteArray& plain); - - ByteArray::size_type encrypt_start(const ByteArray::size_type plain_length, ByteArray& encrypted); - ByteArray::size_type encrypt_continue(const ByteArray& plain, ByteArray& encrypted); - ByteArray::size_type encrypt_continue(const unsigned char* plain, const ByteArray::size_type plain_length, ByteArray& encrypted); - ByteArray::size_type encrypt_end(ByteArray& encrypted); - - ByteArray::size_type decrypt_start(const ByteArray::size_type encrypted_length); - ByteArray::size_type decrypt_continue(const ByteArray& encrypted, ByteArray& plain); - ByteArray::size_type decrypt_continue(const unsigned char* encrypted, const ByteArray::size_type encrypted_length, ByteArray& plain); - ByteArray::size_type decrypt_end(ByteArray& plain); - - private: - ByteArray m_key; - ByteArray m_salt; - ByteArray m_rkey; - - unsigned char m_buffer[3 * BLOCK_SIZE]; - unsigned char m_buffer_pos; - ByteArray::size_type m_remainingLength; - - bool m_decryptInitialized; - - void check_and_encrypt_buffer(ByteArray& encrypted); - void check_and_decrypt_buffer(ByteArray& plain); - - void encrypt(unsigned char *buffer); - void decrypt(unsigned char *buffer); - - void expand_enc_key(unsigned char *rc); - void expand_dec_key(unsigned char *rc); - - void sub_bytes(unsigned char *buffer); - void sub_bytes_inv(unsigned char *buffer); - - void copy_key(); - - void add_round_key(unsigned char *buffer, const unsigned char round); - - void shift_rows(unsigned char *buffer); - void shift_rows_inv(unsigned char *buffer); - - void mix_columns(unsigned char *buffer); - void mix_columns_inv(unsigned char *buffer); -}; - -#endif /* AES256_HPP */ diff --git a/examples/driver-aes.bat b/examples/driver-aes.bat deleted file mode 100644 index a8e5c7e..0000000 --- a/examples/driver-aes.bat +++ /dev/null @@ -1,27 +0,0 @@ -@echo off -set SERVICE_NAME=ksocket -set DRIVER="%~dp0\driver-aes.sys" - -net session >nul 2>&1 -if NOT %ERRORLEVEL% EQU 0 ( - echo ERROR: This script requires Administrator privileges! - pause - exit /b 1 -) - -echo --------------------------------------- -echo -- Service Name: %SERVICE_NAME% -echo -- Driver......: %DRIVER% -echo --------------------------------------- - -sc create %SERVICE_NAME% binPath= %DRIVER% type= kernel -echo --------------------------------------- -sc start %SERVICE_NAME% -echo --------------------------------------- -sc query %SERVICE_NAME% -echo [PRESS A KEY TO STOP THE DRIVER] -pause -sc stop %SERVICE_NAME% -sc delete %SERVICE_NAME% -echo Done. -timeout /t 3 diff --git a/examples/driver-aes.cpp b/examples/driver-aes.cpp deleted file mode 100644 index 06b1dcd..0000000 --- a/examples/driver-aes.cpp +++ /dev/null @@ -1,63 +0,0 @@ -#include <aes/aes256.hpp> - -extern "C" { -#include <ksocket/berkeley.h> -#include <ksocket/ksocket.h> -#include <ksocket/wsk.h> - -DRIVER_INITIALIZE DriverEntry; -DRIVER_UNLOAD DriverUnload; - -#define DebuggerPrint(...) \ - DbgPrintEx(DPFLTR_IHVDRIVER_ID, DPFLTR_ERROR_LEVEL, __VA_ARGS__); - -NTSTATUS -NTAPI -DriverEntry(_In_ PDRIVER_OBJECT DriverObject, - _In_ PUNICODE_STRING RegistryPath) { - UNREFERENCED_PARAMETER(DriverObject); - UNREFERENCED_PARAMETER(RegistryPath); - - { - ByteArray key = {0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, - 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, - 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, - 0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f}; - ByteArray enc, dec; - const char secret_message[] = "This is a top secret message."; - const size_t secret_message_len = strlen(secret_message); - - enc.clear(); - dec.clear(); - - { - Aes256 aes(key); - aes.encrypt_start(secret_message_len, enc); - aes.encrypt_continue((unsigned char *)secret_message, secret_message_len, - enc); - aes.encrypt_end(enc); - } - - { - Aes256 aes(key); - aes.decrypt_start(enc.size()); - aes.decrypt_continue(enc.data(), enc.size(), dec); - aes.decrypt_end(dec); - } - - if (memcmp(secret_message, dec.data(), secret_message_len) != 0) { - DebuggerPrint("%s\n", "AES secret message differs!"); - DebuggerPrint("Original.: %s\n", secret_message); - } - DebuggerPrint("Decrypted: %s\n", dec.data()); - } - - return STATUS_SUCCESS; -} - -VOID DriverUnload(_In_ struct _DRIVER_OBJECT *DriverObject) { - UNREFERENCED_PARAMETER(DriverObject); - - DebuggerPrint("Bye."); -} -} |