PsAdjustBasicEnclaveThreadList

INT64 __fastcall PsAdjustBasicEnclaveThreadList(INT64 a1, UINT64 a2, INT64 a3){
  _QWORD *v3; 
  int v4; 
  _QWORD *PoolWithTag; 
  _ETHREAD *CurrentThread; 
  INT64 v10; 
  bool v11; 
  __int64 *v12; 
  bool v13; 
  __int64 v14; 
  int v15; 
  _ETHREAD *v16; 
  unsigned int SessionId; 
  char v18; 
  int v19; 
  unsigned __int64 v20; 
  __int64 v21; 
  __int64 v22; 
  int v23; 
  v3 = 0i64;
  v4 = a3;
  if( (_DWORD)a3 )
  {
    PoolWithTag = ExAllocatePoolWithTag(PagedPool, 0x28ui64, 0x74457350ui64);
    v3 = PoolWithTag;
    if( !PoolWithTag )
      return 3221225626i64;
    *((_DWORD *)PoolWithTag + 9) = 0;
    PoolWithTag[3] = a2;
    *((_DWORD *)PoolWithTag + 8) = v4;
  }
  CurrentThread = (_ETHREAD *)KeGetCurrentThread();
  --*((_WORD *)CurrentThread + 242);
  v10 = a1 + 112;
  ExAcquirePushLockExclusiveEx(a1 + 112, 0i64);
  v11 = v4 == 0;
  v12 = *(__int64 **)(a1 + 104);
  if( v11 )
  {
    while( v12 )
    {
      v15 = MiHotPatchImageTreeCompare(a2, (INT64)v12);
      if( v15 >= 0 )
      {
        if( v15 <= 0 )
          break;
        v12 = (__int64 *)v12[1];
      }
      else
      {
        v12 = (__int64 *)*v12;
      }
    }
    RtlAvlRemoveNode((unsigned __int64 *)(a1 + 104), v12);
    goto LABEL_20;
  }
  v13 = 0;
  if( !v12 )
    goto LABEL_19;
  while( (int)MiHotPatchImageTreeCompare(a2, (INT64)v12) >= 0 )
  {
    v14 = v12[1];
    if( !v14 )
    {
      v13 = 1;
      goto LABEL_19;
    }
LABEL_17:
    v12 = (__int64 *)v14;
  }
  v14 = *v12;
  if( *v12 )
    goto LABEL_17;
  v13 = 0;
LABEL_19:
  RtlAvlInsertNodeEx((unsigned __int64 *)(a1 + 104), (__int64)v12, v13, v3);
  v12 = 0i64;
LABEL_20:
  if( (_InterlockedExchangeAdd64((volatile signed __int64 *)v10, 0xFFFFFFFFFFFFFFFFui64) & 6) == 2 )
    ExfTryToWakePushLock((volatile INT64 *)(a1 + 112));
  v23 = 0;
  v16 = (_ETHREAD *)KeGetCurrentThread();
  if( (unsigned int)MiGetSystemRegionType(a1 + 112) == 1 )
    SessionId = MmGetSessionIdEx(*((_QWORD *)v16 + 23));
  else
    SessionId = -1;
  --*((_WORD *)v16 + 243);
  v18 = ++*((_BYTE *)v16 + 794);
  v19 = (*((char *)v16 + 792) | *((char *)v16 + 870)) ^ 0x3F;
  while( 1 )
  {
    v11 = !_BitScanReverse((unsigned int *)&v21, v19);
    if( v11 )
      goto LABEL_32;
    v20 = *((_QWORD *)v16 + 100) + 96 * v21;
    v19 &= ~(1 << v21);
    if( (*(_BYTE *)(v20 + 26) & 1) != 0
      && (*(_DWORD *)(v20 + 32) & 1) == 0
      && (*(_QWORD *)(v20 + 32) & 0x7FFFFFFFFFFFFFFCi64) == (v10 & 0x7FFFFFFFFFFFFFFCi64)
      && *(_DWORD *)(v20 + 40) == SessionId )
    {
      *(_BYTE *)(v20 + 26) &= ~1u;
      if( *(_QWORD *)(v20 + 32) )
        break;
    }
  }
  if( !v20 )
  {
LABEL_32:
    if( (*((_DWORD *)v16 + 30) & 0x10000) == 0 )
      KeBugCheckEx(0x162u, (ULONG_PTR)v16, v10, SessionId, 0i64);
    goto LABEL_39;
  }
  *(_BYTE *)(v20 + 32) |= 2u;
  if( *(__int64 *)(v20 + 32) < 0 )
    KiAbEntryRemoveFromTree(v20);
  v23 = *(_DWORD *)(v20 + 88) & 0x1FFFF;
  *(_DWORD *)(v20 + 88) &= 0xFFFE0000;
  *(_BYTE *)(v20 + 25) &= ~1u;
  *(_QWORD *)(v20 + 32) = 0i64;
  v22 = (__int64)(v20 - *((_QWORD *)v16 + 100)) / 96;
  if( v18 == 1 )
    *((_BYTE *)v16 + 792) |= 1 << v22;
  else
    _InterlockedOr8((volatile signed __int8 *)v16 + 870, 1 << v22);
LABEL_39:
  --*((_BYTE *)v16 + 794);
  KiAbThreadRemoveBoosts((ULONG_PTR)v16, v10, (unsigned int *)&v23);
  v11 = (*((_WORD *)v16 + 243))++ == 0xFFFF;
  if( v11 && *((_ETHREAD **)v16 + 19) != (_ETHREAD *)((char *)v16 + 152) )
    KiCheckForKernelApcDelivery();
  KeLeaveCriticalRegion();
  if( v12 )
    ExFreePoolWithTag(v12, 0);
  return 0i64;
}

Referenced by: