From 16f549aaa5050a7b799e1130bae3fac534d8bfd5 Mon Sep 17 00:00:00 2001 From: Andras Gemes Date: Sun, 6 Oct 2024 11:34:26 +0200 Subject: [PATCH 1/2] FunctionID: Add scripts for creating and attaching FID databases --- .../ghidra_scripts/AttachFidDatabase.java | 35 ++++++++++++++++++ .../CreateEmptyFidDatabase.java | 36 +++++++++++++++++++ 2 files changed, 71 insertions(+) create mode 100644 Ghidra/Features/FunctionID/ghidra_scripts/AttachFidDatabase.java create mode 100644 Ghidra/Features/FunctionID/ghidra_scripts/CreateEmptyFidDatabase.java diff --git a/Ghidra/Features/FunctionID/ghidra_scripts/AttachFidDatabase.java b/Ghidra/Features/FunctionID/ghidra_scripts/AttachFidDatabase.java new file mode 100644 index 0000000000..361c3e8983 --- /dev/null +++ b/Ghidra/Features/FunctionID/ghidra_scripts/AttachFidDatabase.java @@ -0,0 +1,35 @@ +/* ### + * 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. + */ +// Attach an existing FID database. +// The functionality is similar to "Tools -> Function ID -> Attach existing FidDb..." +// but this script can be executed in both GUI and headless modes. +// The attached database can later be populated, +// for example by `CreateMultipleLibraries.java` when running in headless mode or +// by "Tools -> Function ID -> Populate FidDb from programs..." when running in GUI mode. +//@category FunctionID +import ghidra.app.script.GhidraScript; +import ghidra.feature.fid.db.FidFileManager; +import java.io.File; + +public class AttachFidDatabase extends GhidraScript { + + @Override + protected void run() throws Exception { + File dbFile = askFile("Attach existing FidDb", "Attach"); + + FidFileManager.getInstance().addUserFidFile(dbFile); + } +} diff --git a/Ghidra/Features/FunctionID/ghidra_scripts/CreateEmptyFidDatabase.java b/Ghidra/Features/FunctionID/ghidra_scripts/CreateEmptyFidDatabase.java new file mode 100644 index 0000000000..1f656b44ab --- /dev/null +++ b/Ghidra/Features/FunctionID/ghidra_scripts/CreateEmptyFidDatabase.java @@ -0,0 +1,36 @@ +/* ### + * 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. + */ +// Create and add (attach) an empty FID database. +// The functionality is similar to "Tools -> Function ID -> Create new empty FidDb..." +// but this script can be executed in both GUI and headless modes. +// The created database can later be populated, +// for example by `CreateMultipleLibraries.java` when running in headless mode or +// by "Tools -> Function ID -> Populate FidDb from programs..." when running in GUI mode. +//@category FunctionID +import ghidra.app.script.GhidraScript; +import ghidra.feature.fid.db.FidFileManager; +import java.io.File; + +public class CreateEmptyFidDatabase extends GhidraScript { + + @Override + protected void run() throws Exception { + File dbFile = askFile("Create new FidDb file", "Create"); + + FidFileManager.getInstance().createNewFidDatabase(dbFile); + FidFileManager.getInstance().addUserFidFile(dbFile); + } +} From 6b2932ccbb45ff354ab77d30f632ae311cd3dfed Mon Sep 17 00:00:00 2001 From: Ryan Kurtz Date: Wed, 30 Oct 2024 07:38:49 -0400 Subject: [PATCH 2/2] GP-0: Certify --- .../Features/FunctionID/ghidra_scripts/AttachFidDatabase.java | 4 ++-- .../FunctionID/ghidra_scripts/CreateEmptyFidDatabase.java | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/Ghidra/Features/FunctionID/ghidra_scripts/AttachFidDatabase.java b/Ghidra/Features/FunctionID/ghidra_scripts/AttachFidDatabase.java index 361c3e8983..2709a22cd1 100644 --- a/Ghidra/Features/FunctionID/ghidra_scripts/AttachFidDatabase.java +++ b/Ghidra/Features/FunctionID/ghidra_scripts/AttachFidDatabase.java @@ -4,9 +4,9 @@ * 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. diff --git a/Ghidra/Features/FunctionID/ghidra_scripts/CreateEmptyFidDatabase.java b/Ghidra/Features/FunctionID/ghidra_scripts/CreateEmptyFidDatabase.java index 1f656b44ab..f67bb84ad6 100644 --- a/Ghidra/Features/FunctionID/ghidra_scripts/CreateEmptyFidDatabase.java +++ b/Ghidra/Features/FunctionID/ghidra_scripts/CreateEmptyFidDatabase.java @@ -4,9 +4,9 @@ * 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.