tripwire-open-source/src/fs/fsdatasourceiter.cpp

261 lines
7.9 KiB
C++

//
// The developer of the original code and/or files is Tripwire, Inc.
// Portions created by Tripwire, Inc. are copyright (C) 2000 Tripwire,
// Inc. Tripwire is a registered trademark of Tripwire, Inc. All rights
// reserved.
//
// This program is free software. The contents of this file are subject
// to the terms of the GNU General Public License as published by the
// Free Software Foundation; either version 2 of the License, or (at your
// option) any later version. You may redistribute it and/or modify it
// only in compliance with the GNU General Public License.
//
// This program is distributed in the hope that it will be useful.
// However, this program is distributed AS-IS WITHOUT ANY
// WARRANTY; INCLUDING THE IMPLIED WARRANTY OF MERCHANTABILITY OR FITNESS
// FOR A PARTICULAR PURPOSE. Please see the GNU General Public License
// for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
// USA.
//
// Nothing in the GNU General Public License or any other license to use
// the code or files shall permit you to use Tripwire's trademarks,
// service marks, or other intellectual property without Tripwire's
// prior written consent.
//
// If you have any questions, please contact Tripwire, Inc. at either
// info@tripwire.org or www.tripwire.org.
//
///////////////////////////////////////////////////////////////////////////////
// fsdatasourceiter.cpp
//
//=========================================================================
// INCLUDES
//=========================================================================
#include "stdfs.h"
#include "fsdatasourceiter.h"
#include "fco/fcodatasourceiter.h"
#include "fsobject.h"
#include "core/fsservices.h"
#include "core/errorbucket.h"
#include "core/corestrings.h"
#include "core/usernotify.h"
#include "fco/twfactory.h"
#include "fco/fconametranslator.h"
#include "fco/fconameinfo.h"
#include "fsstrings.h"
//=========================================================================
// METHOD CODE
//=========================================================================
cFSDataSourceIter::cFSDataSourceIter()
{
// set the case sensitiveness of the parent...
//
mDev = 0;
mParentName.SetCaseSensitive( iTWFactory::GetInstance()->GetNameInfo()->IsCaseSensitive() );
}
cFSDataSourceIter::~cFSDataSourceIter()
{
}
cFSDataSourceIter::cFSDataSourceIter( const cFSDataSourceIter& rhs )
{
// set the case sensitiveness of the parent...
//
mParentName.SetCaseSensitive( iTWFactory::GetInstance()->GetNameInfo()->IsCaseSensitive() );
*this = rhs;
}
///////////////////////////////////////////////////////////////////////////////
// operator =
///////////////////////////////////////////////////////////////////////////////
cFSDataSourceIter& cFSDataSourceIter::operator=( const cFSDataSourceIter& rhs )
{
if( this == &rhs )
return *this;
// copy base
cFCODataSourceIterImpl::operator=(rhs);
// copy derived
mDev = rhs.mDev;
return *this;
}
static bool gCrossFileSystems = false;
// Call this to set the property where cFSDataSourceIter does not automatically recurse
// across file system boundaries. Currently this is by default is set to false.
/*static*/ void cFSDataSourceIter::SetFileSystemCrossing(bool crossFS)
{
gCrossFileSystems = crossFS;
}
///////////////////////////////////////////////////////////////////////////////
// CreateCopy
///////////////////////////////////////////////////////////////////////////////
iFCODataSourceIter* cFSDataSourceIter::CreateCopy() const
{
return (new cFSDataSourceIter(*this));
}
///////////////////////////////////////////////////////////////////////////////
// CreateObject -- creates the named object and fills out its properties
// appropriately. Returns NULL if any errors occur and fills up the
// error queue.
//
// This does not create the object if dev > 0 and the object's mDev is not
// equal to it (thus preventing the crossing of file systems)
//
// TODO -- in the future, this should become some kind of lazy evaluation so
// that we don't have to get all the stats() we don't need to.
///////////////////////////////////////////////////////////////////////////////
iFCO * cFSDataSourceIter::CreateObject(const cFCOName& name, bool bCreatePeers )
{
cFSObject* pNewObj = new cFSObject(name);
if( ! bCreatePeers )
{
// when bCreatePeers is false, it means we should set mDev to the
// device number of the current object (ie -- it is a new "start point")
// If we don't do this here, InitializeTypeInfo() will reject creating the
// node.
// -- 9 June 99 mdb
//
mDev = 0;
if( ! InitializeTypeInfo( pNewObj ) )
{
pNewObj->Release();
return 0;
}
mDev = pNewObj->GetFSPropSet().GetDev();
}
return pNewObj;
}
void cFSDataSourceIter::GetChildrenNames( const TSTRING& strParentName, std::vector<TSTRING>& vChildrenNames )
{
try
{
iFSServices::GetInstance()->ReadDir( strParentName, vChildrenNames, false );
}
catch( eFSServices& e )
{
cDebug d("cFSDataSourceIter::GeneratePeers");
d.TraceError("**** ReadDir failed for %s\n", strParentName.c_str() );
if( mpErrorBucket )
{
eFSDataSourceIterReadDir eReadDir(e.GetMsg(), eError::NON_FATAL);
mpErrorBucket->AddError( eReadDir );
}
}
}
///////////////////////////////////////////////////////////////////////////////
// InitializeTypeInfo
///////////////////////////////////////////////////////////////////////////////
bool cFSDataSourceIter::InitializeTypeInfo(iFCO* pFCO)
{
cFSObject* pObj = (cFSObject*)pFCO;
iFCONameTranslator* pTrans = iTWFactory::GetInstance()->GetNameTranslator();
if( pObj->GetFSPropSet().GetValidVector().ContainsItem( cFSPropSet::PROP_FILETYPE) )
return true;
//
// assume invalid by default...
//
cFSPropSet& propSet = pObj->GetFSPropSet();
propSet.SetFileType(cFSPropSet::FT_INVALID);
cFSStatArgs statArgs;
try
{
iFSServices::GetInstance()->Stat( pTrans->ToStringAPI( pObj->GetName() ), statArgs);
}
catch(eFSServices& e)
{
cDebug d("CreateObject");
d.TraceError( "*** Stat of %s failed!!!\n", pObj->GetName().AsString().c_str() );
if( mpErrorBucket )
{
e.SetFatality( false );
mpErrorBucket->AddError( e );
}
return false;
}
//
// don't create the object if it is on a different file system...
//
if( gCrossFileSystems == false && (mDev != 0) && (statArgs.dev != mDev) )
{
TW_NOTIFY_NORMAL( TSS_GetString( cFS, fs::STR_DIFFERENT_FILESYSTEM ).c_str(),
pTrans->ToStringDisplay( pObj->GetName() ).c_str() );
return false;
}
//
// fill out all of the appropriate properties....
//
propSet.SetDev (statArgs.dev);
propSet.SetRDev (statArgs.rdev);
propSet.SetInode (statArgs.ino);
propSet.SetMode (statArgs.mode);
propSet.SetNLink (statArgs.nlink);
propSet.SetUID (statArgs.uid);
propSet.SetGID (statArgs.gid);
propSet.SetSize (statArgs.size);
propSet.SetAccessTime (statArgs.atime);
propSet.SetModifyTime (statArgs.mtime);
propSet.SetCreateTime (statArgs.ctime);
propSet.SetBlockSize (statArgs.blksize);
propSet.SetBlocks (statArgs.blocks);
propSet.SetGrowingFile (statArgs.size);
// set the file type
switch(statArgs.mFileType)
{
case cFSStatArgs::TY_FILE:
propSet.SetFileType(cFSPropSet::FT_FILE);
break;
case cFSStatArgs::TY_DIR:
propSet.SetFileType(cFSPropSet::FT_DIR);
break;
case cFSStatArgs::TY_BLOCKDEV:
propSet.SetFileType(cFSPropSet::FT_BLOCKDEV);
break;
case cFSStatArgs::TY_CHARDEV:
propSet.SetFileType(cFSPropSet::FT_CHARDEV);
break;
case cFSStatArgs::TY_SYMLINK:
propSet.SetFileType(cFSPropSet::FT_SYMLINK);
break;
case cFSStatArgs::TY_FIFO:
propSet.SetFileType(cFSPropSet::FT_FIFO);
break;
case cFSStatArgs::TY_SOCK:
propSet.SetFileType(cFSPropSet::FT_SOCK);
break;
default:
// set it to invalid
propSet.SetFileType(cFSPropSet::FT_INVALID);
}
return true;
}