r102:

[carnet-tools-cn.git] / functions.sh

diff --git a/functions.sh b/functions.sh
index 47ad62f..2ab04ff 100644 (file)
--- a/functions.sh
+++ b/functions.sh
@@ -88,9 +88,19 @@ cp_get_ifdefault() {
 
   RET=""
   case $1 in
-    dev)  RET=`route -n | awk '/^0.0.0.0/{print $8}'` ;;
-    addr) RET=`route -n | awk '/^0.0.0.0/{print $2}'` ;;
-    *)    echo "cp_get_ifdefault: Argument required (dev or addr)" >&2 ;;
+    dev)
+      RET=`route -n | awk 'BEGIN {m = 2^32}
+                           /^0\.0\.0\.0[ \t]/ {if ($5 < m) {r = $8; m = $5}}
+                             END {print r}'`
+      ;;
+    addr)
+      RET=`route -n | awk 'BEGIN {m = 2^32}
+                           /^0\.0\.0\.0[ \t]/ {if ($5 < m) {r = $2; m = $5}}
+                                    END {print r}'`
+      ;;
+    *)
+      echo "cp_get_ifdefault: Argument required (dev or addr)" >&2
+      ;;
   esac
 
   [ -z "$RET" -a "$1" ] && echo "cp_get_ifdefault: No default route" >&2
@@ -173,6 +183,7 @@ cp_backup_conffile() {
     -p)
       shift
       opt_p=1
+      ;;
     -d)
       shift
       opt_d=1
@@ -187,11 +198,13 @@ cp_backup_conffile() {
     -n)
       shift
       ext=
+      ;;
     esac
   done
   if [ "$opt_d" ]; then
     if [ ! -e "$dir" -a "$opt_p" ]; then
       mkdir "$dir"
+    fi
     if [ $? -ne 0 ]; then
       echo "cp_backup_conffile: Error creating backup directory $dir" 1>&2
       return 3
@@ -467,3 +480,41 @@ cp_get_ldap_realm() {
   CP_ECHO_RETURN="$echo_return"
 }
 
+_cp_customize() {
+  [ "$CP_SCRIPT_DEBUG" ] && set -vx
+  # < input functions > output_functions
+  # new-prefix list-of-functions
+  perl -e '
+    $prefix = shift; %keep = (); $output = "";
+    $keep{$_} = 1 for @ARGV;
+    # add dependent functions
+    %deps = (
+      get_ifaddr       => [ "get_ifdefault" ],
+      get_netaddr      => [ "get_ifdefault", "get_ifaddr", "get_ifmask" ],
+      check_and_backup => [ "backup_conffile" ],
+      check_and_sed    => [ "mv" ],
+      get_ldap_realm   => [ "get_ldap_suffix" ],
+    );
+    foreach my $f (keys(%keep)) {
+      if (exists($deps{$f})) {
+        $keep{$_} = 1 for (@{$deps{$f}});
+      }
+    }
+    while (<STDIN>) {
+      if (/^_?cp_(\w+) *\(\) *\{/) {
+        if (! exists($keep{$1})) {
+          while (<STDIN>) { last if /^\}/; } # skip it
+          next;
+        } else {
+          $output .= "\n";
+        }
+      }
+      next if /^\s*(\#|$)/;
+      $output .= $_;
+    }
+    $output =~ s/\b(_?)cp_/$1${prefix}_/sg;
+    $prefix = uc($prefix);
+    $output =~ s/\b(_?)CP_/$1${prefix}_/sg;
+    print $output;
+  ' -- $*
+}