Merge pull request #179 from i-am-not-an-ai/master

Impersonation support

Author: l0ss

SnaffCore/Concurrency/BlockingTaskScheduler.cs

@@ -1,6 +1,7 @@
 using System;
 using System.Collections.Generic;
 using System.Numerics;
+using System.Security.Principal;
 using System.Threading;
 using System.Threading.Tasks;
 
@@ -62,7 +63,13 @@ public void New(Action action)
                     // okay, let's add the thing
                     proceed = true;
 
-                    _taskFactory.StartNew(action, _cancellationSource.Token);
+                    WindowsIdentity impersonatedUser = WindowsIdentity.GetCurrent();
+                    _taskFactory.StartNew(() => {
+                        using (WindowsImpersonationContext ctx = impersonatedUser.Impersonate())
+                        {
+                            action();
+                        }
+                    }, _cancellationSource.Token);
                 }
             }
         }

Snaffler/SnaffleRunner.cs

@@ -1,4 +1,4 @@
-using NLog;
+using NLog;
 using NLog.Config;
 using NLog.Layouts;
 using NLog.Targets;
@@ -8,9 +8,10 @@
 using System;
 using System.Diagnostics;
 using System.IO;
-using System.Threading.Tasks;
+using System.Security.Principal;
 using System.Text.RegularExpressions;
 using System.Threading;
+using System.Threading.Tasks;
 
 namespace Snaffler
 {
@@ -201,7 +202,14 @@ public void Run(string[] args)
 
                 var tokenSource = new CancellationTokenSource();
                 var token = tokenSource.Token;
-                Task thing = Task.Factory.StartNew(() => { controller.Execute(); }, token);
+
+                WindowsIdentity impersonatedUser = WindowsIdentity.GetCurrent();
+                Task thing = Task.Factory.StartNew(() => {
+                    using (WindowsImpersonationContext ctx = impersonatedUser.Impersonate())
+                    {
+                        controller.Execute();
+                    }
+                }, token);
                 bool exit = false;
 
                 while (exit == false)