4 files changed, 121 insertions, 0 deletions

diff --git a/Makefile.am b/Makefile.am
index f907031..768e984 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -35,6 +35,8 @@ vdagent_SOURCES =			\
         vdagent/file_xfer.cpp		\
         vdagent/file_xfer.h		\
 	vdagent/vdagent.cpp		\
+	vdagent/as_user.cpp		\
+	vdagent/as_user.h		\
 	$(NULL)
 
 vdagent_rc.$(OBJEXT): vdagent/vdagent.rc
diff --git a/vdagent/as_user.cpp b/vdagent/as_user.cpp
new file mode 100644
index 0000000..d2d6c58
--- /dev/null
+++ b/vdagent/as_user.cpp
@@ -0,0 +1,73 @@
+/*
+   Copyright (C) 2013 Red Hat, Inc.
+
+   This program is free software; you can redistribute it and/or
+   modify it under the terms of the GNU General Public License as
+   published by the Free Software Foundation; either version 2 of
+   the License, or (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "vdcommon.h"
+#include "as_user.h"
+
+#include <wtsapi32.h>
+
+AsUser::AsUser(DWORD session_id):
+    _started(false),
+    _session_id(session_id),
+    _token(INVALID_HANDLE_VALUE)
+{
+}
+
+bool AsUser::begin()
+{
+    BOOL ret;
+
+    if (_session_id == (DWORD)-1) {
+        ret = ProcessIdToSessionId(GetCurrentProcessId(), &_session_id);
+        if (!ret) {
+            vd_printf("ProcessIdToSessionId failed %lu", GetLastError());
+            return false;
+        }
+    }
+    if (_token == INVALID_HANDLE_VALUE) {
+        ret = WTSQueryUserToken(_session_id, &_token);
+        if (!ret) {
+            vd_printf("WTSQueryUserToken failed -- %lu", GetLastError());
+        return false;
+        }
+    }
+
+    ret = ImpersonateLoggedOnUser(_token);
+    if (!ret) {
+        vd_printf("ImpersonateLoggedOnUser failed: %lu", GetLastError());
+        return false;
+    }
+
+    _started = true;
+    return true;
+}
+
+void AsUser::end()
+{
+    if (_started) {
+        RevertToSelf();
+        _started = false;
+    }
+}
+
+AsUser::~AsUser()
+{
+    end();
+    if (_token != INVALID_HANDLE_VALUE) {
+        CloseHandle(_token);
+    }
+}
diff --git a/vdagent/as_user.h b/vdagent/as_user.h
new file mode 100644
index 0000000..6243e98
--- /dev/null
+++ b/vdagent/as_user.h
@@ -0,0 +1,38 @@
+/*
+   Copyright (C) 2013 Red Hat, Inc.
+
+   This program is free software; you can redistribute it and/or
+   modify it under the terms of the GNU General Public License as
+   published by the Free Software Foundation; either version 2 of
+   the License, or (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef _H_AS_USER_
+#define _H_AS_USER_
+
+/** AsUser runs a task as the user logged on in session_id.
+ *  Constructor calls Impersonate, Destructor calls Revert, so
+ *  the caller needs not worry about that.
+ */
+class AsUser {
+public:
+    ~AsUser();
+    AsUser(DWORD session_id = (DWORD)-1);
+    bool begin();
+    void end();
+
+private:
+    DWORD _session_id;
+    HANDLE _token;
+    bool _started;
+};
+
+#endif
diff --git a/vdagent/vdagent.vcproj b/vdagent/vdagent.vcproj
index 6943e5e..f9f4228 100644
--- a/vdagent/vdagent.vcproj
+++ b/vdagent/vdagent.vcproj
@@ -342,6 +342,10 @@
 			UniqueIdentifier="{4FC737F1-C7A5-4376-A066-2A32D752A2FF}"

 			>

 			<File

+				RelativePath=".\as_user.cpp"

+				>

+			</File>

+			<File

 				RelativePath=".\desktop_layout.cpp"

 				>

 			</File>

@@ -372,6 +376,10 @@
 			UniqueIdentifier="{93995380-89BD-4b04-88EB-625FBE52EBFB}"

 			>

 			<File

+				RelativePath=".\as_user.h"

+				>

+			</File>

+			<File

 				RelativePath=".\desktop_layout.h"

 				>

 			</File>