Divecomputer download: try to offer only those devices that make sense

If the user selects a Uemis divecomputer, don't show serial devices.
If the user selects a serial divecomputer, don't show the Uemis
filesystem.

Signed-off-by: Dirk Hohndel <dirk@hohndel.org>

android.cpp

@@ -29,7 +29,7 @@ const char *system_default_filename(void)
 	return strdup(system_default_filename.toUtf8().data());
 }
 
-int enumerate_devices (device_callback_t callback, void *userdata)
+int enumerate_devices (device_callback_t callback, void *userdata, int dc_type)
 {
 	/* FIXME: we need to enumerate in some other way on android */
 	/* qtserialport maybee? */

display.h

@@ -55,7 +55,12 @@ extern int is_default_dive_computer_device(const char *);
 extern int is_default_dive_computer(const char *, const char *);
 
 typedef void (*device_callback_t)(const char *name, void *userdata);
-int enumerate_devices(device_callback_t callback, void *userdata);
+
+#define DC_TYPE_SERIAL 1
+#define DC_TYPE_UEMIS 2
+#define DC_TYPE_OTHER 3
+
+int enumerate_devices(device_callback_t callback, void *userdata, int dc_type);
 
 extern const char *default_dive_computer_vendor;
 extern const char *default_dive_computer_product;

linux.c

@@ -49,12 +49,17 @@ const char *system_default_filename(void)
 	return buffer;
 }
 
-int enumerate_devices(device_callback_t callback, void *userdata)
+int enumerate_devices(device_callback_t callback, void *userdata, int dc_type)
 {
 	int index = -1;
 	DIR *dp = NULL;
 	struct dirent *ep = NULL;
 	size_t i;
+	FILE *file;
+	char *line = NULL;
+	char *fname;
+	size_t len;
+	if (dc_type != DC_TYPE_UEMIS) {
 		const char *dirname = "/dev";
 		const char *patterns[] = {
 			"ttyUSB*",
@@ -63,10 +68,6 @@ int enumerate_devices(device_callback_t callback, void *userdata)
 			"rfcomm*",
 			NULL
 		};
-	FILE *file;
-	char *line = NULL;
-	char *fname;
-	size_t len;
 
 		dp = opendir(dirname);
 		if (dp == NULL) {
@@ -90,7 +91,8 @@ int enumerate_devices(device_callback_t callback, void *userdata)
 			}
 		}
 		closedir(dp);
-
+	}
+	if (dc_type != DC_TYPE_SERIAL) {
 		file = fopen("/proc/mounts", "r");
 		if (file == NULL)
 			return index;
@@ -120,7 +122,7 @@ int enumerate_devices(device_callback_t callback, void *userdata)
 
 		free(line);
 		fclose(file);
-
+	}
 	return index;
 }
 

macos.c

@@ -45,12 +45,13 @@ const char *system_default_filename(void)
 	return buffer;
 }
 
-int enumerate_devices(device_callback_t callback, void *userdata)
+int enumerate_devices(device_callback_t callback, void *userdata, int dc_type)
 {
 	int index = -1;
 	DIR *dp = NULL;
 	struct dirent *ep = NULL;
 	size_t i;
+	if (dc_type != DC_TYPE_UEMIS) {
 		const char *dirname = "/dev";
 		const char *patterns[] = {
 			"tty.*",
@@ -79,9 +80,31 @@ int enumerate_devices(device_callback_t callback, void *userdata)
 				}
 			}
 		}
-	// TODO: list UEMIS mount point from /proc/mounts
-
 		closedir(dp);
+	}
+	if (dc_type != DC_TYPE_SERIAL) {
+		const char *dirname = "/Volumes";
+		dp = opendir(dirname);
+		if (dp == NULL) {
+			return -1;
+		}
+
+		while ((ep = readdir(dp)) != NULL) {
+			if (fnmatch("UEMISSDA", ep->d_name, 0) == 0) {
+				char filename[1024];
+				int n = snprintf(filename, sizeof(filename), "%s/%s", dirname, ep->d_name);
+				if (n >= sizeof(filename)) {
+					closedir(dp);
+					return -1;
+				}
+				callback(filename, userdata);
+				if (is_default_dive_computer_device(filename))
+					index = i;
+				break;
+			}
+		}
+		closedir(dp);
+	}
 	return index;
 }
 

qt-ui/downloadfromdivecomputer.cpp

@@ -56,7 +56,7 @@ DownloadFromDCWidget::DownloadFromDCWidget(QWidget *parent, Qt::WindowFlags f) :
 
 	progress_bar_text = "";
 
