Hex Artifact Content

Not logged in

Artifact 6b2f129d530b6aee348c6479c2ea49e86c9a579d:


0000: ef bb bf 69 6d 70 6f 72 74 20 63 6f 72 65 2e 72  ...import core.r
0010: 75 6e 74 69 6d 65 3b 0a 69 6d 70 6f 72 74 20 77  untime;.import w
0020: 69 6e 33 32 2e 77 69 6e 64 6f 77 73 3b 0a 69 6d  in32.windows;.im
0030: 70 6f 72 74 20 73 74 64 2e 73 74 72 69 6e 67 3b  port std.string;
0040: 0a 0a 63 6c 61 73 73 20 57 69 6e 44 4c 4c 45 78  ..class WinDLLEx
0050: 63 65 70 74 69 6f 6e 20 3a 20 45 78 63 65 70 74  ception : Except
0060: 69 6f 6e 0a 7b 0a 09 70 72 69 76 61 74 65 20 74  ion.{..private t
0070: 68 69 73 28 20 73 74 72 69 6e 67 20 6d 73 67 20  his( string msg 
0080: 29 20 7b 20 73 75 70 65 72 28 6d 73 67 29 3b 20  ) { super(msg); 
0090: 7d 0a 7d 0a 0a 63 6c 61 73 73 20 57 69 6e 44 4c  }.}..class WinDL
00a0: 4c 0a 7b 0a 09 2f 2f 20 63 6f 6e 73 74 72 75 63  L.{..// construc
00b0: 74 6f 72 3a 0a 09 2f 2f 20 20 20 4c 6f 61 64 20  tor:..//   Load 
00c0: 61 20 44 4c 4c 20 6f 66 20 73 70 65 63 69 66 69  a DLL of specifi
00d0: 65 64 20 6e 61 6d 65 2e 0a 09 2f 2f 20 20 20 41  ed name...//   A
00e0: 76 6f 69 64 20 6c 6f 61 64 69 6e 67 20 74 68 65  void loading the
00f0: 20 44 4c 4c 20 66 72 6f 6d 20 74 68 65 20 63 75   DLL from the cu
0100: 72 72 65 6e 74 20 64 69 72 65 63 74 6f 72 79 20  rrent directory 
0110: 66 6f 72 20 73 65 63 75 72 69 74 79 2e 0a 0a 09  for security....
0120: 74 68 69 73 28 20 73 74 72 69 6e 67 20 64 6c 6c  this( string dll
0130: 6e 61 6d 65 20 29 0a 09 7b 0a 09 09 63 68 61 72  name )..{...char
0140: 20 6f 72 69 67 69 6e 61 6c 5f 63 75 72 5b 4d 41   original_cur[MA
0150: 58 5f 50 41 54 48 5d 3b 0a 09 09 63 68 61 72 20  X_PATH];...char 
0160: 73 79 73 5b 4d 41 58 5f 50 41 54 48 5d 3b 0a 09  sys[MAX_PATH];..
0170: 09 47 65 74 43 75 72 72 65 6e 74 44 69 72 65 63  .GetCurrentDirec
0180: 74 6f 72 79 41 28 6f 72 69 67 69 6e 61 6c 5f 63  toryA(original_c
0190: 75 72 2e 6c 65 6e 67 74 68 2c 20 6f 72 69 67 69  ur.length, origi
01a0: 6e 61 6c 5f 63 75 72 2e 70 74 72 29 3b 0a 09 09  nal_cur.ptr);...
01b0: 47 65 74 53 79 73 74 65 6d 44 69 72 65 63 74 6f  GetSystemDirecto
01c0: 72 79 41 28 73 79 73 2e 70 74 72 2c 20 73 79 73  ryA(sys.ptr, sys
01d0: 2e 6c 65 6e 67 74 68 29 3b 0a 09 09 53 65 74 43  .length);...SetC
01e0: 75 72 72 65 6e 74 44 69 72 65 63 74 6f 72 79 41  urrentDirectoryA
01f0: 28 73 79 73 2e 70 74 72 29 3b 0a 09 09 68 61 6e  (sys.ptr);...han
0200: 64 6c 65 20 3d 20 63 61 73 74 28 48 49 4e 53 54  dle = cast(HINST
0210: 41 4e 43 45 29 20 52 75 6e 74 69 6d 65 2e 6c 6f  ANCE) Runtime.lo
0220: 61 64 4c 69 62 72 61 72 79 28 64 6c 6c 6e 61 6d  adLibrary(dllnam
0230: 65 29 3b 0a 09 09 53 65 74 43 75 72 72 65 6e 74  e);...SetCurrent
0240: 44 69 72 65 63 74 6f 72 79 41 28 6f 72 69 67 69  DirectoryA(origi
0250: 6e 61 6c 5f 63 75 72 2e 70 74 72 29 3b 0a 09 09  nal_cur.ptr);...
0260: 69 66 28 20 68 61 6e 64 6c 65 20 69 73 20 6e 75  if( handle is nu
0270: 6c 6c 20 29 0a 09 09 09 74 68 72 6f 77 20 6e 65  ll )....throw ne
0280: 77 20 57 69 6e 44 4c 4c 45 78 63 65 70 74 69 6f  w WinDLLExceptio
0290: 6e 28 20 64 6c 6c 6e 61 6d 65 20 7e 20 22 20 6e  n( dllname ~ " n
02a0: 6f 74 20 66 6f 75 6e 64 22 20 29 3b 0a 09 7d 0a  ot found" );..}.
02b0: 0a 09 73 74 61 74 69 63 20 57 69 6e 44 4c 4c 20  ..static WinDLL 
02c0: 6c 6f 61 64 28 20 73 74 72 69 6e 67 20 6e 61 6d  load( string nam
02d0: 65 20 29 0a 09 7b 0a 09 09 74 72 79 20 7b 0a 09  e )..{...try {..
02e0: 09 09 72 65 74 75 72 6e 20 6e 65 77 20 57 69 6e  ..return new Win
02f0: 44 4c 4c 28 6e 61 6d 65 29 3b 0a 09 09 7d 20 63  DLL(name);...} c
0300: 61 74 63 68 28 57 69 6e 44 4c 4c 45 78 63 65 70  atch(WinDLLExcep
0310: 74 69 6f 6e 20 65 29 20 7b 20 72 65 74 75 72 6e  tion e) { return
0320: 20 6e 75 6c 6c 3b 20 7d 0a 09 7d 0a 0a 09 2f 2f   null; }..}...//
0330: 20 63 6c 6f 73 65 3a 0a 09 2f 2f 20 20 20 66 72   close:..//   fr
0340: 65 65 20 74 68 65 20 44 4c 4c 0a 0a 09 76 6f 69  ee the DLL...voi
0350: 64 20 63 6c 6f 73 65 28 29 0a 09 7b 0a 09 09 69  d close()..{...i
0360: 66 28 20 68 61 6e 64 6c 65 20 29 0a 09 09 7b 0a  f( handle )...{.
0370: 09 09 09 52 75 6e 74 69 6d 65 2e 75 6e 6c 6f 61  ...Runtime.unloa
0380: 64 4c 69 62 72 61 72 79 28 68 61 6e 64 6c 65 29  dLibrary(handle)
0390: 3b 0a 09 09 09 68 61 6e 64 6c 65 20 3d 20 6e 75  ;....handle = nu
03a0: 6c 6c 3b 0a 09 09 7d 0a 09 7d 0a 0a 09 2f 2f 20  ll;...}..}...// 
03b0: 67 65 74 5f 61 70 69 3a 0a 09 2f 2f 20 20 20 65  get_api:..//   e
03c0: 78 74 72 61 63 74 20 61 20 66 75 6e 63 74 69 6f  xtract a functio
03d0: 6e 20 77 69 74 68 20 73 70 65 63 69 66 69 65 64  n with specified
03e0: 20 6e 61 6d 65 20 66 72 6f 6d 20 74 68 65 20 44   name from the D
03f0: 4c 4c 2e 0a 09 2f 2f 20 20 20 6d 61 79 20 72 65  LL...//   may re
0400: 74 75 72 6e 20 6e 75 6c 6c 2e 0a 0a 09 74 65 6d  turn null....tem
0410: 70 6c 61 74 65 20 67 65 74 5f 61 70 69 28 46 6e  plate get_api(Fn
0420: 54 29 0a 09 7b 0a 09 09 46 6e 54 20 67 65 74 5f  T)..{...FnT get_
0430: 61 70 69 28 20 73 74 72 69 6e 67 20 61 70 69 6e  api( string apin
0440: 61 6d 65 20 29 0a 09 09 09 69 6e 20 7b 20 61 73  ame )....in { as
0450: 73 65 72 74 28 68 61 6e 64 6c 65 29 3b 20 7d 0a  sert(handle); }.
0460: 09 09 09 62 6f 64 79 0a 09 09 09 7b 0a 09 09 09  ...body....{....
0470: 09 72 65 74 75 72 6e 20 63 61 73 74 28 46 6e 54  .return cast(FnT
0480: 29 20 47 65 74 50 72 6f 63 41 64 64 72 65 73 73  ) GetProcAddress
0490: 28 20 68 61 6e 64 6c 65 2c 20 61 70 69 6e 61 6d  ( handle, apinam
04a0: 65 2e 74 6f 53 74 72 69 6e 67 7a 28 29 20 29 3b  e.toStringz() );
04b0: 0a 09 09 09 7d 0a 09 7d 0a 0a 09 2f 2f 20 6c 6f  ....}..}...// lo
04c0: 61 64 5f 72 65 73 6f 75 72 63 65 3a 0a 09 2f 2f  ad_resource:..//
04d0: 20 20 20 65 78 74 72 61 63 74 20 61 20 72 65 73     extract a res
04e0: 6f 75 72 63 65 0a 0a 09 76 6f 69 64 2a 20 6c 6f  ource...void* lo
04f0: 61 64 5f 72 65 73 6f 75 72 63 65 28 20 73 74 72  ad_resource( str
0500: 69 6e 67 20 72 65 73 6e 61 6d 65 2c 20 63 6f 6e  ing resname, con
0510: 73 74 20 63 68 61 72 2a 20 74 79 70 65 20 29 0a  st char* type ).
0520: 09 09 69 6e 20 7b 20 61 73 73 65 72 74 28 68 61  ..in { assert(ha
0530: 6e 64 6c 65 29 3b 20 7d 0a 09 09 62 6f 64 79 0a  ndle); }...body.
0540: 09 09 7b 0a 09 09 09 69 66 28 72 65 73 6e 61 6d  ..{....if(resnam
0550: 65 20 69 6e 20 72 73 72 63 5f 6d 61 70 29 0a 09  e in rsrc_map)..
0560: 09 09 09 72 65 74 75 72 6e 20 72 73 72 63 5f 6d  ...return rsrc_m
0570: 61 70 5b 72 65 73 6e 61 6d 65 5d 3b 0a 0a 09 09  ap[resname];....
0580: 09 48 52 53 52 43 20 68 69 20 3d 20 46 69 6e 64  .HRSRC hi = Find
0590: 52 65 73 6f 75 72 63 65 28 20 68 61 6e 64 6c 65  Resource( handle
05a0: 2c 20 72 65 73 6e 61 6d 65 2e 74 6f 53 74 72 69  , resname.toStri
05b0: 6e 67 7a 28 29 2c 20 74 79 70 65 20 29 3b 0a 09  ngz(), type );..
05c0: 09 09 69 66 28 20 68 69 20 69 73 20 6e 75 6c 6c  ..if( hi is null
05d0: 20 29 0a 09 09 09 09 72 65 74 75 72 6e 20 6e 75   ).....return nu
05e0: 6c 6c 3b 0a 09 09 09 48 47 4c 4f 42 41 4c 20 68  ll;....HGLOBAL h
05f0: 72 20 3d 20 4c 6f 61 64 52 65 73 6f 75 72 63 65  r = LoadResource
0600: 28 20 68 61 6e 64 6c 65 2c 20 68 69 20 29 3b 0a  ( handle, hi );.
0610: 09 09 09 69 66 28 20 68 72 20 69 73 20 6e 75 6c  ...if( hr is nul
0620: 6c 20 29 0a 09 09 09 09 72 65 74 75 72 6e 20 6e  l ).....return n
0630: 75 6c 6c 3b 0a 09 09 09 76 6f 69 64 2a 20 76 20  ull;....void* v 
0640: 3d 20 4c 6f 63 6b 52 65 73 6f 75 72 63 65 28 20  = LockResource( 
0650: 68 72 20 29 3b 0a 09 09 09 72 73 72 63 5f 6d 61  hr );....rsrc_ma
0660: 70 5b 72 65 73 6e 61 6d 65 5d 20 3d 20 76 3b 0a  p[resname] = v;.
0670: 09 09 09 72 65 74 75 72 6e 20 76 3b 0a 09 09 7d  ...return v;...}
0680: 0a 0a 09 2f 2f 20 6c 6f 61 64 5f 64 69 61 67 6c  ...// load_diagl
0690: 6f 67 3a 0a 09 2f 2f 20 20 20 73 70 65 63 69 61  og:..//   specia
06a0: 6c 69 7a 65 64 20 76 65 72 73 69 6f 6e 20 6f 66  lized version of
06b0: 20 6c 6f 61 64 5f 72 65 73 6f 75 72 63 65 0a 0a   load_resource..
06c0: 09 76 6f 69 64 2a 20 6c 6f 61 64 5f 64 69 61 6c  .void* load_dial
06d0: 6f 67 28 20 73 74 72 69 6e 67 20 72 65 73 6e 61  og( string resna
06e0: 6d 65 20 29 0a 09 7b 0a 09 09 72 65 74 75 72 6e  me )..{...return
06f0: 20 6c 6f 61 64 5f 72 65 73 6f 75 72 63 65 28 20   load_resource( 
0700: 72 65 73 6e 61 6d 65 2c 20 52 54 5f 44 49 41 4c  resname, RT_DIAL
0710: 4f 47 20 29 3b 0a 09 7d 0a 0a 09 70 72 69 76 61  OG );..}...priva
0720: 74 65 20 48 49 4e 53 54 41 4e 43 45 20 68 61 6e  te HINSTANCE han
0730: 64 6c 65 3b 0a 09 70 72 69 76 61 74 65 20 76 6f  dle;..private vo
0740: 69 64 2a 5b 73 74 72 69 6e 67 5d 20 72 73 72 63  id*[string] rsrc
0750: 5f 6d 61 70 3b 0a 7d 0a                          _map;.}.