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

1 files changed, 214 insertions, 0 deletions

diff --git a/MySpace/NetMessage.cpp b/MySpace/NetMessage.cpp
new file mode 100644
index 0000000..92ce21e
--- /dev/null
+++ b/MySpace/NetMessage.cpp
@@ -0,0 +1,214 @@
+#include "common.h"

+#include "NetMessage.h"

+#include <cstdlib>

+

+NMString::NMString(): text(0) {

+}

+

+NMString::NMString(const NMString &other) {

+	int len = strlen(other.text) + 1;

+	text = new char[len + 1];

+	mir_snprintf(text, len, "%s", other.text);

+}

+

+NMString::NMString(char *t) {

+	int len = strlen(t) + 1;

+	text = new char[len + 1];

+	mir_snprintf(text, len, "%s", t);

+}

+

+const bool NMString::operator<(const NMString &other) const {

+	return strcmp(text, other.text) < 0;

+}

+

+const bool NMString::operator==(const NMString &other) const {

+	return strcmp(text, other.text) == 0;

+}

+

+NMString &NMString::operator=(const NMString &other) {

+	if(text) delete[] text;

+

+	int len = strlen(other.text) + 1;

+	text = new char[len + 1];

+	mir_snprintf(text, len, "%s", other.text);

+

+	return *this;

+}

+

+NMString::~NMString() {

+	delete[] text;

+}

+

+NetMessage::NetMessage(): Map<NMString, NMString>() {

+}

+

+NetMessage::~NetMessage()

+{

+}

+

+//static

+char *NetMessage::unescape_inplace(char *val) {

+	int read_pos = 0, write_pos = 0, len = strlen(val);

+	while(read_pos < len) {

+		if(val[read_pos] == '/' && read_pos + 1 < len) {

+			read_pos++;

+			if(val[read_pos] == '1') val[write_pos++] = '/';

+			else if(val[read_pos] == '2') val[write_pos++] = '\\';

+			else val[write_pos++] = '?';

+			read_pos++;

+		} else {

+			val[write_pos++] = val[read_pos++];

+		}

+	}

+	val[write_pos] = 0;

+

+	return val;

+}

+

+int NetMessage::parse(char *data, int size) {

+	int pos = 0;

+	bool finished = false;

+	char *key, *value;

+	int start, end = 0;

+	while(end < size && !finished) {

+		start = end + 1;

+		end = start;

+		while(end < size && data[end] != '\\') end++;

+		if(end < size) {

+			data[end] = 0;

+			key = &data[start];

+

+			start = end + 1;

+			end = start;

+			while(end < size && data[end] != '\\') end++;

+			if(end < size) {

+				data[end] = 0;

+				value = &data[start];

+			} else

+				value = 0;

+		} else {

+			key = value = 0;

+		}

+

+		if(key && strcmp(key, "final") == 0)

+			finished = true;

+		else if(key && value)

+			put(NMString(key), NMString(value));

+	}

+	return pos;

+}

+

+bool NetMessage::get_string(char *key, char *buff, int buffsize) {

+	NMString val;

+	if(get(NMString(key), val)) {

+		mir_snprintf(buff, buffsize, "%s", val.text);

+		unescape_inplace(buff);

+		return true;

+	}

+	return false;

+}

+

+bool NetMessage::get_data(char *key, char *buff, int *size) {

+	NMString val;

+	if(get(NMString(key), val)) {

+		NETLIBBASE64 nbd = {0};

+		nbd.pszEncoded = val.text;

+		nbd.cchEncoded = strlen(nbd.pszEncoded);

+		nbd.pbDecoded = (BYTE *)buff;

+		nbd.cbDecoded = *size;

+

+		//*size = Netlib_GetBase64DecodedBufferSize(nbd.cchEncoded);

+		CallService(MS_NETLIB_BASE64DECODE, 0, (LPARAM)&nbd);

+		*size = nbd.cbDecoded;

+

+		return true;

+	}

+	return false;

+}

+

+int NetMessage::get_int(char *key) {

+	NMString val;

+	if(get(NMString(key), val)) {

+		return atoi(val.text);

+	}

+	return 0;

+}

+

+ClientNetMessage::ClientNetMessage(): LinkedList< KeyValue >() {

+}

+

+ClientNetMessage::~ClientNetMessage() {

+}

+

+void ClientNetMessage::add_string(char *key, char *buff) {

+	char *temp = escape(buff);

+	KeyValue dat = KeyValue(NMString(key), NMString(temp));

+	delete temp;

+	add(dat);

+}

+

+void ClientNetMessage::add_data(char *key, char *data, int size) {

+	int len = Netlib_GetBase64EncodedBufferSize(size);

+	char *buff = new char[len];

+

+	NETLIBBASE64 nbd = {0};

+	nbd.pszEncoded = buff;

+	nbd.cchEncoded = len;

+	nbd.pbDecoded = (BYTE *)data;

+	nbd.cbDecoded = size;

+

+	//*size = Netlib_GetBase64DecodedBufferSize(nbd.cchEncoded);

+	CallService(MS_NETLIB_BASE64ENCODE, 0, (LPARAM)&nbd);

+

+	KeyValue dat = KeyValue(NMString(key), NMString(buff));

+	add(dat);

+}

+

+void ClientNetMessage::add_int(char *key, int i) {

+	char buff[254];

+	KeyValue dat(NMString(key), NMString(_itoa(i, buff, 10)));

+	add(dat);

+}

+

+//static 

+char *ClientNetMessage::escape(char *val) {

+	int len = strlen(val);

+	char *buff = new char[len * 2 + 1];

+	int read_pos = 0, write_pos = 0;

+	while(read_pos < len) {

+		if(val[read_pos] == '/') {

+			buff[write_pos++] = '/';

+			buff[write_pos++] = '1';

+			read_pos++;

+		} else if(val[read_pos] == '\\') {

+			buff[write_pos++] = '/';

+			buff[write_pos++] = '2';

+			read_pos++;

+		} else {

+			buff[write_pos++] = val[read_pos++];

+		}

+	}

+	buff[write_pos] = 0;

+	return buff;

+}

+

+int ClientNetMessage::make_packet(char *buff, int size) {

+	int pos = 0, key_len, val_len, len;

+	for(Iterator i = start(); i.has_val() && pos < size; i.next()) {

+		KeyValue v = i.val();

+

+		key_len = strlen(v.first.text);

+		val_len = strlen(v.second.text);

+		len = key_len + val_len + 2;

+

+		if(size - pos - len > 0)

+			mir_snprintf(buff + pos, size - pos, "\\%s\\%s", v.first.text, v.second.text);

+		pos += len;

+	}

+

+	if(size - pos > 7) {

+		mir_snprintf(buff + pos, size - pos, "\\final\\");

+		pos += 7;

+	}

+	return pos;

+}