The Tor Browser: toolkit/mozapps/update/tests/TestAUSHelper.cpp@129ffea94266 (annotated)

toolkit/mozapps/update/tests/TestAUSHelper.cpp@129ffea94266 (annotated)

toolkit/mozapps/update/tests/TestAUSHelper.cpp

Sat, 03 Jan 2015 20:18:00 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Sat, 03 Jan 2015 20:18:00 +0100
branch: TOR_BUG_3246
changeset 7: 129ffea94266
permissions: -rw-r--r--

Conditionally enable double key logic according to:
private browsing mode or privacy.thirdparty.isolate preference and
implement in GetCookieStringCommon and FindCookie where it counts...
With some reservations of how to convince FindCookie users to test
condition and pass a nullptr when disabling double key logic.

 /* Any copyright is dedicated to the Public Domain.
  * http://creativecommons.org/publicdomain/zero/1.0/
  */
 #ifdef XP_WIN
 # include <windows.h>
 # include <wintrust.h>
 # include <tlhelp32.h>
 # include <softpub.h>
 # include <direct.h>
 # include <io.h>
   typedef WCHAR NS_tchar;
 # define NS_main wmain
 # define F_OK 00
 # define W_OK 02
 # define R_OK 04
 # define stat _stat
 # define NS_T(str) L ## str
 # define NS_tsnprintf(dest, count, fmt, ...) \
   { \
     int _count = count - 1; \
     _snwprintf(dest, _count, fmt, ##__VA_ARGS__); \
     dest[_count] = L'\0'; \
   }
 # define NS_taccess _waccess
 # define NS_tchdir _wchdir
 # define NS_tfopen _wfopen
 # define NS_tstrcmp wcscmp
 # define NS_ttoi _wtoi
 # define NS_tstat _wstat
 # define NS_tgetcwd _wgetcwd
 # define LOG_S "%S"
 #include "../common/updatehelper.h"
 #else
 # include <unistd.h>
 # define NS_main main
   typedef char NS_tchar;
 # define NS_T(str) str
 # define NS_tsnprintf snprintf
 # define NS_taccess access
 # define NS_tchdir chdir
 # define NS_tfopen fopen
 # define NS_tstrcmp strcmp
 # define NS_ttoi atoi
 # define NS_tstat stat
 # define NS_tgetcwd getcwd
 # define NS_tfputs fputs
 # define LOG_S "%s"
 #endif
 #include "mozilla/NullPtr.h"
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #ifndef MAXPATHLEN
 # ifdef PATH_MAX
 #  define MAXPATHLEN PATH_MAX
 # elif defined(MAX_PATH)
 #  define MAXPATHLEN MAX_PATH
 # elif defined(_MAX_PATH)
 #  define MAXPATHLEN _MAX_PATH
 # elif defined(CCHMAXPATH)
 #  define MAXPATHLEN CCHMAXPATH
 # else
 #  define MAXPATHLEN 1024
 # endif
 #endif
 static void
 WriteMsg(const NS_tchar *path, const char *status)
 {
   FILE* outFP = NS_tfopen(path, NS_T("wb"));
   if (!outFP)
     return;
   fprintf(outFP, "%s\n", status);
   fclose(outFP);
   outFP = nullptr;
 }
 static bool
 CheckMsg(const NS_tchar *path, const char *expected)
 {
   if (NS_taccess(path, F_OK)) {
     return false;
   }
   FILE *inFP = NS_tfopen(path, NS_T("rb"));
   if (!inFP) {
     return false;
   }
   struct stat ms;
   if (fstat(fileno(inFP), &ms)) {
     return false;
   }
   char *mbuf = (char *) malloc(ms.st_size + 1);
   if (!mbuf) {
     return false;
   }
   size_t r = ms.st_size;
   char *rb = mbuf;
   size_t c = fread(rb, sizeof(char), 50, inFP);
   r -= c;
   rb += c;
   if (c == 0 && r) {
     return false;
   }
   mbuf[ms.st_size] = '\0';
   rb = mbuf;
   fclose(inFP);
   inFP = nullptr;
   return strcmp(rb, expected) == 0;
 }
 #ifdef XP_WIN
 /**
  * Verifies the trust of the specified file path.
  *
  * @param  filePath  The file path to check.
  * @return ERROR_SUCCESS if successful, or the last error code otherwise.
  */
 DWORD
 VerifyCertificateTrustForFile(LPCWSTR filePath)
 {
   // Setup the file to check.
   WINTRUST_FILE_INFO fileToCheck;
   ZeroMemory(&fileToCheck, sizeof(fileToCheck));
   fileToCheck.cbStruct = sizeof(WINTRUST_FILE_INFO);
   fileToCheck.pcwszFilePath = filePath;
   // Setup what to check, we want to check it is signed and trusted.
   WINTRUST_DATA trustData;
   ZeroMemory(&trustData, sizeof(trustData));
   trustData.cbStruct = sizeof(trustData);
   trustData.pPolicyCallbackData = nullptr;
   trustData.pSIPClientData = nullptr;
   trustData.dwUIChoice = WTD_UI_NONE;
   trustData.fdwRevocationChecks = WTD_REVOKE_NONE;
   trustData.dwUnionChoice = WTD_CHOICE_FILE;
   trustData.dwStateAction = 0;
   trustData.hWVTStateData = nullptr;
   trustData.pwszURLReference = nullptr;
   // no UI
   trustData.dwUIContext = 0;
   trustData.pFile = &fileToCheck;
   GUID policyGUID = WINTRUST_ACTION_GENERIC_VERIFY_V2;
   // Check if the file is signed by something that is trusted.
   return WinVerifyTrust(nullptr, &policyGUID, &trustData);
 }
 #endif
 int NS_main(int argc, NS_tchar **argv)
 {
   if (argc < 3) {
     fprintf(stderr, \
             "\n" \
             "Application Update Service Test Helper\n" \
             "\n" \
             "Usage: WORKINGDIR INFILE OUTFILE -s SECONDS [FILETOLOCK]\n"  \
             "   or: WORKINGDIR LOGFILE [ARG2 ARG3...]\n" \
             "   or: signature-check filepath\n" \
             "   or: setup-symlink dir1 dir2 file symlink\n" \
             "   or: remove-symlink dir1 dir2 file symlink\n" \
             "   or: check-symlink symlink\n" \
             "\n" \
             "  WORKINGDIR  \tThe relative path to the working directory to use.\n" \
             "  INFILE      \tThe relative path from the working directory for the file to\n" \
             "              \tread actions to perform such as finish.\n" \
             "  OUTFILE     \tThe relative path from the working directory for the file to\n" \
             "              \twrite status information.\n" \
             "  SECONDS     \tThe number of seconds to sleep.\n" \
             "  FILETOLOCK  \tThe relative path from the working directory to an existing\n" \
             "              \tfile to open exlusively.\n" \
             "              \tOnly available on Windows platforms and silently ignored on\n" \
             "              \tother platforms.\n" \
             "  LOGFILE     \tThe relative path from the working directory to log the\n" \
             "              \tcommand line arguments.\n" \
             "  ARG2 ARG3...\tArguments to write to the LOGFILE after the preceding command\n" \
             "              \tline arguments.\n" \
             "\n" \
             "Note: All paths must be relative.\n" \
             "\n");
     return 1;
   }
   if (!NS_tstrcmp(argv[1], NS_T("check-signature"))) {
 #ifdef XP_WIN
     if (ERROR_SUCCESS == VerifyCertificateTrustForFile(argv[2])) {
       return 0;
     } else {
       return 1;
     }
 #else
     // Not implemented on non-Windows platforms
     return 1;
 #endif
   }
   if (!NS_tstrcmp(argv[1], NS_T("setup-symlink"))) {
 #ifdef XP_UNIX
     NS_tchar path[MAXPATHLEN];
     NS_tsnprintf(path, sizeof(path)/sizeof(path[0]),
                  NS_T("%s/%s"), NS_T("/tmp"), argv[2]);
     mkdir(path, 0755);
     NS_tsnprintf(path, sizeof(path)/sizeof(path[0]),
                  NS_T("%s/%s/%s"), NS_T("/tmp"), argv[2], argv[3]);
     mkdir(path, 0755);
     NS_tsnprintf(path, sizeof(path)/sizeof(path[0]),
                  NS_T("%s/%s/%s/%s"), NS_T("/tmp"), argv[2], argv[3], argv[4]);
     FILE * file = NS_tfopen(path, NS_T("w"));
     if (file) {
       NS_tfputs(NS_T("test"), file);
       fclose(file);
     }
     symlink(path, argv[5]);
     NS_tsnprintf(path, sizeof(path)/sizeof(path[0]),
                  NS_T("%s/%s"), NS_T("/tmp"), argv[2]);
     if (argc > 6 && !NS_tstrcmp(argv[6], NS_T("change-perm"))) {
       chmod(path, 0644);
     }
     return 0;
 #else
     // Not implemented on non-Unix platforms
     return 1;
 #endif
   }
   if (!NS_tstrcmp(argv[1], NS_T("remove-symlink"))) {
 #ifdef XP_UNIX
     NS_tchar path[MAXPATHLEN];
     NS_tsnprintf(path, sizeof(path)/sizeof(path[0]),
                  NS_T("%s/%s"), NS_T("/tmp"), argv[2]);
     chmod(path, 0755);
     NS_tsnprintf(path, sizeof(path)/sizeof(path[0]),
                  NS_T("%s/%s/%s/%s"), NS_T("/tmp"), argv[2], argv[3], argv[4]);
     unlink(path);
     NS_tsnprintf(path, sizeof(path)/sizeof(path[0]),
                  NS_T("%s/%s/%s"), NS_T("/tmp"), argv[2], argv[3]);
     rmdir(path);
     NS_tsnprintf(path, sizeof(path)/sizeof(path[0]),
                  NS_T("%s/%s"), NS_T("/tmp"), argv[2]);
     rmdir(path);
     return 0;
 #else
     // Not implemented on non-Unix platforms
     return 1;
 #endif
   }
   if (!NS_tstrcmp(argv[1], NS_T("check-symlink"))) {
 #ifdef XP_UNIX
     struct stat ss;
     lstat(argv[2], &ss);
     return S_ISLNK(ss.st_mode) ? 0 : 1;
 #else
     // Not implemented on non-Unix platforms
     return 1;
 #endif
   }
   if (!NS_tstrcmp(argv[1], NS_T("wait-for-service-stop"))) {
 #ifdef XP_WIN
     const int maxWaitSeconds = NS_ttoi(argv[3]);
     LPCWSTR serviceName = argv[2];
     DWORD serviceState = WaitForServiceStop(serviceName, maxWaitSeconds);
     if (SERVICE_STOPPED == serviceState) {
       return 0;
     } else {
       return serviceState;
     }
 #else
     // Not implemented on non-Windows platforms
     return 1;
 #endif
   }
   if (!NS_tstrcmp(argv[1], NS_T("wait-for-application-exit"))) {
 #ifdef XP_WIN
     const int maxWaitSeconds = NS_ttoi(argv[3]);
     LPCWSTR application = argv[2];
     DWORD ret = WaitForProcessExit(application, maxWaitSeconds);
     if (ERROR_SUCCESS == ret) {
       return 0;
     } else if (WAIT_TIMEOUT == ret) {
       return 1;
     } else {
       return 2;
     }
 #else
     // Not implemented on non-Windows platforms
     return 1;
 #endif
   }
   int i = 0;
   if (NS_tchdir(argv[1]) != 0) {
     return 1;
   }
   // File in use test helper section
   if (!NS_tstrcmp(argv[4], NS_T("-s"))) {
     NS_tchar *cwd = NS_tgetcwd(nullptr, 0);
     NS_tchar inFilePath[MAXPATHLEN];
     NS_tsnprintf(inFilePath, sizeof(inFilePath)/sizeof(inFilePath[0]),
                  NS_T("%s/%s"), cwd, argv[2]);
     NS_tchar outFilePath[MAXPATHLEN];
     NS_tsnprintf(outFilePath, sizeof(outFilePath)/sizeof(outFilePath[0]),
                  NS_T("%s/%s"), cwd, argv[3]);
     int seconds = NS_ttoi(argv[5]);
 #ifdef XP_WIN
     HANDLE hFile = INVALID_HANDLE_VALUE;
     if (argc == 7) {
       hFile = CreateFileW(argv[6],
                           DELETE | GENERIC_WRITE, 0,
                           nullptr, OPEN_EXISTING, 0, nullptr);
       if (hFile == INVALID_HANDLE_VALUE) {
         WriteMsg(outFilePath, "error_locking");
         return 1;
       }
     }
     WriteMsg(outFilePath, "sleeping");
     while (!CheckMsg(inFilePath, "finish\n") && i++ <= seconds)  {
       Sleep(1000);
     }
     if (argc == 7) {
       CloseHandle(hFile);
     }
 #else
     WriteMsg(outFilePath, "sleeping");
     while (!CheckMsg(inFilePath, "finish\n") && i++ <= seconds)  {
       sleep(1);
     }
 #endif
     WriteMsg(outFilePath, "finished");
     return 0;
   }
   // Command line argument test helper section
   NS_tchar logFilePath[MAXPATHLEN];
   NS_tsnprintf(logFilePath, sizeof(logFilePath)/sizeof(logFilePath[0]),
                NS_T("%s"), argv[2]);
   FILE* logFP = NS_tfopen(logFilePath, NS_T("wb"));
   for (i = 1; i < argc; ++i) {
     fprintf(logFP, LOG_S "\n", argv[i]);
   }
   fclose(logFP);
   logFP = nullptr;
   return 0;
 }

The Tor Browser / annotate

toolkit/mozapps/update/tests/TestAUSHelper.cpp@129ffea94266 (annotated)

toolkit/mozapps/update/tests/TestAUSHelper.cpp