[refactor] unify path validation

1 files changed, 17 insertions, 15 deletions

diff --git a/server/src/leap/soledad/server/_blobs.py b/server/src/leap/soledad/server/_blobs.py
index 72125d5a..1595a549 100644
--- a/server/src/leap/soledad/server/_blobs.py
+++ b/server/src/leap/soledad/server/_blobs.py
@@ -114,9 +114,7 @@ class FilesystemBlobsBackend(object):
 
     def list_blobs(self, user, request):
         blob_ids = []
-        assert VALID_STRINGS.match(user)
-        base_path = os.path.join(self.path, user)
-        assert self._valid_subdir(base_path)
+        base_path = self._get_path(user)
         for _, _, filenames in os.walk(base_path):
             blob_ids += filenames
         return json.dumps(blob_ids)
@@ -157,8 +155,7 @@ class FilesystemBlobsBackend(object):
         yield fbp.startProducing(open(path, 'wb'))
 
     def get_total_storage(self, user):
-        assert VALID_STRINGS.match(user)
-        return self._get_disk_usage(os.path.join(self.path, user))
+        return self._get_disk_usage(self._get_path(user))
 
     def delete_blob(user, blob_id):
         raise NotImplementedError
@@ -170,26 +167,31 @@ class FilesystemBlobsBackend(object):
     def _get_disk_usage(self, start_path):
         if not os.path.isdir(start_path):
             defer.returnValue(0)
-        assert self._valid_subdir(start_path)
         cmd = ['/usr/bin/du', '-s', '-c', start_path]
         output = yield utils.getProcessOutput(cmd[0], cmd[1:])
         size = output.split()[0]
         defer.returnValue(int(size))
 
-    def _valid_subdir(self, desired_path):
+    def _validate_path(self, desired_path, user, blob_id):
+        if not VALID_STRINGS.match(user):
+            raise Exception("Invalid characters on user: %s" % user)
+        if blob_id and not VALID_STRINGS.match(blob_id):
+            raise Exception("Invalid characters on blob_id: %s" % blob_id)
         desired_path = os.path.realpath(desired_path)  # expand path references
         root = os.path.realpath(self.path)
-        return desired_path.startswith(root + os.sep)
+        if not desired_path.startswith(root + os.sep + user):
+            err = "User %s tried accessing a invalid path: %s" % (user,
+                                                                  desired_path)
+            raise Exception(err)
+        return desired_path
 
-    def _get_path(self, user, blob_id):
-        assert VALID_STRINGS.match(user)
-        assert VALID_STRINGS.match(blob_id)
+    def _get_path(self, user, blob_id=False):
         parts = [user]
-        parts += [blob_id[0], blob_id[0:3], blob_id[0:6]]
-        parts += [blob_id]
+        if blob_id:
+            parts += [blob_id[0], blob_id[0:3], blob_id[0:6]]
+            parts += [blob_id]
         path = os.path.join(self.path, *parts)
-        assert self._valid_subdir(path)
-        return path
+        return self._validate_path(path, user, blob_id)
 
 
 class BlobsResource(resource.Resource):