git-svn-id: https://server.scottellis.com.au/svn/mim_plugs@12 4f64403b-2f21-0410-a795-97e2b3489a10

1 files changed, 324 insertions, 0 deletions

diff --git a/ping_protocol/utils.cpp b/ping_protocol/utils.cpp
new file mode 100644
index 0000000..1f9c839
--- /dev/null
+++ b/ping_protocol/utils.cpp
@@ -0,0 +1,324 @@
+#include "stdafx.h"

+#include "utils.h"

+

+bool use_raw_ping = true;

+/*

+std::string trim_string(const std::string &in) {

+	std::string out;

+

+	int i = 0;

+	while(i < in.size() && in[i] == ' ')

+		i++;

+	for(; i < in.size(); i++) {

+		out += in[i];

+	}

+

+	while(out.size() && out[out.size() - 1] == ' ') {

+		out = out.substr(0, out.size() - 1);

+	}

+

+

+	return out;

+}

+

+void capitalize(std::string &in) {

+	for(int i = 0; i < in.length(); i++) {

+		if(in[i] >= 'a' && in[i] <= 'z')

+			in[i] = 'A' + (in[i] - 'a');

+	}

+}

+*/

+LRESULT CALLBACK NullWindowProc( HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam )

+{

+	switch( message ) {

+		case WM_COMMAND: {

+			PUDeletePopUp( hWnd );

+			break;

+		}

+

+		case WM_CONTEXTMENU:

+			PUDeletePopUp( hWnd );

+			break;

+	}

+

+	return DefWindowProc(hWnd, message, wParam, lParam);

+}

+

+void CALLBACK sttMainThreadCallback( ULONG dwParam )

+{

+	POPUPDATAEX* ppd = ( POPUPDATAEX* )dwParam;

+

+	if ( ServiceExists(MS_POPUP_ADDPOPUPEX) )

+		CallService( MS_POPUP_ADDPOPUPEX, ( WPARAM )ppd, 0 );

+	else 

+		if ( ServiceExists(MS_POPUP_ADDPOPUP) )

+			CallService( MS_POPUP_ADDPOPUP, ( WPARAM )ppd, 0 );

+

+	free( ppd );

+}

+

+void __stdcall	ShowPopup( const char* line1, const char* line2, int flags )

+{

+	if(CallService(MS_SYSTEM_TERMINATED, 0, 0)) return;

+

+	if ( !ServiceExists( MS_POPUP_ADDPOPUP )) {	

+		MessageBox( NULL, line2, PROTO " Message", MB_OK | MB_ICONINFORMATION );

+		return;

+	}

+

+	POPUPDATAEX* ppd = ( POPUPDATAEX* )calloc( sizeof( POPUPDATAEX ), 1 );

+

+	ppd->lchContact = NULL;

+	ppd->lchIcon = (flags ? LoadSkinnedProtoIcon(PROTO, options.rstatus) : LoadSkinnedProtoIcon(PROTO, options.nrstatus));

+	strcpy( ppd->lpzContactName, line1 );

+	strcpy( ppd->lpzText, line2 );

+

+	ppd->colorBack = GetSysColor( COLOR_BTNFACE );

+	ppd->colorText = GetSysColor( COLOR_WINDOWTEXT );

+	ppd->iSeconds = 10;

+

+	ppd->PluginWindowProc = ( WNDPROC )NullWindowProc;

+	ppd->PluginData = NULL;

+

+	QueueUserAPC( sttMainThreadCallback , mainThread, ( ULONG )ppd );

+}

+

+// service functions

+

+// wParam is zero

+// lParam is address of PINGADDRESS structure where ping result is placed (i.e. modifies 'responding' 

+// and 'round_trip_time')

+int PluginPing(WPARAM wParam,LPARAM lParam)

