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

#
# Copyright (c) 2010 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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55
#

use English;
use strict;

#
# Configure variables
#
my $TB			= "@prefix@";

#
# 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() {
56
    print STDERR "Usage: $0 <owner-uid> <filename>\n";
57 58 59
    exit( 1 );
}

60 61
usage() unless @ARGV == 2;
my ( $uid, $filename ) = @ARGV;
62 63 64
#
# Must taint check!
#
65 66 67 68 69 70 71 72
if ($uid =~ /^([-\w #%&*+,.\/:;=?@\[\\\]^{|}]+)$/) {
    $uid = $1;
}
else {
    print STDERR "Bad character in uid\n";
    exit( 1 );
}

73 74 75 76 77 78 79 80 81 82 83 84 85 86
if ($filename =~ /^([-\w #%&*+,.\/:;=?@\[\\\]^{|}]+)$/) {
    $filename = $1;
}
else {
    print STDERR "Bad character in filename\n";
    exit( 1 );
}

# We could use MySQL's UUID() function, but if we call it in the INSERT it
# becomes a pain to retrieve it.  So we do the job ourselves.
my $uuid = `@UUIDGEN@`;
chomp $uuid;

my $result = DBQueryWarn( "INSERT INTO blobs SET uuid='$uuid', " .
87
			  "filename='$filename', owner_uid='$uid';" );
88 89 90 91 92 93 94 95
unless( $result ) {
    print STDERR "Could not insert record.\n";
    exit( 1 );
}

print "$uuid\n";

exit( 0 );