From 64bf8236d1d15437837e584616b10b5aa41b9f67 Mon Sep 17 00:00:00 2001
From: Fred Hebert <mononcqc@ferd.ca>
Date: Mon, 29 Jul 2024 00:40:25 +0000
Subject: [PATCH] Fix multipart failures

multipart files didn't ensure directories existed, which caused failures
in nested directory structures.

Additionally, when crashing, multipart error messages weren't truncated
and nodes could use a lot of memory.
---
 apps/revault/src/revault_fsm.erl | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/apps/revault/src/revault_fsm.erl b/apps/revault/src/revault_fsm.erl
index 7e64091..187c0d0 100644
--- a/apps/revault/src/revault_fsm.erl
+++ b/apps/revault/src/revault_fsm.erl
@@ -931,6 +931,9 @@ format_status(Status) ->
 format_status_msg({info, {revault, Marker, {file, F, Meta, Bin}}}) when byte_size(Bin) > 64 ->
     BinSize = integer_to_binary(byte_size(Bin)),
     {revault, Marker, {file, F, Meta, <<"[truncated: ", BinSize/binary, " bytes]">>}};
+format_status_msg({info, {revault, Marker, {file, F, Meta, PartNum, PartTotal, Bin}}}) when byte_size(Bin) > 64 ->
+    BinSize = integer_to_binary(byte_size(Bin)),
+    {revault, Marker, {file, F, Meta, PartNum, PartTotal, <<"[truncated: ", BinSize/binary, " bytes]">>}};
 format_status_msg(Term) -> Term.
 
 %%%%%%%%%%%%%%%
@@ -1137,6 +1140,7 @@ handle_multipart_file(MPState, F, Hash, PartNum, PartTotal, Bin) when PartNum ==
     %% use an unpredictable rand file since this may be one of many incidentally
     %% conflicting files sharing the same name, even if unlikely. Better to protect
     %% against overlapping runs.
+    ok = revault_file:ensure_dir(F),
     State = revault_file:multipart_init(F, PartTotal, Hash),
     {ok, NewState} = revault_file:multipart_update(State, F, PartNum, PartTotal, Hash, Bin),
     {new, MPState#{F => NewState}};