소스 검색

Beginning of mesh setup

Bob Mottram 9 년 전
부모
커밋
a8d6ffa091
3개의 변경된 파일322개의 추가작업 그리고 25개의 파일을 삭제
  1. 5
    5
      src/freedombone-image
  2. 95
    20
      src/freedombone-image-customise
  3. 222
    0
      src/freedombone-mesh-batman

+ 5
- 5
src/freedombone-image 파일 보기

@@ -220,11 +220,11 @@ if [[ $GENERIC_IMAGE == "yes" ]]; then
220 220
 fi
221 221
 
222 222
 # If this is a mesh variant then create an appropriate script
223
-if [[ $VARIANT == "mesh"* ]]; then
224
-    CONFIG_FILENAME=/tmp/${PROJECT_NAME}_mesh.cfg
225
-    mesh_router_setup_script $CONFIG_FILENAME
226
-    DEFAULT_DOMAIN_NAME=$(cat $CONFIG_FILENAME | grep 'DEFAULT_DOMAIN_NAME' | awk -F '=' '{print $2}')
227
-fi
223
+#if [[ $VARIANT == "mesh"* ]]; then
224
+    #CONFIG_FILENAME=/tmp/${PROJECT_NAME}_mesh.cfg
225
+    #mesh_router_setup_script $CONFIG_FILENAME
226
+    #DEFAULT_DOMAIN_NAME=$(cat $CONFIG_FILENAME | grep 'DEFAULT_DOMAIN_NAME' | awk -F '=' '{print $2}')
227
+#fi
228 228
 
229 229
 if [ ! $PASSWORD ]; then
230 230
     # generate a random password

+ 95
- 20
src/freedombone-image-customise 파일 보기

@@ -362,30 +362,105 @@ atheros_wifi() {
362 362
     fi
363 363
 }
364 364
 
