111 lines
2.6 KiB
C++
111 lines
2.6 KiB
C++
|
|
||
|
|
#include "PTY_Interface.h"
|
||
|
|
#include <pty.h>
|
||
|
|
#include <iostream>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <boost/algorithm/string.hpp>
|
||
|
|
#include <termios.h>
|
||
|
|
using namespace boost::algorithm;
|
||
|
|
|
||
|
|
static void set_noecho(int fd) { // Turn off Slave echo
|
||
|
|
struct termios stermios;
|
||
|
|
|
||
|
|
if (tcgetattr(fd, &stermios) < 0)
|
||
|
|
std::cout << "Error tcgetattr()" << std::endl;
|
||
|
|
|
||
|
|
stermios.c_lflag &= ~(ECHO | ECHOE | ECHOK | ECHONL);
|
||
|
|
stermios.c_oflag &= ~(ONLCR);
|
||
|
|
/* also turn off NL to CR/NL mapping on output */
|
||
|
|
|
||
|
|
if (tcsetattr(fd, TCSANOW, &stermios) < 0)
|
||
|
|
std::cout << "Error tcsetattr()" << std::endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
Interface::Interface(Device *default_device,
|
||
|
|
int default_address) {
|
||
|
|
char name[40];
|
||
|
|
|
||
|
|
this->addr = default_address;
|
||
|
|
this->addDevice(default_device, default_address);
|
||
|
|
|
||
|
|
if (openpty(&this->m, &this->s, name, NULL, NULL) < 0)
|
||
|
|
std::cerr << "Could not open pty device" << std::endl;
|
||
|
|
else {
|
||
|
|
set_noecho(this->s);
|
||
|
|
this->pname = name;
|
||
|
|
this->running = false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
Interface::~Interface() {
|
||
|
|
close(this->m);
|
||
|
|
}
|
||
|
|
|
||
|
|
string Interface::printFilename() {
|
||
|
|
std::cout << this->pname << std::endl;
|
||
|
|
return this->pname;
|
||
|
|
}
|
||
|
|
|
||
|
|
void Interface::run(int readLen) {
|
||
|
|
char *tty_in = new char[readLen+1];
|
||
|
|
vector<string> tty_out;
|
||
|
|
string tty_full;
|
||
|
|
int i;
|
||
|
|
|
||
|
|
this->running = true;
|
||
|
|
while (this->running) {
|
||
|
|
tty_full = "";
|
||
|
|
while (read(this->m, tty_in, readLen) >= readLen) {
|
||
|
|
std::cout << tty_in << std::endl;
|
||
|
|
tty_full += tty_in;
|
||
|
|
}
|
||
|
|
tty_full += tty_in;
|
||
|
|
i = tty_full.find("\n");
|
||
|
|
if ( i != string::npos )
|
||
|
|
tty_full = tty_full.substr(0, i);
|
||
|
|
|
||
|
|
tty_out = this->handleInput(tty_full);
|
||
|
|
for (std::vector<string>::iterator it = tty_out.begin();
|
||
|
|
it != tty_out.end();
|
||
|
|
it++) {
|
||
|
|
write(this->m, it->c_str(), it->length());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
delete[] tty_in;
|
||
|
|
}
|
||
|
|
|
||
|
|
vector<string> Interface::handleInput(string tty_in) {
|
||
|
|
vector<string> out;
|
||
|
|
Device *dev = NULL;
|
||
|
|
vector<string> cmd;
|
||
|
|
vector<string> p;
|
||
|
|
string first;
|
||
|
|
int i;
|
||
|
|
|
||
|
|
trim(tty_in);
|
||
|
|
if (tty_in != "" && this->devices.count(this->addr) > 0) {
|
||
|
|
dev = this->devices.at(this->addr);
|
||
|
|
if (dev == NULL) {
|
||
|
|
out.push_back("ERROR! No Device at address: " +
|
||
|
|
to_string(this->addr));
|
||
|
|
}
|
||
|
|
|
||
|
|
boost::split(cmd, tty_in, boost::is_any_of(";\n"));
|
||
|
|
for (i=0;i<cmd.size();i++) {
|
||
|
|
boost::split(p, cmd.at(i), boost::is_any_of(" "));
|
||
|
|
first = p.at(0);
|
||
|
|
p.erase(p.begin());
|
||
|
|
out.push_back(dev->handleCMD(this, first, &p) + "\n");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
out.push_back("");
|
||
|
|
return out;
|
||
|
|
}
|
||
|
|
|
||
|
|
void Interface::addDevice(Device *dev, int addr) {
|
||
|
|
this->devices.insert( std::pair<int, Device*>(addr, dev));
|
||
|
|
}
|
||
|
|
|