r8141: Update volker's valgrind fix in r8097. Same effect, just helps me
[ira/wip.git] / source3 / lib / privileges.c
index b9d4df301d95e24b12df8378af21a3aad4b8d1b9..05dff33307c6a0ba71115f33ea1bcd25bd6f4e5d 100644 (file)
@@ -1,8 +1,9 @@
 /*
    Unix SMB/CIFS implementation.
    Privileges handling functions
-   Copyright (C) Jean François Micouleau       1998-2001
+   Copyright (C) Jean François Micouleau      1998-2001
    Copyright (C) Simo Sorce                    2002-2003
+   Copyright (C) Gerald (Jerry) Carter          2005
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */
 
 #include "includes.h"
 
-/* defines */
+#define PRIVPREFIX              "PRIV_"
+
+static const SE_PRIV se_priv_all  = SE_ALL_PRIVS;
+static const SE_PRIV se_priv_end  = SE_END;
+
+/* Define variables for all privileges so we can use the
+   SE_PRIV* in the various se_priv_XXX() functions */
+
+const SE_PRIV se_priv_none       = SE_NONE;
+const SE_PRIV se_machine_account = SE_MACHINE_ACCOUNT;
+const SE_PRIV se_print_operator  = SE_PRINT_OPERATOR;
+const SE_PRIV se_add_users       = SE_ADD_USERS;
+const SE_PRIV se_disk_operators  = SE_DISK_OPERATOR;
+const SE_PRIV se_remote_shutdown = SE_REMOTE_SHUTDOWN;
+const SE_PRIV se_restore         = SE_RESTORE;
+
+/********************************************************************
+ This is a list of privileges reported by a WIndows 2000 SP4 AD DC
+ just for reference purposes (and I know the LUID is not guaranteed 
+ across reboots):
+
+            SeCreateTokenPrivilege  Create a token object ( 0x0, 0x2 )
+     SeAssignPrimaryTokenPrivilege  Replace a process level token ( 0x0, 0x3 )
+             SeLockMemoryPrivilege  Lock pages in memory ( 0x0, 0x4 )
+          SeIncreaseQuotaPrivilege  Increase quotas ( 0x0, 0x5 )
+         SeMachineAccountPrivilege  Add workstations to domain ( 0x0, 0x6 )
+                    SeTcbPrivilege  Act as part of the operating system ( 0x0, 0x7 )
+               SeSecurityPrivilege  Manage auditing and security log ( 0x0, 0x8 )
+          SeTakeOwnershipPrivilege  Take ownership of files or other objects ( 0x0, 0x9 )
+             SeLoadDriverPrivilege  Load and unload device drivers ( 0x0, 0xa )
+          SeSystemProfilePrivilege  Profile system performance ( 0x0, 0xb )
+             SeSystemtimePrivilege  Change the system time ( 0x0, 0xc )
+   SeProfileSingleProcessPrivilege  Profile single process ( 0x0, 0xd )
+   SeIncreaseBasePriorityPrivilege  Increase scheduling priority ( 0x0, 0xe )
+         SeCreatePagefilePrivilege  Create a pagefile ( 0x0, 0xf )
+        SeCreatePermanentPrivilege  Create permanent shared objects ( 0x0, 0x10 )
+                 SeBackupPrivilege  Back up files and directories ( 0x0, 0x11 )
+                SeRestorePrivilege  Restore files and directories ( 0x0, 0x12 )
+               SeShutdownPrivilege  Shut down the system ( 0x0, 0x13 )
+                  SeDebugPrivilege  Debug programs ( 0x0, 0x14 )
+                  SeAuditPrivilege  Generate security audits ( 0x0, 0x15 )
+      SeSystemEnvironmentPrivilege  Modify firmware environment values ( 0x0, 0x16 )
+           SeChangeNotifyPrivilege  Bypass traverse checking ( 0x0, 0x17 )
+         SeRemoteShutdownPrivilege  Force shutdown from a remote system ( 0x0, 0x18 )
+                 SeUndockPrivilege  Remove computer from docking station ( 0x0, 0x19 )
+              SeSyncAgentPrivilege  Synchronize directory service data ( 0x0, 0x1a )
+       SeEnableDelegationPrivilege  Enable computer and user accounts to be trusted for delegation ( 0x0, 0x1b )
+           SeManageVolumePrivilege  Perform volume maintenance tasks ( 0x0, 0x1c )
+            SeImpersonatePrivilege  Impersonate a client after authentication ( 0x0, 0x1d )
+           SeCreateGlobalPrivilege  Create global objects ( 0x0, 0x1e )
+
+ ********************************************************************/
+
+/* we have to define the LUID here due to a horrible check by printmig.exe
+   that requires the SeBackupPrivilege match what is in Windows.  So match
+   those that we implement and start Samba privileges at 0x1001 */
+   
+PRIVS privs[] = {
+#if 0  /* usrmgr will display these twice if you include them.  We don't 
+          use them but we'll keep the bitmasks reserved in privileges.h anyways */
+          
+       {SE_NETWORK_LOGON,      "SeNetworkLogonRight",          "Access this computer from network",       { 0x0, 0x0 }},
+       {SE_INTERACTIVE_LOGON,  "SeInteractiveLogonRight",      "Log on locally",                          { 0x0, 0x0 }},
+       {SE_BATCH_LOGON,        "SeBatchLogonRight",            "Log on as a batch job",                   { 0x0, 0x0 }},
+       {SE_SERVICE_LOGON,      "SeServiceLogonRight",          "Log on as a service",                     { 0x0, 0x0 }},
+#endif
+       {SE_MACHINE_ACCOUNT,    "SeMachineAccountPrivilege",    "Add machines to domain",                  { 0x0, 0x0006 }},
+       {SE_TAKE_OWNERSHIP,     "SeTakeOwnershipPrivilege",     "Take ownership of files or other objects",{ 0x0, 0x0009 }},
+        {SE_BACKUP,             "SeBackupPrivilege",            "Back up files and directories",          { 0x0, 0x0011 }},
+        {SE_RESTORE,            "SeRestorePrivilege",           "Restore files and directories",          { 0x0, 0x0012 }},
+       {SE_REMOTE_SHUTDOWN,    "SeRemoteShutdownPrivilege",    "Force shutdown from a remote system",     { 0x0, 0x0018 }},
+       
+       {SE_PRINT_OPERATOR,     "SePrintOperatorPrivilege",     "Manage printers",                         { 0x0, 0x1001 }},
+       {SE_ADD_USERS,          "SeAddUsersPrivilege",          "Add users and groups to the domain",      { 0x0, 0x1002 }},
+       {SE_DISK_OPERATOR,      "SeDiskOperatorPrivilege",      "Manage disk shares",                      { 0x0, 0x1003 }},
 
-#define ALLOC_CHECK(ptr, err, label, str) do { if ((ptr) == NULL) { DEBUG(0, ("%s: out of memory!\n", str)); err = NT_STATUS_NO_MEMORY; goto label; } } while(0)
-#define NTSTATUS_CHECK(err, label, str1, str2) do { if (!NT_STATUS_IS_OK(err)) { DEBUG(0, ("%s: %s failed!\n", str1, str2)); } } while(0)
+       {SE_END, "", "", { 0x0, 0x0 }}
+};
 
-/****************************************************************************
- Check if a user is a mapped group.
-
-   This function will check if the group SID is mapped onto a
-   system managed gid or onto a winbind manged sid.
-   In the first case it will be threated like a mapped group
-   and the backend should take the member list with a getgrgid
-   and ignore any user that have been possibly set into the group
-   object.
-
-   In the second case, the group is a fully SAM managed group
-   served back to the system through winbind. In this case the
-   members of a Local group are "unrolled" to cope with the fact
-   that unix cannot contain groups inside groups.
-   The backend MUST never call any getgr* / getpw* function or
-   loops with winbind may happen. 
- ****************************************************************************/
+typedef struct {
+       int count;
+       DOM_SID *list;
+} SID_LIST;
 
-#if 0
-NTSTATUS is_mapped_group(BOOL *mapped, const DOM_SID *sid)
+typedef struct {
+       SE_PRIV privilege;
+       SID_LIST sids;
+} PRIV_SID_LIST;
+
+/***************************************************************************
+ copy an SE_PRIV structure
+****************************************************************************/
+
+BOOL se_priv_copy( SE_PRIV *dst, const SE_PRIV *src )
 {
-       NTSTATUS result;
-       gid_t id;
+       if ( !dst || !src )
+               return False;
+               
+       memcpy( dst, src, sizeof(SE_PRIV) );
+       
+       return True;
+}
+
+/***************************************************************************
+ combine 2 SE_PRIV structures and store the resulting set in mew_mask
+****************************************************************************/
+
+void se_priv_add( SE_PRIV *mask, const SE_PRIV *addpriv )
+{
+       int i;
 
-       /* look if mapping exist, do not make idmap alloc an uid if SID is not found */
-       result = idmap_get_gid_from_sid(&id, sid, False);
-       if (NT_STATUS_IS_OK(result)) {
-               *mapped = gid_is_in_winbind_range(id);
-       } else {
-               *mapped = False;
+       for ( i=0; i<SE_PRIV_MASKSIZE; i++ ) {
+               mask->mask[i] |= addpriv->mask[i];
        }
+}
 
-       return result;
+/***************************************************************************
+ remove one SE_PRIV sytucture from another and store the resulting set 
+ in mew_mask
+****************************************************************************/
+
+void se_priv_remove( SE_PRIV *mask, const SE_PRIV *removepriv )
+{      
+       int i;
+
+       for ( i=0; i<SE_PRIV_MASKSIZE; i++ ) {
+               mask->mask[i] &= ~removepriv->mask[i];
+       }
 }
-#endif
 
-/****************************************************************************
- duplicate alloc luid_attr
- ****************************************************************************/
-NTSTATUS dupalloc_luid_attr(TALLOC_CTX *mem_ctx, LUID_ATTR **new_la, LUID_ATTR *old_la)
-{
-       NTSTATUS ret;
+/***************************************************************************
+ invert a given SE_PRIV and store the set in new_mask
+****************************************************************************/
+
+static void se_priv_invert( SE_PRIV *new_mask, const SE_PRIV *mask )
+{      
+       SE_PRIV allprivs;
+       
+       se_priv_copy( &allprivs, &se_priv_all );
+       se_priv_remove( &allprivs, mask );
+       se_priv_copy( new_mask, &allprivs );
+}
 
-       /* don't crash if the source pointer is NULL (since we don't
-          do priviledges now anyways) */
+/***************************************************************************
+ check if 2 SE_PRIV structure are equal
+****************************************************************************/
 
-       if ( !old_la )
-               return NT_STATUS_OK;
+static BOOL se_priv_equal( const SE_PRIV *mask1, const SE_PRIV *mask2 )
+{      
+       return ( memcmp(mask1, mask2, sizeof(SE_PRIV)) == 0 );
+}
 
-       *new_la = (LUID_ATTR *)talloc(mem_ctx, sizeof(LUID_ATTR));
-       ALLOC_CHECK(new_la, ret, done, "dupalloc_luid_attr");
+/***************************************************************************
+ check if a SE_PRIV has any assigned privileges
+****************************************************************************/
 
-       (*new_la)->luid.high = old_la->luid.high;
-       (*new_la)->luid.low = old_la->luid.low;
-       (*new_la)->attr = old_la->attr;
+static BOOL se_priv_empty( const SE_PRIV *mask )
+{
+       SE_PRIV p1;
+       int i;
        
-       ret = NT_STATUS_OK;
+       se_priv_copy( &p1, mask );
 
-done:
-       return ret;
+       for ( i=0; i<SE_PRIV_MASKSIZE; i++ ) {
+               p1.mask[i] &= se_priv_all.mask[i];
+       }
+       
+       return se_priv_equal( &p1, &se_priv_none );
 }
 
-/****************************************************************************
- initialise a privilege list
- ****************************************************************************/
-NTSTATUS init_privilege(PRIVILEGE_SET **priv_set)
+/*********************************************************************
+ Lookup the SE_PRIV value for a privilege name 
+*********************************************************************/
+
+BOOL se_priv_from_name( const char *name, SE_PRIV *mask )
 {
-       NTSTATUS ret;
-       TALLOC_CTX *mem_ctx = talloc_init("privilege set");
-       ALLOC_CHECK(mem_ctx, ret, done, "init_privilege");
+       int i;
 
-       *priv_set = talloc_zero(mem_ctx, sizeof(PRIVILEGE_SET));
-       ALLOC_CHECK(*priv_set, ret, done, "init_privilege");
+       for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
+               if ( strequal( privs[i].name, name ) ) {
+                       se_priv_copy( mask, &privs[i].se_priv );
+                       return True;
+               }
+       }
 
-       (*priv_set)->mem_ctx = mem_ctx;
+       return False;
+}
 
-       ret = NT_STATUS_OK;
+/***************************************************************************
+ dump an SE_PRIV structure to the log files
+****************************************************************************/
 
-done:
-       return ret;
+void dump_se_priv( int dbg_cl, int dbg_lvl, const SE_PRIV *mask )
+{
+       int i;
+       
+       DEBUGADDC( dbg_cl, dbg_lvl,("SE_PRIV "));
+       
+       for ( i=0; i<SE_PRIV_MASKSIZE; i++ ) {
+               DEBUGADDC( dbg_cl, dbg_lvl,(" 0x%x", mask->mask[i] ));
+       }
+               
+       DEBUGADDC( dbg_cl, dbg_lvl, ("\n"));
 }
 
-NTSTATUS init_priv_with_ctx(TALLOC_CTX *mem_ctx, PRIVILEGE_SET **priv_set)
+/***************************************************************************
+ Retrieve the privilege mask (set) for a given SID
+****************************************************************************/
+
+static BOOL get_privileges( const DOM_SID *sid, SE_PRIV *mask )
 {
-       NTSTATUS ret;
+       TDB_CONTEXT *tdb = get_account_pol_tdb();
+       fstring keystr;
+       TDB_DATA key, data;
 
-       *priv_set = talloc_zero(mem_ctx, sizeof(PRIVILEGE_SET));
-       ALLOC_CHECK(*priv_set, ret, done, "init_privilege");
+       /* Fail if the admin has not enable privileges */
+       
+       if ( !lp_enable_privileges() ) {
+               return False;
+       }
+       
+       if ( !tdb )
+               return False;
+
+       /* PRIV_<SID> (NULL terminated) as the key */
+       
+       fstr_sprintf( keystr, "%s%s", PRIVPREFIX, sid_string_static(sid) );
+       key.dptr = keystr;
+       key.dsize = strlen(keystr) + 1;
+
+       data = tdb_fetch( tdb, key );
+       
+       if ( !data.dptr ) {
+               DEBUG(3,("get_privileges: No privileges assigned to SID [%s]\n",
+                       sid_string_static(sid)));
+               return False;
+       }
+       
+       SMB_ASSERT( data.dsize == sizeof( SE_PRIV ) );
+       
+       se_priv_copy( mask, (SE_PRIV*)data.dptr );
+       SAFE_FREE(data.dptr);
 
-       (*priv_set)->mem_ctx = mem_ctx;
-       (*priv_set)->ext_ctx = True;
+       return True;
+}
+
+/***************************************************************************
+ Store the privilege mask (set) for a given SID
+****************************************************************************/
+
+static BOOL set_privileges( const DOM_SID *sid, SE_PRIV *mask )
+{
+       TDB_CONTEXT *tdb = get_account_pol_tdb();
+       fstring keystr;
+       TDB_DATA key, data;
+       
+       if ( !lp_enable_privileges() )
+               return False;
+
+       if ( !tdb )
+               return False;
 
-       ret = NT_STATUS_OK;
+       /* PRIV_<SID> (NULL terminated) as the key */
+       
+       fstr_sprintf( keystr, "%s%s", PRIVPREFIX, sid_string_static(sid) );
+       key.dptr = keystr;
+       key.dsize = strlen(keystr) + 1;
+       
+       /* no packing.  static size structure, just write it out */
+       
+       data.dptr  = (char*)mask;
+       data.dsize = sizeof(SE_PRIV);
 
-done:
-       return ret;
+       return ( tdb_store(tdb, key, data, TDB_REPLACE) != -1 );
 }
 
-void reset_privilege(PRIVILEGE_SET *priv_set)
+/****************************************************************************
+ check if the privilege is in the privilege list
+****************************************************************************/
+
+static BOOL is_privilege_assigned( SE_PRIV *privileges, const SE_PRIV *check )
 {
-       priv_set->count = 0;
-       priv_set->control = 0;
-       priv_set->set = NULL;
+       SE_PRIV p1, p2;
+
+       if ( !privileges || !check )
+               return False;
+       
+       /* everyone has privileges if you aren't checking for any */
+       
+       if ( se_priv_empty( check ) ) {
+               DEBUG(1,("is_privilege_assigned: no privileges in check_mask!\n"));
+               return True;
+       }
+       
+       se_priv_copy( &p1, check );
+       
+       /* invert the SE_PRIV we want to check for and remove that from the 
+          original set.  If we are left with the SE_PRIV we are checking 
+          for then return True */
+          
+       se_priv_invert( &p1, check );
+       se_priv_copy( &p2, privileges );
+       se_priv_remove( &p2, &p1 );
+       
+       return se_priv_equal( &p2, check );
 }
 
-void destroy_privilege(PRIVILEGE_SET **priv_set)
+/****************************************************************************
+ check if the privilege is in the privilege list
+****************************************************************************/
+
+static BOOL is_any_privilege_assigned( SE_PRIV *privileges, const SE_PRIV *check )
 {
-       reset_privilege(*priv_set);
-       if (!((*priv_set)->ext_ctx))
-               /* mem_ctx is local, destroy it */
-               talloc_destroy((*priv_set)->mem_ctx);
-       *priv_set = NULL;
+       SE_PRIV p1, p2;
+
+       if ( !privileges || !check )
+               return False;
+       
+       /* everyone has privileges if you aren't checking for any */
+       
+       if ( se_priv_empty( check ) ) {
+               DEBUG(1,("is_any_privilege_assigned: no privileges in check_mask!\n"));
+               return True;
+       }
+       
+       se_priv_copy( &p1, check );
+       
+       /* invert the SE_PRIV we want to check for and remove that from the 
+          original set.  If we are left with the SE_PRIV we are checking 
+          for then return True */
+          
+       se_priv_invert( &p1, check );
+       se_priv_copy( &p2, privileges );
+       se_priv_remove( &p2, &p1 );
+       
+       /* see if we have any bits left */
+       
+       return !se_priv_empty( &p2 );
 }
 
 /****************************************************************************
  add a privilege to a privilege array
  ****************************************************************************/
-NTSTATUS add_privilege(PRIVILEGE_SET *priv_set, LUID_ATTR set)
+
+static BOOL privilege_set_add(PRIVILEGE_SET *priv_set, LUID_ATTR set)
 {
-       NTSTATUS ret;
        LUID_ATTR *new_set;
 
-       /* check if the privilege is not already in the list */
-       if (NT_STATUS_IS_OK(check_priv_in_privilege(priv_set, set)))
-               return NT_STATUS_UNSUCCESSFUL;
-
        /* we can allocate memory to add the new privilege */
 
-       new_set = (LUID_ATTR *)talloc_realloc(priv_set->mem_ctx, priv_set->set, (priv_set->count + 1) * (sizeof(LUID_ATTR)));
-       ALLOC_CHECK(new_set, ret, done, "add_privilege");
+       new_set = TALLOC_REALLOC_ARRAY(priv_set->mem_ctx, priv_set->set, LUID_ATTR, priv_set->count + 1);
+       if ( !new_set ) {
+               DEBUG(0,("privilege_set_add: failed to allocate memory!\n"));
+               return False;
+       }       
 
        new_set[priv_set->count].luid.high = set.luid.high;
        new_set[priv_set->count].luid.low = set.luid.low;
@@ -164,187 +386,464 @@ NTSTATUS add_privilege(PRIVILEGE_SET *priv_set, LUID_ATTR set)
        priv_set->count++;
        priv_set->set = new_set;
 
-       ret = NT_STATUS_OK;
+       return True;
+}
+
+/*********************************************************************
+ Generate the LUID_ATTR structure based on a bitmask
+*********************************************************************/
+
+LUID_ATTR get_privilege_luid( SE_PRIV *mask )
+{
+       LUID_ATTR priv_luid;
+       int i;
+
+       priv_luid.attr = 0;
+       priv_luid.luid.high = 0;
+       
+       for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
+       
+               if ( se_priv_equal( &privs[i].se_priv, mask ) ) {
+                       priv_luid.luid = privs[i].luid;
+                       break;
+               }
+       }
 
-done:
-       return ret;
+       return priv_luid;
 }
 
