From 3197ca54e44f2c428ac7a326cc87e2b98cbf9829 Mon Sep 17 00:00:00 2001
From: nelisky <nelisky.btc@gmail.com>
Date: Sun, 18 Mar 2012 22:35:10 +0000
Subject: [PATCH] Better cleanup and error handling

---
 libztex.c | 19 +++++++++++++++----
 libztex.h |  1 +
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/libztex.c b/libztex.c
index aca07ac5..acc50259 100644
--- a/libztex.c
+++ b/libztex.c
@@ -53,7 +53,7 @@ static bool libztex_checkDevice (struct libusb_device *dev) {
     applog(LOG_ERR, "Ztex check device: Failed to open read descriptor with error %d", err);
     return false;
   }
-  if (!(desc.idVendor == 0x221A && desc.idProduct == 0x0100)) {
+  if (!(desc.idVendor == LIBZTEX_IDVENDOR && desc.idProduct == LIBZTEX_IDPRODUCT)) {
     return false;
   }
   return true;
@@ -264,14 +264,18 @@ int libztex_prepare_device (struct libusb_device *dev, struct libztex_device** z
     return 1;
   }
 
-  libusb_open(dev, &newdev->hndl);
+  err = libusb_open(dev, &newdev->hndl);
+  if (unlikely(err != 0)) {
+    applog(LOG_ERR, "Ztex check device: Failed to open handle with error %d", err);
+    return err;
+  }
+
   cnt = libusb_get_string_descriptor_ascii (newdev->hndl, newdev->descriptor.iSerialNumber, newdev->snString,
                                             LIBZTEX_SNSTRING_LEN+1);
   if (unlikely(cnt < 0)) {
     applog(LOG_ERR, "Ztex check device: Failed to read device snString with err %d", cnt);
     return cnt;
   }
-  applog(LOG_WARNING, "-- %s", newdev->snString);
 
   cnt = libusb_control_transfer(newdev->hndl, 0xc0, 0x22, 0, 0, buf, 40, 500);
   if (unlikely(cnt < 0)) {
@@ -356,6 +360,7 @@ void libztex_destroy_device (struct libztex_device* ztex) {
   }
   if (ztex->bitFileName != NULL) {
     free(ztex->bitFileName);
+    ztex->bitFileName = NULL;
   }
   free(ztex);
 }
@@ -415,7 +420,9 @@ int libztex_scanDevices (struct libztex_dev_list*** devs_p) {
 
 int libztex_sendHashData (struct libztex_device *ztex, unsigned char *sendbuf) {
   int cnt;
-
+  if (ztex->hndl == NULL) {
+    return 0;
+  }
   cnt = libusb_control_transfer(ztex->hndl, 0x40, 0x80, 0, 0, sendbuf, 44, 1000);
   if (unlikely(cnt < 0)) {
     applog(LOG_ERR, "%s: Failed sendHashData with err %d", ztex->repr, cnt);
@@ -428,6 +435,10 @@ int libztex_readHashData (struct libztex_device *ztex, struct libztex_hash_data
   // length of buf must be 8 * (numNonces + 1)
   unsigned char rbuf[12*8];
   int cnt, i;
+
+  if (ztex->hndl == NULL) {
+    return 0;
+  }
   
   cnt = libusb_control_transfer(ztex->hndl, 0xc0, 0x81, 0, 0, rbuf, 12*ztex->numNonces, 1000);
   if (unlikely(cnt < 0)) {
diff --git a/libztex.h b/libztex.h
index dbcbe882..ec6d56fa 100644
--- a/libztex.h
+++ b/libztex.h
@@ -84,6 +84,7 @@ struct libztex_hash_data {
 
 extern int libztex_scanDevices (struct libztex_dev_list ***devs);
 extern void libztex_freeDevList (struct libztex_dev_list **devs);
+extern int libztex_prepare_device (struct libusb_device *dev, struct libztex_device** ztex);
 extern void libztex_destroy_device (struct libztex_device* ztex);
 extern int libztex_configureFpga (struct libztex_device *dev);
 extern int libztex_setFreq (struct libztex_device *ztex, uint16_t freq);