mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 16:46:47 +00:00
93 lines
3.0 KiB
C++
93 lines
3.0 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
|
* *
|
|
* This software is distributed under the terms of the *
|
|
* GNU Lesser General Public Licence (LGPL) version 3, *
|
|
* copied verbatim in the file "LICENSE" *
|
|
********************************************************************************/
|
|
|
|
#include <fairmq/shmem/Common.h>
|
|
#include <fairmq/shmem/UnmanagedRegion.h>
|
|
#include <fairmq/shmem/Segment.h>
|
|
|
|
#include <fairmq/ProgOptions.h>
|
|
#include <fairmq/tools/Unique.h>
|
|
#include <fairmq/TransportFactory.h>
|
|
|
|
#include <fairlogger/Logger.h>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <unistd.h>
|
|
|
|
#include <string>
|
|
|
|
namespace
|
|
{
|
|
|
|
using namespace std;
|
|
|
|
struct ShmRemover
|
|
{
|
|
ShmRemover(std::string _shmId) : shmId(std::move(_shmId)) {}
|
|
~ShmRemover() { fair::mq::shmem::Monitor::Cleanup(fair::mq::shmem::ShmId{shmId}); }
|
|
|
|
std::string shmId;
|
|
};
|
|
|
|
void Preallocate()
|
|
{
|
|
const string session = to_string(fair::mq::tools::UuidHash());
|
|
// generate shmId out of session id + user id (geteuid).
|
|
const string shmId = fair::mq::shmem::makeShmIdStr(session);
|
|
|
|
// const uint64_t shmId64 = 0;
|
|
// const string shmId = fair::mq::shmem::makeShmIdStr(shmId64);
|
|
// LOG(info) << shmId;
|
|
|
|
const uint16_t s1id = 0;
|
|
const uint64_t s1size = 10000000;
|
|
const uint16_t s2id = 1;
|
|
const uint64_t s2size = 20000000;
|
|
|
|
const uint16_t r1id = 0;
|
|
const uint64_t r1size = 10000000;
|
|
const uint16_t r2id = 1;
|
|
const uint64_t r2size = 20000000;
|
|
|
|
// cleanup when done
|
|
ShmRemover shmRemover(shmId);
|
|
|
|
// managed segments
|
|
fair::mq::shmem::Segment segment1(shmId, s1id, s1size, fair::mq::shmem::rbTreeBestFit);
|
|
segment1.Lock();
|
|
segment1.Zero();
|
|
LOG(info) << "Created segment " << s1id << " of size " << segment1.GetSize() << " starting at " << segment1.GetData();
|
|
|
|
fair::mq::shmem::Segment segment2(shmId, s2id, s2size, fair::mq::shmem::rbTreeBestFit);
|
|
segment2.Lock();
|
|
segment2.Zero();
|
|
LOG(info) << "Created segment " << s2id << " of size " << segment2.GetSize() << " starting at " << segment2.GetData();
|
|
|
|
// unmanaged regions
|
|
fair::mq::shmem::UnmanagedRegion region1(shmId, r1id, r1size);
|
|
region1.Lock();
|
|
region1.Zero();
|
|
LOG(info) << "Created region " << r1id << " of size " << region1.GetSize() << " starting at " << region1.GetData();
|
|
|
|
fair::mq::shmem::UnmanagedRegion region2(shmId, r2id, r2size);
|
|
region2.Lock();
|
|
region2.Zero();
|
|
LOG(info) << "Created region " << r2id << " of size " << region2.GetSize() << " starting at " << region2.GetData();
|
|
|
|
// for a "soft reset" call (shmem should not be in active use by (no messages in flight) devices during this call):
|
|
// fair::mq::shmem::Monitor::ResetContent(fair::mq::shmem::ShmId{shmId});
|
|
}
|
|
|
|
TEST(PreallocateInsideSession, shmem)
|
|
{
|
|
Preallocate();
|
|
}
|
|
|
|
}
|