2 * Unix SMB/CIFS implementation.
3 * Support for OneFS system interfaces.
5 * Copyright (C) Zack Kirsch, 2009
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.
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, see <http://www.gnu.org/licenses/>.
22 #include "smbd/smbd.h"
25 #include <ifs/ifs_syscalls.h>
26 #include <sys/isi_cifs_brl.h>
27 #include <isi_ecs/isi_ecs_cbrl.h>
30 #define DBGC_CLASS DBGC_LOCKING
32 static uint64_t onefs_get_new_id(void) {
33 static uint64_t id = 0;
40 enum onefs_cbrl_lock_state {ONEFS_CBRL_NONE, ONEFS_CBRL_ASYNC, ONEFS_CBRL_DONE,
43 struct onefs_cbrl_blr_state {
45 enum onefs_cbrl_lock_state state;
48 static char *onefs_cbrl_blr_state_str(const struct blocking_lock_record *blr)
50 static fstring result;
51 struct onefs_cbrl_blr_state *bs;
55 bs = (struct onefs_cbrl_blr_state *)blr->blr_private;
58 fstrcpy(result, "NULL CBRL BLR state - Posix lock?");
64 fstr_sprintf(result, "CBRL BLR id=%llu: state=NONE", bs->id);
66 case ONEFS_CBRL_ASYNC:
67 fstr_sprintf(result, "CBRL BLR id=%llu: state=ASYNC", bs->id);
70 fstr_sprintf(result, "CBRL BLR id=%llu: state=DONE", bs->id);
72 case ONEFS_CBRL_ERROR:
73 fstr_sprintf(result, "CBRL BLR id=%llu: state=ERROR", bs->id);
76 fstr_sprintf(result, "CBRL BLR id=%llu: unknown state %d",
84 static void onefs_cbrl_enumerate_blq(const char *fn)
86 struct smbd_server_connection *sconn = smbd_server_conn;
87 struct blocking_lock_record *blr;
92 DEBUG(10, ("CBRL BLR records (%s):\n", fn));
94 if (sconn->using_smb2) {
95 struct smbd_smb2_request *smb2req;
96 for (smb2req = sconn->smb2.requests; smb2req; smb2req = nextreq) {
97 blr = get_pending_smb2req_blr(smb2req);
99 DEBUGADD(10, ("%s\n", onefs_cbrl_blr_state_str(blr)));
103 for (blr = sconn->smb1.locks.blocking_lock_queue; blr; blr = blr->next)
104 DEBUGADD(10, ("%s\n", onefs_cbrl_blr_state_str(blr)));
108 static struct blocking_lock_record *onefs_cbrl_find_blr(uint64_t id)
110 struct smbd_server_connection *sconn = smbd_server_conn;
111 struct blocking_lock_record *blr;
112 struct onefs_cbrl_blr_state *bs;
114 onefs_cbrl_enumerate_blq("onefs_cbrl_find_blr");
116 if (sconn->using_smb2) {
117 struct smbd_smb2_request *smb2req;
118 for (smb2req = sconn->smb2.requests; smb2req; smb2req = nextreq) {
119 blr = get_pending_smb2req_blr(smb2req);
123 bs = (struct onefs_cbrl_blr_state *)blr->blr_private;
128 DEBUG(10, ("found %s\n",
129 onefs_cbrl_blr_state_str(blr)));
134 for (blr = sconn->smb1.locks.blocking_lock_queue; blr; blr = blr->next) {
135 bs = (struct onefs_cbrl_blr_state *)blr->blr_private;
137 /* We don't control all of the BLRs on the BLQ. */
142 DEBUG(10, ("found %s\n",
143 onefs_cbrl_blr_state_str(blr)));
150 DEBUG(5, ("Could not find CBRL BLR for id %llu\n", id));
157 static void onefs_cbrl_async_success(uint64_t id)
159 struct blocking_lock_record *blr;
160 struct onefs_cbrl_blr_state *bs;
163 DEBUG(10, ("CBRL async success!\n"));
165 /* Find BLR with id. Its okay not to find one (race with cancel) */
166 blr = onefs_cbrl_find_blr(id);
170 bs = (struct onefs_cbrl_blr_state *)blr->blr_private;
172 SMB_ASSERT(bs->state == ONEFS_CBRL_ASYNC);
176 num_locks = SVAL(blr->req->vwv+7, 0);
178 if (blr->lock_num == num_locks)
179 bs->state = ONEFS_CBRL_DONE;
181 bs->state = ONEFS_CBRL_NONE;
183 /* Self contend our own level 2 oplock. The kernel handles
184 * contention of other opener's level 2 oplocks. */
185 contend_level2_oplocks_begin(blr->fsp,
186 LEVEL2_CONTEND_WINDOWS_BRL);
188 /* Process the queue, to try the next lock or finish up. */
189 process_blocking_lock_queue(smbd_server_conn);
192 static void onefs_cbrl_async_failure(uint64_t id)
194 struct blocking_lock_record *blr;
195 struct onefs_cbrl_blr_state *bs;
197 DEBUG(10, ("CBRL async failure!\n"));
199 /* Find BLR with id. Its okay not to find one (race with cancel) */
200 blr = onefs_cbrl_find_blr(id);
204 bs = (struct onefs_cbrl_blr_state *)blr->blr_private;
207 SMB_ASSERT(bs->state == ONEFS_CBRL_ASYNC);
208 bs->state = ONEFS_CBRL_ERROR;
210 /* Process the queue. It will end up trying to retake the same lock,
211 * see the error in onefs_cbrl_lock_windows() and fail. */
212 process_blocking_lock_queue(smbd_server_conn);
215 static struct cbrl_event_ops cbrl_ops =
216 {.cbrl_async_success = onefs_cbrl_async_success,
217 .cbrl_async_failure = onefs_cbrl_async_failure};
219 static void onefs_cbrl_events_handler(struct event_context *ev,
220 struct fd_event *fde,
224 DEBUG(10, ("onefs_cbrl_events_handler\n"));
226 if (cbrl_event_dispatcher(&cbrl_ops)) {
227 DEBUG(0, ("cbrl_event_dispatcher failed: %s\n",
232 static void onefs_init_cbrl(void)
234 static bool init_done = false;
235 static int cbrl_event_fd;
236 static struct fd_event *cbrl_fde;
241 DEBUG(10, ("onefs_init_cbrl\n"));
243 /* Register the event channel for CBRL. */
244 cbrl_event_fd = cbrl_event_register();
245 if (cbrl_event_fd == -1) {
246 DEBUG(0, ("cbrl_event_register failed: %s\n",
251 DEBUG(10, ("cbrl_event_fd = %d\n", cbrl_event_fd));
253 /* Register the CBRL event_fd with samba's event system */
254 cbrl_fde = event_add_fd(server_event_context(),
258 onefs_cbrl_events_handler,
266 * Blocking PID. As far as I can tell, the blocking_pid is only used to tell
267 * whether a posix lock or a CIFS lock blocked us. If it was a posix lock,
268 * Samba polls every 10 seconds, which we don't want. -zkirsch
270 #define ONEFS_BLOCKING_PID 0xABCDABCD
273 * @param[in] br_lck Contains the fsp.
274 * @param[in] plock Lock request.
275 * @param[in] blocking_lock Only used for figuring out the error.
276 * @param[in,out] blr The BLR for the already-deferred operation.
278 NTSTATUS onefs_brl_lock_windows(vfs_handle_struct *handle,
279 struct byte_range_lock *br_lck,
280 struct lock_struct *plock,
282 struct blocking_lock_record *blr)
284 int fd = br_lck->fsp->fh->fd;
286 enum cbrl_lock_type type;
288 bool pending = false;
289 bool pending_async = false;
291 struct onefs_cbrl_blr_state *bs;
294 START_PROFILE(syscall_brl_lock);
296 SMB_ASSERT(plock->lock_flav == WINDOWS_LOCK);
297 SMB_ASSERT(plock->lock_type != UNLOCK_LOCK);
299 onefs_cbrl_enumerate_blq("onefs_brl_lock_windows");
301 /* Will only initialize the first time its called. */
304 switch (plock->lock_type) {
311 case PENDING_WRITE_LOCK:
312 /* Called when a blocking lock request is added - do an
318 case PENDING_READ_LOCK:
319 /* Called when a blocking lock request is added - do an
326 /* UNLOCK_LOCK: should only be used for a POSIX_LOCK */
327 smb_panic("Invalid plock->lock_type passed in to "
328 "onefs_brl_lock_windows");
331 /* Figure out if we're actually doing the lock or a no-op. We need to
332 * do a no-op when process_blocking_lock_queue calls back into us.
334 * We know process_* is calling into us if a blr is passed in and
335 * pending is false. */
336 if (!pending && blr) {
337 /* Check the BLR state. */
338 bs = (struct onefs_cbrl_blr_state *)blr->blr_private;
341 /* ASYNC still in progress: The process_* calls will keep
342 * calling even if we haven't gotten the lock. Keep erroring
343 * without calling ifs_cbrl, or getting/setting an id. */
344 if (bs->state == ONEFS_CBRL_ASYNC) {
347 else if (bs->state == ONEFS_CBRL_ERROR) {
348 END_PROFILE(syscall_brl_lock);
349 return NT_STATUS_NO_MEMORY;
352 SMB_ASSERT(bs->state == ONEFS_CBRL_NONE);
357 SMB_ASSERT(blocking_lock);
359 id = onefs_get_new_id();
362 DEBUG(10, ("Calling ifs_cbrl(LOCK)...\n"));
363 error = ifs_cbrl(fd, CBRL_OP_LOCK, type, plock->start,
364 plock->size, async, id, plock->context.smbpid, plock->context.tid,
368 } else if (errno == EWOULDBLOCK) {
370 } else if (errno == EINPROGRESS) {
374 /* Talloc a new BLR private state. */
375 blr->blr_private = talloc(blr, struct onefs_cbrl_blr_state);
376 pending_async = true;
379 /* Store the new id in the BLR private state. */
380 bs = (struct onefs_cbrl_blr_state *)blr->blr_private;
382 bs->state = ONEFS_CBRL_ASYNC;
384 DEBUG(0, ("onefs_brl_lock_windows failure: error=%d (%s).\n",
385 errno, strerror(errno)));
390 END_PROFILE(syscall_brl_lock);
392 /* Failure - error or async. */
393 plock->context.smbpid = (uint32) ONEFS_BLOCKING_PID;
396 status = NT_STATUS_OK;
398 status = brl_lock_failed(br_lck->fsp, plock, blocking_lock);
400 DEBUG(10, ("returning %s.\n", nt_errstr(status)));
404 /* Self contend our own level 2 oplock. The kernel handles
405 * contention of other opener's level 2 oplocks. */
406 contend_level2_oplocks_begin(br_lck->fsp,
407 LEVEL2_CONTEND_WINDOWS_BRL);
409 END_PROFILE(syscall_brl_lock);
412 onefs_cbrl_enumerate_blq("onefs_brl_unlock_windows");
413 DEBUG(10, ("returning NT_STATUS_OK.\n"));
417 bool onefs_brl_unlock_windows(vfs_handle_struct *handle,
418 struct messaging_context *msg_ctx,
419 struct byte_range_lock *br_lck,
420 const struct lock_struct *plock)
423 int fd = br_lck->fsp->fh->fd;
425 START_PROFILE(syscall_brl_unlock);
427 SMB_ASSERT(plock->lock_flav == WINDOWS_LOCK);
428 SMB_ASSERT(plock->lock_type == UNLOCK_LOCK);
430 DEBUG(10, ("Calling ifs_cbrl(UNLOCK)...\n"));
431 error = ifs_cbrl(fd, CBRL_OP_UNLOCK, CBRL_LK_SH,
432 plock->start, plock->size, false, 0, plock->context.smbpid,
433 plock->context.tid, plock->fnum);
435 END_PROFILE(syscall_brl_unlock);
438 DEBUG(10, ("returning false.\n"));
442 /* For symmetry purposes, end our oplock contention even though its
443 * currently a no-op. */
444 contend_level2_oplocks_end(br_lck->fsp, LEVEL2_CONTEND_WINDOWS_BRL);
446 DEBUG(10, ("returning true.\n"));
449 /* Problem with storing things in TDB: I won't know what BRL to unlock in the TDB.
451 * - I could send Samba a message with which lock is being unlocked?
452 * - I could *easily* make the "id" something you always pass in to
453 * lock, unlock or cancel -- it identifies a lock. Makes sense!
457 /* Default implementation only calls this on PENDING locks. */
458 bool onefs_brl_cancel_windows(vfs_handle_struct *handle,
459 struct byte_range_lock *br_lck,
460 struct lock_struct *plock,
461 struct blocking_lock_record *blr)
464 int fd = br_lck->fsp->fh->fd;
465 struct onefs_cbrl_blr_state *bs;
467 START_PROFILE(syscall_brl_cancel);
470 SMB_ASSERT(plock->lock_flav == WINDOWS_LOCK);
473 onefs_cbrl_enumerate_blq("onefs_brl_cancel_windows");
475 bs = ((struct onefs_cbrl_blr_state *)blr->blr_private);
478 if (bs->state == ONEFS_CBRL_DONE || bs->state == ONEFS_CBRL_ERROR) {
480 DEBUG(10, ("State=%d, returning true\n", bs->state));
481 END_PROFILE(syscall_brl_cancel);
485 SMB_ASSERT(bs->state == ONEFS_CBRL_NONE ||
486 bs->state == ONEFS_CBRL_ASYNC);
489 DEBUG(10, ("Calling ifs_cbrl(CANCEL)...\n"));
490 error = ifs_cbrl(fd, CBRL_OP_CANCEL, CBRL_LK_UNSPEC, plock->start,
491 plock->size, false, bs->id, plock->context.smbpid,
492 plock->context.tid, plock->fnum);
494 END_PROFILE(syscall_brl_cancel);
497 DEBUG(10, ("returning false\n"));
498 bs->state = ONEFS_CBRL_ERROR;
502 bs->state = ONEFS_CBRL_DONE;
503 onefs_cbrl_enumerate_blq("onefs_brl_cancel_windows");
504 DEBUG(10, ("returning true\n"));
508 bool onefs_strict_lock(vfs_handle_struct *handle,
510 struct lock_struct *plock)
514 START_PROFILE(syscall_strict_lock);
516 SMB_ASSERT(plock->lock_type == READ_LOCK ||
517 plock->lock_type == WRITE_LOCK);
519 if (!lp_locking(handle->conn->params) ||
520 !lp_strict_locking(handle->conn->params)) {
521 END_PROFILE(syscall_strict_lock);
525 if (plock->lock_flav == POSIX_LOCK) {
526 END_PROFILE(syscall_strict_lock);
527 return SMB_VFS_NEXT_STRICT_LOCK(handle, fsp, plock);
530 if (plock->size == 0) {
531 END_PROFILE(syscall_strict_lock);
535 error = ifs_cbrl(fsp->fh->fd, CBRL_OP_LOCK,
536 plock->lock_type == READ_LOCK ? CBRL_LK_RD : CBRL_LK_WR,
537 plock->start, plock->size, 0, 0, plock->context.smbpid,
538 plock->context.tid, plock->fnum);
540 END_PROFILE(syscall_strict_lock);
545 void onefs_strict_unlock(vfs_handle_struct *handle,
547 struct lock_struct *plock)
549 START_PROFILE(syscall_strict_unlock);
551 SMB_ASSERT(plock->lock_type == READ_LOCK ||
552 plock->lock_type == WRITE_LOCK);
554 if (!lp_locking(handle->conn->params) ||
555 !lp_strict_locking(handle->conn->params)) {
556 END_PROFILE(syscall_strict_unlock);
560 if (plock->lock_flav == POSIX_LOCK) {
561 SMB_VFS_NEXT_STRICT_UNLOCK(handle, fsp, plock);
562 END_PROFILE(syscall_strict_unlock);
566 if (plock->size == 0) {
567 END_PROFILE(syscall_strict_unlock);
572 ifs_cbrl(fsp->fh->fd, CBRL_OP_UNLOCK,
573 plock->lock_type == READ_LOCK ? CBRL_LK_RD : CBRL_LK_WR,
574 plock->start, plock->size, 0, 0, plock->context.smbpid,
575 plock->context.tid, plock->fnum);
578 END_PROFILE(syscall_strict_unlock);
581 /* TODO Optimization: Abstract out brl_get_locks() in the Windows case.
582 * We'll malloc some memory or whatever (can't return NULL), but not actually
585 /* XXX brl_locktest: CBRL does not support calling this, but its only for
586 * strict locking. Add empty VOP? */
588 /* XXX brl_lockquery: CBRL does not support calling this for WINDOWS LOCKS, but
589 * its only called for POSIX LOCKS. Add empty VOP? */
591 /* XXX brl_close_fnum: CBRL will do this automatically. I think this is a NO-OP
592 * for us, we could add an empty VOP. */