libztex.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688
  1. /**
  2. * libztex.c - Ztex 1.15x fpga board support library
  3. *
  4. * Copyright (c) 2012 nelisky.btc@gmail.com
  5. *
  6. * This work is based upon the Java SDK provided by ztex which is
  7. * Copyright (C) 2009-2011 ZTEX GmbH.
  8. * http://www.ztex.de
  9. *
  10. * This program is free software; you can redistribute it and/or modify
  11. * it under the terms of the GNU General Public License version 2 as
  12. * published by the Free Software Foundation.
  13. *
  14. * This program is distributed in the hope that it will be useful, but
  15. * WITHOUT ANY WARRANTY; without even the implied warranty of
  16. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  17. * General Public License for more details.
  18. *
  19. * You should have received a copy of the GNU General Public License
  20. * along with this program; if not, see http://www.gnu.org/licenses/.
  21. **/
  22. #include <stdio.h>
  23. #include <unistd.h>
  24. #include "dynclock.h"
  25. #include "fpgautils.h"
  26. #include "miner.h"
  27. #include "libztex.h"
  28. #define BUFSIZE 256
  29. //* Capability index for EEPROM support.
  30. #define CAPABILITY_EEPROM 0,0
  31. //* Capability index for FPGA configuration support.
  32. #define CAPABILITY_FPGA 0,1
  33. //* Capability index for FLASH memory support.
  34. #define CAPABILITY_FLASH 0,2
  35. //* Capability index for DEBUG helper support.
  36. #define CAPABILITY_DEBUG 0,3
  37. //* Capability index for AVR XMEGA support.
  38. #define CAPABILITY_XMEGA 0,4
  39. //* Capability index for AVR XMEGA support.
  40. #define CAPABILITY_HS_FPGA 0,5
  41. //* Capability index for AVR XMEGA support.
  42. #define CAPABILITY_MAC_EEPROM 0,6
  43. //* Capability index for multi FPGA support.
  44. #define CAPABILITY_MULTI_FPGA 0,7
  45. static bool libztex_checkDevice(struct libusb_device *dev)
  46. {
  47. struct libusb_device_descriptor desc;
  48. int err;
  49. err = libusb_get_device_descriptor(dev, &desc);
  50. if (unlikely(err != 0)) {
  51. applog(LOG_ERR, "Ztex check device: Failed to open read descriptor with error %d", err);
  52. return false;
  53. }
  54. if (!(desc.idVendor == LIBZTEX_IDVENDOR && desc.idProduct == LIBZTEX_IDPRODUCT)) {
  55. applog(LOG_DEBUG, "Not a ZTEX device %0.4x:%0.4x", desc.idVendor, desc.idProduct);
  56. return false;
  57. }
  58. return true;
  59. }
  60. static bool libztex_checkCapability(struct libztex_device *ztex, int i, int j)
  61. {
  62. if (!((i >= 0) && (i <= 5) && (j >= 0) && (j < 8) &&
  63. (((ztex->interfaceCapabilities[i] & 255) & (1 << j)) != 0))) {
  64. applog(LOG_ERR, "%s: capability missing: %d %d", ztex->repr, i, j);
  65. return false;
  66. }
  67. return true;
  68. }
  69. static int libztex_detectBitstreamBitOrder(const unsigned char *buf, int size)
  70. {
  71. int i;
  72. for (i = 0; i < size - 4; i++) {
  73. if (((buf[i] & 255) == 0xaa) && ((buf[i + 1] & 255) == 0x99) && ((buf[i + 2] & 255) == 0x55) && ((buf[i + 3] & 255) == 0x66))
  74. return 1;
  75. if (((buf[i] & 255) == 0x55) && ((buf[i + 1] & 255) == 0x99) && ((buf[i + 2] & 255) == 0xaa) && ((buf[i + 3] & 255) == 0x66))
  76. return 0;
  77. }
  78. applog(LOG_WARNING, "Unable to determine bitstream bit order: no signature found");
  79. return 0;
  80. }
  81. static void libztex_swapBits(unsigned char *buf, int size)
  82. {
  83. unsigned char c;
  84. int i;
  85. for (i = 0; i < size; i++) {
  86. c = buf[i];
  87. buf[i] = ((c & 128) >> 7) |
  88. ((c & 64) >> 5) |
  89. ((c & 32) >> 3) |
  90. ((c & 16) >> 1) |
  91. ((c & 8) << 1) |
  92. ((c & 4) << 3) |
  93. ((c & 2) << 5) |
  94. ((c & 1) << 7);
  95. }
  96. }
  97. static int libztex_getFpgaState(struct libztex_device *ztex, struct libztex_fpgastate *state)
  98. {
  99. unsigned char buf[9];
  100. int cnt;
  101. if (!libztex_checkCapability(ztex, CAPABILITY_FPGA))
  102. return -1;
  103. cnt = libusb_control_transfer(ztex->hndl, 0xc0, 0x30, 0, 0, buf, 9, 1000);
  104. if (unlikely(cnt < 0)) {
  105. applog(LOG_ERR, "%s: Failed getFpgaState with err %d", ztex->repr, cnt);
  106. return cnt;
  107. }
  108. state->fpgaConfigured = (buf[0] == 0);
  109. state->fpgaChecksum = buf[1] & 0xff;
  110. state->fpgaBytes = ((buf[5] & 0xff) << 24) | ((buf[4] & 0xff) << 16) | ((buf[3] & 0xff) << 8) | (buf[2] & 0xff);
  111. state->fpgaInitB = buf[6] & 0xff;
  112. state->fpgaFlashResult = buf[7];
  113. state->fpgaFlashBitSwap = (buf[8] != 0);
  114. return 0;
  115. }
  116. static int libztex_configureFpgaHS(struct libztex_device *ztex, const char* firmware, bool force, char bs)
  117. {
  118. struct libztex_fpgastate state;
  119. const int transactionBytes = 65536;
  120. unsigned char buf[transactionBytes], settings[2];
  121. int tries, cnt, buf_p, i;
  122. ssize_t pos = 0;
  123. FILE *fp;
  124. if (!libztex_checkCapability(ztex, CAPABILITY_HS_FPGA))
  125. return -1;
  126. libztex_getFpgaState(ztex, &state);
  127. if (!force && state.fpgaConfigured) {
  128. applog(LOG_INFO, "Bitstream already configured");
  129. return 1;
  130. }
  131. cnt = libusb_control_transfer(ztex->hndl, 0xc0, 0x33, 0, 0, settings, 2, 1000);
  132. if (unlikely(cnt < 0)) {
  133. applog(LOG_ERR, "%s: Failed getHSFpgaSettings with err %d", ztex->repr, cnt);
  134. return cnt;
  135. }
  136. libusb_claim_interface(ztex->hndl, settings[1]);
  137. for (tries = 3; tries > 0; tries--) {
  138. fp = open_bitstream("ztex", firmware);
  139. if (!fp) {
  140. applog(LOG_ERR, "%s: failed to read firmware '%s'", ztex->repr, firmware);
  141. return -2;
  142. }
  143. while (pos < transactionBytes && !feof(fp)) {
  144. buf[pos++] = getc(fp);
  145. }
  146. if (feof(fp))
  147. pos--;
  148. if (bs != 0 && bs != 1)
  149. bs = libztex_detectBitstreamBitOrder(buf, transactionBytes < pos? transactionBytes: pos);
  150. if (bs == 1)
  151. libztex_swapBits(buf, pos);
  152. libusb_control_transfer(ztex->hndl, 0x40, 0x34, 0, 0, NULL, 0, 1000);
  153. // 0x34 - initHSFPGAConfiguration
  154. buf_p = pos;
  155. while (1) {
  156. i = 0;
  157. while (i < buf_p) {
  158. if (libusb_bulk_transfer(ztex->hndl,
  159. settings[0],
  160. &buf[i],
  161. buf_p - i,
  162. &cnt, 1000) != 0) {
  163. applog(LOG_ERR, "%s: Failed send hs fpga data", ztex->repr);
  164. break;
  165. }
  166. usleep(500);
  167. i += cnt;
  168. }
  169. if (i < buf_p || buf_p < transactionBytes)
  170. break;
  171. buf_p = 0;
  172. while (buf_p < transactionBytes && !feof(fp)) {
  173. buf[buf_p++] = getc(fp);
  174. }
  175. if (feof(fp))
  176. buf_p--;
  177. pos += buf_p;
  178. if (buf_p == 0)
  179. break;
  180. if (bs == 1)
  181. libztex_swapBits(buf, buf_p);
  182. }
  183. libusb_control_transfer(ztex->hndl, 0x40, 0x35, 0, 0, NULL, 0, 1000);
  184. // 0x35 - finishHSFPGAConfiguration
  185. if (cnt >= 0)
  186. tries = 0;
  187. fclose(fp);
  188. libztex_getFpgaState(ztex, &state);
  189. if (!state.fpgaConfigured) {
  190. applog(LOG_ERR, "%s: HS FPGA configuration failed: DONE pin does not go high", ztex->repr);
  191. return -3;
  192. }
  193. }
  194. libusb_release_interface(ztex->hndl, settings[1]);
  195. usleep(200000);
  196. applog(LOG_INFO, "%s: HS FPGA configuration done", ztex->repr);
  197. return 0;
  198. }
  199. static int libztex_configureFpgaLS(struct libztex_device *ztex, const char* firmware, bool force, char bs)
  200. {
  201. struct libztex_fpgastate state;
  202. const int transactionBytes = 2048;
  203. unsigned char buf[transactionBytes], cs;
  204. int tries, cnt, buf_p, i;
  205. ssize_t pos = 0;
  206. FILE *fp;
  207. if (!libztex_checkCapability(ztex, CAPABILITY_FPGA))
  208. return -1;
  209. libztex_getFpgaState(ztex, &state);
  210. if (!force && state.fpgaConfigured) {
  211. applog(LOG_DEBUG, "Bitstream already configured");
  212. return 1;
  213. }
  214. for (tries = 10; tries > 0; tries--) {
  215. fp = open_bitstream("ztex", firmware);
  216. if (!fp) {
  217. applog(LOG_ERR, "%s: failed to read firmware '%s'", ztex->repr, firmware);
  218. return -2;
  219. }
  220. cs = 0;
  221. while (pos < transactionBytes && !feof(fp)) {
  222. buf[pos] = getc(fp);
  223. cs += buf[pos++];
  224. }
  225. if (feof(fp))
  226. pos--;
  227. if (bs != 0 && bs != 1)
  228. bs = libztex_detectBitstreamBitOrder(buf, transactionBytes < pos? transactionBytes: pos);
  229. //* Reset fpga
  230. cnt = libztex_resetFpga(ztex);
  231. if (unlikely(cnt < 0)) {
  232. applog(LOG_ERR, "%s: Failed reset fpga with err %d", ztex->repr, cnt);
  233. continue;
  234. }
  235. if (bs == 1)
  236. libztex_swapBits(buf, pos);
  237. buf_p = pos;
  238. while (1) {
  239. i = 0;
  240. while (i < buf_p) {
  241. cnt = libusb_control_transfer(ztex->hndl, 0x40, 0x32, 0, 0, &buf[i], buf_p - i, 5000);
  242. if (unlikely(cnt < 0)) {
  243. applog(LOG_ERR, "%s: Failed send fpga data with err %d", ztex->repr, cnt);
  244. break;
  245. }
  246. i += cnt;
  247. }
  248. if (i < buf_p || buf_p < transactionBytes)
  249. break;
  250. buf_p = 0;
  251. while (buf_p < transactionBytes && !feof(fp)) {
  252. buf[buf_p] = getc(fp);
  253. cs += buf[buf_p++];
  254. }
  255. if (feof(fp))
  256. buf_p--;
  257. pos += buf_p;
  258. if (buf_p == 0)
  259. break;
  260. if (bs == 1)
  261. libztex_swapBits(buf, buf_p);
  262. }
  263. if (cnt >= 0)
  264. tries = 0;
  265. fclose(fp);
  266. }
  267. libztex_getFpgaState(ztex, &state);
  268. if (!state.fpgaConfigured) {
  269. applog(LOG_ERR, "%s: FPGA configuration failed: DONE pin does not go high", ztex->repr);
  270. return 3;
  271. }
  272. usleep(200000);
  273. applog(LOG_INFO, "%s: FPGA configuration done", ztex->repr);
  274. return 0;
  275. }
  276. int libztex_configureFpga(struct libztex_device *ztex)
  277. {
  278. char buf[256];
  279. int rv;
  280. strcpy(buf, ztex->bitFileName);
  281. strcat(buf, ".bit");
  282. rv = libztex_configureFpgaHS(ztex, buf, true, 2);
  283. if (rv != 0)
  284. rv = libztex_configureFpgaLS(ztex, buf, true, 2);
  285. return rv;
  286. }
  287. int libztex_numberOfFpgas(struct libztex_device *ztex) {
  288. int cnt;
  289. unsigned char buf[3];
  290. if (ztex->numberOfFpgas < 0) {
  291. if (libztex_checkCapability(ztex, CAPABILITY_MULTI_FPGA)) {
  292. cnt = libusb_control_transfer(ztex->hndl, 0xc0, 0x50, 0, 0, buf, 3, 1000);
  293. if (unlikely(cnt < 0)) {
  294. applog(LOG_ERR, "%s: Failed getMultiFpgaInfo with err %d", ztex->repr, cnt);
  295. return cnt;
  296. }
  297. ztex->numberOfFpgas = buf[0] + 1;
  298. ztex->selectedFpga = -1;//buf[1];
  299. ztex->parallelConfigSupport = (buf[2] == 1);
  300. } else {
  301. ztex->numberOfFpgas = 1;
  302. ztex->selectedFpga = -1;//0;
  303. ztex->parallelConfigSupport = false;
  304. }
  305. }
  306. return ztex->numberOfFpgas;
  307. }
  308. int libztex_selectFpga(struct libztex_device *ztex) {
  309. int cnt, fpgacnt = libztex_numberOfFpgas(ztex->root);
  310. int number = ztex->fpgaNum;
  311. if (number < 0 || number >= fpgacnt) {
  312. applog(LOG_WARNING, "%s: Trying to select wrong fpga (%d in %d)", ztex->repr, number, fpgacnt);
  313. return 1;
  314. }
  315. if (ztex->root->selectedFpga != number && libztex_checkCapability(ztex->root, CAPABILITY_MULTI_FPGA)) {
  316. cnt = libusb_control_transfer(ztex->root->hndl, 0x40, 0x51, number, 0, NULL, 0, 500);
  317. if (unlikely(cnt < 0)) {
  318. applog(LOG_ERR, "Ztex check device: Failed to set fpga with err %d", cnt);
  319. return cnt;
  320. }
  321. ztex->root->selectedFpga = number;
  322. }
  323. return 0;
  324. }
  325. int libztex_setFreq(struct libztex_device *ztex, uint16_t freq) {
  326. int cnt;
  327. uint16_t oldfreq = ztex->dclk.freqM;
  328. if (freq > ztex->dclk.freqMaxM)
  329. freq = ztex->dclk.freqMaxM;
  330. cnt = libusb_control_transfer(ztex->hndl, 0x40, 0x83, freq, 0, NULL, 0, 500);
  331. if (unlikely(cnt < 0)) {
  332. applog(LOG_ERR, "Ztex check device: Failed to set frequency with err %d", cnt);
  333. return cnt;
  334. }
  335. ztex->dclk.freqM = freq;
  336. if (oldfreq > ztex->dclk.freqMaxM)
  337. applog(LOG_WARNING, "%s: Frequency set to %0.2f Mhz",
  338. ztex->repr,
  339. ztex->freqM1 * (ztex->dclk.freqM + 1)
  340. );
  341. else
  342. applog(LOG_WARNING, "%s: Frequency change from %0.2f to %0.2f Mhz",
  343. ztex->repr,
  344. ztex->freqM1 * (oldfreq + 1), ztex->freqM1 * (ztex->dclk.freqM + 1)
  345. );
  346. return 0;
  347. }
  348. int libztex_resetFpga(struct libztex_device *ztex)
  349. {
  350. return libusb_control_transfer(ztex->hndl, 0x40, 0x31, 0, 0, NULL, 0, 1000);
  351. }
  352. int libztex_suspend(struct libztex_device *ztex) {
  353. if (ztex->suspendSupported) {
  354. return libusb_control_transfer(ztex->hndl, 0x40, 0x84, 0, 0, NULL, 0, 1000);
  355. } else {
  356. return 0;
  357. }
  358. }
  359. int libztex_prepare_device(struct libusb_device *dev, struct libztex_device** ztex) {
  360. struct libztex_device *newdev;
  361. int i, cnt, err;
  362. unsigned char buf[64];
  363. newdev = malloc(sizeof(struct libztex_device));
  364. dclk_prepare(&newdev->dclk);
  365. newdev->bitFileName = NULL;
  366. newdev->numberOfFpgas = -1;
  367. newdev->valid = false;
  368. newdev->hndl = NULL;
  369. *ztex = newdev;
  370. err = libusb_get_device_descriptor(dev, &newdev->descriptor);
  371. if (unlikely(err != 0)) {
  372. applog(LOG_ERR, "Ztex check device: Failed to open read descriptor with error %d", err);
  373. return err;
  374. }
  375. // Check vendorId and productId
  376. if (!(newdev->descriptor.idVendor == LIBZTEX_IDVENDOR &&
  377. newdev->descriptor.idProduct == LIBZTEX_IDPRODUCT)) {
  378. applog(LOG_ERR, "Not a ztex device? %0.4X, %0.4X", newdev->descriptor.idVendor, newdev->descriptor.idProduct);
  379. return 1;
  380. }
  381. err = libusb_open(dev, &newdev->hndl);
  382. if (unlikely(err != 0)) {
  383. applog(LOG_ERR, "Ztex check device: Failed to open handle with error %d", err);
  384. return err;
  385. }
  386. cnt = libusb_get_string_descriptor_ascii (newdev->hndl, newdev->descriptor.iSerialNumber, newdev->snString,
  387. LIBZTEX_SNSTRING_LEN + 1);
  388. if (unlikely(cnt < 0)) {
  389. applog(LOG_ERR, "Ztex check device: Failed to read device snString with err %d", cnt);
  390. return cnt;
  391. }
  392. cnt = libusb_control_transfer(newdev->hndl, 0xc0, 0x22, 0, 0, buf, 40, 500);
  393. if (unlikely(cnt < 0)) {
  394. applog(LOG_ERR, "Ztex check device: Failed to read ztex descriptor with err %d", cnt);
  395. return cnt;
  396. }
  397. if (buf[0] != 40 || buf[1] != 1 || buf[2] != 'Z' || buf[3] != 'T' || buf[4] != 'E' || buf[5] != 'X') {
  398. applog(LOG_ERR, "Ztex check device: Error reading ztex descriptor");
  399. return 2;
  400. }
  401. newdev->productId[0] = buf[6];
  402. newdev->productId[1] = buf[7];
  403. newdev->productId[2] = buf[8];
  404. newdev->productId[3] = buf[9];
  405. newdev->fwVersion = buf[10];
  406. newdev->interfaceVersion = buf[11];
  407. newdev->interfaceCapabilities[0] = buf[12];
  408. newdev->interfaceCapabilities[1] = buf[13];
  409. newdev->interfaceCapabilities[2] = buf[14];
  410. newdev->interfaceCapabilities[3] = buf[15];
  411. newdev->interfaceCapabilities[4] = buf[16];
  412. newdev->interfaceCapabilities[5] = buf[17];
  413. newdev->moduleReserved[0] = buf[18];
  414. newdev->moduleReserved[1] = buf[19];
  415. newdev->moduleReserved[2] = buf[20];
  416. newdev->moduleReserved[3] = buf[21];
  417. newdev->moduleReserved[4] = buf[22];
  418. newdev->moduleReserved[5] = buf[23];
  419. newdev->moduleReserved[6] = buf[24];
  420. newdev->moduleReserved[7] = buf[25];
  421. newdev->moduleReserved[8] = buf[26];
  422. newdev->moduleReserved[9] = buf[27];
  423. newdev->moduleReserved[10] = buf[28];
  424. newdev->moduleReserved[11] = buf[29];
  425. cnt = libusb_control_transfer(newdev->hndl, 0xc0, 0x82, 0, 0, buf, 64, 500);
  426. if (unlikely(cnt < 0)) {
  427. applog(LOG_ERR, "Ztex check device: Failed to read ztex descriptor with err %d", cnt);
  428. return cnt;
  429. }
  430. if (unlikely(buf[0] != 5)) {
  431. if (unlikely(buf[0] != 2 && buf[0] != 4)) {
  432. applog(LOG_ERR, "Invalid BTCMiner descriptor version. Firmware must be updated (%d).", buf[0]);
  433. return 3;
  434. }
  435. applog(LOG_WARNING, "Firmware out of date (%d).", buf[0]);
  436. }
  437. i = buf[0] > 4? 11: (buf[0] > 2? 10: 8);
  438. while (cnt < 64 && buf[cnt] != 0)
  439. cnt++;
  440. if (cnt < i + 1) {
  441. applog(LOG_ERR, "Invalid bitstream file name .");
  442. return 4;
  443. }
  444. newdev->bitFileName = malloc(sizeof(char) * (cnt + 1));
  445. memcpy(newdev->bitFileName, &buf[i], cnt);
  446. newdev->bitFileName[cnt] = 0;
  447. newdev->numNonces = buf[1] + 1;
  448. newdev->offsNonces = ((buf[2] & 255) | ((buf[3] & 255) << 8)) - 10000;
  449. newdev->freqM1 = ((buf[4] & 255) | ((buf[5] & 255) << 8) ) * 0.01;
  450. newdev->dclk.freqMaxM = (buf[7] & 255);
  451. newdev->dclk.freqM = (buf[6] & 255);
  452. newdev->dclk.freqMDefault = newdev->dclk.freqM;
  453. newdev->suspendSupported = (buf[0] == 5);
  454. newdev->hashesPerClock = buf[0] > 2? (((buf[8] & 255) | ((buf[9] & 255) << 8)) + 1) / 128.0: 1.0;
  455. newdev->extraSolutions = buf[0] > 4? buf[10]: 0;
  456. applog(LOG_DEBUG, "PID: %d numNonces: %d offsNonces: %d freqM1: %f freqMaxM: %d freqM: %d suspendSupported: %s hashesPerClock: %f extraSolutions: %d",
  457. buf[0], newdev->numNonces, newdev->offsNonces, newdev->freqM1, newdev->dclk.freqMaxM, newdev->dclk.freqM, newdev->suspendSupported ? "T": "F",
  458. newdev->hashesPerClock, newdev->extraSolutions);
  459. if (buf[0] < 4) {
  460. if (strncmp(newdev->bitFileName, "ztex_ufm1_15b", 13) != 0)
  461. newdev->hashesPerClock = 0.5;
  462. applog(LOG_WARNING, "HASHES_PER_CLOCK not defined, assuming %0.2f", newdev->hashesPerClock);
  463. }
  464. newdev->usbbus = libusb_get_bus_number(dev);
  465. newdev->usbaddress = libusb_get_device_address(dev);
  466. sprintf(newdev->repr, "ZTEX %s-1", newdev->snString);
  467. newdev->valid = true;
  468. return 0;
  469. }
  470. void libztex_destroy_device(struct libztex_device* ztex)
  471. {
  472. if (ztex->hndl != NULL) {
  473. libusb_close(ztex->hndl);
  474. ztex->hndl = NULL;
  475. }
  476. if (ztex->bitFileName != NULL) {
  477. free(ztex->bitFileName);
  478. ztex->bitFileName = NULL;
  479. }
  480. free(ztex);
  481. }
  482. int libztex_scanDevices(struct libztex_dev_list*** devs_p)
  483. {
  484. int usbdevices[LIBZTEX_MAX_DESCRIPTORS];
  485. struct libztex_dev_list **devs;
  486. struct libztex_device *ztex;
  487. int found = 0, pos = 0, err;
  488. libusb_device **list;
  489. ssize_t cnt, i = 0;
  490. cnt = libusb_get_device_list(NULL, &list);
  491. if (unlikely(cnt < 0)) {
  492. applog(LOG_ERR, "Ztex scan devices: Failed to list usb devices with err %d", cnt);
  493. return 0;
  494. }
  495. for (i = 0; i < cnt; i++) {
  496. if (libztex_checkDevice(list[i])) {
  497. // Got one!
  498. usbdevices[found] = i;
  499. found++;
  500. }
  501. }
  502. devs = malloc(sizeof(struct libztex_dev_list *) * found);
  503. if (devs == NULL) {
  504. applog(LOG_ERR, "Ztex scan devices: Failed to allocate memory");
  505. return 0;
  506. }
  507. for (i = 0; i < found; i++) {
  508. err = libztex_prepare_device(list[usbdevices[i]], &ztex);
  509. if (unlikely(err != 0))
  510. applog(LOG_ERR, "prepare device: %d", err);
  511. // check if valid
  512. if (!ztex->valid) {
  513. libztex_destroy_device(ztex);
  514. continue;
  515. }
  516. devs[pos] = malloc(sizeof(struct libztex_dev_list));
  517. devs[pos]->dev = ztex;
  518. devs[pos]->next = NULL;
  519. if (pos > 0)
  520. devs[pos - 1]->next = devs[pos];
  521. pos++;
  522. }
  523. libusb_free_device_list(list, 1);
  524. *devs_p = devs;
  525. return pos;
  526. }
  527. int libztex_sendHashData(struct libztex_device *ztex, unsigned char *sendbuf)
  528. {
  529. int cnt, ret, len;
  530. if (ztex == NULL || ztex->hndl == NULL)
  531. return 0;
  532. ret = 44; len = 0;
  533. while (ret > 0) {
  534. cnt = libusb_control_transfer(ztex->hndl, 0x40, 0x80, 0, 0, sendbuf + len, ret, 1000);
  535. if (cnt >= 0) {
  536. ret -= cnt;
  537. len += cnt;
  538. } else
  539. break;
  540. }
  541. if (unlikely(cnt < 0))
  542. applog(LOG_ERR, "%s: Failed sendHashData with err %d", ztex->repr, cnt);
  543. return cnt;
  544. }
  545. int libztex_readHashData(struct libztex_device *ztex, struct libztex_hash_data nonces[]) {
  546. int bufsize = 12 + ztex->extraSolutions * 4;
  547. int cnt = 0, i, j, ret, len;
  548. unsigned char *rbuf;
  549. if (ztex->hndl == NULL)
  550. return 0;
  551. rbuf = malloc(sizeof(unsigned char) * (ztex->numNonces * bufsize));
  552. if (rbuf == NULL) {
  553. applog(LOG_ERR, "%s: Failed to allocate memory for reading nonces", ztex->repr);
  554. return 0;
  555. }
  556. ret = bufsize * ztex->numNonces; len = 0;
  557. while (ret > 0) {
  558. cnt = libusb_control_transfer(ztex->hndl, 0xc0, 0x81, 0, 0, rbuf + len, ret, 1000);
  559. if (cnt >= 0) {
  560. ret -= cnt;
  561. len += cnt;
  562. } else
  563. break;
  564. }
  565. if (unlikely(cnt < 0)) {
  566. applog(LOG_ERR, "%s: Failed readHashData with err %d", ztex->repr, cnt);
  567. free(rbuf);
  568. return cnt;
  569. }
  570. for (i=0; i<ztex->numNonces; i++) {
  571. memcpy((char*)&nonces[i].goldenNonce[0], &rbuf[i*bufsize], 4);
  572. nonces[i].goldenNonce[0] -= ztex->offsNonces;
  573. //applog(LOG_DEBUG, "W %d:0 %0.8x", i, nonces[i].goldenNonce[0]);
  574. memcpy((char*)&nonces[i].nonce, &rbuf[(i*bufsize)+4], 4);
  575. nonces[i].nonce -= ztex->offsNonces;
  576. memcpy((char*)&nonces[i].hash7, &rbuf[(i*bufsize)+8], 4);
  577. for (j=0; j<ztex->extraSolutions; j++) {
  578. memcpy((char*)&nonces[i].goldenNonce[j+1], &rbuf[(i*bufsize)+12+(j*4)], 4);
  579. nonces[i].goldenNonce[j+1] -= ztex->offsNonces;
  580. //applog(LOG_DEBUG, "W %d:%d %0.8x", i, j+1, nonces[i].goldenNonce[j+1]);
  581. }
  582. }
  583. free(rbuf);
  584. return cnt;
  585. }
  586. void libztex_freeDevList(struct libztex_dev_list **devs)
  587. {
  588. bool done = false;
  589. ssize_t cnt = 0;
  590. while (!done) {
  591. if (devs[cnt]->next == NULL)
  592. done = true;
  593. free(devs[cnt++]);
  594. }
  595. free(devs);
  596. }