Remove get_cut_time function from multiple places and refactor code

2 files changed, 40 insertions, 35 deletions

diff --git a/include/afl-fuzz.h b/include/afl-fuzz.h
index 8c0e7ca9..d5ad4653 100644
--- a/include/afl-fuzz.h
+++ b/include/afl-fuzz.h
@@ -461,8 +461,9 @@ extern s32 cmplog_child_pid, cmplog_forksrv_pid;
 /* Custom mutators */
 
 struct custom_mutator {
+
   const char* name;
-  void* dh;
+  void*       dh;
 
   /* hooks for the custom mutator function */
 
@@ -485,8 +486,8 @@ struct custom_mutator {
    * @param[in] buf_size Size of the input/output data
    * @param[in] add_buf Buffer containing the additional test case
    * @param[in] add_buf_size Size of the additional test case
-   * @param[in] max_size Maximum size of the mutated output. The mutation must not
-   *     produce data larger than max_size.
+   * @param[in] max_size Maximum size of the mutated output. The mutation must
+   * not produce data larger than max_size.
    * @return Size of the mutated output.
    */
   size_t (*afl_custom_fuzz)(u8** buf, size_t buf_size, u8* add_buf,
@@ -560,7 +561,7 @@ struct custom_mutator {
    *     steps returned in init_trim)
    */
   u32 (*afl_custom_post_trim)(u8 success);
-  
+
   /**
    * Perform a single custom mutation on a given input.
    * This mutation is stacked with the other muatations in havoc.
@@ -574,8 +575,9 @@ struct custom_mutator {
    *     not produce data larger than max_size.
    * @return Size of the mutated output.
    */
-  size_t (*afl_custom_havoc_mutation)(u8** buf, size_t buf_size, size_t max_size);
-  
+  size_t (*afl_custom_havoc_mutation)(u8** buf, size_t buf_size,
+                                      size_t max_size);
+
   /**
    * Return the probability (in percentage) that afl_custom_havoc_mutation
    * is called in havoc. By default it is 6 %.
@@ -598,7 +600,7 @@ struct custom_mutator {
   u8 (*afl_custom_queue_get)(const u8* filename);
 
   /**
-   * Allow for additional analysis (e.g. calling a different tool that does a 
+   * Allow for additional analysis (e.g. calling a different tool that does a
    * different kind of coverage and saves this for the custom mutator).
    *
    * (Optional)
@@ -609,6 +611,7 @@ struct custom_mutator {
    */
   void (*afl_custom_queue_new_entry)(const u8* filename_new_queue,
                                      const u8* filename_orig_queue);
+
 };
 
 extern struct custom_mutator* mutator;
@@ -680,8 +683,8 @@ u8   trim_case_custom(char** argv, struct queue_entry* q, u8* in_buf);
 /* Python */
 #ifdef USE_PYTHON
 
-int    init_py_module(u8*);
-void   finalize_py_module();
+int  init_py_module(u8*);
+void finalize_py_module();
 
 void   init_py(unsigned int);
 size_t fuzz_py(u8**, size_t, u8*, size_t, size_t);
@@ -855,32 +858,6 @@ static u64 next_p2(u64 val) {
 
 }
 
-/* Get unix time in milliseconds */
-
-static u64 get_cur_time(void) {
-
-  struct timeval  tv;
-  struct timezone tz;
-
-  gettimeofday(&tv, &tz);
-
-  return (tv.tv_sec * 1000ULL) + (tv.tv_usec / 1000);
-
-}
-
-/* Get unix time in microseconds */
-
-static u64 get_cur_time_us(void) {
-
-  struct timeval  tv;
-  struct timezone tz;
-
-  gettimeofday(&tv, &tz);
-
-  return (tv.tv_sec * 1000000ULL) + tv.tv_usec;
-
-}
-
 #ifdef _AFL_DOCUMENT_MUTATIONS
 extern u8  do_document;
 extern u32 document_counter;
diff --git a/include/common.h b/include/common.h
index ad1f81fb..0d7f4f0b 100644
--- a/include/common.h
+++ b/include/common.h
@@ -25,6 +25,8 @@
 
 #ifndef __AFLCOMMON_H
 #define __AFLCOMMON_H
+
+#include <sys/time.h>
 #include "types.h"
 
 extern u8* target_path;                 /* Path to target binary            */
@@ -37,3 +39,29 @@ char** get_wine_argv(u8* own_loc, char** argv, int argc);
 char*  get_afl_env(char* env);
 #endif
 
+/* Get unix time in milliseconds */
+
+static u64 get_cur_time(void) {
+
+  struct timeval  tv;
+  struct timezone tz;
+
+  gettimeofday(&tv, &tz);
+
+  return (tv.tv_sec * 1000ULL) + (tv.tv_usec / 1000);
+
+}
+
+/* Get unix time in microseconds */
+
+static u64 get_cur_time_us(void) {
+
+  struct timeval  tv;
+  struct timezone tz;
+
+  gettimeofday(&tv, &tz);
+
+  return (tv.tv_sec * 1000000ULL) + tv.tv_usec;
+
+}
+