2 Unix SMB/CIFS implementation.
3 session handling for utmp and PAM
5 Copyright (C) tridge@samba.org 2001
6 Copyright (C) abartlet@samba.org 2001
7 Copyright (C) Gerald (Jerry) Carter 2006
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 3 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program. If not, see <http://www.gnu.org/licenses/>.
23 /* a "session" is claimed when we do a SessionSetupX operation
24 and is yielded when the corresponding vuid is destroyed.
26 sessions are used to populate utmp and PAM session structures
31 /********************************************************************
32 ********************************************************************/
34 static struct db_context
*session_db_ctx(void)
36 static struct db_context
*ctx
;
41 ctx
= db_open(NULL
, lock_path("sessionid.tdb"), 0,
42 TDB_CLEAR_IF_FIRST
|TDB_DEFAULT
,
43 O_RDWR
| O_CREAT
, 0644);
47 BOOL
session_init(void)
49 if (session_db_ctx() == NULL
) {
50 DEBUG(1,("session_init: failed to open sessionid tdb\n"));
57 /********************************************************************
58 called when a session is created
59 ********************************************************************/
61 BOOL
session_claim(user_struct
*vuser
)
66 struct in_addr
*client_ip
;
67 struct sessionid sessionid
;
68 struct server_id pid
= procid_self();
71 struct db_context
*ctx
;
72 struct db_record
*rec
;
75 vuser
->session_keystr
= NULL
;
77 /* don't register sessions for the guest user - its just too
78 expensive to go through pam session code for browsing etc */
83 if (!(ctx
= session_db_ctx())) {
87 ZERO_STRUCT(sessionid
);
94 for (i
=1;i
<MAX_SESSION_ID
;i
++) {
97 * This is very inefficient and needs fixing -- vl
100 struct server_id sess_pid
;
102 snprintf(keystr
, sizeof(keystr
), "ID/%d", i
);
103 key
= string_term_tdb_data(keystr
);
105 rec
= ctx
->fetch_locked(ctx
, NULL
, key
);
108 DEBUG(1, ("Could not lock \"%s\"\n", keystr
));
112 if (rec
->value
.dsize
!= sizeof(sessionid
)) {
113 DEBUG(1, ("Re-using invalid record\n"));
117 sess_pid
= ((struct sessionid
*)rec
->value
.dptr
)->pid
;
119 if (!process_exists(sess_pid
)) {
120 DEBUG(5, ("%s has died -- re-using session\n",
121 procid_str_static(&sess_pid
)));
128 if (i
== MAX_SESSION_ID
) {
129 SMB_ASSERT(rec
== NULL
);
130 DEBUG(1,("session_claim: out of session IDs "
131 "(max is %d)\n", MAX_SESSION_ID
));
135 snprintf(sessionid
.id_str
, sizeof(sessionid
.id_str
),
136 SESSION_UTMP_TEMPLATE
, i
);
139 snprintf(keystr
, sizeof(keystr
), "ID/%s/%u",
140 procid_str_static(&pid
), vuser
->vuid
);
141 key
= string_term_tdb_data(keystr
);
143 rec
= ctx
->fetch_locked(ctx
, NULL
, key
);
146 DEBUG(1, ("Could not lock \"%s\"\n", keystr
));
150 snprintf(sessionid
.id_str
, sizeof(sessionid
.id_str
),
151 SESSION_TEMPLATE
, (long unsigned int)sys_getpid(),
155 SMB_ASSERT(rec
!= NULL
);
157 /* If 'hostname lookup' == yes, then do the DNS lookup. This is
158 needed because utmp and PAM both expect DNS names
160 client_name() handles this case internally.
163 hostname
= client_name();
164 if (strcmp(hostname
, "UNKNOWN") == 0) {
165 hostname
= client_addr();
168 fstrcpy(sessionid
.username
, vuser
->user
.unix_name
);
169 fstrcpy(sessionid
.hostname
, hostname
);
170 sessionid
.id_num
= i
; /* Only valid for utmp sessions */
172 sessionid
.uid
= vuser
->uid
;
173 sessionid
.gid
= vuser
->gid
;
174 fstrcpy(sessionid
.remote_machine
, get_remote_machine_name());
175 fstrcpy(sessionid
.ip_addr
, client_addr());
176 sessionid
.connect_start
= time(NULL
);
178 client_ip
= client_inaddr(&sa
);
180 if (!smb_pam_claim_session(sessionid
.username
, sessionid
.id_str
,
181 sessionid
.hostname
)) {
182 DEBUG(1,("pam_session rejected the session for %s [%s]\n",
183 sessionid
.username
, sessionid
.id_str
));
189 data
.dptr
= (uint8
*)&sessionid
;
190 data
.dsize
= sizeof(sessionid
);
192 status
= rec
->store(rec
, data
, TDB_REPLACE
);
196 if (!NT_STATUS_IS_OK(status
)) {
197 DEBUG(1,("session_claim: unable to create session id "
198 "record: %s\n", nt_errstr(status
)));
203 sys_utmp_claim(sessionid
.username
, sessionid
.hostname
,
205 sessionid
.id_str
, sessionid
.id_num
);
210 vuser
->session_keystr
= talloc_strdup(vuser
, keystr
);
211 if (!vuser
->session_keystr
) {
212 DEBUG(0, ("session_claim: talloc_strdup() failed for "
213 "session_keystr\n"));
219 /********************************************************************
220 called when a session is destroyed
221 ********************************************************************/
223 void session_yield(user_struct
*vuser
)
226 struct sessionid sessionid
;
227 struct in_addr
*client_ip
;
228 struct db_context
*ctx
;
229 struct db_record
*rec
;
231 if (!(ctx
= session_db_ctx())) return;
233 if (!vuser
->session_keystr
) {
237 key
= string_term_tdb_data(vuser
->session_keystr
);
239 if (!(rec
= ctx
->fetch_locked(ctx
, NULL
, key
))) {
243 if (rec
->value
.dsize
!= sizeof(sessionid
))
246 memcpy(&sessionid
, rec
->value
.dptr
, sizeof(sessionid
));
248 client_ip
= interpret_addr2(sessionid
.ip_addr
);
251 sys_utmp_yield(sessionid
.username
, sessionid
.hostname
,
253 sessionid
.id_str
, sessionid
.id_num
);
256 smb_pam_close_session(sessionid
.username
, sessionid
.id_str
,
259 rec
->delete_rec(rec
);
264 /********************************************************************
265 ********************************************************************/
267 static BOOL
session_traverse(int (*fn
)(struct db_record
*db
,
271 struct db_context
*ctx
;
273 if (!(ctx
= session_db_ctx())) {
274 DEBUG(3, ("No tdb opened\n"));
278 ctx
->traverse_read(ctx
, fn
, private_data
);
282 /********************************************************************
283 ********************************************************************/
285 struct session_list
{
288 struct sessionid
*sessions
;
291 static int gather_sessioninfo(struct db_record
*rec
, void *state
)
293 struct session_list
*sesslist
= (struct session_list
*) state
;
294 const struct sessionid
*current
=
295 (const struct sessionid
*) rec
->value
.dptr
;
297 sesslist
->sessions
= TALLOC_REALLOC_ARRAY(
298 sesslist
->mem_ctx
, sesslist
->sessions
, struct sessionid
,
301 if (!sesslist
->sessions
) {
306 memcpy(&sesslist
->sessions
[sesslist
->count
], current
,
307 sizeof(struct sessionid
));
311 DEBUG(7,("gather_sessioninfo session from %s@%s\n",
312 current
->username
, current
->remote_machine
));
317 /********************************************************************
318 ********************************************************************/
320 int list_sessions(TALLOC_CTX
*mem_ctx
, struct sessionid
**session_list
)
322 struct session_list sesslist
;
324 sesslist
.mem_ctx
= mem_ctx
;
326 sesslist
.sessions
= NULL
;
328 if (!session_traverse(gather_sessioninfo
, (void *) &sesslist
)) {
329 DEBUG(3, ("Session traverse failed\n"));
330 SAFE_FREE(sesslist
.sessions
);
331 *session_list
= NULL
;
335 *session_list
= sesslist
.sessions
;
336 return sesslist
.count
;