1 | /*
|
---|
2 | Unix SMB/CIFS implementation.
|
---|
3 | client change notify operations
|
---|
4 | Copyright (C) Andrew Tridgell 2003
|
---|
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 | #include "libcli/raw/libcliraw.h"
|
---|
22 | #include "libcli/raw/raw_proto.h"
|
---|
23 | #include "../lib/util/dlinklist.h"
|
---|
24 |
|
---|
25 | /****************************************************************************
|
---|
26 | change notify (async send)
|
---|
27 | ****************************************************************************/
|
---|
28 | _PUBLIC_ struct smbcli_request *smb_raw_changenotify_send(struct smbcli_tree *tree, union smb_notify *parms)
|
---|
29 | {
|
---|
30 | struct smb_nttrans nt;
|
---|
31 | uint8_t setup[8];
|
---|
32 |
|
---|
33 | if (parms->nttrans.level != RAW_NOTIFY_NTTRANS) {
|
---|
34 | return NULL;
|
---|
35 | }
|
---|
36 |
|
---|
37 | nt.in.max_setup = 0;
|
---|
38 | nt.in.max_param = parms->nttrans.in.buffer_size;
|
---|
39 | nt.in.max_data = 0;
|
---|
40 | nt.in.setup_count = 4;
|
---|
41 | nt.in.setup = setup;
|
---|
42 | SIVAL(setup, 0, parms->nttrans.in.completion_filter);
|
---|
43 | SSVAL(setup, 4, parms->nttrans.in.file.fnum);
|
---|
44 | SSVAL(setup, 6, parms->nttrans.in.recursive);
|
---|
45 | nt.in.function = NT_TRANSACT_NOTIFY_CHANGE;
|
---|
46 | nt.in.params = data_blob(NULL, 0);
|
---|
47 | nt.in.data = data_blob(NULL, 0);
|
---|
48 |
|
---|
49 | return smb_raw_nttrans_send(tree, &nt);
|
---|
50 | }
|
---|
51 |
|
---|
52 | /****************************************************************************
|
---|
53 | change notify (async recv)
|
---|
54 | ****************************************************************************/
|
---|
55 | _PUBLIC_ NTSTATUS smb_raw_changenotify_recv(struct smbcli_request *req,
|
---|
56 | TALLOC_CTX *mem_ctx, union smb_notify *parms)
|
---|
57 | {
|
---|
58 | struct smb_nttrans nt;
|
---|
59 | NTSTATUS status;
|
---|
60 | uint32_t ofs, i;
|
---|
61 | struct smbcli_session *session = req?req->session:NULL;
|
---|
62 |
|
---|
63 | if (parms->nttrans.level != RAW_NOTIFY_NTTRANS) {
|
---|
64 | return NT_STATUS_INVALID_LEVEL;
|
---|
65 | }
|
---|
66 |
|
---|
67 | status = smb_raw_nttrans_recv(req, mem_ctx, &nt);
|
---|
68 | if (!NT_STATUS_IS_OK(status)) {
|
---|
69 | return status;
|
---|
70 | }
|
---|
71 |
|
---|
72 | parms->nttrans.out.changes = NULL;
|
---|
73 | parms->nttrans.out.num_changes = 0;
|
---|
74 |
|
---|
75 | /* count them */
|
---|
76 | for (ofs=0; nt.out.params.length - ofs > 12; ) {
|
---|
77 | uint32_t next = IVAL(nt.out.params.data, ofs);
|
---|
78 | parms->nttrans.out.num_changes++;
|
---|
79 | if (next == 0 ||
|
---|
80 | ofs + next >= nt.out.params.length) break;
|
---|
81 | ofs += next;
|
---|
82 | }
|
---|
83 |
|
---|
84 | /* allocate array */
|
---|
85 | parms->nttrans.out.changes = talloc_array(mem_ctx, struct notify_changes, parms->nttrans.out.num_changes);
|
---|
86 | if (!parms->nttrans.out.changes) {
|
---|
87 | return NT_STATUS_NO_MEMORY;
|
---|
88 | }
|
---|
89 |
|
---|
90 | for (i=ofs=0; i<parms->nttrans.out.num_changes; i++) {
|
---|
91 | parms->nttrans.out.changes[i].action = IVAL(nt.out.params.data, ofs+4);
|
---|
92 | smbcli_blob_pull_string(session, mem_ctx, &nt.out.params,
|
---|
93 | &parms->nttrans.out.changes[i].name,
|
---|
94 | ofs+8, ofs+12, STR_UNICODE);
|
---|
95 | ofs += IVAL(nt.out.params.data, ofs);
|
---|
96 | }
|
---|
97 |
|
---|
98 | return NT_STATUS_OK;
|
---|
99 | }
|
---|
100 |
|
---|
101 | /****************************************************************************
|
---|
102 | handle ntcancel replies from the server,
|
---|
103 | as the MID of the real reply and the ntcancel reply is the same
|
---|
104 | we need to do find out to what request the reply belongs
|
---|
105 | ****************************************************************************/
|
---|
106 | struct smbcli_request *smbcli_handle_ntcancel_reply(struct smbcli_request *req,
|
---|
107 | size_t len, const uint8_t *hdr)
|
---|
108 | {
|
---|
109 | struct smbcli_request *ntcancel;
|
---|
110 |
|
---|
111 | if (!req) return req;
|
---|
112 |
|
---|
113 | if (!req->ntcancel) return req;
|
---|
114 |
|
---|
115 | if (len >= MIN_SMB_SIZE + NBT_HDR_SIZE &&
|
---|
116 | (CVAL(hdr, HDR_FLG) & FLAG_REPLY) &&
|
---|
117 | CVAL(hdr,HDR_COM) == SMBntcancel) {
|
---|
118 | ntcancel = req->ntcancel;
|
---|
119 | DLIST_REMOVE(req->ntcancel, ntcancel);
|
---|
120 |
|
---|
121 | /*
|
---|
122 | * TODO: untill we understand how the
|
---|
123 | * smb_signing works for this case we
|
---|
124 | * return NULL, to just ignore the packet
|
---|
125 | */
|
---|
126 | /*return ntcancel;*/
|
---|
127 | return NULL;
|
---|
128 | }
|
---|
129 |
|
---|
130 | return req;
|
---|
131 | }
|
---|
132 |
|
---|
133 | /****************************************************************************
|
---|
134 | Send a NT Cancel request - used to hurry along a pending request. Usually
|
---|
135 | used to cancel a pending change notify request
|
---|
136 | note that this request does not expect a response!
|
---|
137 | ****************************************************************************/
|
---|
138 | NTSTATUS smb_raw_ntcancel(struct smbcli_request *oldreq)
|
---|
139 | {
|
---|
140 | struct smbcli_request *req;
|
---|
141 |
|
---|
142 | req = smbcli_request_setup_transport(oldreq->transport, SMBntcancel, 0, 0);
|
---|
143 |
|
---|
144 | SSVAL(req->out.hdr, HDR_MID, SVAL(oldreq->out.hdr, HDR_MID));
|
---|
145 | SSVAL(req->out.hdr, HDR_PID, SVAL(oldreq->out.hdr, HDR_PID));
|
---|
146 | SSVAL(req->out.hdr, HDR_TID, SVAL(oldreq->out.hdr, HDR_TID));
|
---|
147 | SSVAL(req->out.hdr, HDR_UID, SVAL(oldreq->out.hdr, HDR_UID));
|
---|
148 |
|
---|
149 | /* this request does not expect a reply, so tell the signing
|
---|
150 | subsystem not to allocate an id for a reply */
|
---|
151 | req->sign_single_increment = 1;
|
---|
152 | req->one_way_request = 1;
|
---|
153 |
|
---|
154 | /*
|
---|
155 | * smbcli_request_send() free's oneway requests
|
---|
156 | * but we want to keep it under oldreq->ntcancel
|
---|
157 | */
|
---|
158 | if (!talloc_reference(oldreq, req)) {
|
---|
159 | talloc_free(req);
|
---|
160 | return NT_STATUS_NO_MEMORY;
|
---|
161 | }
|
---|
162 |
|
---|
163 | smbcli_request_send(req);
|
---|
164 |
|
---|
165 | DLIST_ADD_END(oldreq->ntcancel, req, struct smbcli_request *);
|
---|
166 |
|
---|
167 | return NT_STATUS_OK;
|
---|
168 | }
|
---|