2 Unix SMB/CIFS implementation.
3 Manage connections_struct structures
4 Copyright (C) Andrew Tridgell 1998
5 Copyright (C) Alexander Bokovoy 2002
6 Copyright (C) Jeremy Allison 2010
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.
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.
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/>.
23 #include "smbd/smbd.h"
24 #include "smbd/globals.h"
25 #include "lib/util/bitmap.h"
27 /* The connections bitmap is expanded in increments of BITMAP_BLOCK_SZ. The
28 * maximum size of the bitmap is the largest positive integer, but you will hit
29 * the "max connections" limit, looong before that.
32 #define BITMAP_BLOCK_SZ 128
34 /****************************************************************************
35 Init the conn structures.
36 ****************************************************************************/
38 void conn_init(struct smbd_server_connection
*sconn
)
40 sconn
->smb1
.tcons
.Connections
= NULL
;
41 sconn
->smb1
.tcons
.bmap
= bitmap_talloc(sconn
, BITMAP_BLOCK_SZ
);
44 /****************************************************************************
45 Return the number of open connections.
46 ****************************************************************************/
48 int conn_num_open(struct smbd_server_connection
*sconn
)
50 return sconn
->num_tcons_open
;
53 /****************************************************************************
54 Check if a snum is in use.
55 ****************************************************************************/
57 bool conn_snum_used(struct smbd_server_connection
*sconn
,
60 if (sconn
->using_smb2
) {
62 struct smbd_smb2_session
*sess
;
63 for (sess
= sconn
->smb2
.sessions
.list
; sess
; sess
= sess
->next
) {
64 struct smbd_smb2_tcon
*ptcon
;
66 for (ptcon
= sess
->tcons
.list
; ptcon
; ptcon
= ptcon
->next
) {
67 if (ptcon
->compat_conn
&&
68 ptcon
->compat_conn
->params
&&
69 (ptcon
->compat_conn
->params
->service
= snum
)) {
76 connection_struct
*conn
;
77 for (conn
=sconn
->smb1
.tcons
.Connections
;conn
;conn
=conn
->next
) {
78 if (conn
->params
->service
== snum
) {
86 /****************************************************************************
87 Find a conn given a cnum.
88 ****************************************************************************/
90 connection_struct
*conn_find(struct smbd_server_connection
*sconn
,unsigned cnum
)
92 if (sconn
->using_smb2
) {
94 struct smbd_smb2_session
*sess
;
95 for (sess
= sconn
->smb2
.sessions
.list
; sess
; sess
= sess
->next
) {
96 struct smbd_smb2_tcon
*ptcon
;
98 for (ptcon
= sess
->tcons
.list
; ptcon
; ptcon
= ptcon
->next
) {
99 if (ptcon
->compat_conn
&&
100 ptcon
->compat_conn
->cnum
== cnum
) {
101 return ptcon
->compat_conn
;
108 connection_struct
*conn
;
109 for (conn
=sconn
->smb1
.tcons
.Connections
;conn
;conn
=conn
->next
,count
++) {
110 if (conn
->cnum
== cnum
) {
112 DLIST_PROMOTE(sconn
->smb1
.tcons
.Connections
,
123 /****************************************************************************
124 Find first available connection slot, starting from a random position.
125 The randomisation stops problems with the server dieing and clients
126 thinking the server is still available.
127 ****************************************************************************/
129 connection_struct
*conn_new(struct smbd_server_connection
*sconn
)
131 connection_struct
*conn
;
135 if (sconn
->using_smb2
) {
137 if (!(conn
=talloc_zero(NULL
, connection_struct
)) ||
138 !(conn
->params
= talloc(conn
, struct share_params
))) {
139 DEBUG(0,("TALLOC_ZERO() failed!\n"));
149 i
= bitmap_find(sconn
->smb1
.tcons
.bmap
, find_offset
);
152 /* Expand the connections bitmap. */
153 int oldsz
= sconn
->smb1
.tcons
.bmap
->n
;
154 int newsz
= sconn
->smb1
.tcons
.bmap
->n
+
156 struct bitmap
* nbmap
;
158 if (newsz
<= oldsz
) {
160 DEBUG(0,("ERROR! Out of connection structures\n"));
164 DEBUG(4,("resizing connections bitmap from %d to %d\n",
167 nbmap
= bitmap_talloc(sconn
, newsz
);
169 DEBUG(0,("ERROR! malloc fail.\n"));
173 bitmap_copy(nbmap
, sconn
->smb1
.tcons
.bmap
);
174 TALLOC_FREE(sconn
->smb1
.tcons
.bmap
);
176 sconn
->smb1
.tcons
.bmap
= nbmap
;
177 find_offset
= oldsz
; /* Start next search in the new portion. */
182 /* The bitmap position is used below as the connection number
183 * conn->cnum). This ends up as the TID field in the SMB header,
184 * which is limited to 16 bits (we skip 0xffff which is the
188 DEBUG(0, ("Maximum connection limit reached\n"));
192 if (!(conn
=talloc_zero(NULL
, connection_struct
)) ||
193 !(conn
->params
= talloc(conn
, struct share_params
))) {
194 DEBUG(0,("TALLOC_ZERO() failed!\n"));
200 conn
->force_group_gid
= (gid_t
)-1;
202 bitmap_set(sconn
->smb1
.tcons
.bmap
, i
);
204 sconn
->num_tcons_open
++;
206 string_set(&conn
->connectpath
,"");
207 string_set(&conn
->origpath
,"");
209 DLIST_ADD(sconn
->smb1
.tcons
.Connections
, conn
);
214 /****************************************************************************
215 Clear a vuid out of the connection's vuid cache
216 ****************************************************************************/
218 static void conn_clear_vuid_cache(connection_struct
*conn
, uint16_t vuid
)
222 for (i
=0; i
<VUID_CACHE_SIZE
; i
++) {
223 struct vuid_cache_entry
*ent
;
225 ent
= &conn
->vuid_cache
.array
[i
];
227 if (ent
->vuid
== vuid
) {
228 ent
->vuid
= UID_FIELD_INVALID
;
230 * We need to keep conn->session_info around
231 * if it's equal to ent->session_info as a SMBulogoff
232 * is often followed by a SMBtdis (with an invalid
233 * vuid). The debug code (or regular code in
234 * vfs_full_audit) wants to refer to the
235 * conn->session_info pointer to print debug
236 * statements. Theoretically this is a bug,
237 * as once the vuid is gone the session_info
238 * on the conn struct isn't valid any more,
239 * but there's enough code that assumes
240 * conn->session_info is never null that
241 * it's easier to hold onto the old pointer
242 * until we get a new sessionsetupX.
243 * As everything is hung off the
244 * conn pointer as a talloc context we're not
245 * leaking memory here. See bug #6315. JRA.
247 if (conn
->session_info
== ent
->session_info
) {
248 ent
->session_info
= NULL
;
250 TALLOC_FREE(ent
->session_info
);
252 ent
->read_only
= False
;
257 /****************************************************************************
258 Clear a vuid out of the validity cache, and as the 'owner' of a connection.
260 Called from invalidate_vuid()
261 ****************************************************************************/
263 void conn_clear_vuid_caches(struct smbd_server_connection
*sconn
,uint16_t vuid
)
265 connection_struct
*conn
;
267 if (sconn
->using_smb2
) {
269 struct smbd_smb2_session
*sess
;
270 for (sess
= sconn
->smb2
.sessions
.list
; sess
; sess
= sess
->next
) {
271 struct smbd_smb2_tcon
*ptcon
;
273 for (ptcon
= sess
->tcons
.list
; ptcon
; ptcon
= ptcon
->next
) {
274 if (ptcon
->compat_conn
) {
275 if (ptcon
->compat_conn
->vuid
== vuid
) {
276 ptcon
->compat_conn
->vuid
= UID_FIELD_INVALID
;
278 conn_clear_vuid_cache(ptcon
->compat_conn
, vuid
);
284 for (conn
=sconn
->smb1
.tcons
.Connections
;conn
;conn
=conn
->next
) {
285 if (conn
->vuid
== vuid
) {
286 conn
->vuid
= UID_FIELD_INVALID
;
288 conn_clear_vuid_cache(conn
, vuid
);
293 /****************************************************************************
294 Free a conn structure - internal part.
295 ****************************************************************************/
297 static void conn_free_internal(connection_struct
*conn
)
299 vfs_handle_struct
*handle
= NULL
, *thandle
= NULL
;
300 struct trans_state
*state
= NULL
;
302 /* Free vfs_connection_struct */
303 handle
= conn
->vfs_handles
;
305 thandle
= handle
->next
;
306 DLIST_REMOVE(conn
->vfs_handles
, handle
);
307 if (handle
->free_data
)
308 handle
->free_data(&handle
->data
);
312 /* Free any pending transactions stored on this conn. */
313 for (state
= conn
->pending_trans
; state
; state
= state
->next
) {
314 /* state->setup is a talloc child of state. */
315 SAFE_FREE(state
->param
);
316 SAFE_FREE(state
->data
);
319 free_namearray(conn
->veto_list
);
320 free_namearray(conn
->hide_list
);
321 free_namearray(conn
->veto_oplock_list
);
322 free_namearray(conn
->aio_write_behind_list
);
324 string_free(&conn
->connectpath
);
325 string_free(&conn
->origpath
);
328 talloc_destroy(conn
);
331 /****************************************************************************
332 Free a conn structure.
333 ****************************************************************************/
335 void conn_free(connection_struct
*conn
)
337 if (conn
->sconn
== NULL
) {
338 conn_free_internal(conn
);
342 if (conn
->sconn
->using_smb2
) {
344 conn_free_internal(conn
);
349 DLIST_REMOVE(conn
->sconn
->smb1
.tcons
.Connections
, conn
);
351 if (conn
->sconn
->smb1
.tcons
.bmap
!= NULL
) {
353 * Can be NULL for fake connections created by
354 * create_conn_struct()
356 bitmap_clear(conn
->sconn
->smb1
.tcons
.bmap
, conn
->cnum
);
359 SMB_ASSERT(conn
->sconn
->num_tcons_open
> 0);
360 conn
->sconn
->num_tcons_open
--;
362 conn_free_internal(conn
);