Changeset 3267 in MondoRescue

Timestamp:

Apr 29, 2014, 11:52:55 AM (11 years ago)

Author:

Bruno Cornec

Message:

• First fully working version of mr-net-get-config

File:

• branches/3.2/MondoRescue/bin/mr-net-get-config (modified) (6 diffs)

branches/3.2/MondoRescue/bin/mr-net-get-config

@@ -7 +7 @@
 # Get the complete network configuration of the system
 #
-use strict;
+use strict 'vars';
 use IO::Interface::Simple;
 use Net::IPv4Addr qw( :all );;
+use Socket;
 use ProjectBuilder::Base;
+use ProjectBuilder::Version;
+use Data::Dumper;
+use Getopt::Long qw(:config auto_abbrev no_ignore_case);
+use English;
+
 
 =pod
@@ -24 +30 @@
 =head1 SYNOPSIS
 
-mr-net-get-config server-ip:/export /path/to/file
+mr-net-get-config [-v][-h][-f /path/to/file] proto://[user@]server/export|[user@]server:/export
 
 =head1 ARGUMENTS
@@ -30 +36 @@
 =over 4
 
-=item B<server-ip:/export> 
+=item B<proto://[user@]server/export|[user@]server:/export>
 
-This is the IP address of the network server, followed by a ':' and the exported directory
+This is the name of the network server, followed by a ':' and the exported directory if nfs or you have to specify the protocol used (nfs, sshfs or smbfs) followed by the network server and the exported directory (without ':'). An optional user name may be provided in case it's needed.
 
-=item B</path/to/file> 
+=back 
 
-This is the path of the configuration file into which we want to store the network configuration
+=head1 OPTIONS
+
+=over 4
+
+=item B<-v|--verbose>
+
+Increase verbosity
+
+=item B<-h|--help>
+
+Print a brief help message and exits.
+
+=item B<--man>
+
+Prints the manual page and exits.
+
+=item B<-f /path/to/file> 
+
+This is the path of the output file into which we want to store the network configuration.
+By default the command prints on the standard output.
 
 =back
@@ -59 +84 @@
 =cut
 