-	fill_device_list();
+	fill_device_list(DC_TYPE_OTHER);
 	fill_computer_list();
 
 	ui.chooseDumpFile->setEnabled(ui.dumpToFile->isChecked());
@@ -104,7 +104,7 @@ void DownloadFromDCWidget::updateState(states state)
 		return;
 
 	if (state == INITIAL) {
-		fill_device_list();
+		fill_device_list(DC_TYPE_OTHER);
 		ui.progressBar->hide();
 		markChildrenAsEnabled();
 		timer->stop();
@@ -176,6 +176,7 @@ void DownloadFromDCWidget::updateState(states state)
 
 void DownloadFromDCWidget::on_vendor_currentIndexChanged(const QString &vendor)
 {
+	int dcType = DC_TYPE_SERIAL;
 	QAbstractItemModel *currentModel = ui.product->model();
 	if (!currentModel)
 		return;
@@ -183,6 +184,10 @@ void DownloadFromDCWidget::on_vendor_currentIndexChanged(const QString &vendor)
 	productModel = new QStringListModel(productList[vendor]);
 	ui.product->setModel(productModel);
 
+	if (vendor == QString("Uemis"))
+		dcType = DC_TYPE_UEMIS;
+	fill_device_list(dcType);
+
 	// Memleak - but deleting gives me a crash.
 	//currentModel->deleteLater();
 }
@@ -429,11 +434,11 @@ static void fillDeviceList(const char *name, void *data)
 	comboBox->addItem(name);
 }
 
-void DownloadFromDCWidget::fill_device_list()
+void DownloadFromDCWidget::fill_device_list(int dc_type)
 {
 	int deviceIndex;
 	ui.device->clear();
-	deviceIndex = enumerate_devices(fillDeviceList, ui.device);
+	deviceIndex = enumerate_devices(fillDeviceList, ui.device, dc_type);
 	if (deviceIndex >= 0)
 		ui.device->setCurrentIndex(deviceIndex);
 }

qt-ui/downloadfromdivecomputer.h

@@ -68,7 +68,7 @@ private:
 	QStringListModel *vendorModel;
 	QStringListModel *productModel;
 	void fill_computer_list();
-	void fill_device_list();
+	void fill_device_list(int dc_type);
 	QString logFile;
 	QString dumpFile;
 	QTimer *timer;

windows.c

@@ -36,11 +36,13 @@ const char *system_default_filename(void)
 	return buffer;
 }
 
-int enumerate_devices(device_callback_t callback, void *userdata)
+int enumerate_devices(device_callback_t callback, void *userdata, int dc_type)
 {
+	int index = -1;
+	DWORD i;
+	if (dc_type != DC_TYPE_UEMIS) {
 		// Open the registry key.
 		HKEY hKey;
-	int index = -1;
 		LONG rc = RegOpenKeyEx(HKEY_LOCAL_MACHINE, "HARDWARE\\DEVICEMAP\\SERIALCOMM", 0, KEY_QUERY_VALUE, &hKey);
 		if (rc != ERROR_SUCCESS) {
 			return -1;
@@ -53,7 +55,6 @@ int enumerate_devices(device_callback_t callback, void *userdata)
 			RegCloseKey(hKey);
 			return -1;
 		}
-	DWORD i;
 		for (i = 0; i < count; ++i) {
 			// Get the value name, data and type.
 			char name[512], data[512];
@@ -86,6 +87,37 @@ int enumerate_devices(device_callback_t callback, void *userdata)
 		}
 
 		RegCloseKey(hKey);
+	}
+	if (dc_type != DC_TYPE_SERIAL) {
+		int i;
+		const int bufdef = 512;
+		const char *dlabels[] = {"UEMISSDA", NULL};
+		char bufname[bufdef], bufval[bufdef], *p;
+		DWORD bufname_len;
+
+		/* add drive letters that match labels */
+		memset(bufname, 0, bufdef);
+		bufname_len = bufdef;
+		if (GetLogicalDriveStringsA(bufname_len, bufname)) {
+			p = bufname;
+
+			while (*p) {
+				memset(bufval, 0, bufdef);
+				if (GetVolumeInformationA(p, bufval, bufdef, NULL, NULL, NULL, NULL, 0)) {
+					for (i = 0; dlabels[i] != NULL; i++)
+						if (!strcmp(bufval, dlabels[i])) {
+							char data[512];
+							snprintf(data, sizeof(data), "%s (%s)", p, dlabels[i]);
+							callback(data, userdata);
+							if (is_default_dive_computer_device(p))
+								index = i;
+							i++;
+						}
+				}
+				p = &p[strlen(p) + 1];
+			}
+		}
+	}
 	return index;
 }