2
0
mirror of https://github.com/msantos/xmppipe synced 2024-11-16 00:12:59 +00:00
xmppipe/src/xmppipe_send.c
Michael Santos 1bb03b563e xmppipe_send_message: fix memory leak
Fix a memory leak caused by improper usage of
xmpp_stanza_new()/xmpp_stanza_release() by replacing usage with the
simpler xmpp_message_new()/xmpp_message_set_body() API available in
libstrophe 0.9.0, as advised by @pasis.

Fixes https://github.com/msantos/xmppipe/issues/3.
2019-06-17 10:19:36 -04:00

186 lines
4.3 KiB
C

/* Copyright (c) 2019, Michael Santos <michael.santos@gmail.com>
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include "xmppipe.h"
void xmppipe_send_stanza(xmppipe_state_t *state, char *buf, size_t len) {
switch (state->format) {
case XMPPIPE_FMT_TEXT:
xmppipe_send_message(
state, state->out,
(state->opt & XMPPIPE_OPT_GROUPCHAT) ? "groupchat" : "chat", buf, len);
return;
case XMPPIPE_FMT_CSV:
xmppipe_send_stanza_fmt(state, buf, len);
break;
default:
if (state->verbose)
(void)fprintf(stderr, "unsupported format: %d\n", state->format);
return;
}
}
void xmppipe_send_stanza_fmt(xmppipe_state_t *state, char *buf, size_t len) {
char *to = NULL;
char *type = NULL;
char *default_type;
int i;
size_t n;
char *tmp;
char *start;
char *end;
char *body = NULL;
int valid = 0;
default_type = (state->opt & XMPPIPE_OPT_GROUPCHAT) ? "groupchat" : "chat";
tmp = xmppipe_strdup(buf);
start = tmp;
/* trailing newline */
end = strchr(start, '\n');
if (end != NULL)
*end = '\0';
for (i = 0; start != NULL; i++) {
end = strchr(start, ':');
if (end != NULL)
*end++ = '\0';
n = strlen(start);
if (state->verbose)
(void)fprintf(stderr, "message:%d:%s\n", i, n == 0 ? "<empty>" : start);
switch (i) {
case 0:
if (n != 1) {
if (state->verbose)
(void)fprintf(stderr, "stanza required\n");
goto XMPPIPE_DONE;
}
switch (start[0]) {
case 'm':
break;
case 'p':
/* unsupported: fall through */
default:
if (state->verbose)
(void)fprintf(stderr, "unsupported stanza: %c\n", start[0]);
goto XMPPIPE_DONE;
}
break;
case 1:
type = xmppipe_fmt_decode((n == 0) ? default_type : start);
if (type == NULL)
goto XMPPIPE_DONE;
break;
case 2:
to = xmppipe_fmt_decode((n == 0) ? state->out : start);
if (to == NULL)
goto XMPPIPE_DONE;
break;
case 3:
break;
case 4:
body = xmppipe_fmt_decode(start);
if (body == NULL)
goto XMPPIPE_DONE;
valid = 1;
break;
default:
goto XMPPIPE_DONE;
}
start = end;
}
XMPPIPE_DONE:
if (valid == 1)
xmppipe_send_message(state, to, type, body, strlen(body));
else if (state->verbose)
(void)fprintf(stderr, "invalid input\n");
free(tmp);
free(type);
free(to);
free(body);
}
void xmppipe_send_message(xmppipe_state_t *state, char *to, char *type,
char *buf, size_t len) {
xmpp_stanza_t *message = NULL;
char *id = NULL;
id = xmpp_uuid_gen(state->ctx);
if (id == NULL) {
errx(EXIT_FAILURE, "unable to allocate message id");
}
message = xmppipe_message_new(state->ctx, type, to, id);
if (state->encode) {
size_t len = strlen(buf);
char *b64 = xmpp_base64_encode(state->ctx, (unsigned char *)buf, len);
if (b64 == NULL)
errx(EXIT_FAILURE, "encode: invalid input: %zu", len);
xmppipe_message_set_body(message, b64);
xmpp_free(state->ctx, b64);
} else {
xmppipe_message_set_body(message, buf);
}
xmppipe_send(state, message);
xmpp_free(state->ctx, id);
}
void xmppipe_send(xmppipe_state_t *state, xmpp_stanza_t *const stanza) {
xmpp_stanza_t *r = NULL;
state->sm_request++;
xmpp_send(state->conn, stanza);
if (state->sm_enabled == 0)
return;
if (state->sm_request % state->sm_request_interval != 0)
return;
r = xmppipe_stanza_new(state->ctx);
xmppipe_stanza_set_name(r, "r");
xmppipe_stanza_set_ns(r, "urn:xmpp:sm:3");
xmpp_send(state->conn, r);
state->sm_request_unack++;
(void)xmpp_stanza_release(r);
}