blob: bb678f0bbb528ea161dc77bb4ecf5b2db566a4db [file] [log] [blame]
Jeff Thompson687ea652013-12-06 11:28:23 -08001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil -*- */
2/**
3 * Copyright (C) 2013 Regents of the University of California.
4 * @author: Jeff Thompson <jefft0@remap.ucla.edu>
5 * See COPYING for copyright and distribution information.
6 */
7
8#include <iostream>
9#include <time.h>
10#include <sys/time.h>
Jeff Thompson70e80802013-12-10 20:21:03 -080011#include <sstream>
Jeff Thompson18dba8b2013-12-09 12:17:40 -080012#include <stdexcept>
Yingdi Yu61ec2722014-01-20 14:22:32 -080013#include <ndn-cpp-dev/data.hpp>
14#include <ndn-cpp-dev/security/key-chain.hpp>
15// #include <ndn-cpp-dev/security/policy/self-verify-policy-manager.hpp>
Alexander Afanasyev84c2bd42014-01-09 22:35:35 -080016
Jeff Thompson687ea652013-12-06 11:28:23 -080017// Hack: Hook directly into non-API functions.
18#include "../src/c/encoding/binary-xml-decoder.h"
19#include "../src/c/data.h"
20#include "../src/c/encoding/binary-xml-data.h"
Yingdi Yu61ec2722014-01-20 14:22:32 -080021#include <ndn-cpp-dev/c/util/crypto.h>
Jeff Thompson687ea652013-12-06 11:28:23 -080022
23using namespace std;
24using namespace ndn;
Jeff Thompson687ea652013-12-06 11:28:23 -080025
26static double
Jeff Thompson18dba8b2013-12-09 12:17:40 -080027getNowSeconds()
Jeff Thompson687ea652013-12-06 11:28:23 -080028{
29 struct timeval t;
30 gettimeofday(&t, 0);
Jeff Thompson18dba8b2013-12-09 12:17:40 -080031 return t.tv_sec + t.tv_usec / 1000000.0;
Jeff Thompson687ea652013-12-06 11:28:23 -080032}
33
Jeff Thompson30d3e6c2013-12-11 12:35:54 -080034static bool
35verifyRsaSignature
36 (uint8_t* signedPortion, size_t signedPortionLength, uint8_t* signatureBits, size_t signatureBitsLength,
37 uint8_t* publicKeyDer, size_t publicKeyDerLength)
38{
39 // Set signedPortionDigest to the digest of the signed portion of the wire encoding.
40 uint8_t signedPortionDigest[SHA256_DIGEST_LENGTH];
41 ndn_digestSha256(signedPortion, signedPortionLength, signedPortionDigest);
42
43 // Verify the signedPortionDigest.
44 // Use a temporary pointer since d2i updates it.
45 const uint8_t *derPointer = publicKeyDer;
46 RSA *rsaPublicKey = d2i_RSA_PUBKEY(NULL, &derPointer, publicKeyDerLength);
47 if (!rsaPublicKey) {
48 // Don't expect this to happen.
49 cout << "Error decoding public key in d2i_RSAPublicKey" << endl;
50 return 0;
51 }
52 int success = RSA_verify
53 (NID_sha256, signedPortionDigest, sizeof(signedPortionDigest), signatureBits, signatureBitsLength, rsaPublicKey);
54 // Free the public key before checking for success.
55 RSA_free(rsaPublicKey);
56
57 // RSA_verify returns 1 for a valid signature.
58 return (success == 1);
59}
60
Jeff Thompson70e80802013-12-10 20:21:03 -080061static uint8_t DEFAULT_PUBLIC_KEY_DER[] = {
620x30, 0x81, 0x9F, 0x30, 0x0D, 0x06, 0x09, 0x2A, 0x86, 0x48, 0x86, 0xF7, 0x0D, 0x01, 0x01, 0x01, 0x05, 0x00, 0x03, 0x81,
630x8D, 0x00, 0x30, 0x81, 0x89, 0x02, 0x81, 0x81, 0x00, 0xE1, 0x7D, 0x30, 0xA7, 0xD8, 0x28, 0xAB, 0x1B, 0x84, 0x0B, 0x17,
640x54, 0x2D, 0xCA, 0xF6, 0x20, 0x7A, 0xFD, 0x22, 0x1E, 0x08, 0x6B, 0x2A, 0x60, 0xD1, 0x6C, 0xB7, 0xF5, 0x44, 0x48, 0xBA,
650x9F, 0x3F, 0x08, 0xBC, 0xD0, 0x99, 0xDB, 0x21, 0xDD, 0x16, 0x2A, 0x77, 0x9E, 0x61, 0xAA, 0x89, 0xEE, 0xE5, 0x54, 0xD3,
660xA4, 0x7D, 0xE2, 0x30, 0xBC, 0x7A, 0xC5, 0x90, 0xD5, 0x24, 0x06, 0x7C, 0x38, 0x98, 0xBB, 0xA6, 0xF5, 0xDC, 0x43, 0x60,
670xB8, 0x45, 0xED, 0xA4, 0x8C, 0xBD, 0x9C, 0xF1, 0x26, 0xA7, 0x23, 0x44, 0x5F, 0x0E, 0x19, 0x52, 0xD7, 0x32, 0x5A, 0x75,
680xFA, 0xF5, 0x56, 0x14, 0x4F, 0x9A, 0x98, 0xAF, 0x71, 0x86, 0xB0, 0x27, 0x86, 0x85, 0xB8, 0xE2, 0xC0, 0x8B, 0xEA, 0x87,
690x17, 0x1B, 0x4D, 0xEE, 0x58, 0x5C, 0x18, 0x28, 0x29, 0x5B, 0x53, 0x95, 0xEB, 0x4A, 0x17, 0x77, 0x9F, 0x02, 0x03, 0x01,
Jeff Thompson9e387fe2013-12-11 13:00:36 -0800700x00, 0x01
Jeff Thompson687ea652013-12-06 11:28:23 -080071};
72
Jeff Thompson70e80802013-12-10 20:21:03 -080073static uint8_t DEFAULT_PRIVATE_KEY_DER[] = {
740x30, 0x82, 0x02, 0x5d, 0x02, 0x01, 0x00, 0x02, 0x81, 0x81, 0x00, 0xe1, 0x7d, 0x30, 0xa7, 0xd8, 0x28, 0xab, 0x1b, 0x84,
750x0b, 0x17, 0x54, 0x2d, 0xca, 0xf6, 0x20, 0x7a, 0xfd, 0x22, 0x1e, 0x08, 0x6b, 0x2a, 0x60, 0xd1, 0x6c, 0xb7, 0xf5, 0x44,
760x48, 0xba, 0x9f, 0x3f, 0x08, 0xbc, 0xd0, 0x99, 0xdb, 0x21, 0xdd, 0x16, 0x2a, 0x77, 0x9e, 0x61, 0xaa, 0x89, 0xee, 0xe5,
770x54, 0xd3, 0xa4, 0x7d, 0xe2, 0x30, 0xbc, 0x7a, 0xc5, 0x90, 0xd5, 0x24, 0x06, 0x7c, 0x38, 0x98, 0xbb, 0xa6, 0xf5, 0xdc,
780x43, 0x60, 0xb8, 0x45, 0xed, 0xa4, 0x8c, 0xbd, 0x9c, 0xf1, 0x26, 0xa7, 0x23, 0x44, 0x5f, 0x0e, 0x19, 0x52, 0xd7, 0x32,
790x5a, 0x75, 0xfa, 0xf5, 0x56, 0x14, 0x4f, 0x9a, 0x98, 0xaf, 0x71, 0x86, 0xb0, 0x27, 0x86, 0x85, 0xb8, 0xe2, 0xc0, 0x8b,
800xea, 0x87, 0x17, 0x1b, 0x4d, 0xee, 0x58, 0x5c, 0x18, 0x28, 0x29, 0x5b, 0x53, 0x95, 0xeb, 0x4a, 0x17, 0x77, 0x9f, 0x02,
810x03, 0x01, 0x00, 0x01, 0x02, 0x81, 0x80, 0x1a, 0x4b, 0xfa, 0x4f, 0xa8, 0xc2, 0xdd, 0x69, 0xa1, 0x15, 0x96, 0x0b, 0xe8,
820x27, 0x42, 0x5a, 0xf9, 0x5c, 0xea, 0x0c, 0xac, 0x98, 0xaa, 0xe1, 0x8d, 0xaa, 0xeb, 0x2d, 0x3c, 0x60, 0x6a, 0xfb, 0x45,
830x63, 0xa4, 0x79, 0x83, 0x67, 0xed, 0xe4, 0x15, 0xc0, 0xb0, 0x20, 0x95, 0x6d, 0x49, 0x16, 0xc6, 0x42, 0x05, 0x48, 0xaa,
840xb1, 0xa5, 0x53, 0x65, 0xd2, 0x02, 0x99, 0x08, 0xd1, 0x84, 0xcc, 0xf0, 0xcd, 0xea, 0x61, 0xc9, 0x39, 0x02, 0x3f, 0x87,
850x4a, 0xe5, 0xc4, 0xd2, 0x07, 0x02, 0xe1, 0x9f, 0xa0, 0x06, 0xc2, 0xcc, 0x02, 0xe7, 0xaa, 0x6c, 0x99, 0x8a, 0xf8, 0x49,
860x00, 0xf1, 0xa2, 0x8c, 0x0c, 0x8a, 0xb9, 0x4f, 0x6d, 0x73, 0x3b, 0x2c, 0xb7, 0x9f, 0x8a, 0xa6, 0x7f, 0x9b, 0x9f, 0xb7,
870xa1, 0xcc, 0x74, 0x2e, 0x8f, 0xb8, 0xb0, 0x26, 0x89, 0xd2, 0xe5, 0x66, 0xe8, 0x8e, 0xa1, 0x02, 0x41, 0x00, 0xfc, 0xe7,
880x52, 0xbc, 0x4e, 0x95, 0xb6, 0x1a, 0xb4, 0x62, 0xcc, 0xd8, 0x06, 0xe1, 0xdc, 0x7a, 0xa2, 0xb6, 0x71, 0x01, 0xaa, 0x27,
890xfc, 0x99, 0xe5, 0xf2, 0x54, 0xbb, 0xb2, 0x85, 0xe1, 0x96, 0x54, 0x2d, 0xcb, 0xba, 0x86, 0xfa, 0x80, 0xdf, 0xcf, 0x39,
900xe6, 0x74, 0xcb, 0x22, 0xce, 0x70, 0xaa, 0x10, 0x00, 0x73, 0x1d, 0x45, 0x0a, 0x39, 0x51, 0x84, 0xf5, 0x15, 0x8f, 0x37,
910x76, 0x91, 0x02, 0x41, 0x00, 0xe4, 0x3f, 0xf0, 0xf4, 0xde, 0x79, 0x77, 0x48, 0x9b, 0x9c, 0x28, 0x45, 0x26, 0x57, 0x3c,
920x71, 0x40, 0x28, 0x6a, 0xa1, 0xfe, 0xc3, 0xe5, 0x37, 0xa1, 0x03, 0xf6, 0x2d, 0xbe, 0x80, 0x64, 0x72, 0x69, 0x2e, 0x9b,
930x4d, 0xe3, 0x2e, 0x1b, 0xfe, 0xe7, 0xf9, 0x77, 0x8c, 0x18, 0x53, 0x9f, 0xe2, 0xfe, 0x00, 0xbb, 0x49, 0x20, 0x47, 0xdf,
940x01, 0x61, 0x87, 0xd6, 0xe3, 0x44, 0xb5, 0x03, 0x2f, 0x02, 0x40, 0x54, 0xec, 0x7c, 0xbc, 0xdd, 0x0a, 0xaa, 0xde, 0xe6,
950xc9, 0xf2, 0x8d, 0x6c, 0x2a, 0x35, 0xf6, 0x3c, 0x63, 0x55, 0x29, 0x40, 0xf1, 0x32, 0x82, 0x9f, 0x53, 0xb3, 0x9e, 0x5f,
960xc1, 0x53, 0x52, 0x3e, 0xac, 0x2e, 0x28, 0x51, 0xa1, 0x16, 0xdb, 0x90, 0xe3, 0x99, 0x7e, 0x88, 0xa4, 0x04, 0x7c, 0x92,
970xae, 0xd2, 0xe7, 0xd4, 0xe1, 0x55, 0x20, 0x90, 0x3e, 0x3c, 0x6a, 0x63, 0xf0, 0x34, 0xf1, 0x02, 0x41, 0x00, 0x84, 0x5a,
980x17, 0x6c, 0xc6, 0x3c, 0x84, 0xd0, 0x93, 0x7a, 0xff, 0x56, 0xe9, 0x9e, 0x98, 0x2b, 0xcb, 0x5a, 0x24, 0x4a, 0xff, 0x21,
990xb4, 0x9e, 0x87, 0x3d, 0x76, 0xd8, 0x9b, 0xa8, 0x73, 0x96, 0x6c, 0x2b, 0x5c, 0x5e, 0xd3, 0xa6, 0xff, 0x10, 0xd6, 0x8e,
1000xaf, 0xa5, 0x8a, 0xcd, 0xa2, 0xde, 0xcb, 0x0e, 0xbd, 0x8a, 0xef, 0xae, 0xfd, 0x3f, 0x1d, 0xc0, 0xd8, 0xf8, 0x3b, 0xf5,
1010x02, 0x7d, 0x02, 0x41, 0x00, 0x8b, 0x26, 0xd3, 0x2c, 0x7d, 0x28, 0x38, 0x92, 0xf1, 0xbf, 0x15, 0x16, 0x39, 0x50, 0xc8,
1020x6d, 0x32, 0xec, 0x28, 0xf2, 0x8b, 0xd8, 0x70, 0xc5, 0xed, 0xe1, 0x7b, 0xff, 0x2d, 0x66, 0x8c, 0x86, 0x77, 0x43, 0xeb,
1030xb6, 0xf6, 0x50, 0x66, 0xb0, 0x40, 0x24, 0x6a, 0xaf, 0x98, 0x21, 0x45, 0x30, 0x01, 0x59, 0xd0, 0xc3, 0xfc, 0x7b, 0xae,
1040x30, 0x18, 0xeb, 0x90, 0xfb, 0x17, 0xd3, 0xce, 0xb5
105};
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800106
Jeff Thompson70e80802013-12-10 20:21:03 -0800107/**
108 * Loop to encode a data packet nIterations times using C++.
109 * @param nIterations The number of iterations.
110 * @param useComplex If true, use a large name, large content and all fields. If false, use a small name, small content
111 * and only required fields.
112 * @param useCrypto If true, sign the data packet. If false, use a blank signature.
113 * @param encoding Set this to the wire encoding.
114 * @return The number of seconds for all iterations.
115 */
116static double
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800117benchmarkEncodeDataSecondsCpp(int nIterations, bool useComplex, bool useCrypto, Block& encoding)
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800118{
Jeff Thompson70e80802013-12-10 20:21:03 -0800119 Name name;
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800120 Block content;
Jeff Thompson70e80802013-12-10 20:21:03 -0800121 if (useComplex) {
122 // Use a large name and content.
123 name = Name("/ndn/ucla.edu/apps/lwndn-test/numbers.txt/%FD%05%05%E8%0C%CE%1D/%00");
124
125 ostringstream contentStream;
126 int count = 1;
127 contentStream << (count++);
128 while (contentStream.str().length() < 1170)
129 contentStream << " " << (count++);
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800130 content = dataBlock(Tlv::Content, contentStream.str().c_str(), contentStream.str().length());
Jeff Thompson70e80802013-12-10 20:21:03 -0800131 }
132 else {
133 // Use a small name and content.
134 name = Name("/test");
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800135 content = dataBlock(Tlv::Content, "abc", 3);
Jeff Thompson70e80802013-12-10 20:21:03 -0800136 }
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800137 std::cout << "Content size: " << content.value_size() << std::endl;
Jeff Thompson70e80802013-12-10 20:21:03 -0800138
139 // Initialize the KeyChain storage in case useCrypto is true.
Yingdi Yuf4330ee2014-01-14 14:34:46 -0800140 KeyChainImpl<SecPublicInfoMemory, SecTpmMemory> keyChain;
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800141
142 Name keyName("/testname/dsk-123");
143
144 // Initialize the storage.
Yingdi Yuf4330ee2014-01-14 14:34:46 -0800145 keyChain.addPublicKey(keyName, KEY_TYPE_RSA,
146 PublicKey(DEFAULT_PUBLIC_KEY_DER, sizeof(DEFAULT_PUBLIC_KEY_DER)));
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800147
Yingdi Yuf4330ee2014-01-14 14:34:46 -0800148 keyChain.setKeyPairForKeyName(keyName,
149 DEFAULT_PUBLIC_KEY_DER, sizeof(DEFAULT_PUBLIC_KEY_DER),
150 DEFAULT_PRIVATE_KEY_DER, sizeof(DEFAULT_PRIVATE_KEY_DER));
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800151
Yingdi Yuf4330ee2014-01-14 14:34:46 -0800152 keyChain.addCertificateAsKeyDefault(*keyChain.selfSign(keyName));
Alexander Afanasyev3e7788b2014-01-12 23:00:41 -0800153 Name certificateName = keyChain.getDefaultCertificateName();
Jeff Thompson70e80802013-12-10 20:21:03 -0800154
155 // Set up publisherPublicKeyDigest and signatureBits in case useCrypto is false.
Jeff Thompson70e80802013-12-10 20:21:03 -0800156 uint8_t signatureBitsArray[128];
157 memset(signatureBitsArray, 0, sizeof(signatureBitsArray));
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800158 Block signatureValue = dataBlock(Tlv::SignatureValue, signatureBitsArray, sizeof(signatureBitsArray));
Jeff Thompson70e80802013-12-10 20:21:03 -0800159
160 double start = getNowSeconds();
161 for (int i = 0; i < nIterations; ++i) {
162 Data data(name);
163 data.setContent(content);
164 if (useComplex) {
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800165 data.setFreshnessPeriod(1000000);
Jeff Thompson70e80802013-12-10 20:21:03 -0800166 }
167
168 if (useCrypto)
169 // This sets the signature fields.
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800170 keyChain.sign(data);
Jeff Thompson70e80802013-12-10 20:21:03 -0800171 else {
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800172 // Imitate real sign method to set up the signature fields, but don't actually sign.
173 SignatureSha256WithRsa signature;
174 signature.setKeyLocator(certificateName);
175 signature.setValue(signatureValue);
176 data.setSignature(signature);
Jeff Thompson70e80802013-12-10 20:21:03 -0800177 }
178
179 encoding = data.wireEncode();
180 }
181 double finish = getNowSeconds();
182
183 return finish - start;
184}
185
186static void
Jeff Thompson06d17802013-12-11 16:34:27 -0800187onVerified(const ptr_lib::shared_ptr<Data>& data)
Jeff Thompson70e80802013-12-10 20:21:03 -0800188{
189 // Do nothing since we expect it to verify.
190}
191
192static void
Jeff Thompson06d17802013-12-11 16:34:27 -0800193onVerifyFailed(const ptr_lib::shared_ptr<Data>& data)
Jeff Thompson70e80802013-12-10 20:21:03 -0800194{
195 cout << "Signature verification: FAILED" << endl;
196}
197
198/**
199 * Loop to decode a data packet nIterations times using C++.
200 * @param nIterations The number of iterations.
201 * @param useCrypto If true, verify the signature. If false, don't verify.
202 * @param encoding The wire encoding to decode.
203 * @return The number of seconds for all iterations.
204 */
205static double
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800206benchmarkDecodeDataSecondsCpp(int nIterations, bool useCrypto, const ConstBufferPtr &encoding)
Jeff Thompson70e80802013-12-10 20:21:03 -0800207{
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800208 // // Initialize the KeyChain storage in case useCrypto is true.
209 // ptr_lib::shared_ptr<MemoryIdentityStorage> identityStorage(new MemoryIdentityStorage());
210 // identityStorage->addKey(keyName, KEY_TYPE_RSA, Blob(DEFAULT_PUBLIC_KEY_DER, sizeof(DEFAULT_PUBLIC_KEY_DER)));
Alexander Afanasyev84c2bd42014-01-09 22:35:35 -0800211
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800212 // ptr_lib::shared_ptr<MemoryPrivateKeyStorage> privateKeyStorage(new MemoryPrivateKeyStorage());
213 // KeyChain keyChain(identityStorage, privateKeyStorage);
Alexander Afanasyev84c2bd42014-01-09 22:35:35 -0800214
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800215 // Name keyName("/testname/DSK-123");
Jeff Thompson70e80802013-12-10 20:21:03 -0800216
Jeff Thompson687ea652013-12-06 11:28:23 -0800217 size_t nameSize = 0;
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800218 double start = getNowSeconds();
Jeff Thompson687ea652013-12-06 11:28:23 -0800219 for (int i = 0; i < nIterations; ++i) {
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800220 Data data;
221 data.wireDecode(encoding);
Jeff Thompson70e80802013-12-10 20:21:03 -0800222
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800223 // if (useCrypto)
224 // keyChain.verifyData(data, onVerified, onVerifyFailed);
Jeff Thompson687ea652013-12-06 11:28:23 -0800225 }
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800226 double finish = getNowSeconds();
227
Jeff Thompson70e80802013-12-10 20:21:03 -0800228 return finish - start;
229}
230
231/**
232 * Loop to encode a data packet nIterations times using C.
233 * @param nIterations The number of iterations.
234 * @param useComplex If true, use a large name, large content and all fields. If false, use a small name, small content
235 * and only required fields.
236 * @param useCrypto If true, sign the data packet. If false, use a blank signature.
237 * @param encoding Output buffer for the wire encoding.
238 * @param maxEncodingLength The size of the encoding buffer.
239 * @param encodingLength Return the number of output bytes in encoding.
240 * @return The number of seconds for all iterations.
241 */
242static double
243benchmarkEncodeDataSecondsC
244 (int nIterations, bool useComplex, bool useCrypto, uint8_t* encoding, size_t maxEncodingLength, size_t *encodingLength)
245{
246 struct ndn_Blob finalBlockId;
247 ndn_Blob_initialize(&finalBlockId, (uint8_t*)"\x00", 1);
248
249 struct ndn_NameComponent nameComponents[20];
250 struct ndn_Name name;
251 ndn_Name_initialize(&name, nameComponents, sizeof(nameComponents) / sizeof(nameComponents[0]));
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800252 Buffer contentBlob;
Jeff Thompson70e80802013-12-10 20:21:03 -0800253 struct ndn_Blob content;
254 if (useComplex) {
255 // Use a large name and content.
256 ndn_Name_appendString(&name, (char*)"ndn");
257 ndn_Name_appendString(&name, (char*)"ucla.edu");
258 ndn_Name_appendString(&name, (char*)"apps");
259 ndn_Name_appendString(&name, (char*)"lwndn-test");
260 ndn_Name_appendString(&name, (char*)"numbers.txt");
261 ndn_Name_appendString(&name, (char*)"\xFD\x05\x05\xE8\x0C\xCE\x1D");
262 ndn_Name_appendBlob(&name, &finalBlockId);
263
264 ostringstream contentStream;
265 int count = 1;
266 contentStream << (count++);
267 while (contentStream.str().length() < 1170)
268 contentStream << " " << (count++);
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800269 contentBlob = Buffer((uint8_t*)contentStream.str().c_str(), contentStream.str().length());
Jeff Thompson70e80802013-12-10 20:21:03 -0800270 }
271 else {
272 // Use a small name and content.
273 ndn_Name_appendString(&name, (char*)"test");
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800274 contentBlob = Buffer((uint8_t*)"abc", 3);
Jeff Thompson70e80802013-12-10 20:21:03 -0800275 }
276 ndn_Blob_initialize(&content, (uint8_t*)contentBlob.buf(), contentBlob.size());
277
278 struct ndn_NameComponent certificateNameComponents[20];
279 struct ndn_Name certificateName;
280 ndn_Name_initialize(&certificateName, certificateNameComponents, sizeof(certificateNameComponents) / sizeof(certificateNameComponents[0]));
281 ndn_Name_appendString(&certificateName, (char*)"testname");
282 ndn_Name_appendString(&certificateName, (char*)"KEY");
283 ndn_Name_appendString(&certificateName, (char*)"DSK-123");
284 ndn_Name_appendString(&certificateName, (char*)"ID-CERT");
285 ndn_Name_appendString(&certificateName, (char*)"0");
286
287 // Set up publisherPublicKeyDigest and signatureBits in case useCrypto is false.
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800288 uint8_t* publicKeyDer = DEFAULT_PUBLIC_KEY_DER;
289 size_t publicKeyDerLength = sizeof(DEFAULT_PUBLIC_KEY_DER);
290 uint8_t publisherPublicKeyDigestArray[SHA256_DIGEST_LENGTH];
291 ndn_digestSha256(publicKeyDer, publicKeyDerLength, publisherPublicKeyDigestArray);
Jeff Thompson70e80802013-12-10 20:21:03 -0800292 struct ndn_Blob publisherPublicKeyDigest;
293 ndn_Blob_initialize(&publisherPublicKeyDigest, publisherPublicKeyDigestArray, sizeof(publisherPublicKeyDigestArray));
294 uint8_t signatureBitsArray[128];
295 memset(signatureBitsArray, 0, sizeof(signatureBitsArray));
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800296
297 // Set up the private key now in case useCrypto is true.
298 // Use a temporary pointer since d2i updates it.
299 const uint8_t *privateKeyDerPointer = DEFAULT_PRIVATE_KEY_DER;
300 RSA *privateKey = d2i_RSAPrivateKey(NULL, &privateKeyDerPointer, sizeof(DEFAULT_PRIVATE_KEY_DER));
301 if (!privateKey) {
302 // Don't expect this to happen.
303 cout << "Error decoding private key DER" << endl;
304 return 0;
305 }
Jeff Thompson70e80802013-12-10 20:21:03 -0800306
307 double start = getNowSeconds();
308 for (int i = 0; i < nIterations; ++i) {
309 struct ndn_Data data;
310 ndn_Data_initialize(&data, name.components, name.maxComponents, certificateName.components, certificateName.maxComponents);
311
312 data.name = name;
313 data.content = content;
314 if (useComplex) {
315 data.metaInfo.timestampMilliseconds = 1.3e+12;
316 data.metaInfo.freshnessSeconds = 1000;
317 ndn_NameComponent_initialize(&data.metaInfo.finalBlockID, finalBlockId.value, finalBlockId.length);
318 }
319
Jeff Thompson70e80802013-12-10 20:21:03 -0800320 struct ndn_DynamicUInt8Array output;
Jeff Thompson70e80802013-12-10 20:21:03 -0800321 struct ndn_BinaryXmlEncoder encoder;
Jeff Thompson70e80802013-12-10 20:21:03 -0800322 size_t signedPortionBeginOffset, signedPortionEndOffset;
323 ndn_Error error;
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800324
325 data.signature.keyLocator.type = ndn_KeyLocatorType_KEYNAME;
326 data.signature.keyLocator.keyName = certificateName;
327 data.signature.keyLocator.keyNameType = (ndn_KeyNameType)-1;
328 data.signature.publisherPublicKeyDigest.publisherPublicKeyDigest = publisherPublicKeyDigest;
329 if (useCrypto) {
330 // Encode once to get the signed portion.
331 ndn_DynamicUInt8Array_initialize(&output, encoding, maxEncodingLength, 0);
332 ndn_BinaryXmlEncoder_initialize(&encoder, &output);
333 if ((error = ndn_encodeBinaryXmlData(&data, &signedPortionBeginOffset, &signedPortionEndOffset, &encoder))) {
334 cout << "Error in ndn_encodeBinaryXmlData: " << ndn_getErrorString(error) << endl;
335 return 0;
336 }
337
338 // Imitate MemoryPrivateKeyStorage::sign.
339 uint8_t digest[SHA256_DIGEST_LENGTH];
340 ndn_digestSha256(encoding + signedPortionBeginOffset, signedPortionEndOffset - signedPortionBeginOffset, digest);
341 unsigned int signatureBitsLength;
342 if (!RSA_sign(NID_sha256, digest, sizeof(digest), signatureBitsArray, &signatureBitsLength, privateKey)) {
343 // Don't expect this to happen.
344 cout << "Error in RSA_sign" << endl;
345 return 0;
346 }
347
348 ndn_Blob_initialize(&data.signature.signature, signatureBitsArray, signatureBitsLength);
349 }
350 else
351 // Set up the signature, but don't sign.
352 ndn_Blob_initialize(&data.signature.signature, signatureBitsArray, sizeof(signatureBitsArray));
353
354 // Assume the encoding buffer is big enough so we don't need to dynamically reallocate.
355 ndn_DynamicUInt8Array_initialize(&output, encoding, maxEncodingLength, 0);
356 ndn_BinaryXmlEncoder_initialize(&encoder, &output);
Jeff Thompson70e80802013-12-10 20:21:03 -0800357 if ((error = ndn_encodeBinaryXmlData(&data, &signedPortionBeginOffset, &signedPortionEndOffset, &encoder))) {
358 cout << "Error in ndn_encodeBinaryXmlData: " << ndn_getErrorString(error) << endl;
359 return 0;
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800360 }
Jeff Thompson70e80802013-12-10 20:21:03 -0800361 *encodingLength = encoder.offset;
362 }
363 double finish = getNowSeconds();
Jeff Thompson687ea652013-12-06 11:28:23 -0800364
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800365 if (privateKey)
366 RSA_free(privateKey);
367
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800368 return finish - start;
Jeff Thompson687ea652013-12-06 11:28:23 -0800369}
370
Jeff Thompson70e80802013-12-10 20:21:03 -0800371/**
372 * Loop to decode a data packet nIterations times using C.
373 * @param nIterations The number of iterations.
374 * @param useCrypto If true, verify the signature. If false, don't verify.
375 * @param encoding The buffer with wire encoding to decode.
376 * @param encodingLength The number of bytes in the encoding.
377 * @return The number of seconds for all iterations.
378 */
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800379static double
Jeff Thompson70e80802013-12-10 20:21:03 -0800380benchmarkDecodeDataSecondsC(int nIterations, bool useCrypto, uint8_t* encoding, size_t encodingLength)
Jeff Thompson687ea652013-12-06 11:28:23 -0800381{
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800382 double start = getNowSeconds();
Jeff Thompson687ea652013-12-06 11:28:23 -0800383 for (int i = 0; i < nIterations; ++i) {
384 struct ndn_NameComponent nameComponents[100];
385 struct ndn_NameComponent keyNameComponents[100];
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800386 struct ndn_Data data;
Jeff Thompson687ea652013-12-06 11:28:23 -0800387 ndn_Data_initialize
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800388 (&data, nameComponents, sizeof(nameComponents) / sizeof(nameComponents[0]),
Jeff Thompson687ea652013-12-06 11:28:23 -0800389 keyNameComponents, sizeof(keyNameComponents) / sizeof(keyNameComponents[0]));
390
391 ndn_BinaryXmlDecoder decoder;
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800392 ndn_BinaryXmlDecoder_initialize(&decoder, encoding, encodingLength);
Jeff Thompson687ea652013-12-06 11:28:23 -0800393 size_t signedPortionBeginOffset, signedPortionEndOffset;
394 ndn_Error error;
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800395 if ((error = ndn_decodeBinaryXmlData(&data, &signedPortionBeginOffset, &signedPortionEndOffset, &decoder))) {
Jeff Thompson687ea652013-12-06 11:28:23 -0800396 cout << "Error in ndn_decodeBinaryXmlData: " << ndn_getErrorString(error) << endl;
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800397 return 0;
Jeff Thompson687ea652013-12-06 11:28:23 -0800398 }
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800399
400 if (useCrypto) {
401 if (!verifyRsaSignature
402 (encoding + signedPortionBeginOffset, signedPortionEndOffset - signedPortionBeginOffset,
403 data.signature.signature.value, data.signature.signature.length,
404 DEFAULT_PUBLIC_KEY_DER, sizeof(DEFAULT_PUBLIC_KEY_DER)))
405 cout << "Signature verification: FAILED" << endl;
406 }
Jeff Thompson687ea652013-12-06 11:28:23 -0800407 }
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800408 double finish = getNowSeconds();
409
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800410 return finish - start;
Jeff Thompson687ea652013-12-06 11:28:23 -0800411}
412
Jeff Thompson70e80802013-12-10 20:21:03 -0800413/**
Jeff Thompson4c66e712013-12-11 13:37:41 -0800414 * Call benchmarkEncodeDataSecondsCpp and benchmarkDecodeDataSecondsCpp with appropriate nInterations. Print the
Jeff Thompson70e80802013-12-10 20:21:03 -0800415 * results to cout.
Jeff Thompsonb33b21e2013-12-11 17:37:55 -0800416 * @param useComplex See benchmarkEncodeDataSecondsCpp.
Jeff Thompson4c66e712013-12-11 13:37:41 -0800417 * @param useCrypto See benchmarkEncodeDataSecondsCpp and benchmarkDecodeDataSecondsCpp.
Jeff Thompson70e80802013-12-10 20:21:03 -0800418 */
419static void
420benchmarkEncodeDecodeDataCpp(bool useComplex, bool useCrypto)
421{
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800422 Block encoding;
Jeff Thompson70e80802013-12-10 20:21:03 -0800423 {
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800424 int nIterations = useCrypto ? 20000 : 200000;
Jeff Thompson70e80802013-12-10 20:21:03 -0800425 double duration = benchmarkEncodeDataSecondsCpp(nIterations, useComplex, useCrypto, encoding);
426 cout << "Encode " << (useComplex ? "complex" : "simple ") << " data C++: Crypto? " << (useCrypto ? "yes" : "no ")
427 << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl;
428 }
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800429
430 BufferPtr wire = ptr_lib::make_shared<Buffer>(encoding.wire(), encoding.size());
Jeff Thompson70e80802013-12-10 20:21:03 -0800431 {
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800432 int nIterations = useCrypto ? 10000 : 1000000;
433 double duration = benchmarkDecodeDataSecondsCpp(nIterations, useCrypto, wire);
Jeff Thompson70e80802013-12-10 20:21:03 -0800434 cout << "Decode " << (useComplex ? "complex" : "simple ") << " data C++: Crypto? " << (useCrypto ? "yes" : "no ")
435 << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl;
436 }
437}
438
439/**
Jeff Thompson4c66e712013-12-11 13:37:41 -0800440 * Call benchmarkEncodeDataSecondsC and benchmarkDecodeDataSecondsC with appropriate nInterations. Print the
Jeff Thompson70e80802013-12-10 20:21:03 -0800441 * results to cout.
Jeff Thompsonb33b21e2013-12-11 17:37:55 -0800442 * @param useComplex See benchmarkEncodeDataSecondsC.
Jeff Thompson4c66e712013-12-11 13:37:41 -0800443 * @param useCrypto See benchmarkEncodeDataSecondsC and benchmarkDecodeDataSecondsC.
Jeff Thompson70e80802013-12-10 20:21:03 -0800444 */
445static void
446benchmarkEncodeDecodeDataC(bool useComplex, bool useCrypto)
447{
448 uint8_t encoding[1500];
449 size_t encodingLength;
450 {
451 int nIterations = useCrypto ? 20000 : 10000000;
452 double duration = benchmarkEncodeDataSecondsC(nIterations, useComplex, useCrypto, encoding, sizeof(encoding), &encodingLength);
453 cout << "Encode " << (useComplex ? "complex" : "simple ") << " data C: Crypto? " << (useCrypto ? "yes" : "no ")
454 << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl;
455 }
456 {
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800457 int nIterations = useCrypto ? 150000 : 15000000;
Jeff Thompson70e80802013-12-10 20:21:03 -0800458 double duration = benchmarkDecodeDataSecondsC(nIterations, useCrypto, encoding, encodingLength);
459 cout << "Decode " << (useComplex ? "complex" : "simple ") << " data C: Crypto? " << (useCrypto ? "yes" : "no ")
460 << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl;
461 }
462}
463
Jeff Thompson687ea652013-12-06 11:28:23 -0800464int
465main(int argc, char** argv)
466{
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800467 try {
Jeff Thompson70e80802013-12-10 20:21:03 -0800468 benchmarkEncodeDecodeDataCpp(false, false);
469 benchmarkEncodeDecodeDataCpp(true, false);
470 benchmarkEncodeDecodeDataCpp(false, true);
471 benchmarkEncodeDecodeDataCpp(true, true);
472
473 benchmarkEncodeDecodeDataC(false, false);
474 benchmarkEncodeDecodeDataC(true, false);
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800475 benchmarkEncodeDecodeDataC(false, true);
476 benchmarkEncodeDecodeDataC(true, true);
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800477 } catch (std::exception& e) {
478 cout << "exception: " << e.what() << endl;
479 }
Jeff Thompson687ea652013-12-06 11:28:23 -0800480 return 0;
481}