2 files changed, 61 insertions, 35 deletions

diff --git a/partitionmanager.cpp b/partitionmanager.cpp
index fdc3fba44..9b4de4e07 100644
--- a/partitionmanager.cpp
+++ b/partitionmanager.cpp
@@ -1625,48 +1625,74 @@ int TWPartitionManager::usb_storage_enable(void) {
 	DataManager::GetValue(TW_HAS_DUAL_STORAGE, has_dual);
 	DataManager::GetValue(TW_HAS_DATA_MEDIA, has_data_media);
 	if (has_dual == 1 && has_data_media == 0) {
-		Part = Find_Partition_By_Path(DataManager::GetSettingsStoragePath());
-		if (Part == NULL) {
-			LOGE("Unable to locate volume information.");
-			return false;
-		}
-		if (!Part->UnMount(true))
-			return false;
+		sprintf(lun_file, CUSTOM_LUN_FILE, 1);
+		if (!TWFunc::Path_Exists(lun_file)) {
+			// Device doesn't have multiple lun files, mount current storage
+			Part = Find_Partition_By_Path(DataManager::GetCurrentStoragePath());
+			if (Part == NULL) {
+				LOGE("Unable to locate volume information for USB storage mode.");
+				return false;
+			}
+			if (!Part->UnMount(true))
+				return false;
 
-		sprintf(lun_file, CUSTOM_LUN_FILE, 0);
-		if ((fd = open(lun_file, O_WRONLY)) < 0) {
-			LOGE("Unable to open ums lunfile '%s': (%s)\n", lun_file, strerror(errno));
-			return false;
-		}
+			sprintf(lun_file, CUSTOM_LUN_FILE, 0);
+			if ((fd = open(lun_file, O_WRONLY)) < 0) {
+				LOGE("Unable to open ums lunfile '%s': (%s)\n", lun_file, strerror(errno));
+				return false;
+			}
 
-		if (write(fd, Part->Actual_Block_Device.c_str(), Part->Actual_Block_Device.size()) < 0) {
-			LOGE("Unable to write to ums lunfile '%s': (%s)\n", lun_file, strerror(errno));
+			if (write(fd, Part->Actual_Block_Device.c_str(), Part->Actual_Block_Device.size()) < 0) {
+				LOGE("Unable to write to ums lunfile '%s': (%s)\n", lun_file, strerror(errno));
+				close(fd);
+				return false;
+			}
 			close(fd);
-			return false;
-		}
-		close(fd);
+		} else {
+			// Device has multiple lun files
+			Part = Find_Partition_By_Path(DataManager::GetSettingsStoragePath());
+			if (Part == NULL) {
+				LOGE("Unable to locate volume information.");
+				return false;
+			}
+			if (!Part->UnMount(true))
+				return false;
 
-		DataManager::GetValue(TW_EXTERNAL_PATH, ext_path);
-		Part = Find_Partition_By_Path(ext_path);
-		if (Part == NULL) {
-			LOGE("Unable to locate volume information.\n");
-			return false;
-		}
-		if (!Part->UnMount(true))
-			return false;
+			sprintf(lun_file, CUSTOM_LUN_FILE, 0);
+			if ((fd = open(lun_file, O_WRONLY)) < 0) {
+				LOGE("Unable to open ums lunfile '%s': (%s)\n", lun_file, strerror(errno));
+				return false;
+			}
 
-		sprintf(lun_file, CUSTOM_LUN_FILE, 1);
-		if ((fd = open(lun_file, O_WRONLY)) < 0) {
-			LOGE("Unable to open ums lunfile '%s': (%s)\n", lun_file, strerror(errno));
-			return false;
-		}
+			if (write(fd, Part->Actual_Block_Device.c_str(), Part->Actual_Block_Device.size()) < 0) {
+				LOGE("Unable to write to ums lunfile '%s': (%s)\n", lun_file, strerror(errno));
+				close(fd);
+				return false;
+			}
+			close(fd);
 
-		if (write(fd, Part->Actual_Block_Device.c_str(), Part->Actual_Block_Device.size()) < 0) {
-			LOGE("Unable to write to ums lunfile '%s': (%s)\n", lun_file, strerror(errno));
+			DataManager::GetValue(TW_EXTERNAL_PATH, ext_path);
+			Part = Find_Partition_By_Path(ext_path);
+			if (Part == NULL) {
+				LOGE("Unable to locate volume information.\n");
+				return false;
+			}
+			if (!Part->UnMount(true))
+				return false;
+
+			sprintf(lun_file, CUSTOM_LUN_FILE, 1);
+			if ((fd = open(lun_file, O_WRONLY)) < 0) {
+				LOGE("Unable to open ums lunfile '%s': (%s)\n", lun_file, strerror(errno));
+				return false;
+			}
+
+			if (write(fd, Part->Actual_Block_Device.c_str(), Part->Actual_Block_Device.size()) < 0) {
+				LOGE("Unable to write to ums lunfile '%s': (%s)\n", lun_file, strerror(errno));
+				close(fd);
+				return false;
+			}
 			close(fd);
-			return false;
 		}
-		close(fd);
 	} else {
 		if (has_data_media == 0)
 			ext_path = DataManager::GetCurrentStoragePath();
diff --git a/variables.h b/variables.h
index 2418a2492..5de6cd6e5 100644
--- a/variables.h
+++ b/variables.h
@@ -17,7 +17,7 @@
 #ifndef _VARIABLES_HEADER_
 #define _VARIABLES_HEADER_
 
-#define TW_VERSION_STR              "2.3.0.3"
+#define TW_VERSION_STR              "2.3.0.4"
 
 #define TW_USE_COMPRESSION_VAR      "tw_use_compression"
 #define TW_IGNORE_IMAGE_SIZE        "tw_ignore_image_size"