trustbridge: cinst/nssstore

comparison cinst/nssstore_win.c @ 675:4ad764bfb39c

Add writing of the NSS line into the registry

author	Andre Heinecke <andre.heinecke@intevation.de>
date	Fri, 27 Jun 2014 18:53:52 +0200
parents	f1795a232418
children	cb40af11ec3a

comparison

equal deleted inserted replaced

-:f1795a232418
+:4ad764bfb39c
 #include "strhelp.h"
 #ifndef APPNAME
 #define APPNAME L"cinst"
 #endif
+/**@def The name of the nss installation process */
+#define NSS_APP_NAME L"mozilla.exe"
 #ifndef SELECTION_FILE_NAME
 #define SELECTION_FILE_NAME L"currently_selected.txt"
 #endif
 {
 ERRORPRINTF ("Failed to enumeratre profile list. Error: %i", ret);
 goto done;
 }
+error = false;
 done:
 xfree (current_user);
 RegCloseKey (profile_list);
 if (error)
 {
 strv_free (retval);
 retval = NULL;
 }
+return retval;
+}
+/** @brief Build the command line for the NSS installation process
+*
+* Caller has to free the return value
+*
+* @param [in] selection_file the certificates to install
+*
+* @returns the command line to install the certificates. */
+static wchar_t*
+get_command_line(wchar_t *selection_file)
+{
+LPWSTR retval;
+char *install_dir = get_install_dir();
+wchar_t *w_inst_dir;
+size_t cmd_line_len = 0;
+if (install_dir == NULL)
+{
+ERRORPRINTF ("Failed to get installation directory");
+return NULL;
+}
+w_inst_dir = utf8_to_wchar (install_dir, strlen(install_dir));
+xfree (install_dir);
+if (w_inst_dir == NULL)
+{
+ERRORPRINTF ("Failed to convert installation directory");
+return NULL;
+}
+/* installdir + dirsep +  quotes + process name + space + quotes + selection_file
++ NULL */
+cmd_line_len = wcslen (w_inst_dir) + 1 + 2 + wcslen (NSS_APP_NAME) +
++ 1 + 2 + wcslen(selection_file) + 1;
+retval = xmalloc (cmd_line_len * sizeof(wchar_t));
+wcscpy_s (retval, cmd_line_len, L"\"");
+wcscat_s (retval, cmd_line_len, w_inst_dir);
+wcscat_s (retval, cmd_line_len, L"\\");
+wcscat_s (retval, cmd_line_len, NSS_APP_NAME);
+wcscat_s (retval, cmd_line_len, L"\" \"");
+wcscat_s (retval, cmd_line_len, selection_file);
+wcscat_s (retval, cmd_line_len, L"\"");
 return retval;
 }
 /**@brief Register NSS process as runOnce for other users
 */
 void
 register_proccesses_for_others (wchar_t *selection_file)
 {
 char **hives = locate_other_hives();
+int i = 0;
+wchar_t *run_command = NULL;
+if (hives == NULL)
+{
+DEBUGPRINTF ("No hives found.");
+return;
+}
+run_command = get_command_line (selection_file);
+for (i = 0; hives[i] != NULL; i++)
+{
+LONG ret = 0;
+wchar_t *hivepath = utf8_to_wchar (hives[i], strlen(hives[i]));
+HKEY key_handle = NULL;
+if (hivepath == NULL)
+{
+ERRORPRINTF ("Failed to read hive path");
+continue;
+}
+ret = RegLoadKeyW (HKEY_LOCAL_MACHINE, APPNAME L"_tmphive", hivepath);
+xfree (hivepath);
+hivepath = NULL;
+if (ret != ERROR_SUCCESS)
+{
+/* This is somewhat expected if the registry is not located
+in the standard location. Failure is accepted in that case. */
+PRINTLASTERROR ("Failed to load hive.");
+continue;
+}
+ret = RegOpenKeyExW (HKEY_LOCAL_MACHINE,
+APPNAME L"_tmphive\\Software\\Microsoft\\Windows\\CurrentVersion\\RunOnce",
+0,
+KEY_WRITE,
+&key_handle);
+if (ret != ERROR_SUCCESS)
+{
+ERRORPRINTF ("Failed to find RunOnce key in other registry.");
+RegUnLoadKey (HKEY_LOCAL_MACHINE, APPNAME L"_tmphive");
+continue;
+}
+ret = RegSetValueExW (key_handle, APPNAME, 0, REG_SZ, (LPBYTE) run_command,
+(wcslen(run_command) + 1) * sizeof(wchar_t));
+if (ret != ERROR_SUCCESS)
+{
+ERRORPRINTF ("Failed to write RunOnce key.");
+}
+RegCloseKey (key_handle);
+RegUnLoadKey (HKEY_LOCAL_MACHINE, APPNAME L"_tmphive");
+}
+xfree (run_command);
 strv_free (hives);
-printf("Selection file %S", selection_file);
 }
 /**@brief Start the process to install / remove
 *
 * Starts the NSS installation process for the current user
 LPWSTR lpApplicationName = L"mozilla.exe",
 lpCommandLine;
 PROCESS_INFORMATION piProcInfo = {0};
 STARTUPINFOW siStartInfo = {0};
 BOOL success = FALSE;
-size_t cmd_line_len = 0;
 if (!selection_file)
 {
 ERRORPRINTF ("Invalid call\n");
 return false;
 }
 /* TODO! if (is_elevated())
 restrict token -> hChildToken
 */
-cmd_line_len = wcslen (lpApplicationName) + wcslen(selection_file) + 2;
+lpCommandLine = get_command_line (selection_file);
-lpCommandLine = xmalloc (cmd_line_len * sizeof(wchar_t));
+if (lpCommandLine == NULL)
-wcscpy_s (lpCommandLine, cmd_line_len, lpApplicationName);
+{
-wcscpy_s (lpCommandLine, cmd_line_len, L" ");
+ERRORPRINTF ("Failed to build command line.");
-wcscat_s (lpCommandLine, cmd_line_len, selection_file);
+return false;
+}
 DEBUGPRINTF ("Starting %S with command line %S\n", lpApplicationName, lpCommandLine);
 success = CreateProcessAsUserW (hToken,
 lpApplicationName,

Mercurial > trustbridge

comparison cinst/nssstore_win.c @ 675:4ad764bfb39c