compat / win32 / lazyload.hon commit Merge branch 'tg/t0021-racefix' (8f3ba42)
   1#ifndef LAZYLOAD_H
   2#define LAZYLOAD_H
   3
   4/*
   5 * A pair of macros to simplify loading of DLL functions. Example:
   6 *
   7 *   DECLARE_PROC_ADDR(kernel32.dll, BOOL, CreateHardLinkW,
   8 *                     LPCWSTR, LPCWSTR, LPSECURITY_ATTRIBUTES);
   9 *
  10 *   if (!INIT_PROC_ADDR(CreateHardLinkW))
  11 *           return error("Could not find CreateHardLinkW() function";
  12 *
  13 *   if (!CreateHardLinkW(source, target, NULL))
  14 *           return error("could not create hardlink from %S to %S",
  15 *                        source, target);
  16 */
  17
  18struct proc_addr {
  19        const char *const dll;
  20        const char *const function;
  21        FARPROC pfunction;
  22        unsigned initialized : 1;
  23};
  24
  25/* Declares a function to be loaded dynamically from a DLL. */
  26#define DECLARE_PROC_ADDR(dll, rettype, function, ...) \
  27        static struct proc_addr proc_addr_##function = \
  28        { #dll, #function, NULL, 0 }; \
  29        static rettype (WINAPI *function)(__VA_ARGS__)
  30
  31/*
  32 * Loads a function from a DLL (once-only).
  33 * Returns non-NULL function pointer on success.
  34 * Returns NULL + errno == ENOSYS on failure.
  35 * This function is not thread-safe.
  36 */
  37#define INIT_PROC_ADDR(function) \
  38        (function = get_proc_addr(&proc_addr_##function))
  39
  40static inline void *get_proc_addr(struct proc_addr *proc)
  41{
  42        /* only do this once */
  43        if (!proc->initialized) {
  44                HANDLE hnd;
  45                proc->initialized = 1;
  46                hnd = LoadLibraryExA(proc->dll, NULL,
  47                                     LOAD_LIBRARY_SEARCH_SYSTEM32);
  48                if (hnd)
  49                        proc->pfunction = GetProcAddress(hnd, proc->function);
  50        }
  51        /* set ENOSYS if DLL or function was not found */
  52        if (!proc->pfunction)
  53                errno = ENOSYS;
  54        return proc->pfunction;
  55}
  56
  57#endif