clair/CVM-Sentry
1
0
Fork 0
Code Activity

changes changes

Browse Source
This commit is contained in:
Clair Delafuente
2025-10-14 13:28:59 -04:00
parent 167ac1858b
commit 299aa4e0b1
1 changed files with 14 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
cvmsentry.py
View File

@@ -58,7 +58,7 @@ async def periodic_snapshot_task():
log.info("Starting periodic snapshot task") log.info("Starting periodic snapshot task")
while True: while True:
try: try:
await asyncio.sleep(5) await asyncio.sleep(config.snapshot_cadence)
log.debug("Running periodic framebuffer snapshot capture...") log.debug("Running periodic framebuffer snapshot capture...")
save_tasks = [] save_tasks = []
@@ -77,23 +77,23 @@ async def periodic_snapshot_task():
filename = f"{epoch_timestamp}.webp" filename = f"{epoch_timestamp}.webp"
filepath = os.path.join(snapshot_dir, filename) filepath = os.path.join(snapshot_dir, filename)
# Create a hash of the framebuffer for comparison # Get framebuffer reference (no copy needed)
framebuffer = vm_data["framebuffer"] framebuffer = vm_data["framebuffer"]
if not framebuffer: if not framebuffer:
continue continue
# Calculate difference hash for the image # Calculate difference hash asynchronously to avoid blocking
current_hash = str(imagehash.dhash(framebuffer)) current_hash = await asyncio.to_thread(
lambda: str(imagehash.dhash(framebuffer))
)
# Only save if the framebuffer has changed since last snapshot # Only save if the framebuffer has changed since last snapshot
if current_hash != vm_data.get("last_frame_hash"): if current_hash != vm_data.get("last_frame_hash"):
# Create a copy of the image to avoid race conditions # Pass framebuffer directly without copying
img_copy = framebuffer.copy()
# Create and store task for asynchronous saving
save_tasks.append( save_tasks.append(
asyncio.create_task( asyncio.create_task(
save_image_async( save_image_async(
img_copy, filepath, vm_name, vm_data, current_hash framebuffer, filepath, vm_name, vm_data, current_hash
) )
) )
) )
@@ -349,14 +349,11 @@ async def connect(vm_name: str):
"RGB", (expected_width, expected_height) "RGB", (expected_width, expected_height)
) )
# Only update the portion that was received # Only update the portion that was received - modify in place
if vms[vm_name]["framebuffer"]: if vms[vm_name]["framebuffer"]:
# Create a copy of the current framebuffer to modify # Paste directly onto existing framebuffer
updated_img = vms[vm_name]["framebuffer"].copy() vms[vm_name]["framebuffer"].paste(frame_img, (0, 0))
# Paste the new partial frame at position (0,0) frame_img = vms[vm_name]["framebuffer"]
updated_img.paste(frame_img, (0, 0))
# Use this as our new framebuffer
frame_img = updated_img
# Update the framebuffer with the new image # Update the framebuffer with the new image
vms[vm_name]["framebuffer"] = frame_img vms[vm_name]["framebuffer"] = frame_img
@@ -385,12 +382,8 @@ async def connect(vm_name: str):
# If we have a valid framebuffer, update it with the fragment # If we have a valid framebuffer, update it with the fragment
if vms[vm_name]["framebuffer"]: if vms[vm_name]["framebuffer"]:
# Create a copy to modify # Paste directly onto existing framebuffer (no copy needed)
updated_img = vms[vm_name]["framebuffer"].copy() vms[vm_name]["framebuffer"].paste(fragment_img, (x, y))
# Paste the fragment at the specified position
updated_img.paste(fragment_img, (x, y))
# Update the framebuffer
vms[vm_name]["framebuffer"] = updated_img
log.debug( log.debug(
f"({STATE.name} - {vm_name}) Updated framebuffer with fragment at ({x}, {y}), fragment size: {fragment_img.size}" f"({STATE.name} - {vm_name}) Updated framebuffer with fragment at ({x}, {y}), fragment size: {fragment_img.size}"
) )