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;