将 2D 矢量转换为 C 类型的最佳方法(适用于 SGX 飞地)



我正在尝试编写一个SGX飞地,它接受int的2D向量(表示灰度图像),但edger8r只接受C类型。

我知道这个和这个,但有人可以提供有关如何转换类型的示例,例如:

vector< vector <int> > ImagePixels;

转换为 C 类型,以便我可以将其传递给 SGX 飞地?

std::vector有一个data()方法,它将为您提供指向其数据的普通指针。但是,您需要对第二层向量执行一些操作,因此您需要创建一个临时数组,该数组将存储来自内部向量的纯指针列表。

#include <algorithm>
#include <iostream>
#include <vector>

extern "C" int my_c_function(int **double_pointer, int rows, int cols) {
for (int i = 0; i < rows; ++i) {
for (int j = 0; j < cols; ++j) {
printf("%d ", double_pointer[i][j]);
}
printf("n");
}
}

int main() {
std::vector<std::vector<int>> ImagePixels{{1, 2, 3}, {4, 5, 6}};
// transform a vector of vectors into a vector of pointers
std::vector<int*> ptrs(ImagePixels.size());
std::transform(begin(ImagePixels), end(ImagePixels), begin(ptrs),
[](std::vector<int> &inner_vec) {
return inner_vec.data();
});
my_c_function(ptrs.data(), ImagePixels.size(), ImagePixels[0].size());
}

现场示例:https://onlinegdb.com/BJvb3yG3f

但是,如果您可以控制整个代码,我建议您使用平面结构,您只需将整个图片逐行存储到一个 faltstd::vector<int>中。你甚至可以在 std::vector 周围做一个简单的包装器,以便更容易地操作这样的结构:

class image {
public:
image(int rows = 0; int cols = 0)
: rows_{rows}, cols_{cols}, data_(rows * cols) {}
int &operator ()(int row, int col) {
return data_[row * cols_ + col];
}
int *data() { return data_.data(); };
// needs a few more utilities and constant versions of the methods
private:
std::vector<int> data_;
int rows_;
int cols_;
};

然后你可以做:

image im(3, 2);
// set pixels
im(0, 0) = 1;
im(0, 1) = 2;
int *d = im.data(); // a plain pointer to an array of data

最新更新