summaryrefslogtreecommitdiffstats
path: root/remote/modules/partitioner/partitioner.sh
blob: 19d594a1294bd69ee4b1cdfeeededbe93d399e7c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
#!/bin/sh

PARTITIONSPATH="/proc/partitions"
CONFIGPATH="./config"
DEFPARTSIZE=2 #in GB
DEFSCRATCHID=44
DEFHOMEID=45
DEFDNBD3ID=46
DEFBOOTID=47

# BLOCKSIZE=$(sfdisk -l /dev/$CHOSENDISK | grep -o -e "blocks of [0-9]* bytes" | cut -d ' ' -f3)
BLOCKSIZE=1024

. $CONFIGPATH
# testing if the sizes and ID's of the disk is defined in the config file
test $SCRATCHID
if [ $? -eq 1 ]; then
  SCRATCHID=$DEFSCRATCHID
fi
test $SCRATCHSIZE
if [ $? -eq 1 ]; then
  SCRATCHSIZEGB=$DEFPARTSIZE
  SCRATCHSIZE=$(($DEFPARTSIZE*1024*1024*1024))
else
  SCRATCHSIZEGB=$SCRATCHSIZE
  SCRATCHSIZE=$(($SCRATCHSIZE*1024*1024*1024))
fi

test $HOMEID
if [ $? -eq 1 ]; then
  HOMEID=$DEFHOMEID
fi
test $HOMESIZE
if [ $? -eq 1 ]; then
  HOMESIZEGB=$DEFPARTSIZE
  HOMESIZE=$(($DEFPARTSIZE*1024*1024*1024))
else
  HOMESIZEGB=$HOMESIZE
  HOMESIZE=$(($HOMESIZE*1024*1024*1024))
fi

test $DNBD3ID
if [ $? -eq 1 ]; then
  DNBD3ID=$DEFDNBD3ID
fi
test $DNBD3SIZE
if [ $? -eq 1 ]; then
  DNBD3SIZEGB=$DEFPARTSIZE
  DNBD3SIZE=$(($DEFPARTSIZE*1024*1024*1024))
else
  DNBD3SIZEGB=$DNBD3SIZE
  DNBD3SIZE=$(($DNBD3SIZE*1024*1024*1024))
fi

test $BOOTID
if [ $? -eq 1 ]; then
  BOOTID=$DEFBOOTID
fi
test $BOOTSIZE
if [ $? -eq 1 ]; then
  BOOTSIZEGB=$DEFPARTSIZE
  BOOTSIZE=$(($DEFPARTSIZE*1024*1024*1024))
else
  BOOTSIZEGB=$BOOTSIZE
  BOOTSIZE=$(($BOOTSIZE*1024*1024*1024))
fi

# picking disk that will be used
DISKS=$(cat $PARTITIONSPATH | tr -s ' ' | cut -d ' ' -f5 | grep -e "[a-z]$")
test $DISKS 2> /dev/null
if [ $? -eq 1 ]; then
  dialog --msgbox "ERROR: Can't find an hard disk." 5 40
  return 1
fi
test $CHOSENDISK
if [ $? -eq 1 ]; then
  DIALOGSTR="This computer has the following partitions:\n"
  for disk in $DISKS; do
    DISKSIZE=$(($(cat $PARTITIONSPATH | grep -e $disk$ | tr -s ' ' | cut -d ' ' -f4)*$BLOCKSIZE))
    PARTS=$(cat $PARTITIONSPATH | grep -e $disk[0-9] | tr -s ' ' | cut -d ' ' -f5)
    GBSIZE=$(echo "scale=2; $DISKSIZE/1024/1024/1024" | bc -l)
    DIALOGSTR=$DIALOGSTR"$disk  $DISKSIZE Bytes ($GBSIZE GB)\n"
    USED=0
    for PART in $PARTS; do
      PARTSIZE=$(($(cat $PARTITIONSPATH | grep -e $PART$ | tr -s ' ' | cut -d ' ' -f4)*1024))
      USED=$(($USED+$PARTSIZE))
      GBSIZE=$(echo "scale=2; $PARTSIZE/1024/1024/1024" | bc -l)
      DIALOGSTR=$DIALOGSTR"   $PART   $PARTSIZE ($GBSIZE GB)\n"
    done
    DIALOGSTR=$DIALOGSTR"   ----------\n"
    FREESPACE=$(($DISKSIZE-$USED))
    GBSIZE=$(echo "scale=2; $USED/1024/1024/1024" | bc -l)
    DIALOGSTR=$DIALOGSTR"   Used  $USED ($GBSIZE GB)\n"
    GBSIZE=$(echo "scale=2; $FREESPACE/1024/1024/1024" | bc -l)
    DIALOGSTR=$DIALOGSTR"   Free  $FREESPACE ($GBSIZE GB)\n\n"
  done

  if [ $(echo $DISKS | tr ' ' \\n | wc -l) -gt 1 ]; then
    DIALOGSTR=$DIALOGSTR"Which of these disks you want to use?\n"
    CHOSENDISK=$(dialog --no-collapse --cr-wrap --inputbox "$DIALOGSTR" 0 0 $(echo $DISKS | cut -d ' ' -f1) 3>&1 1>&2 2>&3)
    if [ $? -eq 1 ]; then
      return 1;
    fi
    echo $DISKS | tr ' ' \\n | grep -e ^$CHOSENDISK$ > /dev/null 2>/dev/null
    while [ $? -ne 0 ]; do
      dialog --msgbox "Invalid option.\nTry again." 0 0
      CHOSENDISK=$(dialog --no-collapse --cr-wrap --inputbox "$DIALOGSTR" 0 0 $(echo $DISKS | cut -d ' ' -f1) 3>&1 1>&2 2>&3)
      if [ $? -eq 1 ]; then
        return 1;
      fi
      echo $DISKS | tr ' ' \\n | grep -e ^$CHOSENDISK$ > /dev/null 2>/dev/null
    done
    # ask a confirmation (all your data will be lost etc)
  elif [ $(echo $DISKS | tr ' ' \\n | wc -l) -eq 1 ]; then
    CHOSENDISK=$(echo $DISKS | tr ' ' \\n | head -n1)
  else
    dialog --msgbox "ERROR: Can't find an hard disk." 5 40
    return 1
  fi
