Check network on start and reset db

services/chainstorage/src/main/java/tech/pegasys/teku/services/chainstorage/StorageService.java

@@ -16,6 +16,9 @@
 import static tech.pegasys.teku.infrastructure.async.AsyncRunnerFactory.DEFAULT_MAX_QUEUE_SIZE;
 import static tech.pegasys.teku.spec.config.Constants.STORAGE_QUERY_CHANNEL_PARALLELISM;
 
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
 import java.util.Optional;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
@@ -42,6 +45,7 @@
 import tech.pegasys.teku.storage.server.RetryingStorageUpdateChannel;
 import tech.pegasys.teku.storage.server.StorageConfiguration;
 import tech.pegasys.teku.storage.server.VersionedDatabaseFactory;
+import tech.pegasys.teku.storage.server.network.EphemeryException;
 import tech.pegasys.teku.storage.server.pruner.BlobSidecarPruner;
 import tech.pegasys.teku.storage.server.pruner.BlockPruner;
 import tech.pegasys.teku.storage.server.pruner.StatePruner;
@@ -80,14 +84,24 @@ protected SafeFuture<?> doStart() {
                       1,
                       DEFAULT_MAX_QUEUE_SIZE,
                       Thread.NORM_PRIORITY - 1);
-              final VersionedDatabaseFactory dbFactory =
+              VersionedDatabaseFactory dbFactory =
                   new VersionedDatabaseFactory(
                       serviceConfig.getMetricsSystem(),
                       serviceConfig.getDataDirLayout().getBeaconDataDirectory(),
                       config);
-              database = dbFactory.createDatabase();
 
-              database.migrate();
+              try {
+                database = dbFactory.createDatabase();
+                database.migrate();
+              } catch (EphemeryException e) {
+                try {
+                  resetDatabaseDirectories(serviceConfig);
+                  database = dbFactory.createDatabase();
+                  database.migrate();
+                } catch (Exception ex) {
+                  throw new RuntimeException("Failed to reset and recreate the database.", ex);
+                }
+              }
 
               final SettableLabelledGauge pruningTimingsLabelledGauge =
                   SettableLabelledGauge.create(
@@ -222,4 +236,30 @@ protected SafeFuture<?> doStop() {
   public ChainStorage getChainStorage() {
     return chainStorage;
   }
+
+  /** This method is called only on Ephemery network when reset is due. */
+  void resetDatabaseDirectories(final ServiceConfig serviceConfig) throws IOException {
+    final Path beaconDataDir = serviceConfig.getDataDirLayout().getBeaconDataDirectory();
+    final Path slashProtectionDir =
+        serviceConfig.getDataDirLayout().getValidatorDataDirectory().resolve("slashprotection");
+    deleteDirectoryRecursively(beaconDataDir);
+    deleteDirectoryRecursively(slashProtectionDir);
+  }
+
+  private void deleteDirectoryRecursively(final Path path) throws IOException {
+    if (Files.isDirectory(path)) {
+      try (var stream = Files.walk(path)) {
+        stream
+            .sorted((o1, o2) -> o2.compareTo(o1))
+            .forEach(
+                p -> {
+                  try {
+                    Files.delete(p);
+                  } catch (IOException e) {
+                    throw new RuntimeException("Failed to delete file: " + p, e);
+                  }
+                });
+      }
+    }
+  }
 }

storage/src/main/java/tech/pegasys/teku/storage/server/VersionedDatabaseFactory.java

@@ -46,7 +46,6 @@ public class VersionedDatabaseFactory implements DatabaseFactory {
   @VisibleForTesting static final String STORAGE_MODE_PATH = "data-storage-mode.txt";
   @VisibleForTesting static final String METADATA_FILENAME = "metadata.yml";
   @VisibleForTesting static final String NETWORK_FILENAME = "network.yml";
-
   private final MetricsSystem metricsSystem;
   private final File dataDirectory;
   private final int maxKnownNodeCacheSize;

storage/src/main/java/tech/pegasys/teku/storage/server/network/DatabaseNetwork.java

@@ -45,6 +45,9 @@ public class DatabaseNetwork {
   @VisibleForTesting
   final Long depositChainId;
 
+  private static final String EPHEMERY_DEPOSIT_CONTRACT_ADDRESS =
+      "0x4242424242424242424242424242424242424242";
+
   @JsonCreator
   DatabaseNetwork(
       @JsonProperty(value = "fork_version") final String forkVersion,
@@ -84,6 +87,20 @@ public static DatabaseNetwork init(
             formatMessage(
                 "deposit contract", depositContractString, databaseNetwork.depositContract));
       }
+      if (databaseNetwork.depositChainId != null
+          && !depositContractString.equals(EPHEMERY_DEPOSIT_CONTRACT_ADDRESS)
+          && !databaseNetwork.depositChainId.equals(depositChainId)) {
+        throw DatabaseStorageException.unrecoverable(
+            formatMessage(
+                "deposit chain id",
+                String.valueOf(depositChainId),
+                String.valueOf(databaseNetwork.depositChainId)));
+      }
+      if (databaseNetwork.depositChainId != null
+          && depositContractString.equals(EPHEMERY_DEPOSIT_CONTRACT_ADDRESS)
+          && !databaseNetwork.depositChainId.equals(depositChainId)) {
+        throw new EphemeryException();
+      }
       return databaseNetwork;
     } else {
       DatabaseNetwork databaseNetwork =
@@ -93,6 +110,10 @@ public static DatabaseNetwork init(
     }
   }
 
+  public Long getDepositChainId() {
+    return depositChainId;
+  }
+
   private static String formatMessage(
       final String fieldName, final String expected, final String actual) {
     return String.format(

storage/src/main/java/tech/pegasys/teku/storage/server/network/EphemeryException.java

@@ -0,0 +1,16 @@
+/*
+ * Copyright Consensys Software Inc., 2024
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+ * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations under the License.
+ */
+
+package tech.pegasys.teku.storage.server.network;
+
+public class EphemeryException extends RuntimeException {}