folly/io/async/test/AsyncSocketExceptionTest.cpp

   1 /*
   2  * Copyright 2017 Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 #include <array>
  17
  18 #include <folly/io/async/AsyncSocketException.h>
  19 #include <folly/io/async/ssl/SSLErrors.h>
  20 #include <folly/portability/GTest.h>
  21
  22 #include <openssl/ssl.h>
  23
  24 using namespace testing;
  25
  26 namespace folly {
  27
  28 TEST(AsyncSocketException, SimpleTest) {
  29   AsyncSocketException ex1(
  30       AsyncSocketException::AsyncSocketExceptionType::NOT_OPEN,
  31       "test exception 1");
  32
  33   EXPECT_EQ(AsyncSocketException::AsyncSocketExceptionType::NOT_OPEN,
  34             ex1.getType());
  35   EXPECT_EQ(0, ex1.getErrno());
  36   EXPECT_EQ("AsyncSocketException: test exception 1, type = Socket not open",
  37             std::string(ex1.what()));
  38
  39   AsyncSocketException ex2(
  40       AsyncSocketException::AsyncSocketExceptionType::BAD_ARGS,
  41       "test exception 2",
  42       111 /*ECONNREFUSED*/);
  43
  44   EXPECT_EQ(AsyncSocketException::AsyncSocketExceptionType::BAD_ARGS,
  45             ex2.getType());
  46   EXPECT_EQ(111, ex2.getErrno());
  47   EXPECT_EQ(
  48       "AsyncSocketException: test exception 2, type = Invalid arguments, "
  49       "errno = 111 (Connection refused)",
  50       std::string(ex2.what()));
  51 }
  52
  53 TEST(AsyncSocketException, SSLExceptionType) {
  54   {
  55     SSLException eof(SSL_ERROR_ZERO_RETURN, 0, 0, 0);
  56     EXPECT_EQ(eof.getType(), AsyncSocketException::END_OF_FILE);
  57
  58     SSLException netEof(SSL_ERROR_SYSCALL, 0, 0, 0);
  59     EXPECT_EQ(netEof.getType(), AsyncSocketException::END_OF_FILE);
  60
  61     SSLException netOther(SSL_ERROR_SYSCALL, 0, 1, 0);
  62     EXPECT_EQ(netOther.getType(), AsyncSocketException::NETWORK_ERROR);
  63
  64     std::array<int, 6> sslErrs{{SSL_ERROR_SSL,
  65                                 SSL_ERROR_WANT_READ,
  66                                 SSL_ERROR_WANT_WRITE,
  67                                 SSL_ERROR_WANT_X509_LOOKUP,
  68                                 SSL_ERROR_WANT_CONNECT,
  69                                 SSL_ERROR_WANT_ACCEPT}};
  70
  71     for (auto& e : sslErrs) {
  72       SSLException sslEx(e, 0, 0, 0);
  73       EXPECT_EQ(sslEx.getType(), AsyncSocketException::SSL_ERROR);
  74     }
  75   }
  76
  77   {
  78     SSLException eof(SSLError::EOF_ERROR);
  79     EXPECT_EQ(eof.getType(), AsyncSocketException::END_OF_FILE);
  80
  81     SSLException net(SSLError::NETWORK_ERROR);
  82     EXPECT_EQ(net.getType(), AsyncSocketException::NETWORK_ERROR);
  83
  84     std::array<SSLError, 4> errs{{SSLError::CLIENT_RENEGOTIATION,
  85                                   SSLError::INVALID_RENEGOTIATION,
  86                                   SSLError::EARLY_WRITE,
  87                                   SSLError::SSL_ERROR}};
  88
  89     for (auto& e : errs) {
  90       SSLException sslEx(e);
  91       EXPECT_EQ(sslEx.getType(), AsyncSocketException::SSL_ERROR);
  92     }
  93   }
  94 }
  95
  96 } // namespace folly