/*
    Copyright (C) 2011 de4dot@gmail.com

    This file is part of de4dot.

    de4dot 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 3 of the License, or
    (at your option) any later version.

    de4dot is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with de4dot.  If not, see <http://www.gnu.org/licenses/>.
*/

using System;
using System.Collections.Generic;
using Mono.Cecil;
using Mono.Cecil.Cil;
using de4dot.blocks;

namespace de4dot.code.deobfuscators {
	class InitializedDataCreator {
		ModuleDefinition module;
		Dictionary<long, TypeDefinition> sizeToArrayType = new Dictionary<long, TypeDefinition>();
		TypeDefinition ourType;
		TypeReference valueType;
		int unique = 0;
		MethodReference initializeArrayMethod;

		public MethodReference InitializeArrayMethod {
			get { return createInitializeArrayMethod(); }
		}

		public InitializedDataCreator(ModuleDefinition module) {
			this.module = module;
		}

		MethodReference createInitializeArrayMethod() {
			if (initializeArrayMethod == null) {
				var runtimeHelpersType = DotNetUtils.findOrCreateTypeReference(module, module.TypeSystem.Corlib as AssemblyNameReference, "System.Runtime.CompilerServices", "RuntimeHelpers", false);
				initializeArrayMethod = new MethodReference("InitializeArray", module.TypeSystem.Void, runtimeHelpersType);
				var systemArrayType = DotNetUtils.findOrCreateTypeReference(module, module.TypeSystem.Corlib as AssemblyNameReference, "System", "Array", false);
				var runtimeFieldHandleType = DotNetUtils.findOrCreateTypeReference(module, module.TypeSystem.Corlib as AssemblyNameReference, "System", "RuntimeFieldHandle", true);
				initializeArrayMethod.Parameters.Add(new ParameterDefinition(systemArrayType));
				initializeArrayMethod.Parameters.Add(new ParameterDefinition(runtimeFieldHandleType));
			}
			return initializeArrayMethod;
		}

		public void addInitializeArrayCode(Block block, int start, int numToRemove, TypeReference elementType, byte[] data) {
			int index = start;
			block.replace(index++, numToRemove, DotNetUtils.createLdci4(data.Length / elementType.PrimitiveSize));
			block.insert(index++, Instruction.Create(OpCodes.Newarr, elementType));
			block.insert(index++, Instruction.Create(OpCodes.Dup));
			block.insert(index++, Instruction.Create(OpCodes.Ldtoken, create(data)));
			block.insert(index++, Instruction.Create(OpCodes.Call, InitializeArrayMethod));
		}

		void createOurType() {
			if (ourType != null)
				return;

			var attrs = TypeAttributes.NotPublic | TypeAttributes.AutoLayout |
						TypeAttributes.Class | TypeAttributes.AnsiClass;
			ourType = new TypeDefinition("", string.Format("<PrivateImplementationDetails>{0}", Guid.NewGuid().ToString("B")), attrs, module.TypeSystem.Object);
			ourType.MetadataToken = DotNetUtils.nextTypeDefToken();
			module.Types.Add(ourType);
		}

		TypeDefinition getArrayType(long size) {
			createOurType();

			TypeDefinition arrayType;
			if (sizeToArrayType.TryGetValue(size, out arrayType))
				return arrayType;

			if (valueType == null)
				valueType = DotNetUtils.findOrCreateTypeReference(module, module.TypeSystem.Corlib as AssemblyNameReference, "System", "ValueType", false);
			var attrs = TypeAttributes.NestedPrivate | TypeAttributes.ExplicitLayout |
						TypeAttributes.Class | TypeAttributes.Sealed | TypeAttributes.AnsiClass;
			arrayType = new TypeDefinition("", string.Format("__StaticArrayInitTypeSize={0}", size), attrs, valueType);
			arrayType.MetadataToken = DotNetUtils.nextTypeDefToken();
			ourType.NestedTypes.Add(arrayType);
			sizeToArrayType[size] = arrayType;
			arrayType.ClassSize = (int)size;
			arrayType.PackingSize = 1;
			arrayType.IsValueType = true;
			return arrayType;
		}

		public FieldDefinition create(byte[] data) {
			var arrayType = getArrayType(data.LongLength);
			var attrs = FieldAttributes.Assembly | FieldAttributes.Static;
			var field = new FieldDefinition(string.Format("field_{0}", unique++), attrs, arrayType);
			field.Attributes |= FieldAttributes.HasFieldRVA;
			field.MetadataToken = DotNetUtils.nextFieldToken();
			ourType.Fields.Add(field);
			var iv = new byte[data.Length];
			Array.Copy(data, iv, data.Length);
			field.InitialValue = iv;
			return field;
		}
	}
}