~youscribe/parted/3.1

« back to all changes in this revision

Viewing changes to build-aux/mktempd

  • Committer: Guilhem Lettron
  • Date: 2012-10-22 14:37:59 UTC
  • Revision ID: guilhem+ubuntu@lettron.fr-20121022143759-m403kecgz13sknvp
3.1 from tarball

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#!/bin/sh
 
2
# Create a temporary directory, much like mktemp -d does.
 
3
 
 
4
# Copyright (C) 2007-2012 Free Software Foundation, Inc.
 
5
 
 
6
# This program is free software: you can redistribute it and/or modify
 
7
# it under the terms of the GNU General Public License as published by
 
8
# the Free Software Foundation, either version 3 of the License, or
 
9
# (at your option) any later version.
 
10
 
 
11
# This program is distributed in the hope that it will be useful,
 
12
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
13
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
14
# GNU General Public License for more details.
 
15
 
 
16
# You should have received a copy of the GNU General Public License
 
17
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
18
 
 
19
# Written by Jim Meyering.
 
20
 
 
21
# Usage: mktempd /tmp phoey.XXXXXXXXXX
 
22
 
 
23
# First, try to use the mktemp program.
 
24
# Failing that, we'll roll our own mktemp-like function:
 
25
#  - try to get random bytes from /dev/urandom
 
26
#  - failing that, generate output from a combination of quickly-varying
 
27
#      sources and gzip.  Ignore non-varying gzip header, and extract
 
28
#      "random" bits from there.
 
29
#  - given those bits, map to file-name bytes using tr, and try to create
 
30
#      the desired directory.
 
31
#  - make only $MAX_TRIES attempts
 
32
 
 
33
ME=`basename "$0"`
 
34
die() { echo >&2 "$ME: $@"; exit 1; }
 
35
 
 
36
MAX_TRIES=4
 
37
 
 
38
rand_bytes()
 
39
{
 
40
  n=$1
 
41
 
 
42
  chars=abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789
 
43
 
 
44
  dev_rand=/dev/urandom
 
45
  if test -r "$dev_rand"; then
 
46
    # Note: 256-length($chars) == 194; 3 copies of $chars is 186 + 8 = 194.
 
47
    dd ibs=$n count=1 if="$dev_rand" 2>/dev/null \
 
48
      | tr -c $chars 01234567$chars$chars$chars
 
49
    return
 
50
  fi
 
51
 
 
52
  cmds='date; date +%N; free; who -a; w; ps auxww; ps ef; netstat -n'
 
53
  data=` (eval "$cmds") 2>&1 | gzip `
 
54
 
 
55
  n_plus_50=`expr $n + 50`
 
56
 
 
57
  # Ensure that $data has length at least 50+$n
 
58
  while :; do
 
59
    len=`echo "$data"|wc -c`
 
60
    test $n_plus_50 -le $len && break;
 
61
    data=` (echo "$data"; eval "$cmds") 2>&1 | gzip `
 
62
  done
 
63
 
 
64
  echo "$data" \
 
65
    | dd bs=1 skip=50 count=$n 2>/dev/null \
 
66
    | tr -c $chars 01234567$chars$chars$chars
 
67
}
 
68
 
 
69
mktempd()
 
70
{
 
71
  case $# in
 
72
  2);;
 
73
  *) die "Usage: $ME DIR TEMPLATE";;
 
74
  esac
 
75
 
 
76
  destdir=$1
 
77
  template=$2
 
78
 
 
79
  # Disallow any trailing slash on specified destdir:
 
80
  # it would subvert the post-mktemp "case"-based destdir test.
 
81
  case $destdir in
 
82
  /) ;;
 
83
  */) die "invalid destination dir: remove trailing slash(es)";;
 
84
  esac
 
85
 
 
86
  case $template in
 
87
  *XXXX) ;;
 
88
  *) die "invalid template: $template (must have a suffix of at least 4 X's)";;
 
89
  esac
 
90
 
 
91
  fail=0
 
92
 
 
93
  # First, try to use mktemp.
 
94
  d=`env -u TMPDIR mktemp -d -t -p "$destdir" "$template" 2>/dev/null` \
 
95
    || fail=1
 
96
 
 
97
  # The resulting name must be in the specified directory.
 
98
  case $d in "$destdir"*);; *) fail=1;; esac
 
99
 
 
100
  # It must have created the directory.
 
101
  test -d "$d" || fail=1
 
102
 
 
103
  # It must have 0700 permissions.  Handle sticky "S" bits.
 
104
  perms=`ls -dgo "$d" 2>/dev/null|tr S -` || fail=1
 
105
  case $perms in drwx------*) ;; *) fail=1;; esac
 
106
 
 
107
  test $fail = 0 && {
 
108
    echo "$d"
 
109
    return
 
110
  }
 
111
 
 
112
  # If we reach this point, we'll have to create a directory manually.
 
113
 
 
114
  # Get a copy of the template without its suffix of X's.
 
115
  base_template=`echo "$template"|sed 's/XX*$//'`
 
116
 
 
117
  # Calculate how many X's we've just removed.
 
118
  template_length=`echo "$template" | wc -c`
 
119
  nx=`echo "$base_template" | wc -c`
 
120
  nx=`expr $template_length - $nx`
 
121
 
 
122
  err=
 
123
  i=1
 
124
  while :; do
 
125
    X=`rand_bytes $nx`
 
126
    candidate_dir="$destdir/$base_template$X"
 
127
    err=`mkdir -m 0700 "$candidate_dir" 2>&1` \
 
128
      && { echo "$candidate_dir"; return; }
 
129
    test $MAX_TRIES -le $i && break;
 
130
    i=`expr $i + 1`
 
131
  done
 
132
  die "$err"
 
133
}
 
134
 
 
135
mktempd "$@"