1 /* 2 * Copyright (C) 2017 The Android Open Source Project 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 17 #ifndef NETUTILS_MOCK_SYSCALLS_H 18 #define NETUTILS_MOCK_SYSCALLS_H 19 20 #include <atomic> 21 #include <cassert> 22 #include <memory> 23 24 #include <gmock/gmock.h> 25 #include <gtest/gtest.h> 26 27 #include "netdutils/Syscalls.h" 28 29 namespace android { 30 namespace netdutils { 31 32 class MockSyscalls : public Syscalls { 33 public: 34 virtual ~MockSyscalls() = default; 35 // Use Return(ByMove(...)) to deal with movable return types. 36 MOCK_CONST_METHOD3(open, 37 StatusOr<UniqueFd>(const std::string& pathname, int flags, mode_t mode)); 38 MOCK_CONST_METHOD3(socket, StatusOr<UniqueFd>(int domain, int type, int protocol)); 39 MOCK_CONST_METHOD3(getsockname, Status(Fd sock, sockaddr* addr, socklen_t* addrlen)); 40 MOCK_CONST_METHOD5(setsockopt, Status(Fd sock, int level, int optname, const void* optval, 41 socklen_t optlen)); 42 43 MOCK_CONST_METHOD3(bind, Status(Fd sock, const sockaddr* addr, socklen_t addrlen)); 44 MOCK_CONST_METHOD3(connect, Status(Fd sock, const sockaddr* addr, socklen_t addrlen)); 45 46 // Use Return(ByMove(...)) to deal with movable return types. 47 MOCK_CONST_METHOD2(eventfd, StatusOr<UniqueFd>(unsigned int initval, int flags)); 48 MOCK_CONST_METHOD3(ppoll, StatusOr<int>(pollfd* fds, nfds_t nfds, double timeout)); 49 MOCK_CONST_METHOD2(write, StatusOr<size_t>(Fd fd, const Slice buf)); 50 MOCK_CONST_METHOD2(read, StatusOr<Slice>(Fd fd, const Slice buf)); 51 MOCK_CONST_METHOD5(sendto, StatusOr<size_t>(Fd sock, const Slice buf, int flags, 52 const sockaddr* dst, socklen_t dstlen)); 53 MOCK_CONST_METHOD5(recvfrom, StatusOr<Slice>(Fd sock, const Slice dst, int flags, sockaddr* src, 54 socklen_t* srclen)); 55 MOCK_CONST_METHOD2(shutdown, Status(Fd fd, int how)); 56 MOCK_CONST_METHOD1(close, Status(Fd fd)); 57 58 MOCK_CONST_METHOD2(fopen, 59 StatusOr<UniqueFile>(const std::string& path, const std::string& mode)); 60 MOCK_CONST_METHOD3(vfprintf, StatusOr<int>(FILE* file, const char* format, va_list ap)); 61 MOCK_CONST_METHOD3(vfscanf, StatusOr<int>(FILE* file, const char* format, va_list ap)); 62 MOCK_CONST_METHOD1(fclose, Status(FILE* file)); 63 MOCK_CONST_METHOD0(fork, StatusOr<pid_t>()); 64 }; 65 66 // For the lifetime of this mock, replace the contents of sSyscalls 67 // with a pointer to this mock. Behavior is undefined if multiple 68 // ScopedMockSyscalls instances exist concurrently. 69 class ScopedMockSyscalls : public MockSyscalls { 70 public: ScopedMockSyscalls()71 ScopedMockSyscalls() : mOld(sSyscalls.swap(*this)) { assert((mRefcount++) == 1); } ~ScopedMockSyscalls()72 virtual ~ScopedMockSyscalls() { 73 sSyscalls.swap(mOld); 74 assert((mRefcount--) == 0); 75 } 76 77 private: 78 std::atomic<int> mRefcount{0}; 79 Syscalls& mOld; 80 }; 81 82 } // namespace netdutils 83 } // namespace android 84 85 #endif /* NETUTILS_MOCK_SYSCALLS_H */ 86