-/****************************************************************************
- add all the privileges to a privilege array
- ****************************************************************************/
-NTSTATUS add_all_privilege(PRIVILEGE_SET *priv_set)
+/*********************************************************************
+ Generate the LUID_ATTR structure based on a bitmask
+*********************************************************************/
+
+const char* get_privilege_dispname( const char *name )
 {
-       NTSTATUS result = NT_STATUS_OK;
-       LUID_ATTR set;
+       int i;
 
-       set.attr = 0;
-       set.luid.high = 0;
+       for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
+       
+               if ( strequal( privs[i].name, name ) ) {
+                       return privs[i].description;
+               }
+       }
 
-       /* TODO: set a proper list of privileges */
-       set.luid.low = SE_PRIV_ADD_USERS;
-       result = add_privilege(priv_set, set);
-       NTSTATUS_CHECK(result, done, "add_all_privilege", "add_privilege");
+       return NULL;
+}
 
-       set.luid.low = SE_PRIV_ADD_MACHINES;
-       result = add_privilege(priv_set, set);
-       NTSTATUS_CHECK(result, done, "add_all_privilege", "add_privilege");
+/*********************************************************************
+ get a list of all privleges for all sids the in list
+*********************************************************************/
 
-       set.luid.low = SE_PRIV_PRINT_OPERATOR;
-       result = add_privilege(priv_set, set);
-       NTSTATUS_CHECK(result, done, "add_all_privilege", "add_privilege");
+BOOL get_privileges_for_sids(SE_PRIV *privileges, DOM_SID *slist, int scount)
+{
+       SE_PRIV mask;
+       int i;
+       BOOL found = False;
+
+       se_priv_copy( privileges, &se_priv_none );
+       
+       for ( i=0; i<scount; i++ ) {
+               /* don't add unless we actually have a privilege assigned */
+
+               if ( !get_privileges( &slist[i], &mask ) )
+                       continue;
+
+               DEBUG(5,("get_privileges_for_sids: sid = %s\nPrivilege set:\n", 
+                       sid_string_static(&slist[i])));
+               dump_se_priv( DBGC_ALL, 5, &mask );
+                       
+               se_priv_add( privileges, &mask );
+               found = True;
+       }
 
-       return result;
+       return found;
 }
 
