1 //
2 // EmbeddedResource.cs
3 //
4 // Author:
5 //   Jb Evain (jbevain@gmail.com)
6 //
7 // Copyright (c) 2008 - 2011 Jb Evain
8 //
9 // Permission is hereby granted, free of charge, to any person obtaining
10 // a copy of this software and associated documentation files (the
11 // "Software"), to deal in the Software without restriction, including
12 // without limitation the rights to use, copy, modify, merge, publish,
13 // distribute, sublicense, and/or sell copies of the Software, and to
14 // permit persons to whom the Software is furnished to do so, subject to
15 // the following conditions:
16 //
17 // The above copyright notice and this permission notice shall be
18 // included in all copies or substantial portions of the Software.
19 //
20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
27 //
28 
29 using System;
30 using System.IO;
31 
32 namespace Mono.Cecil {
33 
34 	public sealed class EmbeddedResource : Resource {
35 
36 		readonly MetadataReader reader;
37 
38 		uint? offset;
39 		byte [] data;
40 		Stream stream;
41 
42 		public override ResourceType ResourceType {
43 			get { return ResourceType.Embedded; }
44 		}
45 
EmbeddedResource(string name, ManifestResourceAttributes attributes, byte [] data)46 		public EmbeddedResource (string name, ManifestResourceAttributes attributes, byte [] data) :
47 			base (name, attributes)
48 		{
49 			this.data = data;
50 		}
51 
EmbeddedResource(string name, ManifestResourceAttributes attributes, Stream stream)52 		public EmbeddedResource (string name, ManifestResourceAttributes attributes, Stream stream) :
53 			base (name, attributes)
54 		{
55 			this.stream = stream;
56 		}
57 
EmbeddedResource(string name, ManifestResourceAttributes attributes, uint offset, MetadataReader reader)58 		internal EmbeddedResource (string name, ManifestResourceAttributes attributes, uint offset, MetadataReader reader)
59 			: base (name, attributes)
60 		{
61 			this.offset = offset;
62 			this.reader = reader;
63 		}
64 
GetResourceStream()65 		public Stream GetResourceStream ()
66 		{
67 			if (stream != null)
68 				return stream;
69 
70 			if (data != null)
71 				return new MemoryStream (data);
72 
73 			if (offset.HasValue)
74 				return reader.GetManagedResourceStream (offset.Value);
75 
76 			throw new InvalidOperationException ();
77 		}
78 
GetResourceData()79 		public byte [] GetResourceData ()
80 		{
81 			if (stream != null)
82 				return ReadStream (stream);
83 
84 			if (data != null)
85 				return data;
86 
87 			if (offset.HasValue)
88 				return reader.GetManagedResourceStream (offset.Value).ToArray ();
89 
90 			throw new InvalidOperationException ();
91 		}
92 
ReadStream(Stream stream)93 		static byte [] ReadStream (Stream stream)
94 		{
95 			var length = (int) stream.Length;
96 			var data = new byte [length];
97 			int offset = 0, read;
98 
99 			while ((read = stream.Read (data, offset, length - offset)) > 0)
100 				offset += read;
101 
102 			return data;
103 		}
104 	}
105 }
106