git.sur5r.net Git - bacula/bacula/blob - bacula/src/lib/guid_to_name.c

   1 /*
   2    Bacula® - The Network Backup Solution
   3
   4    Copyright (C) 2007 Kern Sibbald
   5
   6    The main author of Bacula is Kern Sibbald, with contributions from
   7    many others, a complete list can be found in the file AUTHORS.
   8    This program is Free Software; you can redistribute it and/or
   9    modify it under the terms of version two of the GNU General Public
  10    License as published by the Free Software Foundation, which is
  11    listed in the file LICENSE.
  12
  13    This program is distributed in the hope that it will be useful, but
  14    WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16    General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  21    02110-1301, USA.
  22
  23    Bacula® is a registered trademark of Kern Sibbald.
  24    The licensor of Bacula is the Free Software Foundation Europe
  25    (FSFE), Fiduciary Program, Sumatrastrasse 25, 8006 Zürich,
  26    Switzerland, email:ftf@fsfeurope.org.
  27 */
  28 /*
  29  * Written by Kern Sibbald, July 2007 to replace idcache.c
  30  *
  31  *  Program to convert uid and gid into names, and cache the results
  32  *   for preformance reasons.
  33  *
  34  *  Version $Id$
  35  */
  36
  37 #include "bacula.h"
  38
  39 #ifndef WIN32
  40 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
  41 #endif
  42
  43 struct guitem {
  44    dlink link;
  45    char *name;
  46    union {
  47       uid_t uid;
  48       gid_t gid;
  49    };
  50 };
  51
  52
  53 guid_list *new_guid_list()
  54 {
  55    guid_list *list;
  56    guitem *item = NULL;
  57    list = (guid_list *)malloc(sizeof(guid_list));
  58    list->uid_list = New(dlist(item, &item->link));
  59    list->gid_list = New(dlist(item, &item->link));
  60    return list;
  61 }
  62
  63 void free_guid_list(guid_list *list)
  64 {
  65    guitem *item;
  66    foreach_dlist(item, list->uid_list) {
  67       free(item->name);
  68    }
  69    foreach_dlist(item, list->gid_list) {
  70       free(item->name);
  71    }
  72    delete list->uid_list;
  73    delete list->gid_list;
  74    free(list);
  75 }
  76
  77 static int uid_compare(void *item1, void *item2)
  78 {
  79    guitem *i1 = (guitem *)item1;
  80    guitem *i2 = (guitem *)item2;
  81    if (i1->uid < i2->uid) {
  82       return -1;
  83    } else if (i1->uid > i2->uid) {
  84       return 1;
  85    } else {
  86       return 0;
  87    }
  88 }
  89
  90 static int gid_compare(void *item1, void *item2)
  91 {
  92    guitem *i1 = (guitem *)item1;
  93    guitem *i2 = (guitem *)item2;
  94    if (i1->gid < i2->gid) {
  95       return -1;
  96    } else if (i1->gid > i2->gid) {
  97       return 1;
  98    } else {
  99       return 0;
 100    }
 101 }
 102
 103
 104 static void get_uidname(uid_t uid, guitem *item)
 105 {
 106 #ifndef HAVE_WIN32
 107    struct passwd *pwbuf;
 108    P(mutex);
 109    pwbuf = getpwuid(uid);
 110    if (pwbuf != NULL && strcmp(pwbuf->pw_name, "????????") != 0) {
 111       item->name = bstrdup(pwbuf->pw_name);
 112    }
 113    V(mutex);
 114 #endif
 115 }
 116
 117 static void get_gidname(gid_t gid, guitem *item)
 118 {
 119 #ifndef HAVE_WIN32
 120    struct group *grbuf;
 121    P(mutex);
 122    grbuf = getgrgid(gid);
 123    if (grbuf != NULL && strcmp(grbuf->gr_name, "????????") != 0) {
 124       item->name = bstrdup(grbuf->gr_name);
 125    }
 126    V(mutex);
 127 #endif
 128 }
 129
 130
 131 char *guid_list::uid_to_name(uid_t uid, char *name, int maxlen)
 132 {
 133    guitem sitem, *item, *fitem;
 134    sitem.uid = uid;
 135    char buf[50];
 136
 137    item = (guitem *)uid_list->binary_search(&sitem, uid_compare);
 138    Dmsg2(900, "uid=%d item=%p\n", uid, item);
 139    if (!item) {
 140       item = (guitem *)malloc(sizeof(guitem));
 141       item->uid = uid;
 142       item->name = NULL;
 143       get_uidname(uid, item);
 144       if (!item->name) {
 145          item->name = bstrdup(edit_int64(uid, buf));
 146          Dmsg2(900, "set uid=%d name=%s\n", uid, item->name);
 147       }
 148       fitem = (guitem *)uid_list->binary_insert(item, uid_compare);
 149       if (fitem != item) {               /* item already there this shouldn't happen */
 150          free(item->name);
 151          free(item);
 152          item = fitem;
 153       }
 154    }
 155    bstrncpy(name, item->name, maxlen);
 156    return name;
 157 }
 158
 159 char *guid_list::gid_to_name(gid_t gid, char *name, int maxlen)
 160 {
 161    guitem sitem, *item, *fitem;
 162    sitem.gid = gid;
 163    char buf[50];
 164
 165    item = (guitem *)gid_list->binary_search(&sitem, gid_compare);
 166    if (!item) {
 167       item = (guitem *)malloc(sizeof(guitem));
 168       item->gid = gid;
 169       item->name = NULL;
 170       get_gidname(gid, item);
 171       if (!item->name) {
 172          item->name = bstrdup(edit_int64(gid, buf));
 173       }
 174       fitem = (guitem *)gid_list->binary_insert(item, gid_compare);
 175       if (fitem != item) {               /* item already there this shouldn't happen */
 176          free(item->name);
 177          free(item);
 178          item = fitem;
 179       }
 180    }
 181
 182    bstrncpy(name, item->name, maxlen);
 183    return name;
 184 }
 185
 186 #ifdef TEST_PROGRAM
 187
 188 int main()
 189 {
 190    int i;
 191    guid_list *list;
 192    char ed1[50], ed2[50];
 193    list = new_guid_list();
 194    for (i=0; i<1001; i++) {
 195       printf("uid=%d name=%s  gid=%d name=%s\n", i, list->uid_to_name(i, ed1, sizeof(ed1)),
 196          i, list->gid_to_name(i, ed2, sizeof(ed2)));
 197       printf("uid=%d name=%s  gid=%d name=%s\n", i, list->uid_to_name(i, ed1, sizeof(ed1)),
 198          i, list->gid_to_name(i, ed2, sizeof(ed2)));
 199    }
 200
 201    free_guid_list(list);
 202    sm_dump(false);
 203
 204    return 0;
 205 }
 206
 207 #endif