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