1*8d67ca89SAndroid Build Coastguard Worker /*
2*8d67ca89SAndroid Build Coastguard Worker * Copyright (C) 2015 The Android Open Source Project
3*8d67ca89SAndroid Build Coastguard Worker *
4*8d67ca89SAndroid Build Coastguard Worker * Licensed under the Apache License, Version 2.0 (the "License");
5*8d67ca89SAndroid Build Coastguard Worker * you may not use this file except in compliance with the License.
6*8d67ca89SAndroid Build Coastguard Worker * You may obtain a copy of the License at
7*8d67ca89SAndroid Build Coastguard Worker *
8*8d67ca89SAndroid Build Coastguard Worker * http://www.apache.org/licenses/LICENSE-2.0
9*8d67ca89SAndroid Build Coastguard Worker *
10*8d67ca89SAndroid Build Coastguard Worker * Unless required by applicable law or agreed to in writing, software
11*8d67ca89SAndroid Build Coastguard Worker * distributed under the License is distributed on an "AS IS" BASIS,
12*8d67ca89SAndroid Build Coastguard Worker * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*8d67ca89SAndroid Build Coastguard Worker * See the License for the specific language governing permissions and
14*8d67ca89SAndroid Build Coastguard Worker * limitations under the License.
15*8d67ca89SAndroid Build Coastguard Worker */
16*8d67ca89SAndroid Build Coastguard Worker
17*8d67ca89SAndroid Build Coastguard Worker #include <gtest/gtest.h>
18*8d67ca89SAndroid Build Coastguard Worker #include <pthread.h>
19*8d67ca89SAndroid Build Coastguard Worker #include <fcntl.h>
20*8d67ca89SAndroid Build Coastguard Worker #include <unistd.h>
21*8d67ca89SAndroid Build Coastguard Worker #include <sys/types.h>
22*8d67ca89SAndroid Build Coastguard Worker #include <sys/stat.h>
23*8d67ca89SAndroid Build Coastguard Worker #include <sys/prctl.h>
24*8d67ca89SAndroid Build Coastguard Worker
25*8d67ca89SAndroid Build Coastguard Worker #include <string>
26*8d67ca89SAndroid Build Coastguard Worker
27*8d67ca89SAndroid Build Coastguard Worker #include <android-base/scopeguard.h>
28*8d67ca89SAndroid Build Coastguard Worker
29*8d67ca89SAndroid Build Coastguard Worker extern "C" pid_t gettid();
30*8d67ca89SAndroid Build Coastguard Worker
ProcSelfReadlinkBody()31*8d67ca89SAndroid Build Coastguard Worker static void ProcSelfReadlinkBody() {
32*8d67ca89SAndroid Build Coastguard Worker char buf[100];
33*8d67ca89SAndroid Build Coastguard Worker char buf2[1024];
34*8d67ca89SAndroid Build Coastguard Worker int fd = open("/dev/null", O_RDWR | O_CLOEXEC);
35*8d67ca89SAndroid Build Coastguard Worker ASSERT_NE(-1, fd);
36*8d67ca89SAndroid Build Coastguard Worker snprintf(buf, sizeof(buf), "/proc/self/fd/%d", fd);
37*8d67ca89SAndroid Build Coastguard Worker const char* ERRORMSG = "Please apply the following two kernel patches:\n"
38*8d67ca89SAndroid Build Coastguard Worker "* https://git.kernel.org/cgit/linux/kernel/git/torvalds/linux.git/commit/?id=73af963f9f3036dffed55c3a2898598186db1045\n"
39*8d67ca89SAndroid Build Coastguard Worker "* https://git.kernel.org/cgit/linux/kernel/git/torvalds/linux.git/commit/?id=96d0df79f2644fc823f26c06491e182d87a90c2a\n";
40*8d67ca89SAndroid Build Coastguard Worker ssize_t length = readlink(buf, buf2, sizeof(buf2));
41*8d67ca89SAndroid Build Coastguard Worker ASSERT_LT(0, length) << ERRORMSG;
42*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ("/dev/null", std::string(buf2, length));
43*8d67ca89SAndroid Build Coastguard Worker close(fd);
44*8d67ca89SAndroid Build Coastguard Worker }
45*8d67ca89SAndroid Build Coastguard Worker
ProcSelfReadlink(void *)46*8d67ca89SAndroid Build Coastguard Worker static void* ProcSelfReadlink(void*) {
47*8d67ca89SAndroid Build Coastguard Worker ProcSelfReadlinkBody();
48*8d67ca89SAndroid Build Coastguard Worker return nullptr;
49*8d67ca89SAndroid Build Coastguard Worker }
50*8d67ca89SAndroid Build Coastguard Worker
TEST(bug_26110743,ProcSelfReadlink)51*8d67ca89SAndroid Build Coastguard Worker TEST(bug_26110743, ProcSelfReadlink) {
52*8d67ca89SAndroid Build Coastguard Worker pthread_t t;
53*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(0, pthread_create(&t, nullptr, ProcSelfReadlink, nullptr));
54*8d67ca89SAndroid Build Coastguard Worker void* result;
55*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(0, pthread_join(t, &result));
56*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(nullptr, result);
57*8d67ca89SAndroid Build Coastguard Worker }
58*8d67ca89SAndroid Build Coastguard Worker
TEST(bug_26110743,ProcSelfReadlink_NotDumpable)59*8d67ca89SAndroid Build Coastguard Worker TEST(bug_26110743, ProcSelfReadlink_NotDumpable) {
60*8d67ca89SAndroid Build Coastguard Worker int dumpable = prctl(PR_GET_DUMPABLE, 0, 0, 0, 0);
61*8d67ca89SAndroid Build Coastguard Worker prctl(PR_SET_DUMPABLE, 0, 0, 0, 0);
62*8d67ca89SAndroid Build Coastguard Worker auto guard = android::base::make_scope_guard([&]() {
63*8d67ca89SAndroid Build Coastguard Worker // restore dumpable
64*8d67ca89SAndroid Build Coastguard Worker prctl(PR_SET_DUMPABLE, dumpable, 0, 0, 0);
65*8d67ca89SAndroid Build Coastguard Worker });
66*8d67ca89SAndroid Build Coastguard Worker
67*8d67ca89SAndroid Build Coastguard Worker pthread_t t;
68*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(0, pthread_create(&t, nullptr, ProcSelfReadlink, nullptr));
69*8d67ca89SAndroid Build Coastguard Worker void* result;
70*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(0, pthread_join(t, &result));
71*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(nullptr, result);
72*8d67ca89SAndroid Build Coastguard Worker }
73*8d67ca89SAndroid Build Coastguard Worker
ProcTaskFdReadlinkBody()74*8d67ca89SAndroid Build Coastguard Worker static void ProcTaskFdReadlinkBody() {
75*8d67ca89SAndroid Build Coastguard Worker char buf[200];
76*8d67ca89SAndroid Build Coastguard Worker char buf2[1024];
77*8d67ca89SAndroid Build Coastguard Worker int fd = open("/dev/null", O_RDWR | O_CLOEXEC);
78*8d67ca89SAndroid Build Coastguard Worker ASSERT_NE(-1, fd);
79*8d67ca89SAndroid Build Coastguard Worker pid_t mypid = getpid();
80*8d67ca89SAndroid Build Coastguard Worker pid_t mytid = gettid();
81*8d67ca89SAndroid Build Coastguard Worker ASSERT_NE(mypid, mytid);
82*8d67ca89SAndroid Build Coastguard Worker snprintf(buf, sizeof(buf), "/proc/%d/task/%d/fd/%d", mypid, mytid, fd);
83*8d67ca89SAndroid Build Coastguard Worker const char* ERRORMSG = "Please apply the following kernel patch:\n"
84*8d67ca89SAndroid Build Coastguard Worker "* https://git.kernel.org/cgit/linux/kernel/git/torvalds/linux.git/commit/?id=54708d2858e79a2bdda10bf8a20c80eb96c20613\n";
85*8d67ca89SAndroid Build Coastguard Worker ssize_t length = readlink(buf, buf2, sizeof(buf2));
86*8d67ca89SAndroid Build Coastguard Worker ASSERT_LT(0, length) << ERRORMSG;
87*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ("/dev/null", std::string(buf2, length));
88*8d67ca89SAndroid Build Coastguard Worker close(fd);
89*8d67ca89SAndroid Build Coastguard Worker }
90*8d67ca89SAndroid Build Coastguard Worker
ProcTaskFdReadlink(void *)91*8d67ca89SAndroid Build Coastguard Worker static void* ProcTaskFdReadlink(void*) {
92*8d67ca89SAndroid Build Coastguard Worker ProcTaskFdReadlinkBody();
93*8d67ca89SAndroid Build Coastguard Worker return nullptr;
94*8d67ca89SAndroid Build Coastguard Worker }
95*8d67ca89SAndroid Build Coastguard Worker
TEST(bug_26110743,ProcTaskFdReadlink)96*8d67ca89SAndroid Build Coastguard Worker TEST(bug_26110743, ProcTaskFdReadlink) {
97*8d67ca89SAndroid Build Coastguard Worker pthread_t t;
98*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(0, pthread_create(&t, nullptr, ProcTaskFdReadlink, nullptr));
99*8d67ca89SAndroid Build Coastguard Worker void* result;
100*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(0, pthread_join(t, &result));
101*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(nullptr, result);
102*8d67ca89SAndroid Build Coastguard Worker }
103*8d67ca89SAndroid Build Coastguard Worker
TEST(bug_26110743,ProcTaskFdReadlink_NotDumpable)104*8d67ca89SAndroid Build Coastguard Worker TEST(bug_26110743, ProcTaskFdReadlink_NotDumpable) {
105*8d67ca89SAndroid Build Coastguard Worker int dumpable = prctl(PR_GET_DUMPABLE, 0, 0, 0, 0);
106*8d67ca89SAndroid Build Coastguard Worker prctl(PR_SET_DUMPABLE, 0, 0, 0, 0);
107*8d67ca89SAndroid Build Coastguard Worker auto guard = android::base::make_scope_guard([&]() {
108*8d67ca89SAndroid Build Coastguard Worker // restore dumpable
109*8d67ca89SAndroid Build Coastguard Worker prctl(PR_SET_DUMPABLE, dumpable, 0, 0, 0);
110*8d67ca89SAndroid Build Coastguard Worker });
111*8d67ca89SAndroid Build Coastguard Worker
112*8d67ca89SAndroid Build Coastguard Worker pthread_t t;
113*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(0, pthread_create(&t, nullptr, ProcTaskFdReadlink, nullptr));
114*8d67ca89SAndroid Build Coastguard Worker void* result;
115*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(0, pthread_join(t, &result));
116*8d67ca89SAndroid Build Coastguard Worker ASSERT_EQ(nullptr, result);
117*8d67ca89SAndroid Build Coastguard Worker }
118