1/*-
2 * Copyright (c) 2004-2006 Pawel Jakub Dawidek <pjd@FreeBSD.org>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 *
26 * $FreeBSD$
27 */
28
29#ifndef	_G_MIRROR_H_
30#define	_G_MIRROR_H_
31
32#include <sys/endian.h>
33#include <sys/md5.h>
34
35#define	G_MIRROR_CLASS_NAME	"MIRROR"
36
37#define	G_MIRROR_MAGIC		"GEOM::MIRROR"
38/*
39 * Version history:
40 * 0 - Initial version number.
41 * 1 - Added 'prefer' balance algorithm.
42 * 2 - Added md_genid field to metadata.
43 * 3 - Added md_provsize field to metadata.
44 * 4 - Added 'no failure synchronization' flag.
45 */
46#define	G_MIRROR_VERSION	4
47
48#define	G_MIRROR_BALANCE_NONE		0
49#define	G_MIRROR_BALANCE_ROUND_ROBIN	1
50#define	G_MIRROR_BALANCE_LOAD		2
51#define	G_MIRROR_BALANCE_SPLIT		3
52#define	G_MIRROR_BALANCE_PREFER		4
53#define	G_MIRROR_BALANCE_MIN		G_MIRROR_BALANCE_NONE
54#define	G_MIRROR_BALANCE_MAX		G_MIRROR_BALANCE_PREFER
55
56#define	G_MIRROR_DISK_FLAG_DIRTY		0x0000000000000001ULL
57#define	G_MIRROR_DISK_FLAG_SYNCHRONIZING	0x0000000000000002ULL
58#define	G_MIRROR_DISK_FLAG_FORCE_SYNC		0x0000000000000004ULL
59#define	G_MIRROR_DISK_FLAG_INACTIVE		0x0000000000000008ULL
60#define	G_MIRROR_DISK_FLAG_HARDCODED		0x0000000000000010ULL
61#define	G_MIRROR_DISK_FLAG_BROKEN		0x0000000000000020ULL
62#define	G_MIRROR_DISK_FLAG_CANDELETE		0x0000000000000040ULL
63#define	G_MIRROR_DISK_FLAG_MASK		(G_MIRROR_DISK_FLAG_DIRTY |	\
64					 G_MIRROR_DISK_FLAG_SYNCHRONIZING | \
65					 G_MIRROR_DISK_FLAG_FORCE_SYNC | \
66					 G_MIRROR_DISK_FLAG_INACTIVE | \
67					 G_MIRROR_DISK_FLAG_CANDELETE)
68
69#define	G_MIRROR_DEVICE_FLAG_NOAUTOSYNC	0x0000000000000001ULL
70#define	G_MIRROR_DEVICE_FLAG_NOFAILSYNC	0x0000000000000002ULL
71#define	G_MIRROR_DEVICE_FLAG_MASK	(G_MIRROR_DEVICE_FLAG_NOAUTOSYNC | \
72					 G_MIRROR_DEVICE_FLAG_NOFAILSYNC)
73
74#ifdef _KERNEL
75extern u_int g_mirror_debug;
76
77#define	G_MIRROR_DEBUG(lvl, ...)	do {				\
78	if (g_mirror_debug >= (lvl)) {					\
79		printf("GEOM_MIRROR");					\
80		if (g_mirror_debug > 0)					\
81			printf("[%u]", lvl);				\
82		printf(": ");						\
83		printf(__VA_ARGS__);					\
84		printf("\n");						\
85	}								\
86} while (0)
87#define	G_MIRROR_LOGREQ(lvl, bp, ...)	do {				\
88	if (g_mirror_debug >= (lvl)) {					\
89		printf("GEOM_MIRROR");					\
90		if (g_mirror_debug > 0)					\
91			printf("[%u]", lvl);				\
92		printf(": ");						\
93		printf(__VA_ARGS__);					\
94		printf(" ");						\
95		g_print_bio(bp);					\
96		printf("\n");						\
97	}								\
98} while (0)
99
100#define	G_MIRROR_BIO_FLAG_REGULAR	0x01
101#define	G_MIRROR_BIO_FLAG_SYNC		0x02
102
103/*
104 * Informations needed for synchronization.
105 */
106struct g_mirror_disk_sync {
107	struct g_consumer *ds_consumer;	/* Consumer connected to our mirror. */
108	off_t		  ds_offset;	/* Offset of next request to send. */
109	off_t		  ds_offset_done; /* Offset of already synchronized
110					   region. */
111	u_int		  ds_syncid;	/* Disk's synchronization ID. */
112	u_int		  ds_inflight;	/* Number of in-flight sync requests. */
113	struct bio	**ds_bios;	/* BIOs for synchronization I/O. */
114};
115
116/*
117 * Informations needed for synchronization.
118 */
119struct g_mirror_device_sync {
120	struct g_geom	*ds_geom;	/* Synchronization geom. */
121	u_int		 ds_ndisks;	/* Number of disks in SYNCHRONIZING
122					   state. */
123};
124
125#define	G_MIRROR_DISK_STATE_NONE		0
126#define	G_MIRROR_DISK_STATE_NEW			1
127#define	G_MIRROR_DISK_STATE_ACTIVE		2
128#define	G_MIRROR_DISK_STATE_STALE		3
129#define	G_MIRROR_DISK_STATE_SYNCHRONIZING	4
130#define	G_MIRROR_DISK_STATE_DISCONNECTED	5
131#define	G_MIRROR_DISK_STATE_DESTROY		6
132struct g_mirror_disk {
133	uint32_t	 d_id;		/* Disk ID. */
134	struct g_consumer *d_consumer;	/* Consumer. */
135	struct g_mirror_softc	*d_softc; /* Back-pointer to softc. */
136	int		 d_state;	/* Disk state. */
137	u_int		 d_priority;	/* Disk priority. */
138	u_int		 load;		/* Averaged queue length */
139	off_t		 d_last_offset;	/* Last read offset */
140	uint64_t	 d_flags;	/* Additional flags. */
141	u_int		 d_genid;	/* Disk's generation ID. */
142	struct g_mirror_disk_sync d_sync;/* Sync information. */
143	LIST_ENTRY(g_mirror_disk) d_next;
144};
145#define	d_name	d_consumer->provider->name
146
147#define	G_MIRROR_EVENT_DONTWAIT	0x1
148#define	G_MIRROR_EVENT_WAIT	0x2
149#define	G_MIRROR_EVENT_DEVICE	0x4
150#define	G_MIRROR_EVENT_DONE	0x8
151struct g_mirror_event {
152	struct g_mirror_disk	*e_disk;
153	int			 e_state;
154	int			 e_flags;
155	int			 e_error;
156	TAILQ_ENTRY(g_mirror_event) e_next;
157};
158
159#define	G_MIRROR_DEVICE_FLAG_DESTROY	0x0100000000000000ULL
160#define	G_MIRROR_DEVICE_FLAG_WAIT	0x0200000000000000ULL
161#define	G_MIRROR_DEVICE_FLAG_DESTROYING	0x0400000000000000ULL
162#define	G_MIRROR_DEVICE_FLAG_TASTING	0x0800000000000000ULL
163#define	G_MIRROR_DEVICE_FLAG_WIPE	0x1000000000000000ULL
164
165#define	G_MIRROR_DEVICE_STATE_STARTING		0
166#define	G_MIRROR_DEVICE_STATE_RUNNING		1
167
168/* Bump syncid on first write. */
169#define	G_MIRROR_BUMP_SYNCID	0x1
170/* Bump genid immediately. */
171#define	G_MIRROR_BUMP_GENID	0x2
172struct g_mirror_softc {
173	u_int		sc_state;	/* Device state. */
174	uint32_t	sc_slice;	/* Slice size. */
175	uint8_t		sc_balance;	/* Balance algorithm. */
176	uint64_t	sc_mediasize;	/* Device size. */
177	uint32_t	sc_sectorsize;	/* Sector size. */
178	uint64_t	sc_flags;	/* Additional flags. */
179
180	struct g_geom	*sc_geom;
181	struct g_provider *sc_provider;
182
183	uint32_t	sc_id;		/* Mirror unique ID. */
184
185	struct sx	 sc_lock;
186	struct bio_queue_head sc_queue;
187	struct mtx	 sc_queue_mtx;
188	struct proc	*sc_worker;
189	struct bio_queue_head sc_regular_delayed; /* Delayed I/O requests due
190						     collision with sync
191						     requests. */
192	struct bio_queue_head sc_inflight; /* In-flight regular write
193					      requests. */
194	struct bio_queue_head sc_sync_delayed; /* Delayed sync requests due
195						  collision with regular
196						  requests. */
197
198	LIST_HEAD(, g_mirror_disk) sc_disks;
199	u_int		sc_ndisks;	/* Number of disks. */
200	struct g_mirror_disk *sc_hint;
201
202	u_int		sc_genid;	/* Generation ID. */
203	u_int		sc_syncid;	/* Synchronization ID. */
204	int		sc_bump_id;
205	struct g_mirror_device_sync sc_sync;
206	int		sc_idle;	/* DIRTY flags removed. */
207	time_t		sc_last_write;
208	u_int		sc_writes;
209
210	TAILQ_HEAD(, g_mirror_event) sc_events;
211	struct mtx	sc_events_mtx;
212
213	struct callout	sc_callout;
214
215	struct root_hold_token *sc_rootmount;
216};
217#define	sc_name	sc_geom->name
218
219u_int g_mirror_ndisks(struct g_mirror_softc *sc, int state);
220#define	G_MIRROR_DESTROY_SOFT		0
221#define	G_MIRROR_DESTROY_DELAYED	1
222#define	G_MIRROR_DESTROY_HARD		2
223int g_mirror_destroy(struct g_mirror_softc *sc, int how);
224int g_mirror_event_send(void *arg, int state, int flags);
225struct g_mirror_metadata;
226int g_mirror_add_disk(struct g_mirror_softc *sc, struct g_provider *pp,
227    struct g_mirror_metadata *md);
228int g_mirror_read_metadata(struct g_consumer *cp, struct g_mirror_metadata *md);
229void g_mirror_fill_metadata(struct g_mirror_softc *sc,
230    struct g_mirror_disk *disk, struct g_mirror_metadata *md);
231void g_mirror_update_metadata(struct g_mirror_disk *disk);
232
233g_ctl_req_t g_mirror_config;
234#endif	/* _KERNEL */
235
236struct g_mirror_metadata {
237	char		md_magic[16];	/* Magic value. */
238	uint32_t	md_version;	/* Version number. */
239	char		md_name[16];	/* Mirror name. */
240	uint32_t	md_mid;		/* Mirror unique ID. */
241	uint32_t	md_did;		/* Disk unique ID. */
242	uint8_t		md_all;		/* Number of disks in mirror. */
243	uint32_t	md_genid;	/* Generation ID. */
244	uint32_t	md_syncid;	/* Synchronization ID. */
245	uint8_t		md_priority;	/* Disk priority. */
246	uint32_t	md_slice;	/* Slice size. */
247	uint8_t		md_balance;	/* Balance type. */
248	uint64_t	md_mediasize;	/* Size of the smallest
249					   disk in mirror. */
250	uint32_t	md_sectorsize;	/* Sector size. */
251	uint64_t	md_sync_offset;	/* Synchronized offset. */
252	uint64_t	md_mflags;	/* Additional mirror flags. */
253	uint64_t	md_dflags;	/* Additional disk flags. */
254	char		md_provider[16]; /* Hardcoded provider. */
255	uint64_t	md_provsize;	/* Provider's size. */
256	u_char		md_hash[16];	/* MD5 hash. */
257};
258static __inline void
259mirror_metadata_encode(struct g_mirror_metadata *md, u_char *data)
260{
261	MD5_CTX ctx;
262
263	bcopy(md->md_magic, data, 16);
264	le32enc(data + 16, md->md_version);
265	bcopy(md->md_name, data + 20, 16);
266	le32enc(data + 36, md->md_mid);
267	le32enc(data + 40, md->md_did);
268	*(data + 44) = md->md_all;
269	le32enc(data + 45, md->md_genid);
270	le32enc(data + 49, md->md_syncid);
271	*(data + 53) = md->md_priority;
272	le32enc(data + 54, md->md_slice);
273	*(data + 58) = md->md_balance;
274	le64enc(data + 59, md->md_mediasize);
275	le32enc(data + 67, md->md_sectorsize);
276	le64enc(data + 71, md->md_sync_offset);
277	le64enc(data + 79, md->md_mflags);
278	le64enc(data + 87, md->md_dflags);
279	bcopy(md->md_provider, data + 95, 16);
280	le64enc(data + 111, md->md_provsize);
281	MD5Init(&ctx);
282	MD5Update(&ctx, data, 119);
283	MD5Final(md->md_hash, &ctx);
284	bcopy(md->md_hash, data + 119, 16);
285}
286static __inline int
287mirror_metadata_decode_v0v1(const u_char *data, struct g_mirror_metadata *md)
288{
289	MD5_CTX ctx;
290
291	bcopy(data + 20, md->md_name, 16);
292	md->md_mid = le32dec(data + 36);
293	md->md_did = le32dec(data + 40);
294	md->md_all = *(data + 44);
295	md->md_syncid = le32dec(data + 45);
296	md->md_priority = *(data + 49);
297	md->md_slice = le32dec(data + 50);
298	md->md_balance = *(data + 54);
299	md->md_mediasize = le64dec(data + 55);
300	md->md_sectorsize = le32dec(data + 63);
301	md->md_sync_offset = le64dec(data + 67);
302	md->md_mflags = le64dec(data + 75);
303	md->md_dflags = le64dec(data + 83);
304	bcopy(data + 91, md->md_provider, 16);
305	bcopy(data + 107, md->md_hash, 16);
306	MD5Init(&ctx);
307	MD5Update(&ctx, data, 107);
308	MD5Final(md->md_hash, &ctx);
309	if (bcmp(md->md_hash, data + 107, 16) != 0)
310		return (EINVAL);
311
312	/* New fields. */
313	md->md_genid = 0;
314	md->md_provsize = 0;
315
316	return (0);
317}
318static __inline int
319mirror_metadata_decode_v2(const u_char *data, struct g_mirror_metadata *md)
320{
321	MD5_CTX ctx;
322
323	bcopy(data + 20, md->md_name, 16);
324	md->md_mid = le32dec(data + 36);
325	md->md_did = le32dec(data + 40);
326	md->md_all = *(data + 44);
327	md->md_genid = le32dec(data + 45);
328	md->md_syncid = le32dec(data + 49);
329	md->md_priority = *(data + 53);
330	md->md_slice = le32dec(data + 54);
331	md->md_balance = *(data + 58);
332	md->md_mediasize = le64dec(data + 59);
333	md->md_sectorsize = le32dec(data + 67);
334	md->md_sync_offset = le64dec(data + 71);
335	md->md_mflags = le64dec(data + 79);
336	md->md_dflags = le64dec(data + 87);
337	bcopy(data + 95, md->md_provider, 16);
338	bcopy(data + 111, md->md_hash, 16);
339	MD5Init(&ctx);
340	MD5Update(&ctx, data, 111);
341	MD5Final(md->md_hash, &ctx);
342	if (bcmp(md->md_hash, data + 111, 16) != 0)
343		return (EINVAL);
344
345	/* New fields. */
346	md->md_provsize = 0;
347
348	return (0);
349}
350static __inline int
351mirror_metadata_decode_v3v4(const u_char *data, struct g_mirror_metadata *md)
352{
353	MD5_CTX ctx;
354
355	bcopy(data + 20, md->md_name, 16);
356	md->md_mid = le32dec(data + 36);
357	md->md_did = le32dec(data + 40);
358	md->md_all = *(data + 44);
359	md->md_genid = le32dec(data + 45);
360	md->md_syncid = le32dec(data + 49);
361	md->md_priority = *(data + 53);
362	md->md_slice = le32dec(data + 54);
363	md->md_balance = *(data + 58);
364	md->md_mediasize = le64dec(data + 59);
365	md->md_sectorsize = le32dec(data + 67);
366	md->md_sync_offset = le64dec(data + 71);
367	md->md_mflags = le64dec(data + 79);
368	md->md_dflags = le64dec(data + 87);
369	bcopy(data + 95, md->md_provider, 16);
370	md->md_provsize = le64dec(data + 111);
371	bcopy(data + 119, md->md_hash, 16);
372	MD5Init(&ctx);
373	MD5Update(&ctx, data, 119);
374	MD5Final(md->md_hash, &ctx);
375	if (bcmp(md->md_hash, data + 119, 16) != 0)
376		return (EINVAL);
377	return (0);
378}
379static __inline int
380mirror_metadata_decode(const u_char *data, struct g_mirror_metadata *md)
381{
382	int error;
383
384	bcopy(data, md->md_magic, 16);
385	md->md_version = le32dec(data + 16);
386	switch (md->md_version) {
387	case 0:
388	case 1:
389		error = mirror_metadata_decode_v0v1(data, md);
390		break;
391	case 2:
392		error = mirror_metadata_decode_v2(data, md);
393		break;
394	case 3:
395	case 4:
396		error = mirror_metadata_decode_v3v4(data, md);
397		break;
398	default:
399		error = EINVAL;
400		break;
401	}
402	return (error);
403}
404
405static __inline const char *
406balance_name(u_int balance)
407{
408	static const char *algorithms[] = {
409		[G_MIRROR_BALANCE_NONE] = "none",
410		[G_MIRROR_BALANCE_ROUND_ROBIN] = "round-robin",
411		[G_MIRROR_BALANCE_LOAD] = "load",
412		[G_MIRROR_BALANCE_SPLIT] = "split",
413		[G_MIRROR_BALANCE_PREFER] = "prefer",
414		[G_MIRROR_BALANCE_MAX + 1] = "unknown"
415	};
416
417	if (balance > G_MIRROR_BALANCE_MAX)
418		balance = G_MIRROR_BALANCE_MAX + 1;
419
420	return (algorithms[balance]);
421}
422
423static __inline int
424balance_id(const char *name)
425{
426	static const char *algorithms[] = {
427		[G_MIRROR_BALANCE_NONE] = "none",
428		[G_MIRROR_BALANCE_ROUND_ROBIN] = "round-robin",
429		[G_MIRROR_BALANCE_LOAD] = "load",
430		[G_MIRROR_BALANCE_SPLIT] = "split",
431		[G_MIRROR_BALANCE_PREFER] = "prefer"
432	};
433	int n;
434
435	for (n = G_MIRROR_BALANCE_MIN; n <= G_MIRROR_BALANCE_MAX; n++) {
436		if (strcmp(name, algorithms[n]) == 0)
437			return (n);
438	}
439	return (-1);
440}
441
442static __inline void
443mirror_metadata_dump(const struct g_mirror_metadata *md)
444{
445	static const char hex[] = "0123456789abcdef";
446	char hash[16 * 2 + 1];
447	u_int i;
448
449	printf("     magic: %s\n", md->md_magic);
450	printf("   version: %u\n", (u_int)md->md_version);
451	printf("      name: %s\n", md->md_name);
452	printf("       mid: %u\n", (u_int)md->md_mid);
453	printf("       did: %u\n", (u_int)md->md_did);
454	printf("       all: %u\n", (u_int)md->md_all);
455	printf("     genid: %u\n", (u_int)md->md_genid);
456	printf("    syncid: %u\n", (u_int)md->md_syncid);
457	printf("  priority: %u\n", (u_int)md->md_priority);
458	printf("     slice: %u\n", (u_int)md->md_slice);
459	printf("   balance: %s\n", balance_name((u_int)md->md_balance));
460	printf(" mediasize: %jd\n", (intmax_t)md->md_mediasize);
461	printf("sectorsize: %u\n", (u_int)md->md_sectorsize);
462	printf("syncoffset: %jd\n", (intmax_t)md->md_sync_offset);
463	printf("    mflags:");
464	if (md->md_mflags == 0)
465		printf(" NONE");
466	else {
467		if ((md->md_mflags & G_MIRROR_DEVICE_FLAG_NOFAILSYNC) != 0)
468			printf(" NOFAILSYNC");
469		if ((md->md_mflags & G_MIRROR_DEVICE_FLAG_NOAUTOSYNC) != 0)
470			printf(" NOAUTOSYNC");
471	}
472	printf("\n");
473	printf("    dflags:");
474	if (md->md_dflags == 0)
475		printf(" NONE");
476	else {
477		if ((md->md_dflags & G_MIRROR_DISK_FLAG_DIRTY) != 0)
478			printf(" DIRTY");
479		if ((md->md_dflags & G_MIRROR_DISK_FLAG_SYNCHRONIZING) != 0)
480			printf(" SYNCHRONIZING");
481		if ((md->md_dflags & G_MIRROR_DISK_FLAG_FORCE_SYNC) != 0)
482			printf(" FORCE_SYNC");
483		if ((md->md_dflags & G_MIRROR_DISK_FLAG_INACTIVE) != 0)
484			printf(" INACTIVE");
485	}
486	printf("\n");
487	printf("hcprovider: %s\n", md->md_provider);
488	printf("  provsize: %ju\n", (uintmax_t)md->md_provsize);
489	bzero(hash, sizeof(hash));
490	for (i = 0; i < 16; i++) {
491		hash[i * 2] = hex[md->md_hash[i] >> 4];
492		hash[i * 2 + 1] = hex[md->md_hash[i] & 0x0f];
493	}
494	printf("  MD5 hash: %s\n", hash);
495}
496#endif	/* !_G_MIRROR_H_ */
497