🔍 Scanned IPs: {{ stats.found }}
✅ Verified Ollama: {{ stats.verified_models }}
🚀 Queue Size: {{ queue_size }}
🔄 Active Checks: {{ active_checks }}
⏱️ Uptime: {{ uptime }}s
''', stats=stats, queue_size=ip_queue.qsize(), active_checks=active_checks, uptime=uptime)
# --- Verify Ollama ---
def verify_ollama(ip):
global active_checks, stats
active_checks += 1
try:
url = f"http://{ip}:{TARGET_PORT}/api/tags"
response = requests.get(url, timeout=3)
if response.status_code == 200 and "models" in response.json():
stats["verified_models"] += 1
conn = db_pool.getconn()
try:
with conn.cursor() as cur:
cur.execute("""
INSERT INTO providers (ip, port, first_seen)
VALUES (%s, %s, %s)
ON CONFLICT (ip) DO NOTHING
""", (ip, TARGET_PORT, datetime.utcnow()))
conn.commit()
stats["api_success"] += 1
except Exception as e:
print(f"DB Error: {e}")
stats["api_failed"] += 1
finally:
db_pool.putconn(conn)
except:
stats["verification_failed"] += 1
finally:
active_checks -= 1
# --- Worker ---
def worker():
while True:
if not ip_queue.empty():
ip = ip_queue.get()
verify_ollama(ip)
time.sleep(0.01)
# --- ZMap Scanner ---
def run_zmap():
print("=== Starting ZMap Scan ===")
process = subprocess.Popen(
["sudo", "zmap", "-p", str(TARGET_PORT), "-o", "-"],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
universal_newlines=True
)
for line in process.stdout:
ip = line.strip()
if re.match(r"^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$", ip):
if ip not in processed_ips:
processed_ips.add(ip)
stats["found"] += 1
ip_queue.put(ip)
print("ZMap finished. Restarting in 5 sec...")
time.sleep(5)
run_zmap()
# --- Start ---
if __name__ == '__main__':
# Start workers
for _ in range(NUM_CHECKERS):
threading.Thread(target=worker, daemon=True).start()
# Start ZMap
threading.Thread(target=run_zmap, daemon=True).start()
app.run(host='0.0.0.0', port=5000, threaded=True)