1 files changed, 102 insertions, 0 deletions
diff --git a/include/iprt/ldrlazy.h b/include/iprt/ldrlazy.h
new file mode 100644
index 00000000..addfa36e
--- /dev/null
+++ b/include/iprt/ldrlazy.h
@@ -0,0 +1,102 @@
+/** @file
+ * IPRT - Lazy share library linking (2nd try).
+ */
+
+/*
+ * Copyright (C) 2013 Oracle Corporation
+ *
+ * This file is part of VirtualBox Open Source Edition (OSE), as
+ * available from http://www.virtualbox.org. This file is free software;
+ * you can redistribute it and/or modify it under the terms of the GNU
+ * General Public License (GPL) as published by the Free Software
+ * Foundation, in version 2 as it comes in the "COPYING" file of the
+ * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
+ * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
+ *
+ * The contents of this file may alternatively be used under the terms
+ * of the Common Development and Distribution License Version 1.0
+ * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
+ * VirtualBox OSE distribution, in which case the provisions of the
+ * CDDL are applicable instead of those of the GPL.
+ *
+ * You may elect to license modified versions of this file under the
+ * terms and conditions of either the GPL or the CDDL or both.
+ */
+
+#ifndef ___iprt_ldrlazy_h
+#define ___iprt_ldrlazy_h
+
+#include <iprt/ldr.h>
+
+/** @defgroup grp_rt_ldrlazy        Lazy shared library linking.
+ * @ingroup grp_rt
+ *
+ *
+ * @{
+ */
+
+
+/**
+ * Defines a module for use in lazy resolving.
+ *
+ * @param   a_Mod       The module name (C name).
+ * @param   a_pszFile   The file to tell RTLdrLoad to load.
+ */
+#define RTLDRLAZY_MODULE(a_Mod, a_pszFile) \
+    RTLDRLAZY_MODULE_EX(a_Mod, a_pszFile, RTLdrLoad)
+
+/**
+ * Defines a module for use in lazy resolving.
+ *
+ * @param   a_Mod       The module name (C name).
+ * @param   a_pszFile   The file to tell RTLdrLoad to load.
+ * @param   a_pfnLoadIt Function to call for loading the DLL, replacing
+ *                      RTLdrLoad.
+ */
+#define RTLDRLAZY_MODULE_EX(a_Mod, a_pszFile, a_pfnLoadIt) \
+    static bool rtLdrLazy_##a_Mod##_Resolve(const char *pszName, void **ppvSymbol) \
+    { \
+        static RTLDRMOD volatile    s_hMod    = NIL_RTLDRMOD; \
+        static bool volatile        s_fLoaded = false; \
+        RTLDRMOD    hMod; \
+        int         rc; \
+        if (!s_fLoaded) \
+        { \
+            rc = a_pfnLoadIt(a_pszFile, &hMod); \
+            s_hMod = RT_SUCCESS(rc) ? hMod : NIL_RTLDRMOD; \
+            s_fLoaded = true; \
+            if (RT_FAILURE(rc)) \
+                return false; \
+        } \
+        hMod = s_hMod; \
+        if (hMod == NIL_RTLDRMOD) \
+            return false; \
+        rc = RTLdrGetSymbol(hMod, pszName, ppvSymbol); \
+        return RT_SUCCESS(rc); \
+    }
+
+
+
+/** Function name mangler for preventing collision with system prototypes. */
+#define RTLDRLAZY_FUNC_NAME(a_Mod, a_Name) a_Mod##__##a_Name
+
+/**
+ * Defines a function that should be lazily resolved.
+ */
+#define RTLDRLAZY_FUNC(a_Mod, a_RetType, a_CallConv, a_Name, a_ParamDecl, a_ParamNames, a_ErrRet) \
+    DECLINLINE(a_RetType) RTLDRLAZY_FUNC_NAME(a_Mod, a_Name) a_ParamDecl \
+    { \
+        static a_RetType (a_CallConv * s_pfn) a_ParamDecl; \
+        if (!s_pfn) \
+        { \
+            if (!rtLdrLazy_##a_Mod##_Resolve(#a_Name, (void **)&s_pfn)) \
+                return a_ErrRet; \
+        } \
+        return s_pfn a_ParamNames; \
+    }
+
+
+/** @} */
+
+#endif
+