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 extern fstring remote_machine;
32 static TDB_CONTEXT *tdb;
33 /* called when a session is created */
34 BOOL session_claim(user_struct *vuser)
38 struct sessionid sessionid;
39 uint32 pid = (uint32)sys_getpid();
44 vuser->session_id = 0;
46 /* don't register sessions for the guest user - its just too
47 expensive to go through pam session code for browsing etc */
53 tdb = tdb_open_log(lock_path("sessionid.tdb"), 0, TDB_CLEAR_IF_FIRST|TDB_DEFAULT,
54 O_RDWR | O_CREAT, 0644);
56 DEBUG(1,("session_claim: failed to open sessionid tdb\n"));
61 ZERO_STRUCT(sessionid);
66 for (i=1;i<MAX_SESSION_ID;i++) {
67 slprintf(keystr, sizeof(keystr)-1, "ID/%d", i);
69 key.dsize = strlen(keystr)+1;
71 if (tdb_store(tdb, key, data, TDB_INSERT) == 0) break;
74 if (i == MAX_SESSION_ID) {
75 DEBUG(1,("session_claim: out of session IDs (max is %d)\n",
80 /* If 'hostname lookup' == yes, then do the DNS lookup. This is
81 needed becouse utmp and PAM both expect DNS names
83 client_name() handles this case internally.
86 hostname = client_name();
87 if (strcmp(hostname, "UNKNOWN") == 0) {
88 hostname = client_addr();
91 fstrcpy(sessionid.username, vuser->user.unix_name);
92 fstrcpy(sessionid.hostname, hostname);
93 slprintf(sessionid.id_str, sizeof(sessionid.id_str)-1, SESSION_TEMPLATE, i);
96 sessionid.uid = vuser->uid;
97 sessionid.gid = vuser->gid;
98 fstrcpy(sessionid.remote_machine, remote_machine);
99 fstrcpy(sessionid.ip_addr, client_addr());
101 if (!smb_pam_claim_session(sessionid.username, sessionid.id_str, sessionid.hostname)) {
102 DEBUG(1,("pam_session rejected the session for %s [%s]\n",
103 sessionid.username, sessionid.id_str));
104 tdb_delete(tdb, key);
108 data.dptr = (char *)&sessionid;
109 data.dsize = sizeof(sessionid);
110 if (tdb_store(tdb, key, data, TDB_MODIFY) != 0) {
111 DEBUG(1,("session_claim: unable to create session id record\n"));
117 sys_utmp_claim(sessionid.username, sessionid.hostname,
118 sessionid.id_str, sessionid.id_num);
122 vuser->session_id = i;
126 /* called when a session is destroyed */
127 void session_yield(user_struct *vuser)
130 struct sessionid sessionid;
136 if (vuser->session_id == 0) {
140 slprintf(keystr, sizeof(keystr)-1, "ID/%d", vuser->session_id);
143 key.dsize = strlen(keystr)+1;
145 dbuf = tdb_fetch(tdb, key);
147 if (dbuf.dsize != sizeof(sessionid))
150 memcpy(&sessionid, dbuf.dptr, sizeof(sessionid));
152 SAFE_FREE(dbuf.dptr);
156 sys_utmp_yield(sessionid.username, sessionid.hostname,
157 sessionid.id_str, sessionid.id_num);
161 smb_pam_close_session(sessionid.username, sessionid.id_str, sessionid.hostname);
163 tdb_delete(tdb, key);
166 static BOOL session_traverse(int (*fn)(TDB_CONTEXT *, TDB_DATA, TDB_DATA, void *), void *state)
169 DEBUG(3, ("No tdb opened\n"));
173 tdb_traverse(tdb, fn, state);
177 struct session_list {
179 struct sessionid *sessions;
182 static int gather_sessioninfo(TDB_CONTEXT *stdb, TDB_DATA kbuf, TDB_DATA dbuf,
185 struct session_list *sesslist = (struct session_list *) state;
186 const struct sessionid *current = (const struct sessionid *) dbuf.dptr;
188 sesslist->count += 1;
189 sesslist->sessions = REALLOC(sesslist->sessions, sesslist->count *
190 sizeof(struct sessionid));
192 memcpy(&sesslist->sessions[sesslist->count - 1], current,
193 sizeof(struct sessionid));
194 DEBUG(7,("gather_sessioninfo session from %s@%s\n",
195 current->username, current->remote_machine));
199 int list_sessions(struct sessionid **session_list)
201 struct session_list sesslist;
204 sesslist.sessions = NULL;
206 if (!session_traverse(gather_sessioninfo, (void *) &sesslist)) {
207 DEBUG(3, ("Session traverse failed\n"));
208 SAFE_FREE(sesslist.sessions);
209 *session_list = NULL;
213 *session_list = sesslist.sessions;
214 return sesslist.count;