2 Unix SMB/CIFS implementation.
3 Manage connections_struct structures
4 Copyright (C) Andrew Tridgell 1998
5 Copyright (C) Alexander Bokovoy 2002
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.
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.
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/>.
22 #include "smbd/globals.h"
24 /* The connections bitmap is expanded in increments of BITMAP_BLOCK_SZ. The
25 * maximum size of the bitmap is the largest positive integer, but you will hit
26 * the "max connections" limit, looong before that.
28 #define BITMAP_BLOCK_SZ 128
30 /****************************************************************************
31 init the conn structures
32 ****************************************************************************/
33 void conn_init(struct smbd_server_connection
*sconn
)
35 sconn
->smb1
.tcons
.Connections
= NULL
;
36 sconn
->smb1
.tcons
.num_open
= 0;
37 sconn
->smb1
.tcons
.bmap
= bitmap_allocate(BITMAP_BLOCK_SZ
);
40 /****************************************************************************
41 return the number of open connections
42 ****************************************************************************/
43 int conn_num_open(struct smbd_server_connection
*sconn
)
45 return sconn
->smb1
.tcons
.num_open
;
49 /****************************************************************************
50 check if a snum is in use
51 ****************************************************************************/
52 bool conn_snum_used(int snum
)
54 struct smbd_server_connection
*sconn
= smbd_server_conn
;
55 connection_struct
*conn
;
56 for (conn
=sconn
->smb1
.tcons
.Connections
;conn
;conn
=conn
->next
) {
57 if (conn
->params
->service
== snum
) {
64 /****************************************************************************
65 Find a conn given a cnum.
66 ****************************************************************************/
68 connection_struct
*conn_find(struct smbd_server_connection
*sconn
,unsigned cnum
)
71 connection_struct
*conn
;
73 for (conn
=sconn
->smb1
.tcons
.Connections
;conn
;conn
=conn
->next
,count
++) {
74 if (conn
->cnum
== cnum
) {
76 DLIST_PROMOTE(sconn
->smb1
.tcons
.Connections
,
86 /****************************************************************************
87 find first available connection slot, starting from a random position.
88 The randomisation stops problems with the server dieing and clients
89 thinking the server is still available.
90 ****************************************************************************/
91 connection_struct
*conn_new(struct smbd_server_connection
*sconn
)
93 connection_struct
*conn
;
97 if (sconn
->allow_smb2
) {
98 if (!(conn
=TALLOC_ZERO_P(NULL
, connection_struct
)) ||
99 !(conn
->params
= TALLOC_P(conn
, struct share_params
))) {
100 DEBUG(0,("TALLOC_ZERO() failed!\n"));
109 i
= bitmap_find(sconn
->smb1
.tcons
.bmap
, find_offset
);
112 /* Expand the connections bitmap. */
113 int oldsz
= sconn
->smb1
.tcons
.bmap
->n
;
114 int newsz
= sconn
->smb1
.tcons
.bmap
->n
+
116 struct bitmap
* nbmap
;
118 if (newsz
<= oldsz
) {
120 DEBUG(0,("ERROR! Out of connection structures\n"));
124 DEBUG(4,("resizing connections bitmap from %d to %d\n",
127 nbmap
= bitmap_allocate(newsz
);
129 DEBUG(0,("ERROR! malloc fail.\n"));
133 bitmap_copy(nbmap
, sconn
->smb1
.tcons
.bmap
);
134 bitmap_free(sconn
->smb1
.tcons
.bmap
);
136 sconn
->smb1
.tcons
.bmap
= nbmap
;
137 find_offset
= oldsz
; /* Start next search in the new portion. */
142 /* The bitmap position is used below as the connection number
143 * conn->cnum). This ends up as the TID field in the SMB header,
144 * which is limited to 16 bits (we skip 0xffff which is the
148 DEBUG(0, ("Maximum connection limit reached\n"));
152 if (!(conn
=TALLOC_ZERO_P(NULL
, connection_struct
)) ||
153 !(conn
->params
= TALLOC_P(conn
, struct share_params
))) {
154 DEBUG(0,("TALLOC_ZERO() failed!\n"));
160 conn
->force_group_gid
= (gid_t
)-1;
162 bitmap_set(sconn
->smb1
.tcons
.bmap
, i
);
164 sconn
->smb1
.tcons
.num_open
++;
166 string_set(&conn
->connectpath
,"");
167 string_set(&conn
->origpath
,"");
169 DLIST_ADD(sconn
->smb1
.tcons
.Connections
, conn
);
174 /****************************************************************************
175 Close all conn structures.
176 return true if any were closed
177 ****************************************************************************/
178 bool conn_close_all(struct smbd_server_connection
*sconn
)
180 connection_struct
*conn
, *next
;
182 for (conn
=sconn
->smb1
.tcons
.Connections
;conn
;conn
=next
) {
184 set_current_service(conn
, 0, True
);
185 close_cnum(conn
, conn
->vuid
);
191 /****************************************************************************
192 Idle inactive connections.
193 ****************************************************************************/
195 bool conn_idle_all(struct smbd_server_connection
*sconn
,time_t t
)
197 int deadtime
= lp_deadtime()*60;
198 pipes_struct
*plist
= NULL
;
199 connection_struct
*conn
;
203 deadtime
= DEFAULT_SMBD_TIMEOUT
;
205 for (conn
=sconn
->smb1
.tcons
.Connections
;conn
;conn
=conn
->next
) {
207 time_t age
= t
- conn
->lastused
;
209 /* Update if connection wasn't idle. */
210 if (conn
->lastused
!= conn
->lastused_count
) {
212 conn
->lastused_count
= t
;
216 /* close dirptrs on connections that are idle */
217 if (age
> DPTR_IDLE_TIMEOUT
) {
221 if (conn
->num_files_open
> 0 || age
< deadtime
) {
227 * Check all pipes for any open handles. We cannot
228 * idle with a handle open.
231 for (plist
= get_first_internal_pipe(); plist
;
232 plist
= get_next_internal_pipe(plist
)) {
233 if (num_pipe_handles(plist
->pipe_handles
) != 0) {
242 /****************************************************************************
243 Clear a vuid out of the validity cache, and as the 'owner' of a connection.
244 ****************************************************************************/
246 void conn_clear_vuid_caches(struct smbd_server_connection
*sconn
,uint16_t vuid
)
248 connection_struct
*conn
;
250 for (conn
=sconn
->smb1
.tcons
.Connections
;conn
;conn
=conn
->next
) {
251 if (conn
->vuid
== vuid
) {
252 conn
->vuid
= UID_FIELD_INVALID
;
254 conn_clear_vuid_cache(conn
, vuid
);
258 /****************************************************************************
259 Free a conn structure - internal part.
260 ****************************************************************************/
262 static void conn_free_internal(connection_struct
*conn
)
264 vfs_handle_struct
*handle
= NULL
, *thandle
= NULL
;
265 struct trans_state
*state
= NULL
;
267 /* Free vfs_connection_struct */
268 handle
= conn
->vfs_handles
;
270 thandle
= handle
->next
;
271 DLIST_REMOVE(conn
->vfs_handles
, handle
);
272 if (handle
->free_data
)
273 handle
->free_data(&handle
->data
);
277 /* Free any pending transactions stored on this conn. */
278 for (state
= conn
->pending_trans
; state
; state
= state
->next
) {
279 /* state->setup is a talloc child of state. */
280 SAFE_FREE(state
->param
);
281 SAFE_FREE(state
->data
);
284 free_namearray(conn
->veto_list
);
285 free_namearray(conn
->hide_list
);
286 free_namearray(conn
->veto_oplock_list
);
287 free_namearray(conn
->aio_write_behind_list
);
289 string_free(&conn
->connectpath
);
290 string_free(&conn
->origpath
);
293 talloc_destroy(conn
);
296 /****************************************************************************
297 Free a conn structure.
298 ****************************************************************************/
300 void conn_free(connection_struct
*conn
)
302 if (conn
->sconn
== NULL
) {
303 conn_free_internal(conn
);
307 if (conn
->sconn
->allow_smb2
) {
308 conn_free_internal(conn
);
312 DLIST_REMOVE(conn
->sconn
->smb1
.tcons
.Connections
, conn
);
314 bitmap_clear(conn
->sconn
->smb1
.tcons
.bmap
, conn
->cnum
);
316 SMB_ASSERT(conn
->sconn
->smb1
.tcons
.num_open
> 0);
317 conn
->sconn
->smb1
.tcons
.num_open
--;
319 conn_free_internal(conn
);
322 /****************************************************************************
323 receive a smbcontrol message to forcibly unmount a share
324 the message contains just a share name and all instances of that
326 the special sharename '*' forces unmount of all shares
327 ****************************************************************************/
328 void msg_force_tdis(struct messaging_context
*msg
,
331 struct server_id server_id
,
334 struct smbd_server_connection
*sconn
= smbd_server_conn
;
335 connection_struct
*conn
, *next
;
338 fstrcpy(sharename
, (const char *)data
->data
);
340 if (strcmp(sharename
, "*") == 0) {
341 DEBUG(1,("Forcing close of all shares\n"));
342 conn_close_all(sconn
);
346 for (conn
=sconn
->smb1
.tcons
.Connections
;conn
;conn
=next
) {
348 if (strequal(lp_servicename(SNUM(conn
)), sharename
)) {
349 DEBUG(1,("Forcing close of share %s cnum=%d\n",
350 sharename
, conn
->cnum
));
351 close_cnum(conn
, (uint16
)-1);