1 files changed, 21 insertions, 2 deletions

diff --git a/src/server.cpp b/src/server.cpp
index 6feb55b..3766906 100644
--- a/src/server.cpp
+++ b/src/server.cpp
@@ -31,6 +31,8 @@ int g_srv_health = -1;
 int g_srv_maxhealth = -1;
 int g_srv_lifestate = -1;
 int g_srv_modelindex = -1;
+int g_srv_absvelocity = -1;
+int g_srv_velocity = -1;
 static bool g_srv_ready = false;
 static bool g_srv_checked = false;
 
@@ -105,12 +107,14 @@ bool resolve_server_netvars() {
   g_srv_maxhealth = datamap_offset("m_iMaxHealth", FIELD_INTEGER);
   g_srv_lifestate = datamap_offset("m_lifeState", FIELD_CHARACTER);
   g_srv_modelindex = datamap_offset("m_nModelIndex", 7); // FIELD_SHORT
+  g_srv_absvelocity = datamap_offset("m_vecAbsVelocity", 3); // FIELD_VECTOR
+  g_srv_velocity = datamap_offset("m_vecVelocity", 3);       // FIELD_VECTOR
   g_srv_ready = g_srv_health > 0;
   if (g_srv_ready) {
     logf("server datamap (%s): health=%d maxhealth=%d lifeste=%d "
-         "modelindex=%d",
+         "modelindex=%d absvel=%d vel=%d",
          g_server.path.c_str(), g_srv_health, g_srv_maxhealth, g_srv_lifestate,
-         g_srv_modelindex);
+         g_srv_modelindex, g_srv_absvelocity, g_srv_velocity);
     fail_logged = false;
   } else if (!fail_logged) {
     fail_logged = true;
@@ -175,3 +179,18 @@ int server_model_index(void *base) {
     return -1;
   return (int)*(short *)((uintptr_t)base + g_srv_modelindex);
 }
+
+// Set the authoritative vertical velocity of a server entity. m_vecAbsVelocity
+// is what CGameMovement::SetupMove reads through GetAbsVelocity(); writing the
+// networked m_vecVelocity directly is deliberately avoided because the server
+// datamap scan can find a same-named field on another class first. The movement
+// code syncs m_vecVelocity through SetAbsVelocity after the command.
+bool server_set_velocity_z(int idx, float vz) {
+  if (!g_srv_ready || !g_iface.engineserver || g_srv_absvelocity <= 0)
+    return false;
+  void *base = nullptr;
+  if (server_entity_health(idx, nullptr, nullptr, &base) < 0 || !base)
+    return false;
+  *(float *)((uintptr_t)base + g_srv_absvelocity + 8) = vz;
+  return true;
+}