//Arguments:
// hConWnd - Хэндл КОНСОЛЬНОГО окна (по нему формируется имя пайпа для GUI)
// pIn - выполняемая команда
// nTimeout- таймаут подключения
//Returns:
// CESERVER_REQ. Его необходимо освободить через free(...);
//WARNING!!!
// Эта процедура не может получить с сервера более 600 байт данных!
// В заголовке hOwner в дебаге может быть отображена ошибка
CESERVER_REQ* ExecuteCmd(const wchar_t* szPipeName, CESERVER_REQ* pIn, DWORD nWaitPipe, HWND hOwner, BOOL bAsyncNoResult, DWORD nServerPID)
{
CESERVER_REQ* pOut = NULL;
HANDLE hPipe = NULL;
BYTE cbReadBuf[600]; // чтобы CESERVER_REQ_OUTPUTFILE поместился
wchar_t szErr[MAX_PATH*2]; szErr[0] = 0;
BOOL fSuccess = FALSE;
DWORD cbRead = 0, /*dwMode = 0,*/ dwErr = 0;
if (!pIn || !szPipeName)
{
_ASSERTE(pIn && szPipeName);
return NULL;
}
pIn->hdr.bAsync = bAsyncNoResult;
_ASSERTE(pIn->hdr.nSrcPID && pIn->hdr.nSrcThreadId);
_ASSERTE(pIn->hdr.cbSize >= sizeof(pIn->hdr));
hPipe = ExecuteOpenPipe(szPipeName, szErr, NULL/*Сюда хорошо бы имя модуля подкрутить*/, nServerPID, nWaitPipe);
if (hPipe == NULL || hPipe == INVALID_HANDLE_VALUE)
{
#ifdef _DEBUG
dwErr = GetLastError();
// в заголовке "чисто" запущенного фара появляются отладочные(?) сообщения
// по идее - не должны, т.к. все должно быть через мэппинг
_ASSERTEX(hPipe != NULL && hPipe != INVALID_HANDLE_VALUE);
#ifdef CONEMU_MINIMAL
SetConsoleTitle(szErr);
#else
if (hOwner)
{
if (hOwner == myGetConsoleWindow())
SetConsoleTitle(szErr);
else
SetWindowText(hOwner, szErr);
}
#endif
#endif
return NULL;
}
//// Try to open a named pipe; wait for it, if necessary.
//while (1)
//{
// hPipe = CreateFile(
// szPipeName, // pipe name
// GENERIC_READ | // read and write access
// GENERIC_WRITE,
// 0, // no sharing
// NULL, // default security attributes
// OPEN_EXISTING, // opens existing pipe
// 0, // default attributes
// NULL); // no template file
//
// // Break if the pipe handle is valid.
// if (hPipe != INVALID_HANDLE_VALUE)
// break;
//
// // Exit if an error other than ERROR_PIPE_BUSY occurs.
// dwErr = GetLastError();
// if (dwErr != ERROR_PIPE_BUSY)
// {
// return NULL;
// }
//
// // All pipe instances are busy, so wait for 1 second.
// if (!WaitNamedPipe(szPipeName, nWaitPipe) )
// {
// return NULL;
// }
//}
//
//// The pipe connected; change to message-read mode.
//dwMode = PIPE_READMODE_MESSAGE;
//fSuccess = SetNamedPipeHandleState(
// hPipe, // pipe handle
// &dwMode, // new pipe mode
// NULL, // don't set maximum bytes
// NULL); // don't set maximum time
//if (!fSuccess)
//{
// CloseHandle(hPipe);
// return NULL;
//}
_ASSERTE(pIn->hdr.nSrcThreadId==GetCurrentThreadId());
if (bAsyncNoResult)
{
// Если нас не интересует возврат и нужно сразу вернуться
fSuccess = WriteFile(hPipe, pIn, pIn->hdr.cbSize, &cbRead, NULL);
#ifdef _DEBUG
dwErr = GetLastError();
_ASSERTE(fSuccess && (cbRead == pIn->hdr.cbSize));
#endif
return NULL;
}
else
{
WARNING("При Overlapped часто виснет в этом месте.");
// Send a message to the pipe server and read the response.
fSuccess = TransactNamedPipe(
hPipe, // pipe handle
(LPVOID)pIn, // message to server
pIn->hdr.cbSize, // message length
cbReadBuf, // buffer to receive reply
sizeof(cbReadBuf), // size of read buffer
&cbRead, // bytes read
NULL); // not overlapped
dwErr = GetLastError();
//CloseHandle(hPipe);
if (!fSuccess && (dwErr != ERROR_MORE_DATA))
{
//_ASSERTE(fSuccess || (dwErr == ERROR_MORE_DATA));
CloseHandle(hPipe);
return NULL;
}
}
if (cbRead < sizeof(CESERVER_REQ_HDR))
{
CloseHandle(hPipe);
return NULL;
}
pOut = (CESERVER_REQ*)cbReadBuf; // temporary
if (pOut->hdr.cbSize < cbRead)
{
CloseHandle(hPipe);
if (pOut->hdr.cbSize)
{
_ASSERTE(pOut->hdr.cbSize == 0 || pOut->hdr.cbSize >= cbRead);
DEBUGSTR(L"!!! Wrong nSize received from GUI server !!!\n");
}
return NULL;
}
if (pOut->hdr.nVersion != CESERVER_REQ_VER)
{
CloseHandle(hPipe);
DEBUGSTR(L"!!! Wrong nVersion received from GUI server !!!\n");
return NULL;
}
int nAllSize = pOut->hdr.cbSize;
pOut = (CESERVER_REQ*)malloc(nAllSize);
_ASSERTE(pOut);
if (!pOut)
{
CloseHandle(hPipe);
return NULL;
}
memmove(pOut, cbReadBuf, cbRead);
LPBYTE ptrData = ((LPBYTE)pOut)+cbRead;
nAllSize -= cbRead;
while (nAllSize>0)
{
// Break if TransactNamedPipe or ReadFile is successful
if (fSuccess)
break;
// Read from the pipe if there is more data in the message.
fSuccess = ReadFile(
hPipe, // pipe handle
ptrData, // buffer to receive reply
nAllSize, // size of buffer
&cbRead, // number of bytes read
NULL); // not overlapped
// Exit if an error other than ERROR_MORE_DATA occurs.
if (!fSuccess && (GetLastError() != ERROR_MORE_DATA))
break;
ptrData += cbRead;
nAllSize -= cbRead;
}
CloseHandle(hPipe);
if (pOut && (pOut->hdr.nCmd != pIn->hdr.nCmd))
{
_ASSERTE(pOut->hdr.nCmd == pIn->hdr.nCmd);
if (pOut->hdr.nCmd == 0)
{
ExecuteFreeResult(pOut);
pOut = NULL;
}
}
return pOut;
}
BOOL CConEmuPipe::Init(LPCTSTR asOp, BOOL abSilent)
{
wchar_t szErr[MAX_PATH*2];
mh_Pipe = ExecuteOpenPipe(ms_PipeName, szErr, gpConEmu->GetDefaultTitle());
if (!mh_Pipe || mh_Pipe == INVALID_HANDLE_VALUE)
{
MBoxA(szErr);
return FALSE;
}
return TRUE;
//DWORD dwErr = 0;
//
//// Try to open a named pipe; wait for it, if necessary.
//while (1)
//{
// mh_Pipe = CreateFile(
// ms_PipeName, // pipe name
// GENERIC_READ | // read and write access
// GENERIC_WRITE,
// 0, // no sharing
// NULL, // default security attributes
// OPEN_EXISTING, // opens existing pipe
// 0, // default attributes
// NULL); // no template file
//
// // Break if the pipe handle is valid.
// if (mh_Pipe != INVALID_HANDLE_VALUE)
// break;
//
// // Exit if an error other than ERROR_PIPE_BUSY occurs.
// dwErr = GetLastError();
// if (dwErr != ERROR_PIPE_BUSY)
// {
// if (!abSilent) {
// WCHAR szMsg[128];
// swprintf(szMsg, _T("Can't open plugin pipe! Plugin is not installed?\r\nFAR PID: %i, ErrCode: 0x%08X"), mn_FarPID, dwErr);
// MBoxA(szMsg);
// }
// return FALSE;
// }
//
// // All pipe instances are busy, so wait for 1 second.
// if (!WaitNamedPipe(ms_PipeName, 1000) )
// {
// if (!abSilent) {
// WCHAR szMsg[128];
// swprintf(szMsg, _T("Plugin pipe timeout!\r\nFAR PID: %i"), mn_FarPID);
// MBoxA(szMsg);
// }
// return FALSE;
// }
//}
//
//// The pipe connected; change to message-read mode.
//DWORD dwMode = CE_PIPE_READMODE;
//BOOL fSuccess = SetNamedPipeHandleState(
// mh_Pipe, // pipe handle
// &dwMode, // new pipe mode
// NULL, // don't set maximum bytes
// NULL); // don't set maximum time
//if (!fSuccess)
//{
// SafeCloseHandle(mh_Pipe);
// return FALSE;
//}
//
//return TRUE;
}
//Arguments:
// hConWnd - Хэндл КОНСОЛЬНОГО окна (по нему формируется имя пайпа для GUI)
// pIn - выполняемая команда
// nTimeout- таймаут подключения
//Returns:
// CESERVER_REQ. Его необходимо освободить через free(...);
//WARNING!!!
// Эта процедура не может получить с сервера более 600 байт данных!
// В заголовке hOwner в дебаге может быть отображена ошибка
CESERVER_REQ* ExecuteCmd(const wchar_t* szPipeName, CESERVER_REQ* pIn, DWORD nWaitPipe, HWND hOwner, BOOL bAsyncNoResult, DWORD nServerPID, BOOL bIgnoreAbsence /*= FALSE*/)
{
CESERVER_REQ* pOut = NULL;
HANDLE hPipe = NULL;
BYTE cbReadBuf[600]; // чтобы CESERVER_REQ_OUTPUTFILE поместился
wchar_t szErr[MAX_PATH*2]; szErr[0] = 0;
BOOL fSuccess = FALSE;
DWORD cbRead = 0, /*dwMode = 0,*/ dwErr = 0;
int nAllSize;
LPBYTE ptrData;
#ifdef _DEBUG
bool bIsAltSrvCmd;
wchar_t szDbgPrefix[64], szDbgResult[64], *pszDbgMsg = NULL;
#endif
if (!pIn || !szPipeName)
{
_ASSERTE(pIn && szPipeName);
pOut = NULL;
goto wrap;
}
#ifdef _DEBUG
_wsprintf(szDbgPrefix, SKIPLEN(countof(szDbgPrefix)) L">> ExecCmd: PID=%5u TID=%5u Cmd=%3u ", GetCurrentProcessId(), GetCurrentThreadId(), pIn->hdr.nCmd);
pszDbgMsg = lstrmerge(szDbgPrefix, szPipeName, L"\n");
if (pszDbgMsg) { DEBUGSTRCMD(pszDbgMsg); free(pszDbgMsg); }
#endif
pIn->hdr.bAsync = bAsyncNoResult;
_ASSERTE(pIn->hdr.nSrcPID && pIn->hdr.nSrcThreadId);
_ASSERTE(pIn->hdr.cbSize >= sizeof(pIn->hdr));
hPipe = ExecuteOpenPipe(szPipeName, szErr, NULL/*Сюда хорошо бы имя модуля подкрутить*/, nServerPID, nWaitPipe, FALSE, NULL, bIgnoreAbsence);
if (hPipe == NULL || hPipe == INVALID_HANDLE_VALUE)
{
#ifdef _DEBUG
dwErr = GetLastError();
// в заголовке "чисто" запущенного фара появляются отладочные(?) сообщения
// по идее - не должны, т.к. все должно быть через мэппинг
// *** _ASSERTEX(hPipe != NULL && hPipe != INVALID_HANDLE_VALUE); - no need in assert, it was already shown
#ifdef CONEMU_MINIMAL
SetConsoleTitle(szErr);
#else
if (hOwner)
{
if (hOwner == myGetConsoleWindow())
SetConsoleTitle(szErr);
else
SetWindowText(hOwner, szErr);
}
#endif
#endif
pOut = NULL;
goto wrap;
}
#ifdef _DEBUG
bIsAltSrvCmd = (pIn->hdr.nCmd==CECMD_ALTBUFFER || pIn->hdr.nCmd==CECMD_ALTBUFFERSTATE || pIn->hdr.nCmd==CECMD_SETCONSCRBUF || pIn->hdr.nCmd == CECMD_LOCKSTATION || pIn->hdr.nCmd == CECMD_UNLOCKSTATION);
_ASSERTE(pIn->hdr.nSrcThreadId==GetCurrentThreadId() || (bIsAltSrvCmd && pIn->hdr.nSrcPID!=GetCurrentProcessId()));
#endif
if (bAsyncNoResult)
{
// Если нас не интересует возврат и нужно сразу вернуться
fSuccess = WriteFile(hPipe, pIn, pIn->hdr.cbSize, &cbRead, NULL);
#ifdef _DEBUG
dwErr = GetLastError();
_ASSERTE(fSuccess && (cbRead == pIn->hdr.cbSize));
#endif
// -- Do not close hPipe, otherwise the reader may fail with that packet
// -- with error ‘pipe was closed before end’.
// -- Handle leak, yeah, however this is rarely used op.
// -- Must be refactored, but not so critical...
// -- CloseHandle(hPipe);
pOut = NULL;
goto wrap;
}
else
{
WARNING("При Overlapped часто виснет в этом месте.");
// Send a message to the pipe server and read the response.
fSuccess = TransactNamedPipe(
hPipe, // pipe handle
(LPVOID)pIn, // message to server
pIn->hdr.cbSize, // message length
cbReadBuf, // buffer to receive reply
sizeof(cbReadBuf), // size of read buffer
&cbRead, // bytes read
NULL); // not overlapped
dwErr = GetLastError();
//CloseHandle(hPipe);
if (!fSuccess && (dwErr != ERROR_MORE_DATA))
{
//_ASSERTE(fSuccess || (dwErr == ERROR_MORE_DATA));
CloseHandle(hPipe);
pOut = NULL;
goto wrap;
}
}
if (cbRead < sizeof(CESERVER_REQ_HDR))
{
CloseHandle(hPipe);
pOut = NULL;
goto wrap;
}
pOut = (CESERVER_REQ*)cbReadBuf; // temporary
if (pOut->hdr.cbSize < cbRead)
{
CloseHandle(hPipe);
if (pOut->hdr.cbSize)
{
_ASSERTE(pOut->hdr.cbSize == 0 || pOut->hdr.cbSize >= cbRead);
DEBUGSTR(L"!!! Wrong nSize received from GUI server !!!\n");
}
pOut = NULL;
goto wrap;
}
if (pOut->hdr.nVersion != CESERVER_REQ_VER)
{
CloseHandle(hPipe);
DEBUGSTR(L"!!! Wrong nVersion received from GUI server !!!\n");
pOut = NULL;
goto wrap;
}
nAllSize = pOut->hdr.cbSize;
pOut = (CESERVER_REQ*)malloc(nAllSize);
_ASSERTE(pOut);
if (!pOut)
{
CloseHandle(hPipe);
_ASSERTE(pOut == NULL);
goto wrap;
}
memmove(pOut, cbReadBuf, cbRead);
ptrData = ((LPBYTE)pOut)+cbRead;
nAllSize -= cbRead;
while (nAllSize>0)
{
// Break if TransactNamedPipe or ReadFile is successful
if (fSuccess)
break;
// Read from the pipe if there is more data in the message.
fSuccess = ReadFile(
hPipe, // pipe handle
ptrData, // buffer to receive reply
nAllSize, // size of buffer
&cbRead, // number of bytes read
NULL); // not overlapped
// Exit if an error other than ERROR_MORE_DATA occurs.
if (!fSuccess && ((dwErr = GetLastError()) != ERROR_MORE_DATA))
break;
ptrData += cbRead;
nAllSize -= cbRead;
}
CloseHandle(hPipe);
if (pOut && (pOut->hdr.nCmd != pIn->hdr.nCmd))
{
_ASSERTE(pOut->hdr.nCmd == pIn->hdr.nCmd);
if (pOut->hdr.nCmd == 0)
{
ExecuteFreeResult(pOut);
pOut = NULL;
}
}
wrap:
#ifdef _DEBUG
if (pOut)
_wsprintf(szDbgResult, SKIPLEN(countof(szDbgResult)) L"- Data=%5u Err=%u\n", pOut->DataSize(), dwErr);
else
lstrcpyn(szDbgResult, L"[NULL]\n", countof(szDbgResult));
pszDbgMsg = lstrmerge(szDbgPrefix, szDbgResult);
if (pszDbgMsg) { DEBUGSTRCMD(pszDbgMsg); free(pszDbgMsg); }
#endif
return pOut;
}
