From 32600793b3325960dfd1ecc57471f79d7f6ad95c Mon Sep 17 00:00:00 2001
From: Thomas Waldmann <tw@waldmann-edv.de>
Date: Tue, 1 Oct 2024 21:23:29 +0200
Subject: [PATCH] make sure the store gets closed in case of exceptions, fixes
 #8413

---
 src/borg/repository.py | 32 +++++++++++++++++++++-----------
 1 file changed, 21 insertions(+), 11 deletions(-)

diff --git a/src/borg/repository.py b/src/borg/repository.py
index 82a049d7a..319744ffb 100644
--- a/src/borg/repository.py
+++ b/src/borg/repository.py
@@ -130,6 +130,7 @@ def __init__(
             self.store = Store(url, levels=levels_config)
         except StoreBackendError as e:
             raise Error(str(e))
+        self.store_opened = False
         self.version = None
         # long-running repository methods which emit log or progress output are responsible for calling
         # the ._send_log method periodically to get log and progress output transferred to the borg client
@@ -156,7 +157,11 @@ def __enter__(self):
             self.do_create = False
             self.create()
             self.created = True
-        self.open(exclusive=bool(self.exclusive), lock_wait=self.lock_wait, lock=self.do_lock)
+        try:
+            self.open(exclusive=bool(self.exclusive), lock_wait=self.lock_wait, lock=self.do_lock)
+        except Exception:
+            self.close()
+            raise
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
@@ -170,11 +175,13 @@ def create(self):
         """Create a new empty repository"""
         self.store.create()
         self.store.open()
-        self.store.store("config/readme", REPOSITORY_README.encode())
-        self.version = 3
-        self.store.store("config/version", str(self.version).encode())
-        self.store.store("config/id", bin_to_hex(os.urandom(32)).encode())
-        self.store.close()
+        try:
+            self.store.store("config/readme", REPOSITORY_README.encode())
+            self.version = 3
+            self.store.store("config/version", str(self.version).encode())
+            self.store.store("config/id", bin_to_hex(os.urandom(32)).encode())
+        finally:
+            self.store.close()
 
     def _set_id(self, id):
         # for testing: change the id of an existing repository
@@ -207,6 +214,8 @@ def open(self, *, exclusive, lock_wait=None, lock=True):
             self.store.open()
         except StoreBackendDoesNotExist:
             raise self.DoesNotExist(str(self._location)) from None
+        else:
+            self.store_opened = True
         if lock:
             self.lock = Lock(self.store, exclusive, timeout=lock_wait).acquire()
         else:
@@ -224,12 +233,13 @@ def open(self, *, exclusive, lock_wait=None, lock=True):
         self.opened = True
 
     def close(self):
-        if self.opened:
-            if self.lock:
-                self.lock.release()
-                self.lock = None
+        if self.lock:
+            self.lock.release()
+            self.lock = None
+        if self.store_opened:
             self.store.close()
-            self.opened = False
+            self.store_opened = False
+        self.opened = False
 
     def info(self):
         """return some infos about the repo (must be opened first)"""