mirror of
https://git.tartarus.org/simon/putty.git
synced 2025-01-09 09:27:59 +00:00
c336643576
On a similar theme of separating the query operation from the attempted change, backend_send() now no longer has the side effect of returning the current size of the send buffer. Instead, you have to call backend_sendbuffer() every time you want to know that.
390 lines
10 KiB
C
390 lines
10 KiB
C
/*
|
|
* Backend to run a Windows console session using ConPTY.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <limits.h>
|
|
|
|
#include "putty.h"
|
|
|
|
#include <windows.h>
|
|
#include <consoleapi.h>
|
|
|
|
typedef struct ConPTY ConPTY;
|
|
struct ConPTY {
|
|
HPCON pseudoconsole;
|
|
HANDLE outpipe, inpipe, hprocess;
|
|
struct handle *out, *in;
|
|
HandleWait *subprocess;
|
|
bool exited;
|
|
DWORD exitstatus;
|
|
Seat *seat;
|
|
LogContext *logctx;
|
|
int bufsize;
|
|
Backend backend;
|
|
};
|
|
|
|
static void conpty_terminate(ConPTY *conpty)
|
|
{
|
|
if (conpty->out) {
|
|
handle_free(conpty->out);
|
|
conpty->out = NULL;
|
|
}
|
|
if (conpty->outpipe != INVALID_HANDLE_VALUE) {
|
|
CloseHandle(conpty->outpipe);
|
|
conpty->outpipe = INVALID_HANDLE_VALUE;
|
|
}
|
|
if (conpty->in) {
|
|
handle_free(conpty->in);
|
|
conpty->in = NULL;
|
|
}
|
|
if (conpty->inpipe != INVALID_HANDLE_VALUE) {
|
|
CloseHandle(conpty->inpipe);
|
|
conpty->inpipe = INVALID_HANDLE_VALUE;
|
|
}
|
|
if (conpty->subprocess) {
|
|
delete_handle_wait(conpty->subprocess);
|
|
conpty->subprocess = NULL;
|
|
conpty->hprocess = INVALID_HANDLE_VALUE;
|
|
}
|
|
if (conpty->pseudoconsole != INVALID_HANDLE_VALUE) {
|
|
ClosePseudoConsole(conpty->pseudoconsole);
|
|
conpty->pseudoconsole = INVALID_HANDLE_VALUE;
|
|
}
|
|
}
|
|
|
|
static void conpty_process_wait_callback(void *vctx)
|
|
{
|
|
ConPTY *conpty = (ConPTY *)vctx;
|
|
|
|
if (!GetExitCodeProcess(conpty->hprocess, &conpty->exitstatus))
|
|
return;
|
|
conpty->exited = true;
|
|
|
|
/*
|
|
* We can stop waiting for the process now.
|
|
*/
|
|
if (conpty->subprocess) {
|
|
delete_handle_wait(conpty->subprocess);
|
|
conpty->subprocess = NULL;
|
|
conpty->hprocess = INVALID_HANDLE_VALUE;
|
|
}
|
|
|
|
/*
|
|
* Once the contained process exits, close the pseudo-console as
|
|
* well. But don't close the pipes yet, since apparently
|
|
* ClosePseudoConsole can trigger a final bout of terminal output
|
|
* as things clean themselves up.
|
|
*/
|
|
if (conpty->pseudoconsole != INVALID_HANDLE_VALUE) {
|
|
ClosePseudoConsole(conpty->pseudoconsole);
|
|
conpty->pseudoconsole = INVALID_HANDLE_VALUE;
|
|
}
|
|
}
|
|
|
|
static size_t conpty_gotdata(
|
|
struct handle *h, const void *data, size_t len, int err)
|
|
{
|
|
ConPTY *conpty = (ConPTY *)handle_get_privdata(h);
|
|
if (err || len == 0) {
|
|
char *error_msg;
|
|
|
|
conpty_terminate(conpty);
|
|
|
|
seat_notify_remote_exit(conpty->seat);
|
|
|
|
if (!err && conpty->exited) {
|
|
/*
|
|
* The clean-exit case: our subprocess terminated, we
|
|
* deleted the PseudoConsole ourself, and now we got the
|
|
* expected EOF on the pipe.
|
|
*/
|
|
return 0;
|
|
}
|
|
|
|
if (err)
|
|
error_msg = dupprintf("Error reading from console pty: %s",
|
|
win_strerror(err));
|
|
else
|
|
error_msg = dupprintf(
|
|
"Unexpected end of file reading from console pty");
|
|
|
|
logevent(conpty->logctx, error_msg);
|
|
seat_connection_fatal(conpty->seat, "%s", error_msg);
|
|
sfree(error_msg);
|
|
|
|
return 0;
|
|
} else {
|
|
return seat_stdout(conpty->seat, data, len);
|
|
}
|
|
}
|
|
|
|
static void conpty_sentdata(struct handle *h, size_t new_backlog, int err)
|
|
{
|
|
ConPTY *conpty = (ConPTY *)handle_get_privdata(h);
|
|
if (err) {
|
|
const char *error_msg = "Error writing to conpty device";
|
|
|
|
conpty_terminate(conpty);
|
|
|
|
seat_notify_remote_exit(conpty->seat);
|
|
|
|
logevent(conpty->logctx, error_msg);
|
|
|
|
seat_connection_fatal(conpty->seat, "%s", error_msg);
|
|
} else {
|
|
conpty->bufsize = new_backlog;
|
|
}
|
|
}
|
|
|
|
static char *conpty_init(const BackendVtable *vt, Seat *seat,
|
|
Backend **backend_handle, LogContext *logctx,
|
|
Conf *conf, const char *host, int port,
|
|
char **realhost, bool nodelay, bool keepalive)
|
|
{
|
|
ConPTY *conpty;
|
|
char *err = NULL;
|
|
|
|
HANDLE in_r = INVALID_HANDLE_VALUE;
|
|
HANDLE in_w = INVALID_HANDLE_VALUE;
|
|
HANDLE out_r = INVALID_HANDLE_VALUE;
|
|
HANDLE out_w = INVALID_HANDLE_VALUE;
|
|
|
|
HPCON pcon;
|
|
bool pcon_needs_cleanup = false;
|
|
|
|
STARTUPINFOEX si;
|
|
memset(&si, 0, sizeof(si));
|
|
|
|
if (!CreatePipe(&in_r, &in_w, NULL, 0)) {
|
|
err = dupprintf("CreatePipe: %s", win_strerror(GetLastError()));
|
|
goto out;
|
|
}
|
|
if (!CreatePipe(&out_r, &out_w, NULL, 0)) {
|
|
err = dupprintf("CreatePipe: %s", win_strerror(GetLastError()));
|
|
goto out;
|
|
}
|
|
|
|
COORD size;
|
|
size.X = conf_get_int(conf, CONF_width);
|
|
size.Y = conf_get_int(conf, CONF_height);
|
|
|
|
HRESULT result = CreatePseudoConsole(size, in_r, out_w, 0, &pcon);
|
|
if (FAILED(result)) {
|
|
if (HRESULT_FACILITY(result) == FACILITY_WIN32)
|
|
err = dupprintf("CreatePseudoConsole: %s",
|
|
win_strerror(HRESULT_CODE(result)));
|
|
else
|
|
err = dupprintf("CreatePseudoConsole failed: HRESULT=0x%08x",
|
|
(unsigned)result);
|
|
goto out;
|
|
}
|
|
pcon_needs_cleanup = true;
|
|
|
|
CloseHandle(in_r);
|
|
in_r = INVALID_HANDLE_VALUE;
|
|
CloseHandle(out_w);
|
|
out_w = INVALID_HANDLE_VALUE;
|
|
|
|
si.StartupInfo.cb = sizeof(si);
|
|
|
|
size_t attrsize = 0;
|
|
InitializeProcThreadAttributeList(NULL, 1, 0, &attrsize);
|
|
si.lpAttributeList = smalloc(attrsize);
|
|
if (!InitializeProcThreadAttributeList(
|
|
si.lpAttributeList, 1, 0, &attrsize)) {
|
|
err = dupprintf("InitializeProcThreadAttributeList: %s",
|
|
win_strerror(GetLastError()));
|
|
goto out;
|
|
}
|
|
if (!UpdateProcThreadAttribute(
|
|
si.lpAttributeList,
|
|
0, PROC_THREAD_ATTRIBUTE_PSEUDOCONSOLE,
|
|
pcon, sizeof(pcon), NULL, NULL)) {
|
|
err = dupprintf("UpdateProcThreadAttribute: %s",
|
|
win_strerror(GetLastError()));
|
|
goto out;
|
|
}
|
|
|
|
PROCESS_INFORMATION pi;
|
|
memset(&pi, 0, sizeof(pi));
|
|
|
|
char *command;
|
|
const char *conf_cmd = conf_get_str(conf, CONF_remote_cmd);
|
|
if (*conf_cmd) {
|
|
command = dupstr(conf_cmd);
|
|
} else {
|
|
command = dupcat(get_system_dir(), "\\cmd.exe");
|
|
}
|
|
bool created_ok = CreateProcess(NULL, command, NULL, NULL,
|
|
false, EXTENDED_STARTUPINFO_PRESENT,
|
|
NULL, NULL, &si.StartupInfo, &pi);
|
|
sfree(command);
|
|
if (!created_ok) {
|
|
err = dupprintf("CreateProcess: %s",
|
|
win_strerror(GetLastError()));
|
|
goto out;
|
|
}
|
|
|
|
/* No local authentication phase in this protocol */
|
|
seat_set_trust_status(seat, false);
|
|
|
|
conpty = snew(ConPTY);
|
|
conpty->pseudoconsole = pcon;
|
|
pcon_needs_cleanup = false;
|
|
conpty->outpipe = in_w;
|
|
conpty->out = handle_output_new(in_w, conpty_sentdata, conpty, 0);
|
|
in_w = INVALID_HANDLE_VALUE;
|
|
conpty->inpipe = out_r;
|
|
conpty->in = handle_input_new(out_r, conpty_gotdata, conpty, 0);
|
|
out_r = INVALID_HANDLE_VALUE;
|
|
conpty->subprocess = add_handle_wait(
|
|
pi.hProcess, conpty_process_wait_callback, conpty);
|
|
conpty->hprocess = pi.hProcess;
|
|
CloseHandle(pi.hThread);
|
|
conpty->exited = false;
|
|
conpty->exitstatus = 0;
|
|
conpty->bufsize = 0;
|
|
conpty->backend.vt = vt;
|
|
*backend_handle = &conpty->backend;
|
|
|
|
conpty->seat = seat;
|
|
conpty->logctx = logctx;
|
|
|
|
*realhost = dupstr("");
|
|
|
|
/*
|
|
* Specials are always available.
|
|
*/
|
|
seat_update_specials_menu(conpty->seat);
|
|
|
|
out:
|
|
if (in_r != INVALID_HANDLE_VALUE)
|
|
CloseHandle(in_r);
|
|
if (in_w != INVALID_HANDLE_VALUE)
|
|
CloseHandle(in_w);
|
|
if (out_r != INVALID_HANDLE_VALUE)
|
|
CloseHandle(out_r);
|
|
if (out_w != INVALID_HANDLE_VALUE)
|
|
CloseHandle(out_w);
|
|
if (pcon_needs_cleanup)
|
|
ClosePseudoConsole(pcon);
|
|
sfree(si.lpAttributeList);
|
|
return err;
|
|
}
|
|
|
|
static void conpty_free(Backend *be)
|
|
{
|
|
ConPTY *conpty = container_of(be, ConPTY, backend);
|
|
|
|
conpty_terminate(conpty);
|
|
expire_timer_context(conpty);
|
|
sfree(conpty);
|
|
}
|
|
|
|
static void conpty_reconfig(Backend *be, Conf *conf)
|
|
{
|
|
}
|
|
|
|
static void conpty_send(Backend *be, const char *buf, size_t len)
|
|
{
|
|
ConPTY *conpty = container_of(be, ConPTY, backend);
|
|
|
|
if (conpty->out == NULL)
|
|
return;
|
|
|
|
conpty->bufsize = handle_write(conpty->out, buf, len);
|
|
}
|
|
|
|
static size_t conpty_sendbuffer(Backend *be)
|
|
{
|
|
ConPTY *conpty = container_of(be, ConPTY, backend);
|
|
return conpty->bufsize;
|
|
}
|
|
|
|
static void conpty_size(Backend *be, int width, int height)
|
|
{
|
|
ConPTY *conpty = container_of(be, ConPTY, backend);
|
|
COORD size;
|
|
size.X = width;
|
|
size.Y = height;
|
|
ResizePseudoConsole(conpty->pseudoconsole, size);
|
|
}
|
|
|
|
static void conpty_special(Backend *be, SessionSpecialCode code, int arg)
|
|
{
|
|
}
|
|
|
|
static const SessionSpecial *conpty_get_specials(Backend *be)
|
|
{
|
|
static const SessionSpecial specials[] = {
|
|
{NULL, SS_EXITMENU}
|
|
};
|
|
return specials;
|
|
}
|
|
|
|
static bool conpty_connected(Backend *be)
|
|
{
|
|
return true; /* always connected */
|
|
}
|
|
|
|
static bool conpty_sendok(Backend *be)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
static void conpty_unthrottle(Backend *be, size_t backlog)
|
|
{
|
|
ConPTY *conpty = container_of(be, ConPTY, backend);
|
|
if (conpty->in)
|
|
handle_unthrottle(conpty->in, backlog);
|
|
}
|
|
|
|
static bool conpty_ldisc(Backend *be, int option)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
static void conpty_provide_ldisc(Backend *be, Ldisc *ldisc)
|
|
{
|
|
}
|
|
|
|
static int conpty_exitcode(Backend *be)
|
|
{
|
|
ConPTY *conpty = container_of(be, ConPTY, backend);
|
|
|
|
if (conpty->exited &&
|
|
0 <= conpty->exitstatus &&
|
|
conpty->exitstatus <= INT_MAX)
|
|
return conpty->exitstatus;
|
|
else
|
|
return -1;
|
|
}
|
|
|
|
static int conpty_cfg_info(Backend *be)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
const BackendVtable conpty_backend = {
|
|
.init = conpty_init,
|
|
.free = conpty_free,
|
|
.reconfig = conpty_reconfig,
|
|
.send = conpty_send,
|
|
.sendbuffer = conpty_sendbuffer,
|
|
.size = conpty_size,
|
|
.special = conpty_special,
|
|
.get_specials = conpty_get_specials,
|
|
.connected = conpty_connected,
|
|
.exitcode = conpty_exitcode,
|
|
.sendok = conpty_sendok,
|
|
.ldisc_option_state = conpty_ldisc,
|
|
.provide_ldisc = conpty_provide_ldisc,
|
|
.unthrottle = conpty_unthrottle,
|
|
.cfg_info = conpty_cfg_info,
|
|
.id = "conpty",
|
|
.displayname = "ConPTY",
|
|
.protocol = -1,
|
|
};
|