s3-privileges: add privilege_delete_account().
[ira/wip.git] / source3 / lib / privileges.c
index c0f7857c95b8c231504a7b39035da86fb1b1dfc2..c8be360dc643c16ef6847bb48e7e428f3c1c83ae 100644 (file)
    Copyright (C) Jean François Micouleau      1998-2001
    Copyright (C) Simo Sorce                    2002-2003
    Copyright (C) Gerald (Jerry) Carter          2005
-   
+   Copyright (C) Michael Adam                  2007
+
    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
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
-   
+
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
-   
+
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
+
 #include "includes.h"
 
 #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;
-const SE_PRIV se_take_ownership  = SE_TAKE_OWNERSHIP;
-
-/********************************************************************
- 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 }},
-
-       {SE_END, "", "", { 0x0, 0x0 }}
-};
-
 typedef struct {
        size_t count;
        DOM_SID *list;
 } SID_LIST;
 
 typedef struct {
+       TALLOC_CTX *mem_ctx;
        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 )
-{
-       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;
-
-       for ( i=0; i<SE_PRIV_MASKSIZE; i++ ) {
-               mask->mask[i] |= addpriv->mask[i];
-       }
-}
-
-/***************************************************************************
- 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];
-       }
-}
-
-/***************************************************************************
- 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 );
-}
-
-/***************************************************************************
- check if 2 SE_PRIV structure are equal
-****************************************************************************/
-
-static BOOL se_priv_equal( const SE_PRIV *mask1, const SE_PRIV *mask2 )
-{      
-       return ( memcmp(mask1, mask2, sizeof(SE_PRIV)) == 0 );
-}
-
-/***************************************************************************
- check if a SE_PRIV has any assigned privileges
-****************************************************************************/
-
-static BOOL se_priv_empty( const SE_PRIV *mask )
-{
-       SE_PRIV p1;
-       int i;
-       
-       se_priv_copy( &p1, mask );
-
-       for ( i=0; i<SE_PRIV_MASKSIZE; i++ ) {
-               p1.mask[i] &= se_priv_all.mask[i];
-       }
-       
-       return se_priv_equal( &p1, &se_priv_none );
-}
-
-/*********************************************************************
- Lookup the SE_PRIV value for a privilege name 
-*********************************************************************/
-
-BOOL se_priv_from_name( const char *name, SE_PRIV *mask )
-{
-       int i;
-
-       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;
-               }
-       }
-
-       return False;
-}
-
-/***************************************************************************
- dump an SE_PRIV structure to the log files
-****************************************************************************/
-
-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"));
-}
-
-/***************************************************************************
- Retrieve the privilege mask (set) for a given SID
-****************************************************************************/
 
-static BOOL get_privileges( const DOM_SID *sid, SE_PRIV *mask )
+static bool get_privileges( const DOM_SID *sid, SE_PRIV *mask )
 {
-       TDB_CONTEXT *tdb = get_account_pol_tdb();
-       fstring keystr;
-       TDB_DATA key, data;
+       struct db_context *db = get_account_pol_db();
+       fstring tmp, keystr;
+       TDB_DATA data;
 
        /* Fail if the admin has not enable privileges */
-       
+
        if ( !lp_enable_privileges() ) {
                return False;
        }
-       
-       if ( !tdb )
+
+       if ( db == NULL )
                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 );
-       
+       fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
+
+       data = dbwrap_fetch_bystring( db, talloc_tos(), keystr );
+
        if ( !data.dptr ) {
-               DEBUG(3,("get_privileges: No privileges assigned to SID [%s]\n",
-                       sid_string_static(sid)));
+               DEBUG(3, ("get_privileges: No privileges assigned to SID "
+                         "[%s]\n", sid_string_dbg(sid)));
                return False;
        }
-       
+
        SMB_ASSERT( data.dsize == sizeof( SE_PRIV ) );
-       
+
        se_priv_copy( mask, (SE_PRIV*)data.dptr );
-       SAFE_FREE(data.dptr);
+       TALLOC_FREE(data.dptr);
 
        return True;
 }
