yetangitu/ghidra
1
0
Fork 0
mirror of https://github.com/NationalSecurityAgency/ghidra.git synced 2025-10-05 02:39:44 +02:00
Code Issues Releases Wiki Activity

Merge remote-tracking branch

Browse source 
'origin/GP-774-dragonmacher-dt-manager-deadlock--SQUASHED' (Closes
#2832)
This commit is contained in:
ghidra1 2021-03-15 17:55:26 -04:00
commit 28138bcba2
1 changed files with 48 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

76
Ghidra/Framework/SoftwareModeling/src/main/java/ghidra/program/database/data/DataTypeManagerDB.java
View file

@ -1695,34 +1695,41 @@ abstract public class DataTypeManagerDB implements DataTypeManager {
@Override
public void disassociate(DataType dataType) {
UniversalID oldDtID = dataType.getUniversalID();
SourceArchive sourceArchive = dataType.getSourceArchive();
sourceArchive = resolveSourceArchive(sourceArchive);
UniversalID id = sourceArchive == null ? DataTypeManager.LOCAL_ARCHIVE_UNIVERSAL_ID
: sourceArchive.getSourceArchiveID();
if (id.equals(getUniversalID())) {
id = DataTypeManager.LOCAL_ARCHIVE_UNIVERSAL_ID;
}
if (id == DataTypeManager.LOCAL_ARCHIVE_UNIVERSAL_ID) {
// Already local data type so no source archive associated.
return;
}
// Set the source archive to null indicating no associated archive.
dataType.setSourceArchive(null);
lock.acquire();
try {
UniversalID oldDtID = dataType.getUniversalID();
SourceArchive sourceArchive = dataType.getSourceArchive();
sourceArchive = resolveSourceArchive(sourceArchive);
UniversalID id = sourceArchive == null ? DataTypeManager.LOCAL_ARCHIVE_UNIVERSAL_ID
: sourceArchive.getSourceArchiveID();
if (id.equals(getUniversalID())) {
id = DataTypeManager.LOCAL_ARCHIVE_UNIVERSAL_ID;
}
if (id == DataTypeManager.LOCAL_ARCHIVE_UNIVERSAL_ID) {
// Already local data type so no source archive associated.
return;
}
// Set the datatype's universal ID to a newly generated universal ID,
// since we no longer want the source archive data type's universal ID.
if (dataType instanceof DataTypeDB) {
DataTypeDB dt = (DataTypeDB) dataType;
dt.setUniversalID(UniversalIdGenerator.nextID());
// Set the source archive to null indicating no associated archive.
dataType.setSourceArchive(null);
// Set the datatype's universal ID to a newly generated universal ID,
// since we no longer want the source archive data type's universal ID.
if (dataType instanceof DataTypeDB) {
DataTypeDB dt = (DataTypeDB) dataType;
dt.setUniversalID(UniversalIdGenerator.nextID());
}
if (oldDtID != null) {
idsToDataTypeMap.removeDataType(sourceArchive, oldDtID);
}
dataTypeChanged(dataType);
}
if (oldDtID != null) {
idsToDataTypeMap.removeDataType(sourceArchive, oldDtID);
finally {
lock.release();
}
dataTypeChanged(dataType);
}
private Collection<DataType> filterOutNonSourceSettableDataTypes(
@ -2694,9 +2701,10 @@ abstract public class DataTypeManagerDB implements DataTypeManager {
}
try {
creatingDataType++;
DBRecord record = functionDefAdapter.createRecord(name, funDef.getComment(), cat.getID(),
DEFAULT_DATATYPE_ID, funDef.hasVarArgs(), funDef.getGenericCallingConvention(),
sourceArchiveIdValue, universalIdValue, funDef.getLastChangeTime());
DBRecord record =
functionDefAdapter.createRecord(name, funDef.getComment(), cat.getID(),
DEFAULT_DATATYPE_ID, funDef.hasVarArgs(), funDef.getGenericCallingConvention(),
sourceArchiveIdValue, universalIdValue, funDef.getLastChangeTime());
FunctionDefinitionDB funDefDb =
new FunctionDefinitionDB(this, dtCache, functionDefAdapter, paramAdapter, record);
@ -3696,7 +3704,13 @@ abstract public class DataTypeManagerDB implements DataTypeManager {
@Override
public DataType getDataType(SourceArchive sourceArchive, UniversalID datatypeID) {
UniversalID sourceID = sourceArchive == null ? null : sourceArchive.getSourceArchiveID();
return idsToDataTypeMap.getDataType(sourceID, datatypeID);
lock.acquire();
try {
return idsToDataTypeMap.getDataType(sourceID, datatypeID);
}
finally {
lock.release();
}
}
@Override
@ -4162,7 +4176,11 @@ abstract public class DataTypeManagerDB implements DataTypeManager {
Map<UniversalID, DataType> idMap =
map.computeIfAbsent(sourceID, k -> new ConcurrentHashMap<>());
final UniversalID sourceArchiveID = sourceID;
UniversalID sourceArchiveID = sourceID;
// note: this call is atomic and has a lock on the 'idMap'. It may call to a method
// that requires a db lock. As such, the call to computeIfAbsent() must be
// made while holding the db lock.
return idMap.computeIfAbsent(dataTypeID,
k -> findDataTypeForIDs(sourceArchiveID, dataTypeID));
}