s4-streams: fixed handling of stream rename and overwrite
[ira/wip.git] / source4 / ntvfs / posix / pvfs_streams.c
index 12f783e1729c7112642031f261a45edb659bfcc0..4da95432c1ff46c8842d91a871de822669a9910d 100644 (file)
@@ -7,7 +7,7 @@
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
    GNU General Public License for more details.
    
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
-#include "system/filesys.h"
 #include "vfs_posix.h"
-#include "librpc/gen_ndr/ndr_xattr.h"
+#include "librpc/gen_ndr/xattr.h"
+
+/*
+  normalise a stream name, removing a :$DATA suffix if there is one 
+  Note: this returns the existing pointer to the name if the name does 
+        not need normalising
+ */
+static const char *stream_name_normalise(TALLOC_CTX *ctx, const char *name)
+{
+       const char *c = strchr_m(name, ':');
+       if (c == NULL || strcasecmp_m(c, ":$DATA") != 0) {
+               return name;
+       }
+       return talloc_strndup(ctx, name, c-name);
+}
+
+/*
+  compare two stream names, taking account of the default $DATA extension
+ */
+static int stream_name_cmp(const char *name1, const char *name2)
+{
+       const char *c1, *c2;
+       int l1, l2, ret;
+       c1 = strchr_m(name1, ':');
+       c2 = strchr_m(name2, ':');
+       
+       /* check the first part is the same */
+       l1 = c1?(c1 - name1):strlen(name1);
+       l2 = c2?(c2 - name2):strlen(name2);
+       if (l1 != l2) {
+               return l1 - l2;
+       }
+       ret = strncasecmp_m(name1, name2, l1);
+       if (ret != 0) {
+               return ret;
+       }
+
+       /* the first parts are the same, check the suffix */
+       if (c1 && c2) {
+               return strcasecmp_m(c1, c2);
+       }
+
+       if (c1) {
+               return strcasecmp_m(c1, ":$DATA");
+       }
+       if (c2) {
+               return strcasecmp_m(c2, ":$DATA");
+       }
+
+       /* neither names have a suffix */
+       return 0;
+}
 
 
 /*
@@ -38,7 +87,14 @@ NTSTATUS pvfs_stream_information(struct pvfs_state *pvfs,
        int i;
        NTSTATUS status;
 
-       streams = talloc_p(mem_ctx, struct xattr_DosStreams);
+       /* directories don't have streams */
+       if (name->dos.attrib & FILE_ATTRIBUTE_DIRECTORY) {
+               info->num_streams = 0;
+               info->streams = NULL;
+               return NT_STATUS_OK;
+       }
+
+       streams = talloc(mem_ctx, struct xattr_DosStreams);
        if (streams == NULL) {
                return NT_STATUS_NO_MEMORY;
        }
@@ -49,7 +105,7 @@ NTSTATUS pvfs_stream_information(struct pvfs_state *pvfs,
        }
 
        info->num_streams = streams->num_streams+1;
-       info->streams = talloc_array_p(mem_ctx, struct stream_struct, info->num_streams);
+       info->streams = talloc_array(mem_ctx, struct stream_struct, info->num_streams);
        if (!info->streams) {
                return NT_STATUS_NO_MEMORY;
        }
