s4-kcc: Sets the isRecyled attributes on deleted entries when needed
[mat/samba.git] / source4 / dsdb / kcc / kcc_deleted.c
1 /*
2    Unix SMB/CIFS implementation.
3
4    handle removal of deleted objects
5
6    Copyright (C) 2009 Andrew Tridgell
7
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 3 of the License, or
11    (at your option) any later version.
12
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17
18    You should have received a copy of the GNU General Public License
19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
20
21 */
22
23 #include "includes.h"
24 #include "lib/events/events.h"
25 #include "dsdb/samdb/samdb.h"
26 #include "auth/auth.h"
27 #include "smbd/service.h"
28 #include "lib/messaging/irpc.h"
29 #include "dsdb/kcc/kcc_connection.h"
30 #include "dsdb/kcc/kcc_service.h"
31 #include <ldb_errors.h>
32 #include "../lib/util/dlinklist.h"
33 #include "librpc/gen_ndr/ndr_misc.h"
34 #include "librpc/gen_ndr/ndr_drsuapi.h"
35 #include "librpc/gen_ndr/ndr_drsblobs.h"
36 #include "param/param.h"
37 #include "dsdb/common/util.h"
38
39 /*
40   check to see if any deleted objects need scavenging
41  */
42 NTSTATUS kccsrv_check_deleted(struct kccsrv_service *s, TALLOC_CTX *mem_ctx)
43 {
44         struct kccsrv_partition *part;
45         int ret;
46         uint32_t tombstoneLifetime;
47         const struct dsdb_schema *schema;
48         bool add_recycled = false;
49         bool rcbin_enabled = false;
50
51         time_t t = time(NULL);
52         if (t - s->last_deleted_check < lpcfg_parm_int(s->task->lp_ctx, NULL, "kccsrv",
53                                                     "check_deleted_interval", 600)) {
54                 return NT_STATUS_OK;
55         }
56         s->last_deleted_check = t;
57         DEBUG(11, ("Check deleted has kicked in !\n"));
58
59         ret = dsdb_tombstone_lifetime(s->samdb, &tombstoneLifetime);
60         if (ret != LDB_SUCCESS) {
61                 DEBUG(1,(__location__ ": Failed to get tombstone lifetime\n"));
62                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
63         }
64
65         schema = dsdb_get_schema(s->samdb, mem_ctx);
66         if (!schema) {
67                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
68         }
69
70         if (dsdb_attribute_by_lDAPDisplayName(schema, "isRecycled") != NULL) {
71                 add_recycled = true;
72         }
73
74         if (dsdb_functional_level(s->samdb) >= DS_DOMAIN_FUNCTION_2008_R2) {
75                 ret = samdb_recyclebin_enabled(s->samdb , &rcbin_enabled);
76                 if (ret != LDB_SUCCESS) {
77                         return NT_STATUS_INTERNAL_DB_CORRUPTION;
78                 }
79         }
80
81         for (part=s->partitions; part; part=part->next) {
82                 struct ldb_dn *do_dn;
83                 struct ldb_result *res;
84                 const char *attrs[] = { "whenChanged", "isRecycled", NULL };
85                 unsigned int i;
86
87                 ret = dsdb_get_deleted_objects_dn(s->samdb, mem_ctx, part->dn, &do_dn);
88                 if (ret != LDB_SUCCESS) {
89                         /* some partitions have no Deleted Objects
90                            container */
91                         continue;
92                 }
93                 ret = dsdb_search(s->samdb, do_dn, &res, do_dn, LDB_SCOPE_ONELEVEL, attrs,
94                                   DSDB_SEARCH_SHOW_RECYCLED, NULL);
95
96                 if (ret != LDB_SUCCESS) {
97                         DEBUG(1,(__location__ ": Failed to search for deleted objects in %s\n",
98                                  ldb_dn_get_linearized(do_dn)));
99                         talloc_free(do_dn);
100                         continue;
101                 }
102
103                 for (i=0; i<res->count; i++) {
104                         const char *tstring;
105                         time_t whenChanged = 0;
106
107                         if (ldb_dn_compare(do_dn, res->msgs[i]->dn) == 0) {
108                                 /* Skip the Deleted Object Container */
109                                 continue;
110                         }
111                         tstring = ldb_msg_find_attr_as_string(res->msgs[i], "whenChanged", NULL);
112                         if (tstring) {
113                                 whenChanged = ldb_string_to_time(tstring);
114                         }
115                         if (t - whenChanged > tombstoneLifetime*60*60*24) {
116                                 ret = dsdb_delete(s->samdb, res->msgs[i]->dn, DSDB_SEARCH_SHOW_DELETED);
117                                 if (ret != LDB_SUCCESS) {
118                                         DEBUG(1,(__location__ ": Failed to remove deleted object %s\n",
119                                                  ldb_dn_get_linearized(res->msgs[i]->dn)));
120                                 } else {
121                                         DEBUG(4,("Removed deleted object %s\n",
122                                                  ldb_dn_get_linearized(res->msgs[i]->dn)));
123                                 }
124                         } else if (add_recycled & !ldb_msg_find_ldb_val(res->msgs[i], "isRecycled") && !rcbin_enabled) {
125                                 /* There is no isRecycled attribute and the recycle-bin
126                                  * is not enabled so set isRecycled to TRUE.
127                                  * We used to not set this attribute before
128                                  * let's set it
129                                  */
130
131                                 DEBUG(1, ("Adding isRecycled\n"));
132                                 struct ldb_message *msg = ldb_msg_new(mem_ctx);
133                                 if (msg == NULL) {
134                                         talloc_free(do_dn);
135                                         return NT_STATUS_NO_MEMORY;
136                                 }
137
138                                 msg->dn = res->msgs[i]->dn;
139                                 ret = ldb_msg_add_string(msg, "isRecycled", "TRUE");
140                                 if (ret != LDB_SUCCESS) {
141                                         talloc_free(do_dn);
142                                         return NT_STATUS_NO_MEMORY;
143                                 }
144                                 msg->elements[msg->num_elements - 1].flags = LDB_FLAG_MOD_ADD;
145
146                                 ret = dsdb_modify(s->samdb, msg, DSDB_SEARCH_SHOW_DELETED);
147                                 if (ret != LDB_SUCCESS) {
148                                         talloc_free(do_dn);
149                                         return NT_STATUS_INTERNAL_DB_CORRUPTION;
150                                 }
151                         }
152                 }
153
154                 talloc_free(do_dn);
155         }
156
157         return NT_STATUS_OK;
158 }