ddde558e01a1a97173c9d99daeb4f4d96316d487
[carnet-upgrade.git] / src / functions.sh
1 # last update: 2005-07-03
2 # Ivan 'ico' Rako <irako@srce.hr>
3 # Zoran 'jelly' Dzelajlija <jelly@srce.hr>
4
5 notice () {
6   dialog --backtitle "$title" --msgbox "$*" 18 75
7 }
8
9 first_warning () {
10   dialog --backtitle "$title" --yesno "$msg_first_warning" 18 75
11 }
12
13 pkg() {
14   local version
15   dpkg -l "$1" 2> /dev/null | egrep -q "^.i" || return 1
16   [ -z "$2" ] && return 0
17   version=$(dpkg -s "$1" | awk '/^Version:/ {print $2}')
18   dpkg --compare-versions "$version" "$2" "$3" 2>&1 > /dev/null
19 }
20
21 log() {
22   logfile=${logfile:=/var/log/carnet-upgrade.log}
23   touch $logfile
24   echo "$(date +'%Y-%m-%d %H:%M:%S') $*" >> $logfile
25   echo "CN: $*"
26 }
27
28 # find first free uid/gid in range
29 # find_id passwd 100 999
30 find_id() {
31   local db first last ids
32   db=$1
33   first=$2
34   last=$3
35   ids=$(getent $db | awk -F: "\$3 >= $first && \$3 <= $last {print \$3}")
36   for i in $(seq $first $last)
37   do
38     if ! echo $ids |grep -q $i; then
39       echo $i
40       return 0
41     fi
42   done
43   return 1
44 }
45
46 # safe in-place s///
47 check_and_sed() {
48   [ "$DEBIAN_SCRIPT_DEBUG" ] && set -vx
49   local s sedcmd ret i
50   s="$1"
51   shift
52   sedcmd="$1"
53   shift
54   ret=2
55   for i in $*
56   do
57     [ -e "$i" ]        || continue
58     egrep -q "$s" "$i" || continue
59     [ -h "$i" ]        && i=$(readlink -f "$i")
60     sed "$sedcmd" "$i" > "$i.dpkg-tmp"
61     chown --reference "$i" "$i.dpkg-tmp"
62     chmod --reference "$i" "$i.dpkg-tmp"
63     mv "$i.dpkg-tmp" "$i"
64     ret=0
65   done
66   if [ "$ret" -eq 0 ]; then
67     log "Done \"$sedcmd\" to $*"
68   fi
69   return $ret
70 }
71
72 #
73 # Update uid for user from reserved system range (0-99) to dynamic system
74 # range (100-999).  Optionally update ownerships of given directories.
75 # $0 user [directory ...]
76 #
77 check_and_update_ugid() {
78   local user newgid newuid
79   user=$1
80   if [ "$(getent passwd $user | awk -F: '$3 >= 100 {print "ok"; exit 0}')" ]; then
81     return 0
82   fi
83   shift
84   newgid=$(find_id group 100 999)
85   newuid=$(find_id passwd 100 999)
86   # other directories/files
87   chown -R $newuid:$newgid $*
88   groupmod -g $newgid $user
89   usermod -u $newuid -g $newgid $user
90   log "Fixed $user user uid/gid."
91 }
92
93 pkgadd () {
94   [ -n "$*" ] || return 0
95   log "Installing $* ..."
96   apt-get -y$s install $* || {
97     dpkg --configure -a
98     apt-get -yf install
99     apt-get -y$s install $*
100   }
101 }
102
103 pkgrm () {
104   [ -n "$*" ] || return 0
105   log "Removing $* ..."
106   apt-get -y$s --purge remove $* || {
107     dpkg --configure -a
108     apt-get -yf install
109     apt-get -y$s --purge remove $*
110   }
111 }
112
113 update() {
114   log "Running apt-get update ..."
115   apt-get -y$s update || {
116     apt-get -yf install
117     apt-get -y$s update
118   }
119   apt-get autoclean
120   apt-get -dy$s dist-upgrade
121 }
122
123 dist_upgrade () {
124   log "Running dist-upgrade ..."
125   apt-get -y$s dist-upgrade || {
126     dpkg --configure -a
127     apt-get -y$s dist-upgrade
128   }
129 }
130
131 remove_bloat() {
132   local to_remove add_this i j n text to_show cmd tmpfile installed
133   local update_selections pkgs_to_deselect oldifs
134   text="$msg_remove_bloat"
135
136   [ "$DEBUG" ] && set -x
137   n=0
138   for i in $(echo "$bloats" | sed 's/           */      /g'| awk -F'    ' '{print $2}')
139   do
140     if pkg $i; then
141       installed="$installed $i"
142       add_this="$(echo "$bloats" | egrep "      .*\b${i}\b" | sed 's/           */      /g'| awk -F'    ' '{print $1}')"
143       if ! echo "$to_show" | grep -q "$add_this"; then
144         to_show="$to_show \"$add_this\" \"\" on"
145         n=$(($n+1))
146       fi
147     fi
148   done
149   [ -z "$to_show" ] && return 0
150   tmpfile=$(mktemp /var/lib/carnet-upgrade/dialog-tmp.XXXXXX)
151   cmd="dialog --nocancel --backtitle \""$title"\" --checklist \""$text"\" 20 75 $n $to_show"
152   [ "$DEBUG" ] && set +x
153     eval $cmd 2> $tmpfile
154   [ "$DEBUG" ] && set -x
155   to_remove="$(cat $tmpfile | sed 's/\"\ \"/    /g; s/\"\ *//g; s/\\//g')"
156   [ "$DEBUG" ] || rm -f $tmpfile 
157   [ "$DEBUG" ] && echo "DEBUG: to_remove=$to_remove" 1>&2
158   oldifs="$IFS"
159   IFS=" " # tab
160   for i in $(echo "$to_remove")
161   do
162     IFS="$oldifs"
163     for j in $(echo "$bloats" | grep "^$i" | sed 's/            */      /g'| awk -F'    ' '{print $2}')
164     do
165       # hopefully this will tell apt not to reinstall them at upgrade time
166       update_selections="${update_selections}
167         $j      deinstall"
168       pkgs_to_deselect="$pkgs_to_deselect $j"
169       # only remove those that really are installed
170       if echo "$installed" | grep -q "$j"; then
171         pkgs_to_remove="$pkgs_to_remove $j"
172       fi
173     done
174   done
175   [ "$DEBUG" ] && echo "DEBUG: pkgs_to_remove=$pkgs_to_remove" 1>&2
176   # Finally, remove those
177   if [ -n "$update_selections" ]; then
178     echo "$update_selections" | dpkg --set-selections
179     log "Deselected:${pkgs_to_deselect}"
180   fi
181   eval pkgrm $pkgs_to_remove
182 }
183
184 post_upgrade() {
185   touch "$post_upgrade_queuetab"
186   if ! grep -q "true    $*" "$post_upgrade_queuetab"; then
187     echo "true  $*" >> "$post_upgrade_queuetab"
188   fi
189 }
190
191 run_post_upgrade() {
192   if [ -f "$post_upgrade_queuetab" ]; then
193     run_actions post_upgrade "$post_upgrade_queuetab"
194     mv "$post_upgrade_queuetab" "${post_upgrade_queuetab}.done.$(date '+%Y-%m-%d.%H:%M:%S')"
195   fi      
196 }
197
198 ###
199
200 run_actions() {
201   local line lineno lineno2 f n name action testmsg
202   [ "$DEBUG" ] && set -vx
203   name="$1"
204   f="$2"
205   [ -f "$f" ] ||   return 2 # no such file
206   n=$(wc -l "$f" | awk '{print $1}')
207   [ "$n" -lt 1 ] && return 1 # no lines to process
208   log "run_actions ($name) started, processing $n lines from \"$f\"."
209   log "-----"
210   lineno=0
211   lineno2=0
212   while [ "$lineno" -lt "$n" ]
213   do
214     lineno=$(($lineno+1))
215     lineno2=$lineno
216     line=$(head -$lineno "$f" | tail -1)
217     if echo $line | grep -q '^#'; then  # comment
218       continue
219     fi
220     while echo "$line" | egrep -q '\\$' # continuation
221     do
222       lineno2=$(($lineno2+1))
223       line=$(echo -n "$line"| sed 's/\\$//'; head -$lineno2 "$f" | tail -1)
224     done
225     test=$(echo "$line" | sed 's/               */      /g'| awk -F'    ' '{print $1}')
226     action=$(echo "$line" | sed 's/             */      /g'| awk -F'    ' '{print $2}')
227     testmsg=""
228     [ "$test" != "true" ] && testmsg="if '$test'"
229     if eval $test; then
230       log "$(printf 'action %02d' $lineno) starting: '${action}' $testmsg "
231       eval $action
232       log "$(printf 'action %02d' $lineno) finished: '${action}' $testmsg"
233     else
234       log "$(printf 'action %02d' $lineno) skipping: '${action}' $testmsg"
235     fi
236     [ "$lineno2" -gt "$lineno" ] && lineno=$lineno2
237   done
238   log "run_actions ($1) finished"
239   log "-----"
240 }
241
242 # funkcija vraca popis paketa koje treba upgradeati
243 what_upgrade () {
244   local IFSOLD name version package
245
246   for package in `echo "$needed_versions" | awk '{print $1}'| grep -v ^$`; do
247     IFSOLD="$IFS"
248     IFS="       " # tab
249     read name version <<-EOPTS
250         $(echo "$needed_versions" | sed 's/             */      /g' | grep ^$package)
251         EOPTS
252     IFS="$IFSOLD"
253     if pkg "$name" lt "$version"; then
254         echo -n " $name"
255     fi
256   done
257 }
258
259 # funkcija za ispis paketa koje treba upgradeati
260 check_upgrade () {
261   local upgrade
262   upgrade="`what_upgrade`"
263
264   if [ "$upgrade" ]; then
265     notice "
266 Prije prelaska na Debian Etch potrebno je napraviti upgrade ovih paketa:
267   $upgrade
268
269 Pokrenite:
270   'apt-get dist-upgrade'
271 ili:
272   'apt-get install${upgrade}'
273   
274 Nakon toga ponovno pozovite ovu skriptu."
275     log "carnet-upgrade aborted, upgrade of$upgrade needed."
276     exit 1
277   fi
278 }
279
280 upgrade () {
281   local to_install packages
282   case "$1" in
283     first)
284       packages="$packages_first"
285     ;;
286     second)
287       packages="$packages_second"
288     ;;
289     *)
290     return 1
291   esac
292
293   for i in $packages
294   do
295     pkg $i && to_install="$to_install $i"
296   done
297
298   if [ -n "$to_install" ];then
299     eval pkgadd $to_install
300   else
301     return 0
302   fi
303 }
304
305 # funkcija utrpa ispravan sources.list
306 create_sources_list () {
307     [ -f /etc/apt/sources.list -a ! -f /etc/apt/sources.list.cn4-upgrade ] && mv /etc/apt/sources.list /etc/apt/sources.list.cn4-upgrade
308     cp /usr/share/carnet-upgrade/files/etc/apt/sources.list /etc/apt
309     notice "Novi sadrzaj datoteke /etc/apt/sources.list:
310
311 `cat /etc/apt/sources.list`"
312 }
313
314
315 # funkcija koja brise grupu proc
316 remove_group_proc () {
317   if getent group proc > /dev/null; then
318     groupdel proc || true
319     log "groupdel proc (temporarily)"
320     post_upgrade add_group_proc
321   fi
322 }
323
324 add_group_proc () {
325   if ! getent group proc > /dev/null; then
326     # grsec uses gid 99 for /proc files
327     groupadd -g 99 proc 
328     log "groupadd -g 99 proc"
329
330     # update oidentd so it uses the proc group
331     if [ -x /etc/init.d/oidentd -a -f /etc/default/oidentd ]; then
332         if check_and_sed '^OIDENT_GROUP=oident' \
333                          's/^\(OIDENT_GROUP\)=.*/\1=proc/' \
334                          /etc/default/oidentd;
335         then
336             /etc/init.d/oidentd restart
337         fi
338     fi
339   fi
340 }
341
342 # backup_slapd_db
343 backup_slapd_db () {
344   local ldap_backup=/var/backups/slapcat.cn4-upgrade
345
346   if [ ! -f $ldap_backup ]; then
347     [ -x /etc/init.d/freeradius ] && /etc/init.d/freeradius stop || true
348
349     if [ -x /etc/init.d/slapd -a -d /var/lib/ldap ]; then
350       /etc/init.d/slapd stop || true
351       if slapcat -l $ldap_backup; then
352         log "$ldap_backup created"
353         notice "Napravljen backup slapd baze u $ldap_backup."
354       fi
355       /etc/init.d/slapd start || true
356     fi
357
358     [ -x /etc/init.d/freeradius ] && /etc/init.d/freeradius start || true
359   else
360     log "slapd backup already at $ldap_backup, skipping"
361     notice "Backup slapd baze u $ldap_backup vec postoji!"
362   fi
363 }
364
365 # backup etc direktorija
366 backup_etc_dir () {
367   local etc_backup=/var/backups/etc.cn4-upgrade.tar.gz
368
369   if [ ! -f $etc_backup ]; then
370     if tar cfz $etc_backup /etc 2> /dev/null; then
371       log "$etc_backup created"
372       notice "Napravljen backup /etc direktorija u $etc_backup."
373     fi
374   else
375     log "/etc backup already present in $etc_backup, skipping."
376     notice "Backup /etc direktorija u $etc_backup vec postoji!"
377   fi
378 }
379
380 # fixaj /etc/default/raid2
381 fix_etc_default_raid2 () {
382   if [ -f /etc/default/raid2 ]; then
383     sed "s/^AUTOSTART.*/AUTOSTART=true/g" /etc/default/raid2 > /etc/default/raid2.cn4-upgrade.$$
384     chown --reference=/etc/default/raid2 /etc/default/raid2.cn4-upgrade.$$
385     chmod --reference=/etc/default/raid2 /etc/default/raid2.cn4-upgrade.$$
386
387     if ! cmp -s /etc/default/raid2.cn4-upgrade.$$ /etc/default/raid2; then
388       log "Fixed /etc/default/raid2 AUTOSTART option."
389       mv /etc/default/raid2.cn4-upgrade.$$ /etc/default/raid2
390     else
391       rm /etc/default/raid2.cn4-upgrade.$$
392     fi
393   fi
394 }
395
396 # restore distribution config file
397 restore_config () {
398   local file file_backup
399
400   for file in "$@"; do
401     file_backup=$file.cn4-upgrade
402     if [ -e $file -a ! -e $file_backup ]; then
403       # backup file
404       mv $file $file_backup
405
406       # install original
407       log Restoring config file $file
408       cp /usr/share/carnet-upgrade/files/$file $file
409     fi
410   done
411 }
412
413 # restore modified config to their package defaults
414 # so the upgrade doesn't complain so much
415 restore_configs () {
416   restore_config /etc/bind/named.conf.options
417   restore_config /etc/default/ntpdate
418   restore_config /etc/default/oidentd
419   restore_config /etc/default/postgrey
420   restore_config /etc/default/saslauthd
421   restore_config /etc/default/slapd
422   restore_config /etc/dovecot/dovecot.conf
423   restore_config /etc/init.d/mysql
424   restore_config /etc/init.d/slapd
425   restore_config /etc/issue
426   restore_config /etc/issue.net
427   restore_config /etc/logrotate.d/mysql-server
428   restore_config /etc/mysql/my.cnf
429   restore_config /etc/ntp.conf
430   restore_config /etc/pam.d/login
431   restore_config /etc/php4/apache/php.ini
432   restore_config /etc/php4/cgi/php.ini
433   restore_config /etc/php4/cli/php.ini
434   restore_config /etc/postgrey/whitelist_clients
435   restore_config /etc/security/limits.conf
436   restore_config /etc/squirrelmail/apache.conf
437   restore_config /etc/sysctl.conf
438   restore_config /etc/vsftpd.conf
439   restore_config /etc/xinetd.conf
440
441   # orphaned config file - no owner
442   rm -f /etc/logcheck/ignore.d.server/imap
443
444   # aide switched to ucf, move old configs aside
445   for file in /etc/aide/aide.conf /etc/cron.daily/aide /etc/default/aide; do
446       [ ! -e "$file.cn4-upgrade" ] && mv "$file" "$file.cn4-upgrade"
447       rm -f "$file"
448   done
449 }
450
451 # make a silent installation of carnet and srce keyrings
452 install_keyrings () {
453   pkgadd carnet-keyring srce-keyring
454   pkgadd dpkg apt debconf
455
456   dpkg-reconfigure carnet-keyring srce-keyring
457   update
458 }
459
460 # make a silent upgrade to new libc6
461 upgrade_libc () {
462   DEBIAN_FRONTEND=noninteractive pkgadd libc6
463 }
464
465 # upgrade apache -> apache2
466 upgrade_apache () {
467
468   # bugfix for apache2-cn postinst
469   mkdir -p /etc/apache2
470   touch /etc/apache2/httpd.conf
471
472   # temporary remove packages conflicting with apache2
473   delpkg=""
474   for p in aosi-aai aosi-www-aai \
475            php4-cn php4-odbc php4-xslt \
476            squirrelmail-cn; do
477      pkg p && delpkg="$delpkg $p"
478   done
479   eval pkgrm apache $delpkg
480
481   # install new packages
482   eval pkgadd apache2-cn apache2-mpm-prefork \
483               php4-cn libapache2-mod-php4 \
484               $delpkg
485 }
486
487 upgrade_amavis () {
488   # remove init script diversion
489   if [ -L /etc/init.d/amavis -a -f /etc/init.d/amavis.amavisd-new ]; then
490     rm -f /etc/init.d/amavis
491     dpkg-divert --quiet --remove /etc/init.d/amavis
492   fi
493
494   # move old config aside
495   local conf=/etc/amavis/amavisd.conf
496   if [ -e $conf -a ! -e $conf.cn4-upgrade ]; then
497     mv $conf $conf.cn4-upgrade
498   fi
499
500   # install new packages
501   pkgadd amavisd-cn amavisd-new
502
503   # fix new packages
504   check_and_sed '^clamd.*5.clamav.log$' \
505                 's/^\(clamd.*\)5.clamav.log$/\14\tsocket/g' \
506                 /etc/init.d/amavisd-cn || true
507
508   # start new packages
509   /etc/init.d/amavis restart
510 }
511
512 # fix openldap-aai postinst user handling
513 upgrade_openldap () {
514   #pkgrm openldap-cn # deprecated
515   apt-get -y$s install openldap-aai || true # postinst fails
516   chown -R openldap:openldap /var/lib/ldap  # fix slapd
517   /etc/init.d/slapd restart
518   dpkg --configure -a # try postinst again
519 }
520
521 # fix /etc/dpkg/dpkg.cfg
522 comment_force_overwrite () {
523   if [ -f /etc/dpkg/dpkg.cfg ]; then
524     if check_and_sed '^force-overwrite' \
525                   's,\(^force-overwrite\),#\1,' \
526                   /etc/dpkg/dpkg.cfg; then
527       notice "$msg_comment_force_overwrite"
528     fi
529   fi
530 }
531
532 fix_issue () {
533   cat > /etc/issue <<EOF
534 Debian GNU/Linux 4.0 (CARNet Debian 4.0) \\n \\l
535 EOF
536
537   cat > /etc/issue.net <<EOF
538 Debian GNU/Linux 4.0 (CARNet Debian 4.0) %h
539 EOF
540 }
541
542 check_disk_space() {
543   local available_disk_space
544   available_disk_space=$(df -m /var|tail -1|awk '{print $4}')
545   if [ $available_disk_space -lt $min_var_space ]; then
546     log "carnet-upgrade aborted: not enough space on /var (need ${min_var_space}MB)"
547     notice "Premalo mjesta na /var particiji. Potrebno je imati barem ${min_var_space}MB."
548     exit 1
549   fi
550 }