rage: set up fb environment when dumping traces

Reviewed By: chadaustin

Differential Revision: D35325659

fbshipit-source-id: 17bfc262c11aa6f620416c130ac7bbf94011f881
This commit is contained in:
Zeyi (Rice) Fan 2022-04-20 15:11:01 -07:00 committed by Facebook GitHub Bot
parent f556079bca
commit 647c5aacbe

View File

@ -18,7 +18,7 @@ import sys
import traceback
from datetime import datetime, timedelta
from pathlib import Path
from typing import Callable, Generator, IO, List, Tuple, Optional, cast
from typing import Callable, Generator, IO, List, Tuple, Optional, cast, Dict
from . import (
debug as debug_mod,
@ -33,12 +33,15 @@ from . import (
from .config import EdenInstance
try:
from .facebook.rage import find_fb_cdb
from .facebook.rage import find_fb_cdb, setup_fb_env
except ImportError:
def find_fb_cdb() -> Optional[Path]:
return None
def setup_fb_env(env: Dict[str, str]) -> Dict[str, str]:
return env
def section_title(message: str, out: IO[bytes]) -> None:
out.write(util_mod.underlined(message).encode())
@ -486,7 +489,10 @@ def print_cdb_backtrace(pid: int, sink: IO[bytes]) -> None:
]
cdb_cmd += [";".join(debugger_command)]
subprocess.run(cdb_cmd, check=True, stderr=subprocess.STDOUT, stdout=sink)
env = os.environ.copy()
env = setup_fb_env(env)
subprocess.run(cdb_cmd, check=True, stderr=subprocess.STDOUT, stdout=sink, env=env)
def print_sample_trace(pid: int, sink: IO[bytes]) -> None: