1 /*******************************************************************************
\r
2 TPI - flexible but useless plug-in framework.
\r
3 Copyright (C) 2002-2009 Silky
\r
5 This library is free software; you can redistribute it and/or modify it under
\r
6 the terms of the GNU Lesser General Public License as published by the Free
\r
7 Software Foundation; either version 2.1 of the License, or (at your option)
\r
10 This library is distributed in the hope that it will be useful, but WITHOUT
\r
11 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
\r
12 FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
\r
15 You should have received a copy of the GNU Lesser General Public License along
\r
16 with this library; if not, write to the Free Software Foundation, Inc.,
\r
17 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
\r
20 *******************************************************************************/
\r
22 //******************************************************************************
\r
24 //******************************************************************************
\r
26 #define MYUSE_LIBPATH 1
\r
27 #include "../../common/header/plugin.h"
\r
28 #include "../../common/header/plugin-extra.h"
\r
29 #include "../../common/library/library.h"
\r
30 #include "../../common/library/xmldoc.h"
\r
31 #include <wx/tokenzr.h>
\r
32 #include "cuiWrapper.h"
\r
33 #define min(a, b) ((a) < (b) ? (a) : (b))
\r
35 //******************************************************************************
\r
37 //******************************************************************************
\r
42 wxString szExeFileAlt;
\r
43 wxString szListCommand;
\r
44 wxULongLong_t nLibIndex;
\r
50 //******************************************************************************
\r
52 //******************************************************************************
\r
54 int myExecute(const wxString & szCommandLine, wxString * szOutput, const wxString & szCwd, bool fWine)
\r
57 wxString sz = ::wxGetCwd();
\r
58 ::wxSetWorkingDirectory(szCwd);
\r
59 FILE * fp = popen(fWine ? (wxT("wine ") + szCommandLine).ToUTF8() : szCommandLine.ToUTF8(), "r");
\r
60 ::wxSetWorkingDirectory(sz);
\r
63 wxLogError(L"Error :\n\nCommandLine:\n%s", szCommandLine.c_str());
\r
64 return TPI_ERROR_U_USE_LIBRARY;
\r
67 if (szOutput != NULL)
\r
72 memset(sz, 0, sizeof(sz));
\r
73 fread(sz, sizeof(char), sizeof(sz) - 1, fp);
\r
74 * szOutput += UTF82String(sz);
\r
76 // ::wxMessageBox(* szOutput);
\r
79 // 127を返した場合はコマンドが存在しない。
\r
80 int nErrorCode = pclose(fp);
\r
81 return (WIFEXITED(nErrorCode) && WEXITSTATUS(nErrorCode) != 127) ? TPI_ERROR_SUCCESS : TPI_ERROR_U_USE_LIBRARY;
\r
83 SECURITY_ATTRIBUTES sa;
\r
84 memset(& sa, 0, sizeof(SECURITY_ATTRIBUTES));
\r
85 sa.bInheritHandle = TRUE;
\r
86 sa.nLength = sizeof(SECURITY_ATTRIBUTES);
\r
87 HANDLE hRead, hWrite;
\r
88 if (! ::CreatePipe(& hRead, & hWrite, & sa, 4096))
\r
90 return TPI_ERROR_U_USE_LIBRARY;
\r
93 memset(& si, 0, sizeof(STARTUPINFO));
\r
94 si.cb = sizeof(STARTUPINFO);
\r
95 si.dwFlags = STARTF_USESTDHANDLES;
\r
96 si.hStdOutput = hWrite;
\r
97 si.hStdError = hWrite;
\r
98 PROCESS_INFORMATION pi;
\r
99 if (! ::CreateProcess(NULL, szCommandLine.wchar_str(), NULL, NULL, TRUE, CREATE_NO_WINDOW, NULL, szCwd.IsEmpty() ? NULL : szCwd.wchar_str(), & si, & pi))
\r
101 return TPI_ERROR_U_USE_LIBRARY;
\r
103 ::CloseHandle(pi.hThread);
\r
105 if (szOutput != NULL)
\r
111 bSignal = ::WaitForSingleObject(pi.hProcess, 100) == WAIT_OBJECT_0;
\r
112 ::PeekNamedPipe(hRead, NULL, 0, NULL, & dwSize, NULL);
\r
116 memset(sz, 0, sizeof(sz));
\r
117 ::ReadFile(hRead, & sz, sizeof(sz) - 1, & dwSize, NULL);
\r
118 * szOutput += wxString(sz);
\r
119 // UTF-8以外の文字コードだと、UTF82Stringを使うと書庫が開けなくなる。データの切り出しにも影響。
\r
120 // * szOutput += UTF82String(sz);
\r
121 // ::MessageBoxA(NULL, sz, NULL, 0);
\r
122 ::PeekNamedPipe(hRead, NULL, 0, NULL, & dwSize, NULL);
\r
128 ::CloseHandle(pi.hProcess);
\r
129 ::CloseHandle(hRead);
\r
130 ::CloseHandle(hWrite);
\r
131 return TPI_ERROR_SUCCESS;
\r
135 PosInfo MakePosInfo(const wxString & szPrefix)
\r
138 pi.nStart = myGetAttributeInt(& g_LibInfo.node, wxT("list-") + szPrefix + wxT("-s"));
\r
139 pi.nCount = myGetAttributeInt(& g_LibInfo.node, wxT("list-") + szPrefix + wxT("-c"));
\r
140 pi.nLine = myGetAttributeInt(& g_LibInfo.node, wxT("list-") + szPrefix + wxT("-l"));
\r
144 wxULongLong_t GetSize(const PosInfo & pi, wxULongLong_t nCurrent, const wxArrayString & as)
\r
146 if (pi.nStart == 0 && pi.nCount == 0)
\r
151 wxULongLong_t nTemp = 0, nPos = nCurrent + pi.nLine;
\r
152 wxString sz = pi.nCount == 0 ? as[nPos].Mid(pi.nStart) : as[nPos].Mid(pi.nStart, pi.nCount);
\r
153 sz.ToULongLong(& nTemp);
\r
157 //******************************************************************************
\r
159 //******************************************************************************
\r
166 int __stdcall GetPluginInformation
\r
168 unsigned int _uInfoId,
\r
175 return TPI_ERROR_D_PARAMETER;
\r
179 case TPI_INFO_VERSION_MAJOR:
\r
180 case TPI_INFO_VERSION_MINOR:
\r
181 * (int *) _pPtr = 0;
\r
183 case TPI_INFO_VERSION_API:
\r
184 * (int *) _pPtr = 2;
\r
186 case TPI_INFO_HANDLE_ON_COMMAND:
\r
187 * (int *) _pPtr = 0;
\r
190 return TPI_ERROR_D_UNSUPPORTED;
\r
192 return TPI_ERROR_SUCCESS;
\r
195 int __stdcall GetFormatInformation(TPI_FORMATINFO * _fiInfo, bool _bFirst)
\r
197 static wxULongLong_t s_nFileId;
\r
198 static wxXmlDocument xmlDoc(myMakeXMLName(wxT("cuiWrapper")));
\r
199 static wxXmlNode * xmlLibrary;
\r
204 xmlLibrary = myGetFirstLib(& xmlDoc);
\r
208 xmlLibrary = myGetNextLib(xmlLibrary);
\r
210 if (xmlLibrary == NULL)
\r
213 return TPI_ERROR_S_ENDOFDATA;
\r
216 MakeFormatInfo(xmlLibrary, wxT("cuiWrapper"), _fiInfo, s_nFileId++);
\r
217 if (myExecute(_fiInfo->szEngineName, NULL, wxEmptyString, _fiInfo->szEngineName.MakeLower().EndsWith(wxT(".exe"))) != TPI_ERROR_SUCCESS)
\r
219 _fiInfo->eSupportedCommand = 0;
\r
221 return TPI_ERROR_SUCCESS;
\r
224 int __stdcall LoadPlugin
\r
226 const wxString & _szArcName,
\r
228 wxULongLong_t _nTypeId
\r
232 wxXmlDocument xmlDoc(myMakeXMLName(wxT("cuiWrapper")));
\r
233 wxXmlNode * xmlLibrary;
\r
236 if (_prProc != NULL)
\r
238 g_prProc = * _prProc;
\r
241 // 対象が存在するならば対応するライブラリを調査、
\r
242 // 対象が存在しないならば指示されたライブラリをロード。
\r
244 if (_szArcName.IsEmpty())
\r
246 return TPI_ERROR_SUCCESS;
\r
248 if (! ::wxFileExists(_szArcName))
\r
250 xmlLibrary = myGetFirstLib(& xmlDoc, _nTypeId);
\r
251 if (xmlLibrary == NULL)
\r
254 return TPI_ERROR_UNDEFINED;
\r
256 g_LibInfo.szExeFile = xmlLibrary->GetAttribute(wxT("name"), wxEmptyString);
\r
257 g_LibInfo.szExeFileAlt = xmlLibrary->GetAttribute(wxT("name-alt"), wxEmptyString);
\r
258 g_LibInfo.node = * xmlLibrary;
\r
259 g_LibInfo.nLibIndex = _nTypeId;
\r
260 return TPI_ERROR_SUCCESS;
\r
263 // 無限ループに陥らないよう上限を設定。
\r
264 xmlLibrary = myGetFirstLib(& xmlDoc);
\r
265 for (g_LibInfo.nLibIndex = 0; g_LibInfo.nLibIndex < 300 && xmlLibrary != NULL; g_LibInfo.nLibIndex++)
\r
268 wxFileName fnArchive(_szArcName);
\r
269 wxArrayString asExt = ::wxStringTokenize(xmlLibrary->GetAttribute(wxT("suffix"), wxEmptyString), wxT(";"));
\r
270 if (xmlLibrary->HasAttribute(wxT("list")))
\r
272 for (size_t i = 0; i < asExt.GetCount(); i++)
\r
274 // .tar.XXXなど二重判定への対応。
\r
275 // if (asExt[i].IsSameAs(fnArchive.GetExt(), false))
\r
276 if (fnArchive.GetFullName().MakeLower().EndsWith(wxT('.') + asExt[i].MakeLower()))
\r
279 g_LibInfo.szExeFile = xmlLibrary->GetAttribute(wxT("name"), wxEmptyString);
\r
280 g_LibInfo.szExeFileAlt = xmlLibrary->GetAttribute(wxT("name-alt"), wxEmptyString);
\r
281 g_LibInfo.node = * xmlLibrary;
\r
282 g_LibInfo.nLibIndex = _nTypeId;
\r
283 return TPI_ERROR_SUCCESS;
\r
287 xmlLibrary = myGetNextLib(xmlLibrary);
\r
289 return TPI_ERROR_U_LOAD_LIBRARY;
\r
292 int __stdcall FreePlugin
\r
294 void * // _pReserved
\r
297 return TPI_ERROR_SUCCESS;
\r
300 int __stdcall OpenArchive
\r
302 const wxString & _szArcName,
\r
303 void * * _hArchive,
\r
304 wxULongLong_t * _nFileCount
\r
308 TPI_SWITCHES swInfo;
\r
309 swInfo.szArcName = _szArcName;
\r
310 if (myExecute(g_LibInfo.szExeFile + wxT(" ") + MakeCommandLineSend(g_LibInfo.node.GetAttribute(wxT("list"), wxEmptyString), & swInfo), & szOutput, wxEmptyString, g_LibInfo.szExeFile.MakeLower().EndsWith(wxT(".exe"))) != TPI_ERROR_SUCCESS)
\r
312 return TPI_ERROR_U_USE_LIBRARY;
\r
314 wxArrayString * as = new wxArrayString(::wxStringTokenize(szOutput, wxT("\r\n")));
\r
315 * _hArchive = (void *) as;
\r
317 if (_nFileCount != NULL)
\r
319 * _nFileCount = as->GetCount();
\r
322 wxString szStartLine = g_LibInfo.node.GetAttribute(wxT("list-line-s"), wxEmptyString);
\r
323 if ((! szStartLine.IsEmpty() && as->Index(szStartLine) == wxNOT_FOUND) || as->IsEmpty())
\r
327 return TPI_ERROR_ARC_UNSUPPORTED;
\r
330 return TPI_ERROR_SUCCESS;
\r
333 int __stdcall CloseArchive
\r
338 wxArrayString * as = (wxArrayString *) _hArchive;
\r
341 return TPI_ERROR_SUCCESS;
\r
344 int __stdcall GetFileInformation
\r
347 TPI_FILEINFO * _fiInfo,
\r
351 static size_t s_nCurrentLine;
\r
352 static wxULongLong_t s_nFileId;
\r
353 if (_hArchive == NULL)
\r
355 return TPI_ERROR_UNDEFINED;
\r
357 wxArrayString asOutput = * (wxArrayString *) _hArchive;
\r
359 // XMLからの読み込みは初回に行う。
\r
360 static wxString szEndLine, szDateFormat;
\r
361 static wxULongLong_t nProcessPerLine;
\r
362 static PosInfo piFName, piPSize, piUSize, piDate;
\r
366 wxString szStartLine = g_LibInfo.node.GetAttribute(wxT("list-line-s"), wxEmptyString);
\r
367 if (! szStartLine.IsEmpty())
\r
369 // 開始行の次の行にセット。エラーはOpenArchiveでチェック済み。
\r
370 s_nCurrentLine = asOutput.Index(szStartLine) + 1;
\r
374 nProcessPerLine = myGetAttributeInt(& g_LibInfo.node, wxT("list-line-c"), 1);
\r
375 szEndLine = g_LibInfo.node.GetAttribute(wxT("list-line-e"), szStartLine);
\r
376 szDateFormat = g_LibInfo.node.GetAttribute(wxT("list-date-f"), wxDefaultDateTimeFormat);
\r
377 piFName = MakePosInfo(wxT("fname"));
\r
378 piPSize = MakePosInfo(wxT("psize"));
\r
379 piUSize = MakePosInfo(wxT("usize"));
\r
380 piDate = MakePosInfo(wxT("date"));
\r
383 if (s_nCurrentLine >= asOutput.GetCount())
\r
385 // 空行で終わるとき以外はエラーとする。
\r
386 return szEndLine.IsEmpty() ? TPI_ERROR_S_ENDOFDATA : TPI_ERROR_ARC_UNSUPPORTED;
\r
390 if (asOutput[s_nCurrentLine] == szEndLine)
\r
392 return TPI_ERROR_S_ENDOFDATA;
\r
396 _fiInfo->szStoredName = piFName.nCount == 0 ? asOutput[s_nCurrentLine + piFName.nLine].Mid(piFName.nStart) : asOutput[s_nCurrentLine + piFName.nLine].Mid(piFName.nStart, piFName.nCount);
\r
397 _fiInfo->szStoredName.Trim();
\r
398 _fiInfo->fnFileName = wxFileName(_fiInfo->szStoredName, wxPATH_DOS);
\r
401 _fiInfo->nPackedSize = GetSize(piPSize, s_nCurrentLine, asOutput);
\r
402 _fiInfo->nUnpackedSize = GetSize(piUSize, s_nCurrentLine, asOutput);
\r
405 if (piDate.nStart != 0 || piDate.nCount != 0)
\r
407 _fiInfo->tmModify.ParseFormat(piDate.nCount == 0 ? asOutput[s_nCurrentLine + piDate.nLine].Mid(piDate.nStart) : asOutput[s_nCurrentLine + piDate.nLine].Mid(piDate.nStart, piDate.nCount), szDateFormat);
\r
411 _fiInfo->nFileId = s_nFileId++;
\r
412 s_nCurrentLine += nProcessPerLine;
\r
414 return TPI_ERROR_SUCCESS;
\r
417 int __stdcall GetArchiveInformation
\r
420 TPI_ARCHIVEINFO * _aiInfo
\r
424 MakeFormatInfo(& g_LibInfo.node, wxT("cuiWrapper"), & _aiInfo->fiInfo, 0);
\r
425 return TPI_ERROR_SUCCESS;
\r
428 int __stdcall Command
\r
430 wxULongLong_t _eCommand,
\r
431 TPI_SWITCHES * _swInfo,
\r
432 void *,// _hArchive,
\r
433 const wxArrayString & _szFiles
\r
436 // xmlからコマンドラインを取得。
\r
437 wxString szPath, szCommandLine;
\r
440 if (! g_LibInfo.node.GetAttribute(
\r
441 _eCommand == TPI_COMMAND_CREATE ? wxT("create") :
\r
442 _eCommand == TPI_COMMAND_ADD ? wxT("add") :
\r
443 _eCommand == TPI_COMMAND_EXTRACT ? wxT("extract") :
\r
444 _eCommand == TPI_COMMAND_DELETE ? wxT("delete") :
\r
445 _eCommand == TPI_COMMAND_UPDATE ? wxT("update") :
\r
446 _eCommand == TPI_COMMAND_TEST ? wxT("test") :
\r
447 _eCommand == TPI_COMMAND_REPAIR ? wxT("repair") :
\r
448 _eCommand == TPI_COMMAND_MOVE ? wxT("move") :
\r
449 _eCommand == TPI_COMMAND_SFX ? wxT("sfx") :
\r
450 _eCommand == TPI_COMMAND_UNSFX ? wxT("unsfx") : wxEmptyString, & szCommandLine))
\r
452 g_LibInfo.node.GetAttribute(
\r
453 _eCommand == TPI_COMMAND_CREATE ? wxT("create-alt") :
\r
454 _eCommand == TPI_COMMAND_ADD ? wxT("add-alt") :
\r
455 _eCommand == TPI_COMMAND_EXTRACT ? wxT("extract-alt") :
\r
456 _eCommand == TPI_COMMAND_DELETE ? wxT("delete-alt") :
\r
457 _eCommand == TPI_COMMAND_UPDATE ? wxT("update-alt") :
\r
458 _eCommand == TPI_COMMAND_TEST ? wxT("test-alt") :
\r
459 _eCommand == TPI_COMMAND_REPAIR ? wxT("repair-alt") :
\r
460 _eCommand == TPI_COMMAND_MOVE ? wxT("move-alt") :
\r
461 _eCommand == TPI_COMMAND_SFX ? wxT("sfx-alt") :
\r
462 _eCommand == TPI_COMMAND_UNSFX ? wxT("unsfx-alt") : wxEmptyString, & szCommandLine);
\r
465 if (szCommandLine.IsEmpty())
\r
467 return TPI_ERROR_U_USE_LIBRARY;
\r
470 // コマンドライン・レスポンスファイル作成。
\r
472 szResponceFileName = MakeResponceFile(_szFiles, myGetAttributeBool(& g_LibInfo.node, wxT("quote-resp"), true)),
\r
473 szCommandLineSend = MakeCommandLineSend(szCommandLine, _swInfo, _szFiles, szResponceFileName);
\r
477 int nErrorCode = myExecute(g_LibInfo.szExeFile + wxT(" ") + szCommandLineSend, & szOutput, _swInfo->fnDestinationDirectory.GetFullPath(), g_LibInfo.szExeFile.MakeLower().EndsWith(wxT(".exe")));
\r
480 ::wxRemoveFile(szResponceFileName);
\r
482 if (nErrorCode != TPI_ERROR_SUCCESS)
\r
484 wxLogError(L"Error :\n%x\n\nCommandLine:\n%s\n\nOutput:\n%s", nErrorCode, szCommandLineSend.c_str(), szOutput.c_str());
\r