Line data Source code
1 : /*
2 : Unix SMB/CIFS implementation.
3 : Manage smbsrv_tcon structures
4 : Copyright (C) Andrew Tridgell 1998
5 : Copyright (C) Alexander Bokovoy 2002
6 : Copyright (C) Stefan Metzmacher 2005-2006
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 "smb_server/smb_server.h"
24 : #include "samba/service_stream.h"
25 : #include "lib/tsocket/tsocket.h"
26 : #include "ntvfs/ntvfs.h"
27 :
28 : /****************************************************************************
29 : init the tcon structures
30 : ****************************************************************************/
31 4004 : static NTSTATUS smbsrv_init_tcons(struct smbsrv_tcons_context *tcons_ctx, TALLOC_CTX *mem_ctx, uint32_t limit)
32 : {
33 : /*
34 : * the idr_* functions take 'int' as limit,
35 : * and only work with a max limit 0x00FFFFFF
36 : */
37 4004 : limit &= 0x00FFFFFF;
38 :
39 4004 : tcons_ctx->idtree_tid = idr_init(mem_ctx);
40 4004 : NT_STATUS_HAVE_NO_MEMORY(tcons_ctx->idtree_tid);
41 4004 : tcons_ctx->idtree_limit = limit;
42 4004 : tcons_ctx->list = NULL;
43 :
44 4004 : return NT_STATUS_OK;
45 : }
46 :
47 2253 : NTSTATUS smbsrv_smb_init_tcons(struct smbsrv_connection *smb_conn)
48 : {
49 2253 : return smbsrv_init_tcons(&smb_conn->smb_tcons, smb_conn, UINT16_MAX);
50 : }
51 :
52 1751 : NTSTATUS smbsrv_smb2_init_tcons(struct smbsrv_session *smb_sess)
53 : {
54 1751 : return smbsrv_init_tcons(&smb_sess->smb2_tcons, smb_sess, UINT32_MAX);
55 : }
56 :
57 : /****************************************************************************
58 : find a tcon given a tid for SMB
59 : ****************************************************************************/
60 829237 : static struct smbsrv_tcon *smbsrv_tcon_find(struct smbsrv_tcons_context *tcons_ctx,
61 : uint32_t tid, struct timeval request_time)
62 : {
63 : void *p;
64 : struct smbsrv_tcon *tcon;
65 :
66 829237 : if (tid == 0) return NULL;
67 :
68 822620 : if (tid > tcons_ctx->idtree_limit) return NULL;
69 :
70 822620 : p = idr_find(tcons_ctx->idtree_tid, tid);
71 822620 : if (!p) return NULL;
72 :
73 822517 : tcon = talloc_get_type(p, struct smbsrv_tcon);
74 822517 : if (!tcon) return NULL;
75 :
76 822517 : tcon->statistics.last_request_time = request_time;
77 :
78 822517 : return tcon;
79 : }
80 :
81 423436 : struct smbsrv_tcon *smbsrv_smb_tcon_find(struct smbsrv_connection *smb_conn,
82 : uint32_t tid, struct timeval request_time)
83 : {
84 423436 : return smbsrv_tcon_find(&smb_conn->smb_tcons, tid, request_time);
85 : }
86 :
87 408140 : struct smbsrv_tcon *smbsrv_smb2_tcon_find(struct smbsrv_session *smb_sess,
88 : uint32_t tid, struct timeval request_time)
89 : {
90 408140 : if (!smb_sess) return NULL;
91 405801 : return smbsrv_tcon_find(&smb_sess->smb2_tcons, tid, request_time);
92 : }
93 :
94 : /*
95 : destroy a connection structure
96 : */
97 2622 : static int smbsrv_tcon_destructor(struct smbsrv_tcon *tcon)
98 : {
99 : struct smbsrv_tcons_context *tcons_ctx;
100 : struct tsocket_address *client_addr;
101 :
102 2622 : client_addr = tcon->smb_conn->connection->remote_address;
103 :
104 2622 : DEBUG(3,("%s closed connection to service %s\n",
105 : tsocket_address_string(client_addr, tcon),
106 : tcon->share_name));
107 :
108 : /* tell the ntvfs backend that we are disconnecting */
109 2622 : if (tcon->ntvfs) {
110 2622 : ntvfs_disconnect(tcon->ntvfs);
111 2622 : tcon->ntvfs = NULL;
112 : }
113 :
114 2622 : if (tcon->smb2.session) {
115 1645 : tcons_ctx = &tcon->smb2.session->smb2_tcons;
116 : } else {
117 977 : tcons_ctx = &tcon->smb_conn->smb_tcons;
118 : }
119 :
120 2622 : idr_remove(tcons_ctx->idtree_tid, tcon->tid);
121 2622 : DLIST_REMOVE(tcons_ctx->list, tcon);
122 2622 : return 0;
123 : }
124 :
125 : /*
126 : find first available connection slot
127 : */
128 2622 : static struct smbsrv_tcon *smbsrv_tcon_new(struct smbsrv_connection *smb_conn,
129 : struct smbsrv_session *smb_sess,
130 : const char *share_name)
131 : {
132 : TALLOC_CTX *mem_ctx;
133 : struct smbsrv_tcons_context *tcons_ctx;
134 : uint32_t handle_uint_max;
135 : struct smbsrv_tcon *tcon;
136 : NTSTATUS status;
137 : int i;
138 :
139 2622 : if (smb_sess) {
140 1645 : mem_ctx = smb_sess;
141 1645 : tcons_ctx = &smb_sess->smb2_tcons;
142 1645 : handle_uint_max = UINT32_MAX;
143 : } else {
144 977 : mem_ctx = smb_conn;
145 977 : tcons_ctx = &smb_conn->smb_tcons;
146 977 : handle_uint_max = UINT16_MAX;
147 : }
148 :
149 2622 : tcon = talloc_zero(mem_ctx, struct smbsrv_tcon);
150 2622 : if (!tcon) return NULL;
151 2622 : tcon->smb_conn = smb_conn;
152 2622 : tcon->smb2.session = smb_sess;
153 2622 : tcon->share_name = talloc_strdup(tcon, share_name);
154 2622 : if (!tcon->share_name) goto failed;
155 :
156 : /*
157 : * the use -1 here, because we don't want to give away the wildcard
158 : * fnum used in SMBflush
159 : */
160 2622 : status = smbsrv_init_handles(tcon, handle_uint_max - 1);
161 2622 : if (!NT_STATUS_IS_OK(status)) {
162 0 : DEBUG(1,("ERROR! failed to init handles: %s\n", nt_errstr(status)));
163 0 : goto failed;
164 : }
165 :
166 2622 : i = idr_get_new_random(tcons_ctx->idtree_tid, tcon, tcons_ctx->idtree_limit);
167 2622 : if (i == -1) {
168 0 : DEBUG(1,("ERROR! Out of connection structures\n"));
169 0 : goto failed;
170 : }
171 2622 : tcon->tid = i;
172 :
173 2622 : DLIST_ADD(tcons_ctx->list, tcon);
174 2622 : talloc_set_destructor(tcon, smbsrv_tcon_destructor);
175 :
176 : /* now fill in some statistics */
177 2622 : tcon->statistics.connect_time = timeval_current();
178 :
179 2622 : return tcon;
180 :
181 0 : failed:
182 0 : talloc_free(tcon);
183 0 : return NULL;
184 : }
185 :
186 977 : struct smbsrv_tcon *smbsrv_smb_tcon_new(struct smbsrv_connection *smb_conn, const char *share_name)
187 : {
188 977 : return smbsrv_tcon_new(smb_conn, NULL, share_name);
189 : }
190 :
191 1645 : struct smbsrv_tcon *smbsrv_smb2_tcon_new(struct smbsrv_session *smb_sess, const char *share_name)
192 : {
193 1645 : return smbsrv_tcon_new(smb_sess->smb_conn, smb_sess, share_name);
194 : }
|
