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
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 2 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, write to the Free Software
20 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
24 #include "lib/util/dlinklist.h"
25 #include "smb_server/smb_server.h"
26 #include "smbd/service_stream.h"
27 #include "ntvfs/ntvfs.h"
29 struct socket_address
*smbsrv_get_my_addr(void *p
, TALLOC_CTX
*mem_ctx
)
31 struct smbsrv_connection
*smb_conn
= talloc_get_type(p
,
32 struct smbsrv_connection
);
34 return socket_get_my_addr(smb_conn
->connection
->socket
, mem_ctx
);
37 struct socket_address
*smbsrv_get_peer_addr(void *p
, TALLOC_CTX
*mem_ctx
)
39 struct smbsrv_connection
*smb_conn
= talloc_get_type(p
,
40 struct smbsrv_connection
);
42 return socket_get_peer_addr(smb_conn
->connection
->socket
, mem_ctx
);
45 /****************************************************************************
46 init the tcon structures
47 ****************************************************************************/
48 static NTSTATUS
smbsrv_init_tcons(struct smbsrv_tcons_context
*tcons_ctx
, TALLOC_CTX
*mem_ctx
, uint32_t limit
)
51 * the idr_* functions take 'int' as limit,
52 * and only work with a max limit 0x00FFFFFF
56 tcons_ctx
->idtree_tid
= idr_init(mem_ctx
);
57 NT_STATUS_HAVE_NO_MEMORY(tcons_ctx
->idtree_tid
);
58 tcons_ctx
->idtree_limit
= limit
;
59 tcons_ctx
->list
= NULL
;
64 NTSTATUS
smbsrv_smb_init_tcons(struct smbsrv_connection
*smb_conn
)
66 return smbsrv_init_tcons(&smb_conn
->smb_tcons
, smb_conn
, UINT16_MAX
);
69 NTSTATUS
smbsrv_smb2_init_tcons(struct smbsrv_session
*smb_sess
)
71 return smbsrv_init_tcons(&smb_sess
->smb2_tcons
, smb_sess
, UINT32_MAX
);
74 /****************************************************************************
75 find a tcon given a tid for SMB
76 ****************************************************************************/
77 static struct smbsrv_tcon
*smbsrv_tcon_find(struct smbsrv_tcons_context
*tcons_ctx
,
78 uint32_t tid
, struct timeval request_time
)
81 struct smbsrv_tcon
*tcon
;
83 if (tid
== 0) return NULL
;
85 if (tid
> tcons_ctx
->idtree_limit
) return NULL
;
87 p
= idr_find(tcons_ctx
->idtree_tid
, tid
);
90 tcon
= talloc_get_type(p
, struct smbsrv_tcon
);
91 if (!tcon
) return NULL
;
93 tcon
->statistics
.last_request_time
= request_time
;
98 struct smbsrv_tcon
*smbsrv_smb_tcon_find(struct smbsrv_connection
*smb_conn
,
99 uint32_t tid
, struct timeval request_time
)
101 return smbsrv_tcon_find(&smb_conn
->smb_tcons
, tid
, request_time
);
104 struct smbsrv_tcon
*smbsrv_smb2_tcon_find(struct smbsrv_session
*smb_sess
,
105 uint32_t tid
, struct timeval request_time
)
107 if (!smb_sess
) return NULL
;
108 return smbsrv_tcon_find(&smb_sess
->smb2_tcons
, tid
, request_time
);
112 destroy a connection structure
114 static int smbsrv_tcon_destructor(struct smbsrv_tcon
*tcon
)
116 struct smbsrv_tcons_context
*tcons_ctx
;
117 struct socket_address
*client_addr
;
119 client_addr
= socket_get_peer_addr(tcon
->smb_conn
->connection
->socket
, tcon
);
120 DEBUG(3,("%s closed connection to service %s\n",
121 client_addr
? client_addr
->addr
: "(unknown)",
124 /* tell the ntvfs backend that we are disconnecting */
126 ntvfs_disconnect(tcon
->ntvfs
);
130 if (tcon
->smb2
.session
) {
131 tcons_ctx
= &tcon
->smb2
.session
->smb2_tcons
;
133 tcons_ctx
= &tcon
->smb_conn
->smb_tcons
;
136 idr_remove(tcons_ctx
->idtree_tid
, tcon
->tid
);
137 DLIST_REMOVE(tcons_ctx
->list
, tcon
);
142 find first available connection slot
144 static struct smbsrv_tcon
*smbsrv_tcon_new(struct smbsrv_connection
*smb_conn
,
145 struct smbsrv_session
*smb_sess
,
146 const char *share_name
)
149 struct smbsrv_tcons_context
*tcons_ctx
;
150 struct smbsrv_tcon
*tcon
;
156 tcons_ctx
= &smb_sess
->smb2_tcons
;
159 tcons_ctx
= &smb_conn
->smb_tcons
;
162 tcon
= talloc_zero(mem_ctx
, struct smbsrv_tcon
);
163 if (!tcon
) return NULL
;
164 tcon
->smb_conn
= smb_conn
;
165 tcon
->smb2
.session
= smb_sess
;
166 tcon
->share_name
= talloc_strdup(tcon
, share_name
);
167 if (!tcon
->share_name
) goto failed
;
170 * the use -1 here, because we don't want to give away the wildcard
171 * fnum used in SMBflush
173 status
= smbsrv_init_handles(tcon
, UINT16_MAX
- 1);
174 if (!NT_STATUS_IS_OK(status
)) {
175 DEBUG(1,("ERROR! failed to init handles: %s\n", nt_errstr(status
)));
179 i
= idr_get_new_random(tcons_ctx
->idtree_tid
, tcon
, tcons_ctx
->idtree_limit
);
181 DEBUG(1,("ERROR! Out of connection structures\n"));
186 DLIST_ADD(tcons_ctx
->list
, tcon
);
187 talloc_set_destructor(tcon
, smbsrv_tcon_destructor
);
189 /* now fill in some statistics */
190 tcon
->statistics
.connect_time
= timeval_current();
199 struct smbsrv_tcon
*smbsrv_smb_tcon_new(struct smbsrv_connection
*smb_conn
, const char *share_name
)
201 return smbsrv_tcon_new(smb_conn
, NULL
, share_name
);
204 struct smbsrv_tcon
*smbsrv_smb2_tcon_new(struct smbsrv_session
*smb_sess
, const char *share_name
)
206 return smbsrv_tcon_new(smb_sess
->smb_conn
, smb_sess
, share_name
);