1 | /*
|
---|
2 | Unix SMB/CIFS implementation.
|
---|
3 | SMB2 Find
|
---|
4 | Copyright (C) Andrew Tridgell 2003
|
---|
5 | Copyright (c) Stefan Metzmacher 2006
|
---|
6 |
|
---|
7 | This program is free software; you can redistribute it and/or modify
|
---|
8 | it under the terms of the GNU General Public License as published by
|
---|
9 | the Free Software Foundation; either version 3 of the License, or
|
---|
10 | (at your option) any later version.
|
---|
11 |
|
---|
12 | This program is distributed in the hope that it will be useful,
|
---|
13 | but WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
14 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
---|
15 | GNU General Public License for more details.
|
---|
16 |
|
---|
17 | You should have received a copy of the GNU General Public License
|
---|
18 | along with this program. If not, see <http://www.gnu.org/licenses/>.
|
---|
19 | */
|
---|
20 | /*
|
---|
21 | This file handles the parsing of transact2 requests
|
---|
22 | */
|
---|
23 |
|
---|
24 | #include "includes.h"
|
---|
25 | #include "libcli/smb2/smb2.h"
|
---|
26 | #include "libcli/smb2/smb2_calls.h"
|
---|
27 | #include "smb_server/smb_server.h"
|
---|
28 | #include "smb_server/smb2/smb2_server.h"
|
---|
29 | #include "ntvfs/ntvfs.h"
|
---|
30 |
|
---|
31 |
|
---|
32 | /* a structure to encapsulate the state information about an in-progress ffirst/fnext operation */
|
---|
33 | struct smb2srv_find_state {
|
---|
34 | struct smb2srv_request *req;
|
---|
35 | struct smb2_find *info;
|
---|
36 | union smb_search_first *ff;
|
---|
37 | union smb_search_next *fn;
|
---|
38 | uint32_t last_entry_offset;
|
---|
39 | };
|
---|
40 |
|
---|
41 | /* callback function for SMB2 Find */
|
---|
42 | static bool smb2srv_find_callback(void *private_data, const union smb_search_data *file)
|
---|
43 | {
|
---|
44 | struct smb2srv_find_state *state = talloc_get_type(private_data, struct smb2srv_find_state);
|
---|
45 | struct smb2_find *info = state->info;
|
---|
46 | uint32_t old_length;
|
---|
47 | NTSTATUS status;
|
---|
48 |
|
---|
49 | old_length = info->out.blob.length;
|
---|
50 |
|
---|
51 | status = smbsrv_push_passthru_search(state, &info->out.blob, info->data_level, file, STR_UNICODE);
|
---|
52 | if (!NT_STATUS_IS_OK(status) ||
|
---|
53 | info->out.blob.length > info->in.max_response_size) {
|
---|
54 | /* restore the old length and tell the backend to stop */
|
---|
55 | smbsrv_blob_grow_data(state, &info->out.blob, old_length);
|
---|
56 | return false;
|
---|
57 | }
|
---|
58 |
|
---|
59 | state->last_entry_offset = old_length;
|
---|
60 |
|
---|
61 | return true;
|
---|
62 | }
|
---|
63 |
|
---|
64 | static void smb2srv_find_send(struct ntvfs_request *ntvfs)
|
---|
65 | {
|
---|
66 | struct smb2srv_request *req;
|
---|
67 | struct smb2srv_find_state *state;
|
---|
68 |
|
---|
69 | SMB2SRV_CHECK_ASYNC_STATUS(state, struct smb2srv_find_state);
|
---|
70 | SMB2SRV_CHECK(smb2srv_setup_reply(req, 0x08, true, state->info->out.blob.length));
|
---|
71 |
|
---|
72 | if (state->info->out.blob.length > 0) {
|
---|
73 | SIVAL(state->info->out.blob.data + state->last_entry_offset, 0, 0);
|
---|
74 | }
|
---|
75 |
|
---|
76 | SMB2SRV_CHECK(smb2_push_o16s32_blob(&req->out, 0x02, state->info->out.blob));
|
---|
77 |
|
---|
78 | smb2srv_send_reply(req);
|
---|
79 | }
|
---|
80 |
|
---|
81 | static NTSTATUS smb2srv_find_backend(struct smb2srv_find_state *state)
|
---|
82 | {
|
---|
83 | struct smb2_find *info = state->info;
|
---|
84 |
|
---|
85 | switch (info->in.level) {
|
---|
86 | case SMB2_FIND_DIRECTORY_INFO:
|
---|
87 | info->data_level = RAW_SEARCH_DATA_DIRECTORY_INFO;
|
---|
88 | break;
|
---|
89 |
|
---|
90 | case SMB2_FIND_FULL_DIRECTORY_INFO:
|
---|
91 | info->data_level = RAW_SEARCH_DATA_FULL_DIRECTORY_INFO;
|
---|
92 | break;
|
---|
93 |
|
---|
94 | case SMB2_FIND_BOTH_DIRECTORY_INFO:
|
---|
95 | info->data_level = RAW_SEARCH_DATA_BOTH_DIRECTORY_INFO;
|
---|
96 | break;
|
---|
97 |
|
---|
98 | case SMB2_FIND_NAME_INFO:
|
---|
99 | info->data_level = RAW_SEARCH_DATA_NAME_INFO;
|
---|
100 | break;
|
---|
101 |
|
---|
102 | case SMB2_FIND_ID_BOTH_DIRECTORY_INFO:
|
---|
103 | info->data_level = RAW_SEARCH_DATA_ID_BOTH_DIRECTORY_INFO;
|
---|
104 | break;
|
---|
105 |
|
---|
106 | case SMB2_FIND_ID_FULL_DIRECTORY_INFO:
|
---|
107 | info->data_level = RAW_SEARCH_DATA_ID_FULL_DIRECTORY_INFO;
|
---|
108 | break;
|
---|
109 |
|
---|
110 | default:
|
---|
111 | return NT_STATUS_FOOBAR;
|
---|
112 | }
|
---|
113 |
|
---|
114 | if (info->in.continue_flags & SMB2_CONTINUE_FLAG_REOPEN) {
|
---|
115 | state->ff = talloc(state, union smb_search_first);
|
---|
116 | NT_STATUS_HAVE_NO_MEMORY(state->ff);
|
---|
117 |
|
---|
118 | state->ff->smb2 = *info;
|
---|
119 | state->info = &state->ff->smb2;
|
---|
120 | ZERO_STRUCT(state->ff->smb2.out);
|
---|
121 |
|
---|
122 | return ntvfs_search_first(state->req->ntvfs, state->ff, state, smb2srv_find_callback);
|
---|
123 | } else {
|
---|
124 | state->fn = talloc(state, union smb_search_next);
|
---|
125 | NT_STATUS_HAVE_NO_MEMORY(state->fn);
|
---|
126 |
|
---|
127 | state->fn->smb2 = *info;
|
---|
128 | state->info = &state->fn->smb2;
|
---|
129 | ZERO_STRUCT(state->fn->smb2.out);
|
---|
130 |
|
---|
131 | return ntvfs_search_next(state->req->ntvfs, state->fn, state, smb2srv_find_callback);
|
---|
132 | }
|
---|
133 |
|
---|
134 | /* should not be reached */
|
---|
135 | return NT_STATUS_INTERNAL_ERROR;
|
---|
136 | }
|
---|
137 |
|
---|
138 | void smb2srv_find_recv(struct smb2srv_request *req)
|
---|
139 | {
|
---|
140 | struct smb2srv_find_state *state;
|
---|
141 | struct smb2_find *info;
|
---|
142 |
|
---|
143 | SMB2SRV_CHECK_BODY_SIZE(req, 0x20, true);
|
---|
144 | SMB2SRV_TALLOC_IO_PTR(info, struct smb2_find);
|
---|
145 | /* this overwrites req->io_ptr !*/
|
---|
146 | SMB2SRV_TALLOC_IO_PTR(state, struct smb2srv_find_state);
|
---|
147 | state->req = req;
|
---|
148 | state->info = info;
|
---|
149 | state->ff = NULL;
|
---|
150 | state->fn = NULL;
|
---|
151 | state->last_entry_offset= 0;
|
---|
152 | SMB2SRV_SETUP_NTVFS_REQUEST(smb2srv_find_send, NTVFS_ASYNC_STATE_MAY_ASYNC);
|
---|
153 |
|
---|
154 | info->level = RAW_SEARCH_SMB2;
|
---|
155 | info->data_level = RAW_SEARCH_DATA_GENERIC;/* will be overwritten later */
|
---|
156 | info->in.level = CVAL(req->in.body, 0x02);
|
---|
157 | info->in.continue_flags = CVAL(req->in.body, 0x03);
|
---|
158 | info->in.file_index = IVAL(req->in.body, 0x04);
|
---|
159 | info->in.file.ntvfs = smb2srv_pull_handle(req, req->in.body, 0x08);
|
---|
160 | SMB2SRV_CHECK(smb2_pull_o16s16_string(&req->in, info, req->in.body+0x18, &info->in.pattern));
|
---|
161 | info->in.max_response_size = IVAL(req->in.body, 0x1C);
|
---|
162 |
|
---|
163 | /* the VFS backend does not yet handle NULL patterns */
|
---|
164 | if (info->in.pattern == NULL) {
|
---|
165 | info->in.pattern = "";
|
---|
166 | }
|
---|
167 |
|
---|
168 | SMB2SRV_CHECK_FILE_HANDLE(info->in.file.ntvfs);
|
---|
169 | SMB2SRV_CALL_NTVFS_BACKEND(smb2srv_find_backend(state));
|
---|
170 | }
|
---|