1 | /*
|
---|
2 | Unix SMB/CIFS implementation.
|
---|
3 | client transaction calls
|
---|
4 | Copyright (C) Andrew Tridgell 1994-1998
|
---|
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 "libsmb/libsmb.h"
|
---|
22 | #include "../lib/util/tevent_ntstatus.h"
|
---|
23 | #include "async_smb.h"
|
---|
24 | #include "../libcli/smb/smbXcli_base.h"
|
---|
25 |
|
---|
26 | struct cli_trans_state {
|
---|
27 | struct cli_state *cli;
|
---|
28 | struct tevent_req *req;
|
---|
29 | struct cli_trans_state **ptr;
|
---|
30 | };
|
---|
31 |
|
---|
32 | static int cli_trans_state_destructor(struct cli_trans_state *state)
|
---|
33 | {
|
---|
34 | talloc_set_destructor(state->ptr, NULL);
|
---|
35 | talloc_free(state->ptr);
|
---|
36 | return 0;
|
---|
37 | }
|
---|
38 |
|
---|
39 | static int cli_trans_state_ptr_destructor(struct cli_trans_state **ptr)
|
---|
40 | {
|
---|
41 | struct cli_trans_state *state = *ptr;
|
---|
42 | void *parent = talloc_parent(state);
|
---|
43 |
|
---|
44 | talloc_set_destructor(state, NULL);
|
---|
45 |
|
---|
46 | talloc_reparent(state, parent, state->req);
|
---|
47 | talloc_free(state);
|
---|
48 | return 0;
|
---|
49 | }
|
---|
50 |
|
---|
51 | struct tevent_req *cli_trans_send(
|
---|
52 | TALLOC_CTX *mem_ctx, struct tevent_context *ev,
|
---|
53 | struct cli_state *cli, uint8_t cmd,
|
---|
54 | const char *pipe_name, uint16_t fid, uint16_t function, int flags,
|
---|
55 | uint16_t *setup, uint8_t num_setup, uint8_t max_setup,
|
---|
56 | uint8_t *param, uint32_t num_param, uint32_t max_param,
|
---|
57 | uint8_t *data, uint32_t num_data, uint32_t max_data)
|
---|
58 | {
|
---|
59 | struct cli_trans_state *state;
|
---|
60 | uint8_t additional_flags = 0;
|
---|
61 | uint8_t clear_flags = 0;
|
---|
62 | uint16_t additional_flags2 = 0;
|
---|
63 | uint16_t clear_flags2 = 0;
|
---|
64 |
|
---|
65 | state = talloc_zero(mem_ctx, struct cli_trans_state);
|
---|
66 | if (state == NULL) {
|
---|
67 | return NULL;
|
---|
68 | }
|
---|
69 | state->cli = cli;
|
---|
70 | state->ptr = talloc(state, struct cli_trans_state *);
|
---|
71 | if (state->ptr == NULL) {
|
---|
72 | talloc_free(state);
|
---|
73 | return NULL;
|
---|
74 | }
|
---|
75 | *state->ptr = state;
|
---|
76 |
|
---|
77 | state->req = smb1cli_trans_send(state, ev,
|
---|
78 | cli->conn, cmd,
|
---|
79 | additional_flags, clear_flags,
|
---|
80 | additional_flags2, clear_flags2,
|
---|
81 | cli->timeout,
|
---|
82 | cli->smb1.pid,
|
---|
83 | cli->smb1.tcon,
|
---|
84 | cli->smb1.session,
|
---|
85 | pipe_name, fid, function, flags,
|
---|
86 | setup, num_setup, max_setup,
|
---|
87 | param, num_param, max_param,
|
---|
88 | data, num_data, max_data);
|
---|
89 | if (state->req == NULL) {
|
---|
90 | talloc_free(state);
|
---|
91 | return NULL;
|
---|
92 | }
|
---|
93 |
|
---|
94 | talloc_reparent(state, state->req, state->ptr);
|
---|
95 | talloc_set_destructor(state, cli_trans_state_destructor);
|
---|
96 | talloc_set_destructor(state->ptr, cli_trans_state_ptr_destructor);
|
---|
97 |
|
---|
98 | return state->req;
|
---|
99 | }
|
---|
100 |
|
---|
101 | NTSTATUS cli_trans_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
|
---|
102 | uint16_t *recv_flags2,
|
---|
103 | uint16_t **setup, uint8_t min_setup,
|
---|
104 | uint8_t *num_setup,
|
---|
105 | uint8_t **param, uint32_t min_param,
|
---|
106 | uint32_t *num_param,
|
---|
107 | uint8_t **data, uint32_t min_data,
|
---|
108 | uint32_t *num_data)
|
---|
109 | {
|
---|
110 | NTSTATUS status;
|
---|
111 | void *parent = talloc_parent(req);
|
---|
112 | struct cli_trans_state *state =
|
---|
113 | talloc_get_type(parent,
|
---|
114 | struct cli_trans_state);
|
---|
115 | bool map_dos_errors = true;
|
---|
116 |
|
---|
117 | status = smb1cli_trans_recv(req, mem_ctx, recv_flags2,
|
---|
118 | setup, min_setup, num_setup,
|
---|
119 | param, min_param, num_param,
|
---|
120 | data, min_data, num_data);
|
---|
121 |
|
---|
122 | if (state) {
|
---|
123 | map_dos_errors = state->cli->map_dos_errors;
|
---|
124 | state->cli->raw_status = status;
|
---|
125 | talloc_free(state->ptr);
|
---|
126 | state = NULL;
|
---|
127 | }
|
---|
128 |
|
---|
129 | if (NT_STATUS_IS_DOS(status) && map_dos_errors) {
|
---|
130 | uint8_t eclass = NT_STATUS_DOS_CLASS(status);
|
---|
131 | uint16_t ecode = NT_STATUS_DOS_CODE(status);
|
---|
132 | /*
|
---|
133 | * TODO: is it really a good idea to do a mapping here?
|
---|
134 | *
|
---|
135 | * The old cli_pull_error() also does it, so I do not change
|
---|
136 | * the behavior yet.
|
---|
137 | */
|
---|
138 | status = dos_to_ntstatus(eclass, ecode);
|
---|
139 | }
|
---|
140 |
|
---|
141 | return status;
|
---|
142 | }
|
---|
143 |
|
---|
144 | NTSTATUS cli_trans(TALLOC_CTX *mem_ctx, struct cli_state *cli,
|
---|
145 | uint8_t trans_cmd,
|
---|
146 | const char *pipe_name, uint16_t fid, uint16_t function,
|
---|
147 | int flags,
|
---|
148 | uint16_t *setup, uint8_t num_setup, uint8_t max_setup,
|
---|
149 | uint8_t *param, uint32_t num_param, uint32_t max_param,
|
---|
150 | uint8_t *data, uint32_t num_data, uint32_t max_data,
|
---|
151 | uint16_t *recv_flags2,
|
---|
152 | uint16_t **rsetup, uint8_t min_rsetup, uint8_t *num_rsetup,
|
---|
153 | uint8_t **rparam, uint32_t min_rparam, uint32_t *num_rparam,
|
---|
154 | uint8_t **rdata, uint32_t min_rdata, uint32_t *num_rdata)
|
---|
155 | {
|
---|
156 | NTSTATUS status;
|
---|
157 | uint8_t additional_flags = 0;
|
---|
158 | uint8_t clear_flags = 0;
|
---|
159 | uint16_t additional_flags2 = 0;
|
---|
160 | uint16_t clear_flags2 = 0;
|
---|
161 |
|
---|
162 | status = smb1cli_trans(mem_ctx,
|
---|
163 | cli->conn, trans_cmd,
|
---|
164 | additional_flags, clear_flags,
|
---|
165 | additional_flags2, clear_flags2,
|
---|
166 | cli->timeout,
|
---|
167 | cli->smb1.pid,
|
---|
168 | cli->smb1.tcon,
|
---|
169 | cli->smb1.session,
|
---|
170 | pipe_name, fid, function, flags,
|
---|
171 | setup, num_setup, max_setup,
|
---|
172 | param, num_param, max_param,
|
---|
173 | data, num_data, max_data,
|
---|
174 | recv_flags2,
|
---|
175 | rsetup, min_rsetup, num_rsetup,
|
---|
176 | rparam, min_rparam, num_rparam,
|
---|
177 | rdata, min_rdata, num_rdata);
|
---|
178 |
|
---|
179 | cli->raw_status = status;
|
---|
180 |
|
---|
181 | if (NT_STATUS_IS_DOS(status) && cli->map_dos_errors) {
|
---|
182 | uint8_t eclass = NT_STATUS_DOS_CLASS(status);
|
---|
183 | uint16_t ecode = NT_STATUS_DOS_CODE(status);
|
---|
184 | /*
|
---|
185 | * TODO: is it really a good idea to do a mapping here?
|
---|
186 | *
|
---|
187 | * The old cli_pull_error() also does it, so I do not change
|
---|
188 | * the behavior yet.
|
---|
189 | */
|
---|
190 | status = dos_to_ntstatus(eclass, ecode);
|
---|
191 | }
|
---|
192 |
|
---|
193 | return status;
|
---|
194 | }
|
---|