Source code
Revision control
Copy as Markdown
Other Tools
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "gtest/gtest.h"
#include "mozilla/UniquePtrExtensions.h"
#ifdef XP_UNIX
# include "fcntl.h"
# include "unistd.h"
#endif
#ifdef XP_WIN
# include "namedpipeapi.h"
#endif
using namespace mozilla;
static UniqueFileHandle CreateArbitraryFileHandle() {
#ifdef XP_UNIX
return UniqueFileHandle(dup(0));
#endif
#ifdef XP_WIN
UniqueFileHandle hnd0, hnd1;
if (::CreatePipe(getter_Transfers(hnd0), getter_Transfers(hnd1),
/* lpPipeAttributes = */ nullptr,
/* nSize = */ 0)) {
return hnd0;
}
return nullptr;
#endif
}
// Test duplicating a null UniqueFileHandle
TEST(MFBT_UniquePtrExtensions, UFH_DupNull)
{
UniqueFileHandle fd0;
ASSERT_FALSE(fd0);
UniqueFileHandle fd1 = DuplicateFileHandle(fd0);
EXPECT_FALSE(fd1);
}
// Test that DuplicateFileHandle returns a valid fd/handle which isn't
// the same as the input.
TEST(MFBT_UniquePtrExtensions, UFH_DupBasic)
{
UniqueFileHandle fd0 = CreateArbitraryFileHandle();
ASSERT_TRUE(fd0);
UniqueFileHandle fd1 = DuplicateFileHandle(fd0);
EXPECT_TRUE(fd1);
EXPECT_NE(fd0.get(), fd1.get());
}
#ifdef XP_UNIX
// Test that SetCloseOnExec works
TEST(MFBT_UniquePtrExtensions, UFH_SetCloExec)
{
UniqueFileHandle fd0(dup(0));
ASSERT_TRUE(fd0);
int rv0 = fcntl(fd0.get(), F_GETFD);
ASSERT_GE(rv0, 0);
EXPECT_FALSE(rv0 & FD_CLOEXEC);
SetCloseOnExec(fd0);
ASSERT_TRUE(fd0);
int rv1 = fcntl(fd0.get(), F_GETFD);
ASSERT_GE(rv1, 0);
EXPECT_TRUE(rv1 & FD_CLOEXEC);
}
// Test that DuplicateFileHandle sets close-on-exec
TEST(MFBT_UniquePtrExtensions, UFH_DupCloExec)
{
UniqueFileHandle fd0(dup(0));
ASSERT_TRUE(fd0);
int rv0 = fcntl(fd0.get(), F_GETFD);
ASSERT_GE(rv0, 0);
EXPECT_FALSE(rv0 & FD_CLOEXEC);
UniqueFileHandle fd1 = DuplicateFileHandle(fd0);
ASSERT_TRUE(fd1);
int rv1 = fcntl(fd1.get(), F_GETFD);
ASSERT_GE(rv1, 0);
EXPECT_TRUE(rv1 & FD_CLOEXEC);
}
#endif // XP_UNIX