From 49c7681c58dd6dbd85afcdb9884a5dc8ed766192 Mon Sep 17 00:00:00 2001
From: Jay Berkenbilt <ejb@ql.org>
Date: Mon, 11 Mar 2013 12:34:31 -0400
Subject: Windows install: check DLL type

When copying dlls, make sure to only consider DLLs whose type matches
the type of what is loading them.
---
 copy_dlls | 42 +++++++++++++++++++++++++++++++++++++++---
 1 file changed, 39 insertions(+), 3 deletions(-)

(limited to 'copy_dlls')

diff --git a/copy_dlls b/copy_dlls
index e93f1613..c85e44b9 100755
--- a/copy_dlls
+++ b/copy_dlls
@@ -12,6 +12,7 @@ my ($file, $destdir, $objdump) = @ARGV;
 my $filedir = dirname($file);
 
 my %dlls = ();
+my $format = undef;
 open(O, "$objdump -p $file|") or die "$whoami: can't run objdump\n";
 while (<O>)
 {
@@ -22,11 +23,21 @@ while (<O>)
 	next if $dll =~ m/^(kernel32|user32|msvcrt)\.dll$/;
 	$dlls{$dll} = 1;
     }
+    elsif (m/^Magic.*\((PE.+?)\)/)
+    {
+        $format = $1;
+    }
 }
 close(O);
+if (! defined $format)
+{
+    die "$whoami: can't determine format of $file\n";
+}
 
-# Search the file's directory, the current directory, and the path for
-# dlls since that's what Windows does.
+# Search the directories named in the file's manifest (if present),
+# the file's directory, the current directory, and the path for dlls
+# since that's what Windows does.  Be sure to only capture compatible
+# DLLs.
 my $sep = ($^O eq 'MSWin32' ? ';' : ':');
 my @path = ($filedir, '.', split($sep, $ENV{'PATH'}));
 if (-f "$file.manifest")
@@ -41,7 +52,7 @@ foreach my $dll (sort keys %dlls)
     my $found = 0;
     foreach my $dir (@path)
     {
-	if (-f "$dir/$dll")
+	if ((-f "$dir/$dll") && is_format("$dir/$dll", $format))
 	{
 	    push(@final, "$dir/$dll");
 	    $found = 1;
@@ -68,6 +79,31 @@ foreach my $f (@final)
 	die "$whoami: copy $f to $destdir failed\n";
 }
 
+sub is_format
+{
+    my ($file, $format) = @_;
+    $file =~ s,\\,/,g;
+    # Special case: msvc*.dll seem to be able to behave both as 32-bit
+    # and 64-bit DLLs.  Either that, or this logic is wrong for those
+    # DLLs and it doesn't matter because they're already installed on
+    # my test system (which doesn't have msvc installed on it).
+    if ($file =~ m,/msvc,i)
+    {
+        return 1;
+    }
+    my $result = 0;
+    my $file_format = `file $file`;
+    print "$file $format $file_format\n";
+    if ($? == 0)
+    {
+        if ($file_format =~ m/\Q${format}\E executable/)
+        {
+            $result = 1;
+        }
+    }
+    $result;
+}
+
 sub get_manifest_dirs
 {
     # Find all system directories in which to search for DLLs based on
-- 
cgit v1.2.3-54-g00ecf