Start converting to Rocky Linux base image

slurm/controller/docker-entrypoint.sh

@@ -12,44 +12,49 @@ _sshd_host() {
 
 # setup worker ssh to be passwordless
 _ssh_worker() {
-  if [[ ! -d /home/worker ]]; then
-    mkdir -p /home/worker
-    chown -R worker:worker /home/worker
-  fi
-  cat > /home/worker/setup-worker-ssh.sh <<'EOF2'
-mkdir -p ~/.ssh
-chmod 0700 ~/.ssh
-ssh-keygen -b 2048 -t rsa -f ~/.ssh/id_rsa -q -N "" -C "$(whoami)@$(hostname)-$(date -I)"
-cat ~/.ssh/id_rsa.pub > ~/.ssh/authorized_keys
-chmod 0640 ~/.ssh/authorized_keys
-cat >> ~/.ssh/config <<EOF
+    if [[ ! -d /home/worker  ]]; then
+        mkdir -p /home/worker
+        chown -R worker:worker /home/worker
+    fi
+    cat > /home/worker/setup-worker-ssh.sh <<EOF2
+   mkdir -p ~/.ssh
+   chmod 0700 ~/.ssh
+   ssh-keygen -b 2048 -t rsa -f ~/.ssh/id_rsa -q -N "" -C "$(whoami)@$(hostname)-$(date -I)"
+   cat ~/.ssh/id_rsa.pub > ~/.ssh/authorized_keys
+   chmod 0640 ~/.ssh/authorized_keys
+   cat >> ~/.ssh/config <<EOF
 Host *
-  StrictHostKeyChecking no
-  UserKnownHostsFile /dev/null
-  LogLevel QUIET
-EOF
-chmod 0644 ~/.ssh/config
-cd ~/
-tar -czvf ~/worker-secret.tar.gz .ssh
-cd -
+StrictHostKeyChecking no
+UserKnownHostsFile /dev/null
+LogLevel QUIET
 EOF2
-  chmod +x /home/worker/setup-worker-ssh.sh
-  chown worker: /home/worker/setup-worker-ssh.sh
-  sudo -u worker /home/worker/setup-worker-ssh.sh
+    chmod 0644 ~/.ssh/config
+    cd ~/
+    tar -czvf ~/worker-secret.tar.gz .ssh
+    cd -
+    EOF2
+    chmod +x /home/worker/setup-worker-ssh.sh
+    chown worker: /home/worker/setup-worker-ssh.sh
+    sudo -u worker /home/worker/setup-worker-ssh.sh
 }
 
 # start munge and generate key
 _munge_start() {
-  chown -R munge: /etc/munge /var/lib/munge /var/log/munge /var/run/munge
-  chmod 0700 /etc/munge
-  chmod 0711 /var/lib/munge
-  chmod 0700 /var/log/munge
-  chmod 0755 /var/run/munge
-  /sbin/create-munge-key -f
-  sudo -u munge /sbin/munged
-  munge -n
-  munge -n | unmunge
-  remunge
+    chown -R munge: /etc/munge /var/lib/munge /var/log/munge /var/run/munge
+    chmod 0700 /etc/munge
+    chmod 0711 /var/lib/munge
+    chmod 0700 /var/log/munge
+    chmod 0755 /var/run/munge
+    /sbin/create-munge-key -f
+    rngd -r /dev/urandom
+    /usr/sbin/create-munge-key -r -f
+    sh -c  "dd if=/dev/urandom bs=1 count=1024 > /etc/munge/munge.key"
+    chown munge: /etc/munge/munge.key
+    chmod 400 /etc/munge/munge.key
+    sudo -u munge /sbin/munged
+    munge -n
+    munge -n | unmunge
+    remunge
 }
 
 # copy secrets to /.secret directory for other nodes
@@ -76,15 +81,10 @@ _generate_slurm_conf() {
 #
 ClusterName=$CLUSTER_NAME
 SlurmctldHost=$CONTROL_MACHINE
-#SlurmctldHostr=
-#
 SlurmUser=slurm
-#SlurmdUser=root
 SlurmctldPort=$SLURMCTLD_PORT
 SlurmdPort=$SLURMD_PORT
 AuthType=auth/munge
-#JobCredentialPrivateKey=
-#JobCredentialPublicCertificate=
 StateSaveLocation=/var/spool/slurm/ctld
 SlurmdSpoolDir=/var/spool/slurm/d
 SwitchType=switch/none
@@ -92,25 +92,7 @@ MpiDefault=none
 SlurmctldPidFile=/var/run/slurmctld.pid
 SlurmdPidFile=/var/run/slurmd.pid
 ProctrackType=proctrack/pgid
-#PluginDir=
-#FirstJobId=
 ReturnToService=0
-#MaxJobCount=
-#PlugStackConfig=
-#PropagatePrioProcess=
-#PropagateResourceLimits=
-#PropagateResourceLimitsExcept=
-#Prolog=
-#Epilog=
-#SrunProlog=
-#SrunEpilog=
-#TaskProlog=
-#TaskEpilog=
-#TaskPlugin=
-#TrackWCKey=no
-#TreeWidth=50
-#TmpFS=
-#UsePAM=
 #
 # TIMERS
 SlurmctldTimeout=300
@@ -122,17 +104,7 @@ Waittime=0
 #
 # SCHEDULING
 SchedulerType=sched/backfill
-#SchedulerAuth=
-#SelectType=select/linear
 FastSchedule=1
-#PriorityType=priority/multifactor
-#PriorityDecayHalfLife=14-0
-#PriorityUsageResetPeriod=14-0
-#PriorityWeightFairshare=100000
-#PriorityWeightAge=1000
-#PriorityWeightPartition=10000
-#PriorityWeightJobSize=1000
-#PriorityMaxAge=1-0
 #
 # LOGGING
 SlurmctldDebug=3
@@ -161,33 +133,31 @@ EOF
 
 # run slurmctld
 _slurmctld() {
-  if $USE_SLURMDBD; then
-    echo -n "cheking for slurmdbd.conf"
-    while [ ! -f /.secret/slurmdbd.conf ]; do
-      echo -n "."
-      sleep 1
-    done
-    echo ""
-  fi
-  mkdir -p /var/spool/slurm/ctld \
-    /var/spool/slurm/d \
-    /var/log/slurm
-  chown -R slurm: /var/spool/slurm/ctld \
-    /var/spool/slurm/d \
-    /var/log/slurm
-  touch /var/log/slurmctld.log
-  chown slurm: /var/log/slurmctld.log
-  if [[ ! -f /home/config/slurm.conf ]]; then
-    echo "### generate slurm.conf ###"
-    _generate_slurm_conf
-  else
-    echo "### use provided slurm.conf ###"
-    cp /home/config/slurm.conf /etc/slurm/slurm.conf
-  fi
-  sacctmgr -i add cluster "${CLUSTER_NAME}"
-  sleep 2s
-  /usr/sbin/slurmctld
-  cp -f /etc/slurm/slurm.conf /.secret/
+    cd /root/rpmbuild/RPMS/aarch64
+    yum -y --nogpgcheck localinstall slurm-22.05.6-1.el8.aarch64.rpm slurm-perlapi-22.05.6-1.el8.aarch64.rpm  slurm-slurmctld-22.05.6-1.el8.aarch64.rpm
+    if $USE_SLURMDBD; then
+        echo -n "checking for slurmdbd.conf"
+        while [ ! -f /.secret/slurmdbd.conf ]; do
+            echo -n "."
+            sleep 1
+        done
+        echo ""
+    fi
+    mkdir -p /var/spool/slurm/ctld /var/spool/slurmd  /var/log/slurm /etc/slurm
+    chown -R slurm: /var/spool/slurm/ctld /var/spool/slurmd  /var/log/slurm
+    touch /var/log/slurmctld.log
+    chown slurm: /var/log/slurmctld.log
+    if [[ ! -f /home/config/slurm.conf ]]; then
+        echo "### generate slurm.conf ###"
+        _generate_slurm_conf
+    else
+        echo "### use provided slurm.conf ###"
+        cp /home/config/slurm.conf /etc/slurm/slurm.conf
+    fi
+    sacctmgr -i add cluster "${CLUSTER_NAME}"
+    sleep 2s
+    /usr/sbin/slurmctld
+    cp -f /etc/slurm/slurm.conf /.secret/
 }
 
 ### main ###