-my $verbose = undef;
-if ((defined $ARGV[0]) && ($ARGV[0] eq "-v")) {
-    $verbose = 1;
-    shift @ARGV;
+# Global variables
+my ($mrver,$mrrev) = pb_version_init();
+my $appname = "mr-net-get-config";
+my %opts;                   # CLI Options
+
+# Initialize the syntax string
+
+pb_syntax_init("$appname Version $mrver-$mrrev\n");
+
+GetOptions("help|?|h" => \$opts{'h'}, 
+        "man" => \$opts{'man'},
+        "verbose|v+" => \$opts{'v'},
+        "file|f=s" => \$opts{'f'},
+) || pb_syntax(-1,0);
+
+if (defined $opts{'h'}) {
+    pb_syntax(0,1);
+}
+if (defined $opts{'man'}) {
+    pb_syntax(0,2);
+}
+if (defined $opts{'v'}) {
+    $pbdebug = $opts{'v'};
+}
+pb_log_init($pbdebug, $pbLOG);
+
+my $fd;
+if (defined $opts{'f'}) {
+    open(CFG, "> $opts{'f'}") || die "Unable to write into $opts{'f'}: $!";
+    $fd=\*CFG;
+} else {
+    $fd=\*STDOUT;
 }
 
-if ((not defined $ARGV[0]) || (not defined $ARGV[1])) {
-    die "Syntax: mr-net-get-config [proto:]server-ip:/export /path/to/file";
+# Normalize URI
+my $url = $ARGV[0];
+if ($url !~ /:\/\//) {
+    # No Protocol given
+    # Protocol for sharing is NFS by default
+    # Format is without ':' then
+    $url =~ s/://;
+    $url = "nfs://".$url;
 }
 
-open(CFG, ">> $ARGV[1]") || die "Unable to append into $ARGV[1]: $!";
+# Analyze URI
+my ($scheme, $account, $host, $port, $path) = pb_get_uri($url);
 
-my ($netfs_proto,$netfs_server_ip,$netfs_server_mount);
-# Protocol for sharing is NFS by default
-$netfs_proto = "nfs";
-
-my ($scheme, $account, $host, $port, $path) = pb_get_uri($ARGV[0]);
+# If hot is a name transform into an IP
+my $hostip = inet_ntoa(scalar gethostbyname($host || 'localhost'));
 
 # We need to loop on all if to see which one is on the same LAN as the server
@@ -85 +142 @@
     my $netfs_broadcast = $if->broadcast;
     my $netfs_hwaddr = $if->hwaddr;
-    if (defined $verbose) {
-        print "interface = $if\n";
-        print "--\n";
-        print "addr =      ",$netfs_ipaddr,"\n",
-            "broadcast = ",$netfs_broadcast,"\n",
-            "netmask =   ",$netfs_netmask,"\n",
-            "dstaddr =   ",$if->dstaddr,"\n",
-            "hwaddr =    ",$netfs_hwaddr,"\n",
-            "mtu =       ",$if->mtu,"\n",
-            "metric =    ",$if->metric,"\n",
-            "index =     ",$if->index,"\n";
-        print "is running\n"     if $if->is_running;
-        print "is broadcast\n"   if $if->is_broadcast;
-        print "is p-to-p\n"      if $if->is_pt2pt;
-        print "is loopback\n"    if $if->is_loopback;
-        print "is promiscuous\n" if $if->is_promiscuous;
-        print "is multicast\n"   if $if->is_multicast;
-        print "is notrailers\n"  if $if->is_notrailers;
-        print "is noarp\n"       if $if->is_noarp;
-        print "--\n";
-    }
-    if (ipv4_in_network($if->address, $if->netmask, $host)) {
-        print "Netfs server $host is in network $netfs_ipaddr/$netfs_netmask\n";
+    pb_log(2, "interface = $if\n");
+    pb_log(2, "--\n");
+    pb_log(2, "addr =      $netfs_ipaddr\n",
+        "broadcast = $netfs_broadcast\n",
+        "netmask =   $netfs_netmask\n",
+        "dstaddr =   ",$if->dstaddr,"\n",
+        "hwaddr =    $netfs_hwaddr\n",
+        "mtu =       ",$if->mtu,"\n",
+        "metric =    ",$if->metric,"\n",
+        "index =     ",$if->index,"\n");
+    pb_log(2, "is running\n")     if $if->is_running;
+    pb_log(2, "is broadcast\n")   if $if->is_broadcast;
+    pb_log(2, "is p-to-p\n")      if $if->is_pt2pt;
+    pb_log(2, "is loopback\n")    if $if->is_loopback;
+    pb_log(2, "is promiscuous\n") if $if->is_promiscuous;
+    pb_log(2, "is multicast\n")   if $if->is_multicast;
+    pb_log(2, "is notrailers\n")  if $if->is_notrailers;
+    pb_log(2, "is noarp\n")       if $if->is_noarp;
+    pb_log(2, "--\n");
+    if (ipv4_in_network($if->address, $if->netmask, $hostip)) {
+        pb_log(1, "Netfs server $host is in network $netfs_ipaddr/$netfs_netmask\n");
         # So generate conf file
-        print CFG "netfs-dev $if\n";
-        print CFG "netfs-proto $scheme\n";
-        print CFG "netfs-client-ipaddr $netfs_ipaddr\n";
-        print CFG "netfs-client-netmask $netfs_netmask\n";
-        print CFG "netfs-client-broadcast $netfs_broadcast\n";
-        print CFG "netfs-client-hwaddr $netfs_hwaddr\n";
-        print CFG "netfs-server-mount $ARGV[0]\n";
-        print CFG "netfs-server-user $account\n";
-        print CFG "netfs-server-path $path\n";
-        print CFG "netfs-server-ipaddr $host\n";
+        print $fd "netfs-dev $if\n";
+        print $fd "netfs-proto $scheme\n";
+        print $fd "netfs-client-ipaddr $netfs_ipaddr\n";
+        print $fd "netfs-client-netmask $netfs_netmask\n";
+        print $fd "netfs-client-broadcast $netfs_broadcast\n";
+        print $fd "netfs-client-hwaddr $netfs_hwaddr\n";
+        print $fd "netfs-server-mount $ARGV[0]\n";
+        print $fd "netfs-server-user $account\n";
+        print $fd "netfs-server-path $path\n";
+        print $fd "netfs-server-ipaddr $hostip\n";
         open(ROUTE, "/proc/net/route") || die "Unable to read /proc/net/route: $!";
         while (<ROUTE>) {
@@ -125 +180 @@
             next if ($dest !~ /^00000000$/);
             $gw =~ s/([A-z0-9]{2})([A-z0-9]{2})([A-z0-9]{2})([A-z0-9]{2})/hex($4).".".hex($3).".".hex($2).".".hex($1)/eg;
-            print CFG "netfs-client-defgw $gw\n";
+            print $fd "netfs-client-defgw $gw\n";
         }
         close(ROUTE);
-        last if (not defined $verbose);
+        last if ($pbdebug > 1);
     }
 }
-close(CFG);
+close($fd);