Merge pull request #169 from exfinen/update-c

Introduce file-level lock for pem file exchange

Author: exfinen

config-docker-compose.py

@@ -22,7 +22,7 @@ def gen_env_consumer_api(
 """
   return output
 
-def gen_env_coord(
+def gen_env_coord(
   transport: str,
   party_hosts: list[str],
   party_ports: list[int],

gen-self-signed-cert.sh

@@ -9,6 +9,7 @@
 import asyncio
 import os
 import glob
+from filelock import FileLock
 
 import requests
 from fastapi import APIRouter, Depends, HTTPException
@@ -464,7 +465,11 @@ def get_party_cert(host: str, port: int, party_id: int):
             logger.error(f"party_id mismatch, expected {party_id}, got {data['party_id']}")
             raise HTTPException(status_code=500, detail=f"Party ID mismatch, expected {party_id}, got {data['party_id']}")
         logger.info(f"Fetched party cert from {host}:{port}")
-        (CERTS_PATH / f"P{party_id}.pem").write_text(data["cert_file"])
+        pem_file = CERTS_PATH / f"P{party_id}.pem"
+        lock_file = f"{pem_file}.lock"
+        with FileLock(lock_file):
+            pem_file.write_text(data["cert_file"])
+
         logger.info(f"Saved party cert to {CERTS_PATH / f'P{party_id}.pem'}")
 
     with ThreadPoolExecutor() as executor:

mpc_demo_infra/computation_party_server/routes.py

@@ -6,7 +6,7 @@ authors = ["mhchia <[email protected]>"]
 readme = "README.md"
 
 [tool.poetry.dependencies]
-python = ">=3.8,<3.14"
+python = ">=3.9,<3.14"
 fastapi = "^0.95.0"
 uvicorn = "^0.22.0"
 sqlalchemy = "^1.4.0"
@@ -17,6 +17,7 @@ aiohttp = "^3.10.9"
 slowapi = "^0.1.9"
 readerwriterlock = "^1.0.9"
 
+filelock = "^3.17.0"
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.2.0"
 pytest-asyncio = "^0.24.0"