plugin/windows: Added TTY Console view
This commit is contained in:
WerWolv
parent
c373174436
commit
c22d1033eb
87
plugins/windows/include/views/view_tty_console.hpp
Normal file
87
plugins/windows/include/views/view_tty_console.hpp
Normal file
@ -0,0 +1,87 @@
|
||||
#pragma once
|
||||
|
||||
#include <hex/views/view.hpp>
|
||||
|
||||
#include <windows.h>
|
||||
|
||||
#include <thread>
|
||||
|
||||
namespace hex {
|
||||
|
||||
namespace prv { class Provider; }
|
||||
|
||||
class ViewTTYConsole : public View {
|
||||
public:
|
||||
ViewTTYConsole();
|
||||
~ViewTTYConsole() override;
|
||||
|
||||
void drawContent() override;
|
||||
void drawMenu() override;
|
||||
|
||||
private:
|
||||
std::vector<std::pair<std::string, std::string>> m_comPorts;
|
||||
|
||||
std::vector<std::pair<std::string, std::string>> getAvailablePorts();
|
||||
bool connect();
|
||||
bool disconnect();
|
||||
|
||||
void transmitData();
|
||||
|
||||
HANDLE m_portHandle = INVALID_HANDLE_VALUE;
|
||||
std::jthread m_receiveThread;
|
||||
|
||||
int m_selectedPort = 0;
|
||||
int m_selectedBaudRate = 11; // 115200
|
||||
int m_selectedNumBits = 3; // 8
|
||||
int m_selectedStopBits = 0; // 1
|
||||
int m_selectedParityBits = 0; // None
|
||||
bool m_hasCTSFlowControl = false; // No
|
||||
|
||||
bool m_shouldAutoScroll = true;
|
||||
|
||||
std::vector<char> m_receiveDataBuffer, m_transmitDataBuffer;
|
||||
std::vector<u32> m_wrapPositions;
|
||||
bool m_transmitting = false;
|
||||
|
||||
bool m_needsScrolling = false;
|
||||
|
||||
constexpr static std::array BaudRates = {
|
||||
"110",
|
||||
"300",
|
||||
"600",
|
||||
"1200",
|
||||
"2400",
|
||||
"4800",
|
||||
"9600",
|
||||
"14400",
|
||||
"19200",
|
||||
"38400",
|
||||
"57600",
|
||||
"115200",
|
||||
"128000",
|
||||
"256000"
|
||||
};
|
||||
|
||||
constexpr static std::array NumBits = {
|
||||
"5",
|
||||
"6",
|
||||
"7",
|
||||
"8"
|
||||
};
|
||||
|
||||
constexpr static std::array StopBits = {
|
||||
"1",
|
||||
"1.5",
|
||||
"2.0"
|
||||
};
|
||||
|
||||
constexpr static std::array ParityBits = {
|
||||
"None",
|
||||
"Odd",
|
||||
"Even",
|
||||
"Mark",
|
||||
"Space"
|
||||
};
|
||||
};
|
||||
|
||||
}
|
||||
@ -5,6 +5,7 @@
|
||||
#include "views/view_tty_console.hpp"
|
||||
|
||||
IMHEX_PLUGIN_SETUP("Windows", "WerWolv", "Windows-only features") {
|
||||
ContentRegistry::Views::add<hex::ViewTTYConsole>();
|
||||
}
|
||||
|
||||
|
||||
|
||||
273
plugins/windows/source/views/view_tty_console.cpp
Normal file
273
plugins/windows/source/views/view_tty_console.cpp
Normal file
@ -0,0 +1,273 @@
|
||||
#include "views/view_tty_console.hpp"
|
||||
|
||||
#include <imgui_imhex_extensions.h>
|
||||
#include <imgui_internal.h>
|
||||
|
||||
namespace hex {
|
||||
|
||||
ViewTTYConsole::ViewTTYConsole() : View("hex.windows.view.tty_console.name") {
|
||||
this->m_comPorts = getAvailablePorts();
|
||||
this->m_transmitDataBuffer.resize(0xFFF, 0x00);
|
||||
this->m_receiveDataBuffer.reserve(0xFFF);
|
||||
this->m_receiveDataBuffer.push_back(0x00);
|
||||
}
|
||||
|
||||
ViewTTYConsole::~ViewTTYConsole() {
|
||||
|
||||
}
|
||||
|
||||
void ViewTTYConsole::drawContent() {
|
||||
if (ImGui::Begin(View::toWindowName("hex.windows.view.tty_console.name").c_str(), &this->getWindowOpenState())) {
|
||||
|
||||
ImGui::TextUnformatted("Configuration");
|
||||
ImGui::Separator();
|
||||
|
||||
bool connected = this->m_portHandle != INVALID_HANDLE_VALUE;
|
||||
|
||||
ImGui::PushItemFlag(ImGuiItemFlags_Disabled, connected);
|
||||
ImGui::PushStyleVar(ImGuiStyleVar_Alpha, connected ? 0.5F : 1.0F);
|
||||
|
||||
ImGui::Combo("Port", &this->m_selectedPort, [](void* data, int idx, const char** out_text) -> bool {
|
||||
auto &ports = *static_cast<std::vector<std::pair<std::string, std::string>>*>(data);
|
||||
|
||||
*out_text = ports[idx].first.c_str();
|
||||
return true;
|
||||
}, &this->m_comPorts, this->m_comPorts.size());
|
||||
|
||||
ImGui::SameLine();
|
||||
if (ImGui::Button("Reload"))
|
||||
this->m_comPorts = getAvailablePorts();
|
||||
|
||||
ImGui::Combo("Baud rate", &this->m_selectedBaudRate, [](void* data, int idx, const char** out_text) -> bool {
|
||||
*out_text = ViewTTYConsole::BaudRates[idx];
|
||||
return true;
|
||||
}, nullptr, ViewTTYConsole::BaudRates.size());
|
||||
|
||||
ImGui::Combo("Data bits", &this->m_selectedNumBits, [](void* data, int idx, const char** out_text) -> bool {
|
||||
*out_text = ViewTTYConsole::NumBits[idx];
|
||||
return true;
|
||||
}, nullptr, ViewTTYConsole::NumBits.size());
|
||||
|
||||
ImGui::Combo("Stop bits", &this->m_selectedStopBits, [](void* data, int idx, const char** out_text) -> bool {
|
||||
*out_text = ViewTTYConsole::StopBits[idx];
|
||||
return true;
|
||||
}, nullptr, ViewTTYConsole::StopBits.size());
|
||||
|
||||
ImGui::Combo("Parity bit", &this->m_selectedParityBits, [](void* data, int idx, const char** out_text) -> bool {
|
||||
*out_text = ViewTTYConsole::ParityBits[idx];
|
||||
return true;
|
||||
}, nullptr, ViewTTYConsole::ParityBits.size());
|
||||
|
||||
ImGui::Checkbox("CTS Flow control", &this->m_hasCTSFlowControl);
|
||||
|
||||
ImGui::PopStyleVar();
|
||||
ImGui::PopItemFlag();
|
||||
|
||||
ImGui::NewLine();
|
||||
|
||||
if (this->m_portHandle == INVALID_HANDLE_VALUE) {
|
||||
if (ImGui::Button("Connect"))
|
||||
if (!this->connect())
|
||||
log::error("Connect error");
|
||||
} else {
|
||||
if (ImGui::Button("Disconnect"))
|
||||
this->disconnect();
|
||||
}
|
||||
|
||||
ImGui::NewLine();
|
||||
|
||||
if (ImGui::Button("Clear")) {
|
||||
this->m_receiveDataBuffer.clear();
|
||||
this->m_wrapPositions.clear();
|
||||
}
|
||||
|
||||
ImGui::SameLine();
|
||||
|
||||
ImGui::Checkbox("Auto Scroll", &this->m_shouldAutoScroll);
|
||||
|
||||
ImGui::NewLine();
|
||||
ImGui::TextUnformatted("Data");
|
||||
ImGui::Separator();
|
||||
|
||||
auto consoleSize = ImGui::GetContentRegionAvail();
|
||||
consoleSize.y -= ImGui::GetTextLineHeight() + ImGui::GetStyle().FramePadding.y * 4;
|
||||
if (ImGui::BeginChild("##scrolling", consoleSize, true, ImGuiWindowFlags_HorizontalScrollbar)) {
|
||||
ImGuiListClipper clipper;
|
||||
clipper.Begin(this->m_wrapPositions.size(), ImGui::GetTextLineHeight());
|
||||
|
||||
while (clipper.Step()) {
|
||||
for (u32 i = clipper.DisplayStart + 1; i < clipper.DisplayEnd; i++) {
|
||||
ImGui::TextUnformatted(this->m_receiveDataBuffer.data() + this->m_wrapPositions[i - 1], this->m_receiveDataBuffer.data() + this->m_wrapPositions[i]);
|
||||
}
|
||||
}
|
||||
|
||||
if (this->m_shouldAutoScroll && this->m_needsScrolling) {
|
||||
ImGui::SetScrollHereY(0.0F);
|
||||
this->m_needsScrolling = false;
|
||||
}
|
||||
|
||||
ImGui::EndChild();
|
||||
}
|
||||
|
||||
ImGui::PushItemWidth(-1);
|
||||
if (ImGui::InputText("##transmit", this->m_transmitDataBuffer.data(), this->m_transmitDataBuffer.size() - 2, ImGuiInputTextFlags_EnterReturnsTrue)) {
|
||||
auto size = strlen(this->m_transmitDataBuffer.data());
|
||||
|
||||
this->m_transmitDataBuffer[size + 0] = '\n';
|
||||
this->m_transmitDataBuffer[size + 1] = 0x00;
|
||||
|
||||
this->transmitData();
|
||||
ImGui::SetKeyboardFocusHere(0);
|
||||
}
|
||||
ImGui::PopItemWidth();
|
||||
|
||||
}
|
||||
ImGui::End();
|
||||
}
|
||||
|
||||
void ViewTTYConsole::drawMenu() {
|
||||
|
||||
}
|
||||
|
||||
std::vector<std::pair<std::string, std::string>> ViewTTYConsole::getAvailablePorts() {
|
||||
std::vector<std::pair<std::string, std::string>> result;
|
||||
std::vector<char> buffer(0xFFF, 0x00);
|
||||
|
||||
for (u16 portNumber = 0; portNumber <= 255; portNumber++) {
|
||||
std::string port = "COM" + std::to_string(portNumber);
|
||||
|
||||
if (::QueryDosDevice(port.c_str(), buffer.data(), buffer.size()) != 0) {
|
||||
result.emplace_back(port, buffer.data());
|
||||
}
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
bool ViewTTYConsole::connect() {
|
||||
this->m_portHandle = ::CreateFile(("\\\\.\\" + this->m_comPorts[this->m_selectedPort].first).c_str(),
|
||||
GENERIC_READ | GENERIC_WRITE,
|
||||
0,
|
||||
nullptr,
|
||||
OPEN_EXISTING,
|
||||
FILE_FLAG_OVERLAPPED,
|
||||
nullptr);
|
||||
|
||||
if (this->m_portHandle == INVALID_HANDLE_VALUE)
|
||||
return false;
|
||||
|
||||
auto closeHandle = SCOPE_GUARD { CloseHandle(this->m_portHandle); };
|
||||
|
||||
DCB serialParams;
|
||||
serialParams.DCBlength = sizeof(DCB);
|
||||
|
||||
if (!::GetCommState(this->m_portHandle, &serialParams))
|
||||
return false;
|
||||
|
||||
serialParams.BaudRate = std::stoi(ViewTTYConsole::BaudRates[this->m_selectedBaudRate]);
|
||||
serialParams.ByteSize = std::stoi(ViewTTYConsole::NumBits[this->m_selectedNumBits]);
|
||||
serialParams.StopBits = this->m_selectedStopBits;
|
||||
serialParams.Parity = this->m_selectedParityBits;
|
||||
serialParams.fOutxCtsFlow = this->m_hasCTSFlowControl;
|
||||
|
||||
if (!::SetCommState(this->m_portHandle, &serialParams))
|
||||
return false;
|
||||
|
||||
COMMTIMEOUTS timeouts;
|
||||
timeouts.ReadIntervalTimeout = 500;
|
||||
timeouts.ReadTotalTimeoutConstant = 500;
|
||||
timeouts.ReadTotalTimeoutMultiplier = 100;
|
||||
timeouts.WriteTotalTimeoutConstant = 500;
|
||||
timeouts.WriteTotalTimeoutMultiplier = 100;
|
||||
|
||||
if (!::SetCommTimeouts(this->m_portHandle, &timeouts))
|
||||
return false;
|
||||
|
||||
if(!::SetupComm(this->m_portHandle, 10000, 10000))
|
||||
return false;
|
||||
|
||||
closeHandle.release();
|
||||
|
||||
this->m_receiveThread = std::jthread([this](std::stop_token token) {
|
||||
bool waitingOnRead = false;
|
||||
OVERLAPPED overlapped = { 0 };
|
||||
|
||||
overlapped.hEvent = ::CreateEvent(nullptr, true, false, nullptr);
|
||||
ON_SCOPE_EXIT { ::CloseHandle(&overlapped); };
|
||||
|
||||
while (!token.stop_requested()) {
|
||||
DWORD bytesRead = 0;
|
||||
|
||||
char byte;
|
||||
if (!waitingOnRead) {
|
||||
if (::ReadFile(this->m_portHandle, &byte, sizeof(char), &bytesRead, &overlapped)) {
|
||||
if (byte >= 0x20 && byte <= 0x7E) {
|
||||
this->m_receiveDataBuffer.back() = byte;
|
||||
this->m_receiveDataBuffer.push_back(0x00);
|
||||
this->m_needsScrolling = true;
|
||||
} else if (byte == '\n' || byte == '\r') {
|
||||
this->m_wrapPositions.push_back(this->m_receiveDataBuffer.size() - 1);
|
||||
}
|
||||
} else if (::GetLastError() == ERROR_IO_PENDING) {
|
||||
waitingOnRead = true;
|
||||
}
|
||||
} else {
|
||||
auto res = ::WaitForSingleObject(overlapped.hEvent, 500);
|
||||
switch (res) {
|
||||
case WAIT_OBJECT_0:
|
||||
if (::GetOverlappedResult(this->m_portHandle, &overlapped, &bytesRead, false)) {
|
||||
if (byte >= 0x20 && byte <= 0x7E) {
|
||||
this->m_receiveDataBuffer.back() = byte;
|
||||
this->m_receiveDataBuffer.push_back(0x00);
|
||||
this->m_needsScrolling = true;
|
||||
}
|
||||
waitingOnRead = false;
|
||||
}
|
||||
default: break;
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool ViewTTYConsole::disconnect() {
|
||||
::SetCommMask(this->m_portHandle, EV_TXEMPTY);
|
||||
this->m_receiveThread.request_stop();
|
||||
this->m_receiveThread.join();
|
||||
|
||||
::CloseHandle(this->m_portHandle);
|
||||
this->m_portHandle = INVALID_HANDLE_VALUE;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void ViewTTYConsole::transmitData() {
|
||||
if (this->m_transmitting)
|
||||
return;
|
||||
|
||||
auto transmitThread = std::thread([&, this]{
|
||||
OVERLAPPED overlapped = { 0 };
|
||||
|
||||
overlapped.hEvent = ::CreateEvent(nullptr, true, false, nullptr);
|
||||
ON_SCOPE_EXIT { ::CloseHandle(&overlapped); };
|
||||
|
||||
this->m_transmitting = true;
|
||||
|
||||
DWORD bytesWritten = 0;
|
||||
if (!::WriteFile(this->m_portHandle, this->m_transmitDataBuffer.data(), strlen(this->m_transmitDataBuffer.data()), &bytesWritten, &overlapped)) {
|
||||
if (::GetLastError() == ERROR_IO_PENDING) {
|
||||
::GetOverlappedResult(this->m_portHandle, &overlapped, &bytesWritten, true);
|
||||
}
|
||||
}
|
||||
|
||||
if (bytesWritten > 0)
|
||||
this->m_transmitDataBuffer[0] = 0x00;
|
||||
|
||||
this->m_transmitting = false;
|
||||
});
|
||||
transmitThread.detach();
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user