X-Git-Url: https://code.citadel.org/?a=blobdiff_plain;f=citadel%2Fmodules%2Fxmpp%2Fserv_xmpp.c;h=b06d11aef107c4017d3849be24ecf9b932ccd44a;hb=85a4d8d0b5af0edda2a0fa27c66457b775eb1209;hp=cfad3489f7f0ec024834a97b54aea5b9c63a798c;hpb=62e5cd8b5bcf983f843a7e27d0ecddc4b11b13c3;p=citadel.git

diff --git a/citadel/modules/xmpp/serv_xmpp.c b/citadel/modules/xmpp/serv_xmpp.c
index cfad3489f..b06d11aef 100644
--- a/citadel/modules/xmpp/serv_xmpp.c
+++ b/citadel/modules/xmpp/serv_xmpp.c
@@ -67,6 +67,9 @@
 struct xmpp_event *xmpp_queue = NULL;
 HashList *XMPP_StartHandlers = NULL;
 HashList *XMPP_EndHandlers = NULL;
+HashList *XMPP_SupportedNamespaces = NULL;
+HashList *XMPP_NameSpaces = 0;
+HashList *FlatToken = NULL;
 
 int XMPPSrvDebugEnable = 0;
 
@@ -92,7 +95,6 @@ void XPut(const char *Str, long Len)
 {
 	StrBufAppendBufPlain(XMPP->OutBuf, Str, Len, 0);
 }
-#define XPUT(CONSTSTR) XPut(CONSTSTR, sizeof(CONSTSTR) -1)
 
 void XPrintf(const char *Format, ...)
 {
@@ -103,6 +105,96 @@ void XPrintf(const char *Format, ...)
 }
 
 
+void XPrint(const char *Token, long tlen,
+	    int Flags,
+	    ...)
+
+{
+	int BodySeen = 0;
+	int ArgType;
+	int Finished = 0;
+	char *PName;
+	long PLen;
+	char *Val;
+	long VLen;
+        va_list arg_ptr;
+
+	XPUT("<");
+	XPut(Token, tlen);
+
+        va_start(arg_ptr, Flags);
+	while (!Finished)
+	{
+		ArgType = va_arg(arg_ptr, int);
+		switch (ArgType)
+		{
+		case TYPE_STR:
+			PName = va_arg(arg_ptr, char*);
+			PLen  = va_arg(arg_ptr, long);
+			Val   = va_arg(arg_ptr, char*);
+			VLen  = va_arg(arg_ptr, long);
+			XPUT(" ");
+			XPut(PName, PLen);
+			XPUT("=\"");
+			XPutProp(Val, VLen);
+			XPUT("\"");
+			break;
+		case TYPE_OPTSTR:
+			PName = va_arg(arg_ptr, char*);
+			PLen  = va_arg(arg_ptr, long);
+			Val   = va_arg(arg_ptr, char*);
+			VLen  = va_arg(arg_ptr, long);
+			if (VLen > 0)
+			{
+				XPUT(" ");
+				XPut(PName, PLen);
+				XPUT("=\"");
+				XPutProp(Val, VLen);
+				XPUT("\"");
+			}
+			break;
+		case TYPE_INT:
+			PName = va_arg(arg_ptr, char*);
+			PLen  = va_arg(arg_ptr, long);
+			VLen  = va_arg(arg_ptr, long);
+			XPUT(" ");
+			XPut(PName, PLen);
+			XPUT("=\"");
+			XPrintf("%ld", VLen);
+			XPUT("\"");
+			break;
+		case TYPE_BODYSTR:
+			BodySeen = 1;
+			XPUT(">");
+			Val   = va_arg(arg_ptr, char*);
+			VLen  = va_arg(arg_ptr, long);
+			XPutBody(Val, VLen);
+			break;
+		case TYPE_ARGEND:
+			Finished = 1;
+			break;
+		}
+	}
+	if (Flags == XCLOSED)
+	{
+		if (BodySeen)
+		{
+			XPUT("</");
+			XPut(Token, tlen);
+			XPUT(">");
+		}
+		else
+		{
+			XPUT("></");
+			XPut(Token, tlen);
+			XPUT(">");
+		}
+	}
+	else
+		XPUT(">");
+	va_end(arg_ptr);
+}
+
 #ifdef HAVE_XML_STOPPARSER
 /* Stop the parser if an entity declaration is hit. */
 static void xmpp_entity_declaration(void *userData, const XML_Char *entityName,
@@ -139,20 +231,20 @@ void xmpp_stream_start(void *data, const char *supplied_el, const char **attr)
 	XPrintf("%08x\" ", CC->cs_pid);
 	XPUT("version=\"1.0\" "
 		  "xmlns:stream=\"http://etherx.jabber.org/streams\" "
-		  "xmlns=\"jabber:client\">");
-
+		  "xmlns=\"jabber:client\">"
 	/* The features of this stream are... */
-	XPUT("<stream:features>");
+	     "<stream:features>");
 
 	/*
 	 * TLS encryption (but only if it isn't already active)
 	 */ 
