1/*-
2 * Copyright (c) 2016-2019 The DragonFly Project
3 * Copyright (c) 2016-2019 Tomohiro Kusumi <tkusumi@netbsd.org>
4 * All rights reserved.
5 *
6 * This software was developed by Edward Tomasz Napierala under sponsorship
7 * from the FreeBSD Foundation.
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 * 2. Redistributions in binary form must reproduce the above copyright
15 *    notice, this list of conditions and the following disclaimer in the
16 *    documentation and/or other materials provided with the distribution.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
19 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
22 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 * SUCH DAMAGE.
29 */
30
31#include <sys/cdefs.h>
32#include <stdio.h>
33#include <stdlib.h>
34#include <string.h>
35#include <err.h>
36#include <assert.h>
37
38#include <sys/types.h>
39
40#include "hammer_disk.h"
41
42#include "fstyp.h"
43
44extern int fsvtyp_hammer(const char *blkdevs, char *label, size_t size);
45extern int fsvtyp_hammer_partial(const char *blkdevs, char *label, size_t size);
46
47static hammer_volume_ondisk_t
48read_ondisk(FILE *fp)
49{
50	hammer_volume_ondisk_t ondisk;
51
52	ondisk = read_buf(fp, 0, sizeof(*ondisk));
53	if (ondisk == NULL)
54		err(1, "failed to read ondisk");
55
56	return (ondisk);
57}
58
59static int
60test_ondisk(const hammer_volume_ondisk_t ondisk)
61{
62	static int count = 0;
63	static hammer_uuid_t fsid, fstype;
64	static char label[64];
65
66	if (ondisk->vol_signature != HAMMER_FSBUF_VOLUME &&
67	    ondisk->vol_signature != HAMMER_FSBUF_VOLUME_REV)
68		return (1);
69	if (ondisk->vol_rootvol != HAMMER_ROOT_VOLNO)
70		return (2);
71	if (ondisk->vol_no < 0 || ondisk->vol_no > HAMMER_MAX_VOLUMES - 1)
72		return (3);
73	if (ondisk->vol_count < 1 || ondisk->vol_count > HAMMER_MAX_VOLUMES)
74		return (4);
75
76	if (count == 0) {
77		count = ondisk->vol_count;
78		assert(count != 0);
79		memcpy(&fsid, &ondisk->vol_fsid, sizeof(fsid));
80		memcpy(&fstype, &ondisk->vol_fstype, sizeof(fstype));
81		strlcpy(label, ondisk->vol_label, sizeof(label));
82	} else {
83		if (ondisk->vol_count != count)
84			return (5);
85		if (memcmp(&ondisk->vol_fsid, &fsid, sizeof(fsid)))
86			return (6);
87		if (memcmp(&ondisk->vol_fstype, &fstype, sizeof(fstype)))
88			return (7);
89		if (strcmp(ondisk->vol_label, label))
90			return (8);
91	}
92
93	return (0);
94}
95
96int
97fstyp_hammer(FILE *fp, char *label, size_t size)
98{
99	hammer_volume_ondisk_t ondisk;
100	int error = 1;
101
102	ondisk = read_ondisk(fp);
103	if (ondisk->vol_no != HAMMER_ROOT_VOLNO)
104		goto fail;
105	if (ondisk->vol_count != 1)
106		goto fail;
107	if (test_ondisk(ondisk))
108		goto fail;
109
110	strlcpy(label, ondisk->vol_label, size);
111	error = 0;
112fail:
113	free(ondisk);
114	return (error);
115}
116
117static int
118test_volume(const char *volpath)
119{
120	hammer_volume_ondisk_t ondisk;
121	FILE *fp;
122	int volno = -1;
123
124	if ((fp = fopen(volpath, "r")) == NULL)
125		err(1, "failed to open %s", volpath);
126
127	ondisk = read_ondisk(fp);
128	fclose(fp);
129	if (test_ondisk(ondisk))
130		goto fail;
131
132	volno = ondisk->vol_no;
133fail:
134	free(ondisk);
135	return (volno);
136}
137
138static int
139__fsvtyp_hammer(const char *blkdevs, char *label, size_t size, int partial)
140{
141	hammer_volume_ondisk_t ondisk = NULL;
142	FILE *fp;
143	char *dup, *p, *volpath, x[HAMMER_MAX_VOLUMES];
144	int i, volno, error = 1;
145
146	if (!blkdevs)
147		goto fail;
148
149	memset(x, 0, sizeof(x));
150	dup = strdup(blkdevs);
151	p = dup;
152
153	volpath = NULL;
154	volno = -1;
155	while (p) {
156		volpath = p;
157		if ((p = strchr(p, ':')) != NULL)
158			*p++ = '\0';
159		if ((volno = test_volume(volpath)) == -1)
160			break;
161		assert(volno >= 0);
162		assert(volno < HAMMER_MAX_VOLUMES);
163		x[volno]++;
164	}
165
166	if (!volpath)
167		err(1, "invalid path %s", blkdevs);
168	if ((fp = fopen(volpath, "r")) == NULL)
169		err(1, "failed to open %s", volpath);
170	ondisk = read_ondisk(fp);
171	fclose(fp);
172
173	free(dup);
174
175	if (volno == -1)
176		goto fail;
177	if (partial)
178		goto success;
179
180	for (i = 0; i < HAMMER_MAX_VOLUMES; i++)
181		if (x[i] > 1)
182			goto fail;
183	for (i = 0; i < HAMMER_MAX_VOLUMES; i++)
184		if (x[i] == 0)
185			break;
186	if (ondisk->vol_count != i)
187		goto fail;
188	for (; i < HAMMER_MAX_VOLUMES; i++)
189		if (x[i] != 0)
190			goto fail;
191success:
192	strlcpy(label, ondisk->vol_label, size);
193	error = 0;
194fail:
195	free(ondisk);
196	return (error);
197}
198
199int
200fsvtyp_hammer(const char *blkdevs, char *label, size_t size)
201{
202	return (__fsvtyp_hammer(blkdevs, label, size, 0));
203}
204
205int
206fsvtyp_hammer_partial(const char *blkdevs, char *label, size_t size)
207{
208	return (__fsvtyp_hammer(blkdevs, label, size, 1));
209}
210