/[winpt]/trunk/PTD/PTD.cpp
ViewVC logotype

Diff of /trunk/PTD/PTD.cpp

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 165 by twoaday, Tue Jan 24 10:10:15 2006 UTC revision 412 by twoaday, Sat Feb 11 15:03:43 2012 UTC
# Line 1  Line 1 
1  /* PTD.cpp - Privacy Tray Dynamic  /* PTD.cpp - Privacy Tray Dynamic
2   *      Copyright (C) 2002-2005 Timo Schulz   *      Copyright (C) 2002-2009, 2012 Timo Schulz
3   *   *
4   * This file is part of WinPT.   * This file is part of WinPT.
5   *   *
# Line 12  Line 12 
12   * but WITHOUT ANY WARRANTY; without even the implied warranty of   * but WITHOUT ANY WARRANTY; without even the implied warranty of
13   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14   * GNU General Public License for more details.   * GNU General Public License for more details.
  *  
  * You should have received a copy of the GNU General Public License  
  * along with WinPT; if not, write to the Free Software Foundation,  
  * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA  
15   */   */
   
16  #ifdef HAVE_CONFIG_H  #ifdef HAVE_CONFIG_H
17  #include <config.h>  #include <config.h>
18  #endif  #endif
19    
20  #include <windows.h>  #include <windows.h>
21  #include <stdio.h>  #include <stdio.h>
 #include <ocidl.h>  
 #include <olectl.h>  
   
 #include "wptJPG.h"  
