Search in sources :

Example 16 with LoadCancelledException

use of com.google.security.zynamics.binnavi.Database.Exceptions.LoadCancelledException in project binnavi by google.

the class CProjectFunctions method addAddressSpace.

/**
   * Adds a new address space with a default name to a given project.
   * 
   * @param parent Parent window used for dialogs.
   * @param project The project where the new address space is added.
   * @param updater Updates the project tree after the execution is complete.
   */
public static void addAddressSpace(final Window parent, final INaviProject project, final INodeSelectionUpdater updater) {
    new Thread() {

        @Override
        public void run() {
            final CDefaultProgressOperation operation = new CDefaultProgressOperation("", false, true);
            operation.getProgressPanel().setMaximum(2);
            try {
                operation.getProgressPanel().setText("Creating new address space");
                operation.getProgressPanel().next();
                final CAddressSpace addressSpace = project.getContent().createAddressSpace("New Address Space");
                operation.getProgressPanel().setText("Loading new address space");
                addressSpace.load();
                operation.getProgressPanel().next();
                operation.stop();
                updater.setObject(addressSpace);
                updater.update();
            } catch (final CouldntSaveDataException exception) {
                CUtilityFunctions.logException(exception);
                final String innerMessage = "E00136: " + "Could not add address space";
                final String innerDescription = CUtilityFunctions.createDescription(String.format("It was not possible to add a new address space to the project '%s'.", project.getConfiguration().getName()), new String[] { "There was a problem with the database connection." }, new String[] { "The address space was not created." });
                NaviErrorDialog.show(parent, innerMessage, innerDescription, exception);
            } catch (final CouldntLoadDataException exception) {
                CUtilityFunctions.logException(exception);
                final String innerMessage = "E00137: " + "Could not load the new address space";
                final String innerDescription = CUtilityFunctions.createDescription(String.format("The new address space in project '%s' was created but it could not be loaded.", project.getConfiguration().getName()), new String[] { "There was a problem with the database connection." }, new String[] { "The address space was created but not loaded." });
                NaviErrorDialog.show(parent, innerMessage, innerDescription, exception);
            } catch (final LoadCancelledException e) {
            // Do nothing
            }
        }
    }.start();
}
Also used : CDefaultProgressOperation(com.google.security.zynamics.binnavi.Gui.Progress.CDefaultProgressOperation) CouldntSaveDataException(com.google.security.zynamics.binnavi.Database.Exceptions.CouldntSaveDataException) CouldntLoadDataException(com.google.security.zynamics.binnavi.Database.Exceptions.CouldntLoadDataException) LoadCancelledException(com.google.security.zynamics.binnavi.Database.Exceptions.LoadCancelledException) CAddressSpace(com.google.security.zynamics.binnavi.disassembly.AddressSpaces.CAddressSpace)

Example 17 with LoadCancelledException

use of com.google.security.zynamics.binnavi.Database.Exceptions.LoadCancelledException in project binnavi by google.

the class CProjectFunctions method createProject.

/**
   * Creates a new project.
   * 
   * @param parent Parent window used for dialogs.
   * @param database Database where the project is created.
   * @param updater Responsible for updating the project tree after the project was created.
   */
public static void createProject(final JFrame parent, final IDatabase database, final INodeSelectionUpdater updater) {
    new Thread() {

        @Override
        public void run() {
            try {
                final CDefaultProgressOperation operation = new CDefaultProgressOperation("", false, true);
                operation.getProgressPanel().setMaximum(3);
                operation.getProgressPanel().setText("Creating new project");
                final INaviProject newProject = database.getContent().addProject("New Project");
                operation.getProgressPanel().next();
                try {
                    newProject.load();
                } catch (final LoadCancelledException e) {
                // Do nothing
                }
                operation.getProgressPanel().next();
                createDefaultAddressSpace(parent, newProject);
                updater.setObject(newProject);
                updater.update();
                operation.getProgressPanel().next();
                operation.stop();
            } catch (final CouldntSaveDataException exception) {
                CUtilityFunctions.logException(exception);
                final String innerMessage = "E00140: " + "New project could not be created";
                final String innerDescription = CUtilityFunctions.createDescription("It was not possible to create a new project in the selected database.", new String[] { "There was a problem with the database connection." }, new String[] { "No new project was created in the selected database." });
                NaviErrorDialog.show(parent, innerMessage, innerDescription, exception);
            } catch (final CouldntLoadDataException exception) {
                CUtilityFunctions.logException(exception);
                final String innerMessage = "E00141: " + "New project could not be loaded";
                final String innerDescription = CUtilityFunctions.createDescription("The new project could not be loaded.", new String[] { "There was a problem with the database connection." }, new String[] { "The new project was created but it could not be loaded." });
                NaviErrorDialog.show(parent, innerMessage, innerDescription, exception);
            }
        }
    }.start();
}
Also used : CDefaultProgressOperation(com.google.security.zynamics.binnavi.Gui.Progress.CDefaultProgressOperation) INaviProject(com.google.security.zynamics.binnavi.disassembly.INaviProject) CouldntSaveDataException(com.google.security.zynamics.binnavi.Database.Exceptions.CouldntSaveDataException) CouldntLoadDataException(com.google.security.zynamics.binnavi.Database.Exceptions.CouldntLoadDataException) LoadCancelledException(com.google.security.zynamics.binnavi.Database.Exceptions.LoadCancelledException)

