BAHAHAHAHAHAHAHH

src/DSi_NAND.cpp

@@ -0,0 +1,173 @@
+/*
+    Copyright 2016-2021 Arisotura
+
+    This file is part of melonDS.
+
+    melonDS is free software: you can redistribute it and/or modify it under
+    the terms of the GNU General Public License as published by the Free
+    Software Foundation, either version 3 of the License, or (at your option)
+    any later version.
+
+    melonDS 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 General Public License for more details.
+
+    You should have received a copy of the GNU General Public License along
+    with melonDS. If not, see http://www.gnu.org/licenses/.
+*/
+
+#include <stdio.h>
+
+#include "DSi.h"
+#include "DSi_AES.h"
+#include "DSi_NAND.h"
+
+#include "sha1/sha1.hpp"
+#include "tiny-AES-c/aes.hpp"
+
+#include "fatfs/ff.h"
+
+
+namespace DSi_NAND
+{
+
+u8 FATIV[16];
+u8 FATKey[16];
+
+
+void Init()
+{
+    // init NAND crypto
+
+    SHA1_CTX sha;
+    u8 tmp[20];
+
+    SHA1Init(&sha);
+    SHA1Update(&sha, DSi::eMMC_CID, 16);
+    SHA1Final(tmp, &sha);
+
+    DSi_AES::Swap16(FATIV, tmp);
+
+    u8 keyX[16];
+    *(u32*)&keyX[0] = (u32)DSi::ConsoleID;
+    *(u32*)&keyX[4] = (u32)DSi::ConsoleID ^ 0x24EE6906;
+    *(u32*)&keyX[8] = (u32)(DSi::ConsoleID >> 32) ^ 0xE65B601D;
+    *(u32*)&keyX[12] = (u32)(DSi::ConsoleID >> 32);
+
+    u8 keyY[16];
+    *(u32*)&keyY[0] = 0x0AB9DC76;
+    *(u32*)&keyY[4] = 0xBD4DC4D3;
+    *(u32*)&keyY[8] = 0x202DDD1D;
+    *(u32*)&keyY[12] = 0xE1A00005;
+
+    DSi_AES::DeriveNormalKey(keyX, keyY, tmp);
+    DSi_AES::Swap16(FATKey, tmp);
+}
+
+void SetupFATCrypto(AES_ctx* ctx, u32 ctr)
+{
+    u8 iv[16];
+    memcpy(iv, FATIV, 16);
+
+    u32 res;
+    res = iv[15] + (ctr & 0xFF);
+    iv[15] = (res & 0xFF);
+    res = iv[14] + ((ctr >> 8) & 0xFF) + (res >> 8);
+    iv[14] = (res & 0xFF);
+    res = iv[13] + ((ctr >> 16) & 0xFF) + (res >> 8);
+    iv[13] = (res & 0xFF);
+    res = iv[12] + (ctr >> 24) + (res >> 8);
+    iv[12] = (res & 0xFF);
+    iv[11] += (res >> 8);
+
+    AES_init_ctx_iv(ctx, FATKey, iv);
+}
+
+u32 ReadFATBlock(u64 addr, u32 len, u8* buf)
+{
+    u32 ctr = (u32)(addr >> 4);
+
+    AES_ctx ctx;
+    SetupFATCrypto(&ctx, ctr);
+
+    fseek(DSi::SDMMCFile, addr, SEEK_SET);
+    u32 res = fread(buf, len, 1, DSi::SDMMCFile);
+    if (!res) return 0;
+
+    for (u32 i = 0; i < len; i += 16)
+    {
+        //printf("BLOCK %d: IV=", i);
+        //for (int k = 0; k < 16; k++) printf("%02X:", ctx.Iv[k]);
+        //printf("\n");
+
+        u8 tmp[16];
+        DSi_AES::Swap16(tmp, &buf[i]);
+        AES_CTR_xcrypt_buffer(&ctx, tmp, 16);
+        DSi_AES::Swap16(&buf[i], tmp);
+    }
+
+    return len;
+}
+
+
+UINT FF_ReadNAND(BYTE* buf, LBA_t sector, UINT num)
+{
+    printf("READ %08X %08X\n", sector, num);
+
+    // TODO: allow selecting other partitions?
+    u64 baseaddr = 0x10EE00;
+
+    u64 blockaddr = baseaddr + (sector * 0x200ULL);
+
+    u32 res = ReadFATBlock(blockaddr, num*0x200, buf);
+    return res >> 9;
+}
+
+UINT FF_WriteNAND(BYTE* buf, LBA_t sector, UINT num)
+{
+    // TODO!
+    printf("!! WRITE %08X %08X\n", sector, num);
+    return 0;
+}
+
+
+void PatchTSC()
+{
+    // TEST ZONE
+
+    u8 dorp[0x200];
+    ReadFATBlock(0, 0x200, dorp);
+
+    for (int i = 0; i < 0x200; i+=16)
+    {
+        for (int j = 0; j < 16; j++)
+        {
+            printf("%02X ", dorp[i+j]);
+        }
+        printf("\n");
+    }
+
+    ReadFATBlock(0x0010EE00, 0x200, dorp);
+    printf("FUKA\n");
+    for (int i = 0; i < 0x200; i+=16)
+    {
+        for (int j = 0; j < 16; j++)
+        {
+            printf("%02X ", dorp[i+j]);
+        }
+        printf("\n");
+    }
+
+    ff_disk_open(FF_ReadNAND, FF_WriteNAND);
+
+    FRESULT res;
+    FATFS fs;
+    res = f_mount(&fs, "0:", 0);
+    printf("mount=%d\n", res);
+
+    FIL file;
+    res = f_open(&file, "0:/shared1/TWLCFG0.dat", FA_READ);
+    printf("blarg=%d\n", res);
+}
+
+}