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