1 /*
2  * Copyright (C) 2020 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 #pragma once
18 
19 #include <linux/netlink.h>
20 #include <sys/socket.h>
21 
22 // We use #defines here so as to get local lamba captures and error message line numbers
23 #define ASSERT_TRUE_CALLBACK                                    \
24     [&](bool success, std::string errMsg) {                     \
25         ASSERT_TRUE(success) << "unexpected error: " << errMsg; \
26     }
27 
28 #define ASSERT_FALSE_CALLBACK \
29     [&](bool success, std::string errMsg) { ASSERT_FALSE(success) << "expected error: " << errMsg; }
30 
31 #define ASSERT_ZERO_BYTES_CALLBACK            \
32     [&](uint64_t rxBytes, uint64_t txBytes) { \
33         EXPECT_EQ(0ULL, rxBytes);             \
34         EXPECT_EQ(0ULL, txBytes);             \
35     }
36 
37 inline const sockaddr* asSockaddr(const sockaddr_nl* nladdr);
38 
39 int conntrackSocket(unsigned groups);