add basic kvg integration, and some of the worst C i've ever written

include/tpm.h

@@ -0,0 +1 @@
+void tpm_write(char* index, char* bytes);

src/KVS/main.c

@@ -12,8 +12,10 @@ int main(int argc, char **argv) {
 	char* gscver = "0.5.229";
 	char* gsctype = "Cr50";
 
+	
+
 	// only allow 2 characters (option & newline)
-	char choice[2];
+	char choice[3];
 
 	ui_header(fwver, kernver, tpmver, fwmp, gscver, gsctype);
 	printf("1) Flash new kernver via /dev/tpm0 (REQ. UNENROLLED)\n");
@@ -24,9 +26,8 @@ int main(int argc, char **argv) {
 	printf("> ");
 	fgets(choice, sizeof(choice), stdin);
 	if (choice[strlen(choice) - 1] == '\n') {
-		choice[strlen(choice) - 1] = '\0';
-	}
-	printf("You entered: %s\n", choice);
+        choice[strlen(choice) - 1] = '\0';
+    }
 
 	if (!strcmp(choice, "1")) {
 		ui_flash("tpm0");

src/KVS/tpm.c

@@ -0,0 +1,5 @@
+#include <stdio.h>
+
+void tpm_write(char* index, char* bytes){
+    printf("wip, index: %s, bytes: %s", index, bytes);
+}

src/KVS/ui.c

@@ -1,7 +1,78 @@
 #include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <stdbool.h>
+#include <stdint.h>
+#include <unistd.h>
+
+
+#include "tpm.h"
+#include "hex_utils.h"
 
 void ui_flash(char* flashtype) {
-	printf("wip\n");
+	// i feel like this is some of the dirtiest C that
+	// i've ever written, please ignore it
+	#define V0_SIZE 38
+	#define V1_SIZE 120
+
+	char kvgout_v0[V0_SIZE + 1];
+	char kvgout_v1[V1_SIZE + 1];
+
+	char kerninput[12];
+	char structtype[4];
+
+	printf("What kernver would you like to flash? \n");
+	printf("> ");
+	fgets(kerninput, sizeof(kerninput), stdin);
+	// nya
+	if (kerninput[strlen(kerninput) - 1] == '\n') {
+        kerninput[strlen(kerninput) - 1] = '\0';
+    }
+	if (!is_valid_hex(kerninput)){
+		fprintf(stderr, "Your kernver, %s, was an invalid input. Not hex.", kerninput);
+		exit(1);
+	}
+
+	printf("Does your device have lightmode (v0) or darkmode (v1) recovery? Please type either v0 or v1.\n");
+	printf("> ");
+	fgets(structtype, sizeof(structtype), stdin);
+	if (structtype[strlen(structtype) - 1] == '\n') {
+        structtype[strlen(structtype) - 1] = '\0';
+    }
+
+	// the output of strcmp if it fails is True
+	if (strcmp(structtype, "v0") && strcmp(structtype, "v1")){
+		fprintf(stderr, "Invalid struct type %s, valid types are v0 and v1\n", structtype);
+		exit(1);
+	}
+
+	// we check if its *false* since strcmp returns true if failing
+	if (!strcmp(structtype, "v0")){ 
+		char cmd[128];
+
+		snprintf(cmd, sizeof(cmd), "kvg %s --ver=0", kerninput);
+    	FILE* fp = popen(cmd, "r");
+    	fgets(kvgout_v0, sizeof(kvgout_v0), fp);
+		fclose(fp);
+	} else if (!strcmp(structtype, "v1")) {
+		char cmd[128];
+
+		snprintf(cmd, sizeof(cmd), "kvg %s --ver=1", kerninput);
+    	FILE* fp = popen(cmd, "r");
+    	fgets(kvgout_v1, sizeof(kvgout_v1), fp);
+		fclose(fp);
+	}
+	
+
+	if (flashtype == "tpm0"){
+		if (!strcmp(structtype, "v0")) {
+			tpm_write("0x1008", kvgout_v0);
+		} else if (!strcmp(structtype, "v1")) {
+			tpm_write("0x1008", kvgout_v1);
+		}
+	} else if (flashtype == "rmasmoke"){
+		printf("using rmasmoke\n");
+	}
 }
 
 void ui_header(char* fwver, char* kernver, char* tpmver, char* fwmp, char* gscver, char* gsctype){