/[rdesktop]/sourceforge.net/trunk/seamlessrdp/ServerExe/main.c
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /sourceforge.net/trunk/seamlessrdp/ServerExe/main.c

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

revision 1077 by ossman_, Thu Mar 9 15:26:35 2006 UTC revision 1147 by ossman_, Thu Mar 16 14:54:02 2006 UTC
# Line 24  Line 24 
24    
25  #include <windows.h>  #include <windows.h>
26  #include <stdio.h>  #include <stdio.h>
27    #include <wtsapi32.h>
28    #include <cchannel.h>
29    
30  #include "vchannel.h"  #include "vchannel.h"
31    
# Line 31  Line 33 
33    
34  #define APP_NAME "SeamlessRDP Shell"  #define APP_NAME "SeamlessRDP Shell"
35    
36    #ifndef WM_WTSSESSION_CHANGE
37    #define WM_WTSSESSION_CHANGE 0x02B1
38    #endif
39    #ifndef WTS_REMOTE_CONNECT
40    #define WTS_REMOTE_CONNECT 0x3
41    #endif
42    
43  /* Global data */  /* Global data */
44  static HINSTANCE g_instance;  static HINSTANCE g_instance;
45    static HWND g_hwnd;
46    
47  typedef void (*set_hooks_proc_t) ();  typedef void (*set_hooks_proc_t) ();
48  typedef void (*remove_hooks_proc_t) ();  typedef void (*remove_hooks_proc_t) ();
49  typedef int (*get_instance_count_proc_t) ();  typedef int (*get_instance_count_proc_t) ();
50    
51    typedef void (*move_window_proc_t) (HWND hwnd, int x, int y, int width, int height);
52    
53    static move_window_proc_t g_move_window_fn = NULL;
54    
55  static void  static void
56  message(const char *text)  message(const char *text)
57  {  {
# Line 67  get_token(char **s) Line 81  get_token(char **s)
81          return head;          return head;
82  }  }
83    
84  static BOOL CALLBACK enum_cb(HWND hwnd, LPARAM lparam)  static BOOL CALLBACK
85    enum_cb(HWND hwnd, LPARAM lparam)
86  {  {
87          RECT rect;          RECT rect;
88          char title[150];          unsigned short title[150];
89          LONG styles;          LONG styles;
90          int state;          int state;
91            HWND parent;
92    
93          styles = GetWindowLong(hwnd, GWL_STYLE);          styles = GetWindowLong(hwnd, GWL_STYLE);
94    
95          if (!(styles & WS_VISIBLE))          if (!(styles & WS_VISIBLE))
96                  return TRUE;                  return TRUE;
97    
98          vchannel_write("CREATE1,0x%p,0x%x", hwnd, 0);          if (styles & WS_POPUP)
99                    parent = (HWND) GetWindowLong(hwnd, GWL_HWNDPARENT);
100            else
101                    parent = NULL;
102    
103          if (!GetWindowRect(hwnd, &rect)) {          vchannel_write("CREATE,0x%p,0x%p,0x%x", hwnd, parent, 0);
104    
105            if (!GetWindowRect(hwnd, &rect))
106            {
107                  debug("GetWindowRect failed!");                  debug("GetWindowRect failed!");
108                  return TRUE;                  return TRUE;
109          }          }
110    
111          vchannel_write("POSITION1,0x%p,%d,%d,%d,%d,0x%x",          vchannel_write("POSITION,0x%p,%d,%d,%d,%d,0x%x",
112                         hwnd,                         hwnd,
113                         rect.left, rect.top,                         rect.left, rect.top, rect.right - rect.left, rect.bottom - rect.top, 0);
                        rect.right - rect.left,  
                        rect.bottom - rect.top,  
                        0);  
114    
115          GetWindowText(hwnd, title, sizeof(title));          GetWindowTextW(hwnd, title, sizeof(title) / sizeof(*title));
116    
117          /* FIXME: Strip title of dangerous characters */          vchannel_write("TITLE,0x%x,%s,0x%x", hwnd, vchannel_strfilter_unicode(title), 0);
118    
119          if (styles & WS_MAXIMIZE)          if (styles & WS_MAXIMIZE)
120                  state = 2;                  state = 2;
# Line 104  static BOOL CALLBACK enum_cb(HWND hwnd, Line 123  static BOOL CALLBACK enum_cb(HWND hwnd,
123          else          else
124                  state = 0;                  state = 0;
125    
126          vchannel_write("SETSTATE1,0x%p,%s,0x%x,0x%x",          vchannel_write("STATE,0x%p,0x%x,0x%x", hwnd, state, 0);
                        hwnd, title, state, 0);  
127    
128          return TRUE;          return TRUE;
129  }  }
# Line 125  do_sync(void) Line 143  do_sync(void)
143  }  }
144    
145  static void  static void
146    do_state(HWND hwnd, int state)
147    {
148            if (state == 0)
149                    ShowWindow(hwnd, SW_RESTORE);
150            else if (state == 1)
151                    ShowWindow(hwnd, SW_MINIMIZE);
152            else if (state == 2)
153                    ShowWindow(hwnd, SW_MAXIMIZE);
154            else
155                    debug("Invalid state %d sent.", state);
156    }
157    
158    static void
159    do_position(HWND hwnd, int x, int y, int width, int height)
160    {
161            g_move_window_fn(hwnd, x, y, width, height);
162    }
163    
164    static void
165    do_zchange(HWND hwnd, HWND behind)
166    {
167            if (behind == NULL)
168                    behind = HWND_TOP;
169            SetWindowPos(hwnd, behind, 0, 0, 0, 0, SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOSIZE);
170    }
171    
172    static void
173  process_cmds(void)  process_cmds(void)
174  {  {
175          char line[VCHANNEL_MAX_LINE];          char line[VCHANNEL_MAX_LINE];
# Line 132  process_cmds(void) Line 177  process_cmds(void)
177    
178          char *p, *tok1, *tok2, *tok3, *tok4, *tok5, *tok6, *tok7, *tok8;          char *p, *tok1, *tok2, *tok3, *tok4, *tok5, *tok6, *tok7, *tok8;
179    
180          while ((size = vchannel_read(line, sizeof(line))) >= 0) {          while ((size = vchannel_read(line, sizeof(line))) >= 0)
181                  debug("Got: %s", line);          {
   
182                  p = line;                  p = line;
183    
184                  tok1 = get_token(&p);                  tok1 = get_token(&p);
# Line 148  process_cmds(void) Line 192  process_cmds(void)
192    
193                  if (strcmp(tok1, "SYNC") == 0)                  if (strcmp(tok1, "SYNC") == 0)
194                          do_sync();                          do_sync();
195                    else if (strcmp(tok1, "STATE") == 0)
196                            do_state((HWND) strtoul(tok2, NULL, 0), strtol(tok3, NULL, 0));
197                    else if (strcmp(tok1, "POSITION") == 0)
198                            do_position((HWND) strtoul(tok2, NULL, 0), strtol(tok3, NULL, 0),
199                                        strtol(tok4, NULL, 0), strtol(tok5, NULL, 0), strtol(tok6, NULL,
200                                                                                             0));
201                    else if (strcmp(tok1, "ZCHANGE") == 0)
202                            do_zchange((HWND) strtoul(tok2, NULL, 0), (HWND) strtoul(tok3, NULL, 0));
203          }          }
204  }  }
205    
206    static LRESULT CALLBACK
207    wndproc(HWND hwnd, UINT message, WPARAM wparam, LPARAM lparam)
208    {
209            if (message == WM_WTSSESSION_CHANGE)
210            {
211                    if (wparam == WTS_REMOTE_CONNECT)
212                            vchannel_write("HELLO,0x%08x", 1);
213            }
214    
215            return DefWindowProc(hwnd, message, wparam, lparam);
216    }
217    
218    static BOOL
219    register_class(void)
220    {
221            WNDCLASSEX wcx;
222    
223            memset(&wcx, 0, sizeof(wcx));
224    
225            wcx.cbSize = sizeof(wcx);
226            wcx.lpfnWndProc = wndproc;
227            wcx.hInstance = g_instance;
228            wcx.lpszClassName = "SeamlessClass";
229    
230            return RegisterClassEx(&wcx);
231    }
232    
233    static BOOL
234    create_wnd(void)
235    {
236            if (!register_class())
237                    return FALSE;
238    
239            g_hwnd = CreateWindow("SeamlessClass",
240                                  "Seamless Window",
241                                  WS_OVERLAPPEDWINDOW,
242                                  CW_USEDEFAULT,
243                                  CW_USEDEFAULT,
244                                  CW_USEDEFAULT,
245                                  CW_USEDEFAULT, (HWND) NULL, (HMENU) NULL, g_instance, (LPVOID) NULL);
246    
247            if (!g_hwnd)
248                    return FALSE;
249    
250            return TRUE;
251    }
252    
253  int WINAPI  int WINAPI
254  WinMain(HINSTANCE instance, HINSTANCE prev_instance, LPSTR cmdline, int cmdshow)  WinMain(HINSTANCE instance, HINSTANCE prev_instance, LPSTR cmdline, int cmdshow)
255  {  {
# Line 162  WinMain(HINSTANCE instance, HINSTANCE pr Line 261  WinMain(HINSTANCE instance, HINSTANCE pr
261    
262          g_instance = instance;          g_instance = instance;
263    
264          hookdll = LoadLibrary("hookdll.dll");          hookdll = LoadLibrary("seamlessrdp.dll");
265          if (!hookdll)          if (!hookdll)
266          {          {
267                  message("Could not load hook DLL. Unable to continue.");                  message("Could not load hook DLL. Unable to continue.");
# Line 172  WinMain(HINSTANCE instance, HINSTANCE pr Line 271  WinMain(HINSTANCE instance, HINSTANCE pr
271          set_hooks_fn = (set_hooks_proc_t) GetProcAddress(hookdll, "SetHooks");          set_hooks_fn = (set_hooks_proc_t) GetProcAddress(hookdll, "SetHooks");
272          remove_hooks_fn = (remove_hooks_proc_t) GetProcAddress(hookdll, "RemoveHooks");          remove_hooks_fn = (remove_hooks_proc_t) GetProcAddress(hookdll, "RemoveHooks");
273          instance_count_fn = (get_instance_count_proc_t) GetProcAddress(hookdll, "GetInstanceCount");          instance_count_fn = (get_instance_count_proc_t) GetProcAddress(hookdll, "GetInstanceCount");
274            g_move_window_fn = (move_window_proc_t) GetProcAddress(hookdll, "SafeMoveWindow");
275    
276          if (!set_hooks_fn || !remove_hooks_fn || !instance_count_fn)          if (!set_hooks_fn || !remove_hooks_fn || !instance_count_fn || !g_move_window_fn)
277          {          {
278                  FreeLibrary(hookdll);                  FreeLibrary(hookdll);
279                  message("Hook DLL doesn't contain the correct functions. Unable to continue.");                  message("Hook DLL doesn't contain the correct functions. Unable to continue.");
# Line 188  WinMain(HINSTANCE instance, HINSTANCE pr Line 288  WinMain(HINSTANCE instance, HINSTANCE pr
288                  return -1;                  return -1;
289          }          }
290    
291            if (!create_wnd())
292            {
293                    FreeLibrary(hookdll);
294                    message("Couldn't create a window to catch events.");
295                    return -1;
296            }
297    
298            WTSRegisterSessionNotification(g_hwnd, NOTIFY_FOR_THIS_SESSION);
299    
300          vchannel_open();          vchannel_open();
301    
302            vchannel_write("HELLO,0x%08x", 0);
303    
304          set_hooks_fn();          set_hooks_fn();
305    
306          /* Since we don't see the entire desktop we must resize windows          /* Since we don't see the entire desktop we must resize windows
307             immediatly. */             immediatly. */
308          SystemParametersInfo(SPI_SETDRAGFULLWINDOWS, TRUE, NULL, 0);          SystemParametersInfo(SPI_SETDRAGFULLWINDOWS, TRUE, NULL, 0);
309    
310            /* Disable screen saver since we cannot catch its windows. */
311            SystemParametersInfo(SPI_SETSCREENSAVEACTIVE, FALSE, NULL, 0);
312    
313          if (strlen(cmdline) == 0)          if (strlen(cmdline) == 0)
314          {          {
315                  message("No command line specified.");                  message("No command line specified.");
# Line 207  WinMain(HINSTANCE instance, HINSTANCE pr Line 321  WinMain(HINSTANCE instance, HINSTANCE pr
321                  DWORD exitcode;                  DWORD exitcode;
322                  PROCESS_INFORMATION proc_info;                  PROCESS_INFORMATION proc_info;
323                  STARTUPINFO startup_info;                  STARTUPINFO startup_info;
324                    MSG msg;
325    
326                  memset(&startup_info, 0, sizeof(STARTUPINFO));                  memset(&startup_info, 0, sizeof(STARTUPINFO));
327                  startup_info.cb = sizeof(STARTUPINFO);                  startup_info.cb = sizeof(STARTUPINFO);
# Line 218  WinMain(HINSTANCE instance, HINSTANCE pr Line 333  WinMain(HINSTANCE instance, HINSTANCE pr
333                  {                  {
334                          do                          do
335                          {                          {
336                                    while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE))
337                                    {
338                                            TranslateMessage(&msg);
339                                            DispatchMessage(&msg);
340                                    }
341                                  process_cmds();                                  process_cmds();
342                                  Sleep(100);                                  Sleep(100);
343                                  GetExitCodeProcess(proc_info.hProcess, &exitcode);                                  GetExitCodeProcess(proc_info.hProcess, &exitcode);
# Line 238  WinMain(HINSTANCE instance, HINSTANCE pr Line 358  WinMain(HINSTANCE instance, HINSTANCE pr
358                  }                  }
359          }          }
360    
361            WTSUnRegisterSessionNotification(g_hwnd);
362    
363          remove_hooks_fn();          remove_hooks_fn();
364    
365          FreeLibrary(hookdll);          FreeLibrary(hookdll);
Legend:
Removed from v.1077  
changed lines
  Added in v.1147
  ViewVC Help
Powered by ViewVC 1.1.26