source3/lib/privileges.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Privileges handling functions
   4    Copyright (C) Jean François Micouleau       1998-2001
   5    Copyright (C) Simo Sorce                     2002-2003
   6    Copyright (C) Gerald (Jerry) Carter          2005
   7
   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.
  12
  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.
  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., 675 Mass Ave, Cambridge, MA 02139, USA.
  21 */
  22
  23
  24 #include "includes.h"
  25
  26 #define PRIVPREFIX              "PRIV_"
  27
  28 static const SE_PRIV se_priv_all  = SE_ALL_PRIVS;
  29 static const SE_PRIV se_priv_end  = SE_END;
  30
  31 /* Define variables for all privileges so we can use the
  32    SE_PRIV* in the various se_priv_XXX() functions */
  33
  34 const SE_PRIV se_priv_none       = SE_NONE;
  35 const SE_PRIV se_machine_account = SE_MACHINE_ACCOUNT;
  36 const SE_PRIV se_print_operator  = SE_PRINT_OPERATOR;
  37 const SE_PRIV se_add_users       = SE_ADD_USERS;
  38 const SE_PRIV se_disk_operators  = SE_DISK_OPERATOR;
  39 const SE_PRIV se_remote_shutdown = SE_REMOTE_SHUTDOWN;
  40 const SE_PRIV se_restore         = SE_RESTORE;
  41 const SE_PRIV se_take_ownership  = SE_TAKE_OWNERSHIP;
  42
  43 /********************************************************************
  44  This is a list of privileges reported by a WIndows 2000 SP4 AD DC
  45  just for reference purposes (and I know the LUID is not guaranteed
  46  across reboots):
  47
  48             SeCreateTokenPrivilege  Create a token object ( 0x0, 0x2 )
  49      SeAssignPrimaryTokenPrivilege  Replace a process level token ( 0x0, 0x3 )
  50              SeLockMemoryPrivilege  Lock pages in memory ( 0x0, 0x4 )
  51           SeIncreaseQuotaPrivilege  Increase quotas ( 0x0, 0x5 )
  52          SeMachineAccountPrivilege  Add workstations to domain ( 0x0, 0x6 )
  53                     SeTcbPrivilege  Act as part of the operating system ( 0x0, 0x7 )
  54                SeSecurityPrivilege  Manage auditing and security log ( 0x0, 0x8 )
  55           SeTakeOwnershipPrivilege  Take ownership of files or other objects ( 0x0, 0x9 )
  56              SeLoadDriverPrivilege  Load and unload device drivers ( 0x0, 0xa )
  57           SeSystemProfilePrivilege  Profile system performance ( 0x0, 0xb )
  58              SeSystemtimePrivilege  Change the system time ( 0x0, 0xc )
  59    SeProfileSingleProcessPrivilege  Profile single process ( 0x0, 0xd )
  60    SeIncreaseBasePriorityPrivilege  Increase scheduling priority ( 0x0, 0xe )
  61          SeCreatePagefilePrivilege  Create a pagefile ( 0x0, 0xf )
  62         SeCreatePermanentPrivilege  Create permanent shared objects ( 0x0, 0x10 )
  63                  SeBackupPrivilege  Back up files and directories ( 0x0, 0x11 )
  64                 SeRestorePrivilege  Restore files and directories ( 0x0, 0x12 )
  65                SeShutdownPrivilege  Shut down the system ( 0x0, 0x13 )
  66                   SeDebugPrivilege  Debug programs ( 0x0, 0x14 )
  67                   SeAuditPrivilege  Generate security audits ( 0x0, 0x15 )
  68       SeSystemEnvironmentPrivilege  Modify firmware environment values ( 0x0, 0x16 )
  69            SeChangeNotifyPrivilege  Bypass traverse checking ( 0x0, 0x17 )
  70          SeRemoteShutdownPrivilege  Force shutdown from a remote system ( 0x0, 0x18 )
  71                  SeUndockPrivilege  Remove computer from docking station ( 0x0, 0x19 )
  72               SeSyncAgentPrivilege  Synchronize directory service data ( 0x0, 0x1a )
  73        SeEnableDelegationPrivilege  Enable computer and user accounts to be trusted for delegation ( 0x0, 0x1b )
  74            SeManageVolumePrivilege  Perform volume maintenance tasks ( 0x0, 0x1c )
  75             SeImpersonatePrivilege  Impersonate a client after authentication ( 0x0, 0x1d )
  76            SeCreateGlobalPrivilege  Create global objects ( 0x0, 0x1e )
  77
  78  ********************************************************************/
  79
  80 /* we have to define the LUID here due to a horrible check by printmig.exe
  81    that requires the SeBackupPrivilege match what is in Windows.  So match
  82    those that we implement and start Samba privileges at 0x1001 */
  83
  84 PRIVS privs[] = {
  85 #if 0   /* usrmgr will display these twice if you include them.  We don't
  86            use them but we'll keep the bitmasks reserved in privileges.h anyways */
  87
  88         {SE_NETWORK_LOGON,      "SeNetworkLogonRight",          "Access this computer from network",       { 0x0, 0x0 }},
  89         {SE_INTERACTIVE_LOGON,  "SeInteractiveLogonRight",      "Log on locally",                          { 0x0, 0x0 }},
  90         {SE_BATCH_LOGON,        "SeBatchLogonRight",            "Log on as a batch job",                   { 0x0, 0x0 }},
  91         {SE_SERVICE_LOGON,      "SeServiceLogonRight",          "Log on as a service",                     { 0x0, 0x0 }},
  92 #endif
  93         {SE_MACHINE_ACCOUNT,    "SeMachineAccountPrivilege",    "Add machines to domain",                  { 0x0, 0x0006 }},
  94         {SE_TAKE_OWNERSHIP,     "SeTakeOwnershipPrivilege",     "Take ownership of files or other objects",{ 0x0, 0x0009 }},
  95         {SE_BACKUP,             "SeBackupPrivilege",            "Back up files and directories",           { 0x0, 0x0011 }},
  96         {SE_RESTORE,            "SeRestorePrivilege",           "Restore files and directories",           { 0x0, 0x0012 }},
  97         {SE_REMOTE_SHUTDOWN,    "SeRemoteShutdownPrivilege",    "Force shutdown from a remote system",     { 0x0, 0x0018 }},
  98
  99         {SE_PRINT_OPERATOR,     "SePrintOperatorPrivilege",     "Manage printers",                         { 0x0, 0x1001 }},
 100         {SE_ADD_USERS,          "SeAddUsersPrivilege",          "Add users and groups to the domain",      { 0x0, 0x1002 }},
 101         {SE_DISK_OPERATOR,      "SeDiskOperatorPrivilege",      "Manage disk shares",                      { 0x0, 0x1003 }},
 102
 103         {SE_END, "", "", { 0x0, 0x0 }}
 104 };
 105
 106 typedef struct {
 107         int count;
 108         DOM_SID *list;
 109 } SID_LIST;
 110
 111 typedef struct {
 112         SE_PRIV privilege;
 113         SID_LIST sids;
 114 } PRIV_SID_LIST;
 115
 116 /***************************************************************************
 117  copy an SE_PRIV structure
 118 ****************************************************************************/
 119
 120 BOOL se_priv_copy( SE_PRIV *dst, const SE_PRIV *src )
 121 {
 122         if ( !dst || !src )
 123                 return False;
 124
 125         memcpy( dst, src, sizeof(SE_PRIV) );
 126
 127         return True;
 128 }
 129
 130 /***************************************************************************
 131  combine 2 SE_PRIV structures and store the resulting set in mew_mask
 132 ****************************************************************************/
 133
 134 void se_priv_add( SE_PRIV *mask, const SE_PRIV *addpriv )
 135 {
 136         int i;
 137
 138         for ( i=0; i<SE_PRIV_MASKSIZE; i++ ) {
 139                 mask->mask[i] |= addpriv->mask[i];
 140         }
 141 }
 142
 143 /***************************************************************************
 144  remove one SE_PRIV sytucture from another and store the resulting set
 145  in mew_mask
 146 ****************************************************************************/
 147
 148 void se_priv_remove( SE_PRIV *mask, const SE_PRIV *removepriv )
 149 {
 150         int i;
 151
 152         for ( i=0; i<SE_PRIV_MASKSIZE; i++ ) {
 153                 mask->mask[i] &= ~removepriv->mask[i];
 154         }
 155 }
 156
 157 /***************************************************************************
 158  invert a given SE_PRIV and store the set in new_mask
 159 ****************************************************************************/
 160
 161 static void se_priv_invert( SE_PRIV *new_mask, const SE_PRIV *mask )
 162 {
 163         SE_PRIV allprivs;
 164
 165         se_priv_copy( &allprivs, &se_priv_all );
 166         se_priv_remove( &allprivs, mask );
 167         se_priv_copy( new_mask, &allprivs );
 168 }
 169
 170 /***************************************************************************
 171  check if 2 SE_PRIV structure are equal
 172 ****************************************************************************/
 173
 174 static BOOL se_priv_equal( const SE_PRIV *mask1, const SE_PRIV *mask2 )
 175 {
 176         return ( memcmp(mask1, mask2, sizeof(SE_PRIV)) == 0 );
 177 }
 178
 179 /***************************************************************************
 180  check if a SE_PRIV has any assigned privileges
 181 ****************************************************************************/
 182
 183 static BOOL se_priv_empty( const SE_PRIV *mask )
 184 {
 185         SE_PRIV p1;
 186         int i;
 187
 188         se_priv_copy( &p1, mask );
 189
 190         for ( i=0; i<SE_PRIV_MASKSIZE; i++ ) {
 191                 p1.mask[i] &= se_priv_all.mask[i];
 192         }
 193
 194         return se_priv_equal( &p1, &se_priv_none );
 195 }
 196
 197 /*********************************************************************
 198  Lookup the SE_PRIV value for a privilege name
 199 *********************************************************************/
 200
 201 BOOL se_priv_from_name( const char *name, SE_PRIV *mask )
 202 {
 203         int i;
 204
 205         for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
 206                 if ( strequal( privs[i].name, name ) ) {
 207                         se_priv_copy( mask, &privs[i].se_priv );
 208                         return True;
 209                 }
 210         }
 211
 212         return False;
 213 }
 214
 215 /***************************************************************************
 216  dump an SE_PRIV structure to the log files
 217 ****************************************************************************/
 218
 219 void dump_se_priv( int dbg_cl, int dbg_lvl, const SE_PRIV *mask )
 220 {
 221         int i;
 222
 223         DEBUGADDC( dbg_cl, dbg_lvl,("SE_PRIV "));
 224
 225         for ( i=0; i<SE_PRIV_MASKSIZE; i++ ) {
 226                 DEBUGADDC( dbg_cl, dbg_lvl,(" 0x%x", mask->mask[i] ));
 227         }
 228
 229         DEBUGADDC( dbg_cl, dbg_lvl, ("\n"));
 230 }
 231
 232 /***************************************************************************
 233  Retrieve the privilege mask (set) for a given SID
 234 ****************************************************************************/
 235
 236 static BOOL get_privileges( const DOM_SID *sid, SE_PRIV *mask )
 237 {
 238         TDB_CONTEXT *tdb = get_account_pol_tdb();
 239         fstring keystr;
 240         TDB_DATA key, data;
 241
 242         /* Fail if the admin has not enable privileges */
 243
 244         if ( !lp_enable_privileges() ) {
 245                 return False;
 246         }
 247
 248         if ( !tdb )
 249                 return False;
 250
 251         /* PRIV_<SID> (NULL terminated) as the key */
 252
 253         fstr_sprintf( keystr, "%s%s", PRIVPREFIX, sid_string_static(sid) );
 254         key.dptr = keystr;
 255         key.dsize = strlen(keystr) + 1;
 256
 257         data = tdb_fetch( tdb, key );
 258
 259         if ( !data.dptr ) {
 260                 DEBUG(3,("get_privileges: No privileges assigned to SID [%s]\n",
 261                         sid_string_static(sid)));
 262                 return False;
 263         }
 264
 265         SMB_ASSERT( data.dsize == sizeof( SE_PRIV ) );
 266
 267         se_priv_copy( mask, (SE_PRIV*)data.dptr );
 268         SAFE_FREE(data.dptr);
 269
 270         return True;
 271 }
 272
 273 /***************************************************************************
 274  Store the privilege mask (set) for a given SID
 275 ****************************************************************************/
 276
 277 static BOOL set_privileges( const DOM_SID *sid, SE_PRIV *mask )
 278 {
 279         TDB_CONTEXT *tdb = get_account_pol_tdb();
 280         fstring keystr;
 281         TDB_DATA key, data;
 282
 283         if ( !lp_enable_privileges() )
 284                 return False;
 285
 286         if ( !tdb )
 287                 return False;
 288
 289         if ( !sid || (sid->num_auths == 0) ) {
 290                 DEBUG(0,("set_privileges: Refusing to store empty SID!\n"));
 291                 return False;
 292         }
 293
 294         /* PRIV_<SID> (NULL terminated) as the key */
 295
 296         fstr_sprintf( keystr, "%s%s", PRIVPREFIX, sid_string_static(sid) );
 297         key.dptr = keystr;
 298         key.dsize = strlen(keystr) + 1;
 299
 300         /* no packing.  static size structure, just write it out */
 301
 302         data.dptr  = (char*)mask;
 303         data.dsize = sizeof(SE_PRIV);
 304
 305         return ( tdb_store(tdb, key, data, TDB_REPLACE) != -1 );
 306 }
 307
 308 /****************************************************************************
 309  check if the privilege is in the privilege list
 310 ****************************************************************************/
 311
 312 static BOOL is_privilege_assigned( SE_PRIV *privileges, const SE_PRIV *check )
 313 {
 314         SE_PRIV p1, p2;
 315
 316         if ( !privileges || !check )
 317                 return False;
 318
 319         /* everyone has privileges if you aren't checking for any */
 320
 321         if ( se_priv_empty( check ) ) {
 322                 DEBUG(1,("is_privilege_assigned: no privileges in check_mask!\n"));
 323                 return True;
 324         }
 325
 326         se_priv_copy( &p1, check );
 327
 328         /* invert the SE_PRIV we want to check for and remove that from the
 329            original set.  If we are left with the SE_PRIV we are checking
 330            for then return True */
 331
 332         se_priv_invert( &p1, check );
 333         se_priv_copy( &p2, privileges );
 334         se_priv_remove( &p2, &p1 );
 335
 336         return se_priv_equal( &p2, check );
 337 }
 338
 339 /****************************************************************************
 340  check if the privilege is in the privilege list
 341 ****************************************************************************/
 342
 343 static BOOL is_any_privilege_assigned( SE_PRIV *privileges, const SE_PRIV *check )
 344 {
 345         SE_PRIV p1, p2;
 346
 347         if ( !privileges || !check )
 348                 return False;
 349
 350         /* everyone has privileges if you aren't checking for any */
 351
 352         if ( se_priv_empty( check ) ) {
 353                 DEBUG(1,("is_any_privilege_assigned: no privileges in check_mask!\n"));
 354                 return True;
 355         }
 356
 357         se_priv_copy( &p1, check );
 358
 359         /* invert the SE_PRIV we want to check for and remove that from the
 360            original set.  If we are left with the SE_PRIV we are checking
 361            for then return True */
 362
 363         se_priv_invert( &p1, check );
 364         se_priv_copy( &p2, privileges );
 365         se_priv_remove( &p2, &p1 );
 366
 367         /* see if we have any bits left */
 368
 369         return !se_priv_empty( &p2 );
 370 }
 371
 372 /****************************************************************************
 373  add a privilege to a privilege array
 374  ****************************************************************************/
 375
 376 static BOOL privilege_set_add(PRIVILEGE_SET *priv_set, LUID_ATTR set)
 377 {
 378         LUID_ATTR *new_set;
 379
 380         /* we can allocate memory to add the new privilege */
 381
 382         new_set = TALLOC_REALLOC_ARRAY(priv_set->mem_ctx, priv_set->set, LUID_ATTR, priv_set->count + 1);
 383         if ( !new_set ) {
 384                 DEBUG(0,("privilege_set_add: failed to allocate memory!\n"));
 385                 return False;
 386         }
 387
 388         new_set[priv_set->count].luid.high = set.luid.high;
 389         new_set[priv_set->count].luid.low = set.luid.low;
 390         new_set[priv_set->count].attr = set.attr;
 391
 392         priv_set->count++;
 393         priv_set->set = new_set;
 394
 395         return True;
 396 }
 397
 398 /*********************************************************************
 399  Generate the LUID_ATTR structure based on a bitmask
 400 *********************************************************************/
 401
 402 LUID_ATTR get_privilege_luid( SE_PRIV *mask )
 403 {
 404         LUID_ATTR priv_luid;
 405         int i;
 406
 407         priv_luid.attr = 0;
 408         priv_luid.luid.high = 0;
 409
 410         for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
 411
 412                 if ( se_priv_equal( &privs[i].se_priv, mask ) ) {
 413                         priv_luid.luid = privs[i].luid;
 414                         break;
 415                 }
 416         }
 417
 418         return priv_luid;
 419 }
 420
 421 /*********************************************************************
 422  Generate the LUID_ATTR structure based on a bitmask
 423 *********************************************************************/
 424
 425 const char* get_privilege_dispname( const char *name )
 426 {
 427         int i;
 428
 429         for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
 430
 431                 if ( strequal( privs[i].name, name ) ) {
 432                         return privs[i].description;
 433                 }
 434         }
 435
 436         return NULL;
 437 }
 438
 439 /*********************************************************************
 440  get a list of all privleges for all sids the in list
 441 *********************************************************************/
 442
 443 BOOL get_privileges_for_sids(SE_PRIV *privileges, DOM_SID *slist, int scount)
 444 {
 445         SE_PRIV mask;
 446         int i;
 447         BOOL found = False;
 448
 449         se_priv_copy( privileges, &se_priv_none );
 450
 451         for ( i=0; i<scount; i++ ) {
 452                 /* don't add unless we actually have a privilege assigned */
 453
 454                 if ( !get_privileges( &slist[i], &mask ) )
 455                         continue;
 456
 457                 DEBUG(5,("get_privileges_for_sids: sid = %s\nPrivilege set:\n",
 458                         sid_string_static(&slist[i])));
 459                 dump_se_priv( DBGC_ALL, 5, &mask );
 460
 461                 se_priv_add( privileges, &mask );
 462                 found = True;
 463         }
 464
 465         return found;
 466 }
 467
 468
 469 /*********************************************************************
 470  travseral functions for privilege_enumerate_accounts
 471 *********************************************************************/
 472
 473 static int priv_traverse_fn(TDB_CONTEXT *t, TDB_DATA key, TDB_DATA data, void *state)
 474 {
 475         PRIV_SID_LIST *priv = state;
 476         int  prefixlen = strlen(PRIVPREFIX);
 477         DOM_SID sid;
 478         fstring sid_string;
 479
 480         /* easy check first */
 481
 482         if ( data.dsize != sizeof(SE_PRIV) )
 483                 return 0;
 484
 485         /* check we have a PRIV_+SID entry */
 486
 487         if ( strncmp(key.dptr, PRIVPREFIX, prefixlen) != 0)
 488                 return 0;
 489
 490         /* check to see if we are looking for a particular privilege */
 491
 492         if ( !se_priv_equal(&priv->privilege, &se_priv_none) ) {
 493                 SE_PRIV mask;
 494
 495                 se_priv_copy( &mask, (SE_PRIV*)data.dptr );
 496
 497                 /* if the SID does not have the specified privilege
 498                    then just return */
 499
 500                 if ( !is_privilege_assigned( &mask, &priv->privilege) )
 501                         return 0;
 502         }
 503
 504         fstrcpy( sid_string, &key.dptr[strlen(PRIVPREFIX)] );
 505
 506         /* this is a last ditch safety check to preventing returning
 507            and invalid SID (i've somehow run into this on development branches) */
 508
 509         if ( strcmp( "S-0-0", sid_string ) == 0 )
 510                 return 0;
 511
 512         if ( !string_to_sid(&sid, sid_string) ) {
 513                 DEBUG(0,("travsersal_fn_enum__acct: Could not convert SID [%s]\n",
 514                         sid_string));
 515                 return 0;
 516         }
 517
 518         add_sid_to_array( NULL, &sid, &priv->sids.list, &priv->sids.count );
 519
 520         return 0;
 521 }
 522
 523 /*********************************************************************
 524  Retreive list of privileged SIDs (for _lsa_enumerate_accounts()
 525 *********************************************************************/
 526
 527 NTSTATUS privilege_enumerate_accounts(DOM_SID **sids, int *num_sids)
 528 {
 529         TDB_CONTEXT *tdb = get_account_pol_tdb();
 530         PRIV_SID_LIST priv;
 531
 532         ZERO_STRUCT(priv);
 533
 534         se_priv_copy( &priv.privilege, &se_priv_none );
 535
 536         tdb_traverse( tdb, priv_traverse_fn, &priv);
 537
 538         /* give the memory away; caller will free */
 539
 540         *sids      = priv.sids.list;
 541         *num_sids  = priv.sids.count;
 542
 543         return NT_STATUS_OK;
 544 }
 545
 546 /***************************************************************************
 547  Add privilege to sid
 548 ****************************************************************************/
 549
 550 BOOL grant_privilege(const DOM_SID *sid, const SE_PRIV *priv_mask)
 551 {
 552         SE_PRIV old_mask, new_mask;
 553
 554         ZERO_STRUCT( old_mask );
 555         ZERO_STRUCT( new_mask );
 556
 557         if ( get_privileges( sid, &old_mask ) )
 558                 se_priv_copy( &new_mask, &old_mask );
 559         else
 560                 se_priv_copy( &new_mask, &se_priv_none );
 561
 562         se_priv_add( &new_mask, priv_mask );
 563
 564         DEBUG(10,("grant_privilege: %s\n", sid_string_static(sid)));
 565
 566         DEBUGADD( 10, ("original privilege mask:\n"));
 567         dump_se_priv( DBGC_ALL, 10, &old_mask );
 568
 569         DEBUGADD( 10, ("new privilege mask:\n"));
 570         dump_se_priv( DBGC_ALL, 10, &new_mask );
 571
 572         return set_privileges( sid, &new_mask );
 573 }
 574
 575 /*********************************************************************
 576  Add a privilege based on its name
 577 *********************************************************************/
 578
 579 BOOL grant_privilege_by_name(DOM_SID *sid, const char *name)
 580 {
 581         int i;
 582
 583         for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
 584                 if ( strequal(privs[i].name, name) ) {
 585                         return grant_privilege( sid, &privs[i].se_priv );
 586                 }
 587         }
 588
 589         DEBUG(3, ("grant_privilege_by_name: No Such Privilege Found (%s)\n", name));
 590
 591         return False;
 592 }
 593
 594 /***************************************************************************
 595  Remove privilege from sid
 596 ****************************************************************************/
 597
 598 BOOL revoke_privilege(const DOM_SID *sid, const SE_PRIV *priv_mask)
 599 {
 600         SE_PRIV mask;
 601
 602         /* if the user has no privileges, then we can't revoke any */
 603
 604         if ( !get_privileges( sid, &mask ) )
 605                 return True;
 606
 607         DEBUG(10,("revoke_privilege: %s\n", sid_string_static(sid)));
 608
 609         DEBUGADD( 10, ("original privilege mask:\n"));
 610         dump_se_priv( DBGC_ALL, 10, &mask );
 611
 612         se_priv_remove( &mask, priv_mask );
 613
 614         DEBUGADD( 10, ("new privilege mask:\n"));
 615         dump_se_priv( DBGC_ALL, 10, &mask );
 616
 617         return set_privileges( sid, &mask );
 618 }
 619
 620 /*********************************************************************
 621  Revoke all privileges
 622 *********************************************************************/
 623
 624 BOOL revoke_all_privileges( DOM_SID *sid )
 625 {
 626         return revoke_privilege( sid, &se_priv_all );
 627 }
 628
 629 /*********************************************************************
 630  Add a privilege based on its name
 631 *********************************************************************/
 632
 633 BOOL revoke_privilege_by_name(DOM_SID *sid, const char *name)
 634 {
 635         int i;
 636
 637         for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
 638                 if ( strequal(privs[i].name, name) ) {
 639                         return revoke_privilege( sid, &privs[i].se_priv );
 640                 }
 641         }
 642
 643         DEBUG(3, ("revoke_privilege_by_name: No Such Privilege Found (%s)\n", name));
 644
 645         return False;
 646 }
 647
 648 /***************************************************************************
 649  Retrieve the SIDs assigned to a given privilege
 650 ****************************************************************************/
 651
 652 NTSTATUS privilege_create_account(const DOM_SID *sid )
 653 {
 654         return ( grant_privilege(sid, &se_priv_none) ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL);
 655 }
 656
 657 /****************************************************************************
 658  initialise a privilege list and set the talloc context
 659  ****************************************************************************/
 660
 661 NTSTATUS privilege_set_init(PRIVILEGE_SET *priv_set)
 662 {
 663         TALLOC_CTX *mem_ctx;
 664
 665         ZERO_STRUCTP( priv_set );
 666
 667         mem_ctx = talloc_init("privilege set");
 668         if ( !mem_ctx ) {
 669                 DEBUG(0,("privilege_set_init: failed to initialize talloc ctx!\n"));
 670                 return NT_STATUS_NO_MEMORY;
 671         }
 672
 673         priv_set->mem_ctx = mem_ctx;
 674
 675         return NT_STATUS_OK;
 676 }
 677
 678 /****************************************************************************
 679   initialise a privilege list and with someone else's talloc context
 680 ****************************************************************************/
 681
 682 NTSTATUS privilege_set_init_by_ctx(TALLOC_CTX *mem_ctx, PRIVILEGE_SET *priv_set)
 683 {
 684         ZERO_STRUCTP( priv_set );
 685
 686         priv_set->mem_ctx = mem_ctx;
 687         priv_set->ext_ctx = True;
 688
 689         return NT_STATUS_OK;
 690 }
 691
 692 /****************************************************************************
 693  Free all memory used by a PRIVILEGE_SET
 694 ****************************************************************************/
 695
 696 void privilege_set_free(PRIVILEGE_SET *priv_set)
 697 {
 698         if ( !priv_set )
 699                 return;
 700
 701         if ( !( priv_set->ext_ctx ) )
 702                 talloc_destroy( priv_set->mem_ctx );
 703
 704         ZERO_STRUCTP( priv_set );
 705 }
 706
 707 /****************************************************************************
 708  duplicate alloc luid_attr
 709  ****************************************************************************/
 710
 711 NTSTATUS dup_luid_attr(TALLOC_CTX *mem_ctx, LUID_ATTR **new_la, LUID_ATTR *old_la, int count)
 712 {
 713         int i;
 714
 715         if ( !old_la )
 716                 return NT_STATUS_OK;
 717
 718         *new_la = TALLOC_ARRAY(mem_ctx, LUID_ATTR, count);
 719         if ( !*new_la ) {
 720                 DEBUG(0,("dup_luid_attr: failed to alloc new LUID_ATTR array [%d]\n", count));
 721                 return NT_STATUS_NO_MEMORY;
 722         }
 723
 724         for (i=0; i<count; i++) {
 725                 (*new_la)[i].luid.high = old_la[i].luid.high;
 726                 (*new_la)[i].luid.low = old_la[i].luid.low;
 727                 (*new_la)[i].attr = old_la[i].attr;
 728         }
 729
 730         return NT_STATUS_OK;
 731 }
 732
 733 /****************************************************************************
 734  Does the user have the specified privilege ?  We only deal with one privilege
 735  at a time here.
 736 *****************************************************************************/
 737
 738 BOOL user_has_privileges(NT_USER_TOKEN *token, const SE_PRIV *privilege)
 739 {
 740         if ( !token )
 741                 return False;
 742
 743         return is_privilege_assigned( &token->privileges, privilege );
 744 }
 745
 746 /****************************************************************************
 747  Does the user have any of the specified privileges ?  We only deal with one privilege
 748  at a time here.
 749 *****************************************************************************/
 750
 751 BOOL user_has_any_privilege(NT_USER_TOKEN *token, const SE_PRIV *privilege)
 752 {
 753         if ( !token )
 754                 return False;
 755
 756         return is_any_privilege_assigned( &token->privileges, privilege );
 757 }
 758
 759 /****************************************************************************
 760  Convert a LUID to a named string
 761 ****************************************************************************/
 762
 763 char* luid_to_privilege_name(const LUID *set)
 764 {
 765         static fstring name;
 766         int i;
 767
 768         if (set->high != 0)
 769                 return NULL;
 770
 771         for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
 772                 if ( set->low == privs[i].luid.low ) {
 773                         fstrcpy( name, privs[i].name );
 774                         return name;
 775                 }
 776         }
 777
 778         return NULL;
 779 }
 780
 781 /*******************************************************************
 782  return the number of elements in the privlege array
 783 *******************************************************************/
 784
 785 int count_all_privileges( void )
 786 {
 787         static int count;
 788
 789         if ( count )
 790                 return count;
 791
 792         /* loop over the array and count it */
 793         for ( count=0; !se_priv_equal(&privs[count].se_priv, &se_priv_end); count++ ) ;
 794
 795         return count;
 796 }
 797
 798 /*******************************************************************
 799 *******************************************************************/
 800
 801 BOOL se_priv_to_privilege_set( PRIVILEGE_SET *set, SE_PRIV *mask )
 802 {
 803         int i;
 804         uint32 num_privs = count_all_privileges();
 805         LUID_ATTR luid;
 806
 807         luid.attr = 0;
 808         luid.luid.high = 0;
 809
 810         for ( i=0; i<num_privs; i++ ) {
 811                 if ( !is_privilege_assigned(mask, &privs[i].se_priv) )
 812                         continue;
 813
 814                 luid.luid = privs[i].luid;
 815
 816                 if ( !privilege_set_add( set, luid ) )
 817                         return False;
 818         }
 819
 820         return True;
 821 }
 822
 823 /*******************************************************************
 824 *******************************************************************/
 825
 826 static BOOL luid_to_se_priv( LUID *luid, SE_PRIV *mask )
 827 {
 828         int i;
 829         uint32 num_privs = count_all_privileges();
 830
 831         for ( i=0; i<num_privs; i++ ) {
 832                 if ( luid->low == privs[i].luid.low ) {
 833                         se_priv_copy( mask, &privs[i].se_priv );
 834                         return True;
 835                 }
 836         }
 837
 838         return False;
 839 }
 840
 841 /*******************************************************************
 842 *******************************************************************/
 843
 844 BOOL privilege_set_to_se_priv( SE_PRIV *mask, PRIVILEGE_SET *privset )
 845 {
 846         int i;
 847
 848         ZERO_STRUCTP( mask );
 849
 850         for ( i=0; i<privset->count; i++ ) {
 851                 SE_PRIV r;
 852
 853                 /* sanity check for invalid privilege.  we really
 854                    only care about the low 32 bits */
 855
 856                 if ( privset->set[i].luid.high != 0 )
 857                         return False;
 858
 859                 if ( luid_to_se_priv( &privset->set[i].luid, &r ) )
 860                         se_priv_add( mask, &r );
 861         }
 862
 863         return True;
 864 }
 865
 866 /*******************************************************************
 867 *******************************************************************/
 868
 869 BOOL is_privileged_sid( DOM_SID *sid )
 870 {
 871         SE_PRIV mask;
 872
 873         return get_privileges( sid, &mask );
 874 }