xref: /freebsd/sys/contrib/openzfs/udev/zvol_id.c (revision 7a7741af18d6c8a804cc643cb7ecda9d730c6aa6)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or https://opensource.org/licenses/CDDL-1.0.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 /*
22  * Copyright (c) 2011, Fajar A. Nugraha.  All rights reserved.
23  * Use is subject to license terms.
24  */
25 
26 #include <ctype.h>
27 #include <errno.h>
28 #include <fcntl.h>
29 #include <stdio.h>
30 #include <string.h>
31 #include <unistd.h>
32 #include <sys/fs/zfs.h>
33 #include <sys/ioctl.h>
34 #include <sys/stat.h>
35 
36 #if defined(ZFS_ASAN_ENABLED)
37 /*
38  * zvol_id is invoked by udev with the help of ptrace()
39  * making sanitized binary with leak detection croak
40  * because of tracing mechanisms collision
41  */
42 extern const char *__asan_default_options(void);
43 
__asan_default_options(void)44 const char *__asan_default_options(void) {
45 	return ("abort_on_error=true:halt_on_error=true:"
46 		"allocator_may_return_null=true:disable_coredump=false:"
47 		"detect_stack_use_after_return=true:detect_leaks=false");
48 }
49 #endif
50 
51 int
main(int argc,const char * const * argv)52 main(int argc, const char *const *argv)
53 {
54 	if (argc != 2 || strncmp(argv[1], "/dev/zd", 7) != 0) {
55 		fprintf(stderr, "usage: %s /dev/zdX\n", argv[0]);
56 		return (1);
57 	}
58 	const char *dev_name = argv[1];
59 	size_t i, len;
60 
61 	int fd;
62 	struct stat sb;
63 	if ((fd = open(dev_name, O_RDONLY|O_CLOEXEC)) == -1 ||
64 	    fstat(fd, &sb) != 0) {
65 		fprintf(stderr, "%s: %s\n", dev_name, strerror(errno));
66 		return (1);
67 	}
68 
69 	char zvol_name[MAXNAMELEN + strlen("-part") + 10];
70 	if (ioctl(fd, BLKZNAME, zvol_name) == -1) {
71 		fprintf(stderr, "%s: BLKZNAME: %s\n",
72 		    dev_name, strerror(errno));
73 		return (1);
74 	}
75 
76 	const char *dev_part = strrchr(dev_name, 'p');
77 	len = strlen(zvol_name);
78 	if (dev_part != NULL) {
79 		sprintf(zvol_name + len, "-part%s", dev_part + 1);
80 		len = strlen(zvol_name);
81 	}
82 
83 	for (i = 0; i < len; ++i)
84 		if (isblank(zvol_name[i]))
85 			zvol_name[i] = '+';
86 
87 	puts(zvol_name);
88 
89 	return (0);
90 }
91