X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=hardware%2Fcommon%2Fezusb.c;h=30b43beb6240300e1410a3efa3afa37fc30d1d22;hb=6d754b6d3a8f843cd5254617674f484cac9b12db;hp=fe204a00208ab738fd7a41e3767ea3f319e482eb;hpb=9d2933fbe9719df679cde8fbffde8c191d6a56d3;p=libsigrok.git

diff --git a/hardware/common/ezusb.c b/hardware/common/ezusb.c
index fe204a00..30b43beb 100644
--- a/hardware/common/ezusb.c
+++ b/hardware/common/ezusb.c
@@ -21,24 +21,28 @@
  * Helper functions for the Cypress EZ-USB / FX2 series chips.
  */
 
+#include <sigrok.h>
+#include <sigrok-internal.h>
 #include <libusb.h>
 #include <glib.h>
+#include <glib/gstdio.h>
 #include <stdio.h>
 #include <errno.h>
 #include <string.h>
 #include "config.h"
 
+
 int ezusb_reset(struct libusb_device_handle *hdl, int set_clear)
 {
 	int err;
 	unsigned char buf[1];
 
-	g_message("setting CPU reset mode %s...", set_clear ? "on" : "off");
+	sr_info("setting CPU reset mode %s...", set_clear ? "on" : "off");
 	buf[0] = set_clear ? 1 : 0;
 	err = libusb_control_transfer(hdl, LIBUSB_REQUEST_TYPE_VENDOR, 0xa0,
 				      0xe600, 0x0000, buf, 1, 100);
 	if (err < 0)
-		g_warning("Unable to send control request: %d", err);
+		sr_warn("Unable to send control request: %d", err);
 
 	return err;
 }
@@ -49,32 +53,72 @@ int ezusb_install_firmware(libusb_device_handle *hdl, const char *filename)
 	int offset, chunksize, err, result;
 	unsigned char buf[4096];
 
-	g_message("Uploading firmware at %s", filename);
-	if ((fw = fopen(filename, "r")) == NULL) {
-		g_warning("Unable to open firmware file %s for reading: %s",
-			  filename, strerror(errno));
-		return 1;
+	sr_info("Uploading firmware at %s", filename);
+	if ((fw = g_fopen(filename, "rb")) == NULL) {
+		sr_warn("Unable to open firmware file %s for reading: %s",
+			filename, strerror(errno));
+		return SR_ERR;
 	}
 
-	result = 0;
+	result = SR_OK;
 	offset = 0;
 	while (1) {
 		chunksize = fread(buf, 1, 4096, fw);
 		if (chunksize == 0)
 			break;
 		err = libusb_control_transfer(hdl, LIBUSB_REQUEST_TYPE_VENDOR |
-				LIBUSB_ENDPOINT_OUT, 0xa0, offset, 0x0000,
-				buf, chunksize, 100);
+					      LIBUSB_ENDPOINT_OUT, 0xa0, offset,
+					      0x0000, buf, chunksize, 100);
 		if (err < 0) {
-			g_warning("Unable to send firmware to device: %d", err);
-			result = 1;
+			sr_warn("Unable to send firmware to device: %d", err);
+			result = SR_ERR;
 			break;
 		}
-		g_message("Uploaded %d bytes", chunksize);
+		sr_info("Uploaded %d bytes", chunksize);
 		offset += chunksize;
 	}
 	fclose(fw);
-	g_message("Firmware upload done");
+	sr_info("Firmware upload done");
 
 	return result;
 }
+
+int ezusb_upload_firmware(libusb_device *dev, int configuration,
+			  const char *filename)
+{
+	struct libusb_device_handle *hdl;
+	int err;
+
+	sr_info("uploading firmware to device on %d.%d",
+		  libusb_get_bus_number(dev), libusb_get_device_address(dev));
+
+	if ((err = libusb_open(dev, &hdl)) < 0) {
+		sr_warn("failed to open device: %d", err);
+		return SR_ERR;
+	}
+
+	if (libusb_kernel_driver_active(hdl, 0)) {
+		if ((err = libusb_detach_kernel_driver(hdl, 0)) < 0) {
+			sr_warn("failed to detach kernel driver: %d", err);
+			return SR_ERR;
+		}
+	}
+
+	if ((err = libusb_set_configuration(hdl, configuration)) < 0) {
+		sr_warn("Unable to set configuration: %d", err);
+		return SR_ERR;
+	}
+
+	if ((ezusb_reset(hdl, 1)) < 0)
+		return SR_ERR;
+
+	if (ezusb_install_firmware(hdl, filename) < 0)
+		return SR_ERR;
+
+	if ((ezusb_reset(hdl, 0)) < 0)
+		return SR_ERR;
+
+	libusb_close(hdl);
+
+	return SR_OK;
+}