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