| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007 |
- /**
- * PANDA 3D SOFTWARE
- * Copyright (c) Carnegie Mellon University. All rights reserved.
- *
- * All use of this software is subject to the terms of the revised BSD
- * license. You should have received a copy of this license along
- * with this source code in a file named "LICENSE."
- *
- * @file cConnectionRepository.cxx
- * @author drose
- * @date 2004-05-17
- */
- #include "cConnectionRepository.h"
- #include "dcmsgtypes.h"
- #include "dcClass.h"
- #include "dcPacker.h"
- #include "config_distributed.h"
- #include "config_downloader.h"
- #include "httpChannel.h"
- #include "urlSpec.h"
- #include "datagramIterator.h"
- #include "throw_event.h"
- #include "pStatTimer.h"
- #ifdef HAVE_PYTHON
- #include "py_panda.h"
- #endif
- const string CConnectionRepository::_overflow_event_name = "CRDatagramOverflow";
- #ifndef CPPPARSER
- PStatCollector CConnectionRepository::_update_pcollector("App:Show code:readerPollTask:Update");
- #endif // CPPPARSER
- /**
- *
- */
- CConnectionRepository::
- CConnectionRepository(bool has_owner_view, bool threaded_net) :
- _lock("CConnectionRepository::_lock"),
- #ifdef HAVE_PYTHON
- _python_repository(NULL),
- #endif
- #ifdef HAVE_OPENSSL
- _http_conn(NULL),
- #endif
- #ifdef HAVE_NET
- _cw(&_qcm, threaded_net ? 1 : 0),
- _qcr(&_qcm, threaded_net ? 1 : 0),
- #endif
- #ifdef WANT_NATIVE_NET
- _bdc(4096000,4096000,1400),
- _native(false),
- #endif
- _client_datagram(true),
- _handle_datagrams_internally(handle_datagrams_internally),
- _simulated_disconnect(false),
- _verbose(distributed_cat.is_spam()),
- _time_warning(0.0),
- // _msg_channels(),
- _msg_sender(0),
- _msg_type(0),
- _has_owner_view(has_owner_view),
- _handle_c_updates(true),
- _want_message_bundling(true),
- _bundling_msgs(0),
- _in_quiet_zone(0)
- {
- #if defined(HAVE_NET) && defined(SIMULATE_NETWORK_DELAY)
- if (min_lag != 0.0 || max_lag != 0.0) {
- _qcr.start_delay(min_lag, max_lag);
- }
- #endif
- _tcp_header_size = tcp_header_size;
- }
- /**
- *
- */
- CConnectionRepository::
- ~CConnectionRepository() {
- disconnect();
- }
- /**
- * Sets the header size of TCP packets. At the present, legal values for this
- * are 0, 2, or 4; this specifies the number of bytes to use encode the
- * datagram length at the start of each TCP datagram. Sender and receiver
- * must independently agree on this.
- */
- void CConnectionRepository::
- set_tcp_header_size(int tcp_header_size) {
- _tcp_header_size = tcp_header_size;
- #ifdef HAVE_OPENSSL
- if (_http_conn != (SocketStream *)NULL) {
- _http_conn->set_tcp_header_size(tcp_header_size);
- }
- #endif
- #ifdef HAVE_NET
- _cw.set_tcp_header_size(tcp_header_size);
- _qcr.set_tcp_header_size(tcp_header_size);
- #endif
- }
- #ifdef HAVE_OPENSSL
- /**
- * Once a connection has been established via the HTTP interface, gets the
- * connection and uses it. The supplied HTTPChannel object must have a
- * connection available via get_connection().
- */
- void CConnectionRepository::
- set_connection_http(HTTPChannel *channel) {
- ReMutexHolder holder(_lock);
- disconnect();
- nassertv(channel->is_connection_ready());
- _http_conn = channel->get_connection();
- _http_conn->set_tcp_header_size(_tcp_header_size);
- #ifdef SIMULATE_NETWORK_DELAY
- if (min_lag != 0.0 || max_lag != 0.0) {
- _http_conn->start_delay(min_lag, max_lag);
- }
- #endif
- }
- #endif // HAVE_OPENSSL
- #ifdef HAVE_OPENSSL
- /**
- * Returns the SocketStream that internally represents the already-established
- * HTTP connection. Returns NULL if there is no current HTTP connection.
- */
- SocketStream *CConnectionRepository::
- get_stream() {
- ReMutexHolder holder(_lock);
- return _http_conn;
- }
- #endif // HAVE_OPENSSL
- #ifdef HAVE_NET
- /**
- * Uses Panda's "net" library to try to connect to the server and port named
- * in the indicated URL. Returns true if successful, false otherwise.
- */
- bool CConnectionRepository::
- try_connect_net(const URLSpec &url) {
- ReMutexHolder holder(_lock);
- disconnect();
- _net_conn =
- _qcm.open_TCP_client_connection(url.get_server(), url.get_port(),
- game_server_timeout_ms);
- if (_net_conn != (Connection *)NULL) {
- _net_conn->set_no_delay(true);
- _qcr.add_connection(_net_conn);
- return true;
- }
- return false;
- }
- #endif // HAVE_NET
- #ifdef WANT_NATIVE_NET
- /**
- * Connects to the server using Panda's low-level and fast "native net"
- * library.
- */
- bool CConnectionRepository::
- connect_native(const URLSpec &url) {
- ReMutexHolder holder(_lock);
- _native=true;
- Socket_Address addr;
- addr.set_host(url.get_server(),url.get_port());
- _bdc.ClearAddresses();
- _bdc.AddAddress(addr);
- return _bdc.DoConnect();
- }
- #endif //WANT NATIVE NET
- #ifdef SIMULATE_NETWORK_DELAY
- /**
- * Enables a simulated network latency. All datagrams received from this
- * point on will be held for a random interval of least min_delay seconds, and
- * no more than max_delay seconds, before being visible. It is as if
- * datagrams suddenly took much longer to arrive.
- *
- * This should *only* be called if the underlying socket is non-blocking. If
- * you call this on a blocking socket, it will force all datagrams to be held
- * up until the socket closes.
- *
- * This has no effect if the connection method is via the "native net"
- * library.
- */
- void CConnectionRepository::
- start_delay(double min_delay, double max_delay) {
- ReMutexHolder holder(_lock);
- if (min_delay != 0.0 || max_delay != 0.0) {
- #ifdef HAVE_NET
- _qcr.start_delay(min_delay, max_delay);
- #endif // HAVE_NET
- #ifdef HAVE_OPENSSL
- if (_http_conn != (SocketStream *)NULL) {
- _http_conn->start_delay(min_delay, max_delay);
- }
- #endif // HAVE_OPENSSL
- } else {
- stop_delay();
- }
- }
- #endif // SIMULATE_NETWORK_DELAY
- #ifdef SIMULATE_NETWORK_DELAY
- /**
- * Disables the simulated network latency started by a previous call to
- * start_delay(). Datagrams will once again be visible as soon as they are
- * received.
- */
- void CConnectionRepository::
- stop_delay() {
- ReMutexHolder holder(_lock);
- #ifdef HAVE_NET
- _qcr.stop_delay();
- #endif // HAVE_NET
- #ifdef HAVE_OPENSSL
- if (_http_conn != (SocketStream *)NULL) {
- _http_conn->stop_delay();
- }
- #endif // HAVE_OPENSSL
- }
- #endif // SIMULATE_NETWORK_DELAY
- /**
- * Returns true if a new datagram is available, false otherwise. If the
- * return value is true, the new datagram may be retrieved via get_datagram(),
- * or preferably, with get_datagram_iterator() and get_msg_type().
- */
- bool CConnectionRepository::
- check_datagram() {
- ReMutexHolder holder(_lock);
- if (_simulated_disconnect) {
- return false;
- }
- #ifdef WANT_NATIVE_NET
- if(_native)
- _bdc.Flush();
- #endif //WANT_NATIVE_NET
- while (do_check_datagram()) {
- if (get_verbose()) {
- describe_message(nout, "RECV", _dg);
- }
- // Start breaking apart the datagram.
- _di = DatagramIterator(_dg);
- if (!_client_datagram) {
- unsigned char wc_cnt;
- wc_cnt = _di.get_uint8();
- _msg_channels.clear();
- for (unsigned char lp1 = 0; lp1 < wc_cnt; lp1++) {
- CHANNEL_TYPE schan = _di.get_uint64();
- _msg_channels.push_back(schan);
- }
- _msg_sender = _di.get_uint64();
- #ifdef HAVE_PYTHON
- // For now, we need to stuff this field onto the Python structure, to
- // support legacy code that expects to find it there.
- if (_python_repository != (PyObject *)NULL) {
- #if defined(HAVE_THREADS) && !defined(SIMPLE_THREADS)
- PyGILState_STATE gstate;
- gstate = PyGILState_Ensure();
- #endif
- PyObject *value = PyLong_FromUnsignedLongLong(_msg_sender);
- PyObject_SetAttrString(_python_repository, "msgSender", value);
- Py_DECREF(value);
- #if defined(HAVE_THREADS) && !defined(SIMPLE_THREADS)
- PyGILState_Release(gstate);
- #endif
- }
- #endif // HAVE_PYTHON
- }
- _msg_type = _di.get_uint16();
- // Is this a message that we can process directly?
- if (!_handle_datagrams_internally) {
- return true;
- }
- switch (_msg_type) {
- #ifdef HAVE_PYTHON
- case CLIENT_OBJECT_SET_FIELD:
- case STATESERVER_OBJECT_SET_FIELD:
- if (_handle_c_updates) {
- if (_has_owner_view) {
- if (!handle_update_field_owner()) {
- return false;
- }
- } else {
- if (!handle_update_field()) {
- return false;
- }
- }
- } else {
- // Let the caller (Python) deal with this update.
- return true;
- }
- break;
- #endif // HAVE_PYTHON
- default:
- // Some unknown message; let the caller deal with it.
- return true;
- }
- }
- // No datagrams available.
- return false;
- }
- /**
- * Returns true if the connection to the gameserver is established and still
- * good, false if we are not connected. A false value means either (a) we
- * never successfully connected, (b) we explicitly called disconnect(), or (c)
- * we were connected, but the connection was spontaneously lost.
- */
- bool CConnectionRepository::
- is_connected() {
- ReMutexHolder holder(_lock);
- #ifdef WANT_NATIVE_NET
- if(_native)
- return (_bdc.IsConnected());
- #endif
- #ifdef HAVE_NET
- if (_net_conn) {
- if (_qcm.reset_connection_available()) {
- PT(Connection) reset_connection;
- if (_qcm.get_reset_connection(reset_connection)) {
- _qcm.close_connection(reset_connection);
- if (reset_connection == _net_conn) {
- // Whoops, lost our connection.
- _net_conn = NULL;
- return false;
- }
- }
- }
- return true;
- }
- #endif // HAVE_NET
- #ifdef HAVE_OPENSSL
- if (_http_conn) {
- if (!_http_conn->is_closed()) {
- return true;
- }
- // Connection lost.
- delete _http_conn;
- _http_conn = NULL;
- }
- #endif // HAVE_OPENSSL
- return false;
- }
- /**
- * Queues the indicated datagram for sending to the server. It may not get
- * sent immediately if collect_tcp is in effect; call flush() to guarantee it
- * is sent now.
- */
- bool CConnectionRepository::
- send_datagram(const Datagram &dg) {
- ReMutexHolder holder(_lock);
- if (_simulated_disconnect) {
- distributed_cat.warning()
- << "Unable to send datagram during simulated disconnect.\n";
- return false;
- }
- if (get_verbose()) {
- describe_message(nout, "SEND", dg);
- }
- if (is_bundling_messages() && get_want_message_bundling()) {
- bundle_msg(dg);
- return true;
- }
- #ifdef WANT_NATIVE_NET
- if (_native) {
- bool result = _bdc.SendMessage(dg);
- if (!result && _bdc.IsConnected()) {
- #ifdef HAVE_PYTHON
- ostringstream s;
- #if PY_VERSION_HEX >= 0x03030000
- PyObject *exc_type = PyExc_ConnectionError;
- #else
- PyObject *exc_type = PyExc_OSError;
- #endif
- s << endl << "Error sending message: " << endl;
- dg.dump_hex(s);
- s << "Message data: " << dg.get_data() << endl;
- string message = s.str();
- PyErr_SetString(exc_type, message.c_str());
- #endif
- }
- return result;
- }
- #endif
- #ifdef HAVE_NET
- if (_net_conn) {
- _cw.send(dg, _net_conn);
- return true;
- }
- #endif // HAVE_NET
- #ifdef HAVE_OPENSSL
- if (_http_conn) {
- if (!_http_conn->send_datagram(dg)) {
- distributed_cat.warning()
- << "Could not send datagram.\n";
- return false;
- }
- return true;
- }
- #endif // HAVE_OPENSSL
- distributed_cat.warning()
- << "Unable to send datagram after connection is closed.\n";
- return false;
- }
- /**
- * Send a set of messages to the state server that will be processed
- * atomically. For instance, you can do a combined setLocation/setPos and
- * prevent race conditions where clients briefly get the setLocation but not
- * the setPos, because the state server hasn't processed the setPos yet
- */
- void CConnectionRepository::
- start_message_bundle() {
- ReMutexHolder holder(_lock);
- // store up network messages until sendMessageBundle is called all updates
- // in between must be sent from the same doId (updates must all affect the
- // same DistributedObject) it is an error to call this again before calling
- // sendMessageBundle
- if (get_verbose()) {
- nout << "CR::SEND:BUNDLE_START(" << _bundling_msgs << ")" << endl;
- }
- if (_bundling_msgs == 0) {
- _bundle_msgs.clear();
- }
- ++_bundling_msgs;
- }
- /**
- * Send network messages queued up since startMessageBundle was called.
- */
- void CConnectionRepository::
- send_message_bundle(unsigned int channel, unsigned int sender_channel) {
- ReMutexHolder holder(_lock);
- nassertv(_bundling_msgs);
- --_bundling_msgs;
- if (get_verbose()) {
- nout << "CR::SEND:BUNDLE_FINISH(" << _bundling_msgs << ")" << endl;
- }
- // if _bundling_msgs ref count is zero, send the bundle out
- if (_bundling_msgs == 0 && get_want_message_bundling()) {
- Datagram dg;
- // add server header (see PyDatagram.addServerHeader)
- dg.add_int8(1);
- dg.add_uint64(channel);
- dg.add_uint64(sender_channel);
- //dg.add_uint16(STATESERVER_BOUNCE_MESSAGE);
- // add each bundled message
- BundledMsgVector::const_iterator bmi;
- for (bmi = _bundle_msgs.begin(); bmi != _bundle_msgs.end(); bmi++) {
- dg.add_string(*bmi);
- }
- send_datagram(dg);
- }
- }
- /**
- * throw out any msgs that have been queued up for message bundles
- */
- void CConnectionRepository::
- abandon_message_bundles() {
- ReMutexHolder holder(_lock);
- nassertv(is_bundling_messages());
- _bundling_msgs = 0;
- _bundle_msgs.clear();
- }
- /**
- *
- */
- void CConnectionRepository::
- bundle_msg(const Datagram &dg) {
- ReMutexHolder holder(_lock);
- nassertv(is_bundling_messages());
- _bundle_msgs.push_back(dg.get_message());
- }
- /**
- * Sends the most recently queued data if enough time has elapsed. This only
- * has meaning if set_collect_tcp() has been set to true.
- */
- bool CConnectionRepository::
- consider_flush() {
- ReMutexHolder holder(_lock);
- if (_simulated_disconnect) {
- return false;
- }
- #ifdef WANT_NATIVE_NET
- if(_native)
- return true; //Maybe we should just flush here for now?
- #endif
- #ifdef HAVE_NET
- if (_net_conn) {
- return _net_conn->consider_flush();
- }
- #endif // HAVE_NET
- #ifdef HAVE_OPENSSL
- if (_http_conn) {
- return _http_conn->consider_flush();
- }
- #endif // HAVE_OPENSSL
- return false;
- }
- /**
- * Sends the most recently queued data now. This only has meaning if
- * set_collect_tcp() has been set to true.
- */
- bool CConnectionRepository::
- flush() {
- ReMutexHolder holder(_lock);
- if (_simulated_disconnect) {
- return false;
- }
- #ifdef WANT_NATIVE_NET
- if(_native)
- return _bdc.Flush();
- #endif
- #ifdef HAVE_NET
- if (_net_conn) {
- return _net_conn->flush();
- }
- #endif // HAVE_NET
- #ifdef HAVE_OPENSSL
- if (_http_conn) {
- return _http_conn->flush();
- }
- #endif // HAVE_OPENSSL
- return false;
- }
- /**
- * Closes the connection to the server.
- */
- void CConnectionRepository::
- disconnect() {
- ReMutexHolder holder(_lock);
- #ifdef WANT_NATIVE_NET
- if(_native) {
- _bdc.Reset();
- _bdc.ClearAddresses();
- }
- #endif
- #ifdef HAVE_NET
- if (_net_conn) {
- _qcm.close_connection(_net_conn);
- _net_conn = NULL;
- }
- #endif // HAVE_NET
- #ifdef HAVE_OPENSSL
- if (_http_conn) {
- _http_conn->close();
- delete _http_conn;
- _http_conn = NULL;
- }
- #endif // HAVE_OPENSSL
- _simulated_disconnect = false;
- }
- /**
- * May be called at application shutdown to ensure all threads are cleaned up.
- */
- void CConnectionRepository::
- shutdown() {
- disconnect();
- #ifdef HAVE_NET
- _cw.shutdown();
- _qcr.shutdown();
- #endif // HAVE_NET
- }
- /**
- * The private implementation of check_datagram(), this gets one datagram if
- * it is available.
- */
- bool CConnectionRepository::
- do_check_datagram() {
- #ifdef WANT_NATIVE_NET
- if(_native) {
- return _bdc.GetMessage(_dg);
- }
- #endif
- #ifdef HAVE_NET
- if (_net_conn) {
- _net_conn->consider_flush();
- if (_qcr.get_overflow_flag()) {
- throw_event(get_overflow_event_name());
- _qcr.reset_overflow_flag();
- }
- return (_qcr.data_available() && _qcr.get_data(_dg));
- }
- #endif // HAVE_NET
- #ifdef HAVE_OPENSSL
- if (_http_conn) {
- _http_conn->consider_flush();
- return _http_conn->receive_datagram(_dg);
- }
- #endif // HAVE_OPENSSL
- return false;
- }
- /**
- * Directly handles an update message on a field. Python never touches the
- * datagram; it just gets its distributed method called with the appropriate
- * parameters. Returns true if everything is ok, false if there was an error
- * processing the field's update method.
- */
- bool CConnectionRepository::
- handle_update_field() {
- #ifdef HAVE_PYTHON
- #if defined(HAVE_THREADS) && !defined(SIMPLE_THREADS)
- PyGILState_STATE gstate;
- gstate = PyGILState_Ensure();
- #endif
- PStatTimer timer(_update_pcollector);
- unsigned int do_id = _di.get_uint32();
- if (_python_repository != (PyObject *)NULL)
- {
- PyObject *doId2do =
- PyObject_GetAttrString(_python_repository, "doId2do");
- nassertr(doId2do != NULL, false);
- #ifdef USE_PYTHON_2_2_OR_EARLIER
- PyObject *doId = PyInt_FromLong(do_id);
- #else
- PyObject *doId = PyLong_FromUnsignedLong(do_id);
- #endif
- PyObject *distobj = PyDict_GetItem(doId2do, doId);
- Py_DECREF(doId);
- Py_DECREF(doId2do);
- if (distobj != NULL) {
- PyObject *dclass_obj = PyObject_GetAttrString(distobj, "dclass");
- nassertr(dclass_obj != NULL, false);
- PyObject *dclass_this = PyObject_GetAttrString(dclass_obj, "this");
- Py_DECREF(dclass_obj);
- nassertr(dclass_this != NULL, false);
- DCClass *dclass = (DCClass *)PyLong_AsVoidPtr(dclass_this);
- Py_DECREF(dclass_this);
- // If in quiet zone mode, throw update away unless distobj has
- // 'neverDisable' attribute set to non-zero
- if (_in_quiet_zone) {
- PyObject *neverDisable = PyObject_GetAttrString(distobj, "neverDisable");
- nassertr(neverDisable != NULL, false);
- unsigned int cNeverDisable = PyLong_AsLong(neverDisable);
- if (!cNeverDisable) {
- // in quiet zone and distobj is disable-able drop update on the
- // floor
- #if defined(HAVE_THREADS) && !defined(SIMPLE_THREADS)
- PyGILState_Release(gstate);
- #endif
- return true;
- }
- }
- // It's a good idea to ensure the reference count to distobj is raised
- // while we call the update method--otherwise, the update method might
- // get into trouble if it tried to delete the object from the doId2do
- // map.
- Py_INCREF(distobj);
- dclass->receive_update(distobj, _di);
- Py_DECREF(distobj);
- if (PyErr_Occurred()) {
- #if defined(HAVE_THREADS) && !defined(SIMPLE_THREADS)
- PyGILState_Release(gstate);
- #endif
- return false;
- }
- }
- }
- #if defined(HAVE_THREADS) && !defined(SIMPLE_THREADS)
- PyGILState_Release(gstate);
- #endif
- #endif // HAVE_PYTHON
- return true;
- }
- /**
- * Directly handles an update message on a field. Supports 'owner' views of
- * objects, separate from 'visible' view, and forwards fields to the
- * appropriate view(s) based on DC flags. Python never touches the datagram;
- * it just gets its distributed method called with the appropriate parameters.
- * Returns true if everything is ok, false if there was an error processing
- * the field's update method.
- */
- bool CConnectionRepository::
- handle_update_field_owner() {
- #ifdef HAVE_PYTHON
- #if defined(HAVE_THREADS) && !defined(SIMPLE_THREADS)
- PyGILState_STATE gstate;
- gstate = PyGILState_Ensure();
- #endif
- PStatTimer timer(_update_pcollector);
- unsigned int do_id = _di.get_uint32();
- if (_python_repository != (PyObject *)NULL) {
- PyObject *doId2do =
- PyObject_GetAttrString(_python_repository, "doId2do");
- nassertr(doId2do != NULL, false);
- PyObject *doId2ownerView =
- PyObject_GetAttrString(_python_repository, "doId2ownerView");
- nassertr(doId2ownerView != NULL, false);
- #ifdef USE_PYTHON_2_2_OR_EARLIER
- PyObject *doId = PyInt_FromLong(do_id);
- #else
- PyObject *doId = PyLong_FromUnsignedLong(do_id);
- #endif
- // pass the update to the owner view first
- PyObject *distobjOV = PyDict_GetItem(doId2ownerView, doId);
- Py_DECREF(doId2ownerView);
- if (distobjOV != NULL) {
- PyObject *dclass_obj = PyObject_GetAttrString(distobjOV, "dclass");
- nassertr(dclass_obj != NULL, false);
- PyObject *dclass_this = PyObject_GetAttrString(dclass_obj, "this");
- Py_DECREF(dclass_obj);
- nassertr(dclass_this != NULL, false);
- DCClass *dclass = (DCClass *)PyLong_AsVoidPtr(dclass_this);
- Py_DECREF(dclass_this);
- // check if we should forward this update to the owner view
- vector_uchar data = _di.get_remaining_bytes();
- DCPacker packer;
- packer.set_unpack_data((const char *)data.data(), data.size(), false);
- int field_id = packer.raw_unpack_uint16();
- DCField *field = dclass->get_field_by_index(field_id);
- if (field->is_ownrecv()) {
- // It's a good idea to ensure the reference count to distobjOV is
- // raised while we call the update method--otherwise, the update
- // method might get into trouble if it tried to delete the object from
- // the doId2do map.
- Py_INCREF(distobjOV);
- // make a copy of the datagram iterator so that we can use the main
- // iterator for the non-owner update
- DatagramIterator _odi(_di);
- dclass->receive_update(distobjOV, _odi);
- Py_DECREF(distobjOV);
- if (PyErr_Occurred()) {
- #if defined(HAVE_THREADS) && !defined(SIMPLE_THREADS)
- PyGILState_Release(gstate);
- #endif
- return false;
- }
- }
- }
- // now pass the update to the visible view
- PyObject *distobj = PyDict_GetItem(doId2do, doId);
- Py_DECREF(doId);
- Py_DECREF(doId2do);
- if (distobj != NULL) {
- PyObject *dclass_obj = PyObject_GetAttrString(distobj, "dclass");
- nassertr(dclass_obj != NULL, false);
- PyObject *dclass_this = PyObject_GetAttrString(dclass_obj, "this");
- Py_DECREF(dclass_obj);
- nassertr(dclass_this != NULL, false);
- DCClass *dclass = (DCClass *)PyLong_AsVoidPtr(dclass_this);
- Py_DECREF(dclass_this);
- // check if we should forward this update to the owner view
- vector_uchar data = _di.get_remaining_bytes();
- DCPacker packer;
- packer.set_unpack_data((const char *)data.data(), data.size(), false);
- int field_id = packer.raw_unpack_uint16();
- DCField *field = dclass->get_field_by_index(field_id);
- if (true) {//field->is_broadcast()) {
- // It's a good idea to ensure the reference count to distobj is raised
- // while we call the update method--otherwise, the update method might
- // get into trouble if it tried to delete the object from the doId2do
- // map.
- Py_INCREF(distobj);
- dclass->receive_update(distobj, _di);
- Py_DECREF(distobj);
- if (PyErr_Occurred()) {
- #if defined(HAVE_THREADS) && !defined(SIMPLE_THREADS)
- PyGILState_Release(gstate);
- #endif
- return false;
- }
- }
- }
- }
- #if defined(HAVE_THREADS) && !defined(SIMPLE_THREADS)
- PyGILState_Release(gstate);
- #endif
- #endif // HAVE_PYTHON
- return true;
- }
- /**
- * Unpacks the message and reformats it for user consumption, writing a
- * description on the indicated output stream.
- */
- void CConnectionRepository::
- describe_message(ostream &out, const string &prefix,
- const Datagram &dg) const {
- DCPacker packer;
- packer.set_unpack_data(dg.get_message());
- CHANNEL_TYPE do_id;
- int msg_type;
- bool is_update = false;
- string full_prefix = "CR::" + prefix;
- if (!_client_datagram)
- {
- unsigned char mcnt = packer.raw_unpack_uint8();
- for( ;mcnt > 0; mcnt--)
- packer.RAW_UNPACK_CHANNEL(); // msg_channel
- packer.RAW_UNPACK_CHANNEL(); // msg_sender
- msg_type = packer.raw_unpack_uint16();
- is_update = (msg_type == STATESERVER_OBJECT_SET_FIELD);
- } else {
- msg_type = packer.raw_unpack_uint16();
- is_update = (msg_type == CLIENT_OBJECT_SET_FIELD);
- }
- if (!is_update) {
- // figure out the name of the message TODO: print out the arguments to the
- // message
- string msgName;
- #ifdef HAVE_PYTHON
- if (_python_repository != (PyObject *)NULL) {
- PyObject *msgId = PyLong_FromLong(msg_type);
- nassertv(msgId != NULL);
- #if PY_MAJOR_VERSION >= 3
- PyObject *methodName = PyUnicode_FromString("_getMsgName");
- #else
- PyObject *methodName = PyString_FromString("_getMsgName");
- #endif
- nassertv(methodName != NULL);
- PyObject *result = PyObject_CallMethodObjArgs(_python_repository, methodName,
- msgId, NULL);
- nassertv(result != NULL);
- #if PY_MAJOR_VERSION >= 3
- msgName += string(PyUnicode_AsUTF8(result));
- #else
- msgName += string(PyString_AsString(result));
- #endif
- Py_DECREF(methodName);
- Py_DECREF(msgId);
- Py_DECREF(result);
- }
- #endif
- if (msgName.length() == 0) {
- msgName += "unknown message ";
- msgName += msg_type;
- msgName += "\n";
- }
- out << full_prefix << ":" << msgName << "\n";
- dg.dump_hex(out, 2);
- } else {
- // It's an update message. Figure out what dclass the object is based on
- // its doId, so we can decode the rest of the message.
- do_id = packer.raw_unpack_uint32();
- DCClass *dclass = NULL;
- #ifdef HAVE_PYTHON
- if (_python_repository != (PyObject *)NULL) {
- PyObject *doId2do =
- PyObject_GetAttrString(_python_repository, "doId2do");
- nassertv(doId2do != NULL);
- #ifdef USE_PYTHON_2_2_OR_EARLIER
- PyObject *doId = PyInt_FromLong(do_id);
- #else
- PyObject *doId = PyLong_FromUnsignedLong(do_id);
- #endif
- PyObject *distobj = PyDict_GetItem(doId2do, doId);
- Py_DECREF(doId);
- Py_DECREF(doId2do);
- if (distobj != NULL) {
- PyObject *dclass_obj = PyObject_GetAttrString(distobj, "dclass");
- nassertv(dclass_obj != NULL);
- PyObject *dclass_this = PyObject_GetAttrString(dclass_obj, "this");
- Py_DECREF(dclass_obj);
- nassertv(dclass_this != NULL);
- dclass = (DCClass *)PyLong_AsVoidPtr(dclass_this);
- Py_DECREF(dclass_this);
- }
- }
- #endif // HAVE_PYTHON
- int field_id = packer.raw_unpack_uint16();
- if (dclass == (DCClass *)NULL) {
- out << full_prefix << "update for unknown object " << do_id
- << ", field " << field_id << "\n";
- } else {
- out << full_prefix <<
- ":" << dclass->get_name() << "(" << do_id << ").";
- DCField *field = dclass->get_field_by_index(field_id);
- if (field == (DCField *)NULL) {
- out << "unknown field " << field_id << "\n";
- } else {
- out << field->get_name();
- packer.begin_unpack(field);
- packer.unpack_and_format(out);
- packer.end_unpack();
- out << "\n";
- }
- }
- }
- }
|