2 * Unix SMB/CIFS implementation.
4 * Copyright (C) Volker Lendecke 2015
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/>.
21 #include "smbd_cleanupd.h"
22 #include "lib/util_procid.h"
23 #include "lib/util/tevent_ntstatus.h"
24 #include "lib/util/debug.h"
25 #include "smbprofile.h"
27 #include "locking/proto.h"
28 #include "cleanupdb.h"
30 struct smbd_cleanupd_state
{
34 static void smbd_cleanupd_shutdown(struct messaging_context
*msg
,
35 void *private_data
, uint32_t msg_type
,
36 struct server_id server_id
,
38 static void smbd_cleanupd_process_exited(struct messaging_context
*msg
,
39 void *private_data
, uint32_t msg_type
,
40 struct server_id server_id
,
42 static void smbd_cleanupd_unlock(struct messaging_context
*msg
,
43 void *private_data
, uint32_t msg_type
,
44 struct server_id server_id
,
47 struct tevent_req
*smbd_cleanupd_send(TALLOC_CTX
*mem_ctx
,
48 struct tevent_context
*ev
,
49 struct messaging_context
*msg
,
52 struct tevent_req
*req
;
53 struct smbd_cleanupd_state
*state
;
56 req
= tevent_req_create(mem_ctx
, &state
, struct smbd_cleanupd_state
);
60 state
->parent_pid
= parent_pid
;
62 status
= messaging_register(msg
, req
, MSG_SHUTDOWN
,
63 smbd_cleanupd_shutdown
);
64 if (tevent_req_nterror(req
, status
)) {
65 return tevent_req_post(req
, ev
);
68 status
= messaging_register(msg
, req
, MSG_SMB_NOTIFY_CLEANUP
,
69 smbd_cleanupd_process_exited
);
70 if (tevent_req_nterror(req
, status
)) {
71 return tevent_req_post(req
, ev
);
74 status
= messaging_register(msg
, NULL
, MSG_SMB_BRL_VALIDATE
,
75 smbd_cleanupd_unlock
);
76 if (tevent_req_nterror(req
, status
)) {
77 return tevent_req_post(req
, ev
);
83 static void smbd_cleanupd_shutdown(struct messaging_context
*msg
,
84 void *private_data
, uint32_t msg_type
,
85 struct server_id server_id
,
88 struct tevent_req
*req
= talloc_get_type_abort(
89 private_data
, struct tevent_req
);
93 static void smbd_cleanupd_unlock(struct messaging_context
*msg
,
94 void *private_data
, uint32_t msg_type
,
95 struct server_id server_id
,
98 DBG_WARNING("Cleaning up brl and lock database after unclean "
101 brl_revalidate(msg
, private_data
, msg_type
, server_id
, data
);
104 struct cleanup_child
{
105 struct cleanup_child
*prev
, *next
;
110 struct cleanupdb_traverse_state
{
113 struct cleanup_child
*childs
;
116 static int cleanupdb_traverse_fn(const pid_t pid
,
120 struct cleanupdb_traverse_state
*cleanup_state
=
121 (struct cleanupdb_traverse_state
*)private_data
;
122 struct cleanup_child
*child
= NULL
;
124 child
= talloc_zero(cleanup_state
->mem_ctx
, struct cleanup_child
);
126 DBG_ERR("talloc_zero failed\n");
131 child
->unclean
= unclean
;
132 DLIST_ADD(cleanup_state
->childs
, child
);
137 static void smbd_cleanupd_process_exited(struct messaging_context
*msg
,
138 void *private_data
, uint32_t msg_type
,
139 struct server_id server_id
,
142 struct tevent_req
*req
= talloc_get_type_abort(
143 private_data
, struct tevent_req
);
144 struct smbd_cleanupd_state
*state
= tevent_req_data(
145 req
, struct smbd_cleanupd_state
);
147 struct cleanupdb_traverse_state cleanup_state
;
148 TALLOC_CTX
*frame
= talloc_stackframe();
149 struct cleanup_child
*child
= NULL
;
151 cleanup_state
= (struct cleanupdb_traverse_state
) {
156 * This merely collect childs in a list, whatever we're
157 * supposed to cleanup for every child, it has to take place
158 * *after* the db traverse in a list loop. This is to minimize
159 * locking interaction between the traverse and writers (ie
162 ret
= cleanupdb_traverse_read(cleanupdb_traverse_fn
, &cleanup_state
);
164 DBG_ERR("cleanupdb_traverse_read failed\n");
174 for (child
= cleanup_state
.childs
;
180 ok
= cleanupdb_delete_child(child
->pid
);
182 DBG_ERR("failed to delete pid %d\n", (int)child
->pid
);
185 smbprofile_cleanup(child
->pid
, state
->parent_pid
);
187 ret
= messaging_cleanup(msg
, child
->pid
);
189 if ((ret
!= 0) && (ret
!= ENOENT
)) {
190 DBG_DEBUG("messaging_cleanup returned %s\n",
194 DBG_DEBUG("cleaned up pid %d\n", (int)child
->pid
);
200 NTSTATUS
smbd_cleanupd_recv(struct tevent_req
*req
)
202 return tevent_req_simple_recv_ntstatus(req
);