1 /* 2 * Copyright 2014 Google, Inc 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 <sys/cdefs.h> 20 #include <sys/types.h> 21 #include <string> 22 #include <vector> 23 24 __BEGIN_DECLS 25 26 static constexpr const char* CGROUPV2_CONTROLLER_NAME = "cgroup2"; 27 28 bool CgroupGetControllerPath(const std::string& cgroup_name, std::string* path); 29 bool CgroupGetControllerFromPath(const std::string& path, std::string* cgroup_name); 30 bool CgroupGetAttributePath(const std::string& attr_name, std::string* path); 31 bool CgroupGetAttributePathForTask(const std::string& attr_name, int tid, std::string* path); 32 33 bool SetTaskProfiles(int tid, const std::vector<std::string>& profiles, bool use_fd_cache = false); 34 bool SetProcessProfiles(uid_t uid, pid_t pid, const std::vector<std::string>& profiles); 35 36 #ifndef __ANDROID_VNDK__ 37 38 bool SetProcessProfilesCached(uid_t uid, pid_t pid, const std::vector<std::string>& profiles); 39 40 static constexpr const char* CGROUPS_RC_PATH = "/dev/cgroup_info/cgroup.rc"; 41 42 bool UsePerAppMemcg(); 43 44 // Drop the fd cache of cgroup path. It is used for when resource caching is enabled and a process 45 // loses the access to the path, the access checking (See SetCgroupAction::EnableResourceCaching) 46 // should be active again. E.g. Zygote specialization for child process. 47 void DropTaskProfilesResourceCaching(); 48 49 // Return 0 and removes the cgroup if there are no longer any processes in it. 50 // Returns -1 in the case of an error occurring or if there are processes still running 51 // even after retrying for up to 200ms. 52 // If max_processes is not nullptr, it returns the maximum number of processes seen in the cgroup 53 // during the killing process. Note that this can be 0 if all processes from the process group have 54 // already been terminated. 55 int killProcessGroup(uid_t uid, int initialPid, int signal, int* max_processes = nullptr); 56 57 // Returns the same as killProcessGroup(), however it does not retry, which means 58 // that it only returns 0 in the case that the cgroup exists and it contains no processes. 59 int killProcessGroupOnce(uid_t uid, int initialPid, int signal, int* max_processes = nullptr); 60 61 int createProcessGroup(uid_t uid, int initialPid, bool memControl = false); 62 63 // Set various properties of a process group. For these functions to work, the process group must 64 // have been created by passing memControl=true to createProcessGroup. 65 bool setProcessGroupSwappiness(uid_t uid, int initialPid, int swappiness); 66 bool setProcessGroupSoftLimit(uid_t uid, int initialPid, int64_t softLimitInBytes); 67 bool setProcessGroupLimit(uid_t uid, int initialPid, int64_t limitInBytes); 68 69 void removeAllProcessGroups(void); 70 void removeAllEmptyProcessGroups(void); 71 72 // Provides the path for an attribute in a specific process group 73 // Returns false in case of error, true in case of success 74 bool getAttributePathForTask(const std::string& attr_name, int tid, std::string* path); 75 76 #endif // __ANDROID_VNDK__ 77 78 __END_DECLS 79