From d2da43aa823e1f6b9f95e37dd6db1c0b884d170b Mon Sep 17 00:00:00 2001
From: ruti <>
Date: Thu, 2 May 2024 21:23:14 +0300
Subject: [PATCH] fix: invalid index

---
 extend.c | 9 +++++----
 main.c   | 6 +++++-
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/extend.c b/extend.c
index 08b0939..9db28b9 100644
--- a/extend.c
+++ b/extend.c
@@ -17,8 +17,8 @@
 #include "error.h"
 #include "params.h"
 
-#include <desync.h>
-#include <packets.h>
+#include "desync.h"
+#include "packets.h"
 
 
 int set_timeout(int fd, unsigned int s)
@@ -90,7 +90,7 @@ int ext_connect(struct poolhd *pool, struct eval *val,
         struct sockaddr_ina *dst, int next, int m)
 {
     struct desync_params *dp = &params.dp[m];
-    if (dp->to_ip == 2) {
+    if (dp->to_ip) {
         struct sockaddr_ina addr = { .in6 = dp->addr };
         if (!addr.in.sin_port) {
             addr.in.sin_port = dst->in.sin_port;
@@ -106,7 +106,8 @@ int connect_hook(struct poolhd *pool, struct eval *val,
 {
     int m = mode_add_get(dst, -1);
     val->cache = (m == 0);
-    val->attempt = m < 0 ? 0 : m;
+    m = m < 0 ? 0 : m;
+    val->attempt = m;
     
     if (params.late_conn) {
         val->type = EV_DESYNC;
diff --git a/main.c b/main.c
index b5e71e4..fbf020b 100644
--- a/main.c
+++ b/main.c
@@ -416,6 +416,10 @@ void clear_params(void)
                 free(s.parts);
                 s.parts = 0;
             }
+            if (s.tlsrec != 0) {
+                free(s.tlsrec);
+                s.tlsrec = 0;
+            }
             if (s.fake_data.data) {
                 FREE_MAP(s.fake_data.data, s.fake_data.size);
                 s.fake_data.data = 0;
@@ -659,7 +663,7 @@ int main(int argc, char **argv)
             if (get_addr_with_port(optarg, (struct sockaddr_ina *)&dp->addr) < 0)
                 invalid = 1;
             else
-                dp->to_ip = 2;
+                dp->to_ip = 1;
             break;
             
         case 's':