From 31a29103939b2f0c7abee973b0dfaa500a544c53 Mon Sep 17 00:00:00 2001
From: "Hatter Jiang@Pixelbook" <jht5945@gmail.com>
Date: Sun, 2 Aug 2020 14:43:13 +0800
Subject: [PATCH] fix: add loop count

---
 Cargo.toml       |  2 +-
 src/util_file.rs | 22 +++++++++++++++++++++-
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index ce115aa..0338048 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "rust_util"
-version = "0.6.1"
+version = "0.6.2"
 authors = ["Hatter Jiang <jht5945@gmail.com>"]
 edition = "2018"
 description = "Hatter's Rust Util"
diff --git a/src/util_file.rs b/src/util_file.rs
index 75e93d4..6a1cad6 100644
--- a/src/util_file.rs
+++ b/src/util_file.rs
@@ -77,10 +77,30 @@ impl Iterator for JoinFilesReader {
     }
 }
 
-pub fn find_parents_exists_dir(dir: &str) -> Option<PathBuf> {
+pub fn find_parents_exists_file(file: &str) -> Option<PathBuf> {
+    let mut loop_count = 0_usize;
     match PathBuf::from(".").canonicalize() {
         Err(_) => None,
         Ok(mut path) => loop {
+            loop_count += 1;
+            if loop_count > 100 { panic!("Loop count more than 100!"); }
+            if path.join(file).is_file() {
+                return Some(path);
+            }
+            if !path.pop() {
+                return None;
+            }
+        }
+    }
+}
+
+pub fn find_parents_exists_dir(dir: &str) -> Option<PathBuf> {
+    let mut loop_count = 0_usize;
+    match PathBuf::from(".").canonicalize() {
+        Err(_) => None,
+        Ok(mut path) => loop {
+            loop_count += 1;
+            if loop_count > 100 { panic!("Loop count more than 100!"); }
             if path.join(dir).is_dir() {
                 return Some(path);
             }