1. int main(int argc, const char* argv[]) {
2. int result = 0;
3. try {
4. // - Initialise the available loggers
5. //freopen("\\\\drupe\\tjr\\WinVNC4.log","ab",stderr);
6. //setbuf(stderr, 0);
7. initStdIOLoggers();
8. initFileLogger("C:\\temp\\WinVNC4.log");
9. rfb::win32::initEventLogLogger(VNCServerService::Name);
10. // - By default, just log errors to stderr
11. logParams.setParam("*:stderr:0");
12. // - Print program details and process the command line
13. programInfo();
14. processParams(argc, argv);
15. // - Run the server if required
16. if (runServer) {
17. if (close_console) {
18. vlog.info("closing console");
19. if (!FreeConsole())
20. vlog.info("unable to close console:%u", GetLastError());
21. }
22. network::TcpSocket::initTcpSockets();
23. VNCServerWin32 server;
24. if (runAsService) {
25. printf("Starting Service-Mode VNC Server.\n");
26. VNCServerService service(server);
27. ///>启动服务
28. service.start();
29. result = service.getStatus().dwWin32ExitCode;
30. } else {
31. printf("Starting User-Mode VNC Server.\n");
32. ///>启动server
33. result = server.run();
34. }
35. }
36. vlog.debug("WinVNC service destroyed");
37. } catch (rdr::Exception& e) {
38. try {
39. vlog.error("Fatal Error: %s", e.str());
40. } catch (...) {
41. fprintf(stderr, "WinVNC: Fatal Error: %s\n", e.str());
42. }
43. if (!runAsService)
44. MsgBox(0, TStr(e.str()), MB_ICONERROR | MB_OK);
45. }
46. vlog.debug("WinVNC process quitting");
47. return result;
48. }

1. processParams(argc, argv);

1. void
2. processParams(int argc, const char* argv[]) {
3. for (int i=1; i<argc; i++) {
4. try {
5. if (strcasecmp(argv[i], "-connect") == 0) {
6. runServer = false;
7. CharArray host;
8. if (i+1 < argc) {
9. host.buf = strDup(argv[i+1]);
10. } else {
11. AddNewClientDialog ancd;
12. if (ancd.showDialog())
13. host.buf = strDup(ancd.getHostName());
14. }
15. if (host.buf) {
16. HWND hwnd = FindWindow(0, _T("winvnc::IPC_Interface"));
17. COPYDATASTRUCT copyData;
18. copyData.dwData = 1; // *** AddNewClient
19. copyData.cbData = strlen(host.buf);
20. copyData.lpData = (void*)host.buf;
21. i++;
22. SendMessage(hwnd, WM_COPYDATA, 0, (LPARAM)&copyData);
23. printf("Sent connect request to VNC Server...\n");
24. }
25. } else if (strcasecmp(argv[i], "-disconnect") == 0) {
26. HWND hwnd = FindWindow(0, _T("winvnc::IPC_Interface"));
27. COPYDATASTRUCT copyData;
28. copyData.dwData = 2; // *** DisconnectClients
29. copyData.lpData = 0;
30. copyData.cbData = 0;
31. SendMessage(hwnd, WM_COPYDATA, 0, (LPARAM)©Data);
32. printf("Sent disconnect request to VNC Server...\n");
33. runServer = false;
34. } else if (strcasecmp(argv[i], "-start") == 0) {
35. printf("Attempting to start service...\n");
36. runServer = false;
37. if (rfb::win32::startService(VNCServerService::Name))
38. printf("Started service successfully\n");
39. } else if (strcasecmp(argv[i], "-stop") == 0) {
40. printf("Attempting to stop service...\n");
41. runServer = false;
42. if (rfb::win32::stopService(VNCServerService::Name))
43. printf("Stopped service successfully\n");
44. } else if (strcasecmp(argv[i], "-status") == 0) {
45. printf("Querying service status...\n");
46. runServer = false;
47. rfb::win32::printServiceStatus(VNCServerService::Name);
48. } else if (strcasecmp(argv[i], "-service") == 0) {
49. printf("Run in service mode\n");
50. runAsService = true;
51. } else if (strcasecmp(argv[i], "-register") == 0) {
52. printf("Attempting to register service...\n");
53. runServer = false;
54. int j = i;
55. i = argc;
56. if (rfb::win32::registerService(VNCServerService::Name,
57. _T("VNC Server Version 4"),
58. argc-(j+1), &argv[j+1]))
59. printf("Registered service successfully\n");
60. } else if (strcasecmp(argv[i], "-unregister") == 0) {
61. printf("Attempting to unregister service...\n");
62. runServer = false;
63. if (rfb::win32::unregisterService(VNCServerService::Name))
64. printf("Unregistered service successfully\n");
65. } else if (strcasecmp(argv[i], "-noconsole") == 0) {
66. close_console = true;
67. } else if ((strcasecmp(argv[i], "-help") == 0) ||
68. (strcasecmp(argv[i], "--help") == 0) ||
69. (strcasecmp(argv[i], "-h") == 0) ||
70. (strcasecmp(argv[i], "/?") == 0)) {
71. runServer = false;
72. programUsage();
73. break;
74. } else {
75. // Try to process <option>=<value>, or -<bool>
76. if (Configuration::setParam(argv[i], true))
77. continue;
78. // Try to process -<option> <value>
79. if ((argv[i][0] == '-') && (i+1 < argc)) {
80. if (Configuration::setParam(&argv[i][1], argv[i+1], true)) {
81. i++;
82. continue;
83. }
84. }
85. // Nope.  Show them usage and don't run the server
86. runServer = false;
87. programUsage();
88. break;
89. }
90. } catch (rdr::Exception& e) {
91. vlog.error(e.str());
92. }
93. }
94. }

