Developing lightweight computation at the DSG edge

qmp_control.sh 4.92 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
configure_wifi() {
	qmp_configure_wifi_initial
	qmp_configure_wifi
35
	configure_network
36
	/etc/init.d/network restart
37
	/etc/init.d/network reload
38
39
40
41
	if /etc/init.d/gwck enabled
	then
		/etc/init.d/gwck restart
	fi
42
43
}

44
45
46
47
configure_gw() {
	qmp_gw_apply
}

48
49
apply_services() {
	qmp_set_services
50
51
52
53
}

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

configure_system() {
	qmp_configure_system
67
	apply_services
68
	qmp_bmx6_reload
69
70
71
	/etc/init.d/uhttpd restart
}

72
enable_ns_xm_ppt() {
73
74
75
76
77
	echo 8 > /sys/class/gpio/export
	echo out > /sys/class/gpio/gpio8/direction
	echo 1 > /sys/class/gpio/gpio8/value
}

78
79
80
81
82
83
enable_ns_xw_ppt() {
	echo 2 > /sys/class/gpio/export
	echo out > /sys/class/gpio/gpio2/direction
	echo 1 > /sys/class/gpio/gpio2/value
}

84
85
86
87
88
89
90
91
92
93
94
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
95
	qmp_update_upgrade_system $1 && hard_reboot
96
97
98
99
100
}

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

104
105
106
configure_all() {
	configure_system
	configure_wifi
107
	configure_network
108
109
}

p4u's avatar
p4u committed
110
111
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!" \
112
	|| {
113
114
115
		echo "Cannot found saved state, saving it..."
		save_state
	}
116

117
	touch /tmp/qmp.safe.test
p4u's avatar
p4u committed
118
	echo "------------------------------------------------------------------------------------"
119
	echo "File /tmp/qmp.safe.test has been created, after configuring the system you will have
p4u's avatar
p4u committed
120
121
	180 seconds to remove it or the previous state will be recovered"
	echo "------------------------------------------------------------------------------------"
122

123
124
	read -p'Do you agree?[y,N] ' a
	[ "$a" != "y" ] && return
125

126
	configure_all
127

p4u's avatar
p4u committed
128
	( sleep 180
129
		[ -e "/tmp/qmp.safe.test" ] && {
p4u's avatar
p4u committed
130
			cp -f /etc/config/qmp /tmp/qmp.wrong
131
			recover_state
p4u's avatar
p4u committed
132
			cp -f /tmp/qmp.wrong /etc/config/qmp.wrong
133
134
135
136
137
138
139
			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
140
141
142
	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
143
144
145
146
}

recover_state() {
	echo "Recovering state at $(date) from /tmp/qmp.save.tar.gz"
p4u's avatar
p4u committed
147
148
	[ -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"
149
150
}

151
152
help() {
	echo "Use: $0 <function> [params]"
153

154
	echo ""
155
	echo "Configuration:"
p4u's avatar
p4u committed
156
157
158
159
160
	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"
161
	echo " configure_gw			: Configure and apply gateways settings"
162

163
	echo ""
p4u's avatar
p4u committed
164
165
166
167
	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"
168

169
170
	echo ""
	echo "Gateways:"
p4u's avatar
p4u committed
171
172
173
174
175
	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>"
176
177
178

	echo ""
	echo "Other:"
179
	echo " apply_services			: Start/stop services depending on qmp configuration"
180
181
	echo " enable_ns_xm_ppt		: Enable PoE passtrough from NanoStation XM M2/M5 devices. Be careful with this option!"
	echo " enable_ns_xw_ppt		: Enable PoE passtrough from NanoStation XW M2/M5 devices. Be careful with this option!"
p4u's avatar
p4u committed
182
	echo " upgrade [URL]			: Upgrade system. By default to the last version, but image url can be provided to force"
183
	echo " hard_reboot				: Performs a hard reboot (using kernel sysrq)"
184

185
	echo ""
186
	exit 0
187
188
189
190
191
}


[ -z "$1" ] && help

192
echo "executing function $1..."
193

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