fi

DISKSIZE=$(($(cat $PARTITIONSPATH | grep -e $CHOSENDISK$ | tr -s ' ' | cut -d ' ' -f4)*$BLOCKSIZE))
echo "$DISKSIZE -lt $(($SCRATCHSIZE+$HOMESIZE+$DNBD3SIZE+$BOOTSIZE+4*1024*1024))"
if [ $DISKSIZE -lt $(($SCRATCHSIZE+$HOMESIZE+$DNBD3SIZE+$BOOTSIZE+4*1024*1024)) ]; then
  dialog --msgbox "ERROR: Insufficient space in disk /dev/$CHOSENDISK." 6 40
  return 1
fi

# choosing the partition type (MSDOS or GPT) if it wasn't set in config file
test $PARTITIONTYPE
if [ $? -eq 1 ]; then
  ANS=$(dialog --no-tags --menu "Choose the partitions type" 0 0 0 "1" " MSDOS " "2" " GPT " 3>&1 1>&2 2>&3)
  if [ $? -eq 1 ]; then
    return 1
  fi
  if [ $ANS -eq 1 ]; then
    PARTITIONTYPE=MSDOS
  else
    PARTITIONTYPE=GPT
  fi
fi

# asking confirmation to create the partitions
GBSIZE=$(echo "scale=2; $DISKSIZE/1024/1024/1024" | bc -l)
DIALOGSTR="$CHOSENDISK  $DISKSIZE Bytes ($GBSIZE GB)\n"
PARTS=$(cat $PARTITIONSPATH | grep -e $CHOSENDISK[0-9] | tr -s ' ' | cut -d ' ' -f5)
USED=0
for PART in $PARTS; do
  PARTSIZE=$(($(cat $PARTITIONSPATH | grep -e $PART$ | tr -s ' ' | cut -d ' ' -f4)*$BLOCKSIZE))
  USED=$(($USED+$PARTSIZE))
  GBSIZE=$(echo "scale=2; $PARTSIZE/1024/1024/1024" | bc -l)
  DIALOGSTR=$DIALOGSTR"   $PART   $PARTSIZE ($GBSIZE GB)\n"
done
DIALOGSTR=$DIALOGSTR"   ----------\n"
FREESPACE=$(($DISKSIZE-$USED))
GBSIZE=$(echo "scale=2; $USED/1024/1024/1024" | bc -l)
DIALOGSTR=$DIALOGSTR"   Used  $USED ($GBSIZE GB)\n"
GBSIZE=$(echo "scale=2; $FREESPACE/1024/1024/1024" | bc -l)
DIALOGSTR=$DIALOGSTR"   Free  $FREESPACE ($GBSIZE GB)\n\n"
DIALOGSTR=$DIALOGSTR"All /dev/$CHOSENDISK partitions will be deleted, ok?"

dialog --defaultno --yesno "$DIALOGSTR" 0 0
if [ $? -eq 1 ]; then
  return 1
fi

# pick the size of a sector (usually 512 bytes)
# SECTORSIZE=$(sfdisk /dev/$CHOSENDISK -l -u S 2> /dev/null | grep Units | cut -d ' ' -f5)
SECTORSIZE=512

# delete partition table
echo "apaga"
sgdisk -Z /dev/$CHOSENDISK > /dev/null 2> /dev/null

