xref: /netbsd-src/lib/libutil/getdiskrawname.c (revision eaac9e3d28859567590b7fef63a96c293f76dbf9)
1 /*	$NetBSD: getdiskrawname.c,v 1.2 2013/12/22 14:31:51 mlelstv Exp $	*/
2 
3 /*-
4  * Copyright (c) 2012 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Christos Zoulas.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29  * POSSIBILITY OF SUCH DAMAGE.
30  */
31 #include <sys/cdefs.h>
32 __RCSID("$NetBSD: getdiskrawname.c,v 1.2 2013/12/22 14:31:51 mlelstv Exp $");
33 
34 #include <sys/stat.h>
35 
36 #include <stdio.h>
37 #include <string.h>
38 #include <errno.h>
39 #include <util.h>
40 #include <unistd.h>
41 
42 const char *
43 getdiskrawname(char *buf, size_t bufsiz, const char *name)
44 {
45 	const char *dp = strrchr(name, '/');
46 	struct stat st;
47 	ssize_t len;
48 
49 	if (dp == NULL) {
50 		errno = EINVAL;
51 		return NULL;
52 	}
53 
54 	len = readlink(name, buf, bufsiz-1);
55 	if (len > 0) {
56 		buf[len] = '\0';
57 		name = buf;
58 	}
59 
60 	if (stat(name, &st) == -1)
61 		return NULL;
62 
63 	if (!S_ISBLK(st.st_mode)) {
64 		errno = EFTYPE;
65 		return NULL;
66 	}
67 
68 	(void)snprintf(buf, bufsiz, "%.*s/r%s", (int)(dp - name), name, dp + 1);
69 
70 	return buf;
71 }
72 
73 const char *
74 getdiskcookedname(char *buf, size_t bufsiz, const char *name)
75 {
76 	const char *dp;
77 	struct stat st;
78 	ssize_t len;
79 
80 	if ((dp = strrchr(name, '/')) == NULL) {
81 		errno = EINVAL;
82 		return NULL;
83 	}
84 
85 	len = readlink(name, buf, bufsiz-1);
86 	if (len > 0) {
87 		buf[len] = '\0';
88 		name = buf;
89 	}
90 
91 	if (stat(name, &st) == -1)
92 		return NULL;
93 
94 	if (!S_ISCHR(st.st_mode)) {
95 		errno = EFTYPE;
96 		return NULL;
97 	}
98 
99 	if (dp[1] != 'r') {
100 		errno = EINVAL;
101 		return NULL;
102 	}
103 
104 	(void)snprintf(buf, bufsiz, "%.*s/%s", (int)(dp - name), name, dp + 2);
105 
106 	return buf;
107 }
108