from flask import Flask, request, jsonify, send_file, make_response
import os
import json
import uuid
import time
import threading
import logging
from main import get_symbol_address, patch_address, copy_file_to_src, zip_src_files
app = Flask(__name__)
PATCHED_LIBRARIES = {}
PERMALINK_EXPIRY = 600 # 10 minutes
PATCHES_JSON = 'patches.json'
# Configure logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger()
def save_patch_info(permalink_id, file_path):
patch_info = {
'permalink_id': permalink_id,
'file_path': file_path,
'timestamp': time.time()
}
if os.path.exists(PATCHES_JSON):
with open(PATCHES_JSON, 'r') as f:
patches = json.load(f)
else:
patches = []
patches.append(patch_info)
with open(PATCHES_JSON, 'w') as f:
json.dump(patches, f, indent=4)
@app.route('/')
def index():
return '''
Library Patcher
Upload a library to patch
'''
@app.route('/patch', methods=['POST'])
def patch():
if 'file' not in request.files:
return jsonify({"error": "No file part"}), 400
file = request.files['file']
if file.filename == '':
return jsonify({"error": "No selected file"}), 400
if not file.filename.endswith('.so'):
return jsonify({"error": "Invalid file type. Only .so files are allowed."}), 400
file_path = os.path.join('uploads', file.filename)
file.save(file_path)
# Patch the file
try:
l2c_fcr_chk_chan_modes_address = get_symbol_address(file_path, "l2c_fcr_chk_chan_modes")
patch_address(file_path, l2c_fcr_chk_chan_modes_address, "20008052c0035fd6")
l2cu_send_peer_info_req_address = get_symbol_address(file_path, "l2cu_send_peer_info_req")
patch_address(file_path, l2cu_send_peer_info_req_address, "c0035fd6")
except Exception as e:
logger.error(f"Error patching file: {str(e)}")
return jsonify({"error": f"Error patching file: {str(e)}"}), 500
# Create permalink
permalink_id = str(uuid.uuid4())
PATCHED_LIBRARIES[permalink_id] = {
'file_path': file_path,
'timestamp': time.time()
}
# Save patch info
save_patch_info(permalink_id, file_path)
# Schedule deletion
threading.Timer(PERMALINK_EXPIRY, delete_expired_permalink, args=[permalink_id]).start()
return jsonify({'permalink': f'/download/{permalink_id}'})
@app.route('/download/', methods=['GET'])
def download(permalink_id):
if permalink_id not in PATCHED_LIBRARIES:
return "Permalink expired or invalid", 404
file_path = PATCHED_LIBRARIES[permalink_id]['file_path']
if not os.path.exists(file_path):
return "File not found", 404
try:
copy_file_to_src(file_path)
zip_src_files()
except Exception as e:
logger.error(f"Error preparing download: {str(e)}")
return f"Error preparing download: {str(e)}", 500
resp = make_response(send_file('btl2capfix.zip', as_attachment=True))
resp.headers['Content-Disposition'] = f'attachment; filename=btl2capfix.zip'
return resp
def delete_expired_permalink(permalink_id):
if permalink_id in PATCHED_LIBRARIES:
if os.path.exists(PATCHED_LIBRARIES[permalink_id]['file_path']):
os.remove(PATCHED_LIBRARIES[permalink_id]['file_path'])
del PATCHED_LIBRARIES[permalink_id]
if not os.path.exists('uploads'):
os.makedirs('uploads')
if __name__ == '__main__':
app.run(debug=True, host='0.0.0.0', port=8080)