1. if (host.buf) {
2. HWND hwnd = FindWindow(0, _T("winvnc::IPC_Interface"));
3. COPYDATASTRUCT copyData;
4. copyData.dwData = 1; // *** AddNewClient
5. copyData.cbData = strlen(host.buf);
6. copyData.lpData = (void*)host.buf;
7. i++;
8. SendMessage(hwnd, WM_COPYDATA, 0, (LPARAM)©Data);
9. printf("Sent connect request to VNC Server...\n");
10. }

1. if (runAsService) {
2. printf("Starting Service-Mode VNC Server.\n");
3. VNCServerService service(server);
4. ///>启动服务
5. service.start();
6. result = service.getStatus().dwWin32ExitCode;
7. } else {
8. printf("Starting User-Mode VNC Server.\n");
9. ///>启动server
10. result = server.run();
11. }

1. if (!FreeConsole())

1. VNCServerWin32 server;
2. if (runAsService) {
3. printf("Starting Service-Mode VNC Server.\n");
4. VNCServerService service(server);
5. ///>启动服务
6. service.start();
7. result = service.getStatus().dwWin32ExitCode;

1. result = server.run();

1. ///>服务器启动
2. int VNCServerWin32::run() {
3. { Lock l(runLock);
4. hostThread = Thread::self();
5. runServer = true;
6. }
7. // - Register for notification of configuration changes
8. ///>注册表设置
9. if (isServiceProcess())
10. config.setKey(HKEY_LOCAL_MACHINE, RegConfigPath);
11. else
12. config.setKey(HKEY_CURRENT_USER, RegConfigPath);
13. config.setNotifyThread(Thread::self(), VNCM_REG_CHANGED);
14. // - Create the tray icon if possible
15. STrayIconThread trayIcon(*this, IDI_ICON, IDI_CONNECTED, IDR_TRAY);
16. DWORD result = 0;
17. try {
18. // - Create some managed listening sockets
19. ///>创建监听socket
20. ManagedListener rfb(&sockMgr, &vncServer);
21. ManagedListener http(&sockMgr, httpServer);
22. // - Continue to operate until WM_QUIT is processed
23. MSG msg;
24. do {
25. // -=- Make sure we're listening on the right ports.
26. ///>设置监听端口
27. rfb.setPort(port_number, localHost);
28. http.setPort(http_port, localHost);
29. // -=- Update the Java viewer's web page port number.
30. httpServer->setRFBport(rfb.sock ? port_number : 0);
31. // -=- Update the TCP address filter for both ports, if open.
32. CharArray pattern;
33. pattern.buf = hosts.getData();
34. if (!localHost) {
35. rfb.setFilter(pattern.buf);
36. http.setFilter(pattern.buf);
37. }
38. // - If there is a listening port then add the address to the
39. //   tray icon's tool-tip text.
40. {
41. const TCHAR* prefix = isServiceProcess() ?
42. _T("VNC Server (Service):") : _T("VNC Server (User):");
43. std::list<char*> addrs;
44. if (rfb.sock)
45. rfb.sock->getMyAddresses(&addrs);
46. else
47. addrs.push_front(strDup("Not accepting connections"));
48. std::list<char*>::iterator i, next_i;
49. int length = _tcslen(prefix)+1;
50. for (i=addrs.begin(); i!= addrs.end(); i++)
51. length += strlen(*i) + 1;
52. TCharArray toolTip(length);
53. _tcscpy(toolTip.buf, prefix);
54. for (i=addrs.begin(); i!= addrs.end(); i=next_i) {
55. next_i = i; next_i ++;
56. TCharArray addr = *i;    // Assumes ownership of string
57. _tcscat(toolTip.buf, addr.buf);
58. if (next_i != addrs.end())
59. _tcscat(toolTip.buf, _T(","));
60. }
61. trayIcon.setToolTip(toolTip.buf);
62. }
63. vlog.debug("Entering message loop");
64. // - Run the server until the registry changes, or we're told to quit
65. ///>进入socket监听，进入消息循环
66. while (sockMgr.getMessage(&msg, NULL, 0, 0)) {
67. if (msg.hwnd == 0) {
68. if (msg.message == VNCM_REG_CHANGED)
69. break;
70. if (msg.message == VNCM_COMMAND)
71. doCommand();
72. }
73. TranslateMessage(&msg);
74. DispatchMessage(&msg);
75. }
76. } while ((msg.message != WM_QUIT) || runServer);
77. vlog.debug("Server exited cleanly");
78. } catch (rdr::SystemException &s) {
79. vlog.error(s.str());
80. result = s.err;
81. } catch (rdr::Exception &e) {
82. vlog.error(e.str());
83. }
84. { Lock l(runLock);
85. runServer = false;
86. hostThread = 0;
87. }
88. return result;
89. }

1. if (isServiceProcess())
2. config.setKey(HKEY_LOCAL_MACHINE, RegConfigPath);
3. else
4. config.setKey(HKEY_CURRENT_USER, RegConfigPath);
5. config.setNotifyThread(Thread::self(), VNCM_REG_CHANGED);

1. // - Create the tray icon if possible
2. STrayIconThread trayIcon(*this, IDI_ICON, IDI_CONNECTED, IDR_TRAY);

1. STrayIconThread::STrayIconThread(VNCServerWin32& sm, UINT inactiveIcon_, UINT activeIcon_, UINT menu_)
2. : server(sm), inactiveIcon(inactiveIcon_), activeIcon(activeIcon_), menu(menu_),
3. windowHandle(0), runTrayIcon(true) {
4. start();
5. }

1. void STrayIconThread::run() {
2. while (runTrayIcon) {
3. if (rfb::win32::desktopChangeRequired() &&
4. !rfb::win32::changeDesktop())
5. Sleep(2000);
6. STrayIcon icon(*this);
7. windowHandle = icon.getHandle();
8. MSG msg;
9. while (runTrayIcon && ::GetMessage(&msg, 0, 0, 0) > 0) {
10. TranslateMessage(&msg);
11. DispatchMessage(&msg);
12. }
13. windowHandle = 0;
14. }
15. }

1. if (rfb::win32::desktopChangeRequired() &&
2. !rfb::win32::changeDesktop())
3. Sleep(2000);

1. STrayIcon icon(*this);
2. windowHandle = icon.getHandle();

1. class winvnc::STrayIcon : public TrayIcon

1. class TrayIcon : public MsgWindow

1. class MsgWindow {
2. public:
3. MsgWindow(const TCHAR* _name);
4. virtual ~MsgWindow();
5. const TCHAR* getName() {return name.buf;}
6. HWND getHandle() const {return handle;}
7. virtual LRESULT processMessage(UINT msg, WPARAM wParam, LPARAM lParam);
8. protected:
9. TCharArray name;
10. HWND handle;
11. };

1. MsgWindow::MsgWindow(const TCHAR* name_) : name(tstrDup(name_)), handle(0) {
2. vlog.debug("creating window \"%s\"", (const char*)CStr(name.buf));
3. handle = CreateWindow((const TCHAR*)baseClass.classAtom, name.buf, WS_OVERLAPPED,
4. 0, 0, 10, 10, 0, 0, baseClass.instance, this);
5. if (!handle) {
6. throw rdr::SystemException("unable to create WMNotifier window instance", GetLastError());
7. }
8. vlog.debug("created window \"%s\" (%x)", (const char*)CStr(name.buf), handle);
9. }

1. MsgWindowClass::MsgWindowClass() : classAtom(0) {
2. WNDCLASS wndClass;
3. wndClass.style = 0;
4. wndClass.lpfnWndProc = MsgWindowProc;
5. wndClass.cbClsExtra = 0;
6. wndClass.cbWndExtra = 0;
7. wndClass.hInstance = instance = GetModuleHandle(0);
8. wndClass.hIcon = 0;
9. wndClass.hCursor = 0;
10. wndClass.hbrBackground = 0;
11. wndClass.lpszMenuName = 0;
12. wndClass.lpszClassName = _T("rfb::win32::MsgWindowClass");
13. classAtom = RegisterClass(&wndClass);
14. if (!classAtom) {
15. throw rdr::SystemException("unable to register MsgWindow window class", GetLastError());
16. }
17. }

1. LRESULT CALLBACK MsgWindowProc(HWND wnd, UINT msg, WPARAM wParam, LPARAM lParam) {
2. LRESULT result;
3. if (msg == WM_CREATE)
4. SetWindowLong(wnd, GWL_USERDATA, (long)((CREATESTRUCT*)lParam)->lpCreateParams);
5. else if (msg == WM_DESTROY)
6. SetWindowLong(wnd, GWL_USERDATA, 0);
7. MsgWindow* _this = (MsgWindow*) GetWindowLong(wnd, GWL_USERDATA);
8. if (!_this) {
9. vlog.info("null _this in %x, message %x", wnd, msg);
10. return SafeDefWindowProc(wnd, msg, wParam, lParam);
11. }
12. try {
13. result = _this->processMessage(msg, wParam, lParam);
14. } catch (rdr::Exception& e) {
15. vlog.error("untrapped: %s", e.str());
16. }
17. return result;
18. };

1. TrayIcon() : MsgWindow(_T("VNCTray")) {
2. fdef NOTIFYICONDATA_V1_SIZE
3. nid.cbSize = NOTIFYICONDATA_V1_SIZE;
4. lse
5. nid.cbSize = sizeof(NOTIFYICONDATA);
6. ndif
7. nid.hWnd = getHandle();
8. nid.uID = 0;
9. nid.hIcon = 0;
10. nid.uFlags = NIF_ICON | NIF_MESSAGE;
11. nid.uCallbackMessage = WM_USER;
12. }

1. STrayIcon(STrayIconThread& t) : thread(t),
2. vncConfig(_T("vncconfig.exe"), isServiceProcess() ? _T("-noconsole -service") : _T("-noconsole")),
3. vncConnect(_T("winvnc4.exe"), _T("-connect")) {
4. // ***
5. SetWindowText(getHandle(), _T("winvnc::IPC_Interface"));
6. // ***
7. SetTimer(getHandle(), 1, 3000, 0);
8. PostMessage(getHandle(), WM_TIMER, 1, 0);
9. PostMessage(getHandle(), WM_SET_TOOLTIP, 0, 0);
10. }

1. LaunchProcess vncConfig;
2. LaunchProcess vncConnect;

1. CurrentUserToken token;
2. if (token.isValid())
3. vncConnect.start(isServiceProcess() ? (HANDLE)token : 0);
4. else
5. vlog.error("Options: unknown current user");

1. void LaunchProcess::start(HANDLE userToken) {
2. if (procInfo.hProcess && (WaitForSingleObject(procInfo.hProcess, 0) != WAIT_OBJECT_0))
3. return;
4. await();
5. // - Create storage for the process startup information
6. STARTUPINFO sinfo;
7. memset(&sinfo, 0, sizeof(sinfo));
8. sinfo.cb = sizeof(sinfo);
9. // - Concoct a suitable command-line
10. TCharArray exePath;
11. if (!tstrContains(exeName.buf, _T('\\'))) {
12. ModuleFileName filename;
13. TCharArray path; splitPath(filename.buf, &path.buf, 0);
14. exePath.buf = new TCHAR[_tcslen(path.buf) + _tcslen(exeName.buf) + 2];
15. _stprintf(exePath.buf, _T("%s\\%s"), path.buf, exeName.buf);
16. } else {
17. exePath.buf = tstrDup(exeName.buf);
18. }
19. // - Start the VNC server
20. // Note: We specify the exe's precise path in the ApplicationName parameter,
21. //       AND include the name as the first part of the CommandLine parameter,
22. //       because CreateProcess doesn't make ApplicationName argv[0] in C programs.
23. TCharArray cmdLine(_tcslen(exeName.buf) + 3 + _tcslen(params.buf) + 1);
24. _stprintf(cmdLine.buf, _T("\"%s\" %s"), exeName.buf, params.buf);
25. #ifdef _DEBUG
26. DWORD flags = CREATE_NEW_CONSOLE;
27. #else
28. DWORD flags = CREATE_NO_WINDOW;
29. #endif
30. BOOL success;
31. if (userToken)
32. success = CreateProcessAsUser(userToken, exePath.buf, cmdLine.buf, 0, 0, FALSE, flags, 0, 0, &sinfo, &procInfo);
33. else
34. success = CreateProcess(exePath.buf, cmdLine.buf, 0, 0, FALSE, flags, 0, 0, &sinfo, &procInfo);
35. if (!success)
36. throw rdr::SystemException("unable to launch process", GetLastError());
37. // Wait for it to finish initialising
38. WaitForInputIdle(procInfo.hProcess, 15000);
39. }

1. if (rfb::win32::desktopChangeRequired()) {
2. SendMessage(getHandle(), WM_CLOSE, 0, 0);
3. return 0;
4. }
5. setIcon(thread.server.isServerInUse() ? thread.activeIcon : thread.inactiveIcon);
6. return 0;

1. case WM_SET_TOOLTIP:
2. {
3. rfb::Lock l(thread.lock);
4. if (thread.toolTip.buf)
5. setToolTip(thread.toolTip.buf);
6. }

1. MSG msg;
2. while (runTrayIcon && ::GetMessage(&msg, 0, 0, 0) > 0) {
3. TranslateMessage(&msg);
4. DispatchMessage(&msg);
5. }

1. ManagedListener rfb(&sockMgr, &vncServer);
2. ManagedListener http(&sockMgr, httpServer);

1. MSG msg;
2. do {
3. // 第一件事：设置监听端口
4. // 第二件事：Update the Java viewer's web page port number
5. //           Update the TCP address filter for both ports, if open.
6. //
7. // 第三件事：如果有一个正在监听的端口，那么就增加地址到icon的tooltip(提示信息)
8. //
9. // - Run the server until the registry changes, or we're told to quit
10. ///>进入socket监听，进入消息循环
11. while (sockMgr.getMessage(&msg, NULL, 0, 0)) {
12. if (msg.hwnd == 0) {
13. if (msg.message == VNCM_REG_CHANGED)
14. break;
15. if (msg.message == VNCM_COMMAND)
16. doCommand();
17. }
18. TranslateMessage(&msg);
19. DispatchMessage(&msg);
20. }
21. } while ((msg.message != WM_QUIT) || runServer);

缩写为TLS。进程中的全局变量与函数内定义的静态(static)变量，是各个线程都可以访问的共享变量。在一个线程修改的内存内容，对所有线程都生效。这是一个优点也是一个缺点。说它是优点，线程的数据交换变得非常快捷。说它是缺点，一个线程死掉了，其它线程也性命不保； 多个线程访问共享数据，需要昂贵的同步开销，也容易造成同步相关的BUG。