Beispiel #1
0
        public virtual async Task CommitAsync(
            bool gotDestAttributes,
            Attributes sourceAttributes,
            Func <object, object, Task> setCustomAttributes,
            CancellationToken cancellationToken)
        {
            FileRequestOptions fileRequestOptions = Utils.GenerateFileRequestOptions(this.destLocation.FileRequestOptions);
            OperationContext   operationContext   = Utils.GenerateOperationContext(this.transferContext);

            if (!gotDestAttributes)
            {
                await this.destFile.FetchAttributesAsync(
                    Utils.GenerateConditionWithCustomerCondition(this.destLocation.AccessCondition),
                    fileRequestOptions,
                    operationContext,
                    cancellationToken);
            }

            var originalMetadata = new Dictionary <string, string>(this.destFile.Metadata);

            SingleObjectTransfer transferInstance = this.transferJob.Transfer;

            Utils.SetAttributes(this.destFile, sourceAttributes, transferInstance.PreserveSMBAttributes);

            if ((PreserveSMBPermissions.None != transferInstance.PreserveSMBPermissions) &&
                !string.IsNullOrEmpty(sourceAttributes.PortableSDDL))
            {
                if (sourceAttributes.PortableSDDL.Length >= Constants.MaxSDDLLengthInProperties)
                {
                    string permissionKey = null;
                    var    sddlCache     = transferInstance.SDDLCache;
                    if (null != sddlCache)
                    {
                        sddlCache.TryGetValue(sourceAttributes.PortableSDDL, out permissionKey);

                        if (null == permissionKey)
                        {
                            permissionKey = await this.destFile.Share.CreateFilePermissionAsync(sourceAttributes.PortableSDDL,
                                                                                                fileRequestOptions,
                                                                                                operationContext,
                                                                                                cancellationToken).ConfigureAwait(false);

                            sddlCache.TryAddValue(sourceAttributes.PortableSDDL, permissionKey);
                        }
                    }
                    else
                    {
                        permissionKey = await this.destFile.Share.CreateFilePermissionAsync(sourceAttributes.PortableSDDL,
                                                                                            fileRequestOptions,
                                                                                            operationContext,
                                                                                            cancellationToken).ConfigureAwait(false);
                    }

                    this.destFile.Properties.FilePermissionKey = permissionKey;
                }
                else
                {
                    this.destFile.FilePermission = sourceAttributes.PortableSDDL;
                }
            }

            await setCustomAttributes(this.transferJob.Source.Instance, this.destFile);

            await this.destFile.SetPropertiesAsync(
                Utils.GenerateConditionWithCustomerCondition(this.destLocation.AccessCondition),
                fileRequestOptions,
                operationContext,
                cancellationToken);

            if (!originalMetadata.DictionaryEquals(this.destFile.Metadata))
            {
                await this.destFile.SetMetadataAsync(
                    Utils.GenerateConditionWithCustomerCondition(this.destLocation.AccessCondition),
                    fileRequestOptions,
                    operationContext,
                    cancellationToken);
            }
        }
        public async Task FetchAttributesAsync(CancellationToken cancellationToken)
        {
            try
            {
                await this.sourceFile.FetchAttributesAsync(
                    null,
                    Utils.GenerateFileRequestOptions(this.sourceLocation.FileRequestOptions),
                    Utils.GenerateOperationContext(this.transferContext),
                    cancellationToken);
            }
#if EXPECT_INTERNAL_WRAPPEDSTORAGEEXCEPTION
            catch (Exception ex) when(ex is StorageException || (ex is AggregateException && ex.InnerException is StorageException))
            {
                var e = ex as StorageException ?? ex.InnerException as StorageException;
#else
            catch (StorageException e)
            {
#endif
                HandleFetchSourceAttributesException(e);
                throw;
            }

            if (string.IsNullOrEmpty(this.sourceLocation.ETag))
            {
                if (0 != this.transferJob.CheckPoint.EntryTransferOffset)
                {
                    throw new InvalidOperationException(Resources.RestartableInfoCorruptedException);
                }

                this.sourceLocation.ETag = this.sourceFile.Properties.ETag;
            }
            else if ((this.transferJob.CheckPoint.EntryTransferOffset > this.sourceFile.Properties.Length) ||
                     (this.transferJob.CheckPoint.EntryTransferOffset < 0))
            {
                throw new InvalidOperationException(Resources.RestartableInfoCorruptedException);
            }

            SingleObjectTransfer transferInstance = this.transferJob.Transfer;

            this.sourceAttributes = Utils.GenerateAttributes(this.sourceFile, transferInstance.PreserveSMBAttributes);

            if (transferInstance.PreserveSMBPermissions != PreserveSMBPermissions.None)
            {
                if (!string.IsNullOrEmpty(this.sourceFile.FilePermission))
                {
                    this.sourceAttributes.PortableSDDL = this.sourceFile.FilePermission;
                }
                else if (!string.IsNullOrEmpty(this.sourceFile.Properties.FilePermissionKey))
                {
                    var sddlCache = transferInstance.SDDLCache;

                    if (null != sddlCache)
                    {
                        string portableSDDL = null;
                        sddlCache.TryGetValue(this.sourceFile.Properties.FilePermissionKey, out portableSDDL);

                        if (null == portableSDDL)
                        {
                            portableSDDL = await this.sourceFile.Share.GetFilePermissionAsync(this.sourceFile.Properties.FilePermissionKey,
                                                                                              Utils.GenerateFileRequestOptions(this.sourceLocation.FileRequestOptions),
                                                                                              Utils.GenerateOperationContext(this.transferContext),
                                                                                              cancellationToken).ConfigureAwait(false);

                            sddlCache.TryAddValue(this.sourceFile.Properties.FilePermissionKey, portableSDDL);
                        }

                        this.sourceAttributes.PortableSDDL = portableSDDL;
                    }
                    else
                    {
                        this.sourceAttributes.PortableSDDL = await this.sourceFile.Share.GetFilePermissionAsync(this.sourceFile.Properties.FilePermissionKey,
                                                                                                                Utils.GenerateFileRequestOptions(this.sourceLocation.FileRequestOptions),
                                                                                                                Utils.GenerateOperationContext(this.transferContext),
                                                                                                                cancellationToken).ConfigureAwait(false);
                    }
                }
                else
                {
                    this.sourceAttributes.PortableSDDL = null;
                }
            }

            this.totalLength = this.sourceFile.Properties.Length;
        }