71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
import hmac
|
|
import hashlib
|
|
import sys
|
|
from pathlib import Path
|
|
import requests
|
|
import subprocess
|
|
import base64
|
|
|
|
def load_secret_key(keyb64: str):
|
|
# Decode the base64 encoded key
|
|
decoded_key = base64.b64decode(keyb64)
|
|
|
|
return decoded_key
|
|
|
|
def compute_hmac(file_path, secret_key):
|
|
"""Compute the HMAC of a file using HMAC-SHA256."""
|
|
hmac_obj = hmac.new(secret_key, digestmod=hashlib.sha256)
|
|
with open(file_path, 'rb') as f:
|
|
while chunk := f.read(4096):
|
|
hmac_obj.update(chunk)
|
|
return hmac_obj.hexdigest()
|
|
|
|
def upload_file(url, file_path, secret_key):
|
|
"""Upload a file to the server with an HMAC signature."""
|
|
# Compute HMAC of the file
|
|
signature = compute_hmac(file_path, secret_key)
|
|
print(signature)
|
|
# Prepare headers
|
|
headers = {'X-Signature': signature}
|
|
|
|
# Read file and prepare files dictionary for request
|
|
with open(file_path, 'rb') as f:
|
|
files = {'file': (Path(file_path).name, f)}
|
|
response = requests.post(url, files=files, headers=headers)
|
|
|
|
return response
|
|
|
|
|
|
def get_short_git_commit_hash():
|
|
try:
|
|
# Run the Git command to get the short commit hash
|
|
result = subprocess.run(['git', 'rev-parse', '--short', 'HEAD'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, check=True)
|
|
# Extract and return the commit hash
|
|
return result.stdout.strip()
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Error getting Git commit hash: {e}")
|
|
return None
|
|
|
|
if __name__ == "__main__":
|
|
url = "https://fw.revwal.de/master/upload" # Adjust this to your server's URL
|
|
|
|
|
|
fw_path = "build/CLS_Master.bin" # Path to the file you want to upload
|
|
|
|
# Get the Git commit hash
|
|
commit_hash = get_short_git_commit_hash()
|
|
if commit_hash is None:
|
|
print("Error: Could not get Git commit hash")
|
|
sys.exit(1)
|
|
|
|
# replace the file name with the commit hash
|
|
file_path = fw_path.replace("CLS_Master", commit_hash)
|
|
|
|
# copy the file to the new name
|
|
subprocess.run(['cp', fw_path, file_path])
|
|
|
|
|
|
secret_key = load_secret_key(sys.argv[1])
|
|
|
|
response = upload_file(url, file_path, secret_key)
|
|
print(f"Server response: {response.text}") |