addvpubaddr.in 2.63 KB
Newer Older
1 2 3 4
#!/usr/bin/perl -w

#
# Copyright (c) 2012 University of Utah and the Flux Group.
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
# 
# {{{EMULAB-LICENSE
# 
# This file is part of the Emulab network testbed software.
# 
# This file is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or (at
# your option) any later version.
# 
# This file is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public
# License for more details.
# 
# You should have received a copy of the GNU Affero General Public License
# along with this file.  If not, see <http://www.gnu.org/licenses/>.
# 
# }}}
24 25 26 27 28 29 30 31 32 33
#

use English;
use strict;
use Socket;

#
# Configure variables
#
my $TB			= "@prefix@";
34
my $control_netmask     = "@CONTROL_NETMASK@";
35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57

#
# Turn off line buffering on output
#
$| = 1;

#
# Untaint the path
# 
$ENV{'PATH'} = "$TB/bin:$TB/sbin:/bin:/usr/bin:/sbin:/usr/sbin";
delete @ENV{'IFS', 'CDPATH', 'ENV', 'BASH_ENV'};

#
# Testbed Support libraries
#
use lib "@prefix@/lib";
use libdb;
use libtestbed;

#
# Handle command-line options.
#
sub usage() {
58 59 60 61 62 63
    print STDERR "Usage: $0 [-d] aaa.bbb.ccc.ddd eee.fff.ggg.hhh [mmm.nnn.ooo.ppp]\n" .
	"    where aaa.bbb.ccc.ddd is the first addresses to add,\n" .
	"          eee.fff.ggg.hhh is the last address to add,\n" .
	"          mmm.nnn.ooo.ppp is the optional netmask to apply\n" .
	"              (default is control network netmask).\n" .
	"    if -d is specified, the addresses in range are deleted.\n";
64 65 66
    exit( 1 );
}

67
my $delete;
68

69 70 71
if( $ARGV[ 0 ] eq "-d" ) {
    $delete = 1;
    shift;
72 73
}

74 75 76 77 78 79 80 81
usage() unless @ARGV == 2 || @ARGV == 3;

my $first = unpack( "N", inet_aton( $ARGV[ 0 ] ) );
my $last = unpack( "N", inet_aton( $ARGV[ 1 ] ) );
my $netmask;

if( @ARGV == 2 ) {
    $netmask = $control_netmask;
82
} else {
83 84
    $netmask = $ARGV[ 2 ];
    $netmask =~ /^[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+$/ or usage();
85 86 87 88 89 90
}

my $i;

for( $i = $first; $i <= $last; $i++ ) {
    my $addr = inet_ntoa( pack( "N", $i ) );
91 92 93 94
    my $result;

    if( $delete ) {
	$result = DBQueryWarn( "DELETE FROM virt_node_public_addr WHERE " .
95 96
			       "IP='$addr' AND node_id IS NULL AND " .
			       "pid IS NULL AND eid IS NULL" );
97 98 99 100 101 102 103 104 105
    } else {
	$result = DBQueryWarn( "INSERT INTO virt_node_public_addr SET " .
			       "IP='$addr', mask='$netmask'" );
    }

    if( !defined( $result ) or $result->affectedrows < 1 ) {
	print STDERR "warning: unable to " . ( $delete ? "delete" : "add" ) .
	    " address $addr\n";
    }
106 107 108
}

exit( 0 );