xref: /freebsd/usr.sbin/bsdconfig/share/media/common.subr (revision 734e82fe33aa764367791a7d603b383996c6b40b)
1if [ ! "$_MEDIA_COMMON_SUBR" ]; then _MEDIA_COMMON_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#
28############################################################ INCLUDES
29
30BSDCFG_SHARE="/usr/share/bsdconfig"
31. $BSDCFG_SHARE/common.subr || exit 1
32f_dprintf "%s: loading includes..." media/common.subr
33f_include $BSDCFG_SHARE/device.subr
34f_include $BSDCFG_SHARE/media/any.subr
35f_include $BSDCFG_SHARE/struct.subr
36
37############################################################ GLOBALS
38
39#
40# Where to mount media
41#
42MOUNTPOINT=/dist
43
44#
45# Media probe values to use for `f_media_get_TYPE media $file $PROBE' or
46# `f_device_get device_media $file $PROBE' (where $PROBE is one of the below
47# values).
48#
49PROBE_EXIST=1
50PROBE_SIZE=2
51
52############################################################ FUNCTIONS
53
54# f_media_open
55#
56# Returms success if able to initialize the media device.
57#
58f_media_open()
59{
60	f_dprintf "f_media_open: Verifying and initiliazing media device"
61	{ # Verify and initialize device media if-defined
62		f_struct device_media &&
63		f_media_verify &&
64		f_device_init device_media
65	} || return $FAILURE
66}
67
68# f_media_close
69#
70# Shuts down the media device, see f_device_shutdown() from device.subr for
71# more details.
72#
73f_media_close()
74{
75	f_dprintf "f_media_close: Shutting down media device"
76	f_struct device_media &&
77		f_device_shutdown device_media
78	f_struct_free device_media
79}
80
81# f_media_verify
82#
83# Returns success if the media device is available, and if not, prompts the
84# user to select a media type. See f_media_get_type() from media/any.subr for
85# more details.
86#
87f_media_verify()
88{
89	f_dprintf "f_media_verify: Verifying media device"
90	f_struct device_media || f_media_get_type
91}
92
93# f_media_generic_get $base $file [$probe_type]
94#
95# A generic open which follows a well-known "path" of places to look. If
96# $probe_type is present and non-NULL, returns success if $file exists. If
97# $probe_type is equal to $PROBE_SIZE, prints the size of $file in bytes to
98# standard-out.
99#
100f_media_generic_get()
101{
102	local funcname=f_media_generic_get
103	local base="$1" file="$2" probe_type="$3"
104
105	local fname=f_media_generic_get
106	f_dprintf "%s: base=[%s] files=[%s] probe_type=%s" \
107	          $fname "$base" "$file" "$probe_type"
108
109	local rel path
110	f_getvar $VAR_RELNAME rel
111	for path in \
112		"$base/$file" \
113		"$base/FreeBSD/$file" \
114		"$base/releases/$file" \
115		"$base/$rel/$file" \
116	; do
117		if [ -f "$path" -a -r "$path" ]; then
118			f_dprintf "%s: file exists path=[%s]" $fname "$path"
119			if [ "$probe_type" = "$PROBE_SIZE" ]; then
120				local size
121				f_eval_catch -dk size $funcname stat \
122					'stat -f %%z "%s"' "$path" || size=-1
123				f_isinteger "$size" || size=-1
124				echo $size
125			fi
126			[ "$probe_type" ] && return $SUCCESS
127			cat "$path"
128			return $?
129		fi
130	done
131
132	path="$base/releases/$rel/$file" # Final path to try
133	if [ -f "$path" -a -r "$path" ]; then
134		f_dprintf "%s: file exists path=[%s]" $fname "$path"
135		if [ "$probe_type" = "$PROBE_SIZE" ]; then
136			local size
137			f_eval_catch -dk size $funcname stat \
138				'stat -f %%z "%s"' "$path" || size=-1
139			f_isinteger "$size" || size=-1
140			echo $size
141		fi
142		[ "$probe_type" ] && return $SUCCESS
143	elif [ "$probe_type" ]; then
144		[ "$probe_type" = "$PROBE_SIZE" ] && echo "-1"
145		return $FAILURE
146	fi
147	cat "$base/releases/$rel/$file" # Final path to try
148}
149
150############################################################ MAIN
151
152f_dprintf "%s: Successfully loaded." media/common.subr
153
154fi # ! $_MEDIA_COMMON_SUBR
155