瀏覽代碼

Added incoming queue size limit in TLS transport

Paul-Louis Ageneau 3 年之前
父節點
當前提交
526f73089e
共有 1 個文件被更改,包括 5 次插入5 次删除
  1. 5 5
      src/impl/tlstransport.cpp

+ 5 - 5
src/impl/tlstransport.cpp

@@ -59,7 +59,8 @@ void TlsTransport::Cleanup() {
 
 TlsTransport::TlsTransport(shared_ptr<TcpTransport> lower, optional<string> host,
                            certificate_ptr certificate, state_callback callback)
-    : Transport(lower, std::move(callback)), mHost(std::move(host)), mIsClient(lower->isActive()) {
+    : Transport(lower, std::move(callback)), mHost(std::move(host)), mIsClient(lower->isActive()),
+      mIncomingQueue(RECV_QUEUE_LIMIT, message_size_func) {
 
 	PLOG_DEBUG << "Initializing TLS transport (GnuTLS)";
 
@@ -304,7 +305,8 @@ void TlsTransport::Cleanup() {
 
 TlsTransport::TlsTransport(shared_ptr<TcpTransport> lower, optional<string> host,
                            certificate_ptr certificate, state_callback callback)
-    : Transport(lower, std::move(callback)), mHost(std::move(host)), mIsClient(lower->isActive()) {
+    : Transport(lower, std::move(callback)), mHost(std::move(host)), mIsClient(lower->isActive()),
+      mIncomingQueue(RECV_QUEUE_LIMIT, message_size_func) {
 
 	PLOG_DEBUG << "Initializing TLS transport (OpenSSL)";
 
@@ -430,9 +432,7 @@ void TlsTransport::incoming(message_ptr message) {
 	mIncomingQueue.push(message);
 }
 
-bool TlsTransport::outgoing(message_ptr message) {
-	return Transport::outgoing(std::move(message));
-}
+bool TlsTransport::outgoing(message_ptr message) { return Transport::outgoing(std::move(message)); }
 
 void TlsTransport::postHandshake() {
 	// Dummy