bool operator<( const QTreeWidgetItem & other ) const
{
int col = this->treeWidget()->sortColumn();
if ( col != 1)
return this->text(col) < other.text(col);
FWObject *otherobj = db->findInIndex(other.data(0, Qt::UserRole).toInt());
FWObject *thisobj = db->findInIndex(this->data(0, Qt::UserRole).toInt());
if (otherobj->getTypeName() != thisobj->getTypeName())
return thisobj->getTypeName() < otherobj->getTypeName();
if (IPv4::isA(thisobj) || IPv6::isA(thisobj))
{
return compare_addrs(Address::cast(thisobj)->getAddressPtr(), Address::cast(otherobj)->getAddressPtr());
}
if (Service::isA(thisobj))
{
return Service::cast(thisobj)->getProtocolNumber() < Service::cast(otherobj)->getProtocolNumber();
}
if(AddressRange::isA(thisobj))
{
return compare_addrs(&AddressRange::cast(thisobj)->getRangeStart(),
&AddressRange::cast(otherobj)->getRangeStart());
}
if (Host::isA(thisobj))
{
return compare_addrs(Host::cast(thisobj)->getAddressPtr(),
Host::cast(otherobj)->getAddressPtr());
}
return this->text(col) < other.text(col);
}
void PolicyRule::setDummySource()
{
FWObjectDatabase *root = getRoot();
FWObject *dummySource = root->findInIndex(FWObjectDatabase::DUMMY_ADDRESS_ID);
if (!dummySource || (root->getStringId(dummySource->getId()) != "dummyaddressid0"))
return;
FWObject::iterator i1 = begin();
(*i1)->addRef(dummySource);
src_re = RuleElementSrc::cast(*i1);
}
void PolicyRule::setDummyDestination()
{
FWObjectDatabase *root = getRoot();
FWObject *dummyDestination = root->findInIndex(FWObjectDatabase::DUMMY_ADDRESS_ID);
if (!dummyDestination || (root->getStringId(dummyDestination->getId()) != "dummyaddressid0"))
return;
FWObject::iterator i1 = begin();
i1++;
(*i1)->addRef(dummyDestination);
dst_re = RuleElementDst::cast(*i1);
}
void PolicyRule::setDummyService()
{
FWObjectDatabase *root = getRoot();
FWObject *dummyService = root->findInIndex(FWObjectDatabase::DUMMY_SERVICE_ID);
if (!dummyService || (root->getStringId(dummyService->getId()) != "dummyserviceid0"))
return;
FWObject::iterator i1 = begin();
i1++;
i1++;
(*i1)->addRef(dummyService);
srv_re = RuleElementSrv::cast(*i1);
}
void PolicyRule::setDummyInterface()
{
FWObjectDatabase *root = getRoot();
FWObject *dummyInterface = root->findInIndex(FWObjectDatabase::DUMMY_INTERFACE_ID);
if (!dummyInterface || (root->getStringId(dummyInterface->getId()) != "dummyinterfaceid0"))
return;
FWObject::iterator i1 = begin();
i1++;
i1++;
i1++;
(*i1)->addRef(dummyInterface);
itf_re = RuleElementItf::cast(*i1);
}
void ProjectPanel::exportLibraryTo(QString fname,list<FWObject*> &selectedLibs, bool rof)
{
QApplication::setOverrideCursor( QCursor( Qt::WaitCursor) );
FWObjectDatabase *ndb = db()->exportSubtree( selectedLibs );
QApplication::restoreOverrideCursor();
if (rof)
{
for (list<FWObject*>::iterator i=selectedLibs.begin(); i!=selectedLibs.end(); ++i)
{
FWObject *nlib= ndb->findInIndex( (*i)->getId() );
if (nlib && nlib->getId()!=FWObjectDatabase::DELETED_OBJECTS_ID)
nlib->setReadOnly( true );
}
}
try
{
xmlSetCompressMode(st->getCompression() ? 9 : 0);
ndb->saveFile( fname.toLocal8Bit().constData() );
}
catch (FWException &ex)
{
/* error saving the file. Since XMLTools does not return any useful
* error message in the exception, let's check for obvious problems here
*/
QString err;
if (access( fname.toLocal8Bit().constData(), W_OK)!=0 && errno==EACCES)
err=QObject::tr("File is read-only");
QMessageBox::critical(
this,"Firewall Builder",
QObject::tr("Error saving file %1: %2")
.arg(fname).arg(err),
"&Continue", QString::null, QString::null,
0, 1 );
}
}
/**
* Load library or several libraries from an external file. Return
* pointer to the last new imported library.
*/
FWObject* ProjectPanel::loadLibrary(const string &libfpath)
{
MessageBoxUpgradePredicate upgrade_predicate(mainW);
FWObject *last_new_lib = nullptr;
try
{
FWObjectDatabase *ndb = new FWObjectDatabase();
ndb->load(libfpath, &upgrade_predicate, Constants::getDTDDirectory());
FWObject *dobj = ndb->findInIndex(FWObjectDatabase::DELETED_OBJECTS_ID);
if (dobj) ndb->remove(dobj, false);
set<int> duplicate_ids;
db()->findDuplicateIds(ndb, duplicate_ids);
map<int, int> id_mapping;
for (set<int>::iterator it=duplicate_ids.begin(); it!=duplicate_ids.end();
++it)
{
FWObject *obj = ndb->findInIndex(*it);
assert(obj!=nullptr);
int new_id = FWObjectDatabase::generateUniqueId();
obj->setId(new_id);
id_mapping[*it] = new_id;
// cerr << "Duplicate ID: " << *it
// << " " << FWObjectDatabase::getStringId(*it)
// << obj->getPath()
// << endl;
}
ndb->fixReferences(ndb, id_mapping);
int new_lib_id = -1;
// check for duplicate library names
FWObjectTypedChildIterator it2 = ndb->findByType(Library::TYPENAME);
for (; it2!=it2.end(); ++it2)
{
QString new_name = m_panel->om->makeNameUnique(
db(), QString::fromUtf8((*it2)->getName().c_str()), Library::TYPENAME);
(*it2)->setName(string(new_name.toUtf8()));
if ((*it2)->getId() != FWObjectDatabase::STANDARD_LIB_ID)
new_lib_id = (*it2)->getId();
}
MergeConflictRes mcr(this);
db()->merge(ndb, &mcr);
delete ndb;
last_new_lib = db()->findInIndex(new_lib_id);
} catch(FWException &ex)
{
QString error_txt = ex.toString().c_str();
if (error_txt.length() > LONG_ERROR_CUTOFF)
{
error_txt.truncate(LONG_ERROR_CUTOFF);
error_txt += "\n\n" + tr("(Long error message was truncated)");
}
QMessageBox::critical(
this,"Firewall Builder",
tr("The program encountered error trying to load file %1.\n"
"The file has not been loaded. Error:\n%2").
arg(libfpath.c_str()).arg(error_txt),
tr("&Continue"), QString::null,QString::null,
0, 1 );
}
return last_new_lib;
}
