181 lines
3.6 KiB
Perl
181 lines
3.6 KiB
Perl
|
#---------------------------------------------------------------------
|
||
|
|
package Updcat;
|
||
|
|
#
|
||
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
||
|
|
#
|
||
|
|
# Version: 1.00 Update catalogs by adding and/or removing hashes
|
||
|
|
#---------------------------------------------------------------------
|
||
|
|
use strict;
|
||
|
|
use Carp;
|
||
|
|
use lib $ENV{RAZZLETOOLPATH}. "\\sp";
|
||
|
|
use File::Temp qw(tempfile);
|
||
|
|
use vars qw($VERSION);
|
||
|
|
use Data::Dumper;
|
||
|
|
$VERSION = '1.00';
|
||
|
|
|
||
|
|
my $LastError;
|
||
|
|
|
||
|
|
sub GetLastError
|
||
|
|
{
|
||
|
|
return $LastError;
|
||
|
|
}
|
||
|
|
|
||
|
|
#
|
||
|
|
# Parameters:
|
||
|
|
#
|
||
|
|
# catalog - path to catalog to update
|
||
|
|
#
|
||
|
|
# remove_hashes -
|
||
|
|
# option 1: an array of hashes
|
||
|
|
# option 2: path of a file containing hashes
|
||
|
|
# option 3: undef, no hashes to remove
|
||
|
|
#
|
||
|
|
# add_file_signatures -
|
||
|
|
# option 1: an array of file paths
|
||
|
|
# option 2: path of file containing file paths
|
||
|
|
# option 3: undef, no file signatures to add
|
||
|
|
#
|
||
|
|
sub Update
|
||
|
|
{
|
||
|
|
my $catalog = shift;
|
||
|
|
my $remove_hashes = shift;
|
||
|
|
my $add_file_signatures = shift;
|
||
|
|
my $add_drm_list = shift;
|
||
|
|
|
||
|
|
my $command_line = "";
|
||
|
|
my ($fn1, $fn2);
|
||
|
|
|
||
|
|
if ( 'ARRAY' eq ref $remove_hashes &&
|
||
|
|
@$remove_hashes )
|
||
|
|
{
|
||
|
|
my $fh;
|
||
|
|
($fh, $fn1) = tempfile();
|
||
|
|
if ( $! )
|
||
|
|
{
|
||
|
|
$LastError = "Failed opening temporary file ($!)";
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
print $fh "$_\n" foreach (@$remove_hashes);
|
||
|
|
close $fh;
|
||
|
|
|
||
|
|
$command_line .= " -d @". $fn1;
|
||
|
|
}
|
||
|
|
elsif ( 'SCALAR' eq ref $remove_hashes &&
|
||
|
|
defined $remove_hashes )
|
||
|
|
{
|
||
|
|
$command_line .= " -d @". $$remove_hashes;
|
||
|
|
}
|
||
|
|
elsif ( defined $remove_hashes )
|
||
|
|
{
|
||
|
|
$command_line .= " -d @". $remove_hashes;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( 'ARRAY' eq ref $add_file_signatures &&
|
||
|
|
@$add_file_signatures )
|
||
|
|
{
|
||
|
|
my $fh;
|
||
|
|
($fh, $fn2) = tempfile();
|
||
|
|
if ( $! )
|
||
|
|
{
|
||
|
|
$LastError = "Failed opening temporary file ($!)";
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
print $fh "$_\n" foreach (@$add_file_signatures);
|
||
|
|
close $fh;
|
||
|
|
|
||
|
|
$command_line .= " -a @". $fn2;
|
||
|
|
}
|
||
|
|
elsif ( 'SCALAR' eq ref $add_file_signatures &&
|
||
|
|
defined $add_file_signatures )
|
||
|
|
{
|
||
|
|
$command_line .= " -a @". $$add_file_signatures;
|
||
|
|
}
|
||
|
|
elsif ( defined $add_file_signatures )
|
||
|
|
{
|
||
|
|
$command_line .= " -a @". $add_file_signatures;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( !$command_line )
|
||
|
|
{
|
||
|
|
carp "Invalid (empty) arguments specified.";
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
$command_line = "updcat.exe $catalog $command_line";
|
||
|
|
|
||
|
|
my $retval;
|
||
|
|
sys($command_line) || return;
|
||
|
|
|
||
|
|
if ('ARRAY' eq ref $add_drm_list && @$add_drm_list ){
|
||
|
|
addDrmAttrib($add_drm_list ,$catalog) ||return ;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
unlink $fn1 if $fn1;
|
||
|
|
unlink $fn2 if $fn2;
|
||
|
|
|
||
|
|
return 1 ;
|
||
|
|
}
|
||
|
|
|
||
|
|
sub addDrmAttrib {
|
||
|
|
my $drm_file = shift;
|
||
|
|
my $catalog = shift;
|
||
|
|
my $cmd_line;
|
||
|
|
foreach ( @$drm_file) {
|
||
|
|
$cmd_line = "updcat $catalog -attr $$_[0] DRMLevel $$_[1]";
|
||
|
|
sys("$cmd_line") || return;
|
||
|
|
|
||
|
|
}
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
sub sys {
|
||
|
|
my $cmd_line = shift;
|
||
|
|
my $retval;
|
||
|
|
|
||
|
|
print("Running $cmd_line \n");
|
||
|
|
system($cmd_line);
|
||
|
|
if ( $! )
|
||
|
|
{
|
||
|
|
$LastError = "Unable to execute updcat.exe ($!)";
|
||
|
|
return ;
|
||
|
|
|
||
|
|
}
|
||
|
|
elsif ( $? )
|
||
|
|
{
|
||
|
|
$LastError = "'$cmd_line' failed (". ($?>>8). ")";
|
||
|
|
return ;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
1;
|
||
|
|
|
||
|
|
__END__
|
||
|
|
|
||
|
|
=head1 NAME
|
||
|
|
|
||
|
|
<<mypackage>> - <<short description>>
|
||
|
|
|
||
|
|
=head1 SYNOPSIS
|
||
|
|
|
||
|
|
<<A short code example>>
|
||
|
|
|
||
|
|
=head1 DESCRIPTION
|
||
|
|
|
||
|
|
<<The purpose and functionality of this module>>
|
||
|
|
|
||
|
|
=head1 AUTHOR
|
||
|
|
|
||
|
|
<<your alias>>
|
||
|
|
|
||
|
|
=head1 COPYRIGHT
|
||
|
|
|
||
|
|
Copyright (c) Microsoft Corporation. All rights reserved.
|
||
|
|
|
||
|
|
=cut
|