Textractor_test/vnr/texthook/host/pipe.cc

// pipe.cc
// 8/24/2013 jichi
// Branch IHF/pipe.cpp, rev 93
// 8/24/2013 TODO: Clean up this file

#include "host_p.h"
#include "hookman.h"
#include "vnrhook/include/defs.h"
#include "vnrhook/include/const.h"
#include "ithsys/ithsys.h"
#include <stdio.h>
#include "growl.h"
#include <atlbase.h>
//#include "CommandQueue.h"
//#include <QtCore/QDebug>

#define DEBUG "vnrhost/pipe.cc"
#include "sakurakit/skdebug.h"

//DWORD WINAPI UpdateWindows(LPVOID lpThreadParameter);

namespace
{ // unnamed

	// jichi 10/27/2013
	// Check if text has leading space
	enum { FILTER_LIMIT = 0x20 }; // The same as the orignal ITH filter. So, I don't have to check \u3000
	//enum { FILTER_LIMIT = 0x19 };
	inline bool HasLeadingSpace(const BYTE *text, int len)
	{
		return len == 1 ? *text <= FILTER_LIMIT : // 1 byte
			*(WORD*)text <= FILTER_LIMIT; // 2 bytes
	}

	// jichi 9/28/2013: Skip leading garbage
	// Note:
	// - Modifying limit will break manual translation. The orignal one is 0x20
	// - Eliminating 0x20 will break English-translated games
	const BYTE* Filter(const BYTE *str, int len)
	{
#ifdef ITH_DISABLE_FILTER // jichi 9/28/2013: only for debugging purpose
		return str;
#endif
		while (true)
		{
			if (len >= 2)
			{
				if (*(WORD*)str <= FILTER_LIMIT)
				{ // jichi 10/27/2013: two bytes
					str += 2;
					len -= 2;
				}
				else
				{
					break;
				}
			}
			else if (*str <= FILTER_LIMIT)
			{ // jichi 10/27/2013: 1 byte
				str++;
				len--;
			}
			else
			{
				break;
			}
		}
		return str;
	}

} // unnamed namespace

CRITICAL_SECTION detachCs; // jichi 9/27/2013: also used in main
//HANDLE hDetachEvent;
extern HANDLE pipeExistsEvent;

void CreateNewPipe()
{
	HANDLE hookPipe, hostPipe, TextReceivingThread;

	hookPipe = CreateNamedPipeW(ITH_TEXT_PIPE, PIPE_ACCESS_INBOUND, PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE, PIPE_UNLIMITED_INSTANCES, 0x1000, 0x1000, MAXDWORD, NULL);
	hostPipe = CreateNamedPipeW(ITH_COMMAND_PIPE, PIPE_ACCESS_OUTBOUND, 0, PIPE_UNLIMITED_INSTANCES, 0x1000, 0x1000, MAXDWORD, NULL);
	TextReceivingThread = CreateThread(nullptr, 0, TextReceiver, hookPipe, 0, nullptr);
	man->RegisterPipe(hookPipe, hostPipe, TextReceivingThread);
}

DWORD WINAPI TextReceiver(LPVOID lpThreadParameter)
{
	HANDLE hookPipe = (HANDLE)lpThreadParameter;
	ConnectNamedPipe(hookPipe, nullptr);

	enum { PIPE_BUFFER_SIZE = 0x1000 };
	BYTE* buffer = new BYTE[PIPE_BUFFER_SIZE];
	DWORD bytesRead, processId;

	// Artikash 5/20/2018: Shouldn't Windows automatically close the handles when the host process stops running?
	//if (!::running) {
	//  NtClose(hookPipe);
	//  return 0;
	//}

	ReadFile(hookPipe, &processId, sizeof(processId), &bytesRead, nullptr);
	man->RegisterProcess(processId);

	// jichi 9/27/2013: why recursion?
	// Artikash 5/20/2018: To create a new pipe for another process
	CreateNewPipe();

	while (::running)
	{
		if (!ReadFile(hookPipe, buffer, PIPE_BUFFER_SIZE, &bytesRead, nullptr))
		{
			break;
		}

		enum { DATA_OFFSET = 0xc }; // jichi 10/27/2013: Seem to be the data offset in the pipe

		if (bytesRead < DATA_OFFSET)
		{
			break;
		}

		union { DWORD retn; DWORD commandType; };
		DWORD hook = *(DWORD *)buffer;
		retn = *(DWORD *)(buffer + 4);
		DWORD split = *(DWORD *)(buffer + 8);

		buffer[bytesRead] = 0;
		buffer[bytesRead + 1] = 0;

		if (hook == HOST_NOTIFICATION)
		{
			switch (commandType) 
			{
			case HOST_NOTIFICATION_NEWHOOK:
			{
				static long lock;
				while (InterlockedExchange(&lock, 1) == 1);
				man->ProcessNewHook()(processId);
				lock = 0;
			}
			break;
			case HOST_NOTIFICATION_TEXT:
				USES_CONVERSION;
				man->AddConsoleOutput(A2W((LPCSTR)(buffer + 8)));
				break;
			}
		}
		else
		{
			// jichi 9/28/2013: Debug raw data
			//ITH_DEBUG_DWORD9(RecvLen - 0xc,
			//    buffer[0xc], buffer[0xd], buffer[0xe], buffer[0xf],
			//    buffer[0x10], buffer[0x11], buffer[0x12], buffer[0x13]);

			const BYTE *data = buffer + DATA_OFFSET; // th
			int dataLength = bytesRead - DATA_OFFSET;
			bool space = ::HasLeadingSpace(data, dataLength);
			if (space)
			{
				const BYTE *it = ::Filter(data, dataLength);
				dataLength -= it - data;
				data = it;
			}
			man->DispatchText(processId, data, hook, retn, split, dataLength, space);
		}
	}

	EnterCriticalSection(&detachCs);

	DisconnectNamedPipe(hookPipe);
	DisconnectNamedPipe(man->GetCmdHandleByPID(processId));
	man->UnRegisterProcess(processId);

	LeaveCriticalSection(&detachCs);
	delete[] buffer;

	return 0;
}

// EOF