1 files changed, 14 insertions, 3 deletions

diff --git a/src/menu.cpp b/src/menu.cpp
index 0ed89d4..4f76529 100644
--- a/src/menu.cpp
+++ b/src/menu.cpp
@@ -63,6 +63,7 @@ static bool g_bhop_autostrafe = true;
 
 static FeatureState g_abh{true, 0};
 static FeatureState g_jumpbug{true, 0};
+static bool g_jumpbug_assist = false;
 static FeatureState g_esp{true, 0};
 static FeatureState g_aimbot{true, 604}; // default Mouse5, rebindable in menu
 static FeatureState g_triggerbot{true, 603}; // default Mouse4
@@ -99,6 +100,7 @@ static MenuNode g_abh_feat{NODE_FEATURE, "ABH", false, 1, {}, nullptr, nullptr,
 static MenuNode g_abh_active{NODE_TOGGLE, "Active", false, 0, {}, &g_abh.active, nullptr, nullptr};
 static MenuNode g_jumpbug_feat{NODE_FEATURE, "Jumpbug", false, 1, {}, nullptr, nullptr, nullptr};
 static MenuNode g_jumpbug_active{NODE_TOGGLE, "Active", false, 0, {}, &g_jumpbug.active, nullptr, nullptr};
+static MenuNode g_jumpbug_assist_node{NODE_TOGGLE, "Assist timing", false, 0, {}, &g_jumpbug_assist, nullptr, nullptr};
 static MenuNode g_esp_feat{NODE_FEATURE, "ESP", false, 1, {}, nullptr, nullptr, nullptr};
 static MenuNode g_esp_active{NODE_TOGGLE, "Active", false, 0, {}, &g_esp.active, nullptr, nullptr};
 static MenuNode g_esp_opt_nodes[ESP_OPT_COUNT] = {
@@ -150,7 +152,8 @@ static void build_tree() {
   g_abh_feat.children[0] = &g_abh_active;
   g_abh_feat.child_count = 1;
   g_jumpbug_feat.children[0] = &g_jumpbug_active;
-  g_jumpbug_feat.child_count = 1;
+  g_jumpbug_feat.children[1] = &g_jumpbug_assist_node;
+  g_jumpbug_feat.child_count = 2;
   g_esp_feat.children[0] = &g_esp_active;
   g_aimbot_feat.children[0] = &g_aimbot_active;
   g_aimbot_feat.children[1] = &g_aimbot_bind;
@@ -302,7 +305,8 @@ static void config_save() {
   fprintf(f, "[bhop]\nactive=%d\nbind=%d\nautostrafe=%d\n", g_bhop.active,
           g_bhop.bind, g_bhop_autostrafe);
   fprintf(f, "[abh]\nactive=%d\nbind=%d\n", g_abh.active, g_abh.bind);
-  fprintf(f, "[jumpbug]\nactive=%d\n", g_jumpbug.active);
+  fprintf(f, "[jumpbug]\nactive=%d\nassist=%d\n", g_jumpbug.active,
+          (int)g_jumpbug_assist);
   fprintf(f, "[aimbot]\nactive=%d\nbind=%d\nsmooth=%d\nvischeck=%d\n"
              "fov=%d\nfov_circle=%d\n"
              "bone_head=%d\nbone_torso=%d\nbone_larm=%d\nbone_rarm=%d\n"
@@ -332,7 +336,8 @@ static void config_load() {
     return;
   char line[256];
   FeatureState *cur = nullptr;
-  bool in_esp = false, in_bhop = false, in_aim = false, in_silent = false;
+  bool in_esp = false, in_bhop = false, in_aim = false, in_silent = false,
+       in_jumpbug = false;
   while (fgets(line, sizeof(line), f)) {
     line[strcspn(line, "\r\n")] = 0;
     if (line[0] == '[') {
@@ -356,6 +361,7 @@ static void config_load() {
       in_bhop = (cur == &g_bhop);
       in_aim = (cur == &g_aimbot);
       in_silent = (cur == &g_silentaim);
+      in_jumpbug = (cur == &g_jumpbug);
     } else if (cur) {
       char *eq = strchr(line, '=');
       if (!eq)
@@ -365,6 +371,9 @@ static void config_load() {
       if (in_bhop && strcmp(key, "autostrafe") == 0) {
         g_bhop_autostrafe = atoi(val) != 0;
       }
+      if (in_jumpbug && strcmp(key, "assist") == 0) {
+        g_jumpbug_assist = atoi(val) != 0;
+      }
       if (in_aim) {
         if (strcmp(key, "smooth") == 0) {
           int v = atoi(val);
@@ -557,6 +566,8 @@ bool esp_option(int opt) {
 
 bool bhop_autostrafe_enabled() { return g_bhop_autostrafe; }
 
+bool jumpbug_assist_enabled() { return g_jumpbug_assist; }
+
 unsigned aimbot_bone_mask() {
   unsigned m = 0;
   if (g_aim_bones[AIM_BONE_HEAD])