source: drbl-virt/sbin/functions_drbl_virt @ 190

Last change on this file since 190 was 190, checked in by rock, 14 years ago

Milestone: drbl-virt deployment completion.
Next Milestone: drbl-virt VM management interface.

File size: 9.8 KB
Line 
1#!/bin/bash
2# Program:
3#   DRBL virt module
4# Author:
5#   Jazz, Rock {jazz, rock}@nchc.org.tw
6# Version:
7#    1.0
8# History:
9#   2010/07/20  Rock    First release (1.0)
10
11# [Variable Declation]
12cpu_flag="no"
13Linux_bit=""
14Linux_Distribution=""
15Linux_Version=""
16VMM_select=""
17
18# [Check Root]
19function check_root(){     
20  if [ $USER != "root" ]; then
21    echo -e "Please change root to run it!"
22    exit
23  fi
24} 
25
26function check_root_run(){
27    if [ $USER != "root" ]; then
28    echo -e "Please change root to run it!"
29    sudo su -c ~/"$0" "$@"
30    exit
31fi
32}
33
34
35# [Check CPU support]
36function cpu_check(){
37egrep '(vmx|svm)' --color=always /proc/cpuinfo > /dev/null
38if [ $? == 0 ]; then
39    cpu_flag="yes"
40fi
41}
42
43# [Check System Version]
44function check_systemInfo(){
45Linux_bit=$(uname -m)
46Linux_Distribution=$(lsb_release -i | awk '{print $3}')
47Linux_Version=$(lsb_release -r | awk '{print $2}')
48}
49
50# [Select Hypervisior(Xen/KVM)]
51function VMM_select(){
52echo ""
53if [ $cpu_flag == "yes" ]; then
54    while [ "$VMM_select" != "1" -a "$VMM_select" != "2"  ]
55    do
56       read -p "Which VMM/Hypervisior will be used in DRBL (1)Xen (2)KVM: " VMM_select
57    done
58  if [ $VMM_select == "1"  ]; then
59    echo "Xen" > $Work_Home/etc/hypervisior
60  else
61    echo "KVM" > $Work_Home/etc/hypervisior
62  fi 
63else
64    echo "Xen will be the only Hypervisior in your DRBL"
65    echo -e "Because CPU don't support virtualization, it can't choose KVM to be the Hypervisior"
66    echo ""
67    VMM_select=1
68    echo "Xen" > $Work_Home/etc/hypervisior
69fi 
70}   
71
72# [Check Hypervisior]
73function check_hypervisior(){
74    hypervisior=$(cat $Work_Home/etc/hypervisior)   
75}
76
77
78# [Check Debian Xen package]
79function check_debian_xen(){
80aptitude update
81check_debian_xen_pkg=$(aptitude search xen-linux-system)
82if [ -n "$check_debian_xen_pkg"  ]; then
83    check_debian_xen_pkg="yes"
84else
85    check_debian_xen_pkg="no"
86fi
87}
88
89
90# [Check Xen kernel]
91function check_xen_nu(){
92Kernels=$(ls /boot | grep vmlinuz)
93declare -i Kernels_nu=$(echo $Kernels | wc -w)
94Kernel_choose=""
95
96Xens=$(ls /boot | grep -i ^xen)
97declare -i Xen_nu=$(echo $Xen | wc -w)
98Xen_choose=""
99
100jude="no"
101
102while [ $jude != "yes" -a $jude != "y" ]
103do
104    declare -i i=1
105    declare -i j=1
106    echo ""
107    echo "Choose Xen-enabled Kernel"
108    if [ $Kernels_nu -gt 1  ]; then
109        for Kernel in $Kernels
110        do
111            echo "($i) $Kernel" 
112            i=i+1
113        done
114
115        read -p "Xen Kenrel is (1/2/...): " Kernel_choose
116        Xen_Kernel=$(echo $Kernels | cut -d " " -f${Kernel_choose})
117
118    else
119        Xen_Kernel=$Kernels
120    fi
121
122    if [ $Xen_nu -gt 1  ]; then
123        for Xen in $Xens
124        do
125            echo "($i) $Xen"
126            j=j+1
127        done
128
129        read -p "Xen is (1/2/...): " Xen_choose
130        Xen=$(echo $Xens | cut -d " " -f${Xen_choose})
131    else
132        Xen=$Xens
133    fi
134
135    read -p "Are you sure (yes/no): " jude
136done
137
138#echo "debug: Xen_Kernel=$Xen_Kernel"
139}
140
141
142function debain-lenny_xen_patch(){
143echo xen.independent_wallclock=1 >> /etc/sysctl.conf
144echo loop max_loop=255 >> /etc/modules
145echo xenblktap >> /etc/modules
146ln -s /usr/lib/$/bin/tapdisk /usr/sbin
147}
148
149
150function get_DRBL_eth(){
151eths=$(ls /etc/drbl | grep macadr-eth[0-9] | grep [0-9].txt$ | cut -d "-" -f2 | cut -d "." -f1)
152eths_nu=$(echo $eths | wc -w)
153
154if [ $eths_nu -gt 1 ]; then
155    echo ""
156    echo "There are multiple eth for DRBL environment: "
157    echo "$eths"
158fi
159
160}
161
162function get_Host_IP_range(){
163if [ ! -e /usr/bin/ipcalc ] || [ ! -e /opt/drbl/bin/drbl-get-network ] || [ ! -e /opt/drbl/bin/drbl-get-ipadd ]; then
164    echo "Don't find /usr/bin/ipcalc /opt/drbl/bin/drbl-get-network & /opt/drbl/bin/drbl-get-ipadd"
165    exit 0
166fi
167
168# get eth network X.X.X
169eth_network_f3=$(echo $eth_network | cut -d "." -f 1-3)
170
171IP_nu=$(cat /etc/drbl/IP_HOST_TABLE | grep $eth_network_f3 | wc -l)
172Host_first_IP=$(cat /etc/drbl/IP_HOST_TABLE | grep $eth_network_f3 | awk "NR==1 {print $1}" | awk '{print $1}')
173Host_last_IP=$(cat /etc/drbl/IP_HOST_TABLE | grep $eth_network_f3 | awk "NR==$IP_nu {print $1}"| awk '{print $1}')
174
175}
176
177function get_VM_IP_range_and_prefix_name(){
178
179echo ""
180echo "DRBL client IP range -> $eth: [$Host_first_IP ~ $Host_last_IP]"
181
182Host_last_IP_4=$(echo $Host_last_IP | cut -d "." -f4)
183Host_last_IP_4=$(($Host_last_IP_4+1))
184VM_first_IP=$(echo $Host_last_IP | cut -d "." -f 1-3)
185VM_first_IP="${VM_first_IP}.${Host_last_IP_4}"
186
187echo "The VM IP Range will start from -> [$VM_first_IP]"
188jude="no"
189while [ $jude != "yes" -a $jude != "y" ];
190do
191    echo ""
192    echo "Input VM number/IP range for $eth (ex. 40)."
193
194    jude_2="no"
195    while [ $jude_2 != "yes" -a $jude_2 != "y" ]
196    do
197        jude_2="yes"
198        read -p "(If you have 10 PCs(4-core), suggestion is 30): " VM_IP_range
199        if [ $((${VM_IP_range}+${Host_last_IP_4})) -ge 254 ]; then
200            echo "range can't greater than 254"
201            jude_2="no"
202        fi
203    done
204
205    #read -p "Input Prefix name for VM for $eth (ex. drblvm): " VM_prefix_name
206    VM_prefix_name=$(cat /etc/drbl/drblpush.conf  | grep hostname= | sed 's/hostname=//')
207    read -p "Is it correct? (yes/no): " jude
208done
209
210#echo "VM_first_IP=$VM_first_IP"
211#echo "VM_IP_range=$VM_IP_range"
212}
213
214
215#function get_VM_prefix_name(){
216#jude="no"
217#while [ $jude != "yes" -a $jude != "y" ];
218#do
219#    echo ""
220#    read -p "Input Prefix name for VM for $eth (ex. drblvm): " VM_prefix_name
221#    read -p "Is it correct? (yes/no): " jude
222#done
223#echo "VM_prefix_name=$VM_prefix_name"
224#}
225
226function create_VM_IP_table(){
227#/etc/drbl-virt/etc/IP_VM_eth
228VM_prefix_eth=$(echo $eth |  sed 's/eth//g')
229VM_Host_totoal_nu=$((${Host_last_IP_4}-1+${VM_IP_range}))
230if [ -e /opt/drbl-virt/etc/IP_VM_$eth ]; then
231    mv -f /opt/drbl-virt/etc/IP_VM_$eth /opt/drbl-virt/etc/IP_VM_${eth}.$(date +%Y-%m-%d-%H-%M-%S).drbl-virt_bak
232fi
233declare -i VM_IP_0=$VM_prefix_eth
234declare -i VM_IP_1=""
235declare -i VM_IP_2=""
236declare -i VM_IP_3=""
237
238# 2 bit
239if [ $VM_Host_totoal_nu -lt 100 ]; then
240    for (( i=$Host_last_IP_4 ; i<="$VM_Host_totoal_nu" ; i++ ))
241    do
242        if [ $i -lt 10 ]; then
243            VM_IP_1=0
244            VM_IP_2=$i
245            echo "${eth_network_f3}.$VM_IP_2 ${VM_prefix_name}${VM_IP_0}${VM_IP_1}${VM_IP_2}" >> /opt/drbl-virt/etc/IP_VM_$eth
246        else
247            VM_IP_1=$i
248            echo "${eth_network_f3}.$VM_IP_1 ${VM_prefix_name}${VM_IP_0}${VM_IP_1}" >> /opt/drbl-virt/etc/IP_VM_$eth
249        fi
250    done
251# 3 bit
252elif [ $VM_Host_totoal_nu -ge 100 ] && [ $VM_Host_totoal_nu -lt 254 ]; then
253    for (( i=$Host_last_IP_4 ; i<=$VM_Host_totoal_nu ; i++ ))
254    do
255        if [ $i -lt 10 ]; then
256            VM_IP_1=0
257            VM_IP_2=0
258            VM_IP_3=$i
259            echo "${eth_network_f3}.${VM_IP_3} ${VM_prefix_name}${VM_IP_0}${VM_IP_1}${VM_IP_2}${VM_IP_3}" >> /opt/drbl-virt/etc/IP_VM_$eth
260        elif [ $i -ge 10 ] && [ $i -lt 100 ]; then
261            VM_IP_1=0
262            VM_IP_2=$i
263            echo "${eth_network_f3}.${VM_IP_2} ${VM_prefix_name}${VM_IP_0}${VM_IP_1}${VM_IP_2}" >> /opt/drbl-virt/etc/IP_VM_$eth
264        else
265            VM_IP_1=$i
266            echo "${eth_network_f3}.${VM_IP_1} ${VM_prefix_name}${VM_IP_0}${VM_IP_1}" >> /opt/drbl-virt/etc/IP_VM_$eth
267        fi
268    done
269
270fi
271}
272
273function generate_Xen_MAC_address(){
274#echo "eth $eth"
275#echo "VM_Host_totoal_nu $VM_Host_totoal_nu"
276#echo "Host_last_IP_4 $Host_last_IP_4"
277#echo "VM_prefix_eth $VM_prefix_eth"
278
279# /opt/drbl-virt/etc/macadr-VM-ethX.txt
280# MAC address [00:16:3e:xx:xx:xx] is for Xen use
281declare -i Xen_MAC_1="0"
282declare -i Xen_MAC_2="$VM_prefix_eth"
283declare -i Xen_MAC_3="0"
284declare -i Xen_MAC_4="0"
285Xen_MAC_5=""
286Xen_MAC_6=""
287
288# backup
289if [ -e /opt/drbl-virt/etc/macadr-VM-$eth.txt ]; then
290mv -f /opt/drbl-virt/etc/macadr-VM-$eth.txt /opt/drbl-virt/etc/macadr-VM-$eth.txt.$(date +%Y-%m-%d-%H-%M-%S).drbl-virt_bak
291fi
292
293# generate MAC
294for (( i=1 ; i<"$VM_Host_totoal_nu" ; i++ ))
295do
296    if [ $i -lt 10 ]; then
297        Xen_MAC_5=0
298        Xen_MAC_6=$i
299        echo "00:16:3e:${Xen_MAC_1}${Xen_MAC_2}:${Xen_MAC_3}${Xen_MAC_4}:${Xen_MAC_5}${Xen_MAC_6}" >> /opt/drbl-virt/etc/macadr-VM-$eth.txt
300    elif [ $i -ge 10 ] && [ $i -le 15 ] ; then
301        Xen_MAC_5=0
302        # translate 10(dec) to 16(hex)
303        printf '%X\n' $i > /tmp/drbl-virt_MAC
304        Xen_MAC_6=$(cat /tmp/drbl-virt_MAC)
305        echo "00:16:3e:${Xen_MAC_1}${Xen_MAC_2}:${Xen_MAC_3}${Xen_MAC_4}:${Xen_MAC_5}${Xen_MAC_6}" >> /opt/drbl-virt/etc/macadr-VM-$eth.txt
306    else
307        printf '%X\n' $i > /tmp/drbl-virt_MAC
308        Xen_MAC_5=$(cat /tmp/drbl-virt_MAC)
309        echo "00:16:3e:${Xen_MAC_1}${Xen_MAC_2}:${Xen_MAC_3}${Xen_MAC_4}:${Xen_MAC_5}" >> /opt/drbl-virt/etc/macadr-VM-$eth.txt
310    fi
311done
312}
313
314#function add_VM_dhcpd_conf(){
315## backup dhcpd.conf
316#if [ -e /etc/dhcp3/dhcpd.conf ]; then
317#cp -f /etc/dhcp3/dhcpd.conf /etc/dhcp3/dhcpd.conf.drbl-virt_bak
318#fi
319#subnet_lines=$(cat -n /etc/dhcp3/dhcpd.conf | grep $eth_network | awk '{print $1}')
320#host_lines=$(cat -n /etc/dhcp3/dhcpd.conf | grep 'host ' | grep '{' | awk '{print $1}')
321#
322#for subnet_line in $subnet_lines
323#do
324#    host_lines cat -n /etc/dhcp3/dhcpd.conf | sed -n "$subnet_line,\$p" | grep 'host .* {' | awk '{print $1}'
325#done
326#}
327
328function delete_duplicating_MAC(){
329drbl_mac_file="/etc/drbl/macadr-$eth.txt"
330# if macadr-ethX.txt.drbl-virt_bak already have, backup it
331# else cp default backup
332if [ -e $drbl_mac_file.drbl-virt_bak ]; then
333    cp $drbl_mac_file $drbl_mac_file.$(date +%Y-%m-%d-%H-%M-%S).drbl-virt_bak
334    cp -f $drbl_mac_file.drbl-virt_bak $drbl_mac_file
335else 
336    cp $drbl_mac_file $drbl_mac_file.drbl-virt_bak
337fi
338
339    # Delete duplating VM MAC
340    for vm_mac in $(echo $VM_MACs)
341    do
342        vm_mac_lines=$(cat -n $drbl_mac_file | grep $vm_mac | awk '{print $1}' )
343        # no duplicated vlaue, doesn't do it
344        if [ -n "$vm_mac_lines" ]; then
345            vm_mac_count=$(echo $vm_mac_lines | wc -w)
346            for (( i=1 ; i<=${vm_mac_count} ; i++ ))
347            do
348                del_line=$(echo $vm_mac_lines | cut -d " " -f${i})
349                sed -i "${del_line}d" $drbl_mac_file
350            done
351        fi
352    done
353} 
354
Note: See TracBrowser for help on using the repository browser.