BOOL TMakeVirtualStorePathW(WCHAR *org_path, WCHAR *buf)
{
if (!IsWinVista()) return FALSE;
if (!TIsVirtualizedDirW(org_path)
|| !SHGetSpecialFolderPathW(NULL, buf, CSIDL_LOCAL_APPDATA, FALSE)
|| org_path[1] != ':' || org_path[2] != '\\') {
wcscpy(buf, org_path);
return FALSE;
}
swprintf(buf + wcslen(buf), L"\\VirtualStore%s", org_path + 2);
return TRUE;
}
BOOL TInstDlg::Install(void)
{
char buf[MAX_PATH_U8], setupDir[MAX_PATH_U8], setupPath[MAX_PATH_U8];
char installPath[MAX_PATH_U8];
BOOL extract_only = IsDlgButtonChecked(EXTRACT_CHECK);
// 現在、起動中の ipmsg を終了
int st = extract_only ? 0 : TerminateIPMsg();
if (st == 1) return FALSE;
// インストールパス設定
GetDlgItemTextU8(FILE_EDIT, setupDir, sizeof(setupDir));
if (IsWinVista() && !::IsUserAnAdmin() && TIsEnableUAC()
&& TIsVirtualizedDirW(U8toWs(setupDir))) {
if (MessageBox(GetLoadStr(IDS_REQUIREADMIN), INSTALL_STR,
MB_OKCANCEL|MB_ICONINFORMATION) != IDOK) return FALSE;
return RunAsAdmin(hWnd, TRUE);
}
CreateDirectoryU8(setupDir, NULL);
DWORD attr = GetFileAttributesU8(setupDir);
if (attr == 0xffffffff || (attr & FILE_ATTRIBUTE_DIRECTORY) == 0)
return MessageBox(GetLoadStr(IDS_NOTCREATEDIR), INSTALL_STR), FALSE;
MakePath(setupPath, setupDir, IPMSG_EXENAME);
if (st == 2) {
MessageBox(GetLoadStr(IDS_CANTTERMINATE), INSTALL_STR);
return FALSE;
}
if (!runasImm &&
MessageBox(GetLoadStr(IDS_START), INSTALL_STR, MB_OKCANCEL|MB_ICONINFORMATION) != IDOK) {
return FALSE;
}
runasImm = FALSE;
// ファイル生成
for (int i=0; SetupFiles[i]; i++) {
MakePath(installPath, setupDir, SetupFiles[i]);
CreateStatus cs = CreateFileBySelf(installPath, SetupFiles[i]);
if (cs == CS_BROKEN) {
MessageBox(GetLoadStr(IDS_BROKENARCHIVE), INSTALL_STR);
return FALSE;
}
else if (cs == CS_ACCESS) {
const char *msg = Fmt("%s\r\n%s", GetLoadStrU8(IDS_NOTCREATEFILE), installPath);
MessageBoxU8(msg, INSTALL_STR);
return FALSE;
}
}
// 展開のみ
if (extract_only) {
ShellExecuteW(NULL, NULL, U8toWs(setupDir), 0, 0, SW_SHOW);
return TRUE;
}
// スタートアップ＆デスクトップに登録
TRegistry reg(HKEY_CURRENT_USER);
if (reg.OpenKey(REGSTR_SHELLFOLDERS)) {
char *regStr[] = { REGSTR_STARTUP, REGSTR_PROGRAMS, REGSTR_DESKTOP, NULL };
BOOL resId[] = { STARTUP_CHECK, PROGRAM_CHECK, DESKTOP_CHECK, NULL };
for (int i=0; regStr[i]; i++) {
if (reg.GetStr(regStr[i], buf, sizeof(buf))) {
if (i != 0 || !RemoveSameLink(buf, buf))
::wsprintf(buf + strlen(buf), "\\%s", IPMSG_SHORTCUT_NAME);
if (IsDlgButtonChecked(resId[i]))
SymLink(setupPath, buf);
else
DeleteLink(buf);
}
}
reg.CloseKey();
}
// レジストリにアプリケーション情報を登録
reg.ChangeTopKey(HKEY_LOCAL_MACHINE);
if (reg.OpenKey(REGSTR_PATH_APPPATHS)) {
if (reg.CreateKey(IPMSG_EXENAME)) {
reg.SetStr(NULL, setupPath);
reg.SetStr(REGSTR_PATH, setupDir);
reg.CloseKey();
}
reg.CloseKey();
}
// レジストリにアンインストール情報を登録
if (reg.OpenKey(REGSTR_PATH_UNINSTALL)) {
if (reg.CreateKey(IPMSG_NAME)) {
MakePath(buf, setupDir, SETUP_EXENAME);
strcat(buf, " /r");
reg.SetStr(REGSTR_VAL_UNINSTALLER_DISPLAYNAME, IPMSG_FULLNAME);
reg.SetStr(REGSTR_VAL_UNINSTALLER_COMMANDLINE, buf);
reg.CloseKey();
}
reg.CloseKey();
}
// コピーしたアプリケーションを起動
const char *msg = GetLoadStr(IDS_SETUPCOMPLETE);
int flg = MB_OKCANCEL|MB_ICONINFORMATION;
// if (IsWinVista() && ::IsUserAnAdmin() && TIsEnableUAC()) {
// msg = Fmt("%s%s", msg, GetLoadStr(IDS_COMPLETE_UACADD));
// flg |= MB_DEFBUTTON2;
// }
TLaunchDlg dlg(msg, this);
if (dlg.Exec() == IDOK) {
if (runasWnd) {
Wstr wbuf(setupDir);
if (::SendDlgItemMessageW(runasWnd, FILE_EDIT, WM_SETTEXT, 0, (LPARAM)wbuf.Buf())) {
::PostMessage(runasWnd, WM_IPMSG_QUIT, 1, 0);
runasWnd = NULL;
}
}
else {
AppKick();
}
}
else {
HWND hHelp = ShowHelpU8(0, setupDir, GetLoadStrU8(IDS_IPMSGHELP), "#history");
if (hHelp) {
Show(SW_HIDE);
while (::IsWindow(hHelp)) {
this->Sleep(100);
}
}
}
if (runasWnd) {
::PostMessage(runasWnd, WM_IPMSG_QUIT, 0, 0);
runasWnd = NULL;
}
// ShellExecuteU8(NULL, NULL, setupDir, 0, 0, SW_SHOW);
::PostQuitMessage(0);
return TRUE;
}
BOOL Cfg::Init(WCHAR *user_dir, WCHAR *virtual_dir)
{
WCHAR buf[MAX_PATH], path[MAX_PATH], *fname = NULL;
::GetModuleFileNameW(NULL, buf, MAX_PATH);
::GetFullPathNameW(buf, MAX_PATH, path, &fname);
if (!fname) return FALSE;
execPath = wcsdup(path);
fname[-1] = 0; // remove '\\' of dir\\fname
execDir = wcsdup(path);
errLogPath = NULL;
userDir = NULL;
virtualDir = NULL;
if (IsWinVista() && TIsVirtualizedDirW(execDir)) {
if (user_dir) {
userDir = wcsdup(user_dir);
if (virtual_dir) virtualDir = wcsdup(virtual_dir);
}
else {
WCHAR virtual_store[MAX_PATH];
WCHAR fastcopy_dir[MAX_PATH];
WCHAR *fastcopy_dirname = NULL;
::GetFullPathNameW(path, MAX_PATH, fastcopy_dir, &fastcopy_dirname);
::SHGetSpecialFolderPathW(NULL, buf, CSIDL_APPDATA, FALSE);
MakePathW(path, buf, fastcopy_dirname);
userDir = wcsdup(path);
wcscpy(buf, execDir);
#ifdef _WIN64
ConvertToX86Dir(buf);
#endif
if (!TMakeVirtualStorePathW(buf, virtual_store)) return FALSE;
virtualDir = wcsdup(virtual_store);
}
ConvertVirtualStoreConf(execDir, userDir, virtualDir);
}
if (!userDir) userDir = wcsdup(execDir);
WCHAR ini_path[MAX_PATH];
MakePathW(ini_path, userDir, FASTCOPY_INI);
ini.Init(ini_path);
ini.SetSection(MAIN_SECTION);
needIniConvert = FALSE;
if (ini.GetInt(INI_VERSION_KEY, -1) < CUR_INI_VERSION) {
if (::GetFileAttributesW(ini_path) != 0xffffffff) {
WCHAR bak_path[MAX_PATH];
wcscpy(bak_path, ini_path);
wcscat(bak_path, L".bak");
if (!::MoveFileExW(ini_path, bak_path, MOVEFILE_REPLACE_EXISTING)) {
::DeleteFileW(ini_path);
}
ini.UnInit();
ini.Init(ini_path);
}
WCHAR ini_path_old[MAX_PATH];
MakePathW(ini_path_old, userDir, FASTCOPY_INI_OLD);
if (::GetFileAttributesW(ini_path_old) != 0xffffffff) {
ini.UnInit();
ini.Init(ini_path_old); // loading done
ini.SetIniFileNameW(ini_path);
needIniConvert = TRUE;
// obsolete message file
MakePathW(ini_path_old, userDir, FASTCOPY_INI_OBSOLETE);
HANDLE hFile = ::CreateFileW(ini_path_old, GENERIC_WRITE, 0, 0, CREATE_ALWAYS,
FILE_ATTRIBUTE_NORMAL, 0);
DWORD size;
::WriteFile(hFile, FASTCOPY_OBSOLETE_MSG, sizeof(FASTCOPY_OBSOLETE_MSG), &size, 0);
::CloseHandle(hFile);
}
}
return TRUE;
}
BOOL TInstDlg::UnInstall(void)
{
char buf[MAX_PATH];
char setupDir[MAX_PATH] = "";
::GetModuleFileName(NULL, setupDir, sizeof(setupDir));
GetParentDir(setupDir, setupDir);
BOOL is_shext = FALSE;
is_shext = ShellExtFunc(setupDir, CHECK_SHELLEXT);
if (is_shext && IsWinVista() && !::IsUserAnAdmin()) {
RunAsAdmin(TRUE);
return TRUE;
}
if (MessageBox(GetLoadStr(IDS_START), UNINSTALL_STR, MB_OKCANCEL|MB_ICONINFORMATION) != IDOK)
return FALSE;
// スタートメニュー＆デスクトップから削除
TRegistry reg(HKEY_CURRENT_USER, BY_MBCS);
if (reg.OpenKey(REGSTR_SHELLFOLDERS)) {
char *regStr[] = { REGSTR_PROGRAMS, REGSTR_DESKTOP, NULL };
for (int cnt=0; regStr[cnt] != NULL; cnt++) {
if (reg.GetStr(regStr[cnt], buf, sizeof(buf))) {
if (cnt == 0)
RemoveSameLink(buf);
::wsprintf(buf + strlen(buf), "\\%s", FASTCOPY_SHORTCUT);
Wstr w_buf(buf, BY_MBCS);
DeleteLinkW(w_buf.Buf());
}
}
reg.CloseKey();
}
ShellExtFunc(setupDir, UNREGISTER_SHELLEXT);
#ifdef _WIN64
if (1) {
#else
if (TIsWow64()) {
#endif
SHELLEXECUTEINFO sei = { sizeof(sei) };
char arg[1024];
sprintf(arg, "\"%s\\%s\",%s", setupDir, CURRENT_SHEXTDLL_EX, "DllUnregisterServer");
sei.lpFile = "rundll32.exe";
sei.lpParameters = arg;
ShellExecuteEx(&sei);
}
// レジストリからアンインストール情報を削除
if (reg.OpenKey(REGSTR_PATH_UNINSTALL)) {
if (reg.OpenKey(FASTCOPY)) {
if (reg.GetStr(REGSTR_VAL_UNINSTALLER_COMMANDLINE, setupDir, sizeof(setupDir)))
GetParentDir(setupDir, setupDir);
reg.CloseKey();
}
reg.DeleteKey(FASTCOPY);
reg.CloseKey();
}
// 終了メッセージ
MessageBox(is_shext ? GetLoadStr(IDS_UNINSTSHEXTFIN) : GetLoadStr(IDS_UNINSTFIN));
// インストールディレクトリを開く
if (GetFileAttributes(setupDir) != 0xffffffff) {
::ShellExecute(NULL, NULL, setupDir, 0, 0, SW_SHOW);
}
// AppDataディレクトリを開く
if (IsWinVista()) {
WCHAR wbuf[MAX_PATH] = L"", upath[MAX_PATH] = L"";
WCHAR fastcopy_dir[MAX_PATH] = L"", *fastcopy_dirname = NULL;
Wstr w_setup(setupDir);
if (TIsVirtualizedDirW(w_setup.Buf())) {
if (::SHGetSpecialFolderPathW(NULL, wbuf, CSIDL_APPDATA, FALSE)) {
::GetFullPathNameW(w_setup.Buf(), MAX_PATH, fastcopy_dir, &fastcopy_dirname);
MakePathW(upath, wbuf, fastcopy_dirname);
if (::GetFileAttributesW(upath) != 0xffffffff) {
::ShellExecuteW(NULL, NULL, upath, 0, 0, SW_SHOW);
}
}
}
}
::PostQuitMessage(0);
return TRUE;
}
BOOL ReadLink(char *src, char *dest, char *arg=NULL)
{
IShellLink *shellLink;
IPersistFile *persistFile;
WCHAR wbuf[MAX_PATH];
BOOL ret = FALSE;
if (SUCCEEDED(CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink,
(void **)&shellLink))) {
if (SUCCEEDED(shellLink->QueryInterface(IID_IPersistFile, (void **)&persistFile))) {
AtoW(src, wbuf, MAX_PATH);
if (SUCCEEDED(persistFile->Load(wbuf, STGM_READ))) {
if (SUCCEEDED(shellLink->GetPath(dest, MAX_PATH, NULL, 0))) {
if (arg) {
shellLink->GetArguments(arg, MAX_PATH);
}
ret = TRUE;
}
}
persistFile->Release();
}
shellLink->Release();
}
return ret;
}
BOOL TInstDlg::Install(void)
{
char buf[MAX_PATH], setupDir[MAX_PATH], setupPath[MAX_PATH];
BOOL is_delay_copy = FALSE;
int len;
// インストールパス設定
len = GetDlgItemText(FILE_EDIT, setupDir, sizeof(setupDir));
if (setupDir[len-1] == '\\') setupDir[len-1] = 0;
Wstr w_setup(setupDir);
if (IsWinVista() && TIsVirtualizedDirW(w_setup.Buf())) {
if (!::IsUserAnAdmin()) {
return RunAsAdmin(TRUE);
}
else if (cfg.runImme && cfg.setupDir && lstrcmpiW(w_setup.Buf(), cfg.setupDir)) {
return MessageBox(GetLoadStr(IDS_ADMINCHANGE)), FALSE;
}
}
CreateDirectory(setupDir, NULL);
DWORD attr = GetFileAttributes(setupDir);
if (attr == 0xffffffff || (attr & FILE_ATTRIBUTE_DIRECTORY) == 0)
return MessageBox(GetLoadStr(IDS_NOTCREATEDIR)), FALSE;
MakePath(setupPath, setupDir, FASTCOPY_EXE);
if (MessageBox(GetLoadStr(IDS_START), INSTALL_STR, MB_OKCANCEL|MB_ICONINFORMATION) != IDOK)
return FALSE;
// ファイルコピー
if (cfg.mode == SETUP_MODE) {
char installPath[MAX_PATH], orgDir[MAX_PATH];
::GetModuleFileName(NULL, orgDir, sizeof(orgDir));
GetParentDir(orgDir, orgDir);
for (int cnt=0; SetupFiles[cnt] != NULL; cnt++) {
MakePath(buf, orgDir, SetupFiles[cnt]);
MakePath(installPath, setupDir, SetupFiles[cnt]);
if (MiniCopy(buf, installPath) || IsSameFile(buf, installPath))
continue;
if ((strcmp(SetupFiles[cnt], CURRENT_SHEXTDLL_EX) == 0 ||
strcmp(SetupFiles[cnt], CURRENT_SHEXTDLL) == 0) && DelayCopy(buf, installPath)) {
is_delay_copy = TRUE;
continue;
}
return MessageBox(installPath, GetLoadStr(IDS_NOTCREATEFILE)), FALSE;
}
}
// スタートメニュー＆デスクトップに登録
char *linkPath[] = { WtoA(cfg.startMenu), WtoA(cfg.deskTop), NULL };
BOOL execFlg[] = { cfg.programLink, cfg.desktopLink, NULL };
for (int cnt=0; linkPath[cnt]; cnt++) {
strcpy(buf, linkPath[cnt]);
if (cnt != 0 || RemoveSameLink(linkPath[cnt], buf) == FALSE) {
::wsprintf(buf + strlen(buf), "\\%s", FASTCOPY_SHORTCUT);
}
if (execFlg[cnt]) {
Wstr w_setup(setupPath, BY_MBCS);
Wstr w_buf(buf, BY_MBCS);
SymLinkW(w_setup.Buf(), w_buf.Buf());
}
else {
Wstr w_buf(buf, BY_MBCS);
DeleteLinkW(w_buf.Buf());
}
}
#if 0
// レジストリにアンインストール情報を登録
if (reg.OpenKey(REGSTR_PATH_UNINSTALL)) {
if (reg.CreateKey(FASTCOPY)) {
MakePath(buf, setupDir, INSTALL_EXE);
strcat(buf, " /r");
reg.SetStr(REGSTR_VAL_UNINSTALLER_DISPLAYNAME, FASTCOPY);
reg.SetStr(REGSTR_VAL_UNINSTALLER_COMMANDLINE, buf);
reg.CloseKey();
}
reg.CloseKey();
}
#endif
if (IsWinVista() && TIsVirtualizedDirW(w_setup.Buf())) {
WCHAR wbuf[MAX_PATH] = L"", old_path[MAX_PATH] = L"", usr_path[MAX_PATH] = L"";
WCHAR fastcopy_dir[MAX_PATH], *fastcopy_dirname = NULL;
::GetFullPathNameW(w_setup.Buf(), MAX_PATH, fastcopy_dir, &fastcopy_dirname);
if (cfg.appData) {
wcscpy(usr_path, cfg.appData);
}
else {
::SHGetSpecialFolderPathW(NULL, wbuf, CSIDL_APPDATA, FALSE);
MakePathW(usr_path, wbuf, fastcopy_dirname);
}
ConvertVirtualStoreConf(w_setup.Buf(), usr_path, cfg.virtualDir);
}
// コピーしたアプリケーションを起動
const char *msg = GetLoadStr(is_delay_copy ? IDS_DELAYSETUPCOMPLETE : IDS_SETUPCOMPLETE);
int flg = MB_OKCANCEL|MB_ICONINFORMATION;
TLaunchDlg dlg(msg, this);
if (dlg.Exec() == IDOK) {
ShellExecute(NULL, "open", setupPath, "", setupDir, SW_SHOW);
}
::PostQuitMessage(0);
return TRUE;
}
