xref: /freebsd/usr.sbin/bsdconfig/share/media/nfs.subr (revision 4ec234c813eed05c166859bba82c882e40826eb9) 
1if [ ! "$_MEDIA_NFS_SUBR" ]; then _MEDIA_NFS_SUBR=1
2#
3# Copyright (c) 2012-2013 Devin Teske
4# All rights reserved.
5#
6# Redistribution and use in source and binary forms, with or without
7# modification, are permitted provided that the following conditions
8# are met:
9# 1. Redistributions of source code must retain the above copyright
10#    notice, this list of conditions and the following disclaimer.
11# 2. Redistributions in binary form must reproduce the above copyright
12#    notice, this list of conditions and the following disclaimer in the
13#    documentation and/or other materials provided with the distribution.
14#
15# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25# SUCH DAMAGE.
26#
27# $FreeBSD$
28#
29############################################################ INCLUDES
30
31BSDCFG_SHARE="/usr/share/bsdconfig"
32. $BSDCFG_SHARE/common.subr || exit 1
33f_dprintf "%s: loading includes..." media/nfs.subr
34f_include $BSDCFG_SHARE/device.subr
35f_include $BSDCFG_SHARE/dialog.subr
36f_include $BSDCFG_SHARE/media/common.subr
37f_include $BSDCFG_SHARE/media/network.subr
38f_include $BSDCFG_SHARE/media/tcpip.subr
39f_include $BSDCFG_SHARE/struct.subr
40f_include $BSDCFG_SHARE/variable.subr
41
42BSDCFG_LIBE="/usr/libexec/bsdconfig"
43f_include_lang $BSDCFG_LIBE/include/messages.subr
44
45############################################################ GLOBALS
46
47NFS_MOUNTED=
48
49############################################################ FUNCTIONS
50
51# f_media_set_nfs
52#
53# Return success if we both found and set the media type to be an NFS server.
54# Variables from variable.subr that can be used to script user input:
55#
56# 	VAR_NFS_PATH
57# 		The NFS path specification (host:path) to use when mounting the
58# 		remote repository.
59# 	VAR_NAMESERVER [Optional]
60# 		Automatically populated from resolv.conf(5) but can be
61# 		overridden. If set, the host portion of VAR_NFS_PATH is
62# 		looked up using f_host_lookup() from `tcpip.subr'.
63#
64# Meanwhile, the following variables from variable.subr are set after
65# successful execution:
66#
67# 	VAR_NFS_HOST
68# 		The host portion of the NFS path specification, parsed from
69# 		VAR_NFS_PATH.
70#
71f_media_set_nfs()
72{
73	local nfs
74
75	f_media_close
76
77	f_variable_get_value $VAR_NFS_PATH \
78		"$msg_please_enter_the_full_nfs_file_specification"
79	f_getvar $VAR_NFS_PATH nfs
80	[ "$nfs" ] || return $FAILURE
81
82	case "$nfs" in
83	*:*) : valid NFS path ;;
84	*)
85		f_show_msg "$msg_invalid_nfs_path_specification"
86		return $FAILURE
87	esac
88
89	f_struct_new DEVICE device_nfs
90	device_nfs set name "$nfs"
91
92	if ! f_struct device_network ||
93	   ! f_dialog_yesno "$msg_youve_already_done_the_network_configuration"
94	then
95		f_struct device_network &&
96			f_device_shutdown network
97		f_device_select_tcp || return $FAILURE
98		local dev
99		f_getvar $VAR_NETWORK_DEVICE dev
100		f_struct_copy "device_$dev" device_network
101	fi
102	f_device_init network ||
103		f_dprintf "%s: $msg_net_device_init_failed\n" f_media_set_nfs
104
105	local hostname="${nfs%%:*}"
106	if f_isset $VAR_NAMESERVER && ! {
107		f_validate_ipaddr "$hostname" || f_validate_ipaddr6 "$hostname"
108	}; then
109		f_show_info "$msg_looking_up_host" "$hostname"
110		f_dprintf "%s Looking up hostname, %s, using host(1)" \
111		          "f_media_set_nfs" "$hostname"
112		if ! f_quietly f_host_lookup "$hostname"; then
113			f_show_msg "$msg_cannot_resolve_hostname" "$hostname"
114			f_struct device_network &&
115				f_device_shutdown network
116			f_struct_free device_network
117			unset $VAR_NFS_PATH
118			return $FAILURE
119		fi
120		f_dprintf "Found DNS entry for %s successfully." "$hostname"
121	fi
122
123	setvar $VAR_NFS_HOST "$hostname"
124
125	device_nfs set type     $DEVICE_TYPE_NFS
126	device_nfs set init     f_media_init_nfs
127	device_nfs set get      f_media_get_nfs
128	device_nfs set shutdown f_media_shutdown_nfs
129	device_nfs set private  device_network # in name only (deref'd later)
130
131	f_struct_copy device_nfs device_media
132	f_struct_free device_nfs
133
134	return $SUCCESS
135}
136
137# f_media_init_nfs $device
138#
139# Initializes the NFS media device. Returns success if able to mount the NFS
140# device using mount_nfs(1).
141#
142# The variables (from variable.subr) used to initialize the NFS mount are as
143# follows (all of which are configured manually/optionally from the options
144# menu):
145#
146# 	VAR_NFS_TCP [Optional]
147# 		If non-NULL, adds the "tcp" option via `-o' to mount_nfs(8).
148# 	VAR_NFS_V3 [Optional]
149# 		If non-NULL, adds the "nfsv3" option via `-o' to mount_nfs(8).
150# 	VAR_NFS_SECURE [Optional]
151# 		If non-NULL, adds the "-P" flag to mount_nfs(8).
152# 	VAR_SLOW_ETHER [Optional]
153# 		If non-NULL, adjusts the read/write size to avoid timeouts.
154#
155f_media_init_nfs()
156{
157	local funcname=f_media_init_nfs
158	local dev="$1" name err
159
160	device_$dev get name name || return $FAILURE
161	f_dprintf "Init routine called for NFS device. name=[%s]" \
162	          "$name"
163
164	if [ "$NFS_MOUNTED" ]; then
165		f_dprintf "NFS device already mounted."
166		return $SUCCESS
167	fi
168
169	if ! f_device_init network; then
170		f_dprintf "f_media_init_nfs: %s" "$msg_net_device_init_failed"
171		return $FAILURE
172	fi
173
174	if [ ! -e "$MOUNTPOINT" ]; then
175		f_eval_catch $funcname mkdir 'mkdir -p "%s"' "$MOUNTPOINT" ||
176			return $FAILURE
177	fi
178
179	local cp tcp="" use3="" secure="" readsize=4096 writesize=4096
180	f_getvar $VAR_NFS_TCP cp
181	[ "$cp" = "YES" ] && tcp=1
182	f_getvar $VAR_NFS_V3 cp
183	[ "$cp" = "YES" ] && use3=1
184	f_getvar $VAR_NFS_SECURE cp
185	[ "$cp" = "YES" ] && secure=1
186	f_getvar $VAR_SLOW_ETHER cp
187	[ "$cp" = "YES" ] && readsize=1024 writesize=1024
188
189	local options="rsize=$readsize,wsize=$writesize"
190	[ "$use3" ] && options="$options,nfsv3"
191	[ "$tcp" ] && options="$options,tcp"
192
193	if ! f_eval_catch -dk err $funcname mount_nfs \
194		'mount_nfs %s -o "%s" "%s" "%s"' \
195		"${secure:+-P}" "$options" "$name" "$MOUNTPOINT"
196	then
197		err="${err#mount_nfs: }"
198		f_show_msg "$msg_error_mounting_device" \
199		           "$name" "$MOUNTPOINT" "$err"
200		f_struct device_network &&
201			f_device_shutdown network
202		return $FAILURE
203	fi
204	NFS_MOUNTED=1
205
206	f_dprintf "Mounted NFS device %s onto %s" "$name" "$MOUNTPOINT"
207
208	return $SUCCESS
209}
210
211# f_media_get_nfs $device $file [$probe_type]
212#
213# Returns data from $file on a mounted NFS device. Similar to cat(1). If
214# $probe_type is present and non-NULL, returns success if $file exists. If
215# $probe_type is equal to $PROBE_SIZE, prints the size of $file in bytes to
216# standard-out.
217#
218f_media_get_nfs()
219{
220	local dev="$1" file="$2" probe_type="$3"
221
222	f_dprintf "f_media_get_nfs: dev=[%s] file=[%s] probe_type=%s" \
223	          "$dev" "$file" "$probe_type"
224
225	f_media_generic_get "$MOUNTPOINT" "$file" "$probe_type"
226}
227
228# f_media_shutdown_nfs $device
229#
230# Shuts down the NFS device using umount(8). Return status should be ignored.
231#
232f_media_shutdown_nfs()
233{
234	local funcname=f_media_shutdown_nfs
235	local dev="$1" err
236
237	[ "$NFS_MOUNTED" ] || return $FAILURE
238
239	f_dprintf "Unmounting NFS partition on %s" "$MOUNTPOINT"
240	if ! f_eval_catch -dk err $funcname umount \
241		'umount -f "%s"' "$MOUNTPOINT"
242	then
243		err="${err#umount: }"; err="${err#*: }"
244		f_show_msg "$msg_could_not_unmount_the_nfs_partition" \
245		           "$MOUNTPOINT" "$err"
246	else
247		NFS_MOUNTED=
248	fi
249}
250
251############################################################ MAIN
252
253f_dprintf "%s: Successfully loaded." media/nfs.subr
254
255fi # ! $_MEDIA_NFS_SUBR
256