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 Squeeze 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 grep -q "$running_release .* $running_version" $default_kernel; then
 484       log "Latest kernel version is running."
 485       return 1
 486   else
 487       log "Reboot to latest version required."
 488       return 0
 489   fi
 490 }
 491
 492 upgrade () {
 493   local packages
 494   case "$1" in
 495     first)
 496       packages="$packages_first"
 497     ;;
 498     second)
 499       packages="$packages_second"
 500     ;;
 501     *)
 502     return 1
 503   esac
 504
 505   pkgupgrade $packages
 506 }
 507
 508 # install packages introduced in this release
 509 install_new () {
 510   if [ "$new_packages" ]; then
 511     pkgadd $new_packages
 512   fi
 513 }
 514
 515 # funkcija utrpa ispravan sources.list
 516 create_sources_list () {
 517     local sl sl_new
 518     sl=/etc/apt/sources.list
 519     sl_new=$( copy_template $sl )
 520
 521     # comment aai repository if not in use
 522     if ! pkg srce-keyring; then
 523         sed -i 's/^\(deb.*ftp.srce.hr\)/#\1/' $sl_new
 524     fi
 525
 526     # check if already installed
 527     if ! cmp $sl $sl_new >/dev/null; then
 528
 529         # backup old version
 530         if [ -f $sl -a ! -f $sl.$backup_ext ]; then
 531             cp -av $sl $sl.$backup_ext
 532         fi
 533
 534         # install new version
 535         cp -av $sl_new $sl
 536         notice "Novi sadrzaj datoteke /etc/apt/sources.list:
 537
 538 `cat /etc/apt/sources.list`"
 539     fi
 540
 541     rm $sl_new
 542 }
 543
 544 # funkcija koja brise grupu proc
 545 remove_group_proc () {
 546   if getent group proc > /dev/null; then
 547     groupdel proc || true
 548     log "groupdel proc (temporarily)"
 549     post_upgrade add_group_proc
 550   fi
 551 }
 552
 553 add_group_proc () {
 554   if ! getent group proc > /dev/null; then
 555     # grsec uses gid 99 for /proc files
 556     groupadd -g 99 proc
 557     log "groupadd -g 99 proc"
 558
 559     # update oidentd so it uses the proc group
 560     if [ -x /etc/init.d/oidentd ]; then
 561         if gpasswd -a oident proc; then
 562             /etc/init.d/oidentd restart
 563         fi
 564     fi
 565   fi
 566 }
 567
 568 # backup ldap database
 569 backup_slapd_db () {
 570   local ldap_backup=/var/backups/slapcat.$backup_ext.gz
 571   local old_umask result
 572
 573   if [ ! -f $ldap_backup ]; then
 574     [ -x /etc/init.d/freeradius ] && /etc/init.d/freeradius stop || true
 575
 576     if [ -x /etc/init.d/slapd -a -d /var/lib/ldap ]; then
 577       /etc/init.d/slapd stop || true
 578       old_umask=`umask`
 579       umask 0077 # protect the backup file
 580       slapcat | gzip --best > $ldap_backup
 581       result=$?
 582       umask $old_umask
 583       /etc/init.d/slapd start || true
 584
 585       if [ "$result" -eq 0 ]; then
 586         log "$ldap_backup created"
 587         notice "Napravljen backup slapd baze u $ldap_backup."
 588       else
 589         log "$ldap_backup was not created"
 590         notice "Backup slapd baze u $ldap_backup nije uspio."
 591         exit 1
 592       fi
 593     fi
 594
 595     [ -x /etc/init.d/freeradius ] && /etc/init.d/freeradius start || true
 596   else
 597     log "slapd backup already at $ldap_backup, skipping."
 598     #notice "Backup slapd baze u $ldap_backup vec postoji!"
 599   fi
 600 }
 601
 602 # backup etc direktorija
 603 backup_etc_dir () {
 604   local etc_backup=/var/backups/etc.$backup_ext.tar.gz
 605   local old_umask
 606
 607   if [ ! -f $etc_backup ]; then
 608     old_umask=`umask`
 609     umask 0077 # protect the backup file
 610
 611     if tar cfz $etc_backup /etc 2> /dev/null; then
 612       log "$etc_backup created"
 613       notice "Napravljen backup /etc direktorija u $etc_backup."
 614     else
 615       log "$etc_backup was not created"
 616       notice "Backup /etc direktorija u $etc_backup nije uspio."
 617       exit 1
 618     fi
 619     umask $old_umask
 620   else
 621     log "/etc backup already present in $etc_backup, skipping."
 622     #notice "Backup /etc direktorija u $etc_backup vec postoji!"
 623   fi
 624 }
 625
 626 # restore a configuration file if it contains only CN modifications
 627 restore_file () {
 628   local file file_expect file_restore file_backup
 629
 630   file=$1
 631   file_expect=$2
 632   file_restore=$3
 633   file_backup=$file.$backup_ext
 634
 635   [ "$file_expect" ] || file_expect=$cnup/files/$file.expect
 636   [ "$file_restore" ] || file_restore=$cnup/files/$file.restore
 637
 638   # check if all required files are there
 639   if ! [ -e $file -a -e $file_expect -a -e $file_restore ]; then
 640     log "Required files missing, skipping $file restore."
 641     return 1
 642   fi
 643
 644   # is restore needed
 645   if ! cmp $file_expect $file >/dev/null; then
 646     log "Config doesn't match the template, skipping $file restore."
 647     return 1
 648   fi
 649
 650   # backup and restore
 651   if [ ! -e $file_backup ]; then
 652     log "Restoring config file $file."
 653
 654     # backup local changes
 655     cp -av $file $file_backup
 656
 657     # restore original
 658     cp -v $file_restore $file
 659
 660     # restore succedded
 661     return 0
 662   else
 663     log "Backup file exists, skipping $file restore."
 664     return 1
 665   fi
 666 }
 667
 668 # force reconfiguration at the end if package is not upgraded automatically
 669 postupgrade_reconfigure () {
 670   local cn_package cn_version
 671
 672   cn_package=$1
 673   cn_version=$( dpkg -s "$cn_package" | awk '/^Version:/ {print $2}' )
 674   post_upgrade "pkg $cn_package gt $cn_version || dpkg-reconfigure $cn_package"
 675 }
 676
 677 # copy template into a temporary file
 678 copy_template () {
 679   local path file template config_new
 680
 681   path=$1
 682   file=$(basename $path)
 683   template=$cnup/files/$path.template
 684   config_new=$(mktemp /var/lib/carnet-upgrade/$file.XXXXXX)
 685   cp $template $config_new
 686
 687   echo $config_new
 688 }
 689
 690 # restore modified config to their package defaults
 691 # so the upgrade doesn't complain so much
 692 restore_configs () {
 693   local hostname domain config_new memtotal memlimit
 694
 695   # restore simple configs
 696   if pkg ntp-cn && pkg ntp lt 1:4.2.6.p2+dfsg-1+b1; then
 697      if restore_file /etc/ntp.conf; then
 698         postupgrade_reconfigure ntp-cn
 699      fi
 700   fi
 701
 702   if pkg kernel-2.6-cn && pkg procps lt 1:3.2.8-9; then
 703      if restore_file /etc/sysctl.conf; then
 704         postupgrade_reconfigure kernel-2.6-cn
 705      fi
 706   fi
 707
 708   if pkg kernel-2.6-cn && pkg libpam-modules lt 1.1.1-6.1; then
 709      if restore_file /etc/security/limits.conf; then
 710         postupgrade_reconfigure kernel-2.6-cn
 711      fi
 712   fi
 713
 714   if pkg vsftpd-cn && pkg vsftpd lt 2.3.2-3+squeeze2; then
 715      if restore_file /etc/vsftpd.conf; then
 716         postupgrade_reconfigure vsftpd-cn
 717      fi
 718   fi
 719
 720   if pkg squirrelmail-cn && pkg squirrelmail lt 2:1.4.21-2; then
 721      if restore_file /etc/squirrelmail/apache.conf; then
 722         postupgrade_reconfigure squirrelmail-cn
 723      fi
 724   fi
 725
 726   if pkg spamassassin-cn && pkg spamassassin lt 3.3.1-1; then
 727      if restore_file /etc/spamassassin/v310.pre; then
 728         postupgrade_reconfigure spamassassin-cn
 729      fi
 730   fi
 731
 732   if pkg amavisd-cn && pkg amavisd-new lt 1:2.6.4-3; then
 733      if restore_file /etc/cron.daily/amavisd-new; then
 734         rm -f /etc/cron.daily/amavisd-new.$backup_ext
 735         postupgrade_reconfigure amavisd-cn
 736      fi
 737
 738      if restore_file /etc/amavis/conf.d/15-av_scanners; then
 739         rm -f /etc/amavis/conf.d/15-av_scanners.$backup_ext
 740         postupgrade_reconfigure amavisd-cn
 741      fi
 742   fi
 743
 744   if pkg amavisd-cn lt 3:2.6.5; then
 745      restore_file /etc/init.d/amavisd-cn
 746      rm -f /etc/init.d/amavisd-cn.$backup_ext
 747   fi
 748
 749   if pkg console-tools lt 1:0.2.3dbs-69.1; then
 750      restore_file /etc/console-tools/config
 751   fi
 752
 753   if pkg base-files lt 6.0squeeze2; then
 754      restore_file /etc/issue
 755      restore_file /etc/issue.net
 756   fi
 757
 758   if pkg slapd lt 2.4.23-7.2; then
 759      restore_file /etc/default/slapd
 760   fi
 761
 762   # dovecot won't start with these options
 763   if pkg dovecot-cn && pkg dovecot-common lt 1:1.2.15-7; then
 764      sed -i 's/^\( *\)\(sieve\(_storage\)\?=.*\)/\1#\2/' \
 765          /etc/dovecot/dovecot.conf
 766   fi
 767
 768   # check if monitrc is template based
 769   if [ -e /etc/monit/monitrc ]; then # monit is removed at this point
 770      # regenerate config from template
 771      hostname=$(hostname)
 772      domain=$(hostname --domain)
 773      config_new=$(copy_template /etc/monit/monitrc)
 774      check_and_sed "@localhost" \
 775        "s/@localhost/@$hostname.$domain/g" $config_new || true
 776      restore_file /etc/monit/monitrc $config_new
 777      rm -f $config_new
 778   fi
 779 }
 780
 781 # temporary disable listchanges packages to reduce clutter during upgrade
 782 apt_listchanges () {
 783   local file command=$1
 784
 785   for file in /etc/apt/apt.conf.d/20listchanges \
 786               /etc/apt/apt.conf.d/20listchanges-cn
 787   do
 788       case $command in
 789            disable)
 790                if [ -f $file ]; then
 791                    dpkg-divert --local --rename --divert $file.disabled \
 792                                --add $file || true
 793                fi
 794                ;;
 795
 796            enable)
 797                # cleanup the diversion even if the file is already removed
 798                dpkg-divert --rename --remove $file || true
 799                ;;
 800      esac
 801   done
 802 }
 803
 804 # make a silent installation of carnet and srce keyrings
 805 install_keyrings () {
 806   pkgadd carnet-keyring debian-archive-keyring
 807   dpkg-reconfigure carnet-keyring debian-archive-keyring
 808
 809   if pkg srce-keyring; then
 810     pkgadd srce-keyring
 811     dpkg-reconfigure srce-keyring
 812   fi
 813
 814   apt_update
 815 }
 816
 817 # check if package is orphaned (nothing depends on it)
 818 is_orphaned () {
 819   local package deps
 820
 821   package=$1
 822   deps=$(apt-get remove -s $package | grep ^Remv | wc -l)
 823   if [ "$deps" -eq 1 ]; then
 824     return 0
 825   else
 826     return 1
 827   fi
 828 }
 829
 830 # remove old and unused libraries
 831 remove_orphaned () {
 832   local package remove
 833
 834   apt_autoremove
 835
 836   remove=
 837   for package in $orphaned_packages; do
 838     if is_orphaned $package; then
 839       remove="$remove $package"
 840     fi
 841   done
 842
 843   if [ "$remove" ]; then
 844     pkgrm $remove
 845   fi
 846 }
 847
 848 # monit it causing problems for postinst scripts
 849 # restarting daemons so try to disable it
 850 disable_monit () {
 851   if pkg monit-cn; then
 852     pkgrm monit-cn # disables monit in prerm
 853   fi
 854
 855   # if configured manually
 856   if pkg monit; then
 857     # remove the binary but leave the config
 858     apt-get remove --yes monit
 859     # stop the binary
 860     pkill -9 monit || true
 861     # move away init for insserv to work
 862     mv /etc/init.d/monit /var/lib/carnet-upgrade/
 863   fi
 864 }
 865
 866 # allow monit to run again
 867 enable_monit () {
 868   if [ -f /var/lib/carnet-upgrade/monit ]; then
 869     mv /var/lib/carnet-upgrade/monit /etc/init.d/monit
 870     pkgadd monit-cn
 871   fi
 872 }
 873
 874 # check which of the given packages are installed
 875 grep_installed () {
 876   local package installed
 877
 878   for package in $*; do
 879     if pkg $package; then
 880       installed="$installed $package"
 881     fi
 882   done
 883
 884   echo $installed
 885 }
 886
 887 # only upgrade packages if they are already installed
 888 pkgupgrade () {
 889   local packages
 890
 891   packages=$( grep_installed $* )
 892
 893   if [ "$packages" ]; then
 894     eval pkgadd $packages
 895   fi
 896 }
 897
 898 # make a silent upgrade to new libc6
 899 upgrade_libc () {
 900   DEBIAN_FRONTEND=noninteractive pkgupgrade libc6 libc6-dev tzdata
 901 }
 902
 903 # upgrade apache2/php5
 904 upgrade_apache2 () {
 905   pkgrm apache-common # prevents installation of apache2-suexec
 906   pkgupgrade libapache2-mod-php5 php5-cli php5-cn apache2-cn \
 907     php5-odbc php5-suhosin php-suhosin-cn
 908 }
 909
 910 upgrade_amavis () {
 911   local conf
 912
 913   # move old config aside
 914   for conf in /etc/amavisd.conf /etc/amavis/amavisd.conf; do
 915     if [ -e $conf -a ! -e $conf.$backup_ext ]; then
 916       mv -v $conf $conf.$backup_ext
 917     fi
 918   done
 919
 920   # install new packages
 921   pkgadd amavisd-cn amavisd-new
 922
 923   # start new packages
 924   /etc/init.d/amavisd-cn restart
 925 }
 926
 927 # handle freerdius upgrade
 928 upgrade_freeradius () {
 929   local password
 930
 931   pkg freeradius-aai lt 2.1.10-1 || return 0
 932
 933   # aai team prevents a normal upgrade
 934   cp -av /etc/freeradius /etc/freeradius.$backup_ext
 935   pkgrm freeradius-aai freeradius-ldap freeradius
 936   rm -rf /etc/freeradius/certs
 937   pkgadd freeradius-aai
 938
 939   # sync localhost passwords
 940   if pkg libpam-radius-auth && [ -f "/etc/pam_radius_auth.conf" ]; then
 941     pkgadd libpam-radius-auth
 942     password=$(
 943       sed -n '/^[[:space:]]*client[[:space:]]\+localhost/,/^[[:space:]]*}/ { /^[[:space:]]*secret[[:space:]]*=[[:space:]]*\([^[:space:]]*\)/s//\1/p }' /etc/freeradius/clients.conf
 944     )
 945     if [ "$password" ] \
 946        && ! grep -q "^127.0.0.1\(:[[:digit:]]\+\)\?[[:space:]]\+$password[[:space:]]\+" \
 947          /etc/pam_radius_auth.conf
 948     then
 949       sed -i.$backup_ext "s/^\(127.0.0.1\(:[[:digit:]]\+\)\?[[:space:]]\+\)[^[:space:]]\+\(.*\)/\1$password\3/" /etc/pam_radius_auth.conf
 950     fi
 951   fi
 952 }
 953
 954 # handle mysql 5.0 to 5.1 upgrade
 955 upgrade_mysql () {
 956   if pkg mysql-server || pkg mysql-server-5.0; then
 957      pkgadd mysql-server
 958      dpkg -P mysql-server-5.0
 959      /etc/init.d/mysql restart
 960   fi
 961 }
 962
 963 # fix postgresql deinstallation during upgrade
 964 upgrade_postgresql () {
 965   if check_services show 2>&1 | grep -qw postgresql; then
 966      apt-get install -y postgresql || apt-get -yf install
 967   fi
 968 }
 969
 970 # upgrade the MTA
 971 upgrade_postfix () {
 972   pkgadd sasl2-bin postfix-cn
 973
 974   if pkg postgrey; then
 975      pkgrm postgrey # not used by postfix-cn anymore
 976   fi
 977 }
 978
 979 # upgrade bind separately so DNS is not down for too long
 980 # or breaks postinst scripts of other cn packages that depend on
 981 # working resolver
 982 upgrade_bind() {
 983   pkgadd bind9-cn
 984 }
 985
 986 get_variable () {
 987   local name=$1 file=$2 val
 988
 989   if [ -f "$file" ]; then
 990      val=$(sed -n "/^[[:space:]]*$name[[:space:]]*=/s/[[:space:]]*//gp" $file)
 991      val=${val#*=}
 992   fi
 993
 994   echo $val
 995 }
 996
 997 # restore munin configs for silent upgrade
 998 upgrade_munin () {
 999   local config backup hostname domain
1000
1001   # new htmldir location
1002   if [ -d /var/www/munin -a ! -e /var/cache/munin/www ]; then
1003     mkdir -p /var/cache/munin
1004     mv /var/www/munin /var/cache/munin/www
1005   fi
1006
1007   for pkg_name in munin munin-node; do
1008      pkg $pkg_name lt '1.4.5-3' || continue
1009
1010      config=/etc/munin/$pkg_name.conf
1011      backup=$config.$backup_ext
1012      if [ -f $config -a ! -f $backup ]; then
1013         hostname=$(hostname)
1014         domain=$(hostname --domain)
1015
1016         cp -av $config $backup
1017         check_and_sed "$hostname.$domain" \
1018           "s/$hostname.$domain/localhost.localdomain/g" $config
1019      fi
1020      pkgadd $pkg_name
1021   done
1022
1023   # force localhost -> hostname replacement in config files
1024   pkgadd munin-cn
1025   dpkg-reconfigure munin-cn
1026 }
1027
1028 # try not to overwrite user modified issue files
1029 fix_issue () {
1030   if cmp /etc/issue $cnup/files/etc/issue.expect.new >/dev/null; then
1031     cp -av /etc/issue /etc/issue.$backup_ext
1032     cat > /etc/issue <<EOF
1033 Debian GNU/Linux 6.0 (CARNet Debian 6.0) \\n \\l
1034
1035 EOF
1036   fi
1037
1038   if cmp /etc/issue.net $cnup/files/etc/issue.net.expect.new >/dev/null; then
1039     cp -av /etc/issue.net /etc/issue.net.$backup_ext
1040     cat > /etc/issue.net <<EOF
1041 Debian GNU/Linux 6.0 (CARNet Debian 6.0) %h
1042 EOF
1043   fi
1044 }
1045
1046 check_archives_space() {
1047   local available_disk_space archives_size
1048   available_disk_space=$(free_space /var/cache/apt/archives/)
1049   archives_size=$(download_size dist-upgrade)
1050
1051   if [ $? -ne 0 ]; then
1052     log "Failed to get download size, skipping check."
1053     return 0
1054   fi
1055
1056   log "Archives space: ${available_disk_space}MB"
1057   log "Download size: ${archives_size}MB"
1058
1059   if [ $available_disk_space -lt $archives_size ]; then
1060     log "carnet-upgrade aborted: not enough space in /var/cache/apt/archives/ (have ${available_disk_space}MB, need ${archives_size}MB)"
1061     notice "Premalo slobodnog mjesta u /var/cache/apt/archives/ za skidanje paketa (potrebno je barem ${archives_size}MB). Procitajte poglavlje 'Izvanredne situacije' u dokumentaciji."
1062     exit 1
1063   fi
1064 }
1065
1066 # calculate the list of linux-image packages from kernel-2.6-cn dependencies
1067 get_cn_kernels() {
1068   local cn_kernels linux_images metapkg pkg
1069
1070   cn_kernels=
1071   linux_images=$(apt-cache show kernel-2.6-cn | grep ^Depends: \
1072     | grep -o 'linux-image[^, ]*')
1073   for metapkg in $linux_images; do
1074     pkg=$(apt-cache show $metapkg | grep ^Depends: \
1075           | grep -o 'linux-image[^, ]*' | head -1)
1076     [ "$pkg" ] && cn_kernels="$cn_kernels $pkg"
1077   done
1078
1079   echo $cn_kernels
1080 }
1081
1082 # check if there is enough space on / for the new kernel package
1083 check_kernel_space() {
1084   local available_disk_space kernel_size linux_images pkg ret
1085   available_disk_space=$(free_space /)
1086
1087   linux_images=$( get_cn_kernels )
1088   for pkg in $linux_images; do
1089     kernel_size=$(installed_size $pkg)
1090     ret=$?
1091     [ $ret -eq 0 ] && break
1092   done
1093
1094   if [ $ret -ne 0 ]; then
1095     log "Failed to get installed size, skipping check."
1096     return 0
1097   fi
1098
1099   log "Free space in /: ${available_disk_space}MB"
1100   log "Kernel package: $pkg"
1101   log "Kernel size: ${kernel_size}MB"
1102
1103   if LC_ALL=C apt-get -s install kernel-2.6-cn 2>/dev/null \
1104      | grep -q '^kernel-2.6-cn is already the newest version.$'
1105   then
1106      log "Not checking free space in /: kernel-2.6-cn already installed"
1107      return
1108   fi
1109
1110   if [ $available_disk_space -lt $kernel_size ]; then
1111     log "carnet-upgrade aborted: not enough space in / (have ${available_disk_space}MB, need ${kernel_size}MB)"
1112     notice "Premalo mjesta u / za instalaciju kernela (potrebno je barem ${kernel_size}MB). Procitajte poglavlje 'Izvanredne situacije' u dokumentaciji."
1113     exit 1
1114   fi
1115 }
1116
1117 # free some space on / partition by cleaning old unused kernels
1118 clean_old_kernels() {
1119   local installed keep pkg keep remove delete name dialog_list selection
1120
1121   # find all installed kernels
1122   installed=$(
1123     dpkg -l | egrep '^ii  linux-image-[0-9]+[.][0-9]+[.][0-9]+-' \
1124     | awk '{print $2}'
1125   )
1126   log "Found kernel packages: $installed"
1127
1128   # leave current and new kernels
1129   keep="linux-image-$(uname -r) $(get_cn_kernels)"
1130   log "Keep kernel packages: $keep"
1131
1132   # check what to remove
1133   remove=
1134   for pkg in $installed; do
1135     delete=yes
1136     for name in $keep; do
1137       [ "$pkg" = "$name" ] && delete=
1138     done
1139     if [ "$delete" = yes ]; then
1140       remove="$remove $pkg"
1141     fi
1142   done
1143   log "Obsolete kernel packages: $remove"
1144
1145   if [ "$remove" ]; then
1146     for pkg in $remove; do
1147       dialog_list="$dialog_list $pkg '' off"
1148     done
1149
1150     # user selects the packages to delete
1151     selection=$( mktemp /var/lib/carnet-upgrade/selection.XXXXXX )
1152     eval LC_MESSAGES=hr_HR dialog --nocancel --backtitle \""$title"\" \
1153       --checklist \""$msg_remove_kernels"\" 20 75 6 $dialog_list 2>$selection
1154
1155     selected=$( tr -d \" < $selection )
1156     rm -f $selection
1157
1158     if [ "$selected" ]; then
1159       log "Removing kernel packages: $selected"
1160       pkgrm $selected
1161     fi
1162   fi
1163 }
1164
1165 # prevent sysv-rc migration problems
1166 clean_initd_packages() {
1167   local list removed pkg dialog_list selection selected
1168
1169   list=$( dpkg -S /etc/init.d/\* | grep -v ^diversion | cut -d: -f1 | sort -u )
1170   removed=$( dpkg -l $list | sed -n 's/^r.[[:space:]]\+\([^[:space:]]\+\).*/\1/p' )
1171   log "Found removed packages: $removed"
1172
1173   if [ "$removed" ]; then
1174     for pkg in $removed; do
1175       dialog_list="$dialog_list $pkg '' off"
1176     done
1177
1178     # user selects the packages to delete
1179     selection=$( mktemp /var/lib/carnet-upgrade/selection.XXXXXX )
1180     eval LC_MESSAGES=hr_HR dialog --nocancel --backtitle \""$title"\" \
1181       --checklist \""$msg_remove_initd"\" 20 75 6 $dialog_list 2>$selection
1182
1183     selected=$( tr -d \" < $selection )
1184     rm -f $selection
1185
1186     if [ "$selected" ]; then
1187       log "Purging packages: $selected"
1188       dpkg -P $selected
1189     fi
1190   fi
1191 }
1192
1193 free_space() {
1194   df --portability --block-size=1M "$1" | tail -1 | awk '{print $4}'
1195 }
1196
1197 download_size() {
1198   local size
1199   size=$(LC_ALL=C apt-get --yes --print-uris "$@" | grep '^Need to get ')
1200
1201   if [ $? -ne 0 ]; then
1202     log "Failed to get download size for 'apt-get $@'."
1203     return 1
1204   fi
1205
1206   size=${size#Need to get }
1207   size=${size%%B*}
1208
1209   size_to_mb "${size}B"
1210 }
1211
1212 installed_size() {
1213   local package size
1214   package=$1
1215   size=$(apt-cache show $package | grep '^Installed-Size:')
1216
1217   if [ $? -ne 0 ]; then
1218     log "Failed to get installed size for '$package'."
1219     return 1
1220   fi
1221
1222   size=${size#*: }
1223
1224   size_to_mb "${size}kB"
1225 }
1226
1227 size_to_mb() {
1228   local size
1229   size=$1
1230   if [ -z "$size" ]; then
1231     size=0;
1232   fi
1233
1234   if [ "${size%kB}" != "$size" ]; then
1235     size=$(round ${size%kB})
1236     size=$(($size/1024+1))
1237   elif [ "${size%MB}" != "$size" ]; then
1238     size=$(round ${size%MB})
1239   elif [ "${size%GB}" != "$size" ]; then
1240     size=$(round ${size%GB})
1241     size=$(($size*1024))
1242   elif [ "${size%[0-9]B}" != "$size" ]; then
1243     size=$(round ${size%B})
1244     size=$(($size/1024/1024+1))
1245   else
1246     log "unrecognized size unit: $size"
1247     return 1
1248   fi
1249
1250   echo $size
1251 }
1252
1253 round() {
1254   local num
1255   num=$1
1256
1257   if [ "${num%.*}" != "$num" ]; then
1258     num=${num%.*}
1259     num=$(($num+1))
1260   fi
1261
1262   echo $num
1263 }
1264
1265 # make sure carnet-upgrade doesn't get upgraded until the end
1266 hold_carnet_upgrade() {
1267   echo carnet-upgrade hold | dpkg --set-selections
1268 }
1269
1270 release_carnet_upgrade() {
1271   echo carnet-upgrade install | dpkg --set-selections
1272 }