Files
intel-linux-sgx/external/CppMicroServices/framework/test/gtest/OpenFileHandleTest.cpp
T
Li, Xun d10cabebb5 Linux 2.5 Open Source Gold Release
Added new APIs in sgx_uae_service.h and sgx_ukey_exchange.h to support ECDSA quote based remote attestation.
  The set of legacy APIs supports EPID only and the set of new APIs supports ECDSA quotes.
Enhanced Edger8r with structure deep-copy feature.
Fixed bugs.

Signed-off-by: Li, Xun <xun.li@intel.com>
2019-04-02 13:26:58 +08:00

101 lines
3.2 KiB
C++

/*=============================================================================
Library: CppMicroServices
Copyright (c) The CppMicroServices developers. See the COPYRIGHT
file at the top-level directory of this distribution and at
https://github.com/CppMicroServices/CppMicroServices/COPYRIGHT .
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
=============================================================================*/
#include <chrono>
#include "TestUtilBundleListener.h"
#include "TestUtils.h"
#include "cppmicroservices/Bundle.h"
#include "cppmicroservices/BundleContext.h"
#include "cppmicroservices/BundleEvent.h"
#include "cppmicroservices/Constants.h"
#include "cppmicroservices/Framework.h"
#include "cppmicroservices/FrameworkEvent.h"
#include "cppmicroservices/FrameworkFactory.h"
#include "cppmicroservices/util/FileSystem.h"
#include "gtest/gtest.h"
#if defined(US_PLATFORM_WINDOWS)
#include "windows.h"
#elif defined(US_PLATFORM_POSIX)
#include <stdio.h>
#include <sys/types.h>
#include <unistd.h>
#endif
using namespace cppmicroservices;
static unsigned long GetHandleCountForCurrentProcess()
{
#if defined(US_PLATFORM_WINDOWS)
auto processHandle = GetCurrentProcess();
unsigned long handleCount{ 0 };
if (!GetProcessHandleCount(processHandle, &handleCount)) {
throw std::runtime_error("GetProcessHandleCount failed to retrieve the number of open handles.");
}
return handleCount;
#elif defined(US_PLATFORM_POSIX)
auto pid_t = getpid();
std::string command("lsof -p " + std::to_string(pid_t) + " | wc -l");
FILE* fd = popen(command.c_str(), "r");
if (nullptr == fd) {
throw std::runtime_error("popen failed.");
}
std::string result;
char buf[PATH_MAX];
while (nullptr != fgets(buf, PATH_MAX, fd)) {
result += buf;
}
if (-1 == pclose(fd)) {
throw std::runtime_error("pclose failed.");
}
return stoul(result);
#else
throw std::runtime_error("unsupported platform - can't get handle count for current process.")
#endif
}
// Test that the file handle count doesn't change after installing a bundle.
// Installing a bundle should not hold the file open.
TEST(OpenFileHandleTest, InstallBundle)
{
auto f = FrameworkFactory().NewFramework();
ASSERT_TRUE(f);
f.Start();
auto handleCountBefore = GetHandleCountForCurrentProcess();
#if defined(US_BUILD_SHARED_LIBS)
auto bundle =
cppmicroservices::testing::InstallLib(f.GetBundleContext(), "TestBundleA");
#else
auto bundle =
cppmicroservices::testing::GetBundle("TestBundleA", f.GetBundleContext());
#endif
auto handleCountAfter = GetHandleCountForCurrentProcess();
ASSERT_EQ(handleCountBefore, handleCountAfter) << "The handle counts before and after installing a bundle should not differ.";
f.Stop();
f.WaitForStop(std::chrono::seconds::zero());
}