#!/bin/sh
###########################################################################
#
# Copyright (c) 2021-2023 Diality Inc. - All Rights Reserved.
#
# THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
# WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
#
# @file    rootsshaccess_set.sh
#
# @author  (last)      Behrouz NematiPour
# @date    (last)      17-Jun-2023
# @author  (original)  Behrouz NematiPour
# @date    (original)  17-Jun-2023
#
############################################################################


SSHD_CONFIG="/etc/ssh/sshd_config"
ERR_EXECUTION=201

if [ "$1" == "YES" ]; then
  sed -i '/PermitRootLogin/c\PermitRootLogin yes' $SSHD_CONFIG
else
  sed -i '/PermitRootLogin/c\PermitRootLogin no'  $SSHD_CONFIG
fi

if [ $? != 0 ]; then
  echo "Execution Failed"
  exit $ERR_EXECUTION
fi

echo ""
exit 0