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