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