.net 程序集重新签名及引用Publickeytoken更改

背景：

1、标题的好像有点长；

2、这个工具是干什么用的？能看明白标题的人估计能用得上，看懂的估计也不需要。

3、可以根据签名文件更改、也可以根据程序集文件更改。

4、欢迎交流，尽请留言。

原始代码：

代码

1 using System;

2 using System.Linq;

3 using System.Windows.Forms;

4 using System.IO;

5 using Mono.Cecil;

6 using System.Diagnostics;

7 using System.Runtime.InteropServices;

8 using System.Reflection;

9 namespace AssemblyResign

10 {

11 public partial class FormMain : Form

12 {

13 public FormMain()

14 {

15 InitializeComponent();

16 }

17

18 private void btnReSign_Click(object sender, EventArgs e)

19 {

20 if (tBoxSourceAse.Text == "" && tBoxSourceSign.Text == "")

21 {

22 MessageBox.Show("Please select a source assembly or signfile!", "Error");

23 return;

24 }

25 if (lBoxObjectAse.Items.Count == 0)

26 {

27 MessageBox.Show("Please select object assembly!", "Error");

28 return;

29 }

30 AssemblyInfo[] assemblyInfoList = new AssemblyInfo[lBoxObjectAse.Items.Count];

31 for (int i = 0; i < lBoxObjectAse.Items.Count; i++)

32 {

33 assemblyInfoList[i] = new AssemblyInfo { FileName = lBoxObjectAse.Items[i].ToString() };

34 }

35 try

36 {

37 if (tBoxSourceSign.Text != "")

38 {

39 ReSignByKeyFile(assemblyInfoList, tBoxSourceSign.Text.Trim());

40 MessageBox.Show("Resign by signfile ucceed!", "Information");

41 }

42 else

43 {

44 ReSignByAssembly(assemblyInfoList, tBoxSourceAse.Text.Trim());

45 MessageBox.Show("Resign by assembly ucceed!", "Information");

46 }

47 }

48 catch(Exception ex)

49 {

50 MessageBox.Show("Resign error,Detail Infor:" + ex.ToString(), "Information");

51 }

52

53 }

54 private void btnRelink_Click(object sender, EventArgs e)

55 {

56 AssemblyInfo[] assemblyInfoList = new AssemblyInfo[lBoxObjectAse.Items.Count];

57 for (int i = 0; i < lBoxObjectAse.Items.Count; i++)

58 {

59 assemblyInfoList[i] = new AssemblyInfo { FileName = lBoxObjectAse.Items[i].ToString() };

60 }

61 try

62 {

63 ReSignature rs = new ReSignature();

64 rs.ReLink(assemblyInfoList, tBoxSourceSign.Text.Trim());

65 MessageBox.Show("Relink by singfile succeed!", "Information");

66 }

67 catch (Exception ex)

68 {

69 MessageBox.Show("Relink error,Detail Infor:" + ex.ToString(), "Error");

70 }

71 }

72 private void btnSourceAse_Click(object sender, EventArgs e)

73 {

74 OpenFileDialog ofd = new OpenFileDialog();

75 ofd.Filter = "Assembly(*.exe,*.dll)|*.exe;*.dll";

76 ofd.ShowDialog();

77 if (ofd.FileName != "")

78 tBoxSourceAse.Text = ofd.FileName;

79 }

80 private void btnSourceSign_Click(object sender, EventArgs e)

81 {

82 OpenFileDialog ofd = new OpenFileDialog();

83 ofd.Filter = "签名文件(*.snk)|*.snk";

84 ofd.ShowDialog();

85 if (ofd.FileName != "")

86 tBoxSourceSign.Text = ofd.FileName;

87 }

88 private void btnObjectAse_Click(object sender, EventArgs e)

89 {

90 OpenFileDialog ofd = new OpenFileDialog();

91 ofd.Multiselect = true;

92 ofd.Filter = "应用程序集(*.exe,*.dll)|*.exe;*.dll";

93 ofd.ShowDialog();

94 if (ofd.FileNames.Length > 0)

95 {

96 lBoxObjectAse.Items.Clear();

97 foreach(string strFileName in ofd.FileNames)

98 {

99 lBoxObjectAse.Items.Add(strFileName);

100 }

101 }

102 }

103

104 private void ReSignByAssembly(AssemblyInfo[] assemblyInfoList, string assemblyFileName)

105 {

106 byte[] publicKey = null;

107 byte[] publicKeyToken = null;

108 Assembly assembly = Assembly.LoadFile(assemblyFileName);

109 publicKey = assembly.GetName().GetPublicKey();

110 publicKeyToken = assembly.GetName().GetPublicKeyToken();

111

112 foreach (AssemblyInfo assemblyInfo in assemblyInfoList)

113 {

114 AssemblyDefinition ass = AssemblyFactory.GetAssembly(assemblyInfo.FileName);

115 ass.Name.PublicKey = publicKey;

116 ass.Name.PublicKeyToken = publicKeyToken;

117 AssemblyFactory.SaveAssembly(ass, assemblyInfo.FileName);

118 }

119 }

120 private void ReSignByKeyFile(AssemblyInfo[] assemblyInfoList, string keyFileName)

121 {

122 ReSignature rs = new ReSignature();

123 rs.ReSign(assemblyInfoList, keyFileName);

124 }

125

126 }

127

128 #region 重新签名

129 public class ReSignature

130 {

131 [DllImport("mscoree.dll", EntryPoint = "StrongNameKeyDelete", CharSet = CharSet.Auto)]

132 public static extern bool StrongNameKeyDelete(string wszKeyContainer);

133 [DllImport("mscoree.dll", EntryPoint = "StrongNameKeyInstall", CharSet = CharSet.Auto)]

134 public static extern bool StrongNameKeyInstall([MarshalAs(UnmanagedType.LPWStr)] string wszKeyContainer, [MarshalAs(UnmanagedType.LPArray, SizeParamIndex = 2, SizeConst = 0)] byte[]
pbKeyBlob, int arg0);

135 [DllImport("mscoree.dll", EntryPoint = "StrongNameSignatureGeneration", CharSet = CharSet.Auto)]

136 public static extern bool StrongNameSignatureGeneration(string wszFilePath, string wszKeyContainer, IntPtr pbKeyBlob, int cbKeyBlob, int ppbSignatureBlob, int pcbSignatureBlob);

137 [DllImport("mscoree.dll", EntryPoint = "StrongNameErrorInfo", CharSet = CharSet.Auto)]

138 public static extern uint StrongNameErrorInfo();

139 [DllImport("mscoree.dll", EntryPoint = "StrongNameTokenFromPublicKey", CharSet = CharSet.Auto)]

140 public static extern bool StrongNameTokenFromPublicKey(byte[] pbPublicKeyBlob, int cbPublicKeyBlob, out IntPtr ppbStrongNameToken, out int pcbStrongNameToken);

141 [DllImport("mscoree.dll", EntryPoint = "StrongNameFreeBuffer", CharSet = CharSet.Auto)]

142 public static extern void StrongNameFreeBuffer(IntPtr pbMemory);

143

144 private byte[] GetNewKey(string keyFileName)

145 {

146 using (FileStream keyPairStream = File.OpenRead(keyFileName))

147 {

148 return new StrongNameKeyPair(keyPairStream).PublicKey;

149 }

150 }

151 private byte[] GetPublicKeyToken(string keyFileName)

152 {

153 try

154 {

155 byte[] newPublicKey;

156 using (FileStream keyPairStream = File.OpenRead(keyFileName))

157 {

158 newPublicKey = new StrongNameKeyPair(keyPairStream).PublicKey;

159 }

160 int pcbStrongNameToken;

161 IntPtr ppbStrongNameToken;

162 StrongNameTokenFromPublicKey(newPublicKey, newPublicKey.Length, out ppbStrongNameToken, out pcbStrongNameToken);

163 var token = new byte[pcbStrongNameToken];

164 Marshal.Copy(ppbStrongNameToken, token, 0, pcbStrongNameToken);

165 StrongNameFreeBuffer(ppbStrongNameToken);

166 return token;

167 }

168 catch (Exception)

169 {

170 return null;

171 }

172 }

173 public void RunSnTool(string assemblyName, string keyFileName)

174 {

175 //SN工具进程

176 Process snProcess = new Process

177 {

178 StartInfo =

179 {

180 FileName = "SN.exe",

181 Arguments = string.Format("/Ra {0} {1}", assemblyName, keyFileName)

182 }

183 };

184 snProcess.Start();

185 snProcess.WaitForExit();

186 }

187 public void ReSign(AssemblyInfo[] assemblyInfoList, string keyFileName)

188 {

189 //用KEY文件建立密钥容器

190 string wszKeyContainer = Guid.NewGuid().ToString();

191 byte[] pbKeyBlob = File.ReadAllBytes(keyFileName);

192 StrongNameKeyInstall(wszKeyContainer, pbKeyBlob, pbKeyBlob.Length);

193 byte[] publicKey = GetNewKey(keyFileName);

194

195 foreach (AssemblyInfo assemblyInfo in assemblyInfoList)

196 {

197 string assemblyName = assemblyInfo.FileName;

198 AssemblyDefinition asm = AssemblyFactory.GetAssembly(assemblyName);

199 asm.Name.PublicKey = publicKey;

200 AssemblyFactory.SaveAssembly(asm, assemblyName);

201 StrongNameSignatureGeneration(assemblyName, wszKeyContainer, IntPtr.Zero, 0, 0, 0);

202 }

203 //删除新建的密钥容器

204 StrongNameKeyDelete(wszKeyContainer);

205 }

206 public void ReLink(AssemblyInfo[] assemblyInfoList, string keyFileName)

207 {

208 byte[] publicKeyToken = GetPublicKeyToken(keyFileName);

209

210 if (publicKeyToken == null)

211 {

212 return;

213 } //初始化一个要重新签名的程序信息列表

214 //获得每个程序集的名称

215 foreach (AssemblyInfo assemblyInfo in assemblyInfoList)

216 {

217 assemblyInfo.FullName = AssemblyFactory.GetAssembly(assemblyInfo.FileName).Name.FullName;

218 }

219 //检查是否被引用,是的话,就替换PublicKeyToken

220 foreach (AssemblyInfo assemblyInfo in assemblyInfoList)

221 {

222 AssemblyDefinition assembly = AssemblyFactory.GetAssembly(assemblyInfo.FileName);

223 foreach (ModuleDefinition module in assembly.Modules)

224

225 foreach (AssemblyNameReference reference in module.AssemblyReferences)

226 {

227 if (assemblyInfoList.Any(a => a.FullName == reference.FullName))

228 {

229 reference.PublicKeyToken = publicKeyToken;

230 AssemblyFactory.SaveAssembly(assembly, assemblyInfo.FileName);

231 }

232 }

233 }

234 }

235 }

236 #endregion

237

238 public class AssemblyInfo

239 {

240 public string FileName { get; set; }

241 public string FullName { get; set; }

242 }

243 }