autoload.c

   1 /*
   2  * Copyright (c) 2000, Red Hat, Inc.
   3  *
   4  *     This program is free software; you can redistribute it and/or modify
   5  *     it under the terms of the GNU General Public License as published by
   6  *     the Free Software Foundation; either version 2 of the License, or
   7  *     (at your option) any later version.
   8  *
   9  *     A copy of the GNU General Public License can be found at
  10  *     http://www.gnu.org/
  11  *
  12  * Written by DJ Delorie <dj@cygnus.com>
  13  *
  14  */
  15 #if 0
  16 static const char *cvsid = "\n%%% $Id$\n";
  17 #endif
  18
  19 #define WIN32_LEAN_AND_MEAN
  20 #include <windows.h>
  21
  22 typedef struct {
  23   const char *name;
  24   HINSTANCE handle;
  25 } DllInfo;
  26
  27 #define DLL(n) __attribute__ ((used)) static DllInfo n ## _info __asm__ (#n "_info") = { #n, 0}
  28
  29 #define Auto(dll, func, size) \
  30         __asm__ ("\t.data"); \
  31         __asm__ ("\t.global\t_" #func "@" #size); \
  32         __asm__ ("_" #func "@" #size ":"); \
  33         __asm__ ("\tcall\tautoload_common"); \
  34         __asm__ ("\t.long\t" #dll "_info"); \
  35         __asm__ ("\t.ascii\t\"" #func "\\0\"")
  36
  37 DLL (wininet);
  38
  39 Auto (wininet, InternetAttemptConnect, 4);
  40 Auto (wininet, InternetCloseHandle, 4);
  41 Auto (wininet, InternetGetLastResponseInfoA, 12);
  42 Auto (wininet, InternetOpenA, 20);
  43 Auto (wininet, InternetOpenUrlA, 24);
  44 Auto (wininet, InternetReadFile, 16);
  45 Auto (wininet, InternetSetOptionA, 16);
  46 Auto (wininet, InternetQueryOptionA, 16);
  47 Auto (wininet, HttpQueryInfoA, 20);
  48 Auto (wininet, HttpSendRequestA, 20);
  49
  50 DLL (advapi32);
  51
  52 Auto (advapi32, AddAccessAllowedAce, 16);
  53 Auto (advapi32, AllocateAndInitializeSid, 44);
  54 Auto (advapi32, FreeSid, 4);
  55 Auto (advapi32, InitializeAcl, 12);
  56 Auto (advapi32, OpenProcessToken, 12);
  57 Auto (advapi32, SetTokenInformation, 16);
  58 Auto (advapi32, OpenSCManagerA, 16);
  59 Auto (advapi32, CloseServiceHandle, 4);
  60 Auto (advapi32, OpenServiceA, 16);
  61 Auto (advapi32, QueryServiceStatus, 8);
  62 Auto (advapi32, StartServiceA, 16);
  63
  64 DLL (ntdll);
  65
  66 Auto (ntdll, NtCreateFile, 44);
  67 Auto (ntdll, NtOpenFile, 24);
  68 Auto (ntdll, NtClose, 4);
  69 Auto (ntdll, NtQueryAttributesFile, 8);
  70 Auto (ntdll, NtQueryInformationFile, 20);
  71 Auto (ntdll, NtSetInformationFile, 20);
  72 Auto (ntdll, RtlInitUnicodeString, 8);
  73 Auto (ntdll, RtlNtStatusToDosError, 4);
  74
  75 typedef struct {
  76   DllInfo *dll;
  77   char name[100];
  78 } AutoEntry;
  79
  80 __asm__ ("\t.text");
  81
  82 __attribute__ ((used)) static void autoload_common (int x) __asm__ ("autoload_common");
  83
  84 static void
  85 autoload_common (int x)
  86 {
  87   int fp, rel;
  88   unsigned char *proc;
  89   HINSTANCE h;
  90   AutoEntry *a;
  91
  92   a = *(AutoEntry **)(&x - 1);
  93   if (a->dll->handle == 0)
  94     {
  95       h = LoadLibrary (a->dll->name);
  96       a->dll->handle = h;
  97     }
  98   fp = (int) GetProcAddress (a->dll->handle, a->name);
  99   proc = ((unsigned char *)a) - 5;
 100   rel = fp - (int)(a); /* now it's a relative call */
 101   *proc++ = 0xe9; /* jump near 32-bit relative */
 102   *proc++ = rel;
 103   *proc++ = rel>>8;
 104   *proc++ = rel>>16;
 105   *proc++ = rel>>24;
 106
 107   *(int *)(&x-1) = (int)proc-5;
 108 }