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 2 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, write to the Free Software
19 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
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 static connection_struct *Connections;
32 /* number of open connections */
33 static struct bitmap *bmap;
36 /****************************************************************************
37 init the conn structures
38 ****************************************************************************/
41 bmap = bitmap_allocate(BITMAP_BLOCK_SZ);
44 /****************************************************************************
45 return the number of open connections
46 ****************************************************************************/
47 int conn_num_open(void)
53 /****************************************************************************
54 check if a snum is in use
55 ****************************************************************************/
56 BOOL conn_snum_used(int snum)
58 connection_struct *conn;
59 for (conn=Connections;conn;conn=conn->next) {
60 if (conn->params->service == snum) {
67 /****************************************************************************
68 Find a conn given a cnum.
69 ****************************************************************************/
71 connection_struct *conn_find(unsigned cnum)
74 connection_struct *conn;
76 for (conn=Connections;conn;conn=conn->next,count++) {
77 if (conn->cnum == cnum) {
79 DLIST_PROMOTE(Connections, conn);
88 /****************************************************************************
89 Find a conn given a service name.
90 ****************************************************************************/
92 connection_struct *conn_find_byname(const char *service)
94 connection_struct *conn;
96 for (conn=Connections;conn;conn=conn->next) {
97 if (strequal(lp_servicename(SNUM(conn)),service)) {
98 if (conn != Connections) {
99 /* Promote if not first. */
100 DLIST_PROMOTE(Connections, conn);
110 /****************************************************************************
111 find first available connection slot, starting from a random position.
112 The randomisation stops problems with the server dieing and clients
113 thinking the server is still available.
114 ****************************************************************************/
115 connection_struct *conn_new(void)
118 connection_struct *conn;
123 i = bitmap_find(bmap, find_offset);
126 /* Expand the connections bitmap. */
128 int newsz = bmap->n + BITMAP_BLOCK_SZ;
129 struct bitmap * nbmap;
131 if (newsz <= oldsz) {
133 DEBUG(0,("ERROR! Out of connection structures\n"));
137 DEBUG(4,("resizing connections bitmap from %d to %d\n",
140 nbmap = bitmap_allocate(newsz);
142 DEBUG(0,("ERROR! malloc fail.\n"));
146 bitmap_copy(nbmap, bmap);
150 find_offset = oldsz; /* Start next search in the new portion. */
155 /* The bitmap position is used below as the connection number
156 * conn->cnum). This ends up as the TID field in the SMB header,
157 * which is limited to 16 bits (we skip 0xffff which is the
161 DEBUG(0, ("Maximum connection limit reached\n"));
165 if ((mem_ctx=talloc_init("connection_struct"))==NULL) {
166 DEBUG(0,("talloc_init(connection_struct) failed!\n"));
170 if (!(conn=TALLOC_ZERO_P(mem_ctx, connection_struct)) ||
171 !(conn->params = TALLOC_P(mem_ctx, struct share_params))) {
172 DEBUG(0,("TALLOC_ZERO() failed!\n"));
173 TALLOC_FREE(mem_ctx);
176 conn->mem_ctx = mem_ctx;
183 string_set(&conn->user,"");
184 string_set(&conn->dirpath,"");
185 string_set(&conn->connectpath,"");
186 string_set(&conn->origpath,"");
188 DLIST_ADD(Connections, conn);
193 /****************************************************************************
194 Close all conn structures.
195 ****************************************************************************/
197 void conn_close_all(void)
199 connection_struct *conn, *next;
200 for (conn=Connections;conn;conn=next) {
202 set_current_service(conn, 0, True);
203 close_cnum(conn, conn->vuid);
207 /****************************************************************************
208 Idle inactive connections.
209 ****************************************************************************/
211 BOOL conn_idle_all(time_t t, int deadtime)
213 pipes_struct *plist = NULL;
215 connection_struct *conn, *next;
217 for (conn=Connections;conn;conn=next) {
220 /* Update if connection wasn't idle. */
221 if (conn->lastused != conn->lastused_count) {
223 conn->lastused_count = t;
226 /* close dirptrs on connections that are idle */
227 if ((t-conn->lastused) > DPTR_IDLE_TIMEOUT) {
231 if (conn->num_files_open > 0 || (t-conn->lastused)<deadtime) {
237 * Check all pipes for any open handles. We cannot
238 * idle with a handle open.
241 for (plist = get_first_internal_pipe(); plist; plist = get_next_internal_pipe(plist))
242 if (plist->pipe_handles && plist->pipe_handles->count)
248 /****************************************************************************
249 Clear a vuid out of the validity cache, and as the 'owner' of a connection.
250 ****************************************************************************/
252 void conn_clear_vuid_cache(uint16 vuid)
254 connection_struct *conn;
257 for (conn=Connections;conn;conn=conn->next) {
258 if (conn->vuid == vuid) {
259 conn->vuid = UID_FIELD_INVALID;
262 for (i=0;i<conn->vuid_cache.entries && i< VUID_CACHE_SIZE;i++) {
263 if (conn->vuid_cache.array[i].vuid == vuid) {
264 struct vuid_cache_entry *ent = &conn->vuid_cache.array[i];
265 ent->vuid = UID_FIELD_INVALID;
266 ent->read_only = False;
267 ent->admin_user = False;
273 /****************************************************************************
274 Free a conn structure - internal part.
275 ****************************************************************************/
277 void conn_free_internal(connection_struct *conn)
279 vfs_handle_struct *handle = NULL, *thandle = NULL;
280 TALLOC_CTX *mem_ctx = NULL;
281 struct trans_state *state = NULL;
283 /* Free vfs_connection_struct */
284 handle = conn->vfs_handles;
286 DLIST_REMOVE(conn->vfs_handles, handle);
287 thandle = handle->next;
288 if (handle->free_data)
289 handle->free_data(&handle->data);
293 /* Free any pending transactions stored on this conn. */
294 for (state = conn->pending_trans; state; state = state->next) {
295 /* state->setup is a talloc child of state. */
296 SAFE_FREE(state->param);
297 SAFE_FREE(state->data);
300 free_namearray(conn->veto_list);
301 free_namearray(conn->hide_list);
302 free_namearray(conn->veto_oplock_list);
303 free_namearray(conn->aio_write_behind_list);
305 string_free(&conn->user);
306 string_free(&conn->dirpath);
307 string_free(&conn->connectpath);
308 string_free(&conn->origpath);
310 mem_ctx = conn->mem_ctx;
312 talloc_destroy(mem_ctx);
315 /****************************************************************************
316 Free a conn structure.
317 ****************************************************************************/
319 void conn_free(connection_struct *conn)
321 DLIST_REMOVE(Connections, conn);
323 bitmap_clear(bmap, conn->cnum);
326 conn_free_internal(conn);
329 /****************************************************************************
330 receive a smbcontrol message to forcibly unmount a share
331 the message contains just a share name and all instances of that
333 the special sharename '*' forces unmount of all shares
334 ****************************************************************************/
335 void msg_force_tdis(int msg_type, struct process_id pid, void *buf, size_t len,
338 connection_struct *conn, *next;
341 fstrcpy(sharename, (const char *)buf);
343 if (strcmp(sharename, "*") == 0) {
344 DEBUG(1,("Forcing close of all shares\n"));
349 for (conn=Connections;conn;conn=next) {
351 if (strequal(lp_servicename(SNUM(conn)), sharename)) {
352 DEBUG(1,("Forcing close of share %s cnum=%d\n",
353 sharename, conn->cnum));
354 close_cnum(conn, (uint16)-1);