3 files changed, 52 insertions, 9 deletions

diff --git a/src/luarocks/cfg.lua b/src/luarocks/cfg.lua
index 8e26bb46..9227db4b 100644
--- a/src/luarocks/cfg.lua
+++ b/src/luarocks/cfg.lua
@@ -437,6 +437,23 @@ if detected.solaris then
    defaults.variables.MAKE = "gmake"
 end
 
+if type(site_config.LUAROCKS_ADD_LIB_SUBDIRS) == "table" and next(site_config.LUAROCKS_ADD_LIB_SUBDIRS) ~= nil then
+   local lib = defaults.external_deps_subdirs.lib
+   if type(lib) ~= "table" then
+      lib = { lib }
+   end
+   local rt_lib = defaults.runtime_external_deps_subdirs.lib
+   if type(rt_lib) ~= "table" then
+      rt_lib = { rt_lib }
+   end
+   for i,v in ipairs(site_config.LUAROCKS_ADD_LIB_SUBDIRS) do
+      lib[#lib+1] = v
+      rt_lib[#rt_lib+1] = v
+   end
+   defaults.external_deps_subdirs.lib = lib
+   defaults.runtime_external_deps_subdirs.lib = rt_lib
+end
+
 -- Expose some more values detected by LuaRocks for use by rockspec authors.
 defaults.variables.LIB_EXTENSION = defaults.lib_extension
 defaults.variables.OBJ_EXTENSION = defaults.obj_extension
diff --git a/src/luarocks/deps.lua b/src/luarocks/deps.lua
index f24dc507..7f3b44f2 100644
--- a/src/luarocks/deps.lua
+++ b/src/luarocks/deps.lua
@@ -583,7 +583,19 @@ function check_external_deps(rockspec, mode)
                prefix = prefix.prefix
             end
             for dirname, dirdata in pairs(dirs) do
-               dirdata.dir = vars[name.."_"..dirname] or dir.path(prefix, dirdata.subdir)
+               local paths
+               local path_var_value = vars[name.."_"..dirname]
+               if path_var_value then
+                  paths = { path_var_value }
+               elseif type(dirdata.subdir) == "table" then
+                  paths = {}
+                  for i,v in ipairs(dirdata.subdir) do
+                     paths[i] = dir.path(prefix, v)
+                  end
+               else
+                  paths = { dir.path(prefix, dirdata.subdir) }
+               end
+               dirdata.dir = paths[1]
                local file = files[dirdata.testfile]
                if file then
                   local files = {}
@@ -605,16 +617,22 @@ function check_external_deps(rockspec, mode)
                      if f:match("%.so$") or f:match("%.dylib$") or f:match("%.dll$") then
                         f = f:gsub("%.[^.]+$", "."..cfg.external_lib_extension)
                      end
-                     if f:match("%*") then
-                        local replaced = f:gsub("%.", "%%."):gsub("%*", ".*")
-                        for _, entry in ipairs(fs.list_dir(dirdata.dir)) do
-                           if entry:match(replaced) then
-                              found = true
-                              break
+                     for _, d in ipairs(paths) do
+                        if f:match("%*") then
+                           local replaced = f:gsub("%.", "%%."):gsub("%*", ".*")
+                           for _, entry in ipairs(fs.list_dir(d)) do
+                              if entry:match(replaced) then
+                                 found = true
+                                 break
+                              end
                            end
+                        else
+                           found = fs.is_file(dir.path(d, f))
+                        end
+                        if found then
+                           dirdata.dir = d
+                           break
                         end
-                     else
-                        found = fs.is_file(dir.path(dirdata.dir, f))
                      end
                      if found then
                         break
diff --git a/src/luarocks/fs/unix.lua b/src/luarocks/fs/unix.lua
index cccbbd33..cb5e634d 100644
--- a/src/luarocks/fs/unix.lua
+++ b/src/luarocks/fs/unix.lua
@@ -61,6 +61,14 @@ function wrap_script(file, dest, name, version)
    wrapper:write('LUA_PATH="'..package.path..';$LUA_PATH"\n')
    wrapper:write('LUA_CPATH="'..package.cpath..';$LUA_CPATH"\n')
    wrapper:write('export LUA_PATH LUA_CPATH\n')
+   wrapper:write('if [ -n "$LUA_PATH_5_2" ]; then\n')
+   wrapper:write('   LUA_PATH_5_2="'..package.path..';$LUA_PATH_5_2"\n')
+   wrapper:write('   export LUA_PATH_5_2\n')
+   wrapper:write('fi\n')
+   wrapper:write('if [ -n "$LUA_CPATH_5_2" ]; then\n')
+   wrapper:write('   LUA_CPATH_5_2="'..package.cpath..';$LUA_CPATH_5_2"\n')
+   wrapper:write('   export LUA_CPATH_5_2\n')
+   wrapper:write('fi\n')
    wrapper:write('exec "'..dir.path(cfg.variables["LUA_BINDIR"], cfg.lua_interpreter)..'" -lluarocks.loader -e\'luarocks.loader.add_context([['..name..']],[['..version..']])\' "'..file..'" "$@"\n')
    wrapper:close()
    if fs.chmod(wrapname, "0755") then