Ejemplo n.º 1
0
        public IDictionary <IMetadataMember, MetadataToken> UnlockMetadata()
        {
            var buffer = new MetadataBuffer(Image);

            buffer.TableStreamBuffer.AddAssembly(Image.Assembly);
            NetDirectory.ResourcesManifest = buffer.ResourcesBuffer.CreateDirectory();

            var buffers = new MetadataStreamBuffer[]
            {
                buffer.TableStreamBuffer,
                buffer.BlobStreamBuffer,
                buffer.GuidStreamBuffer,
                buffer.StringStreamBuffer,
                buffer.UserStringStreamBuffer
            };

            foreach (var streamBuffer in buffers)
            {
                var header = StreamHeaders.FirstOrDefault(x => x.Name == streamBuffer.Name)
                             ?? new MetadataStreamHeader(streamBuffer.Name);
                header.Stream = streamBuffer.CreateStream();
            }

            Image = null;

            return(buffer.TableStreamBuffer.GetNewTokenMapping());
        }
Ejemplo n.º 2
0
        public IDictionary <IMetadataMember, MetadataToken> UnlockMetadata()
        {
            if (!IsLocked)
            {
                throw new InvalidOperationException("Cannot unlock the metadata if it has not already been locked.");
            }

            var image  = Image;
            var buffer = new MetadataBuffer(image);

            // Add assembly to buffer.
            buffer.TableStreamBuffer.AddAssembly(image.Assembly);

            // Create resources.
            NetDirectory.ResourcesManifest = buffer.ResourcesBuffer.CreateDirectory();

            // Unlock metadata.
            Image = null;

            // Replace old streams with new buffers.
            var buffers = new MetadataStreamBuffer[]
            {
                buffer.TableStreamBuffer,
                buffer.BlobStreamBuffer,
                buffer.GuidStreamBuffer,
                buffer.StringStreamBuffer,
                buffer.UserStringStreamBuffer
            };

            foreach (var streamBuffer in buffers)
            {
                var header = StreamHeaders.FirstOrDefault(x => x.Name == streamBuffer.Name);

                if (header == null)
                {
                    header = new MetadataStreamHeader(streamBuffer.Name);
                    StreamHeaders.Add(header);
                }

                header.Stream = streamBuffer.CreateStream();
            }

            // Update managed entrypoint.
            var newTokenMapping = buffer.TableStreamBuffer.GetNewTokenMapping();

            NetDirectory.EntryPointToken = image.ManagedEntrypoint != null
                ? newTokenMapping[image.ManagedEntrypoint].ToUInt32()
                : 0u;

            return(newTokenMapping);
        }
Ejemplo n.º 3
0
        public IDictionary <IMetadataMember, MetadataToken> UnlockMetadata(IMetadataBuilder builder)
        {
            if (!IsLocked)
            {
                throw new InvalidOperationException("Cannot unlock the metadata if it has not already been locked.");
            }

            var image = Image;

            // Construct new metadata streams.
            var buffer = builder.Rebuild(image);

            // Create resources.
            NetDirectory.ResourcesManifest = buffer.ResourcesBuffer.CreateDirectory();

            // Serialize new streams.
            var newStreams = new MetadataStreamBuffer[]
            {
                buffer.TableStreamBuffer,
                buffer.BlobStreamBuffer,
                buffer.GuidStreamBuffer,
                buffer.StringStreamBuffer,
                buffer.UserStringStreamBuffer
            }.ToDictionary(x => x, x => x.CreateStream());

            // Determine new entrypoint token.
            var  newTokenMapping = buffer.TableStreamBuffer.GetNewTokenMapping();
            uint entrypointToken;

            if (image.ManagedEntrypoint == null)
            {
                entrypointToken = 0u;
            }
            else
            {
                if (newTokenMapping.TryGetValue(image.ManagedEntrypoint, out var token))
                {
                    entrypointToken = token.ToUInt32();
                }
                else
                {
                    throw new MemberNotImportedException(image.ManagedEntrypoint);
                }
            }

            // Unlock metadata, commit changes to streams.
            Image = null;
            foreach (var entry in newStreams)
            {
                var header = StreamHeaders.FirstOrDefault(x => x.Name == entry.Key.Name);

                if (header == null)
                {
                    header = new MetadataStreamHeader(entry.Key.Name);
                    StreamHeaders.Add(header);
                }

                header.Stream = entry.Value;
            }

            // Update managed entrypoint.
            NetDirectory.EntryPointToken = entrypointToken;
            return(newTokenMapping);
        }