mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-12-21 01:42:12 +00:00
73 lines
2.8 KiB
C++
73 lines
2.8 KiB
C++
|
/*
|
||
|
|
* Copyright (c) 2018-2020 Atmosphère-NX
|
||
|
|
*
|
||
|
|
* This program is free software; you can redistribute it and/or modify it
|
||
|
|
* under the terms and conditions of the GNU General Public License,
|
||
|
|
* version 2, as published by the Free Software Foundation.
|
||
|
|
*
|
||
|
|
* This program is distributed in the hope it will be useful, but WITHOUT
|
||
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
||
|
|
* more details.
|
||
|
|
*
|
||
|
|
* You should have received a copy of the GNU General Public License
|
||
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
*/
|
||
|
|
#include <exosphere.hpp>
|
||
|
|
#include "se_execute.hpp"
|
||
|
|
|
||
|
|
namespace ams::se {
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
void SetMessageSize(volatile SecurityEngineRegisters *SE, size_t src_size) {
|
||
|
|
/* Set the message size. */
|
||
|
|
reg::Write(SE->SE_SHA_MSG_LENGTH[0], src_size * BITSIZEOF(u8));
|
||
|
|
reg::Write(SE->SE_SHA_MSG_LENGTH[1], 0);
|
||
|
|
reg::Write(SE->SE_SHA_MSG_LENGTH[2], 0);
|
||
|
|
reg::Write(SE->SE_SHA_MSG_LENGTH[3], 0);
|
||
|
|
|
||
|
|
/* Set the message remaining size. */
|
||
|
|
reg::Write(SE->SE_SHA_MSG_LEFT[0], src_size * BITSIZEOF(u8));
|
||
|
|
reg::Write(SE->SE_SHA_MSG_LEFT[1], 0);
|
||
|
|
reg::Write(SE->SE_SHA_MSG_LEFT[2], 0);
|
||
|
|
reg::Write(SE->SE_SHA_MSG_LEFT[3], 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
void GetHashResult(volatile SecurityEngineRegisters *SE, void *dst, size_t dst_size) {
|
||
|
|
/* Copy out the words. */
|
||
|
|
const int num_words = dst_size / sizeof(u32);
|
||
|
|
for (int i = 0; i < num_words; ++i) {
|
||
|
|
const u32 word = reg::Read(SE->SE_HASH_RESULT[i]);
|
||
|
|
util::StoreBigEndian(static_cast<u32 *>(dst) + i, word);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
void CalculateSha256(Sha256Hash *dst, const void *src, size_t src_size) {
|
||
|
|
/* Get the engine. */
|
||
|
|
auto *SE = GetRegisters();
|
||
|
|
|
||
|
|
/* Configure the engine to perform SHA256 "encryption". */
|
||
|
|
reg::Write(SE->SE_CONFIG, SE_REG_BITS_ENUM(CONFIG_ENC_MODE, SHA256),
|
||
|
|
SE_REG_BITS_ENUM(CONFIG_DEC_MODE, AESMODE_KEY128),
|
||
|
|
SE_REG_BITS_ENUM(CONFIG_ENC_ALG, SHA),
|
||
|
|
SE_REG_BITS_ENUM(CONFIG_DEC_ALG, NOP),
|
||
|
|
SE_REG_BITS_ENUM(CONFIG_DST, HASH_REG));
|
||
|
|
|
||
|
|
/* Begin a hardware hash operation. */
|
||
|
|
reg::Write(SE->SE_SHA_CONFIG, SE_REG_BITS_VALUE(SHA_CONFIG_HW_INIT_HASH, 1));
|
||
|
|
|
||
|
|
/* Set the message size. */
|
||
|
|
SetMessageSize(SE, src_size);
|
||
|
|
|
||
|
|
/* Execute the operation. */
|
||
|
|
ExecuteOperation(SE, SE_OPERATION_OP_START, nullptr, 0, src, src_size);
|
||
|
|
|
||
|
|
/* Get the result. */
|
||
|
|
GetHashResult(SE, dst, sizeof(*dst));
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|