Merge remote-tracking branch 'origin/patch'

This commit is contained in:
Ryan Kurtz 2024-07-08 07:01:45 -04:00
commit 2964a95593
3 changed files with 122 additions and 11 deletions

View file

@ -1459,6 +1459,8 @@ abstract public class DataTypeManagerDB implements DataTypeManager {
* <br>
* NOTE: The original datatype name will be returned unchanged for pointers and arrays since
* they cannot be renamed.
* <br>
* NOTE: Otherwise, if category does not exist the non-conflict name will be returned.
*
* @param path the category path of the category where the new data type live in
* the data type manager.
@ -1466,16 +1468,39 @@ abstract public class DataTypeManagerDB implements DataTypeManager {
* @return the unused conflict name
*/
public String getUnusedConflictName(CategoryPath path, DataType dt) {
String name = dt.getName();
if ((dt instanceof Array) || (dt instanceof Pointer) || (dt instanceof BuiltInDataType)) {
// name not used - anything will do
return name;
return dt.getName();
}
return getUnusedConflictName(getCategory(path), dt);
}
/**
* This method gets a ".conflict" name that is not currently used by any data
* types in the indicated category within this data type manager. If the baseName without
* conflict suffix is not used that name will be returned.
* <br>
* NOTE: The original datatype name will be returned unchanged for pointers and arrays since
* they cannot be renamed.
* <br>
* NOTE: Otherwise, if category does not exist the non-conflict name will be returned.
*
* @param cat the existing category to check.
* @param dt datatype who name is used to establish non-conflict base name
* @return the unused conflict name
*/
private String getUnusedConflictName(Category cat, DataType dt) {
if ((dt instanceof Array) || (dt instanceof Pointer) || (dt instanceof BuiltInDataType)) {
// name not used - anything will do
return dt.getName();
}
String baseName = DataTypeUtilities.getNameWithoutConflict(dt);
if (cat == null) {
return baseName;
}
String testName = baseName;
int count = 0;
while (getDataType(path, testName) != null) {
while (cat.getDataType(testName) != null) {
testName = baseName + DataType.CONFLICT_SUFFIX;
if (count > 0) {
testName += Integer.toString(count);
@ -3130,6 +3155,10 @@ abstract public class DataTypeManagerDB implements DataTypeManager {
flags = (short) TypedefDBAdapter.TYPEDEF_FLAG_AUTONAME;
cat = getCategory(dataType.getCategoryPath()); // force category
}
else if (cat.getDataType(name) != null) {
// force use of conflict name if needed
name = getUnusedConflictName(cat, typedef);
}
DBRecord record = typedefAdapter.createRecord(getID(dataType), name, flags, cat.getID(),
sourceArchiveIdValue, universalIdValue, typedef.getLastChangeTime());
TypedefDB typedefDB = new TypedefDB(this, dtCache, typedefAdapter, record);

View file

@ -0,0 +1,69 @@
/* ###
* IP: GHIDRA
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package ghidra.program.database.data;
import static org.junit.Assert.*;
import org.junit.*;
import generic.test.AbstractGenericTest;
import ghidra.program.model.data.*;
public class TypedefDBTest extends AbstractGenericTest {
private final String NAME = "Test";
private DataTypeManagerDB dataMgr;
private int txId;
@Before
public void setUp() throws Exception {
dataMgr = new StandAloneDataTypeManager("dummyDTM");
txId = dataMgr.startTransaction("Test");
}
@After
public void tearDown() {
if (txId > 0) {
dataMgr.endTransaction(txId, true);
dataMgr.close();
}
}
@Test
public void testDuplicateNameResolve() throws Exception {
Structure struct = new StructureDataType(NAME, 0);
struct.add(new ByteDataType(), "field1", "Comment1");
struct.add(new WordDataType(), null, "Comment2");
struct.add(new DWordDataType(), "field3", null);
struct.add(new ByteDataType(), "field4", "Comment4");
Pointer structPtr = new PointerDataType(struct);
TypeDef typeDef = new TypedefDataType(NAME, structPtr);
TypeDef td = (TypeDef) dataMgr.resolve(typeDef, null);
assertNotNull(td);
assertEquals(NAME + ".conflict", td.getName());
assertTrue(td.isEquivalent(typeDef));
assertEquals("typedef Test.conflict Test *", td.toString());
}
}