19
19
import java .io .File ;
20
20
import java .io .OutputStream ;
21
21
import java .io .PrintStream ;
22
- import java .util .List ;
23
22
24
23
import org .apache .commons .io .FileUtils ;
25
24
import org .apache .tools .ant .BuildException ;
26
25
import org .apache .tools .ant .DefaultLogger ;
27
26
import org .apache .tools .ant .Project ;
28
27
import org .apache .tools .ant .ProjectHelper ;
29
- import org .bouncycastle .cms .CMSSignedData ;
30
28
import org .junit .Before ;
31
29
import org .junit .Test ;
32
30
33
31
import net .jsign .msi .MSIFile ;
34
32
import net .jsign .pe .PEFile ;
35
33
import net .jsign .script .PowerShellScript ;
36
34
35
+ import static net .jsign .DigestAlgorithm .*;
37
36
import static org .junit .Assert .*;
38
37
39
38
public class JsignTaskTest {
@@ -177,13 +176,7 @@ public void testSigning() throws Exception {
177
176
assertTrue ("The file " + targetFile + " wasn't changed" , SOURCE_FILE_CRC32 != FileUtils .checksumCRC32 (targetFile ));
178
177
179
178
try (PEFile peFile = new PEFile (targetFile )) {
180
- List <CMSSignedData > signatures = peFile .getSignatures ();
181
- assertNotNull (signatures );
182
- assertEquals (1 , signatures .size ());
183
-
184
- CMSSignedData signature = signatures .get (0 );
185
-
186
- assertNotNull (signature );
179
+ SignatureAssert .assertSigned (peFile , SHA1 );
187
180
}
188
181
}
189
182
@@ -199,22 +192,10 @@ public void testSigningMultipleFiles() throws Exception {
199
192
assertTrue ("The file " + targetFile2 + " wasn't changed" , SOURCE_FILE_CRC32 != FileUtils .checksumCRC32 (targetFile2 ));
200
193
201
194
try (PEFile peFile = new PEFile (targetFile )) {
202
- List <CMSSignedData > signatures = peFile .getSignatures ();
203
- assertNotNull (signatures );
204
- assertEquals (1 , signatures .size ());
205
-
206
- CMSSignedData signature = signatures .get (0 );
207
-
208
- assertNotNull (signature );
195
+ SignatureAssert .assertSigned (peFile , SHA1 );
209
196
}
210
197
try (PEFile peFile = new PEFile (targetFile2 )) {
211
- List <CMSSignedData > signatures = peFile .getSignatures ();
212
- assertNotNull (signatures );
213
- assertEquals (1 , signatures .size ());
214
-
215
- CMSSignedData signature = signatures .get (0 );
216
-
217
- assertNotNull (signature );
198
+ SignatureAssert .assertSigned (peFile , SHA1 );
218
199
}
219
200
}
220
201
@@ -229,13 +210,7 @@ public void testSigningPowerShell() throws Exception {
229
210
230
211
PowerShellScript script = new PowerShellScript (targetFile );
231
212
232
- List <CMSSignedData > signatures = script .getSignatures ();
233
- assertNotNull (signatures );
234
- assertEquals (1 , signatures .size ());
235
-
236
- CMSSignedData signature = signatures .get (0 );
237
-
238
- assertNotNull (signature );
213
+ SignatureAssert .assertSigned (script , SHA1 );
239
214
}
240
215
241
216
@ Test
@@ -248,13 +223,7 @@ public void testSigningMSI() throws Exception {
248
223
project .executeTarget ("signing-msi" );
249
224
250
225
try (MSIFile file = new MSIFile (targetFile )) {
251
- List <CMSSignedData > signatures = file .getSignatures ();
252
- assertNotNull (signatures );
253
- assertEquals (1 , signatures .size ());
254
-
255
- CMSSignedData signature = signatures .get (0 );
256
-
257
- assertNotNull (signature );
226
+ SignatureAssert .assertSigned (file , SHA1 );
258
227
}
259
228
}
260
229
@@ -265,13 +234,7 @@ public void testSigningPKCS12() throws Exception {
265
234
assertTrue ("The file " + targetFile + " wasn't changed" , SOURCE_FILE_CRC32 != FileUtils .checksumCRC32 (targetFile ));
266
235
267
236
try (PEFile peFile = new PEFile (targetFile )) {
268
- List <CMSSignedData > signatures = peFile .getSignatures ();
269
- assertNotNull (signatures );
270
- assertEquals (1 , signatures .size ());
271
-
272
- CMSSignedData signature = signatures .get (0 );
273
-
274
- assertNotNull (signature );
237
+ SignatureAssert .assertSigned (peFile , SHA256 );
275
238
}
276
239
}
277
240
@@ -282,13 +245,7 @@ public void testSigningPVKSPC() throws Exception {
282
245
assertTrue ("The file " + targetFile + " wasn't changed" , SOURCE_FILE_CRC32 != FileUtils .checksumCRC32 (targetFile ));
283
246
284
247
try (PEFile peFile = new PEFile (targetFile )) {
285
- List <CMSSignedData > signatures = peFile .getSignatures ();
286
- assertNotNull (signatures );
287
- assertEquals (1 , signatures .size ());
288
-
289
- CMSSignedData signature = signatures .get (0 );
290
-
291
- assertNotNull (signature );
248
+ SignatureAssert .assertSigned (peFile , SHA256 );
292
249
}
293
250
}
294
251
@@ -301,13 +258,7 @@ public void testTimestampingAuthenticode() throws Exception {
301
258
assertTrue ("The file " + targetFile2 + " wasn't changed" , SOURCE_FILE_CRC32 != FileUtils .checksumCRC32 (targetFile2 ));
302
259
303
260
try (PEFile peFile = new PEFile (targetFile2 )) {
304
- List <CMSSignedData > signatures = peFile .getSignatures ();
305
- assertNotNull (signatures );
306
- assertEquals (1 , signatures .size ());
307
-
308
- CMSSignedData signature = signatures .get (0 );
309
-
310
- assertNotNull (signature );
261
+ SignatureAssert .assertSigned (peFile , SHA256 );
311
262
}
312
263
}
313
264
@@ -320,13 +271,7 @@ public void testTimestampingRFC3161() throws Exception {
320
271
assertTrue ("The file " + targetFile2 + " wasn't changed" , SOURCE_FILE_CRC32 != FileUtils .checksumCRC32 (targetFile2 ));
321
272
322
273
try (PEFile peFile = new PEFile (targetFile2 )) {
323
- List <CMSSignedData > signatures = peFile .getSignatures ();
324
- assertNotNull (signatures );
325
- assertEquals (1 , signatures .size ());
326
-
327
- CMSSignedData signature = signatures .get (0 );
328
-
329
- assertNotNull (signature );
274
+ SignatureAssert .assertSigned (peFile , SHA256 );
330
275
}
331
276
}
332
277
@@ -339,13 +284,7 @@ public void testReplaceSignature() throws Exception {
339
284
assertTrue ("The file " + targetFile2 + " wasn't changed" , SOURCE_FILE_CRC32 != FileUtils .checksumCRC32 (targetFile2 ));
340
285
341
286
try (PEFile peFile = new PEFile (targetFile2 )) {
342
- List <CMSSignedData > signatures = peFile .getSignatures ();
343
- assertNotNull (signatures );
344
- assertEquals (1 , signatures .size ());
345
-
346
- assertNotNull (signatures .get (0 ));
347
-
348
- assertEquals ("Digest algorithm" , DigestAlgorithm .SHA512 .oid , signatures .get (0 ).getDigestAlgorithmIDs ().iterator ().next ().getAlgorithm ());
287
+ SignatureAssert .assertSigned (peFile , SHA512 );
349
288
}
350
289
}
351
290
0 commit comments