15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch#include "base/process/process.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <errno.h> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/resource.h> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/file_util.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/lazy_instance.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/strings/string_split.h" 14868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/synchronization/lock.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 17a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)namespace base { 18a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kForegroundPriority = 0; 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_CHROMEOS) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// We are more aggressive in our lowering of background process priority 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// for chromeos as we have much more control over other processes running 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// on the machine. 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// TODO(davemoore) Refactor this by adding support for higher levels to set 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the foregrounding / backgrounding process so we don't have to keep 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// chrome / chromeos specific logic here. 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kBackgroundPriority = 19; 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kControlPath[] = "/sys/fs/cgroup/cpu%s/cgroup.procs"; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kForeground[] = "/chrome_renderers/foreground"; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kBackground[] = "/chrome_renderers/background"; 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kProcPath[] = "/proc/%d/cgroup"; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)struct CGroups { 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Check for cgroups files. ChromeOS supports these by default. It creates 382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // a cgroup mount in /sys/fs/cgroup and then configures two cpu task groups, 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // one contains at most a single foreground renderer and the other contains 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // all background renderers. This allows us to limit the impact of background 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // renderers on foreground ones to a greater level than simple renicing. 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool enabled; 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath foreground_file; 442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath background_file; 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CGroups() { 472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) foreground_file = 482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath(base::StringPrintf(kControlPath, kForeground)); 492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) background_file = 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath(base::StringPrintf(kControlPath, kBackground)); 51a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) base::FileSystemType foreground_type; 52a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) base::FileSystemType background_type; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) enabled = 54a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) base::GetFileSystemType(foreground_file, &foreground_type) && 55a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) base::GetFileSystemType(background_file, &background_type) && 56a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) foreground_type == FILE_SYSTEM_CGROUP && 57a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) background_type == FILE_SYSTEM_CGROUP; 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::LazyInstance<CGroups> cgroups = LAZY_INSTANCE_INITIALIZER; 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kBackgroundPriority = 5; 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool Process::IsProcessBackgrounded() const { 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(process_); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_CHROMEOS) 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (cgroups.Get().enabled) { 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string proc; 7358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) if (base::ReadFileToString( 7458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) base::FilePath(StringPrintf(kProcPath, process_)), 7558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) &proc)) { 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<std::string> proc_parts; 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::SplitString(proc, ':', &proc_parts); 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(proc_parts.size() == 3); 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool ret = proc_parts[2] == std::string(kBackground); 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return ret; 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return GetPriority() == kBackgroundPriority; 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool Process::SetProcessBackgrounded(bool background) { 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(process_); 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_CHROMEOS) 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (cgroups.Get().enabled) { 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string pid = StringPrintf("%d", process_); 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath file = 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) background ? 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cgroups.Get().background_file : cgroups.Get().foreground_file; 98a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) return base::WriteFile(file, pid.c_str(), pid.size()) > 0; 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // OS_CHROMEOS 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!CanBackgroundProcesses()) 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int priority = background ? kBackgroundPriority : kForegroundPriority; 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int result = setpriority(PRIO_PROCESS, process_, priority); 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DPCHECK(result == 0); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return result == 0; 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)struct CheckForNicePermission { 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CheckForNicePermission() : can_reraise_priority(false) { 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We won't be able to raise the priority if we don't have the right rlimit. 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The limit may be adjusted in /etc/security/limits.conf for PAM systems. 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) struct rlimit rlim; 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if ((getrlimit(RLIMIT_NICE, &rlim) == 0) && 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (20 - kForegroundPriority) <= static_cast<int>(rlim.rlim_cur)) { 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) can_reraise_priority = true; 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool can_reraise_priority; 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool Process::CanBackgroundProcesses() { 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_CHROMEOS) 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (cgroups.Get().enabled) 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static LazyInstance<CheckForNicePermission> check_for_nice_permission = 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LAZY_INSTANCE_INITIALIZER; 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return check_for_nice_permission.Get().can_reraise_priority; 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace base 138