401-ch31.py

#!/usr/bin/env python3

# Script:                       Create Signature-based Malware Detection part 1
# Author:                       Scotty Jokon
# Date of latest revision:      2/20/2024
# Purpose:                      Challenge 31
# Resources:                    https://github.com/codefellows/seattle-cybersecurity-401d10/blob/main/class-31/challenges/DEMO.md, tps://chat.openai.com/share/176700ef-faf8-455b-aee6-82a4f1151781, https://github.com/ymoua27/401-challenges/blob/main/ch31.py

import os

def search_file(file_name, directory):
    hits = 0
    searched_files = 0
    for root, _, files in os.walk(directory):
        for file in files:
            if file == file_name:
                hits += 1
                print(f"Found: {file} at {os.path.join(root, file)}")
            searched_files += 1
    return searched_files, hits

def main():
    file_name = input("Enter the file name to search for: ")
    directory = input("Enter the directory to search in: ")

    if not os.path.exists(directory):
        print("Directory not found.")
        return

    searched_files, hits = search_file(file_name, directory)
    
    print(f"\nSearched {searched_files} files. Found {hits} hits.")

if __name__ == "__main__":
    main()