entrypoint.sh

#!/bin/bash
set -e
PHOTODNA=false
HASHESDB=false
COUNTRY='BR'



echo -n Populating IPED plugins directory with extra plugins...
if [ -d /mnt/plugins ] && [ ! -z "$(ls /mnt/plugins)" ]
then
        ORIGDIR=${PWD} 
        cd /opt/IPED/plugins/ && find /mnt/plugins -type f \
                | xargs -I% sh -c 'ln -s "$@" > /dev/null 2>&1 && echo -n $@[OK]...|| echo -n $@[FAILED]...' _ %
        cd ${ORIGDIR}
        echo "Done."
fi

if [ ! -z "$(ls /opt/IPED/plugins/ | grep -i photodna | grep -i '\.jar$' )" ]
then
        PHOTODNA=true

        echo -n Setting PhotoDNA related flags to $PHOTODNA... && \
        sed -i -e "s/enablePhotoDNA =.*/enablePhotoDNA = $PHOTODNA/" /opt/IPED/iped/IPEDConfig.txt && \
        sed -i -e "s/enablePhotoDNALookup =.*/enablePhotoDNALookup = $PHOTODNA/" /opt/IPED/iped/IPEDConfig.txt && \
        echo Done. || echo Failed.
fi


if [ -f /mnt/hashesdb/iped-hashes.db ]
then
        HASHESDB=true         

        echo -n Setting HASHDB related flags to $HASHESDB... && \
        sed -i -e "s/enableHashDBLookup =.*/enableHashDBLookup = $HASHESDB/" /opt/IPED/iped/IPEDConfig.txt && \
        sed -i -e "s/enableLedCarving =.*/enableLedCarving = $HASHESDB/" /opt/IPED/iped/IPEDConfig.txt && \
        echo Done. || echo Failed.

        # check if HASHESDBONTMP is setted, if it is, copy it to tmp dir
        # can be used in cases that hashesdb in on the network and the only way
        # to accelerate things is to put it on tmpdir, that is mandatory to be local
        if [ "$HASHESDBONTMP" == "true" ] 
        then
                echo -n "Copying iped-hashes.db to /mnt/ipedtmp..." && \
                cp -p --update /mnt/hashesdb/iped-hashes.db /mnt/ipedtmp/ && echo -n OK... && \
                echo -n "Updating config..." && \
                sed -i -e "s/hashesDB =.*/hashesDB = \/mnt\/ipedtmp\/iped-hashes.db/" /opt/IPED/iped/LocalConfig.txt && \
                echo OK. || -n echo Failed.

        fi

fi


# Custom flags to be used to modify configuration on runtime
# LocalConfig.txt variables (with iped_ prefix)
for v in \
        iped_locale \
        iped_indexTemp \
        iped_indexTempOnSSD \
        iped_outputOnSSD \
        iped_numThreads \
        iped_hashesDB \
        iped_tskJarPath \
        iped_mplayerPath \
        iped_pluginFolder \
        iped_regripperFolder
do
        echo ${v}=${!v}
        if [ "${!v}" ]
        then
                sed -i -e "s|.*${v#iped_} =.*|${v#iped_} = ${!v}|" /opt/IPED/iped/LocalConfig.txt
        fi
done

# IPEDConfig.txt variables (with iped_ prefix)
for v in \
        iped_enableHash \
        iped_enablePhotoDNA \
        iped_enableHashDBLookup \
        iped_enablePhotoDNALookup \
        iped_enableLedDie \
        iped_enableYahooNSFWDetection \
        iped_enableQRCode \
        iped_ignoreDuplicates \
        iped_exportFileProps \
        iped_processFileSignatures \
        iped_enableFileParsing \
        iped_expandContainers \
        iped_processEmbeddedDisks \
        iped_enableRegexSearch \
        iped_enableAutomaticExportFiles \
        iped_enableLanguageDetect \
        iped_enableNamedEntityRecogniton \
        iped_enableGraphGeneration \
        iped_entropyTest \
        iped_enableSplitLargeBinary \
        iped_indexFileContents \
        iped_enableIndexToElasticSearch \
        iped_enableMinIO \
        iped_enableAudioTranscription \
        iped_enableCarving \
        iped_enableLedCarving \
        iped_enableKnownMetCarving \
        iped_enableImageThumbs \
        iped_enableImageSimilarity \
        iped_enableFaceRecognition \
        iped_enableVideoThumbs \
        iped_enableDocThumbs \
        iped_enableHTMLReport \
        iped_enableOCR
do
        echo ${v}=${!v}
        if [ "${!v}" ]
        then
                sed -i -e "s|.*${v#iped_} =.*|${v#iped_} = ${!v}|" /opt/IPED/iped/IPEDConfig.txt
        fi
done

# IPED variables setting on the config dir (with iped_ prefix).
# supportedMimes, host and port are repeatedly used on configs
# so they were removed. IPED Variables with dots cannot be used 
# as environment variables, removed also
for v in $( for file in $( find /opt/IPED/iped/conf/ -type f | grep Config.txt \
                | grep -v -i regex); do grep "=" $file | grep -v "^host =" \
                | grep -v "^port = " | cut -d "=" -f 1 \
                | grep -v "\." | grep -v "^#" | grep -v supportedMimes \
                | awk '{ if ($0 != "\r" ) {print "iped_"$0;} }';\
            done )        
do
        echo ${v}=${!v}
        if [ "${!v}" ]
        then
                find /opt/IPED/iped/conf/ -type f | grep Config.txt | grep -v -i regex | xargs sed -i -e "s|${v#iped_} =.*|${v#iped_} = ${!v}|" 
        fi
done

echo Setting GraphConfig...
for v in \
        iped_phone_region
do
        echo ${v}=${!v}
        if [ "${!v}" ]
        then
                sed -i -e "s|.*\"$(echo ${v#iped_}| sed 's/_/-/g')\":.*|\"$(echo ${v#iped_}| sed 's/_/-/g')\":\"${!v}\",|" /opt/IPED/iped/conf/GraphConfig.json 
        else 
                sed -i -e "s|.*\"$(echo ${v#iped_}| sed 's/_/-/g')\":.*|\"$(echo ${v#iped_}| sed 's/_/-/g')\":\"${COUNTRY}\",|" /opt/IPED/iped/conf/GraphConfig.json
        fi

done

 
#
# Test for UID presence and, if exist, change the execution for this user id
# 
if [ "${USERID}" ]
then
        echo -n "Adding user for command execution..." && \
        useradd --uid ${USERID} -U tmpuser -m && echo "user added with UID ${USERID}." && \
        echo -n "Creating mplayer config on user environment..." && \
        sudo -u tmpuser mplayer >/dev/null 2>&1 && echo "OK" && \
        echo -n "Configuring cache of tmpuser ..." && \
        ln -fs /root/.cache /home/tmpuser/.cache && \
        chown -RL tmpuser:tmpuser /root/.cache && chmod +x /root && \
        echo "Executing command as UID $USERID..." && \
        sudo -u tmpuser --chdir=${PWD} --preserve-env=SAL_USE_VCLPLUGIN,JAVA_HOME,LD_LIBRARY_PATH,IPED_VERSION $@ || \
        echo "Running as UID $USERID Failed."         
        
else                 
        # no arguments = bash, otherwise exec then
        echo "Executing command as ROOT..."
        exec "$@"
fi