]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - usr.sbin/pc-sysinstall/backend/functions-networking.sh
Import new version of libc++. Among other improvements, this comes with an
[FreeBSD/FreeBSD.git] / usr.sbin / pc-sysinstall / backend / functions-networking.sh
1 #!/bin/sh
2 #-
3 # Copyright (c) 2010 iXsystems, Inc.  All rights reserved.
4 # Copyright (c) 2011 The FreeBSD Foundation
5 # All rights reserved.
6 #
7 # Portions of this software were developed by Bjoern Zeeb
8 # under sponsorship from the FreeBSD Foundation.
9 #
10 # Redistribution and use in source and binary forms, with or without
11 # modification, are permitted provided that the following conditions
12 # are met:
13 # 1. Redistributions of source code must retain the above copyright
14 #    notice, this list of conditions and the following disclaimer.
15 # 2. Redistributions in binary form must reproduce the above copyright
16 #    notice, this list of conditions and the following disclaimer in the
17 #    documentation and/or other materials provided with the distribution.
18 #
19 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 # ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 # SUCH DAMAGE.
30 #
31 # $FreeBSD$
32
33 # Functions which perform our networking setup
34
35 # Function which creates a kde4 .desktop file for the PC-BSD net tray
36 create_desktop_nettray()
37 {
38   NIC="${1}"
39   echo "#!/usr/bin/env xdg-open
40 [Desktop Entry]
41 Exec=/usr/local/kde4/bin/pc-nettray ${NIC}
42 Icon=network
43 StartupNotify=false
44 Type=Application" > ${FSMNT}/usr/share/skel/.kde4/Autostart/tray-${NIC}.desktop
45   chmod 744 ${FSMNT}/usr/share/skel/.kde4/Autostart/tray-${NIC}.desktop
46
47 };
48
49 # Function which checks is a nic is wifi or not
50 check_is_wifi()
51 {
52   NIC="$1"
53   ifconfig ${NIC} | grep -q "802.11" 2>/dev/null
54   if [ $? -eq 0 ]
55   then
56     return 0
57   else 
58     return 1
59   fi
60 };
61
62 # Function to get the first available wired nic, used for setup
63 get_first_wired_nic()
64 {
65   rm ${TMPDIR}/.niclist >/dev/null 2>/dev/null
66   # start by getting a list of nics on this system
67   ${QUERYDIR}/detect-nics.sh > ${TMPDIR}/.niclist
68   if [ -e "${TMPDIR}/.niclist" ]
69   then
70     while read line
71     do
72       NIC="`echo $line | cut -d ':' -f 1`"
73       check_is_wifi ${NIC}
74       if [ $? -ne 0 ]
75       then
76         export VAL="${NIC}"
77         return
78       fi
79     done < ${TMPDIR}/.niclist
80   fi
81
82   export VAL=""
83   return
84 };
85
86
87 # Function which simply enables plain dhcp on all detected nics
88 enable_dhcp_all()
89 {
90   rm ${TMPDIR}/.niclist >/dev/null 2>/dev/null
91   # start by getting a list of nics on this system
92   ${QUERYDIR}/detect-nics.sh > ${TMPDIR}/.niclist
93   if [ -e "${TMPDIR}/.niclist" ]
94   then
95     echo "# Auto-Enabled NICs from pc-sysinstall" >>${FSMNT}/etc/rc.conf
96     WLANCOUNT="0"
97     while read line
98     do
99       NIC="`echo $line | cut -d ':' -f 1`"
100       DESC="`echo $line | cut -d ':' -f 2`"
101       echo_log "Setting $NIC to DHCP on the system."
102       check_is_wifi ${NIC}
103       if [ $? -eq 0 ]
104       then
105         # We have a wifi device, setup a wlan* entry for it
106         WLAN="wlan${WLANCOUNT}"
107         cat ${FSMNT}/etc/rc.conf | grep -q "wlans_${NIC}="
108         if [ $? -ne 0 ] ; then
109           echo "wlans_${NIC}=\"${WLAN}\"" >>${FSMNT}/etc/rc.conf
110         fi
111         echo "ifconfig_${WLAN}=\"DHCP\"" >>${FSMNT}/etc/rc.conf
112         CNIC="${WLAN}"
113         WLANCOUNT=$((WLANCOUNT+1))
114       else
115         echo "ifconfig_${NIC}=\"DHCP\"" >>${FSMNT}/etc/rc.conf
116         CNIC="${NIC}"
117       fi
118  
119     done < ${TMPDIR}/.niclist 
120   fi
121 };
122
123
124 # Function which detects available nics, and enables dhcp on them
125 save_auto_dhcp()
126 {
127   enable_dhcp_all
128 };
129
130 # Function which simply enables iPv6 SLAAC on all detected nics
131 enable_slaac_all()
132 {
133   rm ${TMPDIR}/.niclist >/dev/null 2>/dev/null
134   # start by getting a list of nics on this system
135   ${QUERYDIR}/detect-nics.sh > ${TMPDIR}/.niclist
136   if [ -e "${TMPDIR}/.niclist" ]
137   then
138     echo "# Auto-Enabled NICs from pc-sysinstall" >>${FSMNT}/etc/rc.conf
139     WLANCOUNT="0"
140     while read line
141     do
142       NIC="`echo $line | cut -d ':' -f 1`"
143       DESC="`echo $line | cut -d ':' -f 2`"
144       echo_log "Setting $NIC to accepting RAs on the system."
145       check_is_wifi ${NIC}
146       if [ $? -eq 0 ]
147       then
148         # We have a wifi device, setup a wlan* entry for it
149         # Given we cannot have DHCP and SLAAC the same time currently
150         # it's save to just duplicate.
151         WLAN="wlan${WLANCOUNT}"
152         cat ${FSMNT}/etc/rc.conf | grep -q "wlans_${NIC}="
153         if [ $? -ne 0 ] ; then
154           echo "wlans_${NIC}=\"${WLAN}\"" >>${FSMNT}/etc/rc.conf
155         fi
156         #echo "ifconfig_${NIC}=\"up\"" >>${FSMNT}/etc/rc.conf
157         echo "ifconfig_${WLAN}_ipv6=\"inet6 accept_rtadv\"" >>${FSMNT}/etc/rc.conf
158         CNIC="${WLAN}"
159         WLANCOUNT=$((WLANCOUNT+1))
160       else
161         #echo "ifconfig_${NIC}=\"up\"" >>${FSMNT}/etc/rc.conf
162         echo "ifconfig_${NIC}_ipv6=\"inet6 accept_rtadv\"" >>${FSMNT}/etc/rc.conf
163         CNIC="${NIC}"
164       fi
165  
166     done < ${TMPDIR}/.niclist 
167   fi
168
169   # Given we cannot yet rely on RAs to provide DNS information as much
170   # as we can in the DHCP world, we should append a given nameserver.
171   : > ${FSMNT}/etc/resolv.conf
172   get_value_from_cfg netSaveIPv6NameServer
173   NAMESERVER="${VAL}"
174   if [ -n "${NAMESERVER}" ]
175   then
176     echo "nameserver ${NAMESERVER}" >>${FSMNT}/etc/resolv.conf
177   fi
178
179 };
180
181
182 # Function which detects available nics, and enables IPv6 SLAAC on them
183 save_auto_slaac()
184 {
185   enable_slaac_all
186 };
187
188
189 # Function which saves a manual nic setup to the installed system
190 save_manual_nic()
191 {
192   # Get the target nic
193   NIC="$1"
194
195   get_value_from_cfg netSaveIP
196   NETIP="${VAL}"
197  
198   if [ "$NETIP" = "DHCP" ]
199   then
200     echo_log "Setting $NIC to DHCP on the system."
201     echo "ifconfig_${NIC}=\"DHCP\"" >>${FSMNT}/etc/rc.conf
202     return 0
203   fi
204
205   # If we get here, we have a manual setup, lets do so now
206   IFARGS=""
207   IF6ARGS=""
208
209   # Set the manual IP
210   if [ -n "${NETIP}" ]
211   then
212     IFARGS="inet ${NETIP}"
213
214     # Check if we have a netmask to set
215     get_value_from_cfg netSaveMask
216     NETMASK="${VAL}"
217     if [ -n "${NETMASK}" ]
218     then
219       IFARGS="${IFARGS} netmask ${NETMASK}"
220     fi
221   fi
222
223   get_value_from_cfg netSaveIPv6
224   NETIP6="${VAL}"
225   if [ -n "${NETIP6}" ]
226   then
227     # Make sure we have one inet6 prefix.
228     IF6ARGS=`echo "${NETIP6}" | awk '{ if ("^inet6 ") { print $0; } else
229       { printf "inet6 %s", $0; } }'`
230   fi
231
232   echo "# Auto-Enabled NICs from pc-sysinstall" >>${FSMNT}/etc/rc.conf
233   if [ -n "${IFARGS}" ]
234   then
235     echo "ifconfig_${NIC}=\"${IFARGS}\"" >>${FSMNT}/etc/rc.conf
236   fi
237   if [ -n "${IF6ARGS}" ]
238   then
239     echo "ifconfig_${NIC}_ipv6=\"${IF6ARGS}\"" >>${FSMNT}/etc/rc.conf
240   fi
241
242   # Check if we have a default router to set
243   get_value_from_cfg netSaveDefaultRouter
244   NETROUTE="${VAL}"
245   if [ -n "${NETROUTE}" ]
246   then
247     echo "defaultrouter=\"${NETROUTE}\"" >>${FSMNT}/etc/rc.conf
248   fi
249   get_value_from_cfg netSaveIPv6DefaultRouter
250   NETROUTE="${VAL}"
251   if [ -n "${NETROUTE}" ]
252   then
253     echo "ipv6_defaultrouter=\"${NETROUTE}\"" >>${FSMNT}/etc/rc.conf
254   fi
255
256   # Check if we have a nameserver to enable
257   : > ${FSMNT}/etc/resolv.conf
258   get_value_from_cfg netSaveNameServer
259   NAMESERVER="${VAL}"
260   if [ -n "${NAMESERVER}" ]
261   then
262     echo "nameserver ${NAMESERVER}" >>${FSMNT}/etc/resolv.conf
263   fi
264   get_value_from_cfg netSaveIPv6NameServer
265   NAMESERVER="${VAL}"
266   if [ -n "${NAMESERVER}" ]
267   then
268     echo "nameserver ${NAMESERVER}" >>${FSMNT}/etc/resolv.conf
269   fi
270
271 };
272
273 # Function which determines if a nic is active / up
274 is_nic_active()
275 {
276   ifconfig ${1} | grep -q "status: active" 2>/dev/null
277   if [ $? -eq 0 ] ; then
278     return 0
279   else
280     return 1
281   fi
282 };
283
284
285 # Function which detects available nics, and runs DHCP on them until
286 # a success is found
287 enable_auto_dhcp()
288 {
289   # start by getting a list of nics on this system
290   ${QUERYDIR}/detect-nics.sh > ${TMPDIR}/.niclist
291   while read line
292   do
293     NIC="`echo $line | cut -d ':' -f 1`"
294     DESC="`echo $line | cut -d ':' -f 2`"
295
296     is_nic_active "${NIC}"
297     if [ $? -eq 0 ] ; then
298       echo_log "Trying DHCP on $NIC $DESC"
299       dhclient ${NIC} >/dev/null 2>/dev/null
300       if [ $? -eq 0 ] ; then
301         # Got a valid DHCP IP, we can return now
302             export WRKNIC="$NIC"
303             return 0
304           fi
305     fi
306   done < ${TMPDIR}/.niclist 
307
308 };
309
310 # Function which detects available nics, and runs rtsol on them.
311 enable_auto_slaac()
312 {
313
314   # start by getting a list of nics on this system
315   ${QUERYDIR}/detect-nics.sh > ${TMPDIR}/.niclist
316   ALLNICS=""
317   while read line
318   do
319     NIC="`echo $line | cut -d ':' -f 1`"
320     DESC="`echo $line | cut -d ':' -f 2`"
321
322     is_nic_active "${NIC}"
323     if [ $? -eq 0 ] ; then
324       echo_log "Will try IPv6 SLAAC on $NIC $DESC"
325       ifconfig ${NIC} inet6 -ifdisabled accept_rtadv up
326       ALLNICS="${ALLNICS} ${NIC}"
327     fi
328   done < ${TMPDIR}/.niclist 
329
330   # XXX once we support it in-tree call /sbin/resovconf here.
331   echo_log "Running rtsol on ${ALLNICS}"
332   rtsol -F ${ALLNICS} >/dev/null 2>/dev/null
333 }
334
335 # Get the mac address of a target NIC
336 get_nic_mac()
337 {
338   FOUNDMAC="`ifconfig ${1} | grep 'ether' | tr -d '\t' | cut -d ' ' -f 2`"
339   export FOUNDMAC
340 }
341
342 # Function which performs the manual setup of a target nic in the cfg
343 enable_manual_nic()
344 {
345   # Get the target nic
346   NIC="$1"
347
348   # Check that this NIC exists
349   rc_halt "ifconfig ${NIC}"
350
351   get_value_from_cfg netIP
352   NETIP="${VAL}"
353   
354   if [ "$NETIP" = "DHCP" ]
355   then
356     echo_log "Enabling DHCP on $NIC"
357     rc_halt "dhclient ${NIC}"
358     return 0
359   fi
360
361   # If we get here, we have a manual setup, lets do so now
362
363   # IPv4:
364
365   # Set the manual IP
366   if [ -n "${NETIP}" ]
367   then
368     # Check if we have a netmask to set
369     get_value_from_cfg netMask
370     NETMASK="${VAL}"
371     if [ -n "${NETMASK}" ]
372     then
373       rc_halt "ifconfig inet ${NIC} netmask ${NETMASK}"
374     else
375       rc_halt "ifconfig inet ${NIC} ${NETIP}"
376     fi
377   fi
378
379   # Check if we have a default router to set
380   get_value_from_cfg netDefaultRouter
381   NETROUTE="${VAL}"
382   if [ -n "${NETROUTE}" ]
383   then
384     rc_halt "route add -inet default ${NETROUTE}"
385   fi
386
387   # IPv6:
388
389   # Set static IPv6 address
390   get_value_from_cfg netIPv6
391   NETIP="${VAL}"
392   if [ -n ${NETIP} ]
393   then
394       rc_halt "ifconfig inet6 ${NIC} ${NETIP} -ifdisabled up"
395   fi
396
397   # Default router
398   get_value_from_cfg netIPv6DefaultRouter
399   NETROUTE="${VAL}"
400   if [ -n "${NETROUTE}" ]
401   then
402     rc_halt "route add -inet6 default ${NETROUTE}"
403   fi
404
405   # Check if we have a nameserver to enable
406   : >/etc/resolv.conf
407   get_value_from_cfg netNameServer
408   NAMESERVER="${VAL}"
409   if [ -n "${NAMESERVER}" ]
410   then
411     echo "nameserver ${NAMESERVER}" >>/etc/resolv.conf
412   fi
413   get_value_from_cfg netIPv6NameServer
414   NAMESERVER="${VAL}"
415   if [ -n "${NAMESERVER}" ]
416   then
417     echo "nameserver ${NAMESERVER}" >>/etc/resolv.conf
418   fi
419
420 };
421
422
423 # Function which parses the cfg and enables networking per specified
424 start_networking()
425 {
426   # Check if we have any networking requested
427   get_value_from_cfg netDev
428   if [ -z "${VAL}" ]
429   then
430     return 0
431   fi
432
433   NETDEV="${VAL}"
434   if [ "$NETDEV" = "AUTO-DHCP" ]
435   then
436     enable_auto_dhcp
437   elif [ "$NETDEV" = "IPv6-SLAAC" ]
438   then
439     enable_auto_slaac
440   elif [ "$NETDEV" = "AUTO-DHCP-SLAAC" ]
441   then
442     enable_auto_dhcp
443     enable_auto_slaac
444   else
445     enable_manual_nic ${NETDEV}
446   fi
447
448 };
449
450
451 # Function which checks the cfg and enables the specified networking on
452 # the installed system
453 save_networking_install()
454 {
455
456   # Check if we have any networking requested to save
457   get_value_from_cfg netSaveDev
458   if [ -z "${VAL}" ]
459   then
460     return 0
461   fi
462
463   NETDEV="${VAL}"
464   if [ "$NETDEV" = "AUTO-DHCP" ]
465   then
466     save_auto_dhcp
467   elif [ "$NETDEV" = "IPv6-SLAAC" ]
468   then
469     save_auto_slaac
470   elif [ "$NETDEV" = "AUTO-DHCP-SLAAC" ]
471   then
472     save_auto_dhcp
473     save_auto_slaac
474   else
475     save_manual_nic ${NETDEV}
476   fi
477
478 };