59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
from flask import Flask, request, send_file, after_this_request
|
|
import os
|
|
import subprocess
|
|
import socket
|
|
|
|
app = Flask(__name__)
|
|
app.config['UPLOAD_FOLDER'] = 'uploads'
|
|
app.config['MODIFIED_FOLDER'] = 'modified'
|
|
|
|
# Make sure the uploaded and modified folders exist
|
|
os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
|
|
os.makedirs(app.config['MODIFIED_FOLDER'], exist_ok=True)
|
|
|
|
@app.route('/api/v1/pcap/comment', methods=['POST'])
|
|
def upload_file():
|
|
if 'file' not in request.files:
|
|
return "No file part", 400
|
|
|
|
file = request.files['file']
|
|
if file.filename == '':
|
|
return "No selected file", 400
|
|
|
|
if file:
|
|
# Save uploaded files
|
|
filepath = os.path.join(app.config['UPLOAD_FOLDER'], file.filename)
|
|
file.save(filepath)
|
|
|
|
appsketch_api = request.form.get('url', 'default_value')
|
|
pcap_file_id = request.form.get('id', 'default_value')
|
|
|
|
# Use PcapNGFormatAnalys.py
|
|
modified_filepath = modify_file_with_script(filepath, appsketch_api, pcap_file_id)
|
|
|
|
# Delete uploaded and modified files after response
|
|
|
|
@after_this_request
|
|
def remove_file(response):
|
|
try:
|
|
os.remove(filepath) # 删除上传的文件
|
|
os.remove(modified_filepath) # 删除修改后的文件
|
|
except Exception as e:
|
|
app.logger.error(f'Error deleting file: {e}')
|
|
return response
|
|
|
|
# Provide modified file download
|
|
return send_file(modified_filepath, as_attachment=True)
|
|
|
|
def modify_file_with_script(input_filepath, appsketch_api, pcap_file_id):
|
|
modified_filepath = os.path.join(app.config['MODIFIED_FOLDER'], os.path.basename(input_filepath))
|
|
# Use PcapNGFormatAnalys.py to add Remark
|
|
script_path = 'PcapNGFormatAnalys.py'
|
|
subprocess.run(['python3', script_path, input_filepath, modified_filepath, appsketch_api, pcap_file_id])
|
|
return modified_filepath
|
|
|
|
if __name__ == '__main__':
|
|
hostname = socket.gethostname()
|
|
local_ip = socket.gethostbyname(hostname)
|
|
app.run(host=local_ip, port=5000, debug=True)
|