-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathinstall_tool.sh
199 lines (176 loc) · 6.13 KB
/
install_tool.sh
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
#!/bin/bash
none='\033[0m'
green='\033[0;32m'
red='\033[0;31m'
yellow='\033[33m'
course="CYB102"
branch=${1:-"main"}
scripts_repo="https://raw.githubusercontent.com/codepath/cyb102-vm-setup/${branch}/Scripts/"
# Function to check and perform system updates
perform_system_updates() {
echo -e "Checking for system updates..."
# Check the last update time by examining the modification time of /var/lib/apt/periodic/update-success-stamp
if [ -f /var/lib/apt/periodic/update-success-stamp ]; then
last_update=$(date -r /var/lib/apt/periodic/update-success-stamp +%s)
current_time=$(date +%s)
update_age=$(( (current_time - last_update) / 86400 )) # Convert seconds to days
if [ $update_age -ge 7 ]; then # More than 7 days since last update
echo -e "${yellow}INFO:${none} The system needs to update."
echo -e "${yellow}INFO:${none} During that time, you will need to leave your machine on and connected to the internet."
read -p "Do you want to continue with the update? (y/n) " -n 1 -r
echo
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
echo "Exiting the program. Goodbye!"
exit 1
fi
echo -e "Downloading and running the update script..."
wget "https://raw.githubusercontent.com/codepath/cyb102-vm-setup/main/Scripts/update.sh" -O update.sh
chmod +x update.sh
./update.sh
else
echo -e "${green}System is up-to-date.${none}"
fi
else
echo -e "${yellow}INFO:${none}No update history found... the system needs to update."
echo -e "${yellow}INFO:${none} During that time, you will need to leave your machine on and connected to the internet."
read -p "Do you want to continue with update? (y/n) " -n 1 -r
echo
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
echo "Exiting the program. Goodbye!"
exit 1
fi
wget "https://raw.githubusercontent.com/codepath/cyb102-vm-setup/main/Scripts/update.sh" -O update.sh
chmod +x update.sh
./update.sh
fi
}
# Welcome message
echo -e "Welcome to ${green}CodePath Cybersecurity${none}!"
echo -e "This tool will help you set up your environment for the ${course} course."
if [ "$branch" != "main" ]; then
echo -e "${yellow}WARNING: You are using the a non-default branch ${branch}.${none}"
echo -e "${yellow}This is intended for development purposes only.${none}"
fi
# Check if user is running Ubuntu
if [ -f /etc/os-release ]; then
. /etc/os-release
if [ "$ID" != "ubuntu" ]; then
echo -e "${red}WARNING: You are not running Ubuntu. This script is intended for use on Ubuntu.${none}"
read -p "Do you want to continue anyway? (y/n) " -n 1 -r
echo
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
echo "Exiting the program. Goodbye!"
exit 1
fi
fi
else
echo -e "${yellow}WARNING: Unable to determine Linux distribution. This script is intended for use on Ubuntu.${none}"
read -p "Do you want to continue anyway? (y/n) " -n 1 -r
echo
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
echo "Exiting the program. Goodbye!"
exit 1
fi
fi
# Check if wget is installed
if ! command -v wget &> /dev/null; then
echo -e "${red}ERROR: wget is not installed.${none}"
read -p "Do you want to install wget? (y/n) " -n 1 -r
echo
if [[ $REPLY =~ ^[Yy]$ ]]; then
sudo apt install -y wget
else
echo "Exiting the program. Goodbye!"
exit 1
fi
fi
# Function to install all scripts
install_all_scripts() {
echo -e "Installing all ${course} scripts..."
for i in {1..7} ; do
install_specific_unit "$i"
done
}
# Function to install a specific script
install_specific_unit() {
unit=$1
dirname="tmp_$course_${unit}"
# Move to a temporary directory to download the script
mkdir -p $dirname
cd $dirname
# Download and run Lab script
install_specific_script $unit lab
# Download and run Project script
install_specific_script $unit project
# Move back to the original directory and remove the temporary directory
cd ..
rm -rf $dirname
}
# Function to install a specific script
install_specific_script() {
unit=$1
type=$2
script_name="unit${unit}_${type}.sh"
# Set the script name to rdp_setup.sh if unit is 0
if [ "$unit" == "0" ]; then
script_name="rdp_setup.sh"
fi
# Download and run script
wget "${scripts_repo}${script_name}"
chmod +x ${script_name}
./${script_name} ${branch}
# Check the exit status of the script
status=$?
if [ $status -ne 0 ] ; then
echo -e "${red}Error:${none} ${script_name} exited with status $status"
read -p "Do you want to continue anyway? (y/n) " -n 1 -r
echo
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
echo "Exiting the program. Goodbye!"
cd ..
rm -rf "tmp_$course_${unit}"
exit 1
fi
fi
}
# Function to show the menu and process user input
show_menu() {
echo "Please choose an option:"
echo "1. Install all units"
echo "2. Install a specific unit"
echo "3. Install RDP (Remote Desktop Protocol)"
echo "4. Exit"
read -p "Enter your choice (1-4): " choice
case $choice in
1)
install_all_scripts
;;
2)
unit_number=""
while [[ ! $unit_number =~ ^[1-7]$ ]]; do
read -p "Enter the number of the unit to install (1-7): " unit_number
if [[ ! $unit_number =~ ^[1-7]$ ]]; then
echo "Invalid input, please enter a number between 1 and 7."
fi
done
install_specific_unit "$unit_number"
;;
3)
install_specific_script 0 "rdp"
;;
4)
echo "Exiting the program. Goodbye!"
exit 0
;;
*)
echo "Invalid choice, please try again."
show_menu
;;
esac
}
# Ensure the system is updated before proceeding
perform_system_updates
# Main program loop
while true; do
show_menu
done