Handle config upgrades.
[carnet-upgrade.git] / src / functions.sh
1 # Ivan 'ico' Rako <irako@srce.hr>
2 # Zoran 'jelly' Dzelajlija <jelly@srce.hr>
3
4 notice () {
5   LC_MESSAGES=hr_HR dialog --backtitle "$title" --msgbox "$*" 18 75
6 }
7
8 first_warning () {
9   LC_MESSAGES=hr_HR dialog --backtitle "$title" --yesno "$msg_first_warning" 18 75
10   if [ $? -ne 0 ]; then
11     log "Upgrade canceled."
12     exit 0
13   fi
14 }
15
16 pkg() {
17   local version
18   dpkg -l "$1" 2> /dev/null | egrep -q "^.i" || return 1
19   [ -z "$2" ] && return 0
20   version=$(dpkg -s "$1" | awk '/^Version:/ {print $2}')
21   dpkg --compare-versions "$version" "$2" "$3" 2>&1 > /dev/null
22 }
23
24 log() {
25   local old_umask
26   logfile=${logfile:=/var/log/carnet-upgrade.log}
27
28   old_umask=$(umask)
29   umask 0077
30
31   echo "$(date +'%Y-%m-%d %H:%M:%S') $*" >> $logfile
32   echo "CN: $*"
33
34   umask $old_umask
35   chmod og= $logfile
36 }
37
38 # find first free uid/gid in range
39 # find_id passwd 100 999
40 find_id() {
41   local i db first last ids
42   db=$1
43   first=$2
44   last=$3
45   ids=$(getent $db | awk -F: "\$3 >= $first && \$3 <= $last {print \$3}")
46   for i in $(seq $first $last)
47   do
48     if ! echo $ids | grep -q $i; then
49       echo $i
50       return 0
51     fi
52   done
53   return 1
54 }
55
56 # safe in-place s///
57 check_and_sed() {
58   [ "$DEBIAN_SCRIPT_DEBUG" ] && set -vx
59   local s sedcmd ret i
60   s="$1"
61   shift
62   sedcmd="$1"
63   shift
64   ret=2
65   for i in $*
66   do
67     [ -e "$i" ]        || continue
68     egrep -q "$s" "$i" || continue
69     [ -h "$i" ]        && i=$(readlink -f "$i")
70     sed "$sedcmd" "$i" > "$i.dpkg-tmp"
71     chown --reference "$i" "$i.dpkg-tmp"
72     chmod --reference "$i" "$i.dpkg-tmp"
73     mv "$i.dpkg-tmp" "$i"
74     ret=0
75   done
76   if [ "$ret" -eq 0 ]; then
77     log "Done \"$sedcmd\" to $*"
78   fi
79   return $ret
80 }
81
82 #
83 # Update uid for user from reserved system range (0-99) to dynamic system
84 # range (100-999).  Optionally update ownerships of given directories.
85 # $0 user [directory ...]
86 #
87 check_and_update_ugid() {
88   local user newgid newuid
89   user=$1
90   if [ "$(getent passwd $user | awk -F: '$3 >= 100 {print "ok"; exit 0}')" ]; then
91     return 0
92   fi
93   shift
94   newgid=$(find_id group 100 999)
95   newuid=$(find_id passwd 100 999)
96   # other directories/files
97   chown -R $newuid:$newgid $*
98   groupmod -g $newgid $user
99   usermod -u $newuid -g $newgid $user
100   log "Fixed $user user uid/gid."
101 }
102
103 pkgadd () {
104   [ -n "$*" ] || return 0
105   log "Installing $* ..."
106   apt-get -y$s install $* || {
107     dpkg --configure -a
108     apt-get -yf install
109     apt-get -y$s install $*
110   }
111 }
112
113 pkgrm () {
114   [ -n "$*" ] || return 0
115   log "Purging $* ..."
116   apt-get -y$s --purge remove $* || {
117     dpkg --configure -a
118     apt-get -yf install
119     apt-get -y$s --purge remove $*
120   }
121 }
122
123 pkgrm_only () {
124   [ -n "$*" ] || return 0
125   log "Removing $* ..."
126   apt-get -y$s remove $* || {
127     dpkg --configure -a
128     apt-get -yf install
129     apt-get -y$s remove $*
130   }
131 }
132
133
134 apt_update() {
135   log "Running apt-get update ..."
136   apt-get -y$s update || {
137     apt-get -yf install
138     apt-get -y$s update
139   }
140   apt-get autoclean
141 }
142
143 apt_download() {
144   log "Running apt-get download ..."
145   apt-get -dy$s dist-upgrade
146 }
147
148 dist_upgrade () {
149   log "Running upgrade ..."
150   apt-get -y$s upgrade || {
151     dpkg --configure -a
152     apt-get -y$s upgrade
153   }
154
155   log "Running dist-upgrade ..."
156   apt-get -y$s dist-upgrade || {
157     dpkg --configure -a
158     apt-get -y$s dist-upgrade
159   }
160 }
161
162 remove_x() {
163   LC_MESSAGES=hr_HR dialog --backtitle "$title" --yesno "$msg_remove_x" 18 75
164   if [ $? -eq 0 ]; then
165     log "Removing X Window System"
166     pkgrm x11-common
167   else
168     log "Skipping X Window System removal"
169   fi
170 }
171
172 remove_bloat() {
173   local to_remove add_this i j n text to_show cmd tmpfile installed
174   local update_selections pkgs_to_deselect pkgs_to_remove oldifs pkglist
175   text="$msg_remove_bloat"
176
177   [ "$DEBUG" ] && set -x
178   n=0
179   pkglist=$(mktemp /var/lib/carnet-upgrade/pkglist.XXXXXX)
180   COLUMNS=200 dpkg -l | awk '$1 ~ /^.i/ { print $2 }' > $pkglist
181   for i in $(echo "$bloats" | sed 's/           */      /g'| awk -F'    ' '{print $2}')
182   do
183     if grep -q "^$i$" $pkglist; then
184       installed="$installed $i"
185       add_this="$(echo "$bloats" | egrep "      .*\b${i}\b" | sed 's/           */      /g'| awk -F'    ' '{print $1}')"
186       if ! echo "$to_show" | grep -q "$add_this"; then
187         to_show="$to_show \"$add_this\" \"\" on"
188         n=$(($n+1))
189       fi
190     fi
191   done
192   [ "$DEBUG" ] || rm -f $pkglist
193   [ -z "$to_show" ] && return 0
194   test $n -gt 8 && n=8
195   tmpfile=$(mktemp /var/lib/carnet-upgrade/dialog-tmp.XXXXXX)
196   cmd="LC_MESSAGES=hr_HR dialog --nocancel --backtitle \""$title"\" --checklist \""$text"\" 20 75 $n $to_show"
197   [ "$DEBUG" ] && set +x
198     eval $cmd 2> $tmpfile
199   [ "$DEBUG" ] && set -x
200   to_remove="$(cat $tmpfile | sed 's/\"\ \"/    /g; s/\"\ *//g; s/\\//g')"
201   [ "$DEBUG" ] || rm -f $tmpfile 
202   [ "$DEBUG" ] && echo "DEBUG: to_remove=$to_remove" 1>&2
203   oldifs="$IFS"
204   IFS=" " # tab
205   for i in $(echo "$to_remove")
206   do
207     IFS="$oldifs"
208     for j in $(echo "$bloats" | grep "^$i" | sed 's/            */      /g'| awk -F'    ' '{print $2}')
209     do
210       # hopefully this will tell apt not to reinstall them at upgrade time
211       update_selections="${update_selections}
212         $j      deinstall"
213       pkgs_to_deselect="$pkgs_to_deselect $j"
214       # only remove those that really are installed
215       if echo "$installed" | grep -q "$j"; then
216         pkgs_to_remove="$pkgs_to_remove $j"
217       fi
218     done
219   done
220   [ "$DEBUG" ] && echo "DEBUG: pkgs_to_remove=$pkgs_to_remove" 1>&2
221   # Finally, remove those
222   if [ -n "$update_selections" ]; then
223     echo "$update_selections" | dpkg --set-selections
224     log "Deselected:${pkgs_to_deselect}"
225   fi
226   eval pkgrm $pkgs_to_remove
227 }
228
229 remove_cn_bloat() {
230   local installed pkg cn_pkgs dialog_list selection cn_del to_del del_pkgs
231
232   # get installed packages
233   installed=$(mktemp /var/lib/carnet-upgrade/pkglist.XXXXXX)
234   COLUMNS=200 dpkg -l | awk '$1 ~ /^.i/ { print $2 }' >$installed
235
236   # detect installed CN packagees
237   for pkg in $(echo "$cn_bloats" | cut -d' ' -f1); do
238     if grep -qx $pkg $installed; then
239       cn_pkgs="$cn_pkgs $pkg"
240       dialog_list="$dialog_list $pkg '' off"
241     fi
242   done
243   rm -f $installed
244
245   # report status or finish if no CN packages found
246   if [ "$cn_pkgs" ]; then
247     log "Found CN packages:$cn_pkgs"
248   else
249     log "No CN packages found."
250     return
251   fi
252
253   # user selects the packages to delete
254   selection=$(mktemp /var/lib/carnet-upgrade/selection.XXXXXX)
255   eval LC_MESSAGES=hr_HR dialog --nocancel --backtitle \""$title"\" \
256     --checklist \""$msg_remove_cn_bloat"\" 20 75 6 $dialog_list 2>$selection
257
258   # expand the package list to include the services
259   for pkg in $(cat $selection); do
260     pkg=${pkg%\"}
261     pkg=${pkg#\"}
262     cn_del="$cn_del $pkg"
263     to_del=$(echo "$cn_bloats" | grep ^$pkg)
264     del_pkgs="$del_pkgs $to_del"
265   done
266   rm -f $selection
267
268   # report selection or finish if no service selected
269   if [ "$cn_del" ]; then
270     log "Removing CN services:$cn_del"
271   else
272     log "No CN services selected for removal."
273     return
274   fi
275
276   # delete the cn packages and relevant services
277   pkgrm_only $del_pkgs
278 }
279
280 post_upgrade() {
281   touch "$post_upgrade_queuetab"
282   if ! grep -q "true    $*" "$post_upgrade_queuetab"; then
283     echo "true  $*" >> "$post_upgrade_queuetab"
284   fi
285 }
286
287 run_post_upgrade() {
288   if [ -f "$post_upgrade_queuetab" ]; then
289     run_actions post_upgrade "$post_upgrade_queuetab"
290     mv "$post_upgrade_queuetab" "${post_upgrade_queuetab}.done.$(date '+%Y-%m-%d.%H:%M:%S')"
291   fi      
292 }
293
294 ###
295
296 run_actions() {
297   local line lineno lineno2 f n name action checkmsg check
298   [ "$DEBUG" ] && set -vx
299   name="$1"
300   f="$2"
301   [ -f "$f" ] ||   return 2 # no such file
302   n=$(wc -l "$f" | awk '{print $1}')
303   [ "$n" -lt 1 ] && return 1 # no lines to process
304   log "run_actions ($name) started, processing $n lines from \"$f\"."
305   log "-----"
306   lineno=0
307   lineno2=0
308   while [ "$lineno" -lt "$n" ]
309   do
310     lineno=$(($lineno+1))
311     lineno2=$lineno
312     line=$(head -$lineno "$f" | tail -1)
313     if echo $line | grep -q '^#'; then  # comment
314       continue
315     fi
316     while echo "$line" | egrep -q '\\$' # continuation
317     do
318       lineno2=$(($lineno2+1))
319       line=$(echo -n "$line"| sed 's/\\$//'; head -$lineno2 "$f" | tail -1)
320     done
321     check=$(echo "$line" | sed 's/              */      /g'| awk -F'    ' '{print $1}')
322     action=$(echo "$line" | sed 's/             */      /g'| awk -F'    ' '{print $2}')
323     checkmsg=""
324     [ "$check" != "true" ] && checkmsg="if '$check'"
325     if eval $check; then
326       log "$(printf 'action %02d' $lineno) starting: '${action}' $checkmsg "
327       if ! eval $action; then
328         log "$(printf 'action %02d' $lineno) failed: '${action}' $checkmsg "
329         display_action_failure "$action"
330         exit 1
331       else  
332         log "$(printf 'action %02d' $lineno) finished: '${action}' $checkmsg"
333       fi
334     else
335       log "$(printf 'action %02d' $lineno) skipping: '${action}' $checkmsg"
336     fi
337     [ "$lineno2" -gt "$lineno" ] && lineno=$lineno2
338   done
339   log "run_actions ($1) finished"
340   log "-----"
341 }
342
343 # report action failure
344 display_action_failure () {
345   local action=$1
346
347   notice "
348 Doslo je do greske prilikom izvrsavanja akcije '$action'. Detalje problema moguce je vidjeti u log datotekama:
349
350   /var/log/carnet-upgrade.log
351   /var/log/carnet-upgrade.typescript
352
353 Da bi dovrsili zapocete operacije nad paketima pokrenite:
354
355   # dpkg --configure -a
356   # apt-get -f install
357
358 Nakon toga ponovno pozovite carnet-upgrade skriptu."
359 }
360
361 # funkcija vraca popis paketa koje treba upgradeati
362 what_upgrade () {
363   local IFSOLD name version package
364
365   for package in `echo "$needed_versions" | awk '{print $1}'| grep -v ^$`; do
366     IFSOLD="$IFS"
367     IFS="       " # tab
368     read name version <<-EOPTS
369         $(echo "$needed_versions" | sed 's/             */      /g' | grep ^$package)
370         EOPTS
371     IFS="$IFSOLD"
372     if pkg "$name" lt "$version"; then
373         echo -n " $name"
374     fi
375   done
376 }
377
378 # funkcija za ispis paketa koje treba upgradeati
379 check_upgrade () {
380   local upgrade
381   upgrade=${1:-"`what_upgrade`"}
382
383   if [ "$upgrade" ]; then
384     notice "
385 Prije prelaska na Debian Squeeze potrebno je napraviti nadogradnju ovih paketa:
386   $upgrade
387
388 Pokrenite:
389   # apt-get update
390   # apt-get dist-upgrade
391 ili:
392   # apt-get update
393   # apt-get install${upgrade}
394   
395 Nakon toga ponovno pozovite ovu skriptu."
396     log "carnet-upgrade aborted, upgrade of$upgrade needed."
397     exit 1
398   fi
399 }
400
401 # update sarge urls in sources list
402 fix_sarge_sources () {
403   local apt_sources='/etc/apt/sources.list'
404   local fixed_sources=$(mktemp /var/lib/carnet-upgrade/sources.list.XXXXXX)
405
406   # sarge (oldstable) was archived recently -> sources.list needs update
407   awk -v archive='http://archive.debian.org/debian' \
408       -v security='http://archive.debian.org/debian-security' \
409       '$3 == "sarge" && $2 ~ /[/]debian$/ { $2 = archive }
410        $3 == "sarge/updates" && $2 ~ /security\.debian\.org$/ { $2 = security }
411        { print }' $apt_sources > $fixed_sources
412
413   # did anything get updated
414   if ! cmp $apt_sources $fixed_sources >/dev/null; then
415       log "sources.list fixed to use archive.debian.org"
416
417       # backup current version
418       if [ ! -f "$apt_sources.$backup_ext.sarge" ]; then
419           cp -av $apt_sources "$apt_sources.$backup_ext.sarge"
420       fi
421
422       # install fixed version
423       cp -av $fixed_sources $apt_sources
424
425       # update the database
426       apt_update
427   fi
428
429   # delete temporary
430   rm -f $fixed_sources
431 }
432
433 # try to check if we are using the latest version of carnet-upgrade script
434 check_my_version () {
435   local packages latest_version
436
437   packages=$(mktemp /var/lib/carnet-upgrade/Packages.XXXXXX)
438   if wget -q -O$packages $packages_uri; then
439     latest_version=$(sed -ne \
440         '/Package: carnet-upgrade/,/Description:/{ /Version: /s///p }' \
441         $packages)
442     rm -f $packages
443
444     if [ "$latest_version" ]; then
445       log "Running carnet-upgrade version is $version"
446       log "Latest carnet-upgrade version is $latest_version"
447       if dpkg --compare-versions "$version" lt "$latest_version"; then
448         check_upgrade " carnet-upgrade"
449       else log "Already using the latest carnet-upgrade version."; fi
450     else log "Failed to get latest version, skipping check."; fi
451   else log "Failed to download Packages file, skipping check."; fi
452
453   rm -f $packages
454 }
455
456 # if php is installed make sure it's version 5
457 check_php_version () {
458
459   if pkg php4-cn && ! pkg php5-cn; then
460     # prepare for php upgrade by installing php5 extensions
461     install_php5_extensions
462
463     notice "
464 Prije prelaska na Debian Squeeze potrebno je napraviti nadogradnju sa PHP 4 na PHP 5.
465
466 Pokrenite:
467   # apt-get update
468   # apt-get install php5-cn
469   
470 Nakon sto provjerite da sve web aplikacije rade ispravno, ponovno pozovite ovu skriptu."
471     log "carnet-upgrade aborted, upgrade to php5-cn needed."
472     exit 1
473   fi
474 }
475
476 # install php5 extensions based on installed php4 extensions
477 install_php5_extensions () {
478   local php4_list php5_pkg
479
480   # get a list of installed php4 packages
481   php4_list=$(mktemp /var/lib/carnet-upgrade/php4_list.XXXXXX)
482   dpkg -l php4-\* | awk '/^ii/ {
483     pkg=$2; sub("^php4", "php5", pkg); print pkg }' > $php4_list
484
485   # compare it with the list of available php5 packages
486   php5_pkg=$(apt-cache search -n ^php5- | cut -d' ' -f 1 \
487     | grep -F -x -f $php4_list | grep -F -x -v php5-cn)
488
489   # handle special cases
490   if pkg php4-xslt; then
491      php5_pkg="$php5_pkg php5-xsl"
492   fi
493
494   # install the required php5 modules
495   if [ "$php5_pkg" ]; then
496      pkgadd $php5_pkg
497   fi
498
499   # cleanup
500   rm -f $php4_list
501 }
502
503 # if grsec is installed make sure it is fresh
504 check_grsec_kernel () {
505   local ver=`uname -r`
506
507   log "Kernel version: $ver"
508   if [ "$ver" = "${ver%-grsec}" ]; then
509     # not grsec variant
510     return 0
511   fi
512
513   if [ "$ver" = "${ver#2.6.2[4-9]}" ]; then
514     # kernel too old
515     notice "
516 Prije prelaska na Debian Squeeze potrebno je napraviti nadogradnju na najnoviji kernel.
517
518 Pokrenite:
519   # apt-get update
520   # apt-get install kernel-2.6-cn
521   # reboot
522   
523 Nakon toga ponovno pozovite ovu skriptu."
524     log "carnet-upgrade aborted, kernel upgrade needed."
525     exit 1
526   fi
527 }
528
529 check_reboot () {
530   reboot_required || return 0
531
532   LC_MESSAGES=hr_HR dialog --backtitle "$title" --yesno "$reboot_warning" 18 +75
533   if [ $? -eq 0 ]; then
534     log "Reboot before continuing."
535     exit 0
536   else
537     log "Proceed without reboot."
538   fi
539 }
540
541 # check if the latest kernel is booted
542 reboot_required () {
543   local default_kernel kernel_package running_release running_version
544
545   # try to get default kernel from grub2
546   if [ -z "$default_kernel" -a -r "/boot/grub/grub.cfg" ]; then
547       default_kernel=$( awk '
548            $0 ~ /^set default=".*"/ { 
549                def = gensub(/default="(.*)"/, "\\1", "1", $2);
550                if (def !~ /^[0-9]+$/) { exit }
551            }
552            $1 == "menuentry" { menuentry+=1 }
553            $1 == "linux" && menuentry==def+1 { print $2; exit }' \
554            /boot/grub/grub.cfg 2>/dev/null )
555   fi
556
557   # try to get default kernel from grub-legacy
558   if [ -z "$default_kernel" -a -r "/boot/grub/menu.lst" ]; then
559       default_kernel=$( awk '
560            $1 == "default" { def = $2; 
561                              if (def !~ /^[0-9]+$/) { exit } }
562            $1 == "title" { title+=1 }
563            $1 == "kernel" && title==def+1 { print $2; exit }' \
564            /boot/grub/menu.lst 2>/dev/null )
565   fi
566
567   # try to get default kernel from lilo
568   if [ -z "$default_kernel" ] && type lilo >/dev/null 2>&1; then
569       default_kernel=$( lilo -t -v 2>/dev/null \
570                         | grep -B3 '[*]$' \
571                         | sed -n -e '/^Boot image: / { s///p; q; }' )
572   fi
573
574   if [ -z "$default_kernel" ]; then
575       log "Could not find default kernel"
576       return 1
577   fi
578
579   log "Default kernel image: $default_kernel"
580
581   if [ ! -e "$default_kernel" ]; then
582       # /boot prefix might be missing
583       default_kernel="/boot/${default_kernel#/}"
584       if [ ! -e "$default_kernel" ]; then
585           log "Default kernel image doesn't exist!?"
586           return 1
587       fi
588   fi
589
590   kernel_package=$( dpkg -S $default_kernel | cut -d: -f 1 )
591
592   if [ -z "$kernel_package" ]; then
593       log "Default kernel not installed from a package"
594       return 1
595   fi
596
597   log "Default kernel package: $kernel_package"
598
599   running_release=$( uname -r )
600   running_version=$( uname -v )
601   log "Running kernel: $running_release $running_version"
602
603   if grep -q "$running_release .* $running_version" $default_kernel; then
604       log "Latest kernel version is running." 
605       return 1
606   else
607       log "Reboot to latest version required."
608       return 0
609   fi
610 }
611
612 upgrade () {
613   local packages
614   case "$1" in
615     first)
616       packages="$packages_first"
617     ;;
618     second)
619       packages="$packages_second"
620     ;;
621     *)
622     return 1
623   esac
624
625   pkgupgrade $packages
626 }
627
628 # install packages introduced in this release
629 install_new () {
630   if [ "$new_packages" ]; then
631     pkgadd $new_packages
632   fi
633 }
634
635 # funkcija utrpa ispravan sources.list
636 create_sources_list () {
637     local sl sl_new
638     sl=/etc/apt/sources.list
639     sl_new=/usr/share/carnet-upgrade/files/etc/apt/sources.list
640
641     # check if already installed
642     if ! cmp $sl $sl_new >/dev/null; then
643
644         # backup old version
645         if [ -f $sl -a ! -f $sl.$backup_ext ]; then
646             cp -av $sl $sl.$backup_ext
647         fi
648
649         # install new version
650         cp -av $sl_new $sl
651         notice "Novi sadrzaj datoteke /etc/apt/sources.list:
652
653 `cat /etc/apt/sources.list`"
654     fi
655 }
656
657
658 # funkcija koja brise grupu proc
659 remove_group_proc () {
660   if getent group proc > /dev/null; then
661     groupdel proc || true
662     log "groupdel proc (temporarily)"
663     post_upgrade add_group_proc
664   fi
665 }
666
667 add_group_proc () {
668   if ! getent group proc > /dev/null; then
669     # grsec uses gid 99 for /proc files
670     groupadd -g 99 proc 
671     log "groupadd -g 99 proc"
672
673     # update oidentd so it uses the proc group
674     if [ -x /etc/init.d/oidentd ]; then
675         if gpasswd -a oident proc; then
676             /etc/init.d/oidentd restart
677         fi
678     fi
679   fi
680 }
681
682 # backup ldap database
683 backup_slapd_db () {
684   local ldap_backup=/var/backups/slapcat.$backup_ext.gz
685   local old_umask result
686
687   if [ ! -f $ldap_backup ]; then
688     [ -x /etc/init.d/freeradius ] && /etc/init.d/freeradius stop || true
689
690     if [ -x /etc/init.d/slapd -a -d /var/lib/ldap ]; then
691       /etc/init.d/slapd stop || true
692       old_umask=`umask`
693       umask 0077 # protect the backup file
694       slapcat | gzip --best > $ldap_backup
695       result=$?
696       umask $old_umask
697       /etc/init.d/slapd start || true
698
699       if [ "$result" -eq 0 ]; then
700         log "$ldap_backup created"
701         notice "Napravljen backup slapd baze u $ldap_backup."
702       else
703         log "$ldap_backup was not created"
704         notice "Backup slapd baze u $ldap_backup nije uspio."
705         exit 1
706       fi
707     fi
708
709     [ -x /etc/init.d/freeradius ] && /etc/init.d/freeradius start || true
710   else
711     log "slapd backup already at $ldap_backup, skipping."
712     #notice "Backup slapd baze u $ldap_backup vec postoji!"
713   fi
714 }
715
716 # backup etc direktorija
717 backup_etc_dir () {
718   local etc_backup=/var/backups/etc.$backup_ext.tar.gz
719   local old_umask
720
721   if [ ! -f $etc_backup ]; then
722     old_umask=`umask`
723     umask 0077 # protect the backup file
724
725     if tar cfz $etc_backup /etc 2> /dev/null; then
726       log "$etc_backup created"
727       notice "Napravljen backup /etc direktorija u $etc_backup."
728     else
729       log "$etc_backup was not created"
730       notice "Backup /etc direktorija u $etc_backup nije uspio."
731       exit 1
732     fi
733     umask $old_umask
734   else
735     log "/etc backup already present in $etc_backup, skipping."
736     #notice "Backup /etc direktorija u $etc_backup vec postoji!"
737   fi
738 }
739
740 # fixaj /etc/default/raid2
741 fix_etc_default_raid2 () {
742   if [ -f /etc/default/raid2 ]; then
743     sed "s/^AUTOSTART.*/AUTOSTART=true/g" /etc/default/raid2 > /etc/default/raid2.$backup_ext.$$
744     chown --reference=/etc/default/raid2 /etc/default/raid2.$backup_ext.$$
745     chmod --reference=/etc/default/raid2 /etc/default/raid2.$backup_ext.$$
746
747     if ! cmp -s /etc/default/raid2.$backup_ext.$$ /etc/default/raid2; then
748       log "Fixed /etc/default/raid2 AUTOSTART option."
749       mv -v /etc/default/raid2.$backup_ext.$$ /etc/default/raid2
750     else
751       rm -v /etc/default/raid2.$backup_ext.$$
752     fi
753   fi
754 }
755
756 # restore a configuration file if it contains only CN modifications
757 restore_file () {
758   local file file_expect file_restore file_backup
759
760   file=$1
761   file_expect=$2
762   file_restore=$3
763   file_backup=$file.$backup_ext
764
765   [ "$file_expect" ] || file_expect=$cnup/files/$file.expect
766   [ "$file_restore" ] || file_restore=$cnup/files/$file.restore
767
768   # check if all required files are there
769   if ! [ -e $file -a -e $file_expect -a -e $file_restore ]; then
770     log "Required files missing, skipping $file restore."
771     return 1
772   fi 
773
774   # is restore needed
775   if ! cmp $file_expect $file >/dev/null; then
776     log "Config doesn't match the template, skipping $file restore."
777     return 1
778   fi
779
780   # backup and restore
781   if [ ! -e $file_backup ]; then
782     log "Restoring config file $file."
783
784     # backup local changes
785     cp -av $file $file_backup
786
787     # restore original
788     cp -v $file_restore $file
789
790     # restore succedded
791     return 0
792   else 
793     log "Backup file exists, skipping $file restore."
794     return 1
795   fi
796 }
797
798 # force reconfiguration at the end if package is not upgraded automatically
799 postupgrade_reconfigure () {
800   local cn_package cn_version
801
802   cn_package=$1
803   cn_version=$( dpkg -s "$cn_package" | awk '/^Version:/ {print $2}' )
804   post_upgrade "pkg $cn_package gt $cn_version || dpkg-reconfigure $cn_package"
805 }
806
807 # copy template into a temporary file
808 copy_template () {
809   local path file template config_new
810
811   path=$1
812   file=$(basename $path)
813   template=$cnup/files/$path.template
814   config_new=$(mktemp /var/lib/carnet-upgrade/$file.XXXXXX)
815   cp $template $config_new
816
817   echo $config_new
818 }
819
820 # restore modified config to their package defaults
821 # so the upgrade doesn't complain so much
822 restore_configs () {
823   local hostname domain config_new memtotal memlimit
824
825   # restore simple configs
826   if pkg ntp-cn && pkg ntp lt 1:4.2.6.p2+dfsg-1+b1; then
827      if restore_file /etc/ntp.conf; then
828         postupgrade_reconfigure ntp-cn
829      fi
830   fi
831
832   if pkg kernel-2.6-cn && pkg procps lt 1:3.2.8-9; then
833      if restore_file /etc/sysctl.conf; then
834         postupgrade_reconfigure kernel-2.6-cn
835      fi
836   fi
837
838   if pkg kernel-2.6-cn && pkg libpam-modules lt 1.1.1-6.1; then
839      if restore_file /etc/security/limits.conf; then
840         postupgrade_reconfigure kernel-2.6-cn
841      fi
842   fi
843
844   if pkg vsftpd-cn && pkg vsftpd lt 2.3.2-3+squeeze2; then
845      if restore_file /etc/vsftpd.conf; then
846         postupgrade_reconfigure vsftpd-cn
847      fi
848   fi
849
850   if pkg squirrelmail-cn && pkg squirrelmail lt 2:1.4.21-2; then
851      if restore_file /etc/squirrelmail/apache.conf; then
852         postupgrade_reconfigure squirrelmail-cn
853      fi
854   fi
855
856   if pkg spamassassin-cn && pkg spamassassin lt 3.3.1-1; then
857      if restore_file /etc/spamassassin/v310.pre; then
858         postupgrade_reconfigure spamassassin-cn
859      fi
860   fi
861
862   if pkg amavisd-cn && pkg amavisd-new lt 1:2.6.4-3; then
863      if restore_file /etc/cron.daily/amavisd-new; then
864         rm -f /etc/cron.daily/amavisd-new.$backup_ext
865         postupgrade_reconfigure amavisd-cn
866      fi
867
868      if restore_file /etc/amavis/conf.d/15-av_scanners; then
869         rm -f /etc/amavis/conf.d/15-av_scanners.$backup_ext
870         postupgrade_reconfigure amavisd-cn
871      fi
872   fi
873
874   if pkg amavisd-cn lt 3:2.6.5; then
875      restore_file /etc/init.d/amavisd-cn
876   fi
877
878   if pkg console-tools lt 1:0.2.3dbs-69.1; then
879      restore_file /etc/console-tools/config
880   fi
881
882   if pkg base-files lt 6.0squeeze2; then
883      restore_file /etc/issue
884      restore_file /etc/issue.net
885   fi
886
887   # dovecot won't start with these options
888   if pkg dovecot-cn && pkg dovecot-common lt 1:1.2.15-7; then
889      sed -i 's/^\( *\)\(sieve\(_storage\)\?=.*\)/\1#\2/' \
890          /etc/dovecot/dovecot.conf
891   fi
892
893   # check if monitrc is template based
894   if [ -e /etc/monit/monitrc ]; then # monit is removed at this point
895      # regenerate config from template
896      hostname=$(hostname)
897      domain=$(hostname --domain)
898      config_new=$(copy_template /etc/monit/monitrc)
899      check_and_sed "@localhost" \
900        "s/@localhost/@$hostname.$domain/g" $config_new || true
901      restore_file /etc/monit/monitrc $config_new
902      rm -f $config_new
903   fi
904 }
905
906 # temporary disable listchanges packages to reduce clutter during upgrade
907 apt_listchanges () {
908   local file command=$1
909
910   for file in /etc/apt/apt.conf.d/20listchanges \
911               /etc/apt/apt.conf.d/20listchanges-cn
912   do
913       case $command in
914            disable)
915                if [ -f $file ]; then
916                    dpkg-divert --local --rename --divert $file.$backup_ext \
917                                --add $file || true
918                fi
919                ;;
920
921            enable)
922                if [ -f $file.$backup_ext ]; then
923                    dpkg-divert --remove $file || true
924                fi
925                ;;
926      esac
927   done
928 }
929
930 # make a silent installation of carnet and srce keyrings
931 install_keyrings () {
932   pkgadd carnet-keyring srce-keyring debian-archive-keyring
933   pkgupgrade dpkg apt debconf python-apt dpkg-dev
934
935   dpkg-reconfigure carnet-keyring srce-keyring debian-archive-keyring
936   apt_update
937 }
938
939 # remove skey (not supported anymore)
940 remove_skey () {
941   pkgrm skey-cn libpam-skey
942
943   if getent group skey > /dev/null; then
944     groupdel skey || true
945     log "groupdel skey"
946   fi
947 }
948
949 # remove unsupported php version
950 # but leave the configuration just in case
951 remove_php4 () {
952   if pkg php5-cn && pkg php4-common; then
953      pkgrm_only php4-common
954   fi
955 }
956
957 # check if package is orphaned (nothing depends on it)
958 is_orphaned () {
959   local package deps
960
961   package=$1
962   deps=$(apt-get remove -s $package | grep ^Remv | wc -l)
963   if [ "$deps" -eq 1 ]; then 
964     return 0
965   else 
966     return 1
967   fi
968 }
969
970 # remove old and unused libraries
971 remove_orphaned () {
972   local package
973
974   for package in $orphaned_packages; do
975     if is_orphaned $package; then
976       pkgrm $package
977     fi
978   done
979 }
980
981 # monit it causing problems for postinst scripts
982 # restarting daemons so try to disable it
983 disable_monit () {
984   if pkg monit-cn; then
985     pkgrm monit-cn # disables monit in prerm
986   fi
987
988   # if configured manually
989   if pkg monit; then
990     # remove the binary but leave the config
991     apt-get remove --yes monit
992     # stop the binary
993     pkill -9 monit || true
994   fi
995 }
996
997 # check which of the given packages are installed
998 grep_installed () {
999   local package installed
1000
1001   for package in $*; do
1002     if pkg $package; then
1003       installed="$installed $package"
1004     fi
1005   done
1006
1007   echo $installed
1008 }
1009
1010 # only upgrade packages if they are already installed
1011 pkgupgrade () {
1012   local packages
1013
1014   packages=$( grep_installed $* )
1015
1016   if [ "$packages" ]; then
1017     eval pkgadd $packages
1018   fi
1019 }
1020
1021 # make a silent upgrade to new libc6
1022 upgrade_libc () {
1023   DEBIAN_FRONTEND=noninteractive pkgupgrade libc6 libc6-dev tzdata
1024 }
1025
1026 # upgrade apache2/php5
1027 upgrade_apache2 () {
1028   pkgrm apache-common # prevents installation of apache2-suexec
1029   pkgupgrade libapache2-mod-php5 php5-cli php5-cn apache2-cn \
1030     php5-odbc php5-suhosin php-suhosin-cn
1031 }
1032
1033 upgrade_amavis () {
1034   local conf
1035
1036   # move old config aside
1037   for conf in /etc/amavisd.conf /etc/amavis/amavisd.conf; do
1038     if [ -e $conf -a ! -e $conf.$backup_ext ]; then
1039       mv -v $conf $conf.$backup_ext
1040     fi
1041   done
1042
1043   # install new packages
1044   pkgadd amavisd-cn amavisd-new
1045
1046   # start new packages
1047   /etc/init.d/amavisd-cn restart
1048 }
1049
1050 # handle freerdius upgrade
1051 upgrade_freeradius () {
1052   local template config_new password realm hostname basedn
1053
1054   pkg freeradius-aai lt 2.1.3-0lenny0 || return 0
1055
1056   # aai team prevents a clean upgrade
1057   cp -av /etc/freeradius /etc/freeradius.$backup_ext
1058   pkgrm freeradius-aai freeradius-ldap freeradius
1059   pkgadd freeradius-aai
1060 }
1061
1062 # prepare for slapd upgrade (caused by freeradius installation)
1063 prepare_openldap () {
1064   pkg openldap-aai lt 2.4 || return 0
1065
1066   # openldap-aai expects org.ldif to exists
1067   local ldif=/var/lib/ldap/org.ldif
1068   local ldif_backup=/var/lib/carnet-upgrade/org.ldif
1069   if [ -f $ldif ]; then
1070     cp -v $ldif $ldif_backup
1071   fi
1072 }
1073
1074 # fix openldap-aai postinst user handling
1075 upgrade_openldap () {
1076   pkg openldap-aai lt 2.4 || return 0
1077
1078   # slapd postinst fails if move_old_database is false
1079   echo 'slapd slapd/move_old_database boolean true' | debconf-set-selections 
1080   pkgadd slapd
1081
1082   # slapd removes org.ldif during switch from ldap -> openldap user
1083   local ldif=/var/lib/ldap/org.ldif
1084   local ldif_backup=/var/lib/carnet-upgrade/org.ldif
1085   if [ -f $ldif_backup -a ! -f $ldif ]; then
1086     cp -v $ldif_backup $ldif
1087   fi
1088
1089   # openldap-aai postinst fails here because of database permissions
1090   apt-get -y install openldap-aai || true
1091   chown -R openldap:openldap /var/lib/ldap
1092   /etc/init.d/slapd restart
1093   dpkg --configure -a
1094
1095   # remove ldap user if openldap works
1096   if getent passwd ldap      > /dev/null && \
1097      getent passwd openldap  > /dev/null && \
1098      pgrep -U openldap slapd > /dev/null
1099   then
1100      log "userdel ldap (replaced by openldap)"
1101      userdel ldap || true
1102   fi
1103 }
1104
1105 # fix postgresql deinstallation during upgrade
1106 upgrade_postgresql () {
1107   if check_services show 2>&1 | grep -qw postgresql; then
1108      apt-get install -y postgresql || apt-get -yf install
1109   fi
1110 }
1111
1112 # upgrade the MTA
1113 upgrade_postfix () {
1114   pkgadd sasl2-bin postfix-cn
1115
1116   if pkg postgrey; then
1117      pkgrm postgrey # not used by postfix-cn anymore
1118   fi
1119 }
1120
1121 # upgrade bind separately so DNS is not down for too long
1122 # or breaks postinst scripts of other cn packages that depend on
1123 # working resolver
1124 upgrade_bind() {
1125   pkgadd bind9-cn
1126 }
1127
1128 get_variable () {
1129   local name=$1 file=$2 val
1130
1131   if [ -f "$file" ]; then
1132      val=$(sed -n "/^[[:space:]]*$name[[:space:]]*=/s/[[:space:]]*//gp" $file)
1133      val=${val#*=}
1134   fi
1135
1136   echo $val
1137 }
1138
1139 # restore munin configs for silent upgrade
1140 upgrade_munin () {
1141   local config backup hostname domain
1142
1143   for pkg_name in munin munin-node; do
1144      pkg $pkg_name lt '1.4.5-3' || continue
1145
1146      config=/etc/munin/$pkg_name.conf
1147      backup=$config.$backup_ext
1148      if [ -f $config -a ! -f $backup ]; then
1149         hostname=$(hostname)
1150         domain=$(hostname --domain)
1151
1152         cp -av $config $backup
1153         check_and_sed "$hostname.$domain" \
1154           "s/$hostname.$domain/localhost.localdomain/g" $config
1155      fi
1156      pkgadd $pkg_name
1157   done
1158
1159   # force localhost -> hostname replacement in config files
1160   pkgadd munin-cn
1161   dpkg-reconfigure munin-cn
1162 }
1163
1164 # udev.preinst fails if devfs configs still exist
1165 upgrade_udev () {
1166   local config
1167
1168   for config in /etc/udev/rules.d/devfs.rules \
1169                 /etc/udev/rules.d/compat-full.rules \
1170                 /etc/udev/rules.d/compat.rules
1171   do
1172      if [ -e $config ]; then
1173         rm -v -f $config
1174      fi
1175   done
1176
1177   if [ -f /etc/udev/udev.rules -a ! -f /etc/udev/rules.d/udev.rules ]; then
1178      ln -sfv ../udev.rules /etc/udev/rules.d/
1179   fi
1180
1181   pkgadd udev
1182 }
1183
1184 # fix /etc/dpkg/dpkg.cfg
1185 comment_force_overwrite () {
1186   if [ -f /etc/dpkg/dpkg.cfg ]; then
1187     if check_and_sed '^force-overwrite' \
1188                   's,\(^force-overwrite\),#\1,' \
1189                   /etc/dpkg/dpkg.cfg; then
1190       notice "$msg_comment_force_overwrite"
1191     fi
1192   fi
1193 }
1194
1195 # try not to overwrite user modified issue files
1196 fix_issue () {
1197   if cmp /etc/issue $cnup/files/etc/issue.expect.new >/dev/null; then
1198     cp -av /etc/issue /etc/issue.$backup_ext
1199     cat > /etc/issue <<EOF
1200 Debian GNU/Linux 6.0 (CARNet Debian 6.0) \\n \\l
1201
1202 EOF
1203   fi
1204
1205   if cmp /etc/issue.net $cnup/files/etc/issue.net.expect.new >/dev/null; then
1206     cp -av /etc/issue.net /etc/issue.net.$backup_ext
1207     cat > /etc/issue.net <<EOF
1208 Debian GNU/Linux 6.0 (CARNet Debian 6.0) %h
1209 EOF
1210   fi
1211 }
1212
1213 check_archives_space() {
1214   local available_disk_space archives_size
1215   available_disk_space=$(free_space /var/cache/apt/archives/)
1216   archives_size=$(download_size dist-upgrade)
1217
1218   if [ $? -ne 0 ]; then
1219     log "Failed to get download size, skipping check."
1220     return 0
1221   fi
1222
1223   log "Archives space: ${available_disk_space}MB"
1224   log "Download size: ${archives_size}MB"
1225
1226   if [ $available_disk_space -lt $archives_size ]; then
1227     log "carnet-upgrade aborted: not enough space in /var/cache/apt/archives/ (have ${available_disk_space}MB, need ${archives_size}MB)"
1228     notice "Premalo slobodnog mjesta u /var/cache/apt/archives/ za skidanje paketa (potrebno je barem ${archives_size}MB). Procitajte poglavlje 'Izvanredne situacije' u dokumentaciji."
1229     exit 1
1230   fi
1231 }
1232
1233 check_kernel_space() {
1234   local available_disk_space kernel_size linux_images metapkg pkg ret
1235   available_disk_space=$(free_space /)
1236   linux_images=$(apt-cache show kernel-2.6-cn | grep ^Depends: \
1237     | grep -o 'linux-image[^, ]*')
1238   for metapkg in $linux_images; do
1239     pkg=$(apt-cache show $metapkg | grep ^Depends: \
1240           | grep -o 'linux-image[^, ]*' | head -1)
1241     [ "$pkg" ] || continue
1242
1243     kernel_size=$(installed_size $pkg)
1244     ret=$?
1245     [ $ret -eq 0 ] && break
1246   done
1247
1248   if [ $ret -ne 0 ]; then
1249     log "Failed to get installed size, skipping check."
1250     return 0
1251   fi
1252
1253   log "Free space in /: ${available_disk_space}MB"
1254   log "Kernel package: $pkg"
1255   log "Kernel size: ${kernel_size}MB"
1256
1257   if LC_ALL=C apt-get -s install kernel-2.6-cn \
1258      | grep -q '^kernel-2.6-cn is already the newest version.$'
1259   then
1260      log "Not checking free space in /: kernel-2.6-cn already installed"
1261      return
1262   fi
1263
1264   if [ $available_disk_space -lt $kernel_size ]; then
1265     log "carnet-upgrade aborted: not enough space in / (have ${available_disk_space}MB, need ${kernel_size}MB)"
1266     notice "Premalo mjesta u / za instalaciju kernela (potrebno je barem ${kernel_size}MB). Procitajte poglavlje 'Izvanredne situacije' u dokumentaciji."
1267     exit 1
1268   fi
1269 }
1270
1271 free_space() {
1272   df --portability --block-size=1M "$1" | tail -1 | awk '{print $4}'
1273 }
1274
1275 download_size() {
1276   local size
1277   size=$(LC_ALL=C apt-get --yes --print-uris "$@" | grep '^Need to get ')
1278
1279   if [ $? -ne 0 ]; then
1280     log "Failed to get download size for 'apt-get $@'."
1281     return 1
1282   fi
1283
1284   size=${size#Need to get }
1285   size=${size%%B*}
1286
1287   size_to_mb "${size}B"
1288 }
1289
1290 installed_size() {
1291   local package size
1292   package=$1
1293   size=$(apt-cache show $package | grep '^Installed-Size:')
1294
1295   if [ $? -ne 0 ]; then
1296     log "Failed to get installed size for '$package'."
1297     return 1
1298   fi
1299
1300   size=${size#*: }
1301
1302   size_to_mb "${size}kB"
1303 }
1304
1305 size_to_mb() {
1306   local size
1307   size=$1
1308   if [ -z "$size" ]; then
1309     size=0;
1310   fi
1311
1312   if [ "${size%kB}" != "$size" ]; then
1313     size=$(round ${size%kB})
1314     size=$(($size/1024+1))
1315   elif [ "${size%MB}" != "$size" ]; then
1316     size=$(round ${size%MB})
1317   elif [ "${size%GB}" != "$size" ]; then
1318     size=$(round ${size%GB})
1319     size=$(($size*1024))
1320   elif [ "${size%[0-9]B}" != "$size" ]; then
1321     size=$(round ${size%B})
1322     size=$(($size/1024/1024+1))
1323   else
1324     log "unrecognized size unit: $size"
1325     return 1
1326   fi
1327
1328   echo $size
1329 }
1330
1331 round() {
1332   local num
1333   num=$1
1334
1335   if [ "${num%.*}" != "$num" ]; then
1336     num=${num%.*}
1337     num=$(($num+1))
1338   fi
1339
1340   echo $num
1341 }
1342
1343 # make sure carnet-upgrade doesn't get upgraded until the end
1344 hold_carnet_upgrade() {
1345   echo carnet-upgrade hold | dpkg --set-selections
1346 }
1347
1348 release_carnet_upgrade() {
1349   echo carnet-upgrade install | dpkg --set-selections
1350 }