irecovery: Make sure to send certain commands with bRequest set to 1

1 files changed, 20 insertions, 2 deletions

diff --git a/tools/irecovery.c b/tools/irecovery.c
index 392f912..faf0a92 100644
--- a/tools/irecovery.c
+++ b/tools/irecovery.c
@@ -199,6 +199,16 @@ static void print_devices() {
 	}
 }
 
+static int _is_breq_command(const char* cmd)
+{
+	return (
+		!strcmp(cmd, "go")
+		|| !strcmp(cmd, "bootx")
+		|| !strcmp(cmd, "reboot")
+		|| !strcmp(cmd, "memboot")
+	);
+}
+
 static void parse_command(irecv_client_t client, unsigned char* command, unsigned int size) {
 	char* cmd = strdup((char*)command);
 	char* action = strtok(cmd, " ");
@@ -269,7 +279,11 @@ static void init_shell(irecv_client_t client) {
 
 		char* cmd = readline("> ");
 		if (cmd && *cmd) {
-			error = irecv_send_command(client, cmd);
+			if (_is_breq_command(cmd)) {
+				error = irecv_send_command_breq(client, cmd, 1);
+			} else {
+				error = irecv_send_command(client, cmd);
+			}
 			if (error != IRECV_E_SUCCESS) {
 				quit = 1;
 			}
@@ -560,7 +574,11 @@ int main(int argc, char* argv[]) {
 			break;
 
 		case kSendCommand:
-			error = irecv_send_command(client, argument);
+			if (_is_breq_command(argument)) {
+				error = irecv_send_command_breq(client, argument, 1);
+			} else {
+				error = irecv_send_command(client, argument);
+			}
 			debug("%s\n", irecv_strerror(error));
 			break;