2 Unix SMB/CIFS implementation.
3 session handling for utmp and PAM
4 Copyright (C) tridge@samba.org 2001
5 Copyright (C) abartlet@pcug.org.au 2001
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.
22 /* a "session" is claimed when we do a SessionSetupX operation
23 and is yielded when the corresponding vuid is destroyed.
25 sessions are used to populate utmp and PAM session structures
30 static TDB_CONTEXT
*tdb
;
31 /* called when a session is created */
32 BOOL
session_claim(user_struct
*vuser
)
37 struct in_addr
*client_ip
;
38 struct sessionid sessionid
;
39 uint32 pid
= (uint32
)sys_getpid();
43 int tdb_store_flag
; /* If using utmp, we do an inital 'lock hold' store,
44 but we don't need this if we are just using the
45 (unique) pid/vuid combination */
47 vuser
->session_keystr
= NULL
;
49 /* don't register sessions for the guest user - its just too
50 expensive to go through pam session code for browsing etc */
56 tdb
= tdb_open_log(lock_path("sessionid.tdb"), 0, TDB_CLEAR_IF_FIRST
|TDB_DEFAULT
,
57 O_RDWR
| O_CREAT
, 0644);
59 DEBUG(1,("session_claim: failed to open sessionid tdb\n"));
64 ZERO_STRUCT(sessionid
);
71 for (i
=1;i
<MAX_SESSION_ID
;i
++) {
72 slprintf(keystr
, sizeof(keystr
)-1, "ID/%d", i
);
74 key
.dsize
= strlen(keystr
)+1;
76 if (tdb_store(tdb
, key
, data
, TDB_INSERT
) == 0) break;
79 if (i
== MAX_SESSION_ID
) {
80 DEBUG(1,("session_claim: out of session IDs (max is %d)\n",
84 slprintf(sessionid
.id_str
, sizeof(sessionid
.id_str
)-1, SESSION_UTMP_TEMPLATE
, i
);
85 tdb_store_flag
= TDB_MODIFY
;
89 slprintf(keystr
, sizeof(keystr
)-1, "ID/%lu/%u",
90 (long unsigned int)sys_getpid(),
92 slprintf(sessionid
.id_str
, sizeof(sessionid
.id_str
)-1,
93 SESSION_TEMPLATE
, (long unsigned int)sys_getpid(),
97 key
.dsize
= strlen(keystr
)+1;
99 tdb_store_flag
= TDB_REPLACE
;
102 /* If 'hostname lookup' == yes, then do the DNS lookup. This is
103 needed becouse utmp and PAM both expect DNS names
105 client_name() handles this case internally.
108 hostname
= client_name();
109 if (strcmp(hostname
, "UNKNOWN") == 0) {
110 hostname
= client_addr();
113 fstrcpy(sessionid
.username
, vuser
->user
.unix_name
);
114 fstrcpy(sessionid
.hostname
, hostname
);
115 sessionid
.id_num
= i
; /* Only valid for utmp sessions */
117 sessionid
.uid
= vuser
->uid
;
118 sessionid
.gid
= vuser
->gid
;
119 fstrcpy(sessionid
.remote_machine
, get_remote_machine_name());
120 fstrcpy(sessionid
.ip_addr
, client_addr());
122 client_ip
= client_inaddr(&sa
);
124 if (!smb_pam_claim_session(sessionid
.username
, sessionid
.id_str
, sessionid
.hostname
)) {
125 DEBUG(1,("pam_session rejected the session for %s [%s]\n",
126 sessionid
.username
, sessionid
.id_str
));
127 if (tdb_store_flag
== TDB_MODIFY
) {
128 tdb_delete(tdb
, key
);
133 data
.dptr
= (char *)&sessionid
;
134 data
.dsize
= sizeof(sessionid
);
135 if (tdb_store(tdb
, key
, data
, tdb_store_flag
) != 0) {
136 DEBUG(1,("session_claim: unable to create session id record\n"));
142 sys_utmp_claim(sessionid
.username
, sessionid
.hostname
,
144 sessionid
.id_str
, sessionid
.id_num
);
148 vuser
->session_keystr
= strdup(keystr
);
149 if (!vuser
->session_keystr
) {
150 DEBUG(0, ("session_claim: strdup() failed for session_keystr\n"));
156 /* called when a session is destroyed */
157 void session_yield(user_struct
*vuser
)
160 struct sessionid sessionid
;
161 struct in_addr
*client_ip
;
166 if (!vuser
->session_keystr
) {
170 key
.dptr
= vuser
->session_keystr
;
171 key
.dsize
= strlen(vuser
->session_keystr
)+1;
173 dbuf
= tdb_fetch(tdb
, key
);
175 if (dbuf
.dsize
!= sizeof(sessionid
))
178 memcpy(&sessionid
, dbuf
.dptr
, sizeof(sessionid
));
180 client_ip
= interpret_addr2(sessionid
.ip_addr
);
182 SAFE_FREE(dbuf
.dptr
);
186 sys_utmp_yield(sessionid
.username
, sessionid
.hostname
,
188 sessionid
.id_str
, sessionid
.id_num
);
192 smb_pam_close_session(sessionid
.username
, sessionid
.id_str
, sessionid
.hostname
);
194 tdb_delete(tdb
, key
);
197 static BOOL
session_traverse(int (*fn
)(TDB_CONTEXT
*, TDB_DATA
, TDB_DATA
, void *), void *state
)
200 DEBUG(3, ("No tdb opened\n"));
204 tdb_traverse(tdb
, fn
, state
);
208 struct session_list
{
210 struct sessionid
*sessions
;
213 static int gather_sessioninfo(TDB_CONTEXT
*stdb
, TDB_DATA kbuf
, TDB_DATA dbuf
,
216 struct session_list
*sesslist
= (struct session_list
*) state
;
217 const struct sessionid
*current
= (const struct sessionid
*) dbuf
.dptr
;
219 sesslist
->count
+= 1;
220 sesslist
->sessions
= REALLOC(sesslist
->sessions
, sesslist
->count
*
221 sizeof(struct sessionid
));
223 memcpy(&sesslist
->sessions
[sesslist
->count
- 1], current
,
224 sizeof(struct sessionid
));
225 DEBUG(7,("gather_sessioninfo session from %s@%s\n",
226 current
->username
, current
->remote_machine
));
230 int list_sessions(struct sessionid
**session_list
)
232 struct session_list sesslist
;
235 sesslist
.sessions
= NULL
;
237 if (!session_traverse(gather_sessioninfo
, (void *) &sesslist
)) {
238 DEBUG(3, ("Session traverse failed\n"));
239 SAFE_FREE(sesslist
.sessions
);
240 *session_list
= NULL
;
244 *session_list
= sesslist
.sessions
;
245 return sesslist
.count
;