Add enable DK Menu option

This commit is contained in:
GaryOderNichts 2024-06-19 00:36:04 +02:00
parent e054e05787
commit fe99770abb
6 changed files with 227 additions and 8 deletions

View File

@ -16,7 +16,7 @@ TOPDIR ?= $(CURDIR)
APP_NAME := DRXUtil
APP_SHORTNAME := DRXUtil
APP_AUTHOR := GaryOderNichts
APP_VERSION := 1.0
APP_VERSION := 1.1
include $(DEVKITPRO)/wut/share/wut_rules

View File

@ -0,0 +1,184 @@
#include "EnableDKMenuScreen.hpp"
#include "Utils.hpp"
#include "Gfx.hpp"
#include <nsysccr/cdc.h>
#include <nsysccr/cfg.h>
namespace {
bool SetBoardConfig(uint8_t byte)
{
CCRCDCUicConfig cfg{};
// custom config id which was added by the gamepad cfw
cfg.configId = 4;
// board config byte + crc16
cfg.size = 3;
cfg.data[0] = byte;
uint16_t crc = CCRCDCCalcCRC16(cfg.data, 1);
cfg.data[1] = crc & 0xff;
cfg.data[2] = (crc >> 8) & 0xff;
if (CCRCDCPerSetUicConfig(CCR_CDC_DESTINATION_DRC0, &cfg) != 0) {
return false;
}
// Also update the cached eeprom
return CCRCFGSetCachedEeprom(0, 0x106, cfg.data, cfg.size) == 0;
}
bool GetBoardConfig(uint8_t& boardConfig)
{
uint8_t data[3];
if (CCRCFGGetCachedEeprom(0, 0x106, data, 3) != 0) {
return false;
}
uint16_t crc = (uint16_t) data[2] << 8 | data[1];
if (CCRCDCCalcCRC16(&data[0], 1) != crc) {
return false;
}
boardConfig = data[0];
return true;
}
}
EnableDKMenuScreen::EnableDKMenuScreen()
: mConfirm(false),
mBoardConfig(0),
mDKMenuEnabled(false),
mErrorText()
{
}
EnableDKMenuScreen::~EnableDKMenuScreen()
{
}
void EnableDKMenuScreen::Draw()
{
DrawTopBar("EnableDKMenuScreen");
switch (mState)
{
case STATE_GET_BOARD_CONFIG:
Gfx::Print(Gfx::SCREEN_WIDTH / 2, Gfx::SCREEN_HEIGHT / 2, 64, Gfx::COLOR_TEXT, "Retrieving board config...", Gfx::ALIGN_CENTER);
break;
case STATE_SELECT:
Gfx::Print(Gfx::SCREEN_WIDTH / 2, 128, 64, Gfx::COLOR_TEXT,
Utils::sprintf("DK Menu is currently %s.\nDo you want to %s it?",
mDKMenuEnabled ? "enabled" : "disabled",
mDKMenuEnabled ? "disable" : "enable"),
Gfx::ALIGN_HORIZONTAL | Gfx::ALIGN_TOP);
for (int i = 0; i < 2; i++) {
int yOff = 360 + static_cast<int>(i) * 100;
Gfx::DrawRectFilled(0, yOff, Gfx::SCREEN_WIDTH, 100, Gfx::COLOR_ALT_BACKGROUND);
Gfx::Print(68, yOff + 100 / 2, 50, Gfx::COLOR_TEXT, i == 0 ? "Back" : (mDKMenuEnabled ? "Disable" : "Enable"), Gfx::ALIGN_VERTICAL);
if (mConfirm == !!i) {
Gfx::DrawRect(0, yOff, Gfx::SCREEN_WIDTH, 100, 8, Gfx::COLOR_HIGHLIGHTED);
}
}
break;
case STATE_CONFIRM:
Gfx::Print(Gfx::SCREEN_WIDTH / 2, Gfx::SCREEN_HEIGHT / 2, 64, Gfx::COLOR_TEXT,
Utils::sprintf("Are you sure you want to %s the DK Menu?\n"
"(Note that updating the board config\nonly works with a modified firmware)",
mDKMenuEnabled ? "disable" : "enable"), Gfx::ALIGN_CENTER);
break;
case STATE_UPDATE:
Gfx::Print(Gfx::SCREEN_WIDTH / 2, Gfx::SCREEN_HEIGHT / 2, 64, Gfx::COLOR_TEXT, "Updating board config...", Gfx::ALIGN_CENTER);
break;
case STATE_DONE:
Gfx::Print(Gfx::SCREEN_WIDTH / 2, Gfx::SCREEN_HEIGHT / 2, 64, Gfx::COLOR_TEXT,
Utils::sprintf("Done! DK Menu has been %s.\n%s",
mDKMenuEnabled ? "disabled" : "enabled",
mDKMenuEnabled ? "" : "(Hold L + ZL while powering on the Gamepad to open.)"),
Gfx::ALIGN_CENTER);
break;
case STATE_ERROR:
Gfx::Print(Gfx::SCREEN_WIDTH / 2, Gfx::SCREEN_HEIGHT / 2, 64, Gfx::COLOR_ERROR, "Error!\n" + mErrorText, Gfx::ALIGN_CENTER);
break;
}
if (mState == STATE_SELECT) {
DrawBottomBar("\ue07d Navigate", "\ue044 Exit", "\ue000 Confirm / \ue001 Back");
} else if (mState == STATE_CONFIRM) {
DrawBottomBar(nullptr, "\ue044 Exit", "\ue000 Confirm / \ue001 Back");
} else if (mState == STATE_UPDATE || mState == STATE_GET_BOARD_CONFIG) {
DrawBottomBar(nullptr, "Please wait...", nullptr);
} else {
DrawBottomBar(nullptr, "\ue044 Exit", "\ue001 Back");
}
}
bool EnableDKMenuScreen::Update(VPADStatus& input)
{
switch (mState)
{
case STATE_GET_BOARD_CONFIG:
if (!GetBoardConfig(mBoardConfig)) {
mErrorText = "Failed to retrieve current board configuration.";
mState = STATE_ERROR;
break;
}
mDKMenuEnabled = (mBoardConfig & 0x0C) == 0;
mState = STATE_SELECT;
break;
case STATE_SELECT:
if (input.trigger & VPAD_BUTTON_B) {
return false;
}
if (input.trigger & VPAD_BUTTON_A) {
if (mConfirm) {
mState = STATE_CONFIRM;
} else {
return false;
}
break;
}
if (input.trigger & (VPAD_BUTTON_DOWN | VPAD_BUTTON_UP)) {
mConfirm = !mConfirm;
}
break;
case STATE_CONFIRM:
if (input.trigger & VPAD_BUTTON_B) {
return false;
}
if (input.trigger & VPAD_BUTTON_A) {
mState = STATE_UPDATE;
break;
}
break;
case STATE_UPDATE: {
uint8_t newBoardConfig = mBoardConfig;
// Clear/Set the is retail bits
if (mDKMenuEnabled) {
newBoardConfig |= 0x04;
} else {
newBoardConfig &= ~0x0C;
}
if (!SetBoardConfig(newBoardConfig)) {
mErrorText = "Failed to set board config.\n(Is the firmware modified properly?)";
mState = STATE_ERROR;
break;
}
mState = STATE_DONE;
break;
}
case STATE_DONE:
case STATE_ERROR:
if (input.trigger & VPAD_BUTTON_B) {
return false;
}
break;
}
return true;
}

