Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-12-27 09:34:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

[NDK] Add Affinity helper inline functions

Browse source
This commit is contained in:
Timo Kreuzer 2023-11-26 17:32:27 +02:00
parent 57e7f0b321
commit d87b45bee7
3 changed files with 47 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
hal/halx86/include/smp.h
View file

@ -7,15 +7,6 @@
#pragma once
#define AFFINITY_MASK(Id) ((KAFFINITY)1 << (Id))
/* Helper to find the lowest CPU in a KAFFINITY */
#ifdef _WIN64
#define BitScanForwardAffinity BitScanForward64
#else
#define BitScanForwardAffinity BitScanForward
#endif
/* This table is filled for each physical processor on system */
typedef struct _PROCESSOR_IDENTITY
{

1
ntoskrnl/include/internal/ke.h
View file

@ -155,7 +155,6 @@ extern VOID __cdecl KiInterruptTemplate(VOID);
/* MACROS *************************************************************************/
#define AFFINITY_MASK(ProcessorIndex) ((KAFFINITY)1 << (ProcessorIndex))
#define PRIORITY_MASK(Priority) (1UL << (Priority))
/* Tells us if the Timer or Event is a Syncronization or Notification Object */

47
sdk/include/ndk/kefuncs.h
View file

@ -33,6 +33,53 @@ extern "C"
#ifndef NTOS_MODE_USER
//
// Affinity helpers
//
FORCEINLINE KAFFINITY AFFINITY_MASK(ULONG Index)
{
ASSERT(Index < sizeof(KAFFINITY) * 8);
return (KAFFINITY)1 << Index;
}
FORCEINLINE BOOLEAN BitScanForwardAffinity(PULONG Index, KAFFINITY Mask)
{
#ifdef _WIN64
return BitScanForward64(Index, Mask);
#else
return BitScanForward(Index, Mask);
#endif
}
FORCEINLINE BOOLEAN BitScanReverseAffinity(PULONG Index, KAFFINITY Mask)
{
#ifdef _WIN64
return BitScanReverse64(Index, Mask);
#else
return BitScanReverse(Index, Mask);
#endif
}
FORCEINLINE BOOLEAN InterlockedBitTestAndSetAffinity(volatile KAFFINITY *Affinity, ULONG Index)
{
ASSERT(Index < sizeof(KAFFINITY) * 8);
#ifdef _WIN64
return InterlockedBitTestAndSet64((PLONG64)Affinity, Index);
#else
return InterlockedBitTestAndSet((PLONG)Affinity, Index);
#endif
}
FORCEINLINE BOOLEAN InterlockedBitTestAndResetAffinity(volatile KAFFINITY *Affinity, ULONG Index)
{
ASSERT(Index < sizeof(KAFFINITY) * 8);
#ifdef _WIN64
return InterlockedBitTestAndReset64((PLONG64)Affinity, Index);
#else
return InterlockedBitTestAndReset((PLONG)Affinity, Index);
#endif
}
//
// APC Functions
//