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(getsockopt, Status(Fd sock, int level, int optname, void* optval, 41 socklen_t *optlen)); 42 MOCK_CONST_METHOD5(setsockopt, Status(Fd sock, int level, int optname, const void* optval, 43 socklen_t optlen)); 44 45 MOCK_CONST_METHOD3(bind, Status(Fd sock, const sockaddr* addr, socklen_t addrlen)); 46 MOCK_CONST_METHOD3(connect, Status(Fd sock, const sockaddr* addr, socklen_t addrlen)); 47 MOCK_CONST_METHOD3(ioctl, StatusOr<ifreq>(Fd sock, unsigned long request, ifreq* ifr)); 48 49 // Use Return(ByMove(...)) to deal with movable return types. 50 MOCK_CONST_METHOD2(eventfd, StatusOr<UniqueFd>(unsigned int initval, int flags)); 51 MOCK_CONST_METHOD3(ppoll, StatusOr<int>(pollfd* fds, nfds_t nfds, double timeout)); 52 53 MOCK_CONST_METHOD2(writev, StatusOr<size_t>(Fd fd, const std::vector<iovec>& iov)); 54 MOCK_CONST_METHOD2(write, StatusOr<size_t>(Fd fd, const Slice buf)); 55 MOCK_CONST_METHOD2(read, StatusOr<Slice>(Fd fd, const Slice buf)); 56 MOCK_CONST_METHOD5(sendto, StatusOr<size_t>(Fd sock, const Slice buf, int flags, 57 const sockaddr* dst, socklen_t dstlen)); 58 MOCK_CONST_METHOD5(recvfrom, StatusOr<Slice>(Fd sock, const Slice dst, int flags, sockaddr* src, 59 socklen_t* srclen)); 60 MOCK_CONST_METHOD2(shutdown, Status(Fd fd, int how)); 61 MOCK_CONST_METHOD1(close, Status(Fd fd)); 62 63 MOCK_CONST_METHOD2(fopen, 64 StatusOr<UniqueFile>(const std::string& path, const std::string& mode)); 65 MOCK_CONST_METHOD3(vfprintf, StatusOr<int>(FILE* file, const char* format, va_list ap)); 66 MOCK_CONST_METHOD3(vfscanf, StatusOr<int>(FILE* file, const char* format, va_list ap)); 67 MOCK_CONST_METHOD1(fclose, Status(FILE* file)); 68 MOCK_CONST_METHOD0(fork, StatusOr<pid_t>()); 69 }; 70 71 // For the lifetime of this mock, replace the contents of sSyscalls 72 // with a pointer to this mock. Behavior is undefined if multiple 73 // ScopedMockSyscalls instances exist concurrently. 74 class ScopedMockSyscalls : public MockSyscalls { 75 public: ScopedMockSyscalls()76 ScopedMockSyscalls() : mOld(sSyscalls.swap(*this)) { assert((mRefcount++) == 1); } ~ScopedMockSyscalls()77 virtual ~ScopedMockSyscalls() { 78 sSyscalls.swap(mOld); 79 assert((mRefcount--) == 0); 80 } 81 82 private: 83 std::atomic<int> mRefcount{0}; 84 Syscalls& mOld; 85 }; 86 87 } // namespace netdutils 88 } // namespace android 89 90 #endif /* NETUTILS_MOCK_SYSCALLS_H */ 91