365
+mesh_avahi() {
366
+    chroot "$rootdir" apt-get -y install avahi-utils avahi-autoipd avahi-dnsconfd
367
+
368
+    decarray=( 1 2 3 4 5 6 7 8 9 0 )
369
+    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]}
370
+    sed -i "s|#host-name=.*|host-name=P$PEER_ID|g" $rootdir/etc/avahi/avahi-daemon.conf
371
+
372
+    if [ ! -d $rootdir/etc/avahi/services ]; then
373
+        mkdir -p $rootdir/etc/avahi/services
374
+    fi
375
+
376
+    # remove an avahi service which isn't used
377
+    if [ -f $rootdir/etc/avahi/services/udisks.service ]; then
378
+        rm $rootdir/etc/avahi/services/udisks.service
379
+    fi
380
+
381
+    # Add an ssh service
382
+    echo '<?xml version="1.0" standalone="no"?><!--*-nxml-*-->' > $rootdir/etc/avahi/services/ssh.service
383
+    echo '<!DOCTYPE service-group SYSTEM "avahi-service.dtd">' >> $rootdir/etc/avahi/services/ssh.service
384
+    echo '<service-group>' >> $rootdir/etc/avahi/services/ssh.service
385
+    echo '  <name replace-wildcards="yes">%h SSH</name>' >> $rootdir/etc/avahi/services/ssh.service
386
+    echo '  <service>' >> $rootdir/etc/avahi/services/ssh.service
387
+    echo '    <type>_ssh._tcp</type>' >> $rootdir/etc/avahi/services/ssh.service
388
+    echo "    <port>$SSH_PORT</port>" >> $rootdir/etc/avahi/services/ssh.service
389
+    echo '  </service>' >> $rootdir/etc/avahi/services/ssh.service
390
+    echo '</service-group>' >> $rootdir/etc/avahi/services/ssh.service
391
+
392
+    # keep the daemon running
393
+	WATCHDOG_SCRIPT_NAME="keepon"
394
+    echo '' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
395
+    echo '# keep avahi daemon running' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
396
+    echo 'AVAHI_RUNNING=$(pgrep avahi-daemon > /dev/null && echo Running)' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
397
+    echo 'if [ ! $AVAHI_RUNNING ]; then' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
398
+    echo '  systemctl start avahi-daemon' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
399
+    echo '  echo -n $CURRENT_DATE >> $LOGFILE' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
400
+    echo '  echo " Avahi daemon restarted" >> $LOGFILE' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
401
+    echo 'fi' >> $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
402
+	chmod +x $rootdir/usr/bin/$WATCHDOG_SCRIPT_NAME
403
+}
404
+
405
+function mesh_batman {
406
+    chroot "$rootdir" apt-get -y install iproute bridge-utils libnetfilter-conntrack3 batctl
407
+    chroot "$rootdir" apt-get -y install python-dev libevent-dev ebtables python-pip git
408
+    chroot "$rootdir" apt-get -y install wireless-tools rfkill
409
+
410
+    if ! grep -q "batman_adv" $rootdir/etc/modules; then
411
+        echo 'batman_adv' >> $rootdir/etc/modules
412
+    fi
413
+
414
+	if [ -f /usr/local/bin/${PROJECT_NAME}-mesh-batman ]; then
415
+        cp /usr/local/bin/${PROJECT_NAME}-mesh-batman $rootdir/var/lib/batman
416
+	else
417
+        cp /usr/bin/${PROJECT_NAME}-mesh-batman $rootdir/var/lib/batman
418
+	fi
419
+
420
+    echo '[Unit]' > $rootdir/etc/systemd/system/batman.service
421
+    echo 'Description=B.A.T.M.A.N. Advanced' >> $rootdir/etc/systemd/system/batman.service
422
+    echo '' >> $rootdir/etc/systemd/system/batman.service
423
+    echo '[Service]' >> $rootdir/etc/systemd/system/batman.service
424
+    echo 'Type=oneshot' >> $rootdir/etc/systemd/system/batman.service
425
+    echo "ExecStart=/var/lib/batman start" >> $rootdir/etc/systemd/system/batman.service
426
+    echo "ExecStop=/var/lib/batman stop" >> $rootdir/etc/systemd/system/batman.service
427
+    echo 'RemainAfterExit=yes' >> $rootdir/etc/systemd/system/batman.service
428
+    echo '' >> $rootdir/etc/systemd/system/batman.service
429
+    echo '# Allow time for the server to start/stop' >> $rootdir/etc/systemd/system/batman.service
430
+    echo 'TimeoutSec=300' >> $rootdir/etc/systemd/system/batman.service
431
+    echo '' >> $rootdir/etc/systemd/system/batman.service
432
+    echo '[Install]' >> $rootdir/etc/systemd/system/batman.service
433
+    echo 'WantedBy=multi-user.target' >> $rootdir/etc/systemd/system/batman.service
434
+    chroot "$rootdir" systemctl enable batman
435
+}
436
+
365 437
 initialise_mesh() {
366 438
     if [[ $VARIANT != "mesh" ]]; then
367 439
         return
368 440
     fi
369
-    MESH_SERVICE='mesh-setup.service'
370
-    MESH_SETUP_DAEMON=$rootdir/etc/systemd/system/$MESH_SERVICE
371
-
372
-    echo '[Unit]' > $MESH_SETUP_DAEMON
373
-    echo 'Description=Initial mesh router configuration' >> $MESH_SETUP_DAEMON
374
-    echo 'After=syslog.target' >> $MESH_SETUP_DAEMON
375
-    echo 'After=network.target' >> $MESH_SETUP_DAEMON
376
-    echo '[Service]' >> $MESH_SETUP_DAEMON
377
-    echo 'Type=simple' >> $MESH_SETUP_DAEMON
378
-    echo 'User=root' >> $MESH_SETUP_DAEMON
379
-    echo 'Group=root' >> $MESH_SETUP_DAEMON
380
-    echo 'WorkingDirectory=/root' >> $MESH_SETUP_DAEMON
381
-    echo "ExecStart=/usr/local/bin/${PROJECT_NAME}-image-mesh > /var/log/mesh-setup.log" >> $MESH_SETUP_DAEMON
382
-    echo '' >> $MESH_SETUP_DAEMON
383
-    echo 'TimeoutSec=99999' >> $MESH_SETUP_DAEMON
384
-    echo '' >> $MESH_SETUP_DAEMON
385
-    echo '[Install]' >> $MESH_SETUP_DAEMON
386
-    echo 'WantedBy=multi-user.target' >> $MESH_SETUP_DAEMON
387
-
388
-    chroot "$rootdir" systemctl enable $MESH_SERVICE
441
+    mesh_avahi
442
+	mesh_batman
443
+
444
+	#MESH_SERVICE='mesh-setup.service'
445
+    #MESH_SETUP_DAEMON=$rootdir/etc/systemd/system/$MESH_SERVICE
446
+
447
+    #echo '[Unit]' > $MESH_SETUP_DAEMON
448
+    #echo 'Description=Initial mesh router configuration' >> $MESH_SETUP_DAEMON
449
+    #echo 'After=syslog.target' >> $MESH_SETUP_DAEMON
450
+    #echo 'After=network.target' >> $MESH_SETUP_DAEMON
451
+    #echo '[Service]' >> $MESH_SETUP_DAEMON
452
+    #echo 'Type=simple' >> $MESH_SETUP_DAEMON
453
+    #echo 'User=root' >> $MESH_SETUP_DAEMON
454
+    #echo 'Group=root' >> $MESH_SETUP_DAEMON
455
+    #echo 'WorkingDirectory=/root' >> $MESH_SETUP_DAEMON
456
+    #echo "ExecStart=/usr/local/bin/${PROJECT_NAME}-image-mesh > /var/log/mesh-setup.log" >> $MESH_SETUP_DAEMON
457
+    #echo '' >> $MESH_SETUP_DAEMON
458
+    #echo 'TimeoutSec=99999' >> $MESH_SETUP_DAEMON
459
+    #echo '' >> $MESH_SETUP_DAEMON
460
+    #echo '[Install]' >> $MESH_SETUP_DAEMON
461
+    #echo 'WantedBy=multi-user.target' >> $MESH_SETUP_DAEMON
462
+
463
+    #chroot "$rootdir" systemctl enable $MESH_SERVICE
389 464
 }
