2 files changed, 115 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);
diff --git a/test/meson.build b/test/meson.build
index 77518715..0a1d434e 100644
--- a/test/meson.build
+++ b/test/meson.build
@@ -31,6 +31,7 @@ internal_tests = [
     ['file',                            'file.cpp'],
     ['file-descriptor',                 'file-descriptor.cpp'],
     ['hotplug-cameras',                 'hotplug-cameras.cpp'],
+    ['mapped-buffer',                   'mapped-buffer.cpp'],
     ['message',                         'message.cpp'],
     ['object',                          'object.cpp'],
     ['object-delete',                   'object-delete.cpp'],