167 lines
3 KiB
C
167 lines
3 KiB
C
|
/************************************************************************
|
||
|
|
|
||
|
|
Copyright (c) 2000 - 2000 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name :
|
||
|
|
|
||
|
|
logontable.h
|
||
|
|
|
||
|
|
Abstract :
|
||
|
|
|
||
|
|
Header file for the logon table
|
||
|
|
|
||
|
|
Author :
|
||
|
|
|
||
|
|
Revision History :
|
||
|
|
|
||
|
|
***********************************************************************/
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include <wtypes.h>
|
||
|
|
#include <unknwn.h>
|
||
|
|
|
||
|
|
#include <utility>
|
||
|
|
#include <set>
|
||
|
|
#include <map>
|
||
|
|
#include "tasksched.h"
|
||
|
|
|
||
|
|
class CUser
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
|
||
|
|
CUser( HANDLE Token );
|
||
|
|
~CUser();
|
||
|
|
|
||
|
|
long IncrementRefCount();
|
||
|
|
long DecrementRefCount();
|
||
|
|
|
||
|
|
void SetCookie( DWORD cookie )
|
||
|
|
{
|
||
|
|
_Cookie = cookie;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD GetCookie()
|
||
|
|
{
|
||
|
|
return _Cookie;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD CopyToken( HANDLE * pToken )
|
||
|
|
{
|
||
|
|
if (!DuplicateHandle( GetCurrentProcess(),
|
||
|
|
_Token,
|
||
|
|
GetCurrentProcess(),
|
||
|
|
pToken,
|
||
|
|
TOKEN_ALL_ACCESS,
|
||
|
|
FALSE, // no inheritance
|
||
|
|
0 // no extra options
|
||
|
|
))
|
||
|
|
{
|
||
|
|
return GetLastError();
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT Impersonate()
|
||
|
|
{
|
||
|
|
if (!ImpersonateLoggedOnUser(_Token))
|
||
|
|
{
|
||
|
|
return HRESULT_FROM_WIN32( GetLastError() );
|
||
|
|
}
|
||
|
|
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
SidHandle & QuerySid()
|
||
|
|
{
|
||
|
|
return _Sid;
|
||
|
|
}
|
||
|
|
|
||
|
|
void Dump();
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
LaunchProcess(
|
||
|
|
StringHandle CmdLine
|
||
|
|
);
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
long _ReferenceCount;
|
||
|
|
HANDLE _Token;
|
||
|
|
SidHandle _Sid;
|
||
|
|
DWORD _Cookie;
|
||
|
|
|
||
|
|
//--------------------------------------------------------------------
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
class CLoggedOnUsers
|
||
|
|
{
|
||
|
|
class CSessionList : public std::map<DWORD, CUser *>
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
|
||
|
|
void Dump();
|
||
|
|
};
|
||
|
|
|
||
|
|
class CUserList : public std::multimap<SidHandle, CUser *, CSidSorter>
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
|
||
|
|
~CUserList();
|
||
|
|
|
||
|
|
CUser *
|
||
|
|
RemoveByCookie(
|
||
|
|
SidHandle sid,
|
||
|
|
DWORD cookie
|
||
|
|
);
|
||
|
|
|
||
|
|
bool
|
||
|
|
RemovePair(
|
||
|
|
SidHandle sid,
|
||
|
|
CUser * user
|
||
|
|
);
|
||
|
|
|
||
|
|
CUser *
|
||
|
|
FindSid(
|
||
|
|
SidHandle sid
|
||
|
|
);
|
||
|
|
|
||
|
|
void Dump();
|
||
|
|
};
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
CLoggedOnUsers( TaskScheduler & sched );
|
||
|
|
~CLoggedOnUsers();
|
||
|
|
|
||
|
|
HRESULT AddServiceAccounts();
|
||
|
|
HRESULT AddActiveUsers();
|
||
|
|
|
||
|
|
HRESULT LogonSession( DWORD session );
|
||
|
|
HRESULT LogoffSession( DWORD session );
|
||
|
|
|
||
|
|
HRESULT LogonService( HANDLE token, DWORD * cookie );
|
||
|
|
HRESULT LogoffService( SidHandle Sid, DWORD cookie );
|
||
|
|
|
||
|
|
CUser * FindUser( SidHandle sid, DWORD session );
|
||
|
|
|
||
|
|
void Dump();
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
//--------------------------------------------------------------------
|
||
|
|
|
||
|
|
TaskScheduler & m_TaskScheduler;
|
||
|
|
CSessionList m_ActiveSessions;
|
||
|
|
CUserList m_ActiveUsers;
|
||
|
|
CUserList m_ActiveServiceAccounts;
|
||
|
|
long m_CurrentCookie;
|
||
|
|
|
||
|
|
CLogonNotification * m_SensNotifier;
|
||
|
|
};
|
||
|
|
|
||
|
|
|