Implemented custom command line arguments in unit tests.

end2endzone · end2endzone · commit 7412ab8f231e · 2020-11-07T09:22:08.000-05:00
diff --git a/test/ArgumentsHandler.cpp b/test/ArgumentsHandler.cpp
@@ -0,0 +1,131 @@
+/**********************************************************************************
+ * MIT License
+ * 
+ * Copyright (c) 2018 Antoine Beauchamp
+ * 
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ * 
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ * 
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ *********************************************************************************/
+
+#include "ArgumentsHandler.h"
+#include "rapidassist/process.h"
+#include "rapidassist/environment.h"
+#include "rapidassist/filesystem.h"
+
+#ifndef WIN32_LEAN_AND_MEAN
+#define WIN32_LEAN_AND_MEAN 1
+#endif
+#include <Windows.h>
+#undef GetEnvironmentVariable
+#undef DeleteFile
+#undef CreateDirectory
+#undef CopyFile
+#undef CreateFile
+#undef GetCurrentDirectory
+
+
+namespace shellanything
+{
+
+  //https://stackoverflow.com/questions/8046097/how-to-check-if-a-process-has-the-administrative-rights
+  bool IsProcessElevated()
+  {
+    BOOL fRet = FALSE;
+    HANDLE hToken = NULL;
+    if( OpenProcessToken( GetCurrentProcess( ),TOKEN_QUERY,&hToken ) )
+    {
+      TOKEN_ELEVATION Elevation;
+      DWORD cbSize = sizeof( TOKEN_ELEVATION );
+      if( GetTokenInformation( hToken, TokenElevation, &Elevation, sizeof( Elevation ), &cbSize ) )
+      {
+        fRet = Elevation.TokenIsElevated;
+      }
+    }
+    if( hToken )
+    {
+      CloseHandle( hToken );
+    }
+    return (fRet == TRUE);
+  }
+
+  //http://blog.aaronballman.com/2011/08/how-to-check-access-rights/
+  bool CanAccessFolder( LPCTSTR folderName, DWORD genericAccessRights )
+  {
+    bool bRet = false;
+    DWORD length = 0;
+    if (!::GetFileSecurity( folderName, OWNER_SECURITY_INFORMATION | GROUP_SECURITY_INFORMATION | DACL_SECURITY_INFORMATION, NULL, NULL, &length ) && ERROR_INSUFFICIENT_BUFFER == ::GetLastError())
+    {
+      PSECURITY_DESCRIPTOR security = static_cast< PSECURITY_DESCRIPTOR >( ::malloc( length ) );
+      if (security && ::GetFileSecurity( folderName, OWNER_SECURITY_INFORMATION | GROUP_SECURITY_INFORMATION | DACL_SECURITY_INFORMATION, security, length, &length ))
+      {
+        HANDLE hToken = NULL;
+        if (::OpenProcessToken( ::GetCurrentProcess(), TOKEN_IMPERSONATE | TOKEN_QUERY | TOKEN_DUPLICATE | STANDARD_RIGHTS_READ, &hToken ))
+        {
+          HANDLE hImpersonatedToken = NULL;
+          if (::DuplicateToken( hToken, SecurityImpersonation, &hImpersonatedToken ))
+          {
+            GENERIC_MAPPING mapping = { 0xFFFFFFFF };
+            PRIVILEGE_SET privileges = { 0 };
+            DWORD grantedAccess = 0, privilegesLength = sizeof( privileges );
+            BOOL result = FALSE;
+
+            mapping.GenericRead = FILE_GENERIC_READ;
+            mapping.GenericWrite = FILE_GENERIC_WRITE;
+            mapping.GenericExecute = FILE_GENERIC_EXECUTE;
+            mapping.GenericAll = FILE_ALL_ACCESS;
+
+            ::MapGenericMask( &genericAccessRights, &mapping );
+            if (::AccessCheck( security, hImpersonatedToken, genericAccessRights, &mapping, &privileges, &privilegesLength, &grantedAccess, &result ))
+            {
+              bRet = (result == TRUE);
+            }
+            ::CloseHandle( hImpersonatedToken );
+          }
+          ::CloseHandle( hToken );
+        }
+        ::free( security );
+      }
+    }
+
+    return bRet;
+  }
+
+  int PrintStatistics()
+  {
+    std::string process_path = ra::process::GetCurrentProcessPath();
+    ra::process::processid_t process_id = ra::process::GetCurrentProcessId();
+    std::string process_current_directory = ra::filesystem::GetCurrentDirectory();
+    int process_architecture = (ra::environment::IsProcess64Bit() ? 64 : 86);
+    int process_elevated = (IsProcessElevated() ? 1 : 0);
+    const char * build_configuration = (ra::environment::IsConfigurationDebug() ? "debug" : "release");
+    std::string env_temp = ra::environment::GetEnvironmentVariable("TEMP");
+    bool has_windows_write_access = CanAccessFolder("C:\\Windows", GENERIC_WRITE);
+
+    printf("process.path=%s\n", process_path.c_str());
+    printf("process.pid=%d\n", process_id);
+    printf("process.current_dir=%s\n", process_current_directory.c_str());
+    printf("process.architecture=x%d\n", process_architecture);
+    printf("process.elevated=%d\n", process_elevated);
+    printf("build.configuration=%s\n", build_configuration);
+    printf("env.temp=%s\n", env_temp.c_str());
+    printf("has_windows_write_access=%d\n", (int)has_windows_write_access);
+
+    return 0;
+  }
+
+} //namespace shellanything
diff --git a/test/ArgumentsHandler.h b/test/ArgumentsHandler.h
@@ -0,0 +1,39 @@
+/**********************************************************************************
+ * MIT License
+ * 
+ * Copyright (c) 2018 Antoine Beauchamp
+ * 
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ * 
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ * 
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ *********************************************************************************/
+
+#ifndef SA_ARGUMENT_HANDLER_H
+#define SA_ARGUMENT_HANDLER_H
+
+namespace shellanything
+{
+
+  /// <summary>
+  /// Print statistics about this program settings.
+  /// </summary>
+  /// <returns>Returns 0 when the function succeed. Returns a non zero values otherwise.</returns>
+  int PrintStatistics();
+
+} //namespace shellanything
+
+#endif //SA_ARGUMENT_HANDLER_H
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
@@ -48,6 +48,8 @@ add_executable(shellanything_unittest
   ${CONFIGURATION_TEST_FILES}
   ${SHELLANYTHING_PRIVATE_FILES}
   main.cpp
+  ArgumentsHandler.cpp
+  ArgumentsHandler.h
   TestActionExecute.cpp
   TestActionExecute.h
   TestActionFile.cpp
diff --git a/test/main.cpp b/test/main.cpp
@@ -35,11 +35,19 @@
 
 #include "rapidassist/testing.h"
 #include "rapidassist/environment.h"
+#include "rapidassist/cli.h"
+
+#include "ArgumentsHandler.h"
 
 using namespace ra;
 
 int main(int argc, char **argv)
 {
+  // Look for custom command line arguments
+  bool has_printstatistics = ra::cli::ParseArgument("printstatistics", std::string(), argc, argv);
+  if (has_printstatistics)
+    return shellanything::PrintStatistics();
+
   // Prepare Google's logging library.
   fLB::FLAGS_logtostderr = false; //on error, print to stdout instead of stderr
   fLB::FLAGS_log_prefix = 1; //prefix each message in file/console with 'E0405 19:13:07.577863  6652 shellext.cpp:847]'
