xref: /dragonfly/sbin/hammer/cmd_mirror.c (revision e293de53)
1 /*
2  * Copyright (c) 2008 The DragonFly Project.  All rights reserved.
3  *
4  * This code is derived from software contributed to The DragonFly Project
5  * by Matthew Dillon <dillon@backplane.com>
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  *
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in
15  *    the documentation and/or other materials provided with the
16  *    distribution.
17  * 3. Neither the name of The DragonFly Project nor the names of its
18  *    contributors may be used to endorse or promote products derived
19  *    from this software without specific, prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
25  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
26  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
29  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
30  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
31  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  *
34  * $DragonFly: src/sbin/hammer/cmd_mirror.c,v 1.16 2008/11/09 05:22:56 dillon Exp $
35  */
36 
37 #include "hammer.h"
38 
39 #define SERIALBUF_SIZE	(512 * 1024)
40 
41 static int read_mrecords(int fd, char *buf, u_int size,
42 			 hammer_ioc_mrecord_head_t pickup);
43 static hammer_ioc_mrecord_any_t read_mrecord(int fdin, int *errorp,
44 			 hammer_ioc_mrecord_head_t pickup);
45 static void write_mrecord(int fdout, u_int32_t type,
46 			 hammer_ioc_mrecord_any_t mrec, int bytes);
47 static void generate_mrec_header(int fd, int fdout, int pfs_id,
48 			 hammer_tid_t *tid_begp, hammer_tid_t *tid_endp);
49 static int validate_mrec_header(int fd, int fdin, int is_target, int pfs_id,
50 			 struct hammer_ioc_mrecord_head *pickup,
51 			 hammer_tid_t *tid_begp, hammer_tid_t *tid_endp);
52 static void update_pfs_snapshot(int fd, hammer_tid_t snapshot_tid, int pfs_id);
53 static ssize_t writebw(int fd, const void *buf, size_t nbytes,
54 			u_int64_t *bwcount, struct timeval *tv1);
55 static int getyn(void);
56 static void mirror_usage(int code);
57 
58 /*
59  * Generate a mirroring data stream from the specific source over the
60  * entire key range, but restricted to the specified transaction range.
61  *
62  * The HAMMER VFS does most of the work, we add a few new mrecord
63  * types to negotiate the TID ranges and verify that the entire
64  * stream made it to the destination.
65  */
66 void
67 hammer_cmd_mirror_read(char **av, int ac, int streaming)
68 {
69 	struct hammer_ioc_mirror_rw mirror;
70 	struct hammer_ioc_pseudofs_rw pfs;
71 	union hammer_ioc_mrecord_any mrec_tmp;
72 	struct hammer_ioc_mrecord_head pickup;
73 	hammer_ioc_mrecord_any_t mrec;
74 	hammer_tid_t sync_tid;
75 	const char *filesystem;
76 	char *buf = malloc(SERIALBUF_SIZE);
77 	int interrupted = 0;
78 	int error;
79 	int fd;
80 	int n;
81 	int didwork;
82 	int64_t total_bytes;
83 	time_t base_t = time(NULL);
84 	struct timeval bwtv;
85 	u_int64_t bwcount;
86 
87 	if (ac == 0 || ac > 2)
88 		mirror_usage(1);
89 	filesystem = av[0];
90 
91 	pickup.signature = 0;
92 	pickup.type = 0;
93 
94 again:
95 	bzero(&mirror, sizeof(mirror));
96 	hammer_key_beg_init(&mirror.key_beg);
97 	hammer_key_end_init(&mirror.key_end);
98 
99 	fd = getpfs(&pfs, filesystem);
100 
101 	if (streaming && VerboseOpt) {
102 		fprintf(stderr, "\nRunning");
103 		fflush(stderr);
104 	}
105 	total_bytes = 0;
106 	gettimeofday(&bwtv, NULL);
107 	bwcount = 0;
108 
109 	/*
110 	 * Send initial header for the purpose of determining shared-uuid.
111 	 */
112 	generate_mrec_header(fd, 1, pfs.pfs_id, NULL, NULL);
113 
114 	/*
115 	 * In 2-way mode the target will send us a PFS info packet
116 	 * first.  Use the target's current snapshot TID as our default
117 	 * begin TID.
118 	 */
119 	mirror.tid_beg = 0;
120 	if (TwoWayPipeOpt) {
121 		n = validate_mrec_header(fd, 0, 0, pfs.pfs_id, &pickup,
122 					 NULL, &mirror.tid_beg);
123 		if (n < 0) {	/* got TERM record */
124 			relpfs(fd, &pfs);
125 			return;
126 		}
127 		++mirror.tid_beg;
128 	}
129 
130 	/*
131 	 * Write out the PFS header, tid_beg will be updated if our PFS
132 	 * has a larger begin sync.  tid_end is set to the latest source
133 	 * TID whos flush cycle has completed.
134 	 */
135 	generate_mrec_header(fd, 1, pfs.pfs_id,
136 			     &mirror.tid_beg, &mirror.tid_end);
137 
138 	/* XXX streaming mode support w/ cycle or command line arg */
139 	/*
140 	 * A cycle file overrides the beginning TID
141 	 */
142 	hammer_get_cycle(&mirror.key_beg, &mirror.tid_beg);
143 
144 	if (ac == 2)
145 		mirror.tid_beg = strtoull(av[1], NULL, 0);
146 
147 	if (streaming == 0 || VerboseOpt >= 2) {
148 		fprintf(stderr,
149 			"Mirror-read: Mirror from %016llx to %016llx\n",
150 			mirror.tid_beg, mirror.tid_end);
151 	}
152 	if (mirror.key_beg.obj_id != (int64_t)HAMMER_MIN_OBJID) {
153 		fprintf(stderr, "Mirror-read: Resuming at object %016llx\n",
154 			mirror.key_beg.obj_id);
155 	}
156 
157 	/*
158 	 * Nothing to do if begin equals end.
159 	 */
160 	if (mirror.tid_beg >= mirror.tid_end) {
161 		if (streaming == 0 || VerboseOpt >= 2)
162 			fprintf(stderr, "Mirror-read: No work to do\n");
163 		didwork = 0;
164 		goto done;
165 	}
166 	didwork = 1;
167 
168 	/*
169 	 * Write out bulk records
170 	 */
171 	mirror.ubuf = buf;
172 	mirror.size = SERIALBUF_SIZE;
173 
174 	do {
175 		mirror.count = 0;
176 		mirror.pfs_id = pfs.pfs_id;
177 		mirror.shared_uuid = pfs.ondisk->shared_uuid;
178 		if (ioctl(fd, HAMMERIOC_MIRROR_READ, &mirror) < 0) {
179 			fprintf(stderr, "Mirror-read %s failed: %s\n",
180 				filesystem, strerror(errno));
181 			exit(1);
182 		}
183 		if (mirror.head.flags & HAMMER_IOC_HEAD_ERROR) {
184 			fprintf(stderr,
185 				"Mirror-read %s fatal error %d\n",
186 				filesystem, mirror.head.error);
187 			exit(1);
188 		}
189 		if (mirror.count) {
190 			if (BandwidthOpt) {
191 				n = writebw(1, mirror.ubuf, mirror.count,
192 					    &bwcount, &bwtv);
193 			} else {
194 				n = write(1, mirror.ubuf, mirror.count);
195 			}
196 			if (n != mirror.count) {
197 				fprintf(stderr, "Mirror-read %s failed: "
198 						"short write\n",
199 				filesystem);
200 				exit(1);
201 			}
202 		}
203 		total_bytes += mirror.count;
204 		if (streaming && VerboseOpt) {
205 			fprintf(stderr, "\r%016llx %11lld",
206 				mirror.key_cur.obj_id,
207 				total_bytes);
208 			fflush(stderr);
209 		}
210 		mirror.key_beg = mirror.key_cur;
211 		if (TimeoutOpt &&
212 		    (unsigned)(time(NULL) - base_t) > (unsigned)TimeoutOpt) {
213 			fprintf(stderr,
214 				"Mirror-read %s interrupted by timer at"
215 				" %016llx\n",
216 				filesystem,
217 				mirror.key_cur.obj_id);
218 			interrupted = 1;
219 			break;
220 		}
221 	} while (mirror.count != 0);
222 
223 done:
224 	/*
225 	 * Write out the termination sync record - only if not interrupted
226 	 */
227 	if (interrupted == 0) {
228 		if (didwork) {
229 			write_mrecord(1, HAMMER_MREC_TYPE_SYNC,
230 				      &mrec_tmp, sizeof(mrec_tmp.sync));
231 		} else {
232 			write_mrecord(1, HAMMER_MREC_TYPE_IDLE,
233 				      &mrec_tmp, sizeof(mrec_tmp.sync));
234 		}
235 	}
236 
237 	/*
238 	 * If the -2 option was given (automatic when doing mirror-copy),
239 	 * a two-way pipe is assumed and we expect a response mrec from
240 	 * the target.
241 	 */
242 	if (TwoWayPipeOpt) {
243 		mrec = read_mrecord(0, &error, &pickup);
244 		if (mrec == NULL ||
245 		    mrec->head.type != HAMMER_MREC_TYPE_UPDATE ||
246 		    mrec->head.rec_size != sizeof(mrec->update)) {
247 			fprintf(stderr, "mirror_read: Did not get final "
248 					"acknowledgement packet from target\n");
249 			exit(1);
250 		}
251 		if (interrupted) {
252 			if (CyclePath) {
253 				hammer_set_cycle(&mirror.key_cur, mirror.tid_beg);
254 				fprintf(stderr, "Cyclefile %s updated for "
255 					"continuation\n", CyclePath);
256 			}
257 		} else {
258 			sync_tid = mrec->update.tid;
259 			if (CyclePath) {
260 				hammer_key_beg_init(&mirror.key_beg);
261 				hammer_set_cycle(&mirror.key_beg, sync_tid);
262 				fprintf(stderr, "Cyclefile %s updated to 0x%016llx\n",
263 					CyclePath, sync_tid);
264 			}
265 		}
266 	} else if (CyclePath) {
267 		/* NOTE! mirror.tid_beg cannot be updated */
268 		fprintf(stderr, "Warning: cycle file (-c option) cannot be "
269 				"fully updated unless you use mirror-copy\n");
270 		hammer_set_cycle(&mirror.key_beg, mirror.tid_beg);
271 	}
272 	if (streaming && interrupted == 0) {
273 		time_t t1 = time(NULL);
274 		time_t t2;
275 
276 		if (VerboseOpt) {
277 			fprintf(stderr, " W");
278 			fflush(stderr);
279 		}
280 		pfs.ondisk->sync_end_tid = mirror.tid_end;
281 		if (ioctl(fd, HAMMERIOC_WAI_PSEUDOFS, &pfs) < 0) {
282 			fprintf(stderr, "Mirror-read %s: cannot stream: %s\n",
283 				filesystem, strerror(errno));
284 		} else {
285 			t2 = time(NULL) - t1;
286 			if (t2 >= 0 && t2 < DelayOpt) {
287 				if (VerboseOpt) {
288 					fprintf(stderr, "\bD");
289 					fflush(stderr);
290 				}
291 				sleep(DelayOpt - t2);
292 			}
293 			if (VerboseOpt) {
294 				fprintf(stderr, "\b ");
295 				fflush(stderr);
296 			}
297 			relpfs(fd, &pfs);
298 			goto again;
299 		}
300 	}
301 	write_mrecord(1, HAMMER_MREC_TYPE_TERM,
302 		      &mrec_tmp, sizeof(mrec_tmp.sync));
303 	relpfs(fd, &pfs);
304 	fprintf(stderr, "Mirror-read %s succeeded\n", filesystem);
305 }
306 
307 static void
308 create_pfs(const char *filesystem, uuid_t *s_uuid)
309 {
310 	fprintf(stderr, "PFS slave %s does not exist.\n"
311 			"Do you want to create a new slave PFS? (yes|no) ",
312 			filesystem);
313 	fflush(stderr);
314 	if (getyn() != 1) {
315 		fprintf(stderr, "Aborting operation\n");
316 		exit(1);
317 	}
318 
319 	u_int32_t status;
320 	char *shared_uuid = NULL;
321 	uuid_to_string(s_uuid, &shared_uuid, &status);
322 
323 	char *cmd = NULL;
324 	asprintf(&cmd, "/sbin/hammer pfs-slave '%s' shared-uuid=%s 1>&2",
325 		 filesystem, shared_uuid);
326 	free(shared_uuid);
327 
328 	if (cmd == NULL) {
329 		fprintf(stderr, "Failed to alloc memory\n");
330 		exit(1);
331 	}
332 	if (system(cmd) != 0) {
333 		fprintf(stderr, "Failed to create PFS\n");
334 	}
335 	free(cmd);
336 }
337 
338 /*
339  * Pipe the mirroring data stream on stdin to the HAMMER VFS, adding
340  * some additional packet types to negotiate TID ranges and to verify
341  * completion.  The HAMMER VFS does most of the work.
342  *
343  * It is important to note that the mirror.key_{beg,end} range must
344  * match the ranged used by the original.  For now both sides use
345  * range the entire key space.
346  *
347  * It is even more important that the records in the stream conform
348  * to the TID range also supplied in the stream.  The HAMMER VFS will
349  * use the REC, PASS, and SKIP record types to track the portions of
350  * the B-Tree being scanned in order to be able to proactively delete
351  * records on the target within those active areas that are not mentioned
352  * by the source.
353  *
354  * The mirror.key_cur field is used by the VFS to do this tracking.  It
355  * must be initialized to key_beg but then is persistently updated by
356  * the HAMMER VFS on each successive ioctl() call.  If you blow up this
357  * field you will blow up the mirror target, possibly to the point of
358  * deleting everything.  As a safety measure the HAMMER VFS simply marks
359  * the records that the source has destroyed as deleted on the target,
360  * and normal pruning operations will deal with their final disposition
361  * at some later time.
362  */
363 void
364 hammer_cmd_mirror_write(char **av, int ac)
365 {
366 	struct hammer_ioc_mirror_rw mirror;
367 	const char *filesystem;
368 	char *buf = malloc(SERIALBUF_SIZE);
369 	struct hammer_ioc_pseudofs_rw pfs;
370 	struct hammer_ioc_mrecord_head pickup;
371 	struct hammer_ioc_synctid synctid;
372 	union hammer_ioc_mrecord_any mrec_tmp;
373 	hammer_ioc_mrecord_any_t mrec;
374 	struct stat st;
375 	int error;
376 	int fd;
377 	int n;
378 
379 	if (ac != 1)
380 		mirror_usage(1);
381 	filesystem = av[0];
382 
383 	pickup.signature = 0;
384 	pickup.type = 0;
385 
386 again:
387 	bzero(&mirror, sizeof(mirror));
388 	hammer_key_beg_init(&mirror.key_beg);
389 	hammer_key_end_init(&mirror.key_end);
390 	mirror.key_end = mirror.key_beg;
391 
392 	/*
393 	 * Read initial packet
394 	 */
395 	mrec = read_mrecord(0, &error, &pickup);
396 	if (mrec == NULL) {
397 		if (error == 0)
398 			fprintf(stderr, "validate_mrec_header: short read\n");
399 		exit(1);
400 	}
401 	/*
402 	 * Validate packet
403 	 */
404 	if (mrec->head.type == HAMMER_MREC_TYPE_TERM) {
405 		return;
406 	}
407 	if (mrec->head.type != HAMMER_MREC_TYPE_PFSD) {
408 		fprintf(stderr, "validate_mrec_header: did not get expected "
409 				"PFSD record type\n");
410 		exit(1);
411 	}
412 	if (mrec->head.rec_size != sizeof(mrec->pfs)) {
413 		fprintf(stderr, "validate_mrec_header: unexpected payload "
414 				"size\n");
415 		exit(1);
416 	}
417 
418 	/*
419 	 * Create slave PFS if it doesn't yet exist
420 	 */
421 	if (lstat(filesystem, &st) != 0) {
422 		create_pfs(filesystem, &mrec->pfs.pfsd.shared_uuid);
423 	}
424 	free(mrec);
425 	mrec = NULL;
426 
427 	fd = getpfs(&pfs, filesystem);
428 
429 	/*
430 	 * In two-way mode the target writes out a PFS packet first.
431 	 * The source uses our tid_end as its tid_beg by default,
432 	 * picking up where it left off.
433 	 */
434 	mirror.tid_beg = 0;
435 	if (TwoWayPipeOpt) {
436 		generate_mrec_header(fd, 1, pfs.pfs_id,
437 				     &mirror.tid_beg, &mirror.tid_end);
438 	}
439 
440 	/*
441 	 * Read and process the PFS header.  The source informs us of
442 	 * the TID range the stream represents.
443 	 */
444 	n = validate_mrec_header(fd, 0, 1, pfs.pfs_id, &pickup,
445 				 &mirror.tid_beg, &mirror.tid_end);
446 	if (n < 0) {	/* got TERM record */
447 		relpfs(fd, &pfs);
448 		return;
449 	}
450 
451 	mirror.ubuf = buf;
452 	mirror.size = SERIALBUF_SIZE;
453 
454 	/*
455 	 * Read and process bulk records (REC, PASS, and SKIP types).
456 	 *
457 	 * On your life, do NOT mess with mirror.key_cur or your mirror
458 	 * target may become history.
459 	 */
460 	for (;;) {
461 		mirror.count = 0;
462 		mirror.pfs_id = pfs.pfs_id;
463 		mirror.shared_uuid = pfs.ondisk->shared_uuid;
464 		mirror.size = read_mrecords(0, buf, SERIALBUF_SIZE, &pickup);
465 		if (mirror.size <= 0)
466 			break;
467 		if (ioctl(fd, HAMMERIOC_MIRROR_WRITE, &mirror) < 0) {
468 			fprintf(stderr, "Mirror-write %s failed: %s\n",
469 				filesystem, strerror(errno));
470 			exit(1);
471 		}
472 		if (mirror.head.flags & HAMMER_IOC_HEAD_ERROR) {
473 			fprintf(stderr,
474 				"Mirror-write %s fatal error %d\n",
475 				filesystem, mirror.head.error);
476 			exit(1);
477 		}
478 #if 0
479 		if (mirror.head.flags & HAMMER_IOC_HEAD_INTR) {
480 			fprintf(stderr,
481 				"Mirror-write %s interrupted by timer at"
482 				" %016llx\n",
483 				filesystem,
484 				mirror.key_cur.obj_id);
485 			exit(0);
486 		}
487 #endif
488 	}
489 
490 	/*
491 	 * Read and process the termination sync record.
492 	 */
493 	mrec = read_mrecord(0, &error, &pickup);
494 
495 	if (mrec && mrec->head.type == HAMMER_MREC_TYPE_TERM) {
496 		fprintf(stderr, "Mirror-write: received termination request\n");
497 		free(mrec);
498 		return;
499 	}
500 
501 	if (mrec == NULL ||
502 	    (mrec->head.type != HAMMER_MREC_TYPE_SYNC &&
503 	     mrec->head.type != HAMMER_MREC_TYPE_IDLE) ||
504 	    mrec->head.rec_size != sizeof(mrec->sync)) {
505 		fprintf(stderr, "Mirror-write %s: Did not get termination "
506 				"sync record, or rec_size is wrong rt=%d\n",
507 				filesystem, mrec->head.type);
508 		exit(1);
509 	}
510 
511 	/*
512 	 * Update the PFS info on the target so the user has visibility
513 	 * into the new snapshot, and sync the target filesystem.
514 	 */
515 	if (mrec->head.type == HAMMER_MREC_TYPE_SYNC) {
516 		update_pfs_snapshot(fd, mirror.tid_end, pfs.pfs_id);
517 
518 		bzero(&synctid, sizeof(synctid));
519 		synctid.op = HAMMER_SYNCTID_SYNC2;
520 		ioctl(fd, HAMMERIOC_SYNCTID, &synctid);
521 
522 		if (VerboseOpt >= 2) {
523 			fprintf(stderr, "Mirror-write %s: succeeded\n",
524 				filesystem);
525 		}
526 	}
527 
528 	free(mrec);
529 	mrec = NULL;
530 
531 	/*
532 	 * Report back to the originator.
533 	 */
534 	if (TwoWayPipeOpt) {
535 		mrec_tmp.update.tid = mirror.tid_end;
536 		write_mrecord(1, HAMMER_MREC_TYPE_UPDATE,
537 			      &mrec_tmp, sizeof(mrec_tmp.update));
538 	} else {
539 		printf("Source can update synctid to 0x%016llx\n",
540 		       mirror.tid_end);
541 	}
542 	relpfs(fd, &pfs);
543 	goto again;
544 }
545 
546 void
547 hammer_cmd_mirror_dump(void)
548 {
549 	char *buf = malloc(SERIALBUF_SIZE);
550 	struct hammer_ioc_mrecord_head pickup;
551 	hammer_ioc_mrecord_any_t mrec;
552 	int error;
553 	int size;
554 	int offset;
555 	int bytes;
556 
557 	/*
558 	 * Read and process the PFS header
559 	 */
560 	pickup.signature = 0;
561 	pickup.type = 0;
562 
563 	mrec = read_mrecord(0, &error, &pickup);
564 
565 	/*
566 	 * Read and process bulk records
567 	 */
568 	for (;;) {
569 		size = read_mrecords(0, buf, SERIALBUF_SIZE, &pickup);
570 		if (size <= 0)
571 			break;
572 		offset = 0;
573 		while (offset < size) {
574 			mrec = (void *)((char *)buf + offset);
575 			bytes = HAMMER_HEAD_DOALIGN(mrec->head.rec_size);
576 			if (offset + bytes > size) {
577 				fprintf(stderr, "Misaligned record\n");
578 				exit(1);
579 			}
580 
581 			switch(mrec->head.type & HAMMER_MRECF_TYPE_MASK) {
582 			case HAMMER_MREC_TYPE_REC_BADCRC:
583 			case HAMMER_MREC_TYPE_REC:
584 				printf("Record obj=%016llx key=%016llx "
585 				       "rt=%02x ot=%02x",
586 					mrec->rec.leaf.base.obj_id,
587 					mrec->rec.leaf.base.key,
588 					mrec->rec.leaf.base.rec_type,
589 					mrec->rec.leaf.base.obj_type);
590 				if (mrec->head.type ==
591 				    HAMMER_MREC_TYPE_REC_BADCRC) {
592 					printf(" (BAD CRC)");
593 				}
594 				printf("\n");
595 				printf("       tids %016llx:%016llx data=%d\n",
596 					mrec->rec.leaf.base.create_tid,
597 					mrec->rec.leaf.base.delete_tid,
598 					mrec->rec.leaf.data_len);
599 				break;
600 			case HAMMER_MREC_TYPE_PASS:
601 				printf("Pass   obj=%016llx key=%016llx "
602 				       "rt=%02x ot=%02x\n",
603 					mrec->rec.leaf.base.obj_id,
604 					mrec->rec.leaf.base.key,
605 					mrec->rec.leaf.base.rec_type,
606 					mrec->rec.leaf.base.obj_type);
607 				printf("       tids %016llx:%016llx data=%d\n",
608 					mrec->rec.leaf.base.create_tid,
609 					mrec->rec.leaf.base.delete_tid,
610 					mrec->rec.leaf.data_len);
611 				break;
612 			case HAMMER_MREC_TYPE_SKIP:
613 				printf("Skip   obj=%016llx key=%016llx rt=%02x to\n"
614 				       "       obj=%016llx key=%016llx rt=%02x\n",
615 				       mrec->skip.skip_beg.obj_id,
616 				       mrec->skip.skip_beg.key,
617 				       mrec->skip.skip_beg.rec_type,
618 				       mrec->skip.skip_end.obj_id,
619 				       mrec->skip.skip_end.key,
620 				       mrec->skip.skip_end.rec_type);
621 			default:
622 				break;
623 			}
624 			offset += bytes;
625 		}
626 	}
627 
628 	/*
629 	 * Read and process the termination sync record.
630 	 */
631 	mrec = read_mrecord(0, &error, &pickup);
632 	if (mrec == NULL ||
633 	    (mrec->head.type != HAMMER_MREC_TYPE_SYNC &&
634 	     mrec->head.type != HAMMER_MREC_TYPE_IDLE)
635 	 ) {
636 		fprintf(stderr, "Mirror-dump: Did not get termination "
637 				"sync record\n");
638 	}
639 }
640 
641 void
642 hammer_cmd_mirror_copy(char **av, int ac, int streaming)
643 {
644 	pid_t pid1;
645 	pid_t pid2;
646 	int fds[2];
647 	const char *xav[16];
648 	char tbuf[16];
649 	char *ptr;
650 	int xac;
651 
652 	if (ac != 2)
653 		mirror_usage(1);
654 
655 	if (pipe(fds) < 0) {
656 		perror("pipe");
657 		exit(1);
658 	}
659 
660 	TwoWayPipeOpt = 1;
661 
662 	/*
663 	 * Source
664 	 */
665 	if ((pid1 = fork()) == 0) {
666 		dup2(fds[0], 0);
667 		dup2(fds[0], 1);
668 		close(fds[0]);
669 		close(fds[1]);
670 		if ((ptr = strchr(av[0], ':')) != NULL) {
671 			*ptr++ = 0;
672 			xac = 0;
673 			xav[xac++] = "ssh";
674 			xav[xac++] = av[0];
675 			xav[xac++] = "hammer";
676 
677 			switch(VerboseOpt) {
678 			case 0:
679 				break;
680 			case 1:
681 				xav[xac++] = "-v";
682 				break;
683 			case 2:
684 				xav[xac++] = "-vv";
685 				break;
686 			default:
687 				xav[xac++] = "-vvv";
688 				break;
689 			}
690 			xav[xac++] = "-2";
691 			if (TimeoutOpt) {
692 				snprintf(tbuf, sizeof(tbuf), "%d", TimeoutOpt);
693 				xav[xac++] = "-t";
694 				xav[xac++] = tbuf;
695 			}
696 			if (streaming)
697 				xav[xac++] = "mirror-read-stream";
698 			else
699 				xav[xac++] = "mirror-read";
700 			xav[xac++] = ptr;
701 			xav[xac++] = NULL;
702 			execv("/usr/bin/ssh", (void *)xav);
703 		} else {
704 			hammer_cmd_mirror_read(av, 1, streaming);
705 			fflush(stdout);
706 			fflush(stderr);
707 		}
708 		_exit(1);
709 	}
710 
711 	/*
712 	 * Target
713 	 */
714 	if ((pid2 = fork()) == 0) {
715 		dup2(fds[1], 0);
716 		dup2(fds[1], 1);
717 		close(fds[0]);
718 		close(fds[1]);
719 		if ((ptr = strchr(av[1], ':')) != NULL) {
720 			*ptr++ = 0;
721 			xac = 0;
722 			xav[xac++] = "ssh";
723 			xav[xac++] = av[1];
724 			xav[xac++] = "hammer";
725 
726 			switch(VerboseOpt) {
727 			case 0:
728 				break;
729 			case 1:
730 				xav[xac++] = "-v";
731 				break;
732 			case 2:
733 				xav[xac++] = "-vv";
734 				break;
735 			default:
736 				xav[xac++] = "-vvv";
737 				break;
738 			}
739 
740 			xav[xac++] = "-2";
741 			xav[xac++] = "mirror-write";
742 			xav[xac++] = ptr;
743 			xav[xac++] = NULL;
744 			execv("/usr/bin/ssh", (void *)xav);
745 		} else {
746 			hammer_cmd_mirror_write(av + 1, 1);
747 			fflush(stdout);
748 			fflush(stderr);
749 		}
750 		_exit(1);
751 	}
752 	close(fds[0]);
753 	close(fds[1]);
754 
755 	while (waitpid(pid1, NULL, 0) <= 0)
756 		;
757 	while (waitpid(pid2, NULL, 0) <= 0)
758 		;
759 }
760 
761 /*
762  * Read and return multiple mrecords
763  */
764 static int
765 read_mrecords(int fd, char *buf, u_int size, hammer_ioc_mrecord_head_t pickup)
766 {
767 	hammer_ioc_mrecord_any_t mrec;
768 	u_int count;
769 	size_t n;
770 	size_t i;
771 	size_t bytes;
772 	int type;
773 
774 	count = 0;
775 	while (size - count >= HAMMER_MREC_HEADSIZE) {
776 		/*
777 		 * Cached the record header in case we run out of buffer
778 		 * space.
779 		 */
780 		fflush(stdout);
781 		if (pickup->signature == 0) {
782 			for (n = 0; n < HAMMER_MREC_HEADSIZE; n += i) {
783 				i = read(fd, (char *)pickup + n,
784 					 HAMMER_MREC_HEADSIZE - n);
785 				if (i <= 0)
786 					break;
787 			}
788 			if (n == 0)
789 				break;
790 			if (n != HAMMER_MREC_HEADSIZE) {
791 				fprintf(stderr, "read_mrecords: short read on pipe\n");
792 				exit(1);
793 			}
794 			if (pickup->signature != HAMMER_IOC_MIRROR_SIGNATURE) {
795 				fprintf(stderr, "read_mrecords: malformed record on pipe, "
796 					"bad signature\n");
797 				exit(1);
798 			}
799 		}
800 		if (pickup->rec_size < HAMMER_MREC_HEADSIZE ||
801 		    pickup->rec_size > sizeof(*mrec) + HAMMER_XBUFSIZE) {
802 			fprintf(stderr, "read_mrecords: malformed record on pipe, "
803 				"illegal rec_size\n");
804 			exit(1);
805 		}
806 
807 		/*
808 		 * Stop if we have insufficient space for the record and data.
809 		 */
810 		bytes = HAMMER_HEAD_DOALIGN(pickup->rec_size);
811 		if (size - count < bytes)
812 			break;
813 
814 		/*
815 		 * Stop if the record type is not a REC, SKIP, or PASS,
816 		 * which are the only types the ioctl supports.  Other types
817 		 * are used only by the userland protocol.
818 		 *
819 		 * Ignore all flags.
820 		 */
821 		type = pickup->type & HAMMER_MRECF_TYPE_LOMASK;
822 		if (type != HAMMER_MREC_TYPE_PFSD &&
823 		    type != HAMMER_MREC_TYPE_REC &&
824 		    type != HAMMER_MREC_TYPE_SKIP &&
825 		    type != HAMMER_MREC_TYPE_PASS) {
826 			break;
827 		}
828 
829 		/*
830 		 * Read the remainder and clear the pickup signature.
831 		 */
832 		for (n = HAMMER_MREC_HEADSIZE; n < bytes; n += i) {
833 			i = read(fd, buf + count + n, bytes - n);
834 			if (i <= 0)
835 				break;
836 		}
837 		if (n != bytes) {
838 			fprintf(stderr, "read_mrecords: short read on pipe\n");
839 			exit(1);
840 		}
841 
842 		bcopy(pickup, buf + count, HAMMER_MREC_HEADSIZE);
843 		pickup->signature = 0;
844 		pickup->type = 0;
845 		mrec = (void *)(buf + count);
846 
847 		/*
848 		 * Validate the completed record
849 		 */
850 		if (mrec->head.rec_crc !=
851 		    crc32((char *)mrec + HAMMER_MREC_CRCOFF,
852 			  mrec->head.rec_size - HAMMER_MREC_CRCOFF)) {
853 			fprintf(stderr, "read_mrecords: malformed record "
854 					"on pipe, bad crc\n");
855 			exit(1);
856 		}
857 
858 		/*
859 		 * If its a B-Tree record validate the data crc.
860 		 *
861 		 * NOTE: If the VFS passes us an explicitly errorde mrec
862 		 *	 we just pass it through.
863 		 */
864 		type = mrec->head.type & HAMMER_MRECF_TYPE_MASK;
865 
866 		if (type == HAMMER_MREC_TYPE_REC) {
867 			if (mrec->head.rec_size <
868 			    sizeof(mrec->rec) + mrec->rec.leaf.data_len) {
869 				fprintf(stderr,
870 					"read_mrecords: malformed record on "
871 					"pipe, illegal element data_len\n");
872 				exit(1);
873 			}
874 			if (mrec->rec.leaf.data_len &&
875 			    mrec->rec.leaf.data_offset &&
876 			    hammer_crc_test_leaf(&mrec->rec + 1, &mrec->rec.leaf) == 0) {
877 				fprintf(stderr,
878 					"read_mrecords: data_crc did not "
879 					"match data! obj=%016llx key=%016llx\n",
880 					mrec->rec.leaf.base.obj_id,
881 					mrec->rec.leaf.base.key);
882 				fprintf(stderr,
883 					"continuing, but there are problems\n");
884 			}
885 		}
886 		count += bytes;
887 	}
888 	return(count);
889 }
890 
891 /*
892  * Read and return a single mrecord.
893  */
894 static
895 hammer_ioc_mrecord_any_t
896 read_mrecord(int fdin, int *errorp, hammer_ioc_mrecord_head_t pickup)
897 {
898 	hammer_ioc_mrecord_any_t mrec;
899 	struct hammer_ioc_mrecord_head mrechd;
900 	size_t bytes;
901 	size_t n;
902 	size_t i;
903 
904 	if (pickup && pickup->type != 0) {
905 		mrechd = *pickup;
906 		pickup->signature = 0;
907 		pickup->type = 0;
908 		n = HAMMER_MREC_HEADSIZE;
909 	} else {
910 		/*
911 		 * Read in the PFSD header from the sender.
912 		 */
913 		for (n = 0; n < HAMMER_MREC_HEADSIZE; n += i) {
914 			i = read(fdin, (char *)&mrechd + n, HAMMER_MREC_HEADSIZE - n);
915 			if (i <= 0)
916 				break;
917 		}
918 		if (n == 0) {
919 			*errorp = 0;	/* EOF */
920 			return(NULL);
921 		}
922 		if (n != HAMMER_MREC_HEADSIZE) {
923 			fprintf(stderr, "short read of mrecord header\n");
924 			*errorp = EPIPE;
925 			return(NULL);
926 		}
927 	}
928 	if (mrechd.signature != HAMMER_IOC_MIRROR_SIGNATURE) {
929 		fprintf(stderr, "read_mrecord: bad signature\n");
930 		*errorp = EINVAL;
931 		return(NULL);
932 	}
933 	bytes = HAMMER_HEAD_DOALIGN(mrechd.rec_size);
934 	assert(bytes >= sizeof(mrechd));
935 	mrec = malloc(bytes);
936 	mrec->head = mrechd;
937 
938 	while (n < bytes) {
939 		i = read(fdin, (char *)mrec + n, bytes - n);
940 		if (i <= 0)
941 			break;
942 		n += i;
943 	}
944 	if (n != bytes) {
945 		fprintf(stderr, "read_mrecord: short read on payload\n");
946 		*errorp = EPIPE;
947 		return(NULL);
948 	}
949 	if (mrec->head.rec_crc !=
950 	    crc32((char *)mrec + HAMMER_MREC_CRCOFF,
951 		  mrec->head.rec_size - HAMMER_MREC_CRCOFF)) {
952 		fprintf(stderr, "read_mrecord: bad CRC\n");
953 		*errorp = EINVAL;
954 		return(NULL);
955 	}
956 	*errorp = 0;
957 	return(mrec);
958 }
959 
960 static
961 void
962 write_mrecord(int fdout, u_int32_t type, hammer_ioc_mrecord_any_t mrec,
963 	      int bytes)
964 {
965 	char zbuf[HAMMER_HEAD_ALIGN];
966 	int pad;
967 
968 	pad = HAMMER_HEAD_DOALIGN(bytes) - bytes;
969 
970 	assert(bytes >= (int)sizeof(mrec->head));
971 	bzero(&mrec->head, sizeof(mrec->head));
972 	mrec->head.signature = HAMMER_IOC_MIRROR_SIGNATURE;
973 	mrec->head.type = type;
974 	mrec->head.rec_size = bytes;
975 	mrec->head.rec_crc = crc32((char *)mrec + HAMMER_MREC_CRCOFF,
976 				   bytes - HAMMER_MREC_CRCOFF);
977 	if (write(fdout, mrec, bytes) != bytes) {
978 		fprintf(stderr, "write_mrecord: error %d (%s)\n",
979 			errno, strerror(errno));
980 		exit(1);
981 	}
982 	if (pad) {
983 		bzero(zbuf, pad);
984 		if (write(fdout, zbuf, pad) != pad) {
985 			fprintf(stderr, "write_mrecord: error %d (%s)\n",
986 				errno, strerror(errno));
987 			exit(1);
988 		}
989 	}
990 }
991 
992 /*
993  * Generate a mirroring header with the pfs information of the
994  * originating filesytem.
995  */
996 static void
997 generate_mrec_header(int fd, int fdout, int pfs_id,
998 		     hammer_tid_t *tid_begp, hammer_tid_t *tid_endp)
999 {
1000 	struct hammer_ioc_pseudofs_rw pfs;
1001 	union hammer_ioc_mrecord_any mrec_tmp;
1002 
1003 	bzero(&pfs, sizeof(pfs));
1004 	bzero(&mrec_tmp, sizeof(mrec_tmp));
1005 	pfs.pfs_id = pfs_id;
1006 	pfs.ondisk = &mrec_tmp.pfs.pfsd;
1007 	pfs.bytes = sizeof(mrec_tmp.pfs.pfsd);
1008 	if (ioctl(fd, HAMMERIOC_GET_PSEUDOFS, &pfs) != 0) {
1009 		fprintf(stderr, "Mirror-read: not a HAMMER fs/pseudofs!\n");
1010 		exit(1);
1011 	}
1012 	if (pfs.version != HAMMER_IOC_PSEUDOFS_VERSION) {
1013 		fprintf(stderr, "Mirror-read: HAMMER pfs version mismatch!\n");
1014 		exit(1);
1015 	}
1016 
1017 	/*
1018 	 * sync_beg_tid - lowest TID on source after which a full history
1019 	 *	 	  is available.
1020 	 *
1021 	 * sync_end_tid - highest fully synchronized TID from source.
1022 	 */
1023 	if (tid_begp && *tid_begp < mrec_tmp.pfs.pfsd.sync_beg_tid)
1024 		*tid_begp = mrec_tmp.pfs.pfsd.sync_beg_tid;
1025 	if (tid_endp)
1026 		*tid_endp = mrec_tmp.pfs.pfsd.sync_end_tid;
1027 	mrec_tmp.pfs.version = pfs.version;
1028 	write_mrecord(fdout, HAMMER_MREC_TYPE_PFSD,
1029 		      &mrec_tmp, sizeof(mrec_tmp.pfs));
1030 }
1031 
1032 /*
1033  * Validate the pfs information from the originating filesystem
1034  * against the target filesystem.  shared_uuid must match.
1035  *
1036  * return -1 if we got a TERM record
1037  */
1038 static int
1039 validate_mrec_header(int fd, int fdin, int is_target, int pfs_id,
1040 		     struct hammer_ioc_mrecord_head *pickup,
1041 		     hammer_tid_t *tid_begp, hammer_tid_t *tid_endp)
1042 {
1043 	struct hammer_ioc_pseudofs_rw pfs;
1044 	struct hammer_pseudofs_data pfsd;
1045 	hammer_ioc_mrecord_any_t mrec;
1046 	int error;
1047 
1048 	/*
1049 	 * Get the PFSD info from the target filesystem.
1050 	 */
1051 	bzero(&pfs, sizeof(pfs));
1052 	bzero(&pfsd, sizeof(pfsd));
1053 	pfs.pfs_id = pfs_id;
1054 	pfs.ondisk = &pfsd;
1055 	pfs.bytes = sizeof(pfsd);
1056 	if (ioctl(fd, HAMMERIOC_GET_PSEUDOFS, &pfs) != 0) {
1057 		fprintf(stderr, "mirror-write: not a HAMMER fs/pseudofs!\n");
1058 		exit(1);
1059 	}
1060 	if (pfs.version != HAMMER_IOC_PSEUDOFS_VERSION) {
1061 		fprintf(stderr, "mirror-write: HAMMER pfs version mismatch!\n");
1062 		exit(1);
1063 	}
1064 
1065 	mrec = read_mrecord(fdin, &error, pickup);
1066 	if (mrec == NULL) {
1067 		if (error == 0)
1068 			fprintf(stderr, "validate_mrec_header: short read\n");
1069 		exit(1);
1070 	}
1071 	if (mrec->head.type == HAMMER_MREC_TYPE_TERM) {
1072 		free(mrec);
1073 		return(-1);
1074 	}
1075 
1076 	if (mrec->head.type != HAMMER_MREC_TYPE_PFSD) {
1077 		fprintf(stderr, "validate_mrec_header: did not get expected "
1078 				"PFSD record type\n");
1079 		exit(1);
1080 	}
1081 	if (mrec->head.rec_size != sizeof(mrec->pfs)) {
1082 		fprintf(stderr, "validate_mrec_header: unexpected payload "
1083 				"size\n");
1084 		exit(1);
1085 	}
1086 	if (mrec->pfs.version != pfs.version) {
1087 		fprintf(stderr, "validate_mrec_header: Version mismatch\n");
1088 		exit(1);
1089 	}
1090 
1091 	/*
1092 	 * Whew.  Ok, is the read PFS info compatible with the target?
1093 	 */
1094 	if (bcmp(&mrec->pfs.pfsd.shared_uuid, &pfsd.shared_uuid,
1095 		 sizeof(pfsd.shared_uuid)) != 0) {
1096 		fprintf(stderr,
1097 			"mirror-write: source and target have "
1098 			"different shared-uuid's!\n");
1099 		exit(1);
1100 	}
1101 	if (is_target &&
1102 	    (pfsd.mirror_flags & HAMMER_PFSD_SLAVE) == 0) {
1103 		fprintf(stderr, "mirror-write: target must be in slave mode\n");
1104 		exit(1);
1105 	}
1106 	if (tid_begp)
1107 		*tid_begp = mrec->pfs.pfsd.sync_beg_tid;
1108 	if (tid_endp)
1109 		*tid_endp = mrec->pfs.pfsd.sync_end_tid;
1110 	free(mrec);
1111 	return(0);
1112 }
1113 
1114 static void
1115 update_pfs_snapshot(int fd, hammer_tid_t snapshot_tid, int pfs_id)
1116 {
1117 	struct hammer_ioc_pseudofs_rw pfs;
1118 	struct hammer_pseudofs_data pfsd;
1119 
1120 	bzero(&pfs, sizeof(pfs));
1121 	bzero(&pfsd, sizeof(pfsd));
1122 	pfs.pfs_id = pfs_id;
1123 	pfs.ondisk = &pfsd;
1124 	pfs.bytes = sizeof(pfsd);
1125 	if (ioctl(fd, HAMMERIOC_GET_PSEUDOFS, &pfs) != 0) {
1126 		perror("update_pfs_snapshot (read)");
1127 		exit(1);
1128 	}
1129 	if (pfsd.sync_end_tid != snapshot_tid) {
1130 		pfsd.sync_end_tid = snapshot_tid;
1131 		if (ioctl(fd, HAMMERIOC_SET_PSEUDOFS, &pfs) != 0) {
1132 			perror("update_pfs_snapshot (rewrite)");
1133 			exit(1);
1134 		}
1135 		if (VerboseOpt >= 2) {
1136 			fprintf(stderr,
1137 				"Mirror-write: Completed, updated snapshot "
1138 				"to %016llx\n",
1139 				snapshot_tid);
1140 		}
1141 	}
1142 }
1143 
1144 /*
1145  * Bandwidth-limited write in chunks
1146  */
1147 static
1148 ssize_t
1149 writebw(int fd, const void *buf, size_t nbytes,
1150 	u_int64_t *bwcount, struct timeval *tv1)
1151 {
1152 	struct timeval tv2;
1153 	size_t n;
1154 	ssize_t r;
1155 	ssize_t a;
1156 	int usec;
1157 
1158 	a = 0;
1159 	r = 0;
1160 	while (nbytes) {
1161 		if (*bwcount + nbytes > BandwidthOpt)
1162 			n = BandwidthOpt - *bwcount;
1163 		else
1164 			n = nbytes;
1165 		if (n)
1166 			r = write(fd, buf, n);
1167 		if (r >= 0) {
1168 			a += r;
1169 			nbytes -= r;
1170 			buf = (const char *)buf + r;
1171 		}
1172 		if ((size_t)r != n)
1173 			break;
1174 		*bwcount += n;
1175 		if (*bwcount >= BandwidthOpt) {
1176 			gettimeofday(&tv2, NULL);
1177 			usec = (int)(tv2.tv_sec - tv1->tv_sec) * 1000000 +
1178 				(int)(tv2.tv_usec - tv1->tv_usec);
1179 			if (usec >= 0 && usec < 1000000)
1180 				usleep(1000000 - usec);
1181 			gettimeofday(tv1, NULL);
1182 			*bwcount -= BandwidthOpt;
1183 		}
1184 	}
1185 	return(a ? a : r);
1186 }
1187 
1188 /*
1189  * Get a yes or no answer from the terminal.  The program may be run as
1190  * part of a two-way pipe so we cannot use stdin for this operation.
1191  */
1192 static int
1193 getyn(void)
1194 {
1195 	char buf[256];
1196 	FILE *fp;
1197 	int result;
1198 
1199 	fp = fopen("/dev/tty", "r");
1200 	if (fp == NULL) {
1201 		fprintf(stderr, "No terminal for response\n");
1202 		return(-1);
1203 	}
1204 	result = -1;
1205 	while (fgets(buf, sizeof(buf), fp) != NULL) {
1206 		if (buf[0] == 'y' || buf[0] == 'Y') {
1207 			result = 1;
1208 			break;
1209 		}
1210 		if (buf[0] == 'n' || buf[0] == 'N') {
1211 			result = 0;
1212 			break;
1213 		}
1214 		fprintf(stderr, "Response not understood\n");
1215 		break;
1216 	}
1217 	fclose(fp);
1218 	return(result);
1219 }
1220 
1221 static void
1222 mirror_usage(int code)
1223 {
1224 	fprintf(stderr,
1225 		"hammer mirror-read <filesystem> [begin-tid]\n"
1226 		"hammer mirror-read-stream <filesystem> [begin-tid]\n"
1227 		"hammer mirror-write <filesystem>\n"
1228 		"hammer mirror-dump\n"
1229 		"hammer mirror-copy [[user@]host:]<filesystem>"
1230 				  " [[user@]host:]<filesystem>\n"
1231 		"hammer mirror-stream [[user@]host:]<filesystem>"
1232 				    " [[user@]host:]<filesystem>\n"
1233 	);
1234 	exit(code);
1235 }
1236 
1237