Developing lightweight computation at the DSG edge

qmp_control.sh 4.58 KB
Newer Older
Pau Escrich's avatar
Pau Escrich committed
1
#!/bin/sh
2
3
4
5
6
7
8
9
QMP_PATH="/etc/qmp"

. $QMP_PATH/qmp_common.sh
. $QMP_PATH/qmp_functions.sh
. $QMP_PATH/qmp_gw.sh
. $QMP_PATH/qmp_wireless.sh
. $QMP_PATH/qmp_network.sh
. $QMP_PATH/qmp_update.sh
10
. $QMP_PATH/qmp_system.sh
11
12

offer_default_gw() {
13
	qmp_gw_default offer $1
14
15
16
17
	qmp_gw_apply
}

search_default_gw() {
18
	qmp_gw_default search $1
19
20
21
22
	qmp_gw_apply
}

disable_default_gw() {
23
	qmp_gw_default disable $1
24
	qmp_gw_apply
25
26
}

27
28
29
30
31
reset_wifi() {
	qmp_reset_wifi
	configure_wifi
}

32
33
34
35
configure_wifi() {
	qmp_configure_wifi_initial
	qmp_configure_wifi
	/etc/init.d/network reload
36
37
38
39
	if /etc/init.d/gwck enabled
	then
		/etc/init.d/gwck restart
	fi
40
41
}

42
43
44
45
configure_gw() {
	qmp_gw_apply
}

46
47
apply_services() {
	qmp_set_services
48
49
50
51
}

configure_network() {
	qmp_configure
52
	qmp_bmx6_reload
53
	/etc/init.d/network reload
54
55
56
57
	if /etc/init.d/gwck enabled
	then
		/etc/init.d/gwck restart
	fi
58
	/etc/init.d/dnsmasq restart
p4u's avatar
p4u committed
59
	qmp_restart_firewall
60
61
62
63
}

configure_system() {
	qmp_configure_system
64
	apply_services
65
	qmp_bmx6_reload
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
	/etc/init.d/uhttpd restart
}

enable_ns_ppt() {
	echo 8 > /sys/class/gpio/export
	echo out > /sys/class/gpio/gpio8/direction
	echo 1 > /sys/class/gpio/gpio8/value
}

publish_hna() {
	[ -z "$1" ] && help
	qmp_publish_hna_bmx6 $1 $2
}

unpublish_hna() {
	[ -z "$1" ] && help
	qmp_unpublish_hna_bmx6 $1
}

upgrade() {
p4u's avatar
p4u committed
86
	qmp_update_upgrade_system $1 && hard_reboot
87
88
89
90
91
}

hard_reboot() {
	echo "System is gonna be rebooted now!"
	echo 1 > /proc/sys/kernel/sysrq
92
	echo b > /proc/sysrq-trigger
93
94
}

95
96
97
configure_all() {
	configure_system
	configure_wifi
98
	configure_network
99
100
}

p4u's avatar
p4u committed
101
102
safe_apply() {
	[ -e "/tmp/qmp.save.tar.gz" ] && echo "Found saved state at /tmp/qmp.save.tar.gz. Make sure you want to use it!" \
103
	|| {
104
105
106
		echo "Cannot found saved state, saving it..."
		save_state
	}
107

108
	touch /tmp/qmp.safe.test
p4u's avatar
p4u committed
109
	echo "------------------------------------------------------------------------------------"
110
	echo "File /tmp/qmp.safe.test has been created, after configuring the system you will have
p4u's avatar
p4u committed
111
112
	180 seconds to remove it or the previous state will be recovered"
	echo "------------------------------------------------------------------------------------"
113

114
115
	read -p'Do you agree?[y,N] ' a
	[ "$a" != "y" ] && return
116

117
	configure_all
118

p4u's avatar
p4u committed
119
	( sleep 180
120
		[ -e "/tmp/qmp.safe.test" ] && {
p4u's avatar
p4u committed
121
			cp -f /etc/config/qmp /tmp/qmp.wrong
122
			recover_state
p4u's avatar
p4u committed
123
			cp -f /tmp/qmp.wrong /etc/config/qmp.wrong
124
125
126
127
128
129
130
			hard_reboot
		} || rm -f "/tmp/qmp.safe.test"
	) &
}

save_state() {
	echo "Saving state at $(date) in file /tmp/qmp.save.tar.gz"
p4u's avatar
p4u committed
131
132
133
	rm -f /tmp/qmp.save.tar.gz 2>/dev/null
	(cd /overlay/etc && tar czf /tmp/qmp.save.tar.gz .)
	[ ! -e /tmp/qmp.save.tar.gz ] && echo "ERROR: cannot save state, exiting..." && exit 1
134
135
136
137
}

recover_state() {
	echo "Recovering state at $(date) from /tmp/qmp.save.tar.gz"
p4u's avatar
p4u committed
138
139
	[ -e "/tmp/qmp.save.tar.gz" ] && tar xvzf /tmp/qmp.save.tar.gz -C /overlay/etc/ \
	|| echo "Cannot recover state because it has not been saved before"
140
141
}

142
143
help() {
	echo "Use: $0 <function> [params]"
144

145
	echo ""
146
	echo "Configuration:"
p4u's avatar
p4u committed
147
148
149
150
151
	echo " configure_all			: Configure and apply all settings"
	echo " configure_network		: Configure and apply network settings"
	echo " configure_system 		: Configure and apply system settings (qmp.node section and so on)"
	echo " configure_wifi			: Configure all WiFi devices"
	echo " reset_wifi			: Reset, rescan and configure all the WiFi devices"
152
	echo " configure_gw			: Configure and apply gateways settings"
153

154
	echo ""
p4u's avatar
p4u committed
155
156
157
158
	echo "Safe configuration:"
	echo " save_state			: Saves current state of configuration files"
	echo " recover_state			: Recovers previous saved state"
	echo " safe_apply			: Performs a safe configure_all. If something wrong it comes back to old state"
159

160
161
	echo ""
	echo "Gateways:"
p4u's avatar
p4u committed
162
163
164
165
166
	echo " offer_default_gw [ipv4|ipv6]	: Offers default gw to the network IPv4 or IPv6, both versions if no value"
	echo " search_default_gw [ipv4|ipv6]	: Search for a default gw in the network IPv4 or IPv6, both versions if no value"
	echo " disable_default_gw [ipv4|ipv6]	: Disables the search/offer of default GW IPv4 and/or IPv6"
	echo " publish_hna			: Publish an IP range (v4 or v6): publish_hna <IP/NETMASK> [ID]"
	echo " unpublish_hna			: Unpublish a current HNA: unpublish_hna <ID>"
167
168
169

	echo ""
	echo "Other:"
170
	echo " apply_services			: Start/stop services depending on qmp configuration"
p4u's avatar
p4u committed
171
172
173
	echo " enable_ns_ppt			: Enable POE passtrought from NanoStation M2/5 devices. Be careful with this option!"
	echo " upgrade [URL]			: Upgrade system. By default to the last version, but image url can be provided to force"
	echo " hard_reboot			: Performs a hard reboot (using kernel sysrq)"
174

175
	echo ""
176
	exit 0
177
178
179
180
181
}


[ -z "$1" ] && help

182
echo "executing function $1..."
183

184
$@ # 2>/dev/null || echo " Command not found.\n" && help