94 lines
3.1 KiB
C++
94 lines
3.1 KiB
C++
|
#ifndef slic3r_Utils_TCPConsole_hpp_
|
||
|
|
#define slic3r_Utils_TCPConsole_hpp_
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
#include <deque>
|
||
|
|
#include <boost/system/error_code.hpp>
|
||
|
|
#include <boost/system/system_error.hpp>
|
||
|
|
#include <boost/asio/ip/tcp.hpp>
|
||
|
|
#include <boost/asio/streambuf.hpp>
|
||
|
|
|
||
|
|
namespace Slic3r {
|
||
|
|
namespace Utils {
|
||
|
|
|
||
|
|
using boost::asio::ip::tcp;
|
||
|
|
|
||
|
|
// Generic command / response TCP telnet like console class.
|
||
|
|
// Used by the MKS host to send G-code commands to test connection ("M105") and to start printing ("M23 filename", "M24").
|
||
|
|
class TCPConsole
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
TCPConsole() : m_resolver(m_io_context), m_socket(m_io_context) { set_defaults(); }
|
||
|
|
TCPConsole(const std::string& host_name, const std::string& port_name) : m_resolver(m_io_context), m_socket(m_io_context)
|
||
|
|
{ set_defaults(); set_remote(host_name, port_name); }
|
||
|
|
~TCPConsole() = default;
|
||
|
|
|
||
|
|
void set_defaults()
|
||
|
|
{
|
||
|
|
m_newline = "\n";
|
||
|
|
m_done_string = "ok";
|
||
|
|
m_connect_timeout = std::chrono::milliseconds(5000);
|
||
|
|
m_write_timeout = std::chrono::milliseconds(10000);
|
||
|
|
m_read_timeout = std::chrono::milliseconds(10000);
|
||
|
|
}
|
||
|
|
|
||
|
|
void set_line_delimiter(const std::string& newline) {
|
||
|
|
m_newline = newline;
|
||
|
|
}
|
||
|
|
void set_command_done_string(const std::string& done_string) {
|
||
|
|
m_done_string = done_string;
|
||
|
|
}
|
||
|
|
|
||
|
|
void set_remote(const std::string& host_name, const std::string& port_name)
|
||
|
|
{
|
||
|
|
m_host_name = host_name;
|
||
|
|
m_port_name = port_name;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool enqueue_cmd(const std::string& cmd) {
|
||
|
|
// TODO: Add multithread protection to queue
|
||
|
|
m_cmd_queue.push_back(cmd);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool run_queue();
|
||
|
|
std::string error_message() const { return m_error_code.message(); }
|
||
|
|
|
||
|
|
private:
|
||
|
|
void handle_connect(const boost::system::error_code& ec);
|
||
|
|
void handle_read(const boost::system::error_code& ec, std::size_t bytes_transferred);
|
||
|
|
void handle_write(const boost::system::error_code& ec, std::size_t bytes_transferred);
|
||
|
|
|
||
|
|
void transmit_next_command();
|
||
|
|
void wait_next_line();
|
||
|
|
std::string extract_next_line();
|
||
|
|
|
||
|
|
void set_deadline_in(std::chrono::steady_clock::duration);
|
||
|
|
bool is_deadline_over() const;
|
||
|
|
|
||
|
|
std::string m_host_name;
|
||
|
|
std::string m_port_name;
|
||
|
|
std::string m_newline;
|
||
|
|
std::string m_done_string;
|
||
|
|
std::chrono::steady_clock::duration m_connect_timeout;
|
||
|
|
std::chrono::steady_clock::duration m_write_timeout;
|
||
|
|
std::chrono::steady_clock::duration m_read_timeout;
|
||
|
|
|
||
|
|
std::deque<std::string> m_cmd_queue;
|
||
|
|
|
||
|
|
boost::asio::io_context m_io_context;
|
||
|
|
tcp::resolver m_resolver;
|
||
|
|
tcp::socket m_socket;
|
||
|
|
boost::asio::streambuf m_recv_buffer;
|
||
|
|
std::string m_send_buffer;
|
||
|
|
|
||
|
|
bool m_is_connected;
|
||
|
|
boost::system::error_code m_error_code;
|
||
|
|
std::chrono::steady_clock::time_point m_deadline;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // Utils
|
||
|
|
} // Slic3r
|
||
|
|
|
||
|
|
#endif
|