-/****************************************************************************
- check if the privilege list is empty
- ****************************************************************************/
-NTSTATUS check_empty_privilege(PRIVILEGE_SET *priv_set)
+
+/*********************************************************************
+ travseral functions for privilege_enumerate_accounts
+*********************************************************************/
+
+static int priv_traverse_fn(TDB_CONTEXT *t, TDB_DATA key, TDB_DATA data, void *state)
 {
-       if (!priv_set)
-               return NT_STATUS_INVALID_PARAMETER;
+       PRIV_SID_LIST *priv = state;
+       int  prefixlen = strlen(PRIVPREFIX);
+       DOM_SID sid;
+       fstring sid_string;
+       
+       /* easy check first */
+       
+       if ( data.dsize != sizeof(SE_PRIV) )
+               return 0;
 
-       if (priv_set->count == 0)
-               return NT_STATUS_OK;
+       /* check we have a PRIV_+SID entry */
+
+       if ( strncmp(key.dptr, PRIVPREFIX, prefixlen) != 0)
+               return 0;
+               
+       /* check to see if we are looking for a particular privilege */
+
+       if ( !se_priv_equal(&priv->privilege, &se_priv_none) ) {
+               SE_PRIV mask;
+               
+               se_priv_copy( &mask, (SE_PRIV*)data.dptr );
+               
+               /* if the SID does not have the specified privilege 
+                  then just return */
+                  
+               if ( !is_privilege_assigned( &mask, &priv->privilege) )
+                       return 0;
+       }
+               
+       fstrcpy( sid_string, &key.dptr[strlen(PRIVPREFIX)] );
+
+       if ( !string_to_sid(&sid, sid_string) ) {
+               DEBUG(0,("travsersal_fn_enum__acct: Could not convert SID [%s]\n",
+                       sid_string));
+               return 0;
+       }
 
-       return NT_STATUS_UNSUCCESSFUL;
+       add_sid_to_array( NULL, &sid, &priv->sids.list, &priv->sids.count );
+       
+       return 0;
 }
 
