1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
package cz.crcs.ectester.standalone.output;
import cz.crcs.ectester.common.output.BaseYAMLTestWriter;
import cz.crcs.ectester.common.test.TestSuite;
import cz.crcs.ectester.common.test.Testable;
import cz.crcs.ectester.common.util.ByteUtil;
import cz.crcs.ectester.standalone.ECTesterStandalone;
import cz.crcs.ectester.standalone.test.base.KeyAgreementTestable;
import cz.crcs.ectester.standalone.test.base.KeyGeneratorTestable;
import cz.crcs.ectester.standalone.test.base.SignatureTestable;
import cz.crcs.ectester.standalone.test.base.StandaloneTestable;
import cz.crcs.ectester.standalone.test.suites.StandaloneTestSuite;
import java.io.PrintStream;
import java.security.Key;
import java.security.PrivateKey;
import java.security.PublicKey;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Map;
/**
* @author Jan Jancar johny@neuromancer.sk
*/
public class YAMLTestWriter extends BaseYAMLTestWriter {
public YAMLTestWriter(PrintStream output) {
super(output);
}
private Map<String, Object> keyObject(Key key) {
Map<String, Object> kObject = new LinkedHashMap<>();
if (key == null) {
return kObject;
}
kObject.put("algo", key.getAlgorithm());
kObject.put("format", key.getFormat());
kObject.put("raw", ByteUtil.bytesToHex(key.getEncoded()));
return kObject;
}
private Map<String, Object> kaObject(KeyAgreementTestable kat) {
Map<String, Object> katObject = new LinkedHashMap<>();
katObject.put("algo", kat.getKa().getAlgorithm());
katObject.put("secret", ByteUtil.bytesToHex(kat.getSecret()));
PublicKey pkey = kat.getPublicKey();
katObject.put("pubkey", keyObject(pkey));
PrivateKey skey = kat.getPrivateKey();
katObject.put("privkey", keyObject(skey));
return katObject;
}
private Map<String, Object> kgtObject(KeyGeneratorTestable kgt) {
Map<String, Object> kgtObject = new LinkedHashMap<>();
kgtObject.put("algo", kgt.getKpg().getAlgorithm());
Map<String, Object> keypair = new LinkedHashMap<>();
if (kgt.getKeyPair() != null) {
PublicKey pkey = kgt.getKeyPair().getPublic();
Map<String, Object> pubObject = keyObject(pkey);
keypair.put("pubkey", pubObject);
PrivateKey skey = kgt.getKeyPair().getPrivate();
Map<String, Object> privObject = keyObject(skey);
keypair.put("privkey", privObject);
}
kgtObject.put("keypair", keypair);
return kgtObject;
}
private Map<String, Object> sigObject(SignatureTestable sig) {
Map<String, Object> sigObject = new LinkedHashMap<>();
sigObject.put("algo", sig.getSig().getAlgorithm());
sigObject.put("verified", sig.getVerified());
sigObject.put("raw", ByteUtil.bytesToHex(sig.getSignature()));
return sigObject;
}
private String causeObject(Object cause) {
if (cause == null) {
return "";
} else if (cause instanceof Exception) {
Exception ex = ((Exception) cause);
return ex.getClass().getCanonicalName() + " : " + ex.getMessage();
} else {
return cause.toString();
}
}
@Override
protected Map<String, Object> testableObject(Testable t) {
Map<String, Object> result = new LinkedHashMap<>();
if (t instanceof StandaloneTestable) {
StandaloneTestable<?> testable = (StandaloneTestable) t;
if (t instanceof KeyGeneratorTestable) {
result.put("type", "key-pair-generator");
result.put("key-pair-generator", kgtObject((KeyGeneratorTestable) t));
} else if (t instanceof KeyAgreementTestable) {
result.put("type", "key-agreement");
result.put("key-agreement", kaObject((KeyAgreementTestable) t));
} else if (t instanceof SignatureTestable) {
result.put("type", "signature");
result.put("signature", sigObject((SignatureTestable) t));
}
result.put("stage", testable.getStage().name());
result.put("exception", causeObject(testable.getException()) + causeObject(testable.errorCause()));
}
return result;
}
@Override
protected Map<String, Object> deviceObject(TestSuite suite) {
if (suite instanceof StandaloneTestSuite) {
StandaloneTestSuite standaloneSuite = (StandaloneTestSuite) suite;
Map<String, Object> result = new LinkedHashMap<>();
result.put("type", "library");
result.put("ectester", ECTesterStandalone.VERSION);
result.put("name", standaloneSuite.getLibrary().name());
return result;
}
return null;
}
}
|