22    
23    /* Reference to the DLL module handle */
24  HINSTANCE glob_hinst;  HINSTANCE glob_hinst;
25    
26  /* We need a special section in the DLL for storing our shared data.  /* We need a special section in the DLL for storing our shared data.
27     This shared data is required by the hook function becuase theat     This shared data is required by the hook function because that
28     function is called in the context of the actual thread running the     function is called in the context of the actual thread running the
29     hook and thus we need a way to communicate the result of our hook     hook and thus we need a way to communicate the result of our hook
30     function back to us.  We use a shared data segment for this. */     function back to us.  We use a shared data segment for this. */
# Line 54  static HWND  systab_c32    ATTR_SEC = NU Line 46  static HWND  systab_c32    ATTR_SEC = NU
46  #endif  #endif
47  #undef ATTR_SEC  #undef ATTR_SEC
48    
49    /* Static variables that are used to record and replay key strokes */
50  static unsigned *journ_keys = 0;  static unsigned *journ_keys = 0;
51  static unsigned key_idx = 0;  static unsigned keyidx = 0;
52  static unsigned nkeys = 0;  static unsigned nkeys = 0;
53  static HHOOK journ_hook = NULL;     /* Journaling hook  handle. */  static HHOOK journ_hook = NULL;  /* Journaling hook  handle. */
54    
55    
56    /* Callback function for the Computer Based Training (CBT) hook */
57  static LRESULT CALLBACK  static LRESULT CALLBACK
58  PTD_CBT_proc (int code, WPARAM wparam, LPARAM lparam)  PTD_CBT_proc (int code, WPARAM wparam, LPARAM lparam)
59  {  {
60      HWND curr_hwnd = NULL;            HWND curr_hwnd;
61      DWORD proc_id = 0;      DWORD proc_id;
62            
63      if (code >= 0 && code == HCBT_SETFOCUS) {      if (code >= 0 && code == HCBT_SETFOCUS) {
64          /* A window is about to receive the keyboard focus */                    /* A window is about to receive the keyboard focus */          
65                    
66            proc_id = 0;
67          curr_hwnd = (HWND)wparam;          curr_hwnd = (HWND)wparam;
68          GetWindowThreadProcessId( curr_hwnd, &proc_id );          GetWindowThreadProcessId (curr_hwnd, &proc_id);
69          /* Only if the current window is different from this window list,          /* Only if the current window is different from this window list,
70             we set the new focus. For more information read the hint in             we set the new focus. For more information read the hint in
71             PTD_initialize. */             PTD_initialize. */
72          if ((proc_id != tray_proc_id)          if ((proc_id != tray_proc_id) &&
73               && (curr_hwnd != NULL)              (curr_hwnd != NULL) &&
74               && (curr_hwnd != curr_focus)              (curr_hwnd != curr_focus) &&
75               && (curr_hwnd != shell_traywnd)              (curr_hwnd != shell_traywnd) &&
76               && (curr_hwnd != systab_c32)              (curr_hwnd != systab_c32) &&
77               && (curr_hwnd != mstask_swc)              (curr_hwnd != mstask_swc) &&
78               && (curr_hwnd != rebar_w32)) {              (curr_hwnd != rebar_w32)) {
79              curr_focus = curr_hwnd;              curr_focus = curr_hwnd;
80          }          }
81      }      }
# Line 91  PTD_CBT_proc (int code, WPARAM wparam, L Line 86  PTD_CBT_proc (int code, WPARAM wparam, L
86    
87  /* Return the last selected window */  /* Return the last selected window */
88  extern "C" HWND  extern "C" HWND
89  PTD_get_curr_hwnd (void)  PTD_get_current_window (void)
90  {  {
91      return curr_focus;      return curr_focus;
92  }  }
# Line 109  PTD_initialize (void) Line 104  PTD_initialize (void)
104      /* Okay, what are we doing here:      /* Okay, what are we doing here:
105       * In the past I used the Spy++ application to find out why the old current       * In the past I used the Spy++ application to find out why the old current
106       * window doesn't work on so much systems. Now we try to use the computer       * window doesn't work on so much systems. Now we try to use the computer
107       * based training hook, more precise the HCBT_SETFOCUS to find out the       * based training hook, more precisely the HCBT_SETFOCUS to find out the
108       * last window focus. But to succeed we need to ignore some windows       * last window focus. But to succeed we need to ignore some windows
109       * because otherwise our own window or a window from the taskbar       * because otherwise our own window or a window from the taskbar
110       * would be recognized.       * would be recognized.
# Line 123  PTD_initialize (void) Line 118  PTD_initialize (void)
118       * As a result we need to ignore those windows to prevent failured       * As a result we need to ignore those windows to prevent failured
119       * in the code.       * in the code.
120       */       */
121        
122        // FIXME: I doubt this is working for Windows 7
123      shell_traywnd = FindWindowEx (NULL, NULL, "Shell_TrayWnd", NULL);      shell_traywnd = FindWindowEx (NULL, NULL, "Shell_TrayWnd", NULL);
124      rebar_w32 = FindWindowEx (shell_traywnd, NULL, "ReBarWindow32", NULL);      rebar_w32 = FindWindowEx (shell_traywnd, NULL, "ReBarWindow32", NULL);
125      mstask_swc = FindWindowEx (rebar_w32, NULL, "MSTaskSwWClass", NULL);      mstask_swc = FindWindowEx (rebar_w32, NULL, "MSTaskSwWClass", NULL);
126      systab_c32 = FindWindowEx (mstask_swc, NULL, "SysTabControl32", NULL);      systab_c32 = FindWindowEx (mstask_swc, NULL, "SysTabControl32", NULL);
127            
128      /*log_foo ("tray_proc_id=%u cbt_hook=%p shell_traywnd=%p rebar_w32=%p "      /*debug ("tray_proc_id=%u cbt_hook=%p shell_traywnd=%p rebar_w32=%p "
129                 "mstask_swc=%p systab_c32=%p",               "mstask_swc=%p systab_c32=%p",
130                 tray_proc_id, cbt_hook, shell_traywnd, rebar_w32,               tray_proc_id, cbt_hook, shell_traywnd, rebar_w32,
131                 mstask_swc, systab_c32);*/               mstask_swc, systab_c32);*/
132            
133      return TRUE;      return TRUE;
134  }  }
# Line 139  PTD_initialize (void) Line 136  PTD_initialize (void)
136    
137  /* Remove the CTB hook from the system and reset all variables. */  /* Remove the CTB hook from the system and reset all variables. */
138  extern "C" void  extern "C" void
139  PTD_delete (void)  PTD_delete_hook (void)
140  {  {
141      if (!cbt_hook)      if (!cbt_hook)
142          return;          return;
# Line 152  PTD_delete (void) Line 149  PTD_delete (void)
149    
150  /* Return if the CTB hook is currently used. */  /* Return if the CTB hook is currently used. */
151  extern "C" int  extern "C" int
152  PTD_is_used (void)  PTD_is_hook_used (void)
153  {  {
154      return shell_traywnd && cbt_hook;      return shell_traywnd && cbt_hook;
155  }  }
156    
157    /* From the MSDN:
158     * "Typically, an application uses this function to play back a series
159     * of mouse and keyboard messages recorded previously by the
160     * JournalRecordProc hook procedure. As long as a JournalPlaybackProc
161     * hook procedure is installed, regular mouse and keyboard input is disabled."
162     */
163  static LRESULT CALLBACK  static LRESULT CALLBACK
164  PTD_playback_proc (int code, WPARAM wparam, LPARAM lparam)  PTD_playback_proc (int code, WPARAM wparam, LPARAM lparam)
165  {  {
166      LPEVENTMSG em;      LPEVENTMSG em;
167            
168      if (code == HC_SKIP) {      if (code == HC_SKIP) {
169          key_idx++;          keyidx++;
170          if (key_idx == nkeys) {          if (keyidx >= nkeys) {
171              UnhookWindowsHookEx (journ_hook);              UnhookWindowsHookEx (journ_hook);
172              journ_hook = NULL;              journ_hook = NULL;
173          }          }
# Line 178  PTD_playback_proc (int code, WPARAM wpar Line 180  PTD_playback_proc (int code, WPARAM wpar
180           * paramH - specifies the scan code.                                       * paramH - specifies the scan code.                            
181           */           */
182          em = (LPEVENTMSG )lparam;          em = (LPEVENTMSG )lparam;
183          if (journ_keys[key_idx] & 0x8000)          if (journ_keys[keyidx] & 0x8000)
184              em->message = journ_keys[key_idx] & 0x4000 ? WM_SYSKEYDOWN : WM_KEYDOWN;              em->message = journ_keys[keyidx] & 0x4000 ? WM_SYSKEYDOWN : WM_KEYDOWN;
185          else          else
186              em->message = journ_keys[key_idx] & 0x4000 ? WM_SYSKEYUP : WM_KEYUP;              em->message = journ_keys[keyidx] & 0x4000 ? WM_SYSKEYUP : WM_KEYUP;
187          em->paramL = LOBYTE( journ_keys[key_idx] )          em->paramL = LOBYTE (journ_keys[keyidx])
188              | (MapVirtualKey (LOBYTE (journ_keys[key_idx]), 0) << 8);              | (MapVirtualKey (LOBYTE (journ_keys[keyidx]), 0) << 8);
189          em->paramH = 1;          em->paramH = 1;
190          em->time = GetTickCount ();          em->time = GetTickCount ();
191          return 0;          return 0;
192      }      }
193            
194      return CallNextHookEx (journ_hook, code, wparam, lparam);      return CallNextHookEx (journ_hook, code, wparam, lparam);
195  } /* PTD_Playback_proc */  }
196    
197    
198    
199  extern "C" BOOL  extern "C" BOOL
200  PTD_keyb_send (UINT *keys, UINT n_keys)  PTD_keyboard_send_keys (UINT *keys, UINT n_keys)
201  {  {
202      MSG msg;      MSG msg;
203      journ_keys = keys;      journ_keys = keys;
204      key_idx = 0;      keyidx = 0;
205      nkeys =  n_keys;          nkeys = n_keys;
206            
207        /* Return immediately if there is an active hook */
208      if (journ_hook)      if (journ_hook)
209          return FALSE;          return FALSE;
210            
211      while (GetAsyncKeyState (VK_MENU) & 0x8000      while (GetAsyncKeyState (VK_MENU) & 0x8000 ||
212              || GetAsyncKeyState (VK_SHIFT) & 0x8000             GetAsyncKeyState (VK_SHIFT) & 0x8000 ||
213              || GetAsyncKeyState (VK_CONTROL) & 0x8000) {             GetAsyncKeyState (VK_CONTROL) & 0x8000) {
214          if (PeekMessage (&msg, NULL, 0, 0, PM_REMOVE)) {          if (PeekMessage (&msg, NULL, 0, 0, PM_REMOVE)) {
215              if (msg.message == WM_QUIT) {              if (msg.message == WM_QUIT) {
216                  PostMessage (msg.hwnd, msg.message, msg.wParam, msg.lParam);                  PostMessage (msg.hwnd, msg.message, msg.wParam, msg.lParam);
# Line 239  PTD_keyb_send (UINT *keys, UINT n_keys) Line 242  PTD_keyb_send (UINT *keys, UINT n_keys)
242  }  }
243    
244    
245  /* Display a JPG picture in the given window at the given point. */  /* Main entry point for the DLL */
246  extern "C" int  extern "C" BOOL WINAPI
 PTD_jpg_show (HWND hwnd, POINT *p, LPCSTR name)  
 {  
     CJPG jpg;  
     HDC hdc;  
     POINT sizewnd;  
     RECT rwnd;  
     BOOL rc;  
   
     rc = jpg.load (name);  
     if (!rc)  
         return -1; /* XXX: use real return code. */  
     hdc = GetWindowDC (hwnd);  
     rc = jpg.updateSizeOnDC (hdc);  
     if (!rc) {  
         ReleaseDC (hwnd, hdc);  
         return -2; /* XXX: use real return code. */  
     }  
   
     GetWindowRect (hwnd, &rwnd);  
     sizewnd.x = rwnd.right - rwnd.left;  
     sizewnd.y = rwnd.bottom - rwnd.top;  
     rc = jpg.show (hdc, p, &sizewnd, 0, 0);  
   
     ReleaseDC (hwnd, hdc);  
     jpg.freePictureData ();  
     return rc;  
 }  
   
                   
 extern "C" int WINAPI  
247  DllMain (HINSTANCE hinst, DWORD reason, LPVOID reserv)  DllMain (HINSTANCE hinst, DWORD reason, LPVOID reserv)
248  {  {
249      switch (reason)  {      switch (reason)  {
250      case DLL_PROCESS_ATTACH:      case DLL_PROCESS_ATTACH:
251          glob_hinst = hinst;          glob_hinst = hinst;
252          break;          break;
253      case DLL_THREAD_ATTACH:          
254        case DLL_PROCESS_DETACH:
255          break;          break;
256            
257        case DLL_THREAD_ATTACH:  
258      case DLL_THREAD_DETACH:        case DLL_THREAD_DETACH:  
259          break;          break;
260      case DLL_PROCESS_DETACH:          
         break;  
261      }      }
262        
263      return TRUE;      return TRUE;
264  }  }
Legend:
Removed from v.165  
changed lines
  Added in v.412
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26