From cd7418e3969e63bd048c48f0709df44fb1d3fda0 Mon Sep 17 00:00:00 2001
From: tylerjmchugh <Tyler.McHugh@dfo-mpo.gc.ca>
Date: Mon, 30 Dec 2024 14:51:56 -0500
Subject: [PATCH] Rollback for JCloud

---
 .../geonet/api/records/attachments/JCloudStore.java   | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/datastorages/jcloud/src/main/java/org/fao/geonet/api/records/attachments/JCloudStore.java b/datastorages/jcloud/src/main/java/org/fao/geonet/api/records/attachments/JCloudStore.java
index 10c89045643..bd0edfa50cb 100644
--- a/datastorages/jcloud/src/main/java/org/fao/geonet/api/records/attachments/JCloudStore.java
+++ b/datastorages/jcloud/src/main/java/org/fao/geonet/api/records/attachments/JCloudStore.java
@@ -29,7 +29,9 @@
 import jeeves.server.context.ServiceContext;
 
 import org.apache.commons.collections.MapUtils;
+import org.apache.commons.io.input.CountingInputStream;
 import org.fao.geonet.ApplicationContextHolder;
+import org.fao.geonet.api.exception.GeonetMaxUploadSizeExceededException;
 import org.fao.geonet.api.exception.ResourceNotFoundException;
 import org.fao.geonet.constants.Geonet;
 import org.fao.geonet.domain.MetadataResource;
@@ -41,6 +43,7 @@
 import org.fao.geonet.languages.IsoLanguagesMapper;
 import org.fao.geonet.lib.Lib;
 import org.fao.geonet.resources.JCloudConfiguration;
+import org.fao.geonet.util.FileUtil;
 import org.fao.geonet.utils.IO;
 import org.fao.geonet.utils.Log;
 import org.jclouds.blobstore.ContainerNotFoundException;
@@ -312,6 +315,14 @@ protected MetadataResource putResource(final ServiceContext context, final Strin
                 jCloudConfiguration.getClient().getBlobStore().putBlob(jCloudConfiguration.getContainerName(), blob, multipart());
                 Blob blobResults = jCloudConfiguration.getClient().getBlobStore().getBlob(jCloudConfiguration.getContainerName(), key);
 
+                if (is instanceof LimitedInputStream && ((LimitedInputStream) is).isLimitExceeded()) {
+                    delResource(context, metadataUuid, visibility, filename, approved);
+                    throw new GeonetMaxUploadSizeExceededException("uploadedResourceSizeExceededException")
+                        .withMessageKey("exception.maxUploadSizeExceeded",
+                            new String[]{FileUtil.humanizeFileSize(maxUploadSize)})
+                        .withDescriptionKey("exception.maxUploadSizeExceededUnknownSize.description",
+                            new String[]{FileUtil.humanizeFileSize(maxUploadSize)});
+                }
                 return createResourceDescription(context, metadataUuid, visibility, filename, blobResults.getMetadata(), metadataId, approved);
             } finally {
                 locks.remove(key);