1 files changed, 14 insertions, 9 deletions
diff --git a/tools/read-dependencies b/tools/read-dependencies
index 3aa6e286..4c88aa87 100755
--- a/tools/read-dependencies
+++ b/tools/read-dependencies
@@ -2,16 +2,21 @@
 
 set -e
 
-if [ -e "setup.py" ]
-then
-    ROOT_DIR="$PWD"
-elif [ -e "../setup.py" ]
-then
-    ROOT_DIR="$PWD/../"
-else
+find_root() {
+   local topd
+   if [ -z "${CLOUD_INIT_TOP_D}" ]; then
+      topd=$(cd "$(dirname "${0}")" && cd .. && pwd)
+   else
+      topd=$(cd "${CLOUD_INIT_TOP_D}" && pwd)
+   fi
+   [ $? -eq 0 -a -f "${topd}/setup.py" ] || return
+   ROOT_DIR="$topd"
+}
+
+if ! find_root; then
     echo "Unable to locate 'setup.py' file that should" \
-          "exist in the cloud-init root directory."
-    exit 1
+          "exist in the cloud-init root directory." 1>&2
+    exit 1;
 fi
 
 REQUIRES="$ROOT_DIR/Requires"