@@ -274,16 +76,16 @@ static BOOL get_privileges( const DOM_SID *sid, SE_PRIV *mask )
  Store the privilege mask (set) for a given SID
 ****************************************************************************/
 
-static BOOL set_privileges( const DOM_SID *sid, SE_PRIV *mask )
+static bool set_privileges( const DOM_SID *sid, SE_PRIV *mask )
 {
-       TDB_CONTEXT *tdb = get_account_pol_tdb();
-       fstring keystr;
-       TDB_DATA key, data;
-       
+       struct db_context *db = get_account_pol_db();
+       fstring tmp, keystr;
+       TDB_DATA data;
+
        if ( !lp_enable_privileges() )
                return False;
 
-       if ( !tdb )
+       if ( db == NULL )
                return False;
 
        if ( !sid || (sid->num_auths == 0) ) {
@@ -292,174 +94,40 @@ static BOOL set_privileges( const DOM_SID *sid, SE_PRIV *mask )
        }
 
        /* 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);
-
-       return ( tdb_store(tdb, key, data, TDB_REPLACE) != -1 );
-}
-
-/****************************************************************************
- check if the privilege is in the privilege list
-****************************************************************************/
-
-static BOOL is_privilege_assigned( const SE_PRIV *privileges,
-                                  const SE_PRIV *check )
-{
-       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 );
-}
-
-/****************************************************************************
- check if the privilege is in the privilege list
-****************************************************************************/
-
-static BOOL is_any_privilege_assigned( SE_PRIV *privileges, const SE_PRIV *check )
-{
-       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
- ****************************************************************************/
+       fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
 
-static BOOL privilege_set_add(PRIVILEGE_SET *priv_set, LUID_ATTR set)
-{
-       LUID_ATTR *new_set;
-
-       /* we can allocate memory to add the new 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;
-       new_set[priv_set->count].attr = set.attr;
-
-       priv_set->count++;
-       priv_set->set = new_set;
-
-       return True;
-}
-
-/*********************************************************************
- Generate the LUID_ATTR structure based on a bitmask
- The assumption here is that the privilege has already been validated
- so we are guaranteed to find it in the list. 
-*********************************************************************/
-
-LUID_ATTR get_privilege_luid( SE_PRIV *mask )
-{
-       LUID_ATTR priv_luid;
-       int i;
-
-       ZERO_STRUCT( priv_luid );
-       
-       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;
-               }
-       }
-
-       return priv_luid;
-}
-
-/*********************************************************************
- Generate the LUID_ATTR structure based on a bitmask
-*********************************************************************/
-
-const char* get_privilege_dispname( const char *name )
-{
-       int i;
+       /* no packing.  static size structure, just write it out */
 
-       for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
-       
-               if ( strequal( privs[i].name, name ) ) {
-                       return privs[i].description;
-               }
-       }
+       data.dptr  = (uint8 *)mask;
+       data.dsize = sizeof(SE_PRIV);
 
-       return NULL;
+       return NT_STATUS_IS_OK(dbwrap_store_bystring(db, keystr, data,
+                                                    TDB_REPLACE));
 }
 
 /*********************************************************************
- get a list of all privleges for all sids the in list
+ get a list of all privileges for all sids in the list
 *********************************************************************/
 
-BOOL get_privileges_for_sids(SE_PRIV *privileges, DOM_SID *slist, int scount)
+bool get_privileges_for_sids(SE_PRIV *privileges, DOM_SID *slist, int scount)
 {
        SE_PRIV mask;
        int i;
-       BOOL found = False;
+       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])));
+               DEBUG(5,("get_privileges_for_sids: sid = %s\nPrivilege "
+                        "set:\n", sid_string_dbg(&slist[i])));
                dump_se_priv( DBGC_ALL, 5, &mask );
-                       
+
                se_priv_add( privileges, &mask );
                found = True;
        }
@@ -469,41 +137,41 @@ BOOL get_privileges_for_sids(SE_PRIV *privileges, DOM_SID *slist, int scount)
 
 
 /*********************************************************************
- travseral functions for privilege_enumerate_accounts
+ traversal functions for privilege_enumerate_accounts
 *********************************************************************/
 
