mirror of
https://github.com/samba-team/samba.git
synced 2025-01-10 01:18:15 +03:00
790 lines
22 KiB
C
790 lines
22 KiB
C
|
/*
|
||
|
Unix SMB/CIFS implementation.
|
||
|
|
||
|
[MS-RPCH] - RPC over HTTP client
|
||
|
|
||
|
Copyright (C) 2013 Samuel Cabrero <samuelcabrero@kernevil.me>
|
||
|
|
||
|
This program 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.
|
||
|
|
||
|
This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
*/
|
||
|
|
||
|
#include "includes.h"
|
||
|
#include "lib/events/events.h"
|
||
|
#include "lib/util/tevent_ntstatus.h"
|
||
|
#include "lib/tls/tls.h"
|
||
|
#include "libcli/resolve/resolve.h"
|
||
|
#include "libcli/composite/composite.h"
|
||
|
#include "auth/credentials/credentials.h"
|
||
|
#include "tsocket/tsocket.h"
|
||
|
#include "tsocket/tsocket_internal.h"
|
||
|
#include "librpc/rpc/dcerpc.h"
|
||
|
#include "librpc/rpc/dcerpc_roh.h"
|
||
|
#include "librpc/rpc/dcerpc_proto.h"
|
||
|
|
||
|
static ssize_t tstream_roh_pending_bytes(struct tstream_context *stream);
|
||
|
static struct tevent_req * tstream_roh_readv_send(
|
||
|
TALLOC_CTX *mem_ctx,
|
||
|
struct tevent_context *ev,
|
||
|
struct tstream_context *stream,
|
||
|
struct iovec *vector,
|
||
|
size_t count);
|
||
|
static int tstream_roh_readv_recv(struct tevent_req *req, int *perrno);
|
||
|
static struct tevent_req * tstream_roh_writev_send(
|
||
|
TALLOC_CTX *mem_ctx,
|
||
|
struct tevent_context *ev,
|
||
|
struct tstream_context *stream,
|
||
|
const struct iovec *vector,
|
||
|
size_t count);
|
||
|
static int tstream_roh_writev_recv(struct tevent_req *req, int *perrno);
|
||
|
static struct tevent_req * tstream_roh_disconnect_send(
|
||
|
TALLOC_CTX *mem_ctx,
|
||
|
struct tevent_context *ev,
|
||
|
struct tstream_context *stream);
|
||
|
static int tstream_roh_disconnect_recv(struct tevent_req *req, int *perrno);
|
||
|
|
||
|
static const struct tstream_context_ops tstream_roh_ops = {
|
||
|
.name = "roh",
|
||
|
.pending_bytes = tstream_roh_pending_bytes,
|
||
|
.readv_send = tstream_roh_readv_send,
|
||
|
.readv_recv = tstream_roh_readv_recv,
|
||
|
.writev_send = tstream_roh_writev_send,
|
||
|
.writev_recv = tstream_roh_writev_recv,
|
||
|
.disconnect_send = tstream_roh_disconnect_send,
|
||
|
.disconnect_recv = tstream_roh_disconnect_recv,
|
||
|
};
|
||
|
|
||
|
struct tstream_roh_context {
|
||
|
struct roh_connection *roh_conn;
|
||
|
};
|
||
|
|
||
|
struct roh_open_connection_state {
|
||
|
struct tevent_req *req;
|
||
|
struct tevent_context *event_ctx;
|
||
|
struct cli_credentials *credentials;
|
||
|
struct resolve_context *resolve_ctx;
|
||
|
const char **rpcproxy_addresses;
|
||
|
unsigned int rpcproxy_address_index;
|
||
|
|
||
|
struct dcecli_connection *conn;
|
||
|
bool tls;
|
||
|
|
||
|
const char *rpc_proxy;
|
||
|
unsigned int rpc_proxy_port;
|
||
|
const char *rpc_server;
|
||
|
unsigned int rpc_server_port;
|
||
|
const char *target_hostname;
|
||
|
|
||
|
struct roh_connection *roh;
|
||
|
struct tstream_tls_params *tls_params;
|
||
|
struct loadparm_context *lp_ctx;
|
||
|
bool use_ntlm;
|
||
|
};
|
||
|
|
||
|
NTSTATUS dcerpc_pipe_open_roh_recv(struct tevent_req *req,
|
||
|
TALLOC_CTX *mem_ctx,
|
||
|
struct tstream_context **stream,
|
||
|
struct tevent_queue **queue)
|
||
|
{
|
||
|
struct roh_open_connection_state *state;
|
||
|
struct tstream_roh_context *roh_stream_ctx;
|
||
|
NTSTATUS status;
|
||
|
|
||
|
state = tevent_req_data(req, struct roh_open_connection_state);
|
||
|
if (tevent_req_is_nterror(req, &status)) {
|
||
|
tevent_req_received(req);
|
||
|
return status;
|
||
|
}
|
||
|
|
||
|
*stream = tstream_context_create(mem_ctx, &tstream_roh_ops,
|
||
|
&roh_stream_ctx,
|
||
|
struct tstream_roh_context,
|
||
|
__location__);
|
||
|
if (!stream) {
|
||
|
tevent_req_received(req);
|
||
|
return NT_STATUS_NO_MEMORY;
|
||
|
}
|
||
|
ZERO_STRUCTP(roh_stream_ctx);
|
||
|
|
||
|
roh_stream_ctx->roh_conn = talloc_move(mem_ctx, &state->roh);
|
||
|
*queue = roh_stream_ctx->roh_conn->default_channel_in->send_queue;
|
||
|
|
||
|
tevent_req_received(req);
|
||
|
|
||
|
return NT_STATUS_OK;
|
||
|
}
|
||
|
|
||
|
static void roh_continue_resolve_name(struct composite_context *ctx);
|
||
|
|
||
|
/**
|
||
|
* Send rpc pipe open request to given host:port using http transport
|
||
|
*/
|
||
|
struct tevent_req *dcerpc_pipe_open_roh_send(struct dcecli_connection *conn,
|
||
|
const char *localaddr,
|
||
|
const char *rpc_server,
|
||
|
uint32_t rpc_server_port,
|
||
|
const char *rpc_proxy,
|
||
|
uint32_t rpc_proxy_port,
|
||
|
const char *http_proxy,
|
||
|
uint32_t http_proxy_port,
|
||
|
bool use_tls,
|
||
|
bool use_proxy,
|
||
|
struct cli_credentials *credentials,
|
||
|
struct resolve_context *resolve_ctx,
|
||
|
struct loadparm_context *lp_ctx,
|
||
|
bool use_ntlm)
|
||
|
{
|
||
|
NTSTATUS status;
|
||
|
struct tevent_req *req;
|
||
|
struct composite_context *ctx;
|
||
|
struct roh_open_connection_state *state;
|
||
|
struct nbt_name name;
|
||
|
|
||
|
req = tevent_req_create(conn, &state, struct roh_open_connection_state);
|
||
|
if (req == NULL) {
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
/* Set state fields */
|
||
|
state->req = req;
|
||
|
state->event_ctx = conn->event_ctx;
|
||
|
state->lp_ctx = lp_ctx,
|
||
|
state->credentials = credentials;
|
||
|
state->conn = conn;
|
||
|
state->tls = use_tls;
|
||
|
|
||
|
/* Initialize connection structure (3.2.1.3) */
|
||
|
/* TODO Initialize virtual connection cookie table */
|
||
|
state->rpc_server = talloc_strdup(state, rpc_server);
|
||
|
state->rpc_server_port = rpc_server_port;
|
||
|
state->rpc_proxy = talloc_strdup(state, rpc_proxy);
|
||
|
state->rpc_proxy_port = rpc_proxy_port;
|
||
|
state->use_ntlm = use_ntlm;
|
||
|
|
||
|
state->roh = talloc_zero(state, struct roh_connection);
|
||
|
state->roh->protocol_version = ROH_V2;
|
||
|
state->roh->connection_state = ROH_STATE_OPEN_START;
|
||
|
state->roh->connection_cookie = GUID_random();
|
||
|
state->roh->association_group_id_cookie = GUID_random();
|
||
|
|
||
|
/* Additional initialization steps (3.2.2.3) */
|
||
|
state->roh->proxy_use = use_proxy;
|
||
|
state->roh->current_keep_alive_time = 0;
|
||
|
state->roh->current_keep_alive_interval = 0;
|
||
|
|
||
|
/* Initialize TLS */
|
||
|
if (use_tls) {
|
||
|
status = tstream_tls_params_client(state->roh, NULL, NULL,
|
||
|
&state->tls_params);
|
||
|
if (!NT_STATUS_IS_OK(status)) {
|
||
|
DEBUG(0,("%s: Failed tstream_tls_params_client - %s\n",
|
||
|
__func__, nt_errstr(status)));
|
||
|
tevent_req_nterror(req, status);
|
||
|
return tevent_req_post(req, conn->event_ctx);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Resolve RPC proxy server name */
|
||
|
make_nbt_name_server(&name, state->rpc_proxy);
|
||
|
ctx = resolve_name_send(resolve_ctx, state, &name, state->event_ctx);
|
||
|
if (tevent_req_nomem(ctx, req)) {
|
||
|
return tevent_req_post(req, state->event_ctx);
|
||
|
}
|
||
|
ctx->async.fn = roh_continue_resolve_name;
|
||
|
ctx->async.private_data = state;
|
||
|
|
||
|
return req;
|
||
|
}
|
||
|
|
||
|
static void roh_connect_channel_in_done(struct tevent_req *subreq);
|
||
|
static void roh_continue_resolve_name(struct composite_context *ctx)
|
||
|
{
|
||
|
NTSTATUS status;
|
||
|
struct roh_open_connection_state *state;
|
||
|
struct tevent_req *subreq;
|
||
|
|
||
|
state = talloc_get_type_abort(ctx->async.private_data,
|
||
|
struct roh_open_connection_state);
|
||
|
status = resolve_name_multiple_recv(ctx, state,
|
||
|
&state->rpcproxy_addresses);
|
||
|
if (tevent_req_nterror(state->req, status)) {
|
||
|
DEBUG(2, ("%s: No server found: %s\n", __func__,
|
||
|
nt_errstr(status)));
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
state->rpcproxy_address_index = 0;
|
||
|
if (state->rpcproxy_addresses[state->rpcproxy_address_index] == NULL) {
|
||
|
DEBUG(2, ("%s: No server found\n", __func__));
|
||
|
tevent_req_nterror(state->req, NT_STATUS_OBJECT_NAME_NOT_FOUND);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* TODO Determine proxy use
|
||
|
* If state->roh->proxy_use == true, the client has requested to
|
||
|
* always use local proxy. Otherwise, run the proxy use discovery
|
||
|
*/
|
||
|
state->roh->connection_state = ROH_STATE_OPEN_START;
|
||
|
subreq = roh_connect_channel_in_send(state,
|
||
|
state->event_ctx,
|
||
|
state->rpcproxy_addresses[state->rpcproxy_address_index],
|
||
|
state->rpc_proxy_port,
|
||
|
state->credentials,
|
||
|
state->roh, state->tls,
|
||
|
state->tls_params);
|
||
|
if (tevent_req_nomem(subreq, state->req)) {
|
||
|
return;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, roh_connect_channel_in_done, state->req);
|
||
|
}
|
||
|
|
||
|
static void roh_connect_channel_out_done(struct tevent_req *);
|
||
|
static void roh_connect_channel_in_done(struct tevent_req *subreq)
|
||
|
{
|
||
|
NTSTATUS status;
|
||
|
struct tevent_req *req;
|
||
|
struct roh_open_connection_state *state;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct roh_open_connection_state);
|
||
|
|
||
|
status = roh_connect_channel_in_recv(subreq);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (tevent_req_nterror(req, status)) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
subreq = roh_connect_channel_out_send(state,
|
||
|
state->event_ctx,
|
||
|
state->rpcproxy_addresses[state->rpcproxy_address_index],
|
||
|
state->rpc_proxy_port,
|
||
|
state->credentials,
|
||
|
state->roh,
|
||
|
state->tls,
|
||
|
state->tls_params);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
return;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, roh_connect_channel_out_done, req);
|
||
|
}
|
||
|
|
||
|
static void roh_send_RPC_DATA_IN_done(struct tevent_req *);
|
||
|
static void roh_connect_channel_out_done(struct tevent_req *subreq)
|
||
|
{
|
||
|
NTSTATUS status;
|
||
|
struct tevent_req *req;
|
||
|
struct roh_open_connection_state *state;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct roh_open_connection_state);
|
||
|
|
||
|
status = roh_connect_channel_out_recv(subreq);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (tevent_req_nterror(req, status)) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
subreq = roh_send_RPC_DATA_IN_send(state, state->lp_ctx,
|
||
|
state->event_ctx,
|
||
|
state->credentials,
|
||
|
state->roh,
|
||
|
state->rpc_server,
|
||
|
state->rpc_server_port,
|
||
|
state->rpc_proxy,
|
||
|
state->use_ntlm);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
return;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, roh_send_RPC_DATA_IN_done, req);
|
||
|
}
|
||
|
|
||
|
static void roh_send_RPC_DATA_OUT_done(struct tevent_req *);
|
||
|
static void roh_send_RPC_DATA_IN_done(struct tevent_req *subreq)
|
||
|
{
|
||
|
NTSTATUS status;
|
||
|
struct tevent_req *req;
|
||
|
struct roh_open_connection_state *state;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct roh_open_connection_state);
|
||
|
|
||
|
status = roh_send_RPC_DATA_IN_recv(subreq);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (tevent_req_nterror(req, status)) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
subreq = roh_send_RPC_DATA_OUT_send(state,
|
||
|
state->lp_ctx,
|
||
|
state->event_ctx,
|
||
|
state->credentials,
|
||
|
state->roh,
|
||
|
state->rpc_server,
|
||
|
state->rpc_server_port,
|
||
|
state->rpc_proxy,
|
||
|
state->use_ntlm);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
return;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, roh_send_RPC_DATA_OUT_done, req);
|
||
|
}
|
||
|
|
||
|
static void roh_send_CONN_A1_done(struct tevent_req *);
|
||
|
static void roh_send_RPC_DATA_OUT_done(struct tevent_req *subreq)
|
||
|
{
|
||
|
NTSTATUS status;
|
||
|
struct tevent_req *req;
|
||
|
struct roh_open_connection_state *state;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct roh_open_connection_state);
|
||
|
|
||
|
status = roh_send_RPC_DATA_OUT_recv(subreq);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (tevent_req_nterror(req, status)) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
subreq = roh_send_CONN_A1_send(state, state->event_ctx, state->roh);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
return;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, roh_send_CONN_A1_done, req);
|
||
|
}
|
||
|
|
||
|
static void roh_send_CONN_B1_done(struct tevent_req *);
|
||
|
static void roh_send_CONN_A1_done(struct tevent_req *subreq)
|
||
|
{
|
||
|
NTSTATUS status;
|
||
|
struct tevent_req *req;
|
||
|
struct roh_open_connection_state *state;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct roh_open_connection_state);
|
||
|
|
||
|
status = roh_send_CONN_A1_recv(subreq);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (tevent_req_nterror(req, status)) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
subreq = roh_send_CONN_B1_send(state, state->event_ctx, state->roh);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
return;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, roh_send_CONN_B1_done, req);
|
||
|
}
|
||
|
|
||
|
static void roh_recv_out_channel_response_done(struct tevent_req *);
|
||
|
static void roh_send_CONN_B1_done(struct tevent_req *subreq)
|
||
|
{
|
||
|
NTSTATUS status;
|
||
|
struct tevent_req *req;
|
||
|
struct roh_open_connection_state *state;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct roh_open_connection_state);
|
||
|
|
||
|
status = roh_send_CONN_B1_recv(subreq);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (tevent_req_nterror(req, status)) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
state->roh->connection_state = ROH_STATE_OUT_CHANNEL_WAIT;
|
||
|
subreq = roh_recv_out_channel_response_send(state, state->event_ctx,
|
||
|
state->roh);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
return;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, roh_recv_out_channel_response_done, req);
|
||
|
}
|
||
|
|
||
|
static void roh_recv_CONN_A3_done(struct tevent_req *);
|
||
|
static void roh_recv_out_channel_response_done(struct tevent_req *subreq)
|
||
|
{
|
||
|
NTSTATUS status;
|
||
|
char *response;
|
||
|
struct tevent_req *req;
|
||
|
struct roh_open_connection_state *state;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct roh_open_connection_state);
|
||
|
|
||
|
status = roh_recv_out_channel_response_recv(subreq, state, &response);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (tevent_req_nterror(req, status)) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
state->roh->connection_state = ROH_STATE_WAIT_A3W;
|
||
|
subreq = roh_recv_CONN_A3_send(state, state->event_ctx, state->roh);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
return;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, roh_recv_CONN_A3_done, req);
|
||
|
}
|
||
|
|
||
|
static void roh_recv_CONN_C2_done(struct tevent_req *);
|
||
|
static void roh_recv_CONN_A3_done(struct tevent_req *subreq)
|
||
|
{
|
||
|
NTSTATUS status;
|
||
|
struct tevent_req *req;
|
||
|
struct roh_open_connection_state *state;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct roh_open_connection_state);
|
||
|
|
||
|
status = roh_recv_CONN_A3_recv(subreq, &state->roh->default_channel_out->connection_timeout);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (tevent_req_nterror(req, status)) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
state->roh->connection_state = ROH_STATE_WAIT_C2;
|
||
|
subreq = roh_recv_CONN_C2_send(state, state->event_ctx, state->roh);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
return;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, roh_recv_CONN_C2_done, req);
|
||
|
}
|
||
|
|
||
|
static void roh_recv_CONN_C2_done(struct tevent_req *subreq)
|
||
|
{
|
||
|
NTSTATUS status;
|
||
|
struct tevent_req *req;
|
||
|
struct roh_open_connection_state *state;
|
||
|
unsigned int version;
|
||
|
unsigned int recv;
|
||
|
unsigned int timeout;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct roh_open_connection_state);
|
||
|
|
||
|
status = roh_recv_CONN_C2_recv(subreq, &version, &recv, &timeout);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (tevent_req_nterror(req, status)) {
|
||
|
return;
|
||
|
}
|
||
|
state->roh->connection_state = ROH_STATE_OPENED;
|
||
|
|
||
|
tevent_req_done(req);
|
||
|
}
|
||
|
|
||
|
static ssize_t tstream_roh_pending_bytes(struct tstream_context *stream)
|
||
|
{
|
||
|
struct tstream_roh_context *ctx = NULL;
|
||
|
|
||
|
ctx = tstream_context_data(stream, struct tstream_roh_context);
|
||
|
if (!ctx->roh_conn) {
|
||
|
errno = ENOTCONN;
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
return tstream_pending_bytes(ctx->roh_conn->default_channel_out->streams.active);
|
||
|
}
|
||
|
|
||
|
struct tstream_roh_readv_state {
|
||
|
struct roh_connection *roh_conn;
|
||
|
int ret;
|
||
|
};
|
||
|
|
||
|
static void tstream_roh_readv_handler(struct tevent_req *subreq);
|
||
|
static struct tevent_req * tstream_roh_readv_send(TALLOC_CTX *mem_ctx,
|
||
|
struct tevent_context *ev,
|
||
|
struct tstream_context *stream,
|
||
|
struct iovec *vector,
|
||
|
size_t count)
|
||
|
{
|
||
|
struct tstream_roh_context *ctx = NULL;
|
||
|
struct tstream_roh_readv_state *state;
|
||
|
struct tevent_req *req, *subreq;
|
||
|
|
||
|
req = tevent_req_create(mem_ctx, &state, struct tstream_roh_readv_state);
|
||
|
if (!req) {
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
ctx = tstream_context_data(stream, struct tstream_roh_context);
|
||
|
if (!ctx->roh_conn) {
|
||
|
tevent_req_error(req, ENOTCONN);
|
||
|
goto post;
|
||
|
}
|
||
|
if (!ctx->roh_conn->default_channel_out) {
|
||
|
tevent_req_error(req, ENOTCONN);
|
||
|
goto post;
|
||
|
}
|
||
|
if (!ctx->roh_conn->default_channel_out->streams.active) {
|
||
|
tevent_req_error(req, ENOTCONN);
|
||
|
goto post;
|
||
|
}
|
||
|
|
||
|
state->roh_conn = ctx->roh_conn;
|
||
|
|
||
|
subreq = tstream_readv_send(state, ev,
|
||
|
ctx->roh_conn->default_channel_out->streams.active,
|
||
|
vector, count);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
goto post;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, tstream_roh_readv_handler, req);
|
||
|
|
||
|
return req;
|
||
|
post:
|
||
|
tevent_req_post(req, ev);
|
||
|
return req;
|
||
|
}
|
||
|
|
||
|
static void tstream_roh_readv_handler(struct tevent_req *subreq)
|
||
|
{
|
||
|
struct tevent_req *req;
|
||
|
struct tstream_roh_readv_state *state;
|
||
|
int ret;
|
||
|
int sys_errno;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct tstream_roh_readv_state);
|
||
|
ret = tstream_readv_recv(subreq, &sys_errno);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (ret == -1) {
|
||
|
tevent_req_error(req, sys_errno);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
state->ret = ret;
|
||
|
|
||
|
tevent_req_done(req);
|
||
|
}
|
||
|
|
||
|
static int tstream_roh_readv_recv(struct tevent_req *req, int *perrno)
|
||
|
{
|
||
|
struct tstream_roh_readv_state *state;
|
||
|
int ret;
|
||
|
|
||
|
state = tevent_req_data(req, struct tstream_roh_readv_state);
|
||
|
ret = tsocket_simple_int_recv(req, perrno);
|
||
|
if (ret == 0) {
|
||
|
ret = state->ret;
|
||
|
}
|
||
|
|
||
|
tevent_req_received(req);
|
||
|
return ret;
|
||
|
}
|
||
|
|
||
|
struct tstream_roh_writev_state {
|
||
|
struct roh_connection *roh_conn;
|
||
|
int nwritten;
|
||
|
};
|
||
|
|
||
|
static void tstream_roh_writev_handler(struct tevent_req *subreq);
|
||
|
static struct tevent_req * tstream_roh_writev_send(TALLOC_CTX *mem_ctx,
|
||
|
struct tevent_context *ev,
|
||
|
struct tstream_context *stream,
|
||
|
const struct iovec *vector,
|
||
|
size_t count)
|
||
|
{
|
||
|
struct tstream_roh_context *ctx = NULL;
|
||
|
struct tstream_roh_writev_state *state = NULL;
|
||
|
struct tevent_req *req = NULL;
|
||
|
struct tevent_req *subreq = NULL;
|
||
|
|
||
|
req = tevent_req_create(mem_ctx, &state,
|
||
|
struct tstream_roh_writev_state);
|
||
|
if (!req) {
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
ctx = tstream_context_data(stream, struct tstream_roh_context);
|
||
|
if (!ctx->roh_conn) {
|
||
|
tevent_req_error(req, ENOTCONN);
|
||
|
goto post;
|
||
|
}
|
||
|
if (!ctx->roh_conn->default_channel_in) {
|
||
|
tevent_req_error(req, ENOTCONN);
|
||
|
goto post;
|
||
|
}
|
||
|
if (!ctx->roh_conn->default_channel_in->streams.active) {
|
||
|
tevent_req_error(req, ENOTCONN);
|
||
|
goto post;
|
||
|
}
|
||
|
|
||
|
state->roh_conn = ctx->roh_conn;
|
||
|
|
||
|
subreq = tstream_writev_send(state, ev,
|
||
|
ctx->roh_conn->default_channel_in->streams.active,
|
||
|
vector, count);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
goto post;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, tstream_roh_writev_handler, req);
|
||
|
|
||
|
return req;
|
||
|
post:
|
||
|
tevent_req_post(req, ev);
|
||
|
return req;
|
||
|
}
|
||
|
|
||
|
static void tstream_roh_writev_handler(struct tevent_req *subreq)
|
||
|
{
|
||
|
struct tevent_req *req;
|
||
|
struct tstream_roh_writev_state *state;
|
||
|
int nwritten;
|
||
|
int sys_errno;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct tstream_roh_writev_state);
|
||
|
nwritten = tstream_writev_recv(subreq, &sys_errno);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (nwritten == -1) {
|
||
|
tevent_req_error(req, sys_errno);
|
||
|
return;
|
||
|
}
|
||
|
state->nwritten = nwritten;
|
||
|
state->roh_conn->default_channel_in->sent_bytes += nwritten;
|
||
|
|
||
|
tevent_req_done(req);
|
||
|
}
|
||
|
|
||
|
static int tstream_roh_writev_recv(struct tevent_req *req, int *perrno)
|
||
|
{
|
||
|
struct tstream_roh_writev_state *state;
|
||
|
int ret;
|
||
|
|
||
|
state = tevent_req_data(req, struct tstream_roh_writev_state);
|
||
|
ret = tsocket_simple_int_recv(req, perrno);
|
||
|
if (ret == 0) {
|
||
|
ret = state->nwritten;
|
||
|
}
|
||
|
|
||
|
return ret;
|
||
|
}
|
||
|
|
||
|
struct tstream_roh_disconnect_state {
|
||
|
struct tstream_context *stream;
|
||
|
struct tevent_context *ev;
|
||
|
};
|
||
|
|
||
|
static void tstream_roh_disconnect_channel_in_handler(struct tevent_req *subreq);
|
||
|
static struct tevent_req * tstream_roh_disconnect_send(TALLOC_CTX *mem_ctx,
|
||
|
struct tevent_context *ev,
|
||
|
struct tstream_context *stream)
|
||
|
{
|
||
|
struct tstream_roh_context *ctx = NULL;
|
||
|
struct tevent_req *req, *subreq;
|
||
|
struct tstream_roh_disconnect_state *state;
|
||
|
|
||
|
req = tevent_req_create(mem_ctx, &state, struct tstream_roh_disconnect_state);
|
||
|
if (req == NULL) {
|
||
|
return NULL;
|
||
|
}
|
||
|
|
||
|
state->stream = stream;
|
||
|
state->ev = ev;
|
||
|
|
||
|
ctx = tstream_context_data(stream, struct tstream_roh_context);
|
||
|
if (!ctx->roh_conn) {
|
||
|
tevent_req_error(req, ENOTCONN);
|
||
|
goto post;
|
||
|
}
|
||
|
if (!ctx->roh_conn->default_channel_in) {
|
||
|
tevent_req_error(req, ENOTCONN);
|
||
|
goto post;
|
||
|
}
|
||
|
if (!ctx->roh_conn->default_channel_in->streams.active) {
|
||
|
tevent_req_error(req, ENOTCONN);
|
||
|
goto post;
|
||
|
}
|
||
|
|
||
|
subreq = tstream_disconnect_send(state, ev, ctx->roh_conn->default_channel_in->streams.active);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
goto post;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, tstream_roh_disconnect_channel_in_handler, req);
|
||
|
|
||
|
return req;
|
||
|
post:
|
||
|
tevent_req_post(req, ev);
|
||
|
return req;
|
||
|
}
|
||
|
|
||
|
static void tstream_roh_disconnect_channel_out_handler(struct tevent_req *subreq);
|
||
|
|
||
|
static void tstream_roh_disconnect_channel_in_handler(struct tevent_req *subreq)
|
||
|
{
|
||
|
struct tevent_req *req;
|
||
|
struct tstream_roh_disconnect_state *state;
|
||
|
struct tstream_context *stream;
|
||
|
struct tstream_roh_context *roh_stream;
|
||
|
int ret;
|
||
|
int sys_errno;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct tstream_roh_disconnect_state);
|
||
|
stream = state->stream;
|
||
|
roh_stream = tstream_context_data(stream, struct tstream_roh_context);
|
||
|
|
||
|
ret = tstream_disconnect_recv(subreq, &sys_errno);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (ret == -1) {
|
||
|
tevent_req_error(req, sys_errno);
|
||
|
return;
|
||
|
}
|
||
|
TALLOC_FREE(roh_stream->roh_conn->default_channel_in);
|
||
|
|
||
|
subreq = tstream_disconnect_send(state,
|
||
|
state->ev,
|
||
|
roh_stream->roh_conn->default_channel_out->streams.raw);
|
||
|
if (tevent_req_nomem(subreq, req)) {
|
||
|
return;
|
||
|
}
|
||
|
tevent_req_set_callback(subreq, tstream_roh_disconnect_channel_out_handler, req);
|
||
|
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
static void tstream_roh_disconnect_channel_out_handler(struct tevent_req *subreq)
|
||
|
{
|
||
|
struct tevent_req *req;
|
||
|
struct tstream_roh_disconnect_state *state;
|
||
|
struct tstream_context *stream;
|
||
|
struct tstream_roh_context *roh_stream;
|
||
|
int ret;
|
||
|
int sys_errno;
|
||
|
|
||
|
req = tevent_req_callback_data(subreq, struct tevent_req);
|
||
|
state = tevent_req_data(req, struct tstream_roh_disconnect_state);
|
||
|
stream = state->stream;
|
||
|
roh_stream = tstream_context_data(stream, struct tstream_roh_context);
|
||
|
|
||
|
ret = tstream_disconnect_recv(subreq, &sys_errno);
|
||
|
TALLOC_FREE(subreq);
|
||
|
if (ret == -1) {
|
||
|
tevent_req_error(req, sys_errno);
|
||
|
return;
|
||
|
}
|
||
|
TALLOC_FREE(roh_stream->roh_conn->default_channel_out);
|
||
|
|
||
|
tevent_req_done(req);
|
||
|
}
|
||
|
|
||
|
static int tstream_roh_disconnect_recv(struct tevent_req *req, int *perrno)
|
||
|
{
|
||
|
int ret;
|
||
|
|
||
|
ret = tsocket_simple_int_recv(req, perrno);
|
||
|
tevent_req_received(req);
|
||
|
|
||
|
return ret;
|
||
|
}
|