-/****************************************************************************
- check if the privilege is in the privilege list
- ****************************************************************************/
-NTSTATUS check_priv_in_privilege(PRIVILEGE_SET *priv_set, LUID_ATTR set)
+/*********************************************************************
+ Retreive list of privileged SIDs (for _lsa_enumerate_accounts()
+*********************************************************************/
+
+NTSTATUS privilege_enumerate_accounts(DOM_SID **sids, int *num_sids)
+{
+       TDB_CONTEXT *tdb = get_account_pol_tdb();
+       PRIV_SID_LIST priv;
+       
+       ZERO_STRUCT(priv);
+
+       se_priv_copy( &priv.privilege, &se_priv_none );
+
+       tdb_traverse( tdb, priv_traverse_fn, &priv);
+
+       /* give the memory away; caller will free */
+       
+       *sids      = priv.sids.list;
+       *num_sids  = priv.sids.count;
+
+       return NT_STATUS_OK;
+}
+
+/***************************************************************************
+ Add privilege to sid
+****************************************************************************/
+
+BOOL grant_privilege(const DOM_SID *sid, const SE_PRIV *priv_mask)
+{
+       SE_PRIV old_mask, new_mask;
+       
+       ZERO_STRUCT( old_mask );
+       ZERO_STRUCT( new_mask );
+
+       if ( get_privileges( sid, &old_mask ) )
+               se_priv_copy( &new_mask, &old_mask );
+       else
+               se_priv_copy( &new_mask, &se_priv_none );
+
+       se_priv_add( &new_mask, priv_mask );
+
+       DEBUG(10,("grant_privilege: %s\n", sid_string_static(sid)));
+       
+       DEBUGADD( 10, ("original privilege mask:\n"));
+       dump_se_priv( DBGC_ALL, 10, &old_mask );
+       
+       DEBUGADD( 10, ("new privilege mask:\n"));
+       dump_se_priv( DBGC_ALL, 10, &new_mask );
+       
+       return set_privileges( sid, &new_mask );
+}
+
+/*********************************************************************
+ Add a privilege based on its name
+*********************************************************************/
+
+BOOL grant_privilege_by_name(DOM_SID *sid, const char *name)
 {
        int i;
 
-       if (!priv_set)
-               return NT_STATUS_INVALID_PARAMETER;
+       for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
+               if ( strequal(privs[i].name, name) ) {
+                       return grant_privilege( sid, &privs[i].se_priv );
+                }
+        }
 
