1 files changed, 114 insertions, 0 deletions
diff --git a/test/mapped-buffer.cpp b/test/mapped-buffer.cpp
new file mode 100644
index 00000000..39d624cd
--- /dev/null
+++ b/test/mapped-buffer.cpp
@@ -0,0 +1,114 @@
+/* SPDX-License-Identifier: GPL-2.0-or-later */
+/*
+ * Copyright (C) 2020, Google Inc.
+ *
+ * libcamera internal MappedBuffer tests
+ */
+
+#include <iostream>
+
+#include "libcamera/internal/buffer.h"
+
+#include "camera_test.h"
+#include "test.h"
+
+using namespace std;
+
+namespace {
+
+class MappedBufferTest : public CameraTest, public Test
+{
+public:
+	MappedBufferTest()
+		: CameraTest("platform/vimc.0 Sensor B")
+	{
+	}
+
+protected:
+	int init() override
+	{
+		if (status_ != TestPass)
+			return status_;
+
+		config_ = camera_->generateConfiguration({ StreamRole::VideoRecording });
+		if (!config_ || config_->size() != 1) {
+			cout << "Failed to generate default configuration" << endl;
+			return TestFail;
+		}
+
+		allocator_ = new FrameBufferAllocator(camera_);
+
+		StreamConfiguration &cfg = config_->at(0);
+
+		if (camera_->acquire()) {
+			cout << "Failed to acquire the camera" << endl;
+			return TestFail;
+		}
+
+		if (camera_->configure(config_.get())) {
+			cout << "Failed to set default configuration" << endl;
+			return TestFail;
+		}
+
+		stream_ = cfg.stream();
+
+		int ret = allocator_->allocate(stream_);
+		if (ret < 0)
+			return TestFail;
+
+		return TestPass;
+	}
+
+	void cleanup() override
+	{
+		delete allocator_;
+	}
+
+	int run() override
+	{
+		const std::unique_ptr<FrameBuffer> &buffer = allocator_->buffers(stream_).front();
+		std::vector<MappedBuffer> maps;
+
+		MappedFrameBuffer map(buffer.get(), PROT_READ);
+		if (!map.isValid()) {
+			cout << "Failed to successfully map buffer" << endl;
+			return TestFail;
+		}
+
+		/* Make sure we can move it. */
+		maps.emplace_back(std::move(map));
+
+		/* But copying is prevented, it would cause double-unmap. */
+		// MappedFrameBuffer map_copy = map;
+
+		/* Local map should be invalid (after move). */
+		if (map.isValid()) {
+			cout << "Post-move map should not be valid" << endl;
+			return TestFail;
+		}
+
+		/* Test for multiple successful maps on the same buffer. */
+		MappedFrameBuffer write_map(buffer.get(), PROT_WRITE);
+		if (!write_map.isValid()) {
+			cout << "Failed to map write buffer" << endl;
+			return TestFail;
+		}
+
+		MappedFrameBuffer rw_map(buffer.get(), PROT_READ | PROT_WRITE);
+		if (!rw_map.isValid()) {
+			cout << "Failed to map RW buffer" << endl;
+			return TestFail;
+		}
+
+		return TestPass;
+	}
+
+private:
+	std::unique_ptr<CameraConfiguration> config_;
+	FrameBufferAllocator *allocator_;
+	Stream *stream_;
+};
+
+} /* namespace */
+
+TEST_REGISTER(MappedBufferTest);