145 lines
6.1 KiB
Bash
Executable File
145 lines
6.1 KiB
Bash
Executable File
#!/bin/bash
|
|
#
|
|
# .---. . .
|
|
# | | |
|
|
# |--- .--. .-. .-. .-.| .-. .--.--. |.-. .-. .--. .-.
|
|
# | | (.-' (.-' ( | ( )| | | | )( )| | (.-'
|
|
# ' ' --' --' -' - -' ' ' -' -' -' ' - --'
|
|
#
|
|
# Freedom in the Cloud
|
|
#
|
|
# Avahi functions
|
|
#
|
|
# License
|
|
# =======
|
|
#
|
|
# Copyright (C) 2014-2016 Bob Mottram <bob@freedombone.net>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
function create_avahi_service {
|
|
service_name=$1
|
|
service_type=$2
|
|
service_protocol=$3
|
|
service_port=$4
|
|
service_description="$5"
|
|
|
|
if [ ! -d /etc/avahi ]; then
|
|
echo $'create_avahi_service: avahi was not installed'
|
|
exit 52925
|
|
fi
|
|
|
|
echo '<?xml version="1.0" standalone="no"?><!--*-nxml-*-->' > /etc/avahi/services/${service_name}.service
|
|
echo '<!DOCTYPE service-group SYSTEM "avahi-service.dtd">' >> /etc/avahi/services/${service_name}.service
|
|
echo '<service-group>' >> /etc/avahi/services/${service_name}.service
|
|
echo " <name replace-wildcards=\"yes\">%h ${service_type}</name>" >> /etc/avahi/services/${service_name}.service
|
|
echo ' <service>' >> /etc/avahi/services/${service_name}.service
|
|
echo " <type>_${service_type}._${service_protocol}</type>" >> /etc/avahi/services/${service_name}.service
|
|
echo " <port>${service_port}</port>" >> /etc/avahi/services/${service_name}.service
|
|
if [ "$service_description" ]; then
|
|
echo " <txt-record>$service_description</txt-record>" >> /etc/avahi/services/${service_name}.service
|
|
fi
|
|
echo ' </service>' >> /etc/avahi/services/${service_name}.service
|
|
echo '</service-group>' >> /etc/avahi/services/${service_name}.service
|
|
}
|
|
|
|
function mesh_avahi {
|
|
chroot "$rootdir" apt-get -yq install avahi-utils avahi-autoipd avahi-dnsconfd
|
|
|
|
decarray=( 1 2 3 4 5 6 7 8 9 0 )
|
|
PEER_ID=${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}
|
|
sed -i "s|#host-name=.*|host-name=P$PEER_ID|g" $rootdir/etc/avahi/avahi-daemon.conf
|
|
|
|
if [ ! -d $rootdir/etc/avahi/services ]; then
|
|
mkdir -p $rootdir/etc/avahi/services
|
|
fi
|
|
|
|
# remove an avahi service which isn't used
|
|
if [ -f $rootdir/etc/avahi/services/udisks.service ]; then
|
|
rm $rootdir/etc/avahi/services/udisks.service
|
|
fi
|
|
|
|
# Add an ssh service
|
|
echo '<?xml version="1.0" standalone="no"?><!--*-nxml-*-->' > $rootdir/etc/avahi/services/ssh.service
|
|
echo '<!DOCTYPE service-group SYSTEM "avahi-service.dtd">' >> $rootdir/etc/avahi/services/ssh.service
|
|
echo '<service-group>' >> $rootdir/etc/avahi/services/ssh.service
|
|
echo ' <name replace-wildcards="yes">%h SSH</name>' >> $rootdir/etc/avahi/services/ssh.service
|
|
echo ' <service>' >> $rootdir/etc/avahi/services/ssh.service
|
|
echo ' <type>_ssh._tcp</type>' >> $rootdir/etc/avahi/services/ssh.service
|
|
echo " <port>$SSH_PORT</port>" >> $rootdir/etc/avahi/services/ssh.service
|
|
echo ' </service>' >> $rootdir/etc/avahi/services/ssh.service
|
|
echo '</service-group>' >> $rootdir/etc/avahi/services/ssh.service
|
|
|
|
# keep the daemon running
|
|
WATCHDOG_SCRIPT_NAME="keepon"
|
|
echo '' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
|
|
echo '# keep avahi daemon running' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
|
|
echo 'AVAHI_RUNNING=$(pgrep avahi-daemon > /dev/null && echo Running)' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
|
|
echo 'if [ ! $AVAHI_RUNNING ]; then' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
|
|
echo ' systemctl start avahi-daemon' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
|
|
echo ' echo -n $CURRENT_DATE >> $LOGFILE' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
|
|
echo ' echo " Avahi daemon restarted" >> $LOGFILE' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
|
|
echo 'fi' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
|
|
chmod +x $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
|
|
}
|
|
|
|
function install_avahi {
|
|
if [ $INSTALLING_MESH ]; then
|
|
mesh_avahi
|
|
return
|
|
fi
|
|
if [[ $(is_completed $FUNCNAME) == "1" ]]; then
|
|
return
|
|
fi
|
|
# only enable avahi if we're doing mesh networking
|
|
if [[ $ENABLE_BATMAN != "yes" ]]; then
|
|
return
|
|
fi
|
|
|
|
${PROJECT_NAME}-mesh-install -f avahi
|
|
if [ ! "$?" = "0" ]; then
|
|
echo $'Failed to install avahi'
|
|
exit 68442
|
|
fi
|
|
|
|
if [ $DEFAULT_DOMAIN_NAME ]; then
|
|
sed -i "s|#host-name=.*|host-name=$DEFAULT_DOMAIN_NAME|g" /etc/avahi/avahi-daemon.conf
|
|
sed -i "s|host-name=.*|host-name=$DEFAULT_DOMAIN_NAME|g" /etc/avahi/avahi-daemon.conf
|
|
else
|
|
decarray=( 1 2 3 4 5 6 7 8 9 0 )
|
|
PEER_ID=${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}
|
|
sed -i "s|#host-name=.*|host-name=P$PEER_ID|g" /etc/avahi/avahi-daemon.conf
|
|
sed -i "s|host-name=.*|host-name=P$PEER_ID|g" /etc/avahi/avahi-daemon.conf
|
|
fi
|
|
|
|
mark_completed $FUNCNAME
|
|
}
|
|
|
|
function configure_firewall_for_avahi {
|
|
if [[ $(is_completed $FUNCNAME) == "1" ]]; then
|
|
return
|
|
fi
|
|
iptables -A INPUT -p tcp --dport 548 -j ACCEPT
|
|
iptables -A INPUT -p udp --dport 548 -j ACCEPT
|
|
iptables -A INPUT -p tcp --dport 5353 -j ACCEPT
|
|
iptables -A INPUT -p udp --dport 5353 -j ACCEPT
|
|
iptables -A INPUT -p tcp --dport 5354 -j ACCEPT
|
|
iptables -A INPUT -p udp --dport 5354 -j ACCEPT
|
|
function_check save_firewall_settings
|
|
save_firewall_settings
|
|
mark_completed $FUNCNAME
|
|
}
|
|
|
|
# NOTE: deliberately there is no "exit 0"
|