diff options
| -rw-r--r-- | source3/lib/netapi/cm.c | 110 | ||||
| -rw-r--r-- | source3/lib/netapi/netapi_private.h | 4 |
2 files changed, 114 insertions, 0 deletions
diff --git a/source3/lib/netapi/cm.c b/source3/lib/netapi/cm.c index a3b842f96e..96087247d2 100644 --- a/source3/lib/netapi/cm.c +++ b/source3/lib/netapi/cm.c @@ -73,3 +73,113 @@ WERROR libnetapi_shutdown_cm(struct libnetapi_ctx *ctx) return WERR_OK; } + +/******************************************************************** +********************************************************************/ + +struct client_pipe_connection { + struct client_pipe_connection *prev, *next; + struct rpc_pipe_client *pipe; +}; + +static struct client_pipe_connection *pipe_connections; + +/******************************************************************** +********************************************************************/ + +static struct rpc_pipe_client *pipe_cm_find(struct cli_state *cli, + int pipe_idx, + NTSTATUS *status) +{ + struct client_pipe_connection *p; + + for (p = pipe_connections; p; p = p->next) { + + if (!p->pipe->cli) { + *status = NT_STATUS_PIPE_EMPTY; + return NULL; + } + + if (strequal(cli->desthost, p->pipe->cli->desthost) && + pipe_idx == p->pipe->pipe_idx) { + *status = NT_STATUS_OK; + return p->pipe; + } + } + + *status = NT_STATUS_PIPE_NOT_AVAILABLE; + + return NULL; +} + +/******************************************************************** +********************************************************************/ + +static struct rpc_pipe_client *pipe_cm_connect(TALLOC_CTX *mem_ctx, + struct cli_state *cli, + int pipe_idx, + NTSTATUS *status) +{ + struct client_pipe_connection *p; + + p = TALLOC_ZERO_ARRAY(mem_ctx, struct client_pipe_connection, 1); + if (!p) { + *status = NT_STATUS_NO_MEMORY; + return NULL; + } + + p->pipe = cli_rpc_pipe_open_noauth(cli, pipe_idx, status); + if (!p->pipe) { + TALLOC_FREE(p); + return NULL; + } + + DLIST_ADD(pipe_connections, p); + + return p->pipe; +} + +/******************************************************************** +********************************************************************/ + +static struct rpc_pipe_client *pipe_cm_open(TALLOC_CTX *ctx, + struct cli_state *cli, + int pipe_idx, + NTSTATUS *status) +{ + struct rpc_pipe_client *p; + + p = pipe_cm_find(cli, pipe_idx, status); + if (!p) { + p = pipe_cm_connect(ctx, cli, pipe_idx, status); + } + + return p; +} + +/******************************************************************** +********************************************************************/ + +WERROR libnetapi_open_pipe(struct libnetapi_ctx *ctx, + struct cli_state *cli, + int pipe_idx, + struct rpc_pipe_client **pipe_cli) +{ + NTSTATUS status; + + if (!cli || !pipe_cli) { + return WERR_INVALID_PARAM; + } + + *pipe_cli = pipe_cm_open(ctx, cli, pipe_idx, &status); + if (!*pipe_cli) { + libnetapi_set_error_string(ctx, "failed to open PIPE %s: %s", + cli_get_pipe_name(pipe_idx), + get_friendly_nt_error_msg(status)); + return WERR_DEST_NOT_FOUND; + } + + return WERR_OK; +} + + diff --git a/source3/lib/netapi/netapi_private.h b/source3/lib/netapi/netapi_private.h index 2591a7bc09..a575f42f4e 100644 --- a/source3/lib/netapi/netapi_private.h +++ b/source3/lib/netapi/netapi_private.h @@ -29,4 +29,8 @@ WERROR libnetapi_open_ipc_connection(struct libnetapi_ctx *ctx, const char *server_name, struct cli_state **cli); WERROR libnetapi_shutdown_cm(struct libnetapi_ctx *ctx); +WERROR libnetapi_open_pipe(struct libnetapi_ctx *ctx, + struct cli_state *cli, + int pipe_idx, + struct rpc_pipe_client **pipe_cli); #endif |