More fixes to RethinkDB.

controller/EmbeddedNetworkController.cpp

@@ -1185,8 +1185,9 @@ void EmbeddedNetworkController::_request(
 		ms.lastRequestTime = now;
 	}
 
-	OSUtils::ztsnprintf(nwids,sizeof(nwids),"%.16llx",nwid);
+	Utils::hex(nwid,nwids);
-	if (!_db->get(nwid,network,identity.address().toInt(),member,ns)) {
+	_db->get(nwid,network,identity.address().toInt(),member,ns);
+	if ((!network.is_object())||(network.size() == 0)) {
 		_sender->ncSendError(nwid,requestPacketId,identity.address(),NetworkController::NC_ERROR_OBJECT_NOT_FOUND);
 		return;
 	}
@@ -1684,11 +1685,13 @@ void EmbeddedNetworkController::_startThreads()
 		_threads.emplace_back([this]() {
 			for(;;) {
 				_RQEntry *qe = (_RQEntry *)0;
-				if (_queue.get(qe))
+				if (!_queue.get(qe))
 					break;
 				try {
-					if (qe)
+					if (qe) {
 						_request(qe->nwid,qe->fromAddr,qe->requestPacketId,qe->identity,qe->metaData);
+						delete qe;
+					}
 				} catch (std::exception &e) {
 					fprintf(stderr,"ERROR: exception in controller request handling thread: %s" ZT_EOL_S,e.what());
 				} catch ( ... ) {

controller/RethinkDB.cpp

@@ -78,6 +78,7 @@ RethinkDB::RethinkDB(EmbeddedNetworkController *const nc,const Address &myAddres
 									json &ov = tmp["old_val"];
 									json &nv = tmp["new_val"];
 									if (ov.is_object()||nv.is_object()) {
+										//if (nv.is_object()) printf("MEMBER: %s" ZT_EOL_S,nv.dump().c_str());
 										this->_memberChanged(ov,nv);
 									}
 								} catch ( ... ) {} // ignore bad records
@@ -118,6 +119,7 @@ RethinkDB::RethinkDB(EmbeddedNetworkController *const nc,const Address &myAddres
 									json &ov = tmp["old_val"];
 									json &nv = tmp["new_val"];
 									if (ov.is_object()||nv.is_object()) {
+										//if (nv.is_object()) printf("NETWORK: %s" ZT_EOL_S,nv.dump().c_str());
 										this->_networkChanged(ov,nv);
 									}
 								} catch ( ... ) {} // ignore bad records
@@ -148,34 +150,41 @@ RethinkDB::RethinkDB(EmbeddedNetworkController *const nc,const Address &myAddres
 					const std::string objtype = (*config)["objtype"];
 					const char *table;
 					std::string deleteId;
-					if (objtype == "member") {
+					try {
-						const std::string nwid = (*config)["nwid"];
+						if (objtype == "member") {
-						const std::string id = (*config)["id"];
+							const std::string nwid = (*config)["nwid"];
-						record["id"] = nwid + "-" + id;
+							const std::string id = (*config)["id"];
-						record["controllerId"] = this->_myAddressStr;
+							record["id"] = nwid + "-" + id;
-						record["networkId"] = nwid;
+							record["controllerId"] = this->_myAddressStr;
-						record["nodeId"] = id;
+							record["networkId"] = nwid;
-						record["config"] = *config;
+							record["nodeId"] = id;
-						table = "Member";
+							record["config"] = *config;
-					} else if (objtype == "network") {
+							table = "Member";
-						const std::string id = (*config)["id"];
+						} else if (objtype == "network") {
-						record["id"] = id;
+							const std::string id = (*config)["id"];
-						record["controllerId"] = this->_myAddressStr;
+							record["id"] = id;
-						record["config"] = *config;
+							record["controllerId"] = this->_myAddressStr;
-						table = "Network";
+							record["config"] = *config;
-					} else if (objtype == "delete_network") {
+							table = "Network";
-						deleteId = (*config)["id"];
+						} else if (objtype == "delete_network") {
-						table = "Network";
+							deleteId = (*config)["id"];
-					} else if (objtype == "delete_member") {
+							table = "Network";
-						deleteId = (*config)["nwid"];
+						} else if (objtype == "delete_member") {
-						deleteId.push_back('-');
+							deleteId = (*config)["nwid"];
-						const std::string tmp = (*config)["id"];
+							deleteId.push_back('-');
-						deleteId.append(tmp);
+							const std::string tmp = (*config)["id"];
-						table = "Member";
+							deleteId.append(tmp);
-					} else if (objtype == "trace") {
+							table = "Member";
-						record = *config;
+						} else if (objtype == "trace") {
-						table = "RemoteTrace";
+							record = *config;
-					} else {
+							table = "RemoteTrace";
+						} else {
+							delete config;
+							continue;
+						}
+						delete config;
+					} catch ( ... ) {
+						delete config;
 						continue;
 					}
 
@@ -185,10 +194,10 @@ RethinkDB::RethinkDB(EmbeddedNetworkController *const nc,const Address &myAddres
 								rdb = R::connect(this->_host,this->_port,this->_auth);
 							if (rdb) {
 								if (deleteId.length() > 0) {
-									printf("DELETE: %s" ZT_EOL_S,deleteId.c_str());
+									//printf("DELETE: %s" ZT_EOL_S,deleteId.c_str());
 									R::db(this->_db).table(table).get(deleteId).delete_().run(*rdb);
 								} else {
-									printf("UPSERT: %s" ZT_EOL_S,record.dump().c_str());
+									//printf("UPSERT: %s" ZT_EOL_S,record.dump().c_str());
 									R::db(this->_db).table(table).insert(R::Datum::from_json(record.dump()),R::optargs("conflict","update","return_changes",false)).run(*rdb);
 								}
 								break;
@@ -222,6 +231,7 @@ RethinkDB::RethinkDB(EmbeddedNetworkController *const nc,const Address &myAddres
 						rdb = R::connect(this->_host,this->_port,this->_auth);
 					if (rdb) {
 						OSUtils::ztsnprintf(tmp,sizeof(tmp),"{\"id\":\"%s\",\"lastAlive\":%lld}",this->_myAddressStr.c_str(),(long long)OSUtils::now());
+						//printf("HEARTBEAT: %s" ZT_EOL_S,tmp);
 						R::db(this->_db).table("Controller").update(R::Datum::from_json(tmp)).run(*rdb);
 					}
 				} catch ( ... ) {
@@ -240,15 +250,18 @@ RethinkDB::~RethinkDB()
 	_commitQueue.stop();
 	for(int t=0;t<ZT_CONTROLLER_RETHINKDB_COMMIT_THREADS;++t)
 		_commitThread[t].join();
-	_membersDbWatcher.detach();
+	if (_membersDbWatcherConnection)
-	_networksDbWatcher.detach();
+		((R::Connection *)_membersDbWatcherConnection)->close();
+	if (_networksDbWatcherConnection)
+		((R::Connection *)_networksDbWatcherConnection)->close();
+	_membersDbWatcher.join();
+	_networksDbWatcher.join();
 	_heartbeatThread.join();
 }
 
 bool RethinkDB::get(const uint64_t networkId,nlohmann::json &network)
 {
 	waitForReady();
-
 	std::shared_ptr<_Network> nw;
 	{
 		std::lock_guard<std::mutex> l(_networks_l);
@@ -257,17 +270,16 @@ bool RethinkDB::get(const uint64_t networkId,nlohmann::json &network)
 			return false;
 		nw = nwi->second;
 	}
-
+	{
-	std::lock_guard<std::mutex> l2(nw->lock);
+		std::lock_guard<std::mutex> l2(nw->lock);
-	network = nw->config;
+		network = nw->config;
-
+	}
 	return true;
 }
 
 bool RethinkDB::get(const uint64_t networkId,nlohmann::json &network,const uint64_t memberId,nlohmann::json &member)
 {
 	waitForReady();
-
 	std::shared_ptr<_Network> nw;
 	{
 		std::lock_guard<std::mutex> l(_networks_l);
@@ -276,21 +288,20 @@ bool RethinkDB::get(const uint64_t networkId,nlohmann::json &network,const uint6
 			return false;
 		nw = nwi->second;
 	}
-
+	{
-	std::lock_guard<std::mutex> l2(nw->lock);
+		std::lock_guard<std::mutex> l2(nw->lock);
-	auto m = nw->members.find(memberId);
+		network = nw->config;
-	if (m == nw->members.end())
+		auto m = nw->members.find(memberId);
-		return false;
+		if (m == nw->members.end())
-	network = nw->config;
+			return false;
-	member = m->second;
+		member = m->second;
-
+	}
 	return true;
 }
 
 bool RethinkDB::get(const uint64_t networkId,nlohmann::json &network,const uint64_t memberId,nlohmann::json &member,NetworkSummaryInfo &info)
 {
 	waitForReady();
-
 	std::shared_ptr<_Network> nw;
 	{
 		std::lock_guard<std::mutex> l(_networks_l);
@@ -299,22 +310,21 @@ bool RethinkDB::get(const uint64_t networkId,nlohmann::json &network,const uint6
 			return false;
 		nw = nwi->second;
 	}
-
+	{
-	std::lock_guard<std::mutex> l2(nw->lock);
+		std::lock_guard<std::mutex> l2(nw->lock);
-	auto m = nw->members.find(memberId);
+		network = nw->config;
-	if (m == nw->members.end())
+		_fillSummaryInfo(nw,info);
-		return false;
+		auto m = nw->members.find(memberId);
-	network = nw->config;
+		if (m == nw->members.end())
-	member = m->second;
+			return false;
-	_fillSummaryInfo(nw,info);
+		member = m->second;
-
+	}
 	return true;
 }
 
 bool RethinkDB::get(const uint64_t networkId,nlohmann::json &network,std::vector<nlohmann::json> &members)
 {
 	waitForReady();
-
 	std::shared_ptr<_Network> nw;
 	{
 		std::lock_guard<std::mutex> l(_networks_l);
@@ -323,19 +333,18 @@ bool RethinkDB::get(const uint64_t networkId,nlohmann::json &network,std::vector
 			return false;
 		nw = nwi->second;
 	}
-
+	{
-	std::lock_guard<std::mutex> l2(nw->lock);
+		std::lock_guard<std::mutex> l2(nw->lock);
-	network = nw->config;
+		network = nw->config;
-	for(auto m=nw->members.begin();m!=nw->members.end();++m)
+		for(auto m=nw->members.begin();m!=nw->members.end();++m)
-		members.push_back(m->second);
+			members.push_back(m->second);
-
+	}
 	return true;
 }
 
 bool RethinkDB::summary(const uint64_t networkId,NetworkSummaryInfo &info)
 {
 	waitForReady();
-
 	std::shared_ptr<_Network> nw;
 	{
 		std::lock_guard<std::mutex> l(_networks_l);
@@ -344,10 +353,10 @@ bool RethinkDB::summary(const uint64_t networkId,NetworkSummaryInfo &info)
 			return false;
 		nw = nwi->second;
 	}
-
+	{
-	std::lock_guard<std::mutex> l2(nw->lock);
+		std::lock_guard<std::mutex> l2(nw->lock);
-	_fillSummaryInfo(nw,info);
+		_fillSummaryInfo(nw,info);
-
+	}
 	return true;
 }
 

ext/librethinkdbxx/src/connection.cc

@@ -101,6 +101,8 @@ std::unique_ptr<Connection> connect(std::string host, int port, std::string auth
 Connection::Connection(ConnectionPrivate *dd) : d(dd) { }
 Connection::~Connection() {
     // close();
+    if (d->guarded_sockfd >= 0)
+        ::close(d->guarded_sockfd);
 }
 
 size_t ReadLock::recv_some(char* buf, size_t size, double wait) {
@@ -128,7 +130,7 @@ size_t ReadLock::recv_some(char* buf, size_t size, double wait) {
     }
 
     ssize_t numbytes = ::recv(conn->guarded_sockfd, buf, size, 0);
-    if (numbytes == -1) throw Error::from_errno("recv");
+    if (numbytes <= 0) throw Error::from_errno("recv");
     if (debug_net > 1) {
         fprintf(stderr, "<< %s\n", write_datum(std::string(buf, numbytes)).c_str());
     }
@@ -190,6 +192,7 @@ void Connection::close() {
     if (ret == -1) {
         throw Error::from_errno("close");
     }
+    d->guarded_sockfd = -1;
 }
 
 Response ConnectionPrivate::wait_for_response(uint64_t token_want, double wait) {

ext/librethinkdbxx/src/cursor.cc

@@ -21,9 +21,11 @@ CursorPrivate::CursorPrivate(uint64_t token_, Connection *conn_, Datum&& datum)
 Cursor::Cursor(CursorPrivate *dd) : d(dd) {}
 
 Cursor::~Cursor() {
-    if (d && d->conn) {
+    try {
-        close();
+        if (d && d->conn) {
-    }
+            close();
+        }
+    } catch ( ... ) {}
 }
 
 Datum& Cursor::next(double wait) const {