zydronium/SMBLibrary
1
0
Fork 0
mirror of https://github.com/TalAloni/SMBLibrary.git synced 2025-08-11 09:59:19 +02:00
Code Issues Projects Releases Packages Wiki Activity

SSPIHelper: Added GetUserName and GetSessionKey methods

Browse source
This commit is contained in:
Tal Aloni 2017-03-01 22:14:15 +02:00
parent d3c46361d2
commit 104d18b81b
1 changed files with 61 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

66
SMBLibrary/Win32/Security/SSPIHelper.cs
View file

@ -47,6 +47,8 @@ namespace SMBLibrary.Win32.Security
private const uint SEC_WINNT_AUTH_IDENTITY_ANSI = 1; private const uint SEC_WINNT_AUTH_IDENTITY_ANSI = 1;
private const uint SEC_WINNT_AUTH_IDENTITY_UNICODE = 2; private const uint SEC_WINNT_AUTH_IDENTITY_UNICODE = 2;
private const uint SECPKG_ATTR_NAME = 1; // Username
private const uint SECPKG_ATTR_SESSION_KEY = 9;
private const uint SECPKG_ATTR_ACCESS_TOKEN = 18; private const uint SECPKG_ATTR_ACCESS_TOKEN = 18;
[StructLayout(LayoutKind.Sequential)] [StructLayout(LayoutKind.Sequential)]
@ -71,6 +73,13 @@ namespace SMBLibrary.Win32.Security
public uint Flags; public uint Flags;
}; };
[StructLayout(LayoutKind.Sequential)]
public struct SecPkgContext_SessionKey
{
public uint SessionKeyLength;
public IntPtr SessionKey;
}
[DllImport("secur32.dll", SetLastError = true)] [DllImport("secur32.dll", SetLastError = true)]
private static extern uint AcquireCredentialsHandle( private static extern uint AcquireCredentialsHandle(
string pszPrincipal, string pszPrincipal,
@ -141,7 +150,19 @@ namespace SMBLibrary.Win32.Security
private static extern uint QueryContextAttributes( private static extern uint QueryContextAttributes(
ref SecHandle phContext, ref SecHandle phContext,
uint ulAttribute, uint ulAttribute,
out IntPtr pBuffer); out string value);
[DllImport("secur32.Dll", SetLastError = true)]
private static extern uint QueryContextAttributes(
ref SecHandle phContext,
uint ulAttribute,
out IntPtr value);
[DllImport("secur32.Dll", SetLastError = true)]
private static extern uint QueryContextAttributes(
ref SecHandle phContext,
uint ulAttribute,
out SecPkgContext_SessionKey value);
[DllImport("Secur32.dll")] [DllImport("Secur32.dll")]
private extern static uint FreeContextBuffer( private extern static uint FreeContextBuffer(
@ -378,13 +399,48 @@ namespace SMBLibrary.Win32.Security
} }
} }
public static IntPtr GetAccessToken(SecHandle serverContext) public static string GetUserName(SecHandle context)
{ {
IntPtr pBuffer; string userName;
uint result = QueryContextAttributes(ref serverContext, SECPKG_ATTR_ACCESS_TOKEN, out pBuffer); uint result = QueryContextAttributes(ref context, SECPKG_ATTR_NAME, out userName);
if (result == SEC_E_OK) if (result == SEC_E_OK)
{ {
return pBuffer; return userName;
}
else
{
return null;
}
}
/// <summary>
/// Windows Vista or newer is required for SECPKG_ATTR_SESSION_KEY to work.
/// Windows XP / Server 2003 will return SEC_E_INVALID_TOKEN.
/// </summary>
public static byte[] GetSessionKey(SecHandle context)
{
SecPkgContext_SessionKey sessionKey;
uint result = QueryContextAttributes(ref context, SECPKG_ATTR_SESSION_KEY, out sessionKey);
if (result == SEC_E_OK)
{
int length = (int)sessionKey.SessionKeyLength;
byte[] sessionKeyBytes = new byte[length];
Marshal.Copy(sessionKey.SessionKey, sessionKeyBytes, 0, length);
return sessionKeyBytes;
}
else
{
return null;
}
}
public static IntPtr GetAccessToken(SecHandle serverContext)
{
IntPtr accessToken;
uint result = QueryContextAttributes(ref serverContext, SECPKG_ATTR_ACCESS_TOKEN, out accessToken);
if (result == SEC_E_OK)
{
return accessToken;
} }
else else
{ {