+/*
 #ifdef HAVE_OPENSSL
 	if (!CC->redirect_ssl) {
 		XPUT("<starttls xmlns='urn:ietf:params:xml:ns:xmpp-tls'></starttls>");
 	}
 #endif
-
+*/
 	if (!CC->logged_in) {
 		/* If we're not logged in yet, offer SASL as our feature set */
 		xmpp_output_auth_mechs();
@@ -182,6 +274,7 @@ void xmpp_start_bind(void *data, const char *supplied_el, const char **attr)
 
 void xmpp_start_iq(void *data, const char *supplied_el, const char **attr)
 {
+/*
 	int i;
 	for (i=0; attr[i] != NULL; i+=2) {
 		if (!strcasecmp(attr[i], "type")) {
@@ -197,6 +290,7 @@ void xmpp_start_iq(void *data, const char *supplied_el, const char **attr)
 			safestrncpy(XMPP->iq_to, attr[i+1], sizeof XMPP->iq_to);
 		}
 	}
+*/
 }
 
 void xmpp_start_auth(void *data, const char *supplied_el, const char **attr)
@@ -229,21 +323,75 @@ void xmpp_start_html(void *data, const char *supplied_el, const char **attr)
 
 void xmpp_xml_start(void *data, const char *supplied_el, const char **attr)
 {
-	char el[256];
-	long newlen;
+	HashList *ThisNamespace = NULL;
+	const char *pToken;
+	const char *pch;
+	const char *NS = NULL;
+	long NSLen;
 	long len;
-	char *sep = NULL;
 	void *pv;
 	
 	/* Axe the namespace, we don't care about it */
-	newlen = len = safestrncpy(el, supplied_el, sizeof el);
-	while (sep = strchr(el, ':'), sep)
+	pToken = supplied_el;
+	pch = strchr(pToken, ':');
+	while (pch != NULL)
 	{
-		newlen -= ++sep - el;
-		memmove(el, sep, newlen + 1);
-		len = newlen;
+		pToken = pch;
+		pch = strchr(pToken  + 1, ':');
+	}
+
+	if (*pToken == ':')
+	{
+		NS = supplied_el;
+		NSLen = pToken - supplied_el;
+		if (GetHash(XMPP_NameSpaces, NS, NSLen, &pv))
+		{
+			ThisNamespace = pv;
+
+		}
+		
+		pToken ++;
 	}
 
+	len = strlen(pToken);
+
+
+	if (ThisNamespace != NULL)
+	{
+		if (GetHash(ThisNamespace, pToken, len, &pv))
+		{
+			TokenHandler *th;
+			void *value;
+			long i = 0;
+
+			th = (TokenHandler*) pv;
+			value = th->GetToken();
+
+			while (attr[i] != NULL)
+			{
+
+				if (GetHash(th->Properties, attr[i], strlen(attr[i]), &pv))
+				{
+					PropertyHandler* ph = pv;
+					char *val;
+					StrBuf **pVal;
+					long len;
+
+					len = strlen(attr[i+1]);
+					val = value;
+					val += ph->offset;
+					pVal = (StrBuf**) val;
+					if (*pVal != NULL)
+						StrBufPlain(*pVal, attr[i+1], len);
+					else
+						*pVal = NewStrBufPlain(attr[i+1], len);
+				}
+				i+=2;
+			}
+			return;
+		}
+
+	}
 	/*
 	XMPP_syslog(LOG_DEBUG, "XMPP ELEMENT START: <%s>\n", el);
 	for (i=0; attr[i] != NULL; i+=2) {
@@ -251,13 +399,12 @@ void xmpp_xml_start(void *data, const char *supplied_el, const char **attr)
 	}
 	uncomment for more verbosity */
 
-	if (GetHash(XMPP_StartHandlers, el, len, &pv))
+	if (GetHash(XMPP_StartHandlers, pToken, len, &pv))
 	{
 		xmpp_handler *h;
 		h = (xmpp_handler*) pv;
 		h->Handler(data, supplied_el, attr);
 	}
-	XUnbuffer();
 }
 
 void xmpp_end_resource(void *data, const char *supplied_el, const char **attr)
@@ -295,14 +442,13 @@ void xmpp_end_iq(void *data, const char *supplied_el, const char **attr)
 	/*
 	 * iq type="get" (handle queries)
 	 */
-	if (!strcasecmp(Xmpp->iq_type, "get"))
+	if (!strcasecmp(ChrPtr(Xmpp->IQ.type), "get"))
 	{
 		/*
 		 * Query on a namespace
 		 */
 		if (!IsEmptyStr(Xmpp->iq_query_xmlns)) {
-			xmpp_query_namespace(Xmpp->iq_id, Xmpp->iq_from,
-					     Xmpp->iq_to, Xmpp->iq_query_xmlns);
+			xmpp_query_namespace(&Xmpp->IQ, Xmpp->iq_query_xmlns);
 		}
 		
 		/*
@@ -310,18 +456,18 @@ void xmpp_end_iq(void *data, const char *supplied_el, const char **attr)
 		 */
 		else if (Xmpp->ping_requested) {
 			XPUT("<iq type=\"result\" ");
-			if (!IsEmptyStr(Xmpp->iq_from)) {
+			if (StrLength(Xmpp->IQ.from) > 0) {
 				XPUT("to=\"");
-				XPutProp(Xmpp->iq_from, strlen(Xmpp->iq_from));
+				XPutSProp(Xmpp->IQ.from);
 				XPUT("\" ");
 			}
-			if (!IsEmptyStr(Xmpp->iq_to)) {
+			if (StrLength(Xmpp->IQ.to)>0) {
 				XPUT("from=\"");
-				XPutProp(Xmpp->iq_to, strlen(Xmpp->iq_to));
+				XPutSProp(Xmpp->IQ.to);
 				XPUT("\" ");
 			}
 			XPUT("id=\"");
-			XPutProp(Xmpp->iq_id, strlen(Xmpp->iq_id));
+			XPutSProp(Xmpp->IQ.id);
 			XPUT("\"/>");
 		}
 
@@ -336,7 +482,7 @@ void xmpp_end_iq(void *data, const char *supplied_el, const char **attr)
 				);
 */
 			XPUT("<iq type=\"error\" id=\"");
-			XPutProp(Xmpp->iq_id, strlen(Xmpp->iq_id));
+			XPutSProp(Xmpp->IQ.id);
 			XPUT("\">"
 			     "<error code=\"503\" type=\"cancel\">"
 			     "<service-unavailable xmlns=\"urn:ietf:params:xml:ns:xmpp-stanzas\"/>"
@@ -349,12 +495,12 @@ void xmpp_end_iq(void *data, const char *supplied_el, const char **attr)
 	 * Non SASL authentication
 	 */
 	else if (
-		(!strcasecmp(Xmpp->iq_type, "set"))
+		(!strcasecmp(ChrPtr(Xmpp->IQ.type), "set"))
 		&& (!strcasecmp(Xmpp->iq_query_xmlns, "jabber:iq:auth:query"))
 		) {
 		
 		xmpp_non_sasl_authenticate(
-			Xmpp->iq_id,
+			Xmpp->IQ.id,
 			Xmpp->iq_client_username,
 			Xmpp->iq_client_password,
 			Xmpp->iq_client_resource
@@ -366,7 +512,7 @@ void xmpp_end_iq(void *data, const char *supplied_el, const char **attr)
 	 */
 	else if (
 		(Xmpp->bind_requested)
-		&& (!IsEmptyStr(Xmpp->iq_id))
+		&& (StrLength(Xmpp->IQ.id)>0)
 		&& (!IsEmptyStr(Xmpp->iq_client_resource))
 		&& (CC->logged_in)
 		) {
@@ -382,7 +528,7 @@ void xmpp_end_iq(void *data, const char *supplied_el, const char **attr)
 		/* Tell the client what its JID is */
 
 		XPUT("<iq type=\"result\" id=\"");
-		XPutProp(Xmpp->iq_id, strlen(Xmpp->iq_id));
+		XPutSProp(Xmpp->IQ.id);
 		XPUT("\">"
 		     "<bind xmlns=\"urn:ietf:params:xml:ns:xmpp-bind\">");
 		XPUT("<jid>");
@@ -394,26 +540,26 @@ void xmpp_end_iq(void *data, const char *supplied_el, const char **attr)
 
 	else if (Xmpp->iq_session) {
 		XPUT("<iq type=\"result\" id=\"");
-		XPutProp(Xmpp->iq_id, strlen(Xmpp->iq_id));
+		XPutSProp(Xmpp->IQ.id);
 		XPUT("\">"
 		     "</iq>");
 	}
 
 	else {
 		XPUT("<iq type=\"error\" id=\"");
-		XPutProp(Xmpp->iq_id, strlen(Xmpp->iq_id));
+		XPutSProp(Xmpp->IQ.id);
 		XPUT("\">");
 		XPUT("<error>Don't know howto do '");
-		XPutBody(Xmpp->iq_type, strlen(Xmpp->iq_type));
+		XPutBody(SKEY(Xmpp->IQ.type));
 		XPUT("'!</error>"
 		     "</iq>");
 	}
 
 	/* Now clear these fields out so they don't get used by a future stanza */
-	Xmpp->iq_id[0] = 0;
-	Xmpp->iq_from[0] = 0;
-	Xmpp->iq_to[0] = 0;
-	Xmpp->iq_type[0] = 0;
+	FlushStrBuf(Xmpp->IQ.id);
+	FlushStrBuf(Xmpp->IQ.from);
+	FlushStrBuf(Xmpp->IQ.to);
+	FlushStrBuf(Xmpp->IQ.type);
 	Xmpp->iq_client_resource[0] = 0;
 	Xmpp->iq_session = 0;
 	Xmpp->iq_query_xmlns[0] = 0;
@@ -460,6 +606,7 @@ void xmpp_end_starttls(void *data, const char *supplied_el, const char **attr)
 {
 #ifdef HAVE_OPENSSL
 	XPUT("<proceed xmlns='urn:ietf:params:xml:ns:xmpp-tls'/>");
+	XUnbuffer();
 	CtdlModuleStartCryptoMsgs(NULL, NULL, NULL);
 	if (!CC->redirect_ssl) CC->kill_me = KILLME_NO_CRYPTO;
 #else
@@ -483,21 +630,25 @@ void xmpp_end_stream(void *data, const char *supplied_el, const char **attr)
 
 void xmpp_xml_end(void *data, const char *supplied_el)
 {
-	char el[256];
+	const char *pToken;
+	const char *pch;
 	long len;
-	long newlen;
-	char *sep = NULL;
 	void *pv;
 
 	/* Axe the namespace, we don't care about it */
-	newlen = len = safestrncpy(el, supplied_el, sizeof el);
-	while (sep = strchr(el, ':'), sep) {
-
-		newlen -= ++sep - el;
-		memmove(el, sep, newlen + 1);
-		len = newlen;
+	pToken = supplied_el;
+	pch = strchr(pToken, ':');
+	while (pch != NULL)
+	{
+		pToken = pch;
+		pch = strchr(pToken  + 1, ':');
 	}
 
+	if (*pToken == ':')
+		pToken ++;
+
+	len = strlen(pToken);
+
 	/*
 	XMPP_syslog(LOG_DEBUG, "XMPP ELEMENT END  : <%s>\n", el);
 	if (XMPP->chardata_len > 0) {
@@ -505,7 +656,7 @@ void xmpp_xml_end(void *data, const char *supplied_el)
 	}
 	uncomment for more verbosity */
 
-	if (GetHash(XMPP_EndHandlers, el, len, &pv))
+	if (GetHash(XMPP_EndHandlers, pToken, len, &pv))
 	{
 		xmpp_handler *h;
 		h = (xmpp_handler*) pv;
@@ -513,14 +664,13 @@ void xmpp_xml_end(void *data, const char *supplied_el)
 	}
 	else
 	{
-		XMPP_syslog(LOG_DEBUG, "Ignoring unknown tag <%s>\n", el);
+		XMPP_syslog(LOG_DEBUG, "Ignoring unknown tag <%s>\n", pToken);
 	}
 
 	XMPP->chardata_len = 0;
 	if (XMPP->chardata_alloc > 0) {
 		XMPP->chardata[0] = 0;
 	}
-	XUnbuffer();
 }
 
 
@@ -559,7 +709,10 @@ void xmpp_cleanup_function(void) {
 			free(XMPP->message_body);
 		}
 	}
+	free_buf_iq(&XMPP->IQ);
+
 	XML_ParserFree(XMPP->xp);
+	FreeStrBuf(&XMPP->OutBuf);
 	free(XMPP);
 }
 
@@ -574,7 +727,7 @@ void xmpp_greeting(void) {
 	CC->session_specific_data = malloc(sizeof(citxmpp));
 	memset(XMPP, 0, sizeof(citxmpp));
 	XMPP->last_event_processed = queue_event_seq;
-
+	XMPP->OutBuf = NewStrBufPlain(NULL, SIZ);
 	/* XMPP does not use a greeting, but we still have to initialize some things. */
 
 	XMPP->xp = XML_ParserCreateNS("UTF-8", ':');
@@ -600,6 +753,7 @@ void xmpp_greeting(void) {
 #endif
 
 	CC->can_receive_im = 1;		/* This protocol is capable of receiving instant messages */
+	XUnbuffer();
 }
 
 
@@ -620,6 +774,7 @@ void xmpp_command_loop(void) {
 		CC->kill_me = KILLME_CLIENT_DISCONNECTED;
 	}
 	FreeStrBuf(&stream_input);
+	XUnbuffer();
 }
 
 
@@ -685,6 +840,78 @@ void AddXMPPEndHandler(const char *key,
 	Put(XMPP_EndHandlers, key, len, h, NULL);
 }
 
+void HFreePropertyHandler(void *FreeMe)
+{
+	free(FreeMe);
+}
+
+void HDeleteTokenHandler(void *FreeMe)
+{
+	TokenHandler *th = (TokenHandler *) FreeMe;
+	DeleteHash(&th->Properties);
+	free(th);
+}
+
+void XMPP_RegisterTokenProperty(const char *NS, long NSLen,
+				const char *Token, long TLen,
+				const char *Property, long PLen,
+				GetTokenDataFunc GetToken,
+				long offset)
+{
+	void *pv;
+	HashList *ThisNamespace = NULL;
+	PropertyHandler *h;
+	TokenHandler *th;
+
+	h = (PropertyHandler*) malloc(sizeof(PropertyHandler));
+	h->NameSpace = NS;
+	h->NameSpaceLen = NSLen;
+	h->Token = Token;
+	h->TokenLen = TLen;
+	h->Property = Property;
+	h->PropertyLen = PLen;
+	h->offset = offset;
+
+	if (!GetHash(XMPP_SupportedNamespaces, NS, NSLen, &pv))
+	{
+		Put(XMPP_SupportedNamespaces, NS, NSLen, NewStrBufPlain(NS, NSLen), HFreeStrBuf);
+	}
+		
+	
+	if (GetHash(XMPP_NameSpaces, NS, NSLen, &pv))
+	{
+		ThisNamespace = pv;
+	}
+	else
+	{
+		ThisNamespace = NewHash(1, NULL);
+		Put(XMPP_NameSpaces, NS, NSLen, ThisNamespace, HDeleteHash);
+	}
+
+	if (GetHash(ThisNamespace, Token, TLen, &pv))
+	{
+		th = pv;
+	}
+	else
+	{
+		th = (TokenHandler*) malloc (sizeof(TokenHandler));
+		th->GetToken = GetToken;
+		th->Properties = NewHash(1, NULL);
+		Put(ThisNamespace, Token, TLen, th, HDeleteTokenHandler);
+	}
+
+	Put(th->Properties, Property, PLen, h, HFreePropertyHandler);
+	/*
+	if (!GetHash(FlatToken, Token, TLen, &pv))
+	{
+		// todo mark pv as non uniq
+		Put(FlatToken, Token, TLen, ThisToken, reference_free_handler);
+	}	
+	*/
+}
+
+
+
 CTDL_MODULE_INIT(xmpp)
 {
 	if (!threading) {
@@ -698,6 +925,9 @@ CTDL_MODULE_INIT(xmpp)
 
 		XMPP_StartHandlers = NewHash(1, NULL);
 		XMPP_EndHandlers = NewHash(1, NULL);
+		XMPP_NameSpaces = NewHash(1, NULL);
+		XMPP_SupportedNamespaces = NewHash(1, NULL);
+		FlatToken = NewHash(1, NULL);
 
 		AddXMPPEndHandler(HKEY("resource"),	 xmpp_end_resource, 0);
 		AddXMPPEndHandler(HKEY("username"),	 xmpp_end_username, 0);
@@ -727,9 +957,10 @@ CTDL_MODULE_INIT(xmpp)
                 CtdlRegisterSessionHook(xmpp_login_hook, EVT_UNSTEALTH, PRIO_UNSTEALTH + 1);
                 CtdlRegisterSessionHook(xmpp_logout_hook, EVT_STEALTH, PRIO_STEALTH + 1);
 		CtdlRegisterCleanupHook(xmpp_cleanup_events);
-
 	}
 
 	/* return our module name for the log */
 	return "xmpp";
 }
+
+