1 files changed, 35 insertions, 0 deletions
diff --git a/libraries/common-0/TrapSystemExit.java b/libraries/common-0/TrapSystemExit.java
new file mode 100644
index 0000000..86bc880
--- /dev/null
+++ b/libraries/common-0/TrapSystemExit.java
@@ -0,0 +1,35 @@
+package cbt.reflect;
+
+import java.security.*;
+import java.lang.reflect.InvocationTargetException;
+
+public abstract class TrapSystemExit<T> {
+  public static SecurityManager createSecurityManager(SecurityManager delegateTo) {
+    return new TrapSecurityManager(delegateTo);
+  }
+
+  public static <T> T run(TrapSystemExit<T> runnable) throws Throwable {
+    boolean trapExitCodeBefore = TrapSecurityManager.trapExitCode().get();
+    try {
+      TrapSecurityManager.trapExitCode().set(true);
+      return runnable.run();
+    } catch (InvocationTargetException exception) {
+      Throwable cause = exception.getCause();
+      if (TrapSecurityManager.isTrappedExit(cause)) {
+        return runnable.wrap(TrapSecurityManager.exitCode(cause));
+      }
+      throw exception;
+    } catch (Exception exception) {
+      if (TrapSecurityManager.isTrappedExit(exception)) {
+        return runnable.wrap(TrapSecurityManager.exitCode(exception));
+      }
+      throw exception;
+    } finally {
+      TrapSecurityManager.trapExitCode().set(trapExitCodeBefore);
+    }
+  }
+
+  public abstract T run() throws Throwable;
+
+  public abstract T wrap(int exitCode);
+}