DSM-G600, DNS-3xx and NSA-220 Hack Forum

#!/bin/sh

############
# Settings #
############

# if Transmission consumes more memory (percent) than this it will be restarted
maxTransmissionMemoryUse=90

# These settings control how long will this script wait for Transmission stop to complete
maxTransmissionStopRetryCount=5
maxTransmissionStopRetryWait=60

# Change this to turn on/off emailing what action the script took.
# make sure you configure mutt/msmtp before turning on this feature.
shouldEmail=false
#shouldEmail=true

# Where the email will be sent to
myEmail="my_email@gmail.com"


# Change this to true for verbose output
debug=false
#debug=true


# Set path to Fun_Plug files.
# Fun_Plug 3.0 or 4.0
# ffpPath=/mnt/HD_a2/fun_plug.d
# Fun_Plug 5.0
ffpPath=/ffp
ffpStartPath=${ffpPath}/start

topCmd="${ffpPath}/bin/top"
grepCmd="${ffpPath}/bin/grep"
echoCmd="${ffpPath}/bin/echo"
awkCmd="${ffpPath}/bin/awk"
sedCmd="${ffpPath}/bin/sed"
sleepCmd="${ffpPath}/bin/sleep"
muttCmd="/usr/bin/mutt"
transmissionStart="${ffpStartPath}/transmission.sh"


#########################
# Transmission-Watchdog #
#########################

grepTransmission=$(${topCmd} -b -n 1 | ${grepCmd} transmission-daemon | ${grepCmd} -v grep)
grepRes=$?

# if transmission is running
if [ ${grepRes} -eq 0 ];  then
    ${debug} && ${echoCmd} -n "Transmission is running. "

    transmissionMemoryUse=$(${echoCmd} $grepTransmission | ${awkCmd} '{print $6;}' | ${sedCmd} 's/%//')
    ${debug} && ${echoCmd} "It's using ${transmissionMemoryUse}% of system memory. Allowed: ${maxTransmissionMemoryUse}%"

    if [ ${transmissionMemoryUse} -gt ${maxTransmissionMemoryUse} ]; then
        ${debug} && ${echoCmd} -n "Stopping Transmission: "
        ${transmissionStart} stop

        retryCount=${maxTransmissionStopRetryCount}
        while [ ${retryCount} -gt 0 ]
        do
            ${topCmd} -b -n 1 | ${grepCmd} transmission-daemon | ${grepCmd} -v grep > /dev/null
            if [ $? -ne 0 ]; then
                ${debug} && ${echoCmd} " Done!"
                break;
            fi

            ${debug} && ${echoCmd} -n "."
            retryCount=$((${retryCount} - 1))
            ${sleepCmd} ${maxTransmissionStopRetryWait}
        done

        if [ ${retryCount} -eq 0 ]; then
            timeOut=$((${maxTransmissionStopRetryWait} * ${maxTransmissionStopRetryCount}))
            ${debug} && ${echoCmd} " Timed-out after ${timeOut} seconds!"
            ${shouldEmail} && ${echoCmd} -e "Transmission Restarted due to excessive memory usage.\nActual ${transmissionMemoryUse}%, allowed: ${maxTransmissionMemoryUse}%. But Transmission stop took too long to complete (more than ${timeOut} seconds).\nEither something's wrong or you need to increase the timeout period." | ${muttCmd} -n -F /ffp/usr/local/share/mutt/Muttrc -s "Transmission restart failed" ${myEmail}
            exit 127
        fi

        ${debug} && ${echoCmd} -n "Starting Transmission: "
        ${transmissionStart} start
        ${shouldEmail} && ${echoCmd} -e "Transmission Restarted due to excessive memory usage.\nActual ${transmissionMemoryUse}%, allowed: ${maxTransmissionMemoryUse}%." | ${muttCmd} -n -F /ffp/usr/local/share/mutt/Muttrc -s "Transmission restarted" ${myEmail}
    fi
else
    ${debug} && ${echoCmd} -n "Transmission isn't running. Start it: "
    ${transmissionStart} start
    ${shouldEmail} && ${echoCmd} -e "Transmission wasn't running so it was started." | ${muttCmd} -n -F /ffp/usr/local/share/mutt/Muttrc -s "Transmission started" ${myEmail}
fi

#!/bin/sh

# PROVIDE: EDITCRON_TRANSMISSION
# REQUIRE: LOGIN

#
# Schedule job
# Important: Always run "/ffp/start/editcron_transmission.sh stop" BEFORE making changes to job.
#            Otherwise, the old job will remain scheduled until you restart.
#
schedule="0 */2 * * *"
job="/ffp/bin/transmission-watchdog.sh"

#
# Commands
#
grepcmd="/bin/grep"
crontabcmd="/bin/crontab"
rmcmd="/bin/rm"
echocmd="/bin/echo"

#
# Tempporary files
#
TMPDIR="/tmp"
TMPCRONTXT="${TMPDIR}/crontab.txt"
TMPOTHERCRONTXT="${TMPDIR}/othercrontab.txt"

#
# FFP Start Functions
#
. /ffp/etc/ffp.subr
start_cmd="editcron_start"
stop_cmd="editcron_stop"
status_cmd="editcron_status"

editcron_start()
{
    # grab existing crontab
    ${crontabcmd} -l > ${TMPCRONTXT}

    # check if already scheduled
    cronjobs=$(${grepcmd} "${job}" ${TMPCRONTXT})
    if test -n "${cronjobs}"; then
        ${echocmd} "${job} already scheduled:"
        ${echocmd} "${cronjobs}"
    else
        # add the cron job and install the new one
        ${echocmd} "Scheduling ${schedule} ${job}"
        ${echocmd} "${schedule} ${job}" >> ${TMPCRONTXT}
        ${crontabcmd} ${TMPCRONTXT}
    fi

    # clean up
    ${rmcmd} ${TMPCRONTXT}
}

editcron_stop()
{
    # grab existing crontab
    ${crontabcmd} -l > ${TMPCRONTXT}

    # check if already scheduled
    cronjobs=$(${grepcmd} "${job}" ${TMPCRONTXT})
    if test -z "${cronjobs}"; then
        ${echocmd} "${job} not scheduled"
    else
        # remove the cron job(s) and install the new one
        ${grepcmd} -v "${job}" ${TMPCRONTXT} > ${TMPOTHERCRONTXT}
        ${crontabcmd} ${TMPOTHERCRONTXT}
        ${echocmd} "${job} not longer scheduled"

        # clean up
        ${rmcmd} ${TMPOTHERCRONTXT}
    fi

    # clean up
    ${rmcmd} ${TMPCRONTXT}
}

editcron_status()
{
    # grab existing crontab
    ${crontabcmd} -l > ${TMPCRONTXT}
    cronjobs=$(${grepcmd} "${job}" ${TMPCRONTXT})

    # check if already scheduled
    if test -n "${cronjobs}"; then
        ${echocmd} "${job} is scheduled:"
        ${echocmd} "${cronjobs}"
    else
        ${echocmd} "${job} not scheduled"
    fi

    # clean up
    ${rmcmd} ${TMPCRONTXT}
}

run_rc_command "$1"