blob: c9617bae80eb52f2908df4f5593900adb96248cc [file] [log] [blame]
Jeff Thompson6c314bc2013-09-23 18:09:38 -07001/* -*- 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
Yingdi Yu61ec2722014-01-20 14:22:32 -08008#include <ndn-cpp-dev/security/sec-tpm-memory.hpp>
9#include <ndn-cpp-dev/security/public-key.hpp>
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080010#include <openssl/ssl.h>
11#include <openssl/sha.h>
12#include <openssl/rsa.h>
Jeff Thompson6c314bc2013-09-23 18:09:38 -070013
14using namespace std;
Jeff Thompson6c314bc2013-09-23 18:09:38 -070015
16namespace ndn {
17
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080018/**
19 * RsaPrivateKey is a simple class to hold an RSA private key.
20 */
Yingdi Yu87581582014-01-14 14:28:39 -080021class SecTpmMemory::RsaPrivateKey {
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080022public:
23 RsaPrivateKey(const uint8_t *keyDer, size_t keyDerLength)
24 {
25 // Use a temporary pointer since d2i updates it.
26 const uint8_t *derPointer = keyDer;
27 privateKey_ = d2i_RSAPrivateKey(NULL, &derPointer, keyDerLength);
28 if (!privateKey_)
29 throw Error("RsaPrivateKey constructor: Error decoding private key DER");
30 }
31
32 ~RsaPrivateKey()
33 {
34 if (privateKey_)
35 RSA_free(privateKey_);
36 }
37
38 rsa_st *
39 getPrivateKey()
40 {
41 return privateKey_;
42 }
43
44private:
45 rsa_st * privateKey_;
46};
47
Yingdi Yu87581582014-01-14 14:28:39 -080048SecTpmMemory::~SecTpmMemory()
Jeff Thompson6c314bc2013-09-23 18:09:38 -070049{
50}
51
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080052void
Yingdi Yu87581582014-01-14 14:28:39 -080053SecTpmMemory::setKeyPairForKeyName(const Name& keyName,
54 uint8_t *publicKeyDer, size_t publicKeyDerLength,
55 uint8_t *privateKeyDer, size_t privateKeyDerLength)
Jeff Thompson6c314bc2013-09-23 18:09:38 -070056{
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080057 publicKeyStore_[keyName.toUri()] = ptr_lib::make_shared<PublicKey>(publicKeyDer, publicKeyDerLength);
Jeff Thompsonce115762013-12-18 14:59:56 -080058 privateKeyStore_[keyName.toUri()] = ptr_lib::make_shared<RsaPrivateKey>(privateKeyDer, privateKeyDerLength);
Jeff Thompson6c314bc2013-09-23 18:09:38 -070059}
60
61void
Yingdi Yu87581582014-01-14 14:28:39 -080062SecTpmMemory::generateKeyPairInTpm(const Name& keyName, KeyType keyType, int keySize)
Jeff Thompson6c314bc2013-09-23 18:09:38 -070063{
64#if 1
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080065 throw Error("MemoryPrivateKeyStorage::generateKeyPair not implemented");
Jeff Thompson6c314bc2013-09-23 18:09:38 -070066#endif
67}
68
Jeff Thompsonce115762013-12-18 14:59:56 -080069ptr_lib::shared_ptr<PublicKey>
Yingdi Yu87581582014-01-14 14:28:39 -080070SecTpmMemory::getPublicKeyFromTpm(const Name& keyName)
Jeff Thompson6c314bc2013-09-23 18:09:38 -070071{
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080072 PublicKeyStore::iterator publicKey = publicKeyStore_.find(keyName.toUri());
Jeff Thompson6c314bc2013-09-23 18:09:38 -070073 if (publicKey == publicKeyStore_.end())
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080074 throw Error(string("MemoryPrivateKeyStorage: Cannot find public key ") + keyName.toUri());
Jeff Thompson6c314bc2013-09-23 18:09:38 -070075 return publicKey->second;
76}
77
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080078Block
Yingdi Yub4bb85a2014-01-16 10:11:04 -080079SecTpmMemory::signInTpm(const uint8_t *data, size_t dataLength,
Yingdi Yu87581582014-01-14 14:28:39 -080080 const Name& keyName,
81 DigestAlgorithm digestAlgorithm)
Jeff Thompson6c314bc2013-09-23 18:09:38 -070082{
83 if (digestAlgorithm != DIGEST_ALGORITHM_SHA256)
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080084 return ConstBufferPtr();
Jeff Thompson6c314bc2013-09-23 18:09:38 -070085
Jeff Thompson6c314bc2013-09-23 18:09:38 -070086 // Find the private key and sign.
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080087 PrivateKeyStore::iterator privateKey = privateKeyStore_.find(keyName.toUri());
Jeff Thompson6c314bc2013-09-23 18:09:38 -070088 if (privateKey == privateKeyStore_.end())
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080089 throw Error(string("MemoryPrivateKeyStorage: Cannot find private key ") + keyName.toUri());
Jeff Thompson6c314bc2013-09-23 18:09:38 -070090
Alexander Afanasyev04b22a92014-01-05 22:40:17 -080091 uint8_t digest[SHA256_DIGEST_LENGTH];
92 SHA256_CTX sha256;
93 SHA256_Init(&sha256);
94 SHA256_Update(&sha256, data, dataLength);
95 SHA256_Final(digest, &sha256);
96
97 BufferPtr signatureBuffer = ptr_lib::make_shared<Buffer>();
98 signatureBuffer->resize(RSA_size(privateKey->second->getPrivateKey()));
99
100 unsigned int signatureBitsLength;
101 if (!RSA_sign(NID_sha256, digest, sizeof(digest),
102 signatureBuffer->buf(),
103 &signatureBitsLength,
104 privateKey->second->getPrivateKey()))
105 {
106 throw Error("Error in RSA_sign");
107 }
108
109 return Block(Tlv::SignatureValue, signatureBuffer);
Jeff Thompson6c314bc2013-09-23 18:09:38 -0700110}
111
Alexander Afanasyev04b22a92014-01-05 22:40:17 -0800112ConstBufferPtr
Yingdi Yub4bb85a2014-01-16 10:11:04 -0800113SecTpmMemory::decryptInTpm(const Name& keyName, const uint8_t* data, size_t dataLength, bool isSymmetric)
Jeff Thompson6c314bc2013-09-23 18:09:38 -0700114{
115#if 1
Alexander Afanasyev04b22a92014-01-05 22:40:17 -0800116 throw Error("MemoryPrivateKeyStorage::decrypt not implemented");
Jeff Thompson6c314bc2013-09-23 18:09:38 -0700117#endif
118}
119
Alexander Afanasyev04b22a92014-01-05 22:40:17 -0800120ConstBufferPtr
Yingdi Yub4bb85a2014-01-16 10:11:04 -0800121SecTpmMemory::encryptInTpm(const Name& keyName, const uint8_t* data, size_t dataLength, bool isSymmetric)
Jeff Thompson6c314bc2013-09-23 18:09:38 -0700122{
123#if 1
Alexander Afanasyev04b22a92014-01-05 22:40:17 -0800124 throw Error("MemoryPrivateKeyStorage::encrypt not implemented");
Jeff Thompson6c314bc2013-09-23 18:09:38 -0700125#endif
126}
127
128void
Yingdi Yub4bb85a2014-01-16 10:11:04 -0800129SecTpmMemory::generateSymmetricKeyInTpm(const Name& keyName, KeyType keyType, int keySize)
Jeff Thompson6c314bc2013-09-23 18:09:38 -0700130{
131#if 1
Alexander Afanasyev04b22a92014-01-05 22:40:17 -0800132 throw Error("MemoryPrivateKeyStorage::generateKey not implemented");
Jeff Thompson6c314bc2013-09-23 18:09:38 -0700133#endif
134}
135
136bool
Yingdi Yub4bb85a2014-01-16 10:11:04 -0800137SecTpmMemory::doesKeyExistInTpm(const Name& keyName, KeyClass keyClass)
Jeff Thompson6c314bc2013-09-23 18:09:38 -0700138{
139 if (keyClass == KEY_CLASS_PUBLIC)
140 return publicKeyStore_.find(keyName.toUri()) != publicKeyStore_.end();
141 else if (keyClass == KEY_CLASS_PRIVATE)
142 return privateKeyStore_.find(keyName.toUri()) != privateKeyStore_.end();
143 else
144 // KEY_CLASS_SYMMETRIC not implemented yet.
145 return false;
146}
147
Jeff Thompson6c314bc2013-09-23 18:09:38 -0700148}