From 28464373b6c4a550837bef0436675ad1d00c3c17 Mon Sep 17 00:00:00 2001 From: Michael Reber Date: Thu, 21 Nov 2019 13:52:55 +0100 Subject: [PATCH] Add root check to run as unprivileged user only --- linuxacademy-dl.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/linuxacademy-dl.py b/linuxacademy-dl.py index 5be212f..db1bdf4 100644 --- a/linuxacademy-dl.py +++ b/linuxacademy-dl.py @@ -9,7 +9,13 @@ import youtube_dl from selenium import webdriver from selenium.common.exceptions import NoSuchElementException from selenium.webdriver.common.keys import Keys +from os.path import expanduser +# if running as root...kick out +if os.geteuid()==0: + sys.exit("\nPlease run this script as unprivileged user only.\n") + +home = expanduser("~") def usage(error=0, msg=None): if msg: @@ -20,11 +26,11 @@ def usage(error=0, msg=None): print("\tOptions may be replaced with environment variables. Command line options take precedence.") print("\t\tLADL_USERNAME, LADL_PASSWORD, LADL_DIR, LADL_COOKIES\n") print("Examples:") - print("\t$ ./linuxacademy-dl.py -u person@exmple.com -p p@ssw0rd https://linuxacademy.com/cp/modules/view/id/346\n") + print("\t$ ./linuxacademy-dl.py -u person@exmple.com -p my_p@ssw0rd https://linuxacademy.com/cp/modules/view/id/388\n") print("\t$ export LADL_USERNAME=person@example.com") - print("\t$ export LADL_PASSWORD=p@ssw0rd") - print("\t$ export LADL_DIR=/home/jdoe/linux-academy") - print("\t$ ./linuxacademy-dl.py https://linuxacademy.com/cp/modules/view/id/346") + print("\t$ export LADL_PASSWORD=my_p@ssw0rd") + print('\t$ export LADL_DIR=' + str(home) + '/linux-academy-videos') + print("\t$ ./linuxacademy-dl.py https://linuxacademy.com/cp/modules/view/id/388") exit(error)