ircDDBGateway/RemoteControl/RemoteControlAppD.cpp
2018-05-09 19:23:17 +01:00

288 lines
8.4 KiB
C++

/*
* Copyright (C) 2013,2014 by Jonathan Naylor G4KLX
*
* This program 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 2 of the License, or
* (at your option) any later version.
*
* This program 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 this program; if not, write to the Free Software
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#include "RemoteControlRemoteControlHandler.h"
#include "RemoteControlConfig.h"
#include "RemoteControlDefs.h"
#include "DStarDefines.h"
#include "SHA256.h"
#include "Defs.h"
#include <wx/cmdline.h>
const wxChar* NAME_OPTION = wxT("name");
const wxChar* REPEATER_PARAM = wxT("Callsign");
const wxChar* ACTION_PARAM = wxT("Action");
const wxChar* RECONNECT_PARAM = wxT("Param1");
const wxChar* REFLECTOR_PARAM = wxT("Param2");
void sendHash(CRemoteControlRemoteControlHandler* handler, const wxString& password, unsigned int rnd)
{
wxASSERT(handler != NULL);
unsigned int len = password.Len() + sizeof(unsigned int);
unsigned char* in = new unsigned char[len];
unsigned char* out = new unsigned char[32U];
::memcpy(in, &rnd, sizeof(unsigned int));
for (unsigned int i = 0U; i < password.Len(); i++)
in[i + sizeof(unsigned int)] = password.GetChar(i);
CSHA256 sha256;
sha256.buffer(in, len, out);
handler->sendHash(out, 32U);
delete[] in;
delete[] out;
}
int main(int argc, char** argv)
{
bool res = ::wxInitialize();
if (!res) {
::fprintf(stderr, "remotecontrold: failed to initialise the wxWidgets library, exiting\n");
return -1;
}
wxCmdLineParser parser(argc, argv);
parser.AddOption(NAME_OPTION, wxEmptyString, wxEmptyString, wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_OPTIONAL);
parser.AddParam(REPEATER_PARAM, wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_OPTIONAL);
parser.AddParam(ACTION_PARAM, wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_OPTIONAL);
parser.AddParam(RECONNECT_PARAM, wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_OPTIONAL);
parser.AddParam(REFLECTOR_PARAM, wxCMD_LINE_VAL_STRING, wxCMD_LINE_PARAM_OPTIONAL);
int cmd = parser.Parse();
if (cmd != 0) {
::wxUninitialize();
return 0;
}
wxString name;
parser.Found(NAME_OPTION, &name);
if (parser.GetParamCount() < 2U) {
::fprintf(stderr, "remotecontrold: invalid command line usage: remotecontrold [-name <name>] <repeater> link <reconnect> <reflector>\n");
::fprintf(stderr, " remotecontrold [-name <name>] <repeater> unlink\n");
::fprintf(stderr, " remotecontrold [-name <name>] <starnet> drop <user>\n");
::fprintf(stderr, " remotecontrold [-name <name>] <starnet> drop all\n");
::wxUninitialize();
return 1;
}
wxString repeater = parser.GetParam(0U);
repeater.Replace(wxT("_"), wxT(" "));
repeater.resize(LONG_CALLSIGN_LENGTH, wxT(' '));
repeater.MakeUpper();
wxString actionText = parser.GetParam(1U);
wxString user; // For STARnet Digital
wxString reflector; // For linking
RECONNECT reconnect = RECONNECT_NEVER; // For linking
if (actionText.IsSameAs(wxT("link"), false)) {
if (parser.GetParamCount() < 4U) {
::fprintf(stderr, "remotecontrold: invalid command line usage: remotecontrold [-name <name>] <repeater> link <reconnect> <reflector>\n");
::wxUninitialize();
return 1;
}
wxString reconnectText = parser.GetParam(2U);
reflector = parser.GetParam(3U);
reflector.Replace(wxT("_"), wxT(" "));
reflector.resize(LONG_CALLSIGN_LENGTH, wxT(' '));
reflector.MakeUpper();
if (reconnectText.IsSameAs(wxT("never"), false)) {
reconnect = RECONNECT_NEVER;
} else if (reconnectText.IsSameAs(wxT("fixed"), false)) {
reconnect = RECONNECT_FIXED;
} else if (reconnectText.IsSameAs(wxT("5"), false)) {
reconnect = RECONNECT_5MINS;
} else if (reconnectText.IsSameAs(wxT("10"), false)) {
reconnect = RECONNECT_10MINS;
} else if (reconnectText.IsSameAs(wxT("15"), false)) {
reconnect = RECONNECT_15MINS;
} else if (reconnectText.IsSameAs(wxT("20"), false)) {
reconnect = RECONNECT_20MINS;
} else if (reconnectText.IsSameAs(wxT("25"), false)) {
reconnect = RECONNECT_25MINS;
} else if (reconnectText.IsSameAs(wxT("30"), false)) {
reconnect = RECONNECT_30MINS;
} else if (reconnectText.IsSameAs(wxT("60"), false)) {
reconnect = RECONNECT_60MINS;
} else if (reconnectText.IsSameAs(wxT("90"), false)) {
reconnect = RECONNECT_90MINS;
} else if (reconnectText.IsSameAs(wxT("120"), false)) {
reconnect = RECONNECT_120MINS;
} else if (reconnectText.IsSameAs(wxT("180"), false)) {
reconnect = RECONNECT_180MINS;
} else {
::fprintf(stderr, "remotecontrold: invalid reconnect value passed\n");
::wxUninitialize();
return 1;
}
} else if (actionText.IsSameAs(wxT("unlink"), false)) {
reconnect = RECONNECT_NEVER;
reflector.Clear();
} else if (actionText.IsSameAs(wxT("drop"), false)) {
if (parser.GetParamCount() < 3U) {
::fprintf(stderr, "remotecontrold: invalid command line usage: remotecontrold [-name <name>] <starnet> drop <user>\n");
::fprintf(stderr, " remotecontrold [-name <name>] <starnet> drop all\n");
::wxUninitialize();
return 1;
}
user = parser.GetParam(2U);
user.Replace(wxT("_"), wxT(" "));
user.resize(LONG_CALLSIGN_LENGTH, wxT(' '));
user.MakeUpper();
} else {
::fprintf(stderr, "remotecontrold: invalid action value passed, only drop, link or unlink are allowed\n");
::wxUninitialize();
return 1;
}
CRemoteControlConfig config(new wxConfig(APPLICATION_NAME), name);
wxString address, password;
unsigned int port;
config.getConfig(address, port, password);
if (address.IsEmpty() || port == 0U || password.IsEmpty()) {
::fprintf(stderr, "remotecontrold: no address, port, or password is set\n");
::wxUninitialize();
return 1;
}
CRemoteControlRemoteControlHandler handler(address, port);
bool ret = handler.open();
if (!ret) {
::fprintf(stderr, "remotecontrold: uanble to open the UDP port\n");
::wxUninitialize();
return 1;
}
ret = handler.login();
if (!ret) {
handler.close();
::fprintf(stderr, "remotecontrold: uanble to login to the gateway/starnetserver\n");
::wxUninitialize();
return 1;
}
unsigned int count = 0U;
while (count < 10U) {
::wxMilliSleep(100UL);
RC_TYPE type = handler.readType();
if (type == RCT_RANDOM)
break;
if (type == RCT_NONE)
handler.retry();
count++;
}
if (count >= 10U) {
handler.close();
::fprintf(stderr, "remotecontrold: unable to get a response from the gateway/starnetserver\n");
::wxUninitialize();
return 1;
}
unsigned int rnd = handler.readRandom();
sendHash(&handler, password, rnd);
count = 0U;
while (count < 10U) {
::wxMilliSleep(100UL);
RC_TYPE type = handler.readType();
if (type == RCT_ACK)
break;
if (type == RCT_NAK) {
handler.close();
::fprintf(stderr, "remotecontrold: invalid password sent to the gateway/starnetserver\n");
::wxUninitialize();
return 1;
}
if (type == RCT_NONE)
handler.retry();
count++;
}
if (count >= 10U) {
handler.close();
::fprintf(stderr, "remotecontrold: unable to get a response from the gateway/starnetserver\n");
::wxUninitialize();
return 1;
}
handler.setLoggedIn(true);
if (actionText.IsSameAs(wxT("drop"), false))
handler.logoff(repeater, user);
else
handler.link(repeater, reconnect, reflector);
count = 0U;
while (count < 10U) {
::wxMilliSleep(100UL);
RC_TYPE type = handler.readType();
if (type == RCT_ACK)
break;
if (type == RCT_NAK) {
handler.close();
::fprintf(stderr, "remotecontrold: drop/link/unlink command rejected by the gateway/starnetserver\n");
::wxUninitialize();
return 1;
}
if (type == RCT_NONE)
handler.retry();
count++;
}
if (count >= 10U) {
handler.close();
::fprintf(stderr, "remotecontrold: unable to get a response from the gateway/starnetserver\n");
::wxUninitialize();
return 1;
}
::fprintf(stdout, "remotecontrold: command accepted by the gateway/starnetserver\n");
handler.logout();
handler.close();
::wxUninitialize();
return 0;
}