390 465
 
391 466
 # Set to true/false to control if eatmydata is used during build

+ 222
- 0
src/freedombone-mesh-batman 파일 보기

@@ -0,0 +1,222 @@
1
+#!/bin/bash
2
+#
3
+# .---.                  .              .
4
+# |                      |              |
5
+# |--- .--. .-.  .-.  .-.|  .-. .--.--. |.-.  .-. .--.  .-.
6
+# |    |   (.-' (.-' (   | (   )|  |  | |   )(   )|  | (.-'
7
+# '    '     --'  --'  -' -  -' '  '   -' -'   -' '   -  --'
8
+#
9
+#                    Freedom in the Cloud
10
+#
11
+# Used to enable or disable batman mesh protocol on wlanX
12
+#
13
+# License
14
+# =======
15
+#
16
+# Copyright (C) 2015-2016 Bob Mottram <bob@robotics.uk.to>
17
+#
18
+# This program is free software: you can redistribute it and/or modify
19
+# it under the terms of the GNU General Public License as published by
20
+# the Free Software Foundation, either version 3 of the License, or
21
+# (at your option) any later version.
22
+#
23
+# This program is distributed in the hope that it will be useful,
24
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
25
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
26
+# GNU General Public License for more details.
27
+#
28
+# You should have received a copy of the GNU General Public License
29
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
30
+
31
+PROJECT_NAME='freedombone'
32
+COMPLETION_FILE=/root/${PROJECT_NAME}-completed.txt
33
+
34
+if [[ $1 == "start" ]]; then
35
+    # install avahi
36
+    sed -i "s|#host-name=.*|host-name=$(hostname)|g" /etc/avahi/avahi-daemon.conf
37
+    sed -i "s|host-name=.*|host-name=$(hostname)|g" /etc/avahi/avahi-daemon.conf
38
+    sed -i "s|use-ipv4=.*|use-ipv4=yes|g" /etc/avahi/avahi-daemon.conf
39
+    sed -i "s|use-ipv6=.*|use-ipv6=no|g" /etc/avahi/avahi-daemon.conf
40
+    sed -i "s|#disallow-other-stacks=.*|disallow-other-stacks=yes|g" /etc/avahi/avahi-daemon.conf
41
+    sed -i "s|hosts:.*|hosts:          files mdns4_minimal dns mdns4 mdns|g" /etc/nsswitch.conf
42
+fi
43
+
44
+# Mesh definition
45
+ESSID=
46
+if ! grep -q "ESSID:" $COMPLETION_FILE; then
47
+    ESSID='mesh'
48
+else
49
+    ESSID=$(cat $COMPLETION_FILE | grep "ESSID:" | awk -F ':' '{print $2}')
50
+fi
51
+sed -i "s|ESSID:.*|ESSID:${ESSID}|g" $COMPLETION_FILE
52
+
53
+CELLID='any'
54
+
55
+CHANNEL=
56
+if ! grep -q "Wifi channel:" $COMPLETION_FILE; then
57
+    CHANNEL=2
58
+else
59
+    CHANNEL=$(cat $COMPLETION_FILE | grep "Wifi channel:" | awk -F ':' '{print $2}')
60
+fi
61
+sed -i "s|Wifi channel:.*|Wifi channel:${CHANNEL}|g" $COMPLETION_FILE
62
+
63
+ZERONET_PORT=15441
64
+IPFS_PORT=4001
65
+
66
+# Ethernet bridge definition (bridged to bat0)
67
+BRIDGE=br-mesh
68
+IFACE='wlan0'
69
+EIFACE=eth0
70
+
71
+if [[ $IFACE == "wlan0" ]]; then
72
+    if grep -q "wlan1" /proc/net/dev; then
73
+        IFACE=wlan1
74
+    fi
75
+fi
76
+if [[ $IFACE == "wlan0" ]]; then
77
+    if grep -q "wlan2" /proc/net/dev; then
78
+        IFACE=wlan2
79
+    fi
80
+fi
81
+if [[ $IFACE == "wlan0" ]]; then
82
+    if grep -q "wlan3" /proc/net/dev; then
83
+        IFACE=wlan3
84
+    fi
85
+fi
86
+
87
+if [ -e /etc/default/batctl ]; then
88
+    . /etc/default/batctl
89
+fi
90
+
91
+start() {
92
+    if [ -z "$IFACE" ] ; then
93
+        echo 'error: unable to find wifi interface, not enabling batman-adv mesh'
94
+        return
95
+    fi
96
+    echo 'info: enabling batman-adv mesh network $ESSID on $IFACE'
97
+
98
+    systemctl stop network-manager
99
+    sleep 5
100
+
101
+    # remove an avahi service which isn't used
102
+    if [ -f /etc/avahi/services/udisks.service ]; then
103
+        sudo rm /etc/avahi/services/udisks.service
104
+    fi
105
+
106
+    # Might have to re-enable wifi
107
+    rfkill unblock $(rfkill list|awk -F: "/phy/ {print $1}") || true
108
+
109
+    ifconfig $IFACE down
110
+    ifconfig $IFACE mtu 1532
111
+    iwconfig $IFACE enc off
112
+    iwconfig $IFACE mode ad-hoc essid $ESSID channel $CHANNEL
113
+    sleep 1
114
+    iwconfig $IFACE ap $CELLID
115
+
116
+    modprobe batman-adv
117
+    batctl if add $IFACE
118
+    ifconfig $IFACE up
119
+    avahi-autoipd --force-bind --daemonize --wait $BRIDGE
120
+    avahi-autoipd --force-bind --daemonize --wait $IFACE
121
+    ifconfig bat0 up promisc
122
+
123
+    #Use persistent HWAddr
124
+    ether_new=$(ifconfig eth0 | grep HWaddr | sed -e "s/.*HWaddr //")
125
+    if [ ! -f /var/lib/mesh-node/bat0 ]; then
126
+        mkdir /var/lib/mesh-node
127
+        echo "${ether_new}" > /var/lib/mesh-node/bat0
128
+    else
129
+        ether=$(cat /var/lib/mesh-node/bat0)
130
+        ifconfig bat0 hw ether ${ether}
131
+    fi
132
+
133
+    if [ "$EIFACE" ] ; then
134
+        brctl addbr $BRIDGE
135
+        brctl addif $BRIDGE bat0
136
+        brctl addif $BRIDGE $EIFACE
137
+        ifconfig bat0 0.0.0.0
138
+        ifconfig $EIFACE 0.0.0.0
139
+        ifconfig $EIFACE up promisc
140
+        ifconfig $BRIDGE up
141
+    fi
142
+
143
+    iptables -A INPUT -p tcp --dport 548 -j ACCEPT
144
+    iptables -A INPUT -p udp --dport 548 -j ACCEPT
145
+    iptables -A INPUT -p tcp --dport 5353 -j ACCEPT
146
+    iptables -A INPUT -p udp --dport 5353 -j ACCEPT
147
+    iptables -A INPUT -p tcp --dport 5354 -j ACCEPT
148
+    iptables -A INPUT -p udp --dport 5354 -j ACCEPT
149
+    iptables -A INPUT -p tcp --dport $ZERONET_PORT -j ACCEPT
150
+    iptables -A INPUT -p udp --dport $ZERONET_PORT -j ACCEPT
151
+    iptables -A INPUT -p tcp --dport $IPFS_PORT -j ACCEPT
152
+
153
+    systemctl restart avahi-daemon
154
+}
155
+
156
+stop() {
157
+    if [ -z "$IFACE" ]; then
158
+        echo 'error: unable to find wifi interface, not enabling batman-adv mesh'
159
+        return
160
+    fi
161
+    if [ "$EIFACE" ]; then
162
+        brctl delif $BRIDGE bat0
163
+        brctl delif $BRIDGE $EIFACE
164
+        ifconfig $BRIDGE down || true
165
+        brctl delbr $BRIDGE
166
+        ifconfig $EIFACE down -promisc
167
+    fi
168
+
169
+    avahi-autoipd -k $BRIDGE
170
+    avahi-autoipd -k $IFACE
171
+    ifconfig bat0 down -promisc
172
+
173
+    batctl if del $IFACE
174
+    rmmod batman-adv
175
+    ifconfig $IFACE mtu 1500
176
+    ifconfig $IFACE down
177
+    iwconfig $IFACE mode managed
178
+
179
+    iptables -D INPUT -p tcp --dport 548 -j ACCEPT
180
+    iptables -D INPUT -p udp --dport 548 -j ACCEPT
181
+    iptables -D INPUT -p tcp --dport 5353 -j ACCEPT
182
+    iptables -D INPUT -p udp --dport 5353 -j ACCEPT
183
+    iptables -D INPUT -p tcp --dport 5354 -j ACCEPT
184
+    iptables -D INPUT -p udp --dport 5354 -j ACCEPT
185
+    iptables -D INPUT -p tcp --dport $ZERONET_PORT -j ACCEPT
186
+    iptables -D INPUT -p udp --dport $ZERONET_PORT -j ACCEPT
187
+    iptables -D INPUT -p tcp --dport $IPFS_PORT -j ACCEPT
188
+
189
+    systemctl restart network-manager
190
+}
191
+
192
+if ! grep -q "$IFACE" /proc/net/dev; then
193
+    echo 'Interface $IFACE was not found'
194
+    stop
195
+    exit 1
196
+fi
197
+
198
+case "$1" in
199
+    start|stop)
200
+        $1
201
+        ;;
202
+    restart)
203
+        stop
204
+        sleep 10
205
+        start
206
+        ;;
207
+    status)
208
+        batctl o
209
+        ;;
210
+    ping)
211
+        batctl ping $2
212
+        ;;
213
+    ls|list)
214
+        avahi-browse -atl
215
+        ;;
216
+    *)
217
+    echo "error: invalid parameter $1"
218
+    echo 'usage: $0 {start|stop|restart|status|ping|ls|list}'
219
+    exit 2
220
+    ;;
221
+esac
222
+exit 0