Statistics
| Branch: | Revision:

root / common.sh.in @ a1d43218

History | View | Annotate | Download (5.9 kB)

1
#
2

    
3
# Copyright (C) 2007, 2008, 2009 Google Inc.
4
#
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
9
#
10
# This program is distributed in the hope that it will be useful, but
11
# WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13
# General Public License for more details.
14
#
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
17
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
18
# 02110-1301, USA.
19

    
20
AWK="@AWK@"
21
DUMP="@DUMP@"
22
LOSETUP="@LOSETUP@"
23
KPARTX="@KPARTX@"
24
SFDISK="@SFDISK@"
25
MKDIR_P="@MKDIR_P@"
26

    
27
CLEANUP=( )
28

    
29
log_error() {
30
  echo "$@" >&2
31
}
32

    
33
get_api5_arguments() {
34
  GETOPT_RESULT=$*
35
  # Note the quotes around `$TEMP': they are essential!
36
  eval set -- "$GETOPT_RESULT"
37
  while true; do
38
    case "$1" in
39
      -i|-n) instance=$2; shift 2;;
40

    
41
      -o) old_name=$2; shift 2;;
42

    
43
      -b) blockdev=$2; shift 2;;
44

    
45
      -s) swapdev=$2; shift 2;;
46

    
47
      --) shift; break;;
48

    
49
      *)  log_error "Internal error!" >&2; exit 1;;
50
    esac
51
  done
52
  if [ -z "$instance" -o -z "$blockdev" ]; then
53
    log_error "Missing OS API Argument (-i, -n, or -b)"
54
    exit 1
55
  fi
56
  if [ "$SCRIPT_NAME" != "export" -a -z "$swapdev"  ]; then
57
    log_error "Missing OS API Argument -s (swapdev)"
58
    exit 1
59
  fi
60
  if [ "$SCRIPT_NAME" = "rename" -a -z "$old_name"  ]; then
61
    log_error "Missing OS API Argument -o (old_name)"
62
    exit 1
63
  fi
