1#!/bin/sh
2#-
3# Copyright (c) 2010 iXsystems, Inc. All rights reserved.
4#
5# Redistribution and use in source and binary forms, with or without
6# modification, are permitted provided that the following conditions
7# are met:
8# 1. Redistributions of source code must retain the above copyright
9# notice, this list of conditions and the following disclaimer.
10# 2. Redistributions in binary form must reproduce the above copyright
11# notice, this list of conditions and the following disclaimer in the
12# documentation and/or other materials provided with the distribution.
13#
14# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24# SUCH DAMAGE.
25#
| 1#!/bin/sh
2#-
3# Copyright (c) 2010 iXsystems, Inc. All rights reserved.
4#
5# Redistribution and use in source and binary forms, with or without
6# modification, are permitted provided that the following conditions
7# are met:
8# 1. Redistributions of source code must retain the above copyright
9# notice, this list of conditions and the following disclaimer.
10# 2. Redistributions in binary form must reproduce the above copyright
11# notice, this list of conditions and the following disclaimer in the
12# documentation and/or other materials provided with the distribution.
13#
14# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24# SUCH DAMAGE.
25#
|
27
28# functions.sh
29# Library of functions which pc-sysinstall may call upon
30
31# Function which displays the help-index file
32display_help()
33{
34 if [ -e "${PROGDIR}/doc/help-index" ]
35 then
36 cat ${PROGDIR}/doc/help-index
37 else
38 echo "Error: ${PROGDIR}/doc/help-index not found"
39 exit 1
40 fi
41};
42
43# Function which displays the help for a specified command
44display_command_help()
45{
46 if [ -z "$1" ]
47 then
48 echo "Error: No command specified to display help for"
49 exit 1
50 fi
51
52 if [ -e "${PROGDIR}/doc/help-${1}" ]
53 then
54 cat ${PROGDIR}/doc/help-${1}
55 else
56 echo "Error: ${PROGDIR}/doc/help-${1} not found"
57 exit 1
58 fi
59};
60
61# Function to convert bytes to megabytes
62convert_byte_to_megabyte()
63{
64 if [ -z "${1}" ]
65 then
66 echo "Error: No bytes specified!"
67 exit 1
68 fi
69
70 expr -e ${1} / 1048576
71};
72
73# Function to convert blocks to megabytes
74convert_blocks_to_megabyte()
75{
76 if [ -z "${1}" ] ; then
77 echo "Error: No blocks specified!"
78 exit 1
79 fi
80
81 expr -e ${1} / 2048
82};
83
84# Takes $1 and strips the whitespace out of it, returns VAL
85strip_white_space()
86{
87 if [ -z "${1}" ]
88 then
89 echo "Error: No value setup to strip whitespace from!"
90
91 exit 1
92 fi
93
94 VAL=`echo "$1" | tr -d ' '`
95 export VAL
96};
97
98# Displays an error message and exits with error 1
99exit_err()
100{
101 # Echo the message for the users benefit
102 echo "$1"
103
104 # Save this error to the log file
105 echo "${1}" >>$LOGOUT
106
107 # Check if we need to unmount any file-systems after this failure
108 unmount_all_filesystems_failure
109
110 echo "For more details see log file: $LOGOUT"
111
112 exit 1
113};
114
115# Run-command, don't halt if command exits with non-0
116rc_nohalt()
117{
118 CMD="$1"
119
120 if [ -z "${CMD}" ]
121 then
122 exit_err "Error: missing argument in rc_nohalt()"
123 fi
124
125 echo "Running: ${CMD}" >>${LOGOUT}
126 ${CMD} >>${LOGOUT} 2>>${LOGOUT}
127
128};
129
130# Run-command, halt if command exits with non-0
131rc_halt()
132{
133 CMD="$1"
134
135 if [ -z "${CMD}" ]
136 then
137 exit_err "Error: missing argument in rc_halt()"
138 fi
139
140 echo "Running: ${CMD}" >>${LOGOUT}
141 ${CMD} >>${LOGOUT} 2>>${LOGOUT}
142 STATUS="$?"
143 if [ "${STATUS}" != "0" ]
144 then
145 exit_err "Error ${STATUS}: ${CMD}"
146 fi
147};
148
149# Run-command w/echo to screen, halt if command exits with non-0
150rc_halt_echo()
151{
152 CMD="$1"
153
154 if [ -z "${CMD}" ]
155 then
156 exit_err "Error: missing argument in rc_halt_echo()"
157 fi
158
159 echo "Running: ${CMD}" >>${LOGOUT}
160 ${CMD} 2>&1 | tee -a ${LOGOUT}
161 STATUS="$?"
162 if [ "$STATUS" != "0" ]
163 then
164 exit_err "Error ${STATUS}: $CMD"
165 fi
166
167};
168
169# Run-command w/echo, don't halt if command exits with non-0
170rc_nohalt_echo()
171{
172 CMD="$1"
173
174 if [ -z "${CMD}" ]
175 then
176 exit_err "Error: missing argument in rc_nohalt_echo()"
177 fi
178
179 echo "Running: ${CMD}" >>${LOGOUT}
180 ${CMD} 2>&1 | tee -a ${LOGOUT}
181
182};
183
184# Echo to the screen and to the log
185echo_log()
186{
187 STR="$1"
188
189 if [ -z "${STR}" ]
190 then
191 exit_err "Error: missing argument in echo_log()"
192 fi
193
194 echo "${STR}" | tee -a ${LOGOUT}
195};
196
197# Make sure we have a numeric
198is_num()
199{
200 expr $1 + 1 2>/dev/null
201 return $?
202}
203
204# Function which uses "fetch" to download a file, and display a progress report
205fetch_file()
206{
207
208 FETCHFILE="$1"
209 FETCHOUTFILE="$2"
210 EXITFAILED="$3"
211
212 SIZEFILE="${TMPDIR}/.fetchSize"
213 EXITFILE="${TMPDIR}/.fetchExit"
214
215 rm ${SIZEFILE} 2>/dev/null >/dev/null
216 rm ${FETCHOUTFILE} 2>/dev/null >/dev/null
217
218 fetch -s "${FETCHFILE}" >${SIZEFILE}
219 SIZE="`cat ${SIZEFILE}`"
220 SIZE="`expr ${SIZE} / 1024`"
221 echo "FETCH: ${FETCHFILE}"
222 echo "FETCH: ${FETCHOUTFILE}" >>${LOGOUT}
223
224 ( fetch -o ${FETCHOUTFILE} "${FETCHFILE}" >/dev/null 2>/dev/null ; echo "$?" > ${EXITFILE} ) &
225 PID="$!"
226 while
227 z=1
228 do
229
230 if [ -e "${FETCHOUTFILE}" ]
231 then
232 DSIZE=`du -k ${FETCHOUTFILE} | tr -d '\t' | cut -d '/' -f 1`
233 if [ $(is_num "$DSIZE") ] ; then
234 if [ $SIZE -lt $DSIZE ] ; then DSIZE="$SIZE"; fi
235 echo "SIZE: ${SIZE} DOWNLOADED: ${DSIZE}"
236 echo "SIZE: ${SIZE} DOWNLOADED: ${DSIZE}" >>${LOGOUT}
237 fi
238 fi
239
240 # Check if the download is finished
241 ps -p ${PID} >/dev/null 2>/dev/null
242 if [ "$?" != "0" ]
243 then
244 break;
245 fi
246
247 sleep 2
248 done
249
250 echo "FETCHDONE"
251
252 EXIT="`cat ${EXITFILE}`"
253 if [ "${EXIT}" != "0" -a "$EXITFAILED" = "1" ]
254 then
255 exit_err "Error: Failed to download ${FETCHFILE}"
256 fi
257
258 return $EXIT
259
260};
261
262# Function to return a the zpool name for this device
263get_zpool_name()
264{
265 DEVICE="$1"
266
267 # Set the base name we use for zpools
268 BASENAME="tank"
269
270 if [ ! -d "${TMPDIR}/.zpools" ] ; then
271 mkdir -p ${TMPDIR}/.zpools
272 fi
273
274 if [ -e "${TMPDIR}/.zpools/${DEVICE}" ] ; then
275 cat ${TMPDIR}/.zpools/${DEVICE}
276 return 0
277 else
278 # Need to generate a zpool name for this device
279 NUM=`ls ${TMPDIR}/.zpools/ | wc -l | sed 's| ||g'`
280 NEWNAME="${BASENAME}${NUM}"
281 echo "$NEWNAME" >${TMPDIR}/.zpools/${DEVICE}
282 echo "${NEWNAME}"
283 return
284 fi
285};
| 27
28# functions.sh
29# Library of functions which pc-sysinstall may call upon
30
31# Function which displays the help-index file
32display_help()
33{
34 if [ -e "${PROGDIR}/doc/help-index" ]
35 then
36 cat ${PROGDIR}/doc/help-index
37 else
38 echo "Error: ${PROGDIR}/doc/help-index not found"
39 exit 1
40 fi
41};
42
43# Function which displays the help for a specified command
44display_command_help()
45{
46 if [ -z "$1" ]
47 then
48 echo "Error: No command specified to display help for"
49 exit 1
50 fi
51
52 if [ -e "${PROGDIR}/doc/help-${1}" ]
53 then
54 cat ${PROGDIR}/doc/help-${1}
55 else
56 echo "Error: ${PROGDIR}/doc/help-${1} not found"
57 exit 1
58 fi
59};
60
61# Function to convert bytes to megabytes
62convert_byte_to_megabyte()
63{
64 if [ -z "${1}" ]
65 then
66 echo "Error: No bytes specified!"
67 exit 1
68 fi
69
70 expr -e ${1} / 1048576
71};
72
73# Function to convert blocks to megabytes
74convert_blocks_to_megabyte()
75{
76 if [ -z "${1}" ] ; then
77 echo "Error: No blocks specified!"
78 exit 1
79 fi
80
81 expr -e ${1} / 2048
82};
83
84# Takes $1 and strips the whitespace out of it, returns VAL
85strip_white_space()
86{
87 if [ -z "${1}" ]
88 then
89 echo "Error: No value setup to strip whitespace from!"
90
91 exit 1
92 fi
93
94 VAL=`echo "$1" | tr -d ' '`
95 export VAL
96};
97
98# Displays an error message and exits with error 1
99exit_err()
100{
101 # Echo the message for the users benefit
102 echo "$1"
103
104 # Save this error to the log file
105 echo "${1}" >>$LOGOUT
106
107 # Check if we need to unmount any file-systems after this failure
108 unmount_all_filesystems_failure
109
110 echo "For more details see log file: $LOGOUT"
111
112 exit 1
113};
114
115# Run-command, don't halt if command exits with non-0
116rc_nohalt()
117{
118 CMD="$1"
119
120 if [ -z "${CMD}" ]
121 then
122 exit_err "Error: missing argument in rc_nohalt()"
123 fi
124
125 echo "Running: ${CMD}" >>${LOGOUT}
126 ${CMD} >>${LOGOUT} 2>>${LOGOUT}
127
128};
129
130# Run-command, halt if command exits with non-0
131rc_halt()
132{
133 CMD="$1"
134
135 if [ -z "${CMD}" ]
136 then
137 exit_err "Error: missing argument in rc_halt()"
138 fi
139
140 echo "Running: ${CMD}" >>${LOGOUT}
141 ${CMD} >>${LOGOUT} 2>>${LOGOUT}
142 STATUS="$?"
143 if [ "${STATUS}" != "0" ]
144 then
145 exit_err "Error ${STATUS}: ${CMD}"
146 fi
147};
148
149# Run-command w/echo to screen, halt if command exits with non-0
150rc_halt_echo()
151{
152 CMD="$1"
153
154 if [ -z "${CMD}" ]
155 then
156 exit_err "Error: missing argument in rc_halt_echo()"
157 fi
158
159 echo "Running: ${CMD}" >>${LOGOUT}
160 ${CMD} 2>&1 | tee -a ${LOGOUT}
161 STATUS="$?"
162 if [ "$STATUS" != "0" ]
163 then
164 exit_err "Error ${STATUS}: $CMD"
165 fi
166
167};
168
169# Run-command w/echo, don't halt if command exits with non-0
170rc_nohalt_echo()
171{
172 CMD="$1"
173
174 if [ -z "${CMD}" ]
175 then
176 exit_err "Error: missing argument in rc_nohalt_echo()"
177 fi
178
179 echo "Running: ${CMD}" >>${LOGOUT}
180 ${CMD} 2>&1 | tee -a ${LOGOUT}
181
182};
183
184# Echo to the screen and to the log
185echo_log()
186{
187 STR="$1"
188
189 if [ -z "${STR}" ]
190 then
191 exit_err "Error: missing argument in echo_log()"
192 fi
193
194 echo "${STR}" | tee -a ${LOGOUT}
195};
196
197# Make sure we have a numeric
198is_num()
199{
200 expr $1 + 1 2>/dev/null
201 return $?
202}
203
204# Function which uses "fetch" to download a file, and display a progress report
205fetch_file()
206{
207
208 FETCHFILE="$1"
209 FETCHOUTFILE="$2"
210 EXITFAILED="$3"
211
212 SIZEFILE="${TMPDIR}/.fetchSize"
213 EXITFILE="${TMPDIR}/.fetchExit"
214
215 rm ${SIZEFILE} 2>/dev/null >/dev/null
216 rm ${FETCHOUTFILE} 2>/dev/null >/dev/null
217
218 fetch -s "${FETCHFILE}" >${SIZEFILE}
219 SIZE="`cat ${SIZEFILE}`"
220 SIZE="`expr ${SIZE} / 1024`"
221 echo "FETCH: ${FETCHFILE}"
222 echo "FETCH: ${FETCHOUTFILE}" >>${LOGOUT}
223
224 ( fetch -o ${FETCHOUTFILE} "${FETCHFILE}" >/dev/null 2>/dev/null ; echo "$?" > ${EXITFILE} ) &
225 PID="$!"
226 while
227 z=1
228 do
229
230 if [ -e "${FETCHOUTFILE}" ]
231 then
232 DSIZE=`du -k ${FETCHOUTFILE} | tr -d '\t' | cut -d '/' -f 1`
233 if [ $(is_num "$DSIZE") ] ; then
234 if [ $SIZE -lt $DSIZE ] ; then DSIZE="$SIZE"; fi
235 echo "SIZE: ${SIZE} DOWNLOADED: ${DSIZE}"
236 echo "SIZE: ${SIZE} DOWNLOADED: ${DSIZE}" >>${LOGOUT}
237 fi
238 fi
239
240 # Check if the download is finished
241 ps -p ${PID} >/dev/null 2>/dev/null
242 if [ "$?" != "0" ]
243 then
244 break;
245 fi
246
247 sleep 2
248 done
249
250 echo "FETCHDONE"
251
252 EXIT="`cat ${EXITFILE}`"
253 if [ "${EXIT}" != "0" -a "$EXITFAILED" = "1" ]
254 then
255 exit_err "Error: Failed to download ${FETCHFILE}"
256 fi
257
258 return $EXIT
259
260};
261
262# Function to return a the zpool name for this device
263get_zpool_name()
264{
265 DEVICE="$1"
266
267 # Set the base name we use for zpools
268 BASENAME="tank"
269
270 if [ ! -d "${TMPDIR}/.zpools" ] ; then
271 mkdir -p ${TMPDIR}/.zpools
272 fi
273
274 if [ -e "${TMPDIR}/.zpools/${DEVICE}" ] ; then
275 cat ${TMPDIR}/.zpools/${DEVICE}
276 return 0
277 else
278 # Need to generate a zpool name for this device
279 NUM=`ls ${TMPDIR}/.zpools/ | wc -l | sed 's| ||g'`
280 NEWNAME="${BASENAME}${NUM}"
281 echo "$NEWNAME" >${TMPDIR}/.zpools/${DEVICE}
282 echo "${NEWNAME}"
283 return
284 fi
285};
|