From 00d4f92b717fbcbed6f9eee361975d6ee5380d59 Mon Sep 17 00:00:00 2001 From: Michele Calgaro Date: Sun, 6 Dec 2020 19:28:06 +0900 Subject: Renaming of files in preparation for code style tools. Signed-off-by: Michele Calgaro --- mcop/connection.cpp | 196 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 196 insertions(+) create mode 100644 mcop/connection.cpp (limited to 'mcop/connection.cpp') diff --git a/mcop/connection.cpp b/mcop/connection.cpp new file mode 100644 index 0000000..397ca86 --- /dev/null +++ b/mcop/connection.cpp @@ -0,0 +1,196 @@ + /* + + Copyright (C) 2000 Stefan Westerfeld + stefan@space.twc.de + + This library is free software; you can redistribute it and/or + modify it under the terms of the GNU Library General Public + License as published by the Free Software Foundation; either + version 2 of the License, or (at your option) any later version. + + This library 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 + Library General Public License for more details. + + You should have received a copy of the GNU Library General Public License + along with this library; see the file COPYING.LIB. If not, write to + the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, + Boston, MA 02110-1301, USA. + + */ + +#include "connection.h" +#include "dispatcher.h" +#include "debug.h" +#include +#include // for Solaris +#include +#include + +using namespace Arts; +using namespace std; + +namespace Arts { +class ConnectionPrivate { +public: + struct Data { + Data() : data(0), len(0) { } + Data(unsigned char *data, long len) : data(data), len(len) { } + Data(const Data& d) : data(d.data), len(d.len) { } + unsigned char *data; + long len; + }; + + queue incoming; + map hints; +}; +} + +Connection::Connection() :d(new ConnectionPrivate), _refCnt(1) +{ + _connState = unknown; +} + +Connection::~Connection() +{ + assert(d->incoming.empty()); + assert(_refCnt == 0); + + delete d; +} + +void Connection::_copy() +{ + assert(_refCnt > 0); + _refCnt++; +} + +void Connection::_release() +{ + assert(_refCnt > 0); + _refCnt--; + if(_refCnt == 0) + delete this; +} + +void Connection::initReceive() +{ + rcbuf = 0; + receiveHeader = true; + remaining = 12; +} + +void Connection::receive(unsigned char *newdata, long newlen) +{ + /* + * protect against being freed while receive is running, as there are a + * few points where reentrant event loops may happen (Dispatcher::handle) + */ + _copy(); + + d->incoming.push(ConnectionPrivate::Data(newdata,newlen)); + + do + { + ConnectionPrivate::Data &data = d->incoming.front(); + + // get a buffer for the incoming message + if(!rcbuf) rcbuf = new Buffer; + + // put a suitable amount of input data into the receive buffer + long len = min(remaining, data.len); + + remaining -= len; + rcbuf->write(data.data,len); + + data.len -= len; + data.data += len; + + if(data.len == 0) + d->incoming.pop(); + + // look if it was enough to do something useful with it + if(remaining == 0) + { + if(receiveHeader) + { + long mcopMagic; + + mcopMagic = rcbuf->readLong(); + remaining = rcbuf->readLong() - 12; + messageType = rcbuf->readLong(); + + if(_connState != Connection::established + && (remaining >= 4096 || remaining < 0)) + { + /* + * don't accept large amounts of data on unauthenticated + * connections + */ + remaining = 0; + } + + if(mcopMagic == MCOP_MAGIC) + { + // do we need to receive more data (message body?) + if(remaining) + { + receiveHeader = false; + } + else + { + Buffer *received = rcbuf; + initReceive(); + Dispatcher::the()->handle(this,received,messageType); + } + } + else + { + initReceive(); + Dispatcher::the()->handleCorrupt(this); + } + } + else + { + Buffer *received = rcbuf; + + /* + * it's important to prepare to receive new messages *before* + * calling Dispatcher::the()->handle(...), as handle may + * get into an I/O situation (e.g. when doing an invocation + * itself), and we may receive more messages while handle is + * running + */ + initReceive(); + + // rcbuf is consumed by the dispatcher + Dispatcher::the()->handle(this,received,messageType); + } + } + } while(!d->incoming.empty()); + + _release(); +} + +void Connection::setHints(const vector& hints) +{ + vector::const_iterator i; + + for(i = hints.begin(); i != hints.end(); i++) + { + string key; + vector values; + + if(MCOPUtils::tokenize(*i, key, values)) + { + if(values.size() == 1) + d->hints[key] = values[0]; + } + } +} + +string Connection::findHint(const string& hint) +{ + return d->hints[hint]; +} -- cgit v1.2.1