devs/stabber
2
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Allow stubbing based on query xmlns

Browse Source
This commit is contained in:
James Booth
2015-06-02 00:19:53 +01:00
parent ddb837cdcd
commit fe36df4dbe
13 changed files with 380 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
src/server/parser.c
View File

@@ -20,36 +20,18 @@
*
*/
#include <stdio.h>
#include <string.h>
#include <expat.h>
#include <glib.h>
#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;
}