From d90e4b38cd607161656364bb8aec3019fd23b506 Mon Sep 17 00:00:00 2001
From: Joao Matos <joao@tritao.eu>
Date: Thu, 2 Feb 2017 17:33:44 +0000
Subject: [PATCH] Fixed CleanUnitPass to deal with directory-less translation
 unit paths.

Fixes issue #738.
---
 src/Generator/Passes/CleanUnitPass.cs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/Generator/Passes/CleanUnitPass.cs b/src/Generator/Passes/CleanUnitPass.cs
index 21ca661a..e6a151a4 100644
--- a/src/Generator/Passes/CleanUnitPass.cs
+++ b/src/Generator/Passes/CleanUnitPass.cs
@@ -26,7 +26,11 @@ namespace CppSharp.Passes
             if (unit.IsSystemHeader)
                 return Options.SystemModule;
 
-            var includeDir = Path.GetFullPath(Path.GetDirectoryName(unit.FilePath));
+            var includeDir = Path.GetDirectoryName(unit.FilePath);
+            if (string.IsNullOrWhiteSpace(includeDir))
+                includeDir = ".";
+            includeDir = Path.GetFullPath(includeDir);
+
             return Options.Modules.FirstOrDefault(
                 m => m.IncludeDirs.Any(i => Path.GetFullPath(i) == includeDir)) ??
                 Options.MainModule;