From 311d556a257db3f94bd2c02b7fae7109c238b140 Mon Sep 17 00:00:00 2001
From: Alex Kotov <kotovalexarian@gmail.com>
Date: Sun, 3 Jan 2021 05:36:06 +0500
Subject: [PATCH] Add function "iterate_ttl"

---
 src/main.rs | 74 +++++++++++++++++++++++++++++++----------------------
 1 file changed, 44 insertions(+), 30 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 921b344..2bc6316 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -32,36 +32,12 @@ fn main() {
     let mut current_ttl = options.first_ttl;
 
     while !reached_host && current_ttl <= options.max_ttl {
-        let mut current_address: Option<IpAddr> = None;
-
-        for sequence in 0..options.nqueries {
-            let request = Request::new(0, sequence);
-
-            send_request(socket, current_ttl, &host, &request);
-
-            set_timeout(socket, 2, 0);
-
-            let mut response: Option<Response> = None;
-
-            let time_limit = SystemTime::now() + Duration::new(2, 0);
-
-            while SystemTime::now() < time_limit {
-                if let Some(tmp_response) = recv_response(socket) {
-                    if tmp_response.does_match_request(&request) {
-                        response = Some(tmp_response);
-                        break
-                    }
-                }
-            }
-
-            if let Some(response) = response {
-                if response.type_ != 11 || response.code != 0 {
-                    continue
-                }
-
-                current_address = Some(response.source);
-            }
-        }
+        let current_address = iterate_ttl(
+            &options,
+            &host,
+            socket,
+            current_ttl,
+        );
 
         if let Some(ip_addr) = current_address {
             println!("{} {}", current_ttl, ip_addr);
@@ -74,6 +50,44 @@ fn main() {
     }
 }
 
+fn iterate_ttl(
+    options: &Options,
+    host: &IpAddr,
+    socket: libc::c_int,
+    current_ttl: u8,
+) -> Option<IpAddr> {
+    for sequence in 0..options.nqueries {
+        let request = Request::new(0, sequence);
+
+        send_request(socket, current_ttl, &host, &request);
+
+        set_timeout(socket, 2, 0);
+
+        let mut response: Option<Response> = None;
+
+        let time_limit = SystemTime::now() + Duration::new(2, 0);
+
+        while SystemTime::now() < time_limit {
+            if let Some(tmp_response) = recv_response(socket) {
+                if tmp_response.does_match_request(&request) {
+                    response = Some(tmp_response);
+                    break
+                }
+            }
+        }
+
+        if let Some(response) = response {
+            if response.type_ != 11 || response.code != 0 {
+                continue
+            }
+
+            return Some(response.source)
+        }
+    }
+
+    return None
+}
+
 fn send_request(
     socket: libc::c_int,
     current_ttl: u8,