1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* Copyright (C) 2018 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifdef _WIN32
// nothing to see here
#else
#include <memory>
#include <string>
#include <vector>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>
#include "android-base/logging.h"
#include "androidfw/PosixUtils.h"
namespace {
std::unique_ptr<std::string> ReadFile(int fd) {
std::unique_ptr<std::string> str(new std::string());
char buf[1024];
ssize_t r;
while ((r = read(fd, buf, sizeof(buf))) > 0) {
str->append(buf, r);
}
if (r != 0) {
return nullptr;
}
return str;
}
}
namespace android {
namespace util {
std::unique_ptr<ProcResult> ExecuteBinary(const std::vector<std::string>& argv) {
int stdout[2]; // stdout[0] read, stdout[1] write
if (pipe(stdout) != 0) {
PLOG(ERROR) << "pipe";
return nullptr;
}
int stderr[2]; // stdout[0] read, stdout[1] write
if (pipe(stderr) != 0) {
PLOG(ERROR) << "pipe";
close(stdout[0]);
close(stdout[1]);
return nullptr;
}
auto gid = getgid();
auto uid = getuid();
char const** argv0 = (char const**)malloc(sizeof(char*) * (argv.size() + 1));
for (size_t i = 0; i < argv.size(); i++) {
argv0[i] = argv[i].c_str();
}
argv0[argv.size()] = nullptr;
int pid = fork();
switch (pid) {
case -1: // error
free(argv0);
PLOG(ERROR) << "fork";
return nullptr;
case 0: // child
if (setgid(gid) != 0) {
PLOG(ERROR) << "setgid";
exit(1);
}
if (setuid(uid) != 0) {
PLOG(ERROR) << "setuid";
exit(1);
}
close(stdout[0]);
if (dup2(stdout[1], STDOUT_FILENO) == -1) {
abort();
}
close(stderr[0]);
if (dup2(stderr[1], STDERR_FILENO) == -1) {
abort();
}
execvp(argv0[0], const_cast<char* const*>(argv0));
PLOG(ERROR) << "execv";
abort();
default: // parent
free(argv0);
close(stdout[1]);
close(stderr[1]);
int status;
waitpid(pid, &status, 0);
if (!WIFEXITED(status)) {
close(stdout[0]);
close(stderr[0]);
return nullptr;
}
std::unique_ptr<ProcResult> result(new ProcResult());
result->status = status;
const auto out = ReadFile(stdout[0]);
result->stdout_str = out ? *out : "";
close(stdout[0]);
const auto err = ReadFile(stderr[0]);
result->stderr_str = err ? *err : "";
close(stderr[0]);
return result;
}
}
} // namespace util
} // namespace android
#endif
|
