Fix zerocopy AsyncSocket memory leaks

[folly.git] / folly / io / async / test / ZeroCopy.cpp

diff --git a/folly/io/async/test/ZeroCopy.cpp b/folly/io/async/test/ZeroCopy.cpp
new file mode 100644 (file)
index 0000000..c06badd
--- /dev/null
+++ b/folly/io/async/test/ZeroCopy.cpp
@@ -0,0 +1,52 @@
+/*
+ * Copyright 2017 Facebook, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <folly/io/async/test/ZeroCopy.h>
+
+namespace folly {
+
+// ZeroCopyTest
+ZeroCopyTest::ZeroCopyTest(int numLoops, bool zeroCopy, size_t bufferSize)
+    : numLoops_(numLoops),
+      zeroCopy_(zeroCopy),
+      bufferSize_(bufferSize),
+      client_(
+          new ZeroCopyTestAsyncSocket(&evb_, numLoops_, bufferSize_, zeroCopy)),
+      listenSock_(new folly::AsyncServerSocket(&evb_)),
+      server_(&evb_, numLoops_, bufferSize_, zeroCopy) {
+  if (listenSock_) {
+    server_.addCallbackToServerSocket(*listenSock_);
+  }
+}
+
+bool ZeroCopyTest::run() {
+  evb_.runInEventBaseThread([this]() {
+    if (listenSock_) {
+      listenSock_->bind(0);
+      listenSock_->setZeroCopy(zeroCopy_);
+      listenSock_->listen(10);
+      listenSock_->startAccepting();
+
+      connectOne();
+    }
+  });
+
+  evb_.loopForever();
+
+  return !client_->isZeroCopyWriteInProgress();
+}
+
+} // namespace folly