@@ -61,9 +117,14 @@ NTSTATUS pvfs_stream_information(struct pvfs_state *pvfs,
        for (i=0;i<streams->num_streams;i++) {
                info->streams[i+1].size          = streams->streams[i].size;
                info->streams[i+1].alloc_size    = streams->streams[i].alloc_size;
-               info->streams[i+1].stream_name.s = talloc_asprintf(streams->streams, 
-                                                                  ":%s:$DATA",
-                                                                  streams->streams[i].name);
+               if (strchr(streams->streams[i].name, ':') == NULL) {
+                       info->streams[i+1].stream_name.s = talloc_asprintf(streams->streams, 
+                                                                          ":%s:$DATA",
+                                                                          streams->streams[i].name);
+               } else {
+                       info->streams[i+1].stream_name.s = talloc_strdup(streams->streams, 
+                                                                        streams->streams[i].name);
+               }
        }
 
        return NT_STATUS_OK;
@@ -81,11 +142,11 @@ NTSTATUS pvfs_stream_info(struct pvfs_state *pvfs, struct pvfs_filename *name, i
 
        /* the NULL stream always exists */
        if (name->stream_name == NULL) {
-               name->stream_exists = True;
+               name->stream_exists = true;
                return NT_STATUS_OK;
        }
 
-       streams = talloc_p(name, struct xattr_DosStreams);
+       streams = talloc(name, struct xattr_DosStreams);
        if (streams == NULL) {
                return NT_STATUS_NO_MEMORY;
        }
@@ -98,10 +159,10 @@ NTSTATUS pvfs_stream_info(struct pvfs_state *pvfs, struct pvfs_filename *name, i
 
        for (i=0;i<streams->num_streams;i++) {
                struct xattr_DosStream *s = &streams->streams[i];
-               if (StrCaseCmp(s->name, name->stream_name) == 0) {
+               if (stream_name_cmp(s->name, name->stream_name) == 0) {
                        name->dos.alloc_size = pvfs_round_alloc_size(pvfs, s->alloc_size);
                        name->st.st_size     = s->size;
-                       name->stream_exists = True;
+                       name->stream_exists = true;
                        talloc_free(streams);
                        return NT_STATUS_OK;
                }
@@ -111,7 +172,7 @@ NTSTATUS pvfs_stream_info(struct pvfs_state *pvfs, struct pvfs_filename *name, i
 
        name->dos.alloc_size = 0;
        name->st.st_size     = 0;
-       name->stream_exists = False;
+       name->stream_exists = false;
 
        return NT_STATUS_OK;
 }
@@ -127,7 +188,7 @@ static NTSTATUS pvfs_stream_update_size(struct pvfs_state *pvfs, struct pvfs_fil
        int i;
        NTSTATUS status;
 
-       streams = talloc_p(name, struct xattr_DosStreams);
+       streams = talloc(name, struct xattr_DosStreams);
        if (streams == NULL) {
                return NT_STATUS_NO_MEMORY;
        }
@@ -139,7 +200,7 @@ static NTSTATUS pvfs_stream_update_size(struct pvfs_state *pvfs, struct pvfs_fil
 
        for (i=0;i<streams->num_streams;i++) {
                struct xattr_DosStream *s = &streams->streams[i];
-               if (StrCaseCmp(s->name, name->stream_name) == 0) {
+               if (stream_name_cmp(s->name, name->stream_name) == 0) {
                        s->size       = size;
                        s->alloc_size = pvfs_round_alloc_size(pvfs, size);
                        break;
@@ -148,7 +209,7 @@ static NTSTATUS pvfs_stream_update_size(struct pvfs_state *pvfs, struct pvfs_fil
 
        if (i == streams->num_streams) {
                struct xattr_DosStream *s;
-               streams->streams = talloc_realloc_p(streams, streams->streams, 
+               streams->streams = talloc_realloc(streams, streams->streams, 
                                                    struct xattr_DosStream,
                                                    streams->num_streams+1);
                if (streams->streams == NULL) {
@@ -161,7 +222,11 @@ static NTSTATUS pvfs_stream_update_size(struct pvfs_state *pvfs, struct pvfs_fil
                s->flags      = XATTR_STREAM_FLAG_INTERNAL;
                s->size       = size;
                s->alloc_size = pvfs_round_alloc_size(pvfs, size);
-               s->name       = name->stream_name;
+               s->name       = stream_name_normalise(streams, name->stream_name);
+               if (s->name == NULL) {
+                       talloc_free(streams);
+                       return NT_STATUS_NO_MEMORY;
+               }
        }
 
        status = pvfs_streams_save(pvfs, name, fd, streams);
@@ -171,6 +236,84 @@ static NTSTATUS pvfs_stream_update_size(struct pvfs_state *pvfs, struct pvfs_fil
 }
 
 
+/*
+  rename a stream
+*/
+NTSTATUS pvfs_stream_rename(struct pvfs_state *pvfs, struct pvfs_filename *name, int fd,
+                           const char *new_name, bool overwrite)
+{
+       struct xattr_DosStreams *streams;
+       int i, found_old, found_new;
+       NTSTATUS status;
+
+       streams = talloc(name, struct xattr_DosStreams);
+       if (streams == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       new_name = stream_name_normalise(streams, new_name);
+       if (new_name == NULL) {
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       status = pvfs_streams_load(pvfs, name, fd, streams);
+       if (!NT_STATUS_IS_OK(status)) {
+               ZERO_STRUCTP(streams);
+       }
+
+       /* the default stream always exists */
+       if (strcmp(new_name, "") == 0 ||
+           strcasecmp_m(new_name, ":$DATA") == 0) {
+               return NT_STATUS_OBJECT_NAME_COLLISION;         
+       }
+
+       /* try to find the old/new names in the list */
+       found_old = found_new = -1;
+       for (i=0;i<streams->num_streams;i++) {
+               struct xattr_DosStream *s = &streams->streams[i];
+               if (stream_name_cmp(s->name, new_name) == 0) {
+                       found_new = i;
+               }
+               if (stream_name_cmp(s->name, name->stream_name) == 0) {
+                       found_old = i;
+               }
+       }
+
+       if (found_old == -1) {
+               talloc_free(streams);
+               return NT_STATUS_OBJECT_NAME_NOT_FOUND;         
+       }
+
+       if (found_new == -1) {
+               /* a simple rename */
+               struct xattr_DosStream *s = &streams->streams[found_old];
+               s->name = new_name;
+       } else {
+               if (!overwrite) {
+                       return NT_STATUS_OBJECT_NAME_COLLISION;
+               }
+               if (found_old != found_new) {
+                       /* remove the old one and replace with the new one */
+                       streams->streams[found_old].name = new_name;
+                       memmove(&streams->streams[found_new],
+                               &streams->streams[found_new+1],
+                               sizeof(streams->streams[0]) *
+                               (streams->num_streams - (found_new+1)));
+                       streams->num_streams--;
+               }
+       }
+
+       status = pvfs_streams_save(pvfs, name, fd, streams);
+       talloc_free(streams);
+
+       /* update the in-memory copy of the name of the open file */
+       talloc_free(name->stream_name);
+       name->stream_name = talloc_strdup(name, new_name);
+
+       return status;
+}
+
+
 /*
   create the xattr for a alternate data stream
 */
@@ -204,7 +347,7 @@ NTSTATUS pvfs_stream_delete(struct pvfs_state *pvfs,
                return status;
        }
 
-       streams = talloc_p(name, struct xattr_DosStreams);
+       streams = talloc(name, struct xattr_DosStreams);
        if (streams == NULL) {
                return NT_STATUS_NO_MEMORY;
        }
@@ -217,7 +360,7 @@ NTSTATUS pvfs_stream_delete(struct pvfs_state *pvfs,
 
        for (i=0;i<streams->num_streams;i++) {
                struct xattr_DosStream *s = &streams->streams[i];
-               if (StrCaseCmp(s->name, name->stream_name) == 0) {
+               if (stream_name_cmp(s->name, name->stream_name) == 0) {
                        memmove(s, s+1, (streams->num_streams - (i+1)) * sizeof(*s));
                        streams->num_streams--;
                        break;
@@ -230,6 +373,54 @@ NTSTATUS pvfs_stream_delete(struct pvfs_state *pvfs,
        return status;
 }
 
+/* 
+   load a stream into a blob
+*/
+static NTSTATUS pvfs_stream_load(struct pvfs_state *pvfs,
+                                TALLOC_CTX *mem_ctx,
+                                struct pvfs_filename *name,
+                                int fd,
+                                size_t estimated_size,
+                                DATA_BLOB *blob)
+{
+       NTSTATUS status;
+
+       status = pvfs_xattr_load(pvfs, mem_ctx, name->full_name, fd, 
+                                XATTR_DOSSTREAM_PREFIX,
+                                name->stream_name, estimated_size, blob);
+
+       if (NT_STATUS_EQUAL(status, NT_STATUS_NOT_FOUND)) {
+               /* try with a case insensitive match */
+               struct xattr_DosStreams *streams;
+               int i;
+
+               streams = talloc(mem_ctx, struct xattr_DosStreams);
+               if (streams == NULL) {
+                       return NT_STATUS_NO_MEMORY;
+               }
+               
+               status = pvfs_streams_load(pvfs, name, fd, streams);
+               if (!NT_STATUS_IS_OK(status)) {
+                       talloc_free(streams);
+                       return NT_STATUS_NOT_FOUND;
+               }
+               for (i=0;i<streams->num_streams;i++) {
+                       struct xattr_DosStream *s = &streams->streams[i];
+                       if (stream_name_cmp(s->name, name->stream_name) == 0) {
+                               status = pvfs_xattr_load(pvfs, mem_ctx, name->full_name, fd, 
+                                                        XATTR_DOSSTREAM_PREFIX,
+                                                        s->name, estimated_size, blob);
+                               talloc_free(streams);
+                               return status;
+                       }
+               }
+               talloc_free(streams);
+               return NT_STATUS_NOT_FOUND;
+       }
+       
+       return status;
+}
+
 /*
   the equvalent of pread() on a stream
 */
@@ -241,8 +432,7 @@ ssize_t pvfs_stream_read(struct pvfs_state *pvfs,
        if (count == 0) {
                return 0;
        }
-       status = pvfs_xattr_load(pvfs, h, h->name->full_name, h->fd, XATTR_DOSSTREAM_PREFIX,
-                                h->name->stream_name, offset+count, &blob);
+       status = pvfs_stream_load(pvfs, h, h->name, h->fd, offset+count, &blob);
        if (!NT_STATUS_IS_OK(status)) {
                errno = EIO;
                return -1;
@@ -271,19 +461,21 @@ ssize_t pvfs_stream_write(struct pvfs_state *pvfs,
        if (count == 0) {
                return 0;
        }
-       if (offset > XATTR_MAX_STREAM_SIZE) {
-               errno = ENOSPC;
-               return -1;
+
+       if (count+offset > XATTR_MAX_STREAM_SIZE) {
+               if (!pvfs->ea_db || count+offset > XATTR_MAX_STREAM_SIZE_TDB) {
+                       errno = ENOSPC;
+                       return -1;
+               }
        }
 
        /* we have to load the existing stream, then modify, then save */
-       status = pvfs_xattr_load(pvfs, h, h->name->full_name, h->fd, XATTR_DOSSTREAM_PREFIX,
-                                h->name->stream_name, offset+count, &blob);
+       status = pvfs_stream_load(pvfs, h, h->name, h->fd, offset+count, &blob);
        if (!NT_STATUS_IS_OK(status)) {
                blob = data_blob(NULL, 0);
        }
        if (count+offset > blob.length) {
-               blob.data = talloc_realloc(blob.data, blob.data, count+offset);
+               blob.data = talloc_realloc(blob.data, blob.data, uint8_t, count+offset);
                if (blob.data == NULL) {
                        errno = ENOMEM;
                        return -1;
@@ -327,19 +519,20 @@ NTSTATUS pvfs_stream_truncate(struct pvfs_state *pvfs,
        DATA_BLOB blob;
 
        if (length > XATTR_MAX_STREAM_SIZE) {
-               return NT_STATUS_DISK_FULL;
+               if (!pvfs->ea_db || length > XATTR_MAX_STREAM_SIZE_TDB) {
+                       return NT_STATUS_DISK_FULL;
+               }
        }
 
        /* we have to load the existing stream, then modify, then save */
-       status = pvfs_xattr_load(pvfs, name, name->full_name, fd, XATTR_DOSSTREAM_PREFIX,
-                                name->stream_name, length, &blob);
+       status = pvfs_stream_load(pvfs, name, name, fd, length, &blob);
        if (!NT_STATUS_IS_OK(status)) {
                return status;
        }
        if (length <= blob.length) {
                blob.length = length;
        } else if (length > blob.length) {
-               blob.data = talloc_realloc(blob.data, blob.data, length);
+               blob.data = talloc_realloc(blob.data, blob.data, uint8_t, length);
                if (blob.data == NULL) {
                        return NT_STATUS_NO_MEMORY;
                }
@@ -349,11 +542,11 @@ NTSTATUS pvfs_stream_truncate(struct pvfs_state *pvfs,
 
        status = pvfs_xattr_save(pvfs, name->full_name, fd, XATTR_DOSSTREAM_PREFIX,
                                 name->stream_name, &blob);
-       data_blob_free(&blob);
 
        if (NT_STATUS_IS_OK(status)) {
                status = pvfs_stream_update_size(pvfs, name, fd, blob.length);
        }
+       data_blob_free(&blob);
 
        return status;
 }