diff options
Diffstat (limited to 'zen/process_priority.cpp')
| -rw-r--r-- | zen/process_priority.cpp | 149 |
1 files changed, 149 insertions, 0 deletions
diff --git a/zen/process_priority.cpp b/zen/process_priority.cpp new file mode 100644 index 00000000..96a6c2de --- /dev/null +++ b/zen/process_priority.cpp @@ -0,0 +1,149 @@ +// ************************************************************************** +// * This file is part of the FreeFileSync project. It is distributed under * +// * GNU General Public License: http://www.gnu.org/licenses/gpl.html * +// * Copyright (C) Zenju (zenju AT gmx DOT de) - All Rights Reserved * +// ************************************************************************** + +#include "process_priority.h" +#include <zen/last_error.h> +#include <zen/i18n.h> + +#ifdef FFS_WIN +#include "win.h" //includes "windows.h" + +#elif defined FFS_LINUX +//#include <sys/syscall.h> + +#elif defined FFS_MAC +#include <sys/resource.h> //getiopolicy_np +#include <IOKit/pwr_mgt/IOPMLib.h> //keep in .cpp file to not pollute global namespace! e.g. with UInt64 +#endif + +using namespace zen; + + +#ifdef FFS_WIN +struct PreventStandby::Pimpl {}; + +PreventStandby::PreventStandby() +{ + if (::SetThreadExecutionState(ES_CONTINUOUS | ES_SYSTEM_REQUIRED) == 0) + throw FileError(_("Failed to suspend system sleep mode.")); //no GetLastError() support? +} + + +PreventStandby::~PreventStandby() +{ + ::SetThreadExecutionState(ES_CONTINUOUS); +} + + +#ifndef PROCESS_MODE_BACKGROUND_BEGIN +#define PROCESS_MODE_BACKGROUND_BEGIN 0x00100000 // Windows Server 2003 and Windows XP/2000: This value is not supported! +#define PROCESS_MODE_BACKGROUND_END 0x00200000 // +#endif + +struct ScheduleForBackgroundProcessing::Pimpl {}; + + +ScheduleForBackgroundProcessing::ScheduleForBackgroundProcessing() +{ + if (!::SetPriorityClass(::GetCurrentProcess(), PROCESS_MODE_BACKGROUND_BEGIN)) //this call lowers CPU priority, too!! + throw FileError(_("Cannot change process I/O priorities.") + L"\n\n" + getLastErrorFormatted()); +} + + +ScheduleForBackgroundProcessing::~ScheduleForBackgroundProcessing() +{ + ::SetPriorityClass(::GetCurrentProcess(), PROCESS_MODE_BACKGROUND_END); +} + +#elif defined FFS_LINUX +struct PreventStandby::Pimpl {}; +PreventStandby::PreventStandby() {} +PreventStandby::~PreventStandby() {} + +//solution for GNOME?: http://people.gnome.org/~mccann/gnome-session/docs/gnome-session.html#org.gnome.SessionManager.Inhibit + +struct ScheduleForBackgroundProcessing::Pimpl {}; +ScheduleForBackgroundProcessing::ScheduleForBackgroundProcessing() {}; +ScheduleForBackgroundProcessing::~ScheduleForBackgroundProcessing() {}; + +/* +struct ScheduleForBackgroundProcessing +{ + - required functions ioprio_get/ioprio_set are not part of glibc: http://linux.die.net/man/2/ioprio_set + - and probably never will: http://sourceware.org/bugzilla/show_bug.cgi?id=4464 + - /usr/include/linux/ioprio.h not available on Ubuntu, so we can't use it instead + + ScheduleForBackgroundProcessing() : oldIoPrio(getIoPriority(IOPRIO_WHO_PROCESS, ::getpid())) + { + if (oldIoPrio != -1) + setIoPriority(::getpid(), IOPRIO_PRIO_VALUE(IOPRIO_CLASS_IDLE, 0)); + } + ~ScheduleForBackgroundProcessing() + { + if (oldIoPrio != -1) + setIoPriority(::getpid(), oldIoPrio); + } + +private: + static int getIoPriority(pid_t pid) + { + return ::syscall(SYS_ioprio_get, IOPRIO_WHO_PROCESS, pid); + } + static int setIoPriority(pid_t pid, int ioprio) + { + return ::syscall(SYS_ioprio_set, IOPRIO_WHO_PROCESS, pid, ioprio); + } + + const int oldIoPrio; +}; +*/ + +#elif defined FFS_MAC +//https://developer.apple.com/library/mac/#qa/qa1340 +struct PreventStandby::Pimpl +{ + Pimpl() : assertionID() {} + IOPMAssertionID assertionID; +}; + +PreventStandby::PreventStandby() : pimpl(make_unique<Pimpl>()) +{ + if (::IOPMAssertionCreateWithName(kIOPMAssertionTypeNoIdleSleep, + kIOPMAssertionLevelOn, + CFSTR("FreeFileSync"), + &pimpl->assertionID) != kIOReturnSuccess) + throw FileError(_("Failed to suspend system sleep mode.")); +} + +PreventStandby::~PreventStandby() +{ + ::IOPMAssertionRelease(pimpl->assertionID); +} + + +struct ScheduleForBackgroundProcessing::Pimpl +{ + Pimpl() : oldIoPrio() {} + int oldIoPrio; +}; + + +ScheduleForBackgroundProcessing::ScheduleForBackgroundProcessing() : pimpl(make_unique<Pimpl>()) +{ + pimpl->oldIoPrio = ::getiopolicy_np(IOPOL_TYPE_DISK, IOPOL_SCOPE_PROCESS); + if (pimpl->oldIoPrio == -1) + throw FileError(_("Cannot change process I/O priorities.") + L" (r)" + L"\n\n" + getLastErrorFormatted()); + + if (::setiopolicy_np(IOPOL_TYPE_DISK, IOPOL_SCOPE_PROCESS, IOPOL_THROTTLE) != 0) + throw FileError(_("Cannot change process I/O priorities.") + L" (w)" + L"\n\n" + getLastErrorFormatted()); +} + + +ScheduleForBackgroundProcessing::~ScheduleForBackgroundProcessing() +{ + ::setiopolicy_np(IOPOL_TYPE_DISK, IOPOL_SCOPE_PROCESS, pimpl->oldIoPrio); +} +#endif
\ No newline at end of file |