ZuluSCSI_initiator.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772
  1. /**
  2. * ZuluSCSI™ - Copyright (c) 2022 Rabbit Hole Computing™
  3. *
  4. * ZuluSCSI™ firmware is licensed under the GPL version 3 or any later version. 
  5. *
  6. * https://www.gnu.org/licenses/gpl-3.0.html
  7. * ----
  8. * This program is free software: you can redistribute it and/or modify
  9. * it under the terms of the GNU General Public License as published by
  10. * the Free Software Foundation, either version 3 of the License, or
  11. * (at your option) any later version. 
  12. *
  13. * This program is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16. * GNU General Public License for more details. 
  17. *
  18. * You should have received a copy of the GNU General Public License
  19. * along with this program.  If not, see <https://www.gnu.org/licenses/>.
  20. **/
  21. /*
  22. * Main program for initiator mode.
  23. */
  24. #include "ZuluSCSI_config.h"
  25. #include "ZuluSCSI_log.h"
  26. #include "ZuluSCSI_log_trace.h"
  27. #include "ZuluSCSI_initiator.h"
  28. #include <ZuluSCSI_platform.h>
  29. #include "SdFat.h"
  30. #include <scsi2sd.h>
  31. extern "C" {
  32. #include <scsi.h>
  33. }
  34. #ifndef PLATFORM_HAS_INITIATOR_MODE
  35. void scsiInitiatorInit()
  36. {
  37. }
  38. void scsiInitiatorMainLoop()
  39. {
  40. }
  41. int scsiInitiatorRunCommand(const uint8_t *command, size_t cmdlen,
  42. uint8_t *bufIn, size_t bufInLen,
  43. const uint8_t *bufOut, size_t bufOutLen)
  44. {
  45. return -1;
  46. }
  47. bool scsiInitiatorReadCapacity(int target_id, uint32_t *sectorcount, uint32_t *sectorsize)
  48. {
  49. return false;
  50. }
  51. #else
  52. /*************************************
  53. * High level initiator mode logic *
  54. *************************************/
  55. static struct {
  56. // Bitmap of all drives that have been imaged
  57. uint32_t drives_imaged;
  58. // Is imaging a drive in progress, or are we scanning?
  59. bool imaging;
  60. // Information about currently selected drive
  61. int target_id;
  62. uint32_t sectorsize;
  63. uint32_t sectorcount;
  64. uint32_t sectorcount_all;
  65. uint32_t sectors_done;
  66. uint32_t max_sector_per_transfer;
  67. // Retry information for sector reads.
  68. // If a large read fails, retry is done sector-by-sector.
  69. int retrycount;
  70. uint32_t failposition;
  71. FsFile target_file;
  72. } g_initiator_state;
  73. extern SdFs SD;
  74. // Initialization of initiator mode
  75. void scsiInitiatorInit()
  76. {
  77. scsiHostPhyReset();
  78. g_initiator_state.drives_imaged = 0;
  79. g_initiator_state.imaging = false;
  80. g_initiator_state.target_id = -1;
  81. g_initiator_state.sectorsize = 0;
  82. g_initiator_state.sectorcount = 0;
  83. g_initiator_state.sectors_done = 0;
  84. g_initiator_state.retrycount = 0;
  85. g_initiator_state.failposition = 0;
  86. g_initiator_state.max_sector_per_transfer = 512;
  87. }
  88. // Update progress bar LED during transfers
  89. static void scsiInitiatorUpdateLed()
  90. {
  91. // Update status indicator, the led blinks every 5 seconds and is on the longer the more data has been transferred
  92. const int period = 256;
  93. int phase = (millis() % period);
  94. int duty = g_initiator_state.sectors_done * period / g_initiator_state.sectorcount;
  95. // Minimum and maximum time to verify that the blink is visible
  96. if (duty < 50) duty = 50;
  97. if (duty > period - 50) duty = period - 50;
  98. if (phase <= duty)
  99. {
  100. LED_ON();
  101. }
  102. else
  103. {
  104. LED_OFF();
  105. }
  106. }
  107. void delay_with_poll(uint32_t ms)
  108. {
  109. uint32_t start = millis();
  110. while ((uint32_t)(millis() - start) < ms)
  111. {
  112. platform_poll();
  113. delay(1);
  114. }
  115. }
  116. // High level logic of the initiator mode
  117. void scsiInitiatorMainLoop()
  118. {
  119. if (g_scsiHostPhyReset)
  120. {
  121. logmsg("Executing BUS RESET after aborted command");
  122. scsiHostPhyReset();
  123. }
  124. if (!g_initiator_state.imaging)
  125. {
  126. // Scan for SCSI drives one at a time
  127. g_initiator_state.target_id = (g_initiator_state.target_id + 1) % 8;
  128. g_initiator_state.sectors_done = 0;
  129. g_initiator_state.retrycount = 0;
  130. g_initiator_state.max_sector_per_transfer = 512;
  131. if (!(g_initiator_state.drives_imaged & (1 << g_initiator_state.target_id)))
  132. {
  133. delay_with_poll(1000);
  134. uint8_t inquiry_data[36];
  135. LED_ON();
  136. bool startstopok =
  137. scsiTestUnitReady(g_initiator_state.target_id) &&
  138. scsiStartStopUnit(g_initiator_state.target_id, true);
  139. bool readcapok = startstopok &&
  140. scsiInitiatorReadCapacity(g_initiator_state.target_id,
  141. &g_initiator_state.sectorcount,
  142. &g_initiator_state.sectorsize);
  143. bool inquiryok = startstopok &&
  144. scsiInquiry(g_initiator_state.target_id, inquiry_data);
  145. LED_OFF();
  146. if (readcapok)
  147. {
  148. logmsg("SCSI id ", g_initiator_state.target_id,
  149. " capacity ", (int)g_initiator_state.sectorcount,
  150. " sectors x ", (int)g_initiator_state.sectorsize, " bytes");
  151. g_initiator_state.sectorcount_all = g_initiator_state.sectorcount;
  152. uint64_t total_bytes = (uint64_t)g_initiator_state.sectorcount * g_initiator_state.sectorsize;
  153. logmsg("Drive total size is ", (int)(total_bytes / (1024 * 1024)), " MiB");
  154. if (total_bytes >= 0xFFFFFFFF && SD.fatType() != FAT_TYPE_EXFAT)
  155. {
  156. // Note: the FAT32 limit is 4 GiB - 1 byte
  157. logmsg("Image files equal or larger than 4 GiB are only possible on exFAT filesystem");
  158. logmsg("Please reformat the SD card with exFAT format to image this drive fully");
  159. g_initiator_state.sectorcount = (uint32_t)0xFFFFFFFF / g_initiator_state.sectorsize;
  160. logmsg("Will image first 4 GiB - 1 = ", (int)g_initiator_state.sectorcount, " sectors");
  161. }
  162. }
  163. else if (startstopok)
  164. {
  165. logmsg("SCSI id ", g_initiator_state.target_id, " responds but ReadCapacity command failed");
  166. logmsg("Possibly SCSI-1 drive? Attempting to read up to 1 GB.");
  167. g_initiator_state.sectorsize = 512;
  168. g_initiator_state.sectorcount = g_initiator_state.sectorcount_all = 2097152;
  169. g_initiator_state.max_sector_per_transfer = 128;
  170. }
  171. else
  172. {
  173. dbgmsg("Failed to connect to SCSI id ", g_initiator_state.target_id);
  174. g_initiator_state.sectorsize = 0;
  175. g_initiator_state.sectorcount = g_initiator_state.sectorcount_all = 0;
  176. }
  177. const char *filename_format = "HD00_imaged.hda";
  178. if (inquiryok)
  179. {
  180. if ((inquiry_data[0] & 0x1F) == 5)
  181. {
  182. filename_format = "CD00_imaged.iso";
  183. }
  184. }
  185. if (g_initiator_state.sectorcount > 0)
  186. {
  187. char filename[32] = {0};
  188. strncpy(filename, filename_format, sizeof(filename) - 1);
  189. filename[2] += g_initiator_state.target_id;
  190. SD.remove(filename);
  191. g_initiator_state.target_file = SD.open(filename, O_RDWR | O_CREAT | O_TRUNC);
  192. if (!g_initiator_state.target_file.isOpen())
  193. {
  194. logmsg("Failed to open file for writing: ", filename);
  195. return;
  196. }
  197. if (SD.fatType() == FAT_TYPE_EXFAT)
  198. {
  199. // Only preallocate on exFAT, on FAT32 preallocating can result in false garbage data in the
  200. // file if write is interrupted.
  201. logmsg("Preallocating image file");
  202. g_initiator_state.target_file.preAllocate((uint64_t)g_initiator_state.sectorcount * g_initiator_state.sectorsize);
  203. }
  204. logmsg("Starting to copy drive data to ", filename);
  205. g_initiator_state.imaging = true;
  206. }
  207. }
  208. }
  209. else
  210. {
  211. // Copy sectors from SCSI drive to file
  212. if (g_initiator_state.sectors_done >= g_initiator_state.sectorcount)
  213. {
  214. scsiStartStopUnit(g_initiator_state.target_id, false);
  215. logmsg("Finished imaging drive with id ", g_initiator_state.target_id);
  216. LED_OFF();
  217. if (g_initiator_state.sectorcount != g_initiator_state.sectorcount_all)
  218. {
  219. logmsg("NOTE: Image size was limited to first 4 GiB due to SD card filesystem limit");
  220. logmsg("Please reformat the SD card with exFAT format to image this drive fully");
  221. }
  222. g_initiator_state.drives_imaged |= (1 << g_initiator_state.target_id);
  223. g_initiator_state.imaging = false;
  224. g_initiator_state.target_file.close();
  225. return;
  226. }
  227. scsiInitiatorUpdateLed();
  228. // How many sectors to read in one batch?
  229. int numtoread = g_initiator_state.sectorcount - g_initiator_state.sectors_done;
  230. if (numtoread > g_initiator_state.max_sector_per_transfer)
  231. numtoread = g_initiator_state.max_sector_per_transfer;
  232. // Retry sector-by-sector after failure
  233. if (g_initiator_state.sectors_done < g_initiator_state.failposition)
  234. numtoread = 1;
  235. uint32_t time_start = millis();
  236. bool status = scsiInitiatorReadDataToFile(g_initiator_state.target_id,
  237. g_initiator_state.sectors_done, numtoread, g_initiator_state.sectorsize,
  238. g_initiator_state.target_file);
  239. if (!status)
  240. {
  241. logmsg("Failed to transfer ", numtoread, " sectors starting at ", (int)g_initiator_state.sectors_done);
  242. if (g_initiator_state.retrycount < 5)
  243. {
  244. logmsg("Retrying.. ", g_initiator_state.retrycount, "/5");
  245. delay_with_poll(200);
  246. scsiHostPhyReset();
  247. delay_with_poll(200);
  248. g_initiator_state.retrycount++;
  249. g_initiator_state.target_file.seek((uint64_t)g_initiator_state.sectors_done * g_initiator_state.sectorsize);
  250. if (g_initiator_state.retrycount > 1 && numtoread > 1)
  251. {
  252. logmsg("Multiple failures, retrying sector-by-sector");
  253. g_initiator_state.failposition = g_initiator_state.sectors_done + numtoread;
  254. }
  255. }
  256. else
  257. {
  258. logmsg("Retry limit exceeded, skipping one sector");
  259. g_initiator_state.retrycount = 0;
  260. g_initiator_state.sectors_done++;
  261. g_initiator_state.target_file.seek((uint64_t)g_initiator_state.sectors_done * g_initiator_state.sectorsize);
  262. }
  263. }
  264. else
  265. {
  266. g_initiator_state.retrycount = 0;
  267. g_initiator_state.sectors_done += numtoread;
  268. g_initiator_state.target_file.flush();
  269. int speed_kbps = numtoread * g_initiator_state.sectorsize / (millis() - time_start);
  270. logmsg("SCSI read succeeded, sectors done: ",
  271. (int)g_initiator_state.sectors_done, " / ", (int)g_initiator_state.sectorcount,
  272. " speed ", speed_kbps, " kB/s");
  273. }
  274. }
  275. }
  276. /*************************************
  277. * Low level command implementations *
  278. *************************************/
  279. int scsiInitiatorRunCommand(int target_id,
  280. const uint8_t *command, size_t cmdLen,
  281. uint8_t *bufIn, size_t bufInLen,
  282. const uint8_t *bufOut, size_t bufOutLen,
  283. bool returnDataPhase)
  284. {
  285. if (!scsiHostPhySelect(target_id))
  286. {
  287. dbgmsg("------ Target ", target_id, " did not respond");
  288. scsiHostPhyRelease();
  289. return -1;
  290. }
  291. SCSI_PHASE phase;
  292. int status = -1;
  293. while ((phase = (SCSI_PHASE)scsiHostPhyGetPhase()) != BUS_FREE)
  294. {
  295. platform_poll();
  296. if (phase == MESSAGE_IN)
  297. {
  298. uint8_t dummy = 0;
  299. scsiHostRead(&dummy, 1);
  300. }
  301. else if (phase == MESSAGE_OUT)
  302. {
  303. uint8_t identify_msg = 0x80;
  304. scsiHostWrite(&identify_msg, 1);
  305. }
  306. else if (phase == COMMAND)
  307. {
  308. scsiHostWrite(command, cmdLen);
  309. }
  310. else if (phase == DATA_IN)
  311. {
  312. if (returnDataPhase) return 0;
  313. if (bufInLen == 0)
  314. {
  315. logmsg("DATA_IN phase but no data to receive!");
  316. status = -3;
  317. break;
  318. }
  319. if (scsiHostRead(bufIn, bufInLen) == 0)
  320. {
  321. logmsg("scsiHostRead failed, tried to read ", (int)bufInLen, " bytes");
  322. status = -2;
  323. break;
  324. }
  325. }
  326. else if (phase == DATA_OUT)
  327. {
  328. if (returnDataPhase) return 0;
  329. if (bufOutLen == 0)
  330. {
  331. logmsg("DATA_OUT phase but no data to send!");
  332. status = -3;
  333. break;
  334. }
  335. if (scsiHostWrite(bufOut, bufOutLen) < bufOutLen)
  336. {
  337. logmsg("scsiHostWrite failed, was writing ", bytearray(bufOut, bufOutLen));
  338. status = -2;
  339. break;
  340. }
  341. }
  342. else if (phase == STATUS)
  343. {
  344. uint8_t tmp = -1;
  345. scsiHostRead(&tmp, 1);
  346. status = tmp;
  347. dbgmsg("------ STATUS: ", tmp);
  348. }
  349. }
  350. scsiHostPhyRelease();
  351. return status;
  352. }
  353. bool scsiInitiatorReadCapacity(int target_id, uint32_t *sectorcount, uint32_t *sectorsize)
  354. {
  355. uint8_t command[10] = {0x25, 0, 0, 0, 0, 0, 0, 0, 0, 0};
  356. uint8_t response[8] = {0};
  357. int status = scsiInitiatorRunCommand(target_id,
  358. command, sizeof(command),
  359. response, sizeof(response),
  360. NULL, 0);
  361. if (status == 0)
  362. {
  363. *sectorcount = ((uint32_t)response[0] << 24)
  364. | ((uint32_t)response[1] << 16)
  365. | ((uint32_t)response[2] << 8)
  366. | ((uint32_t)response[3] << 0);
  367. *sectorcount += 1; // SCSI reports last sector address
  368. *sectorsize = ((uint32_t)response[4] << 24)
  369. | ((uint32_t)response[5] << 16)
  370. | ((uint32_t)response[6] << 8)
  371. | ((uint32_t)response[7] << 0);
  372. return true;
  373. }
  374. else if (status == 2)
  375. {
  376. uint8_t sense_key;
  377. scsiRequestSense(target_id, &sense_key);
  378. logmsg("READ CAPACITY on target ", target_id, " failed, sense key ", sense_key);
  379. return false;
  380. }
  381. else
  382. {
  383. *sectorcount = *sectorsize = 0;
  384. return false;
  385. }
  386. }
  387. // Execute REQUEST SENSE command to get more information about error status
  388. bool scsiRequestSense(int target_id, uint8_t *sense_key)
  389. {
  390. uint8_t command[6] = {0x03, 0, 0, 0, 18, 0};
  391. uint8_t response[18] = {0};
  392. int status = scsiInitiatorRunCommand(target_id,
  393. command, sizeof(command),
  394. response, sizeof(response),
  395. NULL, 0);
  396. dbgmsg("RequestSense response: ", bytearray(response, 18));
  397. *sense_key = response[2];
  398. return status == 0;
  399. }
  400. // Execute UNIT START STOP command to load/unload media
  401. bool scsiStartStopUnit(int target_id, bool start)
  402. {
  403. uint8_t command[6] = {0x1B, 0, 0, 0, 0, 0};
  404. uint8_t response[4] = {0};
  405. if (start) command[4] |= 1;
  406. int status = scsiInitiatorRunCommand(target_id,
  407. command, sizeof(command),
  408. response, sizeof(response),
  409. NULL, 0);
  410. if (status == 2)
  411. {
  412. uint8_t sense_key;
  413. scsiRequestSense(target_id, &sense_key);
  414. logmsg("START STOP UNIT on target ", target_id, " failed, sense key ", sense_key);
  415. }
  416. return status == 0;
  417. }
  418. // Execute INQUIRY command
  419. bool scsiInquiry(int target_id, uint8_t inquiry_data[36])
  420. {
  421. uint8_t command[6] = {0x12, 0, 0, 0, 36, 0};
  422. int status = scsiInitiatorRunCommand(target_id,
  423. command, sizeof(command),
  424. inquiry_data, 36,
  425. NULL, 0);
  426. return status == 0;
  427. }
  428. // Execute TEST UNIT READY command and handle unit attention state
  429. bool scsiTestUnitReady(int target_id)
  430. {
  431. for (int retries = 0; retries < 2; retries++)
  432. {
  433. uint8_t command[6] = {0x00, 0, 0, 0, 0, 0};
  434. int status = scsiInitiatorRunCommand(target_id,
  435. command, sizeof(command),
  436. NULL, 0,
  437. NULL, 0);
  438. if (status == 0)
  439. {
  440. return true;
  441. }
  442. else if (status == -1)
  443. {
  444. // No response to select
  445. return false;
  446. }
  447. else if (status == 2)
  448. {
  449. uint8_t sense_key;
  450. scsiRequestSense(target_id, &sense_key);
  451. if (sense_key == 6)
  452. {
  453. uint8_t inquiry[36];
  454. logmsg("Target ", target_id, " reports UNIT_ATTENTION, running INQUIRY");
  455. scsiInquiry(target_id, inquiry);
  456. }
  457. else if (sense_key == 2)
  458. {
  459. logmsg("Target ", target_id, " reports NOT_READY, running STARTSTOPUNIT");
  460. scsiStartStopUnit(target_id, true);
  461. }
  462. }
  463. else
  464. {
  465. logmsg("Target ", target_id, " TEST UNIT READY response: ", status);
  466. }
  467. }
  468. return false;
  469. }
  470. // This uses callbacks to run SD and SCSI transfers in parallel
  471. static struct {
  472. uint32_t bytes_sd; // Number of bytes that have been transferred on SD card side
  473. uint32_t bytes_sd_scheduled; // Number of bytes scheduled for transfer on SD card side
  474. uint32_t bytes_scsi; // Number of bytes that have been scheduled for transfer on SCSI side
  475. uint32_t bytes_scsi_done; // Number of bytes that have been transferred on SCSI side
  476. uint32_t bytes_per_sector;
  477. bool all_ok;
  478. } g_initiator_transfer;
  479. static void initiatorReadSDCallback(uint32_t bytes_complete)
  480. {
  481. if (g_initiator_transfer.bytes_scsi_done < g_initiator_transfer.bytes_scsi)
  482. {
  483. // How many bytes remaining in the transfer?
  484. uint32_t remain = g_initiator_transfer.bytes_scsi - g_initiator_transfer.bytes_scsi_done;
  485. uint32_t len = remain;
  486. // Limit maximum amount of data transferred at one go, to give enough callbacks to SD driver.
  487. // Select the limit based on total bytes in the transfer.
  488. // Transfer size is reduced towards the end of transfer to reduce the dead time between
  489. // end of SCSI transfer and the SD write completing.
  490. uint32_t limit = g_initiator_transfer.bytes_scsi / 8;
  491. uint32_t bytesPerSector = g_initiator_transfer.bytes_per_sector;
  492. if (limit < PLATFORM_OPTIMAL_MIN_SD_WRITE_SIZE) limit = PLATFORM_OPTIMAL_MIN_SD_WRITE_SIZE;
  493. if (limit > PLATFORM_OPTIMAL_MAX_SD_WRITE_SIZE) limit = PLATFORM_OPTIMAL_MAX_SD_WRITE_SIZE;
  494. if (limit > len) limit = PLATFORM_OPTIMAL_LAST_SD_WRITE_SIZE;
  495. if (limit < bytesPerSector) limit = bytesPerSector;
  496. if (len > limit)
  497. {
  498. len = limit;
  499. }
  500. // Split read so that it doesn't wrap around buffer edge
  501. uint32_t bufsize = sizeof(scsiDev.data);
  502. uint32_t start = (g_initiator_transfer.bytes_scsi_done % bufsize);
  503. if (start + len > bufsize)
  504. len = bufsize - start;
  505. // Don't overwrite data that has not yet been written to SD card
  506. uint32_t sd_ready_cnt = g_initiator_transfer.bytes_sd + bytes_complete;
  507. if (g_initiator_transfer.bytes_scsi_done + len > sd_ready_cnt + bufsize)
  508. len = sd_ready_cnt + bufsize - g_initiator_transfer.bytes_scsi_done;
  509. if (sd_ready_cnt == g_initiator_transfer.bytes_sd_scheduled &&
  510. g_initiator_transfer.bytes_sd_scheduled + bytesPerSector <= g_initiator_transfer.bytes_scsi_done)
  511. {
  512. // Current SD transfer is complete, it is better we return now and offer a chance for the next
  513. // transfer to begin.
  514. return;
  515. }
  516. // Keep transfers a multiple of sector size.
  517. if (remain >= bytesPerSector && len % bytesPerSector != 0)
  518. {
  519. len -= len % bytesPerSector;
  520. }
  521. if (len == 0)
  522. return;
  523. // dbgmsg("SCSI read ", (int)start, " + ", (int)len, ", sd ready cnt ", (int)sd_ready_cnt, " ", (int)bytes_complete, ", scsi done ", (int)g_initiator_transfer.bytes_scsi_done);
  524. if (scsiHostRead(&scsiDev.data[start], len) != len)
  525. {
  526. logmsg("Read failed at byte ", (int)g_initiator_transfer.bytes_scsi_done);
  527. g_initiator_transfer.all_ok = false;
  528. }
  529. g_initiator_transfer.bytes_scsi_done += len;
  530. }
  531. }
  532. static void scsiInitiatorWriteDataToSd(FsFile &file, bool use_callback)
  533. {
  534. // Figure out longest continuous block in buffer
  535. uint32_t bufsize = sizeof(scsiDev.data);
  536. uint32_t start = g_initiator_transfer.bytes_sd % bufsize;
  537. uint32_t len = g_initiator_transfer.bytes_scsi_done - g_initiator_transfer.bytes_sd;
  538. if (start + len > bufsize) len = bufsize - start;
  539. // Try to do writes in multiple of 512 bytes
  540. // This allows better performance for SD card access.
  541. if (len >= 512) len &= ~511;
  542. // Start writing to SD card and simultaneously reading more from SCSI bus
  543. uint8_t *buf = &scsiDev.data[start];
  544. // dbgmsg("SD write ", (int)start, " + ", (int)len);
  545. if (use_callback)
  546. {
  547. platform_set_sd_callback(&initiatorReadSDCallback, buf);
  548. }
  549. g_initiator_transfer.bytes_sd_scheduled = g_initiator_transfer.bytes_sd + len;
  550. if (file.write(buf, len) != len)
  551. {
  552. logmsg("scsiInitiatorReadDataToFile: SD card write failed");
  553. g_initiator_transfer.all_ok = false;
  554. }
  555. platform_set_sd_callback(NULL, NULL);
  556. g_initiator_transfer.bytes_sd += len;
  557. }
  558. bool scsiInitiatorReadDataToFile(int target_id, uint32_t start_sector, uint32_t sectorcount, uint32_t sectorsize,
  559. FsFile &file)
  560. {
  561. int status = -1;
  562. if (start_sector < 0xFFFFFF && sectorcount <= 256)
  563. {
  564. // Use READ6 command for compatibility with old SCSI1 drives
  565. uint8_t command[6] = {0x08,
  566. (uint8_t)(start_sector >> 16),
  567. (uint8_t)(start_sector >> 8),
  568. (uint8_t)start_sector,
  569. (uint8_t)sectorcount,
  570. 0x00
  571. };
  572. // Start executing command, return in data phase
  573. status = scsiInitiatorRunCommand(target_id, command, sizeof(command), NULL, 0, NULL, 0, true);
  574. }
  575. else
  576. {
  577. // Use READ10 command for larger number of blocks
  578. uint8_t command[10] = {0x28, 0x00,
  579. (uint8_t)(start_sector >> 24), (uint8_t)(start_sector >> 16),
  580. (uint8_t)(start_sector >> 8), (uint8_t)start_sector,
  581. 0x00,
  582. (uint8_t)(sectorcount >> 8), (uint8_t)(sectorcount),
  583. 0x00
  584. };
  585. // Start executing command, return in data phase
  586. status = scsiInitiatorRunCommand(target_id, command, sizeof(command), NULL, 0, NULL, 0, true);
  587. }
  588. if (status != 0)
  589. {
  590. uint8_t sense_key;
  591. scsiRequestSense(target_id, &sense_key);
  592. logmsg("scsiInitiatorReadDataToFile: READ failed: ", status, " sense key ", sense_key);
  593. scsiHostPhyRelease();
  594. return false;
  595. }
  596. SCSI_PHASE phase;
  597. g_initiator_transfer.bytes_scsi = sectorcount * sectorsize;
  598. g_initiator_transfer.bytes_per_sector = sectorsize;
  599. g_initiator_transfer.bytes_sd = 0;
  600. g_initiator_transfer.bytes_sd_scheduled = 0;
  601. g_initiator_transfer.bytes_scsi_done = 0;
  602. g_initiator_transfer.all_ok = true;
  603. while (true)
  604. {
  605. platform_poll();
  606. phase = (SCSI_PHASE)scsiHostPhyGetPhase();
  607. if (phase != DATA_IN && phase != BUS_BUSY)
  608. {
  609. break;
  610. }
  611. // Read next block from SCSI bus if buffer empty
  612. if (g_initiator_transfer.bytes_sd == g_initiator_transfer.bytes_scsi_done)
  613. {
  614. initiatorReadSDCallback(0);
  615. }
  616. else
  617. {
  618. // Write data to SD card and simultaneously read more from SCSI
  619. scsiInitiatorUpdateLed();
  620. scsiInitiatorWriteDataToSd(file, true);
  621. }
  622. }
  623. // Write any remaining buffered data
  624. while (g_initiator_transfer.bytes_sd < g_initiator_transfer.bytes_scsi_done)
  625. {
  626. platform_poll();
  627. scsiInitiatorWriteDataToSd(file, false);
  628. }
  629. if (g_initiator_transfer.bytes_sd != g_initiator_transfer.bytes_scsi)
  630. {
  631. logmsg("SCSI read from sector ", (int)start_sector, " was incomplete: expected ",
  632. (int)g_initiator_transfer.bytes_scsi, " got ", (int)g_initiator_transfer.bytes_sd, " bytes");
  633. g_initiator_transfer.all_ok = false;
  634. }
  635. while ((phase = (SCSI_PHASE)scsiHostPhyGetPhase()) != BUS_FREE)
  636. {
  637. platform_poll();
  638. if (phase == MESSAGE_IN)
  639. {
  640. uint8_t dummy = 0;
  641. scsiHostRead(&dummy, 1);
  642. }
  643. else if (phase == MESSAGE_OUT)
  644. {
  645. uint8_t identify_msg = 0x80;
  646. scsiHostWrite(&identify_msg, 1);
  647. }
  648. else if (phase == STATUS)
  649. {
  650. uint8_t tmp = 0;
  651. scsiHostRead(&tmp, 1);
  652. status = tmp;
  653. dbgmsg("------ STATUS: ", tmp);
  654. }
  655. }
  656. scsiHostPhyRelease();
  657. return status == 0 && g_initiator_transfer.all_ok;
  658. }
  659. #endif