Example 18 with LoadCancelledException

use of com.google.security.zynamics.binnavi.Database.Exceptions.LoadCancelledException in project binnavi by google.

the class ModuleHelpersTest method testGetFunction_1.

@Test
public void testGetFunction_1() {
    final Database database = new Database(new MockDatabase());
    @SuppressWarnings("unused") final MockModule mockModule = new MockModule();
    final MockSqlProvider provider = new MockSqlProvider();
    final CModule internalModule = new CModule(1, "", "", new Date(), new Date(), "00000000000000000000000000000000", "0000000000000000000000000000000000000000", 0, 0, new CAddress(0), new CAddress(0), null, null, Integer.MAX_VALUE, false, provider);
    try {
        internalModule.load();
    } catch (final CouldntLoadDataException exception) {
        CUtilityFunctions.logException(exception);
    } catch (final LoadCancelledException exception) {
        CUtilityFunctions.logException(exception);
    }
    @SuppressWarnings("unused") final CFunction parentFunction = new CFunction(internalModule, new MockView(), new CAddress(0x123), "Mock Function", "Mock Function", "Mock Description", 0, 0, 0, 0, FunctionType.NORMAL, "", 0, null, null, null, provider);
    final TagManager nodeTagManager = new TagManager(new MockTagManager(com.google.security.zynamics.binnavi.Tagging.TagType.NODE_TAG));
    final TagManager viewTagManager = new TagManager(new MockTagManager(com.google.security.zynamics.binnavi.Tagging.TagType.VIEW_TAG));
    final Module module = new Module(database, internalModule, nodeTagManager, viewTagManager);
    assertEquals(module.getFunctions().get(0), ModuleHelpers.getFunction(module, 0x123));
    assertNull(ModuleHelpers.getFunction(module, 0x1235));
    try {
        ModuleHelpers.getFunction(null, -1);
        fail();
    } catch (final NullPointerException e) {
    }
    try {
        ModuleHelpers.getFunction(module, -1);
        fail();
    } catch (final IllegalArgumentException e) {
    }
}
Also used : MockView(com.google.security.zynamics.binnavi.disassembly.MockView) CouldntLoadDataException(com.google.security.zynamics.binnavi.Database.Exceptions.CouldntLoadDataException) CFunction(com.google.security.zynamics.binnavi.disassembly.CFunction) MockTagManager(com.google.security.zynamics.binnavi.Tagging.MockTagManager) Date(java.util.Date) CAddress(com.google.security.zynamics.zylib.disassembly.CAddress) MockTagManager(com.google.security.zynamics.binnavi.Tagging.MockTagManager) MockModule(com.google.security.zynamics.binnavi.disassembly.Modules.MockModule) MockSqlProvider(com.google.security.zynamics.binnavi.Database.MockClasses.MockSqlProvider) MockDatabase(com.google.security.zynamics.binnavi.Database.MockClasses.MockDatabase) MockDatabase(com.google.security.zynamics.binnavi.Database.MockClasses.MockDatabase) LoadCancelledException(com.google.security.zynamics.binnavi.Database.Exceptions.LoadCancelledException) MockModule(com.google.security.zynamics.binnavi.disassembly.Modules.MockModule) CModule(com.google.security.zynamics.binnavi.disassembly.Modules.CModule) CModule(com.google.security.zynamics.binnavi.disassembly.Modules.CModule) Test(org.junit.Test)

Example 19 with LoadCancelledException

use of com.google.security.zynamics.binnavi.Database.Exceptions.LoadCancelledException in project binnavi by google.

the class CModule method load.

