mirror of
https://github.com/lihop/godot-xterm.git
synced 2025-06-29 18:55:31 +02:00
added windows pty support
This commit is contained in:
Alexander Treml
Leroy Hopson
parent
77ae56f72d
commit
bd399e5603
23 changed files with 1478 additions and 695 deletions
|
|
@ -12,17 +12,20 @@
|
|||
#if (defined(__linux__) || defined(__APPLE__)) && !defined(_PTY_DISABLED)
|
||||
#include "pty_unix.h"
|
||||
#include <unistd.h>
|
||||
#elif defined(_WIN32) && !defined(_PTY_DISABLED)
|
||||
#include "pty_win.h"
|
||||
#include <io.h>
|
||||
#endif
|
||||
|
||||
// Require buffer to be flushed after reaching this size.
|
||||
#define BUFFER_LIMIT 1048576 // 1MB
|
||||
|
||||
#define UV_ERR_MSG(uv_err) \
|
||||
#define UV_ERR_MSG(uv_err) \
|
||||
String(uv_err_name(uv_err)) + String(": ") + String(uv_strerror(uv_err))
|
||||
|
||||
#define ERR_FAIL_UV_ERR(uv_err) \
|
||||
ERR_FAIL_COND_V_MSG(uv_err < 0, PTY::uv_err_to_godot_err(uv_err), \
|
||||
UV_ERR_MSG(uv_err))
|
||||
#define ERR_FAIL_UV_ERR(uv_err) \
|
||||
ERR_FAIL_COND_V_MSG(uv_err < 0, PTY::uv_err_to_godot_err(uv_err), \
|
||||
UV_ERR_MSG(uv_err))
|
||||
|
||||
using namespace godot;
|
||||
|
||||
|
|
@ -30,60 +33,62 @@ void _alloc_buffer(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf);
|
|||
void _write_cb(uv_write_t *req, int status) { std::free(req); }
|
||||
void _close_cb(uv_handle_t *handle) { /* no-op */ };
|
||||
|
||||
void PTY::_bind_methods() {
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGHUP);
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGINT);
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGQUIT);
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGILL);
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGTRAP);
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGABRT);
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGFPE);
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGKILL);
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGSEGV);
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGPIPE);
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGALRM);
|
||||
BIND_ENUM_CONSTANT(SIGNAL_SIGTERM);
|
||||
void PTY::_bind_methods()
|
||||
{
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGHUP);
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGINT);
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGQUIT);
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGILL);
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGTRAP);
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGABRT);
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGFPE);
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGKILL);
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGSEGV);
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGPIPE);
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGALRM);
|
||||
BIND_ENUM_CONSTANT(IPCSIGNAL_SIGTERM);
|
||||
|
||||
ADD_SIGNAL(MethodInfo("data_received", PropertyInfo(Variant::PACKED_BYTE_ARRAY, "data")));
|
||||
ADD_SIGNAL(MethodInfo("exited", PropertyInfo(Variant::INT, "exit_code"), PropertyInfo(Variant::INT, "signal_code")));
|
||||
|
||||
ClassDB::bind_method(D_METHOD("set_cols", "num_cols"), &PTY::set_cols);
|
||||
ClassDB::bind_method(D_METHOD("get_cols"), &PTY::get_cols);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::INT, "cols"), "set_cols", "get_cols");
|
||||
ClassDB::bind_method(D_METHOD("set_cols", "num_cols"), &PTY::set_cols);
|
||||
ClassDB::bind_method(D_METHOD("get_cols"), &PTY::get_cols);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::INT, "cols"), "set_cols", "get_cols");
|
||||
|
||||
ClassDB::bind_method(D_METHOD("set_rows", "num_rows"), &PTY::set_rows);
|
||||
ClassDB::bind_method(D_METHOD("get_rows"), &PTY::get_rows);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::INT, "rows"), "set_rows", "get_rows");
|
||||
ClassDB::bind_method(D_METHOD("set_rows", "num_rows"), &PTY::set_rows);
|
||||
ClassDB::bind_method(D_METHOD("get_rows"), &PTY::get_rows);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::INT, "rows"), "set_rows", "get_rows");
|
||||
|
||||
ClassDB::bind_method(D_METHOD("get_env"), &PTY::get_env);
|
||||
ClassDB::bind_method(D_METHOD("set_env", "env"), &PTY::set_env);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::DICTIONARY, "env"), "set_env", "get_env");
|
||||
ClassDB::bind_method(D_METHOD("get_env"), &PTY::get_env);
|
||||
ClassDB::bind_method(D_METHOD("set_env", "env"), &PTY::set_env);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::DICTIONARY, "env"), "set_env", "get_env");
|
||||
|
||||
ClassDB::bind_method(D_METHOD("get_use_os_env"), &PTY::get_use_os_env);
|
||||
ClassDB::bind_method(D_METHOD("set_use_os_env", "use_os_env"), &PTY::set_use_os_env);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::BOOL, "use_os_env"), "set_use_os_env", "get_use_os_env");
|
||||
ClassDB::bind_method(D_METHOD("get_use_os_env"), &PTY::get_use_os_env);
|
||||
ClassDB::bind_method(D_METHOD("set_use_os_env", "use_os_env"), &PTY::set_use_os_env);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::BOOL, "use_os_env"), "set_use_os_env", "get_use_os_env");
|
||||
|
||||
ClassDB::bind_method(D_METHOD("set_use_threads", "enabled"), &PTY::set_use_threads);
|
||||
ClassDB::bind_method(D_METHOD("is_using_threads"), &PTY::is_using_threads);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::BOOL, "use_threads"), "set_use_threads", "is_using_threads");
|
||||
ClassDB::bind_method(D_METHOD("set_use_threads", "enabled"), &PTY::set_use_threads);
|
||||
ClassDB::bind_method(D_METHOD("is_using_threads"), &PTY::is_using_threads);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::BOOL, "use_threads"), "set_use_threads", "is_using_threads");
|
||||
|
||||
ClassDB::bind_method(D_METHOD("set_terminal_path", "path"), &PTY::set_terminal_path);
|
||||
ClassDB::bind_method(D_METHOD("get_terminal_path"), &PTY::get_terminal_path);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::NODE_PATH, "terminal_path", PropertyHint::PROPERTY_HINT_NODE_PATH_VALID_TYPES, "Terminal"), "set_terminal_path", "get_terminal_path");
|
||||
ClassDB::bind_method(D_METHOD("set_terminal_path", "path"), &PTY::set_terminal_path);
|
||||
ClassDB::bind_method(D_METHOD("get_terminal_path"), &PTY::get_terminal_path);
|
||||
ClassDB::add_property("PTY", PropertyInfo(Variant::NODE_PATH, "terminal_path", PropertyHint::PROPERTY_HINT_NODE_PATH_VALID_TYPES, "Terminal"), "set_terminal_path", "get_terminal_path");
|
||||
|
||||
ClassDB::bind_method(D_METHOD("get_pts_name"), &PTY::get_pts_name);
|
||||
ClassDB::bind_method(D_METHOD("get_pts_name"), &PTY::get_pts_name);
|
||||
|
||||
ClassDB::bind_method(D_METHOD("fork", "file", "args", "cwd", "cols", "rows"), &PTY::fork, DEFVAL(""), DEFVAL(PackedStringArray()), DEFVAL("."), DEFVAL(80), DEFVAL(24));
|
||||
ClassDB::bind_method(D_METHOD("open", "cols", "rows"), &PTY::open, DEFVAL(80), DEFVAL(24));
|
||||
ClassDB::bind_method(D_METHOD("write", "data"), &PTY::write);
|
||||
ClassDB::bind_method(D_METHOD("resize", "cols", "rows"), &PTY::resize);
|
||||
ClassDB::bind_method(D_METHOD("resizev", "size"), &PTY::resizev);
|
||||
ClassDB::bind_method(D_METHOD("kill", "signal"), &PTY::kill);
|
||||
ClassDB::bind_method(D_METHOD("fork", "file", "args", "cwd", "cols", "rows"), &PTY::fork, DEFVAL(""), DEFVAL(PackedStringArray()), DEFVAL("."), DEFVAL(80), DEFVAL(24));
|
||||
ClassDB::bind_method(D_METHOD("open", "cols", "rows"), &PTY::open, DEFVAL(80), DEFVAL(24));
|
||||
ClassDB::bind_method(D_METHOD("write", "data"), &PTY::write);
|
||||
ClassDB::bind_method(D_METHOD("resize", "cols", "rows"), &PTY::resize);
|
||||
ClassDB::bind_method(D_METHOD("resizev", "size"), &PTY::resizev);
|
||||
ClassDB::bind_method(D_METHOD("kill", "signal"), &PTY::kill);
|
||||
|
||||
ClassDB::bind_method(D_METHOD("_on_exit", "exit_code", "signal_code"), &PTY::_on_exit);
|
||||
ClassDB::bind_method(D_METHOD("_on_exit", "exit_code", "signal_code"), &PTY::_on_exit);
|
||||
}
|
||||
|
||||
PTY::PTY() {
|
||||
PTY::PTY()
|
||||
{
|
||||
use_threads = true;
|
||||
|
||||
set_process_internal(false);
|
||||
|
|
@ -94,76 +99,92 @@ PTY::PTY() {
|
|||
env["TERM"] = "xterm-256color";
|
||||
env["COLORTERM"] = "truecolor";
|
||||
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
uv_loop_init(&loop);
|
||||
uv_async_init(&loop, &async_handle, [](uv_async_t *handle) {});
|
||||
uv_pipe_init(&loop, &pipe, false);
|
||||
#ifdef _WIN32
|
||||
uv_pipe_init(&loop, &pipe_out, false);
|
||||
#endif
|
||||
pipe.data = this;
|
||||
#endif
|
||||
}
|
||||
|
||||
void PTY::set_cols(const int num_cols) {
|
||||
if (cols != num_cols) {
|
||||
void PTY::set_cols(const int num_cols)
|
||||
{
|
||||
if (cols != num_cols)
|
||||
{
|
||||
cols = num_cols;
|
||||
resize(cols, rows);
|
||||
}
|
||||
}
|
||||
|
||||
int PTY::get_cols() const {
|
||||
return cols;
|
||||
int PTY::get_cols() const
|
||||
{
|
||||
return cols;
|
||||
}
|
||||
|
||||
void PTY::set_rows(const int num_rows) {
|
||||
if (rows != num_rows) {
|
||||
void PTY::set_rows(const int num_rows)
|
||||
{
|
||||
if (rows != num_rows)
|
||||
{
|
||||
rows = num_rows;
|
||||
resize(cols, rows);
|
||||
}
|
||||
}
|
||||
|
||||
int PTY::get_rows() const {
|
||||
return rows;
|
||||
int PTY::get_rows() const
|
||||
{
|
||||
return rows;
|
||||
}
|
||||
|
||||
Dictionary PTY::get_env() const {
|
||||
return env;
|
||||
Dictionary PTY::get_env() const
|
||||
{
|
||||
return env;
|
||||
}
|
||||
|
||||
void PTY::set_env(const Dictionary &value) {
|
||||
env = value;
|
||||
void PTY::set_env(const Dictionary &value)
|
||||
{
|
||||
env = value;
|
||||
}
|
||||
|
||||
bool PTY::get_use_os_env() const {
|
||||
return use_os_env;
|
||||
bool PTY::get_use_os_env() const
|
||||
{
|
||||
return use_os_env;
|
||||
}
|
||||
|
||||
void PTY::set_use_os_env(const bool value) {
|
||||
use_os_env = value;
|
||||
void PTY::set_use_os_env(const bool value)
|
||||
{
|
||||
use_os_env = value;
|
||||
}
|
||||
|
||||
void PTY::set_use_threads(bool p_use) {
|
||||
void PTY::set_use_threads(bool p_use)
|
||||
{
|
||||
ERR_FAIL_COND(status != STATUS_CLOSED);
|
||||
use_threads = p_use;
|
||||
}
|
||||
|
||||
bool PTY::is_using_threads() const {
|
||||
bool PTY::is_using_threads() const
|
||||
{
|
||||
return use_threads;
|
||||
}
|
||||
|
||||
void PTY::set_terminal_path(NodePath p_terminal_path) {
|
||||
void PTY::set_terminal_path(NodePath p_terminal_path)
|
||||
{
|
||||
terminal_path = p_terminal_path;
|
||||
|
||||
Callable write = Callable(this, "write");
|
||||
Callable resizev = Callable(this, "resizev");
|
||||
|
||||
// Disconnect the current terminal, if any.
|
||||
if (terminal != nullptr) {
|
||||
if (terminal != nullptr)
|
||||
{
|
||||
disconnect("data_received", Callable(terminal, "write"));
|
||||
terminal->disconnect("data_sent", write);
|
||||
terminal->disconnect("size_changed", resizev);
|
||||
}
|
||||
|
||||
terminal = Object::cast_to<Terminal>(get_node_or_null(terminal_path));
|
||||
if (terminal == nullptr) return;
|
||||
if (terminal == nullptr)
|
||||
return;
|
||||
|
||||
// Connect the new terminal.
|
||||
resize(terminal->get_cols(), terminal->get_rows());
|
||||
|
|
@ -175,23 +196,31 @@ void PTY::set_terminal_path(NodePath p_terminal_path) {
|
|||
connect("data_received", Callable(terminal, "write"), CONNECT_PERSIST);
|
||||
}
|
||||
|
||||
NodePath PTY::get_terminal_path() const {
|
||||
NodePath PTY::get_terminal_path() const
|
||||
{
|
||||
return terminal_path;
|
||||
}
|
||||
|
||||
String PTY::get_pts_name() const {
|
||||
String PTY::get_pts_name() const
|
||||
{
|
||||
return pts_name;
|
||||
}
|
||||
|
||||
Error PTY::fork(const String &file, const PackedStringArray &args, const String &cwd, const int p_cols, const int p_rows) {
|
||||
Error PTY::fork(const String &file, const PackedStringArray &args, const String &cwd, const int p_cols, const int p_rows)
|
||||
{
|
||||
String fork_file = _get_fork_file(file);
|
||||
Dictionary fork_env = _get_fork_env();
|
||||
Dictionary result;
|
||||
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
String helper_path = ProjectSettings::get_singleton()->globalize_path("res://addons/godot_xterm/native/bin/spawn-helper");
|
||||
result = PTYUnix::fork(fork_file, args, _parse_env(fork_env), cwd, p_cols, p_rows, -1, -1, true, helper_path, Callable(this, "_on_exit"));
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#if defined(_WIN32)
|
||||
String helper_path = ProjectSettings::get_singleton()->globalize_path("res://addons/godot_xterm/native/bin/spawn-helper");
|
||||
result = PTYWin::fork(fork_file, args, _parse_env(fork_env), cwd, p_cols, p_rows, -1, -1, true, helper_path, Callable(this, "_on_exit"));
|
||||
#endif
|
||||
|
||||
Error err = static_cast<Error>((int)result["error"]);
|
||||
ERR_FAIL_COND_V_MSG(err != OK, err, "Failed to fork.");
|
||||
|
|
@ -199,15 +228,22 @@ Error PTY::fork(const String &file, const PackedStringArray &args, const String
|
|||
fd = result["fd"];
|
||||
pid = result["pid"];
|
||||
pts_name = result["pty"];
|
||||
|
||||
#ifdef _WIN32
|
||||
fd_out = result["fd_out"];
|
||||
hpc = result["hpc"];
|
||||
#endif
|
||||
|
||||
status = STATUS_OPEN;
|
||||
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
_pipe_open(fd);
|
||||
uv_read_start((uv_stream_t *)&pipe, _alloc_buffer, _read_cb);
|
||||
#endif
|
||||
_pipe_open(fd, &pipe);
|
||||
#ifdef _WIN32
|
||||
_pipe_open(fd_out, &pipe_out);
|
||||
#endif
|
||||
|
||||
if (use_threads) {
|
||||
uv_read_start((uv_stream_t *)&pipe, _alloc_buffer, _read_cb);
|
||||
|
||||
if (use_threads)
|
||||
{
|
||||
stop_thread.clear();
|
||||
thread->start(callable_mp(this, &PTY::_thread_func));
|
||||
}
|
||||
|
|
@ -216,80 +252,97 @@ Error PTY::fork(const String &file, const PackedStringArray &args, const String
|
|||
return OK;
|
||||
}
|
||||
|
||||
void PTY::kill(const int signal) {
|
||||
#if (defined(__linux__) || defined(__APPLE__)) && !defined(_PTY_DISABLED)
|
||||
if (pid > 0) {
|
||||
void PTY::kill(const int signal)
|
||||
{
|
||||
#if !defined(_PTY_DISABLED)
|
||||
if (pid > 0)
|
||||
{
|
||||
uv_kill(pid, signal);
|
||||
}
|
||||
#endif
|
||||
#endif
|
||||
}
|
||||
|
||||
Error PTY::open(const int cols, const int rows) {
|
||||
Dictionary result;
|
||||
Error PTY::open(const int cols, const int rows)
|
||||
{
|
||||
Dictionary result;
|
||||
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
result = PTYUnix::open(cols, rows);
|
||||
#endif
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
result = PTYUnix::open(cols, rows);
|
||||
#endif
|
||||
|
||||
Error err = static_cast<Error>((int)result["error"]);
|
||||
ERR_FAIL_COND_V(err != OK, err);
|
||||
#if defined(_WIN32)
|
||||
result = PTYWin::open(cols, rows);
|
||||
#endif
|
||||
|
||||
fd = result["master"];
|
||||
pts_name = result["pty"];
|
||||
Error err = static_cast<Error>((int)result["error"]);
|
||||
ERR_FAIL_COND_V(err != OK, err);
|
||||
|
||||
return OK;
|
||||
fd = result["master"];
|
||||
pts_name = result["pty"];
|
||||
|
||||
return OK;
|
||||
}
|
||||
|
||||
void PTY::resize(const int p_cols, const int p_rows) {
|
||||
void PTY::resize(const int p_cols, const int p_rows)
|
||||
{
|
||||
cols = p_cols;
|
||||
rows = p_rows;
|
||||
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
if (fd > -1) {
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
if (fd > -1)
|
||||
{
|
||||
PTYUnix::resize(fd, cols, rows);
|
||||
}
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#if defined(_WIN32)
|
||||
if (fd > -1)
|
||||
{
|
||||
PTYWin::resize(hpc, cols, rows);
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
void PTY::write(const Variant &data) const {
|
||||
void PTY::write(const Variant &data) const
|
||||
{
|
||||
PackedByteArray bytes;
|
||||
|
||||
switch (data.get_type())
|
||||
{
|
||||
case Variant::STRING:
|
||||
bytes = ((String)data).to_utf8_buffer();
|
||||
break;
|
||||
case Variant::PACKED_BYTE_ARRAY:
|
||||
bytes = data;
|
||||
break;
|
||||
default:
|
||||
ERR_FAIL_MSG("Data must be a String or PackedByteArray.");
|
||||
}
|
||||
switch (data.get_type())
|
||||
{
|
||||
case Variant::STRING:
|
||||
bytes = ((String)data).to_utf8_buffer();
|
||||
break;
|
||||
case Variant::PACKED_BYTE_ARRAY:
|
||||
bytes = data;
|
||||
break;
|
||||
default:
|
||||
ERR_FAIL_MSG("Data must be a String or PackedByteArray.");
|
||||
}
|
||||
|
||||
if (status == STATUS_OPEN) {
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
if (status == STATUS_OPEN)
|
||||
{
|
||||
uv_buf_t buf;
|
||||
buf.base = (char *)bytes.ptr();
|
||||
buf.len = bytes.size();
|
||||
uv_write_t *req = (uv_write_t *)malloc(sizeof(uv_write_t));
|
||||
req->data = (void *)buf.base;
|
||||
uv_write(req, (uv_stream_t *)&pipe, &buf, 1, _write_cb);
|
||||
uv_run((uv_loop_t*)&loop, UV_RUN_NOWAIT);
|
||||
#endif
|
||||
uv_write(req, (uv_stream_t *)&pipe_out, &buf, 1, _write_cb);
|
||||
uv_run((uv_loop_t *)&loop, UV_RUN_NOWAIT);
|
||||
}
|
||||
}
|
||||
|
||||
void PTY::_notification(int p_what) {
|
||||
void PTY::_notification(int p_what)
|
||||
{
|
||||
switch (p_what)
|
||||
{
|
||||
case NOTIFICATION_INTERNAL_PROCESS:
|
||||
case NOTIFICATION_INTERNAL_PROCESS:
|
||||
{
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
if (!use_threads) uv_run(&loop, UV_RUN_NOWAIT);
|
||||
#endif
|
||||
if (!use_threads)
|
||||
uv_run(&loop, UV_RUN_NOWAIT);
|
||||
|
||||
buffer_write_mutex->lock();
|
||||
if (buffer.size() > 0) {
|
||||
if (buffer.size() > 0)
|
||||
{
|
||||
emit_signal("data_received", buffer);
|
||||
buffer.clear();
|
||||
buffer_cleared->post();
|
||||
|
|
@ -304,46 +357,59 @@ void PTY::_notification(int p_what) {
|
|||
}
|
||||
}
|
||||
|
||||
void PTY::_thread_func() {
|
||||
while (!stop_thread.is_set()) {
|
||||
if (buffer.size() < BUFFER_LIMIT) {
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
void PTY::_thread_func()
|
||||
{
|
||||
while (!stop_thread.is_set())
|
||||
{
|
||||
if (buffer.size() < BUFFER_LIMIT)
|
||||
{
|
||||
uv_run(&loop, UV_RUN_ONCE);
|
||||
#endif
|
||||
} else {
|
||||
}
|
||||
else
|
||||
{
|
||||
buffer_cleared->wait();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void PTY::_close() {
|
||||
if (use_threads) {
|
||||
if (thread->is_started()) {
|
||||
void PTY::_close()
|
||||
{
|
||||
if (use_threads)
|
||||
{
|
||||
if (thread->is_started())
|
||||
{
|
||||
stop_thread.set();
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
uv_async_send(&async_handle);
|
||||
#endif
|
||||
thread->wait_to_finish();
|
||||
}
|
||||
}
|
||||
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
if (!uv_is_closing((uv_handle_t *)&pipe)) {
|
||||
if (!uv_is_closing((uv_handle_t *)&pipe))
|
||||
{
|
||||
uv_close((uv_handle_t *)&pipe, _close_cb);
|
||||
}
|
||||
|
||||
if (!uv_is_closing((uv_handle_t *)&async_handle)) {
|
||||
if (!uv_is_closing((uv_handle_t *)&async_handle))
|
||||
{
|
||||
uv_close((uv_handle_t *)&async_handle, _close_cb);
|
||||
}
|
||||
|
||||
if (status == STATUS_OPEN) {
|
||||
|
||||
if (status == STATUS_OPEN)
|
||||
{
|
||||
uv_run(&loop, UV_RUN_NOWAIT);
|
||||
uv_loop_close(&loop);
|
||||
}
|
||||
|
||||
if (fd > 0) close(fd);
|
||||
if (pid > 0) kill(SIGNAL_SIGHUP);
|
||||
#endif
|
||||
#ifdef _WIN32
|
||||
PTYWin::close(hpc, fd, fd_out);
|
||||
fd_out = -1;
|
||||
hpc = -1;
|
||||
#else
|
||||
if (fd > 0)
|
||||
close(fd);
|
||||
if (pid > 0)
|
||||
kill(IPCSIGNAL_SIGHUP);
|
||||
#endif
|
||||
|
||||
fd = -1;
|
||||
pid = -1;
|
||||
|
|
@ -352,38 +418,46 @@ void PTY::_close() {
|
|||
status = STATUS_CLOSED;
|
||||
}
|
||||
|
||||
String PTY::_get_fork_file(const String &file) const {
|
||||
if (!file.is_empty()) return file;
|
||||
String PTY::_get_fork_file(const String &file) const
|
||||
{
|
||||
if (!file.is_empty())
|
||||
return file;
|
||||
|
||||
String shell_env = OS::get_singleton()->get_environment("SHELL");
|
||||
if (!shell_env.is_empty()) {
|
||||
if (!shell_env.is_empty())
|
||||
{
|
||||
return shell_env;
|
||||
}
|
||||
|
||||
#if defined(__linux__)
|
||||
#if defined(__linux__)
|
||||
return "sh";
|
||||
#elif defined(__APPLE__)
|
||||
#elif defined(__APPLE__)
|
||||
return "zsh";
|
||||
#elif defined(_WIN32)
|
||||
#elif defined(_WIN32)
|
||||
return "cmd.exe";
|
||||
#else
|
||||
#else
|
||||
return "";
|
||||
#endif
|
||||
#endif
|
||||
}
|
||||
|
||||
Dictionary PTY::_get_fork_env() const {
|
||||
if (!use_os_env) return env;
|
||||
Dictionary PTY::_get_fork_env() const
|
||||
{
|
||||
if (!use_os_env)
|
||||
return env;
|
||||
|
||||
#if defined(_PTY_DISABLED)
|
||||
#if defined(_PTY_DISABLED)
|
||||
return env;
|
||||
#endif
|
||||
#endif
|
||||
|
||||
// TODO This might need windows specific adjustment
|
||||
return env;
|
||||
Dictionary os_env;
|
||||
uv_env_item_t *uv_env;
|
||||
int count;
|
||||
|
||||
uv_os_environ(&uv_env, &count);
|
||||
for (int i = 0; i < count; i++) {
|
||||
for (int i = 0; i < count; i++)
|
||||
{
|
||||
os_env[uv_env[i].name] = uv_env[i].value;
|
||||
}
|
||||
uv_os_free_environ(uv_env, count);
|
||||
|
|
@ -394,18 +468,19 @@ Dictionary PTY::_get_fork_env() const {
|
|||
|
||||
// Make sure we didn't start our server from inside screen.
|
||||
// http://web.mit.edu/gnu/doc/html/screen_20.html
|
||||
os_env.erase("STY");
|
||||
os_env.erase("WINDOW");
|
||||
os_env.erase("STY");
|
||||
os_env.erase("WINDOW");
|
||||
|
||||
// Delete some variables that might confuse our terminal.
|
||||
os_env.erase("WINDOWID");
|
||||
os_env.erase("TERMCAP");
|
||||
os_env.erase("COLUMNS");
|
||||
os_env.erase("LINES");
|
||||
os_env.erase("WINDOWID");
|
||||
os_env.erase("TERMCAP");
|
||||
os_env.erase("COLUMNS");
|
||||
os_env.erase("LINES");
|
||||
|
||||
// Merge in our custom environment.
|
||||
PackedStringArray keys = PackedStringArray(env.keys());
|
||||
for (int i = 0; i < keys.size(); i++) {
|
||||
for (int i = 0; i < keys.size(); i++)
|
||||
{
|
||||
String key = keys[i];
|
||||
os_env[key] = env[key];
|
||||
}
|
||||
|
|
@ -413,11 +488,13 @@ Dictionary PTY::_get_fork_env() const {
|
|||
return os_env;
|
||||
}
|
||||
|
||||
PackedStringArray PTY::_parse_env(const Dictionary &env) const {
|
||||
PackedStringArray PTY::_parse_env(const Dictionary &env) const
|
||||
{
|
||||
PackedStringArray parsed_env;
|
||||
PackedStringArray keys = PackedStringArray(env.keys());
|
||||
|
||||
for (int i = 0; i < keys.size(); i++) {
|
||||
for (int i = 0; i < keys.size(); i++)
|
||||
{
|
||||
String key = keys[i];
|
||||
parsed_env.push_back(key + "=" + String(env[key]));
|
||||
}
|
||||
|
|
@ -425,37 +502,41 @@ PackedStringArray PTY::_parse_env(const Dictionary &env) const {
|
|||
return parsed_env;
|
||||
}
|
||||
|
||||
void PTY::_on_exit(int exit_code, int exit_signal) {
|
||||
void PTY::_on_exit(int exit_code, int exit_signal)
|
||||
{
|
||||
call_deferred("emit_signal", "exited", exit_code, exit_signal);
|
||||
}
|
||||
|
||||
#if defined(__linux__) || defined(__APPLE__)
|
||||
|
||||
void _alloc_buffer(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf) {
|
||||
buf->base = (char *)malloc(suggested_size);
|
||||
buf->len = suggested_size;
|
||||
void _alloc_buffer(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf)
|
||||
{
|
||||
buf->base = (char *)malloc(suggested_size);
|
||||
buf->len = suggested_size;
|
||||
}
|
||||
|
||||
void PTY::_read_cb(uv_stream_t *pipe, ssize_t nread, const uv_buf_t *buf) {
|
||||
void PTY::_read_cb(uv_stream_t *pipe, ssize_t nread, const uv_buf_t *buf)
|
||||
{
|
||||
PTY *pty = static_cast<PTY *>(pipe->data);
|
||||
|
||||
if (nread < 0) {
|
||||
switch (nread) {
|
||||
case UV_EOF:
|
||||
// Normal after shell exits.
|
||||
case UV_EIO:
|
||||
// Can happen when the process exits.
|
||||
// As long as PTY has caught it, we should be fine.
|
||||
uv_read_stop(pipe);
|
||||
pty->status = PTY::Status::STATUS_CLOSED;
|
||||
if (nread < 0)
|
||||
{
|
||||
switch (nread)
|
||||
{
|
||||
case UV_EOF:
|
||||
// Normal after shell exits.
|
||||
case UV_EIO:
|
||||
// Can happen when the process exits.
|
||||
// As long as PTY has caught it, we should be fine.
|
||||
uv_read_stop(pipe);
|
||||
pty->status = PTY::Status::STATUS_CLOSED;
|
||||
return;
|
||||
default:
|
||||
pty->status = PTY::Status::STATUS_ERROR;
|
||||
}
|
||||
return;
|
||||
default:
|
||||
pty->status = PTY::Status::STATUS_ERROR;
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
if (nread > 0) {
|
||||
if (nread > 0)
|
||||
{
|
||||
MutexLock lock(*pty->buffer_write_mutex.ptr());
|
||||
|
||||
int old_size = pty->buffer.size();
|
||||
|
|
@ -468,14 +549,11 @@ void PTY::_read_cb(uv_stream_t *pipe, ssize_t nread, const uv_buf_t *buf) {
|
|||
}
|
||||
}
|
||||
|
||||
Error PTY::_pipe_open(const int fd) {
|
||||
Error PTY::_pipe_open(const int fd, uv_pipe_t *pipe)
|
||||
{
|
||||
ERR_FAIL_COND_V_MSG(fd < 0, FAILED, "File descriptor must be a non-negative integer value.");
|
||||
|
||||
ERR_FAIL_UV_ERR(uv_pipe_open(&pipe, fd));
|
||||
ERR_FAIL_UV_ERR(uv_stream_set_blocking((uv_stream_t *)&pipe, false));
|
||||
ERR_FAIL_UV_ERR(uv_read_start((uv_stream_t *)&pipe, _alloc_buffer, _read_cb));
|
||||
ERR_FAIL_UV_ERR(uv_pipe_open(pipe, fd));
|
||||
ERR_FAIL_UV_ERR(uv_stream_set_blocking((uv_stream_t *)pipe, false));
|
||||
|
||||
return OK;
|
||||
}
|
||||
|
||||
#endif
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue