#include "types.hh"
#include "util.hh"
#include "affinity.hh"

#if HAVE_SCHED_H
#include <sched.h>
#endif

namespace nix {


#if HAVE_SCHED_SETAFFINITY
static bool didSaveAffinity = false;
static cpu_set_t savedAffinity;
#endif


void setAffinityTo(int cpu)
{
#if HAVE_SCHED_SETAFFINITY
    if (sched_getaffinity(0, sizeof(cpu_set_t), &savedAffinity) == -1) return;
    didSaveAffinity = true;
    printMsg(lvlDebug, format("locking this thread to CPU %1%") % cpu);
    cpu_set_t newAffinity;
    CPU_ZERO(&newAffinity);
    CPU_SET(cpu, &newAffinity);
    if (sched_setaffinity(0, sizeof(cpu_set_t), &newAffinity) == -1)
        printMsg(lvlError, format("failed to lock thread to CPU %1%") % cpu);
#endif
}


int lockToCurrentCPU()
{
#if HAVE_SCHED_SETAFFINITY
    int cpu = sched_getcpu();
    if (cpu != -1) setAffinityTo(cpu);
    return cpu;
#else
    return -1;
#endif
}


void restoreAffinity()
{
#if HAVE_SCHED_SETAFFINITY
    if (!didSaveAffinity) return;
    if (sched_setaffinity(0, sizeof(cpu_set_t), &savedAffinity) == -1)
        printMsg(lvlError, "failed to restore affinity %1%");
#endif
}


}
f='/guix/refs/?id=58e48c64cf653024fc4e34f8c18c517b3c7e3e59'>refs</a><a class='active' href='/guix/log/gnu/machine'>log</a><a href='/guix/tree/gnu/machine?id=58e48c64cf653024fc4e34f8c18c517b3c7e3e59'>tree</a><a href='/guix/commit/gnu/machine?id=58e48c64cf653024fc4e34f8c18c517b3c7e3e59'>commit</a><a href='/guix/diff/gnu/machine?id=58e48c64cf653024fc4e34f8c18c517b3c7e3e59'>diff</a></td><td class='form'><form class='right' method='get' action='/guix/log/gnu/machine'>
<input type='hidden' name='id' value='58e48c64cf653024fc4e34f8c18c517b3c7e3e59'/><select name='qt'>
<option value='grep'>log msg</option>
<option value='author'>author</option>
<option value='committer'>committer</option>
<option value='range'>range</option>
</select>
<input class='txt' type='search' size='10' name='q' value=''/>
<input type='submit' value='search'/>
</form>
</td></tr></table>
<div class='path'>path: <a href='/guix/log/?id=58e48c64cf653024fc4e34f8c18c517b3c7e3e59'>root</a>/<a href='/guix/log/gnu?id=58e48c64cf653024fc4e34f8c18c517b3c7e3e59'>gnu</a>/<a href='/guix/log/gnu/machine?id=58e48c64cf653024fc4e34f8c18c517b3c7e3e59'>machine</a></div><div class='content'><table class='list nowrap'><tr class='nohover'><th class='left'>Age</th><th class='left'>Commit message (<a href='/guix/log/gnu/machine?id=58e48c64cf653024fc4e34f8c18c517b3c7e3e59&amp;showmsg=1'>Expand</a>)</th><th class='left'>Author</th></tr>
<tr><td><span title='2024-06-04 12:08:34 +0200'>2024-06-04</span></td><td><a href='/guix/commit/gnu/machine?id=3e87b207ce96679e2f289a5522a248d38c4f3962'>file-systems: Add support for mounting CIFS file systems</a><span class='msg-avail'>...</span></td><td>Richard Sent</td></tr>
<tr><td><span title='2024-02-19 11:21:27 +0100'>2024-02-19</span></td><td><a href='/guix/commit/gnu/machine?id=9ce548a26e59d85ad9341c0212fef38392d49ff4'>machine/ssh: Refresh parameterization of %CURRENT-SYSTEM.</a><span class='msg-avail'>...</span></td><td>Ricardo Wurmus</td></tr>
<tr><td><span title='2023-08-08 18:01:54 +0200'>2023-08-08</span></td><td><a href='/guix/commit/gnu/machine?id=5652c2e14728cf746f86c4ec34a84e99eb34f9a0'>system: Do not check initrd modules for pseudo file systems.</a><span class='msg-avail'>...</span></td><td>Ludovic Courtès</td></tr>
<tr><td><span title='2022-11-17 22:27:39 +0100'>2022-11-17</span></td><td><a href='/guix/commit/gnu/machine?id=17b01d546306885ff3c07e7b6aaffb541a8b9043'>machine: ssh: Validate 'system' field.</a><span class='msg-avail'>...</span></td><td>Ludovic Courtès</td></tr>
<tr><td><span title='2022-11-09 13:45:24 +0100'>2022-11-09</span></td><td><a href='/guix/commit/gnu/machine?id=8d7cb7f2a4d0b7ce49764e4ba24026762d640da7'>machine/digital-ocean: Pull operating system definition out of string.</a><span class='msg-avail'>...</span></td><td>Ricardo Wurmus</td></tr>
<tr><td><span title='2022-11-09 13:45:24 +0100'>2022-11-09</span></td><td><a href='/guix/commit/gnu/machine?id=ebb88e2bed889b67ea83893c907ab29e3bc81e8b'>machine/digital-ocean: Set load path to Guile 3.0 directories.</a><span class='msg-avail'>...</span></td><td>Ricardo Wurmus</td></tr>
<tr><td><span title='2022-11-09 13:45:24 +0100'>2022-11-09</span></td><td><a href='/guix/commit/gnu/machine?id=e8dcd008a0be8bd967efabab6960d19e5424d93e'>machine/digital-ocean: Use static-networking-service-type.</a><span class='msg-avail'>...</span></td><td>Ricardo Wurmus</td></tr>
<tr><td><span title='2022-11-09 13:45:24 +0100'>2022-11-09</span></td><td><a href='/guix/commit/gnu/machine?id=c8e41af7be47fdd3ae16765d5cf892582f5a407a'>machine/digital-ocean: Use nightly Guix.</a><span class='msg-avail'>...</span></td><td>Ricardo Wurmus</td></tr>
<tr><td><span title='2022-10-17 07:29:09 +0200'>2022-10-17</span></td><td><a href='/guix/commit/gnu/machine?id=82b0f3755d12e04578a54ee53b3a8051958e8143'>guix: Fix typos.</a><span class='msg-avail'>...</span></td><td>Julien Lepiller</td></tr>
<tr><td><span title='2022-09-26 23:29:35 +0200'>2022-09-26</span></td><td><a href='/guix/commit/gnu/machine?id=1033645e9d3899edd6b052b19e24c0a718b95e88'>machine: ssh: Parameterize '%current-system' early on.</a><span class='msg-avail'>...</span></td><td>Ludovic Courtès</td></tr>
<tr><td><span title='2022-09-04 17:57:02 +0200'>2022-09-04</span></td><td><a href='/guix/commit/gnu/machine?id=a451a809d5537035cc8888cfde5349842a68e934'>Fix misspelling of GUIX_DIGITAL_OCEAN_TOKEN.</a><span class='msg-avail'>...</span></td><td>Matthew James Kraai</td></tr>