bun-memory 1.1.24 → 1.1.26

package/example/benchmark.ts

@@ -0,0 +1,107 @@
+// ! Warning:
+// ! This benchmark may take 1-2+ minutes. It repeatedly traverses the entire entity list in
+// ! Counter-Strike 2, storing it into `EntityClassInfoNames`…
+
+import Memory from 'bun-memory';
+
+// Get the latest client_dll.json and offsets.json from:
+// https://github.com/a2x/cs2-dumper/tree/main/output
+
+import ClientDLLJSON from './offsets/client_dll.json';
+import OffsetsJSON from './offsets/offsets.json';
+
+// Load the needed offsets as bigints… It's ugly but… IntelliSense! 🫠…
+const Client = {
+  ...Object.fromEntries(Object.entries(ClientDLLJSON['client.dll'].classes).map(([class_, { fields }]) => [class_, Object.fromEntries(Object.entries(fields).map(([field, value]) => [field, BigInt(value)]))])),
+  Other: Object.fromEntries(Object.entries(OffsetsJSON['client.dll']).map(([key, value]) => [key, BigInt(value)])),
+} as {
+  [Class in keyof (typeof ClientDLLJSON)['client.dll']['classes']]: { [Field in keyof (typeof ClientDLLJSON)['client.dll']['classes'][Class]['fields']]: bigint };
+} & { Other: { [K in keyof (typeof OffsetsJSON)['client.dll']]: bigint } };
+
+// Open a handle to cs2.exe…
+const cs2 = new Memory('cs2.exe');
+
+// Get the base for client.dll…
+const ClientPtr = cs2.modules['client.dll']?.base;
+
+// Make sure client.dll is loaded…
+if (ClientPtr === undefined) {
+  throw new TypeError('ClientPtr must not be undefined.');
+}
+
+// Warmup…
+console.log('Warming up…');
+
+for (let i = 0; i < 1e6; i++) {
+  const GlobalVarsPtr = cs2.u64(ClientPtr + Client.Other.dwGlobalVars);
+  /* */ const CurTime = cs2.f32(GlobalVarsPtr + 0x30n);
+}
+
+// Create caches and scratches to optimize performance…
+const Cache_Names = new Map<bigint, string>();
+
+const EntityChunkScratch = new BigUint64Array(0xf000 / 0x08);
+const EntityListScratch = new BigUint64Array(0x200 / 0x08);
+
+// Start the test…
+console.log('Starting the test…');
+
+const performance1 = performance.now();
+
+const EntityListPtr = cs2.u64(ClientPtr + Client.Other.dwEntityList);
+
+for (let i = 0; i < 1e6; i++) {
+  const EntityClassInfoNames = new Map<string, bigint[]>();
+  const EntityClassInfoPtrs = new Map<bigint, bigint[]>();
+
+  // Traverse the entity list…
+  cs2.read(EntityListPtr + 0x10n, EntityListScratch);
+
+  for (let i = 0; i < 0x40; i++) {
+    const EntityChunkPtr = EntityListScratch[i];
+
+    if (EntityChunkPtr === 0n) {
+      continue;
+    }
+
+    cs2.read(EntityChunkPtr, EntityChunkScratch);
+
+    for (let j = 0, l = 0; j < 0x200; j++, l += 0x0f) {
+      const BaseEntityPtr = EntityChunkScratch[l];
+
+      if (BaseEntityPtr === 0n) {
+        continue;
+      }
+
+      const EntityClassInfoPtr = EntityChunkScratch[l + 0x01];
+
+      let BaseEntityPtrs = EntityClassInfoPtrs.get(EntityClassInfoPtr);
+
+      if (BaseEntityPtrs === undefined) {
+        BaseEntityPtrs = [];
+
+        EntityClassInfoPtrs.set(EntityClassInfoPtr, BaseEntityPtrs);
+      }
+
+      BaseEntityPtrs.push(BaseEntityPtr);
+    }
+
+    for (const [EntityClassInfoPtr, BaseEntityPtrs] of EntityClassInfoPtrs) {
+      let Name = Cache_Names.get(EntityClassInfoPtr);
+
+      if (Name === undefined) {
+        const SchemaClassInfoDataPtr = cs2.u64(EntityClassInfoPtr + 0x30n);
+        /* */ const NamePtr = cs2.u64(SchemaClassInfoDataPtr + 0x08n);
+        /*       */ Name = cs2.cString(NamePtr, 0x2a).toString();
+
+        Cache_Names.set(EntityClassInfoPtr, Name);
+      }
+
+      EntityClassInfoNames.set(Name, BaseEntityPtrs);
+    }
+  }
+}
+
+const performance2 = performance.now();
+
+console.log('Test completed in %fms…', (performance2 - performance1).toFixed(2));

