From 31dc7101cf0d355e8919991ca8e00b05cdca7ee2 Mon Sep 17 00:00:00 2001 From: Jan-Jaap Korpershoek Date: Fri, 21 Jun 2024 22:17:26 +0200 Subject: [PATCH] Add organization field --- recon/core/base.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/recon/core/base.py b/recon/core/base.py index 446fd7d..0215564 100644 --- a/recon/core/base.py +++ b/recon/core/base.py @@ -238,7 +238,7 @@ def _create_db(self): self.query('CREATE TABLE IF NOT EXISTS locations (latitude TEXT, longitude TEXT, street_address TEXT, notes TEXT, module TEXT)') self.query('CREATE TABLE IF NOT EXISTS vulnerabilities (host TEXT, reference TEXT, example TEXT, publish_date TEXT, category TEXT, status TEXT, notes TEXT, module TEXT)') self.query('CREATE TABLE IF NOT EXISTS ports (ip_address TEXT, host TEXT, port TEXT, protocol TEXT, banner TEXT, notes TEXT, module TEXT)') - self.query('CREATE TABLE IF NOT EXISTS hosts (host TEXT, ip_address TEXT, region TEXT, country TEXT, latitude TEXT, longitude TEXT, notes TEXT, module TEXT)') + self.query('CREATE TABLE IF NOT EXISTS hosts (host TEXT, ip_address TEXT, region TEXT, country TEXT, latitude TEXT, longitude TEXT, organization TEXT, notes TEXT, module TEXT)') self.query('CREATE TABLE IF NOT EXISTS contacts (first_name TEXT, middle_name TEXT, last_name TEXT, email TEXT, title TEXT, region TEXT, country TEXT, phone TEXT, notes TEXT, module TEXT)') self.query('CREATE TABLE IF NOT EXISTS credentials (username TEXT, password TEXT, hash TEXT, type TEXT, leak TEXT, notes TEXT, module TEXT)') self.query('CREATE TABLE IF NOT EXISTS leaks (leak_id TEXT, description TEXT, source_refs TEXT, leak_type TEXT, title TEXT, import_date TEXT, leak_date TEXT, attackers TEXT, num_entries TEXT, score TEXT, num_domains_affected TEXT, attack_method TEXT, target_industries TEXT, password_hash TEXT, password_type TEXT, targets TEXT, media_refs TEXT, notes TEXT, module TEXT)') @@ -321,6 +321,9 @@ def _migrate_db(self): # add phone column to contacts table self.query('ALTER TABLE contacts ADD COLUMN phone TEXT') self.query('PRAGMA user_version = 10') + if db_version(self) == 10: + self.query('ALTER TABLE hosts ADD COLUMN organization TEXT') + self.query('PRAGMA user_version = 11') if db_orig != db_version(self): self.alert(f"Database upgraded to version {db_version(self)}.")