Run apt-get update after fixing sources.list.
[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 "Removing $* ..."
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 apt_update() {
117   log "Running apt-get update ..."
118   apt-get -y$s update || {
119     apt-get -yf install
120     apt-get -y$s update
121   }
122   apt-get autoclean
123 }
124
125 apt_download() {
126   log "Running apt-get download ..."
127   apt-get -dy$s dist-upgrade
128 }
129
130 dist_upgrade () {
131   log "Running dist-upgrade ..."
132   apt-get -y$s dist-upgrade || {
133     dpkg --configure -a
134     apt-get -y$s dist-upgrade
135   }
136 }
137
138 remove_x() {
139   LC_MESSAGES=hr_HR dialog --backtitle "$title" --yesno "$msg_remove_x" 18 75
140   if [ $? -eq 0 ]; then
141     log "Removing X Window System"
142     pkgrm xlibs
143   else
144     log "Skipping X Window System removal"
145   fi
146 }
147
148 remove_bloat() {
149   local to_remove add_this i j n text to_show cmd tmpfile installed
150   local update_selections pkgs_to_deselect pkgs_to_remove oldifs pkglist
151   text="$msg_remove_bloat"
152
153   [ "$DEBUG" ] && set -x
154   n=0
155   pkglist=$(mktemp /var/lib/carnet-upgrade/pkglist.XXXXXX)
156   COLUMNS=200 dpkg -l | awk '$1 ~ /^.i/ { print $2 }' > $pkglist
157   for i in $(echo "$bloats" | sed 's/           */      /g'| awk -F'    ' '{print $2}')
158   do
159     if grep -q "^$i$" $pkglist; then
160       installed="$installed $i"
161       add_this="$(echo "$bloats" | egrep "      .*\b${i}\b" | sed 's/           */      /g'| awk -F'    ' '{print $1}')"
162       if ! echo "$to_show" | grep -q "$add_this"; then
163         to_show="$to_show \"$add_this\" \"\" on"
164         n=$(($n+1))
165       fi
166     fi
167   done
168   [ "$DEBUG" ] || rm -f $pkglist
169   [ -z "$to_show" ] && return 0
170   test $n -gt 8 && n=8
171   tmpfile=$(mktemp /var/lib/carnet-upgrade/dialog-tmp.XXXXXX)
172   cmd="LC_MESSAGES=hr_HR dialog --nocancel --backtitle \""$title"\" --checklist \""$text"\" 20 75 $n $to_show"
173   [ "$DEBUG" ] && set +x
174     eval $cmd 2> $tmpfile
175   [ "$DEBUG" ] && set -x
176   to_remove="$(cat $tmpfile | sed 's/\"\ \"/    /g; s/\"\ *//g; s/\\//g')"
177   [ "$DEBUG" ] || rm -f $tmpfile 
178   [ "$DEBUG" ] && echo "DEBUG: to_remove=$to_remove" 1>&2
179   oldifs="$IFS"
180   IFS=" " # tab
181   for i in $(echo "$to_remove")
182   do
183     IFS="$oldifs"
184     for j in $(echo "$bloats" | grep "^$i" | sed 's/            */      /g'| awk -F'    ' '{print $2}')
185     do
186       # hopefully this will tell apt not to reinstall them at upgrade time
187       update_selections="${update_selections}
188         $j      deinstall"
189       pkgs_to_deselect="$pkgs_to_deselect $j"
190       # only remove those that really are installed
191       if echo "$installed" | grep -q "$j"; then
192         pkgs_to_remove="$pkgs_to_remove $j"
193       fi
194     done
195   done
196   [ "$DEBUG" ] && echo "DEBUG: pkgs_to_remove=$pkgs_to_remove" 1>&2
197   # Finally, remove those
198   if [ -n "$update_selections" ]; then
199     echo "$update_selections" | dpkg --set-selections
200     log "Deselected:${pkgs_to_deselect}"
201   fi
202   eval pkgrm $pkgs_to_remove
203 }
204
205 post_upgrade() {
206   touch "$post_upgrade_queuetab"
207   if ! grep -q "true    $*" "$post_upgrade_queuetab"; then
208     echo "true  $*" >> "$post_upgrade_queuetab"
209   fi
210 }
211
212 run_post_upgrade() {
213   if [ -f "$post_upgrade_queuetab" ]; then
214     run_actions post_upgrade "$post_upgrade_queuetab"
215     mv "$post_upgrade_queuetab" "${post_upgrade_queuetab}.done.$(date '+%Y-%m-%d.%H:%M:%S')"
216   fi      
217 }
218
219 ###
220
221 run_actions() {
222   local line lineno lineno2 f n name action checkmsg check
223   [ "$DEBUG" ] && set -vx
224   name="$1"
225   f="$2"
226   [ -f "$f" ] ||   return 2 # no such file
227   n=$(wc -l "$f" | awk '{print $1}')
228   [ "$n" -lt 1 ] && return 1 # no lines to process
229   log "run_actions ($name) started, processing $n lines from \"$f\"."
230   log "-----"
231   lineno=0
232   lineno2=0
233   while [ "$lineno" -lt "$n" ]
234   do
235     lineno=$(($lineno+1))
236     lineno2=$lineno
237     line=$(head -$lineno "$f" | tail -1)
238     if echo $line | grep -q '^#'; then  # comment
239       continue
240     fi
241     while echo "$line" | egrep -q '\\$' # continuation
242     do
243       lineno2=$(($lineno2+1))
244       line=$(echo -n "$line"| sed 's/\\$//'; head -$lineno2 "$f" | tail -1)
245     done
246     check=$(echo "$line" | sed 's/              */      /g'| awk -F'    ' '{print $1}')
247     action=$(echo "$line" | sed 's/             */      /g'| awk -F'    ' '{print $2}')
248     checkmsg=""
249     [ "$check" != "true" ] && checkmsg="if '$check'"
250     if eval $check; then
251       log "$(printf 'action %02d' $lineno) starting: '${action}' $checkmsg "
252       if ! eval $action; then
253         log "$(printf 'action %02d' $lineno) failed: '${action}' $checkmsg "
254         display_action_failure "$action"
255         exit 1
256       else  
257         log "$(printf 'action %02d' $lineno) finished: '${action}' $checkmsg"
258       fi
259     else
260       log "$(printf 'action %02d' $lineno) skipping: '${action}' $checkmsg"
261     fi
262     [ "$lineno2" -gt "$lineno" ] && lineno=$lineno2
263   done
264   log "run_actions ($1) finished"
265   log "-----"
266 }
267
268 # report action failure
269 display_action_failure () {
270   local action=$1
271
272   notice "
273 Doslo je do greske prilikom izvrsavanja akcije '$action'. Detalje problema moguce je vidjeti u log datotekama:
274
275   /var/log/carnet-upgrade.log
276   /var/log/carnet-upgrade.typescript
277
278 Da bi dovrsili zapocete operacije nad paketima pokrenite:
279
280   # dpkg --configure -a
281   # apt-get -f install
282
283 Nakon toga ponovno pozovite carnet-upgrade skriptu."
284 }
285
286 # funkcija vraca popis paketa koje treba upgradeati
287 what_upgrade () {
288   local IFSOLD name version package
289
290   for package in `echo "$needed_versions" | awk '{print $1}'| grep -v ^$`; do
291     IFSOLD="$IFS"
292     IFS="       " # tab
293     read name version <<-EOPTS
294         $(echo "$needed_versions" | sed 's/             */      /g' | grep ^$package)
295         EOPTS
296     IFS="$IFSOLD"
297     if pkg "$name" lt "$version"; then
298         echo -n " $name"
299     fi
300   done
301 }
302
303 # funkcija za ispis paketa koje treba upgradeati
304 check_upgrade () {
305   local upgrade
306   upgrade=${1:-"`what_upgrade`"}
307
308   if [ "$upgrade" ]; then
309     notice "
310 Prije prelaska na Debian Etch potrebno je napraviti nadogradnju ovih paketa:
311   $upgrade
312
313 Pokrenite:
314   # apt-get update
315   # apt-get dist-upgrade
316 ili:
317   # apt-get update
318   # apt-get install${upgrade}
319   
320 Nakon toga ponovno pozovite ovu skriptu."
321     log "carnet-upgrade aborted, upgrade of$upgrade needed."
322     exit 1
323   fi
324 }
325
326 # update sarge urls in sources list
327 fix_sarge_sources () {
328   local apt_sources='/etc/apt/sources.list'
329   local fixed_sources=$(mktemp /var/lib/carnet-upgrade/sources.list.XXXXXX)
330
331   # sarge (oldstable) was archived recently -> sources.list needs update
332   awk -v archive='http://archive.debian.org/debian' \
333       -v security='http://archive.debian.org/debian-security' \
334       '$3 == "sarge" && $2 ~ /[/]debian$/ { $2 = archive }
335        $3 == "sarge/updates" && $2 ~ /security\.debian\.org$/ { $2 = security }
336        { print }' $apt_sources > $fixed_sources
337
338   # did anything get updated
339   if ! cmp $apt_sources $fixed_sources >/dev/null; then
340       log "sources.list fixed to use archive.debian.org"
341
342       # backup current version
343       if [ ! -f "$apt_sources.$backup_ext.sarge" ]; then
344           cp -av $apt_sources "$apt_sources.$backup_ext.sarge"
345       fi
346
347       # install fixed version
348       cp -av $fixed_sources $apt_sources
349
350       # update the database
351       apt_update
352   fi
353
354   # delete temporary
355   rm -f $fixed_sources
356 }
357
358 # try to check if we are using the latest version of carnet-upgrade script
359 check_my_version () {
360   local packages latest_version
361
362   packages=$(mktemp /var/lib/carnet-upgrade/Packages.XXXXXX)
363   if wget -q -O$packages $packages_uri; then
364     latest_version=$(sed -ne \
365         '/Package: carnet-upgrade/,/Description:/{ /Version: /s///p }' \
366         $packages)
367     rm -f $packages
368
369     if [ "$latest_version" ]; then
370       log "Running carnet-upgrade version is $version"
371       log "Latest carnet-upgrade version is $latest_version"
372       if dpkg --compare-versions "$version" lt "$latest_version"; then
373         check_upgrade " carnet-upgrade"
374       else log "Already using the latest carnet-upgrade version."; fi
375     else log "Failed to get latest version, skipping check."; fi
376   else log "Failed to download Packages file, skipping check."; fi
377
378   rm -f $packages
379 }
380
381 # if apache is installed make sure it's version 2
382 check_apache_version () {
383
384   if pkg apache-cn && ! pkg apache2-cn; then
385     notice "
386 Prije prelaska na Debian Etch potrebno je napraviti nadogradnju sa Apache 1.3 na Apache 2.0.
387
388 Pokrenite:
389   # apt-get update
390   # apt-get install apache2-cn
391   
392 Nakon toga ponovno pozovite ovu skriptu."
393     log "carnet-upgrade aborted, upgrade to apache2-cn needed."
394     exit 1
395   fi
396 }
397
398 # if grsec is installed make sure it is fresh
399 check_grsec_kernel () {
400   local ver=`uname -r`
401
402   log "Kernel version: $ver"
403   if [ "$ver" = "${ver%-grsec}" ]; then
404     # not grsec variant
405     return 0
406   fi
407
408   if [ "$ver" = "${ver#2.6.2[4-9]}" ]; then
409     # kernel too old
410     notice "
411 Prije prelaska na Debian Etch potrebno je napraviti nadogradnju na najnoviji grsec kernel.
412
413 Pokrenite:
414   # apt-get update
415   # apt-get install kernel-2.6-cn
416   # reboot
417   
418 Nakon toga ponovno pozovite ovu skriptu."
419     log "carnet-upgrade aborted, kernel upgrade needed."
420     exit 1
421   fi
422 }
423
424 upgrade () {
425   local i to_install packages
426   case "$1" in
427     first)
428       packages="$packages_first"
429     ;;
430     second)
431       packages="$packages_second"
432     ;;
433     *)
434     return 1
435   esac
436
437   for i in $(echo $packages)
438   do
439     pkg $i && to_install="$to_install $i"
440   done
441
442   if [ -n "$to_install" ];then
443     eval pkgadd $to_install
444   else
445     return 0
446   fi
447 }
448
449 # funkcija utrpa ispravan sources.list
450 create_sources_list () {
451     local sl sl_new
452     sl=/etc/apt/sources.list
453     sl_new=/usr/share/carnet-upgrade/files/etc/apt/sources.list
454
455     # check if already installed
456     if ! cmp $sl $sl_new >/dev/null; then
457
458         # backup old version
459         if [ -f $sl -a ! -f $sl.$backup_ext ]; then
460             cp -av $sl $sl.$backup_ext
461         fi
462
463         # install new version
464         cp -av $sl_new $sl
465         notice "Novi sadrzaj datoteke /etc/apt/sources.list:
466
467 `cat /etc/apt/sources.list`"
468     fi
469 }
470
471
472 # funkcija koja brise grupu proc
473 remove_group_proc () {
474   if getent group proc > /dev/null; then
475     groupdel proc || true
476     log "groupdel proc (temporarily)"
477     post_upgrade add_group_proc
478   fi
479 }
480
481 add_group_proc () {
482   if ! getent group proc > /dev/null; then
483     # grsec uses gid 99 for /proc files
484     groupadd -g 99 proc 
485     log "groupadd -g 99 proc"
486
487     # update oidentd so it uses the proc group
488     if [ -x /etc/init.d/oidentd ]; then
489         if gpasswd -a oident proc; then
490             /etc/init.d/oidentd restart
491         fi
492     fi
493   fi
494 }
495
496 # backup ldap database
497 backup_slapd_db () {
498   local ldap_backup=/var/backups/slapcat.$backup_ext.gz
499   local old_umask result
500
501   if [ ! -f $ldap_backup ]; then
502     [ -x /etc/init.d/freeradius ] && /etc/init.d/freeradius stop || true
503
504     if [ -x /etc/init.d/slapd -a -d /var/lib/ldap ]; then
505       /etc/init.d/slapd stop || true
506       old_umask=`umask`
507       umask 0077 # protect the backup file
508       slapcat | gzip --best > $ldap_backup
509       result=$?
510       umask $old_umask
511       /etc/init.d/slapd start || true
512
513       if [ "$result" -eq 0 ]; then
514         log "$ldap_backup created"
515         notice "Napravljen backup slapd baze u $ldap_backup."
516       else
517         log "$ldap_backup was not created"
518         notice "Backup slapd baze u $ldap_backup nije uspio."
519         exit 1
520       fi
521     fi
522
523     [ -x /etc/init.d/freeradius ] && /etc/init.d/freeradius start || true
524   else
525     log "slapd backup already at $ldap_backup, skipping."
526     notice "Backup slapd baze u $ldap_backup vec postoji!"
527   fi
528 }
529
530 # backup etc direktorija
531 backup_etc_dir () {
532   local etc_backup=/var/backups/etc.$backup_ext.tar.gz
533   local old_umask
534
535   if [ ! -f $etc_backup ]; then
536     old_umask=`umask`
537     umask 0077 # protect the backup file
538
539     if tar cfz $etc_backup /etc 2> /dev/null; then
540       log "$etc_backup created"
541       notice "Napravljen backup /etc direktorija u $etc_backup."
542     else
543       log "$etc_backup was not created"
544       notice "Backup /etc direktorija u $etc_backup nije uspio."
545       exit 1
546     fi
547     umask $old_umask
548   else
549     log "/etc backup already present in $etc_backup, skipping."
550     notice "Backup /etc direktorija u $etc_backup vec postoji!"
551   fi
552 }
553
554 # fixaj /etc/default/raid2
555 fix_etc_default_raid2 () {
556   if [ -f /etc/default/raid2 ]; then
557     sed "s/^AUTOSTART.*/AUTOSTART=true/g" /etc/default/raid2 > /etc/default/raid2.$backup_ext.$$
558     chown --reference=/etc/default/raid2 /etc/default/raid2.$backup_ext.$$
559     chmod --reference=/etc/default/raid2 /etc/default/raid2.$backup_ext.$$
560
561     if ! cmp -s /etc/default/raid2.$backup_ext.$$ /etc/default/raid2; then
562       log "Fixed /etc/default/raid2 AUTOSTART option."
563       mv -v /etc/default/raid2.$backup_ext.$$ /etc/default/raid2
564     else
565       rm -v /etc/default/raid2.$backup_ext.$$
566     fi
567   fi
568 }
569
570 # restore original config file (if the new package version is not
571 # installed already)
572 restore_config () {
573   local file file_backup file_orig
574   local old_umask
575
576   # create new files safely
577   old_umask=`umask`
578   umask 0077
579
580   # restore package files
581   for file in $*; do
582     file_orig=/usr/share/carnet-upgrade/files/$file
583     file_backup=$file.$backup_ext
584
585     # is config there
586     if [ ! -e $file ]; then
587         continue
588     fi 
589
590     # is restore needed
591     if cmp $file_orig $file >/dev/null; then
592         continue
593     fi
594
595     # backup and restore
596     if [ ! -e $file_backup ]; then
597       log "Restoring config file $file"
598
599       # backup local changes
600       mv -v $file $file_backup
601
602       # restore original
603       cp -av $file_orig $file
604       chown --reference $file_backup $file
605       chmod --reference $file_backup $file
606     fi
607   done
608
609   umask $old_umask
610 }
611
612 # restore modified config to their package defaults
613 # so the upgrade doesn't complain so much
614 restore_configs () {
615   local config_backup
616
617   pkg base-files lt 4 && restore_config /etc/issue /etc/issue.net
618   pkg libpam-modules lt 0.79 && restore_config /etc/security/limits.conf
619   pkg login lt 1:4.0.18.1 && restore_config /etc/pam.d/login
620   pkg mysql-server lt 5.0.3 && restore_config /etc/init.d/mysql \
621                                     /etc/logrotate.d/mysql-server \
622                                     /etc/mysql/my.cnf
623   pkg ntp lt 1:4.2.2 && restore_config /etc/ntp.conf
624   pkg ntpdate lt 1:4.2.2 && restore_config /etc/default/ntpdate
625   pkg oidentd lt 2.0.8 && restore_config /etc/default/oidentd
626   pkg postgrey lt 1.27 && restore_config /etc/default/postgrey \
627                                /etc/postgrey/whitelist_clients
628   pkg procps lt 1:3.2.7 && restore_config /etc/sysctl.conf
629   pkg sasl2-bin lt 2.1.22 && restore_config /etc/default/saslauthd
630   pkg slapd lt 2.3.30 && restore_config /etc/default/slapd /etc/init.d/slapd
631   pkg squirrelmail lt 2:1.4.9a && restore_config /etc/squirrelmail/apache.conf
632   pkg vsftpd lt 2.0.5 && restore_config /etc/vsftpd.conf
633   pkg xinetd lt 1:2.3.14 && restore_config /etc/xinetd.conf
634
635   # orphaned config file - no owner
636   if pkg logcheck-database lt 1.2.54; then
637       rm -vf /etc/logcheck/ignore.d.server/imap
638   fi
639
640   # aide switched to ucf, move old configs aside
641   if pkg aide lt 0.13.1; then
642       for file in /etc/aide/aide.conf \
643                   /etc/cron.daily/aide \
644                   /etc/default/aide;
645       do
646           if [ ! -e "$file.$backup_ext" ]; then
647               mv -v "$file" "$file.$backup_ext"
648           fi
649
650           rm -vf "$file"
651       done
652   fi
653
654   # register cn changes in ucf managed files
655   for config in /etc/apache/modules.conf \
656                 /etc/clamav/clamd.conf \
657                 /etc/clamav/freshclam.conf
658   do 
659       config_backup=$config.$backup_ext
660       if [ -e $config -a ! -e $config_backup ]; then
661           log "Saving config file $config"
662           cp -av $config $config_backup
663           ucf $config_backup $config
664       fi
665   done
666 }
667
668 # temporary disable listchanges packages to reduce clutter during upgrade
669 apt_listchanges () {
670   local file command=$1
671
672   for file in /etc/apt/apt.conf.d/20listchanges \
673               /etc/apt/apt.conf.d/20listchanges-cn
674   do
675       case $command in
676            disable)
677                if [ -f $file ]; then
678                    dpkg-divert --local --rename --divert $file.$backup_ext \
679                                --add $file || true
680                fi
681                ;;
682
683            enable)
684                if [ -f $file.$backup_ext ]; then
685                    dpkg-divert --remove $file || true
686                fi
687                ;;
688      esac
689   done
690 }
691
692 # make a silent installation of carnet and srce keyrings
693 install_keyrings () {
694   pkgadd carnet-keyring srce-keyring
695   pkgadd dpkg apt debconf python-apt
696
697   dpkg-reconfigure carnet-keyring srce-keyring
698   apt_update
699 }
700
701 # remove skey (not supported anymore)
702 remove_skey () {
703   pkgrm skey-cn libpam-skey
704
705   if getent group skey > /dev/null; then
706     groupdel skey || true
707     log "groupdel skey"
708   fi
709 }
710
711 # monit it causing problems for postinst scripts
712 # restarting daemons so try to disable it
713 disable_monit () {
714   if pkg monit-cn; then
715     pkgrm monit-cn # disables monit in prerm
716   fi
717
718   # if configured manually
719   if pkg monit; then
720     # remove the binary but leave the config
721     apt-get remove --yes monit
722     # stop the binary
723     pkill -9 monit || true
724   fi
725 }
726
727 # make a silent upgrade to new libc6
728 upgrade_libc () {
729   DEBIAN_FRONTEND=noninteractive pkgadd libc6
730 }
731
732 # upgrade apache2 to etch
733 upgrade_apache2 () {
734   local package packages
735
736   for package in \
737       libapache2-mod-php4 \
738       php4-cli \
739       php4-cgi \
740       php4-cn \
741       apache2-cn \
742       php4-odbc \
743       php4-suhosin \
744       php-suhosin-cn
745   do
746       if pkg $package; then
747           packages="$packages $package"
748       fi
749   done
750
751   # enable suhosin downgrade to etch version
752   if pkg php4-suhosin; then
753       pkgrm php4-suhosin php-suhosin-cn
754   fi
755
756   eval pkgadd $packages
757 }
758
759 upgrade_amavis () {
760   local conf
761
762   # remove init script diversion before upgrade
763   if pkg amavisd-cn lt 3:2.4.2-4; then 
764     if [ -L /etc/init.d/amavis -a -f /etc/init.d/amavis.amavisd-new ]; then
765       rm -vf /etc/init.d/amavis
766       dpkg-divert --remove /etc/init.d/amavis
767     fi
768   fi
769
770   # move old config aside
771   for conf in /etc/amavisd.conf /etc/amavis/amavisd.conf; do
772     if [ -e $conf -a ! -e $conf.$backup_ext ]; then
773       mv -v $conf $conf.$backup_ext
774     fi
775   done
776
777   # install new packages
778   pkgadd amavisd-cn amavisd-new
779
780   # fix new packages
781   check_and_sed '^clamd.*5.clamav.log$' \
782                 's/^\(clamd.*\)5.clamav.log$/\14\tsocket/g' \
783                 /etc/init.d/amavisd-cn || true
784
785   # check if link got lost
786   if pkg amavisd-cn && [ ! -L /etc/init.d/amavis ]; then
787     apt-get install --reinstall --yes amavisd-cn
788   fi
789
790   # start new packages
791   /etc/init.d/amavis restart
792 }
793
794 # fix openldap-aai postinst user handling
795 upgrade_openldap () {
796   pkg openldap-aai lt 2.3 || return 0
797
798   # openldap-aai expects org.ldif to exists
799   local ldif=/var/lib/ldap/org.ldif
800   local ldif_backup=/var/lib/carnet-upgrade/org.ldif
801   if [ -f $ldif ]; then
802     cp -v $ldif $ldif_backup
803   fi
804
805   # slapd postinst fails if move_old_database is false
806   echo 'slapd slapd/move_old_database boolean true' | debconf-set-selections 
807   pkgadd slapd
808
809   # slapd removes org.ldif during switch from ldap -> openldap user
810   if [ -f $ldif_backup -a ! -f $ldif ]; then
811     cp -v $ldif_backup $ldif
812   fi
813
814   # openldap-aai postinst fails here because of database permissions
815   apt-get -y install openldap-aai || true
816   chown -R openldap:openldap /var/lib/ldap
817   /etc/init.d/slapd restart
818   dpkg --configure -a
819
820   # remove ldap user if openldap works
821   if getent passwd ldap      > /dev/null && \
822      getent passwd openldap  > /dev/null && \
823      pgrep -U openldap slapd > /dev/null
824   then
825      log "userdel ldap (replaced by openldap)"
826      userdel ldap || true
827   fi
828 }
829
830 # fix postgresql deinstallation during upgrade
831 upgrade_postgresql () {
832   if check_services show 2>&1 | grep -qw postgresql; then
833      apt-get install -y postgresql || apt-get -yf install
834   fi
835 }
836
837 # upgrade the MTA
838 upgrade_postfix () {
839   pkgadd postfix-cn
840
841   if pkg postgrey; then
842      pkgrm postgrey # not used by postfix-cn anymore
843   fi
844 }
845
846 get_variable () {
847   local name=$1 file=$2 val
848
849   if [ -f "$file" ]; then
850      val=$(sed -n "/^[[:space:]]*$name[[:space:]]*=/s/[[:space:]]*//gp" $file)
851      val=${val#*=}
852   fi
853
854   echo $val
855 }
856
857 # preserve mail location on upgrade
858 upgrade_dovecot () {
859   local default_mail_env mail_location
860
861   # check if already upgraded
862   if ! pkg dovecot-common lt 1.0; then
863       return
864   fi
865
866   # save old mail location
867   default_mail_env=$(get_variable default_mail_env /etc/dovecot/dovecot.conf)
868
869   # silently upgrade package
870   restore_config /etc/dovecot/dovecot.conf
871   pkgadd dovecot-cn
872
873   # insert old mail location
874   mail_location=$(get_variable mail_location /etc/dovecot/dovecot.conf)
875   if [ "$default_mail_env" -a ! "$mail_location" ]; then
876       check_and_sed '^#mail_location[[:space:]]*=' \
877                    "s|#mail_location[[:space:]]*=.*|mail_location = $default_mail_env|" /etc/dovecot/dovecot.conf || return 0
878
879       if [ -x /etc/init.d/dovecot ]; then
880           if [ -x /usr/sbin/invoke-rc.d ]; then
881               /usr/sbin/invoke-rc.d dovecot restart
882           else
883               /etc/init.d/dovecot restart
884           fi
885       fi
886   fi
887 }
888
889 # fix /etc/dpkg/dpkg.cfg
890 comment_force_overwrite () {
891   if [ -f /etc/dpkg/dpkg.cfg ]; then
892     if check_and_sed '^force-overwrite' \
893                   's,\(^force-overwrite\),#\1,' \
894                   /etc/dpkg/dpkg.cfg; then
895       notice "$msg_comment_force_overwrite"
896     fi
897   fi
898 }
899
900 fix_issue () {
901   cat > /etc/issue <<EOF
902 Debian GNU/Linux 4.0 (CARNet Debian 4.0) \\n \\l
903 EOF
904
905   cat > /etc/issue.net <<EOF
906 Debian GNU/Linux 4.0 (CARNet Debian 4.0) %h
907 EOF
908 }
909
910 check_archives_space() {
911   local available_disk_space archives_size
912   available_disk_space=$(free_space /var/cache/apt/archives/)
913   archives_size=$(download_size dist-upgrade)
914
915   if [ $? -ne 0 ]; then
916     log "Failed to get download size, skipping check."
917     return 0
918   fi
919
920   log "Archives space: ${available_disk_space}MB"
921   log "Download size: ${archives_size}MB"
922
923   if [ $available_disk_space -lt $archives_size ]; then
924     log "carnet-upgrade aborted: not enough space in /var/cache/apt/archives/ (have ${available_disk_space}MB, need ${archives_size}MB)"
925     notice "Premalo slobodnog mjesta u /var/cache/apt/archives/ za skidanje paketa (potrebno je barem ${archives_size}MB). Procitajte poglavlje 'Izvanredne situacije' u dokumentaciji."
926     exit 1
927   fi
928 }
929
930 check_kernel_space() {
931   local available_disk_space kernel_size
932   available_disk_space=$(free_space /)
933   kernel_size=$(installed_size kernel-2.6-cn)
934
935   if [ $? -ne 0 ]; then
936     log "Failed to get installed size, skipping check."
937     return 0
938   fi
939
940   log "Free space in /: ${available_disk_space}MB"
941   log "Kernel size: ${kernel_size}MB"
942
943   if LC_ALL=C apt-get --yes --print-uris install kernel-2.6-cn \
944      | grep -q '^kernel-2.6-cn is already the newest version.$'
945   then
946      log "Not checking free space in /: kernel-2.6-cn already installed"
947      return
948   fi
949
950   if [ $available_disk_space -lt $kernel_size ]; then
951     log "carnet-upgrade aborted: not enough space in / (have ${available_disk_space}MB, need ${kernel_size}MB)"
952     notice "Premalo mjesta u / za instalaciju kernela (potrebno je barem ${kernel_size}MB). Procitajte poglavlje 'Izvanredne situacije' u dokumentaciji."
953     exit 1
954   fi
955 }
956
957 free_space() {
958   df --portability --block-size=1M "$1" | tail -1 | awk '{print $4}'
959 }
960
961 download_size() {
962   local size
963   size=$(LC_ALL=C apt-get --yes --print-uris "$@" | grep '^Need to get ')
964
965   if [ $? -ne 0 ]; then
966     log "Failed to get download size for 'apt-get $@'."
967     return 1
968   fi
969
970   size=${size#Need to get }
971   size=${size%%B*}
972
973   size_to_mb "${size}B"
974 }
975
976 installed_size() {
977   local package size
978   package=$1
979   size=$(apt-cache show $package | grep '^Installed-Size:')
980
981   if [ $? -ne 0 ]; then
982     log "Failed to get installed size for '$package'."
983     return 1
984   fi
985
986   size=${size#*: }
987
988   size_to_mb "${size}kB"
989 }
990
991 size_to_mb() {
992   local size
993   size=$1
994   if [ -z "$size" ]; then
995     size=0;
996   fi
997
998   if [ "${size%kB}" != "$size" ]; then
999     size=$(round ${size%kB})
1000     size=$(($size/1024+1))
1001   elif [ "${size%MB}" != "$size" ]; then
1002     size=$(round ${size%MB})
1003   elif [ "${size%GB}" != "$size" ]; then
1004     size=$(round ${size%GB})
1005     size=$(($size*1024))
1006   elif [ "${size%[0-9]B}" != "$size" ]; then
1007     size=$(round ${size%B})
1008     size=$(($size/1024/1024+1))
1009   else
1010     log "unrecognized size unit: $size"
1011     return 1
1012   fi
1013
1014   echo $size
1015 }
1016
1017 round() {
1018   local num
1019   num=$1
1020
1021   if [ "${num%.*}" != "$num" ]; then
1022     num=${num%.*}
1023     num=$(($num+1))
1024   fi
1025
1026   echo $num
1027 }