extern int preserve_gid;
extern int preserve_acls;
extern int numeric_ids;
+extern gid_t our_gid;
extern char *usermap;
extern char *groupmap;
# endif
#endif
-#define GID_NONE ((gid_t)-1)
-
#define NFLAGS_WILD_NAME_MATCH (1<<0)
#define NFLAGS_NAME_MATCH (1<<1)
+union name_or_id {
+ const char *name;
+ id_t max_id;
+};
+
struct idlist {
struct idlist *next;
- const char *name;
+ union name_or_id u;
id_t id, id2;
uint16 flags;
};
static struct idlist *uidlist, *uidmap;
static struct idlist *gidlist, *gidmap;
-static struct idlist *add_to_list(struct idlist **root, id_t id, const char *name,
+static id_t id_parse(const char *num_str)
+{
+ id_t tmp, num = 0;
+ const char *cp = num_str;
+
+ while (*cp) {
+ if (!isDigit(cp)) {
+ invalid_num:
+ rprintf(FERROR, "Invalid ID number: %s\n", num_str);
+ exit_cleanup(RERR_SYNTAX);
+ }
+ tmp = num * 10 + *cp++ - '0';
+ if (tmp < num)
+ goto invalid_num;
+ num = tmp;
+ }
+
+ return num;
+}
+
+static struct idlist *add_to_list(struct idlist **root, id_t id, union name_or_id noiu,
id_t id2, uint16 flags)
{
struct idlist *node = new(struct idlist);
if (!node)
out_of_memory("add_to_list");
node->next = *root;
- node->name = name;
+ node->u = noiu;
node->id = id;
node->id2 = id2;
node->flags = flags;
}
/* turn a uid into a user name */
-static const char *uid_to_name(uid_t uid)
+char *uid_to_user(uid_t uid)
{
struct passwd *pass = getpwuid(uid);
if (pass)
}
/* turn a gid into a group name */
-static const char *gid_to_name(gid_t gid)
+char *gid_to_group(gid_t gid)
{
struct group *grp = getgrgid(gid);
if (grp)
return NULL;
}
+/* Parse a user name or (optionally) a number into a uid */
+int user_to_uid(const char *name, uid_t *uid_p, BOOL num_ok)
+{
+ struct passwd *pass;
+ if (!name || !*name)
+ return 0;
+ if (num_ok && name[strspn(name, "0123456789")] == '\0') {
+ *uid_p = id_parse(name);
+ return 1;
+ }
+ if (!(pass = getpwnam(name)))
+ return 0;
+ *uid_p = pass->pw_uid;
+ return 1;
+}
+
+/* Parse a group name or (optionally) a number into a gid */
+int group_to_gid(const char *name, gid_t *gid_p, BOOL num_ok)
+{
+ struct group *grp;
+ if (!name || !*name)
+ return 0;
+ if (num_ok && name[strspn(name, "0123456789")] == '\0') {
+ *gid_p = id_parse(name);
+ return 1;
+ }
+ if (!(grp = getgrnam(name)))
+ return 0;
+ *gid_p = grp->gr_gid;
+ return 1;
+}
+
static int is_in_group(gid_t gid)
{
#ifdef HAVE_GETGROUPS
- static gid_t last_in = GID_NONE, last_out;
- static int ngroups = -2;
+ static gid_t last_in;
+ static int ngroups = -2, last_out = -1;
static GETGROUPS_T *gidset;
int n;
- if (gid == last_in)
+ if (gid == last_in && last_out >= 0)
return last_out;
if (ngroups < -1) {
- gid_t mygid = MY_GID();
if ((ngroups = getgroups(0, NULL)) < 0)
ngroups = 0;
gidset = new_array(GETGROUPS_T, ngroups+1);
ngroups = getgroups(ngroups, gidset);
/* The default gid might not be in the list on some systems. */
for (n = 0; n < ngroups; n++) {
- if (gidset[n] == mygid)
+ if (gidset[n] == our_gid)
break;
}
if (n == ngroups)
- gidset[ngroups++] = mygid;
+ gidset[ngroups++] = our_gid;
if (DEBUG_GTE(OWN, 2)) {
int pos;
char *gidbuf = new_array(char, ngroups*21+32);
return last_out = 0;
#else
- static gid_t mygid = GID_NONE;
- if (mygid == GID_NONE) {
- mygid = MY_GID();
- if (DEBUG_GTE(OWN, 2))
- rprintf(FINFO, "process has gid %u\n", (unsigned)mygid);
- }
- return gid == mygid;
+ return gid == our_gid;
#endif
}
id_t id, const char *name)
{
struct idlist *node;
+ union name_or_id noiu;
int flag;
id_t id2;
+ noiu.name = name; /* ensure that add_to_list() gets the raw value. */
if (!name)
name = "";
for (node = idmap; node; node = node->next) {
if (node->flags & NFLAGS_WILD_NAME_MATCH) {
- if (!wildmatch(node->name, name))
+ if (!wildmatch(node->u.name, name))
continue;
} else if (node->flags & NFLAGS_NAME_MATCH) {
- if (strcmp(node->name, name) != 0)
+ if (strcmp(node->u.name, name) != 0)
continue;
- } else if (node->name) {
- if (id < node->id || id > (unsigned long)node->name)
+ } else if (node->u.max_id) {
+ if (id < node->id || id > node->u.max_id)
continue;
} else {
if (node->id != id)
id2 = id;
flag = idlist_ptr == &gidlist && !am_root && !is_in_group(id2) ? FLAG_SKIP_GROUP : 0;
- node = add_to_list(idlist_ptr, id, *name ? name : NULL, id2, flag);
+ node = add_to_list(idlist_ptr, id, noiu, id2, flag);
if (DEBUG_GTE(OWN, 2)) {
rprintf(FINFO, "%sid %u(%s) maps to %u\n",
/* this function is a definate candidate for a faster algorithm */
uid_t match_uid(uid_t uid)
{
- static uid_t last_in = -1, last_out = -1;
+ static struct idlist *last = NULL;
struct idlist *list;
- if (uid == last_in)
- return last_out;
-
- last_in = uid;
+ if (last && uid == last->id)
+ return last->id2;
for (list = uidlist; list; list = list->next) {
if (list->id == uid)
if (!list)
list = recv_add_id(&uidlist, uidmap, uid, NULL);
+ last = list;
- return last_out = list->id2;
+ return list->id2;
}
gid_t match_gid(gid_t gid, uint16 *flags_ptr)
{
struct idlist *list;
struct idlist *node;
+ union name_or_id noiu;
if (uid == 0) /* don't map root */
return NULL;
return NULL;
}
- node = add_to_list(&uidlist, uid, uid_to_name(uid), 0, 0);
- return node->name;
+ noiu.name = uid_to_user(uid);
+ node = add_to_list(&uidlist, uid, noiu, 0, 0);
+ return node->u.name;
}
/* Add a gid to the list of gids. Only called on sending side. */
{
struct idlist *list;
struct idlist *node;
+ union name_or_id noiu;
if (gid == 0) /* don't map root */
return NULL;
return NULL;
}
- node = add_to_list(&gidlist, gid, gid_to_name(gid), 0, 0);
- return node->name;
+ noiu.name = gid_to_group(gid);
+ node = add_to_list(&gidlist, gid, noiu, 0, 0);
+ return node->u.name;
}
/* send a complete uid/gid mapping to the peer */
int len;
/* we send sequences of uid/byte-length/name */
for (list = uidlist; list; list = list->next) {
- if (!list->name)
+ if (!list->u.name)
continue;
- len = strlen(list->name);
+ len = strlen(list->u.name);
write_varint30(f, list->id);
write_byte(f, len);
- write_buf(f, list->name, len);
+ write_buf(f, list->u.name, len);
}
/* terminate the uid list with a 0 uid. We explicitly exclude
if (preserve_gid || preserve_acls) {
int len;
for (list = gidlist; list; list = list->next) {
- if (!list->name)
+ if (!list->u.name)
continue;
- len = strlen(list->name);
+ len = strlen(list->u.name);
write_varint30(f, list->id);
write_byte(f, len);
- write_buf(f, list->name, len);
+ write_buf(f, list->u.name, len);
}
write_varint30(f, 0);
}
{
struct idlist **idmap_ptr = usernames ? &uidmap : &gidmap;
struct idlist **idlist_ptr = usernames ? &uidlist : &gidlist;
- char *colon, *end, *name, *cp = map + strlen(map);
+ char *colon, *cp = map + strlen(map);
+ union name_or_id noiu;
id_t id1;
uint16 flags;
/* Parse the list in reverse, so the order in the struct is right. */
while (1) {
- end = cp;
while (cp > map && cp[-1] != ',') cp--;
if (!(colon = strchr(cp, ':'))) {
rprintf(FERROR, "No colon found in --%smap: %s\n",
exit_cleanup(RERR_SYNTAX);
}
if (dash)
- name = (char *)atol(dash+1);
+ noiu.max_id = id_parse(dash+1);
else
- name = (char *)0;
+ noiu.max_id = 0;
flags = 0;
- id1 = atol(cp);
+ id1 = id_parse(cp);
} else if (strpbrk(cp, "*[?")) {
flags = NFLAGS_WILD_NAME_MATCH;
- name = cp;
+ noiu.name = cp;
id1 = 0;
} else {
flags = NFLAGS_NAME_MATCH;
- name = cp;
+ noiu.name = cp;
id1 = 0;
}
if (usernames) {
uid_t uid;
if (user_to_uid(colon+1, &uid, True))
- add_to_list(idmap_ptr, id1, name, uid, flags);
+ add_to_list(idmap_ptr, id1, noiu, uid, flags);
else {
rprintf(FERROR,
"Unknown --usermap name on receiver: %s\n",
} else {
gid_t gid;
if (group_to_gid(colon+1, &gid, True))
- add_to_list(idmap_ptr, id1, name, gid, flags);
+ add_to_list(idmap_ptr, id1, noiu, gid, flags);
else {
rprintf(FERROR,
"Unknown --groupmap name on receiver: %s\n",
/* The 0 user/group doesn't get its name sent, so add it explicitly. */
recv_add_id(idlist_ptr, *idmap_ptr, 0,
- numeric_ids ? NULL : usernames ? uid_to_name(0) : gid_to_name(0));
+ numeric_ids ? NULL : usernames ? uid_to_user(0) : gid_to_group(0));
+}
+
+#ifdef HAVE_GETGROUPLIST
+const char *getallgroups(uid_t uid, gid_t *gid_list, int *size_ptr)
+{
+ struct passwd *pw;
+ if ((pw = getpwuid(uid)) == NULL)
+ return "getpwuid failed";
+ /* Get all the process's groups, with the pw_gid group first. */
+ if (getgrouplist(pw->pw_name, pw->pw_gid, gid_list, size_ptr) < 0)
+ return "getgrouplist failed";
+ /* Paranoia: is the default group not first in the list? */
+ if (gid_list[0] != pw->pw_gid) {
+ int j;
+ for (j = 0; j < *size_ptr; j++) {
+ if (gid_list[j] == pw->pw_gid) {
+ gid_list[j] = gid_list[0];
+ gid_list[0] = pw->pw_gid;
+ break;
+ }
+ }
+ }
+ return NULL;
}
+#endif