From df9984e1234530952069850078bce0aa37b05f8b Mon Sep 17 00:00:00 2001
From: Linus Torvalds <torvalds@linux-foundation.org>
Date: Thu, 8 Jun 2017 22:16:14 -0700
Subject: [PATCH] Add initial Scubapro G2 frontend

The back-end parser seems to be the same as for the Uwatec Smart (aka
Galileo Sol).  At least that's the assumption right now.

The downloader just uses USB HID (very similar to EON Steel) rather than
the horrible IrDA thing.

There's also eventually a BLE thing, but that's for the future.

This is an unholy mixture of the Uwatec Smart downloader logic and the
EON Steel usbhid transfer code.  The back-end is pure Uwatec Smart
(model 0x11, same as Galileo Sol).

I'm not at all sure this gets everything right, but it downloads
*something*.

Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
---
 examples/common.c                |   1 +
 include/libdivecomputer/common.h |   2 +
 src/Makefile.am                  |   1 +
 src/descriptor.c                 |   3 +
 src/device.c                     |   4 +
 src/parser.c                     |   3 +
 src/scubapro_g2.c                | 371 +++++++++++++++++++++++++++++++++++++++
 src/scubapro_g2.h                |  39 ++++
 8 files changed, 424 insertions(+)
 create mode 100644 src/scubapro_g2.c
 create mode 100644 src/scubapro_g2.h

