From 33bdb731adb4f2309ecff1815d4df8d7b80084d4 Mon Sep 17 00:00:00 2001
From: Nikias Bassen
Date: Tue, 9 Jan 2024 02:02:20 +0100
Subject: irecovery: Print a message for unsupported commands in Debug USB
 (KIS) mode

---
 tools/irecovery.c | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/tools/irecovery.c b/tools/irecovery.c
index 5546ac2..6303b55 100644
--- a/tools/irecovery.c
+++ b/tools/irecovery.c
@@ -584,6 +584,8 @@ int main(int argc, char* argv[])
 	if (device)
 		debug("Connected to %s, model %s, cpid 0x%04x, bdid 0x%02x\n", device->product_type, device->hardware_model, device->chip_id, device->board_id);
 
+	const struct irecv_device_info *devinfo = irecv_get_device_info(client);
+
 	switch (action) {
 		case kResetDevice:
 			irecv_reset(client);
@@ -596,6 +598,10 @@ int main(int argc, char* argv[])
 			break;
 
 		case kSendCommand:
+			if (devinfo->pid == 0x1881) {
+				printf("Shell is not available in Debug USB (KIS) mode.\n");
+				break;
+			}
 			if (_is_breq_command(argument)) {
 				error = irecv_send_command_breq(client, argument, 1);
 			} else {
@@ -605,6 +611,10 @@ int main(int argc, char* argv[])
 			break;
 
 		case kSendExploit:
+			if (devinfo->pid == 0x1881) {
+				printf("Shell is not available in Debug USB (KIS) mode.\n");
+				break;
+			}
 			if (argument != NULL) {
 				irecv_event_subscribe(client, IRECV_PROGRESS, &progress_cb, NULL);
 				error = irecv_send_file(client, argument, 0);
@@ -618,10 +628,18 @@ int main(int argc, char* argv[])
 			break;
 
 		case kStartShell:
+			if (devinfo->pid == 0x1881) {
+				printf("This feature is not supported in Debug USB (KIS) mode.\n");
+				break;
+			}
 			init_shell(client);
 			break;
 
 		case kSendScript:
+			if (devinfo->pid == 0x1881) {
+				printf("This feature is not supported in Debug USB (KIS) mode.\n");
+				break;
+			}
 			buffer_read_from_filename(argument, &buffer, &buffer_length);
 			if (buffer) {
 				buffer[buffer_length] = '\0';
@@ -638,7 +656,6 @@ int main(int argc, char* argv[])
 			break;
 
 		case kShowMode: {
-			const struct irecv_device_info *devinfo = irecv_get_device_info(client);
 			irecv_get_mode(client, &mode);
 			printf("%s Mode", mode_to_str(mode));
 			if (devinfo->pid == 0x1881) {
@@ -648,6 +665,10 @@ int main(int argc, char* argv[])
 			break;
 		}
 		case kRebootToNormalMode:
+			if (devinfo->pid == 0x1881) {
+				printf("This feature is not supported in Debug USB (KIS) mode.\n");
+				break;
+			}
 			error = irecv_setenv(client, "auto-boot", "true");
 			if (error != IRECV_E_SUCCESS) {
 				debug("%s\n", irecv_strerror(error));
-- 
cgit v1.1-32-gdbae