# (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA from umit.db.Utils import empty from umit.db.Utils import log_debug debug = log_debug('umit.db.Store') """ Missing methods for: Traceroute insertion. """ class RawStore: """ Store data into database. """ def __init__(self, conn, cursor): """ Expects a conn and cursor from database connection. """
# You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA from umit.db.Utils import empty from umit.db.Utils import log_debug from umit.db.Utils import normalize from umit.db.Connection import ConnectDB """ Missing methods for: Traceroute retrieval. """ debug = log_debug('umit.db.Retrieve') class RawRetrieve: """ Retrieve raw data from database. """ def __init__(self, conn, cursor): """ Expects a conn and cursor from database connection. """ self.conn = conn self.cursor = cursor def get_service_info_id_from_db(self, info, service_name_id):
# but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA from datetime import datetime, timedelta from umit.db.Utils import log_debug from umit.db.Connection import ConnectDB from umit.db.Retrieve import CompositeRetrieve debug = log_debug('umit.db.Remove') class ScanRemover(ConnectDB, CompositeRetrieve): """ Removes scans from database. """ def __init__(self, database): ConnectDB.__init__(self, database) CompositeRetrieve.__init__(self, self.conn, self.cursor) def remove_scan(self, fk_scan): """ Removes a scan from database. """
# # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA from umit.db.Utils import empty from umit.db.Utils import log_debug from umit.db.Utils import normalize from umit.db.Connection import ConnectDB """ Missing methods for: Traceroute retrieval. """ debug = log_debug('umit.db.Retrieve') class RawRetrieve: """ Retrieve raw data from database. """ def __init__(self, conn, cursor): """ Expects a conn and cursor from database connection. """ self.conn = conn self.cursor = cursor def get_service_info_id_from_db(self, info, service_name_id): """
# (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA from umit.db.Utils import empty from umit.db.Utils import log_debug debug = log_debug('umit.db.Store') """ Missing methods for: Traceroute insertion. """ class RawStore: """ Store data into database. """ def __init__(self, conn, cursor): """ Expects a conn and cursor from database connection. """ self.conn = conn
# You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA from datetime import datetime from umit.core.NmapParser import NmapParser from umit.db.Connection import ConnectDB from umit.db.Store import RawStore from umit.db.Retrieve import InventoryRetrieve from umit.db.InventoryChanges import UpdateChanges from umit.db.Utils import empty, log_debug, normalize debug = log_debug('umit.db.XMLStore') class XMLStore(ConnectDB, InventoryRetrieve, RawStore): """ Stores xml into database. """ def __init__(self, database, store_original=False): """ store_original - stores xml files in the database or not """ ConnectDB.__init__(self, database) InventoryRetrieve.__init__(self, self.conn, self.cursor) RawStore.__init__(self, self.conn, self.cursor)
# but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA from umit.db.Utils import log_debug from umit.db.Connection import ConnectDB from umit.db.Retrieve import CompositeRetrieve from umit.core.I18N import _ debug = log_debug('umit.db.Search') change_text = (_("change"), _("changes")) port_text = (_("port"), _("ports"), _("service")) comparison = ("<", ">", "==", ">=", "<=", "!=") def perform_comparison(ports, compare, decider): """ Compare ports against decider using especified comparation. """ decider = int(decider) # ports are numbered in ascendent order. for port in ports: if compare == '>':
# but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA from umit.db.Utils import log_debug from umit.db.Connection import ConnectDB from umit.db.Retrieve import CompositeRetrieve from umit.core.I18N import _ debug = log_debug('umit.db.Search') change_text = (_("change"), _("changes")) port_text = (_("port"), _("ports"), _("service")) comparison = ("<", ">", "==", ">=", "<=", "!=") def perform_comparison(ports, compare, decider): """ Compare ports against decider using especified comparation. """ decider = int(decider) # ports are numbered in ascendent order. for port in ports: if compare == '>': if decider <= port[0]:
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA from umit.core.I18N import _ from umit.db.Utils import log_debug from umit.db.Connection import ConnectDB from umit.db.Store import RawStore from umit.db.Retrieve import InventoryRetrieve debug = log_debug("umit.db.InventoryChanges") class BadParams(Exception): pass class ConnectToUpdate(ConnectDB, InventoryRetrieve, RawStore): """ You should use this when doing inventory changes update exclusively in database. """ def __init__(self, database): ConnectDB.__init__(self, database) InventoryRetrieve.__init__(self, self.conn, self.cursor) RawStore.__init__(self, self.conn, self.cursor)
# but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA from datetime import datetime, timedelta from umit.db.Utils import log_debug from umit.db.Connection import ConnectDB from umit.db.Retrieve import CompositeRetrieve debug = log_debug("umit.db.Remove") class ScanRemover(ConnectDB, CompositeRetrieve): """ Removes scans from database. """ def __init__(self, database): ConnectDB.__init__(self, database) CompositeRetrieve.__init__(self, self.conn, self.cursor) def remove_scan(self, fk_scan): """ Removes a scan from database. """
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA """ This handles connections to the database. """ from umit.db._database import sql from umit.db._database import database from umit.db.Utils import log_debug debug = log_debug('umit.db.Connection') def dict_factory(cursor, row): """ Convert a standard row to a dict. """ d = {} for idx, col in enumerate(cursor.description): d[col[0]] = row[idx] return d class ConnectDB:
# GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 # USA """ This handles connections to the database. """ from umit.db._database import sql from umit.db._database import database from umit.db.Utils import log_debug debug = log_debug('umit.db.Connection') def dict_factory(cursor, row): """ Convert a standard row to a dict. """ d = {} for idx, col in enumerate(cursor.description): d[col[0]] = row[idx] return d class ConnectDB: """