fs.win32.pe-parser: copy get_architecture from install.bat

1 files changed, 15 insertions, 0 deletions

diff --git a/src/luarocks/fs/win32/pe-parser.lua b/src/luarocks/fs/win32/pe-parser.lua
index 9cd36ffc..34556812 100644
--- a/src/luarocks/fs/win32/pe-parser.lua
+++ b/src/luarocks/fs/win32/pe-parser.lua
@@ -550,4 +550,19 @@ function M.msvcrt(infile)
   return nil, "No msvcrt found"
 end
 
+function M.get_architecture(program)
+   -- detect processor arch interpreter was compiled for
+   local proc = (M.parse(program) or {}).Machine
+   if not proc then
+      return nil, "Could not detect processor architecture used in "..program
+   end
+   proc = M.const.Machine[proc]  -- collect name from constant value
+   if proc == "IMAGE_FILE_MACHINE_I386" then
+      proc = "x86"
+   else
+      proc = "x86_64"
+   end
+   return proc
+end
+
 return M