Bug #336: Only accept http/https/ftp/file protocols in sources.

1 files changed, 4 insertions, 6 deletions

diff --git a/pkgmk.in b/pkgmk.in
index 00d8e954..9fdf768e 100755
--- a/pkgmk.in
+++ b/pkgmk.in
@@ -34,13 +34,11 @@ error() {
 }
 
 get_filename() {
-	local FILE="`echo $1 | sed 's|^.*://.*/||g'`"
-
-	if [ "$FILE" != "$1" ]; then
-		FILE="$PKGMK_SOURCE_DIR/$FILE"
+	if [[ $1 =~ ^(http|https|ftp|file)://.*/(.+) ]]; then
+		echo "$PKGMK_SOURCE_DIR/${BASH_REMATCH[2]}"
+	else
+		echo $1
 	fi
-
-	echo $FILE
 }
 
 get_basename() {