svn commit: r225609 - head/tools/tools/zfsboottest

[Andriy Gapon]

Author: avg
Date: Fri Sep 16 08:24:31 2011
New Revision: 225609
URL: http://svn.freebsd.org/changeset/base/225609

Log:
  zfsboottest: some additional enhancements
  
  - redirect diagnostics printfs in the boot code to stderr
  - do not read trailing garbage from a trailing block of a file
  
  Also add my copyright to the file after making so many changes.
  
  Approved by:	re (kib)
  MFC after:	1 week

Modified:
  head/tools/tools/zfsboottest/zfsboottest.c

Modified: head/tools/tools/zfsboottest/zfsboottest.c
==============================================================================
--- head/tools/tools/zfsboottest/zfsboottest.c	Fri Sep 16 08:22:48 2011	(r225608)
+++ head/tools/tools/zfsboottest/zfsboottest.c	Fri Sep 16 08:24:31 2011	(r225609)
@@ -1,5 +1,6 @@
 /*-
  * Copyright (c) 2010 Doug Rabson
+ * Copyright (c) 2011 Andriy Gapon
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
@@ -24,9 +25,6 @@
  * SUCH DAMAGE.
  */
 /* $FreeBSD$ */
-/*
- * Compile with 'cc -I. -I../../cddl/boot/zfs zfstest.c -o zfstest'
- */
 
 #include <sys/param.h>
 #include <sys/queue.h>
@@ -48,7 +46,10 @@ pager_output(const char *line)
 	fprintf(stderr, "%s", line);
 }
 
+#define ZFS_TEST
+#define	printf(...)	 fprintf(stderr, __VA_ARGS__)
 #include "zfsimpl.c"
+#undef printf
 
 static int
 vdev_read(vdev_t *vdev, void *priv, off_t off, void *buf, size_t bytes)
@@ -86,7 +87,9 @@ main(int argc, char** argv)
 	struct stat sb;
 	dnode_phys_t dn;
 	spa_t *spa;
-	int i, n, off;
+	off_t off;
+	ssize_t n;
+	int i;
 
 	zfs_init();
 	if (argc == 1) {
@@ -131,7 +134,9 @@ main(int argc, char** argv)
 
 	off = 0;
 	do {
-		n = zfs_read(spa, &dn, buf, 512, off);
+		n = sb.st_size - off;
+		n = n > sizeof(buf) ? sizeof(buf) : n;
+		n = zfs_read(spa, &dn, buf, n, off);
 		if (n < 0) {
 			fprintf(stderr, "zfs_read failed\n");
 			exit(1);