LineageOS/android_packages_modules_Connectivity
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge "Skip the test if cannot open ctrl file"

Browse Source
This commit is contained in:
Treehugger Robot
2018-07-27 03:48:14 +00:00
committed by Gerrit Code Review
parent d8eb370a89 cc7747d2de
commit 2ab9867d51
1 changed files with 21 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
tests/cts/net/native/qtaguid/src/NativeQtaguidTest.cpp
View File

@@ -18,36 +18,29 @@
#include <error.h> #include <error.h>
#include <errno.h> #include <errno.h>
#include <inttypes.h> #include <inttypes.h>
#include <fcntl.h>
#include <string.h> #include <string.h>
#include <sys/socket.h> #include <sys/socket.h>
#include <sys/utsname.h>
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include <qtaguid/qtaguid.h> #include <qtaguid/qtaguid.h>
int hasQtaguidKernelSupport() { int canAccessQtaguidFile() {
struct utsname buf; int fd = open("/proc/net/xt_qtaguid/ctrl", O_RDONLY | O_CLOEXEC);
int kernel_version_major; close(fd);
int kernel_version_minor; return fd != -1;
int ret = uname(&buf);
if (ret) {
ret = -errno;
return ret;
}
char dummy;
ret = sscanf(buf.release, "%d.%d%c", &kernel_version_major, &kernel_version_minor, &dummy);
if (ret < 3)
return -EINVAL;
if ((kernel_version_major == 4 && kernel_version_minor < 9) ||
(kernel_version_major < 4)) {
return 1;
} else {
return access("/proc/net/xt_qtaguid/ctrl", F_OK) != -1;
}
} }
#define SKIP_IF_QTAGUID_NOT_SUPPORTED() \
do { \
int res = canAccessQtaguidFile(); \
ASSERT_LE(0, res); \
if (!res) { \
GTEST_LOG_(INFO) << "This test is skipped since kernel may not have the module\n"; \
return; \
} \
} while (0)
int getCtrlSkInfo(int tag, uid_t uid, uint64_t* sk_addr, int* ref_cnt) { int getCtrlSkInfo(int tag, uid_t uid, uint64_t* sk_addr, int* ref_cnt) {
FILE *fp; FILE *fp;
fp = fopen("/proc/net/xt_qtaguid/ctrl", "r"); fp = fopen("/proc/net/xt_qtaguid/ctrl", "r");
@@ -95,12 +88,8 @@ void checkNoSocketPointerLeaks(int family) {
} }
TEST (NativeQtaguidTest, close_socket_without_untag) { TEST (NativeQtaguidTest, close_socket_without_untag) {
int res = hasQtaguidKernelSupport(); SKIP_IF_QTAGUID_NOT_SUPPORTED();
ASSERT_LE(0, res);
if (!res) {
GTEST_LOG_(INFO) << "This test is skipped since kernel may not have the module\n";
return;
}
int sockfd = socket(AF_INET, SOCK_STREAM, 0); int sockfd = socket(AF_INET, SOCK_STREAM, 0);
uid_t uid = getuid(); uid_t uid = getuid();
int tag = arc4random(); int tag = arc4random();
@@ -114,12 +103,8 @@ TEST (NativeQtaguidTest, close_socket_without_untag) {
} }
TEST (NativeQtaguidTest, close_socket_without_untag_ipv6) { TEST (NativeQtaguidTest, close_socket_without_untag_ipv6) {
int res = hasQtaguidKernelSupport(); SKIP_IF_QTAGUID_NOT_SUPPORTED();
ASSERT_LE(0, res);
if (!res) {
GTEST_LOG_(INFO) << "This test is skipped since kernel may not have the module\n";
return;
}
int sockfd = socket(AF_INET6, SOCK_STREAM, 0); int sockfd = socket(AF_INET6, SOCK_STREAM, 0);
uid_t uid = getuid(); uid_t uid = getuid();
int tag = arc4random(); int tag = arc4random();
@@ -133,12 +118,8 @@ TEST (NativeQtaguidTest, close_socket_without_untag_ipv6) {
} }
TEST (NativeQtaguidTest, no_socket_addr_leak) { TEST (NativeQtaguidTest, no_socket_addr_leak) {
int res = hasQtaguidKernelSupport(); SKIP_IF_QTAGUID_NOT_SUPPORTED();
ASSERT_LE(0, res);
if (!res) {
GTEST_LOG_(INFO) << "This test is skipped since kernel may not have the module\n";
return;
}
checkNoSocketPointerLeaks(AF_INET); checkNoSocketPointerLeaks(AF_INET);
checkNoSocketPointerLeaks(AF_INET6); checkNoSocketPointerLeaks(AF_INET6);
} }