-
Notifications
You must be signed in to change notification settings - Fork 0
/
updatePoC.py
67 lines (53 loc) · 2.12 KB
/
updatePoC.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# Search GitHub for PoC with CVE by xp101t
import subprocess
import os
import sys
def run_command(command, cwd=None):
"""Run a shell command and print its output."""
try:
result = subprocess.run(command, shell=True, cwd=cwd, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
print(result.stdout)
if result.stderr:
print(result.stderr, file=sys.stderr)
except subprocess.CalledProcessError as e:
print(f"Error executing command '{command}': {e}", file=sys.stderr)
sys.exit(1)
def update_repo(repo_path):
"""Update the cloned GitHub repository."""
if not os.path.isdir(repo_path):
print(f"Repository path '{repo_path}' does not exist.")
sys.exit(1)
print(f"Updating repository at {repo_path}...")
# Navigate to the repository directory
os.chdir(repo_path)
# Fetch the latest changes from the remote repository
run_command('git fetch')
# Pull the latest changes from the remote branch
run_command('git pull')
def find_repo(base_dir_name='PoC-in-GitHub'):
"""Search for the repository in the current and previous directories."""
possible_dirs = [
os.path.join(os.getcwd(), base_dir_name),
os.path.join(os.getcwd(), '..', base_dir_name),
os.path.join(os.getcwd(), '..', '..', base_dir_name),
os.path.join(os.getcwd(), '..', '..', '..', base_dir_name)
]
for directory in possible_dirs:
if os.path.isdir(directory):
return directory
return None
def main():
repo_url = "https://github.com/nomi-sec/PoC-in-GitHub"
repo_name = repo_url.split('/')[-1]
# Search for the repository in the current and previous directories
repo_path = find_repo(repo_name)
if repo_path:
update_repo(repo_path)
else:
print(f"Cloning repository from {repo_url}...")
# Clone the repository if it does not exist in any of the searched directories
run_command(f'git clone {repo_url}')
repo_path = os.path.join(os.getcwd(), repo_name)
update_repo(repo_path)
if __name__ == "__main__":
main()