mirror of
https://github.com/etlegacy/etlegacy-tools.git
synced 2024-11-25 05:21:22 +00:00
110 lines
3 KiB
C++
110 lines
3 KiB
C++
/*
|
|
* ET: Legacy
|
|
* Copyright (C) 2012 Jan Simek <mail@etlegacy.com>
|
|
*
|
|
* This file is part of ET: Legacy - http://www.etlegacy.com
|
|
*
|
|
* ET: Legacy is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* ET: Legacy is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with ET: Legacy. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
#include <boost/asio.hpp>
|
|
#include <boost/program_options.hpp>
|
|
#include <boost/lexical_cast.hpp>
|
|
|
|
#include "connection.h"
|
|
|
|
using boost::asio::ip::udp;
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
try
|
|
{
|
|
/**
|
|
* Program options
|
|
*/
|
|
boost::program_options::options_description desc("[OPTIONS]");
|
|
desc.add_options()
|
|
("help,h", "show this message")
|
|
("server,s", boost::program_options::value<std::string>(),
|
|
"server to query")
|
|
("port,p",
|
|
boost::program_options::value<unsigned int>()->default_value(27960),
|
|
"port on the server")
|
|
("message,m",
|
|
boost::program_options::value<std::string>()->default_value("getstatus"),
|
|
"message to be sent")
|
|
("raw,r", "don't parse the server response")
|
|
;
|
|
boost::program_options::variables_map var_map;
|
|
boost::program_options::store(boost::program_options::parse_command_line(argc, argv, desc), var_map);
|
|
boost::program_options::notify(var_map);
|
|
|
|
if (var_map.count("help"))
|
|
{
|
|
std::cout << desc << std::endl;
|
|
return EXIT_SUCCESS;
|
|
}
|
|
else if (!var_map.count("server"))
|
|
{
|
|
std::cerr << "Usage: " << argv[0] << " -s <server>" << std::endl;
|
|
std::cerr << "Append -h or --help to see all program options." << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
/**
|
|
* Send the request
|
|
*/
|
|
Connection client(var_map["server"].as<std::string>(),
|
|
var_map["port"].as<unsigned int>(),
|
|
var_map["message"].as<std::string>());
|
|
|
|
char data[1024];
|
|
boost::system::error_code ec;
|
|
std::size_t n = client.ReceiveMessage(boost::asio::buffer(data),
|
|
boost::posix_time::seconds(10),
|
|
ec);
|
|
|
|
if (ec)
|
|
{
|
|
std::cout << "Receive error: " << ec.message() << std::endl;
|
|
}
|
|
else
|
|
{
|
|
if (var_map.count("raw"))
|
|
{
|
|
std::cout.write(data, n);
|
|
}
|
|
else
|
|
{
|
|
client.ParseMessage(std::string(data, n));
|
|
}
|
|
}
|
|
}
|
|
catch (std::exception& e)
|
|
{
|
|
std::cerr << "Exception caught: " << e.what() << std::endl;
|
|
// return 1;
|
|
}
|
|
catch (...)
|
|
{
|
|
std::cerr << "Exception of unknown type!" << std::endl;
|
|
// return 1;
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|