Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 1 | /* -*- 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 Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 11 | #include <sstream> |
Jeff Thompson | 18dba8b | 2013-12-09 12:17:40 -0800 | [diff] [blame] | 12 | #include <stdexcept> |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 13 | #include <ndn-cpp/data.hpp> |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 14 | #include <ndn-cpp/sha256-with-rsa-signature.hpp> |
| 15 | #include <ndn-cpp/key.hpp> |
| 16 | #include <ndn-cpp/security/key-chain.hpp> |
| 17 | #include <ndn-cpp/security/identity/memory-identity-storage.hpp> |
| 18 | #include <ndn-cpp/security/identity/memory-private-key-storage.hpp> |
| 19 | #include <ndn-cpp/security/policy/self-verify-policy-manager.hpp> |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 20 | // Hack: Hook directly into non-API functions. |
| 21 | #include "../src/c/encoding/binary-xml-decoder.h" |
| 22 | #include "../src/c/data.h" |
| 23 | #include "../src/c/encoding/binary-xml-data.h" |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 24 | #include "../src/c/util/crypto.h" |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 25 | |
| 26 | using namespace std; |
| 27 | using namespace ndn; |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 28 | |
| 29 | static double |
Jeff Thompson | 18dba8b | 2013-12-09 12:17:40 -0800 | [diff] [blame] | 30 | getNowSeconds() |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 31 | { |
| 32 | struct timeval t; |
| 33 | gettimeofday(&t, 0); |
Jeff Thompson | 18dba8b | 2013-12-09 12:17:40 -0800 | [diff] [blame] | 34 | return t.tv_sec + t.tv_usec / 1000000.0; |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 35 | } |
| 36 | |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 37 | static bool |
| 38 | verifyRsaSignature |
| 39 | (uint8_t* signedPortion, size_t signedPortionLength, uint8_t* signatureBits, size_t signatureBitsLength, |
| 40 | uint8_t* publicKeyDer, size_t publicKeyDerLength) |
| 41 | { |
| 42 | // Set signedPortionDigest to the digest of the signed portion of the wire encoding. |
| 43 | uint8_t signedPortionDigest[SHA256_DIGEST_LENGTH]; |
| 44 | ndn_digestSha256(signedPortion, signedPortionLength, signedPortionDigest); |
| 45 | |
| 46 | // Verify the signedPortionDigest. |
| 47 | // Use a temporary pointer since d2i updates it. |
| 48 | const uint8_t *derPointer = publicKeyDer; |
| 49 | RSA *rsaPublicKey = d2i_RSA_PUBKEY(NULL, &derPointer, publicKeyDerLength); |
| 50 | if (!rsaPublicKey) { |
| 51 | // Don't expect this to happen. |
| 52 | cout << "Error decoding public key in d2i_RSAPublicKey" << endl; |
| 53 | return 0; |
| 54 | } |
| 55 | int success = RSA_verify |
| 56 | (NID_sha256, signedPortionDigest, sizeof(signedPortionDigest), signatureBits, signatureBitsLength, rsaPublicKey); |
| 57 | // Free the public key before checking for success. |
| 58 | RSA_free(rsaPublicKey); |
| 59 | |
| 60 | // RSA_verify returns 1 for a valid signature. |
| 61 | return (success == 1); |
| 62 | } |
| 63 | |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 64 | static uint8_t DEFAULT_PUBLIC_KEY_DER[] = { |
| 65 | 0x30, 0x81, 0x9F, 0x30, 0x0D, 0x06, 0x09, 0x2A, 0x86, 0x48, 0x86, 0xF7, 0x0D, 0x01, 0x01, 0x01, 0x05, 0x00, 0x03, 0x81, |
| 66 | 0x8D, 0x00, 0x30, 0x81, 0x89, 0x02, 0x81, 0x81, 0x00, 0xE1, 0x7D, 0x30, 0xA7, 0xD8, 0x28, 0xAB, 0x1B, 0x84, 0x0B, 0x17, |
| 67 | 0x54, 0x2D, 0xCA, 0xF6, 0x20, 0x7A, 0xFD, 0x22, 0x1E, 0x08, 0x6B, 0x2A, 0x60, 0xD1, 0x6C, 0xB7, 0xF5, 0x44, 0x48, 0xBA, |
| 68 | 0x9F, 0x3F, 0x08, 0xBC, 0xD0, 0x99, 0xDB, 0x21, 0xDD, 0x16, 0x2A, 0x77, 0x9E, 0x61, 0xAA, 0x89, 0xEE, 0xE5, 0x54, 0xD3, |
| 69 | 0xA4, 0x7D, 0xE2, 0x30, 0xBC, 0x7A, 0xC5, 0x90, 0xD5, 0x24, 0x06, 0x7C, 0x38, 0x98, 0xBB, 0xA6, 0xF5, 0xDC, 0x43, 0x60, |
| 70 | 0xB8, 0x45, 0xED, 0xA4, 0x8C, 0xBD, 0x9C, 0xF1, 0x26, 0xA7, 0x23, 0x44, 0x5F, 0x0E, 0x19, 0x52, 0xD7, 0x32, 0x5A, 0x75, |
| 71 | 0xFA, 0xF5, 0x56, 0x14, 0x4F, 0x9A, 0x98, 0xAF, 0x71, 0x86, 0xB0, 0x27, 0x86, 0x85, 0xB8, 0xE2, 0xC0, 0x8B, 0xEA, 0x87, |
| 72 | 0x17, 0x1B, 0x4D, 0xEE, 0x58, 0x5C, 0x18, 0x28, 0x29, 0x5B, 0x53, 0x95, 0xEB, 0x4A, 0x17, 0x77, 0x9F, 0x02, 0x03, 0x01, |
Jeff Thompson | 9e387fe | 2013-12-11 13:00:36 -0800 | [diff] [blame] | 73 | 0x00, 0x01 |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 74 | }; |
| 75 | |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 76 | static uint8_t DEFAULT_PRIVATE_KEY_DER[] = { |
| 77 | 0x30, 0x82, 0x02, 0x5d, 0x02, 0x01, 0x00, 0x02, 0x81, 0x81, 0x00, 0xe1, 0x7d, 0x30, 0xa7, 0xd8, 0x28, 0xab, 0x1b, 0x84, |
| 78 | 0x0b, 0x17, 0x54, 0x2d, 0xca, 0xf6, 0x20, 0x7a, 0xfd, 0x22, 0x1e, 0x08, 0x6b, 0x2a, 0x60, 0xd1, 0x6c, 0xb7, 0xf5, 0x44, |
| 79 | 0x48, 0xba, 0x9f, 0x3f, 0x08, 0xbc, 0xd0, 0x99, 0xdb, 0x21, 0xdd, 0x16, 0x2a, 0x77, 0x9e, 0x61, 0xaa, 0x89, 0xee, 0xe5, |
| 80 | 0x54, 0xd3, 0xa4, 0x7d, 0xe2, 0x30, 0xbc, 0x7a, 0xc5, 0x90, 0xd5, 0x24, 0x06, 0x7c, 0x38, 0x98, 0xbb, 0xa6, 0xf5, 0xdc, |
| 81 | 0x43, 0x60, 0xb8, 0x45, 0xed, 0xa4, 0x8c, 0xbd, 0x9c, 0xf1, 0x26, 0xa7, 0x23, 0x44, 0x5f, 0x0e, 0x19, 0x52, 0xd7, 0x32, |
| 82 | 0x5a, 0x75, 0xfa, 0xf5, 0x56, 0x14, 0x4f, 0x9a, 0x98, 0xaf, 0x71, 0x86, 0xb0, 0x27, 0x86, 0x85, 0xb8, 0xe2, 0xc0, 0x8b, |
| 83 | 0xea, 0x87, 0x17, 0x1b, 0x4d, 0xee, 0x58, 0x5c, 0x18, 0x28, 0x29, 0x5b, 0x53, 0x95, 0xeb, 0x4a, 0x17, 0x77, 0x9f, 0x02, |
| 84 | 0x03, 0x01, 0x00, 0x01, 0x02, 0x81, 0x80, 0x1a, 0x4b, 0xfa, 0x4f, 0xa8, 0xc2, 0xdd, 0x69, 0xa1, 0x15, 0x96, 0x0b, 0xe8, |
| 85 | 0x27, 0x42, 0x5a, 0xf9, 0x5c, 0xea, 0x0c, 0xac, 0x98, 0xaa, 0xe1, 0x8d, 0xaa, 0xeb, 0x2d, 0x3c, 0x60, 0x6a, 0xfb, 0x45, |
| 86 | 0x63, 0xa4, 0x79, 0x83, 0x67, 0xed, 0xe4, 0x15, 0xc0, 0xb0, 0x20, 0x95, 0x6d, 0x49, 0x16, 0xc6, 0x42, 0x05, 0x48, 0xaa, |
| 87 | 0xb1, 0xa5, 0x53, 0x65, 0xd2, 0x02, 0x99, 0x08, 0xd1, 0x84, 0xcc, 0xf0, 0xcd, 0xea, 0x61, 0xc9, 0x39, 0x02, 0x3f, 0x87, |
| 88 | 0x4a, 0xe5, 0xc4, 0xd2, 0x07, 0x02, 0xe1, 0x9f, 0xa0, 0x06, 0xc2, 0xcc, 0x02, 0xe7, 0xaa, 0x6c, 0x99, 0x8a, 0xf8, 0x49, |
| 89 | 0x00, 0xf1, 0xa2, 0x8c, 0x0c, 0x8a, 0xb9, 0x4f, 0x6d, 0x73, 0x3b, 0x2c, 0xb7, 0x9f, 0x8a, 0xa6, 0x7f, 0x9b, 0x9f, 0xb7, |
| 90 | 0xa1, 0xcc, 0x74, 0x2e, 0x8f, 0xb8, 0xb0, 0x26, 0x89, 0xd2, 0xe5, 0x66, 0xe8, 0x8e, 0xa1, 0x02, 0x41, 0x00, 0xfc, 0xe7, |
| 91 | 0x52, 0xbc, 0x4e, 0x95, 0xb6, 0x1a, 0xb4, 0x62, 0xcc, 0xd8, 0x06, 0xe1, 0xdc, 0x7a, 0xa2, 0xb6, 0x71, 0x01, 0xaa, 0x27, |
| 92 | 0xfc, 0x99, 0xe5, 0xf2, 0x54, 0xbb, 0xb2, 0x85, 0xe1, 0x96, 0x54, 0x2d, 0xcb, 0xba, 0x86, 0xfa, 0x80, 0xdf, 0xcf, 0x39, |
| 93 | 0xe6, 0x74, 0xcb, 0x22, 0xce, 0x70, 0xaa, 0x10, 0x00, 0x73, 0x1d, 0x45, 0x0a, 0x39, 0x51, 0x84, 0xf5, 0x15, 0x8f, 0x37, |
| 94 | 0x76, 0x91, 0x02, 0x41, 0x00, 0xe4, 0x3f, 0xf0, 0xf4, 0xde, 0x79, 0x77, 0x48, 0x9b, 0x9c, 0x28, 0x45, 0x26, 0x57, 0x3c, |
| 95 | 0x71, 0x40, 0x28, 0x6a, 0xa1, 0xfe, 0xc3, 0xe5, 0x37, 0xa1, 0x03, 0xf6, 0x2d, 0xbe, 0x80, 0x64, 0x72, 0x69, 0x2e, 0x9b, |
| 96 | 0x4d, 0xe3, 0x2e, 0x1b, 0xfe, 0xe7, 0xf9, 0x77, 0x8c, 0x18, 0x53, 0x9f, 0xe2, 0xfe, 0x00, 0xbb, 0x49, 0x20, 0x47, 0xdf, |
| 97 | 0x01, 0x61, 0x87, 0xd6, 0xe3, 0x44, 0xb5, 0x03, 0x2f, 0x02, 0x40, 0x54, 0xec, 0x7c, 0xbc, 0xdd, 0x0a, 0xaa, 0xde, 0xe6, |
| 98 | 0xc9, 0xf2, 0x8d, 0x6c, 0x2a, 0x35, 0xf6, 0x3c, 0x63, 0x55, 0x29, 0x40, 0xf1, 0x32, 0x82, 0x9f, 0x53, 0xb3, 0x9e, 0x5f, |
| 99 | 0xc1, 0x53, 0x52, 0x3e, 0xac, 0x2e, 0x28, 0x51, 0xa1, 0x16, 0xdb, 0x90, 0xe3, 0x99, 0x7e, 0x88, 0xa4, 0x04, 0x7c, 0x92, |
| 100 | 0xae, 0xd2, 0xe7, 0xd4, 0xe1, 0x55, 0x20, 0x90, 0x3e, 0x3c, 0x6a, 0x63, 0xf0, 0x34, 0xf1, 0x02, 0x41, 0x00, 0x84, 0x5a, |
| 101 | 0x17, 0x6c, 0xc6, 0x3c, 0x84, 0xd0, 0x93, 0x7a, 0xff, 0x56, 0xe9, 0x9e, 0x98, 0x2b, 0xcb, 0x5a, 0x24, 0x4a, 0xff, 0x21, |
| 102 | 0xb4, 0x9e, 0x87, 0x3d, 0x76, 0xd8, 0x9b, 0xa8, 0x73, 0x96, 0x6c, 0x2b, 0x5c, 0x5e, 0xd3, 0xa6, 0xff, 0x10, 0xd6, 0x8e, |
| 103 | 0xaf, 0xa5, 0x8a, 0xcd, 0xa2, 0xde, 0xcb, 0x0e, 0xbd, 0x8a, 0xef, 0xae, 0xfd, 0x3f, 0x1d, 0xc0, 0xd8, 0xf8, 0x3b, 0xf5, |
| 104 | 0x02, 0x7d, 0x02, 0x41, 0x00, 0x8b, 0x26, 0xd3, 0x2c, 0x7d, 0x28, 0x38, 0x92, 0xf1, 0xbf, 0x15, 0x16, 0x39, 0x50, 0xc8, |
| 105 | 0x6d, 0x32, 0xec, 0x28, 0xf2, 0x8b, 0xd8, 0x70, 0xc5, 0xed, 0xe1, 0x7b, 0xff, 0x2d, 0x66, 0x8c, 0x86, 0x77, 0x43, 0xeb, |
| 106 | 0xb6, 0xf6, 0x50, 0x66, 0xb0, 0x40, 0x24, 0x6a, 0xaf, 0x98, 0x21, 0x45, 0x30, 0x01, 0x59, 0xd0, 0xc3, 0xfc, 0x7b, 0xae, |
| 107 | 0x30, 0x18, 0xeb, 0x90, 0xfb, 0x17, 0xd3, 0xce, 0xb5 |
| 108 | }; |
Jeff Thompson | ecaf326 | 2013-12-09 17:13:49 -0800 | [diff] [blame] | 109 | |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 110 | /** |
| 111 | * Loop to encode a data packet nIterations times using C++. |
| 112 | * @param nIterations The number of iterations. |
| 113 | * @param useComplex If true, use a large name, large content and all fields. If false, use a small name, small content |
| 114 | * and only required fields. |
| 115 | * @param useCrypto If true, sign the data packet. If false, use a blank signature. |
| 116 | * @param encoding Set this to the wire encoding. |
| 117 | * @return The number of seconds for all iterations. |
| 118 | */ |
| 119 | static double |
| 120 | benchmarkEncodeDataSecondsCpp(int nIterations, bool useComplex, bool useCrypto, Blob& encoding) |
Jeff Thompson | ecaf326 | 2013-12-09 17:13:49 -0800 | [diff] [blame] | 121 | { |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 122 | Name name; |
| 123 | Blob content; |
| 124 | if (useComplex) { |
| 125 | // Use a large name and content. |
| 126 | name = Name("/ndn/ucla.edu/apps/lwndn-test/numbers.txt/%FD%05%05%E8%0C%CE%1D/%00"); |
| 127 | |
| 128 | ostringstream contentStream; |
| 129 | int count = 1; |
| 130 | contentStream << (count++); |
| 131 | while (contentStream.str().length() < 1170) |
| 132 | contentStream << " " << (count++); |
| 133 | content = Blob((uint8_t*)contentStream.str().c_str(), contentStream.str().length()); |
| 134 | } |
| 135 | else { |
| 136 | // Use a small name and content. |
| 137 | name = Name("/test"); |
| 138 | content = Blob((uint8_t*)"abc", 3); |
| 139 | } |
| 140 | Blob finalBlockId((uint8_t*)"\x00", 1); |
| 141 | |
| 142 | // Initialize the KeyChain storage in case useCrypto is true. |
Jeff Thompson | 06d1780 | 2013-12-11 16:34:27 -0800 | [diff] [blame] | 143 | ptr_lib::shared_ptr<MemoryIdentityStorage> identityStorage(new MemoryIdentityStorage()); |
| 144 | ptr_lib::shared_ptr<MemoryPrivateKeyStorage> privateKeyStorage(new MemoryPrivateKeyStorage()); |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 145 | KeyChain keyChain |
Jeff Thompson | 06d1780 | 2013-12-11 16:34:27 -0800 | [diff] [blame] | 146 | (ptr_lib::make_shared<IdentityManager>(identityStorage, privateKeyStorage), |
| 147 | ptr_lib::make_shared<SelfVerifyPolicyManager>(identityStorage.get())); |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 148 | Name keyName("/testname/DSK-123"); |
| 149 | Name certificateName = keyName.getSubName(0, keyName.size() - 1).append("KEY").append |
| 150 | (keyName.get(keyName.size() - 1)).append("ID-CERT").append("0"); |
| 151 | privateKeyStorage->setKeyPairForKeyName |
| 152 | (keyName, DEFAULT_PUBLIC_KEY_DER, sizeof(DEFAULT_PUBLIC_KEY_DER), DEFAULT_PRIVATE_KEY_DER, sizeof(DEFAULT_PRIVATE_KEY_DER)); |
| 153 | |
| 154 | // Set up publisherPublicKeyDigest and signatureBits in case useCrypto is false. |
| 155 | uint8_t publisherPublicKeyDigestArray[32]; |
| 156 | memset(publisherPublicKeyDigestArray, 0, sizeof(publisherPublicKeyDigestArray)); |
| 157 | Blob publisherPublicKeyDigest(publisherPublicKeyDigestArray, sizeof(publisherPublicKeyDigestArray)); |
| 158 | uint8_t signatureBitsArray[128]; |
| 159 | memset(signatureBitsArray, 0, sizeof(signatureBitsArray)); |
| 160 | Blob signatureBits(signatureBitsArray, sizeof(signatureBitsArray)); |
| 161 | |
| 162 | double start = getNowSeconds(); |
| 163 | for (int i = 0; i < nIterations; ++i) { |
| 164 | Data data(name); |
| 165 | data.setContent(content); |
| 166 | if (useComplex) { |
| 167 | data.getMetaInfo().setTimestampMilliseconds(1.3e+12); |
| 168 | data.getMetaInfo().setFreshnessSeconds(1000); |
| 169 | data.getMetaInfo().setFinalBlockID(finalBlockId); |
| 170 | } |
| 171 | |
| 172 | if (useCrypto) |
| 173 | // This sets the signature fields. |
| 174 | keyChain.sign(data, certificateName); |
| 175 | else { |
| 176 | // Imitate IdentityManager::signByCertificate to set up the signature fields, but don't sign. |
| 177 | KeyLocator keyLocator; |
| 178 | keyLocator.setType(ndn_KeyLocatorType_KEYNAME); |
| 179 | keyLocator.setKeyName(certificateName); |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 180 | keyLocator.setKeyNameType((ndn_KeyNameType)-1); |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 181 | Sha256WithRsaSignature* sha256Signature = (Sha256WithRsaSignature*)data.getSignature(); |
| 182 | sha256Signature->setKeyLocator(keyLocator); |
| 183 | sha256Signature->getPublisherPublicKeyDigest().setPublisherPublicKeyDigest(publisherPublicKeyDigest); |
| 184 | sha256Signature->setSignature(signatureBits); |
| 185 | } |
| 186 | |
| 187 | encoding = data.wireEncode(); |
| 188 | } |
| 189 | double finish = getNowSeconds(); |
| 190 | |
| 191 | return finish - start; |
| 192 | } |
| 193 | |
| 194 | static void |
Jeff Thompson | 06d1780 | 2013-12-11 16:34:27 -0800 | [diff] [blame] | 195 | onVerified(const ptr_lib::shared_ptr<Data>& data) |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 196 | { |
| 197 | // Do nothing since we expect it to verify. |
| 198 | } |
| 199 | |
| 200 | static void |
Jeff Thompson | 06d1780 | 2013-12-11 16:34:27 -0800 | [diff] [blame] | 201 | onVerifyFailed(const ptr_lib::shared_ptr<Data>& data) |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 202 | { |
| 203 | cout << "Signature verification: FAILED" << endl; |
| 204 | } |
| 205 | |
| 206 | /** |
| 207 | * Loop to decode a data packet nIterations times using C++. |
| 208 | * @param nIterations The number of iterations. |
| 209 | * @param useCrypto If true, verify the signature. If false, don't verify. |
| 210 | * @param encoding The wire encoding to decode. |
| 211 | * @return The number of seconds for all iterations. |
| 212 | */ |
| 213 | static double |
| 214 | benchmarkDecodeDataSecondsCpp(int nIterations, bool useCrypto, const Blob& encoding) |
| 215 | { |
| 216 | // Initialize the KeyChain storage in case useCrypto is true. |
Jeff Thompson | 06d1780 | 2013-12-11 16:34:27 -0800 | [diff] [blame] | 217 | ptr_lib::shared_ptr<MemoryIdentityStorage> identityStorage(new MemoryIdentityStorage()); |
| 218 | ptr_lib::shared_ptr<MemoryPrivateKeyStorage> privateKeyStorage(new MemoryPrivateKeyStorage()); |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 219 | KeyChain keyChain |
Jeff Thompson | 06d1780 | 2013-12-11 16:34:27 -0800 | [diff] [blame] | 220 | (ptr_lib::make_shared<IdentityManager>(identityStorage, privateKeyStorage), |
| 221 | ptr_lib::make_shared<SelfVerifyPolicyManager>(identityStorage.get())); |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 222 | Name keyName("/testname/DSK-123"); |
| 223 | identityStorage->addKey(keyName, KEY_TYPE_RSA, Blob(DEFAULT_PUBLIC_KEY_DER, sizeof(DEFAULT_PUBLIC_KEY_DER))); |
| 224 | |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 225 | size_t nameSize = 0; |
Jeff Thompson | 18dba8b | 2013-12-09 12:17:40 -0800 | [diff] [blame] | 226 | double start = getNowSeconds(); |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 227 | for (int i = 0; i < nIterations; ++i) { |
Jeff Thompson | 06d1780 | 2013-12-11 16:34:27 -0800 | [diff] [blame] | 228 | ptr_lib::shared_ptr<Data> data(new Data()); |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 229 | data->wireDecode(*encoding); |
| 230 | |
| 231 | if (useCrypto) |
| 232 | keyChain.verifyData(data, onVerified, onVerifyFailed); |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 233 | } |
Jeff Thompson | 18dba8b | 2013-12-09 12:17:40 -0800 | [diff] [blame] | 234 | double finish = getNowSeconds(); |
| 235 | |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 236 | return finish - start; |
| 237 | } |
| 238 | |
| 239 | /** |
| 240 | * Loop to encode a data packet nIterations times using C. |
| 241 | * @param nIterations The number of iterations. |
| 242 | * @param useComplex If true, use a large name, large content and all fields. If false, use a small name, small content |
| 243 | * and only required fields. |
| 244 | * @param useCrypto If true, sign the data packet. If false, use a blank signature. |
| 245 | * @param encoding Output buffer for the wire encoding. |
| 246 | * @param maxEncodingLength The size of the encoding buffer. |
| 247 | * @param encodingLength Return the number of output bytes in encoding. |
| 248 | * @return The number of seconds for all iterations. |
| 249 | */ |
| 250 | static double |
| 251 | benchmarkEncodeDataSecondsC |
| 252 | (int nIterations, bool useComplex, bool useCrypto, uint8_t* encoding, size_t maxEncodingLength, size_t *encodingLength) |
| 253 | { |
| 254 | struct ndn_Blob finalBlockId; |
| 255 | ndn_Blob_initialize(&finalBlockId, (uint8_t*)"\x00", 1); |
| 256 | |
| 257 | struct ndn_NameComponent nameComponents[20]; |
| 258 | struct ndn_Name name; |
| 259 | ndn_Name_initialize(&name, nameComponents, sizeof(nameComponents) / sizeof(nameComponents[0])); |
| 260 | Blob contentBlob; |
| 261 | struct ndn_Blob content; |
| 262 | if (useComplex) { |
| 263 | // Use a large name and content. |
| 264 | ndn_Name_appendString(&name, (char*)"ndn"); |
| 265 | ndn_Name_appendString(&name, (char*)"ucla.edu"); |
| 266 | ndn_Name_appendString(&name, (char*)"apps"); |
| 267 | ndn_Name_appendString(&name, (char*)"lwndn-test"); |
| 268 | ndn_Name_appendString(&name, (char*)"numbers.txt"); |
| 269 | ndn_Name_appendString(&name, (char*)"\xFD\x05\x05\xE8\x0C\xCE\x1D"); |
| 270 | ndn_Name_appendBlob(&name, &finalBlockId); |
| 271 | |
| 272 | ostringstream contentStream; |
| 273 | int count = 1; |
| 274 | contentStream << (count++); |
| 275 | while (contentStream.str().length() < 1170) |
| 276 | contentStream << " " << (count++); |
| 277 | contentBlob = Blob((uint8_t*)contentStream.str().c_str(), contentStream.str().length()); |
| 278 | } |
| 279 | else { |
| 280 | // Use a small name and content. |
| 281 | ndn_Name_appendString(&name, (char*)"test"); |
| 282 | contentBlob = Blob((uint8_t*)"abc", 3); |
| 283 | } |
| 284 | ndn_Blob_initialize(&content, (uint8_t*)contentBlob.buf(), contentBlob.size()); |
| 285 | |
| 286 | struct ndn_NameComponent certificateNameComponents[20]; |
| 287 | struct ndn_Name certificateName; |
| 288 | ndn_Name_initialize(&certificateName, certificateNameComponents, sizeof(certificateNameComponents) / sizeof(certificateNameComponents[0])); |
| 289 | ndn_Name_appendString(&certificateName, (char*)"testname"); |
| 290 | ndn_Name_appendString(&certificateName, (char*)"KEY"); |
| 291 | ndn_Name_appendString(&certificateName, (char*)"DSK-123"); |
| 292 | ndn_Name_appendString(&certificateName, (char*)"ID-CERT"); |
| 293 | ndn_Name_appendString(&certificateName, (char*)"0"); |
| 294 | |
| 295 | // Set up publisherPublicKeyDigest and signatureBits in case useCrypto is false. |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 296 | uint8_t* publicKeyDer = DEFAULT_PUBLIC_KEY_DER; |
| 297 | size_t publicKeyDerLength = sizeof(DEFAULT_PUBLIC_KEY_DER); |
| 298 | uint8_t publisherPublicKeyDigestArray[SHA256_DIGEST_LENGTH]; |
| 299 | ndn_digestSha256(publicKeyDer, publicKeyDerLength, publisherPublicKeyDigestArray); |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 300 | struct ndn_Blob publisherPublicKeyDigest; |
| 301 | ndn_Blob_initialize(&publisherPublicKeyDigest, publisherPublicKeyDigestArray, sizeof(publisherPublicKeyDigestArray)); |
| 302 | uint8_t signatureBitsArray[128]; |
| 303 | memset(signatureBitsArray, 0, sizeof(signatureBitsArray)); |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 304 | |
| 305 | // Set up the private key now in case useCrypto is true. |
| 306 | // Use a temporary pointer since d2i updates it. |
| 307 | const uint8_t *privateKeyDerPointer = DEFAULT_PRIVATE_KEY_DER; |
| 308 | RSA *privateKey = d2i_RSAPrivateKey(NULL, &privateKeyDerPointer, sizeof(DEFAULT_PRIVATE_KEY_DER)); |
| 309 | if (!privateKey) { |
| 310 | // Don't expect this to happen. |
| 311 | cout << "Error decoding private key DER" << endl; |
| 312 | return 0; |
| 313 | } |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 314 | |
| 315 | double start = getNowSeconds(); |
| 316 | for (int i = 0; i < nIterations; ++i) { |
| 317 | struct ndn_Data data; |
| 318 | ndn_Data_initialize(&data, name.components, name.maxComponents, certificateName.components, certificateName.maxComponents); |
| 319 | |
| 320 | data.name = name; |
| 321 | data.content = content; |
| 322 | if (useComplex) { |
| 323 | data.metaInfo.timestampMilliseconds = 1.3e+12; |
| 324 | data.metaInfo.freshnessSeconds = 1000; |
| 325 | ndn_NameComponent_initialize(&data.metaInfo.finalBlockID, finalBlockId.value, finalBlockId.length); |
| 326 | } |
| 327 | |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 328 | struct ndn_DynamicUInt8Array output; |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 329 | struct ndn_BinaryXmlEncoder encoder; |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 330 | size_t signedPortionBeginOffset, signedPortionEndOffset; |
| 331 | ndn_Error error; |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 332 | |
| 333 | data.signature.keyLocator.type = ndn_KeyLocatorType_KEYNAME; |
| 334 | data.signature.keyLocator.keyName = certificateName; |
| 335 | data.signature.keyLocator.keyNameType = (ndn_KeyNameType)-1; |
| 336 | data.signature.publisherPublicKeyDigest.publisherPublicKeyDigest = publisherPublicKeyDigest; |
| 337 | if (useCrypto) { |
| 338 | // Encode once to get the signed portion. |
| 339 | ndn_DynamicUInt8Array_initialize(&output, encoding, maxEncodingLength, 0); |
| 340 | ndn_BinaryXmlEncoder_initialize(&encoder, &output); |
| 341 | if ((error = ndn_encodeBinaryXmlData(&data, &signedPortionBeginOffset, &signedPortionEndOffset, &encoder))) { |
| 342 | cout << "Error in ndn_encodeBinaryXmlData: " << ndn_getErrorString(error) << endl; |
| 343 | return 0; |
| 344 | } |
| 345 | |
| 346 | // Imitate MemoryPrivateKeyStorage::sign. |
| 347 | uint8_t digest[SHA256_DIGEST_LENGTH]; |
| 348 | ndn_digestSha256(encoding + signedPortionBeginOffset, signedPortionEndOffset - signedPortionBeginOffset, digest); |
| 349 | unsigned int signatureBitsLength; |
| 350 | if (!RSA_sign(NID_sha256, digest, sizeof(digest), signatureBitsArray, &signatureBitsLength, privateKey)) { |
| 351 | // Don't expect this to happen. |
| 352 | cout << "Error in RSA_sign" << endl; |
| 353 | return 0; |
| 354 | } |
| 355 | |
| 356 | ndn_Blob_initialize(&data.signature.signature, signatureBitsArray, signatureBitsLength); |
| 357 | } |
| 358 | else |
| 359 | // Set up the signature, but don't sign. |
| 360 | ndn_Blob_initialize(&data.signature.signature, signatureBitsArray, sizeof(signatureBitsArray)); |
| 361 | |
| 362 | // Assume the encoding buffer is big enough so we don't need to dynamically reallocate. |
| 363 | ndn_DynamicUInt8Array_initialize(&output, encoding, maxEncodingLength, 0); |
| 364 | ndn_BinaryXmlEncoder_initialize(&encoder, &output); |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 365 | if ((error = ndn_encodeBinaryXmlData(&data, &signedPortionBeginOffset, &signedPortionEndOffset, &encoder))) { |
| 366 | cout << "Error in ndn_encodeBinaryXmlData: " << ndn_getErrorString(error) << endl; |
| 367 | return 0; |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 368 | } |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 369 | *encodingLength = encoder.offset; |
| 370 | } |
| 371 | double finish = getNowSeconds(); |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 372 | |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 373 | if (privateKey) |
| 374 | RSA_free(privateKey); |
| 375 | |
Jeff Thompson | ecaf326 | 2013-12-09 17:13:49 -0800 | [diff] [blame] | 376 | return finish - start; |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 377 | } |
| 378 | |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 379 | /** |
| 380 | * Loop to decode a data packet nIterations times using C. |
| 381 | * @param nIterations The number of iterations. |
| 382 | * @param useCrypto If true, verify the signature. If false, don't verify. |
| 383 | * @param encoding The buffer with wire encoding to decode. |
| 384 | * @param encodingLength The number of bytes in the encoding. |
| 385 | * @return The number of seconds for all iterations. |
| 386 | */ |
Jeff Thompson | ecaf326 | 2013-12-09 17:13:49 -0800 | [diff] [blame] | 387 | static double |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 388 | benchmarkDecodeDataSecondsC(int nIterations, bool useCrypto, uint8_t* encoding, size_t encodingLength) |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 389 | { |
Jeff Thompson | 18dba8b | 2013-12-09 12:17:40 -0800 | [diff] [blame] | 390 | double start = getNowSeconds(); |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 391 | for (int i = 0; i < nIterations; ++i) { |
| 392 | struct ndn_NameComponent nameComponents[100]; |
| 393 | struct ndn_NameComponent keyNameComponents[100]; |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 394 | struct ndn_Data data; |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 395 | ndn_Data_initialize |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 396 | (&data, nameComponents, sizeof(nameComponents) / sizeof(nameComponents[0]), |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 397 | keyNameComponents, sizeof(keyNameComponents) / sizeof(keyNameComponents[0])); |
| 398 | |
| 399 | ndn_BinaryXmlDecoder decoder; |
Jeff Thompson | ecaf326 | 2013-12-09 17:13:49 -0800 | [diff] [blame] | 400 | ndn_BinaryXmlDecoder_initialize(&decoder, encoding, encodingLength); |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 401 | size_t signedPortionBeginOffset, signedPortionEndOffset; |
| 402 | ndn_Error error; |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 403 | if ((error = ndn_decodeBinaryXmlData(&data, &signedPortionBeginOffset, &signedPortionEndOffset, &decoder))) { |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 404 | cout << "Error in ndn_decodeBinaryXmlData: " << ndn_getErrorString(error) << endl; |
Jeff Thompson | ecaf326 | 2013-12-09 17:13:49 -0800 | [diff] [blame] | 405 | return 0; |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 406 | } |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 407 | |
| 408 | if (useCrypto) { |
| 409 | if (!verifyRsaSignature |
| 410 | (encoding + signedPortionBeginOffset, signedPortionEndOffset - signedPortionBeginOffset, |
| 411 | data.signature.signature.value, data.signature.signature.length, |
| 412 | DEFAULT_PUBLIC_KEY_DER, sizeof(DEFAULT_PUBLIC_KEY_DER))) |
| 413 | cout << "Signature verification: FAILED" << endl; |
| 414 | } |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 415 | } |
Jeff Thompson | 18dba8b | 2013-12-09 12:17:40 -0800 | [diff] [blame] | 416 | double finish = getNowSeconds(); |
| 417 | |
Jeff Thompson | ecaf326 | 2013-12-09 17:13:49 -0800 | [diff] [blame] | 418 | return finish - start; |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 419 | } |
| 420 | |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 421 | /** |
Jeff Thompson | 4c66e71 | 2013-12-11 13:37:41 -0800 | [diff] [blame] | 422 | * Call benchmarkEncodeDataSecondsCpp and benchmarkDecodeDataSecondsCpp with appropriate nInterations. Print the |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 423 | * results to cout. |
Jeff Thompson | b33b21e | 2013-12-11 17:37:55 -0800 | [diff] [blame] | 424 | * @param useComplex See benchmarkEncodeDataSecondsCpp. |
Jeff Thompson | 4c66e71 | 2013-12-11 13:37:41 -0800 | [diff] [blame] | 425 | * @param useCrypto See benchmarkEncodeDataSecondsCpp and benchmarkDecodeDataSecondsCpp. |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 426 | */ |
| 427 | static void |
| 428 | benchmarkEncodeDecodeDataCpp(bool useComplex, bool useCrypto) |
| 429 | { |
| 430 | Blob encoding; |
| 431 | { |
Jeff Thompson | bebe32c | 2013-12-11 17:18:57 -0800 | [diff] [blame] | 432 | int nIterations = useCrypto ? 20000 : 2000000; |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 433 | double duration = benchmarkEncodeDataSecondsCpp(nIterations, useComplex, useCrypto, encoding); |
| 434 | cout << "Encode " << (useComplex ? "complex" : "simple ") << " data C++: Crypto? " << (useCrypto ? "yes" : "no ") |
| 435 | << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl; |
| 436 | } |
| 437 | { |
Jeff Thompson | bebe32c | 2013-12-11 17:18:57 -0800 | [diff] [blame] | 438 | int nIterations = useCrypto ? 100000 : 2000000; |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 439 | double duration = benchmarkDecodeDataSecondsCpp(nIterations, useCrypto, encoding); |
| 440 | cout << "Decode " << (useComplex ? "complex" : "simple ") << " data C++: Crypto? " << (useCrypto ? "yes" : "no ") |
| 441 | << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl; |
| 442 | } |
| 443 | } |
| 444 | |
| 445 | /** |
Jeff Thompson | 4c66e71 | 2013-12-11 13:37:41 -0800 | [diff] [blame] | 446 | * Call benchmarkEncodeDataSecondsC and benchmarkDecodeDataSecondsC with appropriate nInterations. Print the |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 447 | * results to cout. |
Jeff Thompson | b33b21e | 2013-12-11 17:37:55 -0800 | [diff] [blame] | 448 | * @param useComplex See benchmarkEncodeDataSecondsC. |
Jeff Thompson | 4c66e71 | 2013-12-11 13:37:41 -0800 | [diff] [blame] | 449 | * @param useCrypto See benchmarkEncodeDataSecondsC and benchmarkDecodeDataSecondsC. |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 450 | */ |
| 451 | static void |
| 452 | benchmarkEncodeDecodeDataC(bool useComplex, bool useCrypto) |
| 453 | { |
| 454 | uint8_t encoding[1500]; |
| 455 | size_t encodingLength; |
| 456 | { |
| 457 | int nIterations = useCrypto ? 20000 : 10000000; |
| 458 | double duration = benchmarkEncodeDataSecondsC(nIterations, useComplex, useCrypto, encoding, sizeof(encoding), &encodingLength); |
| 459 | cout << "Encode " << (useComplex ? "complex" : "simple ") << " data C: Crypto? " << (useCrypto ? "yes" : "no ") |
| 460 | << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl; |
| 461 | } |
| 462 | { |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 463 | int nIterations = useCrypto ? 150000 : 15000000; |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 464 | double duration = benchmarkDecodeDataSecondsC(nIterations, useCrypto, encoding, encodingLength); |
| 465 | cout << "Decode " << (useComplex ? "complex" : "simple ") << " data C: Crypto? " << (useCrypto ? "yes" : "no ") |
| 466 | << ", Duration sec, Hz: " << duration << ", " << (nIterations / duration) << endl; |
| 467 | } |
| 468 | } |
| 469 | |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 470 | int |
| 471 | main(int argc, char** argv) |
| 472 | { |
Jeff Thompson | 18dba8b | 2013-12-09 12:17:40 -0800 | [diff] [blame] | 473 | try { |
Jeff Thompson | 70e8080 | 2013-12-10 20:21:03 -0800 | [diff] [blame] | 474 | benchmarkEncodeDecodeDataCpp(false, false); |
| 475 | benchmarkEncodeDecodeDataCpp(true, false); |
| 476 | benchmarkEncodeDecodeDataCpp(false, true); |
| 477 | benchmarkEncodeDecodeDataCpp(true, true); |
| 478 | |
| 479 | benchmarkEncodeDecodeDataC(false, false); |
| 480 | benchmarkEncodeDecodeDataC(true, false); |
Jeff Thompson | 30d3e6c | 2013-12-11 12:35:54 -0800 | [diff] [blame] | 481 | benchmarkEncodeDecodeDataC(false, true); |
| 482 | benchmarkEncodeDecodeDataC(true, true); |
Jeff Thompson | 18dba8b | 2013-12-09 12:17:40 -0800 | [diff] [blame] | 483 | } catch (std::exception& e) { |
| 484 | cout << "exception: " << e.what() << endl; |
| 485 | } |
Jeff Thompson | 687ea65 | 2013-12-06 11:28:23 -0800 | [diff] [blame] | 486 | return 0; |
| 487 | } |