X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=hardware%2Fcommon%2Fezusb.c;h=884ceafdccf9fa693a401aecd0ffa1a170d963e6;hb=9ddb2a125d0cc01f76acc43670c4609cc4a7b49b;hp=e765b83452ec2df5c311ebf904f0d344f36e01b3;hpb=a1bb33afbde769156ad4bef7a60579da64aebbb7;p=libsigrok.git

diff --git a/hardware/common/ezusb.c b/hardware/common/ezusb.c
index e765b834..884ceafd 100644
--- a/hardware/common/ezusb.c
+++ b/hardware/common/ezusb.c
@@ -36,14 +36,13 @@ int ezusb_reset(struct libusb_device_handle *hdl, int set_clear)
 	g_message("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)
+				      0xe600, 0x0000, buf, 1, 100);
+	if (err < 0)
 		g_warning("Unable to send control request: %d", err);
 
 	return err;
 }
 
-
 int ezusb_install_firmware(libusb_device_handle *hdl, const char *filename)
 {
 	FILE *fw;
@@ -51,23 +50,21 @@ int ezusb_install_firmware(libusb_device_handle *hdl, const char *filename)
 	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));
+	if ((fw = fopen(filename, "r")) == NULL) {
+		g_warning("Unable to open firmware file %s for reading: %s",
+			  filename, strerror(errno));
 		return 1;
 	}
 
-	result = 0;
-	offset = 0;
-	while(1)
-	{
+	result = offset = 0;
+	while (1) {
 		chunksize = fread(buf, 1, 4096, fw);
-		if(chunksize == 0)
+		if (chunksize == 0)
 			break;
-		err = libusb_control_transfer(hdl, LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_OUT, 0xa0,
-									  offset, 0x0000, buf, chunksize, 100);
-		if(err < 0)
-		{
+		err = libusb_control_transfer(hdl, LIBUSB_REQUEST_TYPE_VENDOR |
+					      LIBUSB_ENDPOINT_OUT, 0xa0, offset,
+					      0x0000, buf, chunksize, 100);
+		if (err < 0) {
 			g_warning("Unable to send firmware to device: %d", err);
 			result = 1;
 			break;
@@ -81,4 +78,37 @@ int ezusb_install_firmware(libusb_device_handle *hdl, const char *filename)
 	return result;
 }
 
+int ezusb_upload_firmware(libusb_device *dev, int configuration,
+			  const char *filename)
+{
+	struct libusb_device_handle *hdl;
+	int err;
+
+	g_message("uploading firmware to device on %d.%d",
+		  libusb_get_bus_number(dev), libusb_get_device_address(dev));
+
+	err = libusb_open(dev, &hdl);
+	if (err != 0) {
+		g_warning("failed to open device: %d", err);
+		return 1;
+	}
+
+	err = libusb_set_configuration(hdl, configuration);
+	if (err != 0) {
+		g_warning("Unable to set configuration: %d", err);
+		return 1;
+	}
+
+	if ((ezusb_reset(hdl, 1)) < 0)
+		return 1;
+
+	if (ezusb_install_firmware(hdl, filename) != 0)
+		return 1;
 
+	if ((ezusb_reset(hdl, 0)) < 0)
+		return 1;
+
+	libusb_close(hdl);
+
+	return 0;
+}