/rocks/Mono.Cecil.Rocks/MethodBodyRocks.cs

http://github.com/jbevain/cecil · C# · 412 lines · 379 code · 24 blank · 9 comment · 44 complexity · fd516f8c1de2acf2cc59872c2a5a2030 MD5 · raw file 

    

  1. //
    2. 

  2. // Author:
    3. 

  3. //   Jb Evain (jbevain@gmail.com)
    4. 

  4. //
    5. 

  5. // Copyright (c) 2008 - 2015 Jb Evain
    6. 

  6. // Copyright (c) 2008 - 2011 Novell, Inc.
    7. 

  7. //
    8. 

  8. // Licensed under the MIT/X11 license.
    9. 

  9. //
    10. 

  10. 

  11. using System;
    12. 

  12. 

  13. using Mono.Cecil.Cil;
    14. 

  14. 

  15. namespace Mono.Cecil.Rocks {
    16. 

  16. 

  17. #if INSIDE_ROCKS
    18. 

  18. 	public
    19. 

  19. #endif
    20. 

  20. 	static class MethodBodyRocks {
    21. 

  21. 

  22. 		public static void SimplifyMacros (this MethodBody self)
    23. 

  23. 		{
    24. 

  24. 			if (self == null)
    25. 

  25. 				throw new ArgumentNullException ("self");
    26. 

  26. 

  27. 			foreach (var instruction in self.Instructions) {
    28. 

  28. 				if (instruction.OpCode.OpCodeType != OpCodeType.Macro)
    29. 

  29. 					continue;
    30. 

  30. 

  31. 				switch (instruction.OpCode.Code) {
    32. 

  32. 				case Code.Ldarg_0:
    33. 

  33. 					ExpandMacro (instruction, OpCodes.Ldarg, self.GetParameter (0));
    34. 

  34. 					break;
    35. 

  35. 				case Code.Ldarg_1:
    36. 

  36. 					ExpandMacro (instruction, OpCodes.Ldarg, self.GetParameter (1));
    37. 

  37. 					break;
    38. 

  38. 				case Code.Ldarg_2:
    39. 

  39. 					ExpandMacro (instruction, OpCodes.Ldarg, self.GetParameter (2));
    40. 

  40. 					break;
    41. 

  41. 				case Code.Ldarg_3:
    42. 

  42. 					ExpandMacro (instruction, OpCodes.Ldarg, self.GetParameter (3));
    43. 

  43. 					break;
    44. 

  44. 				case Code.Ldloc_0:
    45. 

  45. 					ExpandMacro (instruction, OpCodes.Ldloc, self.Variables [0]);
    46. 

  46. 					break;
    47. 

  47. 				case Code.Ldloc_1:
    48. 

  48. 					ExpandMacro (instruction, OpCodes.Ldloc, self.Variables [1]);
    49. 

  49. 					break;
    50. 

  50. 				case Code.Ldloc_2:
    51. 

  51. 					ExpandMacro (instruction, OpCodes.Ldloc, self.Variables [2]);
    52. 

  52. 					break;
    53. 

  53. 				case Code.Ldloc_3:
    54. 

  54. 					ExpandMacro (instruction, OpCodes.Ldloc, self.Variables [3]);
    55. 

  55. 					break;
    56. 

  56. 				case Code.Stloc_0:
    57. 

  57. 					ExpandMacro (instruction, OpCodes.Stloc, self.Variables [0]);
    58. 

  58. 					break;
    59. 

  59. 				case Code.Stloc_1:
    60. 

  60. 					ExpandMacro (instruction, OpCodes.Stloc, self.Variables [1]);
    61. 

  61. 					break;
    62. 

  62. 				case Code.Stloc_2:
    63. 

  63. 					ExpandMacro (instruction, OpCodes.Stloc, self.Variables [2]);
    64. 

  64. 					break;
    65. 

  65. 				case Code.Stloc_3:
    66. 

  66. 					ExpandMacro (instruction, OpCodes.Stloc, self.Variables [3]);
    67. 

  67. 					break;
    68. 

  68. 				case Code.Ldarg_S:
    69. 

  69. 					instruction.OpCode = OpCodes.Ldarg;
    70. 

  70. 					break;
    71. 

  71. 				case Code.Ldarga_S:
    72. 

  72. 					instruction.OpCode = OpCodes.Ldarga;
    73. 

  73. 					break;
    74. 

  74. 				case Code.Starg_S:
    75. 

  75. 					instruction.OpCode = OpCodes.Starg;
    76. 

  76. 					break;
    77. 

  77. 				case Code.Ldloc_S:
    78. 

  78. 					instruction.OpCode = OpCodes.Ldloc;
    79. 

  79. 					break;
    80. 

  80. 				case Code.Ldloca_S:
    81. 

  81. 					instruction.OpCode = OpCodes.Ldloca;
    82. 

  82. 					break;
    83. 

  83. 				case Code.Stloc_S:
    84. 

  84. 					instruction.OpCode = OpCodes.Stloc;
    85. 

  85. 					break;
    86. 

  86. 				case Code.Ldc_I4_M1:
    87. 

  87. 					ExpandMacro (instruction, OpCodes.Ldc_I4, -1);
    88. 

  88. 					break;
    89. 

  89. 				case Code.Ldc_I4_0:
    90. 

  90. 					ExpandMacro (instruction, OpCodes.Ldc_I4, 0);
    91. 

  91. 					break;
    92. 

  92. 				case Code.Ldc_I4_1:
    93. 

  93. 					ExpandMacro (instruction, OpCodes.Ldc_I4, 1);
    94. 

  94. 					break;
    95. 

  95. 				case Code.Ldc_I4_2:
    96. 

  96. 					ExpandMacro (instruction, OpCodes.Ldc_I4, 2);
    97. 

  97. 					break;
    98. 

  98. 				case Code.Ldc_I4_3:
    99. 

  99. 					ExpandMacro (instruction, OpCodes.Ldc_I4, 3);
    100. 

  100. 					break;
    101. 

  101. 				case Code.Ldc_I4_4:
    102. 

  102. 					ExpandMacro (instruction, OpCodes.Ldc_I4, 4);
    103. 

  103. 					break;
    104. 

  104. 				case Code.Ldc_I4_5:
    105. 

  105. 					ExpandMacro (instruction, OpCodes.Ldc_I4, 5);
    106. 

  106. 					break;
    107. 

  107. 				case Code.Ldc_I4_6:
    108. 

  108. 					ExpandMacro (instruction, OpCodes.Ldc_I4, 6);
    109. 

  109. 					break;
    110. 

  110. 				case Code.Ldc_I4_7:
    111. 

  111. 					ExpandMacro (instruction, OpCodes.Ldc_I4, 7);
    112. 

  112. 					break;
    113. 

  113. 				case Code.Ldc_I4_8:
    114. 

  114. 					ExpandMacro (instruction, OpCodes.Ldc_I4, 8);
    115. 

  115. 					break;
    116. 

  116. 				case Code.Ldc_I4_S:
    117. 

  117. 					ExpandMacro (instruction, OpCodes.Ldc_I4, (int) (sbyte) instruction.Operand);
    118. 

  118. 					break;
    119. 

  119. 				case Code.Br_S:
    120. 

  120. 					instruction.OpCode = OpCodes.Br;
    121. 

  121. 					break;
    122. 

  122. 				case Code.Brfalse_S:
    123. 

  123. 					instruction.OpCode = OpCodes.Brfalse;
    124. 

  124. 					break;
    125. 

  125. 				case Code.Brtrue_S:
    126. 

  126. 					instruction.OpCode = OpCodes.Brtrue;
    127. 

  127. 					break;
    128. 

  128. 				case Code.Beq_S:
    129. 

  129. 					instruction.OpCode = OpCodes.Beq;
    130. 

  130. 					break;
    131. 

  131. 				case Code.Bge_S:
    132. 

  132. 					instruction.OpCode = OpCodes.Bge;
    133. 

  133. 					break;
    134. 

  134. 				case Code.Bgt_S:
    135. 

  135. 					instruction.OpCode = OpCodes.Bgt;
    136. 

  136. 					break;
    137. 

  137. 				case Code.Ble_S:
    138. 

  138. 					instruction.OpCode = OpCodes.Ble;
    139. 

  139. 					break;
    140. 

  140. 				case Code.Blt_S:
    141. 

  141. 					instruction.OpCode = OpCodes.Blt;
    142. 

  142. 					break;
    143. 

  143. 				case Code.Bne_Un_S:
    144. 

  144. 					instruction.OpCode = OpCodes.Bne_Un;
    145. 

  145. 					break;
    146. 

  146. 				case Code.Bge_Un_S:
    147. 

  147. 					instruction.OpCode = OpCodes.Bge_Un;
    148. 

  148. 					break;
    149. 

  149. 				case Code.Bgt_Un_S:
    150. 

  150. 					instruction.OpCode = OpCodes.Bgt_Un;
    151. 

  151. 					break;
    152. 

  152. 				case Code.Ble_Un_S:
    153. 

  153. 					instruction.OpCode = OpCodes.Ble_Un;
    154. 

  154. 					break;
    155. 

  155. 				case Code.Blt_Un_S:
    156. 

  156. 					instruction.OpCode = OpCodes.Blt_Un;
    157. 

  157. 					break;
    158. 

  158. 				case Code.Leave_S:
    159. 

  159. 					instruction.OpCode = OpCodes.Leave;
    160. 

  160. 					break;
    161. 

  161. 				}
    162. 

  162. 			}
    163. 

  163. 		}
    164. 

  164. 

  165. 		static void ExpandMacro (Instruction instruction, OpCode opcode, object operand)
    166. 

  166. 		{
    167. 

  167. 			instruction.OpCode = opcode;
    168. 

  168. 			instruction.Operand = operand;
    169. 

  169. 		}
    170. 

  170. 

  171. 		static void MakeMacro (Instruction instruction, OpCode opcode)
    172. 

  172. 		{
    173. 

  173. 			instruction.OpCode = opcode;
    174. 

  174. 			instruction.Operand = null;
    175. 

  175. 		}
    176. 

  176. 

  177. 		public static void Optimize (this MethodBody self)
    178. 

  178. 		{
    179. 

  179. 			if (self == null)
    180. 

  180. 				throw new ArgumentNullException ("self");
    181. 

  181. 

  182. 			OptimizeLongs (self);
    183. 

  183. 			OptimizeMacros (self);
    184. 

  184. 		}
    185. 

  185. 

  186. 		static void OptimizeLongs (this MethodBody self)
    187. 

  187. 		{
    188. 

  188. 			for (var i = 0; i < self.Instructions.Count; i++) {
    189. 

  189. 				var instruction = self.Instructions [i];
    190. 

  190. 				if (instruction.OpCode.Code != Code.Ldc_I8)
    191. 

  191. 					continue;
    192. 

  192. 				var l = (long) instruction.Operand;
    193. 

  193. 				if (l >= int.MaxValue || l <= int.MinValue)
    194. 

  194. 					continue;
    195. 

  195. 				ExpandMacro (instruction, OpCodes.Ldc_I4, (int) l);
    196. 

  196. 				self.Instructions.Insert (++i, Instruction.Create (OpCodes.Conv_I8));
    197. 

  197. 			}
    198. 

  198. 		}
    199. 

  199. 

  200. 		public static void OptimizeMacros (this MethodBody self)
    201. 

  201. 		{
    202. 

  202. 			if (self == null)
    203. 

  203. 				throw new ArgumentNullException ("self");
    204. 

  204. 

  205. 			var method = self.Method;
    206. 

  206. 

  207. 			foreach (var instruction in self.Instructions) {
    208. 

  208. 				int index;
    209. 

  209. 				switch (instruction.OpCode.Code) {
    210. 

  210. 				case Code.Ldarg:
    211. 

  211. 					index = ((ParameterDefinition) instruction.Operand).Index;
    212. 

  212. 					if (index == -1 && instruction.Operand == self.ThisParameter)
    213. 

  213. 						index = 0;
    214. 

  214. 					else if (method.HasThis)
    215. 

  215. 						index++;
    216. 

  216. 

  217. 					switch (index) {
    218. 

  218. 					case 0:
    219. 

  219. 						MakeMacro (instruction, OpCodes.Ldarg_0);
    220. 

  220. 						break;
    221. 

  221. 					case 1:
    222. 

  222. 						MakeMacro (instruction, OpCodes.Ldarg_1);
    223. 

  223. 						break;
    224. 

  224. 					case 2:
    225. 

  225. 						MakeMacro (instruction, OpCodes.Ldarg_2);
    226. 

  226. 						break;
    227. 

  227. 					case 3:
    228. 

  228. 						MakeMacro (instruction, OpCodes.Ldarg_3);
    229. 

  229. 						break;
    230. 

  230. 					default:
    231. 

  231. 						if (index < 256)
    232. 

  232. 							ExpandMacro (instruction, OpCodes.Ldarg_S, instruction.Operand);
    233. 

  233. 						break;
    234. 

  234. 					}
    235. 

  235. 					break;
    236. 

  236. 				case Code.Ldloc:
    237. 

  237. 					index = ((VariableDefinition) instruction.Operand).Index;
    238. 

  238. 					switch (index) {
    239. 

  239. 					case 0:
    240. 

  240. 						MakeMacro (instruction, OpCodes.Ldloc_0);
    241. 

  241. 						break;
    242. 

  242. 					case 1:
    243. 

  243. 						MakeMacro (instruction, OpCodes.Ldloc_1);
    244. 

  244. 						break;
    245. 

  245. 					case 2:
    246. 

  246. 						MakeMacro (instruction, OpCodes.Ldloc_2);
    247. 

  247. 						break;
    248. 

  248. 					case 3:
    249. 

  249. 						MakeMacro (instruction, OpCodes.Ldloc_3);
    250. 

  250. 						break;
    251. 

  251. 					default:
    252. 

  252. 						if (index < 256)
    253. 

  253. 							ExpandMacro (instruction, OpCodes.Ldloc_S, instruction.Operand);
    254. 

  254. 						break;
    255. 

  255. 					}
    256. 

  256. 					break;
    257. 

  257. 				case Code.Stloc:
    258. 

  258. 					index = ((VariableDefinition) instruction.Operand).Index;
    259. 

  259. 					switch (index) {
    260. 

  260. 					case 0:
    261. 

  261. 						MakeMacro (instruction, OpCodes.Stloc_0);
    262. 

  262. 						break;
    263. 

  263. 					case 1:
    264. 

  264. 						MakeMacro (instruction, OpCodes.Stloc_1);
    265. 

  265. 						break;
    266. 

  266. 					case 2:
    267. 

  267. 						MakeMacro (instruction, OpCodes.Stloc_2);
    268. 

  268. 						break;
    269. 

  269. 					case 3:
    270. 

  270. 						MakeMacro (instruction, OpCodes.Stloc_3);
    271. 

  271. 						break;
    272. 

  272. 					default:
    273. 

  273. 						if (index < 256)
    274. 

  274. 							ExpandMacro (instruction, OpCodes.Stloc_S, instruction.Operand);
    275. 

  275. 						break;
    276. 

  276. 					}
    277. 

  277. 					break;
    278. 

  278. 				case Code.Ldarga:
    279. 

  279. 					index = ((ParameterDefinition) instruction.Operand).Index;
    280. 

  280. 					if (index == -1 && instruction.Operand == self.ThisParameter)
    281. 

  281. 						index = 0;
    282. 

  282. 					else if (method.HasThis)
    283. 

  283. 						index++;
    284. 

  284. 					if (index < 256)
    285. 

  285. 						ExpandMacro (instruction, OpCodes.Ldarga_S, instruction.Operand);
    286. 

  286. 					break;
    287. 

  287. 				case Code.Ldloca:
    288. 

  288. 					if (((VariableDefinition) instruction.Operand).Index < 256)
    289. 

  289. 						ExpandMacro (instruction, OpCodes.Ldloca_S, instruction.Operand);
    290. 

  290. 					break;
    291. 

  291. 				case Code.Ldc_I4:
    292. 

  292. 					int i = (int) instruction.Operand;
    293. 

  293. 					switch (i) {
    294. 

  294. 					case -1:
    295. 

  295. 						MakeMacro (instruction, OpCodes.Ldc_I4_M1);
    296. 

  296. 						break;
    297. 

  297. 					case 0:
    298. 

  298. 						MakeMacro (instruction, OpCodes.Ldc_I4_0);
    299. 

  299. 						break;
    300. 

  300. 					case 1:
    301. 

  301. 						MakeMacro (instruction, OpCodes.Ldc_I4_1);
    302. 

  302. 						break;
    303. 

  303. 					case 2:
    304. 

  304. 						MakeMacro (instruction, OpCodes.Ldc_I4_2);
    305. 

  305. 						break;
    306. 

  306. 					case 3:
    307. 

  307. 						MakeMacro (instruction, OpCodes.Ldc_I4_3);
    308. 

  308. 						break;
    309. 

  309. 					case 4:
    310. 

  310. 						MakeMacro (instruction, OpCodes.Ldc_I4_4);
    311. 

  311. 						break;
    312. 

  312. 					case 5:
    313. 

  313. 						MakeMacro (instruction, OpCodes.Ldc_I4_5);
    314. 

  314. 						break;
    315. 

  315. 					case 6:
    316. 

  316. 						MakeMacro (instruction, OpCodes.Ldc_I4_6);
    317. 

  317. 						break;
    318. 

  318. 					case 7:
    319. 

  319. 						MakeMacro (instruction, OpCodes.Ldc_I4_7);
    320. 

  320. 						break;
    321. 

  321. 					case 8:
    322. 

  322. 						MakeMacro (instruction, OpCodes.Ldc_I4_8);
    323. 

  323. 						break;
    324. 

  324. 					default:
    325. 

  325. 						if (i >= -128 && i < 128)
    326. 

  326. 							ExpandMacro (instruction, OpCodes.Ldc_I4_S, (sbyte) i);
    327. 

  327. 						break;
    328. 

  328. 					}
    329. 

  329. 					break;
    330. 

  330. 				}
    331. 

  331. 			}
    332. 

  332. 

  333. 			OptimizeBranches (self);
    334. 

  334. 		}
    335. 

  335. 

  336. 		static void OptimizeBranches (MethodBody body)
    337. 

  337. 		{
    338. 

  338. 			ComputeOffsets (body);
    339. 

  339. 

  340. 			foreach (var instruction in body.Instructions) {
    341. 

  341. 				if (instruction.OpCode.OperandType != OperandType.InlineBrTarget)
    342. 

  342. 					continue;
    343. 

  343. 

  344. 				if (OptimizeBranch (instruction))
    345. 

  345. 					ComputeOffsets (body);
    346. 

  346. 			}
    347. 

  347. 		}
    348. 

  348. 

  349. 		static bool OptimizeBranch (Instruction instruction)
    350. 

  350. 		{
    351. 

  351. 			var offset = ((Instruction) instruction.Operand).Offset - (instruction.Offset + instruction.OpCode.Size + 4);
    352. 

  352. 			if (!(offset >= -128 && offset <= 127))
    353. 

  353. 				return false;
    354. 

  354. 

  355. 			switch (instruction.OpCode.Code) {
    356. 

  356. 			case Code.Br:
    357. 

  357. 				instruction.OpCode = OpCodes.Br_S;
    358. 

  358. 				break;
    359. 

  359. 			case Code.Brfalse:
    360. 

  360. 				instruction.OpCode = OpCodes.Brfalse_S;
    361. 

  361. 				break;
    362. 

  362. 			case Code.Brtrue:
    363. 

  363. 				instruction.OpCode = OpCodes.Brtrue_S;
    364. 

  364. 				break;
    365. 

  365. 			case Code.Beq:
    366. 

  366. 				instruction.OpCode = OpCodes.Beq_S;
    367. 

  367. 				break;
    368. 

  368. 			case Code.Bge:
    369. 

  369. 				instruction.OpCode = OpCodes.Bge_S;
    370. 

  370. 				break;
    371. 

  371. 			case Code.Bgt:
    372. 

  372. 				instruction.OpCode = OpCodes.Bgt_S;
    373. 

  373. 				break;
    374. 

  374. 			case Code.Ble:
    375. 

  375. 				instruction.OpCode = OpCodes.Ble_S;
    376. 

  376. 				break;
    377. 

  377. 			case Code.Blt:
    378. 

  378. 				instruction.OpCode = OpCodes.Blt_S;
    379. 

  379. 				break;
    380. 

  380. 			case Code.Bne_Un:
    381. 

  381. 				instruction.OpCode = OpCodes.Bne_Un_S;
    382. 

  382. 				break;
    383. 

  383. 			case Code.Bge_Un:
    384. 

  384. 				instruction.OpCode = OpCodes.Bge_Un_S;
    385. 

  385. 				break;
    386. 

  386. 			case Code.Bgt_Un:
    387. 

  387. 				instruction.OpCode = OpCodes.Bgt_Un_S;
    388. 

  388. 				break;
    389. 

  389. 			case Code.Ble_Un:
    390. 

  390. 				instruction.OpCode = OpCodes.Ble_Un_S;
    391. 

  391. 				break;
    392. 

  392. 			case Code.Blt_Un:
    393. 

  393. 				instruction.OpCode = OpCodes.Blt_Un_S;
    394. 

  394. 				break;
    395. 

  395. 			case Code.Leave:
    396. 

  396. 				instruction.OpCode = OpCodes.Leave_S;
    397. 

  397. 				break;
    398. 

  398. 			}
    399. 

  399. 

  400. 			return true;
    401. 

  401. 		}
    402. 

  402. 

  403. 		static void ComputeOffsets (MethodBody body)
    404. 

  404. 		{
    405. 

  405. 			var offset = 0;
    406. 

  406. 			foreach (var instruction in body.Instructions) {
    407. 

  407. 				instruction.Offset = offset;
    408. 

  408. 				offset += instruction.GetSize ();
    409. 

  409. 			}
    410. 

  410. 		}
    411. 

  411. 	}
    412. 

  412. }
    413.