HalpLeaveDmaDomain

NTSTATUS __stdcall HalpLeaveDmaDomain(INT64 a1){
  INT64 v1; 
  UINT64 v3; 
  UINT8 v4; 
  int v5; 
  _ETHREAD *v6; 
  unsigned int v7; 
  int v8; 
  bool v9; 
  __int64 v10; 
  _KLOCK_ENTRY *v11; 
  int v12; 
  unsigned int v13; 
  __int64 v14; 
  _ETHREAD *CurrentThread; 
  unsigned int SessionId; 
  unsigned __int8 v17; 
  int v18; 
  __int64 v19; 
  _KLOCK_ENTRY *v20; 
  int v21; 
  unsigned int v22; 
  __int64 v23; 
  void *v24; 
  void *v25; 
  UINT64 v27; 
  __int64 v28; 
  unsigned __int8 v29; 
  UINT64 a3; 
  int v31; 

  v1 = *(_QWORD *)(a1 + 504);
  v28 = 0i64;
  if( !*(_DWORD *)(v1 + 64) )
    goto LABEL_50;
  v3 = a1 + 312;
  ExAcquirePushLockExclusiveEx((_EX_PUSH_LOCK *)(a1 + 312), 0i64);
  if( *(_BYTE *)(a1 + 338) )
  {
    if( (_InterlockedExchangeAdd64((volatile signed __int64 *)v3, 0xFFFFFFFFFFFFFFFFui64) & 6) == 2 )
      ExfTryToWakePushLock((_EX_PUSH_LOCK *)v3);
    LODWORD(v27) = 0;
    CurrentThread = (_ETHREAD *)KeGetCurrentThread();
    if( MiGetSystemRegionType(v3) == 1 )
      SessionId = MmGetSessionIdEx(CurrentThread->Tcb.ApcState.Process);
    else
      SessionId = -1;
    --CurrentThread->Tcb.SpecialApcDisable;
    v17 = ++CurrentThread->Tcb.AbAllocationRegionCount;
    v18 = ((char)CurrentThread->Tcb.AbEntrySummary | (char)CurrentThread->Tcb.AbOrphanedEntrySummary) ^ 0x3F;
    while( 1 )
    {
      v9 = !_BitScanReverse((unsigned int *)&v19, v18);
      v31 = v19;
      if( v9 )
        break;
      v20 = &CurrentThread->Tcb.LockEntries[v19];
      v18 &= ~(1 << v19);
      if( (v20->AcquiredByte & 1) != 0
        && (v20->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 & 1) == 0
        && (v20->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 & 0x7FFFFFFFFFFFFFFCi64) == (v3 & 0x7FFFFFFFFFFFFFFCi64)
        && v20->LockState.SessionId == SessionId )
      {
        v20->AcquiredByte &= ~1u;
        if( v20->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 )
        {
          if( v20 )
          {
            v20->CrossThreadReleasableAndBusyByte |= 2u;
            if( v20->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 < 0 )
              KiAbEntryRemoveFromTree(v20);
            v21 = v20->BoostBitmap.AllFields & 0x1FFFF;
            v22 = v20->BoostBitmap.AllFields & 0xFFFE0000;
            v20->ThreadLocalFlags &= ~1u;
            LODWORD(v27) = v21;
            v20->BoostBitmap.AllFields = v22;
            v20->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 = 0i64;
            v23 = v20 - CurrentThread->Tcb.LockEntries;
            if( v17 == 1 )
              CurrentThread->Tcb.AbEntrySummary |= 1 << v23;
            else
              _InterlockedOr8((volatile signed __int8 *)&CurrentThread->Tcb.AbOrphanedEntrySummary, 1 << v23);
            goto LABEL_45;
          }
          break;
        }
      }
    }
    if( (*(&CurrentThread->Tcb.MiscFlags + 1) & 0x10000) == 0 )
      KeBugCheckEx(0x162u, CurrentThread, (PVOID)v3, (PVOID)SessionId, 0i64);
LABEL_45:
    --CurrentThread->Tcb.AbAllocationRegionCount;
    KiAbThreadRemoveBoosts(&CurrentThread->Tcb, (VOID *)v3, &v27);
    v9 = CurrentThread->Tcb.SpecialApcDisable++ == -1;
    if( v9
      && ($F25F8C4BA33AF922A5F1AF68CD89DDDF *)CurrentThread->Tcb.ApcState.ApcListHead[0].Flink != &CurrentThread->Tcb.152 )
    {
      KiCheckForKernelApcDelivery();
    }
  }
  else
  {
    v5 = (unsigned int)HalpIommuLeaveDmaDomain(*(_KDEVICE_QUEUE **)(a1 + 536), *(_QWORD *)(v1 + 40), v4);
    if( (_InterlockedExchangeAdd64((volatile signed __int64 *)v3, 0xFFFFFFFFFFFFFFFFui64) & 6) == 2 )
      ExfTryToWakePushLock((_EX_PUSH_LOCK *)v3);
    LODWORD(a3) = 0;
    v6 = (_ETHREAD *)KeGetCurrentThread();
    if( MiGetSystemRegionType(v3) == 1 )
      v7 = MmGetSessionIdEx(v6->Tcb.ApcState.Process);
    else
      v7 = -1;
    --v6->Tcb.SpecialApcDisable;
    v29 = ++v6->Tcb.AbAllocationRegionCount;
    v8 = ((char)v6->Tcb.AbEntrySummary | (char)v6->Tcb.AbOrphanedEntrySummary) ^ 0x3F;
    while( 1 )
    {
      v9 = !_BitScanReverse((unsigned int *)&v10, v8);
      if( v9 )
        break;
      v11 = &v6->Tcb.LockEntries[v10];
      v8 &= ~(1 << v10);
      if( (v11->AcquiredByte & 1) != 0
        && (v11->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 & 1) == 0
        && (v11->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 & 0x7FFFFFFFFFFFFFFCi64) == (v3 & 0x7FFFFFFFFFFFFFFCi64)
        && v11->LockState.SessionId == v7 )
      {
        v11->AcquiredByte &= ~1u;
        if( v11->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 )
        {
          if( v11 )
          {
            v11->CrossThreadReleasableAndBusyByte |= 2u;
            if( v11->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 < 0 )
              KiAbEntryRemoveFromTree(v11);
            v12 = v11->BoostBitmap.AllFields & 0x1FFFF;
            v13 = v11->BoostBitmap.AllFields & 0xFFFE0000;
            v11->ThreadLocalFlags &= ~1u;
            LODWORD(a3) = v12;
            v11->BoostBitmap.AllFields = v13;
            v11->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 = 0i64;
            v14 = v11 - v6->Tcb.LockEntries;
            if( v29 == 1 )
              v6->Tcb.AbEntrySummary |= 1 << v14;
            else
              _InterlockedOr8((volatile signed __int8 *)&v6->Tcb.AbOrphanedEntrySummary, 1 << v14);
            goto LABEL_20;
          }
          break;
        }
      }
    }
    if( (*(&v6->Tcb.MiscFlags + 1) & 0x10000) == 0 )
      KeBugCheckEx(0x162u, v6, (PVOID)v3, (PVOID)v7, 0i64);
LABEL_20:
    --v6->Tcb.AbAllocationRegionCount;
    KiAbThreadRemoveBoosts(&v6->Tcb, (VOID *)v3, &a3);
    v9 = v6->Tcb.SpecialApcDisable++ == -1;
    if( v9 && ($F25F8C4BA33AF922A5F1AF68CD89DDDF *)v6->Tcb.ApcState.ApcListHead[0].Flink != &v6->Tcb.152 )
      KiCheckForKernelApcDelivery();
    if( v5 < 0 )
      return v5;
  }
  v24 = *(void **)(a1 + 552);
  v25 = *(void **)(v1 + 40);
  v28 = *(unsigned int *)(a1 + 244);
  v5 = HalpIommuDomainUnmapLogicalRange(v25, v24);
  if( v5 >= 0 )
  {
    v5 = HalpDomainLaDelete(v1, *(_QWORD *)(a1 + 552));
    if( v5 >= 0 )
    {
LABEL_50:
      HalpDmaDereferenceDomainObject((PVOID)v1);
      *(_QWORD *)(a1 + 504) = 0i64;
      return 0;
    }
  }
  return v5;
}

Referenced by: