static void delete_config_done(Channel *c, void *client_data, int error) {
PortConnection * conn = (PortConnection *) client_data;
Trap trap;
if (set_trap(&trap)) {
if (!error) {
error = read_errno(&c->inp);
json_test_char(&c->inp, MARKER_EOM);
}
clear_trap(&trap);
}
else {
error = trap.error;
}
if (!conn->auto_connect_stream) {
protocol_send_command(conn->server->channel, "Streams", "disconnect", disconnect_stream_done, conn);
json_write_string(&conn->server->channel->out, conn->out_stream_id);
write_stream(&conn->server->channel->out, MARKER_EOA);
write_stream(&conn->server->channel->out, MARKER_EOM);
}
else {
loc_free(conn->out_stream_id);
conn->out_stream_id = NULL;
loc_free(conn->in_stream_id);
conn->in_stream_id = NULL;
port_unlock(conn);
port_connection_close(conn);
}
}
static void subscribe_cb(Channel * c, void * x, int error) {
Trap trap;
PortConnection * conn = (PortConnection *)x;
PortAttribute * attr = conn->server->redir_info->attrs;
OutputStream * out = &conn->server->channel->out;
if (set_trap(&trap)) {
if (!error) {
error = read_errno(&c->inp);
json_test_char(&c->inp, MARKER_EOM);
}
clear_trap(&trap);
}
else {
error = trap.error;
}
/* Ignore error for subscribe since we may already have subscribed for this channel */
conn->pending = protocol_send_command(conn->server->channel, "PortForward",
"create", portcreate_cb, conn);
write_stream(out, '{');
json_write_string(out, "ID");
write_stream(out, ':');
json_write_string(out, conn->id);
write_stream(out, ',');
while (attr != NULL) {
json_write_string(out, attr->name);
write_stream(out, ':');
write_string(out, attr->value);
attr = attr->next;
}
write_stream(out, '}');
write_stream(out, MARKER_EOA);
write_stream(out, MARKER_EOM);
}
static void getcapabilities_cb(Channel * c, void * x, int error) {
Trap trap;
PortConnection * conn = (PortConnection *)x;
PortAttribute * attr = conn->server->redir_info->attrs;
OutputStream * out = &conn->server->channel->out;
if (set_trap(&trap)) {
if (!error) {
error = read_errno(&c->inp);
json_read_struct(&c->inp, read_getcapabilities_struct, (void *)conn);
json_test_char(&c->inp, MARKER_EOA);
json_test_char(&c->inp, MARKER_EOM);
}
clear_trap(&trap);
}
else {
error = trap.error;
}
if (error) {
connect_port_callback(conn, error);
} else {
if (conn->auto_connect_stream) {
conn->pending = protocol_send_command(conn->server->channel, "PortForward",
"create", portcreate_cb, conn);
write_stream(out, '{');
json_write_string(out, "ID");
write_stream(out, ':');
json_write_string(out, conn->id);
while (attr != NULL) {
write_stream(out, ',');
if (strcmp(attr->name, "RemotePort") == 0) {
json_write_string(out, "Port");
}
else {
json_write_string(out, attr->name);
}
write_stream(out, ':');
write_string(out, attr->value);
attr = attr->next;
}
if (conn->auto_connect_stream) {
write_stream(out, ',');
json_write_string(out, "AutoConnect");
write_stream(out, ':');
json_write_boolean(out, 1);
}
write_stream(out, '}');
write_stream(out, MARKER_EOA);
write_stream(out, MARKER_EOM);
}
else {
conn->pending = protocol_send_command(conn->server->channel, "Streams", "subscribe",
subscribe_cb, conn);
json_write_string(out, "PortForward");
write_stream(out, MARKER_EOA);
write_stream(out, MARKER_EOM);
}
}
}
static void read_stream_done(Channel *c, void *client_data, int error) {
PortConnection * conn = ((PortReadInfo *) client_data)->conn;
int idx = ((PortReadInfo *) client_data)->idx;
size_t read_size = 0;
conn->pending_read_request &= ~(1 << idx);
if (error) {
trace(LOG_ALWAYS, "Reply error %d: %s\n", error, errno_to_str(error));
read_packet_callback(conn, error, idx, 0);
}
else {
int end;
InputStream *inp = &conn->server->channel->inp;
int ch = peek_stream(inp);
if (ch == 'n') {
(void) read_stream(inp);
if (read_stream(inp) != 'u') goto err_json_syntax;
if (read_stream(inp) != 'l') goto err_json_syntax;
if (read_stream(inp) != 'l') goto err_json_syntax;
}
else {
JsonReadBinaryState state;
json_read_binary_start(&state, inp);
for (;;) {
size_t rd = json_read_binary_data(&state,
conn->read_buffer[idx] + read_size,
sizeof conn->read_buffer[idx]);
if (rd == 0) break;
read_size += rd;
}
assert(state.size_start <= 0 || read_size == state.size_start);
json_read_binary_end(&state);
}
json_test_char(&c->inp, MARKER_EOA);
error = read_errno(inp);
(void)json_read_long(inp);
if (read_stream(inp) != 0) goto err_json_syntax;
end = json_read_boolean(inp);
json_test_char(&c->inp, MARKER_EOA);
json_test_char(&c->inp, MARKER_EOM);
#if 0
if (read_stream(inp) != 0 || read_stream(inp) != MARKER_EOM) goto err_json_syntax;
#endif
if (end) read_packet_callback(conn, 0, idx, 0);
else read_packet_callback(conn, 0, idx, read_size);
}
return;
err_json_syntax: return;
}
static void getconfig_cb(Channel * c, void * x, int error) {
Trap trap;
PortConnection * conn = (PortConnection *)x;
if (set_trap(&trap)) {
if (!error) {
error = read_errno(&c->inp);
json_read_struct(&c->inp, read_getconfig_struct, (void *)conn);
json_test_char(&c->inp, MARKER_EOA);
json_test_char(&c->inp, MARKER_EOM);
}
clear_trap(&trap);
}
else {
error = trap.error;
}
connect_port_callback(conn, error);
}
static void write_stream_done(Channel *c, void *client_data, int error) {
Trap trap;
PortConnection * conn = (PortConnection *) client_data;;
if (set_trap(&trap)) {
if (!error) {
error = read_errno(&c->inp);
json_test_char(&c->inp, MARKER_EOM);
}
clear_trap(&trap);
}
else {
error = trap.error;
}
assert (conn->pending_write_request <= MAX_STREAM_WRITE && conn->pending_write_request > 0);
if (conn->pending_write_request == MAX_STREAM_WRITE) {
send_packet_callback(conn, error);
}
conn->pending_write_request--;
}
static void delete_config_done(Channel *c, void *client_data, int error) {
PortConnection * conn = (PortConnection *) client_data;
Trap trap;
if (set_trap(&trap)) {
if (!error) {
error = read_errno(&c->inp);
json_test_char(&c->inp, MARKER_EOM);
}
clear_trap(&trap);
}
else {
error = trap.error;
}
loc_free(conn->out_stream_id);
conn->out_stream_id = NULL;
loc_free(conn->in_stream_id);
conn->in_stream_id = NULL;
port_unlock(conn);
port_connection_close(conn);
}
static void portcreate_cb(Channel * c, void * x, int error) {
Trap trap;
PortConnection * conn = (PortConnection *)x;
if (set_trap(&trap)) {
if (!error) {
error = read_errno(&c->inp);
json_test_char(&c->inp, MARKER_EOM);
}
clear_trap(&trap);
}
else {
error = trap.error;
}
if (error) {
connect_port_callback(conn, error);
} else {
conn->pending = protocol_send_command(conn->server->channel, "PortForward",
"getConfig", getconfig_cb, conn);
json_write_string(&conn->server->channel->out, conn->id);
write_stream(&conn->server->channel->out, MARKER_EOA);
write_stream(&conn->server->channel->out, MARKER_EOM);
}
}