ST_STORE::StDmpSinglePageAdd

__int64 __fastcall ST_STORE::StDmpSinglePageAdd(UINT64 CompressedSize, int *a2, __int64 r8_0, int *a4){
  char *v6; 
  int v7; 
  unsigned int v8; 
  int v9; 
  int Space; 
  __int64 v11; 
  UINT64 v12; 
  _ETHREAD *CurrentThread; 
  unsigned int SessionId; 
  unsigned __int8 v15; 
  int v16; 
  bool v17; 
  __int64 v18; 
  _KLOCK_ENTRY *v19; 
  unsigned __int8 v20; 
  int v21; 
  UINT8 *v22; 
  UINT8 *v23; 
  unsigned int v24; 
  unsigned int v25; 
  size_t v26; 
  UINT8 *v27; 
  char *v28; 
  unsigned int v29; 
  __int64 v30; 
  __int64 v31; 
  __int64 v32; 
  int v34; 
  UINT8 *v35; 
  unsigned int v36; 
  UINT64 v37; 
  _SMCR_ENCRYPT_CONTEXT *v38; 
  int v39; 
  PVOID BugCheckParameter4; 
  VOID *UncompressedChunkSize; 
  UINT64 Length; 
  int v43; 
  char a3[12]; 
  __int64 v45; 
  unsigned __int8 v46[8]; 
  int v47; 
  int *v48; 
  int *v49; 
  __int64 Nonce; 
  int v51; 

  v49 = a2;
  v45 = r8_0;
  Nonce = 0i64;
  v51 = 0;
  v6 = 0i64;
  v7 = (unsigned __int8)*(_DWORD *)(CompressedSize + 776);
  v8 = *a2 & 7;
  v48 = a4;
  *(_QWORD *)&a3[4] = 0i64;
  v9 = 0;
  v47 = 0;
  Length = 0i64;
  if( ((unsigned __int8)v7 == 0 ? 7 : 0) < v8 )
    v8 = v7 == 0 ? 7 : 0;
  Space = ST_STORE::StDmpSinglePageFindSpace(CompressedSize, v8, *a4, &a3[4], (__int64)&Length + 4);
  if( Space < 0 )
    goto LABEL_41;
  v11 = *(_QWORD *)(CompressedSize + 800);
  if( (*(_BYTE *)(v11 + 6021) & 4) != 0 )
  {
    v12 = v11 + 6024;
    if( (_InterlockedExchangeAdd64((volatile signed __int64 *)v12, 0xFFFFFFFFFFFFFFFFui64) & 6) == 2 )
      ExfTryToWakePushLock((_EX_PUSH_LOCK *)v12);
    *(_DWORD *)a3 = 0;
    CurrentThread = (_ETHREAD *)KeGetCurrentThread();
    if( MiGetSystemRegionType(v12) == 1 )
      SessionId = MmGetSessionIdEx(CurrentThread->Tcb.ApcState.Process);
    else
      SessionId = -1;
    --CurrentThread->Tcb.SpecialApcDisable;
    v15 = ++CurrentThread->Tcb.AbAllocationRegionCount;
    v16 = ((char)CurrentThread->Tcb.AbEntrySummary | (char)CurrentThread->Tcb.AbOrphanedEntrySummary) ^ 0x3F;
    while( 1 )
    {
      v17 = !_BitScanReverse((unsigned int *)&v18, v16);
      v43 = v18;
      if( v17 )
        break;
      v19 = &CurrentThread->Tcb.LockEntries[v18];
      v16 &= ~(1 << v18);
      if( (v19->AcquiredByte & 1) != 0
        && (v19->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 & 1) == 0
        && (v19->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 & 0x7FFFFFFFFFFFFFFCi64) == (v12 & 0x7FFFFFFFFFFFFFFCi64)
        && v19->LockState.SessionId == SessionId )
      {
        v19->AcquiredByte &= ~1u;
        if( v19->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 )
        {
          if( v19 )
          {
            v19->CrossThreadReleasableAndBusyByte |= 2u;
            if( v19->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 < 0 )
              KiAbEntryRemoveFromTree(v19);
            *(_DWORD *)a3 = v19->BoostBitmap.AllFields & 0x1FFFF;
            v19->BoostBitmap.AllFields &= 0xFFFE0000;
            v19->ThreadLocalFlags &= ~1u;
            v19->LockState.$E8276A2CF8B819ED32D1B6FEB35D730A::_bf_0 = 0i64;
            v20 = 1 << (v19 - CurrentThread->Tcb.LockEntries);
            if( v15 == 1 )
              CurrentThread->Tcb.AbEntrySummary |= v20;
            else
              _InterlockedOr8((volatile signed __int8 *)&CurrentThread->Tcb.AbOrphanedEntrySummary, v20);
            goto LABEL_21;
          }
          break;
        }
      }
    }
    if( (*(&CurrentThread->Tcb.MiscFlags + 1) & 0x10000) == 0 )
      KeBugCheckEx(0x162u, CurrentThread, (PVOID)v12, (PVOID)SessionId, 0i64);
LABEL_21:
    --CurrentThread->Tcb.AbAllocationRegionCount;
    KiAbThreadRemoveBoosts(&CurrentThread->Tcb, (VOID *)v12, (UINT64 *)a3);
    v17 = CurrentThread->Tcb.SpecialApcDisable++ == -1;
    if( v17
      && ($F25F8C4BA33AF922A5F1AF68CD89DDDF *)CurrentThread->Tcb.ApcState.ApcListHead[0].Flink != &CurrentThread->Tcb.152 )
    {
      KiCheckForKernelApcDelivery();
    }
    KiLeaveGuardedRegionUnsafe((__int64)KeGetCurrentThread());
  }
  v9 = 1;
  v21 = *a2;
  v22 = (UINT8 *)(*(_QWORD *)&a3[4] + *(unsigned int *)(CompressedSize + 824));
  v23 = (UINT8 *)*((_QWORD *)a2 + 1);
  if( *a2 >= 0 )
  {
    v34 = *(_DWORD *)(CompressedSize + 776);
    if( (v34 & 0x400) != 0 )
    {
      v24 = 4096;
    }
    else
    {
      if( (v34 & 0x8000) != 0 )
      {
        memmove((VOID *)(*(_QWORD *)(CompressedSize + 1784) + 4096i64), *((const VOID **)a2 + 1), 0x1000u);
        v21 = *a2;
        v23 = (UINT8 *)(*(_QWORD *)(CompressedSize + 1784) + 4096i64);
      }
      v24 = 0;
    }
  }
  else
  {
    v24 = *(_DWORD *)(*(_QWORD *)(v45 + 16) + 40i64);
  }
  v25 = *(_DWORD *)(CompressedSize + 816) - (HIDWORD(Length) & *(_DWORD *)(CompressedSize + 808));
  LODWORD(Length) = v24;
  if( v21 < 0 )
  {
    v26 = v24;
    v27 = v23;
LABEL_29:
    memmove(v22, v27, v26);
    goto LABEL_30;
  }
  if( v25 >= *(_DWORD *)(CompressedSize + 820) )
    v35 = v22;
  else
    v35 = *(UINT8 **)(CompressedSize + 1784);
  if( !v24 )
  {
    LODWORD(UncompressedChunkSize) = 4096;
    LODWORD(BugCheckParameter4) = 4096;
    if( RtlCompressBuffer(
           *(_WORD *)(CompressedSize + 992),
           v23,
           0x1000ui64,
           v35,
           (UINT64)BugCheckParameter4,
           (UINT64)UncompressedChunkSize,
           &Length,
           *(VOID **)(CompressedSize + 896)) >= 0 )
    {
      v24 = Length;
    }
    else
    {
      v24 = 4096;
      LODWORD(Length) = 4096;
    }
  }
  if( v35 != v22 )
  {
    v36 = (v24 + *(_DWORD *)(CompressedSize + 824) + 15) >> 4;
    *v48 = v36;
    if( v36 > v25 )
    {
      Space = -1073741267;
      goto LABEL_41;
    }
  }
  if( v24 > 0xFF0 )
  {
    if( (*(_DWORD *)(CompressedSize + 776) & 0x200) != 0 )
    {
      Space = -1073740762;
      goto LABEL_41;
    }
    v24 = 4096;
    v35 = v23;
    LODWORD(Length) = 4096;
  }
  if( v35 != v22 )
  {
    v26 = v24;
    v27 = v35;
    goto LABEL_29;
  }
LABEL_30:
  if( *(_QWORD *)(CompressedSize + 760)
    || (Space = ST_STORE::StDmpDummyPageRecordAllocate(CompressedSize), Space >= 0) )
  {
    v9 = 0;
    SMKM_STORE::SmStAcquireStoreLockExclusive(*(_QWORD *)(CompressedSize + 800));
    v28 = (char *)SmHpChunkAlloc((_SMHP_CHUNK_HEAP *)(CompressedSize + 192));
    v6 = v28;
    if( !v28 )
    {
      Space = -1073741670;
      goto LABEL_41;
    }
    ST_STORE::StDmPageRecordUnprotect(CompressedSize, (INT64)v28);
    *(_QWORD *)(v6 + 4) = 0i64;
    *(_DWORD *)v6 = HIDWORD(Length);
    v29 = Length;
    if( (unsigned int)Length < 0x1000 )
    {
      *((_DWORD *)v6 + 1) ^= Length & 0xFFF;
      v29 = Length;
    }
    v30 = (__int64)v49;
    if( *v49 >= 0 )
    {
      *(_QWORD *)v46 = 0i64;
      MetroHash64::Hash(v22, v29, v46);
      LODWORD(v31) = *(_DWORD *)v46;
    }
    else
    {
      v31 = *(_QWORD *)(v45 + 96);
    }
    *((_DWORD *)v6 + 2) = v31;
    v32 = *(_QWORD *)(CompressedSize + 1016);
    if( *(_DWORD *)(v32 + 24) )
    {
      v37 = -*(_DWORD *)(v32 + 8) & (unsigned int)(*(_DWORD *)(v32 + 8) + Length - 1);
      ++*(_QWORD *)(CompressedSize + 1024);
      *((_DWORD *)v6 + 3) = *(_DWORD *)(CompressedSize + 1024);
      *((_WORD *)v6 + 3) = *(_WORD *)(CompressedSize + 1028);
      LODWORD(Nonce) = Length;
      HIDWORD(Nonce) = *((_DWORD *)v6 + 3);
      v38 = *(_SMCR_ENCRYPT_CONTEXT **)(CompressedSize + 1016);
      v51 = *((unsigned __int16 *)v6 + 3);
      if( SmCrAuthEncrypt(v38, (INT8 *)v22, (INT8 *)v22, v37, &Nonce, UncompressedChunkSize) < 0 )
      {
        Space = -1073741174;
        goto LABEL_41;
      }
    }
    else if( *(_BYTE *)(CompressedSize + 776) )
    {
      v39 = RtlComputeCrc32(0i64, v22, (unsigned int)Length);
      **(_DWORD **)&a3[4] = v39;
    }
    Space = ST_STORE::StDmpSinglePageInsert(CompressedSize, v30, (unsigned __int64)v6);
    if( Space >= 0 )
    {
      v6 = 0i64;
      Space = 0;
    }
  }
LABEL_41:
  if( v9 )
    SMKM_STORE::SmStAcquireStoreLockExclusive(*(_QWORD *)(CompressedSize + 800));
  if( v6 )
    SmHpChunkFree((_SMHP_CHUNK_HEAP *)(CompressedSize + 192), v6);
  return(unsigned int)Space;
}

Referenced by:

No references.