#include "includes.h"
static TDB_CONTEXT *posix_lock_tdb;
#ifdef WITH_BRLM
struct pending_close {
int fd;
BRLMRef brlm_ref;
};
#endif
static TDB_CONTEXT *posix_pending_close_tdb;
struct posix_lock {
int fd;
SMB_OFF_T start;
SMB_OFF_T size;
int lock_type;
};
struct posix_lock_key {
SMB_DEV_T device;
SMB_INO_T inode;
};
static TDB_DATA locking_key(SMB_DEV_T dev, SMB_INO_T inode)
{
static struct posix_lock_key key;
TDB_DATA kbuf;
memset(&key, '\0', sizeof(key));
key.device = dev;
key.inode = inode;
kbuf.dptr = (char *)&key;
kbuf.dsize = sizeof(key);
return kbuf;
}
static TDB_DATA locking_key_fsp(files_struct *fsp)
{
return locking_key(fsp->dev, fsp->inode);
}
static BOOL add_fd_to_close_entry(files_struct *fsp)
{
TDB_DATA kbuf = locking_key_fsp(fsp);
TDB_DATA dbuf;
char *tp;
void *data = NULL;
#ifdef WITH_BRLM
struct pending_close pc;
unsigned long datalen = sizeof(struct pending_close);
#else
unsigned long datalen = sizeof(int);
#endif
dbuf.dptr = NULL;
dbuf = tdb_fetch(posix_pending_close_tdb, kbuf);
#ifdef WITH_BRLM
pc.fd = fsp->fd;
pc.brlm_ref = fsp->brlm_ref;
data = &pc;
#else
data = &fsp->fd;
#endif
tp = SMB_REALLOC(dbuf.dptr, dbuf.dsize + datalen);
if (!tp) {
DEBUG(0,("add_fd_to_close_entry: Realloc fail !\n"));
SAFE_FREE(dbuf.dptr);
return False;
} else
dbuf.dptr = tp;
memcpy(dbuf.dptr + dbuf.dsize, data, datalen);
dbuf.dsize += datalen;
if (tdb_store(posix_pending_close_tdb, kbuf, dbuf, TDB_REPLACE) == -1) {
DEBUG(0,("add_fd_to_close_entry: tdb_store fail !\n"));
}
SAFE_FREE(dbuf.dptr);
return True;
}
static void delete_close_entries(files_struct *fsp)
{
TDB_DATA kbuf = locking_key_fsp(fsp);
if (tdb_delete(posix_pending_close_tdb, kbuf) == -1)
DEBUG(0,("delete_close_entries: tdb_delete fail !\n"));
}
#ifdef WITH_BRLM
static size_t get_posix_pending_close_entries(files_struct *fsp, struct pending_close **entries)
#else
static size_t get_posix_pending_close_entries(files_struct *fsp, int **entries)
#endif
{
TDB_DATA kbuf = locking_key_fsp(fsp);
TDB_DATA dbuf;
size_t count = 0;
*entries = NULL;
dbuf.dptr = NULL;
dbuf = tdb_fetch(posix_pending_close_tdb, kbuf);
if (!dbuf.dptr) {
return 0;
}
#ifdef WITH_BRLM
*entries = (struct pending_close *)dbuf.dptr;
count = (size_t)(dbuf.dsize / sizeof(struct pending_close));
#else
*entries = (int *)dbuf.dptr;
count = (size_t)(dbuf.dsize / sizeof(int));
#endif
return count;
}
static size_t get_posix_lock_entries(files_struct *fsp, struct posix_lock **entries)
{
TDB_DATA kbuf = locking_key_fsp(fsp);
TDB_DATA dbuf;
size_t count = 0;
*entries = NULL;
dbuf.dptr = NULL;
dbuf = tdb_fetch(posix_lock_tdb, kbuf);
if (!dbuf.dptr) {
return 0;
}
*entries = (struct posix_lock *)dbuf.dptr;
count = (size_t)(dbuf.dsize / sizeof(struct posix_lock));
return count;
}
int fd_close_posix(struct connection_struct *conn, files_struct *fsp)
{
int saved_errno = 0;
int ret;
size_t count, i;
struct posix_lock *entries = NULL;
#ifdef WITH_BRLM
BRLMStatus brlm_status = BRLMMiscErr;
struct pending_close *fd_array = NULL;
#else
int *fd_array = NULL;
#endif
BOOL locks_on_other_fds = False;
if (!lp_posix_locking(SNUM(conn))) {
ret = SMB_VFS_CLOSE(fsp,fsp->fd);
fsp->fd = -1;
return ret;
}
count = get_posix_lock_entries(fsp, &entries);
for (i = 0; i < count; i++) {
if (entries[i].fd != fsp->fd) {
locks_on_other_fds = True;
break;
}
}
if (locks_on_other_fds) {
if (!add_fd_to_close_entry(fsp)) {
SAFE_FREE(entries);
return -1;
}
SAFE_FREE(entries);
fsp->fd = -1;
return 0;
}
SAFE_FREE(entries);
count = get_posix_pending_close_entries(fsp, &fd_array);
if (count) {
DEBUG(10,("fd_close_posix: doing close on %u fd's.\n", (unsigned int)count ));
for(i = 0; i < count; i++) {
#ifdef WITH_BRLM
if (lp_BRLM())
{
if (fd_array[i].brlm_ref) {
brlm_status = BRLMCloseRef(fd_array[i].brlm_ref);
if (BRLMNoErr != brlm_status)
saved_errno = brlm_status;
}
DEBUG(4,("fd_close_posix: [%d]BRLMCloseRef fd(%d) ref(%X)\n", brlm_status, fd_array[i].fd, fd_array[i].brlm_ref));
} else {
if (SMB_VFS_CLOSE(fsp,fd_array[i].fd) == -1) {
saved_errno = errno;
}
}
#else
if (SMB_VFS_CLOSE(fsp,fd_array[i]) == -1) {
saved_errno = errno;
}
#endif
}
delete_close_entries(fsp);
}
SAFE_FREE(fd_array);
ret = SMB_VFS_CLOSE(fsp,fsp->fd);
if (saved_errno != 0) {
errno = saved_errno;
ret = -1;
}
fsp->fd = -1;
return ret;
}
static const char *posix_lock_type_name(int lock_type)
{
return (lock_type == F_RDLCK) ? "READ" : "WRITE";
}
static BOOL delete_posix_lock_entry_by_index(files_struct *fsp, size_t entry)
{
TDB_DATA kbuf = locking_key_fsp(fsp);
TDB_DATA dbuf;
struct posix_lock *locks;
size_t count;
dbuf.dptr = NULL;
dbuf = tdb_fetch(posix_lock_tdb, kbuf);
if (!dbuf.dptr) {
DEBUG(10,("delete_posix_lock_entry_by_index: tdb_fetch failed !\n"));
goto fail;
}
count = (size_t)(dbuf.dsize / sizeof(struct posix_lock));
locks = (struct posix_lock *)dbuf.dptr;
if (count == 1) {
tdb_delete(posix_lock_tdb, kbuf);
} else {
if (entry < count-1) {
memmove(&locks[entry], &locks[entry+1], sizeof(*locks)*((count-1) - entry));
}
dbuf.dsize -= sizeof(*locks);
tdb_store(posix_lock_tdb, kbuf, dbuf, TDB_REPLACE);
}
SAFE_FREE(dbuf.dptr);
return True;
fail:
SAFE_FREE(dbuf.dptr);
return False;
}
static BOOL add_posix_lock_entry(files_struct *fsp, SMB_OFF_T start, SMB_OFF_T size, int lock_type, size_t *pentry_num)
{
TDB_DATA kbuf = locking_key_fsp(fsp);
TDB_DATA dbuf;
struct posix_lock pl;
char *tp;
dbuf.dptr = NULL;
dbuf = tdb_fetch(posix_lock_tdb, kbuf);
*pentry_num = (size_t)(dbuf.dsize / sizeof(pl));
pl.fd = fsp->fd;
pl.start = start;
pl.size = size;
pl.lock_type = lock_type;
tp = SMB_REALLOC(dbuf.dptr, dbuf.dsize + sizeof(pl));
if (!tp) {
DEBUG(0,("add_posix_lock_entry: Realloc fail !\n"));
goto fail;
} else
dbuf.dptr = tp;
memcpy(dbuf.dptr + dbuf.dsize, &pl, sizeof(pl));
dbuf.dsize += sizeof(pl);
if (tdb_store(posix_lock_tdb, kbuf, dbuf, TDB_REPLACE) == -1) {
DEBUG(0,("add_posix_lock: Failed to add lock entry on file %s\n", fsp->fsp_name));
goto fail;
}
SAFE_FREE(dbuf.dptr);
DEBUG(10,("add_posix_lock: File %s: type = %s: start=%.0f size=%.0f: dev=%.0f inode=%.0f\n",
fsp->fsp_name, posix_lock_type_name(lock_type), (double)start, (double)size,
(double)fsp->dev, (double)fsp->inode ));
return True;
fail:
SAFE_FREE(dbuf.dptr);
return False;
}
static BOOL does_lock_overlap(SMB_OFF_T start1, SMB_OFF_T size1, SMB_OFF_T start2, SMB_OFF_T size2)
{
if (start1 >= start2 && start1 <= start2 + size2)
return True;
if (start1 < start2 && start1 + size1 > start2)
return True;
return False;
}
static int delete_posix_lock_entry(files_struct *fsp, SMB_OFF_T start, SMB_OFF_T size, struct posix_lock *pl)
{
TDB_DATA kbuf = locking_key_fsp(fsp);
TDB_DATA dbuf;
struct posix_lock *locks;
size_t i, count;
BOOL found = False;
int num_overlapping_records = 0;
dbuf.dptr = NULL;
dbuf = tdb_fetch(posix_lock_tdb, kbuf);
if (!dbuf.dptr) {
DEBUG(10,("delete_posix_lock_entry: tdb_fetch failed !\n"));
goto fail;
}
locks = (struct posix_lock *)dbuf.dptr;
count = (size_t)(dbuf.dsize / sizeof(*locks));
for (i=0; i<count; i++) {
struct posix_lock *entry = &locks[i];
if (entry->fd == fsp->fd &&
entry->start == start &&
entry->size == size) {
if (pl)
*pl = *entry;
if (count == 1) {
tdb_delete(posix_lock_tdb, kbuf);
} else {
if (i < count-1) {
memmove(&locks[i], &locks[i+1], sizeof(*locks)*((count-1) - i));
}
dbuf.dsize -= sizeof(*locks);
tdb_store(posix_lock_tdb, kbuf, dbuf, TDB_REPLACE);
}
count--;
found = True;
break;
}
}
if (!found)
goto fail;
for (i = 0; i < count; i++) {
struct posix_lock *entry = &locks[i];
if (fsp->fd == entry->fd &&
does_lock_overlap( start, size, entry->start, entry->size))
num_overlapping_records++;
}
DEBUG(10,("delete_posix_lock_entry: type = %s: start=%.0f size=%.0f, num_records = %d\n",
posix_lock_type_name(pl->lock_type), (double)pl->start, (double)pl->size,
(unsigned int)num_overlapping_records ));
SAFE_FREE(dbuf.dptr);
return num_overlapping_records;
fail:
SAFE_FREE(dbuf.dptr);
return -1;
}
static int map_posix_lock_type( files_struct *fsp, enum brl_type lock_type)
{
if((lock_type == WRITE_LOCK) && !fsp->can_write) {
DEBUG(10,("map_posix_lock_type: Downgrading write lock to read due to read-only file.\n"));
return F_RDLCK;
} else if((lock_type == READ_LOCK) && !fsp->can_read) {
DEBUG(10,("map_posix_lock_type: Changing read lock to write due to write-only file.\n"));
return F_WRLCK;
}
return (lock_type == READ_LOCK) ? F_RDLCK : F_WRLCK;
}
static BOOL posix_lock_in_range(SMB_OFF_T *offset_out, SMB_OFF_T *count_out,
SMB_BIG_UINT u_offset, SMB_BIG_UINT u_count)
{
SMB_OFF_T offset = (SMB_OFF_T)u_offset;
SMB_OFF_T count = (SMB_OFF_T)u_count;
#if defined(MAX_POSITIVE_LOCK_OFFSET)
SMB_OFF_T max_positive_lock_offset = (MAX_POSITIVE_LOCK_OFFSET);
#elif defined(LARGE_SMB_OFF_T) && !defined(HAVE_BROKEN_FCNTL64_LOCKS)
SMB_OFF_T mask2 = ((SMB_OFF_T)0x4) << (SMB_OFF_T_BITS-4);
SMB_OFF_T mask = (mask2<<1);
SMB_OFF_T max_positive_lock_offset = ~mask;
#else
SMB_OFF_T max_positive_lock_offset = 0x7FFFFFFF;
#endif
if (count == (SMB_OFF_T)0) {
DEBUG(10,("posix_lock_in_range: count = 0, ignoring.\n"));
return False;
}
if (u_offset & ~((SMB_BIG_UINT)max_positive_lock_offset)) {
DEBUG(10,("posix_lock_in_range: (offset = %.0f) offset > %.0f and we cannot handle this. Ignoring lock.\n",
(double)u_offset, (double)((SMB_BIG_UINT)max_positive_lock_offset) ));
return False;
}
if (u_count & ~((SMB_BIG_UINT)max_positive_lock_offset))
count = max_positive_lock_offset;
if (offset + count < 0 || offset + count > max_positive_lock_offset)
count = max_positive_lock_offset - offset;
if (count == 0) {
DEBUG(10,("posix_lock_in_range: Count = 0. Ignoring lock u_offset = %.0f, u_count = %.0f\n",
(double)u_offset, (double)u_count ));
return False;
}
DEBUG(10,("posix_lock_in_range: offset_out = %.0f, count_out = %.0f\n",
(double)offset, (double)count ));
*offset_out = offset;
*count_out = count;
return True;
}
static BOOL posix_fcntl_lock(files_struct *fsp, int op, SMB_OFF_T offset, SMB_OFF_T count, int type)
{
int ret;
#ifdef WITH_BRLM
BRLMStatus brlm_status = BRLMParamErr;
BRLMLockType brlm_lock_type = 0;
#endif
DEBUG(8,("posix_fcntl_lock %d %d %.0f %.0f %d\n",fsp->fd,op,(double)offset,(double)count,type));
#ifdef WITH_BRLM
if (lp_BRLM())
{
if (op == SMB_F_GETLK)
{
if (type == F_RDLCK)
{
brlm_status = BRLMCanRead(fsp->brlm_ref, offset, count);
DEBUG(8,("posix_fcntl_lock: [%d]BRLMCanRead offset[%d] count[%d]\n",brlm_status, offset, count));
} else if (type == F_WRLCK) {
brlm_status = BRLMCanWrite(fsp->brlm_ref, offset, count);
DEBUG(8,("posix_fcntl_lock: [%d]BRLMCanWrite offset[%d] count[%d]\n",brlm_status, offset, count));
} else {
DEBUG(8,("posix_fcntl_lock: !!!!! SMB_F_GETLK !!!!! type[0x%X]\n", type));
ret = 0;
}
if (BRLMNoErr == brlm_status)
ret = 0;
else
ret = 1;
} else if (op == SMB_F_SETLK) {
if (type == F_RDLCK) {
brlm_status = BRLMByteRangeLock(fsp->brlm_ref, kBRLMRLock, offset, count);
DEBUG(8,("posix_fcntl_lock: [%d]BRLMByteRangeLock kBRLMRLock\n",brlm_status));
} else if (type == F_WRLCK) {
brlm_status = BRLMByteRangeLock(fsp->brlm_ref, kBRLMWLock, offset, count);
DEBUG(8,("posix_fcntl_lock: [%d]BRLMByteRangeLock kBRLMWLock\n",brlm_status));
} else if (type == F_UNLCK) {
brlm_status = BRLMByteRangeUnlock(fsp->brlm_ref, kBRLMFree, offset, count);
DEBUG(8,("posix_fcntl_lock: [%d]BRLMByteRangeUnlock kBRLMFree\n",brlm_status));
} else {
DEBUG(8,("posix_fcntl_lock: !!!!! SMB_F_SETLK !!!!! type[0x%X]\n", type));
ret = 0;
}
if (BRLMNoErr == brlm_status)
ret = 1;
else
ret = 0;
} else {
DEBUG(8,("posix_fcntl_lock: !!!!! OP !!!!! op[0x%X] lock_type[0x%X]\n", op, type));
ret = 0;
}
} else {
#endif
ret = SMB_VFS_LOCK(fsp,fsp->fd,op,offset,count,type);
if (!ret && ((errno == EFBIG) || (errno == ENOLCK) || (errno == EINVAL))) {
DEBUG(0,("posix_fcntl_lock: WARNING: lock request at offset %.0f, length %.0f returned\n",
(double)offset,(double)count));
DEBUG(0,("an %s error. This can happen when using 64 bit lock offsets\n", strerror(errno)));
DEBUG(0,("on 32 bit NFS mounted file systems.\n"));
if (offset & ~((SMB_OFF_T)0x7fffffff)) {
DEBUG(0,("Offset greater than 31 bits. Returning success.\n"));
return True;
}
if (count & ~((SMB_OFF_T)0x7fffffff)) {
DEBUG(0,("Count greater than 31 bits - retrying with 31 bit truncated length.\n"));
errno = 0;
count &= 0x7fffffff;
ret = SMB_VFS_LOCK(fsp,fsp->fd,op,offset,count,type);
}
}
#ifdef WITH_BRLM
}
#endif
DEBUG(8,("posix_fcntl_lock: Lock call %s\n", ret ? "successful" : "failed"));
return ret;
}
BOOL is_posix_locked(files_struct *fsp, SMB_BIG_UINT u_offset, SMB_BIG_UINT u_count, enum brl_type lock_type)
{
SMB_OFF_T offset;
SMB_OFF_T count;
int posix_lock_type = map_posix_lock_type(fsp,lock_type);
DEBUG(10,("is_posix_locked: File %s, offset = %.0f, count = %.0f, type = %s\n",
fsp->fsp_name, (double)u_offset, (double)u_count, posix_lock_type_name(lock_type) ));
if(!posix_lock_in_range(&offset, &count, u_offset, u_count))
return False;
return posix_fcntl_lock(fsp,SMB_F_GETLK,offset,count,posix_lock_type);
}
struct lock_list {
struct lock_list *next;
struct lock_list *prev;
SMB_OFF_T start;
SMB_OFF_T size;
};
static struct lock_list *posix_lock_list(TALLOC_CTX *ctx, struct lock_list *lhead, files_struct *fsp)
{
TDB_DATA kbuf = locking_key_fsp(fsp);
TDB_DATA dbuf;
struct posix_lock *locks;
size_t num_locks, i;
dbuf.dptr = NULL;
dbuf = tdb_fetch(posix_lock_tdb, kbuf);
if (!dbuf.dptr)
return lhead;
locks = (struct posix_lock *)dbuf.dptr;
num_locks = (size_t)(dbuf.dsize / sizeof(*locks));
DEBUG(10,("posix_lock_list: curr: start=%.0f,size=%.0f\n",
(double)lhead->start, (double)lhead->size ));
for (i=0; i<num_locks && lhead; i++) {
struct posix_lock *lock = &locks[i];
struct lock_list *l_curr;
for (l_curr = lhead; l_curr;) {
DEBUG(10,("posix_lock_list: lock: fd=%d: start=%.0f,size=%.0f:type=%s", lock->fd,
(double)lock->start, (double)lock->size, posix_lock_type_name(lock->lock_type) ));
if ( (l_curr->start >= (lock->start + lock->size)) ||
(lock->start >= (l_curr->start + l_curr->size))) {
DEBUG(10,("no overlap case.\n" ));
l_curr = l_curr->next;
} else if ( (l_curr->start >= lock->start) &&
(l_curr->start + l_curr->size <= lock->start + lock->size) ) {
struct lock_list *ul_next = l_curr->next;
DEBUG(10,("delete case.\n" ));
DLIST_REMOVE(lhead, l_curr);
if(lhead == NULL)
break;
l_curr = ul_next;
} else if ( (l_curr->start >= lock->start) &&
(l_curr->start < lock->start + lock->size) &&
(l_curr->start + l_curr->size > lock->start + lock->size) ) {
l_curr->size = (l_curr->start + l_curr->size) - (lock->start + lock->size);
l_curr->start = lock->start + lock->size;
DEBUG(10,("truncate high case: start=%.0f,size=%.0f\n",
(double)l_curr->start, (double)l_curr->size ));
l_curr = l_curr->next;
} else if ( (l_curr->start < lock->start) &&
(l_curr->start + l_curr->size > lock->start) &&
(l_curr->start + l_curr->size <= lock->start + lock->size) ) {
l_curr->size = lock->start - l_curr->start;
DEBUG(10,("truncate low case: start=%.0f,size=%.0f\n",
(double)l_curr->start, (double)l_curr->size ));
l_curr = l_curr->next;
} else if ( (l_curr->start < lock->start) &&
(l_curr->start + l_curr->size > lock->start + lock->size) ) {
struct lock_list *l_new = TALLOC_P(ctx, struct lock_list);
if(l_new == NULL) {
DEBUG(0,("posix_lock_list: talloc fail.\n"));
return NULL;
}
ZERO_STRUCTP(l_new);
l_new->start = lock->start + lock->size;
l_new->size = l_curr->start + l_curr->size - l_new->start;
l_curr->size = lock->start - l_curr->start;
DEBUG(10,("split case: curr: start=%.0f,size=%.0f \
new: start=%.0f,size=%.0f\n", (double)l_curr->start, (double)l_curr->size,
(double)l_new->start, (double)l_new->size ));
l_new->prev = l_curr;
l_new->next = l_curr->next;
l_curr->next = l_new;
l_curr = l_new->next;
} else {
pstring msg;
slprintf(msg, sizeof(msg)-1, "logic flaw in cases: l_curr: start = %.0f, size = %.0f : \
lock: start = %.0f, size = %.0f\n", (double)l_curr->start, (double)l_curr->size, (double)lock->start, (double)lock->size );
smb_panic(msg);
}
}
}
SAFE_FREE(dbuf.dptr);
return lhead;
}
BOOL set_posix_lock(files_struct *fsp, SMB_BIG_UINT u_offset, SMB_BIG_UINT u_count, enum brl_type lock_type)
{
SMB_OFF_T offset;
SMB_OFF_T count;
BOOL ret = True;
size_t entry_num = 0;
size_t lock_count;
TALLOC_CTX *l_ctx = NULL;
struct lock_list *llist = NULL;
struct lock_list *ll = NULL;
int posix_lock_type = map_posix_lock_type(fsp,lock_type);
DEBUG(5,("set_posix_lock: File %s, offset = %.0f, count = %.0f, type = %s\n",
fsp->fsp_name, (double)u_offset, (double)u_count, posix_lock_type_name(lock_type) ));
if(!posix_lock_in_range(&offset, &count, u_offset, u_count))
return True;
if ((l_ctx = talloc_init("set_posix_lock")) == NULL) {
DEBUG(0,("set_posix_lock: unable to init talloc context.\n"));
return True;
}
if ((ll = TALLOC_P(l_ctx, struct lock_list)) == NULL) {
DEBUG(0,("set_posix_lock: unable to talloc unlock list.\n"));
talloc_destroy(l_ctx);
return True;
}
ZERO_STRUCTP(ll);
ll->start = offset;
ll->size = count;
DLIST_ADD(llist, ll);
llist = posix_lock_list(l_ctx, llist, fsp);
if (!add_posix_lock_entry(fsp,offset,count,posix_lock_type,&entry_num)) {
DEBUG(0,("set_posix_lock: Unable to create posix lock entry !\n"));
talloc_destroy(l_ctx);
return False;
}
for (lock_count = 0, ll = llist; ll; ll = ll->next, lock_count++) {
offset = ll->start;
count = ll->size;
DEBUG(5,("set_posix_lock: Real lock: Type = %s: offset = %.0f, count = %.0f\n",
posix_lock_type_name(posix_lock_type), (double)offset, (double)count ));
if (!posix_fcntl_lock(fsp,SMB_F_SETLK,offset,count,posix_lock_type)) {
DEBUG(5,("set_posix_lock: Lock fail !: Type = %s: offset = %.0f, count = %.0f. Errno = %s\n",
posix_lock_type_name(posix_lock_type), (double)offset, (double)count, strerror(errno) ));
ret = False;
break;
}
}
if (!ret) {
for (ll = llist; lock_count; ll = ll->next, lock_count--) {
offset = ll->start;
count = ll->size;
DEBUG(5,("set_posix_lock: Backing out locks: Type = %s: offset = %.0f, count = %.0f\n",
posix_lock_type_name(posix_lock_type), (double)offset, (double)count ));
posix_fcntl_lock(fsp,SMB_F_SETLK,offset,count,F_UNLCK);
}
delete_posix_lock_entry_by_index(fsp,entry_num);
}
talloc_destroy(l_ctx);
return ret;
}
BOOL release_posix_lock(files_struct *fsp, SMB_BIG_UINT u_offset, SMB_BIG_UINT u_count)
{
SMB_OFF_T offset;
SMB_OFF_T count;
BOOL ret = True;
TALLOC_CTX *ul_ctx = NULL;
struct lock_list *ulist = NULL;
struct lock_list *ul = NULL;
struct posix_lock deleted_lock;
int num_overlapped_entries;
DEBUG(5,("release_posix_lock: File %s, offset = %.0f, count = %.0f\n",
fsp->fsp_name, (double)u_offset, (double)u_count ));
if(!posix_lock_in_range(&offset, &count, u_offset, u_count))
return True;
num_overlapped_entries = delete_posix_lock_entry(fsp, offset, count, &deleted_lock);
if (num_overlapped_entries == -1) {
smb_panic("release_posix_lock: unable find entry to delete !\n");
}
if (num_overlapped_entries > 0 && deleted_lock.lock_type == F_WRLCK) {
if (!posix_fcntl_lock(fsp,SMB_F_SETLK,offset,count,F_RDLCK)) {
DEBUG(0,("release_posix_lock: downgrade of lock failed with error %s !\n", strerror(errno) ));
return False;
}
}
if ((ul_ctx = talloc_init("release_posix_lock")) == NULL) {
DEBUG(0,("release_posix_lock: unable to init talloc context.\n"));
return True;
}
if ((ul = TALLOC_P(ul_ctx, struct lock_list)) == NULL) {
DEBUG(0,("release_posix_lock: unable to talloc unlock list.\n"));
talloc_destroy(ul_ctx);
return True;
}
ZERO_STRUCTP(ul);
ul->start = offset;
ul->size = count;
DLIST_ADD(ulist, ul);
ulist = posix_lock_list(ul_ctx, ulist, fsp);
for(; ulist; ulist = ulist->next) {
offset = ulist->start;
count = ulist->size;
DEBUG(5,("release_posix_lock: Real unlock: offset = %.0f, count = %.0f\n",
(double)offset, (double)count ));
if (!posix_fcntl_lock(fsp,SMB_F_SETLK,offset,count,F_UNLCK))
ret = False;
}
talloc_destroy(ul_ctx);
return ret;
}
static void delete_posix_lock_entries(files_struct *fsp)
{
TDB_DATA kbuf = locking_key_fsp(fsp);
if (tdb_delete(posix_lock_tdb, kbuf) == -1)
DEBUG(0,("delete_close_entries: tdb_delete fail !\n"));
}
static void dump_entry(struct posix_lock *pl)
{
DEBUG(10,("entry: start=%.0f, size=%.0f, type=%d, fd=%i\n",
(double)pl->start, (double)pl->size, (int)pl->lock_type, pl->fd ));
}
void posix_locking_close_file(files_struct *fsp)
{
struct posix_lock *entries = NULL;
size_t count, i;
count = get_posix_lock_entries(fsp, &entries);
if (count == 0) {
DEBUG(10,("posix_locking_close_file: file %s has no outstanding locks.\n", fsp->fsp_name ));
return;
}
for (i = 0; i < count; i++) {
if (entries[i].fd != fsp->fd )
break;
dump_entry(&entries[i]);
}
if (i == count) {
DEBUG(10,("posix_locking_close_file: file %s has %u outstanding locks, but all on one fd.\n",
fsp->fsp_name, (unsigned int)count ));
SAFE_FREE(entries);
delete_posix_lock_entries(fsp);
return;
}
for (i = 0; i < count; i++) {
struct posix_lock *pl = &entries[i];
if (pl->fd == fsp->fd)
release_posix_lock(fsp, (SMB_BIG_UINT)pl->start, (SMB_BIG_UINT)pl->size );
}
SAFE_FREE(entries);
}
BOOL posix_locking_init(int read_only)
{
if (posix_lock_tdb && posix_pending_close_tdb)
return True;
if (!posix_lock_tdb)
posix_lock_tdb = tdb_open_log(NULL, 0, TDB_INTERNAL,
read_only?O_RDONLY:(O_RDWR|O_CREAT), 0644);
if (!posix_lock_tdb) {
DEBUG(0,("Failed to open POSIX byte range locking database.\n"));
return False;
}
if (!posix_pending_close_tdb)
posix_pending_close_tdb = tdb_open_log(NULL, 0, TDB_INTERNAL,
read_only?O_RDONLY:(O_RDWR|O_CREAT), 0644);
if (!posix_pending_close_tdb) {
DEBUG(0,("Failed to open POSIX pending close database.\n"));
return False;
}
return True;
}
BOOL posix_locking_end(void)
{
if (posix_lock_tdb && tdb_close(posix_lock_tdb) != 0)
return False;
if (posix_pending_close_tdb && tdb_close(posix_pending_close_tdb) != 0)
return False;
return True;
}