Mark Salyzyn | 1271716 | 2014-04-29 15:49:14 -0700 | [diff] [blame] | 1 | /* |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 2 | * Copyright (C) 2019 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 | */ |
San Mehat | 493dad9 | 2009-09-12 10:06:57 -0700 | [diff] [blame] | 16 | |
Suren Baghdasaryan | 1bd127b | 2019-01-25 05:30:52 +0000 | [diff] [blame] | 17 | #include <processgroup/sched_policy.h> |
Elliott Hughes | 8e9aeb9 | 2017-11-10 10:22:07 -0800 | [diff] [blame] | 18 | |
Jeff Brown | bff8f3f | 2012-05-08 15:05:42 -0700 | [diff] [blame] | 19 | #define LOG_TAG "SchedPolicy" |
| 20 | |
San Mehat | 493dad9 | 2009-09-12 10:06:57 -0700 | [diff] [blame] | 21 | #include <errno.h> |
Mark Salyzyn | 1271716 | 2014-04-29 15:49:14 -0700 | [diff] [blame] | 22 | #include <unistd.h> |
| 23 | |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 24 | #include <android-base/logging.h> |
| 25 | #include <android-base/threads.h> |
| 26 | #include <cgroup_map.h> |
| 27 | #include <processgroup/processgroup.h> |
| 28 | |
| 29 | using android::base::GetThreadId; |
Mark Salyzyn | 1271716 | 2014-04-29 15:49:14 -0700 | [diff] [blame] | 30 | |
Jeff Brown | bff8f3f | 2012-05-08 15:05:42 -0700 | [diff] [blame] | 31 | /* Re-map SP_DEFAULT to the system default policy, and leave other values unchanged. |
| 32 | * Call this any place a SchedPolicy is used as an input parameter. |
| 33 | * Returns the possibly re-mapped policy. |
| 34 | */ |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 35 | static inline SchedPolicy _policy(SchedPolicy p) { |
| 36 | return p == SP_DEFAULT ? SP_SYSTEM_DEFAULT : p; |
Jeff Brown | bff8f3f | 2012-05-08 15:05:42 -0700 | [diff] [blame] | 37 | } |
San Mehat | d2e4e46 | 2009-10-29 11:48:00 -0700 | [diff] [blame] | 38 | |
Suren Baghdasaryan | eca87cb | 2019-02-02 14:19:41 -0800 | [diff] [blame] | 39 | #if defined(__ANDROID__) |
| 40 | |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 41 | int set_cpuset_policy(int tid, SchedPolicy policy) { |
Glenn Kasten | 69bfb1f | 2012-03-16 09:43:19 -0700 | [diff] [blame] | 42 | if (tid == 0) { |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 43 | tid = GetThreadId(); |
Tim Murray | b769c8d | 2015-06-08 14:56:29 -0700 | [diff] [blame] | 44 | } |
| 45 | policy = _policy(policy); |
Tim Murray | b769c8d | 2015-06-08 14:56:29 -0700 | [diff] [blame] | 46 | |
Tim Murray | b769c8d | 2015-06-08 14:56:29 -0700 | [diff] [blame] | 47 | switch (policy) { |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 48 | case SP_BACKGROUND: |
Suren Baghdasaryan | 8a315d2 | 2019-02-14 14:40:41 -0800 | [diff] [blame] | 49 | return SetTaskProfiles(tid, |
| 50 | {"HighEnergySaving", "ProcessCapacityLow", "LowIoPriority", |
| 51 | "TimerSlackHigh"}, |
| 52 | true) |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 53 | ? 0 |
| 54 | : -1; |
| 55 | case SP_FOREGROUND: |
| 56 | case SP_AUDIO_APP: |
| 57 | case SP_AUDIO_SYS: |
Suren Baghdasaryan | 8a315d2 | 2019-02-14 14:40:41 -0800 | [diff] [blame] | 58 | return SetTaskProfiles(tid, |
| 59 | {"HighPerformance", "ProcessCapacityHigh", "HighIoPriority", |
| 60 | "TimerSlackNormal"}, |
| 61 | true) |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 62 | ? 0 |
| 63 | : -1; |
| 64 | case SP_TOP_APP: |
Suren Baghdasaryan | 8a315d2 | 2019-02-14 14:40:41 -0800 | [diff] [blame] | 65 | return SetTaskProfiles(tid, |
| 66 | {"MaxPerformance", "ProcessCapacityMax", "MaxIoPriority", |
| 67 | "TimerSlackNormal"}, |
| 68 | true) |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 69 | ? 0 |
| 70 | : -1; |
| 71 | case SP_SYSTEM: |
Suren Baghdasaryan | 8a315d2 | 2019-02-14 14:40:41 -0800 | [diff] [blame] | 72 | return SetTaskProfiles(tid, {"ServiceCapacityLow", "TimerSlackNormal"}, true) ? 0 : -1; |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 73 | case SP_RESTRICTED: |
Suren Baghdasaryan | 8a315d2 | 2019-02-14 14:40:41 -0800 | [diff] [blame] | 74 | return SetTaskProfiles(tid, {"ServiceCapacityRestricted", "TimerSlackNormal"}, true) |
| 75 | ? 0 |
| 76 | : -1; |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 77 | default: |
| 78 | break; |
Todd Kjos | ba8a475 | 2015-10-26 16:22:11 -0700 | [diff] [blame] | 79 | } |
| 80 | |
Tim Murray | 9991026 | 2015-06-22 14:00:56 -0700 | [diff] [blame] | 81 | return 0; |
Tim Murray | b769c8d | 2015-06-08 14:56:29 -0700 | [diff] [blame] | 82 | } |
| 83 | |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 84 | int set_sched_policy(int tid, SchedPolicy policy) { |
Glenn Kasten | 69bfb1f | 2012-03-16 09:43:19 -0700 | [diff] [blame] | 85 | if (tid == 0) { |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 86 | tid = GetThreadId(); |
Glenn Kasten | 69bfb1f | 2012-03-16 09:43:19 -0700 | [diff] [blame] | 87 | } |
Glenn Kasten | 69bfb1f | 2012-03-16 09:43:19 -0700 | [diff] [blame] | 88 | policy = _policy(policy); |
San Mehat | 493dad9 | 2009-09-12 10:06:57 -0700 | [diff] [blame] | 89 | |
San Mehat | d2e4e46 | 2009-10-29 11:48:00 -0700 | [diff] [blame] | 90 | #if POLICY_DEBUG |
| 91 | char statfile[64]; |
| 92 | char statline[1024]; |
| 93 | char thread_name[255]; |
San Mehat | d2e4e46 | 2009-10-29 11:48:00 -0700 | [diff] [blame] | 94 | |
Raja M | a2f37e4 | 2016-04-19 23:55:14 +0530 | [diff] [blame] | 95 | snprintf(statfile, sizeof(statfile), "/proc/%d/stat", tid); |
San Mehat | d2e4e46 | 2009-10-29 11:48:00 -0700 | [diff] [blame] | 96 | memset(thread_name, 0, sizeof(thread_name)); |
| 97 | |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 98 | unique_fd fd(TEMP_FAILURE_RETRY(open(statfile, O_RDONLY | O_CLOEXEC))); |
San Mehat | d2e4e46 | 2009-10-29 11:48:00 -0700 | [diff] [blame] | 99 | if (fd >= 0) { |
| 100 | int rc = read(fd, statline, 1023); |
San Mehat | d2e4e46 | 2009-10-29 11:48:00 -0700 | [diff] [blame] | 101 | statline[rc] = 0; |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 102 | char* p = statline; |
| 103 | char* q; |
San Mehat | d2e4e46 | 2009-10-29 11:48:00 -0700 | [diff] [blame] | 104 | |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 105 | for (p = statline; *p != '('; p++) |
| 106 | ; |
San Mehat | d2e4e46 | 2009-10-29 11:48:00 -0700 | [diff] [blame] | 107 | p++; |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 108 | for (q = p; *q != ')'; q++) |
| 109 | ; |
San Mehat | d2e4e46 | 2009-10-29 11:48:00 -0700 | [diff] [blame] | 110 | |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 111 | strncpy(thread_name, p, (q - p)); |
San Mehat | d2e4e46 | 2009-10-29 11:48:00 -0700 | [diff] [blame] | 112 | } |
Glenn Kasten | 10ec3c7 | 2012-04-19 15:25:58 -0700 | [diff] [blame] | 113 | switch (policy) { |
Tim Murray | b769c8d | 2015-06-08 14:56:29 -0700 | [diff] [blame] | 114 | case SP_BACKGROUND: |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 115 | SLOGD("vvv tid %d (%s)", tid, thread_name); |
Tim Murray | b769c8d | 2015-06-08 14:56:29 -0700 | [diff] [blame] | 116 | break; |
| 117 | case SP_FOREGROUND: |
| 118 | case SP_AUDIO_APP: |
| 119 | case SP_AUDIO_SYS: |
Tim Murray | 6647bb5 | 2016-01-11 16:16:35 -0800 | [diff] [blame] | 120 | case SP_TOP_APP: |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 121 | SLOGD("^^^ tid %d (%s)", tid, thread_name); |
| 122 | break; |
| 123 | case SP_SYSTEM: |
| 124 | SLOGD("/// tid %d (%s)", tid, thread_name); |
Tim Murray | b769c8d | 2015-06-08 14:56:29 -0700 | [diff] [blame] | 125 | break; |
Joel Fernandes | 88ef9f0 | 2017-03-25 22:46:10 -0700 | [diff] [blame] | 126 | case SP_RT_APP: |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 127 | SLOGD("RT tid %d (%s)", tid, thread_name); |
Tim Murray | b769c8d | 2015-06-08 14:56:29 -0700 | [diff] [blame] | 128 | break; |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 129 | default: |
| 130 | SLOGD("??? tid %d (%s)", tid, thread_name); |
| 131 | break; |
| 132 | } |
| 133 | #endif |
Tim Murray | b769c8d | 2015-06-08 14:56:29 -0700 | [diff] [blame] | 134 | |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 135 | switch (policy) { |
| 136 | case SP_BACKGROUND: |
Suren Baghdasaryan | 8a315d2 | 2019-02-14 14:40:41 -0800 | [diff] [blame] | 137 | return SetTaskProfiles(tid, {"HighEnergySaving", "TimerSlackHigh"}, true) ? 0 : -1; |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 138 | case SP_FOREGROUND: |
| 139 | case SP_AUDIO_APP: |
| 140 | case SP_AUDIO_SYS: |
Suren Baghdasaryan | 8a315d2 | 2019-02-14 14:40:41 -0800 | [diff] [blame] | 141 | return SetTaskProfiles(tid, {"HighPerformance", "TimerSlackNormal"}, true) ? 0 : -1; |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 142 | case SP_TOP_APP: |
Suren Baghdasaryan | 8a315d2 | 2019-02-14 14:40:41 -0800 | [diff] [blame] | 143 | return SetTaskProfiles(tid, {"MaxPerformance", "TimerSlackNormal"}, true) ? 0 : -1; |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 144 | case SP_RT_APP: |
Suren Baghdasaryan | 8a315d2 | 2019-02-14 14:40:41 -0800 | [diff] [blame] | 145 | return SetTaskProfiles(tid, {"RealtimePerformance", "TimerSlackNormal"}, true) ? 0 : -1; |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 146 | default: |
Suren Baghdasaryan | 8a315d2 | 2019-02-14 14:40:41 -0800 | [diff] [blame] | 147 | return SetTaskProfiles(tid, {"TimerSlackNormal"}, true) ? 0 : -1; |
San Mehat | 493dad9 | 2009-09-12 10:06:57 -0700 | [diff] [blame] | 148 | } |
| 149 | |
| 150 | return 0; |
| 151 | } |
Raphael | 0384a98 | 2009-09-15 17:10:17 -0700 | [diff] [blame] | 152 | |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 153 | bool cpusets_enabled() { |
Suren Baghdasaryan | fa7a05f | 2019-05-08 17:59:55 -0700 | [diff] [blame] | 154 | static bool enabled = (CgroupMap::GetInstance().FindController("cpuset").IsUsable()); |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 155 | return enabled; |
| 156 | } |
Jeff Brown | bff8f3f | 2012-05-08 15:05:42 -0700 | [diff] [blame] | 157 | |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 158 | bool schedboost_enabled() { |
Suren Baghdasaryan | fa7a05f | 2019-05-08 17:59:55 -0700 | [diff] [blame] | 159 | static bool enabled = (CgroupMap::GetInstance().FindController("schedtune").IsUsable()); |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 160 | return enabled; |
| 161 | } |
Jeff Brown | bff8f3f | 2012-05-08 15:05:42 -0700 | [diff] [blame] | 162 | |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 163 | static int getCGroupSubsys(int tid, const char* subsys, std::string& subgroup) { |
Yifan Hong | 53e0deb | 2019-03-22 17:01:08 -0700 | [diff] [blame] | 164 | auto controller = CgroupMap::GetInstance().FindController(subsys); |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 165 | |
Suren Baghdasaryan | fa7a05f | 2019-05-08 17:59:55 -0700 | [diff] [blame] | 166 | if (!controller.IsUsable()) return -1; |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 167 | |
Yifan Hong | 53e0deb | 2019-03-22 17:01:08 -0700 | [diff] [blame] | 168 | if (!controller.GetTaskGroup(tid, &subgroup)) { |
Wei Wang | d71d301 | 2019-03-07 11:59:12 -0800 | [diff] [blame] | 169 | LOG(ERROR) << "Failed to find cgroup for tid " << tid; |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 170 | return -1; |
| 171 | } |
Jeff Brown | bff8f3f | 2012-05-08 15:05:42 -0700 | [diff] [blame] | 172 | return 0; |
| 173 | } |
| 174 | |
Suren Baghdasaryan | 82b72a5 | 2018-12-21 11:41:50 -0800 | [diff] [blame] | 175 | int get_sched_policy(int tid, SchedPolicy* policy) { |
| 176 | if (tid == 0) { |
| 177 | tid = GetThreadId(); |
| 178 | } |
| 179 | |
| 180 | std::string group; |
| 181 | if (schedboost_enabled()) { |
| 182 | if (getCGroupSubsys(tid, "schedtune", group) < 0) return -1; |
| 183 | } |
| 184 | if (group.empty() && cpusets_enabled()) { |
| 185 | if (getCGroupSubsys(tid, "cpuset", group) < 0) return -1; |
| 186 | } |
| 187 | |
| 188 | // TODO: replace hardcoded directories |
| 189 | if (group.empty()) { |
| 190 | *policy = SP_FOREGROUND; |
| 191 | } else if (group == "foreground") { |
| 192 | *policy = SP_FOREGROUND; |
| 193 | } else if (group == "system-background") { |
| 194 | *policy = SP_SYSTEM; |
| 195 | } else if (group == "background") { |
| 196 | *policy = SP_BACKGROUND; |
| 197 | } else if (group == "top-app") { |
| 198 | *policy = SP_TOP_APP; |
| 199 | } else if (group == "restricted") { |
| 200 | *policy = SP_RESTRICTED; |
| 201 | } else { |
| 202 | errno = ERANGE; |
| 203 | return -1; |
| 204 | } |
Jeff Brown | bff8f3f | 2012-05-08 15:05:42 -0700 | [diff] [blame] | 205 | return 0; |
| 206 | } |
| 207 | |
Suren Baghdasaryan | eca87cb | 2019-02-02 14:19:41 -0800 | [diff] [blame] | 208 | #else |
| 209 | |
| 210 | /* Stubs for non-Android targets. */ |
| 211 | |
| 212 | int set_sched_policy(int, SchedPolicy) { |
| 213 | return 0; |
| 214 | } |
| 215 | |
| 216 | int get_sched_policy(int, SchedPolicy* policy) { |
| 217 | *policy = SP_SYSTEM_DEFAULT; |
| 218 | return 0; |
| 219 | } |
| 220 | |
| 221 | #endif |
| 222 | |
Elliott Hughes | 9f49508 | 2018-04-25 14:52:50 -0700 | [diff] [blame] | 223 | const char* get_sched_policy_name(SchedPolicy policy) { |
Glenn Kasten | 69bfb1f | 2012-03-16 09:43:19 -0700 | [diff] [blame] | 224 | policy = _policy(policy); |
Elliott Hughes | 9f49508 | 2018-04-25 14:52:50 -0700 | [diff] [blame] | 225 | static const char* const kSchedPolicyNames[] = { |
Tim Murray | 419ba9e | 2018-04-13 10:15:49 -0700 | [diff] [blame] | 226 | [SP_BACKGROUND] = "bg", [SP_FOREGROUND] = "fg", [SP_SYSTEM] = " ", |
| 227 | [SP_AUDIO_APP] = "aa", [SP_AUDIO_SYS] = "as", [SP_TOP_APP] = "ta", |
| 228 | [SP_RT_APP] = "rt", [SP_RESTRICTED] = "rs", |
Glenn Kasten | 86c7cc8 | 2012-03-05 16:14:39 -0800 | [diff] [blame] | 229 | }; |
Elliott Hughes | 9f49508 | 2018-04-25 14:52:50 -0700 | [diff] [blame] | 230 | static_assert(arraysize(kSchedPolicyNames) == SP_CNT, "missing name"); |
| 231 | if (policy < SP_BACKGROUND || policy >= SP_CNT) { |
Glenn Kasten | 86c7cc8 | 2012-03-05 16:14:39 -0800 | [diff] [blame] | 232 | return "error"; |
Elliott Hughes | 9f49508 | 2018-04-25 14:52:50 -0700 | [diff] [blame] | 233 | } |
| 234 | return kSchedPolicyNames[policy]; |
Glenn Kasten | 86c7cc8 | 2012-03-05 16:14:39 -0800 | [diff] [blame] | 235 | } |