-       /* if the list is empty, obviously we can't have it */
-       if (NT_STATUS_IS_OK(check_empty_privilege(priv_set)))
-               return NT_STATUS_UNSUCCESSFUL;
+        DEBUG(3, ("grant_privilege_by_name: No Such Privilege Found (%s)\n", name));
 
-       for (i = 0; i < priv_set->count; i++) {
-               LUID_ATTR *cur_set;
+        return False;
+}
 
-               cur_set = &priv_set->set[i];
-               /* check only the low and high part. Checking the attr field has no meaning */
-               if (    (cur_set->luid.low == set.luid.low) &&
-                       (cur_set->luid.high == set.luid.high)   ) {
-                       return NT_STATUS_OK;
-               }
-       }
+/***************************************************************************
+ Remove privilege from sid
+****************************************************************************/
+
+BOOL revoke_privilege(const DOM_SID *sid, const SE_PRIV *priv_mask)
+{
+       SE_PRIV mask;
+       
+       /* if the user has no privileges, then we can't revoke any */
+       
+       if ( !get_privileges( sid, &mask ) )
+               return True;
+       
+       DEBUG(10,("revoke_privilege: %s\n", sid_string_static(sid)));
+       
+       DEBUGADD( 10, ("original privilege mask:\n"));
+       dump_se_priv( DBGC_ALL, 10, &mask );
 
-       return NT_STATUS_UNSUCCESSFUL;
+       se_priv_remove( &mask, priv_mask );
+       
+       DEBUGADD( 10, ("new privilege mask:\n"));
+       dump_se_priv( DBGC_ALL, 10, &mask );
+       
+       return set_privileges( sid, &mask );
+}
+
+/*********************************************************************
+ Revoke all privileges
+*********************************************************************/
+
+BOOL revoke_all_privileges( DOM_SID *sid )
+{
+       return revoke_privilege( sid, &se_priv_all );
+}
+
+/*********************************************************************
+ Add a privilege based on its name
+*********************************************************************/
+
+BOOL revoke_privilege_by_name(DOM_SID *sid, const char *name)
+{
+       int i;
+
+       for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
+               if ( strequal(privs[i].name, name) ) {
+                       return revoke_privilege( sid, &privs[i].se_priv );
+                }
+        }
+
+        DEBUG(3, ("revoke_privilege_by_name: No Such Privilege Found (%s)\n", name));
+
+        return False;
+}
+
+/***************************************************************************
+ Retrieve the SIDs assigned to a given privilege
+****************************************************************************/
+
+NTSTATUS privilege_create_account(const DOM_SID *sid )
+{
+       return ( grant_privilege(sid, &se_priv_none) ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL);
 }
 
 /****************************************************************************
- remove a privilege from a privilege array
+ initialise a privilege list and set the talloc context 
  ****************************************************************************/