package/example/trigger-bot.ts

@@ -1,7 +1,7 @@
 import { FFIType, dlopen } from 'bun:ffi';
 import { sleep } from 'bun';
 
-import Memory from '../index.ts';
+import Memory from 'bun-memory';
 
 // Get the latest client_dll.json and offsets.json from:
 // https://github.com/a2x/cs2-dumper/tree/main/output
@@ -21,21 +21,12 @@ const {
 });
 
 // Load the needed offsets as bigints…
-const { C_BaseEntity, C_BasePlayerPawn, C_BasePlayerWeapon, C_CSPlayerPawn, C_CSPlayerPawnBase, C_CSWeaponBaseGun, CBasePlayerController, CCSPlayer_WeaponServices, CCSWeaponBaseVData } = Object.fromEntries(
-  Object.entries(ClientDLLJSON['client.dll'].classes).map(([class_, { fields }]) => [class_, Object.fromEntries(Object.entries(fields).map(([field, value]) => [field, BigInt(value)]))])
-) as {
-  [C in keyof (typeof ClientDLLJSON)['client.dll']['classes']]: { [F in keyof (typeof ClientDLLJSON)['client.dll']['classes'][C]['fields']]: bigint };
-};
-
-// Load the needed offsets as bigints…
-const {
-  'client.dll': { dwEntityList, dwGlobalVars, dwLocalPlayerController, dwLocalPlayerPawn },
-  'engine2.dll': {},
-} = Object.fromEntries(
-  Object.entries(OffsetsJSON).map(([name, section]) => [name, Object.fromEntries(Object.entries(section).map(([key, value]) => [key, BigInt(value as number)]))]) //
-) as {
-  [M in keyof typeof OffsetsJSON]: { [K in keyof (typeof OffsetsJSON)[M]]: bigint };
-};
+const Client = {
+  ...Object.fromEntries(Object.entries(ClientDLLJSON['client.dll'].classes).map(([class_, { fields }]) => [class_, Object.fromEntries(Object.entries(fields).map(([field, value]) => [field, BigInt(value)]))])),
+  Other: Object.fromEntries(Object.entries(OffsetsJSON['client.dll']).map(([key, value]) => [key, BigInt(value)])),
+} as {
+  [Class in keyof (typeof ClientDLLJSON)['client.dll']['classes']]: { [Field in keyof (typeof ClientDLLJSON)['client.dll']['classes'][Class]['fields']]: bigint };
+} & { Other: { [K in keyof (typeof OffsetsJSON)['client.dll']]: bigint } };
 
 // Open a handle to cs2.exe…
 const cs2 = new Memory('cs2.exe');
@@ -46,23 +37,6 @@ const ClientPtr = cs2.modules['client.dll']?.base;
 if (ClientPtr === undefined) {
   throw new TypeError('ClientPtr must not be undefined.');
 }
-// !
-
-console.time('test');
-
-// while (true) {
-//   for (const pointer of pointers) {
-//     const cString = cs2.cString(pointer, 32).toString();
-
-//     if (cString.length > 32) {
-//       console.log(cString);
-//     }
-//   }
-// }
-
-console.timeEnd('test');
-
-// !
 
 // Create a cache for class name strings… 🫠…
 const Cache_Names = new Map<bigint, string>();
