1
0
mirror of https://git.tartarus.org/simon/putty.git synced 2025-01-09 01:18:00 +00:00

New facility, platform_start_subprocess.

We already have the ability to start a subprocess and hook it up to a
Socket, for running local proxy commands. Now the same facility is
available as an auxiliary feature, so that a backend can start another
subcommand for a different purpose, and make a separate Socket to
communicate with it.

Just like the local proxy system, this facility captures the
subprocess's stderr, and passes it back to the caller via plug_log. To
make that not look silly, I had to add a system where the "proxy:"
prefix on the usual plug_log messages is reconfigurable, and when you
call platform_start_subprocess(), you get to pass the prefix you want
to use in this case.
This commit is contained in:
Simon Tatham 2022-08-22 18:46:32 +01:00
parent a92aeca111
commit eec350c38b
11 changed files with 96 additions and 2 deletions

View File

@ -397,8 +397,10 @@ void backend_socket_log(Seat *seat, LogContext *logctx,
typedef struct ProxyStderrBuf {
char buf[8192];
size_t size;
const char *prefix; /* must be statically allocated */
} ProxyStderrBuf;
void psb_init(ProxyStderrBuf *psb);
void psb_set_prefix(ProxyStderrBuf *psb, const char *prefix);
void log_proxy_stderr(
Plug *plug, ProxyStderrBuf *psb, const void *vdata, size_t len);
@ -429,4 +431,6 @@ struct DeferredSocketOpenerVtable {
static inline void deferred_socket_opener_free(DeferredSocketOpener *dso)
{ dso->vt->free(dso); }
DeferredSocketOpener *null_deferred_socket_opener(void);
#endif

10
putty.h
View File

@ -2893,6 +2893,16 @@ typedef void (*toplevel_callback_notify_fn_t)(void *ctx);
void request_callback_notifications(toplevel_callback_notify_fn_t notify,
void *ctx);
/*
* Facility provided by the platform to spawn a parallel subprocess
* and present its stdio via a Socket.
*
* 'prefix' indicates the prefix that should appear on messages passed
* to plug_log to provide stderr output from the process.
*/
Socket *platform_start_subprocess(const char *cmd, Plug *plug,
const char *prefix);
/*
* Define no-op macros for the jump list functions, on platforms that
* don't support them. (This is a bit of a hack, and it'd be nicer to

View File

@ -26,5 +26,6 @@ add_sources_from_current_dir(utils
null-cipher.c
null-key.c
null-mac.c
null-opener.c
null-plug.c
null-seat.c)

20
stubs/null-opener.c Normal file
View File

@ -0,0 +1,20 @@
/*
* Null implementation of DeferredSocketOpener. Doesn't even bother to
* allocate and free itself: there's just one static implementation
* which we hand out to any caller.
*/
#include "putty.h"
static void null_opener_free(DeferredSocketOpener *opener) {}
static const DeferredSocketOpenerVtable NullOpener_vt = {
.free = null_opener_free,
};
static DeferredSocketOpener null_opener = { .vt = &NullOpener_vt };
DeferredSocketOpener *null_deferred_socket_opener(void)
{
return &null_opener;
}

View File

@ -371,6 +371,13 @@ void setup_fd_socket(Socket *s, int infd, int outfd, int inerrfd)
queue_toplevel_callback(fdsocket_connect_success_callback, fds);
}
void fd_socket_set_psb_prefix(Socket *s, const char *prefix)
{
FdSocket *fds = container_of(s, FdSocket, sock);
assert(fds->sock.vt == &FdSocket_sockvt);
psb_set_prefix(&fds->psb, prefix);
}
static FdSocket *make_fd_socket_internal(SockAddr *addr, int port, Plug *plug)
{
FdSocket *fds;

View File

@ -96,3 +96,21 @@ Socket *platform_new_connection(SockAddr *addr, const char *hostname,
return NULL;
}
}
Socket *platform_start_subprocess(const char *cmd, Plug *plug,
const char *prefix)
{
Socket *socket = make_deferred_fd_socket(
null_deferred_socket_opener(),
sk_nonamelookup("<local command>"), 0, plug);
char *err = platform_setup_local_proxy(socket, cmd);
fd_socket_set_psb_prefix(socket, prefix);
if (err) {
sk_close(socket);
socket = new_error_socket_fmt(plug, "%s", err);
sfree(err);
}
return socket;
}

View File

@ -389,6 +389,7 @@ Socket *make_fd_socket(int infd, int outfd, int inerrfd,
Socket *make_deferred_fd_socket(DeferredSocketOpener *opener,
SockAddr *addr, int port, Plug *plug);
void setup_fd_socket(Socket *s, int infd, int outfd, int inerrfd);
void fd_socket_set_psb_prefix(Socket *s, const char *prefix);
/*
* Default font setting, which can vary depending on NOT_X_WINDOWS.

View File

@ -7,6 +7,12 @@
void psb_init(ProxyStderrBuf *psb)
{
psb->size = 0;
psb->prefix = "proxy";
}
void psb_set_prefix(ProxyStderrBuf *psb, const char *prefix)
{
psb->prefix = prefix;
}
void log_proxy_stderr(Plug *plug, ProxyStderrBuf *psb,
@ -58,7 +64,7 @@ void log_proxy_stderr(Plug *plug, ProxyStderrBuf *psb,
psb->buf[endpos-1] == '\r'))
endpos--;
char *msg = dupprintf(
"proxy: %.*s", (int)(endpos - pos), psb->buf + pos);
"%s: %.*s", psb->prefix, (int)(endpos - pos), psb->buf + pos);
plug_log(plug, PLUGLOG_PROXY_MSG, NULL, 0, msg, 0);
sfree(msg);
@ -73,7 +79,8 @@ void log_proxy_stderr(Plug *plug, ProxyStderrBuf *psb,
*/
if (pos == 0 && psb->size == lenof(psb->buf)) {
char *msg = dupprintf(
"proxy (partial line): %.*s", (int)psb->size, psb->buf);
"%s (partial line): %.*s", psb->prefix, (int)psb->size,
psb->buf);
plug_log(plug, PLUGLOG_PROXY_MSG, NULL, 0, msg, 0);
sfree(msg);

View File

@ -388,6 +388,13 @@ Socket *make_handle_socket(HANDLE send_H, HANDLE recv_H, HANDLE stderr_H,
return &hs->sock;
}
void handle_socket_set_psb_prefix(Socket *s, const char *prefix)
{
HandleSocket *hs = container_of(s, HandleSocket, sock);
assert(hs->sock.vt == &HandleSocket_sockvt);
psb_set_prefix(&hs->psb, prefix);
}
static void sk_handle_deferred_close(Socket *s)
{
HandleSocket *hs = container_of(s, HandleSocket, sock);

View File

@ -98,3 +98,21 @@ Socket *platform_new_connection(SockAddr *addr, const char *hostname,
local_proxy_opener_set_socket(opener, socket);
return socket;
}
Socket *platform_start_subprocess(const char *cmd, Plug *plug,
const char *prefix)
{
Socket *socket = make_deferred_handle_socket(
null_deferred_socket_opener(),
sk_nonamelookup("<local command>"), 0, plug);
char *err = platform_setup_local_proxy(socket, cmd);
handle_socket_set_psb_prefix(socket, prefix);
if (err) {
sk_close(socket);
socket = new_error_socket_fmt(plug, "%s", err);
sfree(err);
}
return socket;
}

View File

@ -354,6 +354,7 @@ Socket *make_deferred_handle_socket(DeferredSocketOpener *opener,
SockAddr *addr, int port, Plug *plug);
void setup_handle_socket(Socket *s, HANDLE send_H, HANDLE recv_H,
HANDLE stderr_H, bool overlapped);
void handle_socket_set_psb_prefix(Socket *s, const char *prefix);
Socket *new_named_pipe_client(const char *pipename, Plug *plug); /* winnpc */
Socket *new_named_pipe_listener(const char *pipename, Plug *plug); /* winnps */