fwdownload.c revision 302377
1/*-
2 * Copyright (c) 2011 Sandvine Incorporated. All rights reserved.
3 * Copyright (c) 2002-2011 Andre Albsmeier <andre@albsmeier.net>
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 *    notice, this list of conditions and the following disclaimer,
11 *    without modification, immediately at the beginning of the file.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28/*
29 * This software is derived from Andre Albsmeier's fwprog.c which contained
30 * the following note:
31 *
32 * Many thanks goes to Marc Frajola <marc@terasolutions.com> from
33 * TeraSolutions for the initial idea and his programme for upgrading
34 * the firmware of I*M DDYS drives.
35 */
36
37/*
38 * BEWARE:
39 *
40 * The fact that you see your favorite vendor listed below does not
41 * imply that your equipment won't break when you use this software
42 * with it. It only means that the firmware of at least one device type
43 * of each vendor listed has been programmed successfully using this code.
44 *
45 * The -s option simulates a download but does nothing apart from that.
46 * It can be used to check what chunk sizes would have been used with the
47 * specified device.
48 */
49
50#include <sys/cdefs.h>
51__FBSDID("$FreeBSD: stable/10/sbin/camcontrol/fwdownload.c 302377 2016-07-06 17:45:38Z truckman $");
52
53#include <sys/types.h>
54#include <sys/stat.h>
55
56#include <err.h>
57#include <fcntl.h>
58#include <stdio.h>
59#include <stdlib.h>
60#include <string.h>
61#include <unistd.h>
62
63#include <cam/scsi/scsi_all.h>
64#include <cam/scsi/scsi_message.h>
65#include <camlib.h>
66
67#include "progress.h"
68
69#include "camcontrol.h"
70
71#define	WB_TIMEOUT 50000	/* 50 seconds */
72
73typedef enum {
74	VENDOR_HGST,
75	VENDOR_HITACHI,
76	VENDOR_HP,
77	VENDOR_IBM,
78	VENDOR_PLEXTOR,
79	VENDOR_QUALSTAR,
80	VENDOR_QUANTUM,
81	VENDOR_SAMSUNG,
82	VENDOR_SEAGATE,
83	VENDOR_SMART,
84	VENDOR_ATA,
85	VENDOR_UNKNOWN
86} fw_vendor_t;
87
88/*
89 * FW_TUR_READY:     The drive must return good status for a test unit ready.
90 *
91 * FW_TUR_NOT_READY: The drive must return not ready status for a test unit
92 *		     ready.  You may want this in a removable media drive.
93 *
94 * FW_TUR_NA:	     It doesn't matter whether the drive is ready or not.
95 * 		     This may be the case for a removable media drive.
96 */
97typedef enum {
98	FW_TUR_NONE,
99	FW_TUR_READY,
100	FW_TUR_NOT_READY,
101	FW_TUR_NA
102} fw_tur_status;
103
104/*
105 * FW_TIMEOUT_DEFAULT:		Attempt to probe for a WRITE BUFFER timeout
106 *				value from the drive.  If we get an answer,
107 *				use the Recommended timeout.  Otherwise,
108 * 				use the default value from the table.
109 *
110 * FW_TIMEOUT_DEV_REPORTED:	The timeout value was probed directly from
111 *				the device.
112 *
113 * FW_TIMEOUT_NO_PROBE:		Do not ask the device for a WRITE BUFFER
114 * 				timeout value.  Use the device-specific
115 *				value.
116 *
117 * FW_TIMEOUT_USER_SPEC:	The user specified a timeout on the command
118 *				line with the -t option.  This overrides any
119 *				probe or default timeout.
120 */
121typedef enum {
122	FW_TIMEOUT_DEFAULT,
123	FW_TIMEOUT_DEV_REPORTED,
124	FW_TIMEOUT_NO_PROBE,
125	FW_TIMEOUT_USER_SPEC
126} fw_timeout_type;
127
128/*
129 * type: 		Enumeration for the particular vendor.
130 *
131 * pattern:		Pattern to match for the Vendor ID from the SCSI
132 *			Inquiry data.
133 *
134 * dev_type:		SCSI device type to match, or T_ANY to match any
135 *			device from the given vendor.  Note that if there
136 *			is a specific device type listed for a particular
137 *			vendor, it must be listed before a T_ANY entry.
138 *
139 * max_pkt_size:	Maximum packet size when talking to a device.  Note
140 *			that although large data sizes may be supported by
141 *			the target device, they may not be supported by the
142 *			OS or the controller.
143 *
144 * cdb_byte2:		This specifies byte 2 (byte 1 when counting from 0)
145 *			of the CDB.  This is generally the WRITE BUFFER mode.
146 *
147 * cdb_byte2_last:	This specifies byte 2 for the last chunk of the
148 *			download.
149 *
150 * inc_cdb_buffer_id:	Increment the buffer ID by 1 for each chunk sent
151 *			down to the drive.
152 *
153 * inc_cdb_offset:	Increment the offset field in the CDB with the byte
154 *			offset into the firmware file.
155 *
156 * tur_status:		Pay attention to whether the device is ready before
157 *			upgrading the firmware, or not.  See above for the
158 *			values.
159 */
160struct fw_vendor {
161	fw_vendor_t type;
162	const char *pattern;
163	int dev_type;
164	int max_pkt_size;
165	u_int8_t cdb_byte2;
166	u_int8_t cdb_byte2_last;
167	int inc_cdb_buffer_id;
168	int inc_cdb_offset;
169	fw_tur_status tur_status;
170	int timeout_ms;
171	fw_timeout_type timeout_type;
172};
173
174/*
175 * Vendor notes:
176 *
177 * HGST:     The packets need to be sent in multiples of 4K.
178 *
179 * IBM:      For LTO and TS drives, the buffer ID is ignored in mode 7 (and
180 * 	     some other modes).  It treats the request as a firmware download.
181 *           The offset (and therefore the length of each chunk sent) needs
182 *           to be a multiple of the offset boundary specified for firmware
183 *           (buffer ID 4) in the read buffer command.  At least for LTO-6,
184 *           that seems to be 0, but using a 32K chunk size should satisfy
185 *           most any alignment requirement.
186 *
187 * SmrtStor: Mode 5 is also supported, but since the firmware is 400KB or
188 *           so, we can't fit it in a single request in most cases.
189 */
190static struct fw_vendor vendors_list[] = {
191	{VENDOR_HGST,	 	"HGST",		T_DIRECT,
192	0x1000, 0x07, 0x07, 1, 0, FW_TUR_READY, WB_TIMEOUT, FW_TIMEOUT_DEFAULT},
193	{VENDOR_HITACHI, 	"HITACHI",	T_ANY,
194	0x8000, 0x05, 0x05, 1, 0, FW_TUR_READY, WB_TIMEOUT, FW_TIMEOUT_DEFAULT},
195	{VENDOR_HP,	 	"HP",		T_ANY,
196	0x8000, 0x07, 0x07, 0, 1, FW_TUR_READY, WB_TIMEOUT, FW_TIMEOUT_DEFAULT},
197	{VENDOR_IBM,		"IBM",		T_SEQUENTIAL,
198	0x8000, 0x07, 0x07, 0, 1, FW_TUR_NA, 300 * 1000, FW_TIMEOUT_DEFAULT},
199	{VENDOR_IBM,		"IBM",		T_ANY,
200	0x8000, 0x05, 0x05, 1, 0, FW_TUR_READY, WB_TIMEOUT, FW_TIMEOUT_DEFAULT},
201	{VENDOR_PLEXTOR,	"PLEXTOR",	T_ANY,
202	0x2000, 0x04, 0x05, 0, 1, FW_TUR_READY, WB_TIMEOUT, FW_TIMEOUT_DEFAULT},
203	{VENDOR_QUALSTAR,	"QUALSTAR",	T_ANY,
204	0x2030, 0x05, 0x05, 0, 0, FW_TUR_READY, WB_TIMEOUT, FW_TIMEOUT_DEFAULT},
205	{VENDOR_QUANTUM,	"QUANTUM",	T_ANY,
206	0x2000, 0x04, 0x05, 0, 1, FW_TUR_READY, WB_TIMEOUT, FW_TIMEOUT_DEFAULT},
207	{VENDOR_SAMSUNG,	"SAMSUNG",	T_ANY,
208	0x8000, 0x07, 0x07, 0, 1, FW_TUR_READY, WB_TIMEOUT, FW_TIMEOUT_DEFAULT},
209	{VENDOR_SEAGATE,	"SEAGATE",	T_ANY,
210	0x8000, 0x07, 0x07, 0, 1, FW_TUR_READY, WB_TIMEOUT, FW_TIMEOUT_DEFAULT},
211	{VENDOR_SMART,		"SmrtStor",	T_DIRECT,
212	0x8000, 0x07, 0x07, 0, 1, FW_TUR_READY, WB_TIMEOUT, FW_TIMEOUT_DEFAULT},
213
214	/*
215	 * We match any ATA device.  This is really just a placeholder,
216	 * since we won't actually send a WRITE BUFFER with any of the
217	 * listed parameters.  If a SATA device is behind a SAS controller,
218	 * the SCSI to ATA translation code (at least for LSI) doesn't
219	 * generaly translate a SCSI WRITE BUFFER into an ATA DOWNLOAD
220	 * MICROCODE command.  So, we use the SCSI ATA PASS_THROUGH command
221	 * to send the ATA DOWNLOAD MICROCODE command instead.
222	 */
223	{VENDOR_ATA,		"ATA",		T_ANY,
224	 0x8000, 0x07, 0x07, 0, 1, FW_TUR_READY, WB_TIMEOUT,
225	 FW_TIMEOUT_NO_PROBE},
226	{VENDOR_UNKNOWN,	NULL,		T_ANY,
227	0x0000, 0x00, 0x00, 0, 0, FW_TUR_NONE, WB_TIMEOUT, FW_TIMEOUT_DEFAULT}
228};
229
230struct fw_timeout_desc {
231	fw_timeout_type timeout_type;
232	const char *timeout_desc;
233};
234
235static const struct fw_timeout_desc fw_timeout_desc_table[] = {
236	{ FW_TIMEOUT_DEFAULT, "the default" },
237	{ FW_TIMEOUT_DEV_REPORTED, "recommended by this particular device" },
238	{ FW_TIMEOUT_NO_PROBE, "the default" },
239	{ FW_TIMEOUT_USER_SPEC, "what was specified on the command line" }
240};
241
242#ifndef ATA_DOWNLOAD_MICROCODE
243#define ATA_DOWNLOAD_MICROCODE	0x92
244#endif
245
246#define USE_OFFSETS_FEATURE	0x3
247
248#ifndef LOW_SECTOR_SIZE
249#define LOW_SECTOR_SIZE		512
250#endif
251
252#define ATA_MAKE_LBA(o, p)	\
253	((((((o) / LOW_SECTOR_SIZE) >> 8) & 0xff) << 16) | \
254	  ((((o) / LOW_SECTOR_SIZE) & 0xff) << 8) | \
255	  ((((p) / LOW_SECTOR_SIZE) >> 8) & 0xff))
256
257#define ATA_MAKE_SECTORS(p)	(((p) / 512) & 0xff)
258
259#ifndef UNKNOWN_MAX_PKT_SIZE
260#define UNKNOWN_MAX_PKT_SIZE	0x8000
261#endif
262
263static struct fw_vendor *fw_get_vendor(struct cam_device *cam_dev,
264				       struct ata_params *ident_buf);
265static int fw_get_timeout(struct cam_device *cam_dev, struct fw_vendor *vp,
266			  int retry_count, int timeout);
267static int fw_validate_ibm(struct cam_device *dev, int retry_count,
268			   int timeout, int fd, char *buf,
269			    const char *fw_img_path, int quiet);
270static char *fw_read_img(struct cam_device *dev, int retry_count,
271			 int timeout, int quiet, const char *fw_img_path,
272			 struct fw_vendor *vp, int *num_bytes);
273static int fw_check_device_ready(struct cam_device *dev,
274				 camcontrol_devtype devtype,
275				 struct fw_vendor *vp, int printerrors,
276				 int timeout);
277static int fw_download_img(struct cam_device *cam_dev,
278			   struct fw_vendor *vp, char *buf, int img_size,
279			   int sim_mode, int printerrors, int quiet,
280			   int retry_count, int timeout, const char */*name*/,
281			   camcontrol_devtype devtype);
282
283/*
284 * Find entry in vendors list that belongs to
285 * the vendor of given cam device.
286 */
287static struct fw_vendor *
288fw_get_vendor(struct cam_device *cam_dev, struct ata_params *ident_buf)
289{
290	char vendor[42];
291	struct fw_vendor *vp;
292
293	if (cam_dev == NULL)
294		return (NULL);
295
296	if (ident_buf != NULL) {
297		cam_strvis((u_char *)vendor, ident_buf->model,
298		    sizeof(ident_buf->model), sizeof(vendor));
299		for (vp = vendors_list; vp->pattern != NULL; vp++) {
300			if (vp->type == VENDOR_ATA)
301				return (vp);
302		}
303	} else {
304		cam_strvis((u_char *)vendor, (u_char *)cam_dev->inq_data.vendor,
305		    sizeof(cam_dev->inq_data.vendor), sizeof(vendor));
306	}
307	for (vp = vendors_list; vp->pattern != NULL; vp++) {
308		if (!cam_strmatch((const u_char *)vendor,
309		    (const u_char *)vp->pattern, strlen(vendor))) {
310			if ((vp->dev_type == T_ANY)
311			 || (vp->dev_type == SID_TYPE(&cam_dev->inq_data)))
312				break;
313		}
314	}
315	return (vp);
316}
317
318static int
319fw_get_timeout(struct cam_device *cam_dev, struct fw_vendor *vp,
320	       int retry_count, int timeout)
321{
322	struct scsi_report_supported_opcodes_one *one;
323	struct scsi_report_supported_opcodes_timeout *td;
324	uint8_t *buf = NULL;
325	uint32_t fill_len = 0, cdb_len = 0, rec_timeout = 0;
326	int retval = 0;
327
328	/*
329	 * If the user has specified a timeout on the command line, we let
330	 * him override any default or probed value.
331	 */
332	if (timeout != 0) {
333		vp->timeout_type = FW_TIMEOUT_USER_SPEC;
334		vp->timeout_ms = timeout;
335		goto bailout;
336	}
337
338	/*
339	 * Check to see whether we should probe for a timeout for this
340	 * device.
341	 */
342	if (vp->timeout_type == FW_TIMEOUT_NO_PROBE)
343		goto bailout;
344
345	retval = scsigetopcodes(/*device*/ cam_dev,
346				/*opcode_set*/ 1,
347				/*opcode*/ WRITE_BUFFER,
348				/*show_sa_errors*/ 1,
349				/*sa_set*/ 0,
350				/*service_action*/ 0,
351				/*timeout_desc*/ 1,
352				/*retry_count*/ retry_count,
353				/*timeout*/ 10000,
354				/*verbose*/ 0,
355				/*fill_len*/ &fill_len,
356				/*data_ptr*/ &buf);
357	/*
358	 * It isn't an error if we can't get a timeout descriptor.  We just
359	 * continue on with the default timeout.
360	 */
361	if (retval != 0) {
362		retval = 0;
363		goto bailout;
364	}
365
366	/*
367	 * Even if the drive didn't return a SCSI error, if we don't have
368	 * enough data to contain the one opcode descriptor, the CDB
369	 * structure and a timeout descriptor, we don't have the timeout
370	 * value we're looking for.  So we'll just fall back to the
371	 * default value.
372	 */
373	if (fill_len < (sizeof(*one) + sizeof(struct scsi_write_buffer) +
374	    sizeof(*td)))
375		goto bailout;
376
377	one = (struct scsi_report_supported_opcodes_one *)buf;
378
379	/*
380	 * If the drive claims to not support the WRITE BUFFER command...
381	 * fall back to the default timeout value and let things fail on
382	 * the actual firmware download.
383	 */
384	if ((one->support & RSO_ONE_SUP_MASK) == RSO_ONE_SUP_NOT_SUP)
385		goto bailout;
386
387	cdb_len = scsi_2btoul(one->cdb_length);
388	td = (struct scsi_report_supported_opcodes_timeout *)
389	    &buf[sizeof(*one) + cdb_len];
390
391	rec_timeout = scsi_4btoul(td->recommended_time);
392	/*
393	 * If the recommended timeout is 0, then the device has probably
394	 * returned a bogus value.
395	 */
396	if (rec_timeout == 0)
397		goto bailout;
398
399	/* CAM timeouts are in ms */
400	rec_timeout *= 1000;
401
402	vp->timeout_ms = rec_timeout;
403	vp->timeout_type = FW_TIMEOUT_DEV_REPORTED;
404
405bailout:
406	return (retval);
407}
408
409#define	SVPD_IBM_FW_DESIGNATION		0x03
410
411/*
412 * IBM LTO and TS tape drives have an INQUIRY VPD page 0x3 with the following
413 * format:
414 */
415struct fw_ibm_tape_fw_designation {
416	uint8_t	device;
417	uint8_t page_code;
418	uint8_t reserved;
419	uint8_t length;
420	uint8_t ascii_length;
421	uint8_t reserved2[3];
422	uint8_t load_id[4];
423	uint8_t fw_rev[4];
424	uint8_t ptf_number[4];
425	uint8_t patch_number[4];
426	uint8_t ru_name[8];
427	uint8_t lib_seq_num[5];
428};
429
430/*
431 * The firmware for IBM tape drives has the following header format.  The
432 * load_id and ru_name in the header file should match what is returned in
433 * VPD page 0x3.
434 */
435struct fw_ibm_tape_fw_header {
436	uint8_t unspec[4];
437	uint8_t length[4];		/* Firmware and header! */
438	uint8_t load_id[4];
439	uint8_t fw_rev[4];
440	uint8_t reserved[8];
441	uint8_t ru_name[8];
442};
443
444static int
445fw_validate_ibm(struct cam_device *dev, int retry_count, int timeout, int fd,
446		char *buf, const char *fw_img_path, int quiet)
447{
448	union ccb *ccb;
449	struct fw_ibm_tape_fw_designation vpd_page;
450	struct fw_ibm_tape_fw_header *header;
451	char drive_rev[sizeof(vpd_page.fw_rev) + 1];
452	char file_rev[sizeof(vpd_page.fw_rev) + 1];
453	int retval = 1;
454
455	ccb = cam_getccb(dev);
456	if (ccb == NULL) {
457		warnx("couldn't allocate CCB");
458		goto bailout;
459	}
460
461	/* cam_getccb cleans up the header, caller has to zero the payload */
462	CCB_CLEAR_ALL_EXCEPT_HDR(&ccb->csio);
463
464	bzero(&vpd_page, sizeof(vpd_page));
465
466	scsi_inquiry(&ccb->csio,
467		     /*retries*/ retry_count,
468		     /*cbfcnp*/ NULL,
469		     /* tag_action */ MSG_SIMPLE_Q_TAG,
470		     /* inq_buf */ (u_int8_t *)&vpd_page,
471		     /* inq_len */ sizeof(vpd_page),
472		     /* evpd */ 1,
473		     /* page_code */ SVPD_IBM_FW_DESIGNATION,
474		     /* sense_len */ SSD_FULL_SIZE,
475		     /* timeout */ timeout ? timeout : 5000);
476
477	/* Disable freezing the device queue */
478	ccb->ccb_h.flags |= CAM_DEV_QFRZDIS;
479
480	if (retry_count != 0)
481		ccb->ccb_h.flags |= CAM_PASS_ERR_RECOVER;
482
483	if (cam_send_ccb(dev, ccb) < 0) {
484		warn("error getting firmware designation page");
485
486		cam_error_print(dev, ccb, CAM_ESF_ALL,
487				CAM_EPF_ALL, stderr);
488
489		cam_freeccb(ccb);
490		ccb = NULL;
491		goto bailout;
492	}
493
494	if ((ccb->ccb_h.status & CAM_STATUS_MASK) != CAM_REQ_CMP) {
495		cam_error_print(dev, ccb, CAM_ESF_ALL,
496				CAM_EPF_ALL, stderr);
497		goto bailout;
498	}
499
500	/*
501	 * Read the firmware header only.
502	 */
503	if (read(fd, buf, sizeof(*header)) != sizeof(*header)) {
504		warn("unable to read %zu bytes from %s", sizeof(*header),
505		     fw_img_path);
506		goto bailout;
507	}
508
509	/* Rewind the file back to 0 for the full file read. */
510	if (lseek(fd, 0, SEEK_SET) == -1) {
511		warn("Unable to lseek");
512		goto bailout;
513	}
514
515	header = (struct fw_ibm_tape_fw_header *)buf;
516
517	bzero(drive_rev, sizeof(drive_rev));
518	bcopy(vpd_page.fw_rev, drive_rev, sizeof(vpd_page.fw_rev));
519	bzero(file_rev, sizeof(file_rev));
520	bcopy(header->fw_rev, file_rev, sizeof(header->fw_rev));
521
522	if (quiet == 0) {
523		fprintf(stdout, "Current Drive Firmware version: %s\n",
524			drive_rev);
525		fprintf(stdout, "Firmware File version: %s\n", file_rev);
526	}
527
528	/*
529	 * For IBM tape drives the load ID and RU name reported by the
530	 * drive should match what is in the firmware file.
531	 */
532	if (bcmp(vpd_page.load_id, header->load_id,
533		 MIN(sizeof(vpd_page.load_id), sizeof(header->load_id))) != 0) {
534		warnx("Drive Firmware load ID 0x%x does not match firmware "
535		      "file load ID 0x%x", scsi_4btoul(vpd_page.load_id),
536		      scsi_4btoul(header->load_id));
537		goto bailout;
538	}
539
540	if (bcmp(vpd_page.ru_name, header->ru_name,
541		 MIN(sizeof(vpd_page.ru_name), sizeof(header->ru_name))) != 0) {
542		warnx("Drive Firmware RU name 0x%jx does not match firmware "
543		      "file RU name 0x%jx",
544		      (uintmax_t)scsi_8btou64(vpd_page.ru_name),
545		      (uintmax_t)scsi_8btou64(header->ru_name));
546		goto bailout;
547	}
548	if (quiet == 0)
549		fprintf(stdout, "Firmware file is valid for this drive.\n");
550	retval = 0;
551bailout:
552	if (ccb != NULL)
553		cam_freeccb(ccb);
554
555	return (retval);
556}
557
558/*
559 * Allocate a buffer and read fw image file into it
560 * from given path. Number of bytes read is stored
561 * in num_bytes.
562 */
563static char *
564fw_read_img(struct cam_device *dev, int retry_count, int timeout, int quiet,
565	    const char *fw_img_path, struct fw_vendor *vp, int *num_bytes)
566{
567	int fd;
568	struct stat stbuf;
569	char *buf;
570	off_t img_size;
571	int skip_bytes = 0;
572
573	if ((fd = open(fw_img_path, O_RDONLY)) < 0) {
574		warn("Could not open image file %s", fw_img_path);
575		return (NULL);
576	}
577	if (fstat(fd, &stbuf) < 0) {
578		warn("Could not stat image file %s", fw_img_path);
579		goto bailout1;
580	}
581	if ((img_size = stbuf.st_size) == 0) {
582		warnx("Zero length image file %s", fw_img_path);
583		goto bailout1;
584	}
585	if ((buf = malloc(img_size)) == NULL) {
586		warnx("Could not allocate buffer to read image file %s",
587		    fw_img_path);
588		goto bailout1;
589	}
590	/* Skip headers if applicable. */
591	switch (vp->type) {
592	case VENDOR_SEAGATE:
593		if (read(fd, buf, 16) != 16) {
594			warn("Could not read image file %s", fw_img_path);
595			goto bailout;
596		}
597		if (lseek(fd, 0, SEEK_SET) == -1) {
598			warn("Unable to lseek");
599			goto bailout;
600		}
601		if ((strncmp(buf, "SEAGATE,SEAGATE ", 16) == 0) ||
602		    (img_size % 512 == 80))
603			skip_bytes = 80;
604		break;
605	case VENDOR_QUALSTAR:
606		skip_bytes = img_size % 1030;
607		break;
608	case VENDOR_IBM: {
609		if (vp->dev_type != T_SEQUENTIAL)
610			break;
611		if (fw_validate_ibm(dev, retry_count, timeout, fd, buf,
612				    fw_img_path, quiet) != 0)
613			goto bailout;
614		break;
615	}
616	default:
617		break;
618	}
619	if (skip_bytes != 0) {
620		fprintf(stdout, "Skipping %d byte header.\n", skip_bytes);
621		if (lseek(fd, skip_bytes, SEEK_SET) == -1) {
622			warn("Could not lseek");
623			goto bailout;
624		}
625		img_size -= skip_bytes;
626	}
627	/* Read image into a buffer. */
628	if (read(fd, buf, img_size) != img_size) {
629		warn("Could not read image file %s", fw_img_path);
630		goto bailout;
631	}
632	*num_bytes = img_size;
633	close(fd);
634	return (buf);
635bailout:
636	free(buf);
637bailout1:
638	close(fd);
639	*num_bytes = 0;
640	return (NULL);
641}
642
643/*
644 * Returns 0 for "success", where success means that the device has met the
645 * requirement in the vendor structure for being ready or not ready when
646 * firmware is downloaded.
647 *
648 * Returns 1 for a failure to be ready to accept a firmware download.
649 * (e.g., a drive needs to be ready, but returns not ready)
650 *
651 * Returns -1 for any other failure.
652 */
653static int
654fw_check_device_ready(struct cam_device *dev, camcontrol_devtype devtype,
655		      struct fw_vendor *vp, int printerrors, int timeout)
656{
657	union ccb *ccb;
658	int retval = 0;
659	int16_t *ptr = NULL;
660	size_t dxfer_len = 0;
661
662	if ((ccb = cam_getccb(dev)) == NULL) {
663		warnx("Could not allocate CCB");
664		retval = -1;
665		goto bailout;
666	}
667
668	CCB_CLEAR_ALL_EXCEPT_HDR(ccb);
669
670	if (devtype != CC_DT_SCSI) {
671		dxfer_len = sizeof(struct ata_params);
672
673		ptr = (uint16_t *)malloc(dxfer_len);
674		if (ptr == NULL) {
675			warnx("can't malloc memory for identify");
676			retval = -1;
677			goto bailout;
678		}
679		bzero(ptr, dxfer_len);
680	}
681
682	switch (devtype) {
683	case CC_DT_SCSI:
684		scsi_test_unit_ready(&ccb->csio,
685				     /*retries*/ 0,
686				     /*cbfcnp*/ NULL,
687				     /*tag_action*/ MSG_SIMPLE_Q_TAG,
688		    		     /*sense_len*/ SSD_FULL_SIZE,
689				     /*timeout*/ 5000);
690		break;
691	case CC_DT_ATA_BEHIND_SCSI:
692	case CC_DT_ATA: {
693		build_ata_cmd(ccb,
694			     /*retries*/ 1,
695			     /*flags*/ CAM_DIR_IN,
696			     /*tag_action*/ MSG_SIMPLE_Q_TAG,
697			     /*protocol*/ AP_PROTO_PIO_IN,
698			     /*ata_flags*/ AP_FLAG_BYT_BLOK_BYTES |
699					   AP_FLAG_TLEN_SECT_CNT |
700					   AP_FLAG_TDIR_FROM_DEV,
701			     /*features*/ 0,
702			     /*sector_count*/ (uint8_t) dxfer_len,
703			     /*lba*/ 0,
704			     /*command*/ ATA_ATA_IDENTIFY,
705			     /*data_ptr*/ (uint8_t *)ptr,
706			     /*dxfer_len*/ dxfer_len,
707			     /*sense_len*/ SSD_FULL_SIZE,
708			     /*timeout*/ timeout ? timeout : 30 * 1000,
709			     /*is48bit*/ 0,
710			     /*devtype*/ devtype);
711		break;
712	}
713	default:
714		warnx("Unknown disk type %d", devtype);
715		retval = -1;
716		goto bailout;
717		break; /*NOTREACHED*/
718	}
719
720	ccb->ccb_h.flags |= CAM_DEV_QFRZDIS;
721
722	retval = cam_send_ccb(dev, ccb);
723	if (retval != 0) {
724		warn("error sending %s CCB", (devtype == CC_DT_SCSI) ?
725		     "Test Unit Ready" : "Identify");
726		retval = -1;
727		goto bailout;
728	}
729
730	if (((ccb->ccb_h.status & CAM_STATUS_MASK) != CAM_REQ_CMP)
731	 && (vp->tur_status == FW_TUR_READY)) {
732		warnx("Device is not ready");
733		if (printerrors)
734			cam_error_print(dev, ccb, CAM_ESF_ALL,
735			    CAM_EPF_ALL, stderr);
736		retval = 1;
737		goto bailout;
738	} else if (((ccb->ccb_h.status & CAM_STATUS_MASK) == CAM_REQ_CMP)
739		&& (vp->tur_status == FW_TUR_NOT_READY)) {
740		warnx("Device cannot have media loaded when firmware is "
741		    "downloaded");
742		retval = 1;
743		goto bailout;
744	}
745bailout:
746	if (ccb != NULL)
747		cam_freeccb(ccb);
748
749	return (retval);
750}
751
752/*
753 * Download firmware stored in buf to cam_dev. If simulation mode
754 * is enabled, only show what packet sizes would be sent to the
755 * device but do not sent any actual packets
756 */
757static int
758fw_download_img(struct cam_device *cam_dev, struct fw_vendor *vp,
759    char *buf, int img_size, int sim_mode, int printerrors, int quiet,
760    int retry_count, int timeout, const char *imgname,
761    camcontrol_devtype devtype)
762{
763	struct scsi_write_buffer cdb;
764	progress_t progress;
765	int size = 0;
766	union ccb *ccb = NULL;
767	int pkt_count = 0;
768	int max_pkt_size;
769	u_int32_t pkt_size = 0;
770	char *pkt_ptr = buf;
771	u_int32_t offset;
772	int last_pkt = 0;
773	int retval = 0;
774
775	/*
776	 * Check to see whether the device is ready to accept a firmware
777	 * download.
778	 */
779	retval = fw_check_device_ready(cam_dev, devtype, vp, printerrors,
780				       timeout);
781	if (retval != 0)
782		goto bailout;
783
784	if ((ccb = cam_getccb(cam_dev)) == NULL) {
785		warnx("Could not allocate CCB");
786		retval = 1;
787		goto bailout;
788	}
789
790	CCB_CLEAR_ALL_EXCEPT_HDR(ccb);
791
792	max_pkt_size = vp->max_pkt_size;
793	if (max_pkt_size == 0)
794		max_pkt_size = UNKNOWN_MAX_PKT_SIZE;
795
796	pkt_size = max_pkt_size;
797	progress_init(&progress, imgname, size = img_size);
798	/* Download single fw packets. */
799	do {
800		if (img_size <= max_pkt_size) {
801			last_pkt = 1;
802			pkt_size = img_size;
803		}
804		progress_update(&progress, size - img_size);
805		if (((sim_mode == 0) && (quiet == 0))
806		 || ((sim_mode != 0) && (printerrors == 0)))
807			progress_draw(&progress);
808		bzero(&cdb, sizeof(cdb));
809		switch (devtype) {
810		case CC_DT_SCSI:
811			cdb.opcode  = WRITE_BUFFER;
812			cdb.control = 0;
813			/* Parameter list length. */
814			scsi_ulto3b(pkt_size, &cdb.length[0]);
815			offset = vp->inc_cdb_offset ? (pkt_ptr - buf) : 0;
816			scsi_ulto3b(offset, &cdb.offset[0]);
817			cdb.byte2 = last_pkt ? vp->cdb_byte2_last :
818					       vp->cdb_byte2;
819			cdb.buffer_id = vp->inc_cdb_buffer_id ? pkt_count : 0;
820			/* Zero out payload of ccb union after ccb header. */
821			CCB_CLEAR_ALL_EXCEPT_HDR(&ccb->csio);
822			/*
823			 * Copy previously constructed cdb into ccb_scsiio
824			 * struct.
825			 */
826			bcopy(&cdb, &ccb->csio.cdb_io.cdb_bytes[0],
827			    sizeof(struct scsi_write_buffer));
828			/* Fill rest of ccb_scsiio struct. */
829			cam_fill_csio(&ccb->csio,		/* ccb_scsiio*/
830			    retry_count,			/* retries*/
831			    NULL,				/* cbfcnp*/
832			    CAM_DIR_OUT | CAM_DEV_QFRZDIS,	/* flags*/
833			    CAM_TAG_ACTION_NONE,		/* tag_action*/
834			    (u_char *)pkt_ptr,			/* data_ptr*/
835			    pkt_size,				/* dxfer_len*/
836			    SSD_FULL_SIZE,			/* sense_len*/
837			    sizeof(struct scsi_write_buffer),	/* cdb_len*/
838			    timeout ? timeout : WB_TIMEOUT);	/* timeout*/
839			break;
840		case CC_DT_ATA:
841		case CC_DT_ATA_BEHIND_SCSI: {
842			uint32_t	off;
843
844			off = (uint32_t)(pkt_ptr - buf);
845
846			build_ata_cmd(ccb,
847			    /*retry_count*/ retry_count,
848			    /*flags*/ CAM_DIR_OUT | CAM_DEV_QFRZDIS,
849			    /*tag_action*/ CAM_TAG_ACTION_NONE,
850			    /*protocol*/ AP_PROTO_PIO_OUT,
851			    /*ata_flags*/ AP_FLAG_BYT_BLOK_BYTES |
852					  AP_FLAG_TLEN_SECT_CNT |
853					  AP_FLAG_TDIR_TO_DEV,
854			    /*features*/ USE_OFFSETS_FEATURE,
855			    /*sector_count*/ ATA_MAKE_SECTORS(pkt_size),
856			    /*lba*/ ATA_MAKE_LBA(off, pkt_size),
857			    /*command*/ ATA_DOWNLOAD_MICROCODE,
858			    /*data_ptr*/ (uint8_t *)pkt_ptr,
859			    /*dxfer_len*/ pkt_size,
860			    /*sense_len*/ SSD_FULL_SIZE,
861			    /*timeout*/ timeout ? timeout : WB_TIMEOUT,
862			    /*is48bit*/ 0,
863			    /*devtype*/ devtype);
864			break;
865		}
866		default:
867			warnx("Unknown device type %d", devtype);
868			retval = 1;
869			goto bailout;
870			break; /*NOTREACHED*/
871		}
872		if (!sim_mode) {
873			/* Execute the command. */
874			if (cam_send_ccb(cam_dev, ccb) < 0 ||
875			    (ccb->ccb_h.status & CAM_STATUS_MASK) !=
876			    CAM_REQ_CMP) {
877				warnx("Error writing image to device");
878				if (printerrors)
879					cam_error_print(cam_dev, ccb,
880					    CAM_ESF_ALL, CAM_EPF_ALL, stderr);
881				retval = 1;
882				goto bailout;
883			}
884		} else if (printerrors) {
885			cam_error_print(cam_dev, ccb, CAM_ESF_COMMAND, 0,
886			    stdout);
887		}
888
889		/* Prepare next round. */
890		pkt_count++;
891		pkt_ptr += pkt_size;
892		img_size -= pkt_size;
893	} while(!last_pkt);
894bailout:
895	if (quiet == 0)
896		progress_complete(&progress, size - img_size);
897	if (ccb != NULL)
898		cam_freeccb(ccb);
899	return (retval);
900}
901
902int
903fwdownload(struct cam_device *device, int argc, char **argv,
904    char *combinedopt, int printerrors, int retry_count, int timeout)
905{
906	struct fw_vendor *vp;
907	char *fw_img_path = NULL;
908	struct ata_params *ident_buf = NULL;
909	camcontrol_devtype devtype;
910	char *buf = NULL;
911	int img_size;
912	int c;
913	int sim_mode = 0;
914	int confirmed = 0;
915	int quiet = 0;
916	int retval = 0;
917
918	while ((c = getopt(argc, argv, combinedopt)) != -1) {
919		switch (c) {
920		case 'f':
921			fw_img_path = optarg;
922			break;
923		case 'q':
924			quiet = 1;
925			break;
926		case 's':
927			sim_mode = 1;
928			break;
929		case 'y':
930			confirmed = 1;
931			break;
932		default:
933			break;
934		}
935	}
936
937	if (fw_img_path == NULL)
938		errx(1, "you must specify a firmware image file using -f "
939		     "option");
940
941	retval = get_device_type(device, retry_count, timeout, printerrors,
942				 &devtype);
943	if (retval != 0)
944		errx(1, "Unable to determine device type");
945
946	if ((devtype == CC_DT_ATA)
947	 || (devtype == CC_DT_ATA_BEHIND_SCSI)) {
948		union ccb *ccb;
949
950		ccb = cam_getccb(device);
951		if (ccb == NULL) {
952			warnx("couldn't allocate CCB");
953			retval = 1;
954			goto bailout;
955		}
956
957		if (ata_do_identify(device, retry_count, timeout, ccb,
958		    		    &ident_buf) != 0) {
959			cam_freeccb(ccb);
960			retval = 1;
961			goto bailout;
962		}
963	} else if (devtype != CC_DT_SCSI)
964		errx(1, "Unsupported device type %d", devtype);
965
966	vp = fw_get_vendor(device, ident_buf);
967	/*
968	 * Bail out if we have an unknown vendor and this isn't an ATA
969	 * disk.  For a SCSI disk, we have no chance of working properly
970	 * with the default values in the VENDOR_UNKNOWN case.  For an ATA
971	 * disk connected via an ATA transport, we may work for drives that
972	 * support the ATA_DOWNLOAD_MICROCODE command.
973	 */
974	if (((vp == NULL)
975	  || (vp->type == VENDOR_UNKNOWN))
976	 && (devtype == CC_DT_SCSI))
977		errx(1, "Unsupported device");
978
979	retval = fw_get_timeout(device, vp, retry_count, timeout);
980	if (retval != 0) {
981		warnx("Unable to get a firmware download timeout value");
982		goto bailout;
983	}
984
985	buf = fw_read_img(device, retry_count, timeout, quiet, fw_img_path,
986	    vp, &img_size);
987	if (buf == NULL) {
988		retval = 1;
989		goto bailout;
990	}
991
992	if (!confirmed) {
993		fprintf(stdout, "You are about to download firmware image (%s)"
994		    " into the following device:\n",
995		    fw_img_path);
996		if (devtype == CC_DT_SCSI) {
997			if (scsidoinquiry(device, argc, argv, combinedopt, 0,
998					  5000) != 0) {
999				warnx("Error sending inquiry");
1000				retval = 1;
1001				goto bailout;
1002			}
1003		} else {
1004			printf("%s%d: ", device->device_name,
1005			    device->dev_unit_num);
1006			ata_print_ident(ident_buf);
1007			camxferrate(device);
1008			free(ident_buf);
1009		}
1010		fprintf(stdout, "Using a timeout of %u ms, which is %s.\n",
1011			vp->timeout_ms,
1012			fw_timeout_desc_table[vp->timeout_type].timeout_desc);
1013		fprintf(stdout, "\nIt may damage your drive. ");
1014		if (!get_confirmation()) {
1015			retval = 1;
1016			goto bailout;
1017		}
1018	}
1019	if ((sim_mode != 0) && (quiet == 0))
1020		fprintf(stdout, "Running in simulation mode\n");
1021
1022	if (fw_download_img(device, vp, buf, img_size, sim_mode, printerrors,
1023	    quiet, retry_count, vp->timeout_ms, fw_img_path, devtype) != 0) {
1024		fprintf(stderr, "Firmware download failed\n");
1025		retval = 1;
1026		goto bailout;
1027	} else if (quiet == 0)
1028		fprintf(stdout, "Firmware download successful\n");
1029
1030bailout:
1031	free(buf);
1032	return (retval);
1033}
1034
1035