Add upgrade functionality

1 files changed, 13 insertions, 0 deletions

diff --git a/firefox-hardening.install b/firefox-hardening.install
index 6b47178ee9da..ec4d6199281f 100755
--- a/firefox-hardening.install
+++ b/firefox-hardening.install
@@ -1,11 +1,24 @@
 post_install() {
 	if [ ! -d /etc/firefox-hardening ]; then
 		mkdir -p /etc/firefox-hardening/prefs
+	fi
+
+	if [ ! -f /etc/firefox-hardening/ffh.conf ]; then
 		install -m644 -D "/usr/lib/firefox-hardening/ffh.conf.example" "/etc/firefox-hardening/ffh.conf"
 	fi
+
 	/usr/lib/firefox-hardening/beatthedeviloutofit.sh
 }
 
+post_upgrade() {
+	if [ ! -d /etc/firefox-hardening ] || [ ! -f /etc/firefox-hardening/ffh.conf ]; then
+		post_install
+	else
+		install -m644 -D "/usr/lib/firefox-hardening/ffh.conf.example" "/etc/firefox-hardening/ffh.conf.package"
+		/usr/lib/firefox-hardening/beatthedeviloutofit.sh
+	fi
+}
+
 post_remove() {
 	if [ -d /etc/firefox-hardening ]; then
 		rm -r /etc/firefox-hardening