123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287 |
- // Copyright (c) 2012 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #include "butil/posix/file_descriptor_shuffle.h"
- #include <gtest/gtest.h>
- namespace {
- // 'Duplicated' file descriptors start at this number
- const int kDuplicateBase = 1000;
- } // namespace
- namespace butil {
- struct Action {
- enum Type {
- CLOSE,
- MOVE,
- DUPLICATE,
- };
- Action(Type in_type, int in_fd1, int in_fd2 = -1)
- : type(in_type),
- fd1(in_fd1),
- fd2(in_fd2) {
- }
- bool operator==(const Action& other) const {
- return other.type == type &&
- other.fd1 == fd1 &&
- other.fd2 == fd2;
- }
- Type type;
- int fd1;
- int fd2;
- };
- class InjectionTracer : public InjectionDelegate {
- public:
- InjectionTracer()
- : next_duplicate_(kDuplicateBase) {
- }
- virtual bool Duplicate(int* result, int fd) OVERRIDE {
- *result = next_duplicate_++;
- actions_.push_back(Action(Action::DUPLICATE, *result, fd));
- return true;
- }
- virtual bool Move(int src, int dest) OVERRIDE {
- actions_.push_back(Action(Action::MOVE, src, dest));
- return true;
- }
- virtual void Close(int fd) OVERRIDE {
- actions_.push_back(Action(Action::CLOSE, fd));
- }
- const std::vector<Action>& actions() const { return actions_; }
- private:
- int next_duplicate_;
- std::vector<Action> actions_;
- };
- TEST(FileDescriptorShuffleTest, Empty) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- EXPECT_EQ(0u, tracer.actions().size());
- }
- TEST(FileDescriptorShuffleTest, Noop) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 0, false));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- EXPECT_EQ(0u, tracer.actions().size());
- }
- TEST(FileDescriptorShuffleTest, NoopAndClose) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 0, true));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- EXPECT_EQ(0u, tracer.actions().size());
- }
- TEST(FileDescriptorShuffleTest, Simple1) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 1, false));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(1u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] == Action(Action::MOVE, 0, 1));
- }
- TEST(FileDescriptorShuffleTest, Simple2) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 1, false));
- map.push_back(InjectionArc(2, 3, false));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(2u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] == Action(Action::MOVE, 0, 1));
- EXPECT_TRUE(tracer.actions()[1] == Action(Action::MOVE, 2, 3));
- }
- TEST(FileDescriptorShuffleTest, Simple3) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 1, true));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(2u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] == Action(Action::MOVE, 0, 1));
- EXPECT_TRUE(tracer.actions()[1] == Action(Action::CLOSE, 0));
- }
- TEST(FileDescriptorShuffleTest, Simple4) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(10, 0, true));
- map.push_back(InjectionArc(1, 1, true));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(2u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] == Action(Action::MOVE, 10, 0));
- EXPECT_TRUE(tracer.actions()[1] == Action(Action::CLOSE, 10));
- }
- TEST(FileDescriptorShuffleTest, Cycle) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 1, false));
- map.push_back(InjectionArc(1, 0, false));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(4u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] ==
- Action(Action::DUPLICATE, kDuplicateBase, 1));
- EXPECT_TRUE(tracer.actions()[1] == Action(Action::MOVE, 0, 1));
- EXPECT_TRUE(tracer.actions()[2] == Action(Action::MOVE, kDuplicateBase, 0));
- EXPECT_TRUE(tracer.actions()[3] == Action(Action::CLOSE, kDuplicateBase));
- }
- TEST(FileDescriptorShuffleTest, CycleAndClose1) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 1, true));
- map.push_back(InjectionArc(1, 0, false));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(4u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] ==
- Action(Action::DUPLICATE, kDuplicateBase, 1));
- EXPECT_TRUE(tracer.actions()[1] == Action(Action::MOVE, 0, 1));
- EXPECT_TRUE(tracer.actions()[2] == Action(Action::MOVE, kDuplicateBase, 0));
- EXPECT_TRUE(tracer.actions()[3] == Action(Action::CLOSE, kDuplicateBase));
- }
- TEST(FileDescriptorShuffleTest, CycleAndClose2) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 1, false));
- map.push_back(InjectionArc(1, 0, true));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(4u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] ==
- Action(Action::DUPLICATE, kDuplicateBase, 1));
- EXPECT_TRUE(tracer.actions()[1] == Action(Action::MOVE, 0, 1));
- EXPECT_TRUE(tracer.actions()[2] == Action(Action::MOVE, kDuplicateBase, 0));
- EXPECT_TRUE(tracer.actions()[3] == Action(Action::CLOSE, kDuplicateBase));
- }
- TEST(FileDescriptorShuffleTest, CycleAndClose3) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 1, true));
- map.push_back(InjectionArc(1, 0, true));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(4u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] ==
- Action(Action::DUPLICATE, kDuplicateBase, 1));
- EXPECT_TRUE(tracer.actions()[1] == Action(Action::MOVE, 0, 1));
- EXPECT_TRUE(tracer.actions()[2] == Action(Action::MOVE, kDuplicateBase, 0));
- EXPECT_TRUE(tracer.actions()[3] == Action(Action::CLOSE, kDuplicateBase));
- }
- TEST(FileDescriptorShuffleTest, Fanout) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 1, false));
- map.push_back(InjectionArc(0, 2, false));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(2u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] == Action(Action::MOVE, 0, 1));
- EXPECT_TRUE(tracer.actions()[1] == Action(Action::MOVE, 0, 2));
- }
- TEST(FileDescriptorShuffleTest, FanoutAndClose1) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 1, true));
- map.push_back(InjectionArc(0, 2, false));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(3u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] == Action(Action::MOVE, 0, 1));
- EXPECT_TRUE(tracer.actions()[1] == Action(Action::MOVE, 0, 2));
- EXPECT_TRUE(tracer.actions()[2] == Action(Action::CLOSE, 0));
- }
- TEST(FileDescriptorShuffleTest, FanoutAndClose2) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 1, false));
- map.push_back(InjectionArc(0, 2, true));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(3u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] == Action(Action::MOVE, 0, 1));
- EXPECT_TRUE(tracer.actions()[1] == Action(Action::MOVE, 0, 2));
- EXPECT_TRUE(tracer.actions()[2] == Action(Action::CLOSE, 0));
- }
- TEST(FileDescriptorShuffleTest, FanoutAndClose3) {
- InjectiveMultimap map;
- InjectionTracer tracer;
- map.push_back(InjectionArc(0, 1, true));
- map.push_back(InjectionArc(0, 2, true));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &tracer));
- ASSERT_EQ(3u, tracer.actions().size());
- EXPECT_TRUE(tracer.actions()[0] == Action(Action::MOVE, 0, 1));
- EXPECT_TRUE(tracer.actions()[1] == Action(Action::MOVE, 0, 2));
- EXPECT_TRUE(tracer.actions()[2] == Action(Action::CLOSE, 0));
- }
- class FailingDelegate : public InjectionDelegate {
- public:
- virtual bool Duplicate(int* result, int fd) OVERRIDE {
- return false;
- }
- virtual bool Move(int src, int dest) OVERRIDE {
- return false;
- }
- virtual void Close(int fd) OVERRIDE {}
- };
- TEST(FileDescriptorShuffleTest, EmptyWithFailure) {
- InjectiveMultimap map;
- FailingDelegate failing;
- EXPECT_TRUE(PerformInjectiveMultimap(map, &failing));
- }
- TEST(FileDescriptorShuffleTest, NoopWithFailure) {
- InjectiveMultimap map;
- FailingDelegate failing;
- map.push_back(InjectionArc(0, 0, false));
- EXPECT_TRUE(PerformInjectiveMultimap(map, &failing));
- }
- TEST(FileDescriptorShuffleTest, Simple1WithFailure) {
- InjectiveMultimap map;
- FailingDelegate failing;
- map.push_back(InjectionArc(0, 1, false));
- EXPECT_FALSE(PerformInjectiveMultimap(map, &failing));
- }
- } // namespace butil
|