Examples with INaviOperandTreeNode com.google.security.zynamics.binnavi.disassembly.INaviOperandTreeNode used on opensource projects

Example 6 with INaviOperandTreeNode

use of com.google.security.zynamics.binnavi.disassembly.INaviOperandTreeNode in project binnavi by google.

the class ZyOperandBuilder method isVariableAccess.

/**
 * Determines whether a given node represents variable access.
 *
 * @param node The node to check.
 *
 * @return True, if the node represents an expression that accesses a variable. False, otherwise.
 */
private static boolean isVariableAccess(final INaviOperandTreeNode node) {
    try {
        final INaviOperandTreeNode memoryAccessNode = getMemoryAccessNode(node);
        final List<INaviOperandTreeNode> children = memoryAccessNode.getChildren();
        if (children.size() == 1) {
            final INaviOperandTreeNode child = children.get(0);
            switch(child.getType()) {
                case SIZE_PREFIX:
                    return isVariable(child.getChildren().get(0));
                case IMMEDIATE_INTEGER:
                    return isVariable(child);
                case OPERATOR:
                    return isOperatorVariableAccess(child);
                default:
                    return false;
            }
        } else {
            throw new IllegalStateException("IE00703: Invalid tree shape");
        }
    } catch (final MaybeNullException e) {
        return false;
    }
}

Example 7 with INaviOperandTreeNode

use of com.google.security.zynamics.binnavi.disassembly.INaviOperandTreeNode in project binnavi by google.

the class PostgreSQLTypesNotificationParser method findOperandTreeNode.

/**
 * Only used internally to find the necessary {@link INaviOperandTreeNode operand node} where a
 * {@link TypeSubstitution type substitution} is associated to.
 *
 * @param provider The {@link SQLProvider} used to differentiate between the different caches used
 *        for getting {@link INaviInstruction instructions}.
 * @param moduleId The id of the {@link INaviModule module} this {@link TypeSubstitution type
 *        substitution} is associated to.
 * @param address The {@link CAddress} of the {@link INaviInstruction instruction} where the
 *        {@link TypeSubstitution type substitution} is associated to.
 * @param position The {@link INaviOperandTree operand tree} position where the
 *        {@link TypeSubstitution type substitution} is associated to.
 * @param operandNodeId the id of the {@link INaviOperandTreeNode operand node} to which the
 *        {@link TypeSubstitution type substitution} is associated.
 * @return A {@link INaviOperandTreeNode} if it is in the cache otherwise null;
 */
private INaviOperandTreeNode findOperandTreeNode(final SQLProvider provider, final int moduleId, final CAddress address, final int position, final int operandNodeId) {
    final INaviInstruction instruction = InstructionCache.get(provider).getInstructionByAddress(address, moduleId);
    if (instruction != null) {
        final INaviOperandTree operandTree = instruction.getOperands().get(position);
        final INaviOperandTreeNode root = operandTree.getRootNode();
        final OperandOrderIterator iterator = new OperandOrderIterator(root);
        while (iterator.next()) {
            final INaviOperandTreeNode currentNode = (INaviOperandTreeNode) iterator.current();
            if (currentNode.getId() == operandNodeId) {
                return currentNode;
            }
        }
    }
    return null;
}

Example 8 with INaviOperandTreeNode

use of com.google.security.zynamics.binnavi.disassembly.INaviOperandTreeNode in project binnavi by google.

the class PostgreSQLTypesNotificationParser method informExpressionTypesNotification.

/**
 * Informs about necessary state changes related to {@link TypeSubstitution type substitutions}.
 *
 * @param container The {@link TypesNotificationContainer} holding the parsed
 *        {@link PGNotification notification} information.
 * @param provider The {@link SQLProvider} used to access the database with.
 *
 * @throws CouldntLoadDataException if the required information could not be loaded from the
 *         database.
 */
