diff options
-rw-r--r-- | src/local.mk | 1 | ||||
-rw-r--r-- | src/temp.sh | 54 |
2 files changed, 55 insertions, 0 deletions
diff --git a/src/local.mk b/src/local.mk index 0ce1d77..0734fa7 100644 --- a/src/local.mk +++ b/src/local.mk @@ -4,6 +4,7 @@ firman_SOURCES += \ src/flashrom.sh \ src/main.sh \ src/rand.sh \ + src/temp.sh \ src/ui.sh include $(top_srcdir)/src/action/local.mk diff --git a/src/temp.sh b/src/temp.sh new file mode 100644 index 0000000..2e6885d --- /dev/null +++ b/src/temp.sh @@ -0,0 +1,54 @@ +# Temporary file management +# +# Copyright (C) 2015 Patrick "P. J." McDermott +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +temp_dir='' + +temp_init() +{ + local tmpdir='' + + if [ "x${TMPDIR:+set}" = 'xset' ]; then + tmpdir="${TMPDIR}" + else + tmpdir='/tmp' + fi + + rand + temp_dir="${tmpdir}/firman-$(printf '%010d' ${rand_x})" + if ! mkdir "${temp_dir}"; then + err 'Failed to create temporary directory' + return 1 + fi + + return 0 +} + +temp_fini() +{ + if ! rmdir "${temp_dir}"; then + err 'Failed to remove temporary directory' + return 1 + fi + + return 0 +} + +get_temp_dir() +{ + printf '%s' "${temp_dir}" + return 0 +} |