Optimization suggested by Volker. Don't do a stat system call on normal read path.
[Samba.git] / source3 / smbd / msg_idmap.c
blob2a00f1bbb919d2518542308b2e1c2342bddb2174
1 /*
2 * Samba Unix/Linux SMB client library
4 * Copyright (C) Gregor Beck 2011
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 3 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
20 /**
21 * @brief Notify smbd about idmap changes
22 * @file msg_idmap.c
23 * @author Gregor Beck <gb@sernet.de>
24 * @date Feb 2011
28 #include "includes.h"
29 #include "smbd/smbd.h"
30 #include "globals.h"
31 #include "../libcli/security/dom_sid.h"
32 #include "idmap_cache.h"
33 #include "passdb/lookup_sid.h"
34 #include "auth.h"
35 #include "messages.h"
37 struct id {
38 union {
39 uid_t uid;
40 gid_t gid;
41 struct dom_sid sid;
42 } id;
43 enum {UID, GID, SID} type;
46 static bool parse_id(const char* str, struct id* id)
48 struct dom_sid sid;
49 unsigned long ul;
50 char c, trash;
52 if (sscanf(str, "%cID %lu%c", &c, &ul, &trash) == 2) {
53 switch(c) {
54 case 'G':
55 id->id.gid = ul;
56 id->type = GID;
57 return true;
58 case 'U':
59 id->id.uid = ul;
60 id->type = UID;
61 return true;
62 default:
63 break;
65 } else if (string_to_sid(&sid, str)) {
66 id->id.sid = sid;
67 id->type = SID;
68 return true;
70 return false;
73 static bool uid_in_use(const struct user_struct* user, uid_t uid)
75 while (user) {
76 if (user->session_info && (user->session_info->utok.uid == uid)) {
77 return true;
79 user = user->next;
81 return false;
84 static bool gid_in_use(const struct user_struct* user, gid_t gid)
86 while (user) {
87 if (user->session_info != NULL) {
88 int i;
89 struct security_unix_token utok = user->session_info->utok;
90 if (utok.gid == gid) {
91 return true;
93 for(i=0; i<utok.ngroups; i++) {
94 if (utok.groups[i] == gid) {
95 return true;
99 user = user->next;
101 return false;
104 static bool sid_in_use(const struct user_struct* user, const struct dom_sid* psid)
106 uid_t uid;
107 gid_t gid;
108 if (sid_to_gid(psid, &gid)) {
109 return gid_in_use(user, gid);
110 } else if (sid_to_uid(psid, &uid)) {
111 return uid_in_use(user, uid);
113 return false;
117 static bool id_in_use(const struct user_struct* user, const struct id* id)
119 switch(id->type) {
120 case UID:
121 return uid_in_use(user, id->id.uid);
122 case GID:
123 return gid_in_use(user, id->id.gid);
124 case SID:
125 return sid_in_use(user, &id->id.sid);
126 default:
127 break;
129 return false;
132 static void delete_from_cache(const struct id* id)
134 switch(id->type) {
135 case UID:
136 delete_uid_cache(id->id.uid);
137 idmap_cache_del_uid(id->id.uid);
138 break;
139 case GID:
140 delete_gid_cache(id->id.gid);
141 idmap_cache_del_gid(id->id.gid);
142 break;
143 case SID:
144 delete_sid_cache(&id->id.sid);
145 idmap_cache_del_sid(&id->id.sid);
146 break;
147 default:
148 break;
153 static void message_idmap_flush(struct messaging_context *msg_ctx,
154 void* private_data,
155 uint32_t msg_type,
156 struct server_id server_id,
157 DATA_BLOB* data)
159 const char* msg = data ? (const char*)data->data : NULL;
161 if ((msg == NULL) || (msg[0] == '\0')) {
162 flush_gid_cache();
163 flush_uid_cache();
164 } else if (strncmp(msg, "GID", 3)) {
165 flush_gid_cache();
166 } else if (strncmp(msg, "UID", 3)) {
167 flush_uid_cache();
168 } else {
169 DEBUG(0, ("Invalid argument: %s\n", msg));
174 static void message_idmap_delete(struct messaging_context *msg_ctx,
175 void* private_data,
176 uint32_t msg_type,
177 struct server_id server_id,
178 DATA_BLOB* data)
180 const char* msg = (data && data->data) ? (const char*)data->data : "<NULL>";
181 bool do_kill = (msg_type == MSG_IDMAP_KILL);
182 struct user_struct* validated_users = smbd_server_conn->smb1.sessions.validated_users;
183 struct id id;
185 if (!parse_id(msg, &id)) {
186 DEBUG(0, ("Invalid ?ID: %s\n", msg));
187 return;
190 if( do_kill && id_in_use(validated_users, &id) ) {
191 exit_server_cleanly(msg);
192 } else {
193 delete_from_cache(&id);
197 void msg_idmap_register_msgs(struct messaging_context *ctx)
199 messaging_register(ctx, NULL, MSG_IDMAP_FLUSH, message_idmap_flush);
200 messaging_register(ctx, NULL, MSG_IDMAP_DELETE, message_idmap_delete);
201 messaging_register(ctx, NULL, MSG_IDMAP_KILL, message_idmap_delete);