256 lines
9.1 KiB
C++
256 lines
9.1 KiB
C++
/// \file Admin/main.cpp
|
|
/// Attempted administration tool for DDVTECH Clients.
|
|
/// Never finished - perhaps now obsolete...?
|
|
/// \todo This could serve as a basis for a new, more robust, control method for gearbox / the API.
|
|
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <cstdlib>
|
|
#include <cstdio>
|
|
#include <stdlib.h>
|
|
|
|
int getparam (std::string input) {
|
|
if(input.size() <= 3) { return -1;}
|
|
int result = 0;
|
|
for (int i = 3; i < input.size(); i++) {
|
|
if ( input[i] < '0' || input[i] > '9') { return -2; }
|
|
result = result * 10;
|
|
result += input[i] - '0';
|
|
}
|
|
return result;
|
|
}
|
|
|
|
std::string getstringparam(std::string input) {
|
|
std::string result;
|
|
for (int i = 3; i < input.size(); i++) {
|
|
result.push_back(input[i]);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void GetThroughPut(int * Down, int * Up){
|
|
//deze file bevat o.a. totaal bytes verstuurd/ontvangen sinds de interface up ging, voor alle interfaces.
|
|
//voor een snelheidsmeting moet je dus 2X lezen en het verschil gebruiken...
|
|
std::ifstream netdev ("/proc/net/dev");
|
|
std::string line;
|
|
int tmpDown, tmpUp;
|
|
int ret = 0;
|
|
*Down = 0;
|
|
*Up = 0;
|
|
while (netdev.good()){
|
|
getline(netdev, line);
|
|
//probeer een geldige interface te parsen - 2 = 2 matches, en dus goed geparsed.
|
|
ret = sscanf(line.c_str(), " %*[^:]: %i %*i %*i %*i %*i %*i %*i %*i %i %*i %*i %*i %*i %*i %*i %*i", &tmpDown, &tmpUp);
|
|
if (ret == 2){*Down += tmpDown;*Up += tmpUp;}
|
|
}
|
|
netdev.close();
|
|
}//GetThroughPut
|
|
|
|
std::string MeasureThroughPut(){
|
|
std::stringstream output;
|
|
int frstDown, frstUp;
|
|
int totDown, totUp;
|
|
GetThroughPut(&frstDown, &frstUp);
|
|
sleep(5);
|
|
GetThroughPut(&totDown, &totUp);
|
|
//return totaal bytes down, up, gemiddelde bytes per seconde over de afgelopen 5 secs down, up.
|
|
output << totDown << " " << totUp << " " << ((totDown - frstDown)/5) << " " << ((totUp - frstUp)/5);
|
|
return output.str();
|
|
}//MeasureThroughPut
|
|
|
|
std::string GetConnectedUsers(){
|
|
std::string output;
|
|
//laat ps aux de processen Client_PLS opvragen, zonder de grep zelf, en tel het aantal lines.
|
|
system("ps aux | grep Client_PLS | grep -v grep | wc -l > ./tmpfile");
|
|
//lees de file, en return de inhoud
|
|
std::ifstream tmpfile ("./tmpfile");
|
|
tmpfile >> output;
|
|
tmpfile.close();
|
|
return output;
|
|
}//GetConnectedUsers
|
|
|
|
void readpreset( unsigned int values[], std::string & filename ) {
|
|
std::ifstream presetfile ("preset");
|
|
presetfile >> filename;
|
|
for (int i = 0; i < 9; i++ ) { presetfile >> values[i]; }
|
|
presetfile.close();
|
|
}
|
|
|
|
void writepreset( unsigned int values[], std::string filename ) {
|
|
std::ofstream presetfile ("preset");
|
|
presetfile << filename;
|
|
for (int i = 0; i < 9; i++ ) {
|
|
presetfile << values[i];
|
|
presetfile << "\n";
|
|
}
|
|
presetfile.close();
|
|
}
|
|
|
|
void writesh( unsigned int values[], std::string filename ) {
|
|
std::ofstream shfile ("run.sh");
|
|
shfile << "ffmpeg -i " << filename << " -f flv -re ";
|
|
if (values[0] != 0 && values[1] != 0) { shfile << "-s " << values[0] << "x" << values[1] << " "; }
|
|
if (values[2] != 0) { shfile << "-b " << values[2] << " "; }
|
|
if (values[3] != 0) { shfile << "-gop " << values[3] << " "; }
|
|
if (values[4] != 0) { shfile << "-r " << values[4] << " "; }
|
|
if (values[5] != 0) { shfile << "-ab " << values[5] << " "; }
|
|
if (values[6] != 0) { shfile << "-ar " << values[6] << " "; }
|
|
shfile << "rtmp://projectlivestream.com/oflaDemo/test &";
|
|
shfile.close();
|
|
system("sh ./run.sh");
|
|
}
|
|
|
|
int main() {
|
|
unsigned int values[9];
|
|
std::string inputcommand = "";
|
|
bool connection = true;
|
|
int tempresult;
|
|
std::string filename = "";
|
|
readpreset(values, filename);
|
|
while (connection) {
|
|
std::cin >> inputcommand;
|
|
switch (inputcommand[0]) {
|
|
case 'G':
|
|
switch (inputcommand[1]) {
|
|
case 'S':
|
|
switch(inputcommand[2]) {
|
|
case 'H': std::cout << "OK" << values[0] << "\n"; break;
|
|
case 'W': std::cout << "OK" << values[1] << "\n"; break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'V':
|
|
switch(inputcommand[2]) {
|
|
case 'B': std::cout << "OK" << values[2] << "\n"; break;
|
|
case 'G': std::cout << "OK" << values[3] << "\n"; break;
|
|
case 'R': std::cout << "OK" << values[4] << "\n"; break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'A':
|
|
switch(inputcommand[2]) {
|
|
case 'B': std::cout << "OK" << values[5] << "\n"; break;
|
|
case 'R': std::cout << "OK" << values[6] << "\n"; break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'U':
|
|
switch(inputcommand[2]) {
|
|
case 'F': std::cout << "OK" << values[7] << "\n"; break;
|
|
case 'P': std::cout << "OK" << values[8] << "\n"; break;
|
|
case 'C': std::cout << "OK" << GetConnectedUsers() << "\n"; break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'F':
|
|
switch(inputcommand[2]) {
|
|
case 'N': std::cout << "OK" << filename << "\n"; break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'G':
|
|
switch(inputcommand[2]) {
|
|
case 'T': std::cout << "OK" << MeasureThroughPut() << "\n"; break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'S':
|
|
switch (inputcommand[1]) {
|
|
case 'S':
|
|
switch(inputcommand[2]) {
|
|
case 'H':
|
|
tempresult = getparam(inputcommand);
|
|
if (tempresult < 0) { std::cout << "ER\n"; } else { values[0] = tempresult; std::cout << "OK\n"; }
|
|
break;
|
|
case 'W':
|
|
tempresult = getparam(inputcommand);
|
|
tempresult = getparam(inputcommand);
|
|
if (tempresult < 0) { std::cout << "ER\n"; } else { values[1] = tempresult; std::cout << "OK\n"; }
|
|
break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'V':
|
|
switch(inputcommand[2]) {
|
|
case 'B':
|
|
tempresult = getparam(inputcommand);
|
|
if (tempresult < 0) { std::cout << "ER\n"; } else { values[2] = tempresult; std::cout << "OK\n"; }
|
|
break;
|
|
case 'G':
|
|
tempresult = getparam(inputcommand);
|
|
if (tempresult < 0) { std::cout << "ER\n"; } else { values[3] = tempresult; std::cout << "OK\n"; }
|
|
break;
|
|
case 'R':
|
|
tempresult = getparam(inputcommand);
|
|
if (tempresult < 0) { std::cout << "ER\n"; } else { values[4] = tempresult; std::cout << "OK\n"; }
|
|
break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'A':
|
|
switch(inputcommand[2]) {
|
|
case 'B':
|
|
tempresult = getparam(inputcommand);
|
|
if (tempresult < 0) { std::cout << "ER\n"; } else { values[5] = tempresult; std::cout << "OK\n"; }
|
|
break;
|
|
case 'R':
|
|
tempresult = getparam(inputcommand);
|
|
if (tempresult < 0) { std::cout << "ER\n"; } else { values[6] = tempresult; std::cout << "OK\n"; }
|
|
break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'U':
|
|
switch(inputcommand[2]) {
|
|
case 'F':
|
|
tempresult = getparam(inputcommand);
|
|
if (tempresult < 0) { std::cout << "ER\n"; } else { values[7] = tempresult; std::cout << "OK\n"; }
|
|
break;
|
|
case 'P':
|
|
tempresult = getparam(inputcommand);
|
|
if (tempresult < 0) { std::cout << "ER\n"; } else { values[8] = tempresult; std::cout << "OK\n"; }
|
|
break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'F':
|
|
switch(inputcommand[2]) {
|
|
case 'N':
|
|
filename = getstringparam(inputcommand); std::cout << "OK\n";
|
|
break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'C':
|
|
switch (inputcommand[1] ) {
|
|
case 'C':
|
|
switch (inputcommand[2]) {
|
|
case 'D': std::cout << "OK\n"; connection = false; break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
case 'S':
|
|
switch (inputcommand[2]) {
|
|
case 'R': std::cout << "OK\n"; readpreset(values, filename); break;
|
|
case 'S': std::cout << "OK\n"; writepreset(values, filename); break;
|
|
case 'A': std::cout << "OK\n"; writesh(values, filename); break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
break;
|
|
default: std::cout << "ER\n"; break;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|