summaryrefslogtreecommitdiffstats
path: root/remote/modules/partitioner/partitioner.sh
blob: ad4becc3e8f62f97bbaca21c253447f0a7244b02 (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
#!/bin/sh

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

. $CONFIGPATH
# picking disk that will be used
test $CHOSENDISK
if [ $? -eq 1 ]; then
  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
  DIALOGSTR="This computer has the following partitions:\n"
  for disk in $DISKS; do
    DISKSIZE=$(cat $PARTITIONSPATH | grep -e $disk$ | tr -s ' ' | cut -d ' ' -f4)
    PARTS=$(cat $PARTITIONSPATH | grep -e $disk[0-9] | tr -s ' ' | cut -d ' ' -f5)
    GBSIZE=$(echo "scale=2; $DISKSIZE/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)
      USED=$(($USED+$PARTSIZE))
      GBSIZE=$(echo "scale=2; $PARTSIZE/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" | bc -l)
    DIALOGSTR=$DIALOGSTR"   Used  $USED ($GBSIZE GB)\n"
    GBSIZE=$(echo "scale=2; $FREESPACE/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

# asking confirmation to create the partitions
DISKSIZE=$(cat $PARTITIONSPATH | grep -e $CHOSENDISK$ | tr -s ' ' | cut -d ' ' -f4)
GBSIZE=$(echo "scale=2; $DISKSIZE/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)
  USED=$(($USED+$PARTSIZE))
  GBSIZE=$(echo "scale=2; $PARTSIZE/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" | bc -l)
DIALOGSTR=$DIALOGSTR"   Used  $USED ($GBSIZE GB)\n"
GBSIZE=$(echo "scale=2; $FREESPACE/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)

# 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

# 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 "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 /dev/$CHOSENDISK < /tmp/emptydisk.tmp > /dev/null
rm -f /tmp/emptydisk.tmp

# constructing dialog message
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"

dialog --msgbox "$DIALOGSTR" 0 0