dll/cpl/appwiz/createlink.c

   1 /*
   2  * PROJECT:         ReactOS Software Control Panel
   3  * FILE:            dll/cpl/appwiz/createlink.c
   4  * PURPOSE:         ReactOS Software Control Panel
   5  * PROGRAMMER:      Gero Kuehn (reactos.filter@gkware.com)
   6  *                  Dmitry Chapyshev (lentind@yandex.ru)
   7  *                  Johannes Anderwald
   8  *                  Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
   9  * UPDATE HISTORY:
  10  *      06-17-2004  Created
  11  */
  12
  13 #include "appwiz.h"
  14 #include <strsafe.h>
  15
  16 BOOL
  17 IsShortcut(HKEY hKey)
  18 {
  19     WCHAR Value[10];
  20     DWORD Size;
  21     DWORD Type;
  22
  23     Size = sizeof(Value);
  24     if (RegQueryValueExW(hKey, L"IsShortcut", NULL, &Type, (LPBYTE)Value, &Size) != ERROR_SUCCESS)
  25         return FALSE;
  26
  27     if (Type != REG_SZ)
  28         return FALSE;
  29
  30     return (wcsicmp(Value, L"yes") == 0);
  31 }
  32
  33 BOOL
  34 IsExtensionAShortcut(LPWSTR lpExtension)
  35 {
  36     HKEY hKey;
  37     WCHAR Buffer[100];
  38     DWORD Size;
  39     DWORD Type;
  40
  41     if (RegOpenKeyExW(HKEY_CLASSES_ROOT, lpExtension, 0, KEY_READ, &hKey) != ERROR_SUCCESS)
  42         return FALSE;
  43
  44     if (IsShortcut(hKey))
  45     {
  46         RegCloseKey(hKey);
  47         return TRUE;
  48     }
  49
  50     Size = sizeof(Buffer);
  51     if (RegQueryValueEx(hKey, NULL, NULL, &Type, (LPBYTE)Buffer, &Size) != ERROR_SUCCESS || Type != REG_SZ)
  52     {
  53         RegCloseKey(hKey);
  54         return FALSE;
  55     }
  56
  57     RegCloseKey(hKey);
  58
  59     if (RegOpenKeyExW(HKEY_CLASSES_ROOT, Buffer, 0, KEY_READ, &hKey) != ERROR_SUCCESS)
  60         return FALSE;
  61
  62     if (IsShortcut(hKey))
  63     {
  64         RegCloseKey(hKey);
  65         return TRUE;
  66     }
  67
  68     RegCloseKey(hKey);
  69     return FALSE;
  70 }
  71
  72 BOOL
  73 CreateShortcut(PCREATE_LINK_CONTEXT pContext)
  74 {
  75     IShellLinkW *pShellLink, *pSourceShellLink;
  76     IPersistFile *pPersistFile;
  77     HRESULT hr;
  78     WCHAR Path[MAX_PATH];
  79     LPWSTR lpExtension;
  80
  81     /* get the extension */
  82     lpExtension = PathFindExtensionW(pContext->szTarget);
  83
  84     if (IsExtensionAShortcut(lpExtension))
  85     {
  86         hr = CoCreateInstance(&CLSID_ShellLink, NULL, CLSCTX_ALL, &IID_IShellLinkW, (void**)&pSourceShellLink);
  87
  88         if (FAILED(hr))
  89             return FALSE;
  90
  91         hr = IUnknown_QueryInterface(pSourceShellLink, &IID_IPersistFile, (void**)&pPersistFile);
  92         if (FAILED(hr))
  93         {
  94             IUnknown_Release(pSourceShellLink);
  95             return FALSE;
  96         }
  97
  98         hr = pPersistFile->lpVtbl->Load(pPersistFile, (LPCOLESTR)pContext->szTarget, STGM_READ);
  99         IUnknown_Release(pPersistFile);
 100
 101         if (FAILED(hr))
 102         {
 103             IUnknown_Release(pSourceShellLink);
 104             return FALSE;
 105         }
 106
 107         hr = IShellLinkW_GetPath(pSourceShellLink, Path, _countof(Path), NULL, 0);
 108         IUnknown_Release(pSourceShellLink);
 109
 110         if (FAILED(hr))
 111         {
 112             return FALSE;
 113         }
 114     }
 115     else
 116     {
 117         StringCchCopyW(Path, _countof(Path), pContext->szTarget);
 118     }
 119
 120     hr = CoCreateInstance(&CLSID_ShellLink, NULL, CLSCTX_ALL,
 121                           &IID_IShellLinkW, (void**)&pShellLink);
 122
 123     if (hr != S_OK)
 124         return FALSE;
 125
 126     pShellLink->lpVtbl->SetPath(pShellLink, Path);
 127     pShellLink->lpVtbl->SetDescription(pShellLink, pContext->szDescription);
 128     pShellLink->lpVtbl->SetWorkingDirectory(pShellLink, pContext->szWorkingDirectory);
 129
 130     hr = IUnknown_QueryInterface(pShellLink, &IID_IPersistFile, (void**)&pPersistFile);
 131     if (hr != S_OK)
 132     {
 133         IUnknown_Release(pShellLink);
 134         return FALSE;
 135     }
 136
 137     hr = pPersistFile->lpVtbl->Save(pPersistFile, pContext->szLinkName, TRUE);
 138     IUnknown_Release(pPersistFile);
 139     IUnknown_Release(pShellLink);
 140     return (hr == S_OK);
 141 }
 142
 143 BOOL
 144 CreateInternetShortcut(PCREATE_LINK_CONTEXT pContext)
 145 {
 146     IUniformResourceLocatorW *pURL = NULL;
 147     IPersistFile *pPersistFile = NULL;
 148     HRESULT hr;
 149     WCHAR szPath[MAX_PATH];
 150     GetFullPathNameW(pContext->szLinkName, _countof(szPath), szPath, NULL);
 151
 152     hr = CoCreateInstance(&CLSID_InternetShortcut, NULL, CLSCTX_ALL,
 153                           &IID_IUniformResourceLocatorW, (void **)&pURL);
 154     if (FAILED(hr))
 155         return FALSE;
 156
 157     hr = IUnknown_QueryInterface(pURL, &IID_IPersistFile, (void **)&pPersistFile);
 158     if (FAILED(hr))
 159     {
 160         IUnknown_Release(pURL);
 161         return FALSE;
 162     }
 163
 164     pURL->lpVtbl->SetURL(pURL, pContext->szTarget, 0);
 165
 166     hr = pPersistFile->lpVtbl->Save(pPersistFile, szPath, TRUE);
 167
 168     IUnknown_Release(pPersistFile);
 169     IUnknown_Release(pURL);
 170
 171     return SUCCEEDED(hr);
 172 }
 173
 174 BOOL IsInternetLocation(LPCWSTR pszLocation)
 175 {
 176     return (PathIsURLW(pszLocation) || wcsstr(pszLocation, L"www.") == pszLocation);
 177 }
 178
 179 INT_PTR
 180 CALLBACK
 181 WelcomeDlgProc(HWND hwndDlg,
 182                UINT uMsg,
 183                WPARAM wParam,
 184                LPARAM lParam)
 185 {
 186     LPPROPSHEETPAGEW ppsp;
 187     PCREATE_LINK_CONTEXT pContext;
 188     LPPSHNOTIFY lppsn;
 189     WCHAR szPath[MAX_PATH];
 190     WCHAR szDesc[100];
 191     BROWSEINFOW brws;
 192     LPITEMIDLIST pidllist;
 193
 194     switch(uMsg)
 195     {
 196         case WM_INITDIALOG:
 197             ppsp = (LPPROPSHEETPAGEW)lParam;
 198             pContext = (PCREATE_LINK_CONTEXT) ppsp->lParam;
 199             SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG_PTR)pContext);
 200             PropSheet_SetWizButtons(GetParent(hwndDlg), 0);
 201             break;
 202         case WM_COMMAND:
 203             switch(HIWORD(wParam))
 204             {
 205                 case EN_CHANGE:
 206                     if (SendDlgItemMessage(hwndDlg, IDC_SHORTCUT_LOCATION, WM_GETTEXTLENGTH, 0, 0))
 207                     {
 208                         PropSheet_SetWizButtons(GetParent(hwndDlg), PSWIZB_NEXT);
 209                     }
 210                     else
 211                     {
 212                         PropSheet_SetWizButtons(GetParent(hwndDlg), 0);
 213                     }
 214                     break;
 215             }
 216             switch(LOWORD(wParam))
 217             {
 218                 case IDC_SHORTCUT_BROWSE:
 219                     ZeroMemory(&brws, sizeof(brws));
 220                     brws.hwndOwner = hwndDlg;
 221                     brws.pidlRoot = NULL;
 222                     brws.pszDisplayName = szPath;
 223                     brws.ulFlags = BIF_BROWSEINCLUDEFILES;
 224                     brws.lpfn = NULL;
 225                     pidllist = SHBrowseForFolderW(&brws);
 226                     if (!pidllist)
 227                         break;
 228
 229                     if (SHGetPathFromIDListW(pidllist, szPath))
 230                         SetDlgItemTextW(hwndDlg, IDC_SHORTCUT_LOCATION, szPath);
 231
 232                     /* Free memory, if possible */
 233                     CoTaskMemFree(pidllist);
 234                     break;
 235             }
 236             break;
 237         case WM_NOTIFY:
 238             lppsn  = (LPPSHNOTIFY) lParam;
 239             if (lppsn->hdr.code == PSN_WIZNEXT)
 240             {
 241                 LPWSTR pch;
 242                 pContext = (PCREATE_LINK_CONTEXT) GetWindowLongPtr(hwndDlg, DWLP_USER);
 243                 GetDlgItemTextW(hwndDlg, IDC_SHORTCUT_LOCATION, pContext->szTarget, _countof(pContext->szTarget));
 244                 StrTrimW(pContext->szTarget, L" \t");
 245
 246                 if (IsInternetLocation(pContext->szTarget))
 247                 {
 248                     /* internet */
 249                     WCHAR szName[128];
 250                     LoadStringW(hApplet, IDS_NEW_INTERNET_SHORTCUT, szName, _countof(szName));
 251                     StringCchCopyW(pContext->szDescription, _countof(pContext->szDescription), szName);
 252
 253                     pContext->szWorkingDirectory[0] = 0;
 254                 }
 255                 else if (GetFileAttributesW(pContext->szTarget) != INVALID_FILE_ATTRIBUTES)
 256                 {
 257                     /* file */
 258                     SendDlgItemMessage(hwndDlg, IDC_SHORTCUT_LOCATION, EM_SETSEL, 0, -1);
 259                     SetFocus(GetDlgItem(hwndDlg, IDC_SHORTCUT_LOCATION));
 260                     SetWindowLongPtr(hwndDlg, DWLP_MSGRESULT, PSNRET_INVALID_NOCHANGEPAGE);
 261
 262                     /* set working directory */
 263                     StringCchCopyW(pContext->szWorkingDirectory, _countof(pContext->szWorkingDirectory),
 264                                    pContext->szTarget);
 265                     PathRemoveBackslashW(pContext->szWorkingDirectory);
 266                     pch = PathFindFileNameW(pContext->szWorkingDirectory);
 267                     if (pch && *pch)
 268                         *pch = 0;
 269                     PathRemoveBackslashW(pContext->szWorkingDirectory);
 270                 }
 271                 else
 272                 {
 273                     /* not found */
 274                     WCHAR szError[MAX_PATH + 100];
 275                     LoadStringW(hApplet, IDS_CREATE_SHORTCUT, szDesc, _countof(szDesc));
 276                     LoadStringW(hApplet, IDS_ERROR_NOT_FOUND, szPath, _countof(szPath));
 277                     StringCchPrintfW(szError, _countof(szError), szPath, pContext->szTarget);
 278                     MessageBoxW(hwndDlg, szError, szDesc, MB_ICONERROR);
 279                 }
 280             }
 281             break;
 282     }
 283     return FALSE;
 284 }
 285
 286 INT_PTR
 287 CALLBACK
 288 FinishDlgProc(HWND hwndDlg,
 289                UINT uMsg,
 290                WPARAM wParam,
 291                LPARAM lParam)
 292 {
 293     LPPROPSHEETPAGEW ppsp;
 294     PCREATE_LINK_CONTEXT pContext;
 295     LPPSHNOTIFY lppsn;
 296
 297     switch(uMsg)
 298     {
 299         case WM_INITDIALOG:
 300             ppsp = (LPPROPSHEETPAGEW)lParam;
 301             pContext = (PCREATE_LINK_CONTEXT) ppsp->lParam;
 302             SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG_PTR)pContext);
 303             SetDlgItemTextW(hwndDlg, IDC_SHORTCUT_NAME, pContext->szDescription);
 304             PropSheet_SetWizButtons(GetParent(hwndDlg), PSWIZB_BACK | PSWIZB_FINISH);
 305             break;
 306         case WM_COMMAND:
 307             switch(HIWORD(wParam))
 308             {
 309                 case EN_CHANGE:
 310                     if (SendDlgItemMessage(hwndDlg, IDC_SHORTCUT_NAME, WM_GETTEXTLENGTH, 0, 0))
 311                     {
 312                         PropSheet_SetWizButtons(GetParent(hwndDlg), PSWIZB_BACK | PSWIZB_FINISH);
 313                     }
 314                     else
 315                     {
 316                         PropSheet_SetWizButtons(GetParent(hwndDlg), PSWIZB_BACK);
 317                     }
 318                     break;
 319             }
 320             break;
 321         case WM_NOTIFY:
 322             lppsn  = (LPPSHNOTIFY) lParam;
 323             pContext = (PCREATE_LINK_CONTEXT) GetWindowLongPtr(hwndDlg, DWLP_USER);
 324             if (lppsn->hdr.code == PSN_WIZFINISH)
 325             {
 326                 LPWSTR pch;
 327                 DWORD attrs;
 328                 GetDlgItemTextW(hwndDlg, IDC_SHORTCUT_NAME, pContext->szDescription, MAX_PATH);
 329                 StrTrimW(pContext->szDescription, L" \t");
 330
 331                 /* if old shortcut file exists, then delete it now */
 332                 attrs = GetFileAttributesW(pContext->szOldFile);
 333                 if (attrs != INVALID_FILE_ATTRIBUTES && !(attrs & FILE_ATTRIBUTE_DIRECTORY))
 334                 {
 335                     DeleteFileW(pContext->szOldFile);
 336                 }
 337
 338                 if (IsInternetLocation(pContext->szTarget))
 339                 {
 340                     /* internet */
 341                     StringCchCopyW(pContext->szLinkName, _countof(pContext->szLinkName),
 342                                    pContext->szOrigin);
 343                     PathAppendW(pContext->szLinkName, pContext->szDescription);
 344
 345                     /* change extension if any */
 346                     pch = PathFindExtensionW(pContext->szLinkName);
 347                     if (pch && *pch)
 348                         *pch = 0;
 349                     StringCchCatW(pContext->szLinkName, _countof(pContext->szLinkName), L".url");
 350
 351                     if (!CreateInternetShortcut(pContext))
 352                     {
 353                         WCHAR szMessage[128];
 354                         LoadStringW(hApplet, IDS_CANTMAKEINETSHORTCUT, szMessage, _countof(szMessage));
 355                         MessageBoxW(hwndDlg, szMessage, NULL, MB_ICONERROR);
 356                     }
 357                 }
 358                 else
 359                 {
 360                     /* file */
 361                     StringCchCopyW(pContext->szLinkName, _countof(pContext->szLinkName),
 362                                    pContext->szOrigin);
 363                     PathAppendW(pContext->szLinkName, pContext->szDescription);
 364
 365                     /* change extension if any */
 366                     pch = PathFindExtensionW(pContext->szLinkName);
 367                     if (pch && *pch)
 368                         *pch = 0;
 369                     StringCchCatW(pContext->szLinkName, _countof(pContext->szLinkName), L".lnk");
 370
 371                     if (!CreateShortcut(pContext))
 372                     {
 373                         WCHAR szMessage[128];
 374                         LoadStringW(hApplet, IDS_CANTMAKESHORTCUT, szMessage, _countof(szMessage));
 375                         MessageBoxW(hwndDlg, szMessage, NULL, MB_ICONERROR);
 376                     }
 377                 }
 378             }
 379             break;
 380     }
 381     return FALSE;
 382 }
 383
 384 LONG CALLBACK
 385 ShowCreateShortcutWizard(HWND hwndCPl, LPWSTR szPath)
 386 {
 387     PROPSHEETHEADERW psh;
 388     HPROPSHEETPAGE ahpsp[2];
 389     PROPSHEETPAGE psp;
 390     UINT nPages = 0;
 391     UINT nLength;
 392     DWORD attrs;
 393
 394     PCREATE_LINK_CONTEXT pContext = (PCREATE_LINK_CONTEXT) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(CREATE_LINK_CONTEXT));
 395     if (!pContext)
 396     {
 397        /* no memory */
 398        return FALSE;
 399     }
 400
 401     nLength = wcslen(szPath);
 402     if (!nLength)
 403     {
 404         HeapFree(GetProcessHeap(), 0, pContext);
 405
 406         /* no directory given */
 407         return FALSE;
 408     }
 409
 410     attrs = GetFileAttributesW(szPath);
 411     if (attrs == INVALID_FILE_ATTRIBUTES)
 412     {
 413         HeapFree(GetProcessHeap(), 0, pContext);
 414
 415         /* invalid path */
 416         return FALSE;
 417     }
 418
 419     /* build the pContext->szOrigin and pContext->szOldFile */
 420     StringCchCopyW(pContext->szOrigin, _countof(pContext->szOrigin), szPath);
 421     pContext->szOldFile[0] = 0;
 422     if (!(attrs & FILE_ATTRIBUTE_DIRECTORY))
 423     {
 424         LPWSTR pch;
 425         StringCchCopyW(pContext->szOldFile, _countof(pContext->szOldFile), szPath);
 426         pch = PathFindFileNameW(pContext->szOrigin);
 427         if (pch && *pch)
 428             *pch = 0;
 429     }
 430     PathAddBackslashW(pContext->szOrigin);
 431
 432     /* Create the Welcome page */
 433     psp.dwSize = sizeof(PROPSHEETPAGE);
 434     psp.dwFlags = PSP_DEFAULT | PSP_HIDEHEADER;
 435     psp.hInstance = hApplet;
 436     psp.pfnDlgProc = WelcomeDlgProc;
 437     psp.pszTemplate = MAKEINTRESOURCEW(IDD_SHORTCUT_LOCATION);
 438     psp.lParam = (LPARAM)pContext;
 439     ahpsp[nPages++] = CreatePropertySheetPage(&psp);
 440
 441     /* Create the Finish page */
 442     psp.dwFlags = PSP_DEFAULT | PSP_HIDEHEADER;
 443     psp.pfnDlgProc = FinishDlgProc;
 444     psp.pszTemplate = MAKEINTRESOURCEW(IDD_SHORTCUT_FINISH);
 445     ahpsp[nPages++] = CreatePropertySheetPage(&psp);
 446
 447
 448     /* Create the property sheet */
 449     psh.dwSize = sizeof(PROPSHEETHEADER);
 450     psh.dwFlags = PSH_WIZARD97 | PSH_WATERMARK;
 451     psh.hInstance = hApplet;
 452     psh.hwndParent = NULL;
 453     psh.nPages = nPages;
 454     psh.nStartPage = 0;
 455     psh.phpage = ahpsp;
 456     psh.pszbmWatermark = MAKEINTRESOURCEW(IDB_SHORTCUT);
 457
 458     /* Display the wizard */
 459     PropertySheet(&psh);
 460     HeapFree(GetProcessHeap(), 0, pContext);
 461     return TRUE;
 462 }
 463
 464 LONG
 465 CALLBACK
 466 NewLinkHereW(HWND hwndCPl, UINT uMsg, LPARAM lParam1, LPARAM lParam2)
 467 {
 468     return ShowCreateShortcutWizard(hwndCPl, (LPWSTR) lParam1);
 469 }
 470
 471 LONG
 472 CALLBACK
 473 NewLinkHereA(HWND hwndCPl, UINT uMsg, LPARAM lParam1, LPARAM lParam2)
 474 {
 475     WCHAR szFile[MAX_PATH];
 476
 477     if (MultiByteToWideChar(CP_ACP, 0, (LPSTR) lParam1, -1, szFile, MAX_PATH))
 478     {
 479         return ShowCreateShortcutWizard(hwndCPl, szFile);
 480     }
 481     return -1;
 482 }