Create MS.SqlTools.Credentials project (#249)

* Initial credential service files * Clean-up hostloader * Update build scripts to build credentials archive * Move hosting files to new assembly * Add credentials files to MS.SqlTools.Credentials * Remove duplicate files * Update namespace in program.cs * Fix test build breaks * Update extensions visibility. * Remove unused resource strings * Add xproj files to SLN for appveyor builds * Fix appveyor build break in test project * Fix extensibility tests * Fix various typos in latest iteration * Add settings for Integration build * Fix codecoverage.bat to use full pdb for new projects * Fix bug when packing in folder with native images * Fix typos in xproj * Reset XLF to fix build.cmd
2026-01-24 09:35:39 -05:00 · 2017-02-23 16:09:58 -08:00
parent e79a37bdfe
commit 0af7bef66d
112 changed files with 4887 additions and 1870 deletions
--- a/src/Microsoft.SqlTools.Credentials/Credentials/Win32/NativeMethods.cs
+++ b/src/Microsoft.SqlTools.Credentials/Credentials/Win32/NativeMethods.cs
@@ -0,0 +1,110 @@
+//
+// Code originally from http://credentialmanagement.codeplex.com/, 
+// Licensed under the Apache License 2.0 
+//
+
+using System;
+using System.Runtime.InteropServices;
+using System.Text;
+using Microsoft.SqlTools.Credentials;
+
+namespace Microsoft.SqlTools.ServiceLayer.Credentials.Win32
+{
+    internal class NativeMethods
+    {
+        
+        [StructLayout(LayoutKind.Sequential)]
+        internal struct CREDENTIAL
+        {
+            public int Flags;
+            public int Type;
+            [MarshalAs(UnmanagedType.LPWStr)]
+            public string TargetName;
+            [MarshalAs(UnmanagedType.LPWStr)]
+            public string Comment;
+            public long LastWritten;
+            public int CredentialBlobSize;
+            public IntPtr CredentialBlob;
+            public int Persist;
+            public int AttributeCount;
+            public IntPtr Attributes;
+            [MarshalAs(UnmanagedType.LPWStr)]
+            public string TargetAlias;
+            [MarshalAs(UnmanagedType.LPWStr)]
+            public string UserName;
+        }
+        
+        [DllImport("Advapi32.dll", EntryPoint = "CredReadW", CharSet = CharSet.Unicode, SetLastError = true)]
+        internal static extern bool CredRead(string target, CredentialType type, int reservedFlag, out IntPtr CredentialPtr);
+
+        [DllImport("Advapi32.dll", EntryPoint = "CredWriteW", CharSet = CharSet.Unicode, SetLastError = true)]
+        internal static extern bool CredWrite([In] ref CREDENTIAL userCredential, [In] UInt32 flags);
+
+        [DllImport("Advapi32.dll", EntryPoint = "CredFree", SetLastError = true)]
+        internal static extern bool CredFree([In] IntPtr cred);
+
+        [DllImport("advapi32.dll", EntryPoint = "CredDeleteW", CharSet = CharSet.Unicode)]
+        internal static extern bool CredDelete(StringBuilder target, CredentialType type, int flags);
+
+        [DllImport("advapi32.dll", SetLastError = true, CharSet = CharSet.Unicode)]
+        internal static extern bool CredEnumerateW(string filter, int flag, out uint count, out IntPtr pCredentials);
+        
+        [DllImport("ole32.dll")]
+        internal static extern void CoTaskMemFree(IntPtr ptr);
+
+
+        internal abstract class CriticalHandleZeroOrMinusOneIsInvalid : CriticalHandle
+        {
+            protected CriticalHandleZeroOrMinusOneIsInvalid() : base(IntPtr.Zero)
+            {
+            }
+
+            public override bool IsInvalid
+            {
+                get { return handle == new IntPtr(0) || handle == new IntPtr(-1); }
+            }
+        }
+
+        internal sealed class CriticalCredentialHandle : CriticalHandleZeroOrMinusOneIsInvalid
+        {
+            // Set the handle.
+            internal CriticalCredentialHandle(IntPtr preexistingHandle)
+            {
+                SetHandle(preexistingHandle);
+            }
+
+            internal CREDENTIAL GetCredential()
+            {
+                if (!IsInvalid)
+                {
+                    // Get the Credential from the mem location
+                    return (CREDENTIAL)Marshal.PtrToStructure<CREDENTIAL>(handle);
+                }
+                else
+                {
+                    throw new InvalidOperationException(SR.CredentialsServiceInvalidCriticalHandle);
+                }
+            }
+
+            // Perform any specific actions to release the handle in the ReleaseHandle method.
+            // Often, you need to use Pinvoke to make a call into the Win32 API to release the 
+            // handle. In this case, however, we can use the Marshal class to release the unmanaged memory.
+
+            protected override bool ReleaseHandle()
+            {
+                // If the handle was set, free it. Return success.
+                if (!IsInvalid)
+                {
+                    // NOTE: We should also ZERO out the memory allocated to the handle, before free'ing it
+                    // so there are no traces of the sensitive data left in memory.
+                    CredFree(handle);
+                    // Mark the handle as invalid for future users.
+                    SetHandleAsInvalid();
+                    return true;
+                }
+                // Return false. 
+                return false;
+            }
+        }
+    }
+}