@@ -74,26 +48,26 @@ async function tick(ClientPtr: bigint) {
     ticks++;
 
     // Read relevant info from memory…
-    const GlobalVarsPtr = cs2.u64(ClientPtr + dwGlobalVars);
+    const GlobalVarsPtr = cs2.u64(ClientPtr + Client.Other.dwGlobalVars);
     /* */ const CurTime = cs2.f32(GlobalVarsPtr + 0x30n);
 
-    const Local_PlayerControllerPtr = cs2.u64(ClientPtr + dwLocalPlayerController);
-    /* */ const Local_TickBase = cs2.u32(Local_PlayerControllerPtr + CBasePlayerController.m_nTickBase);
-
-    const Local_PlayerPawnPtr = cs2.u64(ClientPtr + dwLocalPlayerPawn);
-    /* */ const Local_FlashOverlayAlpha = cs2.f32(Local_PlayerPawnPtr + C_CSPlayerPawnBase.m_flFlashOverlayAlpha);
-    /* */ const Local_IDEntIndex = cs2.i32(Local_PlayerPawnPtr + C_CSPlayerPawn.m_iIDEntIndex);
-    /* */ const Local_IsScoped = cs2.i32(Local_PlayerPawnPtr + C_CSPlayerPawn.m_bIsScoped);
-    /* */ const Local_Player_WeaponServicesPtr = cs2.u64(Local_PlayerPawnPtr + C_BasePlayerPawn.m_pWeaponServices);
-    /*       */ const Local_NextAttack = cs2.f32(Local_Player_WeaponServicesPtr + CCSPlayer_WeaponServices.m_flNextAttack);
-    /* */ const Local_TeamNum = cs2.u8(Local_PlayerPawnPtr + C_BaseEntity.m_iTeamNum);
-    /* */ const Local_WeaponBasePtr = cs2.u64(Local_PlayerPawnPtr + C_CSPlayerPawn.m_pClippingWeapon);
-    /*       */ const Local_Clip1 = cs2.i32(Local_WeaponBasePtr + C_BasePlayerWeapon.m_iClip1);
-    /*       */ const Local_NextPrimaryAttackTick = cs2.u32(Local_WeaponBasePtr + C_BasePlayerWeapon.m_nNextPrimaryAttackTick);
-    /*       */ const Local_WeaponBaseVDataPtr = cs2.u64(Local_WeaponBasePtr + C_BaseEntity.m_nSubclassID + 0x08n);
-    /*             */ const Local_IsFullAuto = cs2.bool(Local_WeaponBaseVDataPtr + CCSWeaponBaseVData.m_bIsFullAuto);
-    /*             */ const Local_WeaponType = cs2.i32(Local_WeaponBaseVDataPtr + CCSWeaponBaseVData.m_WeaponType);
-    /*       */ const Local_ZoomLevel = cs2.i32(Local_WeaponBasePtr + C_CSWeaponBaseGun.m_zoomLevel);
+    const Local_PlayerControllerPtr = cs2.u64(ClientPtr + Client.Other.dwLocalPlayerController);
+    /* */ const Local_TickBase = cs2.u32(Local_PlayerControllerPtr + Client.CBasePlayerController.m_nTickBase);
+
+    const Local_PlayerPawnPtr = cs2.u64(ClientPtr + Client.Other.dwLocalPlayerPawn);
+    /* */ const Local_FlashOverlayAlpha = cs2.f32(Local_PlayerPawnPtr + Client.C_CSPlayerPawnBase.m_flFlashOverlayAlpha);
+    /* */ const Local_IDEntIndex = cs2.i32(Local_PlayerPawnPtr + Client.C_CSPlayerPawn.m_iIDEntIndex);
+    /* */ const Local_IsScoped = cs2.i32(Local_PlayerPawnPtr + Client.C_CSPlayerPawn.m_bIsScoped);
+    /* */ const Local_Player_WeaponServicesPtr = cs2.u64(Local_PlayerPawnPtr + Client.C_BasePlayerPawn.m_pWeaponServices);
+    /*       */ const Local_NextAttack = cs2.f32(Local_Player_WeaponServicesPtr + Client.CCSPlayer_WeaponServices.m_flNextAttack);
+    /* */ const Local_TeamNum = cs2.u8(Local_PlayerPawnPtr + Client.C_BaseEntity.m_iTeamNum);
+    /* */ const Local_WeaponBasePtr = cs2.u64(Local_PlayerPawnPtr + Client.C_CSPlayerPawn.m_pClippingWeapon);
+    /*       */ const Local_Clip1 = cs2.i32(Local_WeaponBasePtr + Client.C_BasePlayerWeapon.m_iClip1);
+    /*       */ const Local_NextPrimaryAttackTick = cs2.u32(Local_WeaponBasePtr + Client.C_BasePlayerWeapon.m_nNextPrimaryAttackTick);
+    /*       */ const Local_WeaponBaseVDataPtr = cs2.u64(Local_WeaponBasePtr + Client.C_BaseEntity.m_nSubclassID + 0x08n);
+    /*             */ const Local_IsFullAuto = cs2.bool(Local_WeaponBaseVDataPtr + Client.CCSWeaponBaseVData.m_bIsFullAuto);
+    /*             */ const Local_WeaponType = cs2.i32(Local_WeaponBaseVDataPtr + Client.CCSWeaponBaseVData.m_WeaponType);
+    /*       */ const Local_ZoomLevel = cs2.i32(Local_WeaponBasePtr + Client.C_CSWeaponBaseGun.m_zoomLevel);
 
     // Conditions where we should not fire…
     if (CurTime < Local_NextAttack) {
@@ -113,7 +87,7 @@ async function tick(ClientPtr: bigint) {
     // Weapon types: https://swiftlys2.net/sdk/cs2/types/csweapontype
 
     // Get the entity that we're aiming at from the entity list…
-    const EntityListPtr = cs2.u64(ClientPtr + dwEntityList);
+    const EntityListPtr = cs2.u64(ClientPtr + Client.Other.dwEntityList);
     /* */ const EntityChunkPtr = cs2.u64(EntityListPtr + (BigInt(Local_IDEntIndex) >> 0x09n) * 0x08n + 0x10n);
     /*       */ const BaseEntityPtr = cs2.u64(EntityChunkPtr + (BigInt(Local_IDEntIndex) & 0x1ffn) * 0x78n);
     /*             */ const EntityClassInfoPtr = cs2.u64(EntityChunkPtr + (BigInt(Local_IDEntIndex) & 0x1ffn) * 0x78n + 0x08n);
@@ -134,7 +108,7 @@ async function tick(ClientPtr: bigint) {
     }
 
     const PlayerPawnPtr = BaseEntityPtr;
-    /* */ const TeamNum = cs2.u8(PlayerPawnPtr + BigInt(C_BaseEntity.m_iTeamNum));
+    /* */ const TeamNum = cs2.u8(PlayerPawnPtr + Client.C_BaseEntity.m_iTeamNum);
 
     if (TeamNum === Local_TeamNum) {
       return;
@@ -148,7 +122,7 @@ async function tick(ClientPtr: bigint) {
     // If the gun is automatic, hold the trigger until we're no longer aiming at them…
     do {
       await sleep(random() * Delay + Delay);
-    } while (Local_IDEntIndex === cs2.u32(Local_PlayerPawnPtr + C_CSPlayerPawn.m_iIDEntIndex) && Local_IsFullAuto);
+    } while (Local_IDEntIndex === cs2.u32(Local_PlayerPawnPtr + Client.C_CSPlayerPawn.m_iIDEntIndex) && Local_IsFullAuto);
 
     // Release the trigger…
     mouse_event(0x04, 0x00, 0x00, 0x00, 0n);

package/package.json

@@ -22,7 +22,7 @@
     "url": "git://github.com/obscuritysrl/bun-memory.git"
   },
   "type": "module",
-  "version": "1.1.24",
+  "version": "1.1.26",
   "main": "./index.ts",
   "keywords": [
     "bun",
@@ -46,5 +46,9 @@
   ],
   "engines": {
     "bun": ">=1.1.0"
+  },
+  "scripts": {
+    "run:benchmark": "bun ./example/benchmark.ts",
+    "run:trigger-bot": "bun ./example/trigger-bot.ts"
   }
 }