# constructing the sfdisk input file
START1=$((1*1024*1024/$SECTORSIZE))
SIZE1=$(($SCRATCHSIZE/$SECTORSIZE))

START2=$(($START1+$SIZE1))
SIZE2=$(($HOMESIZE/$SECTORSIZE))

START3=$(($START2+$SIZE2))
SIZE3=$(($DNBD3SIZE/$SECTORSIZE))

START4=$(($START3+$SIZE3))
SIZE4=$(($BOOTSIZE/$SECTORSIZE))

echo "cria sfdisk"
echo "unit: sectors

/dev/"$CHOSENDISK"1 : start= "$START1", size= "$SIZE1", Id= "$SCRATCHID"
/dev/"$CHOSENDISK"2 : start= "$START2", size= "$SIZE2", Id= "$HOMEID"
/dev/"$CHOSENDISK"3 : start= "$START3", size= "$SIZE3", Id= "$DNBD3ID"
/dev/"$CHOSENDISK"4 : start= "$START4", size= "$SIZE4", Id= "$BOOTID," bootable
 "> /tmp/emptydisk.tmp
sfdisk -q -f /dev/$CHOSENDISK < /tmp/emptydisk.tmp > /dev/null
rm -f /tmp/emptydisk.tmp

if [ $PARTITIONTYPE = GPT ]; then
  echo "converte e aplica GUID"
  sgdisk -g /dev/$CHOSENDISK\
  -t 1:"$SCRATCHID"000000-0000-0000-0000-000000000000\
  -t 2:"$HOMEID"000000-0000-0000-0000-000000000000\
  -t 3:"$DNBD3ID"000000-0000-0000-0000-000000000000\
  -t 4:"$BOOTID"000000-0000-0000-0000-000000000000 > /dev/null

  DIALOGSTR="New partitions created:\n\n"
  DIALOGSTR=$DIALOGSTR"SCRATCH ("/dev/$CHOSENDISK"1)\n"
  DIALOGSTR=$DIALOGSTR"   GUID: "$SCRATCHID"000000-0000-0000-0000-000000000000\n"
  DIALOGSTR=$DIALOGSTR"   Size: $SCRATCHSIZE Bytes ($SCRATCHSIZEGB GB)\n"
  DIALOGSTR=$DIALOGSTR"HOME ("/dev/$CHOSENDISK"2)\n"
  DIALOGSTR=$DIALOGSTR"   GUID: "$HOMEID"000000-0000-0000-0000-000000000000\n"
  DIALOGSTR=$DIALOGSTR"   Size: $HOMESIZE Bytes ($HOMESIZEGB GB)\n"
  DIALOGSTR=$DIALOGSTR"DNBD3 ("/dev/$CHOSENDISK"3)\n"
  DIALOGSTR=$DIALOGSTR"   GUID: "$DNBD3ID"000000-0000-0000-0000-000000000000\n"
  DIALOGSTR=$DIALOGSTR"   Size: $DNBD3SIZE Bytes ($DNBD3SIZEGB GB)\n"
  DIALOGSTR=$DIALOGSTR"BOOT ("/dev/$CHOSENDISK"4)\n"
  DIALOGSTR=$DIALOGSTR"   GUID: "$BOOTID"000000-0000-0000-0000-000000000000\n"
  DIALOGSTR=$DIALOGSTR"   Size: $BOOTSIZE Bytes ($BOOTSIZEGB GB)\n"
else
  DIALOGSTR="New partitions created:\n\n"
  DIALOGSTR=$DIALOGSTR"SCRATCH ("/dev/$CHOSENDISK"1)\n"
  DIALOGSTR=$DIALOGSTR"   ID: $SCRATCHID\n"
  DIALOGSTR=$DIALOGSTR"   Size: $SCRATCHSIZE Bytes ($SCRATCHSIZEGB GB)\n"
  DIALOGSTR=$DIALOGSTR"HOME ("/dev/$CHOSENDISK"2)\n"
  DIALOGSTR=$DIALOGSTR"   ID: $HOMEID\n"
  DIALOGSTR=$DIALOGSTR"   Size: $HOMESIZE Bytes ($HOMESIZEGB GB)\n"
  DIALOGSTR=$DIALOGSTR"DNBD3 ("/dev/$CHOSENDISK"3)\n"
  DIALOGSTR=$DIALOGSTR"   ID: $DNBD3ID\n"
  DIALOGSTR=$DIALOGSTR"   Size: $DNBD3SIZE Bytes ($DNBD3SIZEGB GB)\n"
  DIALOGSTR=$DIALOGSTR"BOOT ("/dev/$CHOSENDISK"4)\n"
  DIALOGSTR=$DIALOGSTR"   ID: $BOOTID\n"
  DIALOGSTR=$DIALOGSTR"   Size: $BOOTSIZE Bytes ($BOOTSIZEGB GB)\n"
fi

dialog --msgbox "$DIALOGSTR" 0 0