-static int priv_traverse_fn(TDB_CONTEXT *t, TDB_DATA key, TDB_DATA data, void *state)
+static int priv_traverse_fn(struct db_record *rec, void *state)
 {
        PRIV_SID_LIST *priv = (PRIV_SID_LIST *)state;
        int  prefixlen = strlen(PRIVPREFIX);
        DOM_SID sid;
        fstring sid_string;
-       
+
        /* easy check first */
-       
-       if ( data.dsize != sizeof(SE_PRIV) )
+
+       if (rec->value.dsize != sizeof(SE_PRIV) )
                return 0;
 
        /* check we have a PRIV_+SID entry */
 
-       if ( strncmp(key.dptr, PRIVPREFIX, prefixlen) != 0)
+       if ( strncmp((char *)rec->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 
+
+               se_priv_copy( &mask, (SE_PRIV*)rec->value.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)] );
+
+       fstrcpy( sid_string, (char *)&(rec->key.dptr[strlen(PRIVPREFIX)]) );
 
        /* this is a last ditch safety check to preventing returning
           and invalid SID (i've somehow run into this on development branches) */
@@ -517,10 +185,13 @@ static int priv_traverse_fn(TDB_CONTEXT *t, TDB_DATA key, TDB_DATA data, void *s
                return 0;
        }
 
-       if (!add_sid_to_array( NULL, &sid, &priv->sids.list, &priv->sids.count )) {
+       if (!NT_STATUS_IS_OK(add_sid_to_array(priv->mem_ctx, &sid,
+                                             &priv->sids.list,
+                                             &priv->sids.count)))
+       {
                return 0;
        }
-       
+
        return 0;
 }
 
@@ -530,10 +201,10 @@ static int priv_traverse_fn(TDB_CONTEXT *t, TDB_DATA key, TDB_DATA data, void *s
 
 NTSTATUS privilege_enumerate_accounts(DOM_SID **sids, int *num_sids)
 {
-       TDB_CONTEXT *tdb = get_account_pol_tdb();
+       struct db_context *db = get_account_pol_db();
        PRIV_SID_LIST priv;
-       
-       if (!tdb) {
+
+       if (db == NULL) {
                return NT_STATUS_ACCESS_DENIED;
        }
 
@@ -541,10 +212,39 @@ NTSTATUS privilege_enumerate_accounts(DOM_SID **sids, int *num_sids)
 
        se_priv_copy( &priv.privilege, &se_priv_none );
 
-       tdb_traverse( tdb, priv_traverse_fn, &priv);
+       db->traverse_read(db, priv_traverse_fn, &priv);
 
        /* give the memory away; caller will free */
-       
+
+       *sids      = priv.sids.list;
+       *num_sids  = priv.sids.count;
+
+       return NT_STATUS_OK;
+}
+
+/*********************************************************************
+ Retrieve list of SIDs granted a particular privilege
+*********************************************************************/
+
+NTSTATUS privilege_enum_sids(const SE_PRIV *mask, TALLOC_CTX *mem_ctx,
+                            DOM_SID **sids, int *num_sids)
+{
+       struct db_context *db = get_account_pol_db();
+       PRIV_SID_LIST priv;
+
+       if (db == NULL) {
+               return NT_STATUS_ACCESS_DENIED;
+       }
+
+       ZERO_STRUCT(priv);
+
+       se_priv_copy(&priv.privilege, mask);
+       priv.mem_ctx = mem_ctx;
+
+       db->traverse_read(db, priv_traverse_fn, &priv);
+
+       /* give the memory away; caller will free */
+
        *sids      = priv.sids.list;
        *num_sids  = priv.sids.count;
 
@@ -555,10 +255,10 @@ NTSTATUS privilege_enumerate_accounts(DOM_SID **sids, int *num_sids)
  Add privilege to sid
 ****************************************************************************/
 
-BOOL grant_privilege(const DOM_SID *sid, const SE_PRIV *priv_mask)
+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 );
 
@@ -569,14 +269,14 @@ BOOL grant_privilege(const DOM_SID *sid, const SE_PRIV *priv_mask)
 
        se_priv_add( &new_mask, priv_mask );
 
-       DEBUG(10,("grant_privilege: %s\n", sid_string_static(sid)));
-       
+       DEBUG(10,("grant_privilege: %s\n", sid_string_dbg(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 );
 }
 
@@ -584,44 +284,42 @@ BOOL grant_privilege(const DOM_SID *sid, const SE_PRIV *priv_mask)
  Add a privilege based on its name
 *********************************************************************/
 
-BOOL grant_privilege_by_name(DOM_SID *sid, const char *name)
+bool grant_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 grant_privilege( sid, &privs[i].se_priv );
-                }
-        }
+       SE_PRIV mask;
 
-        DEBUG(3, ("grant_privilege_by_name: No Such Privilege Found (%s)\n", name));
+       if (! se_priv_from_name(name, &mask)) {
+               DEBUG(3, ("grant_privilege_by_name: "
+                         "No Such Privilege Found (%s)\n", name));
+               return False;
+       }
 
-        return False;
+       return grant_privilege( sid, &mask );
 }
 
 /***************************************************************************
  Remove privilege from sid
 ****************************************************************************/
 
-BOOL revoke_privilege(const DOM_SID *sid, const SE_PRIV *priv_mask)
+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)));
-       
+
+       DEBUG(10,("revoke_privilege: %s\n", sid_string_dbg(sid)));
+
        DEBUGADD( 10, ("original privilege mask:\n"));
        dump_se_priv( DBGC_ALL, 10, &mask );
 
        se_priv_remove( &mask, priv_mask );