+{

+	PINGADDRESS *pa = (PINGADDRESS *)lParam;

+

+	if(pa->port == -1) {

+		if(use_raw_ping) {

+			pa->round_trip_time = raw_ping(pa->pszName, options.ping_timeout * 1000);

+			pa->responding = (pa->round_trip_time != -1);

+		} else {

+			// ICMP echo

+			IP_ECHO_REPLY result;

+			pa->responding = ICMP::get_instance()->ping(pa->pszName, result);

+			if(pa->responding)

+				pa->round_trip_time = result.RoundTripTime;

+			else

+				pa->round_trip_time = -1;

+		}

+	} else if(hNetlibUser) {

+		// TCP connect

+

+		clock_t start_tcp = clock();

+		

+		//GetLocalTime(&systime);

+		NETLIBOPENCONNECTION conn = {0};

+		conn.cbSize = sizeof(NETLIBOPENCONNECTION);

+		conn.szHost = pa->pszName;

+		conn.wPort = pa->port;

+		conn.flags = NLOCF_V2;

+		conn.timeout = options.ping_timeout; 

+

+		HANDLE s = (HANDLE)CallService(MS_NETLIB_OPENCONNECTION, (WPARAM)hNetlibUser, (LPARAM)&conn);

+

+		clock_t end_tcp = clock();

+		

+		if(s) {

+			LINGER l;

+			char buf[1024];

+			SOCKET socket = (SOCKET)CallService(MS_NETLIB_GETSOCKET, (WPARAM)s, (LPARAM)NLOCF_HTTP);

+			if(socket != INVALID_SOCKET) {

+				l.l_onoff = 1;

+				l.l_linger = 0;

+				setsockopt(socket, SOL_SOCKET, SO_LINGER, (char *)&l, sizeof(l));

+

+				Netlib_Send(s, "OUT\r\n\r\n", 7, 0); //MSG_RAW);

+

+				//Sleep(ICMP::get_instance()->get_timeout());

+				Sleep(options.ping_timeout * 1000);

+				unsigned long bytes_remaining;

+				ioctlsocket(socket, FIONBIO, &bytes_remaining);

+

+				if(bytes_remaining > 0) {

+					int retval, rx = 0;

+					while((retval = Netlib_Recv(s, buf, 1024, 0)) != SOCKET_ERROR && (retval > 0) && rx < 2048) {

+						rx += retval; // recv at most 2kb before closing connection

+					}

+				}

+				//closesocket(socket);

+

+			}

+

+			Netlib_CloseHandle(s);

+

+			pa->responding = true;

+			pa->round_trip_time = (int)(((end_tcp - start_tcp) / (double)CLOCKS_PER_SEC) * 1000);

+		} else {

+			pa->responding = false;

+			pa->round_trip_time = -1;

+		}

+

+	}

+	return 0;

+}

+

+void Lock(CRITICAL_SECTION *cs, char *lab) {

+//	if(logging) {

+//		std::ostringstream oss1;

+//		oss1 << "Locking cs: " << cs << ", " << lab;

+//		CallService(PROTO "/Log", (WPARAM)oss1.str().c_str(), 0);

+//	}

+	EnterCriticalSection(cs);

+//	if(logging) {

+//		std::ostringstream oss2;

+//		oss2 << "Locked cs: " << cs;

+//		CallService(PROTO "/Log", (WPARAM)oss2.str().c_str(), 0);

+//	}

+}

+

+void Unlock(CRITICAL_SECTION *cs) {

+//	if(logging) {

+//		std::ostringstream oss1;

+//		oss1 << "Unlocking cs: " << cs;

+//		CallService(PROTO "/Log", (WPARAM)oss1.str().c_str(), 0);

+//	}

+	LeaveCriticalSection(cs);

+}

+

+int PingDisableAll(WPARAM wParam, LPARAM lParam) {

+	PINGLIST pl;

+	CallService(PROTO "/GetPingList", 0, (LPARAM)&pl);

+	for(PINGLIST::iterator i = pl.begin(); i != pl.end(); i++) {

+		i->status = options.off_status;

+		DBWriteContactSettingWord(i->hContact, PROTO, "Status", i->status);

+		set_changing_clist_handle(TRUE);

+		DBWriteContactSettingString(i->hContact, "CList", "MyHandle", i->pszLabel);

+		set_changing_clist_handle(FALSE);

+	}

+	CallService(PROTO "/SetPingList", (WPARAM)&pl, 0);

+	return 0;

+}

+

+int PingEnableAll(WPARAM wParam, LPARAM lParam) {

+	PINGLIST pl;

+	CallService(PROTO "/GetPingList", 0, (LPARAM)&pl);

+	for(PINGLIST::iterator i = pl.begin(); i != pl.end(); i++) {

+		if(i->status == options.off_status) {

+			i->status = options.nrstatus;

+			DBWriteContactSettingWord(i->hContact, PROTO, "Status", i->status);

+		}

+	}

+	CallService(PROTO "/SetPingList", (WPARAM)&pl, 0);

+	return 0;

+}

+

+

