diff --git a/libagent/ssh/__init__.py b/libagent/ssh/__init__.py index ac54940..674d22a 100644 --- a/libagent/ssh/__init__.py +++ b/libagent/ssh/__init__.py @@ -88,6 +88,8 @@ def create_agent_parser(device_type): g = p.add_mutually_exclusive_group() g.add_argument('-d', '--daemonize', default=False, action='store_true', help='Daemonize the agent and print its UNIX socket path') + g.add_argument('-f', '--foreground', default=False, action='store_true', + help='Run agent in foreground with specified UNIX socket path') g.add_argument('-s', '--shell', default=False, action='store_true', help=('run ${SHELL} as subprocess under SSH agent, allowing ' 'regular SSH-based tools to be used in the shell')) @@ -234,7 +236,11 @@ def main(device_type): sock_path = args.sock_path if not sock_path: - sock_path = tempfile.mktemp(prefix='trezor-ssh-agent-') + if args.foreground: + log.error("if running in foreground you must specify the socket path") + return 1 + else: + sock_path = tempfile.mktemp(prefix='trezor-ssh-agent-') command = args.command context = _dummy_context() @@ -248,6 +254,8 @@ def main(device_type): sys.stdout.flush() context = daemon.DaemonContext() log.info('running the agent as a daemon on %s', sock_path) + elif args.foreground: + log.info('running the agent on %s', sock_path) use_shell = bool(args.shell) if use_shell: @@ -258,7 +266,7 @@ def main(device_type): conn_factory=lambda: client.Client(device_type()), identities=identities, public_keys=public_keys) - if command or args.daemonize: + if command or args.daemonize or args.foreground: with context: return run_server(conn=conn, command=command, sock_path=sock_path, debug=args.debug, timeout=args.timeout)