Jerry Zhao
3 years ago
7 changed files with 144 additions and 112 deletions
@ -0,0 +1,74 @@ |
|||||
|
// See LICENSE for license details.
|
||||
|
|
||||
|
#include "socketif.h" |
||||
|
|
||||
|
#ifdef HAVE_BOOST_ASIO |
||||
|
|
||||
|
#include <iostream> |
||||
|
|
||||
|
socketif_t::socketif_t() |
||||
|
{ |
||||
|
try { // create socket server
|
||||
|
using boost::asio::ip::tcp; |
||||
|
io_service_ptr = new boost::asio::io_service; |
||||
|
acceptor_ptr = new tcp::acceptor(*io_service_ptr, tcp::endpoint(tcp::v4(), 0)); |
||||
|
// acceptor is created passing argument port=0, so O.S. will choose a free port
|
||||
|
std::string name = boost::asio::ip::host_name(); |
||||
|
std::cout << "Listening for debug commands on " << name.substr(0,name.find('.')) |
||||
|
<< " port " << acceptor_ptr->local_endpoint().port() << " ." << std::endl; |
||||
|
// at the end, add space and some other character for convenience of javascript .split(" ")
|
||||
|
} catch (std::exception& e) { |
||||
|
std::cerr << e.what() << std::endl; |
||||
|
exit(-1); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
socketif_t::~socketif_t() |
||||
|
{ |
||||
|
delete io_service_ptr; |
||||
|
delete acceptor_ptr; |
||||
|
} |
||||
|
|
||||
|
// read input command string
|
||||
|
std::string socketif_t::rin(std::ostream &sout_) |
||||
|
{ |
||||
|
std::string s; |
||||
|
try { |
||||
|
socket_ptr.reset(new boost::asio::ip::tcp::socket(*io_service_ptr)); |
||||
|
acceptor_ptr->accept(*socket_ptr); // wait for someone to open connection
|
||||
|
boost::asio::streambuf buf; |
||||
|
boost::asio::read_until(*socket_ptr, buf, "\n"); // wait for command
|
||||
|
s = boost::asio::buffer_cast<const char*>(buf.data()); |
||||
|
boost::erase_all(s, "\r"); // get rid off any cr and lf
|
||||
|
boost::erase_all(s, "\n"); |
||||
|
// The socket client is a web server and it appends the IP of the computer
|
||||
|
// that sent the command from its web browser.
|
||||
|
|
||||
|
// For now, erase the IP if it is there.
|
||||
|
boost::regex re(" ((25[0-5]|2[0-4][0-9]|[01]?[0-9]?[0-9])\\.){3}" |
||||
|
"(25[0-5]|2[0-4][0-9]|[01]?[0-9]?[0-9])$"); |
||||
|
s = boost::regex_replace(s, re, (std::string)""); |
||||
|
|
||||
|
// TODO: check the IP against the IP used to upload RISC-V source files
|
||||
|
} catch (std::exception& e) { |
||||
|
std::cerr << e.what() << std::endl; |
||||
|
} |
||||
|
// output goes to socket
|
||||
|
sout_.rdbuf(&bout); |
||||
|
return s; |
||||
|
} |
||||
|
|
||||
|
// write sout_ to socket (via bout)
|
||||
|
void socketif_t::wout() { |
||||
|
try { |
||||
|
boost::system::error_code ignored_error; |
||||
|
boost::asio::write(*socket_ptr, bout, boost::asio::transfer_all(), ignored_error); |
||||
|
socket_ptr->close(); // close the socket after each command input/ouput
|
||||
|
// This is need to in order to make the socket interface
|
||||
|
// acessible by HTTP GET via a socket client in a web server.
|
||||
|
} catch (std::exception& e) { |
||||
|
std::cerr << e.what() << std::endl; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
#endif |
||||
@ -0,0 +1,32 @@ |
|||||
|
// See LICENSE for license details.
|
||||
|
|
||||
|
#ifndef _RISCV_SOCKETIF_H |
||||
|
#define _RISCV_SOCKETIF_H |
||||
|
|
||||
|
#include "config.h" |
||||
|
|
||||
|
#ifdef HAVE_BOOST_ASIO |
||||
|
|
||||
|
#include <boost/algorithm/string.hpp> |
||||
|
#include <boost/regex.hpp> |
||||
|
#include <boost/asio.hpp> |
||||
|
|
||||
|
class socketif_t |
||||
|
{ |
||||
|
public: |
||||
|
socketif_t(); |
||||
|
~socketif_t(); |
||||
|
|
||||
|
std::string rin(std::ostream &sout_); // read input command string
|
||||
|
void wout(); // write output to socket
|
||||
|
|
||||
|
private: |
||||
|
// the following are needed for command socket interface
|
||||
|
boost::asio::io_service *io_service_ptr; |
||||
|
boost::asio::ip::tcp::acceptor *acceptor_ptr; |
||||
|
std::unique_ptr<boost::asio::ip::tcp::socket> socket_ptr; |
||||
|
boost::asio::streambuf bout; |
||||
|
}; |
||||
|
|
||||
|
#endif |
||||
|
#endif |
||||
Loading…
Reference in new issue