@Override
public void load() throws CouldntLoadDataException, LoadCancelledException {
    synchronized (m_loadReporter) {
        if (isLoaded()) {
            return;
        }
        if (!isInitialized()) {
            throw new IllegalStateException("IE02617: The module is not initialized yet");
        }
        m_isLoading = true;
        try {
            if (!m_loadReporter.report(ModuleLoadEvents.Starting)) {
                throw new LoadCancelledException();
            }
            if (!m_loadReporter.report(ModuleLoadEvents.LoadingCallgraphView)) {
                throw new LoadCancelledException();
            }
            final ICallgraphView nativeCallgraph = m_provider.loadNativeCallgraph(this);
            if (!m_loadReporter.report(ModuleLoadEvents.LoadingFlowgraphs)) {
                throw new LoadCancelledException();
            }
            // When loading views, an empty flow graph (imported function) can not
            // be distinguished
            // from an empty call graph. This leads to problems in modules without
            // functions. In
            // these cases, the empty call graph is loaded as an imported function
            // view. This is
            // obviously wrong, so we are correcting this here.
            final ImmutableList<IFlowgraphView> nativeFlowgraphs = nativeCallgraph.getNodeCount() == 0 ? new ImmutableList.Builder<IFlowgraphView>().build() : m_provider.loadNativeFlowgraphs(this);
            if (!m_loadReporter.report(ModuleLoadEvents.LoadingCallgraphViews)) {
                throw new LoadCancelledException();
            }
            final List<ICallgraphView> userCallgraphs = m_provider.loadCallgraphViews(this);
            if (!m_loadReporter.report(ModuleLoadEvents.LoadingFlowgraphViews)) {
                throw new LoadCancelledException();
            }
            final ImmutableList<IFlowgraphView> userFlowgraphs = m_provider.loadFlowgraphs(this);
            if (!m_loadReporter.report(ModuleLoadEvents.LoadingMixedViews)) {
                throw new LoadCancelledException();
            }
            final List<INaviView> userMixedGraphs = m_provider.loadMixedgraphs(this);
            if (!m_loadReporter.report(ModuleLoadEvents.LoadingCallgraph)) {
                throw new LoadCancelledException();
            }
            if (!m_loadReporter.report(ModuleLoadEvents.LoadingFunctions)) {
                throw new LoadCancelledException();
            }
            // Note: the type manager needs to be loaded prior to functions, since a function might
            // have an associated stack frame, which in turn needs the type system.
            typeManager = new TypeManager(new TypeManagerDatabaseBackend(m_provider, this));
            final List<INaviFunction> functions = m_provider.loadFunctions(this, nativeFlowgraphs);
            final ImmutableBiMap<INaviView, INaviFunction> viewFunctionMap = m_provider.loadViewFunctionMapping(nativeFlowgraphs, functions, this);
            final CCallgraph callgraph = m_provider.loadCallgraph(this, nativeCallgraph.getConfiguration().getId(), functions);
            if (!m_loadReporter.report(ModuleLoadEvents.LoadingTraces)) {
                throw new LoadCancelledException();
            }
            final List<TraceList> traces = m_provider.loadTraces(this);
            if (!m_loadReporter.report(ModuleLoadEvents.LoadingGlobalVariables)) {
                throw new LoadCancelledException();
            }
            final List<INaviView> customViews = new ArrayList<INaviView>();
            customViews.addAll(userCallgraphs);
            customViews.addAll(userFlowgraphs);
            customViews.addAll(userMixedGraphs);
            final List<INaviView> currentViews = new ArrayList<INaviView>(nativeFlowgraphs);
            currentViews.addAll(customViews);
            Collections.sort(currentViews, new Comparator<INaviView>() {

                @Override
                public int compare(final INaviView lhs, final INaviView rhs) {
                    return lhs.getConfiguration().getId() - rhs.getConfiguration().getId();
                }
            });
            // Map viewId to corresponding index in currentViews.
            final Map<Integer, Integer> viewIdToIndex = new HashMap<Integer, Integer>();
            for (int i = 0; i < currentViews.size(); ++i) {
                viewIdToIndex.put(currentViews.get(i).getConfiguration().getId(), i);
            }
            if (!m_loadReporter.report(ModuleLoadEvents.LoadingTypes)) {
                throw new LoadCancelledException();
            }
            final SectionContainer sections = new SectionContainer(new SectionContainerBackend(m_provider, this));
            final TypeInstanceContainer typeInstances = new TypeInstanceContainer(new TypeInstanceContainerBackend(m_provider, this, typeManager, sections), m_provider);
            m_content = new CModuleContent(this, m_provider, m_listeners, callgraph, functions, nativeCallgraph, nativeFlowgraphs, customViews, viewFunctionMap, traces, sections, typeInstances);
            typeInstances.initialize();
        } catch (final CouldntLoadDataException e) {
            m_isLoading = false;
            throw e;
        } catch (final LoadCancelledException e) {
            m_isLoading = false;
            throw e;
        } finally {
            m_loadReporter.report(ModuleLoadEvents.Finished);
        }
        for (final IModuleListener listener : m_listeners) {
            try {
                listener.loadedModule(this);
            } catch (final Exception exception) {
                CUtilityFunctions.logException(exception);
            }
        }
        m_isLoading = false;
    }
}
Also used : IFlowgraphView(com.google.security.zynamics.binnavi.disassembly.IFlowgraphView) HashMap(java.util.HashMap) CouldntLoadDataException(com.google.security.zynamics.binnavi.Database.Exceptions.CouldntLoadDataException) ArrayList(java.util.ArrayList) TypeInstanceContainerBackend(com.google.security.zynamics.binnavi.disassembly.types.TypeInstanceContainerBackend) SectionContainerBackend(com.google.security.zynamics.binnavi.disassembly.types.SectionContainerBackend) CCallgraph(com.google.security.zynamics.binnavi.disassembly.CCallgraph) LoadCancelledException(com.google.security.zynamics.binnavi.Database.Exceptions.LoadCancelledException) CouldntSaveDataException(com.google.security.zynamics.binnavi.Database.Exceptions.CouldntSaveDataException) CouldntLoadDataException(com.google.security.zynamics.binnavi.Database.Exceptions.CouldntLoadDataException) ICallgraphView(com.google.security.zynamics.binnavi.disassembly.ICallgraphView) INaviView(com.google.security.zynamics.binnavi.disassembly.views.INaviView) TypeManagerDatabaseBackend(com.google.security.zynamics.binnavi.disassembly.types.TypeManagerDatabaseBackend) SectionContainer(com.google.security.zynamics.binnavi.disassembly.types.SectionContainer) TypeManager(com.google.security.zynamics.binnavi.disassembly.types.TypeManager) TraceList(com.google.security.zynamics.binnavi.debug.models.trace.TraceList) LoadCancelledException(com.google.security.zynamics.binnavi.Database.Exceptions.LoadCancelledException) TypeInstanceContainer(com.google.security.zynamics.binnavi.disassembly.types.TypeInstanceContainer) INaviFunction(com.google.security.zynamics.binnavi.disassembly.INaviFunction)

