kdecore/util/kuser_win.cpp

   1 /*
   2  *  KUser - represent a user/account (Windows)
   3  *  Copyright (C) 2007 Bernhard Loos <nhuh.put@web.de>
   4  *
   5  *  This library is free software; you can redistribute it and/or
   6  *  modify it under the terms of the GNU Library General Public
   7  *  License as published by the Free Software Foundation; either
   8  *  version 2 of the License, or (at your option) any later version.
   9  *
  10  *  This library is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  *  Library General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU Library General Public License
  16  *  along with this library; see the file COPYING.LIB.  If not, write to
  17  *  the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  18  *  Boston, MA 02110-1301, USA.
  19  */
  20
  21 #include "kuser.h"
  22
  23 #include <QtCore/QMutableStringListIterator>
  24 #include <QtCore/QDir>
  25
  26 #include <windows.h>
  27 #include <lm.h>
  28 #include <userenv.h>
  29
  30 class KUser::Private : public KShared
  31 {
  32     public:
  33         PUSER_INFO_11 userInfo;
  34         PSID sid;
  35
  36         Private() : userInfo(0), sid(0) {}
  37
  38         Private(PUSER_INFO_11 userInfo_, PSID sid_ = 0) : userInfo(userInfo_) {}
  39
  40         Private(const QString &name, PSID sid_ = 0) : userInfo(0), sid(0)
  41         {
  42             if (NetUserGetInfo(NULL, (LPCWSTR) name.utf16(), 11, (LPBYTE *) &userInfo) != NERR_Success)
  43                 goto error;
  44
  45             if (!sid_) {
  46                 DWORD size = 0;
  47                 SID_NAME_USE nameuse;
  48                 DWORD cchReferencedDomainName = 0;
  49
  50                 if (!LookupAccountNameW(NULL, (LPCWSTR) name.utf16(), NULL, &size, NULL, &cchReferencedDomainName, &nameuse))
  51                     goto error;
  52                 sid = (PSID) new BYTE[size];
  53                 if (!LookupAccountNameW(NULL, (LPCWSTR) name.utf16(), sid, &size, NULL, &cchReferencedDomainName, &nameuse))
  54                     goto error;
  55             }
  56             else {
  57                 if (!IsValidSid(sid_))
  58                     goto error;
  59
  60                 DWORD sidlength = GetLengthSid(sid_);
  61                 sid = (PSID) new BYTE[sidlength];
  62                 if (!CopySid(sidlength, sid, sid_))
  63                     goto error;
  64             }
  65
  66             return;
  67
  68           error:
  69             delete[] sid;
  70             sid = 0;
  71             if (userInfo) {
  72                 NetApiBufferFree(userInfo);
  73                 userInfo = 0;
  74             }
  75         }
  76
  77         ~Private()
  78         {
  79             if (userInfo)
  80                 NetApiBufferFree(userInfo);
  81
  82             delete[] sid;
  83         }
  84 };
  85
  86 KUser::KUser(UIDMode mode)
  87         : d(0)
  88 {
  89     Q_UNUSED(mode)
  90
  91     DWORD bufferLen = UNLEN + 1;
  92     ushort buffer[UNLEN + 1];
  93
  94     if (GetUserNameW((LPWSTR) buffer, &bufferLen))
  95         d = new Private(QString::fromUtf16(buffer));
  96 }
  97
  98 KUser::KUser(K_UID uid)
  99     : d(0)
 100 {
 101     DWORD bufferLen = UNLEN + 1;
 102     ushort buffer[UNLEN + 1];
 103     SID_NAME_USE eUse;
 104
 105     if (LookupAccountSidW(NULL, uid, (LPWSTR) buffer, &bufferLen, NULL, NULL, &eUse))
 106         d = new Private(QString::fromUtf16(buffer), uid);
 107 }
 108
 109 KUser::KUser(const QString &name)
 110     : d(new Private(name))
 111 {
 112 }
 113
 114 KUser::KUser(const char *name)
 115     :d(new Private(QString::fromLocal8Bit(name)))
 116 {
 117 }
 118
 119 KUser::KUser(const KUser &user)
 120     : d(user.d)
 121 {
 122 }
 123
 124 KUser &KUser::operator=(const KUser &user)
 125 {
 126     d = user.d;
 127     return *this;
 128 }
 129
 130 bool KUser::operator==(const KUser &user) const
 131 {
 132     if (!isValid() || !user.isValid())
 133         return false;
 134     return EqualSid(d->sid, user.d->sid);
 135 }
 136
 137 bool KUser::operator !=(const KUser &user) const
 138 {
 139     return !operator==(user);
 140 }
 141
 142 bool KUser::isValid() const
 143 {
 144     return d->userInfo != 0 && d->sid != 0;
 145 }
 146
 147 bool KUser::isSuperUser() const
 148 {
 149     return d->userInfo && d->userInfo->usri11_priv == USER_PRIV_ADMIN;
 150 }
 151
 152 QString KUser::loginName() const
 153 {
 154     return (d->userInfo ? QString::fromUtf16((ushort *) d->userInfo->usri11_name) : QString());
 155 }
 156
 157 QString KUser::fullName() const
 158 {
 159     return (d->userInfo ? QString::fromUtf16((ushort *) d->userInfo->usri11_full_name) : QString());
 160 }
 161
 162 QString KUser::homeDir() const
 163 {
 164         return QDir::fromNativeSeparators(qgetenv("USERPROFILE"));
 165 }
 166
 167 QString KUser::faceIconPath() const
 168 {
 169     // FIXME: this needs to be adapted to windows systems (BC changes)
 170     return QString();
 171 }
 172
 173 QString KUser::shell() const
 174 {
 175     return QString::fromAscii("cmd.exe");
 176 }
 177
 178 QList<KUserGroup> KUser::groups() const
 179 {
 180     QList<KUserGroup> result;
 181
 182     Q_FOREACH (const QString &name, groupNames()) {
 183         result.append(KUserGroup(name));
 184     }
 185
 186     return result;
 187 }
 188
 189 QStringList KUser::groupNames() const
 190 {
 191     QStringList result;
 192
 193     if (!d->userInfo) {
 194         return result;
 195     }
 196
 197     PGROUP_USERS_INFO_0 pGroups = NULL;
 198     DWORD dwEntriesRead = 0;
 199     DWORD dwTotalEntries = 0;
 200     NET_API_STATUS nStatus;
 201
 202     nStatus = NetUserGetGroups(NULL, d->userInfo->usri11_name, 0, (LPBYTE *) &pGroups, MAX_PREFERRED_LENGTH, &dwEntriesRead, &dwTotalEntries);
 203
 204     if (nStatus == NERR_Success) {
 205         for (DWORD i = 0; i < dwEntriesRead; ++i) {
 206             result.append(QString::fromUtf16((ushort *) pGroups[i].grui0_name));
 207         }
 208     }
 209
 210     if (pGroups) {
 211         NetApiBufferFree(pGroups);
 212     }
 213
 214     return result;
 215 }
 216
 217 K_UID KUser::uid() const
 218 {
 219     return d->sid;
 220 }
 221
 222 QVariant KUser::property(UserProperty which) const
 223 {
 224     if (which == FullName)
 225         return QVariant(d->userInfo ? QString::fromUtf16((ushort *) d->userInfo->usri11_full_name) : QString());
 226
 227     return QVariant();
 228 }
 229
 230 QList<KUser> KUser::allUsers()
 231 {
 232     QList<KUser> result;
 233
 234     NET_API_STATUS nStatus;
 235     PUSER_INFO_11 pUser = NULL;
 236     DWORD dwEntriesRead = 0;
 237     DWORD dwTotalEntries = 0;
 238     DWORD dwResumeHandle = 0;
 239
 240     KUser tmp;
 241
 242     do {
 243         nStatus = NetUserEnum(NULL, 11, 0, (LPBYTE*) &pUser, 1, &dwEntriesRead, &dwTotalEntries, &dwResumeHandle);
 244
 245         if ((nStatus == NERR_Success || nStatus == ERROR_MORE_DATA) && dwEntriesRead > 0) {
 246             tmp.d = new Private(pUser);
 247             result.append(tmp);
 248         }
 249     } while (nStatus == ERROR_MORE_DATA);
 250
 251     return result;
 252 }
 253
 254 QStringList KUser::allUserNames()
 255 {
 256     QStringList result;
 257
 258     NET_API_STATUS nStatus;
 259     PUSER_INFO_0 pUsers = NULL;
 260     DWORD dwEntriesRead = 0;
 261     DWORD dwTotalEntries = 0;
 262
 263     nStatus = NetUserEnum(NULL, 0, 0, (LPBYTE*) &pUsers, MAX_PREFERRED_LENGTH, &dwEntriesRead, &dwTotalEntries, NULL);
 264
 265     if (nStatus == NERR_Success) {
 266         for (DWORD i = 0; i < dwEntriesRead; ++i) {
 267             result.append(QString::fromUtf16((ushort *) pUsers[i].usri0_name));
 268         }
 269     }
 270
 271     if (pUsers) {
 272         NetApiBufferFree(pUsers);
 273     }
 274
 275     return result;
 276 }
 277
 278 KUser::~KUser()
 279 {
 280 }
 281
 282 class KUserGroup::Private : public KShared
 283 {
 284     public:
 285         PGROUP_INFO_0 groupInfo;
 286
 287         Private() : groupInfo(NULL) {}
 288         Private(PGROUP_INFO_0 groupInfo_) : groupInfo(groupInfo_) {}
 289         Private(const QString &Name) : groupInfo(NULL)
 290         {
 291             NetGroupGetInfo(NULL, (PCWSTR) Name.utf16(), 0, (PBYTE *) &groupInfo);
 292         }
 293
 294         ~Private()
 295         {
 296             if (groupInfo) {
 297                 NetApiBufferFree(groupInfo);
 298             }
 299         }
 300 };
 301
 302 KUserGroup::KUserGroup(const QString &_name)
 303     : d(new Private(_name))
 304 {
 305 }
 306
 307 KUserGroup::KUserGroup(const char *_name)
 308     : d(new Private(QString(_name)))
 309 {
 310 }
 311
 312 KUserGroup::KUserGroup(const KUserGroup &group)
 313     : d(group.d)
 314 {
 315 }
 316
 317 KUserGroup& KUserGroup::operator =(const KUserGroup &group)
 318 {
 319     d = group.d;
 320     return *this;
 321 }
 322
 323 bool KUserGroup::operator==(const KUserGroup &group) const
 324 {
 325     if (d->groupInfo == NULL || group.d->groupInfo == NULL) {
 326         return false;
 327     }
 328     return wcscmp(d->groupInfo->grpi0_name, group.d->groupInfo->grpi0_name) == 0;
 329 }
 330
 331 bool KUserGroup::operator!=(const KUserGroup &group) const
 332 {
 333     return !operator==(group);
 334 }
 335
 336 bool KUserGroup::isValid() const
 337 {
 338     return d->groupInfo != NULL;
 339 }
 340
 341 QString KUserGroup::name() const
 342 {
 343     return QString::fromUtf16((ushort *) d->groupInfo->grpi0_name);
 344 }
 345
 346 QList<KUser> KUserGroup::users() const
 347 {
 348     QList<KUser> Result;
 349
 350     Q_FOREACH(const QString &user, userNames()) {
 351         Result.append(KUser(user));
 352     }
 353
 354     return Result;
 355 }
 356
 357 QStringList KUserGroup::userNames() const
 358 {
 359     QStringList result;
 360
 361     if (!d->groupInfo) {
 362         return result;
 363     }
 364
 365     PGROUP_USERS_INFO_0 pUsers = NULL;
 366     DWORD dwEntriesRead = 0;
 367     DWORD dwTotalEntries = 0;
 368     NET_API_STATUS nStatus;
 369
 370     nStatus = NetGroupGetUsers(NULL, d->groupInfo->grpi0_name, 0, (LPBYTE *) &pUsers, MAX_PREFERRED_LENGTH, &dwEntriesRead, &dwTotalEntries, NULL);
 371
 372     if (nStatus == NERR_Success) {
 373         for (DWORD i = 0; i < dwEntriesRead; ++i) {
 374             result.append(QString::fromUtf16((ushort *) pUsers[i].grui0_name));
 375         }
 376     }
 377
 378     if (pUsers) {
 379         NetApiBufferFree(pUsers);
 380     }
 381
 382     return result;
 383 }
 384
 385 QList<KUserGroup> KUserGroup::allGroups()
 386 {
 387     QList<KUserGroup> result;
 388
 389     NET_API_STATUS nStatus;
 390     PGROUP_INFO_0 pGroup=NULL;
 391     DWORD dwEntriesRead=0;
 392     DWORD dwTotalEntries=0;
 393     DWORD dwResumeHandle=0;
 394
 395     KUserGroup tmp("");
 396
 397     do {
 398         nStatus = NetGroupEnum(NULL, 0, (LPBYTE*) &pGroup, 1, &dwEntriesRead, &dwTotalEntries, &dwResumeHandle);
 399
 400         if ((nStatus == NERR_Success || nStatus == ERROR_MORE_DATA) && dwEntriesRead > 0) {
 401             tmp.d = new Private(pGroup);
 402             result.append(tmp);
 403         }
 404     } while (nStatus == ERROR_MORE_DATA);
 405
 406     return result;
 407 }
 408
 409 QStringList KUserGroup::allGroupNames()
 410 {
 411     QStringList result;
 412
 413     NET_API_STATUS nStatus;
 414     PGROUP_INFO_0 pGroups=NULL;
 415     DWORD dwEntriesRead=0;
 416     DWORD dwTotalEntries=0;
 417
 418     nStatus = NetGroupEnum(NULL, 0, (LPBYTE*) &pGroups, MAX_PREFERRED_LENGTH, &dwEntriesRead, &dwTotalEntries, NULL);
 419
 420     if (nStatus == NERR_Success) {
 421         for (DWORD i = 0; i < dwEntriesRead; ++i) {
 422             result.append(QString::fromUtf16((ushort *) pGroups[i].grpi0_name));
 423         }
 424     }
 425
 426     if (pGroups) {
 427         NetApiBufferFree(pGroups);
 428     }
 429
 430     return result;
 431 }
 432
 433 KUserGroup::~KUserGroup()
 434 {
 435 }