-NTSTATUS remove_privilege(PRIVILEGE_SET *priv_set, LUID_ATTR set)
+NTSTATUS privilege_set_init(PRIVILEGE_SET *priv_set)
 {
-       NTSTATUS ret;
-       LUID_ATTR *new_set;
-       LUID_ATTR *old_set;
-       int i,j;
+       TALLOC_CTX *mem_ctx;
+       
+       ZERO_STRUCTP( priv_set );
 
-       if (!priv_set)
-               return NT_STATUS_INVALID_PARAMETER;
+       mem_ctx = talloc_init("privilege set");
+       if ( !mem_ctx ) {
+               DEBUG(0,("privilege_set_init: failed to initialize talloc ctx!\n"));
+               return NT_STATUS_NO_MEMORY;
+       }
 
-       /* check if the privilege is in the list */
-       if (!NT_STATUS_IS_OK(check_priv_in_privilege(priv_set, set)))
-               return NT_STATUS_UNSUCCESSFUL;
+       priv_set->mem_ctx = mem_ctx;
 
-       /* special case if it's the only privilege in the list */
-       if (priv_set->count == 1) {
-               reset_privilege(priv_set);      
-               return NT_STATUS_OK;
-       }
+       return NT_STATUS_OK;
+}
 
-       /* 
-        * the privilege is there, create a new list,
-        * and copy the other privileges
-        */
+/****************************************************************************
+  initialise a privilege list and with someone else's talloc context 
+****************************************************************************/
 
-       old_set = priv_set->set;
+NTSTATUS privilege_set_init_by_ctx(TALLOC_CTX *mem_ctx, PRIVILEGE_SET *priv_set)
+{
+       ZERO_STRUCTP( priv_set );
+       
+       priv_set->mem_ctx = mem_ctx;
+       priv_set->ext_ctx = True;
 
-       new_set = (LUID_ATTR *)talloc(priv_set->mem_ctx, (priv_set->count - 1) * (sizeof(LUID_ATTR)));
-       ALLOC_CHECK(new_set, ret, done, "remove_privilege");
+       return NT_STATUS_OK;
+}
 
