mirror of
https://github.com/lihop/godot-xterm.git
synced 2025-04-11 19:23:58 +02:00
a1de456240
De-references pty_baton's exit callback after it is called so it can be automatically released, preventing leaked instances. Adds basic implementation for Pipe's get_status() method and forces PTY to wait for child process to exit to ensure exit callback is cleaned up. Adds a test to check that exit callback is still called as usual.
131 lines
No EOL
3 KiB
C++
131 lines
No EOL
3 KiB
C++
// Copyright (c) 2021, Leroy Hopson (MIT License).
|
|
|
|
#include "pipe.h"
|
|
#include "libuv_utils.h"
|
|
#include <Dictionary.hpp>
|
|
#include <InputEventKey.hpp>
|
|
#include <OS.hpp>
|
|
#include <ResourceLoader.hpp>
|
|
#include <Theme.hpp>
|
|
#include <Timer.hpp>
|
|
#include <algorithm>
|
|
#include <thread>
|
|
#include <vector>
|
|
#include <xkbcommon/xkbcommon-keysyms.h>
|
|
|
|
#ifndef ULONG
|
|
#define ULONG size_t
|
|
#endif
|
|
|
|
using namespace godot;
|
|
|
|
void Pipe::_register_methods() {
|
|
register_method("_init", &Pipe::_init);
|
|
|
|
register_method("poll", &Pipe::_poll_connection);
|
|
register_method("open", &Pipe::open);
|
|
register_method("write", &Pipe::write);
|
|
register_method("get_status", &Pipe::get_status);
|
|
register_method("close", &Pipe::close);
|
|
|
|
register_signal<Pipe>("data_received", "data",
|
|
GODOT_VARIANT_TYPE_POOL_BYTE_ARRAY);
|
|
}
|
|
|
|
Pipe::Pipe() {}
|
|
Pipe::~Pipe() { close(); }
|
|
|
|
void Pipe::_init() {}
|
|
|
|
void _poll_connection();
|
|
|
|
void _read_cb(uv_stream_t *handle, ssize_t nread, const uv_buf_t *buf);
|
|
|
|
void _close_cb(uv_handle_t *handle);
|
|
|
|
void _write_cb(uv_write_t *req, int status);
|
|
|
|
void _alloc_buffer(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf);
|
|
|
|
godot_error Pipe::open(int fd, bool ipc = false) {
|
|
RETURN_IF_UV_ERR(uv_pipe_init(uv_default_loop(), &handle, ipc));
|
|
|
|
handle.data = this;
|
|
|
|
RETURN_IF_UV_ERR(uv_pipe_open(&handle, fd));
|
|
RETURN_IF_UV_ERR(uv_stream_set_blocking((uv_stream_t *)&handle, false));
|
|
RETURN_IF_UV_ERR(
|
|
uv_read_start((uv_stream_t *)&handle, _alloc_buffer, _read_cb));
|
|
|
|
status = 1;
|
|
return GODOT_OK;
|
|
}
|
|
|
|
void Pipe::close() {
|
|
uv_close((uv_handle_t *)&handle, _close_cb);
|
|
uv_run(uv_default_loop(), UV_RUN_NOWAIT);
|
|
}
|
|
|
|
godot_error Pipe::write(String p_data) {
|
|
char *s = p_data.alloc_c_string();
|
|
ULONG len = strlen(s);
|
|
|
|
uv_buf_t bufs[1];
|
|
bufs[0].base = s;
|
|
bufs[0].len = len;
|
|
|
|
uv_write_t req;
|
|
|
|
req.data = s;
|
|
|
|
uv_write(&req, (uv_stream_t *)&handle, bufs, 1, _write_cb);
|
|
|
|
uv_run(uv_default_loop(), UV_RUN_NOWAIT);
|
|
|
|
return GODOT_OK;
|
|
}
|
|
|
|
int Pipe::get_status() {
|
|
if (!uv_is_active((uv_handle_t *)&handle))
|
|
status = 0;
|
|
_poll_connection();
|
|
return status;
|
|
}
|
|
|
|
void Pipe::_poll_connection() { uv_run(uv_default_loop(), UV_RUN_NOWAIT); }
|
|
|
|
void _read_cb(uv_stream_t *handle, ssize_t nread, const uv_buf_t *buf) {
|
|
Pipe *pipe = static_cast<Pipe *>(handle->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(handle);
|
|
return;
|
|
default:
|
|
UV_ERR_PRINT(nread);
|
|
}
|
|
return;
|
|
}
|
|
|
|
PoolByteArray data;
|
|
data.resize(nread);
|
|
memcpy(data.write().ptr(), buf->base, nread);
|
|
|
|
pipe->emit_signal("data_received", data);
|
|
}
|
|
|
|
void _write_cb(uv_write_t *req, int status) {}
|
|
|
|
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 _close_cb(uv_handle_t *handle) {
|
|
Pipe *pipe = static_cast<Pipe *>(handle->data);
|
|
pipe->status = 0;
|
|
} |