diff --git a/.gitignore b/.gitignore index 1b4d938..5ddec7d 100644 --- a/.gitignore +++ b/.gitignore @@ -43,5 +43,6 @@ configure.lineno src/server/prime.lo src/server/verify.lo src/server/httpapi.lo +src/server/stream_parser.lo stabbertest diff --git a/Makefile.am b/Makefile.am index 37d3f88..aeb5230 100644 --- a/Makefile.am +++ b/Makefile.am @@ -4,6 +4,7 @@ sources = \ src/server/server.c src/server/server.h \ src/server/httpapi.c src/server/httpapi.h \ src/server/xmppclient.c src/server/xmppclient.h \ + src/server/stream_parser.c src/server/stream_parser.h \ src/server/parser.c src/server/parser.h \ src/server/stanza.c src/server/stanza.h \ src/server/log.c src/server/log.h \ diff --git a/src/client/stabber.c b/src/client/stabber.c index ba815a6..8f91b35 100644 --- a/src/client/stabber.c +++ b/src/client/stabber.c @@ -58,6 +58,13 @@ stbbr_for_id(char *id, char *stream) return 1; } +int +stbbr_for_query(char *query, char *stream) +{ + prime_for_query(query, stream); + return 1; +} + void stbbr_wait_for(char *id) { diff --git a/src/server/parser.c b/src/server/parser.c index 2811ae4..1b5095a 100644 --- a/src/server/parser.c +++ b/src/server/parser.c @@ -20,36 +20,18 @@ * */ -#include #include #include #include -#include "server/parser.h" #include "server/stanza.h" -#include "server/log.h" static int depth = 0; -static int do_reset = 0; - -static XML_Parser parser; -static XMPPStanza *curr_stanza; -static GString *curr_string = NULL; - -static stream_start_func stream_start_cb = NULL; -static auth_func auth_cb = NULL; -static id_func id_cb = NULL; +static XMPPStanza *curr_stanza = NULL; static void start_element(void *data, const char *element, const char **attributes) { - if (g_strcmp0(element, "stream:stream") == 0) { - log_println("RECV: %s", curr_string->str); - stream_start_cb(); - do_reset = 1; - return; - } - XMPPStanza *stanza = stanza_new(element, attributes); if (depth == 0) { @@ -71,19 +53,6 @@ end_element(void *data, const char *element) if (depth > 0) { stanza_add_child(curr_stanza->parent, curr_stanza); curr_stanza = curr_stanza->parent; - } else { - log_println("RECV: %s", curr_string->str); - stanza_add(curr_stanza); - if (stanza_get_child_by_ns(curr_stanza, "jabber:iq:auth")) { - auth_cb(curr_stanza); - } else { - const char *id = stanza_get_id(curr_stanza); - if (id) { - id_cb(id); - } - } - - do_reset = 1; } } @@ -97,45 +66,19 @@ handle_data(void *data, const char *content, int length) g_string_append_len(curr_stanza->content, content, length); } -void -parser_init(stream_start_func startcb, auth_func authcb, id_func idcb) +XMPPStanza * +parse_stanza(char *stanza_text) { - if (curr_string) { - g_string_free(curr_string, TRUE); + depth = 0; + if (curr_stanza) { + stanza_free(curr_stanza); } - curr_string = g_string_new(""); - - stream_start_cb = startcb; - auth_cb = authcb; - id_cb = idcb; - - parser = XML_ParserCreate(NULL); + XML_Parser parser = XML_ParserCreate(NULL); XML_SetElementHandler(parser, start_element, end_element); XML_SetCharacterDataHandler(parser, handle_data); -} -int -parser_feed(char *chunk, int len) -{ - g_string_append_len(curr_string, chunk, len); - int res = XML_Parse(parser, chunk, len, 0); - parser_reset(); - return res; -} - -void -parser_close(void) -{ + XML_Parse(parser, stanza_text, strlen(stanza_text), 0); XML_ParserFree(parser); - parser = NULL; -} -void -parser_reset(void) -{ - if (do_reset == 1) { - parser_close(); - parser_init(stream_start_cb, auth_cb, id_cb); - do_reset = 0; - } + return curr_stanza; } diff --git a/src/server/parser.h b/src/server/parser.h index b5895da..ae4593b 100644 --- a/src/server/parser.h +++ b/src/server/parser.h @@ -24,16 +24,7 @@ #define __H_PARSER #include "server/stanza.h" -#include "server/xmppclient.h" -typedef void (*stream_start_func)(void); -typedef void (*auth_func)(XMPPStanza *stanza); -typedef void (*id_func)(const char *id); - -void parser_init(stream_start_func startcb, auth_func authcb, id_func); -int parser_feed(char *chunk, int len); -void parser_close(void); -void parser_reset(void); -void parser_show_stanzas(void); +XMPPStanza* parse_stanza(char *stanza_text); #endif diff --git a/src/server/prime.c b/src/server/prime.c index ace1e0a..4d48bad 100644 --- a/src/server/prime.c +++ b/src/server/prime.c @@ -25,14 +25,19 @@ #include +#include "server/stanza.h" +#include "server/parser.h" + static char *required_passwd = NULL; static GHashTable *idstubs = NULL; +static GHashTable *querystubs = NULL; void prime_init(void) { required_passwd = strdup("password"); idstubs = g_hash_table_new_full(g_str_hash, g_str_equal, free, free); + querystubs = g_hash_table_new_full(g_str_hash, g_str_equal, free, (GDestroyNotify)stanza_free); } void @@ -44,7 +49,12 @@ prime_free_all(void) if (idstubs) { g_hash_table_destroy(idstubs); } - idstubs = NULL; + + querystubs = NULL; + if (querystubs) { + g_hash_table_destroy(querystubs); + } + querystubs = NULL; } void @@ -69,7 +79,22 @@ prime_for_id(const char *id, char *stream) } char* -prime_get_for(const char *id) +prime_get_for_id(const char *id) { return g_hash_table_lookup(idstubs, id); } + +void +prime_for_query(const char *query, char *stream) +{ + if (querystubs) { + XMPPStanza *stanza = parse_stanza(stream); + g_hash_table_insert(querystubs, strdup(query), stanza); + } +} + +XMPPStanza * +prime_get_for_query(const char *query) +{ + return g_hash_table_lookup(querystubs, query); +} diff --git a/src/server/prime.h b/src/server/prime.h index 926baaf..5c5c7ac 100644 --- a/src/server/prime.h +++ b/src/server/prime.h @@ -23,6 +23,8 @@ #ifndef __H_PRIME #define __H_PRIME +#include "server/stanza.h" + void prime_init(void); void prime_free_all(void); @@ -30,6 +32,9 @@ void prime_required_passwd(char *password); char* prime_get_passwd(void); int prime_for_id(const char *id, char *stream); -char* prime_get_for(const char *id); +char* prime_get_for_id(const char *id); + +int prime_for_query(const char *query, char *stream); +XMPPStanza* prime_get_for_query(const char *query); #endif diff --git a/src/server/server.c b/src/server/server.c index 87401fd..74cdabe 100644 --- a/src/server/server.c +++ b/src/server/server.c @@ -29,7 +29,7 @@ #include #include "server/xmppclient.h" -#include "server/parser.h" +#include "server/stream_parser.h" #include "server/prime.h" #include "server/stanza.h" #include "server/verify.h" @@ -216,13 +216,28 @@ auth_callback(XMPPStanza *stanza) void id_callback(const char *id) { - char *stream = prime_get_for(id); + char *stream = prime_get_for_id(id); if (stream) { log_println("--> ID callback fired for '%s'", id); write_stream(stream); } } +void +query_callback(const char *query, const char *id) +{ + XMPPStanza *stanza = prime_get_for_query(query); + if (stanza) { + log_println("--> QUERY callback fired for '%s'", query); + stanza_set_id(stanza, id); + char *stream = stanza_to_string(stanza); + write_stream(stream); + + stanza_free(stanza); + free(stream); + } +} + void server_wait_for(char *id) { @@ -288,7 +303,7 @@ _start_server_cb(void* userdata) } client = xmppclient_new(client_addr, client_socket); - parser_init(stream_start_callback, auth_callback, id_callback); + parser_init(stream_start_callback, auth_callback, id_callback, query_callback); read_stream(); diff --git a/src/server/stanza.c b/src/server/stanza.c index 5436d8a..c345016 100644 --- a/src/server/stanza.c +++ b/src/server/stanza.c @@ -96,6 +96,58 @@ stanza_show_all(void) pthread_mutex_unlock(&stanzas_lock); } +char * +stanza_to_string(XMPPStanza *stanza) +{ + GString *stanza_str = g_string_new("<"); + + g_string_append(stanza_str, stanza->name); + + if (stanza->attrs) { + GList *curr = stanza->attrs; + while (curr) { + XMPPAttr *attr = curr->data; + g_string_append(stanza_str, " "); + g_string_append(stanza_str, attr->name); + g_string_append(stanza_str, "=\""); + g_string_append(stanza_str, attr->value); + g_string_append(stanza_str, "\""); + + curr = g_list_next(curr); + } + } + + if (stanza->content) { + g_string_append(stanza_str, ">"); + g_string_append(stanza_str, stanza->content->str); + g_string_append(stanza_str, "name); + g_string_append(stanza_str, ">"); + } else if (stanza->children) { + g_string_append(stanza_str, ">"); + + GList *curr = stanza->children; + while (curr) { + XMPPStanza *child = curr->data; + char *child_str = stanza_to_string(child); + g_string_append(stanza_str, child_str); + free(child_str); + curr = g_list_next(curr); + } + + g_string_append(stanza_str, "name); + g_string_append(stanza_str, ">"); + } else { + g_string_append(stanza_str, "/>"); + } + + char *result = stanza_str->str; + g_string_free(stanza_str, FALSE); + + return result; +} + int stanzas_contains_id(char *id) { @@ -214,6 +266,72 @@ stanza_get_id(XMPPStanza *stanza) return NULL; } +void +stanza_set_id(XMPPStanza *stanza, const char *id) +{ + GList *curr_attr = stanza->attrs; + while (curr_attr) { + XMPPAttr *attr = curr_attr->data; + if (g_strcmp0(attr->name, "id") == 0) { + free(attr->value); + attr->value = strdup(id); + return; + } + + curr_attr = g_list_next(curr_attr); + } + + XMPPAttr *attrnew = malloc(sizeof(XMPPAttr)); + attrnew->name = strdup("id"); + attrnew->value = strdup(id); + stanza->attrs = g_list_append(stanza->attrs, attrnew); +} + +const char * +stanza_get_attr(XMPPStanza *stanza, const char *name) +{ + if (!stanza->attrs) { + return NULL; + } + + GList *curr_attr = stanza->attrs; + while (curr_attr) { + XMPPAttr *attr = curr_attr->data; + if (g_strcmp0(attr->name, name) == 0) { + return attr->value; + } + + curr_attr = g_list_next(curr_attr); + } + + return NULL; +} + +const char * +stanza_get_query_request(XMPPStanza *stanza) +{ + if (g_strcmp0(stanza->name, "iq") != 0) { + return NULL; + } + + const char *type = stanza_get_attr(stanza, "type"); + if (g_strcmp0(type, "get") != 0) { + return NULL; + } + + XMPPStanza *query = stanza_get_child_by_name(stanza, "query"); + if (!query) { + return NULL; + } + + const char *xmlns = stanza_get_attr(query, "xmlns"); + if (!xmlns) { + return NULL; + } + + return xmlns; +} + static int _xmpp_attr_equal(XMPPAttr *attr1, XMPPAttr *attr2) { diff --git a/src/server/stanza.h b/src/server/stanza.h index e9b3908..ea6b320 100644 --- a/src/server/stanza.h +++ b/src/server/stanza.h @@ -44,6 +44,8 @@ void stanza_show_all(void); void stanza_add(XMPPStanza *stanza); void stanza_add_child(XMPPStanza *parent, XMPPStanza *child); const char* stanza_get_id(XMPPStanza *stanza); +void stanza_set_id(XMPPStanza *stanza, const char *id); +const char *stanza_get_query_request(XMPPStanza *stanza); XMPPStanza* stanza_get_child_by_ns(XMPPStanza *stanza, char *ns); XMPPStanza* stanza_get_child_by_name(XMPPStanza *stanza, char *name); @@ -56,4 +58,6 @@ int stanzas_contains_id(char *id); void stanza_free(XMPPStanza *stanza); void stanza_free_all(void); +char* stanza_to_string(XMPPStanza *stanza); + #endif diff --git a/src/server/stream_parser.c b/src/server/stream_parser.c new file mode 100644 index 0000000..2afa2dd --- /dev/null +++ b/src/server/stream_parser.c @@ -0,0 +1,147 @@ +/* + * stream_parser.c + * + * Copyright (C) 2015 James Booth + * + * This file is part of Stabber. + * + * Stabber is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * Stabber is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with Stabber. If not, see . + * + */ + +#include +#include +#include +#include + +#include "server/stream_parser.h" +#include "server/stanza.h" +#include "server/log.h" + +static int depth = 0; +static int do_reset = 0; + +static XML_Parser parser; +static XMPPStanza *curr_stanza; +static GString *curr_string = NULL; + +static stream_start_func stream_start_cb = NULL; +static auth_func auth_cb = NULL; +static id_func id_cb = NULL; +static query_func query_cb = NULL; + +static void +start_element(void *data, const char *element, const char **attributes) +{ + if (g_strcmp0(element, "stream:stream") == 0) { + log_println("RECV: %s", curr_string->str); + stream_start_cb(); + do_reset = 1; + return; + } + + XMPPStanza *stanza = stanza_new(element, attributes); + + if (depth == 0) { + curr_stanza = stanza; + curr_stanza->parent = NULL; + } else { + stanza->parent = curr_stanza; + curr_stanza = stanza; + } + + depth++; +} + +static void +end_element(void *data, const char *element) +{ + depth--; + + if (depth > 0) { + stanza_add_child(curr_stanza->parent, curr_stanza); + curr_stanza = curr_stanza->parent; + } else { + log_println("RECV: %s", curr_string->str); + stanza_add(curr_stanza); + if (stanza_get_child_by_ns(curr_stanza, "jabber:iq:auth")) { + auth_cb(curr_stanza); + } else { + const char *id = stanza_get_id(curr_stanza); + if (id) { + id_cb(id); + } + const char *query = stanza_get_query_request(curr_stanza); + if (query) { + query_cb(query, id); + } + } + + do_reset = 1; + } +} + +static void +handle_data(void *data, const char *content, int length) +{ + if (!curr_stanza->content) { + curr_stanza->content = g_string_new(""); + } + + g_string_append_len(curr_stanza->content, content, length); +} + +void +parser_init(stream_start_func startcb, auth_func authcb, id_func idcb, query_func querycb) +{ + if (curr_string) { + g_string_free(curr_string, TRUE); + } + curr_string = g_string_new(""); + + stream_start_cb = startcb; + auth_cb = authcb; + id_cb = idcb; + query_cb = querycb; + + parser = XML_ParserCreate(NULL); + XML_SetElementHandler(parser, start_element, end_element); + XML_SetCharacterDataHandler(parser, handle_data); +} + +int +parser_feed(char *chunk, int len) +{ + g_string_append_len(curr_string, chunk, len); + int res = XML_Parse(parser, chunk, len, 0); + parser_reset(); + return res; +} + +void +parser_close(void) +{ + XML_ParserFree(parser); + parser = NULL; +} + +void +parser_reset(void) +{ + if (do_reset == 1) { + parser_close(); + parser_init(stream_start_cb, auth_cb, id_cb, query_cb); + do_reset = 0; + } +} diff --git a/src/server/stream_parser.h b/src/server/stream_parser.h new file mode 100644 index 0000000..caf8b1e --- /dev/null +++ b/src/server/stream_parser.h @@ -0,0 +1,40 @@ +/* + * stream_parser.h + * + * Copyright (C) 2015 James Booth + * + * This file is part of Stabber. + * + * Stabber is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * Stabber is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with Stabber. If not, see . + * + */ + +#ifndef __H_STREAM_PARSER +#define __H_STREAM_PARSER + +#include "server/stanza.h" +#include "server/xmppclient.h" + +typedef void (*stream_start_func)(void); +typedef void (*auth_func)(XMPPStanza *stanza); +typedef void (*id_func)(const char *id); +typedef void (*query_func)(const char *query, const char *id); + +void parser_init(stream_start_func startcb, auth_func authcb, id_func idcb, query_func querycb); +int parser_feed(char *chunk, int len); +void parser_close(void); +void parser_reset(void); +void parser_show_stanzas(void); + +#endif diff --git a/stabber.h b/stabber.h index 6c911f8..cb7d7e9 100644 --- a/stabber.h +++ b/stabber.h @@ -30,6 +30,7 @@ void stbbr_set_timeout(int seconds); int stbbr_auth_passwd(char *password); void stbbr_for_id(char *id, char *stream); +int stbbr_for_query(char *query, char *stream); void stbbr_wait_for(char *id);