|
Packit |
16808d |
|
|
Packit |
16808d |
/*
|
|
Packit |
16808d |
Redirecting Sametime Faux-Server
|
|
Packit |
16808d |
The Meanwhile Project
|
|
Packit |
16808d |
|
|
Packit |
16808d |
This is a tool which helps to test handling of login-redirects when
|
|
Packit |
16808d |
you don't have a server which will issue them. It will listen for a
|
|
Packit |
16808d |
single incoming connection, will accept a handshake message and send
|
|
Packit |
16808d |
a handshake ack, and will respond to a login message with a redirect
|
|
Packit |
16808d |
message to a server specified on the command line.
|
|
Packit |
16808d |
|
|
Packit |
16808d |
Christopher O'Brien <siege@preoccupied.net>
|
|
Packit |
16808d |
*/
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
#include <stdio.h>
|
|
Packit |
16808d |
#include <stdlib.h>
|
|
Packit |
16808d |
#include <string.h>
|
|
Packit |
16808d |
#include <sys/socket.h>
|
|
Packit |
16808d |
#include <netdb.h>
|
|
Packit |
16808d |
#include <netinet/in.h>
|
|
Packit |
16808d |
#include <unistd.h>
|
|
Packit |
16808d |
|
|
Packit |
16808d |
#include <glib.h>
|
|
Packit |
16808d |
|
|
Packit |
16808d |
#include <mw_common.h>
|
|
Packit |
16808d |
#include <mw_message.h>
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
/** the server socket or the connected socket */
|
|
Packit |
16808d |
static int sock;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
/** the io channel */
|
|
Packit |
16808d |
static GIOChannel *chan;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
/** the listening event on the io channel */
|
|
Packit |
16808d |
static int chan_io;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static char *host;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static guchar *sbuf;
|
|
Packit |
16808d |
static gsize sbuf_size;
|
|
Packit |
16808d |
static gsize sbuf_recv;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static void send_msg(struct mwMessage *msg) {
|
|
Packit |
16808d |
struct mwPutBuffer *b;
|
|
Packit |
16808d |
struct mwOpaque o = { 0, 0 };
|
|
Packit |
16808d |
|
|
Packit |
16808d |
b = mwPutBuffer_new();
|
|
Packit |
16808d |
mwMessage_put(b, msg);
|
|
Packit |
16808d |
mwPutBuffer_finalize(&o, b);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
b = mwPutBuffer_new();
|
|
Packit |
16808d |
mwOpaque_put(b, &o);
|
|
Packit |
16808d |
mwOpaque_clear(&o);
|
|
Packit |
16808d |
mwPutBuffer_finalize(&o, b);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(sock) write(sock, o.data, o.len);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
mwOpaque_clear(&o);
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static void handshake_ack() {
|
|
Packit |
16808d |
struct mwMsgHandshakeAck *msg;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
msg = (struct mwMsgHandshakeAck *)
|
|
Packit |
16808d |
mwMessage_new(mwMessage_HANDSHAKE_ACK);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
send_msg(MW_MESSAGE(msg));
|
|
Packit |
16808d |
mwMessage_free(MW_MESSAGE(msg));
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static void login_redir() {
|
|
Packit |
16808d |
struct mwMsgLoginRedirect *msg;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
msg = (struct mwMsgLoginRedirect *)
|
|
Packit |
16808d |
mwMessage_new(mwMessage_LOGIN_REDIRECT);
|
|
Packit |
16808d |
msg->host = g_strdup(host);
|
|
Packit |
16808d |
msg->server_id = NULL;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
send_msg(MW_MESSAGE(msg));
|
|
Packit |
16808d |
mwMessage_free(MW_MESSAGE(msg));
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static void done() {
|
|
Packit |
16808d |
close(sock);
|
|
Packit |
16808d |
exit(0);
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static void side_process(const guchar *buf, gsize len) {
|
|
Packit |
16808d |
struct mwOpaque o = { .len = len, .data = (guchar *) buf };
|
|
Packit |
16808d |
struct mwGetBuffer *b;
|
|
Packit |
16808d |
guint16 type;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(! len) return;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
b = mwGetBuffer_wrap(&o);
|
|
Packit |
16808d |
type = guint16_peek(b);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
switch(type) {
|
|
Packit |
16808d |
case mwMessage_HANDSHAKE:
|
|
Packit |
16808d |
handshake_ack();
|
|
Packit |
16808d |
break;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
case mwMessage_LOGIN:
|
|
Packit |
16808d |
login_redir();
|
|
Packit |
16808d |
break;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
case mwMessage_CHANNEL_DESTROY:
|
|
Packit |
16808d |
done();
|
|
Packit |
16808d |
break;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
default:
|
|
Packit |
16808d |
;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
mwGetBuffer_free(b);
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static void sbuf_free() {
|
|
Packit |
16808d |
g_free(sbuf);
|
|
Packit |
16808d |
sbuf = NULL;
|
|
Packit |
16808d |
sbuf_size = 0;
|
|
Packit |
16808d |
sbuf_recv = 0;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
#define ADVANCE(b, n, count) { b += count; n -= count; }
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
/* handle input to complete an existing buffer */
|
|
Packit |
16808d |
static gsize side_recv_cont(const guchar *b, gsize n) {
|
|
Packit |
16808d |
|
|
Packit |
16808d |
gsize x = sbuf_size - sbuf_recv;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(n < x) {
|
|
Packit |
16808d |
memcpy(sbuf + sbuf_recv, b, n);
|
|
Packit |
16808d |
sbuf_recv += n;
|
|
Packit |
16808d |
return 0;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
} else {
|
|
Packit |
16808d |
memcpy(sbuf + sbuf_recv, b, x);
|
|
Packit |
16808d |
ADVANCE(b, n, x);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(sbuf_size == 4) {
|
|
Packit |
16808d |
struct mwOpaque o = { .len = 4, .data = sbuf };
|
|
Packit |
16808d |
struct mwGetBuffer *gb = mwGetBuffer_wrap(&o);
|
|
Packit |
16808d |
x = guint32_peek(gb);
|
|
Packit |
16808d |
mwGetBuffer_free(gb);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(n < x) {
|
|
Packit |
16808d |
guchar *t;
|
|
Packit |
16808d |
x += 4;
|
|
Packit |
16808d |
t = (guchar *) g_malloc(x);
|
|
Packit |
16808d |
memcpy(t, sbuf, 4);
|
|
Packit |
16808d |
memcpy(t+4, b, n);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
sbuf_free();
|
|
Packit |
16808d |
|
|
Packit |
16808d |
sbuf = t;
|
|
Packit |
16808d |
sbuf_size = x;
|
|
Packit |
16808d |
sbuf_recv = n + 4;
|
|
Packit |
16808d |
return 0;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
} else {
|
|
Packit |
16808d |
sbuf_free();
|
|
Packit |
16808d |
side_process(b, x);
|
|
Packit |
16808d |
ADVANCE(b, n, x);
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
} else {
|
|
Packit |
16808d |
side_process(sbuf+4, sbuf_size-4);
|
|
Packit |
16808d |
sbuf_free();
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
return n;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
/* handle input when there's nothing previously buffered */
|
|
Packit |
16808d |
static gsize side_recv_empty(const guchar *b, gsize n) {
|
|
Packit |
16808d |
struct mwOpaque o = { .len = n, .data = (guchar *) b };
|
|
Packit |
16808d |
struct mwGetBuffer *gb;
|
|
Packit |
16808d |
gsize x;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(n < 4) {
|
|
Packit |
16808d |
sbuf = (guchar *) g_malloc0(4);
|
|
Packit |
16808d |
memcpy(sbuf, b, n);
|
|
Packit |
16808d |
sbuf_size = 4;
|
|
Packit |
16808d |
sbuf_recv = n;
|
|
Packit |
16808d |
return 0;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
gb = mwGetBuffer_wrap(&o);
|
|
Packit |
16808d |
x = guint32_peek(gb);
|
|
Packit |
16808d |
mwGetBuffer_free(gb);
|
|
Packit |
16808d |
if(! x) return n - 4;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(n < (x + 4)) {
|
|
Packit |
16808d |
|
|
Packit |
16808d |
x += 4;
|
|
Packit |
16808d |
sbuf = (guchar *) g_malloc(x);
|
|
Packit |
16808d |
memcpy(sbuf, b, n);
|
|
Packit |
16808d |
sbuf_size = x;
|
|
Packit |
16808d |
sbuf_recv = n;
|
|
Packit |
16808d |
return 0;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
} else {
|
|
Packit |
16808d |
ADVANCE(b, n, 4);
|
|
Packit |
16808d |
side_process(b, x);
|
|
Packit |
16808d |
ADVANCE(b, n, x);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
return n;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static gsize side_recv(const guchar *b, gsize n) {
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(n && (sbuf_size == 0) && (*b & 0x80)) {
|
|
Packit |
16808d |
ADVANCE(b, n, 1);
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(n == 0) {
|
|
Packit |
16808d |
return 0;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
} else if(sbuf_size > 0) {
|
|
Packit |
16808d |
return side_recv_cont(b, n);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
} else {
|
|
Packit |
16808d |
return side_recv_empty(b, n);
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static void feed_buf(const guchar *buf, gsize n) {
|
|
Packit |
16808d |
guchar *b = (guchar *) buf;
|
|
Packit |
16808d |
gsize remain = 0;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
while(n > 0) {
|
|
Packit |
16808d |
remain = side_recv(b, n);
|
|
Packit |
16808d |
b += (n - remain);
|
|
Packit |
16808d |
n = remain;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static int read_recv() {
|
|
Packit |
16808d |
guchar buf[2048];
|
|
Packit |
16808d |
int len;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
len = read(sock, buf, 2048);
|
|
Packit |
16808d |
if(len > 0) feed_buf(buf, (gsize) len);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
return len;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static gboolean read_cb(GIOChannel *chan,
|
|
Packit |
16808d |
GIOCondition cond,
|
|
Packit |
16808d |
gpointer data) {
|
|
Packit |
16808d |
int ret = 0;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(cond & G_IO_IN) {
|
|
Packit |
16808d |
ret = read_recv();
|
|
Packit |
16808d |
if(ret > 0) return TRUE;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(sock) {
|
|
Packit |
16808d |
g_source_remove(chan_io);
|
|
Packit |
16808d |
close(sock);
|
|
Packit |
16808d |
sock = 0;
|
|
Packit |
16808d |
chan = NULL;
|
|
Packit |
16808d |
chan_io = 0;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
done();
|
|
Packit |
16808d |
|
|
Packit |
16808d |
return FALSE;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static gboolean listen_cb(GIOChannel *chan,
|
|
Packit |
16808d |
GIOCondition cond,
|
|
Packit |
16808d |
gpointer data) {
|
|
Packit |
16808d |
|
|
Packit |
16808d |
struct sockaddr_in rem;
|
|
Packit |
16808d |
guint len = sizeof(rem);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
sock = accept(sock, (struct sockaddr *) &rem, &len;;
|
|
Packit |
16808d |
g_assert(sock > 0);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
g_source_remove(chan_io);
|
|
Packit |
16808d |
chan = g_io_channel_unix_new(sock);
|
|
Packit |
16808d |
chan_io = g_io_add_watch(chan, G_IO_IN | G_IO_ERR | G_IO_HUP,
|
|
Packit |
16808d |
read_cb, NULL);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
return FALSE;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
static void init_socket(int port) {
|
|
Packit |
16808d |
/* start listening on the local port specifier */
|
|
Packit |
16808d |
|
|
Packit |
16808d |
struct sockaddr_in sin;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
sock = socket(PF_INET, SOCK_STREAM, 0);
|
|
Packit |
16808d |
g_assert(sock >= 0);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
memset(&sin, 0, sizeof(struct sockaddr_in));
|
|
Packit |
16808d |
sin.sin_family = PF_INET;
|
|
Packit |
16808d |
sin.sin_port = htons(port);
|
|
Packit |
16808d |
sin.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(bind(sock, (struct sockaddr *)&sin, sizeof(sin)) < 0)
|
|
Packit |
16808d |
g_assert_not_reached();
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(listen(sock, 1) < 0)
|
|
Packit |
16808d |
g_assert_not_reached();
|
|
Packit |
16808d |
|
|
Packit |
16808d |
chan = g_io_channel_unix_new(sock);
|
|
Packit |
16808d |
chan_io = g_io_add_watch(chan, G_IO_IN | G_IO_ERR | G_IO_HUP,
|
|
Packit |
16808d |
listen_cb, NULL);
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
|
|
Packit |
16808d |
int main(int argc, char *argv[]) {
|
|
Packit |
16808d |
int port = 0;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(argc > 1) {
|
|
Packit |
16808d |
char *z;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
host = argv[1];
|
|
Packit |
16808d |
z = host;
|
|
Packit |
16808d |
|
|
Packit |
16808d |
host = strchr(z, ':');
|
|
Packit |
16808d |
if(host) *host++ = '\0';
|
|
Packit |
16808d |
port = atoi(z);
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
if(!host || !*host || !port) {
|
|
Packit |
16808d |
fprintf(stderr,
|
|
Packit |
16808d |
( " Usage: %s local_port:redirect_host\n"
|
|
Packit |
16808d |
" Creates a locally-running sametime proxy which redirects"
|
|
Packit |
16808d |
" logins to the\n specified host\n" ),
|
|
Packit |
16808d |
argv[0]);
|
|
Packit |
16808d |
exit(1);
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|
|
Packit |
16808d |
/* @todo create signal handlers to cleanup socket */
|
|
Packit |
16808d |
|
|
Packit |
16808d |
init_socket(port);
|
|
Packit |
16808d |
|
|
Packit |
16808d |
g_main_loop_run(g_main_loop_new(NULL, FALSE));
|
|
Packit |
16808d |
return 0;
|
|
Packit |
16808d |
}
|
|
Packit |
16808d |
|