blob: f68502dbdf9bc6a3b13bf2a3558ee49058e1beec [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>
Jeff Thompson687ea652013-12-06 11:28:23 -080013#include <ndn-cpp/data.hpp>
Jeff Thompson70e80802013-12-10 20:21:03 -080014#include <ndn-cpp/security/key-chain.hpp>
15#include <ndn-cpp/security/identity/memory-identity-storage.hpp>
16#include <ndn-cpp/security/identity/memory-private-key-storage.hpp>
17#include <ndn-cpp/security/policy/self-verify-policy-manager.hpp>
Alexander Afanasyev84c2bd42014-01-09 22:35:35 -080018
Jeff Thompson687ea652013-12-06 11:28:23 -080019// Hack: Hook directly into non-API functions.
20#include "../src/c/encoding/binary-xml-decoder.h"
21#include "../src/c/data.h"
22#include "../src/c/encoding/binary-xml-data.h"
Alexander Afanasyev94a356e2014-01-12 18:48:06 -080023#include <ndn-cpp/c/util/crypto.h>
Jeff Thompson687ea652013-12-06 11:28:23 -080024
25using namespace std;
26using namespace ndn;
Jeff Thompson687ea652013-12-06 11:28:23 -080027
28static double
Jeff Thompson18dba8b2013-12-09 12:17:40 -080029getNowSeconds()
Jeff Thompson687ea652013-12-06 11:28:23 -080030{
31 struct timeval t;
32 gettimeofday(&t, 0);
Jeff Thompson18dba8b2013-12-09 12:17:40 -080033 return t.tv_sec + t.tv_usec / 1000000.0;
Jeff Thompson687ea652013-12-06 11:28:23 -080034}
35
Jeff Thompson30d3e6c2013-12-11 12:35:54 -080036static bool
37verifyRsaSignature
38 (uint8_t* signedPortion, size_t signedPortionLength, uint8_t* signatureBits, size_t signatureBitsLength,
39 uint8_t* publicKeyDer, size_t publicKeyDerLength)
40{
41 // Set signedPortionDigest to the digest of the signed portion of the wire encoding.
42 uint8_t signedPortionDigest[SHA256_DIGEST_LENGTH];
43 ndn_digestSha256(signedPortion, signedPortionLength, signedPortionDigest);
44
45 // Verify the signedPortionDigest.
46 // Use a temporary pointer since d2i updates it.
47 const uint8_t *derPointer = publicKeyDer;
48 RSA *rsaPublicKey = d2i_RSA_PUBKEY(NULL, &derPointer, publicKeyDerLength);
49 if (!rsaPublicKey) {
50 // Don't expect this to happen.
51 cout << "Error decoding public key in d2i_RSAPublicKey" << endl;
52 return 0;
53 }
54 int success = RSA_verify
55 (NID_sha256, signedPortionDigest, sizeof(signedPortionDigest), signatureBits, signatureBitsLength, rsaPublicKey);
56 // Free the public key before checking for success.
57 RSA_free(rsaPublicKey);
58
59 // RSA_verify returns 1 for a valid signature.
60 return (success == 1);
61}
62
Jeff Thompson70e80802013-12-10 20:21:03 -080063static uint8_t DEFAULT_PUBLIC_KEY_DER[] = {
640x30, 0x81, 0x9F, 0x30, 0x0D, 0x06, 0x09, 0x2A, 0x86, 0x48, 0x86, 0xF7, 0x0D, 0x01, 0x01, 0x01, 0x05, 0x00, 0x03, 0x81,
650x8D, 0x00, 0x30, 0x81, 0x89, 0x02, 0x81, 0x81, 0x00, 0xE1, 0x7D, 0x30, 0xA7, 0xD8, 0x28, 0xAB, 0x1B, 0x84, 0x0B, 0x17,
660x54, 0x2D, 0xCA, 0xF6, 0x20, 0x7A, 0xFD, 0x22, 0x1E, 0x08, 0x6B, 0x2A, 0x60, 0xD1, 0x6C, 0xB7, 0xF5, 0x44, 0x48, 0xBA,
670x9F, 0x3F, 0x08, 0xBC, 0xD0, 0x99, 0xDB, 0x21, 0xDD, 0x16, 0x2A, 0x77, 0x9E, 0x61, 0xAA, 0x89, 0xEE, 0xE5, 0x54, 0xD3,
680xA4, 0x7D, 0xE2, 0x30, 0xBC, 0x7A, 0xC5, 0x90, 0xD5, 0x24, 0x06, 0x7C, 0x38, 0x98, 0xBB, 0xA6, 0xF5, 0xDC, 0x43, 0x60,
690xB8, 0x45, 0xED, 0xA4, 0x8C, 0xBD, 0x9C, 0xF1, 0x26, 0xA7, 0x23, 0x44, 0x5F, 0x0E, 0x19, 0x52, 0xD7, 0x32, 0x5A, 0x75,
700xFA, 0xF5, 0x56, 0x14, 0x4F, 0x9A, 0x98, 0xAF, 0x71, 0x86, 0xB0, 0x27, 0x86, 0x85, 0xB8, 0xE2, 0xC0, 0x8B, 0xEA, 0x87,
710x17, 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 -0800720x00, 0x01
Jeff Thompson687ea652013-12-06 11:28:23 -080073};
74
Jeff Thompson70e80802013-12-10 20:21:03 -080075static uint8_t DEFAULT_PRIVATE_KEY_DER[] = {
760x30, 0x82, 0x02, 0x5d, 0x02, 0x01, 0x00, 0x02, 0x81, 0x81, 0x00, 0xe1, 0x7d, 0x30, 0xa7, 0xd8, 0x28, 0xab, 0x1b, 0x84,
770x0b, 0x17, 0x54, 0x2d, 0xca, 0xf6, 0x20, 0x7a, 0xfd, 0x22, 0x1e, 0x08, 0x6b, 0x2a, 0x60, 0xd1, 0x6c, 0xb7, 0xf5, 0x44,
780x48, 0xba, 0x9f, 0x3f, 0x08, 0xbc, 0xd0, 0x99, 0xdb, 0x21, 0xdd, 0x16, 0x2a, 0x77, 0x9e, 0x61, 0xaa, 0x89, 0xee, 0xe5,
790x54, 0xd3, 0xa4, 0x7d, 0xe2, 0x30, 0xbc, 0x7a, 0xc5, 0x90, 0xd5, 0x24, 0x06, 0x7c, 0x38, 0x98, 0xbb, 0xa6, 0xf5, 0xdc,
800x43, 0x60, 0xb8, 0x45, 0xed, 0xa4, 0x8c, 0xbd, 0x9c, 0xf1, 0x26, 0xa7, 0x23, 0x44, 0x5f, 0x0e, 0x19, 0x52, 0xd7, 0x32,
810x5a, 0x75, 0xfa, 0xf5, 0x56, 0x14, 0x4f, 0x9a, 0x98, 0xaf, 0x71, 0x86, 0xb0, 0x27, 0x86, 0x85, 0xb8, 0xe2, 0xc0, 0x8b,
820xea, 0x87, 0x17, 0x1b, 0x4d, 0xee, 0x58, 0x5c, 0x18, 0x28, 0x29, 0x5b, 0x53, 0x95, 0xeb, 0x4a, 0x17, 0x77, 0x9f, 0x02,
830x03, 0x01, 0x00, 0x01, 0x02, 0x81, 0x80, 0x1a, 0x4b, 0xfa, 0x4f, 0xa8, 0xc2, 0xdd, 0x69, 0xa1, 0x15, 0x96, 0x0b, 0xe8,
840x27, 0x42, 0x5a, 0xf9, 0x5c, 0xea, 0x0c, 0xac, 0x98, 0xaa, 0xe1, 0x8d, 0xaa, 0xeb, 0x2d, 0x3c, 0x60, 0x6a, 0xfb, 0x45,
850x63, 0xa4, 0x79, 0x83, 0x67, 0xed, 0xe4, 0x15, 0xc0, 0xb0, 0x20, 0x95, 0x6d, 0x49, 0x16, 0xc6, 0x42, 0x05, 0x48, 0xaa,
860xb1, 0xa5, 0x53, 0x65, 0xd2, 0x02, 0x99, 0x08, 0xd1, 0x84, 0xcc, 0xf0, 0xcd, 0xea, 0x61, 0xc9, 0x39, 0x02, 0x3f, 0x87,
870x4a, 0xe5, 0xc4, 0xd2, 0x07, 0x02, 0xe1, 0x9f, 0xa0, 0x06, 0xc2, 0xcc, 0x02, 0xe7, 0xaa, 0x6c, 0x99, 0x8a, 0xf8, 0x49,
880x00, 0xf1, 0xa2, 0x8c, 0x0c, 0x8a, 0xb9, 0x4f, 0x6d, 0x73, 0x3b, 0x2c, 0xb7, 0x9f, 0x8a, 0xa6, 0x7f, 0x9b, 0x9f, 0xb7,
890xa1, 0xcc, 0x74, 0x2e, 0x8f, 0xb8, 0xb0, 0x26, 0x89, 0xd2, 0xe5, 0x66, 0xe8, 0x8e, 0xa1, 0x02, 0x41, 0x00, 0xfc, 0xe7,
900x52, 0xbc, 0x4e, 0x95, 0xb6, 0x1a, 0xb4, 0x62, 0xcc, 0xd8, 0x06, 0xe1, 0xdc, 0x7a, 0xa2, 0xb6, 0x71, 0x01, 0xaa, 0x27,
910xfc, 0x99, 0xe5, 0xf2, 0x54, 0xbb, 0xb2, 0x85, 0xe1, 0x96, 0x54, 0x2d, 0xcb, 0xba, 0x86, 0xfa, 0x80, 0xdf, 0xcf, 0x39,
920xe6, 0x74, 0xcb, 0x22, 0xce, 0x70, 0xaa, 0x10, 0x00, 0x73, 0x1d, 0x45, 0x0a, 0x39, 0x51, 0x84, 0xf5, 0x15, 0x8f, 0x37,
930x76, 0x91, 0x02, 0x41, 0x00, 0xe4, 0x3f, 0xf0, 0xf4, 0xde, 0x79, 0x77, 0x48, 0x9b, 0x9c, 0x28, 0x45, 0x26, 0x57, 0x3c,
940x71, 0x40, 0x28, 0x6a, 0xa1, 0xfe, 0xc3, 0xe5, 0x37, 0xa1, 0x03, 0xf6, 0x2d, 0xbe, 0x80, 0x64, 0x72, 0x69, 0x2e, 0x9b,
950x4d, 0xe3, 0x2e, 0x1b, 0xfe, 0xe7, 0xf9, 0x77, 0x8c, 0x18, 0x53, 0x9f, 0xe2, 0xfe, 0x00, 0xbb, 0x49, 0x20, 0x47, 0xdf,
960x01, 0x61, 0x87, 0xd6, 0xe3, 0x44, 0xb5, 0x03, 0x2f, 0x02, 0x40, 0x54, 0xec, 0x7c, 0xbc, 0xdd, 0x0a, 0xaa, 0xde, 0xe6,
970xc9, 0xf2, 0x8d, 0x6c, 0x2a, 0x35, 0xf6, 0x3c, 0x63, 0x55, 0x29, 0x40, 0xf1, 0x32, 0x82, 0x9f, 0x53, 0xb3, 0x9e, 0x5f,
980xc1, 0x53, 0x52, 0x3e, 0xac, 0x2e, 0x28, 0x51, 0xa1, 0x16, 0xdb, 0x90, 0xe3, 0x99, 0x7e, 0x88, 0xa4, 0x04, 0x7c, 0x92,
990xae, 0xd2, 0xe7, 0xd4, 0xe1, 0x55, 0x20, 0x90, 0x3e, 0x3c, 0x6a, 0x63, 0xf0, 0x34, 0xf1, 0x02, 0x41, 0x00, 0x84, 0x5a,
1000x17, 0x6c, 0xc6, 0x3c, 0x84, 0xd0, 0x93, 0x7a, 0xff, 0x56, 0xe9, 0x9e, 0x98, 0x2b, 0xcb, 0x5a, 0x24, 0x4a, 0xff, 0x21,
1010xb4, 0x9e, 0x87, 0x3d, 0x76, 0xd8, 0x9b, 0xa8, 0x73, 0x96, 0x6c, 0x2b, 0x5c, 0x5e, 0xd3, 0xa6, 0xff, 0x10, 0xd6, 0x8e,
1020xaf, 0xa5, 0x8a, 0xcd, 0xa2, 0xde, 0xcb, 0x0e, 0xbd, 0x8a, 0xef, 0xae, 0xfd, 0x3f, 0x1d, 0xc0, 0xd8, 0xf8, 0x3b, 0xf5,
1030x02, 0x7d, 0x02, 0x41, 0x00, 0x8b, 0x26, 0xd3, 0x2c, 0x7d, 0x28, 0x38, 0x92, 0xf1, 0xbf, 0x15, 0x16, 0x39, 0x50, 0xc8,
1040x6d, 0x32, 0xec, 0x28, 0xf2, 0x8b, 0xd8, 0x70, 0xc5, 0xed, 0xe1, 0x7b, 0xff, 0x2d, 0x66, 0x8c, 0x86, 0x77, 0x43, 0xeb,
1050xb6, 0xf6, 0x50, 0x66, 0xb0, 0x40, 0x24, 0x6a, 0xaf, 0x98, 0x21, 0x45, 0x30, 0x01, 0x59, 0xd0, 0xc3, 0xfc, 0x7b, 0xae,
1060x30, 0x18, 0xeb, 0x90, 0xfb, 0x17, 0xd3, 0xce, 0xb5
107};
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800108
Jeff Thompson70e80802013-12-10 20:21:03 -0800109/**
110 * Loop to encode a data packet nIterations times using C++.
111 * @param nIterations The number of iterations.
112 * @param useComplex If true, use a large name, large content and all fields. If false, use a small name, small content
113 * and only required fields.
114 * @param useCrypto If true, sign the data packet. If false, use a blank signature.
115 * @param encoding Set this to the wire encoding.
116 * @return The number of seconds for all iterations.
117 */
118static double
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800119benchmarkEncodeDataSecondsCpp(int nIterations, bool useComplex, bool useCrypto, Block& encoding)
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800120{
Jeff Thompson70e80802013-12-10 20:21:03 -0800121 Name name;
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800122 Block content;
Jeff Thompson70e80802013-12-10 20:21:03 -0800123 if (useComplex) {
124 // Use a large name and content.
125 name = Name("/ndn/ucla.edu/apps/lwndn-test/numbers.txt/%FD%05%05%E8%0C%CE%1D/%00");
126
127 ostringstream contentStream;
128 int count = 1;
129 contentStream << (count++);
130 while (contentStream.str().length() < 1170)
131 contentStream << " " << (count++);
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800132 content = dataBlock(Tlv::Content, contentStream.str().c_str(), contentStream.str().length());
Jeff Thompson70e80802013-12-10 20:21:03 -0800133 }
134 else {
135 // Use a small name and content.
136 name = Name("/test");
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800137 content = dataBlock(Tlv::Content, "abc", 3);
Jeff Thompson70e80802013-12-10 20:21:03 -0800138 }
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800139 std::cout << "Content size: " << content.value_size() << std::endl;
Jeff Thompson70e80802013-12-10 20:21:03 -0800140
141 // Initialize the KeyChain storage in case useCrypto is true.
Jeff Thompson06d17802013-12-11 16:34:27 -0800142 ptr_lib::shared_ptr<MemoryIdentityStorage> identityStorage(new MemoryIdentityStorage());
143 ptr_lib::shared_ptr<MemoryPrivateKeyStorage> privateKeyStorage(new MemoryPrivateKeyStorage());
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800144 KeyChain keyChain(identityStorage, privateKeyStorage);
145
146 Name keyName("/testname/dsk-123");
147
148 // Initialize the storage.
149 identityStorage->addKey(keyName, KEY_TYPE_RSA,
150 PublicKey(DEFAULT_PUBLIC_KEY_DER, sizeof(DEFAULT_PUBLIC_KEY_DER)));
151
152 privateKeyStorage->setKeyPairForKeyName(keyName,
153 DEFAULT_PUBLIC_KEY_DER, sizeof(DEFAULT_PUBLIC_KEY_DER),
154 DEFAULT_PRIVATE_KEY_DER, sizeof(DEFAULT_PRIVATE_KEY_DER));
155
156 keyChain.identities().addCertificateAsDefault(*keyChain.identities().selfSign(keyName));
157 Name certificateName = keyChain.identities().getDefaultCertificateName();
Jeff Thompson70e80802013-12-10 20:21:03 -0800158
159 // Set up publisherPublicKeyDigest and signatureBits in case useCrypto is false.
Jeff Thompson70e80802013-12-10 20:21:03 -0800160 uint8_t signatureBitsArray[128];
161 memset(signatureBitsArray, 0, sizeof(signatureBitsArray));
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800162 Block signatureValue = dataBlock(Tlv::SignatureValue, signatureBitsArray, sizeof(signatureBitsArray));
Jeff Thompson70e80802013-12-10 20:21:03 -0800163
164 double start = getNowSeconds();
165 for (int i = 0; i < nIterations; ++i) {
166 Data data(name);
167 data.setContent(content);
168 if (useComplex) {
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800169 data.setFreshnessPeriod(1000000);
Jeff Thompson70e80802013-12-10 20:21:03 -0800170 }
171
172 if (useCrypto)
173 // This sets the signature fields.
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800174 keyChain.sign(data);
Jeff Thompson70e80802013-12-10 20:21:03 -0800175 else {
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800176 // Imitate real sign method to set up the signature fields, but don't actually sign.
177 SignatureSha256WithRsa signature;
178 signature.setKeyLocator(certificateName);
179 signature.setValue(signatureValue);
180 data.setSignature(signature);
Jeff Thompson70e80802013-12-10 20:21:03 -0800181 }
182
183 encoding = data.wireEncode();
184 }
185 double finish = getNowSeconds();
186
187 return finish - start;
188}
189
190static void
Jeff Thompson06d17802013-12-11 16:34:27 -0800191onVerified(const ptr_lib::shared_ptr<Data>& data)
Jeff Thompson70e80802013-12-10 20:21:03 -0800192{
193 // Do nothing since we expect it to verify.
194}
195
196static void
Jeff Thompson06d17802013-12-11 16:34:27 -0800197onVerifyFailed(const ptr_lib::shared_ptr<Data>& data)
Jeff Thompson70e80802013-12-10 20:21:03 -0800198{
199 cout << "Signature verification: FAILED" << endl;
200}
201
202/**
203 * Loop to decode a data packet nIterations times using C++.
204 * @param nIterations The number of iterations.
205 * @param useCrypto If true, verify the signature. If false, don't verify.
206 * @param encoding The wire encoding to decode.
207 * @return The number of seconds for all iterations.
208 */
209static double
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800210benchmarkDecodeDataSecondsCpp(int nIterations, bool useCrypto, const ConstBufferPtr &encoding)
Jeff Thompson70e80802013-12-10 20:21:03 -0800211{
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800212 // // Initialize the KeyChain storage in case useCrypto is true.
213 // ptr_lib::shared_ptr<MemoryIdentityStorage> identityStorage(new MemoryIdentityStorage());
214 // identityStorage->addKey(keyName, KEY_TYPE_RSA, Blob(DEFAULT_PUBLIC_KEY_DER, sizeof(DEFAULT_PUBLIC_KEY_DER)));
Alexander Afanasyev84c2bd42014-01-09 22:35:35 -0800215
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800216 // ptr_lib::shared_ptr<MemoryPrivateKeyStorage> privateKeyStorage(new MemoryPrivateKeyStorage());
217 // KeyChain keyChain(identityStorage, privateKeyStorage);
Alexander Afanasyev84c2bd42014-01-09 22:35:35 -0800218
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800219 // Name keyName("/testname/DSK-123");
Jeff Thompson70e80802013-12-10 20:21:03 -0800220
Jeff Thompson687ea652013-12-06 11:28:23 -0800221 size_t nameSize = 0;
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800222 double start = getNowSeconds();
Jeff Thompson687ea652013-12-06 11:28:23 -0800223 for (int i = 0; i < nIterations; ++i) {
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800224 Data data;
225 data.wireDecode(encoding);
Jeff Thompson70e80802013-12-10 20:21:03 -0800226
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800227 // if (useCrypto)
228 // keyChain.verifyData(data, onVerified, onVerifyFailed);
Jeff Thompson687ea652013-12-06 11:28:23 -0800229 }
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800230 double finish = getNowSeconds();
231
Jeff Thompson70e80802013-12-10 20:21:03 -0800232 return finish - start;
233}
234
235/**
236 * Loop to encode a data packet nIterations times using C.
237 * @param nIterations The number of iterations.
238 * @param useComplex If true, use a large name, large content and all fields. If false, use a small name, small content
239 * and only required fields.
240 * @param useCrypto If true, sign the data packet. If false, use a blank signature.
241 * @param encoding Output buffer for the wire encoding.
242 * @param maxEncodingLength The size of the encoding buffer.
243 * @param encodingLength Return the number of output bytes in encoding.
244 * @return The number of seconds for all iterations.
245 */
246static double
247benchmarkEncodeDataSecondsC
248 (int nIterations, bool useComplex, bool useCrypto, uint8_t* encoding, size_t maxEncodingLength, size_t *encodingLength)
249{
250 struct ndn_Blob finalBlockId;
251 ndn_Blob_initialize(&finalBlockId, (uint8_t*)"\x00", 1);
252
253 struct ndn_NameComponent nameComponents[20];
254 struct ndn_Name name;
255 ndn_Name_initialize(&name, nameComponents, sizeof(nameComponents) / sizeof(nameComponents[0]));
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800256 Buffer contentBlob;
Jeff Thompson70e80802013-12-10 20:21:03 -0800257 struct ndn_Blob content;
258 if (useComplex) {
259 // Use a large name and content.
260 ndn_Name_appendString(&name, (char*)"ndn");
261 ndn_Name_appendString(&name, (char*)"ucla.edu");
262 ndn_Name_appendString(&name, (char*)"apps");
263 ndn_Name_appendString(&name, (char*)"lwndn-test");
264 ndn_Name_appendString(&name, (char*)"numbers.txt");
265 ndn_Name_appendString(&name, (char*)"\xFD\x05\x05\xE8\x0C\xCE\x1D");
266 ndn_Name_appendBlob(&name, &finalBlockId);
267
268 ostringstream contentStream;
269 int count = 1;
270 contentStream << (count++);
271 while (contentStream.str().length() < 1170)
272 contentStream << " " << (count++);
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800273 contentBlob = Buffer((uint8_t*)contentStream.str().c_str(), contentStream.str().length());
Jeff Thompson70e80802013-12-10 20:21:03 -0800274 }
275 else {
276 // Use a small name and content.
277 ndn_Name_appendString(&name, (char*)"test");
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800278 contentBlob = Buffer((uint8_t*)"abc", 3);
Jeff Thompson70e80802013-12-10 20:21:03 -0800279 }
280 ndn_Blob_initialize(&content, (uint8_t*)contentBlob.buf(), contentBlob.size());
281
282 struct ndn_NameComponent certificateNameComponents[20];
283 struct ndn_Name certificateName;
284 ndn_Name_initialize(&certificateName, certificateNameComponents, sizeof(certificateNameComponents) / sizeof(certificateNameComponents[0]));
285 ndn_Name_appendString(&certificateName, (char*)"testname");
286 ndn_Name_appendString(&certificateName, (char*)"KEY");
287 ndn_Name_appendString(&certificateName, (char*)"DSK-123");
288 ndn_Name_appendString(&certificateName, (char*)"ID-CERT");
289 ndn_Name_appendString(&certificateName, (char*)"0");
290
291 // Set up publisherPublicKeyDigest and signatureBits in case useCrypto is false.
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800292 uint8_t* publicKeyDer = DEFAULT_PUBLIC_KEY_DER;
293 size_t publicKeyDerLength = sizeof(DEFAULT_PUBLIC_KEY_DER);
294 uint8_t publisherPublicKeyDigestArray[SHA256_DIGEST_LENGTH];
295 ndn_digestSha256(publicKeyDer, publicKeyDerLength, publisherPublicKeyDigestArray);
Jeff Thompson70e80802013-12-10 20:21:03 -0800296 struct ndn_Blob publisherPublicKeyDigest;
297 ndn_Blob_initialize(&publisherPublicKeyDigest, publisherPublicKeyDigestArray, sizeof(publisherPublicKeyDigestArray));
298 uint8_t signatureBitsArray[128];
299 memset(signatureBitsArray, 0, sizeof(signatureBitsArray));
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800300
301 // Set up the private key now in case useCrypto is true.
302 // Use a temporary pointer since d2i updates it.
303 const uint8_t *privateKeyDerPointer = DEFAULT_PRIVATE_KEY_DER;
304 RSA *privateKey = d2i_RSAPrivateKey(NULL, &privateKeyDerPointer, sizeof(DEFAULT_PRIVATE_KEY_DER));
305 if (!privateKey) {
306 // Don't expect this to happen.
307 cout << "Error decoding private key DER" << endl;
308 return 0;
309 }
Jeff Thompson70e80802013-12-10 20:21:03 -0800310
311 double start = getNowSeconds();
312 for (int i = 0; i < nIterations; ++i) {
313 struct ndn_Data data;
314 ndn_Data_initialize(&data, name.components, name.maxComponents, certificateName.components, certificateName.maxComponents);
315
316 data.name = name;
317 data.content = content;
318 if (useComplex) {
319 data.metaInfo.timestampMilliseconds = 1.3e+12;
320 data.metaInfo.freshnessSeconds = 1000;
321 ndn_NameComponent_initialize(&data.metaInfo.finalBlockID, finalBlockId.value, finalBlockId.length);
322 }
323
Jeff Thompson70e80802013-12-10 20:21:03 -0800324 struct ndn_DynamicUInt8Array output;
Jeff Thompson70e80802013-12-10 20:21:03 -0800325 struct ndn_BinaryXmlEncoder encoder;
Jeff Thompson70e80802013-12-10 20:21:03 -0800326 size_t signedPortionBeginOffset, signedPortionEndOffset;
327 ndn_Error error;
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800328
329 data.signature.keyLocator.type = ndn_KeyLocatorType_KEYNAME;
330 data.signature.keyLocator.keyName = certificateName;
331 data.signature.keyLocator.keyNameType = (ndn_KeyNameType)-1;
332 data.signature.publisherPublicKeyDigest.publisherPublicKeyDigest = publisherPublicKeyDigest;
333 if (useCrypto) {
334 // Encode once to get the signed portion.
335 ndn_DynamicUInt8Array_initialize(&output, encoding, maxEncodingLength, 0);
336 ndn_BinaryXmlEncoder_initialize(&encoder, &output);
337 if ((error = ndn_encodeBinaryXmlData(&data, &signedPortionBeginOffset, &signedPortionEndOffset, &encoder))) {
338 cout << "Error in ndn_encodeBinaryXmlData: " << ndn_getErrorString(error) << endl;
339 return 0;
340 }
341
342 // Imitate MemoryPrivateKeyStorage::sign.
343 uint8_t digest[SHA256_DIGEST_LENGTH];
344 ndn_digestSha256(encoding + signedPortionBeginOffset, signedPortionEndOffset - signedPortionBeginOffset, digest);
345 unsigned int signatureBitsLength;
346 if (!RSA_sign(NID_sha256, digest, sizeof(digest), signatureBitsArray, &signatureBitsLength, privateKey)) {
347 // Don't expect this to happen.
348 cout << "Error in RSA_sign" << endl;
349 return 0;
350 }
351
352 ndn_Blob_initialize(&data.signature.signature, signatureBitsArray, signatureBitsLength);
353 }
354 else
355 // Set up the signature, but don't sign.
356 ndn_Blob_initialize(&data.signature.signature, signatureBitsArray, sizeof(signatureBitsArray));
357
358 // Assume the encoding buffer is big enough so we don't need to dynamically reallocate.
359 ndn_DynamicUInt8Array_initialize(&output, encoding, maxEncodingLength, 0);
360 ndn_BinaryXmlEncoder_initialize(&encoder, &output);
Jeff Thompson70e80802013-12-10 20:21:03 -0800361 if ((error = ndn_encodeBinaryXmlData(&data, &signedPortionBeginOffset, &signedPortionEndOffset, &encoder))) {
362 cout << "Error in ndn_encodeBinaryXmlData: " << ndn_getErrorString(error) << endl;
363 return 0;
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800364 }
Jeff Thompson70e80802013-12-10 20:21:03 -0800365 *encodingLength = encoder.offset;
366 }
367 double finish = getNowSeconds();
Jeff Thompson687ea652013-12-06 11:28:23 -0800368
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800369 if (privateKey)
370 RSA_free(privateKey);
371
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800372 return finish - start;
Jeff Thompson687ea652013-12-06 11:28:23 -0800373}
374
Jeff Thompson70e80802013-12-10 20:21:03 -0800375/**
376 * Loop to decode a data packet nIterations times using C.
377 * @param nIterations The number of iterations.
378 * @param useCrypto If true, verify the signature. If false, don't verify.
379 * @param encoding The buffer with wire encoding to decode.
380 * @param encodingLength The number of bytes in the encoding.
381 * @return The number of seconds for all iterations.
382 */
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800383static double
Jeff Thompson70e80802013-12-10 20:21:03 -0800384benchmarkDecodeDataSecondsC(int nIterations, bool useCrypto, uint8_t* encoding, size_t encodingLength)
Jeff Thompson687ea652013-12-06 11:28:23 -0800385{
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800386 double start = getNowSeconds();
Jeff Thompson687ea652013-12-06 11:28:23 -0800387 for (int i = 0; i < nIterations; ++i) {
388 struct ndn_NameComponent nameComponents[100];
389 struct ndn_NameComponent keyNameComponents[100];
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800390 struct ndn_Data data;
Jeff Thompson687ea652013-12-06 11:28:23 -0800391 ndn_Data_initialize
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800392 (&data, nameComponents, sizeof(nameComponents) / sizeof(nameComponents[0]),
Jeff Thompson687ea652013-12-06 11:28:23 -0800393 keyNameComponents, sizeof(keyNameComponents) / sizeof(keyNameComponents[0]));
394
395 ndn_BinaryXmlDecoder decoder;
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800396 ndn_BinaryXmlDecoder_initialize(&decoder, encoding, encodingLength);
Jeff Thompson687ea652013-12-06 11:28:23 -0800397 size_t signedPortionBeginOffset, signedPortionEndOffset;
398 ndn_Error error;
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800399 if ((error = ndn_decodeBinaryXmlData(&data, &signedPortionBeginOffset, &signedPortionEndOffset, &decoder))) {
Jeff Thompson687ea652013-12-06 11:28:23 -0800400 cout << "Error in ndn_decodeBinaryXmlData: " << ndn_getErrorString(error) << endl;
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800401 return 0;
Jeff Thompson687ea652013-12-06 11:28:23 -0800402 }
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800403
404 if (useCrypto) {
405 if (!verifyRsaSignature
406 (encoding + signedPortionBeginOffset, signedPortionEndOffset - signedPortionBeginOffset,
407 data.signature.signature.value, data.signature.signature.length,
408 DEFAULT_PUBLIC_KEY_DER, sizeof(DEFAULT_PUBLIC_KEY_DER)))
409 cout << "Signature verification: FAILED" << endl;
410 }
Jeff Thompson687ea652013-12-06 11:28:23 -0800411 }
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800412 double finish = getNowSeconds();
413
Jeff Thompsonecaf3262013-12-09 17:13:49 -0800414 return finish - start;
Jeff Thompson687ea652013-12-06 11:28:23 -0800415}
416
Jeff Thompson70e80802013-12-10 20:21:03 -0800417/**
Jeff Thompson4c66e712013-12-11 13:37:41 -0800418 * Call benchmarkEncodeDataSecondsCpp and benchmarkDecodeDataSecondsCpp with appropriate nInterations. Print the
Jeff Thompson70e80802013-12-10 20:21:03 -0800419 * results to cout.
Jeff Thompsonb33b21e2013-12-11 17:37:55 -0800420 * @param useComplex See benchmarkEncodeDataSecondsCpp.
Jeff Thompson4c66e712013-12-11 13:37:41 -0800421 * @param useCrypto See benchmarkEncodeDataSecondsCpp and benchmarkDecodeDataSecondsCpp.
Jeff Thompson70e80802013-12-10 20:21:03 -0800422 */
423static void
424benchmarkEncodeDecodeDataCpp(bool useComplex, bool useCrypto)
425{
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800426 Block encoding;
Jeff Thompson70e80802013-12-10 20:21:03 -0800427 {
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800428 int nIterations = useCrypto ? 20000 : 200000;
Jeff Thompson70e80802013-12-10 20:21:03 -0800429 double duration = benchmarkEncodeDataSecondsCpp(nIterations, useComplex, useCrypto, encoding);
430 cout << "Encode " << (useComplex ? "complex" : "simple ") << " data C++: Crypto? " << (useCrypto ? "yes" : "no ")
431 << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl;
432 }
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800433
434 BufferPtr wire = ptr_lib::make_shared<Buffer>(encoding.wire(), encoding.size());
Jeff Thompson70e80802013-12-10 20:21:03 -0800435 {
Alexander Afanasyev3f1d2282014-01-10 10:12:07 -0800436 int nIterations = useCrypto ? 10000 : 1000000;
437 double duration = benchmarkDecodeDataSecondsCpp(nIterations, useCrypto, wire);
Jeff Thompson70e80802013-12-10 20:21:03 -0800438 cout << "Decode " << (useComplex ? "complex" : "simple ") << " data C++: Crypto? " << (useCrypto ? "yes" : "no ")
439 << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl;
440 }
441}
442
443/**
Jeff Thompson4c66e712013-12-11 13:37:41 -0800444 * Call benchmarkEncodeDataSecondsC and benchmarkDecodeDataSecondsC with appropriate nInterations. Print the
Jeff Thompson70e80802013-12-10 20:21:03 -0800445 * results to cout.
Jeff Thompsonb33b21e2013-12-11 17:37:55 -0800446 * @param useComplex See benchmarkEncodeDataSecondsC.
Jeff Thompson4c66e712013-12-11 13:37:41 -0800447 * @param useCrypto See benchmarkEncodeDataSecondsC and benchmarkDecodeDataSecondsC.
Jeff Thompson70e80802013-12-10 20:21:03 -0800448 */
449static void
450benchmarkEncodeDecodeDataC(bool useComplex, bool useCrypto)
451{
452 uint8_t encoding[1500];
453 size_t encodingLength;
454 {
455 int nIterations = useCrypto ? 20000 : 10000000;
456 double duration = benchmarkEncodeDataSecondsC(nIterations, useComplex, useCrypto, encoding, sizeof(encoding), &encodingLength);
457 cout << "Encode " << (useComplex ? "complex" : "simple ") << " data C: Crypto? " << (useCrypto ? "yes" : "no ")
458 << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl;
459 }
460 {
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800461 int nIterations = useCrypto ? 150000 : 15000000;
Jeff Thompson70e80802013-12-10 20:21:03 -0800462 double duration = benchmarkDecodeDataSecondsC(nIterations, useCrypto, encoding, encodingLength);
463 cout << "Decode " << (useComplex ? "complex" : "simple ") << " data C: Crypto? " << (useCrypto ? "yes" : "no ")
464 << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl;
465 }
466}
467
Jeff Thompson687ea652013-12-06 11:28:23 -0800468int
469main(int argc, char** argv)
470{
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800471 try {
Jeff Thompson70e80802013-12-10 20:21:03 -0800472 benchmarkEncodeDecodeDataCpp(false, false);
473 benchmarkEncodeDecodeDataCpp(true, false);
474 benchmarkEncodeDecodeDataCpp(false, true);
475 benchmarkEncodeDecodeDataCpp(true, true);
476
477 benchmarkEncodeDecodeDataC(false, false);
478 benchmarkEncodeDecodeDataC(true, false);
Jeff Thompson30d3e6c2013-12-11 12:35:54 -0800479 benchmarkEncodeDecodeDataC(false, true);
480 benchmarkEncodeDecodeDataC(true, true);
Jeff Thompson18dba8b2013-12-09 12:17:40 -0800481 } catch (std::exception& e) {
482 cout << "exception: " << e.what() << endl;
483 }
Jeff Thompson687ea652013-12-06 11:28:23 -0800484 return 0;
485}