View File

@ -0,0 +1,31 @@
#pragma once
#include "Screen.hpp"
#include <map>
#include <nsysccr/cdc.h>
class EnableDKMenuScreen : public Screen
{
public:
EnableDKMenuScreen();
virtual ~EnableDKMenuScreen();
void Draw();
bool Update(VPADStatus& input);
private:
enum State {
STATE_GET_BOARD_CONFIG,
STATE_SELECT,
STATE_CONFIRM,
STATE_UPDATE,
STATE_DONE,
STATE_ERROR,
} mState = STATE_GET_BOARD_CONFIG;
bool mConfirm;
uint8_t mBoardConfig;
bool mDKMenuEnabled;
std::string mErrorText;
};

View File

@ -4,6 +4,7 @@
#include "FlashScreen.hpp"
#include "InfoScreen.hpp"
#include "SetRegionScreen.hpp"
#include "EnableDKMenuScreen.hpp"
#include <vector>
@ -12,6 +13,7 @@ MenuScreen::MenuScreen()
{ MENU_ID_INFO, { 0xf085, "Show DRC/DRH information" }},
{ MENU_ID_FLASH, { 0xf1c9, "Flash firmware" }},
{ MENU_ID_SET_REGION, { 0xf0ac, "Set region" }},
{ MENU_ID_ENABLE_DKMENU, { 0xf188, "Enable DK Menu" }},
{ MENU_ID_ABOUT, { 0xf05a, "About DRXUtil" }},
// { MENU_ID_EXIT, { 0xf057, "Exit" }},
})
@ -78,6 +80,9 @@ bool MenuScreen::Update(VPADStatus& input)
case MENU_ID_SET_REGION:
mSubscreen = std::make_unique<SetRegionScreen>();
break;
case MENU_ID_ENABLE_DKMENU:
mSubscreen = std::make_unique<EnableDKMenuScreen>();
break;
case MENU_ID_ABOUT:
mSubscreen = std::make_unique<AboutScreen>();
break;

View File

@ -21,6 +21,7 @@ private:
MENU_ID_INFO,
MENU_ID_FLASH,
MENU_ID_SET_REGION,
MENU_ID_ENABLE_DKMENU,
MENU_ID_ABOUT,
MENU_ID_MIN = MENU_ID_INFO,

View File

@ -34,6 +34,4 @@ private:
REGION_AUSTRALIA = 6,
} mRegion = REGION_JAPAN;
std::map<Region, std::string> mRegionEntries;
std::string mErrorText;
};