Added validation to image download

CoviDok/BLL/Storage/StorageHandler.cs

@@ -40,9 +40,14 @@ namespace CoviDok.BLL.Storage
         }
 
         public async Task GetImage(string bucketName, string ImageId, Action<Stream> callback)
+        {
+            if (await storageProvider.ObjectExists(bucketName, ImageId))
             {
                 await storageProvider.Download(bucketName, ImageId, callback);
             }
+            else throw new KeyNotFoundException();
+            
+        }
 
         public async Task<bool> ImageExists(string bucketName, string ImageId)
         {

CoviDok/Controllers/ImagesController.cs

@@ -67,8 +67,8 @@ namespace CoviDok.Controllers
                 response.Body["reason"] = "unauthorized";
                 return response;
             }
-                        
-            string res = null; ;
+            try {
+                string res = null;
                 await MinioHandler.GetImage(BucketName, imageGet.ImageId, (stream) => {
                     StreamReader reader = new StreamReader(stream);
                     res = reader.ReadToEnd();
@@ -77,5 +77,12 @@ namespace CoviDok.Controllers
                 response.Body["image"] = res;
                 return response;
             }
+            catch (KeyNotFoundException)
+            {
+                response.Status = Status.Error;
+                response.Body["reason"] = "Image not found!";
+                return response;
+            }
+        }
     }
 }