-       
+
        DEBUGADD( 10, ("new privilege mask:\n"));
        dump_se_priv( DBGC_ALL, 10, &mask );
-       
+
        return set_privileges( sid, &mask );
 }
 
@@ -629,7 +327,7 @@ BOOL revoke_privilege(const DOM_SID *sid, const SE_PRIV *priv_mask)
  Revoke all privileges
 *********************************************************************/
 
-BOOL revoke_all_privileges( DOM_SID *sid )
+bool revoke_all_privileges( DOM_SID *sid )
 {
        return revoke_privilege( sid, &se_priv_all );
 }
@@ -638,19 +336,18 @@ BOOL revoke_all_privileges( DOM_SID *sid )
  Add a privilege based on its name
 *********************************************************************/
 
-BOOL revoke_privilege_by_name(DOM_SID *sid, const char *name)
+bool revoke_privilege_by_name(DOM_SID *sid, const char *name)
 {
-       int i;
+       SE_PRIV mask;
 
-       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 );
-                }
-        }
+       if (! se_priv_from_name(name, &mask)) {
+               DEBUG(3, ("revoke_privilege_by_name: "
+                         "No Such Privilege Found (%s)\n", name));
+               return False;
+       }
 
-        DEBUG(3, ("revoke_privilege_by_name: No Such Privilege Found (%s)\n", name));
+       return revoke_privilege(sid, &mask);
 
-        return False;
 }
 
 /***************************************************************************
@@ -662,14 +359,42 @@ NTSTATUS privilege_create_account(const DOM_SID *sid )
        return ( grant_privilege(sid, &se_priv_none) ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL);
 }
 
+/***************************************************************************
+ Delete a privileged account
+****************************************************************************/
+
+NTSTATUS privilege_delete_account(const struct dom_sid *sid)
+{
+       struct db_context *db = get_account_pol_db();
+       fstring tmp, keystr;
+
+       if (!lp_enable_privileges()) {
+               return NT_STATUS_OK;
+       }
+
+       if (!db) {
+               return NT_STATUS_INVALID_HANDLE;
+       }
+
+       if (!sid || (sid->num_auths == 0)) {
+               return NT_STATUS_INVALID_SID;
+       }
+
+       /* PRIV_<SID> (NULL terminated) as the key */
+
+       fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
+
+       return dbwrap_delete_bystring(db, keystr);
+}
+
 /****************************************************************************
- initialise a privilege list and set the talloc context 
+ initialise a privilege list and set the talloc context
  ****************************************************************************/
