/*
BrowseDlg用 WM_COMMAND 処理
*/
BOOL TBrowseDirDlg::EvCommand(WORD wNotifyCode, WORD wID, LPARAM hwndCtl)
{
switch (wID)
{
case MKDIR_BUTTON:
{
char dirBuf[MAX_PATH_U8], path[MAX_PATH_U8];
TInputDlg dlg(dirBuf, this);
if (dlg.Exec() != IDOK)
return TRUE;
MakePath(path, fileBuf, dirBuf);
if (CreateDirectoryU8(path, NULL))
{
strcpy(fileBuf, path);
dirtyFlg = TRUE;
PostMessage(WM_CLOSE, 0, 0);
}
}
return TRUE;
case RMDIR_BUTTON:
if (RemoveDirectoryU8(fileBuf))
{
GetParentDirU8(fileBuf, fileBuf);
dirtyFlg = TRUE;
PostMessage(WM_CLOSE, 0, 0);
}
return TRUE;
}
return FALSE;
}
BOOL SaveImageFile(Cfg *cfg, const char *fname, VBuf *buf)
{
char path[MAX_PATH_U8];
DWORD size;
HANDLE hFile;
if (!MakeImageFolder(cfg, path)) return FALSE;
CreateDirectoryU8(path, 0);
strcat(path, "\\");
strcat(path, fname);
if ((hFile = CreateFileU8(path, GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL,
CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0)) == INVALID_HANDLE_VALUE) return FALSE;
WriteFile(hFile, buf->Buf(), buf->Size(), &size, 0);
CloseHandle(hFile);
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 TRecvDlg::RecvDirFile(void)
{
#define BIG_ALLOC 50
#define PEEK_SIZE 8
if (fileObj->status == FS_DIRFILESTART || fileObj->status == FS_TRANSINFO)
{
int size;
if (fileObj->infoLen == 0)
{
if ((size = ::recv(fileObj->conInfo->sd, fileObj->info + (int)fileObj->offset, PEEK_SIZE - (int)fileObj->offset, 0)) <= 0)
return FALSE;
if ((fileObj->offset += size) < PEEK_SIZE)
return TRUE;
fileObj->info[fileObj->offset] = 0;
if ((fileObj->infoLen = strtoul(fileObj->info, 0, 16)) >= sizeof(fileObj->info) -1 || fileObj->infoLen <= 0)
return FALSE; // too big or small
}
if (fileObj->offset < fileObj->infoLen)
{
if ((size = ::recv(fileObj->conInfo->sd, fileObj->info + (int)fileObj->offset, fileObj->infoLen - (int)fileObj->offset, 0)) <= 0)
return FALSE;
fileObj->offset += size;
}
if (fileObj->offset == fileObj->infoLen)
{
fileObj->info[fileObj->infoLen] = 0;
if (DecodeDirEntry(fileObj->info, &fileObj->curFileInfo, fileObj->u8fname) == FALSE)
return FALSE; // Illegal entry
fileObj->offset = fileObj->infoLen = 0; // 初期化
if (GET_MODE(fileObj->curFileInfo.Attr()) == IPMSG_FILE_DIR)
{
char buf[MAX_BUF];
const char *fname = fileObj->dirCnt == 0 ? fileObj->fileInfo->Fname() : fileObj->curFileInfo.Fname();
if (MakePath(buf, fileObj->path, fname) >= MAX_PATH_U8)
return MessageBoxU8(buf, GetLoadStrU8(IDS_PATHTOOLONG)), FALSE;
if (IsSafePath(buf, fname) == FALSE)
return FALSE;
if (CreateDirectoryU8(buf, NULL) == FALSE)
return FALSE;
strncpyz(fileObj->path, buf, MAX_PATH_U8);
fileObj->dirCnt++;
}
else if (GET_MODE(fileObj->curFileInfo.Attr()) == IPMSG_FILE_RETPARENT)
{
if (fileObj->curFileInfo.Mtime()) // directory の time stamp をあわせる(NT系のみ)
{
FILETIME ft;
HANDLE hFile;
UnixTime2FileTime(fileObj->curFileInfo.Mtime(), &ft);
if ((hFile = CreateFileU8(fileObj->path, GENERIC_WRITE, FILE_SHARE_READ, 0, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, 0)) != INVALID_HANDLE_VALUE)
{
::SetFileTime(hFile, NULL, NULL, &ft);
::CloseHandle(hFile);
}
}
if (fileObj->curFileInfo.Attr() & IPMSG_FILE_RONLYOPT)
SetFileAttributesU8(fileObj->path, FILE_ATTRIBUTE_READONLY);
if (--fileObj->dirCnt <= 0)
{
fileObj->status = FS_COMPLETE;
return TRUE;
}
if (PathToDir(fileObj->path, fileObj->path) == FALSE)
return FALSE;
}
else {
if (fileObj->dirCnt == 0)
return FALSE;
if (fileObj->curFileInfo.Size() == 0) // 0byte file
{
if (OpenRecvFile()) // 0byteの場合は作成失敗を無視
CloseRecvFile(TRUE);
}
fileObj->status = fileObj->curFileInfo.Size() ? FS_TRANSFILE : FS_TRANSINFO;
}
return TRUE;
}
}
if (fileObj->status == FS_TRANSFILE)
{
if (RecvFile() != TRUE)
{
CloseRecvFile();
return FALSE;
}
if (fileObj->status == FS_ENDFILE)
{
CloseRecvFile(TRUE);
fileObj->status = FS_TRANSINFO;
}
}
return TRUE;
}