private void informExpressionTypesNotification(final TypesNotificationContainer container, final SQLProvider provider) throws CouldntLoadDataException {
    final INaviModule module = provider.findModule(container.getModuleId());
    final TypeManager typeManager = module.getTypeManager();
    final INaviOperandTreeNode node = findOperandTreeNode(provider, container.getModuleId(), new CAddress(container.getAddress().get()), container.position().get(), container.expressionId().get());
    if (node == null) {
        return;
    }
    if (container.getDatabaseOperation().equals("INSERT")) {
        final RawTypeSubstitution rawSubstitution = provider.loadTypeSubstitution(module, container.getAddress().get(), container.position().get(), container.expressionId().get());
        typeManager.initializeTypeSubstitution(node, rawSubstitution);
    } else if (container.getDatabaseOperation().equals("UPDATE")) {
        final RawTypeSubstitution rawSubstitution = provider.loadTypeSubstitution(module, container.getAddress().get(), container.position().get(), container.expressionId().get());
        typeManager.updateTypeSubstitution(node, rawSubstitution.getBaseTypeId(), rawSubstitution.getPath(), rawSubstitution.getOffset());
    } else if (container.getDatabaseOperation().equals("DELETE")) {
        typeManager.removeTypeSubstitutionInstance(node.getTypeSubstitution());
    } else {
        throw new IllegalStateException("Error: the database operation " + container.getDatabaseOperation() + " is currently not supported.");
    }
}

Example 9 with INaviOperandTreeNode

use of com.google.security.zynamics.binnavi.disassembly.INaviOperandTreeNode in project binnavi by google.

the class TypeManagerExpensiveTests method testCreateTypeSubstitution.

private void testCreateTypeSubstitution(final INaviModule module) throws CouldntSaveDataException, CouldntLoadDataException, CouldntDeleteException {
    final TypeManager manager = module.getTypeManager();
    final BaseType baseType = manager.createAtomicType(TypeName, 32, true);
    final INaviOperandTreeNode treeNode = new MockOperandTreeNode();
    final int offset = 0;
    final CAddress address = new CAddress(0x1000);
    final TypeSubstitution substitution = manager.createTypeSubstitution(treeNode, baseType, 0, offset, address);
    Assert.assertEquals(substitution.getAddress(), address);
    Assert.assertTrue(substitution.getBaseType() == baseType);
    Assert.assertTrue(substitution.getOffset() == offset);
    Assert.assertTrue(substitution.getOperandTreeNode() == treeNode);
    Assert.assertTrue(substitution.getExpressionId() == treeNode.getId());
    final List<RawTypeSubstitution> substitutions = getProvider().loadTypeSubstitutions(module);
    final List<RawTypeSubstitution> foundSubstitutions = findRawTypeSubstitutions(substitutions, treeNode.getId());
    Assert.assertTrue(foundSubstitutions.size() == 1);
    final RawTypeSubstitution foundSubstitution = foundSubstitutions.get(0);
    matchTypeSubstitution(substitution, foundSubstitution);
    manager.deleteType(baseType);
}

Example 10 with INaviOperandTreeNode

use of com.google.security.zynamics.binnavi.disassembly.INaviOperandTreeNode in project binnavi by google.

the class PostgreSQLInstructionFunctions method createExpressionTree.

/**
 * Creates a new expression tree in the database.
 *
 * @param provider The connection to the database.
 * @param nodes The nodes of the tree to be saved.
 *
 * @return The ID of the created tree in the database.
 *
 * @throws SQLException Thrown if the tree could not be created.
 */
private static int createExpressionTree(final SQLProvider provider, final Set<INaviOperandTreeNode> nodes, final INaviModule module) throws SQLException {
    final int moduleId = module.getConfiguration().getId();
    final ResultSet resultSet = provider.getConnection().executeQuery("select max(id) + 1 AS id from " + CTableNames.EXPRESSION_TREE_IDS_TABLE + " where module_id = " + moduleId, true);
    int expressionTreeId = -1;
    try {
        if (resultSet.next()) {
            expressionTreeId = resultSet.getInt("id");
        }
    } finally {
        resultSet.close();
    }
    provider.getConnection().executeUpdate(String.format("insert into " + CTableNames.EXPRESSION_TREE_IDS_TABLE + " values(%d , %d)", moduleId, expressionTreeId), true);
    for (final INaviOperandTreeNode node : nodes) {
        provider.getConnection().executeUpdate(String.format("insert into " + CTableNames.EXPRESSION_TREE_MAPPING_TABLE + " values(%d, %d, %d)", moduleId, expressionTreeId, node.getId()), true);
    }
    return expressionTreeId;
}