summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorGeorg Pfuetzenreuter2022-05-08 15:03:42 +0200
committerGeorg Pfuetzenreuter2022-05-08 15:03:42 +0200
commita014a0126fa72c9b1630c718d05f1652dfc13fb8 (patch)
tree26982043e094b21c7a3b66e5be7c218b21e8adb0 /src
parent5bb047c96004d2a017423b093957d2b478c4ce56 (diff)
downloadpubsh-web-a014a0126fa72c9b1630c718d05f1652dfc13fb8.tar.gz
pubsh-web-a014a0126fa72c9b1630c718d05f1652dfc13fb8.tar.bz2
pubsh-web-a014a0126fa72c9b1630c718d05f1652dfc13fb8.zip
Shell creation logic
Signed-off-by: Georg Pfuetzenreuter <mail@georg-pfuetzenreuter.net>
Diffstat (limited to 'src')
-rw-r--r--src/main/java/net/libertacasa/pubsh/web/Docker.java51
-rw-r--r--src/main/java/net/libertacasa/pubsh/web/WebApplication.java32
-rw-r--r--src/main/resources/templates/portal.html2
3 files changed, 72 insertions, 13 deletions
diff --git a/src/main/java/net/libertacasa/pubsh/web/Docker.java b/src/main/java/net/libertacasa/pubsh/web/Docker.java
index 9621978..a889703 100644
--- a/src/main/java/net/libertacasa/pubsh/web/Docker.java
+++ b/src/main/java/net/libertacasa/pubsh/web/Docker.java
@@ -11,11 +11,11 @@ import java.util.stream.Collectors;
import com.github.dockerjava.api.DockerClient;
import com.github.dockerjava.api.command.BuildImageResultCallback;
import com.github.dockerjava.api.command.CreateContainerResponse;
-import com.github.dockerjava.api.command.InspectContainerResponse;
import com.github.dockerjava.api.command.ListContainersCmd;
import com.github.dockerjava.api.command.ListImagesCmd;
import com.github.dockerjava.api.command.WaitContainerResultCallback;
import com.github.dockerjava.api.model.Image;
+import com.github.dockerjava.api.model.SearchItem;
import com.github.dockerjava.core.DefaultDockerClientConfig;
import com.github.dockerjava.core.DockerClientConfig;
import com.github.dockerjava.core.DockerClientImpl;
@@ -74,7 +74,7 @@ public class Docker {
public static void deleteContainer(String id) {
Collection<String> stringifiedId = Arrays.asList(id);
- System.out.println(id);
+ //System.out.println(id);
System.out.println(stringifiedId);
List<com.github.dockerjava.api.model.Container> containerQuery = dockerClient.listContainersCmd().withShowAll(false).withIdFilter(stringifiedId).exec();
//apparently listContainersCmd returns a bunch of garbled crap, hence the less ideal toString/contains
@@ -93,7 +93,6 @@ public class Docker {
dockerClient.stopContainerCmd(id).exec();
}
- //System.out.println(containerQuery);
dockerClient.removeContainerCmd(id).exec();
}
@@ -135,10 +134,15 @@ public class Docker {
}
- public static String createContainer(String imgid) {
+ public static String createContainer(String name, String imgid) {
//String containerid = dockerClient.createContainerCmd(imgid).exec();
WaitContainerResultCallback resultCallback = new WaitContainerResultCallback();
- CreateContainerResponse createContainerResponse = dockerClient.createContainerCmd(imgid).exec();
+ CreateContainerResponse createContainerResponse = dockerClient
+ .createContainerCmd(imgid)
+ .withName(name)
+ .withTty(true)
+ .withAttachStdin(true)
+ .exec();
dockerClient.waitContainerCmd(createContainerResponse.getId()).exec(resultCallback);
try {
resultCallback.awaitCompletion();
@@ -153,24 +157,49 @@ public class Docker {
return(containerid);
}
- public static String deleteShell(String user, String id) {
+ public static String createShell(String user, String os, Integer count) {
+ String name1 = user + "_" + os;
+ String name = name1 + "_" + count;
+ String imageid = null;
+ List<Image> imageQuery = Docker.getImages(user + ":" + os);
+
+ System.out.printf("Image query: %s\n", imageQuery);
+
+ if (imageQuery.isEmpty()) {
+ System.out.println("No result, building new image.\n");
+ imageid = Docker.buildImage(user, os, count);
+ }
+ if (! imageQuery.isEmpty()) {
+ System.out.println("Found existing image.\n");
+ Image image = imageQuery.get(0);
+ imageid = image.getId();
+ }
+
+ String containerid = createContainer(name, imageid);
+ System.out.printf("Initialized shell with image ID %s and container ID %s\n", imageid, containerid);
+ return(containerid);
+ }
+
+ public static Integer deleteShell(String user, String id) {
Docker.deleteContainer(id);
List<Image> images = Docker.getImages(user);
for (Image image : images) {
- System.out.println(image.getLabels());
String imageid = image.getId();
System.out.println(imageid);
try {
+ try {
+ Thread.sleep(200);
+ } catch (InterruptedException e) {
+ // TODO Auto-generated catch block
+ e.printStackTrace();
+ }
Docker.deleteImage(imageid);
System.out.println("Removed.");
} catch (com.github.dockerjava.api.exception.ConflictException exception) {
System.out.println("Still being used, skipping.");
}
-
-
}
-
- return("foo");
+ return(null);
}
}
diff --git a/src/main/java/net/libertacasa/pubsh/web/WebApplication.java b/src/main/java/net/libertacasa/pubsh/web/WebApplication.java
index bd497d9..4c06c29 100644
--- a/src/main/java/net/libertacasa/pubsh/web/WebApplication.java
+++ b/src/main/java/net/libertacasa/pubsh/web/WebApplication.java
@@ -158,7 +158,7 @@ public class WebApplication {
System.out.printf("Deletion triggered for ID %s by %s (%s)\n", id, userid, username);
//Docker.deleteContainer(id);
- Docker.deleteShell(username, id);
+ Docker.deleteShell(username, id); //to-do: move to separate mapping
return("redirect:/portal");
}
@@ -220,6 +220,36 @@ public class WebApplication {
return("redirect:/portal");
}
+ @PostMapping(path="/frontend/shell/add",consumes=MediaType.APPLICATION_FORM_URLENCODED_VALUE)
+ public static String addShell(@RequestBody MultiValueMap<String, String> body, HttpServletRequest request, RedirectAttributes redirectAttributes) {
+ KeycloakAuthenticationToken principal = (KeycloakAuthenticationToken) request.getUserPrincipal();
+ String username = null;
+ String userid = principal.getName();
+ IDToken token = principal.getAccount().getKeycloakSecurityContext().getIdToken();
+ Map<String, Object> customClaims = token.getOtherClaims();
+ username = String.valueOf(customClaims.get("username"));
+ String os = (String) body.getFirst("osChoice");
+
+ System.out.printf("New shell with OS %s requested by %s (%s)\n", os, userid, username);
+
+ Random rand = new Random();
+ Integer randomInt = rand.nextInt(9999999-1111);
+ Integer count = randomInt;
+
+ try {
+ String containerid = Docker.createShell(username, os, count);
+ String returnmessage = "Success - spawned shell " + username + "_" + os + "_" + count + " - internal ID: " + containerid;
+ redirectAttributes.addFlashAttribute("message", returnmessage);
+ } catch (Exception exception) {
+ redirectAttributes.addFlashAttribute("message", "Success!");
+ String returnmessage = "Error - failed to build image :-(";
+ redirectAttributes.addFlashAttribute("message", returnmessage);
+ System.out.println(exception);
+ }
+
+ return("redirect:/portal");
+ }
+
@GetMapping(path = "/logout")
public String logout(HttpServletRequest request) throws ServletException {
request.logout();
diff --git a/src/main/resources/templates/portal.html b/src/main/resources/templates/portal.html
index 0088e82..568e4da 100644
--- a/src/main/resources/templates/portal.html
+++ b/src/main/resources/templates/portal.html
@@ -9,7 +9,7 @@
Hello, <span th:text="${username}"></span>.
</h2>
<h3>Generate new throw-away shell:</h3>
- <form th:object="${osChoice}" th:id="request_pseudoform" action="#" th:action="@{'/frontend/container/add'}" th:method="post" th:os="${osChoice}" th:onsubmit="return confirm('You are about to generate a shell with the OS ' + this.getAttribute('osoption') + ' - please be patient after you confirm, as the generation may take a short while.');">
+ <form th:object="${osChoice}" th:id="request_pseudoform" action="#" th:action="@{'/frontend/shell/add'}" th:method="post" th:os="${osChoice}" th:onsubmit="return confirm('You are about to generate a shell with the OS ' + this.getAttribute('osoption') + ' - please be patient after you confirm, as the generation may take a short while.');">
<select class="form-control" th:object="${osChoice}" name="osChoice">
<option value="">Select operating system ...</option>
<option