2012-11-01 16:32:42 +08:00
|
|
|
|
/*
|
2014-03-12 05:15:43 +08:00
|
|
|
|
Copyright (C) 2011-2014 de4dot@gmail.com
|
2012-11-01 16:32:42 +08:00
|
|
|
|
|
|
|
|
|
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;
|
2012-12-20 09:06:09 +08:00
|
|
|
|
using dnlib.DotNet.MD;
|
|
|
|
|
using dnlib.DotNet;
|
2012-11-01 16:32:42 +08:00
|
|
|
|
|
2012-11-07 05:25:19 +08:00
|
|
|
|
namespace de4dot.blocks {
|
2012-11-01 16:32:42 +08:00
|
|
|
|
[Serializable]
|
|
|
|
|
public class DumpedMethods {
|
|
|
|
|
Dictionary<uint, DumpedMethod> methods = new Dictionary<uint, DumpedMethod>();
|
|
|
|
|
|
2012-11-07 06:52:17 +08:00
|
|
|
|
public int Count {
|
|
|
|
|
get { return methods.Count; }
|
2012-11-01 16:32:42 +08:00
|
|
|
|
}
|
|
|
|
|
|
2013-01-19 20:03:57 +08:00
|
|
|
|
public void Add(uint token, DumpedMethod info) {
|
2012-11-01 16:32:42 +08:00
|
|
|
|
methods[token] = info;
|
|
|
|
|
}
|
|
|
|
|
|
2013-01-19 20:03:57 +08:00
|
|
|
|
public DumpedMethod Get(MethodDef method) {
|
|
|
|
|
return Get(method.MDToken.ToUInt32());
|
2012-11-01 16:32:42 +08:00
|
|
|
|
}
|
|
|
|
|
|
2013-01-19 20:03:57 +08:00
|
|
|
|
public DumpedMethod Get(uint token) {
|
2012-11-01 16:32:42 +08:00
|
|
|
|
DumpedMethod dm;
|
|
|
|
|
methods.TryGetValue(token, out dm);
|
|
|
|
|
return dm;
|
|
|
|
|
}
|
|
|
|
|
|
2013-01-19 20:03:57 +08:00
|
|
|
|
public void Add(DumpedMethod dm) {
|
2012-11-01 16:32:42 +08:00
|
|
|
|
if (MDToken.ToTable(dm.token) != Table.Method || MDToken.ToRID(dm.token) == 0)
|
|
|
|
|
throw new ArgumentException("Invalid token");
|
|
|
|
|
methods[dm.token] = dm;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|