+int ToggleEnabled(WPARAM wParam, LPARAM lParam) {

+	PINGLIST pl;

+	CallService(PROTO "/GetPingList", 0, (LPARAM)&pl);

+	for(PINGLIST::iterator i = pl.begin(); i != pl.end(); i++) {

+		if(i->hContact == (HANDLE)wParam) {

+

+			if(i->status == options.off_status)

+				i->status = options.nrstatus;

+			else {

+				i->status = options.off_status;

+				set_changing_clist_handle(TRUE);

+				DBWriteContactSettingString(i->hContact, "CList", "MyHandle", i->pszLabel);

+				set_changing_clist_handle(FALSE);

+			}

+			DBWriteContactSettingWord(i->hContact, PROTO, "Status", i->status);

+		}

+	}

+	CallService(PROTO "/SetPingList", (WPARAM)&pl, 0);

+	return 0;

+}

+

+int EditContact(WPARAM wParam, LPARAM lParam) {

+	PINGLIST pl;

+	HWND hwndList = (HWND)CallService(MS_CLUI_GETHWND, 0, 0);

+

+	CallService(PROTO "/GetPingList", 0, (LPARAM)&pl);

+	for(PINGLIST::iterator i = pl.begin(); i != pl.end(); i++) {

+		if(i->hContact == (HANDLE)wParam) {

+

+			add_edit_addr = *i;

+	

+			if(DialogBox(hInst, MAKEINTRESOURCE(IDD_DIALOG3), hwndList, DlgProcDestEdit) == IDOK) {

+	

+				*i = add_edit_addr;

+				CallService(PROTO "/SetAndSavePingList", (WPARAM)&pl, 0);

+				return 0;

+			}

+		}

+	}

+	return 1;

+}

+

+int ContactDblClick(WPARAM wParam, LPARAM lParam) {

+	PINGLIST pl;

+	CallService(PROTO "/GetPingList", 0, (LPARAM)&pl);

+	for(PINGLIST::iterator i = pl.begin(); i != pl.end(); i++) {

+		if(i->hContact == (HANDLE)wParam) {

+			if(strlen(i->pszCommand)) {

+				ShellExecute(0, "open", i->pszCommand, i->pszParams, 0, SW_SHOW);

+			} else {

+				CallService(PROTO "/ToggleEnabled", wParam, 0);

+			}

+		}

+	}

+	return 0;

+}

+

+

+void import_ping_address(int index, PINGADDRESS &pa) {

+	std::ostringstream os1, os2, os3, os4, os5, os6, os7;

+	os1 << "Address" << index;

+	os2 << "Label" << index;

+	os3 << "Enabled" << index;

+	os4 << "Port" << index;

+	os5 << "Proto" << index;

+	os6 << "Status" << index;

+	os7 << "Status2" << index;

+

+	DBVARIANT dbv;

+	DBGetContactSetting(0, "PingPlug", os1.str().c_str(), &dbv);

+	strncpy(pa.pszName, dbv.pszVal, MAX_PINGADDRESS_STRING_LENGTH);

+	DBFreeVariant(&dbv);

+	DBGetContactSetting(0, "PingPlug", os2.str().c_str(), &dbv);

+	strncpy(pa.pszLabel, dbv.pszVal, MAX_PINGADDRESS_STRING_LENGTH);

+	DBFreeVariant(&dbv);

+

+	pa.port = (int)DBGetContactSettingDword(0, "PingPlug", os4.str().c_str(), -1);

+

+	DBGetContactSetting(0, "PingPlug", os5.str().c_str(), &dbv);

+	strncpy(pa.pszProto, dbv.pszVal, MAX_PINGADDRESS_STRING_LENGTH);

+	DBFreeVariant(&dbv);

+

+	pa.set_status = DBGetContactSettingWord(0, "PingPlug", os6.str().c_str(), ID_STATUS_ONLINE);

+	pa.get_status = DBGetContactSettingWord(0, "PingPlug", os7.str().c_str(), ID_STATUS_OFFLINE);

+

+	pa.responding = false;

+	pa.round_trip_time = 0;

+	pa.miss_count = 0;

+	pa.index = index;

+	pa.pszCommand[0] = '\0';

+

+	pa.hContact = 0;

+	pa.status = options.nrstatus;

+}

+

+// read in addresses from old pingplug

+void import_ping_addresses() {

+	int count = DBGetContactSettingDword(0, "PingPlug", "NumEntries", 0);

+	PINGADDRESS pa;

+

+	EnterCriticalSection(&list_cs);

+	list_items.clear();

+	for(int index = 0; index < count; index++) {

+		import_ping_address(index, pa);

+		list_items.push_back(pa);

+	}

+	std::sort(list_items.begin(), list_items.end(), SAscendingSort());

+	write_ping_addresses();

+	LeaveCriticalSection(&list_cs);

+}