Merge pull request #58 from ImperialCollegeLondon/ssh-server

Add ssh server support to drunc service

docker-compose.yml

@@ -18,7 +18,8 @@ services:
       - sh
       - -c
       - |
-        drunc-process-manager file:///process-manager-no-kafka.json
+        service ssh start &&
+        drunc-process-manager --log-level debug file:///process-manager-no-kafka.json
     expose:
       - 10054
 volumes:

drunc_docker_service/Dockerfile

@@ -1,10 +1,21 @@
 FROM python:3.11-slim-bookworm
 
 RUN apt-get update \
-  && apt-get -y install --no-install-recommends git ssh \
+  && apt-get -y install --no-install-recommends git ssh openssh-server \
   && rm -fr /var/lib/apt/lists/*
 
 COPY requirements.txt /
 RUN pip install --no-cache-dir -r /requirements.txt
 
 COPY process-manager-no-kafka.json /
+
+EXPOSE 22
+RUN mkdir -p /root/.ssh && \
+    ssh-keygen -b 2048 -t rsa -f /root/.ssh/id_rsa -q -N "" && \
+    cp /root/.ssh/id_rsa.pub /root/.ssh/authorized_keys
+
+RUN echo "PermitRootLogin without-password" >> /etc/ssh/sshd_config && \
+    echo "StrictHostKeyChecking no" >> /etc/ssh/ssh_config && \
+    echo "UserKnownHostsFile /dev/null" >> /etc/ssh/ssh_config
+
+RUN mkdir -p /usr/src/app

scripts/talk_to_process_manager.py

@@ -29,11 +29,11 @@ async def create_session(pmd: ProcessManagerDriver) -> list[ProcessInstance]:
     return [
         item
         async for item in pmd.dummy_boot(
-            user="DUMMY_USER",
+            user="root",
             session_name="sess_name",
             n_processes=1,
-            sleep=2,
-            n_sleeps=2,
+            sleep=5,
+            n_sleeps=4,
         )
     ]