nvmefwupdater/nvmelib.hpp at main · userlandkernel/nvmefwupdater · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include <IOKit/IOKitLib.h>
#include <CoreFoundation/CoreFoundation.h>

static char* kBFHLoaderPaths[] = {
	(char*)"/usr/standalone/firmware/BFH/s3e-bfh-loader.bin",
	(char*)"/usr/standalone/firmware/S4E/msp_fubfh_loader.bin",
	(char*)"/usr/standalone/firmware/M9/msp_fubfh_loader.bin",
	(char*)"/usr/standalone/firmware/Turks/msp_fubfh_loader.bin",
};

static char* kBFHLoaderFiles[] = {
	(char*)"BFH/s3e-bfh-loader.bin",
	(char*)"S4E/msp_fubfh_loader.bin",
	(char*)"M9/msp_fubfh_loader.bin",
	(char*)NULL,
	(char*)NULL,
	(char*)NULL,
	(char*)NULL,
	(char*)NULL,
	(char*)NULL,
	(char*)NULL,
	(char*)NULL,
	(char*)"Turks/msp_fubfh_loader.bin",
};

static const char* mspTypes[] = {
	(char*)"S3E",
	(char*)"S4E",
	(char*)"M9",
	(char*)"t302",
	(char*)"Turks",
};

enum AppleEmbeddedNVMeControllerAction {
    kNVMECTL_sendNVMECommandAction = 2,
    kNVMECTL_isBFHModeAction = 3,
    kNVMECTL_performBFHAction = 4,
    kNVMECTL_getNandDescriptorAction = 5,
    kNVMECTL_setNVMeStateAction = 6,
    kNVMECTL_setPCIPortStateAction = 7,
    kNVMECTL_setBFHGPIOAction = 8,
    kNVMECTL_getMSPTypeAction = 0xA,
    kNVMECTL_firmwareVersionCheckAction = 0xC,
};

typedef struct NVMeIdentifyControllerStruct {
    char unknown[0x1000];
} NVMeIdentifyControllerStruct; // Unimplemented

class NVMeUpdateLib {

	public:
		NVMeUpdateLib();
		~NVMeUpdateLib();
		 void DumpIdentifyData();
		 void FirmwareDownload(void* nandDescriptor, void* responseData);
		 int FirmwareValidate(char* fileBuffer);
		 kern_return_t FirmwareVersionCheck(char* data, size_t size, uint32_t mspType);
		 kern_return_t GetClogMismatch(bool *misMatchout);
		 kern_return_t GetMSPType(uint32_t *mspTypeOut);
		 kern_return_t GetNANDDescriptor(uint64_t* nandDescriptor);
		 kern_return_t IdentifyController(NVMeIdentifyControllerStruct* identity);
		 kern_return_t IsBFHMode(bool* bfhModeOut);
		 kern_return_t PerformBFH(char* bfhData, size_t bfhSize);
		 kern_return_t SendNVMeCommand(uint32_t selector, uint64_t* command, uint64_t commandSize);
		 kern_return_t SetBFHMode(bool bfhMode);
		 kern_return_t SetNVMeState(bool enableA, bool enableB);
		 kern_return_t SetPCIPortState(bool A, bool B, uint32_t C);

	private:
		io_service_t svc;
		io_connect_t conn;


};