diff options
| -rw-r--r-- | mpm/include/mpm/types/mmap_regs_iface.hpp | 58 | ||||
| -rw-r--r-- | mpm/include/mpm/types/types_python.hpp | 8 | ||||
| -rw-r--r-- | mpm/lib/types/CMakeLists.txt | 1 | ||||
| -rw-r--r-- | mpm/lib/types/mmap_regs_iface.cpp | 117 |
4 files changed, 184 insertions, 0 deletions
diff --git a/mpm/include/mpm/types/mmap_regs_iface.hpp b/mpm/include/mpm/types/mmap_regs_iface.hpp new file mode 100644 index 000000000..969eb8160 --- /dev/null +++ b/mpm/include/mpm/types/mmap_regs_iface.hpp @@ -0,0 +1,58 @@ +// +// Copyright 2017 Ettus Research, National Instruments Company +// +// SPDX-License-Identifier: GPL-3.0 +// + +#pragma once + +#include <mpm/types/log_buf.hpp> +#include <boost/noncopyable.hpp> +#include <string> +#include <cstdint> + +namespace mpm { namespace types { + + class mmap_regs_iface : public boost::noncopyable + { + public: + mmap_regs_iface( + const std::string &path, + const size_t length, + const size_t offset, + const bool read_only = true, + const bool open_now = true + ); + + //! Will call close() + ~mmap_regs_iface(); + + //! Open the file descriptor and mmap. Safe to call multiple times. + void open(); + + //! Close the file descriptor and mmap. Safe to call multiple times. + void close(); + + //! Write \p data to \p addr + void poke32(const uint32_t addr, const uint32_t data); + + //! Read data from \p addr + uint32_t peek32(const uint32_t addr); + + private: + void log( + mpm::types::log_level_t level, + const std::string path, + const char *comment + ); + + const std::string _path; + const size_t _length; + const size_t _offset; + const bool _read_only; + int _fd = -1; + uint32_t *_mmap = NULL; + + }; + +}} /* namespace mpm::types */ diff --git a/mpm/include/mpm/types/types_python.hpp b/mpm/include/mpm/types/types_python.hpp index 8cb5b6f6a..de777685c 100644 --- a/mpm/include/mpm/types/types_python.hpp +++ b/mpm/include/mpm/types/types_python.hpp @@ -20,6 +20,7 @@ #include "lockable.hpp" #include "regs_iface.hpp" #include "log_buf.hpp" +#include "mmap_regs_iface.hpp" void export_types() { LIBMPM_BOOST_PREAMBLE("types") @@ -52,5 +53,12 @@ void export_types() { ); }) ; + + bp::class_<mmap_regs_iface, boost::noncopyable, std::shared_ptr<mmap_regs_iface>>("mmap_regs_iface", bp::init<std::string, size_t, size_t, bool, bool>()) + .def("open", &mmap_regs_iface::open) + .def("close", &mmap_regs_iface::close) + .def("peek32", &mmap_regs_iface::peek32) + .def("poke32", &mmap_regs_iface::poke32) + ; } diff --git a/mpm/lib/types/CMakeLists.txt b/mpm/lib/types/CMakeLists.txt index 10ce8a28c..029dddc85 100644 --- a/mpm/lib/types/CMakeLists.txt +++ b/mpm/lib/types/CMakeLists.txt @@ -1,6 +1,7 @@ SET(TYPES_SOURCES ${CMAKE_CURRENT_SOURCE_DIR}/lockable.cpp ${CMAKE_CURRENT_SOURCE_DIR}/log_buf.cpp + ${CMAKE_CURRENT_SOURCE_DIR}/mmap_regs_iface.cpp ) USRP_PERIPHS_ADD_OBJECT(types ${TYPES_SOURCES}) diff --git a/mpm/lib/types/mmap_regs_iface.cpp b/mpm/lib/types/mmap_regs_iface.cpp new file mode 100644 index 000000000..3af47a61a --- /dev/null +++ b/mpm/lib/types/mmap_regs_iface.cpp @@ -0,0 +1,117 @@ +// +// Copyright 2017 Ettus Research, National Instruments Company +// +// SPDX-License-Identifier: GPL-3.0 +// + +#include <mpm/types/mmap_regs_iface.hpp> +#include <mpm/exception.hpp> +#include <boost/format.hpp> +#include <iostream> +#include <sys/types.h> +#include <sys/stat.h> +#include <sys/mman.h> +#include <unistd.h> +#include <fcntl.h> +#include <sstream> + +using namespace mpm::types; + + +mmap_regs_iface::mmap_regs_iface( + const std::string &path, + const size_t length, + const size_t offset, + const bool read_only, + const bool open_now +) : _path(path) + , _length(length) + , _offset(offset) + , _read_only(read_only) +{ + if (open_now) { + open(); + } +} + +mmap_regs_iface::~mmap_regs_iface() +{ + try { + close(); + } catch (...) { + // This is not good + } +} + +void mmap_regs_iface::open() +{ + if (_fd < 0) { + _fd = ::open(_path.c_str(), _read_only ? O_RDONLY : O_RDWR); + if (_fd < 0) { + throw mpm::runtime_error("Could not open file descriptor!"); + } + } + + if (!_mmap) { + _mmap = (uint32_t *) ::mmap( + NULL, + _length, + PROT_READ | (_read_only ? 0 : PROT_WRITE), + MAP_SHARED, + _fd, + (off_t) _offset + ); + if (((void *) _mmap) == MAP_FAILED) { + throw mpm::runtime_error("Failed to mmap!"); + } + } + log(mpm::types::log_level_t::TRACE, _path, + "Opened mmap_regs_iface"); +} + +void mmap_regs_iface::close() +{ + if (_mmap) { + int err = munmap((void *) _mmap, _length); + if (err) { + throw mpm::runtime_error("Couldn't munmap!"); + } + _mmap = NULL; + } + + if (_fd >= 0) { + if (::close(_fd)) { + throw mpm::runtime_error("Couldn't close file descriptor!"); + } + _fd = -1; + } + log(mpm::types::log_level_t::TRACE, _path, + "Closed mmap_regs_iface"); +} + +void mmap_regs_iface::poke32( + const uint32_t addr, + const uint32_t data +) { + MPM_ASSERT_THROW(_mmap); + _mmap[addr / sizeof(uint32_t)] = data; +} + +uint32_t mmap_regs_iface::peek32(const uint32_t addr) +{ + MPM_ASSERT_THROW(_mmap); + return _mmap[addr / sizeof(uint32_t)]; +} + +void mmap_regs_iface::log( + mpm::types::log_level_t level, + const std::string path, + const char *comment +) { + mpm::types::log_buf::make_singleton()->post( + level, + "MMAP_REGS_IFACE", + str(boost::format("[UIO %s] %s") + % path % comment) + ); +} |