f0f803dac5df656ef848c11c89abb2f95757338f
[samba.git] / source4 / winbind / wb_samba3_protocol.c
1 /* 
2    Unix SMB/CIFS implementation.
3    Main winbindd samba3 server routines
4
5    Copyright (C) Stefan Metzmacher      2005
6    Copyright (C) Volker Lendecke        2005
7
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 3 of the License, or
11    (at your option) any later version.
12    
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17    
18    You should have received a copy of the GNU General Public License
19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
20 */
21
22 #include "includes.h"
23 #include "winbind/wb_server.h"
24 #include "smbd/service_stream.h"
25 #include "lib/stream/packet.h"
26 #include "lib/tsocket/tsocket.h"
27
28 /*
29   work out if a packet is complete for protocols that use a 32 bit host byte
30   order length
31 */
32 NTSTATUS wbsrv_samba3_packet_full_request(void *private_data, DATA_BLOB blob, size_t *size)
33 {
34         uint32_t *len;
35         if (blob.length < 4) {
36                 return STATUS_MORE_ENTRIES;
37         }
38         len = (uint32_t *)blob.data;
39         *size = (*len);
40         if (*size > blob.length) {
41                 return STATUS_MORE_ENTRIES;
42         }
43         return NT_STATUS_OK;
44 }
45
46
47 NTSTATUS wbsrv_samba3_pull_request(struct wbsrv_samba3_call *call)
48 {
49         if (call->in.length != sizeof(*call->request)) {
50                 DEBUG(0,("wbsrv_samba3_pull_request: invalid blob length %lu should be %lu\n"
51                          " make sure you use the correct winbind client tools!\n",
52                          (long)call->in.length, (long)sizeof(*call->request)));
53                 return NT_STATUS_INVALID_PARAMETER;
54         }
55
56         call->request = talloc_zero(call, struct winbindd_request);
57         NT_STATUS_HAVE_NO_MEMORY(call->request);
58
59         /* the packet layout is the same as the in memory layout of the request, so just copy it */
60         memcpy(call->request, call->in.data, sizeof(*call->request));
61
62         return NT_STATUS_OK;
63 }
64
65 NTSTATUS wbsrv_samba3_handle_call(struct wbsrv_samba3_call *s3call)
66 {
67         DEBUG(10, ("Got winbind samba3 request %d\n", s3call->request->cmd));
68
69         s3call->response = talloc_zero(s3call, struct winbindd_response);
70         NT_STATUS_HAVE_NO_MEMORY(s3call->request);
71
72         s3call->response->length = sizeof(*s3call->response);
73
74         switch(s3call->request->cmd) {
75         case WINBINDD_INTERFACE_VERSION:
76                 return wbsrv_samba3_interface_version(s3call);
77
78         case WINBINDD_CHECK_MACHACC:
79                 return wbsrv_samba3_check_machacc(s3call);
80
81         case WINBINDD_PING:
82                 return wbsrv_samba3_ping(s3call);
83
84         case WINBINDD_INFO:
85                 return wbsrv_samba3_info(s3call);
86
87         case WINBINDD_DOMAIN_NAME:
88                 return wbsrv_samba3_domain_name(s3call);
89
90         case WINBINDD_NETBIOS_NAME:
91                 return wbsrv_samba3_netbios_name(s3call);
92
93         case WINBINDD_PRIV_PIPE_DIR:
94                 return wbsrv_samba3_priv_pipe_dir(s3call);
95
96         case WINBINDD_LOOKUPNAME:
97                 return wbsrv_samba3_lookupname(s3call);
98
99         case WINBINDD_LOOKUPSID:
100                 return wbsrv_samba3_lookupsid(s3call);
101
102         case WINBINDD_PAM_AUTH:
103                 return wbsrv_samba3_pam_auth(s3call);
104
105         case WINBINDD_PAM_AUTH_CRAP:
106                 return wbsrv_samba3_pam_auth_crap(s3call);
107
108         case WINBINDD_GETDCNAME:
109                 return wbsrv_samba3_getdcname(s3call);
110
111         case WINBINDD_GETUSERDOMGROUPS:
112                 return wbsrv_samba3_userdomgroups(s3call);
113
114         case WINBINDD_GETUSERSIDS:
115                 return wbsrv_samba3_usersids(s3call);
116
117         case WINBINDD_LIST_GROUPS:
118                 return wbsrv_samba3_list_groups(s3call);
119
120         case WINBINDD_LIST_TRUSTDOM:
121                 return wbsrv_samba3_list_trustdom(s3call);
122
123         case WINBINDD_LIST_USERS:
124                 return wbsrv_samba3_list_users(s3call);
125
126         case WINBINDD_GETPWNAM:
127                 return wbsrv_samba3_getpwnam(s3call);
128
129         case WINBINDD_GETPWUID:
130                 return wbsrv_samba3_getpwuid(s3call);
131
132         case WINBINDD_SETPWENT:
133                 return wbsrv_samba3_setpwent(s3call);
134
135         case WINBINDD_GETPWENT:
136                 return wbsrv_samba3_getpwent(s3call);
137
138         case WINBINDD_ENDPWENT:
139                 return wbsrv_samba3_endpwent(s3call);
140
141         case WINBINDD_GETGRNAM:
142                 return wbsrv_samba3_getgrnam(s3call);
143
144         case WINBINDD_GETGRGID:
145                 return wbsrv_samba3_getgrgid(s3call);
146
147         case WINBINDD_GETGROUPS:
148                 return wbsrv_samba3_getgroups(s3call);
149
150         case WINBINDD_SETGRENT:
151                 return wbsrv_samba3_setgrent(s3call);
152
153         case WINBINDD_GETGRENT:
154                 return wbsrv_samba3_getgrent(s3call);
155
156         case WINBINDD_ENDGRENT:
157                 return wbsrv_samba3_endgrent(s3call);
158
159         case WINBINDD_SID_TO_UID:
160         case WINBINDD_DUAL_SID2UID:
161                 return wbsrv_samba3_sid2uid(s3call);
162
163         case WINBINDD_SID_TO_GID:
164         case WINBINDD_DUAL_SID2GID:
165                 return wbsrv_samba3_sid2gid(s3call);
166
167         case WINBINDD_UID_TO_SID:
168         case WINBINDD_DUAL_UID2SID:
169                 return wbsrv_samba3_uid2sid(s3call);
170
171         case WINBINDD_GID_TO_SID:
172         case WINBINDD_DUAL_GID2SID:
173                 return wbsrv_samba3_gid2sid(s3call);
174
175         case WINBINDD_DOMAIN_INFO:
176                 return wbsrv_samba3_domain_info(s3call);
177
178         case WINBINDD_PAM_LOGOFF:
179                 return wbsrv_samba3_pam_logoff(s3call);
180
181         /* Unimplemented commands */
182         case WINBINDD_GETPWSID:
183         case WINBINDD_PAM_CHAUTHTOK:
184         case WINBINDD_PAM_CHNG_PSWD_AUTH_CRAP:
185         case WINBINDD_LOOKUPRIDS:
186         case WINBINDD_LOOKUPSIDS:
187         case WINBINDD_SIDS_TO_XIDS:
188         case WINBINDD_ALLOCATE_UID:
189         case WINBINDD_ALLOCATE_GID:
190         case WINBINDD_SHOW_SEQUENCE:
191         case WINBINDD_WINS_BYIP:
192         case WINBINDD_WINS_BYNAME:
193         case WINBINDD_GETGRLST:
194         case WINBINDD_GETSIDALIASES:
195         case WINBINDD_DSGETDCNAME:
196         case WINBINDD_INIT_CONNECTION:
197         case WINBINDD_DUAL_SIDS2XIDS:
198         case WINBINDD_DUAL_USERINFO:
199         case WINBINDD_DUAL_GETSIDALIASES:
200         case WINBINDD_DUAL_NDRCMD:
201         case WINBINDD_CCACHE_NTLMAUTH:
202         case WINBINDD_NUM_CMDS:
203         case WINBINDD_CHANGE_MACHACC:
204         case WINBINDD_PING_DC:
205         case WINBINDD_DC_INFO:
206         case WINBINDD_CCACHE_SAVE:
207                 DEBUG(10, ("Unimplemented winbind samba3 request %d\n",
208                            s3call->request->cmd));
209                 break;
210         }
211
212         s3call->response->result = WINBINDD_ERROR;
213         return NT_STATUS_OK;
214 }
215
216 static NTSTATUS wbsrv_samba3_push_reply(struct wbsrv_samba3_call *call)
217 {
218         uint8_t *extra_data;
219         size_t extra_data_len = 0;
220
221         extra_data = (uint8_t *)call->response->extra_data.data;
222         if (extra_data != NULL) {
223                 extra_data_len = call->response->length -
224                         sizeof(*call->response);
225         }
226
227         call->out = data_blob_talloc(call, NULL, call->response->length);
228         NT_STATUS_HAVE_NO_MEMORY(call->out.data);
229
230         /* don't push real pointer values into sockets */
231         if (extra_data) {
232                 call->response->extra_data.data = (void *)0xFFFFFFFF;
233         }
234
235         memcpy(call->out.data, call->response, sizeof(*call->response));
236         /* set back the pointer */
237         call->response->extra_data.data = extra_data;
238
239         if (extra_data) {
240                 memcpy(call->out.data + sizeof(*call->response),
241                        extra_data,
242                        extra_data_len);
243         }
244
245         return NT_STATUS_OK;
246 }
247
248 static void wbsrv_samba3_send_reply_done(struct tevent_req *subreq);
249
250 /*
251  * queue a wbsrv_call reply on a wbsrv_connection
252  * NOTE: that this implies talloc_free(call),
253  *       use talloc_reference(call) if you need it after
254  *       calling wbsrv_queue_reply
255  */
256 NTSTATUS wbsrv_samba3_send_reply(struct wbsrv_samba3_call *call)
257 {
258         struct wbsrv_connection *wbsrv_conn = call->wbconn;
259         struct tevent_req *subreq;
260         NTSTATUS status;
261
262         status = wbsrv_samba3_push_reply(call);
263         NT_STATUS_NOT_OK_RETURN(status);
264
265         call->out_iov[0].iov_base = (char *) call->out.data;
266         call->out_iov[0].iov_len = call->out.length;
267
268         subreq = tstream_writev_queue_send(call,
269                                            wbsrv_conn->conn->event.ctx,
270                                            wbsrv_conn->tstream,
271                                            wbsrv_conn->send_queue,
272                                            call->out_iov, 1);
273         if (subreq == NULL) {
274                 wbsrv_terminate_connection(wbsrv_conn, "wbsrv_call_loop: "
275                                 "no memory for tstream_writev_queue_send");
276                 return NT_STATUS_NO_MEMORY;
277         }
278         tevent_req_set_callback(subreq, wbsrv_samba3_send_reply_done, call);
279
280         return status;
281 }
282
283 static void wbsrv_samba3_send_reply_done(struct tevent_req *subreq)
284 {
285         struct wbsrv_samba3_call *call = tevent_req_callback_data(subreq,
286                         struct wbsrv_samba3_call);
287         int sys_errno;
288         int rc;
289
290         rc = tstream_writev_queue_recv(subreq, &sys_errno);
291         TALLOC_FREE(subreq);
292         if (rc == -1) {
293                 const char *reason;
294
295                 reason = talloc_asprintf(call, "wbsrv_samba3_send_reply_done: "
296                                          "tstream_writev_queue_recv() - %d:%s",
297                                          sys_errno, strerror(sys_errno));
298                 if (reason == NULL) {
299                         reason = "wbsrv_samba3_send_reply_done: "
300                                  "tstream_writev_queue_recv() failed";
301                 }
302
303                 wbsrv_terminate_connection(call->wbconn, reason);
304                 return;
305         }
306
307         talloc_free(call);
308 }
309
310 NTSTATUS wbsrv_samba3_process(struct wbsrv_samba3_call *call)
311 {
312         NTSTATUS status;
313
314         status = wbsrv_samba3_pull_request(call);
315         
316         if (!NT_STATUS_IS_OK(status)) {
317                 return status;
318         }
319
320         status = wbsrv_samba3_handle_call(call);
321
322         if (!NT_STATUS_IS_OK(status)) {
323                 talloc_free(call);
324                 return status;
325         }
326
327         if (call->flags & WBSRV_CALL_FLAGS_REPLY_ASYNC) {
328                 return NT_STATUS_OK;
329         }
330
331         status = wbsrv_samba3_send_reply(call);
332         return status;
333 }
334