136 lines
2.9 KiB
C++
136 lines
2.9 KiB
C++
/*
|
|
* synergy -- mouse and keyboard sharing utility
|
|
* Copyright (C) 2012 Nick Bolton
|
|
*
|
|
* This package is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* found in the file COPYING that should have accompanied this file.
|
|
*
|
|
* This package is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "CIpcClientProxy.h"
|
|
#include "IStream.h"
|
|
#include "TMethodEventJob.h"
|
|
#include "Ipc.h"
|
|
#include "CLog.h"
|
|
#include "CIpcMessage.h"
|
|
|
|
CEvent::Type CIpcClientProxy::s_messageReceivedEvent = CEvent::kUnknown;
|
|
|
|
CIpcClientProxy::CIpcClientProxy(IStream& stream) :
|
|
m_stream(stream),
|
|
m_enableLog(false)
|
|
{
|
|
EVENTQUEUE->adoptHandler(m_stream.getInputReadyEvent(),
|
|
stream.getEventTarget(),
|
|
new TMethodEventJob<CIpcClientProxy>(
|
|
this, &CIpcClientProxy::handleData, nullptr));
|
|
}
|
|
|
|
CIpcClientProxy::~CIpcClientProxy()
|
|
{
|
|
EVENTQUEUE->removeHandler(m_stream.getInputReadyEvent(),
|
|
m_stream.getEventTarget());
|
|
}
|
|
|
|
void
|
|
CIpcClientProxy::handleData(const CEvent&, void*)
|
|
{
|
|
UInt8 code[1];
|
|
UInt32 n = m_stream.read(code, 1);
|
|
while (n != 0) {
|
|
UInt8 type = code[0];
|
|
|
|
CIpcMessage* m = new CIpcMessage();
|
|
m->m_type = type;
|
|
|
|
if (m_enableLog) {
|
|
LOG((CLOG_DEBUG "ipc client proxy read: %d", code[0]));
|
|
}
|
|
|
|
switch (type) {
|
|
case kIpcCommand:
|
|
m->m_data = parseCommand();
|
|
break;
|
|
|
|
default:
|
|
delete m;
|
|
disconnect();
|
|
return;
|
|
}
|
|
|
|
// event deletes data.
|
|
EVENTQUEUE->addEvent(CEvent(getMessageReceivedEvent(), this, m));
|
|
|
|
n = m_stream.read(code, 1);
|
|
}
|
|
}
|
|
|
|
void
|
|
CIpcClientProxy::send(const CIpcMessage& message)
|
|
{
|
|
if (m_enableLog) {
|
|
LOG((CLOG_DEBUG "ipc client proxy write: %d", message.m_type));
|
|
}
|
|
|
|
UInt8 code[1];
|
|
code[0] = message.m_type;
|
|
m_stream.write(code, 1);
|
|
|
|
switch (message.m_type) {
|
|
case kIpcLogLine: {
|
|
CString* s = (CString*)message.m_data;
|
|
const char* data = s->c_str();
|
|
int len = strlen(data);
|
|
|
|
UInt8 lenBuf[1];
|
|
lenBuf[0] = len;
|
|
m_stream.write(lenBuf, 1);
|
|
|
|
m_stream.write(data, len);
|
|
}
|
|
break;
|
|
|
|
default:
|
|
if (m_enableLog) {
|
|
LOG((CLOG_ERR "message not supported: %d", message.m_type));
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
void*
|
|
CIpcClientProxy::parseCommand()
|
|
{
|
|
UInt8 len[1];
|
|
m_stream.read(len, 1);
|
|
|
|
UInt8* buffer = new UInt8[len[0]];
|
|
m_stream.read(buffer, len[0]);
|
|
|
|
return new CString((const char*)buffer, len[0]);
|
|
}
|
|
|
|
void
|
|
CIpcClientProxy::disconnect()
|
|
{
|
|
if (m_enableLog) {
|
|
LOG((CLOG_NOTE "disconnect, closing stream"));
|
|
}
|
|
m_stream.close();
|
|
}
|
|
|
|
CEvent::Type
|
|
CIpcClientProxy::getMessageReceivedEvent()
|
|
{
|
|
return EVENTQUEUE->registerTypeOnce(
|
|
s_messageReceivedEvent, "CIpcClientProxy::messageReceived");
|
|
}
|