Initial commit

source/Key.hpp

@@ -0,0 +1,72 @@
+/*
+ * Copyright (c) 2018 shchmue
+ *
+ * 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/>.
+ */
+
+#pragma once
+
+#include <fstream>
+#include <string>
+#include <vector>
+
+#include <switch.h>
+
+#include "Common.hpp"
+
+typedef std::vector<u8> byte_vector;
+
+class Key {
+public:
+    Key(std::string name, u64 xx_hash, byte_vector hash, u8 length, byte_vector key);
+    // init with hash only
+    Key(std::string name, u64 xx_hash, byte_vector hash, u8 length);
+    // init with key only
+    Key(std::string name, u8 length, byte_vector key);
+    // temp key, no name stored
+    Key(byte_vector key, u8 length);
+    // key to be assigned later
+    Key(std::string name, u8 length);
+    // for declaration only
+    Key();
+
+    bool found() const { return is_found; }
+    void set_found() { is_found = true; }
+
+    // write key to file
+    void save_key(std::ofstream &file);
+
+    static const size_t get_saved_key_count() { return saved_key_count; }
+
+    // return CTR-decrypted data
+    byte_vector aes_decrypt_ctr(const byte_vector &data, byte_vector iv);
+    // return ECB-decrypted data
+    byte_vector aes_decrypt_ecb(const byte_vector &data);
+    // return CMAC of data
+    byte_vector cmac(byte_vector data);
+    // find key in buffer by hash, optionally specify start offset
+    void find_key(const byte_vector &buffer);
+    // get key encryption key
+    byte_vector generate_kek(Key &master_key, const Key &kek_seed, const Key &key_seed);
+
+    byte_vector key;
+
+private:
+    std::string name;
+    u64 xx_hash;
+    byte_vector hash;
+    u8 length;
+    bool is_found = false;
+
+    static size_t saved_key_count;
+};