1
0
Fork 0
mirror of https://github.com/lidarr/Lidarr synced 2024-12-23 00:03:05 +00:00
Lidarr/CassiniDev/Core/NtlmAuth.cs
2010-10-02 12:01:43 -07:00

179 lines
No EOL
6.4 KiB
C#

// **********************************************************************************
// CassiniDev - http://cassinidev.codeplex.com
//
// Copyright (c) 2010 Sky Sanders. All rights reserved.
// Copyright (c) Microsoft Corporation. All rights reserved.
//
// This source code is subject to terms and conditions of the Microsoft Public
// License (Ms-PL). A copy of the license can be found in the license.txt file
// included in this distribution.
//
// You must not remove this notice, or any other, from this software.
//
// **********************************************************************************
#region
using System;
using System.Security;
using System.Security.Principal;
#endregion
namespace CassiniDev
{
[SuppressUnmanagedCodeSecurity]
internal sealed class NtlmAuth : IDisposable
{
private readonly bool _credentialsHandleAcquired;
private string _blob;
private bool _completed;
private Interop.SecHandle _credentialsHandle;
private Interop.SecBuffer _inputBuffer;
private Interop.SecBufferDesc _inputBufferDesc;
private Interop.SecBuffer _outputBuffer;
private Interop.SecBufferDesc _outputBufferDesc;
private Interop.SecHandle _securityContext;
private bool _securityContextAcquired;
private uint _securityContextAttributes;
private SecurityIdentifier _sid;
private long _timestamp;
public NtlmAuth()
{
if (
Interop.AcquireCredentialsHandle(null, "NTLM", 1, IntPtr.Zero, IntPtr.Zero, IntPtr.Zero, IntPtr.Zero,
ref _credentialsHandle, ref _timestamp) != 0)
{
throw new InvalidOperationException();
}
_credentialsHandleAcquired = true;
}
public string Blob
{
get { return _blob; }
}
public bool Completed
{
get { return _completed; }
}
public SecurityIdentifier SID
{
get { return _sid; }
}
#region IDisposable Members
void IDisposable.Dispose()
{
FreeUnmanagedResources();
GC.SuppressFinalize(this);
}
#endregion
public unsafe bool Authenticate(string blobString)
{
_blob = null;
byte[] buffer = Convert.FromBase64String(blobString);
byte[] inArray = new byte[0x4000];
fixed (void* ptrRef = &_securityContext)
{
fixed (void* ptrRef2 = &_inputBuffer)
{
fixed (void* ptrRef3 = &_outputBuffer)
{
fixed (void* ptrRef4 = buffer)
{
fixed (void* ptrRef5 = inArray)
{
IntPtr zero = IntPtr.Zero;
if (_securityContextAcquired)
{
zero = (IntPtr) ptrRef;
}
_inputBufferDesc.ulVersion = 0;
_inputBufferDesc.cBuffers = 1;
_inputBufferDesc.pBuffers = (IntPtr) ptrRef2;
_inputBuffer.cbBuffer = (uint) buffer.Length;
_inputBuffer.BufferType = 2;
_inputBuffer.pvBuffer = (IntPtr) ptrRef4;
_outputBufferDesc.ulVersion = 0;
_outputBufferDesc.cBuffers = 1;
_outputBufferDesc.pBuffers = (IntPtr) ptrRef3;
_outputBuffer.cbBuffer = (uint) inArray.Length;
_outputBuffer.BufferType = 2;
_outputBuffer.pvBuffer = (IntPtr) ptrRef5;
int num = Interop.AcceptSecurityContext(ref _credentialsHandle, zero,
ref _inputBufferDesc, 20,
0, ref _securityContext, ref _outputBufferDesc,
ref _securityContextAttributes, ref _timestamp);
if (num == 0x90312)
{
_securityContextAcquired = true;
_blob = Convert.ToBase64String(inArray, 0, (int) _outputBuffer.cbBuffer);
}
else
{
if (num != 0)
{
return false;
}
IntPtr phToken = IntPtr.Zero;
if (Interop.QuerySecurityContextToken(ref _securityContext, ref phToken) != 0)
{
return false;
}
try
{
using (WindowsIdentity identity = new WindowsIdentity(phToken))
{
_sid = identity.User;
}
}
finally
{
Interop.CloseHandle(phToken);
}
_completed = true;
}
}
}
}
}
}
return true;
}
~NtlmAuth()
{
FreeUnmanagedResources();
}
private void FreeUnmanagedResources()
{
if (_securityContextAcquired)
{
Interop.DeleteSecurityContext(ref _securityContext);
}
if (_credentialsHandleAcquired)
{
Interop.FreeCredentialsHandle(ref _credentialsHandle);
}
}
}
}