1. import java.io.IOException;
  2. import java.nio.file.Files;
  3. import java.nio.file.Paths;
  4. import java.util.concurrent.ExecutorService;
  5. import java.util.concurrent.Executors;
  6. import java.util.concurrent.Future;
  8. public class ApiPayloadBootstrapper {
  10. private final String payloadDirectory;
  11. private final int numThreads;
  13. public ApiPayloadBootstrapper(String payloadDirectory, int numThreads) {
  14. this.payloadDirectory = payloadDirectory;
  15. this.numThreads = numThreads;
  16. }
  18. public void bootstrapPayloads() {
  19. ExecutorService executor = Executors.newFixedThreadPool(numThreads);
  21. try {
  22. Files.walkFileTree(Paths.get(payloadDirectory), new FileTreeEvaluator(executor));
  24. } catch (IOException e) {
  25. System.err.println("Error during file processing: " + e.getMessage());
  26. //Consider more robust error handling, logging, or alerting here.
  27. executor.shutdownNow(); // Attempt to shutdown the executor gracefully.
  28. return;
  29. }
  31. executor.shutdown(); // Shutdown the executor after processing all files.
  32. try {
  33. executor.awaitTermination(60, java.util.concurrent.TimeUnit.SECONDS); //Wait for completion
  34. } catch (InterruptedException e) {
  35. System.err.println("Executor termination interrupted: " + e.getMessage());
  36. }
  37. }
  40. private static class FileTreeEvaluator implements java.io.FileTreeEvaluator {
  41. private final ExecutorService executor;
  43. public FileTreeEvaluator(ExecutorService executor) {
  44. this.executor = executor;
  45. }
  47. @Override
  48. public void evaluate(java.nio.file.Path path, java.nio.file.FileVisitor<java.nio.file.Path> visitor) throws IOException {
  49. if (Files.isDirectory(path)) {
  50. visitor.visitFile(path, null); //Visit the directory itself.
  51. java.nio.file.Files.walk(path, java.nio.file.FileVisitOption.CONCURRENT).forEach(f -> visitor.visitFile(f, null));
  52. } else if (Files.isRegularFile(path) && path.toString().endsWith(".sh") || path.toString().endsWith(".py")) { //Example script extensions.
  53. Future<?> future = executor.submit(() -> {
  54. try {
  55. // Execute the script with graceful failure handling
  56. executeScript(path.toString()); //Pass the absolute path
  57. } catch (Exception e) {
  58. System.err.println("Error executing script " + path.toString() + ": " + e.getMessage());
  59. e.printStackTrace(); //Log detailed error information.
  60. }
  61. });
  62. }
  63. }
  64. }
  66. private void executeScript(String scriptPath) throws Exception {
  67. try {
  68. // Execute the script using a separate process.
  69. Process process = new ProcessBuilder("bash", scriptPath).start();
  71. try {
  72. int exitCode = process.waitFor();
  73. if (exitCode != 0) {
  74. throw new Exception("Script exited with code " + exitCode);
  75. }
  76. System.out.println("Script " + scriptPath + " completed successfully.");
  78. } catch (InterruptedException e) {
  79. throw new Exception("Script execution interrupted: " + e.getMessage());
  80. } finally {
  81. // Ensure the process is closed.
  82. process.destroy();
  83. }
  84. } catch (IOException e) {
  85. throw new Exception("Error executing script: " + e.getMessage());
  86. }
  87. }
  90. public static void main(String[] args) {
  91. String payloadDirectory = "scripts"; //Replace with the actual directory
  92. int numThreads = 4;
  93. ApiPayloadBootstrapper bootstrapper = new ApiPayloadBootstrapper(payloadDirectory, numThreads);
  94. bootstrapper.bootstrapPayloads();
  95. }
  96. }

Add your comment