sched_policy.c revision 69bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4d
10384a98d6f61fa47ffde01c15e552f738e88fb94Raphael 2493dad966305a7fb555addd367532dd2af275a27San Mehat/* libs/cutils/sched_policy.c 3493dad966305a7fb555addd367532dd2af275a27San Mehat** 4493dad966305a7fb555addd367532dd2af275a27San Mehat** Copyright 2007, The Android Open Source Project 5493dad966305a7fb555addd367532dd2af275a27San Mehat** 6493dad966305a7fb555addd367532dd2af275a27San Mehat** Licensed under the Apache License, Version 2.0 (the "License"); 7493dad966305a7fb555addd367532dd2af275a27San Mehat** you may not use this file except in compliance with the License. 8493dad966305a7fb555addd367532dd2af275a27San Mehat** You may obtain a copy of the License at 9493dad966305a7fb555addd367532dd2af275a27San Mehat** 10493dad966305a7fb555addd367532dd2af275a27San Mehat** http://www.apache.org/licenses/LICENSE-2.0 11493dad966305a7fb555addd367532dd2af275a27San Mehat** 12493dad966305a7fb555addd367532dd2af275a27San Mehat** Unless required by applicable law or agreed to in writing, software 13493dad966305a7fb555addd367532dd2af275a27San Mehat** distributed under the License is distributed on an "AS IS" BASIS, 14493dad966305a7fb555addd367532dd2af275a27San Mehat** WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15493dad966305a7fb555addd367532dd2af275a27San Mehat** See the License for the specific language governing permissions and 16493dad966305a7fb555addd367532dd2af275a27San Mehat** limitations under the License. 17493dad966305a7fb555addd367532dd2af275a27San Mehat*/ 18493dad966305a7fb555addd367532dd2af275a27San Mehat 19493dad966305a7fb555addd367532dd2af275a27San Mehat#include <stdio.h> 20493dad966305a7fb555addd367532dd2af275a27San Mehat#include <stdlib.h> 21493dad966305a7fb555addd367532dd2af275a27San Mehat#include <unistd.h> 22493dad966305a7fb555addd367532dd2af275a27San Mehat#include <string.h> 23493dad966305a7fb555addd367532dd2af275a27San Mehat#include <errno.h> 24493dad966305a7fb555addd367532dd2af275a27San Mehat#include <fcntl.h> 250384a98d6f61fa47ffde01c15e552f738e88fb94Raphael 26d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat#define LOG_TAG "SchedPolicy" 27d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat#include "cutils/log.h" 28d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat 290384a98d6f61fa47ffde01c15e552f738e88fb94Raphael#ifdef HAVE_SCHED_H 3086b121523d7d3634bf5ac07b6e238b36b3e75f0aBrad Fitzpatrick#ifdef HAVE_PTHREADS 310384a98d6f61fa47ffde01c15e552f738e88fb94Raphael 32493dad966305a7fb555addd367532dd2af275a27San Mehat#include <sched.h> 3386b121523d7d3634bf5ac07b6e238b36b3e75f0aBrad Fitzpatrick#include <pthread.h> 34493dad966305a7fb555addd367532dd2af275a27San Mehat 35493dad966305a7fb555addd367532dd2af275a27San Mehat#include <cutils/sched_policy.h> 36493dad966305a7fb555addd367532dd2af275a27San Mehat 373cd5b66ba05cffe38bfa4e2da9b93292ba0b073aSan Mehat#ifndef SCHED_NORMAL 383cd5b66ba05cffe38bfa4e2da9b93292ba0b073aSan Mehat #define SCHED_NORMAL 0 393cd5b66ba05cffe38bfa4e2da9b93292ba0b073aSan Mehat#endif 403cd5b66ba05cffe38bfa4e2da9b93292ba0b073aSan Mehat 413cd5b66ba05cffe38bfa4e2da9b93292ba0b073aSan Mehat#ifndef SCHED_BATCH 423cd5b66ba05cffe38bfa4e2da9b93292ba0b073aSan Mehat #define SCHED_BATCH 3 433cd5b66ba05cffe38bfa4e2da9b93292ba0b073aSan Mehat#endif 443cd5b66ba05cffe38bfa4e2da9b93292ba0b073aSan Mehat 45805d67ab11c1bb21d494459d6167f5f67f0002ebSan Mehat#define POLICY_DEBUG 0 46d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat 47e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrickstatic pthread_once_t the_once = PTHREAD_ONCE_INIT; 48e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick 49c0dfca7ae1f6016461235552091c2cacacca82a2San Mehatstatic int __sys_supports_schedgroups = -1; 50c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat 51e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick// File descriptors open to /dev/cpuctl/../tasks, setup by initialize, or -1 on error. 52e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrickstatic int normal_cgroup_fd = -1; 53e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrickstatic int bg_cgroup_fd = -1; 54e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick 55e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick/* Add tid to the scheduling group defined by the policy */ 56e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrickstatic int add_tid_to_cgroup(int tid, SchedPolicy policy) 57493dad966305a7fb555addd367532dd2af275a27San Mehat{ 58493dad966305a7fb555addd367532dd2af275a27San Mehat int fd; 59e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick 60e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick if (policy == SP_BACKGROUND) { 61e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick fd = bg_cgroup_fd; 62e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick } else { 63e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick fd = normal_cgroup_fd; 64e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick } 65e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick 66e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick if (fd < 0) { 67e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick SLOGE("add_tid_to_cgroup failed; background=%d\n", 68e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick policy == SP_BACKGROUND ? 1 : 0); 69493dad966305a7fb555addd367532dd2af275a27San Mehat return -1; 70805d67ab11c1bb21d494459d6167f5f67f0002ebSan Mehat } 71493dad966305a7fb555addd367532dd2af275a27San Mehat 72253e27acb6c9e3e0b03c59f25cdf9ecbd64bcdadBrad Fitzpatrick // specialized itoa -- works for tid > 0 73253e27acb6c9e3e0b03c59f25cdf9ecbd64bcdadBrad Fitzpatrick char text[22]; 74253e27acb6c9e3e0b03c59f25cdf9ecbd64bcdadBrad Fitzpatrick char *end = text + sizeof(text) - 1; 75253e27acb6c9e3e0b03c59f25cdf9ecbd64bcdadBrad Fitzpatrick char *ptr = end; 76253e27acb6c9e3e0b03c59f25cdf9ecbd64bcdadBrad Fitzpatrick *ptr = '\0'; 77253e27acb6c9e3e0b03c59f25cdf9ecbd64bcdadBrad Fitzpatrick while (tid > 0) { 78253e27acb6c9e3e0b03c59f25cdf9ecbd64bcdadBrad Fitzpatrick *--ptr = '0' + (tid % 10); 79253e27acb6c9e3e0b03c59f25cdf9ecbd64bcdadBrad Fitzpatrick tid = tid / 10; 80253e27acb6c9e3e0b03c59f25cdf9ecbd64bcdadBrad Fitzpatrick } 81253e27acb6c9e3e0b03c59f25cdf9ecbd64bcdadBrad Fitzpatrick 82253e27acb6c9e3e0b03c59f25cdf9ecbd64bcdadBrad Fitzpatrick if (write(fd, ptr, end - ptr) < 0) { 83e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick /* 84e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick * If the thread is in the process of exiting, 85e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick * don't flag an error 86e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick */ 87e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick if (errno == ESRCH) 88e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick return 0; 89e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick SLOGW("add_tid_to_cgroup failed to write '%s' (%s); background=%d\n", 90e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick ptr, strerror(errno), policy == SP_BACKGROUND ? 1 : 0); 91493dad966305a7fb555addd367532dd2af275a27San Mehat return -1; 92493dad966305a7fb555addd367532dd2af275a27San Mehat } 93493dad966305a7fb555addd367532dd2af275a27San Mehat 94493dad966305a7fb555addd367532dd2af275a27San Mehat return 0; 95493dad966305a7fb555addd367532dd2af275a27San Mehat} 96493dad966305a7fb555addd367532dd2af275a27San Mehat 97e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrickstatic void __initialize(void) { 98e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick char* filename; 99e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick if (!access("/dev/cpuctl/tasks", F_OK)) { 100e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick __sys_supports_schedgroups = 1; 101e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick 102e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick filename = "/dev/cpuctl/tasks"; 103e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick normal_cgroup_fd = open(filename, O_WRONLY); 104e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick if (normal_cgroup_fd < 0) { 105e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick SLOGE("open of %s failed: %s\n", filename, strerror(errno)); 106e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick } 107e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick 108e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick filename = "/dev/cpuctl/bg_non_interactive/tasks"; 109e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick bg_cgroup_fd = open(filename, O_WRONLY); 110e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick if (bg_cgroup_fd < 0) { 111e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick SLOGE("open of %s failed: %s\n", filename, strerror(errno)); 112493dad966305a7fb555addd367532dd2af275a27San Mehat } 113e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick } else { 114e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick __sys_supports_schedgroups = 0; 115493dad966305a7fb555addd367532dd2af275a27San Mehat } 116c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat} 117c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat 118c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat/* 119c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat * Try to get the scheduler group. 120c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat * 121503df2075991cd03ddf43d14e05768a2138b9028San Mehat * The data from /proc/<pid>/cgroup looks (something) like: 122c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat * 2:cpu:/bg_non_interactive 123503df2075991cd03ddf43d14e05768a2138b9028San Mehat * 1:cpuacct:/ 124c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat * 125c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat * We return the part after the "/", which will be an empty string for 126c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat * the default cgroup. If the string is longer than "bufLen", the string 127c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat * will be truncated. 128c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat */ 129c0dfca7ae1f6016461235552091c2cacacca82a2San Mehatstatic int getSchedulerGroup(int tid, char* buf, size_t bufLen) 130c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat{ 131c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat#ifdef HAVE_ANDROID_OS 132c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat char pathBuf[32]; 133503df2075991cd03ddf43d14e05768a2138b9028San Mehat char lineBuf[256]; 134503df2075991cd03ddf43d14e05768a2138b9028San Mehat FILE *fp; 135c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat 136c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat snprintf(pathBuf, sizeof(pathBuf), "/proc/%d/cgroup", tid); 137503df2075991cd03ddf43d14e05768a2138b9028San Mehat if (!(fp = fopen(pathBuf, "r"))) { 138c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat return -1; 139c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat } 140c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat 141503df2075991cd03ddf43d14e05768a2138b9028San Mehat while(fgets(lineBuf, sizeof(lineBuf) -1, fp)) { 142503df2075991cd03ddf43d14e05768a2138b9028San Mehat char *next = lineBuf; 143503df2075991cd03ddf43d14e05768a2138b9028San Mehat char *subsys; 144503df2075991cd03ddf43d14e05768a2138b9028San Mehat char *grp; 145503df2075991cd03ddf43d14e05768a2138b9028San Mehat size_t len; 146c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat 147503df2075991cd03ddf43d14e05768a2138b9028San Mehat /* Junk the first field */ 148503df2075991cd03ddf43d14e05768a2138b9028San Mehat if (!strsep(&next, ":")) { 149503df2075991cd03ddf43d14e05768a2138b9028San Mehat goto out_bad_data; 150503df2075991cd03ddf43d14e05768a2138b9028San Mehat } 151c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat 152503df2075991cd03ddf43d14e05768a2138b9028San Mehat if (!(subsys = strsep(&next, ":"))) { 153503df2075991cd03ddf43d14e05768a2138b9028San Mehat goto out_bad_data; 154503df2075991cd03ddf43d14e05768a2138b9028San Mehat } 155503df2075991cd03ddf43d14e05768a2138b9028San Mehat 156503df2075991cd03ddf43d14e05768a2138b9028San Mehat if (strcmp(subsys, "cpu")) { 157503df2075991cd03ddf43d14e05768a2138b9028San Mehat /* Not the subsys we're looking for */ 158503df2075991cd03ddf43d14e05768a2138b9028San Mehat continue; 159503df2075991cd03ddf43d14e05768a2138b9028San Mehat } 160503df2075991cd03ddf43d14e05768a2138b9028San Mehat 161503df2075991cd03ddf43d14e05768a2138b9028San Mehat if (!(grp = strsep(&next, ":"))) { 162503df2075991cd03ddf43d14e05768a2138b9028San Mehat goto out_bad_data; 163503df2075991cd03ddf43d14e05768a2138b9028San Mehat } 164503df2075991cd03ddf43d14e05768a2138b9028San Mehat grp++; /* Drop the leading '/' */ 165503df2075991cd03ddf43d14e05768a2138b9028San Mehat len = strlen(grp); 166503df2075991cd03ddf43d14e05768a2138b9028San Mehat grp[len-1] = '\0'; /* Drop the trailing '\n' */ 167503df2075991cd03ddf43d14e05768a2138b9028San Mehat 168503df2075991cd03ddf43d14e05768a2138b9028San Mehat if (bufLen <= len) { 169503df2075991cd03ddf43d14e05768a2138b9028San Mehat len = bufLen - 1; 170503df2075991cd03ddf43d14e05768a2138b9028San Mehat } 171503df2075991cd03ddf43d14e05768a2138b9028San Mehat strncpy(buf, grp, len); 172503df2075991cd03ddf43d14e05768a2138b9028San Mehat buf[len] = '\0'; 173503df2075991cd03ddf43d14e05768a2138b9028San Mehat fclose(fp); 174503df2075991cd03ddf43d14e05768a2138b9028San Mehat return 0; 175c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat } 176c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat 1777e8529a8b528fd30586aa037f15a31b29582c537San Mehat SLOGE("Failed to find cpu subsys"); 178503df2075991cd03ddf43d14e05768a2138b9028San Mehat fclose(fp); 179503df2075991cd03ddf43d14e05768a2138b9028San Mehat return -1; 180503df2075991cd03ddf43d14e05768a2138b9028San Mehat out_bad_data: 1817e8529a8b528fd30586aa037f15a31b29582c537San Mehat SLOGE("Bad cgroup data {%s}", lineBuf); 182503df2075991cd03ddf43d14e05768a2138b9028San Mehat fclose(fp); 183503df2075991cd03ddf43d14e05768a2138b9028San Mehat return -1; 184c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat#else 185c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat errno = ENOSYS; 186c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat return -1; 187c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat#endif 188c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat} 189c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat 190c0dfca7ae1f6016461235552091c2cacacca82a2San Mehatint get_sched_policy(int tid, SchedPolicy *policy) 191c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat{ 19269bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten#ifdef HAVE_GETTID 19369bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten if (tid == 0) { 19469bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten tid = gettid(); 19569bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten } 19669bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten#endif 197e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick pthread_once(&the_once, __initialize); 198c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat 199c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat if (__sys_supports_schedgroups) { 200c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat char grpBuf[32]; 201c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat if (getSchedulerGroup(tid, grpBuf, sizeof(grpBuf)) < 0) 202c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat return -1; 203c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat if (grpBuf[0] == '\0') { 204c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat *policy = SP_FOREGROUND; 205c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat } else if (!strcmp(grpBuf, "bg_non_interactive")) { 206c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat *policy = SP_BACKGROUND; 207c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat } else { 208c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat errno = ERANGE; 209c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat return -1; 210c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat } 211c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat } else { 212c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat int rc = sched_getscheduler(tid); 213c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat if (rc < 0) 214c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat return -1; 215c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat else if (rc == SCHED_NORMAL) 216c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat *policy = SP_FOREGROUND; 217c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat else if (rc == SCHED_BATCH) 218c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat *policy = SP_BACKGROUND; 219c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat else { 220c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat errno = ERANGE; 221c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat return -1; 222c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat } 223c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat } 224c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat return 0; 225c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat} 226c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat 22769bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten/* Re-map SP_DEFAULT to the system default policy, and leave other values unchanged. 22869bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten * Call this any place a SchedPolicy is used as an input parameter. 22969bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten * Returns the possibly re-mapped policy. 23069bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten */ 23169bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kastenstatic inline SchedPolicy _policy(SchedPolicy p) 23269bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten{ 23369bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten return p == SP_DEFAULT ? SP_SYSTEM_DEFAULT : p; 23469bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten} 23569bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten 236c0dfca7ae1f6016461235552091c2cacacca82a2San Mehatint set_sched_policy(int tid, SchedPolicy policy) 237c0dfca7ae1f6016461235552091c2cacacca82a2San Mehat{ 23869bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten#ifdef HAVE_GETTID 23969bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten if (tid == 0) { 24069bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten tid = gettid(); 24169bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten } 24269bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten#endif 24369bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten policy = _policy(policy); 244e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick pthread_once(&the_once, __initialize); 245493dad966305a7fb555addd367532dd2af275a27San Mehat 246d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat#if POLICY_DEBUG 247d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat char statfile[64]; 248d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat char statline[1024]; 249d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat char thread_name[255]; 250d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat int fd; 251d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat 252d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat sprintf(statfile, "/proc/%d/stat", tid); 253d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat memset(thread_name, 0, sizeof(thread_name)); 254d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat 255d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat fd = open(statfile, O_RDONLY); 256d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat if (fd >= 0) { 257d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat int rc = read(fd, statline, 1023); 258d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat close(fd); 259d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat statline[rc] = 0; 260d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat char *p = statline; 261d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat char *q; 262d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat 263d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat for (p = statline; *p != '('; p++); 264d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat p++; 265d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat for (q = p; *q != ')'; q++); 266d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat 267d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat strncpy(thread_name, p, (q-p)); 268d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat } 269d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat if (policy == SP_BACKGROUND) { 2707e8529a8b528fd30586aa037f15a31b29582c537San Mehat SLOGD("vvv tid %d (%s)", tid, thread_name); 271d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat } else if (policy == SP_FOREGROUND) { 2727e8529a8b528fd30586aa037f15a31b29582c537San Mehat SLOGD("^^^ tid %d (%s)", tid, thread_name); 273d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat } else { 2747e8529a8b528fd30586aa037f15a31b29582c537San Mehat SLOGD("??? tid %d (%s)", tid, thread_name); 275d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat } 276d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat#endif 277d2e4e46ce4bc29c3328fccfbbbe0590eaa8713f7San Mehat 278493dad966305a7fb555addd367532dd2af275a27San Mehat if (__sys_supports_schedgroups) { 279e43c248316d0faa7bdc23de823b068c2aa3520abBrad Fitzpatrick if (add_tid_to_cgroup(tid, policy)) { 280493dad966305a7fb555addd367532dd2af275a27San Mehat if (errno != ESRCH && errno != ENOENT) 281493dad966305a7fb555addd367532dd2af275a27San Mehat return -errno; 282493dad966305a7fb555addd367532dd2af275a27San Mehat } 283493dad966305a7fb555addd367532dd2af275a27San Mehat } else { 284493dad966305a7fb555addd367532dd2af275a27San Mehat struct sched_param param; 285493dad966305a7fb555addd367532dd2af275a27San Mehat 286493dad966305a7fb555addd367532dd2af275a27San Mehat param.sched_priority = 0; 287493dad966305a7fb555addd367532dd2af275a27San Mehat sched_setscheduler(tid, 288493dad966305a7fb555addd367532dd2af275a27San Mehat (policy == SP_BACKGROUND) ? 289493dad966305a7fb555addd367532dd2af275a27San Mehat SCHED_BATCH : SCHED_NORMAL, 290493dad966305a7fb555addd367532dd2af275a27San Mehat ¶m); 291493dad966305a7fb555addd367532dd2af275a27San Mehat } 292493dad966305a7fb555addd367532dd2af275a27San Mehat 293493dad966305a7fb555addd367532dd2af275a27San Mehat return 0; 294493dad966305a7fb555addd367532dd2af275a27San Mehat} 2950384a98d6f61fa47ffde01c15e552f738e88fb94Raphael 29686c7cc81891a69ace7044de667b0624c284ee82bGlenn Kastenconst char *get_sched_policy_name(SchedPolicy policy) 29786c7cc81891a69ace7044de667b0624c284ee82bGlenn Kasten{ 29869bfb1f099164ae7ba44b5d0f7ba7c47cbbafd4dGlenn Kasten policy = _policy(policy); 29986c7cc81891a69ace7044de667b0624c284ee82bGlenn Kasten static const char * const strings[SP_CNT] = { 30086c7cc81891a69ace7044de667b0624c284ee82bGlenn Kasten [SP_BACKGROUND] = "bg", 30186c7cc81891a69ace7044de667b0624c284ee82bGlenn Kasten [SP_FOREGROUND] = "fg", 30286c7cc81891a69ace7044de667b0624c284ee82bGlenn Kasten }; 30386c7cc81891a69ace7044de667b0624c284ee82bGlenn Kasten if ((policy < SP_CNT) && (strings[policy] != NULL)) 30486c7cc81891a69ace7044de667b0624c284ee82bGlenn Kasten return strings[policy]; 30586c7cc81891a69ace7044de667b0624c284ee82bGlenn Kasten else 30686c7cc81891a69ace7044de667b0624c284ee82bGlenn Kasten return "error"; 30786c7cc81891a69ace7044de667b0624c284ee82bGlenn Kasten} 30886c7cc81891a69ace7044de667b0624c284ee82bGlenn Kasten 30986b121523d7d3634bf5ac07b6e238b36b3e75f0aBrad Fitzpatrick#endif /* HAVE_PTHREADS */ 3100384a98d6f61fa47ffde01c15e552f738e88fb94Raphael#endif /* HAVE_SCHED_H */ 311