mecha.c

/*
 * Copyright (c) 2021 MechaResearch
 *
 * 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 <debug.h>
#include <stdio.h>
#include <string.h>
#include <kernel.h>
#include <sifrpc.h>

#include "mecha.h"
#include "mechaproxy.h"

static SifRpcClientData_t SifRpcClientMechaScmd;

static unsigned char RpcBuffer[0x1000] ALIGNED(64);

#define _printf(args...) //scr_printf(args)

int MechaInit()
{
	SifInitRpc(0);

	nopdelay();
	while(SifBindRpc(&SifRpcClientMechaScmd, MECHAPROXY_SCMD_HEADER, 0) < 0 || SifRpcClientMechaScmd.server == NULL)
	{
		_printf("libsecr: bind failed\n");
	}

	return 1;
}

void MechaDeinit()
{
	memset(&SifRpcClientMechaScmd, 0, sizeof(SifRpcClientData_t));
}

int MechaScmd(u8 cmd, void *input, u8 inputlength, void *output)
{
	struct MechaScmdParams *params = (struct MechaScmdParams *) RpcBuffer;

	params->cmd = cmd;
	params->inputlength = inputlength;
	memcpy(params->input, input, inputlength);

	if(SifCallRpc(&SifRpcClientMechaScmd, 1, 0, RpcBuffer, sizeof(RpcBuffer), RpcBuffer, sizeof(RpcBuffer), NULL, NULL) < 0)
	{
		_printf("MechaScmd: rpc error\n");
		return 0;
	}

	memcpy(output, params->output, 16);

	return params->result;
}



char getMechaVersion(uint8_t *data)
{
	u8 input[1];
	u8 output[16];
	input[0] = 0x00;
	if (MechaScmd(0x03, input, sizeof(input), output) != 1)
	{
		return 0;
	}

	memcpy(data, &output[0], 3);
	
	return 1;
}

char getMechaBuildDate(uint8_t *data)
{
	u8 input[1];
	u8 output[16];
	input[0] = 0xfd;
	if (MechaScmd(0x03, input, sizeof(input), output) != 1 || output[0] != 0)
	{
		_printf("Failed to read the build date!\n");
		return 0;
	}

	memcpy(data, &output[1], 5);
	
	return 1;
}

char OpenConfig(uint8_t id, char write, uint8_t blocks)
{
	u8 input[3];
	u8 output[16];
	
	input[0] = write; // 0 = read, 1 = write
	input[1] = id; // id
	input[2] = blocks; // blocks
	
	if (MechaScmd(0x40, input, sizeof(input), output) != 1)
	{
		if (output[0] != 0)
		{
			CloseConfig();

			if (MechaScmd(0x40, input, sizeof(input), output) != 1 || output[0] != 0)
			{
				_printf("Failed to cmd OpenConfig!\n");
				return 0;
			}
		}
	}
	
	return 1;
}

char ReadConfig(uint8_t *data)
{
	u8 input[0];
	u8 output[16];
	
	if (MechaScmd(0x41, input, sizeof(input), output) != 1)
	{
		_printf("Failed to cmd ReadConfig!\n");
		return 0;
	}
	
	memcpy(data, output, 16);
	
	return 1;
}

char WriteConfig(const uint8_t *data)
{
	u8 input[16];
	u8 output[16];
	
	memcpy(input, data, 16);
	
	if (MechaScmd(0x42, input, sizeof(input), output) != 1 || output[0] != 0)
	{
		_printf("Failed to cmd WriteConfig!\n");
		return 0;
	}
	
	return 1;
}

char CloseConfig()
{
	u8 input[0];
	u8 output[16];
	output[0] = 1;
	
	while (output[0])
	{
		if (MechaScmd(0x43, input, sizeof(input), output) != 1 || (output[0] != 0 && output[0] != 1))
		{
			_printf("Failed to cmd CloseConfig!\n");
			return 0;
		}
	}
	
	return 1;
}

char ReadNVM(uint16_t offset, uint16_t *data)
{
	u8 input[2];
	u8 output[16];
	input[0] = offset >> 8;
	input[1] = offset;
	if (MechaScmd(0x0A, input, sizeof(input), output) != 1 || output[0] != 0)
	{
		_printf("Failed to cmd 0x0A!\n");
		return 0;
	}
	
	*data = output[1];
	*data <<= 8;
	*data |= output[2];
	
	return 1;
}

char WriteNVM(uint16_t offset, uint16_t data)
{
	u8 input[4];
	u8 output[16];
	input[0] = offset >> 8;
	input[1] = offset;
	input[2] = data >> 8;
	input[3] = data;
	output[0] = 1;
	
	while (output[0])
	{
		if (MechaScmd(0x0B, input, sizeof(input), output) != 1 || (output[0] != 0 && output[0] != 1))
		{
			_printf("Failed to cmd WriteNVM! (%d)\n", offset);
			return 0;
		}
	}
	
	return 1;
}

char IsNVMUnlocked()
{
	uint16_t data;
	if (!ReadNVM(400, &data))
		return 0;
		
	if (!WriteNVM(400, data))
		return 0;

	return 1;
}