Merge branch 'checktalloc' of /home/jelmer/samba4
[amitay/samba.git] / source3 / lib / netapi / cm.c
1 /*
2  *  Unix SMB/CIFS implementation.
3  *  NetApi Support
4  *  Copyright (C) Guenther Deschner 2008
5  *
6  *  This program is free software; you can redistribute it and/or modify
7  *  it under the terms of the GNU General Public License as published by
8  *  the Free Software Foundation; either version 3 of the License, or
9  *  (at your option) any later version.
10  *
11  *  This program is distributed in the hope that it will be useful,
12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  *  GNU General Public License for more details.
15  *
16  *  You should have received a copy of the GNU General Public License
17  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
18  */
19
20 #include "includes.h"
21
22 #include "lib/netapi/netapi.h"
23 #include "lib/netapi/netapi_private.h"
24
25 /********************************************************************
26 ********************************************************************/
27
28 static WERROR libnetapi_open_ipc_connection(struct libnetapi_ctx *ctx,
29                                             const char *server_name,
30                                             struct cli_state **cli)
31 {
32         struct cli_state *cli_ipc = NULL;
33
34         if (!ctx || !cli || !server_name) {
35                 return WERR_INVALID_PARAM;
36         }
37
38         cli_cm_set_signing_state(Undefined);
39
40         if (ctx->use_kerberos) {
41                 cli_cm_set_use_kerberos();
42         }
43
44         if (ctx->password) {
45                 cli_cm_set_password(ctx->password);
46         }
47         if (ctx->username) {
48                 cli_cm_set_username(ctx->username);
49         }
50
51         if (ctx->username && ctx->username[0] &&
52             ctx->password && ctx->password[0] &&
53             ctx->use_kerberos) {
54                 cli_cm_set_fallback_after_kerberos();
55         }
56
57         cli_ipc = cli_cm_open(ctx, NULL,
58                               server_name, "IPC$",
59                               false, false,
60                               PROTOCOL_NT1,
61                               0, 0x20);
62         if (!cli_ipc) {
63                 libnetapi_set_error_string(ctx,
64                         "Failed to connect to IPC$ share on %s", server_name);
65                 return WERR_CAN_NOT_COMPLETE;
66         }
67
68         *cli = cli_ipc;
69
70         return WERR_OK;
71 }
72
73 /********************************************************************
74 ********************************************************************/
75
76 struct client_pipe_connection {
77         struct client_pipe_connection *prev, *next;
78         struct rpc_pipe_client *pipe;
79         struct cli_state *cli;
80 };
81
82 static struct client_pipe_connection *pipe_connections;
83
84 /********************************************************************
85 ********************************************************************/
86
87 WERROR libnetapi_shutdown_cm(struct libnetapi_ctx *ctx)
88 {
89         struct client_pipe_connection *p;
90
91         for (p = pipe_connections; p; p = p->next) {
92                 cli_shutdown(p->cli);
93         }
94
95         return WERR_OK;
96 }
97
98 /********************************************************************
99 ********************************************************************/
100
101 static NTSTATUS pipe_cm_find(struct cli_state *cli,
102                              const struct ndr_syntax_id *interface,
103                              struct rpc_pipe_client **presult)
104 {
105         struct client_pipe_connection *p;
106
107         for (p = pipe_connections; p; p = p->next) {
108
109                 if (!rpc_pipe_np_smb_conn(p->pipe)) {
110                         return NT_STATUS_PIPE_EMPTY;
111                 }
112
113                 if (strequal(cli->desthost, p->pipe->desthost)
114                     && ndr_syntax_id_equal(&p->pipe->abstract_syntax,
115                                            interface)) {
116                         *presult = p->pipe;
117                         return NT_STATUS_OK;
118                 }
119         }
120
121         return NT_STATUS_PIPE_NOT_AVAILABLE;
122 }
123
124 /********************************************************************
125 ********************************************************************/
126
127 static NTSTATUS pipe_cm_connect(TALLOC_CTX *mem_ctx,
128                                 struct cli_state *cli,
129                                 const struct ndr_syntax_id *interface,
130                                 struct rpc_pipe_client **presult)
131 {
132         struct client_pipe_connection *p;
133         NTSTATUS status;
134
135         p = TALLOC_ZERO_ARRAY(mem_ctx, struct client_pipe_connection, 1);
136         if (!p) {
137                 return NT_STATUS_NO_MEMORY;
138         }
139
140         status = cli_rpc_pipe_open_noauth(cli, interface, &p->pipe);
141         if (!NT_STATUS_IS_OK(status)) {
142                 TALLOC_FREE(p);
143                 return status;
144         }
145
146         p->cli = cli;
147         DLIST_ADD(pipe_connections, p);
148
149         *presult = p->pipe;
150         return NT_STATUS_OK;
151 }
152
153 /********************************************************************
154 ********************************************************************/
155
156 static NTSTATUS pipe_cm_open(TALLOC_CTX *ctx,
157                              struct cli_state *cli,
158                              const struct ndr_syntax_id *interface,
159                              struct rpc_pipe_client **presult)
160 {
161         if (NT_STATUS_IS_OK(pipe_cm_find(cli, interface, presult))) {
162                 return NT_STATUS_OK;
163         }
164
165         return pipe_cm_connect(ctx, cli, interface, presult);
166 }
167
168 /********************************************************************
169 ********************************************************************/
170
171 WERROR libnetapi_open_pipe(struct libnetapi_ctx *ctx,
172                            const char *server_name,
173                            const struct ndr_syntax_id *interface,
174                            struct rpc_pipe_client **presult)
175 {
176         struct rpc_pipe_client *result = NULL;
177         NTSTATUS status;
178         WERROR werr;
179         struct cli_state *cli = NULL;
180
181         if (!presult) {
182                 return WERR_INVALID_PARAM;
183         }
184
185         werr = libnetapi_open_ipc_connection(ctx, server_name, &cli);
186         if (!W_ERROR_IS_OK(werr)) {
187                 return werr;
188         }
189
190         status = pipe_cm_open(ctx, cli, interface, &result);
191         if (!NT_STATUS_IS_OK(status)) {
192                 libnetapi_set_error_string(ctx, "failed to open PIPE %s: %s",
193                         get_pipe_name_from_iface(interface),
194                         get_friendly_nt_error_msg(status));
195                 return WERR_DEST_NOT_FOUND;
196         }
197
198         *presult = result;
199
200         return WERR_OK;
201 }