64
}
65

    
66
get_api10_arguments() {
67
  if [ -z "$INSTANCE_NAME" -o -z "$HYPERVISOR" -o -z "$DISK_COUNT" ]; then
68
    log_error "Missing OS API Variable:"
69
    log_error "(INSTANCE_NAME HYPERVISOR or DISK_COUNT)"
70
    exit 1
71
  fi
72
  instance=$INSTANCE_NAME
73
  if [ $DISK_COUNT -lt 1 -o -z "$DISK_0_PATH" ]; then
74
    log_error "At least one disk is needed"
75
    exit 1
76
  fi
77
  if [ "$SCRIPT_NAME" = "export" ]; then
78
    if [ -z "$EXPORT_DEVICE" ]; then
79
      log_error "Missing OS API Variable EXPORT_DEVICE"
80
    fi
81
    blockdev=$EXPORT_DEVICE
82
  elif [ "$SCRIPT_NAME" = "import" ]; then
83
    if [ -z "$IMPORT_DEVICE" ]; then
84
       log_error "Missing OS API Variable IMPORT_DEVICE"
85
    fi
86
    blockdev=$IMPORT_DEVICE
87
  else
88
    blockdev=$DISK_0_PATH
89
  fi
90
  if [ "$SCRIPT_NAME" = "rename" -a -z "$OLD_INSTANCE_NAME" ]; then
91
    log_error "Missing OS API Variable OLD_INSTANCE_NAME"
92
  fi
93
  old_name=$OLD_INSTANCE_NAME
94
}
95

    
96
format_disk0() {
97
  # Create three partitions:
98
  # 1 - 100MB /boot, bootable
99
  # 2 - Size of Memory, swap
100
  # 3 - Rest
101
  $SFDISK -uM -H 255 -S 63 --quiet --Linux "$1" <<EOF
102
,100,L,*
103
,$INSTANCE_MEMORY,S
104
,,L
105
EOF
106
}
107

    
108
map_disk0() {
109
  blockdev="$1"
110
  filesystem_dev_base=`$KPARTX -l -p- $blockdev | \
111
                       grep -m 1 -- "-1.*$blockdev" | \
112
                       $AWK '{print $1}'`
113
  if [ -z "$filesystem_dev_base" ]; then
114
    log_error "Cannot interpret kpartx output and get partition mapping"
115
    exit 1
116
  fi
117
  $KPARTX -a -p- $blockdev > /dev/null
118
  filesystem_dev="/dev/mapper/${filesystem_dev_base/-1/}"
119
  if [ ! -b "$filesystem_dev" ]; then
120
    log_error "Can't find kpartx mapped partition: $filesystem_dev"
121
    exit 1
122
  fi
123
  echo "$filesystem_dev"
124
}
125

    
126
unmap_disk0() {
127
  $KPARTX -d -p- $1
128
}
129

    
130
cleanup() {
131
  if [ ${#CLEANUP[*]} -gt 0 ]; then
132
    LAST_ELEMENT=$((${#CLEANUP[*]}-1))
133
    REVERSE_INDEXES=$(seq ${LAST_ELEMENT} -1 0)
134
    for i in $REVERSE_INDEXES; do
135
      ${CLEANUP[$i]}
136
    done
137
  fi
138
}
139

    
140
trap cleanup EXIT
141

    
142
DEFAULT_FILE="@sysconfdir@/default/ganeti-instance-debootstrap"
143
if [ -f "$DEFAULT_FILE" ]; then
144
    . "$DEFAULT_FILE"
145
fi
146

    
147
# note: we don't set a default mirror since debian and ubuntu have
148
# different defaults, and it's better to use the default
149

    
150
# only if the user want to specify a mirror in the defaults file we
151
# will use it, this declaration is to make sure the variable is set
152
: ${CDINSTALL:="yes"}
153
: ${IMAGE:=""}
154
: ${ARCH:=""}
155
: ${CUSTOMIZE_DIR:="@sysconfdir@/ganeti/instance-debootstrap.d"}
156
: ${VARIANTS_DIR:="@sysconfdir@/ganeti/instance-debootstrap/variants"}
157
if [ -z "$OS_API_VERSION" -o "$OS_API_VERSION" = "5" ]; then
158
  DEFAULT_PARTITION_STYLE="none"
159
else
160
  DEFAULT_PARTITION_STYLE="msdos"
161
fi
162
: ${PARTITION_STYLE:=$DEFAULT_PARTITION_STYLE} # disk partition style
163

    
164
SCRIPT_NAME=$(basename $0)
165

    
166
if [ -f /sbin/blkid -a -x /sbin/blkid ]; then
167
  VOL_ID="/sbin/blkid -o value -s UUID"
168
  VOL_TYPE="/sbin/blkid -o value -s TYPE"
169
else
170
  for dir in /lib/udev /sbin; do
171
    if [ -f $dir/vol_id -a -x $dir/vol_id ]; then
172
      VOL_ID="$dir/vol_id -u"
173
      VOL_TYPE="$dir/vol_id -t"
174
    fi
175
  done
176
fi
177

    
178
if [ -z "$VOL_ID" ]; then
179
  log_error "vol_id or blkid not found, please install udev or util-linux"
180
  exit 1
181
fi
182

    
183
if [ -z "$OS_API_VERSION" -o "$OS_API_VERSION" = "5" ]; then
184
  OS_API_VERSION=5
185
  GETOPT_RESULT=`getopt -o o:n:i:b:s: -n '$0' -- "$@"`
186
  if [ $? != 0 ] ; then log_error "Terminating..."; exit 1 ; fi
187
  get_api5_arguments $GETOPT_RESULT
188
elif [ "$OS_API_VERSION" = "10" -o "$OS_API_VERSION" = "15" ]; then
189
  get_api10_arguments
190
else
191
  log_error "Unknown OS API VERSION $OS_API_VERSION"
192
  exit 1
193
fi
194

    
195
if [ -n "$OS_VARIANT" ]; then
196
  if [ ! -d "$VARIANTS_DIR" ]; then
197
    log_error "OS Variants directory $VARIANTS_DIR doesn't exist"
198
    exit 1
199
  fi
200
  VARIANT_CONFIG="$VARIANTS_DIR/$OS_VARIANT.conf"
201
  if [ -f "$VARIANT_CONFIG" ]; then
202
    . "$VARIANT_CONFIG"
203
  else
204
    if grep -qxF "$OS_VARIANT" variants.list; then
205
      log_error "ERROR: instance-debootstrap configuration error"
206
      log_error "  Published variant $OS_VARIANT is missing its config file"
207
      log_error "  Please create $VARIANT_CONFIG or unpublish the variant"
208
      log_error "  (by removing $OS_VARIANT from variants.list)"
209
    else
210
      log_error "Unofficial variant $OS_VARIANT is unsupported"
211
      log_error "Most probably this is a user error, forcing a wrong name"
212
      log_error "To support this variant please create file $VARIANT_CONFIG"
213
    fi
214
    exit 1
215
  fi
216
fi
217