|
|
|
#include "main.h"
|
|
|
|
|
|
|
|
// Main sets up the Application/GUI, and tries to retrieveLogin()
|
|
|
|
int main(int argc, char *argv[]) {
|
|
|
|
QApplication app(argc,argv);
|
|
|
|
QWidget *window = new QWidget;
|
|
|
|
|
|
|
|
domainLE = new QLineEdit;
|
|
|
|
tokenUsername = new QLineEdit;
|
|
|
|
tokenCode = new QLineEdit;
|
|
|
|
ntUsername = new QLineEdit;
|
|
|
|
ntPassword = new QLineEdit;
|
|
|
|
authSystemLE = new QLineEdit;
|
|
|
|
rememberCredentials = new QCheckBox;
|
|
|
|
vboxLayout = new QVBoxLayout;
|
|
|
|
vdiLoginLayout = new QFormLayout;
|
|
|
|
|
|
|
|
domainLE->setPlaceholderText("Horizon Domain");
|
|
|
|
tokenUsername->setPlaceholderText("Email address");
|
|
|
|
tokenUsername->setMinimumWidth(250);
|
|
|
|
tokenCode->setPlaceholderText("RSA Token code");
|
|
|
|
ntUsername->setPlaceholderText("Windows username");
|
|
|
|
ntPassword->setPlaceholderText("Windows password");
|
|
|
|
ntPassword->setEchoMode(QLineEdit::Password);
|
|
|
|
QPushButton loginBtn("Log In");
|
|
|
|
loginBtn.setToolTip("Click here to log in using the data entered.");
|
|
|
|
authSystemLE->setPlaceholderText("Optional Auth System");
|
|
|
|
// Try to populate form before adding to the layout
|
|
|
|
retrieveLogin();
|
|
|
|
|
|
|
|
vdiLoginLayout->addRow("Domain: ", domainLE);
|
|
|
|
vdiLoginLayout->addRow("Token Username:", tokenUsername);
|
|
|
|
vdiLoginLayout->addRow("RSA Token:", tokenCode);
|
|
|
|
vdiLoginLayout->addRow("NT Username:", ntUsername);
|
|
|
|
vdiLoginLayout->addRow("NT Password:", ntPassword);
|
|
|
|
vdiLoginLayout->addRow("Auth System:", authSystemLE);
|
|
|
|
vdiLoginLayout->addRow("Remember Credentials:", rememberCredentials);
|
|
|
|
|
|
|
|
vboxLayout->addItem(vdiLoginLayout);
|
|
|
|
vboxLayout->addWidget(&loginBtn);
|
|
|
|
|
|
|
|
QObject::connect(&loginBtn, &QPushButton::released, tryLogin);
|
|
|
|
|
|
|
|
window->setLayout(vboxLayout);
|
|
|
|
window->show();
|
|
|
|
return app.exec();
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
// walletOpen() returns a pointer to the kwallet, or throws an error if the wallet is unable to open.
|
|
|
|
KWallet::Wallet* walletOpen() {
|
|
|
|
KWallet::Wallet *wallet = KWallet::Wallet::openWallet(KWallet::Wallet::NetworkWallet(), 0, KWallet::Wallet::Synchronous);
|
|
|
|
if (wallet->hasFolder(KWallet::Wallet::FormDataFolder())) {
|
|
|
|
if (!wallet->setFolder(KWallet::Wallet::FormDataFolder())) {
|
|
|
|
std::cout << "Wallet folder could not be set." << std::endl;
|
|
|
|
throw -1;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
std::cout << "Wallet did not have folder, creating." << std::endl;
|
|
|
|
if(wallet->createFolder(KWallet::Wallet::FormDataFolder())) {
|
|
|
|
if (!wallet->setFolder(KWallet::Wallet::FormDataFolder())) {
|
|
|
|
std::cout << "Wallet folder could not be set." << std::endl;
|
|
|
|
throw -1;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
std::cout << "Could not create folder." << std::endl;
|
|
|
|
throw -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return wallet;
|
|
|
|
}
|
|
|
|
|
|
|
|
// retrieve login information from kwallet using walletOpen() or die
|
|
|
|
void retrieveLogin() {
|
|
|
|
try {
|
|
|
|
wallet = walletOpen();
|
|
|
|
} catch (int e) {
|
|
|
|
exit(-1);
|
|
|
|
}
|
|
|
|
QMap<QString, QString> map;
|
|
|
|
if (wallet->readMap("http://rsa-vdi.nightmare.haus/#authInfo", map) == 0) {
|
|
|
|
domainLE->setText(map[QString("domain")]);
|
|
|
|
tokenUsername->setText(map[QString("tokenUsername")]);
|
|
|
|
ntUsername->setText(map[QString("ntUsername")]);
|
|
|
|
ntPassword->setText(map[QString("ntPassword")]);
|
|
|
|
authSystemLE->setText(map[QString("authSystem")]);
|
|
|
|
if (tokenUsername->text().length() > 0) {
|
|
|
|
rememberCredentials->setChecked(true);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
std::cout << "Error loading from wallet, is it populated?" << std::endl;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// save login information to kwallet using walletOpen() or die
|
|
|
|
void saveLogin() {
|
|
|
|
try {
|
|
|
|
wallet = walletOpen();
|
|
|
|
} catch (int e) {
|
|
|
|
exit(-1);
|
|
|
|
}
|
|
|
|
QMap<QString, QString> map;
|
|
|
|
map[QString("domain")] = domainLE->text();
|
|
|
|
map[QString("tokenUsername")] = tokenUsername->text();
|
|
|
|
map[QString("ntUsername")] = ntUsername->text();
|
|
|
|
map[QString("ntPassword")] = ntPassword->text();
|
|
|
|
map[QString("authSystem")] = authSystemLE->text();
|
|
|
|
|
|
|
|
if (!wallet->writeMap("http://rsa-vdi.nightmare.haus/#authInfo", map)) {
|
|
|
|
std::cout << "Error writing to Wallet!" << std::endl;
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
void tryLogin() {
|
|
|
|
if (rememberCredentials->isChecked()) {
|
|
|
|
saveLogin();
|
|
|
|
}
|
|
|
|
if(debug) {
|
|
|
|
std::cout << "tryLogin()" << std::endl;
|
|
|
|
std::cout << "\tToken Username: " << tokenUsername->text().toStdString() << std::endl;
|
|
|
|
std::cout << "\tToken Code: " << tokenCode->text().toStdString() << std::endl;
|
|
|
|
std::cout << "\tntUsername: " << ntUsername->text().toStdString() << std::endl;
|
|
|
|
std::cout << "\tntPassword Length: " << ntPassword->text().length() << std::endl;
|
|
|
|
std::cout << "\tAuth System: " << authSystemLE->text().toStdString() << std::endl;
|
|
|
|
if (rememberCredentials->isChecked()) {
|
|
|
|
std::cout << "\tRemember creds: true" << std::endl;
|
|
|
|
} else {
|
|
|
|
std::cout << "\tRemember creds: false" << std::endl;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// -q, --nonInteractive Connect automatically if enough values are given on the command line.
|
|
|
|
QString command = "vmware-view -q";
|
|
|
|
if (domainLE->text().length() > 0) {
|
|
|
|
// -s, --serverURL=<broker URL> Specify connection broker.
|
|
|
|
command += " -s \'" + domainLE->text() + "\'";
|
|
|
|
}
|
|
|
|
if (authSystemLE->text().length() > 0) {
|
|
|
|
// -d, --domainName=<domain name> Specify domain for password authentication
|
|
|
|
command += " -d \'" + authSystemLE->text() + "\'";
|
|
|
|
}
|
|
|
|
if (tokenUsername->text().length() > 0) {
|
|
|
|
// -t, --tokenUserName=<token user name> Specify user name for RSA or Radius authentication.
|
|
|
|
command += " -t \'" + tokenUsername->text() + "\'";
|
|
|
|
}
|
|
|
|
if (tokenCode->text().length() > 0 ) {
|
|
|
|
// -c, --passcode=<passcode> Specify passcode for RSA or Radius authentication.
|
|
|
|
command += " -c \'" + tokenCode->text() + "\'";
|
|
|
|
}
|
|
|
|
if (ntUsername->text().length() > 0) {
|
|
|
|
// -u, --userName=<user name> Specify user name for password authentication.
|
|
|
|
command += " -u \'" + ntUsername->text() + "\'";
|
|
|
|
}
|
|
|
|
if (ntPassword->text().length() > 0) {
|
|
|
|
//-p, --password=<password> Specify password for password authentication.
|
|
|
|
command += " -p \'" + ntPassword->text() + "\'";
|
|
|
|
}
|
|
|
|
if (debug) {
|
|
|
|
std::cout << command.toStdString() << std::endl;
|
|
|
|
} else {
|
|
|
|
system(command.toStdString().c_str());
|
|
|
|
}
|
|
|
|
exit(0);
|
|
|
|
}
|