+
 NTSTATUS privilege_set_init(PRIVILEGE_SET *priv_set)
 {
        TALLOC_CTX *mem_ctx;
-       
+
        ZERO_STRUCTP( priv_set );
 
        mem_ctx = talloc_init("privilege set");
@@ -684,13 +409,13 @@ NTSTATUS privilege_set_init(PRIVILEGE_SET *priv_set)
 }
 
 /****************************************************************************
-  initialise a privilege list and with someone else's talloc context 
+  initialise a privilege list and with someone else's talloc context
 ****************************************************************************/
 
 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;
 
@@ -723,10 +448,14 @@ NTSTATUS dup_luid_attr(TALLOC_CTX *mem_ctx, LUID_ATTR **new_la, LUID_ATTR *old_l
        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 (count) {
+               *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;
+               }
+       } else {
+               *new_la = NULL;
        }
 
        for (i=0; i<count; i++) {
@@ -734,166 +463,29 @@ NTSTATUS dup_luid_attr(TALLOC_CTX *mem_ctx, LUID_ATTR **new_la, LUID_ATTR *old_l
                (*new_la)[i].luid.low = old_la[i].luid.low;
                (*new_la)[i].attr = old_la[i].attr;
        }
-       
-       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(const NT_USER_TOKEN *token, const SE_PRIV *privilege)
-{
-       if ( !token )
-               return False;
-
-       return is_privilege_assigned( &token->privileges, privilege );
-}
-
-/****************************************************************************
- 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)
-{
-       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 (set->high != 0)
-               return NULL;
-
-       for ( i=0; !se_priv_equal(&privs[i].se_priv, &se_priv_end); i++ ) {
-               if ( set->low == privs[i].luid.low ) {
-                       fstrcpy( name, privs[i].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;
-
-       /* loop over the array and count it */  
-       for ( count=0; !se_priv_equal(&privs[count].se_priv, &se_priv_end); count++ ) ;
-
-       return count;
-}
-
-/*******************************************************************
-*******************************************************************/
-
-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;
-               
-               luid.luid = privs[i].luid;
-               
-               if ( !privilege_set_add( set, luid ) )
-                       return False;
-       }
-
-       return True;
-}
-
-/*******************************************************************
-*******************************************************************/
-
-static BOOL luid_to_se_priv( LUID *luid, SE_PRIV *mask )
-{
-       int i;
-       uint32 num_privs = count_all_privileges();
-       
-       for ( i=0; i<num_privs; i++ ) {
-               if ( luid->low == privs[i].luid.low ) {
-                       se_priv_copy( mask, &privs[i].se_priv );
-                       return True;
-               }
-       }
-
-       return False;
-}
-
-/*******************************************************************
-*******************************************************************/
-
-BOOL privilege_set_to_se_priv( SE_PRIV *mask, PRIVILEGE_SET *privset )
-{
-       int i;
-       
-       ZERO_STRUCTP( mask );
-       
-       for ( i=0; i<privset->count; i++ ) {
-               SE_PRIV r;
-       
-               /* sanity check for invalid privilege.  we really
-                  only care about the low 32 bits */
-                  
-               if ( privset->set[i].luid.high != 0 )
-                       return False;
-               
-               if ( luid_to_se_priv( &privset->set[i].luid, &r ) )             
-                       se_priv_add( mask, &r );
-       }
-
-       return True;
+       return NT_STATUS_OK;
 }
 
 /*******************************************************************
 *******************************************************************/
 
-BOOL is_privileged_sid( const DOM_SID *sid )
+bool is_privileged_sid( const DOM_SID *sid )
 {
        SE_PRIV mask;
-       
+
        return get_privileges( sid, &mask );
 }
 
 /*******************************************************************
 *******************************************************************/
 
-BOOL grant_all_privileges( const DOM_SID *sid )
+bool grant_all_privileges( const DOM_SID *sid )
 {
-       int i;
        SE_PRIV mask;
-       uint32 num_privs = count_all_privileges();
 
-       se_priv_copy( &mask, &se_priv_none );
-       
-       for ( i=0; i<num_privs; i++ ) {
-               se_priv_add(&mask, &privs[i].se_priv); 
+       if (!se_priv_put_all_privileges(&mask)) {
+               return False;
        }
 
        return grant_privilege( sid, &mask );