working on getting the run as service tool to work again.

[feisty_meow.git] / nucleus / library / security / win32_security.cpp

diff --git a/nucleus/library/security/win32_security.cpp b/nucleus/library/security/win32_security.cpp
new file mode 100644 (file)
index 0000000..e62a65c
--- /dev/null
+++ b/nucleus/library/security/win32_security.cpp
@@ -0,0 +1,141 @@
+
+
+
+/*****************************************************************************\
+*                                                                             *
+*  Name   : win32_security                                                    *
+*  Author : Sue Richeson                                                      *
+*  Author : Chris Koeritz                                                     *
+*                                                                             *
+*******************************************************************************
+* Copyright (c) 2000-$now By Author.  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 is online at:      *
+*     http://www.fsf.org/copyleft/gpl.html                                    *
+* Please send any updates to: fred@gruntose.com                               *
+\*****************************************************************************/
+
+#ifdef __WIN32__
+
+#include "win32_security.h"
+
+#include <basis/utf_conversion.h>
+#include <basis/astring.h>
+
+#include <comdef.h>
+#include <lm.h>
+
+// This piece of code is borrowed from the following July 1999 MSDN article
+//   HOWTO: Look Up Current User Name and Domain Name, ID: Q155698 
+//NOTE: It has been modified for inclusion here and it is Win32-specific.
+
+bool win32_security::GetUserAndDomainName(astring &UserName, astring &DomainName)
+{
+   HANDLE hToken;
+
+   // Initialize the return parameters.
+   UserName = "";
+   DomainName = "";
+
+   #define MY_BUFSIZE 512  // highly unlikely to exceed 512 bytes
+   UCHAR InfoBuffer[ MY_BUFSIZE + 1 ];
+   DWORD cbInfoBuffer = MY_BUFSIZE;
+   SID_NAME_USE snu;
+
+   BOOL bSuccess;
+
+   if(!OpenThreadToken(
+       GetCurrentThread(),
+       TOKEN_QUERY,
+       TRUE,
+       &hToken
+       )) {
+
+       if(GetLastError() == ERROR_NO_TOKEN) {
+
+           // 
+           // attempt to open the process token, since no thread token
+           // exists
+           // 
+
+           if(!OpenProcessToken(
+               GetCurrentProcess(),
+               TOKEN_QUERY,
+               &hToken
+               )) return FALSE;
+
+       } else {
+
+           // 
+           // error trying to get thread token
+           // 
+
+           return FALSE;
+       }
+   }
+
+   bSuccess = GetTokenInformation( hToken,
+                                   TokenUser,
+                                   InfoBuffer,
+                                   cbInfoBuffer,
+                                   &cbInfoBuffer);
+
+   if(!bSuccess) {
+       if(GetLastError() == ERROR_INSUFFICIENT_BUFFER) {
+
+           // 
+           // alloc buffer and try GetTokenInformation() again
+           // 
+
+           CloseHandle(hToken);
+           return FALSE;
+
+       } else {
+
+           // 
+           // error getting token info
+           // 
+
+           CloseHandle(hToken);
+           return FALSE;
+       }
+   }
+
+   CloseHandle(hToken);
+
+   TCHAR User[MY_BUFSIZE + 1];;
+   DWORD cchUserName = MY_BUFSIZE;
+   TCHAR Domain[MY_BUFSIZE +  1];
+   DWORD cchDomainName = MY_BUFSIZE;
+                      
+    bSuccess = LookupAccountSid(NULL,
+                           ((PTOKEN_USER)InfoBuffer)->User.Sid,
+                           User,
+                           &cchUserName,
+                           Domain,
+                           &cchDomainName,
+                           &snu);
+
+    if (bSuccess)
+    {
+        UserName = from_unicode_temp(User);
+        DomainName = from_unicode_temp(Domain);
+    }
+
+   return bSuccess;
+}
+
+astring win32_security::full_user()
+{
+  astring user, temp_domain;
+  GetUserAndDomainName(user, temp_domain);
+  user += astring("[") + temp_domain + "]";
+  return user;
+}
+
+#endif // windows.
+
+
+
+