给定一个填充unsigned char **的C函数，如何使用没有中间副本的数据填充std :: vector

Question

C函数是：

int i2d_PrivateKey(EVP_PKEY *a, unsigned char **pp);

int i2d_X509(X509 *a, unsigned char **ppout);

我已经编写了这样的代码来复制到std :: vector中：

// populate PrivateKey
std::vector<uint8_t> PrivateKey;
EVP_PKEY *privatekey = NULL;
int size = i2d_PrivateKey(privatekey, NULL);
if (size > 0)
{
    PrivateKey.reserve(size);
    uint8_t* ptr = &PrivateKey[0];
    i2d_PrivateKey(privatekey, &ptr);
std::cout << "PrivateKey size=" << PrivateKey.size() << '\n';
}

PrivateKey.size（）返回零，所以我知道矢量尚未填充。但是，我知道size是一个正整数，因此if（size块）中的代码被执行。

如果ptr是PrivateKey数组的起始地址，那么这段代码不应该工作吗？

虽然这段代码使用openssl，但我认为它更像是一般指针问题。如果我创建一个临时的uint8_t数组然后它可以工作，但我宁愿直接复制到向量中并节省临时副本的开销。

这是代码：

#include <openssl/ssl.h>
#include <openssl/pem.h>
#include <openssl/ossl_typ.h>
#include <openssl/x509.h>
#include <openssl/x509v3.h>
#include <openssl/pkcs12.h>
#include <openssl/err.h>

#include <iostream>
#include <vector>
#include <string>
#include <cstdint>


int main()
{
    std::vector<uint8_t> input;  // contains pkcs12 data
    std::string Password = "password";
    std::vector<uint8_t> Certificate;
    std::vector<uint8_t> PrivateKey;
    OpenSSL_add_all_algorithms();
    OpenSSL_add_all_ciphers();
    OpenSSL_add_all_digests();
    ERR_load_crypto_strings();

    PKCS12* p12_cert = NULL;
    const uint8_t* p1 = &input[0];
    if(d2i_PKCS12(&p12_cert, &p1, input.size()) != NULL) {
        EVP_PKEY *privatekey = NULL;
        X509 *x509_cert = NULL;
        // additional certs, last arg is CA which we don't care about
        if (PKCS12_parse(p12_cert, Password.c_str(), &privatekey, &x509_cert, NULL))
        {
            // populate m_privateKey
            int size = i2d_PrivateKey(privatekey, NULL);
    std::cout << "privatekey size=" << size << '\n';
            if (size > 0)
            {
                PrivateKey.reserve(size);
                uint8_t* ptr = &PrivateKey[0];
                i2d_PrivateKey(privatekey, &ptr);
        std::cout << "PrivateKey size=" << PrivateKey.size() << '\n';
            }
            // populate certificate
            size = i2d_X509(x509_cert, NULL);
            std::cout << "certificate size=" << size << '\n';
            if(size > 0)
            {
                Certificate.reserve(size);
                uint8_t* ptr = &Certificate[0];
                int ret = i2d_X509(x509_cert, &ptr);
        std::cout << "ret=" << ret <<'\n';
        std::cout << "cert size=" << Certificate.size() << '\n';
            }
        }
        PKCS12_free(p12_cert);
    }
}

更新，可以使用下面的代码来合并Arron的修复：

#include <openssl/ssl.h>
#include <openssl/pem.h>
#include <openssl/ossl_typ.h>
#include <openssl/x509.h>
#include <openssl/x509v3.h>
#include <openssl/pkcs12.h>
#include <openssl/err.h>

#include <iostream>
#include <fstream>
#include <iterator>
#include <iomanip>
#include <vector>
#include <string>
#include <cstdint>

using namespace std;

std::vector<uint8_t>& File2Buffer(const std::string path,
                            std::vector<uint8_t>& buffer) {
    fstream fs(path, ios::in | ios::binary);
    if (fs) {
        // Don't skip new lines
        fs.unsetf(ios::skipws);

        fs.seekg(0, ios::end);
        size_t size = static_cast<size_t>(fs.tellg());
        fs.seekg(0, ios::beg);

        buffer.reserve(size);
        buffer.insert(buffer.begin(),
            istream_iterator<uint8_t>(fs),
            istream_iterator<uint8_t>());
    }
    return buffer;
}


int main(int argc, char* argv[])
{
    if (argc != 3) {
        cout << "Usage: " << argv[0] << " <pkcs12 file> " << "<password>\n";
        exit(0);
    }

    std::vector<uint8_t> input;
    File2Buffer(argv[1], input);
    std::string Password = argv[2];
    std::vector<uint8_t> Certificate;
    std::vector<uint8_t> PrivateKey;
    OpenSSL_add_all_algorithms();
    OpenSSL_add_all_ciphers();
    OpenSSL_add_all_digests();
    ERR_load_crypto_strings();

    PKCS12* p12_cert = NULL;
    const uint8_t* p1 = &input[0];
    if(d2i_PKCS12(&p12_cert, &p1, input.size()) != NULL) {
        EVP_PKEY *privatekey = NULL;
        X509 *x509_cert = NULL;
        // additional certs, last arg is CA which we don't care about
        if (PKCS12_parse(p12_cert, Password.c_str(), &privatekey, &x509_cert, NULL))
        {
            // populate m_privateKey
            int size = i2d_PrivateKey(privatekey, NULL);
    std::cout << "privatekey size=" << size << '\n';
            if (size > 0)
            {
                PrivateKey.resize(size);
                uint8_t* ptr = &PrivateKey[0];
                i2d_PrivateKey(privatekey, &ptr);
        std::cout << "PrivateKey size=" << PrivateKey.size() << '\n';
            }
            // populate certificate
            size = i2d_X509(x509_cert, NULL);
            std::cout << "certificate size=" << size << '\n';
            if(size > 0)
            {
                Certificate.resize(size);
                uint8_t* ptr = &Certificate[0];
                int ret = i2d_X509(x509_cert, &ptr);
        std::cout << "ret=" << ret <<'\n';
        std::cout << "cert size=" << Certificate.size() << '\n';
            }
        }
        PKCS12_free(p12_cert);
    }

    // test it out:
    if (Certificate.size() > 0) {
        cout << "Certificate size=" << Certificate.size() << '\n';
        for (auto& ch : Certificate) {
            cout << hex << ch << " ";
        }
    }
}

Answer 1

使用resize而不是reserve。保留的问题是，如果你执行一个赋值（如PrivateKey [5] = 5），并调用PrivateKey.size（），大小仍将保持为0.（实际上，保留可以与std ::中的back_inserter配对使用复制），但在你的情况下，你应该做一个调整大小。