* Deactivate Multicast-DNS via iptables, thanks for suggestion -
[grml-terminalserver.git] / grml-terminalserver
1 #!/bin/sh
2 # Filename:      terminalserver
3 # Purpose:       Program to do something
4 # Authors:       grml-team (grml.org), (c) Michael Gebetsroither <gebi@grml.org>
5 # Bug-Reports:   see http://grml.org/bugs/
6 # License:       This file is licensed under the GPL v2.
7 # Latest change: Sat Aug 06 14:14:40 CEST 2005
8 ################################################################################
9
10
11 ###
12 ### __INCLUDES
13 ###
14 . /etc/grml/sh-lib
15 #. /etc/grml/sysexits-sh
16
17
18
19 ###
20 ### __VARIABLES
21 ###
22
23 FORCE_='false'
24 verbose_=0
25
26 # this file holds all variable definitions
27 SHARED_PROG_VARS_="/usr/share/grml-terminalserver/shared_prog_vars"
28 isExistent $SHARED_PROG_VARS_ die
29 . $SHARED_PROG_VARS_
30
31
32 ###
33 ### __FUNCTIONS
34 ###
35
36 function printUsage
37 {
38   cat <<EOT
39 Usage: "$PROG_NAME__" [OPTIONS] <command>
40
41 $PROG_NAME__ is the config program for the terminalserver
42 comming with grml.
43
44 COMMANDS:
45    help             This help text
46    start <service>  Start services (if all services are started, configs will be updated)
47    stop <service>   Stop services
48    config <service> Update config of given service (or from all if no services given)
49    clean            Stop all + remove config and boot files from services
50    <default>        interactive
51
52 SERVICES:
53   tftp        Tftp daemon
54   dhcp        Dhcp daemon
55   nfs         All necessary nfs daemons
56   <>          ALL services
57
58 OPTIONS:
59    -v         verbose (show what is going on, v++)
60    -h         this help text
61    -f         Force
62
63 EOT
64 }
65
66 function killPortmapper
67 {
68     /etc/init.d/portmap stop >/dev/null &>/dev/null
69     killall -9 portmap &>/dev/null
70 }
71
72 # DHCP SERVICE {{{
73 function createDhcpConf
74 {
75   if [ -e "$DHCPD_CONFIG_FILE_" ]; then
76     if grep $CONFIG_PATTERN_ $DHCPD_CONFIG_FILE_ &>/dev/null; then
77       execute "mv -fb \"$DHCPD_CONFIG_FILE_\" \"$DHCPD_CONFIG_FILE_.old\"" eprint &>/dev/null
78       execute "source $TEMPLATE_CONFIG_DIR_/dhcpd_config" die
79     else
80       if [[ $FORCE_ == "true" ]]; then
81         execute "mv -fb \"$DHCPD_CONFIG_FILE_\" \"$DHCPD_CONFIG_FILE_.old\"" eprint &>/dev/null
82         execute "source $TEMPLATE_CONFIG_DIR_/dhcpd_config" die
83       else
84         warn "Not updating user edited configfile $DHCPD_CONFIG_FILE_, user -f to override"
85       fi
86     fi
87   else
88     execute "source $TEMPLATE_CONFIG_DIR_/dhcpd_config" die
89   fi
90 }
91
92 function removeDhcpConf
93 {
94   if [ -e "$DHCPD_CONFIG_FILE_" ]; then
95     if grep $CONFIG_PATTERN_ $DHCPD_CONFIG_FILE_ &>/dev/null; then
96       rm -f "$DHCPD_CONFIG_FILE_"
97     else
98       if [[ $FORCE_ == "true" ]]; then
99         rm -f "$DHCPD_CONFIG_FILE_"
100       else
101         warn "Not deleting user edited configfile $DHCPD_CONFIG_FILE_, user -f to override"
102       fi
103     fi
104   fi
105 }
106
107
108 function stopDhcp
109 {
110   start-stop-daemon --stop --quiet --pidfile "$DHCPD_PID_"
111   rm -f $DHCPD_PID_
112   rm -f /var/lib/dhcp3/dhcpd.leases* 2>/dev/null    #FIXME
113   touch /var/lib/dhcp3/dhcpd.leases
114 }
115
116 function startDhcp
117 {
118   local conf_file_="$DHCPD_CONFIG_FILE_"
119
120   test -f $DHCPD_BIN_ || die "could not find dhcpd \"$DHCPD_BIN_\""
121   start-stop-daemon --start --quiet --pidfile "$DHCPD_PID_" \
122     --exec "$DHCPD_BIN_" -- -cf "$conf_file_" -q "$INTERFACE_" || warn "problems starting dhcpd"
123 }
124
125 function runDhcp
126 {
127   isExistent "$DHCPD_CONFIG_FILE_" || \
128     warn "no config for dhcpd: \"$DHCPD_CONFIG_FILE_\" => not starting dhcpd" || return 1
129
130   stopDhcp
131   sleep 1
132   startDhcp
133 }
134
135 # make sure tcp/113 is rejected
136 function runIptables
137 {
138  if [ -x /sbin/iptables ] ; then
139     # something keeps answering all tftp requests with auth requests (SYN
140     # packets to the client tcp/113). Since the PXE client doesn't answer with
141     # RST, the auth query has to wait until it times out. Forbidding the
142     # terminalserver to send out packets to tcp/113 via iptables _greatly_
143     # speeds up the process. But of course the real fix would be to have grml
144     # stop sending out auth queries to tftp clients. according to netstat, it
145     # is in.tftpd itself sending out the auth queries.
146     # Thanks to Marc Haber and Wolfgang Karall for noticing and current fix.
147     if iptables -L | grep -q '^REJECT.*tcp dpt:auth reject-with tcp-reset' ; then
148        echo "Rule for tcp/113 already present, nothing to be done."
149     else
150        echo "Rejecting tcp/113 via iptables to speed up booting via PXE, running:"
151        echo -n '* iptables -A OUTPUT -p tcp --dport 113 -j REJECT --reject-with tcp-reset'
152        iptables -A OUTPUT -p tcp --dport 113 -j REJECT --reject-with tcp-reset && echo done || echo failed
153     fi
154     # deactivate Multicast-DNS
155     if iptables -L | grep -q '^REJECT.*dpt:mdns reject-with icmp-port-unreachable' ; then
156        echo "Rule for udp/5353 already present, nothing to be done."
157     else
158        echo "Rejecting udp/5353 via iptables for deactivating Multicast-DNS, running:"
159        echo -n '* iptables -A OUTPUT -p udp -d 224.0.0.0/8 --dport 5353 -j REJECT'
160        iptables -A OUTPUT -p udp -d 224.0.0.0/8 --dport 5353 -j REJECT && echo done || echo failed
161     fi
162  else
163     warn "iptables executable not avilable"
164  fi
165 }
166 # }}}
167
168 # TFTP SERVICE {{{
169 function removeTftpConf
170 {
171   rm -rf $TFTPD_DATA_DIR_/*
172 }
173 function createTftpConf
174 {
175   removeTftpConf
176
177   execute "mkdir $TFTPD_DATA_DIR_/pxelinux.cfg" die
178   execute "install -m 644 /usr/lib/syslinux/pxelinux.0 $TFTPD_DATA_DIR_" die
179   execute "install -m 644 $PATH_/minirt26.gz $TFTPD_DATA_DIR_" die
180   execute "install -m 644 $KERNEL_IMAGE_ $TFTPD_DATA_DIR_/linux26" die
181   execute "install -m 644 $MEMTEST_IMAGE_ $TFTPD_DATA_DIR_/memtest" die
182   execute "install -m 644 $PXE_BOOT_MSG_ $TFTPD_DATA_DIR_" die
183   execute "install -m 644 $PXE_BOOT_LOGO_ $TFTPD_DATA_DIR_" die
184
185   execute "source $TEMPLATE_CONFIG_DIR_/grub-pxelinux_config" die
186 }
187
188 function stopTftp
189 {
190   start-stop-daemon --stop --quiet --name "${TFTPD_BIN_##*/}"
191 }
192 function startTftp
193 {
194   test -f $TFTPD_BIN_ || die "could not find \"$TFTPD_BIN_\""
195   start-stop-daemon --start --quiet --exec "$TFTPD_BIN_" -- -l -a "$IP_" -s "$TFTPD_DATA_DIR_" || \
196     warn "problems starting tftpd server"
197 }
198 function runTftp
199 {
200   stopTftp
201   sleep 1
202   startTftp
203 }
204 # }}}
205
206
207 # NFS  {{{
208 function createNfsConfig
209 {
210   execute "exportfs -o ro,no_root_squash,async,nohide $NETWORK_/$NETMASK_:$MOUNT_POINT_" warn
211 }
212
213 function removeNfsConfig
214 {
215   execute "exportfs -u -o ro,no_root_squash,async,nohide $NETWORK_/$NETMASK_:$MOUNT_POINT_" warn
216 }
217
218 function startNfs
219 {
220   /etc/init.d/portmap start
221   /etc/init.d/nfs-common start
222   # FIXME /etc/init.d/nfs-kernel-server start
223   $USR_SHARE_/nfs-kernel-server start
224
225   createNfsConfig
226 }
227 function stopNfs
228 {
229   removeNfsConfig
230   if [[ `exportfs |wc -l` > 0 ]]; then
231     dprint "There are other exports, not stopping NFS serivces"
232   else
233     /etc/init.d/nfs-kernel-server stop >/dev/null 2>&1
234     /etc/init.d/nfs-common stop >/dev/null 2>&1
235     killPortmapper
236   fi
237 }
238 # }}}
239
240
241 function createConfig
242 {
243   # FIXME
244   execute "sed -i 's/^ALL/\#ALL/' /etc/hosts.deny" warn
245   execute "sed -i 's/^ALL : ALL@ALL : DENY/ALL : ALL@ALL : ALLOW/' /etc/hosts.allow" warn
246
247   createTftpConf
248   createDhcpConf
249 }
250
251 function actionStart
252 {
253   createConfig
254
255   echo -n "Starting tftpd: "
256   runTftp && echo done || echo failed
257
258   echo -n "Starting dhcpd: "
259   runDhcp && echo done || echo failed
260
261   runIptables
262
263   echo "Finally starting nfs services..."
264   startNfs && echo "Sucessfully finished startup of grml-terminalserver." || echo 'Startup of grml-terminalserver-config failed!'
265 }
266
267 function actionStop
268 {
269   stopTftp
270   stopDhcp
271   stopNfs
272   notice "Terminal-server stopped"
273 }
274
275 function actionClean
276 {
277   actionStop
278   removeTftpConf
279   removeDhcpConf
280   stopNfs
281 }
282
283 function updateConfig
284 {
285   local service_="$1"
286
287   case "$service_" in
288     "") createConfig ;;
289     tftp) createTftpConf ;;
290     dhcp) createDhcpConf ;;
291     nfs) removeNfsConfig; createNfsConfig ;;
292     *) warn "Service $service_ not available" ;;
293   esac
294 }
295
296 # SERVICES {{{
297 function serviceStart
298 {
299   local service_="$1"   # service to start, if nothing => all
300
301   case "$service_" in
302     "") actionStart ;;
303     tftp) runTftp ;;
304     dhcp) runDhcp ;;
305     nfs) startNfs ;;
306     *) warn "Service $service_ not available" ;;
307   esac
308 }
309
310 function serviceStop
311 {
312   local service_="$1"   # service to stop, if nothing => all
313
314   case "$service_" in
315     "") actionStop ;;
316     tftp) stopTftp ;;
317     dhcp) stopDhcp ;;
318     nfs) stopNfs ;;
319     *) warn "Service $service_ not available" ;;
320   esac
321 }
322 # }}}
323
324 ###
325 ### __MAIN
326 ###
327
328 while getopts "fi:hv" opt; do
329   case "$opt" in
330     f) FORCE_='true' ;;
331     h) printUsage; exit ;;
332     v) let verbose_=$verbose_+1 ;;
333     ?) printUsage; exit 64 ;;
334   esac
335 done
336 shift $(($OPTIND - 1))  # set ARGV to the first not parsed commandline parameter
337 setVerbose $verbose_
338
339 case "$1" in
340   help)   printUsage; exit 0 ;;
341 esac
342
343 checkRoot die "You have to be root to use this program"
344 disableSyslog
345
346 isExistent $DEFAULT_CONFIG_ die
347 . $DEFAULT_CONFIG_
348 . $CONFIG_
349 # used config vars:
350 # MOUNT_POINT_
351 # KERNEL_IMAGE_
352 # MEMTEST_IMAGE_
353 # PXE_BOOT_MSG_
354 # PXE_BOOT_LOGO_
355 if [[ $MOUNT_POINT_ == "" || $KERNEL_IMAGE_ == "" || $MEMTEST_IMAGE_ == "" || \
356   $PXE_BOOT_MSG_ == "" || $PXE_BOOT_MSG_ == "" ]]; then
357   warn "MOUNT_POINT_=\"$MOUNT_POINT_\" \
358 KERNEL_IMAGE_=\"$KERNEL_IMAGE_\" \
359 MEMTEST_IMAGE_=\"$MEMTEST_IMAGE_\"
360 PXE_BOOT_MSG_=\"$PXE_BOOT_MSG_\"
361 PXE_BOOT_LOGO_=\"$PXE_BOOT_LOGO_\""
362   die "False configuration, please update $CONFIG_"
363 fi
364
365 case "$1" in
366   clean) actionClean; exit 0 ;;
367 esac
368
369
370 while true; do
371   isExistent "$CONF_FILE_" warn "sorry configfile \"$CONF_FILE_\" not found"
372   if [ $? -eq 1 ]; then
373     $CONFIG_PROG_ && continue
374     $DIALOG_ --clear --backtitle "$BACK_TITLE_" --title "Question" --yesno \
375       "grml-terminalserver-config returned an error, do you want to quit now?" 5 75 && exit 1
376   else
377     break
378   fi
379 done
380 source $CONF_FILE_
381
382 # check for necessary files
383 check_necessary_files_='no'
384 if [ "$1" == 'start' ]; then
385   case "$2" in
386     tftp|"") check_necessary_files_='yes' ;;
387   esac
388 fi
389 if [ $check_necessary_files_ == 'yes' ]; then
390   # test for files absolutly necessary for grml-terminalserver and created from -config
391   problem_=0
392   for i in $PATH_/minirt26.gz; do
393     isExistent $i warn || problem_=1
394   done
395   if [ $problem_ -eq 1 ]; then
396     die 'Some necessary files are missing, please rerun grml-terminalserver-config, or copy the files manually'
397   fi
398 fi
399
400
401 case "$1" in
402   start) serviceStart "$2" ;;
403   stop) serviceStop "$2" ;;
404   config) updateConfig "$2" ;;
405   "")  actionStart ;;
406   *)  printUsage ;;
407 esac
408
409 # END OF FILE
410 ################################################################################
411 # vim:foldmethod=marker tabstop=2 expandtab shiftwidth=2