123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120 |
- /*
- * Copyright 2014-present 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 <iostream>
- #include <folly/io/async/AsyncServerSocket.h>
- #include <folly/io/async/AsyncSocket.h>
- #include <folly/io/async/EventBase.h>
- #include <folly/portability/GTest.h>
- namespace folly {
- #ifndef TCP_SAVE_SYN
- #define TCP_SAVE_SYN 27
- #endif
- TEST(AsyncSocketTest, getSockOpt) {
- EventBase evb;
- std::shared_ptr<AsyncSocket> socket = AsyncSocket::newSocket(&evb, 0);
- int val;
- socklen_t len;
- int expectedRc =
- getsockopt(socket->getFd(), SOL_SOCKET, SO_REUSEADDR, &val, &len);
- int actualRc = socket->getSockOpt(SOL_SOCKET, SO_REUSEADDR, &val, &len);
- EXPECT_EQ(expectedRc, actualRc);
- }
- TEST(AsyncSocketTest, REUSEPORT) {
- EventBase base;
- auto serverSocket = AsyncServerSocket::newSocket(&base);
- serverSocket->bind(0);
- serverSocket->listen(0);
- serverSocket->startAccepting();
- try {
- serverSocket->setReusePortEnabled(true);
- } catch (...) {
- LOG(INFO) << "Reuse port probably not supported";
- return;
- }
- SocketAddress address;
- serverSocket->getAddress(&address);
- int port = address.getPort();
- auto serverSocket2 = AsyncServerSocket::newSocket(&base);
- serverSocket2->setReusePortEnabled(true);
- serverSocket2->bind(port);
- serverSocket2->listen(0);
- serverSocket2->startAccepting();
- }
- TEST(AsyncSocketTest, v4v6samePort) {
- EventBase base;
- auto serverSocket = AsyncServerSocket::newSocket(&base);
- serverSocket->bind(0);
- auto addrs = serverSocket->getAddresses();
- ASSERT_GT(addrs.size(), 0);
- uint16_t port = addrs[0].getPort();
- for (const auto& addr : addrs) {
- EXPECT_EQ(port, addr.getPort());
- }
- }
- TEST(AsyncSocketTest, duplicateBind) {
- EventBase base;
- auto server1 = AsyncServerSocket::newSocket(&base);
- server1->bind(0);
- server1->listen(10);
- SocketAddress address;
- server1->getAddress(std::addressof(address));
- auto server2 = AsyncServerSocket::newSocket(&base);
- EXPECT_THROW(server2->bind(address.getPort()), std::exception);
- }
- TEST(AsyncSocketTest, tosReflect) {
- EventBase base;
- auto server1 = AsyncServerSocket::newSocket(&base);
- server1->bind(0);
- server1->listen(10);
- int fd = server1->getSocket();
- // Verify if tos reflect is disabled by default
- // and the TCP_SAVE_SYN setting is not enabled
- EXPECT_FALSE(server1->getTosReflect());
- int value;
- socklen_t valueLength = sizeof(value);
- int rc = getsockopt(fd, IPPROTO_TCP, TCP_SAVE_SYN, &value, &valueLength);
- ASSERT_EQ(rc, 0);
- ASSERT_EQ(value, 0);
- // Enable TOS reflect on the server socket
- server1->setTosReflect(true);
- // Verify if tos reflect is enabled now
- // and the TCP_SAVE_SYN setting is also enabled
- EXPECT_TRUE(server1->getTosReflect());
- rc = getsockopt(fd, IPPROTO_TCP, TCP_SAVE_SYN, &value, &valueLength);
- ASSERT_EQ(rc, 0);
- ASSERT_EQ(value, 1);
- }
- } // namespace folly
|