как определить на каком ядре выполняется код

Собсна сабж..

вдух (или больше)-ядерный проц. драйвер (ring0) должен знать на каком ядре в данный момент выполняется код.
Есть ли какое-то АПИ для этого? решение нужно как для Вин32 так и для Вин64 (х86-64)

спасибо

 

Код (Text):

1.     oldAffinityMask = KeSetAffinityThread((PKTHREAD)PsGetCurrentThread(), 1);
2.     KeSetAffinityThread((PKTHREAD)PsGetCurrentThread(), oldAffinityMask);

oldAffinityMask - маска процессов на котором может исполнятся поток.
можно еще процессор на котором код исполняется взять из структуры ETHREAD.

 

а про это - можно поподробнее? где там эта инфа лежит и в каком виде?

спасибо..

 

еще можно Local APIC ID прочитать

 

ТС воспользуйся этим.

http://rku.nm.ru/System.rar

 

нашел в одном месте определение текущего процессора таким образом:

movzx eax, byte ptr fs:[51h]

вроде для моего вдухядерника выдает то 0 то 1.. т.е. похоже работает.. что это за структура по этому адресу?

 

ULONG KeGetCurrentProcessorNumber();

только такую проверку имеет смысл делать на irql >= DISPATCH, чтобы нитка гарантированно не была вытеснена и перенесена на другой проц.
Чтобы поставить thread на конкретный процессор можно использовать KeSetSystemAffinityThread.

 

и если предыдущий код действительно возвращает текущий ЦПУ, какое смещение будет для вин х64?

 

FS:[0] это Current PCR
структура официально недокументирована, и может отличаться для разных систем, сервиспаков, и даже хотфиксов.
Пользуй API, что я сказал.

 

ratix
KeGetCurrentProcessorNumber используй, что за мания хардкодить смещения в недокументированные структуры...

 

действительно.. определено как инлайн ф-ция.. (поэтому её даже в экспортах нет)

FORCEINLINE
ULONG
NTAPI
KeGetCurrentProcessorNumber(VOID)
{
__asm { movzx eax, _PCR KPCR.Number }
}


я так понимаю, теперь мне нужна DDK для х64 систем.. чтобы посомтреть как там реализована эта ф-ция..

 

для х64:

Код (Text):

1. __forceinline ULONG KeGetCurrentProcessorNumber (VOID)
2. {
3.     return (ULONG)__readgsbyte(0x184);
4. }

для IA64:

Код (Text):

1. #define KeGetCurrentProcessorNumber() ((ULONG)(PCR->Number))

 

Zufyxe
спасибо.. то что нужно..

я правильно понял, что в винде под х86-64 роль fs: выполняет теперь gs:?

 

связано с появлением инструкции swapgs