base/system/runonce/runonce.c

   1 /*
   2  * COPYRIGHT:       See COPYING in the top level directory
   3  * PROJECT:         ReactOS runonce.exe
   4  * FILE:            base/system/runonce/runonce.c
   5  * PROGRAMMER:      Dmitry Chapyshev (dmitry@reactos.org)
   6  */
   7
   8 #include "runonce.h"
   9
  10
  11 static
  12 DWORD
  13 WINAPI
  14 StartApplication(LPVOID lpDlg)
  15 {
  16     HWND hList = GetDlgItem((HWND)lpDlg, IDC_COMP_LIST);
  17     INT Index, Count = SendMessage(hList, LB_GETCOUNT, 0, 0);
  18     PROCESS_INFORMATION pi;
  19     STARTUPINFOW si;
  20     TCHAR szData[MAX_PATH];
  21
  22     for (Index = 0; Index < Count; Index++)
  23     {
  24         SendMessage(hList, LB_GETTEXT, Index, (LPARAM)szData);
  25
  26         ZeroMemory(&si, sizeof(si));
  27         si.cb = sizeof(si);
  28         si.wShowWindow = SW_SHOW;
  29
  30         if (!CreateProcess(NULL, szData, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi))
  31             continue;
  32
  33         WaitForSingleObjectEx(pi.hProcess, INFINITE, TRUE);
  34
  35         CloseHandle(pi.hProcess);
  36         CloseHandle(pi.hThread);
  37     }
  38
  39     RegDeleteKey(HKEY_LOCAL_MACHINE,
  40                  _T("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\RunOnce\\Setup"));
  41
  42     PostMessage((HWND)lpDlg, WM_CLOSE, 0, 0);
  43
  44     return 0;
  45 }
  46
  47 static
  48 VOID
  49 InitDialog(HWND hDlg)
  50 {
  51     TCHAR szAppPath[MAX_PATH], szData[MAX_PATH];
  52     DWORD dwIndex, dwSize, dwType, dwData, dwThreadId;
  53     HKEY hKey;
  54
  55     if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
  56                      _T("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\RunOnce\\Setup"),
  57                      0,
  58                      KEY_QUERY_VALUE,
  59                      &hKey) == ERROR_SUCCESS)
  60     {
  61         for (dwIndex = 0; ; dwIndex++)
  62         {
  63             dwSize = sizeof(szAppPath);
  64             dwData = sizeof(szData) / sizeof(TCHAR);
  65
  66             if (RegEnumValue(hKey,
  67                              dwIndex,
  68                              szAppPath,
  69                              &dwSize,
  70                              NULL,
  71                              &dwType,
  72                              (LPBYTE)szData,
  73                              &dwData) == ERROR_SUCCESS)
  74             {
  75                 if (dwType != REG_SZ) continue;
  76
  77                 SendMessage(GetDlgItem(hDlg, IDC_COMP_LIST), LB_ADDSTRING, 0, (LPARAM)szData);
  78             }
  79         }
  80
  81         RegCloseKey(hKey);
  82     }
  83
  84     CloseHandle(CreateThread(NULL,
  85                              0,
  86                              StartApplication,
  87                              (LPVOID)hDlg,
  88                              0,
  89                              &dwThreadId));
  90 }
  91
  92 static
  93 INT_PTR
  94 CALLBACK
  95 RunOnceDlgProc(HWND hDlg, UINT Msg, WPARAM wParam, LPARAM lParam)
  96 {
  97     switch (Msg)
  98     {
  99         case WM_INITDIALOG:
 100             InitDialog(hDlg);
 101             break;
 102
 103         case WM_CLOSE:
 104             EndDialog(hDlg, 0);
 105             break;
 106     }
 107
 108     return 0;
 109 }
 110
 111 INT
 112 WINAPI
 113 _tWinMain(HINSTANCE hInst, HINSTANCE hPrevInst, LPTSTR lpCmdLine, INT nCmdShow)
 114 {
 115     LPCTSTR lpCmd = GetCommandLine();
 116     TCHAR szAppPath[MAX_PATH], szData[MAX_PATH];
 117     DWORD dwIndex, dwSize, dwType, dwData;
 118     PROCESS_INFORMATION pi;
 119     STARTUPINFOW si;
 120     BOOL bRunApps = FALSE;
 121     HKEY hKey;
 122
 123     while (*lpCmd)
 124     {
 125         while (*lpCmd && *lpCmd != _T('/') && *lpCmd != _T('-')) lpCmd++;
 126         if (!*lpCmd) break;
 127         if (*++lpCmd == _T('r')) bRunApps = TRUE;
 128         lpCmd++;
 129     }
 130
 131     if (bRunApps)
 132     {
 133         if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
 134                          _T("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\RunOnce"),
 135                          0,
 136                          KEY_QUERY_VALUE,
 137                          &hKey) == ERROR_SUCCESS)
 138         {
 139             for (dwIndex = 0; ; dwIndex++)
 140             {
 141                 dwSize = sizeof(szAppPath);
 142                 dwData = sizeof(szData) / sizeof(TCHAR);
 143
 144                 if (RegEnumValue(hKey,
 145                                  dwIndex,
 146                                  szAppPath,
 147                                  &dwSize,
 148                                  NULL,
 149                                  &dwType,
 150                                  (LPBYTE)szData,
 151                                  &dwData) == ERROR_SUCCESS)
 152                 {
 153                     RegDeleteValue(hKey, szAppPath);
 154
 155                     if (dwType != REG_SZ) continue;
 156
 157                     ZeroMemory(&si, sizeof(si));
 158                     si.cb = sizeof(si);
 159                     si.wShowWindow = SW_SHOW;
 160
 161                     if (!CreateProcess(NULL, szData, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi))
 162                         continue;
 163
 164                     WaitForSingleObjectEx(pi.hProcess, INFINITE, TRUE);
 165
 166                     CloseHandle(pi.hProcess);
 167                     CloseHandle(pi.hThread);
 168                 }
 169             }
 170
 171             RegCloseKey(hKey);
 172         }
 173
 174         return 1;
 175     }
 176
 177     DialogBox(hInst, MAKEINTRESOURCE(IDD_RUNONCE_DLG), NULL, RunOnceDlgProc);
 178
 179     return 0;
 180 }