opendb: pass down open_disposition and break_to_none to odb_open_file()
[kai/samba.git] / source4 / ntvfs / common / opendb_tdb.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    Copyright (C) Andrew Tridgell 2004
5    Copyright (C) Stefan Metzmacher 2008
6
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 3 of the License, or
10    (at your option) any later version.
11    
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.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
19 */
20
21 /*
22   this is the open files database, tdb backend. It implements shared
23   storage of what files are open between server instances, and
24   implements the rules of shared access to files.
25
26   The caller needs to provide a file_key, which specifies what file
27   they are talking about. This needs to be a unique key across all
28   filesystems, and is usually implemented in terms of a device/inode
29   pair.
30
31   Before any operations can be performed the caller needs to establish
32   a lock on the record associated with file_key. That is done by
33   calling odb_lock(). The caller releases this lock by calling
34   talloc_free() on the returned handle.
35
36   All other operations on a record are done by passing the odb_lock()
37   handle back to this module. The handle contains internal
38   information about what file_key is being operated on.
39 */
40
41 #include "includes.h"
42 #include "system/filesys.h"
43 #include "lib/tdb/include/tdb.h"
44 #include "messaging/messaging.h"
45 #include "tdb_wrap.h"
46 #include "lib/messaging/irpc.h"
47 #include "librpc/gen_ndr/ndr_opendb.h"
48 #include "ntvfs/ntvfs.h"
49 #include "ntvfs/common/ntvfs_common.h"
50 #include "cluster/cluster.h"
51 #include "param/param.h"
52
53 struct odb_context {
54         struct tdb_wrap *w;
55         struct ntvfs_context *ntvfs_ctx;
56         bool oplocks;
57 };
58
59 /*
60   an odb lock handle. You must obtain one of these using odb_lock() before doing
61   any other operations. 
62 */
63 struct odb_lock {
64         struct odb_context *odb;
65         TDB_DATA key;
66 };
67
68 /*
69   Open up the openfiles.tdb database. Close it down using
70   talloc_free(). We need the messaging_ctx to allow for pending open
71   notifications.
72 */
73 static struct odb_context *odb_tdb_init(TALLOC_CTX *mem_ctx, 
74                                         struct ntvfs_context *ntvfs_ctx)
75 {
76         struct odb_context *odb;
77
78         odb = talloc(mem_ctx, struct odb_context);
79         if (odb == NULL) {
80                 return NULL;
81         }
82
83         odb->w = cluster_tdb_tmp_open(odb, ntvfs_ctx->lp_ctx, "openfiles.tdb", TDB_DEFAULT);
84         if (odb->w == NULL) {
85                 talloc_free(odb);
86                 return NULL;
87         }
88
89         odb->ntvfs_ctx = ntvfs_ctx;
90
91         /* leave oplocks disabled by default until the code is working */
92         odb->oplocks = lp_parm_bool(ntvfs_ctx->lp_ctx, NULL, "opendb", "oplocks", false);
93
94         return odb;
95 }
96
97 /*
98   destroy a lock on the database
99 */
100 static int odb_lock_destructor(struct odb_lock *lck)
101 {
102         tdb_chainunlock(lck->odb->w->tdb, lck->key);
103         return 0;
104 }
105
106 /*
107   get a lock on a entry in the odb. This call returns a lock handle,
108   which the caller should unlock using talloc_free().
109 */
110 static struct odb_lock *odb_tdb_lock(TALLOC_CTX *mem_ctx,
111                                      struct odb_context *odb, DATA_BLOB *file_key)
112 {
113         struct odb_lock *lck;
114
115         lck = talloc(mem_ctx, struct odb_lock);
116         if (lck == NULL) {
117                 return NULL;
118         }
119
120         lck->odb = talloc_reference(lck, odb);
121         lck->key.dptr = talloc_memdup(lck, file_key->data, file_key->length);
122         lck->key.dsize = file_key->length;
123         if (lck->key.dptr == NULL) {
124                 talloc_free(lck);
125                 return NULL;
126         }
127
128         if (tdb_chainlock(odb->w->tdb, lck->key) != 0) {
129                 talloc_free(lck);
130                 return NULL;
131         }
132
133         talloc_set_destructor(lck, odb_lock_destructor);
134         
135         return lck;
136 }
137
138 static DATA_BLOB odb_tdb_get_key(TALLOC_CTX *mem_ctx, struct odb_lock *lck)
139 {
140         return data_blob_talloc(mem_ctx, lck->key.dptr, lck->key.dsize);
141 }
142
143
144 /*
145   determine if two odb_entry structures conflict
146
147   return NT_STATUS_OK on no conflict
148 */
149 static NTSTATUS share_conflict(struct opendb_entry *e1,
150                                uint32_t stream_id,
151                                uint32_t share_access,
152                                uint32_t access_mask)
153 {
154         /* if either open involves no read.write or delete access then
155            it can't conflict */
156         if (!(e1->access_mask & (SEC_FILE_WRITE_DATA |
157                                  SEC_FILE_APPEND_DATA |
158                                  SEC_FILE_READ_DATA |
159                                  SEC_FILE_EXECUTE |
160                                  SEC_STD_DELETE))) {
161                 return NT_STATUS_OK;
162         }
163         if (!(access_mask & (SEC_FILE_WRITE_DATA |
164                              SEC_FILE_APPEND_DATA |
165                              SEC_FILE_READ_DATA |
166                              SEC_FILE_EXECUTE |
167                              SEC_STD_DELETE))) {
168                 return NT_STATUS_OK;
169         }
170
171         /* data IO access masks. This is skipped if the two open handles
172            are on different streams (as in that case the masks don't
173            interact) */
174         if (e1->stream_id != stream_id) {
175                 return NT_STATUS_OK;
176         }
177
178 #define CHECK_MASK(am, right, sa, share) \
179         if (((am) & (right)) && !((sa) & (share))) return NT_STATUS_SHARING_VIOLATION
180
181         CHECK_MASK(e1->access_mask, SEC_FILE_WRITE_DATA | SEC_FILE_APPEND_DATA,
182                    share_access, NTCREATEX_SHARE_ACCESS_WRITE);
183         CHECK_MASK(access_mask, SEC_FILE_WRITE_DATA | SEC_FILE_APPEND_DATA,
184                    e1->share_access, NTCREATEX_SHARE_ACCESS_WRITE);
185         
186         CHECK_MASK(e1->access_mask, SEC_FILE_READ_DATA | SEC_FILE_EXECUTE,
187                    share_access, NTCREATEX_SHARE_ACCESS_READ);
188         CHECK_MASK(access_mask, SEC_FILE_READ_DATA | SEC_FILE_EXECUTE,
189                    e1->share_access, NTCREATEX_SHARE_ACCESS_READ);
190
191         CHECK_MASK(e1->access_mask, SEC_STD_DELETE,
192                    share_access, NTCREATEX_SHARE_ACCESS_DELETE);
193         CHECK_MASK(access_mask, SEC_STD_DELETE,
194                    e1->share_access, NTCREATEX_SHARE_ACCESS_DELETE);
195 #undef CHECK_MASK
196         return NT_STATUS_OK;
197 }
198
199 /*
200   pull a record, translating from the db format to the opendb_file structure defined
201   in opendb.idl
202 */
203 static NTSTATUS odb_pull_record(struct odb_lock *lck, struct opendb_file *file)
204 {
205         struct odb_context *odb = lck->odb;
206         TDB_DATA dbuf;
207         DATA_BLOB blob;
208         enum ndr_err_code ndr_err;
209
210         dbuf = tdb_fetch(odb->w->tdb, lck->key);
211         if (dbuf.dptr == NULL) {
212                 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
213         }
214
215         blob.data = dbuf.dptr;
216         blob.length = dbuf.dsize;
217
218         ndr_err = ndr_pull_struct_blob(&blob, lck, lp_iconv_convenience(lck->odb->ntvfs_ctx->lp_ctx), file, (ndr_pull_flags_fn_t)ndr_pull_opendb_file);
219         free(dbuf.dptr);
220         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
221                 return ndr_map_error2ntstatus(ndr_err);
222         }
223
224         return NT_STATUS_OK;
225 }
226
227 /*
228   push a record, translating from the opendb_file structure defined in opendb.idl
229 */
230 static NTSTATUS odb_push_record(struct odb_lock *lck, struct opendb_file *file)
231 {
232         struct odb_context *odb = lck->odb;
233         TDB_DATA dbuf;
234         DATA_BLOB blob;
235         enum ndr_err_code ndr_err;
236         int ret;
237
238         if (file->num_entries == 0) {
239                 ret = tdb_delete(odb->w->tdb, lck->key);
240                 if (ret != 0) {
241                         return NT_STATUS_INTERNAL_DB_CORRUPTION;
242                 }
243                 return NT_STATUS_OK;
244         }
245
246         ndr_err = ndr_push_struct_blob(&blob, lck, lp_iconv_convenience(lck->odb->ntvfs_ctx->lp_ctx), file, (ndr_push_flags_fn_t)ndr_push_opendb_file);
247         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
248                 return ndr_map_error2ntstatus(ndr_err);
249         }
250
251         dbuf.dptr = blob.data;
252         dbuf.dsize = blob.length;
253                 
254         ret = tdb_store(odb->w->tdb, lck->key, dbuf, TDB_REPLACE);
255         data_blob_free(&blob);
256         if (ret != 0) {
257                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
258         }
259
260         return NT_STATUS_OK;
261 }
262
263 /*
264   send an oplock break to a client
265 */
266 static NTSTATUS odb_oplock_break_send(struct odb_context *odb,
267                                       struct opendb_entry *e,
268                                       uint8_t level)
269 {
270         NTSTATUS status;
271         struct opendb_oplock_break op_break;
272         DATA_BLOB blob;
273
274         ZERO_STRUCT(op_break);
275
276         /* tell the server handling this open file about the need to send the client
277            a break */
278         op_break.file_handle    = e->file_handle;
279         op_break.level          = level;
280
281         blob = data_blob_const(&op_break, sizeof(op_break));
282
283         status = messaging_send(odb->ntvfs_ctx->msg_ctx, e->server,
284                                 MSG_NTVFS_OPLOCK_BREAK, &blob);
285         NT_STATUS_NOT_OK_RETURN(status);
286
287         return NT_STATUS_OK;
288 }
289
290 static bool access_attributes_only(uint32_t access_mask,
291                                    uint32_t open_disposition)
292 {
293         switch (open_disposition) {
294         case NTCREATEX_DISP_SUPERSEDE:
295         case NTCREATEX_DISP_OVERWRITE_IF:
296         case NTCREATEX_DISP_OVERWRITE:
297                 return false;
298         default:
299                 break;
300         }
301 #define CHECK_MASK(m,g) ((m) && (((m) & ~(g))==0) && (((m) & (g)) != 0))
302         return CHECK_MASK(access_mask,
303                           SEC_STD_SYNCHRONIZE |
304                           SEC_FILE_READ_ATTRIBUTE |
305                           SEC_FILE_WRITE_ATTRIBUTE);
306 #undef CHECK_MASK
307 }
308
309 static NTSTATUS odb_tdb_open_can_internal(struct odb_context *odb,
310                                           const struct opendb_file *file,
311                                           uint32_t stream_id, uint32_t share_access,
312                                           uint32_t access_mask, bool delete_on_close,
313                                           uint32_t open_disposition, bool break_to_none,
314                                           bool *_attrs_only)
315 {
316         NTSTATUS status;
317         uint32_t i;
318         bool attrs_only = false;
319
320         /* see if anyone has an oplock, which we need to break */
321         for (i=0;i<file->num_entries;i++) {
322                 if (file->entries[i].oplock_level == OPLOCK_BATCH) {
323                         bool oplock_return = OPLOCK_BREAK_TO_LEVEL_II;
324                         /* if this is an attribute only access
325                          * it doesn't conflict with a BACTCH oplock
326                          * but we'll not grant the oplock below
327                          */
328                         attrs_only = access_attributes_only(access_mask,
329                                                             open_disposition);
330                         if (attrs_only) {
331                                 break;
332                         }
333                         /* a batch oplock caches close calls, which
334                            means the client application might have
335                            already closed the file. We have to allow
336                            this close to propogate by sending a oplock
337                            break request and suspending this call
338                            until the break is acknowledged or the file
339                            is closed */
340                         if (break_to_none) {
341                                 oplock_return = OPLOCK_BREAK_TO_NONE;
342                         }
343                         odb_oplock_break_send(odb, &file->entries[i],
344                                               oplock_return);
345                         return NT_STATUS_OPLOCK_NOT_GRANTED;
346                 }
347         }
348
349         if (file->delete_on_close) {
350                 /* while delete on close is set, no new opens are allowed */
351                 return NT_STATUS_DELETE_PENDING;
352         }
353
354         if (file->num_entries != 0 && delete_on_close) {
355                 return NT_STATUS_SHARING_VIOLATION;
356         }
357
358         /* check for sharing violations */
359         for (i=0;i<file->num_entries;i++) {
360                 status = share_conflict(&file->entries[i], stream_id,
361                                         share_access, access_mask);
362                 NT_STATUS_NOT_OK_RETURN(status);
363         }
364
365         /* we now know the open could succeed, but we need to check
366            for any exclusive oplocks. We can't grant a second open
367            till these are broken. Note that we check for batch oplocks
368            before checking for sharing violations, and check for
369            exclusive oplocks afterwards. */
370         for (i=0;i<file->num_entries;i++) {
371                 if (file->entries[i].oplock_level == OPLOCK_EXCLUSIVE) {
372                         odb_oplock_break_send(odb, &file->entries[i],
373                                               OPLOCK_BREAK_TO_NONE);
374                         return NT_STATUS_OPLOCK_NOT_GRANTED;
375                 }
376         }
377
378         if (_attrs_only) {
379                 *_attrs_only = attrs_only;
380         }
381         return NT_STATUS_OK;
382 }
383
384 /*
385   register an open file in the open files database. This implements the share_access
386   rules
387
388   Note that the path is only used by the delete on close logic, not
389   for comparing with other filenames
390 */
391 static NTSTATUS odb_tdb_open_file(struct odb_lock *lck,
392                                   void *file_handle, const char *path,
393                                   uint32_t stream_id, uint32_t share_access,
394                                   uint32_t access_mask, bool delete_on_close,
395                                   uint32_t open_disposition, bool break_to_none,
396                                   uint32_t oplock_level, uint32_t *oplock_granted)
397 {
398         struct odb_context *odb = lck->odb;
399         struct opendb_entry e;
400         struct opendb_file file;
401         NTSTATUS status;
402         bool attrs_only = false;
403
404         if (odb->oplocks == false) {
405                 oplock_level = OPLOCK_NONE;
406         }
407
408         status = odb_pull_record(lck, &file);
409         if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
410                 /* initialise a blank structure */
411                 ZERO_STRUCT(file);
412                 file.path = path;
413         } else {
414                 NT_STATUS_NOT_OK_RETURN(status);
415         }
416
417         /* see if it conflicts */
418         status = odb_tdb_open_can_internal(odb, &file, stream_id,
419                                            share_access, access_mask,
420                                            delete_on_close, open_disposition,
421                                            break_to_none, &attrs_only);
422         NT_STATUS_NOT_OK_RETURN(status);
423
424         /* see if it conflicts */
425         e.server          = odb->ntvfs_ctx->server_id;
426         e.file_handle     = file_handle;
427         e.stream_id       = stream_id;
428         e.share_access    = share_access;
429         e.access_mask     = access_mask;
430         e.delete_on_close = delete_on_close;
431         e.oplock_level    = OPLOCK_NONE;
432
433         /*
434           possibly grant an exclusive, batch or level2 oplock
435         */
436         if (oplock_granted) {
437                 if (attrs_only) {
438                         e.oplock_level  = OPLOCK_NONE;
439                         *oplock_granted = NO_OPLOCK_RETURN;
440                 } else if (oplock_level == OPLOCK_EXCLUSIVE) {
441                         if (file.num_entries == 0) {
442                                 e.oplock_level  = OPLOCK_EXCLUSIVE;
443                                 *oplock_granted = EXCLUSIVE_OPLOCK_RETURN;
444                         } else {
445                                 e.oplock_level  = OPLOCK_NONE;
446                                 *oplock_granted = NO_OPLOCK_RETURN;
447                         }
448                 } else if (oplock_level == OPLOCK_BATCH) {
449                         if (file.num_entries == 0) {
450                                 e.oplock_level  = OPLOCK_BATCH;
451                                 *oplock_granted = BATCH_OPLOCK_RETURN;
452                         } else {
453                                 e.oplock_level  = OPLOCK_LEVEL_II;
454                                 *oplock_granted = LEVEL_II_OPLOCK_RETURN;
455                         }
456                 } else if (oplock_level == OPLOCK_LEVEL_II) {
457                         e.oplock_level  = OPLOCK_LEVEL_II;
458                         *oplock_granted = LEVEL_II_OPLOCK_RETURN;
459                 } else {
460                         e.oplock_level  = OPLOCK_NONE;
461                         *oplock_granted = NO_OPLOCK_RETURN;
462                 }
463         }
464
465         /* it doesn't conflict, so add it to the end */
466         file.entries = talloc_realloc(lck, file.entries, struct opendb_entry, 
467                                       file.num_entries+1);
468         NT_STATUS_HAVE_NO_MEMORY(file.entries);
469
470         file.entries[file.num_entries] = e;
471         file.num_entries++;
472
473         return odb_push_record(lck, &file);
474 }
475
476
477 /*
478   register a pending open file in the open files database
479 */
480 static NTSTATUS odb_tdb_open_file_pending(struct odb_lock *lck, void *private)
481 {
482         struct odb_context *odb = lck->odb;
483         struct opendb_file file;
484         NTSTATUS status;
485                 
486         status = odb_pull_record(lck, &file);
487         NT_STATUS_NOT_OK_RETURN(status);
488
489         file.pending = talloc_realloc(lck, file.pending, struct opendb_pending, 
490                                       file.num_pending+1);
491         NT_STATUS_HAVE_NO_MEMORY(file.pending);
492
493         file.pending[file.num_pending].server = odb->ntvfs_ctx->server_id;
494         file.pending[file.num_pending].notify_ptr = private;
495
496         file.num_pending++;
497
498         return odb_push_record(lck, &file);
499 }
500
501
502 /*
503   remove a opendb entry
504 */
505 static NTSTATUS odb_tdb_close_file(struct odb_lock *lck, void *file_handle)
506 {
507         struct odb_context *odb = lck->odb;
508         struct opendb_file file;
509         int i;
510         NTSTATUS status;
511
512         status = odb_pull_record(lck, &file);
513         NT_STATUS_NOT_OK_RETURN(status);
514
515         /* find the entry, and delete it */
516         for (i=0;i<file.num_entries;i++) {
517                 if (file_handle == file.entries[i].file_handle &&
518                     cluster_id_equal(&odb->ntvfs_ctx->server_id, &file.entries[i].server)) {
519                         if (file.entries[i].delete_on_close) {
520                                 file.delete_on_close = true;
521                         }
522                         if (i < file.num_entries-1) {
523                                 memmove(file.entries+i, file.entries+i+1, 
524                                         (file.num_entries - (i+1)) * 
525                                         sizeof(struct opendb_entry));
526                         }
527                         break;
528                 }
529         }
530
531         if (i == file.num_entries) {
532                 return NT_STATUS_UNSUCCESSFUL;
533         }
534
535         /* send any pending notifications, removing them once sent */
536         for (i=0;i<file.num_pending;i++) {
537                 messaging_send_ptr(odb->ntvfs_ctx->msg_ctx, file.pending[i].server, 
538                                    MSG_PVFS_RETRY_OPEN, 
539                                    file.pending[i].notify_ptr);
540         }
541         file.num_pending = 0;
542
543         file.num_entries--;
544         
545         return odb_push_record(lck, &file);
546 }
547
548 /*
549   update the oplock level of the client
550 */
551 static NTSTATUS odb_tdb_update_oplock(struct odb_lock *lck, void *file_handle,
552                                       uint32_t oplock_level)
553 {
554         struct odb_context *odb = lck->odb;
555         struct opendb_file file;
556         int i;
557         NTSTATUS status;
558
559         status = odb_pull_record(lck, &file);
560         NT_STATUS_NOT_OK_RETURN(status);
561
562         /* find the entry, and update it */
563         for (i=0;i<file.num_entries;i++) {
564                 if (file_handle == file.entries[i].file_handle &&
565                     cluster_id_equal(&odb->ntvfs_ctx->server_id, &file.entries[i].server)) {
566                         file.entries[i].oplock_level = oplock_level;
567                         break;
568                 }
569         }
570
571         if (i == file.num_entries) {
572                 return NT_STATUS_UNSUCCESSFUL;
573         }
574
575         /* send any pending notifications, removing them once sent */
576         for (i=0;i<file.num_pending;i++) {
577                 messaging_send_ptr(odb->ntvfs_ctx->msg_ctx,
578                                    file.pending[i].server,
579                                    MSG_PVFS_RETRY_OPEN,
580                                    file.pending[i].notify_ptr);
581         }
582         file.num_pending = 0;
583
584         return odb_push_record(lck, &file);
585 }
586
587 /*
588   send oplocks breaks to none to all level2 holders
589 */
590 static NTSTATUS odb_tdb_break_oplocks(struct odb_lock *lck)
591 {
592         struct odb_context *odb = lck->odb;
593         NTSTATUS status;
594         struct opendb_file file;
595         int i;
596         bool modified = true;
597
598         status = odb_pull_record(lck, &file);
599         if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
600                 return NT_STATUS_OK;
601         }
602         NT_STATUS_NOT_OK_RETURN(status);
603
604         /* see if anyone has an oplock, which we need to break */
605         for (i=0;i<file.num_entries;i++) {
606                 if (file.entries[i].oplock_level == OPLOCK_LEVEL_II) {
607                         /*
608                          * there could be multiple level2 oplocks
609                          * and we just send a break to none to all of them
610                          * without waiting for a release
611                          */
612                         odb_oplock_break_send(odb, &file.entries[i],
613                                               OPLOCK_BREAK_TO_NONE);
614                         file.entries[i].oplock_level = OPLOCK_NONE;
615                         modified = true;
616                 }
617         }
618
619         if (modified) {
620                 return odb_push_record(lck, &file);
621         }
622         return NT_STATUS_OK;
623 }
624
625 /*
626   remove a pending opendb entry
627 */
628 static NTSTATUS odb_tdb_remove_pending(struct odb_lock *lck, void *private)
629 {
630         struct odb_context *odb = lck->odb;
631         int i;
632         NTSTATUS status;
633         struct opendb_file file;
634
635         status = odb_pull_record(lck, &file);
636         NT_STATUS_NOT_OK_RETURN(status);
637
638         /* find the entry, and delete it */
639         for (i=0;i<file.num_pending;i++) {
640                 if (private == file.pending[i].notify_ptr &&
641                     cluster_id_equal(&odb->ntvfs_ctx->server_id, &file.pending[i].server)) {
642                         if (i < file.num_pending-1) {
643                                 memmove(file.pending+i, file.pending+i+1, 
644                                         (file.num_pending - (i+1)) * 
645                                         sizeof(struct opendb_pending));
646                         }
647                         break;
648                 }
649         }
650
651         if (i == file.num_pending) {
652                 return NT_STATUS_UNSUCCESSFUL;
653         }
654
655         file.num_pending--;
656         
657         return odb_push_record(lck, &file);
658 }
659
660
661 /*
662   rename the path in a open file
663 */
664 static NTSTATUS odb_tdb_rename(struct odb_lock *lck, const char *path)
665 {
666         struct opendb_file file;
667         NTSTATUS status;
668
669         status = odb_pull_record(lck, &file);
670         if (NT_STATUS_EQUAL(NT_STATUS_OBJECT_NAME_NOT_FOUND, status)) {
671                 /* not having the record at all is OK */
672                 return NT_STATUS_OK;
673         }
674         NT_STATUS_NOT_OK_RETURN(status);
675
676         file.path = path;
677         return odb_push_record(lck, &file);
678 }
679
680 /*
681   update delete on close flag on an open file
682 */
683 static NTSTATUS odb_tdb_set_delete_on_close(struct odb_lock *lck, bool del_on_close)
684 {
685         NTSTATUS status;
686         struct opendb_file file;
687
688         status = odb_pull_record(lck, &file);
689         NT_STATUS_NOT_OK_RETURN(status);
690
691         file.delete_on_close = del_on_close;
692
693         return odb_push_record(lck, &file);
694 }
695
696 /*
697   return the current value of the delete_on_close bit, and how many
698   people still have the file open
699 */
700 static NTSTATUS odb_tdb_get_delete_on_close(struct odb_context *odb, 
701                                             DATA_BLOB *key, bool *del_on_close, 
702                                             int *open_count, char **path)
703 {
704         NTSTATUS status;
705         struct opendb_file file;
706         struct odb_lock *lck;
707
708         lck = odb_lock(odb, odb, key);
709         NT_STATUS_HAVE_NO_MEMORY(lck);
710
711         status = odb_pull_record(lck, &file);
712         if (NT_STATUS_EQUAL(NT_STATUS_OBJECT_NAME_NOT_FOUND, status)) {
713                 talloc_free(lck);
714                 (*del_on_close) = false;
715                 return NT_STATUS_OK;
716         }
717         if (!NT_STATUS_IS_OK(status)) {
718                 talloc_free(lck);
719                 return status;
720         }
721
722         (*del_on_close) = file.delete_on_close;
723         if (open_count != NULL) {
724                 (*open_count) = file.num_entries;
725         }
726         if (path != NULL) {
727                 *path = talloc_strdup(odb, file.path);
728                 NT_STATUS_HAVE_NO_MEMORY(*path);
729                 if (file.num_entries == 1 && file.entries[0].delete_on_close) {
730                         (*del_on_close) = true;
731                 }
732         }
733
734         talloc_free(lck);
735
736         return NT_STATUS_OK;
737 }
738
739
740 /*
741   determine if a file can be opened with the given share_access,
742   create_options and access_mask
743 */
744 static NTSTATUS odb_tdb_can_open(struct odb_lock *lck,
745                                  uint32_t share_access, uint32_t create_options, 
746                                  uint32_t access_mask)
747 {
748         struct odb_context *odb = lck->odb;
749         NTSTATUS status;
750         struct opendb_file file;
751         uint32_t stream_id = 0;
752         uint32_t open_disposition = 0;
753         bool delete_on_close = false;
754         bool break_to_none = false;
755         bool attrs_only = false;
756
757         status = odb_pull_record(lck, &file);
758         if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
759                 return NT_STATUS_OK;
760         }
761         NT_STATUS_NOT_OK_RETURN(status);
762
763         if (create_options & NTCREATEX_OPTIONS_DELETE_ON_CLOSE) {
764                 delete_on_close = true;
765         }
766
767         status = odb_tdb_open_can_internal(odb, &file, stream_id,
768                                            share_access, access_mask,
769                                            delete_on_close, open_disposition,
770                                            break_to_none, &attrs_only);
771         NT_STATUS_NOT_OK_RETURN(status);
772
773         return NT_STATUS_OK;
774 }
775
776
777 static const struct opendb_ops opendb_tdb_ops = {
778         .odb_init                = odb_tdb_init,
779         .odb_lock                = odb_tdb_lock,
780         .odb_get_key             = odb_tdb_get_key,
781         .odb_open_file           = odb_tdb_open_file,
782         .odb_open_file_pending   = odb_tdb_open_file_pending,
783         .odb_close_file          = odb_tdb_close_file,
784         .odb_remove_pending      = odb_tdb_remove_pending,
785         .odb_rename              = odb_tdb_rename,
786         .odb_set_delete_on_close = odb_tdb_set_delete_on_close,
787         .odb_get_delete_on_close = odb_tdb_get_delete_on_close,
788         .odb_can_open            = odb_tdb_can_open,
789         .odb_update_oplock       = odb_tdb_update_oplock,
790         .odb_break_oplocks       = odb_tdb_break_oplocks
791 };
792
793
794 void odb_tdb_init_ops(void)
795 {
796         odb_set_ops(&opendb_tdb_ops);
797 }