freedombone/src/freedombone-utils-wifi

666 lines
20 KiB
Bash
Executable File

#!/bin/bash
# _____ _ _
# | __|___ ___ ___ _| |___ _____| |_ ___ ___ ___
# | __| _| -_| -_| . | . | | . | . | | -_|
# |__| |_| |___|___|___|___|_|_|_|___|___|_|_|___|
#
# Freedom in the Cloud
#
# Wifi functions
#
# License
# =======
#
# Copyright (C) 2015-2018 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/>.
WIFI_CHANNEL=2
WIFI_INTERFACE=wlan0
WIFI_TYPE='wpa2-psk'
WIFI_SSID=
WIFI_PASSPHRASE=
WIFI_HOTSPOT='no'
WIFI_NETWORKS_FILE=~/${PROJECT_NAME}-wifi.cfg
# repo for atheros AR9271 wifi driver
ATHEROS_WIFI_REPO="https://github.com/qca/open-ath9k-htc-firmware.git"
function default_network_config {
echo '# This file describes the network interfaces available on your system' > /etc/network/interfaces
echo '# and how to activate them. For more information, see interfaces(5).' >> /etc/network/interfaces
echo 'source /etc/network/interfaces.d/*' >> /etc/network/interfaces
}
# Debian stretch has a problem where the formerly predictable wlan0 and eth0
# device names get assigned random names. This is a hacky workaround.
# Also adding net.ifnames=0 to kernel options on bootloader may work.
function enable_predictable_device_names {
test_predictable=$(ls -la /etc/udev/rules.d/80-net-setup-link.rules)
if [[ "$test_predictable" != *"/dev/null" ]]; then
ln -s /dev/null /etc/udev/rules.d/80-net-setup-link.rules
update-initramfs -u
fi
}
function wifi_is_running {
wifi_state=$(wpa_cli status)
if [[ "$wifi_state" == *"COMPLETED"* ]]; then
echo "1"
else
echo "0"
fi
}
function wifi_static_network_interface {
NETWORK_IS_STATIC=0
read_config_param "NETWORK_IS_STATIC"
if [ ${NETWORK_IS_STATIC} -eq 0 ]; then
echo '#this line must always be here' > /etc/network/interfaces.d/static
echo 'iface default inet dhcp' >> /etc/network/interfaces.d/static
else
read_config_param "LOCAL_NETWORK_STATIC_IP_ADDRESS"
read_config_param "ROUTER_IP_ADDRESS"
{ echo 'iface default inet static';
echo " address ${LOCAL_NETWORK_STATIC_IP_ADDRESS}";
echo ' netmask 255.255.255.0';
echo " gateway ${ROUTER_IP_ADDRESS}"; } > /etc/network/interfaces.d/static
fi
}
function setup_wifi_atheros {
if [[ $(running_as_root) == "0" ]]; then
return
fi
apt-get -yq install firmware-linux-free
atheros_drivers_file=drivers/firmware-ath9k-htc.deb
if [ ! -f $atheros_drivers_file ]; then
if [ ! -f "$HOME/${PROJECT_NAME}/$atheros_drivers_file" ]; then
return
else
atheros_drivers_file="$HOME/${PROJECT_NAME}/$atheros_drivers_file"
fi
else
atheros_drivers_file=$(pwd)/$atheros_drivers_file
fi
dpkg -i "$atheros_drivers_file"
reset_usb_devices
update_wifi_adaptors 'wlan'
if [ "$IFACE" ]; then
wpa_action "${IFACE}" stop
wpa_cli -i "${IFACE}" terminate
#ifconfig $IFACE up
ifup "$IFACE"
fi
}
function setup_wifi {
if [[ $SYSTEM_TYPE == "mesh"* ]]; then
return
fi
if [ ! $WIFI_SSID ]; then
return
fi
if [ ${#WIFI_SSID} -lt 2 ]; then
return
fi
if [[ $(is_completed "${FUNCNAME[0]}") == "1" ]]; then
return
fi
HOTSPOT='no'
if [[ $WIFI_HOTSPOT != 'no' ]]; then
HOTSPOT='yes'
fi
if [ -f "$WIFI_NETWORKS_FILE" ]; then
"${PROJECT_NAME}-wifi" --networks "$WIFI_NETWORKS_FILE"
mark_complete "${FUNCNAME[0]}"
return
fi
if [[ $WIFI_TYPE != 'none' ]]; then
if [ ! $WIFI_PASSPHRASE ]; then
echo $'No wifi passphrase was given'
return
fi
if [ ${#WIFI_PASSPHRASE} -lt 2 ]; then
echo $'Wifi passphrase was too short'
return
fi
"${PROJECT_NAME}-wifi" -s "$WIFI_SSID" -t "$WIFI_TYPE" -p "$WIFI_PASSPHRASE" --hotspot "$HOTSPOT" --networks "$WIFI_NETWORKS_FILE"
else
"${PROJECT_NAME}-wifi" -s "$WIFI_SSID" -t "$WIFI_TYPE" --hotspot "$HOTSPOT" --networks "$WIFI_NETWORKS_FILE"
fi
mark_completed "${FUNCNAME[0]}"
}
# ath9k_htc driver
function install_atheros_wifi {
if [[ $(is_completed "${FUNCNAME[0]}") == "1" ]]; then
return
fi
if [ "$INSTALLING_ON_BBB" != "yes" ]; then
return
fi
if [[ $ENABLE_BATMAN != "yes" ]]; then
return
fi
if [ -d "$INSTALL_DIR/open-ath9k-htc-firmware" ]; then
return
fi
# have drivers already been installed ?
if [ -f /lib/firmware/htc_9271.fw ]; then
return
fi
apt-get -yq install build-essential cmake git m4 texinfo
if [ ! -d "$INSTALL_DIR" ]; then
mkdir -p "$INSTALL_DIR"
fi
cd "$INSTALL_DIR" || exit 24678246
if [ ! -d "$INSTALL_DIR/open-ath9k-htc-firmware" ]; then
function_check git_clone
if ! git_clone "$ATHEROS_WIFI_REPO" "$INSTALL_DIR/open-ath9k-htc-firmware"; then
rm -rf "$INSTALL_DIR/open-ath9k-htc-firmware"
exit 74283
fi
fi
cd "$INSTALL_DIR/open-ath9k-htc-firmware" || exit 24682468
git checkout 1.4.0
if ! make toolchain; then
rm -rf "$INSTALL_DIR/open-ath9k-htc-firmware"
exit 24820
fi
if ! make firmware; then
rm -rf "$INSTALL_DIR/open-ath9k-htc-firmware"
exit 63412
fi
if ! cp target_firmware/*.fw /lib/firmware/; then
exit 74681
fi
mark_completed "${FUNCNAME[0]}"
}
function update_wifi_adaptors {
interface_type="$1"
IFACE=
IFACE_SECONDARY=
if [ ! "$interface_type" ]; then
interface_type='wlan'
fi
for i in $(seq 10 -1 0); do
ifdown --force "${interface_type}${i}" 2> /dev/null
done
for i in $(seq 10 -1 0); do
if grep -q "${interface_type}${i}" /proc/net/dev; then
if [ ! $IFACE ]; then
IFACE="wlan${i}"
else
IFACE_SECONDARY="${interface_type}${i}"
return
fi
fi
done
}
function wifi_get_psk {
ssid=$1
passphrase=$2
psk=$(wpa_passphrase "$ssid" "$passphrase" | grep 'psk=' | sed -n 2p | awk -F '=' '{print $2}')
echo "$psk"
}
function hotspot_off {
if [ ! -f /etc/hostapd/hostapd.conf ]; then
return
fi
systemctl stop hostapd
rm /etc/hostapd/hostapd.conf
if [ -f /etc/network/interfaces_original_static ]; then
cp /etc/network/interfaces_original_static /etc/network/interfaces.d/static
else
if [ -f /etc/network/interfaces.d/static ]; then
rm /etc/network/interfaces.d/static
fi
fi
if [ -f /etc/network/interfaces_original_wifi ]; then
cp /etc/network/interfaces_original_wifi /etc/network/interfaces.d/wifi
else
if [ -f /etc/network/interfaces.d/wifi ]; then
rm /etc/network/interfaces.d/wifi
fi
fi
if [ -f /etc/network/interfaces.d/bridge ]; then
rm /etc/network/interfaces.d/bridge
fi
default_network_config
wpa_action ${WIFI_INTERFACE} stop
wpa_cli -i ${WIFI_INTERFACE} terminate
systemctl restart network-manager
}
function hotspot_on {
if [ ! -f /etc/default/hostapd ]; then
echo $'/etc/default/hostapd was not found'
exit 67241
fi
if [ ${#WIFI_PASSPHRASE} -lt 8 ]; then
echo $'Wifi hotspot passphrase is too short'
exit 25719
fi
sed -i 's|#DAEMON_CONF=.*|DAEMON_CONF="/etc/hostapd/hostapd.conf"|g' /etc/default/hostapd
{ echo '### Wireless network name ###';
echo "interface=$WIFI_INTERFACE";
echo '';
echo '### Set your bridge name ###';
echo 'bridge=br0';
echo '';
echo 'driver=nl80211';
echo "country_code=UK";
echo "ssid=$WIFI_SSID";
echo 'hw_mode=g';
echo 'channel=6';
echo 'wpa=2';
echo "wpa_passphrase=$WIFI_PASSPHRASE";
echo '';
echo '## Key management algorithms ##';
echo 'wpa_key_mgmt=WPA-PSK';
echo '';
echo '## Set cipher suites (encryption algorithms) ##';
echo '## TKIP = Temporal Key Integrity Protocol';
echo '## CCMP = AES in Counter mode with CBC-MAC';
echo 'wpa_pairwise=TKIP';
echo 'rsn_pairwise=CCMP';
echo '';
echo '## Shared Key Authentication ##'
echo 'auth_algs=1';
echo '';
echo '## Accept all MAC address ###';
echo 'macaddr_acl=0'; } > /etc/hostapd/hostapd.conf
if [ ! -f /etc/network/interfaces_original_static ]; then
if [ -f /etc/network/interfaces.d/static ]; then
cp /etc/network/interfaces.d/static /etc/network/interfaces_original_static
fi
fi
if [ ! -f /etc/network/interfaces_original_wifi ]; then
if [ -f /etc/network/interfaces.d/wifi ]; then
cp /etc/network/interfaces.d/wifi /etc/network/interfaces_original_wifi
fi
fi
default_network_config
echo "allow-hotplug $WIFI_INTERFACE" > /etc/network/interfaces.d/wifi
echo "iface $WIFI_INTERFACE inet manual" >> /etc/network/interfaces.d/wifi
echo 'allow-hotplug eth0' >> /etc/network/interfaces.d/static
echo 'iface eth0 inet manual' >> /etc/network/interfaces.d/static
echo 'iface br0 inet static' >> /etc/network/interfaces.d/bridge
echo " bridge_ports $WIFI_INTERFACE eth0" >> /etc/network/interfaces.d/bridge
systemctl restart network-manager
ifup $WIFI_INTERFACE
systemctl restart hostapd
}
function wifi_store_original_network_settings {
if [ ! -f /etc/network/interfaces_original ]; then
if ! grep -q "# wifi enabled" /etc/network/interfaces; then
cp /etc/network/interfaces /etc/network/interfaces_original
fi
fi
}
function wifi_original_network_settings {
remove_config_param "NETWORK_IS_STATIC"
if [ -f /etc/network/interfaces.d/static ]; then
rm /etc/network/interfaces.d/static
fi
if [ -f /etc/network/interfaces.d/wifi ]; then
rm /etc/network/interfaces.d/wifi
fi
if [ -f /etc/network/interfaces.d/bridge ]; then
rm /etc/network/interfaces.d/bridge
fi
}
function wifi_wpa2_psk {
ssid=$1
passphrase=$2
wifi_store_original_network_settings
echo 'allow-hotplug eth0' > /etc/network/interfaces.d/static
echo 'iface eth0 inet dhcp' >> /etc/network/interfaces.d/static
echo "allow-hotplug ${WIFI_INTERFACE}" > /etc/network/interfaces.d/wifi
echo "iface ${WIFI_INTERFACE} inet manual" >> /etc/network/interfaces.d/wifi
echo " wpa-roam $WIFI_CONFIG" >> /etc/network/interfaces.d/wifi
wifi_static_network_interface
wpa_passphrase "$ssid" "$passphrase" > "$WIFI_CONFIG"
systemctl restart network-manager
ifup $WIFI_INTERFACE
}
function wifi_none {
ssid=$1
wifi_store_original_network_settings
echo 'allow-hotplug eth0' > /etc/network/interfaces.d/static
echo 'iface eth0 inet dhcp' >> /etc/network/interfaces.d/static
echo "allow-hotplug ${WIFI_INTERFACE}" > /etc/network/interfaces.d/wifi
echo "iface ${WIFI_INTERFACE} inet manual" >> /etc/network/interfaces.d/wifi
echo " wpa-roam $WIFI_CONFIG" >> /etc/network/interfaces.d/wifi
wifi_static_network_interface
{ echo 'ctrl_interface=/run/wpa_supplicant';
echo 'update_config=1';
echo 'eapol_version=1';
echo '';
echo 'network={'; } > "$WIFI_CONFIG"
if [[ "${ssid}" != $'any' && "${ssid}" != $'all' && "${ssid}" != $'open' ]]; then
echo " ssid=\"${ssid}\"" >> "$WIFI_CONFIG"
fi
echo ' key_mgmt=NONE' >> "$WIFI_CONFIG"
echo '}' >> "$WIFI_CONFIG"
systemctl restart network-manager
ifup $WIFI_INTERFACE
}
function networks_from_file {
if [ ! -f "$WIFI_NETWORKS_FILE" ]; then
exit 4
fi
if [[ $(config_param_exists "WIFI_INTERFACE") == "0" ]]; then
exit 5
fi
read_config_param "WIFI_INTERFACE"
wifi_store_original_network_settings
echo 'allow-hotplug eth0' > /etc/network/interfaces.d/static
echo 'iface eth0 inet dhcp' >> /etc/network/interfaces.d/static
{ echo "allow-hotplug ${WIFI_INTERFACE}";
echo "iface ${WIFI_INTERFACE} inet manual";
echo " wpa-roam $WIFI_CONFIG"; } > /etc/network/interfaces.d/wifi
wifi_static_network_interface
# remove wpa_supplicant.conf if it exists
if [ -f "$WIFI_CONFIG" ]; then
rm -f "$WIFI_CONFIG"
fi
{ echo 'ctrl_interface=/run/wpa_supplicant';
echo 'update_config=1';
echo 'eapol_version=1';
echo ''; } > "$WIFI_CONFIG"
ctr=0
while read -r line
do
if [ ${#line} -gt 1 ]; then
if [[ "$line" != '#'* ]]; then
if [ $ctr -eq 0 ]; then
WIFI_SSID="$line"
fi
if [ $ctr -eq 1 ]; then
WIFI_TYPE="$line"
if [[ $WIFI_TYPE == $'none' || $WIFI_TYPE == $'open' ]]; then
echo 'network={' >> "$WIFI_CONFIG"
if [[ "${WIFI_SSID}" != $'any' && "${WIFI_SSID}" != $'all' && "${WIFI_SSID}" != $'open' ]]; then
echo " ssid=\"${WIFI_SSID}\"" >> "$WIFI_CONFIG"
fi
echo ' key_mgmt=NONE' >> "$WIFI_CONFIG"
echo '}' >> "$WIFI_CONFIG"
ctr=0
continue
fi
fi
if [ $ctr -eq 2 ]; then
WIFI_PASSPHRASE="$line"
wpa_passphrase "$WIFI_SSID" "$WIFI_PASSPHRASE" >> "$WIFI_CONFIG"
ctr=0
continue
fi
ctr=$((ctr + 1))
fi
fi
done < "$WIFI_NETWORKS_FILE"
systemctl restart network-manager
#ifconfig ${WIFI_INTERFACE} up
ifup $WIFI_INTERFACE
}
function wifi_networks_file_header {
{ echo $'# Add wifi networks as follows:';
echo '#';
echo $'# MySSID';
echo $'# wpa2-psk';
echo $'# myWifiPassphrase';
echo '#';
echo $'# AnotherSSID';
echo $'# none';
echo '#'; } > "$WIFI_NETWORKS_FILE"
}
function create_networks_interactive {
remove_config_param "WIFI_INTERFACE"
update_wifi_adaptors 'wlan'
if [ ! "$IFACE" ]; then
# Don't try to configure wifi if there are no adaptors
return
fi
if [ -f "$WIFI_NETWORKS_FILE" ]; then
rm "$WIFI_NETWORKS_FILE"
fi
# By default connect to any open wifi
WIFI_SSID=''
WIFI_TYPE=$'open'
wifi_ctr=0
wifi_networks_done=
wifi_settings_defined=
while [ ! $wifi_networks_done ]
do
data=$(mktemp 2>/dev/null)
wifistr=$"\\nTo use this system via wifi (eg. USB dongle) enter the details below, otherwise just select Ok:"
if [ -f /root/.wifi-only ]; then
wifistr=$"\\nEnter your wifi login details below."
fi
dialog --backtitle $"Freedombone Configuration" \
--title $"Wifi Settings ${wifi_ctr}" \
--form "$wifistr" 13 65 4 \
$"SSID (can be 'any'):" 1 1 "$WIFI_SSID" 1 24 30 30 \
$"Type (open/wpa2-psk):" 2 1 "$WIFI_TYPE" 2 24 10 10 \
$"Passphrase:" 3 1 "$WIFI_PASSPHRASE" 3 24 50 50 \
2> "$data"
sel=$?
case $sel in
1) rm -f "$data"
return;;
255) rm -f "$data"
return;;
esac
WIFI_SSID=$(sed -n 1p < "$data")
WIFI_TYPE=$(sed -n 2p < "$data")
WIFI_PASSPHRASE=$(sed -n 3p < "$data")
rm -f "$data"
# If the wifi-only indicator is present then wifi details must
# be specified, otherwise they're optional
if [ ! -f /root/.wifi-only ]; then
# if these fields are empty then there are no more wifi networks
if [ ${#WIFI_SSID} -lt 2 ]; then
wifi_networks_done='yes'
continue
fi
if [ ${#WIFI_TYPE} -lt 2 ]; then
wifi_networks_done='yes'
continue
fi
fi
if [ ! -f "$WIFI_NETWORKS_FILE" ]; then
wifi_networks_file_header
fi
# update the wifi networks file
{ echo '';
echo "$WIFI_SSID";
echo "$WIFI_TYPE"; } >> "$WIFI_NETWORKS_FILE"
if [ ${#WIFI_PASSPHRASE} -gt 1 ]; then
echo "$WIFI_PASSPHRASE" >> "$WIFI_NETWORKS_FILE"
fi
if [ ${#WIFI_SSID} -gt 1 ]; then
if [ ${#WIFI_TYPE} -gt 1 ]; then
if [[ "${WIFI_TYPE}" == $'none' || "${WIFI_TYPE}" == $'open' ]]; then
write_config_param "WIFI_INTERFACE" "$WIFI_INTERFACE"
return
else
if [ ${#WIFI_PASSPHRASE} -gt 1 ]; then
write_config_param "WIFI_INTERFACE" "$WIFI_INTERFACE"
return
fi
fi
fi
fi
# clear values
WIFI_SSID=
WIFI_PASSPHRASE=
wifi_ctr=$((wifi_ctr + 1))
done
}
function disable_wifi {
if [[ ${1} == 'yes' || ${1} == 'y' ]]; then
hotspot_off
if [ -f /etc/network/interfaces.d/static ]; then
rm /etc/network/interfaces.d/static
fi
if [ -f /etc/network/interfaces.d/wifi ]; then
rm /etc/network/interfaces.d/wifi
fi
if [ -f /etc/network/interfaces.d/bridge ]; then
rm /etc/network/interfaces.d/bridge
fi
default_network_config
remove_config_param "WIFI_INTERFACE"
wpa_action ${WIFI_INTERFACE} stop
wpa_cli -i ${WIFI_INTERFACE} terminate
systemctl restart network-manager
else
networks_from_file
fi
}
function count_wlan {
# counts the number of wlan devices
ctr=0
for i in $(seq 0 1 10); do
if grep -q "wlan${i}" /proc/net/dev; then
ctr=$((ctr + 1))
fi
done
echo $ctr
}
function wifi_exists {
if grep -q "wlan" /proc/net/dev; then
echo "1"
else
echo "0"
fi
}
function remove_wifi_startup_script {
remove_config_param "WIFI_INTERFACE"
systemd_file=/etc/systemd/system/wifistart.service
if [ -f $systemd_file ]; then
systemctl stop wifistart
systemctl disable wifistart
rm $systemd_file
fi
if [ -f "$HOME/${PROJECT_NAME}-wifi.cfg" ]; then
rm "$HOME/${PROJECT_NAME}-wifi.cfg"
fi
}
function create_wifi_startup_script {
systemd_file=/etc/systemd/system/wifistart.service
{ echo '[Unit]';
echo 'Description=WifiStartup (Start wifi networking)';
echo 'After=syslog.target';
echo 'After=network.target';
echo 'After=remote-fs.target';
echo '';
echo '[Service]';
echo 'Type=simple';
echo 'User=root';
echo 'Group=root';
echo 'WorkingDirectory=/root';
echo "ExecStart=/usr/local/bin/${PROJECT_NAME}-wifi --wait 5 2> /dev/null";
echo '';
echo '[Install]';
echo 'WantedBy=multi-user.target'; } > $systemd_file
systemctl enable wifistart
systemctl daemon-reload
}
function remove_wifi_startup_script {
systemd_file=/etc/systemd/system/wifistart.service
if [ -f $systemd_file ]; then
systemctl disable wifistart
systemctl daemon-reload
rm $systemd_file
fi
}
# NOTE: deliberately no exit 0