xref: /freebsd/sbin/camcontrol/modeedit.c (revision 069ac184)
1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause
3  *
4  * Copyright (c) 2000 Kelly Yancey <kbyanc@posi.net>
5  * Derived from work done by Julian Elischer <julian@tfs.com,
6  * julian@dialix.oz.au>, 1993, and Peter Dufault <dufault@hda.com>, 1994.
7  * All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer,
14  *    without modification, immediately at the beginning of the file.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
20  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
21  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
22  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
23  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
24  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
28  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30 
31 #include <sys/cdefs.h>
32 #include <sys/queue.h>
33 #include <sys/types.h>
34 #include <sys/sbuf.h>
35 
36 #include <assert.h>
37 #include <ctype.h>
38 #include <err.h>
39 #include <errno.h>
40 #include <stdlib.h>
41 #include <string.h>
42 #include <stdio.h>
43 #include <sysexits.h>
44 #include <unistd.h>
45 
46 #include <cam/scsi/scsi_all.h>
47 #include <cam/cam.h>
48 #include <cam/cam_ccb.h>
49 #include <camlib.h>
50 #include "camcontrol.h"
51 
52 #define	DEFAULT_SCSI_MODE_DB	"/usr/share/misc/scsi_modes"
53 #define	DEFAULT_EDITOR		"vi"
54 #define	MAX_FORMAT_SPEC		4096	/* Max CDB format specifier. */
55 #define	MAX_PAGENUM_LEN		10	/* Max characters in page num. */
56 #define	MAX_PAGENAME_LEN	64	/* Max characters in page name. */
57 #define	PAGEDEF_START		'{'	/* Page definition delimiter. */
58 #define	PAGEDEF_END		'}'	/* Page definition delimiter. */
59 #define	PAGENAME_START		'"'	/* Page name delimiter. */
60 #define	PAGENAME_END		'"'	/* Page name delimiter. */
61 #define	PAGEENTRY_END		';'	/* Page entry terminator (optional). */
62 #define	MAX_DATA_SIZE		4096	/* Mode/Log sense data buffer size. */
63 #define PAGE_CTRL_SHIFT		6	/* Bit offset to page control field. */
64 
65 struct editentry {
66 	STAILQ_ENTRY(editentry) link;
67 	char	*name;
68 	char	type;
69 	int	editable;
70 	int	size;
71 	union {
72 		int	ivalue;
73 		char	*svalue;
74 	} value;
75 };
76 static STAILQ_HEAD(, editentry) editlist; /* List of page entries. */
77 static int editlist_changed = 0;	/* Whether any entries were changed. */
78 
79 struct pagename {
80 	SLIST_ENTRY(pagename) link;
81 	int page;
82 	int subpage;
83 	char *name;
84 };
85 static SLIST_HEAD(, pagename) namelist;	/* Page number to name mappings. */
86 
87 static char format[MAX_FORMAT_SPEC];	/* Buffer for scsi cdb format def. */
88 
89 static FILE *edit_file = NULL;		/* File handle for edit file. */
90 static char edit_path[] = "/tmp/camXXXXXX";
91 
92 
93 /* Function prototypes. */
94 static void		 editentry_create(void *hook, int letter, void *arg,
95 					  int count, char *name);
96 static void		 editentry_update(void *hook, int letter, void *arg,
97 					  int count, char *name);
98 static void		 editentry_create_desc(void *hook, int letter, void *arg,
99 					  int count, char *name);
100 static int		 editentry_save(void *hook, char *name);
101 static struct editentry	*editentry_lookup(char *name);
102 static int		 editentry_set(char *name, char *newvalue,
103 				       int editonly);
104 static void		 editlist_populate(struct cam_device *device,
105 			    int cdb_len, int dbd, int pc, int page, int subpage,
106 			    int task_attr, int retries, int timeout);
107 static void		 editlist_populate_desc(struct cam_device *device,
108 			    int cdb_len, int llbaa, int pc, int page,
109 			    int subpage, int task_attr, int retries,
110 			    int timeout);
111 static void		 editlist_save(struct cam_device *device, int cdb_len,
112 			    int dbd, int pc, int page, int subpage,
113 			    int task_attr, int retries, int timeout);
114 static void		 editlist_save_desc(struct cam_device *device, int cdb_len,
115 			    int llbaa, int pc, int page, int subpage,
116 			    int task_attr, int retries, int timeout);
117 static void		 nameentry_create(int page, int subpage, char *name);
118 static struct pagename	*nameentry_lookup(int page, int subpage);
119 static int		 load_format(const char *pagedb_path, int lpage,
120 			    int lsubpage);
121 static int		 modepage_write(FILE *file, int editonly);
122 static int		 modepage_read(FILE *file);
123 static void		 modepage_edit(void);
124 static void		 modepage_dump(struct cam_device *device, int cdb_len,
125 			    int dbd, int pc, int page, int subpage,
126 			    int task_attr, int retries, int timeout);
127 static void		 modepage_dump_desc(struct cam_device *device,
128 			    int cdb_len, int llbaa, int pc, int page,
129 			    int subpage, int task_attr, int retries,
130 			    int timeout);
131 static void		 cleanup_editfile(void);
132 
133 
134 #define	returnerr(code) do {						\
135 	errno = code;							\
136 	return (-1);							\
137 } while (0)
138 
139 
140 #define	RTRIM(string) do {						\
141 	int _length;							\
142 	while (isspace(string[_length = strlen(string) - 1]))		\
143 		string[_length] = '\0';					\
144 } while (0)
145 
146 
147 static void
148 editentry_create(void *hook __unused, int letter, void *arg, int count,
149 		 char *name)
150 {
151 	struct editentry *newentry;	/* Buffer to hold new entry. */
152 
153 	/* Allocate memory for the new entry and a copy of the entry name. */
154 	if ((newentry = malloc(sizeof(struct editentry))) == NULL ||
155 	    (newentry->name = strdup(name)) == NULL)
156 		err(EX_OSERR, NULL);
157 
158 	/* Trim any trailing whitespace for the entry name. */
159 	RTRIM(newentry->name);
160 
161 	newentry->editable = (arg != NULL);
162 	newentry->type = letter;
163 	newentry->size = count;		/* Placeholder; not accurate. */
164 	newentry->value.svalue = NULL;
165 
166 	STAILQ_INSERT_TAIL(&editlist, newentry, link);
167 }
168 
169 static void
170 editentry_update(void *hook __unused, int letter, void *arg, int count,
171 		 char *name)
172 {
173 	struct editentry *dest;		/* Buffer to hold entry to update. */
174 
175 	dest = editentry_lookup(name);
176 	assert(dest != NULL);
177 
178 	dest->type = letter;
179 	dest->size = count;		/* We get the real size now. */
180 
181 	switch (dest->type) {
182 	case 'i':			/* Byte-sized integral type. */
183 	case 'b':			/* Bit-sized integral types. */
184 	case 't':
185 		dest->value.ivalue = (intptr_t)arg;
186 		break;
187 
188 	case 'c':			/* Character array. */
189 	case 'z':			/* Null-padded string. */
190 		editentry_set(name, (char *)arg, 0);
191 		break;
192 	default:
193 		; /* NOTREACHED */
194 	}
195 }
196 
197 static void
198 editentry_create_desc(void *hook __unused, int letter, void *arg, int count,
199 		 char *name)
200 {
201 	struct editentry *newentry;	/* Buffer to hold new entry. */
202 
203 	/* Allocate memory for the new entry and a copy of the entry name. */
204 	if ((newentry = malloc(sizeof(struct editentry))) == NULL ||
205 	    (newentry->name = strdup(name)) == NULL)
206 		err(EX_OSERR, NULL);
207 
208 	/* Trim any trailing whitespace for the entry name. */
209 	RTRIM(newentry->name);
210 
211 	newentry->editable = 1;
212 	newentry->type = letter;
213 	newentry->size = count;
214 	newentry->value.svalue = NULL;
215 
216 	STAILQ_INSERT_TAIL(&editlist, newentry, link);
217 
218 	switch (letter) {
219 	case 'i':			/* Byte-sized integral type. */
220 	case 'b':			/* Bit-sized integral types. */
221 	case 't':
222 		newentry->value.ivalue = (intptr_t)arg;
223 		break;
224 
225 	case 'c':			/* Character array. */
226 	case 'z':			/* Null-padded string. */
227 		editentry_set(name, (char *)arg, 0);
228 		break;
229 	default:
230 		; /* NOTREACHED */
231 	}
232 }
233 
234 static int
235 editentry_save(void *hook __unused, char *name)
236 {
237 	struct editentry *src;		/* Entry value to save. */
238 
239 	src = editentry_lookup(name);
240 	if (src == 0) {
241 		/*
242 		 * This happens if field does not fit into read page size.
243 		 * It also means that this field won't be written, so the
244 		 * returned value does not really matter.
245 		 */
246 		return (0);
247 	}
248 
249 	switch (src->type) {
250 	case 'i':			/* Byte-sized integral type. */
251 	case 'b':			/* Bit-sized integral types. */
252 	case 't':
253 		return (src->value.ivalue);
254 		/* NOTREACHED */
255 
256 	case 'c':			/* Character array. */
257 	case 'z':			/* Null-padded string. */
258 		return ((intptr_t)src->value.svalue);
259 		/* NOTREACHED */
260 
261 	default:
262 		; /* NOTREACHED */
263 	}
264 
265 	return (0);			/* This should never happen. */
266 }
267 
268 static struct editentry *
269 editentry_lookup(char *name)
270 {
271 	struct editentry *scan;
272 
273 	assert(name != NULL);
274 
275 	STAILQ_FOREACH(scan, &editlist, link) {
276 		if (strcasecmp(scan->name, name) == 0)
277 			return (scan);
278 	}
279 
280 	/* Not found during list traversal. */
281 	return (NULL);
282 }
283 
284 static int
285 editentry_set(char *name, char *newvalue, int editonly)
286 {
287 	struct editentry *dest;	/* Modepage entry to update. */
288 	char *cval;		/* Pointer to new string value. */
289 	char *convertend;	/* End-of-conversion pointer. */
290 	long long ival, newival; /* New integral value. */
291 	int resolution;		/* Resolution in bits for integer conversion. */
292 
293 /*
294  * Macro to determine the maximum value of the given size for the current
295  * resolution.
296  */
297 #define	RESOLUTION_MAX(size)	((1LL << (resolution * (size))) - 1)
298 
299 	assert(newvalue != NULL);
300 	if (*newvalue == '\0')
301 		return (0);	/* Nothing to do. */
302 
303 	if ((dest = editentry_lookup(name)) == NULL)
304 		returnerr(ENOENT);
305 	if (!dest->editable && editonly)
306 		returnerr(EPERM);
307 
308 	switch (dest->type) {
309 	case 'i':		/* Byte-sized integral type. */
310 	case 'b':		/* Bit-sized integral types. */
311 	case 't':
312 		/* Convert the value string to an integer. */
313 		resolution = (dest->type == 'i')? 8: 1;
314 		ival = strtoll(newvalue, &convertend, 0);
315 		if (*convertend != '\0')
316 			returnerr(EINVAL);
317 		if (ival > RESOLUTION_MAX(dest->size) || ival < 0) {
318 			newival = (ival < 0) ? 0 : RESOLUTION_MAX(dest->size);
319 			warnx("value %lld is out of range for entry %s; "
320 			    "clipping to %lld", ival, name, newival);
321 			ival = newival;
322 		}
323 		if (dest->value.ivalue != ival)
324 			editlist_changed = 1;
325 		dest->value.ivalue = ival;
326 		break;
327 
328 	case 'c':		/* Character array. */
329 	case 'z':		/* Null-padded string. */
330 		if ((cval = calloc(1, dest->size + 1)) == NULL)
331 			err(EX_OSERR, NULL);
332 		strlcpy(cval, newvalue, dest->size + 1);
333 		if (dest->type == 'z') {
334 			/* Convert trailing spaces to nulls. */
335 			char *convertend2;
336 
337 			for (convertend2 = cval + dest->size;
338 			    convertend2 >= cval; convertend2--) {
339 				if (*convertend2 == ' ')
340 					*convertend2 = '\0';
341 				else if (*convertend2 != '\0')
342 					break;
343 			}
344 		}
345 		if (strncmp(dest->value.svalue, cval, dest->size) == 0) {
346 			/* Nothing changed, free the newly allocated string. */
347 			free(cval);
348 			break;
349 		}
350 		if (dest->value.svalue != NULL) {
351 			/* Free the current string buffer. */
352 			free(dest->value.svalue);
353 			dest->value.svalue = NULL;
354 		}
355 		dest->value.svalue = cval;
356 		editlist_changed = 1;
357 		break;
358 
359 	default:
360 		; /* NOTREACHED */
361 	}
362 
363 	return (0);
364 #undef RESOLUTION_MAX
365 }
366 
367 static void
368 nameentry_create(int page, int subpage, char *name) {
369 	struct pagename *newentry;
370 
371 	if (page < 0 || subpage < 0 || name == NULL || name[0] == '\0')
372 		return;
373 
374 	/* Allocate memory for the new entry and a copy of the entry name. */
375 	if ((newentry = malloc(sizeof(struct pagename))) == NULL ||
376 	    (newentry->name = strdup(name)) == NULL)
377 		err(EX_OSERR, NULL);
378 
379 	/* Trim any trailing whitespace for the page name. */
380 	RTRIM(newentry->name);
381 
382 	newentry->page = page;
383 	newentry->subpage = subpage;
384 	SLIST_INSERT_HEAD(&namelist, newentry, link);
385 }
386 
387 static struct pagename *
388 nameentry_lookup(int page, int subpage) {
389 	struct pagename *scan;
390 
391 	SLIST_FOREACH(scan, &namelist, link) {
392 		if (page == scan->page && subpage == scan->subpage)
393 			return (scan);
394 	}
395 
396 	/* Not found during list traversal. */
397 	return (NULL);
398 }
399 
400 static int
401 load_format(const char *pagedb_path, int lpage, int lsubpage)
402 {
403 	FILE *pagedb;
404 	char str_page[MAX_PAGENUM_LEN];
405 	char *str_subpage;
406 	char str_pagename[MAX_PAGENAME_LEN];
407 	int page;
408 	int subpage;
409 	int depth;			/* Quoting depth. */
410 	int found;
411 	int lineno;
412 	enum { LOCATE, PAGENAME, PAGEDEF } state;
413 	int ch;
414 	char c;
415 
416 #define	SETSTATE_LOCATE do {						\
417 	str_page[0] = '\0';						\
418 	str_pagename[0] = '\0';						\
419 	page = -1;							\
420 	subpage = -1;							\
421 	state = LOCATE;							\
422 } while (0)
423 
424 #define	SETSTATE_PAGENAME do {						\
425 	str_pagename[0] = '\0';						\
426 	state = PAGENAME;						\
427 } while (0)
428 
429 #define	SETSTATE_PAGEDEF do {						\
430 	format[0] = '\0';						\
431 	state = PAGEDEF;						\
432 } while (0)
433 
434 #define	UPDATE_LINENO do {						\
435 	if (c == '\n')							\
436 		lineno++;						\
437 } while (0)
438 
439 #define	BUFFERFULL(buffer)	(strlen(buffer) + 1 >= sizeof(buffer))
440 
441 	if ((pagedb = fopen(pagedb_path, "r")) == NULL)
442 		returnerr(ENOENT);
443 
444 	SLIST_INIT(&namelist);
445 
446 	c = '\0';
447 	depth = 0;
448 	lineno = 0;
449 	found = 0;
450 	SETSTATE_LOCATE;
451 	while ((ch = fgetc(pagedb)) != EOF) {
452 
453 		/* Keep a line count to make error messages more useful. */
454 		UPDATE_LINENO;
455 
456 		/* Skip over comments anywhere in the mode database. */
457 		if (ch == '#') {
458 			do {
459 				ch = fgetc(pagedb);
460 			} while (ch != '\n' && ch != EOF);
461 			UPDATE_LINENO;
462 			continue;
463 		}
464 		c = ch;
465 
466 		/* Strip out newline characters. */
467 		if (c == '\n')
468 			continue;
469 
470 		/* Keep track of the nesting depth for braces. */
471 		if (c == PAGEDEF_START)
472 			depth++;
473 		else if (c == PAGEDEF_END) {
474 			depth--;
475 			if (depth < 0) {
476 				errx(EX_OSFILE, "%s:%d: %s", pagedb_path,
477 				    lineno, "mismatched bracket");
478 			}
479 		}
480 
481 		switch (state) {
482 		case LOCATE:
483 			/*
484 			 * Locate the page the user is interested in, skipping
485 			 * all others.
486 			 */
487 			if (isspace(c)) {
488 				/* Ignore all whitespace between pages. */
489 				break;
490 			} else if (depth == 0 && c == PAGEENTRY_END) {
491 				/*
492 				 * A page entry terminator will reset page
493 				 * scanning (useful for assigning names to
494 				 * modes without providing a mode definition).
495 				 */
496 				/* Record the name of this page. */
497 				str_subpage = str_page;
498 				strsep(&str_subpage, ",");
499 				page = strtol(str_page, NULL, 0);
500 				if (str_subpage)
501 				    subpage = strtol(str_subpage, NULL, 0);
502 				else
503 				    subpage = 0;
504 				nameentry_create(page, subpage, str_pagename);
505 				SETSTATE_LOCATE;
506 			} else if (depth == 0 && c == PAGENAME_START) {
507 				SETSTATE_PAGENAME;
508 			} else if (c == PAGEDEF_START) {
509 				str_subpage = str_page;
510 				strsep(&str_subpage, ",");
511 				page = strtol(str_page, NULL, 0);
512 				if (str_subpage)
513 				    subpage = strtol(str_subpage, NULL, 0);
514 				else
515 				    subpage = 0;
516 				if (depth == 1) {
517 					/* Record the name of this page. */
518 					nameentry_create(page, subpage,
519 					    str_pagename);
520 					/*
521 					 * Only record the format if this is
522 					 * the page we are interested in.
523 					 */
524 					if (lpage == page &&
525 					    lsubpage == subpage && !found)
526 						SETSTATE_PAGEDEF;
527 				}
528 			} else if (c == PAGEDEF_END) {
529 				/* Reset the processor state. */
530 				SETSTATE_LOCATE;
531 			} else if (depth == 0 && ! BUFFERFULL(str_page)) {
532 				strncat(str_page, &c, 1);
533 			} else if (depth == 0) {
534 				errx(EX_OSFILE, "%s:%d: %s %zd %s", pagedb_path,
535 				    lineno, "page identifier exceeds",
536 				    sizeof(str_page) - 1, "characters");
537 			}
538 			break;
539 
540 		case PAGENAME:
541 			if (c == PAGENAME_END) {
542 				/*
543 				 * Return to LOCATE state without resetting the
544 				 * page number buffer.
545 				 */
546 				state = LOCATE;
547 			} else if (! BUFFERFULL(str_pagename)) {
548 				strncat(str_pagename, &c, 1);
549 			} else {
550 				errx(EX_OSFILE, "%s:%d: %s %zd %s", pagedb_path,
551 				    lineno, "page name exceeds",
552 				    sizeof(str_page) - 1, "characters");
553 			}
554 			break;
555 
556 		case PAGEDEF:
557 			/*
558 			 * Transfer the page definition into a format buffer
559 			 * suitable for use with CDB encoding/decoding routines.
560 			 */
561 			if (depth == 0) {
562 				found = 1;
563 				SETSTATE_LOCATE;
564 			} else if (! BUFFERFULL(format)) {
565 				strncat(format, &c, 1);
566 			} else {
567 				errx(EX_OSFILE, "%s:%d: %s %zd %s", pagedb_path,
568 				    lineno, "page definition exceeds",
569 				    sizeof(format) - 1, "characters");
570 			}
571 			break;
572 
573 		default:
574 			; /* NOTREACHED */
575 		}
576 
577 		/* Repeat processing loop with next character. */
578 	}
579 
580 	if (ferror(pagedb))
581 		err(EX_OSFILE, "%s", pagedb_path);
582 
583 	/* Close the SCSI page database. */
584 	fclose(pagedb);
585 
586 	if (!found)			/* Never found a matching page. */
587 		returnerr(ESRCH);
588 
589 	return (0);
590 }
591 
592 static void
593 editlist_populate(struct cam_device *device, int cdb_len, int dbd, int pc,
594     int page, int subpage, int task_attr, int retries, int timeout)
595 {
596 	uint8_t data[MAX_DATA_SIZE];	/* Buffer to hold mode parameters. */
597 	uint8_t *mode_pars;		/* Pointer to modepage params. */
598 	struct scsi_mode_page_header *mph;
599 	struct scsi_mode_page_header_sp *mphsp;
600 	size_t len;
601 
602 	STAILQ_INIT(&editlist);
603 
604 	/* Fetch changeable values; use to build initial editlist. */
605 	mode_sense(device, &cdb_len, dbd, 0, 1, page, subpage, task_attr,
606 		   retries, timeout, data, sizeof(data));
607 
608 	if (cdb_len == 6) {
609 		struct scsi_mode_header_6 *mh =
610 		    (struct scsi_mode_header_6 *)data;
611 		mph = find_mode_page_6(mh);
612 	} else {
613 		struct scsi_mode_header_10 *mh =
614 		    (struct scsi_mode_header_10 *)data;
615 		mph = find_mode_page_10(mh);
616 	}
617 	if ((mph->page_code & SMPH_SPF) == 0) {
618 		mode_pars = (uint8_t *)(mph + 1);
619 		len = mph->page_length;
620 	} else {
621 		mphsp = (struct scsi_mode_page_header_sp *)mph;
622 		mode_pars = (uint8_t *)(mphsp + 1);
623 		len = scsi_2btoul(mphsp->page_length);
624 	}
625 	len = MIN(len, sizeof(data) - (mode_pars - data));
626 
627 	/* Decode the value data, creating edit_entries for each value. */
628 	buff_decode_visit(mode_pars, len, format, editentry_create, 0);
629 
630 	/* Fetch the current/saved values; use to set editentry values. */
631 	mode_sense(device, &cdb_len, dbd, 0, pc, page, subpage, task_attr,
632 	    retries, timeout, data, sizeof(data));
633 	buff_decode_visit(mode_pars, len, format, editentry_update, 0);
634 }
635 
636 static void
637 editlist_populate_desc(struct cam_device *device, int cdb_len, int llbaa, int pc,
638     int page, int subpage, int task_attr, int retries, int timeout)
639 {
640 	uint8_t data[MAX_DATA_SIZE];	/* Buffer to hold mode parameters. */
641 	uint8_t *desc;			/* Pointer to block descriptor. */
642 	char num[8];
643 	struct sbuf sb;
644 	size_t len;
645 	u_int longlba, dlen, i;
646 
647 	STAILQ_INIT(&editlist);
648 
649 	/* Fetch the current/saved values. */
650 	mode_sense(device, &cdb_len, 0, llbaa, pc, page, subpage, task_attr,
651 	    retries, timeout, data, sizeof(data));
652 
653 	if (cdb_len == 6) {
654 		struct scsi_mode_header_6 *mh =
655 		    (struct scsi_mode_header_6 *)data;
656 		desc = (uint8_t *)(mh + 1);
657 		len = mh->blk_desc_len;
658 		longlba = 0;
659 	} else {
660 		struct scsi_mode_header_10 *mh =
661 		    (struct scsi_mode_header_10 *)data;
662 		desc = (uint8_t *)(mh + 1);
663 		len = scsi_2btoul(mh->blk_desc_len);
664 		longlba = (mh->flags & SMH_LONGLBA) != 0;
665 	}
666 	dlen = longlba ? 16 : 8;
667 	len = MIN(len, sizeof(data) - (desc - data));
668 
669 	sbuf_new(&sb, format, sizeof(format), SBUF_FIXEDLEN);
670 	num[0] = 0;
671 	for (i = 0; i * dlen < len; i++) {
672 		if (i > 0)
673 			snprintf(num, sizeof(num), " %d", i + 1);
674 		if (longlba) {
675 			sbuf_printf(&sb, "{Number of Logical Blocks%s High} i4\n", num);
676 			sbuf_printf(&sb, "{Number of Logical Blocks%s} i4\n", num);
677 			sbuf_cat(&sb, "{Reserved} *i4\n");
678 			sbuf_printf(&sb, "{Logical Block Length%s} i4\n", num);
679 		} else if (device->pd_type == T_DIRECT) {
680 			sbuf_printf(&sb, "{Number of Logical Blocks%s} i4\n", num);
681 			sbuf_cat(&sb, "{Reserved} *i1\n");
682 			sbuf_printf(&sb, "{Logical Block Length%s} i3\n", num);
683 		} else {
684 			sbuf_printf(&sb, "{Density Code%s} i1\n", num);
685 			sbuf_printf(&sb, "{Number of Logical Blocks%s} i3\n", num);
686 			sbuf_cat(&sb, "{Reserved} *i1\n");
687 			sbuf_printf(&sb, "{Logical Block Length%s} i3\n", num);
688 		}
689 	}
690 	sbuf_finish(&sb);
691 	sbuf_delete(&sb);
692 
693 	/* Decode the value data, creating edit_entries for each value. */
694 	buff_decode_visit(desc, len, format, editentry_create_desc, 0);
695 }
696 
697 static void
698 editlist_save(struct cam_device *device, int cdb_len, int dbd, int pc,
699     int page, int subpage, int task_attr, int retries, int timeout)
700 {
701 	uint8_t data[MAX_DATA_SIZE];	/* Buffer to hold mode parameters. */
702 	uint8_t *mode_pars;		/* Pointer to modepage params. */
703 	struct scsi_mode_page_header *mph;
704 	struct scsi_mode_page_header_sp *mphsp;
705 	size_t len, hlen, mphlen;
706 
707 	/* Make sure that something changed before continuing. */
708 	if (! editlist_changed)
709 		return;
710 
711 	/* Preload the CDB buffer with the current mode page data. */
712 	mode_sense(device, &cdb_len, dbd, 0, pc, page, subpage, task_attr,
713 	    retries, timeout, data, sizeof(data));
714 
715 	/* Initial headers & offsets. */
716 	/*
717 	 * Tape drives include write protect (WP), Buffered Mode and Speed
718 	 * settings in the device-specific parameter.  Clearing this
719 	 * parameter on a mode select can have the effect of turning off
720 	 * write protect or buffered mode, or changing the speed setting of
721 	 * the tape drive.
722 	 *
723 	 * Disks report DPO/FUA support via the device specific parameter
724 	 * for MODE SENSE, but the bit is reserved for MODE SELECT.  So we
725 	 * clear this for disks (and other non-tape devices) to avoid
726 	 * potential errors from the target device.
727 	 */
728 	if (cdb_len == 6) {
729 		struct scsi_mode_header_6 *mh =
730 		    (struct scsi_mode_header_6 *)data;
731 		hlen = sizeof(*mh);
732 		/* Eliminate block descriptors. */
733 		if (mh->blk_desc_len > 0) {
734 			bcopy(find_mode_page_6(mh), mh + 1,
735 			    mh->data_length + 1 - hlen -
736 			    mh->blk_desc_len);
737 			mh->blk_desc_len = 0;
738 		}
739 		mh->data_length = 0;	/* Reserved for MODE SELECT command. */
740 		if (device->pd_type != T_SEQUENTIAL)
741 			mh->dev_spec = 0;	/* See comment above */
742 		mph = find_mode_page_6(mh);
743 	} else {
744 		struct scsi_mode_header_10 *mh =
745 		    (struct scsi_mode_header_10 *)data;
746 		hlen = sizeof(*mh);
747 		/* Eliminate block descriptors. */
748 		if (scsi_2btoul(mh->blk_desc_len) > 0) {
749 			bcopy(find_mode_page_10(mh), mh + 1,
750 			    scsi_2btoul(mh->data_length) + 1 - hlen -
751 			    scsi_2btoul(mh->blk_desc_len));
752 			scsi_ulto2b(0, mh->blk_desc_len);
753 		}
754 		scsi_ulto2b(0, mh->data_length); /* Reserved for MODE SELECT. */
755 		if (device->pd_type != T_SEQUENTIAL)
756 			mh->dev_spec = 0;	/* See comment above */
757 		mph = find_mode_page_10(mh);
758 	}
759 	if ((mph->page_code & SMPH_SPF) == 0) {
760 		mphlen = sizeof(*mph);
761 		mode_pars = (uint8_t *)(mph + 1);
762 		len = mph->page_length;
763 	} else {
764 		mphsp = (struct scsi_mode_page_header_sp *)mph;
765 		mphlen = sizeof(*mphsp);
766 		mode_pars = (uint8_t *)(mphsp + 1);
767 		len = scsi_2btoul(mphsp->page_length);
768 	}
769 	len = MIN(len, sizeof(data) - (mode_pars - data));
770 
771 	/* Encode the value data to be passed back to the device. */
772 	buff_encode_visit(mode_pars, len, format, editentry_save, 0);
773 
774 	mph->page_code &= ~SMPH_PS;	/* Reserved for MODE SELECT command. */
775 
776 	/*
777 	 * Write the changes back to the device. If the user editted control
778 	 * page 3 (saved values) then request the changes be permanently
779 	 * recorded.
780 	 */
781 	mode_select(device, cdb_len, (pc << PAGE_CTRL_SHIFT == SMS_PAGE_CTRL_SAVED),
782 	    task_attr, retries, timeout, data, hlen + mphlen + len);
783 }
784 
785 static void
786 editlist_save_desc(struct cam_device *device, int cdb_len, int llbaa, int pc,
787     int page, int subpage, int task_attr, int retries, int timeout)
788 {
789 	uint8_t data[MAX_DATA_SIZE];	/* Buffer to hold mode parameters. */
790 	uint8_t *desc;			/* Pointer to block descriptor. */
791 	size_t len, hlen;
792 
793 	/* Make sure that something changed before continuing. */
794 	if (! editlist_changed)
795 		return;
796 
797 	/* Preload the CDB buffer with the current mode page data. */
798 	mode_sense(device, &cdb_len, 0, llbaa, pc, page, subpage, task_attr,
799 	    retries, timeout, data, sizeof(data));
800 
801 	/* Initial headers & offsets. */
802 	if (cdb_len == 6) {
803 		struct scsi_mode_header_6 *mh =
804 		    (struct scsi_mode_header_6 *)data;
805 		hlen = sizeof(*mh);
806 		desc = (uint8_t *)(mh + 1);
807 		len = mh->blk_desc_len;
808 		mh->data_length = 0;	/* Reserved for MODE SELECT command. */
809 		if (device->pd_type != T_SEQUENTIAL)
810 			mh->dev_spec = 0;	/* See comment above */
811 	} else {
812 		struct scsi_mode_header_10 *mh =
813 		    (struct scsi_mode_header_10 *)data;
814 		hlen = sizeof(*mh);
815 		desc = (uint8_t *)(mh + 1);
816 		len = scsi_2btoul(mh->blk_desc_len);
817 		scsi_ulto2b(0, mh->data_length); /* Reserved for MODE SELECT. */
818 		if (device->pd_type != T_SEQUENTIAL)
819 			mh->dev_spec = 0;	/* See comment above */
820 	}
821 	len = MIN(len, sizeof(data) - (desc - data));
822 
823 	/* Encode the value data to be passed back to the device. */
824 	buff_encode_visit(desc, len, format, editentry_save, 0);
825 
826 	/*
827 	 * Write the changes back to the device. If the user editted control
828 	 * page 3 (saved values) then request the changes be permanently
829 	 * recorded.
830 	 */
831 	mode_select(device, cdb_len, (pc << PAGE_CTRL_SHIFT == SMS_PAGE_CTRL_SAVED),
832 	    task_attr, retries, timeout, data, hlen + len);
833 }
834 
835 static int
836 modepage_write(FILE *file, int editonly)
837 {
838 	struct editentry *scan;
839 	int written = 0;
840 
841 	STAILQ_FOREACH(scan, &editlist, link) {
842 		if (scan->editable || !editonly) {
843 			written++;
844 			if (scan->type == 'c' || scan->type == 'z') {
845 				fprintf(file, "%s:  %s\n", scan->name,
846 				    scan->value.svalue);
847 			} else {
848 				fprintf(file, "%s:  %u\n", scan->name,
849 				    scan->value.ivalue);
850 			}
851 		}
852 	}
853 	return (written);
854 }
855 
856 static int
857 modepage_read(FILE *file)
858 {
859 	char *buffer;			/* Pointer to dynamic line buffer.  */
860 	char *line;			/* Pointer to static fgetln buffer. */
861 	char *name;			/* Name portion of the line buffer. */
862 	char *value;			/* Value portion of line buffer.    */
863 	size_t length;			/* Length of static fgetln buffer.  */
864 
865 #define	ABORT_READ(message, param) do {					\
866 	warnx(message, param);						\
867 	free(buffer);							\
868 	returnerr(EAGAIN);						\
869 } while (0)
870 
871 	while ((line = fgetln(file, &length)) != NULL) {
872 		/* Trim trailing whitespace (including optional newline). */
873 		while (length > 0 && isspace(line[length - 1]))
874 			length--;
875 
876 	    	/* Allocate a buffer to hold the line + terminating null. */
877 	    	if ((buffer = malloc(length + 1)) == NULL)
878 			err(EX_OSERR, NULL);
879 		memcpy(buffer, line, length);
880 		buffer[length] = '\0';
881 
882 		/* Strip out comments. */
883 		if ((value = strchr(buffer, '#')) != NULL)
884 			*value = '\0';
885 
886 		/* The name is first in the buffer. Trim whitespace.*/
887 		name = buffer;
888 		RTRIM(name);
889 		while (isspace(*name))
890 			name++;
891 
892 		/* Skip empty lines. */
893 		if (strlen(name) == 0)
894 			continue;
895 
896 		/* The name ends at the colon; the value starts there. */
897 		if ((value = strrchr(buffer, ':')) == NULL)
898 			ABORT_READ("no value associated with %s", name);
899 		*value = '\0';			/* Null-terminate name. */
900 		value++;			/* Value starts afterwards. */
901 
902 		/* Trim leading and trailing whitespace. */
903 		RTRIM(value);
904 		while (isspace(*value))
905 			value++;
906 
907 		/* Make sure there is a value left. */
908 		if (strlen(value) == 0)
909 			ABORT_READ("no value associated with %s", name);
910 
911 		/* Update our in-memory copy of the modepage entry value. */
912 		if (editentry_set(name, value, 1) != 0) {
913 			if (errno == ENOENT) {
914 				/* No entry by the name. */
915 				ABORT_READ("no such modepage entry \"%s\"",
916 				    name);
917 			} else if (errno == EINVAL) {
918 				/* Invalid value. */
919 				ABORT_READ("Invalid value for entry \"%s\"",
920 				    name);
921 			} else if (errno == ERANGE) {
922 				/* Value out of range for entry type. */
923 				ABORT_READ("value out of range for %s", name);
924 			} else if (errno == EPERM) {
925 				/* Entry is not editable; not fatal. */
926 				warnx("modepage entry \"%s\" is read-only; "
927 				    "skipping.", name);
928 			}
929 		}
930 
931 		free(buffer);
932 	}
933 	return (ferror(file)? -1: 0);
934 
935 #undef ABORT_READ
936 }
937 
938 static void
939 modepage_edit(void)
940 {
941 	const char *editor;
942 	char *commandline;
943 	int fd;
944 	int written;
945 
946 	if (!isatty(fileno(stdin))) {
947 		/* Not a tty, read changes from stdin. */
948 		modepage_read(stdin);
949 		return;
950 	}
951 
952 	/* Lookup editor to invoke. */
953 	if ((editor = getenv("EDITOR")) == NULL)
954 		editor = DEFAULT_EDITOR;
955 
956 	/* Create temp file for editor to modify. */
957 	if ((fd = mkstemp(edit_path)) == -1)
958 		errx(EX_CANTCREAT, "mkstemp failed");
959 
960 	atexit(cleanup_editfile);
961 
962 	if ((edit_file = fdopen(fd, "w")) == NULL)
963 		err(EX_NOINPUT, "%s", edit_path);
964 
965 	written = modepage_write(edit_file, 1);
966 
967 	fclose(edit_file);
968 	edit_file = NULL;
969 
970 	if (written == 0) {
971 		warnx("no editable entries");
972 		cleanup_editfile();
973 		return;
974 	}
975 
976 	/*
977 	 * Allocate memory to hold the command line (the 2 extra characters
978 	 * are to hold the argument separator (a space), and the terminating
979 	 * null character.
980 	 */
981 	commandline = malloc(strlen(editor) + strlen(edit_path) + 2);
982 	if (commandline == NULL)
983 		err(EX_OSERR, NULL);
984 	sprintf(commandline, "%s %s", editor, edit_path);
985 
986 	/* Invoke the editor on the temp file. */
987 	if (system(commandline) == -1)
988 		err(EX_UNAVAILABLE, "could not invoke %s", editor);
989 	free(commandline);
990 
991 	if ((edit_file = fopen(edit_path, "r")) == NULL)
992 		err(EX_NOINPUT, "%s", edit_path);
993 
994 	/* Read any changes made to the temp file. */
995 	modepage_read(edit_file);
996 
997 	cleanup_editfile();
998 }
999 
1000 static void
1001 modepage_dump(struct cam_device *device, int cdb_len, int dbd, int pc,
1002 	      int page, int subpage, int task_attr, int retries, int timeout)
1003 {
1004 	uint8_t data[MAX_DATA_SIZE];	/* Buffer to hold mode parameters. */
1005 	uint8_t *mode_pars;		/* Pointer to modepage params. */
1006 	struct scsi_mode_page_header *mph;
1007 	struct scsi_mode_page_header_sp *mphsp;
1008 	size_t indx, len;
1009 
1010 	mode_sense(device, &cdb_len, dbd, 0, pc, page, subpage, task_attr,
1011 	    retries, timeout, data, sizeof(data));
1012 
1013 	if (cdb_len == 6) {
1014 		struct scsi_mode_header_6 *mh =
1015 		    (struct scsi_mode_header_6 *)data;
1016 		mph = find_mode_page_6(mh);
1017 	} else {
1018 		struct scsi_mode_header_10 *mh =
1019 		    (struct scsi_mode_header_10 *)data;
1020 		mph = find_mode_page_10(mh);
1021 	}
1022 	if ((mph->page_code & SMPH_SPF) == 0) {
1023 		mode_pars = (uint8_t *)(mph + 1);
1024 		len = mph->page_length;
1025 	} else {
1026 		mphsp = (struct scsi_mode_page_header_sp *)mph;
1027 		mode_pars = (uint8_t *)(mphsp + 1);
1028 		len = scsi_2btoul(mphsp->page_length);
1029 	}
1030 	len = MIN(len, sizeof(data) - (mode_pars - data));
1031 
1032 	/* Print the raw mode page data with newlines each 8 bytes. */
1033 	for (indx = 0; indx < len; indx++) {
1034 		printf("%02x%c",mode_pars[indx],
1035 		    (((indx + 1) % 8) == 0) ? '\n' : ' ');
1036 	}
1037 	putchar('\n');
1038 }
1039 static void
1040 modepage_dump_desc(struct cam_device *device, int cdb_len, int llbaa, int pc,
1041 	      int page, int subpage, int task_attr, int retries, int timeout)
1042 {
1043 	uint8_t data[MAX_DATA_SIZE];	/* Buffer to hold mode parameters. */
1044 	uint8_t *desc;			/* Pointer to block descriptor. */
1045 	size_t indx, len;
1046 
1047 	mode_sense(device, &cdb_len, 0, llbaa, pc, page, subpage, task_attr,
1048 	    retries, timeout, data, sizeof(data));
1049 
1050 	if (cdb_len == 6) {
1051 		struct scsi_mode_header_6 *mh =
1052 		    (struct scsi_mode_header_6 *)data;
1053 		desc = (uint8_t *)(mh + 1);
1054 		len = mh->blk_desc_len;
1055 	} else {
1056 		struct scsi_mode_header_10 *mh =
1057 		    (struct scsi_mode_header_10 *)data;
1058 		desc = (uint8_t *)(mh + 1);
1059 		len = scsi_2btoul(mh->blk_desc_len);
1060 	}
1061 	len = MIN(len, sizeof(data) - (desc - data));
1062 
1063 	/* Print the raw mode page data with newlines each 8 bytes. */
1064 	for (indx = 0; indx < len; indx++) {
1065 		printf("%02x%c", desc[indx],
1066 		    (((indx + 1) % 8) == 0) ? '\n' : ' ');
1067 	}
1068 	putchar('\n');
1069 }
1070 
1071 static void
1072 cleanup_editfile(void)
1073 {
1074 	if (edit_file == NULL)
1075 		return;
1076 	if (fclose(edit_file) != 0 || unlink(edit_path) != 0)
1077 		warn("%s", edit_path);
1078 	edit_file = NULL;
1079 }
1080 
1081 void
1082 mode_edit(struct cam_device *device, int cdb_len, int desc, int dbd, int llbaa,
1083     int pc, int page, int subpage, int edit, int binary, int task_attr,
1084     int retry_count, int timeout)
1085 {
1086 	const char *pagedb_path;	/* Path to modepage database. */
1087 
1088 	if (binary) {
1089 		if (edit)
1090 			errx(EX_USAGE, "cannot edit in binary mode.");
1091 	} else if (desc) {
1092 		editlist_populate_desc(device, cdb_len, llbaa, pc, page,
1093 		    subpage, task_attr, retry_count, timeout);
1094 	} else {
1095 		if ((pagedb_path = getenv("SCSI_MODES")) == NULL)
1096 			pagedb_path = DEFAULT_SCSI_MODE_DB;
1097 
1098 		if (load_format(pagedb_path, page, subpage) != 0 &&
1099 		    (edit || verbose)) {
1100 			if (errno == ENOENT) {
1101 				/* Modepage database file not found. */
1102 				warn("cannot open modepage database \"%s\"",
1103 				    pagedb_path);
1104 			} else if (errno == ESRCH) {
1105 				/* Modepage entry not found in database. */
1106 				warnx("modepage 0x%02x,0x%02x not found in "
1107 				    "database \"%s\"", page, subpage,
1108 				    pagedb_path);
1109 			}
1110 			/* We can recover in display mode, otherwise we exit. */
1111 			if (!edit) {
1112 				warnx("reverting to binary display only");
1113 				binary = 1;
1114 			} else
1115 				exit(EX_OSFILE);
1116 		}
1117 
1118 		editlist_populate(device, cdb_len, dbd, pc, page, subpage,
1119 		    task_attr, retry_count, timeout);
1120 	}
1121 
1122 	if (edit) {
1123 		if (pc << PAGE_CTRL_SHIFT != SMS_PAGE_CTRL_CURRENT &&
1124 		    pc << PAGE_CTRL_SHIFT != SMS_PAGE_CTRL_SAVED)
1125 			errx(EX_USAGE, "it only makes sense to edit page 0 "
1126 			    "(current) or page 3 (saved values)");
1127 		modepage_edit();
1128 		if (desc) {
1129 			editlist_save_desc(device, cdb_len, llbaa, pc, page,
1130 			    subpage, task_attr, retry_count, timeout);
1131 		} else {
1132 			editlist_save(device, cdb_len, dbd, pc, page, subpage,
1133 			    task_attr, retry_count, timeout);
1134 		}
1135 	} else if (binary || STAILQ_EMPTY(&editlist)) {
1136 		/* Display without formatting information. */
1137 		if (desc) {
1138 			modepage_dump_desc(device, cdb_len, llbaa, pc, page,
1139 			    subpage, task_attr, retry_count, timeout);
1140 		} else {
1141 			modepage_dump(device, cdb_len, dbd, pc, page, subpage,
1142 			    task_attr, retry_count, timeout);
1143 		}
1144 	} else {
1145 		/* Display with format. */
1146 		modepage_write(stdout, 0);
1147 	}
1148 }
1149 
1150 void
1151 mode_list(struct cam_device *device, int cdb_len, int dbd, int pc, int subpages,
1152 	  int task_attr, int retry_count, int timeout)
1153 {
1154 	uint8_t data[MAX_DATA_SIZE];	/* Buffer to hold mode parameters. */
1155 	struct scsi_mode_page_header *mph;
1156 	struct scsi_mode_page_header_sp *mphsp;
1157 	struct pagename *nameentry;
1158 	const char *pagedb_path;
1159 	int len, off, page, subpage;
1160 
1161 	if ((pagedb_path = getenv("SCSI_MODES")) == NULL)
1162 		pagedb_path = DEFAULT_SCSI_MODE_DB;
1163 
1164 	if (load_format(pagedb_path, 0, 0) != 0 && verbose && errno == ENOENT) {
1165 		/* Modepage database file not found. */
1166 		warn("cannot open modepage database \"%s\"", pagedb_path);
1167 	}
1168 
1169 	/* Build the list of all mode pages by querying the "all pages" page. */
1170 	mode_sense(device, &cdb_len, dbd, 0, pc, SMS_ALL_PAGES_PAGE,
1171 	    subpages ? SMS_SUBPAGE_ALL : 0,
1172 	    task_attr, retry_count, timeout, data, sizeof(data));
1173 
1174 	/* Skip block descriptors. */
1175 	if (cdb_len == 6) {
1176 		struct scsi_mode_header_6 *mh =
1177 		    (struct scsi_mode_header_6 *)data;
1178 		len = mh->data_length;
1179 		off = sizeof(*mh) + mh->blk_desc_len;
1180 	} else {
1181 		struct scsi_mode_header_10 *mh =
1182 		    (struct scsi_mode_header_10 *)data;
1183 		len = scsi_2btoul(mh->data_length);
1184 		off = sizeof(*mh) + scsi_2btoul(mh->blk_desc_len);
1185 	}
1186 	/* Iterate through the pages in the reply. */
1187 	while (off < len) {
1188 		/* Locate the next mode page header. */
1189 		mph = (struct scsi_mode_page_header *)(data + off);
1190 
1191 		if ((mph->page_code & SMPH_SPF) == 0) {
1192 			page = mph->page_code & SMS_PAGE_CODE;
1193 			subpage = 0;
1194 			off += sizeof(*mph) + mph->page_length;
1195 		} else {
1196 			mphsp = (struct scsi_mode_page_header_sp *)mph;
1197 			page = mphsp->page_code & SMS_PAGE_CODE;
1198 			subpage = mphsp->subpage;
1199 			off += sizeof(*mphsp) + scsi_2btoul(mphsp->page_length);
1200 		}
1201 
1202 		nameentry = nameentry_lookup(page, subpage);
1203 		if (subpage == 0) {
1204 			printf("0x%02x\t%s\n", page,
1205 			    nameentry ? nameentry->name : "");
1206 		} else {
1207 			printf("0x%02x,0x%02x\t%s\n", page, subpage,
1208 			    nameentry ? nameentry->name : "");
1209 		}
1210 	}
1211 }
1212