From 1fdcbd3d0a4bc43a598ee59f9572e3bf2327e112 Mon Sep 17 00:00:00 2001
From: Chad Baxter <cbax@doslabelectronics.com>
Date: Fri, 23 Apr 2021 02:47:41 -0400
Subject: [PATCH] bugfixes

---
 src/request_builder.rs |  2 +-
 src/thread_manager.rs  | 22 +++++++++++++++++-----
 2 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/src/request_builder.rs b/src/request_builder.rs
index 5dcaaf8..4072967 100644
--- a/src/request_builder.rs
+++ b/src/request_builder.rs
@@ -32,6 +32,6 @@ pub fn build_form(fields: PostFields, data: PostData) -> multipart::Form {
 }
 
 pub fn build_request(form: multipart::Form, url: String) -> Request {
-    let client = reqwest::blocking::Client::new();
+    let client = reqwest::blocking::Client::builder().user_agent(fakeit::user_agent::random_platform()).build().unwrap();
     return client.post(url).multipart(form).build().unwrap();
 }
diff --git a/src/thread_manager.rs b/src/thread_manager.rs
index 2a84a2d..4fcc0f6 100644
--- a/src/thread_manager.rs
+++ b/src/thread_manager.rs
@@ -2,6 +2,8 @@ use crate::types::*;
 use crate::request_builder::*;
 use crate::generator::*;
 
+use std::io::Write;
+
 pub fn execute(fields: PostFields, url: String, domain: String, threads: u64, debug: bool) {
     for _ in 0..threads {
         let fields_clone = fields.clone();
@@ -30,10 +32,13 @@ pub fn execute(fields: PostFields, url: String, domain: String, threads: u64, de
                 ) {
                     Ok(o) => Some(o),
                     Err(e) => {
+                        println!("{:?}", e);
                         if debug_clone1 {
                             println!("{:?}", e);
+                            std::io::stdout().flush().ok().expect("Could not flush stdout");
                         } else {
-                            eprint!("✘");
+                            eprint!("x");
+                            std::io::stderr().flush().ok().expect("Could not flush stdout");
                         }
                         None
                   }
@@ -42,13 +47,20 @@ pub fn execute(fields: PostFields, url: String, domain: String, threads: u64, de
                     let res = response.unwrap();
                     if res.status().is_success() | res.status().is_redirection() {
                         if debug_clone1 {
-                            println!("{:?}", res.headers());
-                            println!("{:?}", res.text());
+                            println!("{:?}", res);
+                            std::io::stdout().flush().ok().expect("Could not flush stdout");
                         } else {
-                            print!("✔");
+                            print!(".");
+                            std::io::stdout().flush().ok().expect("Could not flush stdout");
                         }
                     } else {
-                        print!("✘");
+                        if debug_clone1 {
+                            println!("{:?}", res);
+                            std::io::stdout().flush().ok().expect("Could not flush stdout");
+                        } else {
+                            print!("x");
+                            std::io::stdout().flush().ok().expect("Could not flush stdout");
+                        }
                     }
                 }
             }