PsAdjustBasicEnclaveThreadList

NTSTATUS __stdcall PsAdjustBasicEnclaveThreadList(INT64 a1, UINT64 a2, INT64 er8){
  _RTL_BALANCED_NODE *v3; 
  int v4; 
  VOID **PoolWithTag; 
  _ETHREAD *CurrentThread; 
  INT64 v10; 
  bool v11; 
  INT64 *v12; 
  BOOL v13; 
  INT64 v14; 
  NTSTATUS v15; 
  _ETHREAD *v16; 
  unsigned int SessionId; 
  unsigned __int8 v18; 
  int v19; 
  _KLOCK_ENTRY *v20; 
  __int64 v21; 
  __int64 v22; 
  UINT64 a3; 

  v3 = 0i64;
  v4 = er8;
  if( (_DWORD)er8 )
  {
    PoolWithTag = ExAllocatePoolWithTag(1ui64, 0x28ui64, 1950708560i64);
    v3 = (_RTL_BALANCED_NODE *)PoolWithTag;
    if( !PoolWithTag )
      return -1073741670;
    *((_DWORD *)PoolWithTag + 9) = 0;
    PoolWithTag[3] = (VOID *)a2;
    *((_DWORD *)PoolWithTag + 8) = v4;
  }
  CurrentThread = (_ETHREAD *)KeGetCurrentThread();
  --CurrentThread->Tcb.KernelApcDisable;
  v10 = a1 + 112;
  ExAcquirePushLockExclusiveEx((_EX_PUSH_LOCK *)(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((UINT64 *)(a1 + 104), v12);
    goto LABEL_20;
  }
  v13 = 0;
  if( !v12 )
    goto LABEL_19;
  while( 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((_RTL_BALANCED_NODE **)(a1 + 104), (UINT64)v12, v13, v3);
  v12 = 0i64;
LABEL_20:
  if( (_InterlockedExchangeAdd64((volatile signed __int64 *)v10, 0xFFFFFFFFFFFFFFFFui64) & 6) == 2 )
    ExfTryToWakePushLock((_EX_PUSH_LOCK *)(a1 + 112));
  LODWORD(a3) = 0;
  v16 = (_ETHREAD *)KeGetCurrentThread();
  if( MiGetSystemRegionType(a1 + 112) == 1 )
    SessionId = MmGetSessionIdEx(v16->Tcb.ApcState.Process);
  else
    SessionId = -1;
  --v16->Tcb.SpecialApcDisable;
  v18 = ++v16->Tcb.AbAllocationRegionCount;
  v19 = ((char)v16->Tcb.AbEntrySummary | (char)v16->Tcb.AbOrphanedEntrySummary) ^ 0x3F;
  while( 1 )
  {
    v11 = !_BitScanReverse((unsigned int *)&v21, v19);
    if( v11 )
      goto LABEL_32;
    v20 = &v16->Tcb.LockEntries[v21];
    v19 &= ~(1 << v21);
    if( (v20->AcquiredByte & 1) != 0
      && (v20->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 & 1) == 0
      && (v20->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 & 0x7FFFFFFFFFFFFFFCi64) == (v10 & 0x7FFFFFFFFFFFFFFCi64)
      && v20->LockState.SessionId == SessionId )
    {
      v20->AcquiredByte &= ~1u;
      if( v20->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 )
        break;
    }
  }
  if( !v20 )
  {
LABEL_32:
    if( (*(&v16->Tcb.MiscFlags + 1) & 0x10000) == 0 )
      KeBugCheckEx(0x162u, v16, (PVOID)v10, (PVOID)SessionId, 0i64);
    goto LABEL_39;
  }
  v20->CrossThreadReleasableAndBusyByte |= 2u;
  if( v20->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 < 0 )
    KiAbEntryRemoveFromTree(v20);
  LODWORD(a3) = v20->BoostBitmap.AllFields & 0x1FFFF;
  v20->BoostBitmap.AllFields &= 0xFFFE0000;
  v20->ThreadLocalFlags &= ~1u;
  v20->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 = 0i64;
  v22 = v20 - v16->Tcb.LockEntries;
  if( v18 == 1 )
    v16->Tcb.AbEntrySummary |= 1 << v22;
  else
    _InterlockedOr8((volatile signed __int8 *)&v16->Tcb.AbOrphanedEntrySummary, 1 << v22);
LABEL_39:
  --v16->Tcb.AbAllocationRegionCount;
  KiAbThreadRemoveBoosts(&v16->Tcb, (VOID *)v10, &a3);
  v11 = v16->Tcb.SpecialApcDisable++ == -1;
  if( v11 && ($F25F8C4BA33AF922A5F1AF68CD89DDDF *)v16->Tcb.ApcState.ApcListHead[0].Flink != &v16->Tcb.152 )
    KiCheckForKernelApcDelivery();
  KeLeaveCriticalRegion();
  if( v12 )
    ExFreePoolWithTag(v12, 0);
  return 0;
}

Referenced by: