summaryrefslogtreecommitdiff
path: root/source3
diff options
context:
space:
mode:
Diffstat (limited to 'source3')
-rw-r--r--source3/include/smb.h1
-rw-r--r--source3/locking/brlock.c103
2 files changed, 103 insertions, 1 deletions
diff --git a/source3/include/smb.h b/source3/include/smb.h
index 0edb1c31e2..c583055a49 100644
--- a/source3/include/smb.h
+++ b/source3/include/smb.h
@@ -447,6 +447,7 @@ typedef struct files_struct {
BOOL is_directory;
BOOL is_stat;
BOOL aio_write_behind;
+ BOOL lockdb_clean;
char *fsp_name;
FAKE_FILE_HANDLE *fake_file_handle;
} files_struct;
diff --git a/source3/locking/brlock.c b/source3/locking/brlock.c
index 0233d60df6..b9d401cad6 100644
--- a/source3/locking/brlock.c
+++ b/source3/locking/brlock.c
@@ -1269,13 +1269,62 @@ static int traverse_fn(TDB_CONTEXT *ttdb, TDB_DATA kbuf, TDB_DATA dbuf, void *st
{
struct lock_struct *locks;
struct lock_key *key;
- int i;
+ unsigned int i;
+ unsigned int num_locks = 0;
+ unsigned int num_valid_entries = 0;
BRLOCK_FN(traverse_callback) = (BRLOCK_FN_CAST())state;
locks = (struct lock_struct *)dbuf.dptr;
key = (struct lock_key *)kbuf.dptr;
+ num_locks = dbuf.dsize/sizeof(*locks);
+
+ /* Ensure the lock db is clean of invalid processes. */
+
+ for (i = 0; i < num_locks; i++) {
+ struct lock_struct *lock_data = &locks[i];
+ if (!process_exists(lock_data->context.pid)) {
+ /* This process no longer exists - mark this
+ entry as invalid by zeroing it. */
+ ZERO_STRUCTP(lock_data);
+ } else {
+ num_valid_entries++;
+ }
+ }
+
+ if (num_valid_entries != num_locks) {
+ struct lock_struct *new_lock_data = NULL;
+
+ if (num_valid_entries) {
+ new_lock_data = SMB_MALLOC_ARRAY(struct lock_struct, num_valid_entries);
+ if (!new_lock_data) {
+ DEBUG(3, ("malloc fail\n"));
+ return 0;
+ }
+ num_valid_entries = 0;
+ for (i = 0; i < num_locks; i++) {
+ struct lock_struct *lock_data = &locks[i];
+ if (lock_data->context.smbpid &&
+ lock_data->context.tid) {
+ /* Valid (nonzero) entry - copy it. */
+ memcpy(&new_lock_data[num_valid_entries],
+ lock_data, sizeof(struct lock_struct));
+ num_valid_entries++;
+ }
+ }
+ }
+ SAFE_FREE(dbuf.dptr);
+ dbuf.dptr = (void *)new_lock_data;
+ dbuf.dsize = (num_valid_entries) * sizeof(*locks);
+
+ if (dbuf.dsize) {
+ tdb_store(ttdb, kbuf, dbuf, TDB_REPLACE);
+ } else {
+ tdb_delete(ttdb, kbuf);
+ }
+ }
+
for (i=0;i<dbuf.dsize/sizeof(*locks);i++) {
traverse_callback(key->device,
key->inode,
@@ -1375,6 +1424,58 @@ struct byte_range_lock *brl_get_locks(files_struct *fsp)
br_lck->lock_data = (void *)data.dptr;
br_lck->num_locks = data.dsize / sizeof(struct lock_struct);
+ if (!fsp->lockdb_clean) {
+
+ /* This is the first time we've accessed this. */
+ /* Go through and ensure all entries exist - remove any that don't. */
+ /* Makes the lockdb self cleaning at low cost. */
+ unsigned int num_valid_entries = 0;
+ unsigned int i;
+
+ for (i = 0; i < br_lck->num_locks; i++) {
+ struct lock_struct *lock_data = &((struct lock_struct *)br_lck->lock_data)[i];
+ if (!process_exists(lock_data->context.pid)) {
+ /* This process no longer exists - mark this
+ entry as invalid by zeroing it. */
+ ZERO_STRUCTP(lock_data);
+ } else {
+ num_valid_entries++;
+ }
+ }
+
+ if (num_valid_entries != br_lck->num_locks) {
+ struct lock_struct *new_lock_data = NULL;
+
+ if (num_valid_entries) {
+ new_lock_data = SMB_MALLOC_ARRAY(struct lock_struct, num_valid_entries);
+ if (!new_lock_data) {
+ DEBUG(3, ("malloc fail\n"));
+ tdb_chainunlock(tdb, key);
+ SAFE_FREE(br_lck->lock_data);
+ SAFE_FREE(br_lck);
+ return NULL;
+ }
+ num_valid_entries = 0;
+ for (i = 0; i < br_lck->num_locks; i++) {
+ struct lock_struct *lock_data = &((struct lock_struct *)br_lck->lock_data)[i];
+ if (lock_data->context.smbpid &&
+ lock_data->context.tid) {
+ /* Valid (nonzero) entry - copy it. */
+ memcpy(&new_lock_data[num_valid_entries],
+ lock_data, sizeof(struct lock_struct));
+ num_valid_entries++;
+ }
+ }
+ }
+ SAFE_FREE(br_lck->lock_data);
+ br_lck->lock_data = (void *)new_lock_data;
+ br_lck->num_locks = num_valid_entries;
+ }
+
+ /* Mark the lockdb as "clean" as seen from this open file. */
+ fsp->lockdb_clean = True;
+ }
+
if (DEBUGLEVEL >= 10) {
unsigned int i;
struct lock_struct *locks = (struct lock_struct *)br_lck->lock_data;