Initial commit

2023-01-28 21:15:17 +11:00
parent eb7b6fa4eb
commit 52e75c1748
90 changed files with 7306 additions and 0 deletions
@@ -0,0 +1,214 @@
+using System;
+using System.Linq;
+using System.Runtime.InteropServices;
+using Lithnet.CredentialProvider.Interop;
+using Microsoft.Extensions.Logging;
+
+namespace Lithnet.CredentialProvider
+{
+    public abstract partial class CredentialProviderBase : ICredentialProvider
+    {
+        int ICredentialProvider.SetUsageScenario(UsageScenario cpus, CredUIWinFlags dwFlags)
+        {
+            try
+            {
+                this.logger.LogTrace($"SetUsageScenario: Usage: {cpus} flags: {dwFlags}");
+
+                this.UsageScenario = cpus;
+
+                if (this.IsUsageScenarioSupported(cpus, (CredUIWinFlags)dwFlags))
+                {
+                    return HRESULT.S_OK;
+                }
+                else
+                {
+                    return HRESULT.E_NOTIMPL;
+                }
+            }
+            catch (Exception ex)
+            {
+                this.logger.LogError(ex, "SetUsageScenario failed");
+                return HRESULT.E_FAIL;
+            }
+        }
+
+        int ICredentialProvider.SetSerialization(ref CredentialSerialization pcpcs)
+        {
+            try
+            {
+                this.InboundSerialization = pcpcs;
+                this.OnSetSerialization(pcpcs);
+                return HRESULT.S_OK;
+            }
+            catch (Exception ex)
+            {
+                this.logger.LogError(ex, "UnAdvSetSerializationise failed");
+                return HRESULT.E_FAIL;
+            }
+        }
+
+        int ICredentialProvider.Advise(ICredentialProviderEvents pcpe, IntPtr upAdviseContext)
+        {
+            try
+            {
+                this.logger.LogTrace($"Advise: {upAdviseContext}");
+                if (pcpe != null)
+                {
+                    this.credentialProviderEventsAdviseContext = upAdviseContext;
+                    this.CredentialProviderEvents = pcpe;
+                    var intPtr = Marshal.GetIUnknownForObject(pcpe);
+                    Marshal.AddRef(intPtr);
+                }
+
+                return HRESULT.S_OK;
+            }
+            catch (Exception ex)
+            {
+                this.logger.LogError(ex, "Advise failed");
+                return HRESULT.E_FAIL;
+            }
+        }
+
+        int ICredentialProvider.UnAdvise()
+        {
+            try
+            {
+                this.logger.LogTrace($"Unadvise");
+
+                if (this.CredentialProviderEvents != null)
+                {
+                    var intPtr = Marshal.GetIUnknownForObject(this.CredentialProviderEvents);
+                    Marshal.Release(intPtr);
+                    this.CredentialProviderEvents = null;
+                    this.credentialProviderEventsAdviseContext = IntPtr.Zero;
+                }
+
+                return HRESULT.S_OK;
+            }
+            catch (Exception ex)
+            {
+                this.logger.LogError(ex, "UnAdvise failed");
+                return HRESULT.E_FAIL;
+            }
+        }
+
+        int ICredentialProvider.GetFieldDescriptorCount(out uint pdwCount)
+        {
+            pdwCount = 0;
+
+            try
+            {
+                this.logger.LogTrace("GetFieldDescriptorCount");
+                this.BuildControls();
+                this.Controls.Lock();
+
+                pdwCount = (uint)this.Controls.Count;
+                this.logger.LogTrace($"GetFieldDescriptorCount return {pdwCount}");
+
+                return HRESULT.S_OK;
+            }
+            catch (Exception ex)
+            {
+                this.logger.LogError(ex, "GetFieldDescriptorCount failed");
+                return HRESULT.E_FAIL;
+            }
+        }
+
+        int ICredentialProvider.GetFieldDescriptorAt(uint dwIndex, out IntPtr ppcpfd)
+        {
+            ppcpfd = IntPtr.Zero;
+
+            try
+            {
+                this.logger.LogTrace($"GetFieldDescriptorAt {dwIndex}");
+
+                if (dwIndex >= this.Controls.Count)
+                {
+                    return HRESULT.E_INVALIDARG;
+                }
+
+                var item = this.Controls[(int)dwIndex].GetDescriptor();
+                var size = Marshal.SizeOf<FieldDescriptor>();
+                var ptr = Marshal.AllocCoTaskMem(size);
+                Marshal.StructureToPtr<FieldDescriptor>(item, ptr, false);
+                ppcpfd = ptr;
+
+                this.logger.LogTrace($"GetFieldDescriptorAt {dwIndex}: returning {item}");
+
+                return HRESULT.S_OK;
+            }
+            catch (Exception ex)
+            {
+                this.logger.LogError(ex, "GetFieldDescriptorAt failed");
+                return HRESULT.E_FAIL;
+            }
+        }
+
+        int ICredentialProvider.GetCredentialCount(out uint pdwCount, out uint pdwDefault, out int pbAutoLogonWithDefault)
+        {
+            pdwCount = 0;
+            pdwDefault = 0;
+            pbAutoLogonWithDefault = 0;
+
+            try
+            {
+                this.logger.LogTrace($"GetCredentialCount");
+
+                if (this.Tiles == null)
+                {
+                    this.logger.LogTrace($"GetCredentialCount is enumerating tiles");
+                    this.SetupTiles();
+                    this.logger.LogTrace($"{this.Tiles.Count} tiles enumerated. {this.Tiles.Count(t => t.IsGenericTile)} generic and {this.tiles.Count(t => !t.IsGenericTile)} personalized");
+                }
+
+                this.notifyOnTileCollectionChange = true;
+
+                var autoLogonTile = this.Tiles.FirstOrDefault(t => t.IsAutoLogon);
+                var defaultTile = this.Tiles.FirstOrDefault(t => t.IsDefault);
+
+                int defaultIndex = 0;
+                if (autoLogonTile != null)
+                {
+                    defaultIndex = this.tiles.IndexOf(autoLogonTile);
+                }
+                else if (defaultTile != null)
+                {
+                    defaultIndex = this.tiles.IndexOf(defaultTile);
+                }
+
+                pdwCount = (uint)this.Tiles.Count;
+                pdwDefault = (uint)defaultIndex;
+                pbAutoLogonWithDefault = autoLogonTile == null ? 0 : 1;
+
+                this.logger.LogTrace($"GetCredentialCount returning pdwCount: {pdwCount}, pdwDefault: {pdwDefault}, pbAutoLogonWithDefault: {pbAutoLogonWithDefault}");
+
+                return HRESULT.S_OK;
+            }
+            catch (Exception ex)
+            {
+                this.logger.LogError(ex, "GetCredentialCount failed");
+                return HRESULT.E_FAIL;
+            }
+        }
+
+        int ICredentialProvider.GetCredentialAt(uint dwIndex, out ICredentialProviderCredential ppcpc)
+        {
+            ppcpc = null;
+
+            try
+            {
+                this.logger.LogTrace($"GetCredentialAt {dwIndex}");
+
+                var tile = this.Tiles[(int)dwIndex];
+
+                ppcpc = (ICredentialProviderCredential)tile;
+                return HRESULT.S_OK;
+            }
+            catch (Exception ex)
+            {
+                this.logger.LogError(ex, "GetCredentialAt failed");
+                return HRESULT.E_FAIL;
+            }
+        }
+    }
+}