diff --git a/examples/common.c b/examples/common.c
index 3c4561b..5ebdd77 100644
--- a/examples/common.c
+++ b/examples/common.c
@@ -54,6 +54,7 @@ static const backend_table_t g_backends[] = {
 	{"aladin",      DC_FAMILY_UWATEC_ALADIN,       0x3F},
 	{"memomouse",   DC_FAMILY_UWATEC_MEMOMOUSE,    0},
 	{"smart",       DC_FAMILY_UWATEC_SMART,        0x10},
+	{"g2",          DC_FAMILY_UWATEC_SMART,        0x10},
 	{"meridian",    DC_FAMILY_UWATEC_MERIDIAN,     0x20},
 	{"sensus",      DC_FAMILY_REEFNET_SENSUS,      1},
 	{"sensuspro",   DC_FAMILY_REEFNET_SENSUSPRO,   2},
diff --git a/include/libdivecomputer/common.h b/include/libdivecomputer/common.h
index 67398d1..fdaaa1b 100644
--- a/include/libdivecomputer/common.h
+++ b/include/libdivecomputer/common.h
@@ -59,6 +59,8 @@ typedef enum dc_family_t {
 	DC_FAMILY_UWATEC_MEMOMOUSE,
 	DC_FAMILY_UWATEC_SMART,
 	DC_FAMILY_UWATEC_MERIDIAN,
+	/* We'll enumerate the Scubapro G2 under Uwatec */
+	DC_FAMILY_SCUBAPRO_G2,
 	/* Oceanic */
 	DC_FAMILY_OCEANIC_VTPRO = (4 << 16),
 	DC_FAMILY_OCEANIC_VEO250,
diff --git a/src/Makefile.am b/src/Makefile.am
index b98e57e..92de94b 100644
--- a/src/Makefile.am
+++ b/src/Makefile.am
@@ -30,6 +30,7 @@ libdivecomputer_la_SOURCES = \
 	suunto_vyper2.h suunto_vyper2.c \
 	suunto_d9.h suunto_d9.c suunto_d9_parser.c \
 	suunto_eonsteel.h suunto_eonsteel.c suunto_eonsteel_parser.c \
+	scubapro_g2.h scubapro_g2.c \
 	reefnet_sensus.h reefnet_sensus.c reefnet_sensus_parser.c \
 	reefnet_sensuspro.h reefnet_sensuspro.c reefnet_sensuspro_parser.c \
 	reefnet_sensusultra.h reefnet_sensusultra.c reefnet_sensusultra_parser.c \
diff --git a/src/descriptor.c b/src/descriptor.c
index 9ae3e77..cb0bed6 100644
--- a/src/descriptor.c
+++ b/src/descriptor.c
@@ -89,6 +89,7 @@ static const dc_descriptor_t g_descriptors[] = {
 	/* Suunto EON Steel */
 #ifdef USBHID
 	{"Suunto", "EON Steel", DC_FAMILY_SUUNTO_EONSTEEL, 0},
+	{"Scubapro", "G2", DC_FAMILY_SCUBAPRO_G2, 0x11},
 #endif
 	/* Uwatec Aladin */
 	{"Uwatec", "Aladin Air Twin",     DC_FAMILY_UWATEC_ALADIN, 0x1C},
@@ -429,6 +430,8 @@ dc_descriptor_get_transport (dc_descriptor_t *descriptor)
 		return DC_TRANSPORT_USB;
 	else if (descriptor->type == DC_FAMILY_SUUNTO_EONSTEEL)
 		return DC_TRANSPORT_USB;
+	else if (descriptor->type == DC_FAMILY_SCUBAPRO_G2)
+		return DC_TRANSPORT_USB;
 	else if (descriptor->type == DC_FAMILY_UWATEC_SMART)
 		return DC_TRANSPORT_IRDA;
 	else
diff --git a/src/device.c b/src/device.c
index 2f3e1af..d36710a 100644
--- a/src/device.c
+++ b/src/device.c
@@ -50,6 +50,7 @@
 #include "cressi_leonardo.h"
 #include "zeagle_n2ition3.h"
 #include "atomics_cobalt.h"
+#include "scubapro_g2.h"
 #include "shearwater_petrel.h"
 #include "shearwater_predator.h"
 #include "diverite_nitekq.h"
@@ -126,6 +127,9 @@ dc_device_open (dc_device_t **out, dc_context_t *context, dc_descriptor_t *descr
 	case DC_FAMILY_SUUNTO_EONSTEEL:
 		rc = suunto_eonsteel_device_open (&device, context);
 		break;
+	case DC_FAMILY_SCUBAPRO_G2:
+		rc = scubapro_g2_device_open (&device, context);
+		break;
 	case DC_FAMILY_UWATEC_ALADIN:
 		rc = uwatec_aladin_device_open (&device, context, name);
 		break;
diff --git a/src/parser.c b/src/parser.c
index ebbc6a6..fea1454 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -96,6 +96,9 @@ dc_parser_new_internal (dc_parser_t **out, dc_context_t *context, dc_family_t fa
 	case DC_FAMILY_UWATEC_MEMOMOUSE:
 		rc = uwatec_memomouse_parser_create (&parser, context, devtime, systime);
 		break;
+	case DC_FAMILY_SCUBAPRO_G2:
+		rc = uwatec_smart_parser_create (&parser, context, 0x11, devtime, systime);
+		break;
 	case DC_FAMILY_UWATEC_SMART:
 	case DC_FAMILY_UWATEC_MERIDIAN:
 		rc = uwatec_smart_parser_create (&parser, context, model, devtime, systime);
diff --git a/src/scubapro_g2.c b/src/scubapro_g2.c
new file mode 100644
index 0000000..31fee20
--- /dev/null
+++ b/src/scubapro_g2.c
@@ -0,0 +1,371 @@
+/*
+ * libdivecomputer
+ *
+ * Copyright (C) 2008 Jef Driesen
+ *           (C) 2017 Linus Torvalds
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
+ * MA 02110-1301 USA
+ */
+
+#include <stdlib.h> // malloc, free
+#include <string.h>	// strncmp, strstr
+
+#include "scubapro_g2.h"
+#include "context-private.h"
+#include "device-private.h"
+#include "array.h"
+#include "usbhid.h"
+
+#define ISINSTANCE(device) dc_device_isinstance((device), &scubapro_g2_device_vtable)
+
+typedef struct scubapro_g2_device_t {
+	dc_device_t base;
+	dc_usbhid_t *usbhid;
+	unsigned int address;
+	unsigned int timestamp;
+	unsigned int devtime;
+	dc_ticks_t systime;
+} scubapro_g2_device_t;
+
+static dc_status_t scubapro_g2_device_set_fingerprint (dc_device_t *device, const unsigned char data[], unsigned int size);
+static dc_status_t scubapro_g2_device_dump (dc_device_t *abstract, dc_buffer_t *buffer);
+static dc_status_t scubapro_g2_device_foreach (dc_device_t *abstract, dc_dive_callback_t callback, void *userdata);
+static dc_status_t scubapro_g2_device_close (dc_device_t *abstract);
+
+static const dc_device_vtable_t scubapro_g2_device_vtable = {
+	sizeof(scubapro_g2_device_t),
+	DC_FAMILY_SCUBAPRO_G2,
+	scubapro_g2_device_set_fingerprint, /* set_fingerprint */
+	NULL, /* read */
+	NULL, /* write */
+	scubapro_g2_device_dump, /* dump */
+	scubapro_g2_device_foreach, /* foreach */
+	scubapro_g2_device_close /* close */
+};
+
+static dc_status_t
+scubapro_g2_extract_dives (dc_device_t *device, const unsigned char data[], unsigned int size, dc_dive_callback_t callback, void *userdata);
+
+#define PACKET_SIZE 64
+static int receive_data(scubapro_g2_device_t *g2, unsigned char *buffer, int size)
+{
+	while (size) {
+		unsigned char buf[PACKET_SIZE];
+		size_t transferred = 0;
+		dc_status_t rc;
+		int len;
+
+		rc = dc_usbhid_read(g2->usbhid, buf, PACKET_SIZE, &transferred);
+		if (rc != DC_STATUS_SUCCESS) {
+			ERROR(g2->base.context, "read interrupt transfer failed");
+			return -1;
+		}
+		if (transferred != PACKET_SIZE) {
+			ERROR(g2->base.context, "incomplete read interrupt transfer (got %zu, expected %d)", transferred, PACKET_SIZE);
+			return -1;
+		}
+		len = buf[0];
+		if (len >= PACKET_SIZE) {
+			ERROR(g2->base.context, "read interrupt transfer returns impossible packet size (%d)", len);
+			return -1;
+		}
+		HEXDUMP (g2->base.context, DC_LOGLEVEL_DEBUG, "rcv", buf+1, len);
+		if (len > size) {
+			ERROR(g2->base.context, "receive result buffer too small - truncating");
+			len = size;
+		}
+		memcpy(buffer, buf+1, len);
+		size -= len;
+		buffer += len;
+	}
+	return 0;
+}
+
+static dc_status_t
+scubapro_g2_transfer(scubapro_g2_device_t *g2, const unsigned char command[], unsigned int csize, unsigned char answer[], unsigned int asize)
+{
+	unsigned char buf[PACKET_SIZE];
+	dc_status_t status = DC_STATUS_SUCCESS;
+	size_t transferred = 0;
+
+	if (csize >= PACKET_SIZE) {
+		ERROR(g2->base.context, "command too big (%d)", csize);
+		return DC_STATUS_INVALIDARGS;
+	}
+
+	buf[0] = csize;
+	memcpy(buf+1, command, csize);
+	status = dc_usbhid_write(g2->usbhid, buf, csize+1, &transferred);
+	if (status != DC_STATUS_SUCCESS) {
+		ERROR(g2->base.context, "Failed to send the command.");
+		return status;
+	}
+
+	if (receive_data(g2, answer, asize) < 0) {
+		ERROR(g2->base.context, "Failed to receive the answer.");
+		return DC_STATUS_IO;
+	}
+
+	return DC_STATUS_SUCCESS;
+}
+
+
+dc_status_t
+scubapro_g2_device_open(dc_device_t **out, dc_context_t *context)
+{
+	dc_status_t status = DC_STATUS_SUCCESS;
+	scubapro_g2_device_t *device = NULL;
+
+	if (out == NULL)
+		return DC_STATUS_INVALIDARGS;
+
+	// Allocate memory.
+	device = (scubapro_g2_device_t *) dc_device_allocate (context, &scubapro_g2_device_vtable);
+	if (device == NULL) {
+		ERROR(context, "Failed to allocate memory.");
+		return DC_STATUS_NOMEMORY;
+	}
+
+	// Set the default values.
+	device->usbhid = NULL;
+	device->address = 0;
+	device->timestamp = 0;
+	device->systime = (dc_ticks_t) -1;
+	device->devtime = 0;
+
+	// Open the irda socket.
+	status = dc_usbhid_open(&device->usbhid, context, 0x2e6c, 0x3201);
+	if (status != DC_STATUS_SUCCESS) {
+		ERROR (context, "Failed to open USB device");
+		goto error_free;
+	}
+
+	*out = (dc_device_t*) device;
+
+	return DC_STATUS_SUCCESS;
+
+error_free:
+	dc_device_deallocate ((dc_device_t *) device);
+	return status;
+}
+
+
+static dc_status_t
+scubapro_g2_device_close (dc_device_t *abstract)
+{
+	scubapro_g2_device_t *device = (scubapro_g2_device_t*) abstract;
+
+	dc_usbhid_close(device->usbhid);
+
+	return DC_STATUS_SUCCESS;
+}
+
+
+static dc_status_t
+scubapro_g2_device_set_fingerprint (dc_device_t *abstract, const unsigned char data[], unsigned int size)
+{
+	scubapro_g2_device_t *device = (scubapro_g2_device_t*) abstract;
+
+	if (size && size != 4)
+		return DC_STATUS_INVALIDARGS;
+
+	if (size)
+		device->timestamp = array_uint32_le (data);
+	else
+		device->timestamp = 0;
+
+	return DC_STATUS_SUCCESS;
+}
+
+
+static dc_status_t
+scubapro_g2_device_dump (dc_device_t *abstract, dc_buffer_t *buffer)
+{
+	scubapro_g2_device_t *device = (scubapro_g2_device_t*) abstract;
+	dc_status_t rc = DC_STATUS_SUCCESS;
+
+	// Erase the current contents of the buffer.
+	if (!dc_buffer_clear (buffer)) {
+		ERROR (abstract->context, "Insufficient buffer space available.");
+		return DC_STATUS_NOMEMORY;
+	}
+
+	// Enable progress notifications.
+	dc_event_progress_t progress = EVENT_PROGRESS_INITIALIZER;
+	device_event_emit (&device->base, DC_EVENT_PROGRESS, &progress);
+
+	// Read the model number.
+	unsigned char cmd_model[1] = {0x10};
+	unsigned char model[1] = {0};
+	rc = scubapro_g2_transfer (device, cmd_model, sizeof (cmd_model), model, sizeof (model));
+	if (rc != DC_STATUS_SUCCESS)
+		return rc;
+
+	// Read the serial number.
+	unsigned char cmd_serial[1] = {0x14};
+	unsigned char serial[4] = {0};
+	rc = scubapro_g2_transfer (device, cmd_serial, sizeof (cmd_serial), serial, sizeof (serial));
+	if (rc != DC_STATUS_SUCCESS)
+		return rc;
+
+	// Read the device clock.
+	unsigned char cmd_devtime[1] = {0x1A};
+	unsigned char devtime[4] = {0};
+	rc = scubapro_g2_transfer (device, cmd_devtime, sizeof (cmd_devtime), devtime, sizeof (devtime));
+	if (rc != DC_STATUS_SUCCESS)
+		return rc;
+
+	// Store the clock calibration values.
+	device->systime = dc_datetime_now ();
+	device->devtime = array_uint32_le (devtime);
+
+	// Update and emit a progress event.
+	progress.current += 9;
+	device_event_emit (&device->base, DC_EVENT_PROGRESS, &progress);
+
+	// Emit a clock event.
+	dc_event_clock_t clock;
+	clock.systime = device->systime;
+	clock.devtime = device->devtime;
+	device_event_emit (&device->base, DC_EVENT_CLOCK, &clock);
+
+	// Emit a device info event.
+	dc_event_devinfo_t devinfo;
+	devinfo.model = model[0];
+	devinfo.firmware = 0;
+	devinfo.serial = array_uint32_le (serial);
+	device_event_emit (&device->base, DC_EVENT_DEVINFO, &devinfo);
+
+	// Command template.
+	unsigned char command[9] = {0x00,
+			(device->timestamp      ) & 0xFF,
+			(device->timestamp >> 8 ) & 0xFF,
+			(device->timestamp >> 16) & 0xFF,
+			(device->timestamp >> 24) & 0xFF,
+			0x10,
+			0x27,
+			0,
+			0};
+
+	// Data Length.
+	command[0] = 0xC6;
+	unsigned char answer[4] = {0};
+	rc = scubapro_g2_transfer (device, command, sizeof (command), answer, sizeof (answer));
+	if (rc != DC_STATUS_SUCCESS)
+		return rc;
+
+	unsigned int length = array_uint32_le (answer);
+
+	// Update and emit a progress event.
+	progress.maximum = 4 + 9 + (length ? length + 4 : 0);
+	progress.current += 4;
+	device_event_emit (&device->base, DC_EVENT_PROGRESS, &progress);
+
+  	if (length == 0)
+		return DC_STATUS_SUCCESS;
+
+	// Allocate the required amount of memory.
+	if (!dc_buffer_resize (buffer, length)) {
+		ERROR (abstract->context, "Insufficient buffer space available.");
+		return DC_STATUS_NOMEMORY;
+	}
+
+	unsigned char *data = dc_buffer_get_data (buffer);
+
+	// Data.
+	command[0] = 0xC4;
+	rc = scubapro_g2_transfer (device, command, sizeof (command), answer, sizeof (answer));
+	if (rc != DC_STATUS_SUCCESS)
+		return rc;
+
+	unsigned int total = array_uint32_le (answer);
+
+	// Update and emit a progress event.
+	progress.current += 4;
+	device_event_emit (&device->base, DC_EVENT_PROGRESS, &progress);
+
+	if (total != length + 4) {
+		ERROR (abstract->context, "Received an unexpected size.");
+		return DC_STATUS_PROTOCOL;
+	}
+
+	if (receive_data(device, data, length)) {
+		ERROR (abstract->context, "Received an unexpected size.");
+		return DC_STATUS_IO;
+	}
+
+	// Update and emit a progress event.
+	progress.current += length;
+	device_event_emit (&device->base, DC_EVENT_PROGRESS, &progress);
+
+	return DC_STATUS_SUCCESS;
+}
+
+
+static dc_status_t
+scubapro_g2_device_foreach (dc_device_t *abstract, dc_dive_callback_t callback, void *userdata)
+{
+	dc_buffer_t *buffer = dc_buffer_new (0);
+	if (buffer == NULL)
+		return DC_STATUS_NOMEMORY;
+
+	dc_status_t rc = scubapro_g2_device_dump (abstract, buffer);
+	if (rc != DC_STATUS_SUCCESS) {
+		dc_buffer_free (buffer);
+		return rc;
+	}
+
+	rc = scubapro_g2_extract_dives (abstract,
+		dc_buffer_get_data (buffer), dc_buffer_get_size (buffer), callback, userdata);
+
+	dc_buffer_free (buffer);
+
+	return rc;
+}
+
+
+static dc_status_t
+scubapro_g2_extract_dives (dc_device_t *abstract, const unsigned char data[], unsigned int size, dc_dive_callback_t callback, void *userdata)
+{
+	if (abstract && !ISINSTANCE (abstract))
+		return DC_STATUS_INVALIDARGS;
+
+	const unsigned char header[4] = {0xa5, 0xa5, 0x5a, 0x5a};
+
+	// Search the data stream for start markers.
+	unsigned int previous = size;
+	unsigned int current = (size >= 4 ? size - 4 : 0);
+	while (current > 0) {
+		current--;
+		if (memcmp (data + current, header, sizeof (header)) == 0) {
+			// Get the length of the profile data.
+			unsigned int len = array_uint32_le (data + current + 4);
+
+			// Check for a buffer overflow.
+			if (current + len > previous)
+				return DC_STATUS_DATAFORMAT;
+
+			if (callback && !callback (data + current, len, data + current + 8, 4, userdata))
+				return DC_STATUS_SUCCESS;
+
+			// Prepare for the next dive.
+			previous = current;
+			current = (current >= 4 ? current - 4 : 0);
+		}
+	}
+
+	return DC_STATUS_SUCCESS;
+}
diff --git a/src/scubapro_g2.h b/src/scubapro_g2.h
new file mode 100644
index 0000000..edc0f73
--- /dev/null
+++ b/src/scubapro_g2.h
@@ -0,0 +1,39 @@
+/*
+ * libdivecomputer
+ *
+ * Copyright (C) 2008 Jef Driesen
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
+ * MA 02110-1301 USA
+ */
+
+#ifndef SCUBAPRO_G2_H
+#define SCUBAPRO_G2_H
+
+#include <libdivecomputer/context.h>
+#include <libdivecomputer/device.h>
+#include <libdivecomputer/parser.h>
+
+#ifdef __cplusplus
+extern "C" {
+#endif /* __cplusplus */
+
+dc_status_t
+scubapro_g2_device_open (dc_device_t **device, dc_context_t *context);
+
+#ifdef __cplusplus
+}
+#endif /* __cplusplus */
+#endif /* SCUBAPRO_G2_H */
-- 
2.12.2.599.gcf11a6797

