2 * Handle the mapping of uid/gid and user/group names between systems.
4 * Copyright (C) 1996 Andrew Tridgell
5 * Copyright (C) 1996 Paul Mackerras
6 * Copyright (C) 2004, 2005, 2006 Wayne Davison
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, write to the Free Software Foundation, Inc.,
20 * 51 Franklin Street - Fifth Floor, Boston, MA 02110-1301, USA.
23 /* If the source username/group does not exist on the target then use
24 * the numeric IDs. Never do any mapping for uid=0 or gid=0 as these
31 # define GETGROUPS_T gid_t
37 extern int preserve_uid;
38 extern int preserve_gid;
39 extern int numeric_ids;
47 static struct idlist *uidlist;
48 static struct idlist *gidlist;
50 static struct idlist *add_to_list(struct idlist **root, int id, char *name,
53 struct idlist *node = new(struct idlist);
55 out_of_memory("add_to_list");
64 /* turn a uid into a user name */
65 static char *uid_to_name(uid_t uid)
67 struct passwd *pass = getpwuid(uid);
69 return strdup(pass->pw_name);
73 /* turn a gid into a group name */
74 static char *gid_to_name(gid_t gid)
76 struct group *grp = getgrgid(gid);
78 return strdup(grp->gr_name);
82 static uid_t map_uid(uid_t id, char *name)
85 if (id != 0 && name_to_uid(name, &uid))
90 static gid_t map_gid(gid_t id, char *name)
93 if (id != 0 && name_to_gid(name, &gid))
98 static int is_in_group(gid_t gid)
100 #ifdef HAVE_GETGROUPS
101 static gid_t last_in = GID_NONE, last_out;
102 static int ngroups = -2;
103 static GETGROUPS_T *gidset;
109 gid_t mygid = MY_GID();
110 if ((ngroups = getgroups(0, NULL)) < 0)
112 gidset = new_array(GETGROUPS_T, ngroups+1);
114 out_of_memory("is_in_group");
116 ngroups = getgroups(ngroups, gidset);
117 /* The default gid might not be in the list on some systems. */
118 for (n = 0; n < ngroups; n++) {
119 if (gidset[n] == mygid)
123 gidset[ngroups++] = mygid;
126 char *gidbuf = new_array(char, ngroups*21+32);
128 out_of_memory("is_in_group");
129 pos = snprintf(gidbuf, 32, "process has %d gid%s: ",
130 ngroups, ngroups == 1? "" : "s");
131 for (n = 0; n < ngroups; n++) {
132 pos += snprintf(gidbuf+pos, 21, " %d", (int)gidset[n]);
134 rprintf(FINFO, "%s\n", gidbuf);
140 for (n = 0; n < ngroups; n++) {
141 if (gidset[n] == gid)
147 static gid_t mygid = GID_NONE;
148 if (mygid == GID_NONE) {
151 rprintf(FINFO, "process has gid %d\n", (int)mygid);
157 /* Add a uid to the list of uids. Only called on receiving side. */
158 static uid_t recv_add_uid(uid_t id, char *name)
160 uid_t id2 = name ? map_uid(id, name) : id;
163 node = add_to_list(&uidlist, (int)id, name, (int)id2);
166 rprintf(FINFO, "uid %d(%s) maps to %d\n",
167 (int)id, name ? name : "", (int)id2);
173 /* Add a gid to the list of gids. Only called on receiving side. */
174 static gid_t recv_add_gid(gid_t id, char *name)
176 gid_t id2 = name ? map_gid(id, name) : id;
179 if (!am_root && !is_in_group(id2))
181 node = add_to_list(&gidlist, (int)id, name, (int)id2);
184 rprintf(FINFO, "gid %d(%s) maps to %d\n",
185 (int)id, name ? name : "", (int)id2);
191 /* this function is a definate candidate for a faster algorithm */
192 uid_t match_uid(uid_t uid)
194 static uid_t last_in, last_out;
205 for (list = uidlist; list; list = list->next) {
206 if (list->id == (int)uid)
207 return last_out = (uid_t)list->id2;
210 return last_out = uid;
213 gid_t match_gid(gid_t gid)
215 static gid_t last_in = GID_NONE, last_out = GID_NONE;
226 for (list = gidlist; list; list = list->next) {
227 if (list->id == (int)gid)
228 return last_out = (gid_t)list->id2;
231 return last_out = recv_add_gid(gid, NULL);
234 /* Add a uid to the list of uids. Only called on sending side. */
235 char *add_uid(uid_t uid)
240 if (uid == 0) /* don't map root */
243 for (list = uidlist; list; list = list->next) {
244 if (list->id == (int)uid)
248 node = add_to_list(&uidlist, (int)uid, uid_to_name(uid), 0);
252 /* Add a gid to the list of gids. Only called on sending side. */
253 char *add_gid(gid_t gid)
258 if (gid == 0) /* don't map root */
261 for (list = gidlist; list; list = list->next) {
262 if (list->id == (int)gid)
266 node = add_to_list(&gidlist, (int)gid, gid_to_name(gid), 0);
270 /* send a complete uid/gid mapping to the peer */
271 void send_uid_list(int f)
277 /* we send sequences of uid/byte-length/name */
278 for (list = uidlist; list; list = list->next) {
281 len = strlen(list->name);
282 write_int(f, list->id);
284 write_buf(f, list->name, len);
287 /* terminate the uid list with a 0 uid. We explicitly exclude
294 for (list = gidlist; list; list = list->next) {
297 len = strlen(list->name);
298 write_int(f, list->id);
300 write_buf(f, list->name, len);
306 uid_t recv_user_name(int f, uid_t uid)
308 int len = read_byte(f);
309 char *name = new_array(char, len+1);
311 out_of_memory("recv_user_name");
312 read_sbuf(f, name, len);
313 return recv_add_uid(uid, name); /* node keeps name's memory */
316 gid_t recv_group_name(int f, gid_t gid)
318 int len = read_byte(f);
319 char *name = new_array(char, len+1);
321 out_of_memory("recv_group_name");
322 read_sbuf(f, name, len);
323 return recv_add_gid(gid, name); /* node keeps name's memory */
326 /* recv a complete uid/gid mapping from the peer and map the uid/gid
327 * in the file list to local names */
328 void recv_uid_list(int f, struct file_list *flist)
332 if (preserve_uid && !numeric_ids) {
333 /* read the uid list */
334 while ((id = read_int(f)) != 0)
335 recv_user_name(f, (uid_t)id);
338 if (preserve_gid && !numeric_ids) {
339 /* read the gid list */
340 while ((id = read_int(f)) != 0)
341 recv_group_name(f, (gid_t)id);
344 /* Now convert all the uids/gids from sender values to our values. */
345 if (am_root && preserve_uid && !numeric_ids) {
346 for (i = 0; i < flist->count; i++)
347 F_OWNER(flist->files[i]) = match_uid(F_UID(flist->files[i]));
349 if (preserve_gid && (!am_root || !numeric_ids)) {
350 for (i = 0; i < flist->count; i++)
351 F_GROUP(flist->files[i]) = match_gid(F_GID(flist->files[i]));