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