def read_connection_from_sql(sql_file, sql_statement, connections): conn = sqlite3.connect(sql_file) cursor = conn.cursor() cursor.execute(sql_statement) for entry in cursor.fetchall(): conn = Connection() conn._dst_IP = entry[1] conn._DNS = entry[2] conn._rDNS = entry[3] conn._current_volume = entry[4] conn._parentBatchID = entry[6] for pattern in cdn_regex: if pattern.search(entry[2]) or pattern.search(entry[3]): conn._is_CDN_connection = True break connections.append(conn)
def cdn_analysis_helper(container, type): sql = "select * from %sConnections" % (type) sql_conn = sqlite3.connect(opts.file) cursor = sql_conn.cursor() cursor.execute(sql) for entry in cursor.fetchall(): conn = Connection() conn._dst_IP = entry[1] conn._DNS = entry[2] conn._rDNS = entry[3] conn._current_volume = entry[4] conn._parentBatchID = entry[6] for pattern in cdn_regex: if pattern.search(entry[2]) or pattern.search(entry[3]): conn._is_CDN_connection = True break container.append(conn) sql_conn.close()