src/common/utils/user_util.cc

   1 // Copyright (c) 2016 Samsung Electronics Co., Ltd All Rights Reserved
   2 // Use of this source code is governed by an apache-2.0 license that can be
   3 // found in the LICENSE file.
   4
   5 #include "common/utils/user_util.h"
   6
   7 #include <manifest_parser/utils/logging.h>
   8
   9 #include <boost/filesystem/path.hpp>
  10
  11 #include <glib.h>
  12 #include <grp.h>
  13 #include <gum/gum-user.h>
  14 #include <gum/gum-user-service.h>
  15 #include <gum/common/gum-user-types.h>
  16
  17 #include <vector>
  18
  19 #include "common/utils/glist_range.h"
  20
  21 namespace bf = boost::filesystem;
  22
  23 namespace {
  24
  25 const int32_t kPWBufSize = sysconf(_SC_GETPW_R_SIZE_MAX);
  26 const int32_t kGRBufSize = sysconf(_SC_GETGR_R_SIZE_MAX);
  27
  28 const int kRetryMax = 5;
  29 const int kRetryDelay = 1000000 / 2;
  30
  31 }
  32
  33 namespace common_installer {
  34
  35 UserList GetUserList() {
  36   gchar** user_type_strv = gum_user_type_to_strv(
  37       GUM_USERTYPE_ADMIN | GUM_USERTYPE_GUEST | GUM_USERTYPE_NORMAL);
  38   GumUserService* service = nullptr;
  39   GumUserList* gum_user_list = nullptr;
  40   int count = 0;
  41   bool is_offline = getuid() == 0;
  42   // FIXME: Temporary retry logic, this should be removed
  43   while (count < kRetryMax) {
  44     service = gum_user_service_create_sync(is_offline ? TRUE : FALSE);
  45     if (service == nullptr) {
  46       if (is_offline) {
  47         LOG(WARNING) << "Failed to create gum user service!";
  48         break;
  49       }
  50       count++;
  51       LOG(WARNING) << "Failed to create gum user service! ("
  52                    << count << "/" << kRetryMax << ")";
  53       usleep(kRetryDelay);
  54       continue;
  55     }
  56
  57     gum_user_list =
  58         gum_user_service_get_user_list_sync(service, user_type_strv);
  59     g_object_unref(service);
  60     if (gum_user_list == nullptr) {
  61       if (is_offline) {
  62         LOG(WARNING) << "Failed to get gum user list!";
  63         break;
  64       }
  65       count++;
  66       LOG(WARNING) << "Failed to get gum user list! ("
  67                    << count << "/" << kRetryMax << ")";
  68       usleep(kRetryDelay);
  69       continue;
  70     } else {
  71       break;
  72     }
  73   }
  74   if (gum_user_list == nullptr)
  75     return {};
  76
  77   UserList list;
  78   for (GumUser* guser : GListRange<GumUser*>(gum_user_list)) {
  79     uid_t uid;
  80     g_object_get(G_OBJECT(guser), "uid", &uid, nullptr);
  81     gid_t gid;
  82     g_object_get(G_OBJECT(guser), "gid", &gid, nullptr);
  83     gchar* homedir = nullptr;
  84     g_object_get(G_OBJECT(guser), "homedir", &homedir, nullptr);
  85     if (homedir == nullptr) {
  86       LOG(WARNING) << "No homedir for uid: " << uid;
  87       continue;
  88     }
  89     list.emplace_back(uid, gid, bf::path(homedir));
  90     g_free(homedir);
  91   }
  92   g_strfreev(user_type_strv);
  93   gum_user_service_list_free(gum_user_list);
  94   return list;
  95 }
  96
  97 bool IsAdminUser(uid_t uid) {
  98   GumUser* guser;
  99   int count = 0;
 100   bool is_offline = getuid() == 0;
 101   do {
 102     guser = gum_user_get_sync(uid, is_offline ? TRUE : FALSE);
 103     if (guser)
 104       break;
 105     if (is_offline) {
 106       LOG(WARNING) << "Failed to get gum user!";
 107       break;
 108     }
 109     count++;
 110     LOG(WARNING) << "Failed to get gum user! ("
 111                  << count << "/" << kRetryMax << ")";
 112     usleep(kRetryDelay);
 113   } while (count < kRetryMax);
 114   if (!guser)
 115     return false;
 116   GumUserType ut;
 117   g_object_get(G_OBJECT(guser), "usertype", &ut, NULL);
 118   bool is_admin;
 119   if (ut == GUM_USERTYPE_ADMIN)
 120     is_admin = true;
 121   else
 122     is_admin = false;
 123   g_object_unref(guser);
 124   return is_admin;
 125 }
 126
 127 boost::optional<gid_t> GetGidByGroupName(const char* groupname) {
 128   char buf[kGRBufSize];
 129   struct group entry;
 130   struct group* ge;
 131   int ret = getgrnam_r(groupname, &entry, buf, sizeof(buf), &ge);
 132   if (ret || ge == nullptr)
 133     return {};
 134   return entry.gr_gid;
 135 }
 136
 137 std::string GetUsernameByUid(uid_t user) {
 138   struct passwd pwd;
 139   struct passwd* pwd_result;
 140   char buf[kPWBufSize];
 141   int ret = getpwuid_r(user, &pwd, buf, sizeof(buf), &pwd_result);
 142   if (ret != 0 || pwd_result == nullptr)
 143     return {};
 144   return pwd.pw_name;
 145 }
 146
 147 boost::optional<uid_t> GetUidByUserName(const char* username) {
 148   struct passwd pwd;
 149   struct passwd* pwd_result;
 150   char buf[kPWBufSize];
 151   int ret = getpwnam_r(username, &pwd, buf, sizeof(buf), &pwd_result);
 152   if (ret != 0 || pwd_result == nullptr)
 153     return {};
 154   return pwd.pw_uid;
 155 }
 156
 157 boost::optional<gid_t> GetGidByUid(uid_t uid) {
 158   struct passwd pwd;
 159   struct passwd* pwd_result;
 160   char buf[kPWBufSize];
 161   int ret = getpwuid_r(uid, &pwd, buf, sizeof(buf), &pwd_result);
 162   if (ret != 0 || pwd_result == nullptr)
 163     return {};
 164   return pwd.pw_gid;
 165 }
 166
 167 std::string GetGroupNameByGid(gid_t gid) {
 168   char buf[kGRBufSize];
 169   struct group entry;
 170   struct group* ge;
 171   int ret = getgrgid_r(gid, &entry, buf, sizeof(buf), &ge);
 172   if (ret || ge == nullptr) {
 173     return {};
 174   }
 175   return entry.gr_name;
 176 }
 177
 178 }  // namespace common_installer