如何将GMP mpz_t / mpz_class保存到vector <byte>?</byte>

时间:2014-06-03 13:35:04

标签: c++ gmp

我需要将mpz_class类型转换为vector<byte>,反之亦然 我实现了这两个功能,但是往返转换被打破了。

typedef unsigned char byte;

std::vector<byte> mpz_to_vector(std::shared_ptr<mpz_class> x) {
    size_t size;
    byte *a = (byte *) mpz_export(NULL, &size, 1, 1, 1, 0,x->get_mpz_t());
    std::vector<byte> p(a,a+size);
    free(a);
    return p;
}

std::shared_ptr<mpz_class>vector_to_mpz(std::vector<byte> d) {
    mpz_class ptr;
    mpz_import(ptr.get_mpz_t(), d.size(), 1, sizeof(mpz_t), 1, 0, (void *) d.data());
    auto tmp = std::shared_ptr<mpz_class>(new mpz_class(ptr));
    return tmp;
}

1 个答案:

答案 0 :(得分:3)

避免动态分配,尤其是手动分类:为什么size是指向动态分配size_t的指针? 您不需要任何临时缓冲区,因此请将其删除。本手册包含用于调整输出缓冲区大小的正确算法 应用这个,我们得到这个出口商:

std::vector<byte> mpz_to_vector(const mpz_t x) {
    size_t size = (mpz_sizeinbase (x, 2) + CHAR_BIT-1) / CHAR_BIT;
    std::vector<byte> v(size);
    mpz_export(&v[0], &size, 1, 1, 0, 0, x);
    v.resize(size);
    return v;
}
inline std::vector<byte> mpz_to_vector(std::shared_ptr<mpz_class>& x) {
    return mpz_to_vector(x->get_mpz_t());
}

进口商也遭受了不必要的复制 但是,唯一的错误正在为mpz_export 提供错误的参数。
纠正所有人给我们:

std::shared_ptr<mpz_class> vector_to_mpz(const std::vector<byte>& d) {
    auto p = make_shared<mpz_class>();
    mpz_import(p->get_mpz_t(), d.size(), 1, 1, 0, 0, &d[0]);
    return p;
}

顺便说一句:我使用了make_shared,因为这样做比手动更有效,并且可以实现异常安全。