blob: 8388be0dadc97332f831a247d67b8eb52a85de52
1 | /* |
2 | Copyright (c) 2001-2006, Gerrit Pape |
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 are met: |
7 | |
8 | 1. Redistributions of source code must retain the above copyright notice, |
9 | 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 | 3. The name of the author may not be used to endorse or promote products |
14 | derived from this software without specific prior written permission. |
15 | |
16 | THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED |
17 | WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF |
18 | MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO |
19 | EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
20 | SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
21 | PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; |
22 | OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, |
23 | WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR |
24 | OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF |
25 | ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
26 | */ |
27 | |
28 | #include "libbb.h" |
29 | |
30 | /* Always sets uid and gid */ |
31 | int FAST_FUNC get_uidgid(struct bb_uidgid_t *u, const char *ug, int numeric_ok) |
32 | { |
33 | struct passwd *pwd; |
34 | struct group *gr; |
35 | char *user, *group; |
36 | unsigned n; |
37 | |
38 | user = (char*)ug; |
39 | group = strchr(ug, ':'); |
40 | if (group) { |
41 | int sz = (++group) - ug; |
42 | user = alloca(sz); |
43 | /* copies sz-1 bytes, stores terminating '\0' */ |
44 | safe_strncpy(user, ug, sz); |
45 | } |
46 | if (numeric_ok) { |
47 | n = bb_strtou(user, NULL, 10); |
48 | if (!errno) { |
49 | u->uid = n; |
50 | pwd = getpwuid(n); |
51 | /* If we have e.g. "500" string without user */ |
52 | /* with uid 500 in /etc/passwd, we set gid == uid */ |
53 | u->gid = pwd ? pwd->pw_gid : n; |
54 | goto skip; |
55 | } |
56 | } |
57 | /* Either it is not numeric, or caller disallows numeric username */ |
58 | pwd = getpwnam(user); |
59 | if (!pwd) |
60 | return 0; |
61 | u->uid = pwd->pw_uid; |
62 | u->gid = pwd->pw_gid; |
63 | |
64 | skip: |
65 | if (group) { |
66 | if (numeric_ok) { |
67 | n = bb_strtou(group, NULL, 10); |
68 | if (!errno) { |
69 | u->gid = n; |
70 | return 1; |
71 | } |
72 | } |
73 | gr = getgrnam(group); |
74 | if (!gr) |
75 | return 0; |
76 | u->gid = gr->gr_gid; |
77 | } |
78 | return 1; |
79 | } |
80 | void FAST_FUNC xget_uidgid(struct bb_uidgid_t *u, const char *ug) |
81 | { |
82 | if (!get_uidgid(u, ug, 1)) |
83 | bb_error_msg_and_die("unknown user/group %s", ug); |
84 | } |
85 | |
86 | /* chown-like: |
87 | * "user" sets uid only, |
88 | * ":group" sets gid only |
89 | * "user:" sets uid and gid (to user's primary group id) |
90 | * "user:group" sets uid and gid |
91 | * ('unset' uid or gid retains the value it has on entry) |
92 | */ |
93 | void FAST_FUNC parse_chown_usergroup_or_die(struct bb_uidgid_t *u, char *user_group) |
94 | { |
95 | char *group; |
96 | |
97 | /* Check if there is a group name */ |
98 | group = strchr(user_group, '.'); /* deprecated? */ |
99 | if (!group) |
100 | group = strchr(user_group, ':'); |
101 | else |
102 | *group = ':'; /* replace '.' with ':' */ |
103 | |
104 | /* Parse "user[:[group]]" */ |
105 | if (!group) { /* "user" */ |
106 | u->uid = get_ug_id(user_group, xuname2uid); |
107 | } else if (group == user_group) { /* ":group" */ |
108 | u->gid = get_ug_id(group + 1, xgroup2gid); |
109 | } else { |
110 | if (!group[1]) /* "user:" */ |
111 | *group = '\0'; |
112 | xget_uidgid(u, user_group); |
113 | } |
114 | } |
115 | |
116 | #if 0 |
117 | #include <stdio.h> |
118 | int main() |
119 | { |
120 | unsigned u; |
121 | struct bb_uidgid_t ug; |
122 | u = get_uidgid(&ug, "apache", 0); |
123 | printf("%u = %u:%u\n", u, ug.uid, ug.gid); |
124 | ug.uid = ug.gid = 1111; |
125 | u = get_uidgid(&ug, "apache", 0); |
126 | printf("%u = %u:%u\n", u, ug.uid, ug.gid); |
127 | ug.uid = ug.gid = 1111; |
128 | u = get_uidgid(&ug, "apache:users", 0); |
129 | printf("%u = %u:%u\n", u, ug.uid, ug.gid); |
130 | ug.uid = ug.gid = 1111; |
131 | u = get_uidgid(&ug, "apache:users", 0); |
132 | printf("%u = %u:%u\n", u, ug.uid, ug.gid); |
133 | return 0; |
134 | } |
135 | #endif |
136 |