Aggregations

LoadCancelledException (com.google.security.zynamics.binnavi.Database.Exceptions.LoadCancelledException)19 CouldntLoadDataException (com.google.security.zynamics.binnavi.Database.Exceptions.CouldntLoadDataException)18 CouldntSaveDataException (com.google.security.zynamics.binnavi.Database.Exceptions.CouldntSaveDataException)8 CPartialLoadException (com.google.security.zynamics.binnavi.Database.Exceptions.CPartialLoadException)6 CTag (com.google.security.zynamics.binnavi.Tagging.CTag)5 CouldntConnectException (com.google.security.zynamics.binnavi.Database.Exceptions.CouldntConnectException)4 CouldntInitializeDatabaseException (com.google.security.zynamics.binnavi.Database.Exceptions.CouldntInitializeDatabaseException)4 CouldntLoadDriverException (com.google.security.zynamics.binnavi.Database.Exceptions.CouldntLoadDriverException)4 InvalidDatabaseException (com.google.security.zynamics.binnavi.Database.Exceptions.InvalidDatabaseException)4 InvalidDatabaseVersionException (com.google.security.zynamics.binnavi.Database.Exceptions.InvalidDatabaseVersionException)4 InvalidExporterDatabaseFormatException (com.google.security.zynamics.binnavi.Database.Exceptions.InvalidExporterDatabaseFormatException)4 CDefaultProgressOperation (com.google.security.zynamics.binnavi.Gui.Progress.CDefaultProgressOperation)4 INaviModule (com.google.security.zynamics.binnavi.disassembly.INaviModule)4 CouldntUpdateDatabaseException (com.google.security.zynamics.binnavi.Database.Exceptions.CouldntUpdateDatabaseException)3 TraceList (com.google.security.zynamics.binnavi.debug.models.trace.TraceList)3 CAddressSpace (com.google.security.zynamics.binnavi.disassembly.AddressSpaces.CAddressSpace)3 INaviFunction (com.google.security.zynamics.binnavi.disassembly.INaviFunction)3 INaviView (com.google.security.zynamics.binnavi.disassembly.views.INaviView)3 ArrayList (java.util.ArrayList)3 CouldntDeleteException (com.google.security.zynamics.binnavi.Database.Exceptions.CouldntDeleteException)2