Encapsulate file descriptor into an object, to ensure that an fd will be closed properly while exception raised, e.g., the block_cache throws exception during the block_manager's construction.
		
			
				
	
	
		
			71 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			71 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
// Copyright (C) 2017 Red Hat, Inc. All rights reserved.
 | 
						|
//
 | 
						|
// This file is part of the thin-provisioning-tools source.
 | 
						|
//
 | 
						|
// thin-provisioning-tools is free software: you can redistribute it
 | 
						|
// and/or modify it under the terms of the GNU General Public License
 | 
						|
// as published by the Free Software Foundation, either version 3 of
 | 
						|
// the License, or (at your option) any later version.
 | 
						|
//
 | 
						|
// thin-provisioning-tools is distributed in the hope that it will be
 | 
						|
// useful, but WITHOUT ANY WARRANTY; without even the implied warranty
 | 
						|
// of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
						|
// GNU General Public License for more details.
 | 
						|
//
 | 
						|
// You should have received a copy of the GNU General Public License along
 | 
						|
// with thin-provisioning-tools.  If not, see
 | 
						|
// <http://www.gnu.org/licenses/>.
 | 
						|
 | 
						|
#include "gmock/gmock.h"
 | 
						|
#include "block-cache/block_cache.h"
 | 
						|
#include "test_utils.h"
 | 
						|
 | 
						|
#include <stdlib.h>
 | 
						|
#include <sys/types.h>
 | 
						|
#include <sys/stat.h>
 | 
						|
#include <fcntl.h>
 | 
						|
 | 
						|
using namespace std;
 | 
						|
using namespace test;
 | 
						|
using namespace testing;
 | 
						|
 | 
						|
//----------------------------------------------------------------
 | 
						|
 | 
						|
TEST(BCacheTests, cleaned_on_demand)
 | 
						|
{
 | 
						|
	using namespace bcache;
 | 
						|
 | 
						|
	unsigned const NR_BLOCKS = 16;
 | 
						|
 | 
						|
	temp_file tmp("bcache_t", 1);
 | 
						|
	file_utils::file_descriptor fd(tmp.get_path().c_str(), O_RDWR | O_DIRECT);
 | 
						|
 | 
						|
	uint64_t bs = 8;
 | 
						|
	block_cache bc(fd, bs, 64, (bs << SECTOR_SHIFT) * NR_BLOCKS);
 | 
						|
	validator::ptr v(new bcache::noop_validator());
 | 
						|
	for (unsigned i = 0; i < NR_BLOCKS; i++) {
 | 
						|
		block_cache::block &b = bc.get(i, block_cache::GF_DIRTY | block_cache::GF_ZERO, v);
 | 
						|
		b.put();
 | 
						|
	}
 | 
						|
 | 
						|
	// Default WRITEBACK_LOW_THRESHOLD_PERCENT == 33
 | 
						|
	// For NR_BLOCKS cache entires, only 6 dirty blocks will be submitted,
 | 
						|
	// so the 7th get() doesn't work.
 | 
						|
	list<block_cache::block *> blocks;
 | 
						|
 | 
						|
	try {
 | 
						|
		for (unsigned i = 0; i < NR_BLOCKS; i++)
 | 
						|
			blocks.push_back(&bc.get(NR_BLOCKS + i, 0, v));
 | 
						|
 | 
						|
	} catch (...) {
 | 
						|
		for (auto b : blocks)
 | 
						|
			b->put();
 | 
						|
 | 
						|
		throw;
 | 
						|
	}
 | 
						|
 | 
						|
	for (auto b : blocks)
 | 
						|
		b->put();
 | 
						|
}
 | 
						|
 |