xref: /openbsd/usr.sbin/pkg_add/OpenBSD/Temp.pm (revision 898184e3) 
1# ex:ts=8 sw=4:
2# $OpenBSD: Temp.pm,v 1.25 2012/04/28 15:22:49 espie Exp $
3#
4# Copyright (c) 2003-2005 Marc Espie <espie@openbsd.org>
5#
6# Permission to use, copy, modify, and distribute this software for any
7# purpose with or without fee is hereby granted, provided that the above
8# copyright notice and this permission notice appear in all copies.
9#
10# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17
18use strict;
19use warnings;
20
21package OpenBSD::Temp;
22
23use OpenBSD::MkTemp;
24use OpenBSD::Paths;
25use OpenBSD::Error;
26
27our $tempbase = $ENV{'PKG_TMPDIR'} || OpenBSD::Paths->vartmp;
28
29my $dirs = {};
30my $files = {};
31
32my $cleanup = sub {
33	while (my ($name, $pid) = each %$files) {
34		unlink($name) if $pid == $$;
35	}
36	while (my ($dir, $pid) = each %$dirs) {
37		OpenBSD::Error->rmtree([$dir]) if $pid == $$;
38	}
39};
40
41END {
42	&$cleanup;
43}
44OpenBSD::Handler->register($cleanup);
45
46sub dir
47{
48	my $caught;
49	my $h = sub { $caught = shift; };
50	my $dir;
51
52	{
53	    local $SIG{'INT'} = $h;
54	    local $SIG{'QUIT'} = $h;
55	    local $SIG{'HUP'} = $h;
56	    local $SIG{'KILL'} = $h;
57	    local $SIG{'TERM'} = $h;
58	    $dir = permanent_dir($tempbase, "pkginfo");
59	    $dirs->{$dir} = $$;
60	}
61	if (defined $caught) {
62		kill $caught, $$;
63	}
64	return "$dir/";
65}
66
67sub file
68{
69	my $caught;
70	my $h = sub { $caught = shift; };
71	my ($fh, $file);
72
73	{
74	    local $SIG{'INT'} = $h;
75	    local $SIG{'QUIT'} = $h;
76	    local $SIG{'HUP'} = $h;
77	    local $SIG{'KILL'} = $h;
78	    local $SIG{'TERM'} = $h;
79	    ($fh, $file) = permanent_file($tempbase, "pkgout");
80	    $files->{$file} = $$;
81	}
82	if (defined $caught) {
83		kill $caught, $$;
84	}
85	return $file;
86}
87
88sub permanent_file
89{
90	my ($dir, $stem) = @_;
91	my $template = "$stem.XXXXXXXXXX";
92	if (defined $dir) {
93		$template = "$dir/$template";
94	}
95	return OpenBSD::MkTemp::mkstemp($template);
96}
97
98sub permanent_dir
99{
100	my ($dir, $stem) = @_;
101	my $template = "$stem.XXXXXXXXXX";
102	if (defined $dir) {
103		$template = "$dir/$template";
104	}
105	return OpenBSD::MkTemp::mkdtemp($template);
106}
107
1081;
109