-       for (i=0, j=0; i < priv_set->count; i++) {
-               if (    (old_set[i].luid.low == set.luid.low) && 
-                       (old_set[i].luid.high == set.luid.high) ) {
-                       continue;
-               }
-               
-               new_set[j].luid.low = old_set[i].luid.low;
-               new_set[j].luid.high = old_set[i].luid.high;
-               new_set[j].attr = old_set[i].attr;
+/****************************************************************************
+ Free all memory used by a PRIVILEGE_SET
+****************************************************************************/
 
-               j++;
+void privilege_set_free(PRIVILEGE_SET *priv_set)
+{
+       if ( !priv_set )
+               return;
+
+       if ( !( priv_set->ext_ctx ) )
+               talloc_destroy( priv_set->mem_ctx );
+
+       ZERO_STRUCTP( priv_set );
+}
+
+/****************************************************************************
+ duplicate alloc luid_attr
+ ****************************************************************************/
+
+NTSTATUS dup_luid_attr(TALLOC_CTX *mem_ctx, LUID_ATTR **new_la, LUID_ATTR *old_la, int count)
+{
+       int i;
+
+       if ( !old_la )
+               return NT_STATUS_OK;
+
+       *new_la = TALLOC_ARRAY(mem_ctx, LUID_ATTR, count);
+       if ( !*new_la ) {
+               DEBUG(0,("dup_luid_attr: failed to alloc new LUID_ATTR array [%d]\n", count));
+               return NT_STATUS_NO_MEMORY;
        }
-       
-       if (j != priv_set->count - 1) {
-               DEBUG(0,("remove_privilege: mismatch ! difference is not -1\n"));
-               DEBUGADD(0,("old count:%d, new count:%d\n", priv_set->count, j));
-               return NT_STATUS_INTERNAL_ERROR;
+
+       for (i=0; i<count; i++) {
+               (*new_la)[i].luid.high = old_la[i].luid.high;
+               (*new_la)[i].luid.low = old_la[i].luid.low;
+               (*new_la)[i].attr = old_la[i].attr;
        }
-               
-       /* ok everything is fine */
-       
-       priv_set->count--;
-       priv_set->set = new_set;
        
-       ret = NT_STATUS_OK;
+       return NT_STATUS_OK;
+}
+
+/****************************************************************************
+ Does the user have the specified privilege ?  We only deal with one privilege
+ at a time here.
+*****************************************************************************/
+
+BOOL user_has_privileges(NT_USER_TOKEN *token, const SE_PRIV *privilege)
+{
+       if ( !token )
+               return False;
 
-done:
-       return ret;
+       return is_privilege_assigned( &token->privileges, privilege );
 }
 
 /****************************************************************************
- duplicates a privilege array
- the new privilege set must be passed inited
- (use init_privilege or init_priv_with_ctx)
- ****************************************************************************/
-NTSTATUS dup_priv_set(PRIVILEGE_SET *new_priv_set, PRIVILEGE_SET *priv_set)
+ Does the user have any of the specified privileges ?  We only deal with one privilege
+ at a time here.
+*****************************************************************************/
+
+BOOL user_has_any_privilege(NT_USER_TOKEN *token, const SE_PRIV *privilege)
 {
-       NTSTATUS ret;
-       LUID_ATTR *new_set;
-       LUID_ATTR *old_set;
+       if ( !token )
+               return False;
+
+       return is_any_privilege_assigned( &token->privileges, privilege );
+}
+
+/****************************************************************************
+ Convert a LUID to a named string
+****************************************************************************/
+
+char* luid_to_privilege_name(const LUID *set)
+{
+       static fstring name;
        int i;
 
-       if (!new_priv_set || !priv_set)
-               return NT_STATUS_INVALID_PARAMETER;
+       if (set->high != 0)
+               return NULL;
 
-       /* special case if there are no privileges in the list */
-       if (priv_set->count == 0) {
-               return NT_STATUS_OK;
+       for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
+               if ( set->low == privs[i].luid.low ) {
+                       fstrcpy( name, privs[set->low - 1].name );
+                       return name;
+               }
        }
+       
+       return NULL;
+}
+
+/*******************************************************************
+ return the number of elements in the privlege array
+*******************************************************************/
+
+int count_all_privileges( void )
+{
+       static int count;
+       
+       if ( count )
+               return count;
 
-       /* 
-        * create a new list,
-        * and copy the other privileges
-        */
+       /* loop over the array and count it */  
+       for ( count=0; !se_priv_equal(&privs[count].se_priv, &se_priv_end); count++ ) ;
 
-       old_set = priv_set->set;
+       return count;
+}
 
-       new_set = (LUID_ATTR *)talloc(new_priv_set->mem_ctx, (priv_set->count - 1) * (sizeof(LUID_ATTR)));
-       ALLOC_CHECK(new_set, ret, done, "dup_priv_set");
+/*******************************************************************
+*******************************************************************/
 
-       for (i=0; i < priv_set->count; i++) {
+BOOL se_priv_to_privilege_set( PRIVILEGE_SET *set, SE_PRIV *mask )
+{
+       int i;
+       uint32 num_privs = count_all_privileges();
+       LUID_ATTR luid;
+       
+       luid.attr = 0;
+       luid.luid.high = 0;
+       
+       for ( i=0; i<num_privs; i++ ) {
+               if ( !is_privilege_assigned(mask, &privs[i].se_priv) )
+                       continue;
                
-               new_set[i].luid.low = old_set[i].luid.low;
-               new_set[i].luid.high = old_set[i].luid.high;
-               new_set[i].attr = old_set[i].attr;
+               luid.luid = privs[i].luid;
+               
+               if ( !privilege_set_add( set, luid ) )
+                       return False;
        }
-                       
-       new_priv_set->count = priv_set->count;
-       new_priv_set->control = priv_set->control;
-       new_priv_set->set = new_set;
+
+       return True;
+}
+
+/*******************************************************************
+*******************************************************************/
+
+BOOL privilege_set_to_se_priv( SE_PRIV *mask, PRIVILEGE_SET *privset )
+{
+       int i;
+       uint32 num_privs = count_all_privileges();
+       
+       ZERO_STRUCTP( mask );
+       
+       for ( i=0; i<privset->count; i++ ) {
+               SE_PRIV r;
        
-       ret = NT_STATUS_OK;
+               /* sanity check for invalid privilege.  we really
+                  only care about the low 32 bits */
+                  
+               if ( privset->set[i].luid.high != 0 )
+                       return False;
+               
+               /* make sure :LUID.low is in range */   
+               if ( privset->set[i].luid.low == 0 || privset->set[i].luid.low > num_privs )
+                       return False;
+               
+               r = privs[privset->set[i].luid.low - 1].se_priv;
+               se_priv_add( mask, &r );
+       }
+
+       return True;
+}
+
+/*******************************************************************
+*******************************************************************/
 
-done:
-       return ret;
+BOOL is_privileged_sid( DOM_SID *sid )
+{
+       SE_PRIV mask;
+       
+       return get_privileges( sid, &mask );
 }