227 lines
10 KiB
C++
227 lines
10 KiB
C++
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <fstream>
|
|
#include <set>
|
|
#include "controller_capabilities.h"
|
|
|
|
///\brief Holds everything unique to the controller.
|
|
namespace Controller {
|
|
///\brief A class storing information about the cpu the server is running on.
|
|
class cpudata{
|
|
public:
|
|
std::string model;///<A string describing the model of the cpu.
|
|
int cores;///<The amount of cores in the cpu.
|
|
int threads;///<The amount of threads this cpu can run.
|
|
int mhz;///<The speed of the cpu in mhz.
|
|
int id;///<The id of the cpu in the system.
|
|
|
|
///\brief The default constructor
|
|
cpudata(){
|
|
model = "Unknown";
|
|
cores = 1;
|
|
threads = 1;
|
|
mhz = 0;
|
|
id = 0;
|
|
}
|
|
;
|
|
|
|
///\brief Fills the structure by parsing a given description.
|
|
///\param data A description of the cpu.
|
|
void fill(char * data){
|
|
int i;
|
|
i = 0;
|
|
if (sscanf(data, "model name : %n", &i) != EOF && i > 0){
|
|
model = (data + i);
|
|
}
|
|
if (sscanf(data, "cpu cores : %d", &i) == 1){
|
|
cores = i;
|
|
}
|
|
if (sscanf(data, "siblings : %d", &i) == 1){
|
|
threads = i;
|
|
}
|
|
if (sscanf(data, "physical id : %d", &i) == 1){
|
|
id = i;
|
|
}
|
|
if (sscanf(data, "cpu MHz : %d", &i) == 1){
|
|
mhz = i;
|
|
}
|
|
}
|
|
;
|
|
};
|
|
|
|
///\brief Checks the capabilities of the system.
|
|
///\param capa The location to store the capabilities.
|
|
void checkCapable(JSON::Value & capa){
|
|
capa.null();
|
|
std::ifstream cpuinfo("/proc/cpuinfo");
|
|
if (cpuinfo){
|
|
std::map<int, cpudata> cpus;
|
|
char line[300];
|
|
int proccount = -1;
|
|
while (cpuinfo.good()){
|
|
cpuinfo.getline(line, 300);
|
|
if (cpuinfo.fail()){
|
|
//empty lines? ignore them, clear flags, continue
|
|
if ( !cpuinfo.eof()){
|
|
cpuinfo.ignore();
|
|
cpuinfo.clear();
|
|
}
|
|
continue;
|
|
}
|
|
if (memcmp(line, "processor", 9) == 0){
|
|
proccount++;
|
|
}
|
|
cpus[proccount].fill(line);
|
|
}
|
|
//fix wrong core counts
|
|
std::map<int, int> corecounts;
|
|
for (int i = 0; i <= proccount; ++i){
|
|
corecounts[cpus[i].id]++;
|
|
}
|
|
//remove double physical IDs - we only want real CPUs.
|
|
std::set<int> used_physids;
|
|
int total_speed = 0;
|
|
int total_threads = 0;
|
|
for (int i = 0; i <= proccount; ++i){
|
|
if ( !used_physids.count(cpus[i].id)){
|
|
used_physids.insert(cpus[i].id);
|
|
JSON::Value thiscpu;
|
|
thiscpu["model"] = cpus[i].model;
|
|
thiscpu["cores"] = cpus[i].cores;
|
|
if (cpus[i].cores < 2 && corecounts[cpus[i].id] > cpus[i].cores){
|
|
thiscpu["cores"] = corecounts[cpus[i].id];
|
|
}
|
|
thiscpu["threads"] = cpus[i].threads;
|
|
if (thiscpu["cores"].asInt() > thiscpu["threads"].asInt()){
|
|
thiscpu["threads"] = thiscpu["cores"];
|
|
}
|
|
thiscpu["mhz"] = cpus[i].mhz;
|
|
capa["cpu"].append(thiscpu);
|
|
total_speed += cpus[i].cores * cpus[i].mhz;
|
|
total_threads += cpus[i].threads;
|
|
}
|
|
}
|
|
capa["speed"] = total_speed;
|
|
capa["threads"] = total_threads;
|
|
}
|
|
std::ifstream meminfo("/proc/meminfo");
|
|
if (meminfo){
|
|
char line[300];
|
|
int bufcache = 0;
|
|
while (meminfo.good()){
|
|
meminfo.getline(line, 300);
|
|
if (meminfo.fail()){
|
|
//empty lines? ignore them, clear flags, continue
|
|
if ( !meminfo.eof()){
|
|
meminfo.ignore();
|
|
meminfo.clear();
|
|
}
|
|
continue;
|
|
}
|
|
long long int i;
|
|
if (sscanf(line, "MemTotal : %Li kB", &i) == 1){
|
|
capa["mem"]["total"] = i / 1024;
|
|
}
|
|
if (sscanf(line, "MemFree : %Li kB", &i) == 1){
|
|
capa["mem"]["free"] = i / 1024;
|
|
}
|
|
if (sscanf(line, "SwapTotal : %Li kB", &i) == 1){
|
|
capa["mem"]["swaptotal"] = i / 1024;
|
|
}
|
|
if (sscanf(line, "SwapFree : %Li kB", &i) == 1){
|
|
capa["mem"]["swapfree"] = i / 1024;
|
|
}
|
|
if (sscanf(line, "Buffers : %Li kB", &i) == 1){
|
|
bufcache += i / 1024;
|
|
}
|
|
if (sscanf(line, "Cached : %Li kB", &i) == 1){
|
|
bufcache += i / 1024;
|
|
}
|
|
}
|
|
capa["mem"]["used"] = capa["mem"]["total"].asInt() - capa["mem"]["free"].asInt() - bufcache;
|
|
capa["mem"]["cached"] = bufcache;
|
|
capa["load"]["memory"] = ((capa["mem"]["used"].asInt() + (capa["mem"]["swaptotal"].asInt() - capa["mem"]["swapfree"].asInt())) * 100)
|
|
/ capa["mem"]["total"].asInt();
|
|
}
|
|
std::ifstream loadavg("/proc/loadavg");
|
|
if (loadavg){
|
|
char line[300];
|
|
int bufcache = 0;
|
|
loadavg.getline(line, 300);
|
|
//parse lines here
|
|
float onemin, fivemin, fifteenmin;
|
|
if (sscanf(line, "%f %f %f", &onemin, &fivemin, &fifteenmin) == 3){
|
|
capa["load"]["one"] = (long long int)(onemin * 100);
|
|
capa["load"]["five"] = (long long int)(onemin * 100);
|
|
capa["load"]["fifteen"] = (long long int)(onemin * 100);
|
|
}
|
|
}
|
|
|
|
//list available protocols and report about them
|
|
capa["connectors"]["RTMP"]["desc"] = "Enables the RTMP protocol which is used by Adobe Flash Player.";
|
|
capa["connectors"]["RTMP"]["deps"] = "";
|
|
capa["connectors"]["RTMP"]["optional"]["port"]["name"] = "TCP port";
|
|
capa["connectors"]["RTMP"]["optional"]["port"]["help"] = "TCP port to listen on - default if unprovided is 1935";
|
|
capa["connectors"]["RTMP"]["optional"]["port"]["type"] = "uint";
|
|
capa["connectors"]["RTMP"]["optional"]["interface"]["name"] = "Interface";
|
|
capa["connectors"]["RTMP"]["optional"]["interface"]["help"] = "Address of the interface to listen on - default if unprovided is all interfaces";
|
|
capa["connectors"]["RTMP"]["optional"]["interface"]["type"] = "str";
|
|
capa["connectors"]["RTMP"]["optional"]["username"]["name"] = "Username";
|
|
capa["connectors"]["RTMP"]["optional"]["username"]["help"] =
|
|
"Username to drop privileges to - default if unprovided means do not drop privileges";
|
|
capa["connectors"]["RTMP"]["optional"]["username"]["type"] = "str";
|
|
capa["connectors"]["TS"]["desc"] = "Enables the raw MPEG Transport Stream protocol over TCP.";
|
|
capa["connectors"]["TS"]["deps"] = "";
|
|
capa["connectors"]["TS"]["required"]["port"]["name"] = "TCP port";
|
|
capa["connectors"]["TS"]["required"]["port"]["help"] = "TCP port to listen on";
|
|
capa["connectors"]["TS"]["required"]["port"]["type"] = "uint";
|
|
capa["connectors"]["TS"]["required"]["args"]["name"] = "Stream";
|
|
capa["connectors"]["TS"]["required"]["args"]["help"] = "What streamname to serve - for multiple streams, add this protocol multiple times.";
|
|
capa["connectors"]["TS"]["required"]["args"]["type"] = "str";
|
|
capa["connectors"]["TS"]["optional"]["interface"]["name"] = "Interface";
|
|
capa["connectors"]["TS"]["optional"]["interface"]["help"] = "Address of the interface to listen on - default if unprovided is all interfaces";
|
|
capa["connectors"]["TS"]["optional"]["interface"]["type"] = "str";
|
|
capa["connectors"]["TS"]["optional"]["username"]["name"] = "Username";
|
|
capa["connectors"]["TS"]["optional"]["username"]["help"] = "Username to drop privileges to - default if unprovided means do not drop privileges";
|
|
capa["connectors"]["TS"]["optional"]["username"]["type"] = "str";
|
|
capa["connectors"]["HTTP"]["desc"] =
|
|
"Enables the generic HTTP listener, required by all other HTTP protocols. Needs other HTTP protocols enabled to do much of anything.";
|
|
capa["connectors"]["HTTP"]["deps"] = "";
|
|
capa["connectors"]["HTTP"]["optional"]["port"]["name"] = "TCP port";
|
|
capa["connectors"]["HTTP"]["optional"]["port"]["help"] = "TCP port to listen on - default if unprovided is 8080";
|
|
capa["connectors"]["HTTP"]["optional"]["port"]["type"] = "uint";
|
|
capa["connectors"]["HTTP"]["optional"]["interface"]["name"] = "Interface";
|
|
capa["connectors"]["HTTP"]["optional"]["interface"]["help"] = "Address of the interface to listen on - default if unprovided is all interfaces";
|
|
capa["connectors"]["HTTP"]["optional"]["interface"]["type"] = "str";
|
|
capa["connectors"]["HTTP"]["optional"]["username"]["name"] = "Username";
|
|
capa["connectors"]["HTTP"]["optional"]["username"]["help"] =
|
|
"Username to drop privileges to - default if unprovided means do not drop privileges";
|
|
capa["connectors"]["HTTP"]["optional"]["username"]["type"] = "str";
|
|
capa["connectors"]["HTTPProgressive"]["desc"] = "Enables HTTP protocol progressive streaming.";
|
|
capa["connectors"]["HTTPProgressive"]["deps"] = "HTTP";
|
|
capa["connectors"]["HTTPProgressive"]["optional"]["username"]["name"] = "Username";
|
|
capa["connectors"]["HTTPProgressive"]["optional"]["username"]["help"] =
|
|
"Username to drop privileges to - default if unprovided means do not drop privileges";
|
|
capa["connectors"]["HTTPProgressive"]["optional"]["username"]["type"] = "str";
|
|
capa["connectors"]["HTTPDynamic"]["desc"] = "Enables HTTP protocol Adobe-specific dynamic streaming (aka HDS).";
|
|
capa["connectors"]["HTTPDynamic"]["deps"] = "HTTP";
|
|
capa["connectors"]["HTTPDynamic"]["optional"]["username"]["name"] = "Username";
|
|
capa["connectors"]["HTTPDynamic"]["optional"]["username"]["help"] =
|
|
"Username to drop privileges to - default if unprovided means do not drop privileges";
|
|
capa["connectors"]["HTTPDynamic"]["optional"]["username"]["type"] = "str";
|
|
capa["connectors"]["HTTPSmooth"]["desc"] = "Enables HTTP protocol MicroSoft-specific smooth streaming through silverlight.";
|
|
capa["connectors"]["HTTPSmooth"]["deps"] = "HTTP";
|
|
capa["connectors"]["HTTPSmooth"]["optional"]["username"]["name"] = "Username";
|
|
capa["connectors"]["HTTPSmooth"]["optional"]["username"]["help"] =
|
|
"Username to drop privileges to - default if unprovided means do not drop privileges";
|
|
capa["connectors"]["HTTPSmooth"]["optional"]["username"]["type"] = "str";
|
|
capa["connectors"]["HTTPLive"]["desc"] = "Enables HTTP protocol Apple-style live streaming.";
|
|
capa["connectors"]["HTTPLive"]["deps"] = "HTTP";
|
|
capa["connectors"]["HTTPLive"]["optional"]["username"]["name"] = "Username";
|
|
capa["connectors"]["HTTPLive"]["optional"]["username"]["help"] =
|
|
"Username to drop privileges to - default if unprovided means do not drop privileges";
|
|
capa["connectors"]["HTTPLive"]["optional"]["username"]["type"] = "str";
|
|
}
|
|
|
|
}
|
|
|