mirror of
https://github.com/slsdetectorgroup/slsDetectorPackage.git
synced 2025-04-26 08:10:02 +02:00
140 lines
3.8 KiB
C++
Executable File
140 lines
3.8 KiB
C++
Executable File
#include "Receiver.h"
|
||
#include "sls_detector_exceptions.h"
|
||
#include "versionAPI.h"
|
||
#include "container_utils.h"
|
||
#include "logger.h"
|
||
|
||
#include <cstdlib>
|
||
#include <fstream>
|
||
#include <getopt.h>
|
||
#include <iostream>
|
||
#include <map>
|
||
#include <sstream>
|
||
#include <string>
|
||
#include <syscall.h>
|
||
|
||
|
||
Receiver::Receiver(int argc, char *argv[]):
|
||
tcpipInterface (nullptr) {
|
||
|
||
// options
|
||
std::map<std::string, std::string> configuration_map;
|
||
int tcpip_port_no = 1954;
|
||
uid_t userid = -1;
|
||
|
||
//parse command line for config
|
||
static struct option long_options[] = {
|
||
// These options set a flag.
|
||
//{"verbose", no_argument, &verbose_flag, 1},
|
||
// These options don’t set a flag. We distinguish them by their indices.
|
||
{"rx_tcpport", required_argument, nullptr, 't'}, //TODO change or backward compatible to "port, p"?
|
||
{"uid", required_argument, nullptr, 'u'},
|
||
{"version", no_argument, nullptr, 'v'},
|
||
{"help", no_argument, nullptr, 'h'},
|
||
{nullptr, 0, nullptr, 0}
|
||
};
|
||
|
||
//initialize global optind variable (required when instantiating multiple receivers in the same process)
|
||
optind = 1;
|
||
// getopt_long stores the option index here.
|
||
int option_index = 0;
|
||
int c = 0;
|
||
|
||
while ( c != -1 ){
|
||
c = getopt_long (argc, argv, "hvf:t:u:", long_options, &option_index);
|
||
|
||
// Detect the end of the options.
|
||
if (c == -1)
|
||
break;
|
||
|
||
switch(c){
|
||
|
||
case 't':
|
||
sscanf(optarg, "%d", &tcpip_port_no);
|
||
break;
|
||
|
||
case 'u':
|
||
if (sscanf(optarg, "%u", &userid) != 1) {
|
||
throw sls::RuntimeError("Could not scan uid");
|
||
}
|
||
break;
|
||
|
||
case 'v':
|
||
std::cout << "SLS Receiver Version: " << GITBRANCH << " (0x" << std::hex << APIRECEIVER << ")" << std::endl;
|
||
FILE_LOG(logINFOBLUE) << "Exiting [ Tid: " << syscall(SYS_gettid) << " ]";
|
||
exit(EXIT_SUCCESS);
|
||
|
||
case 'h':
|
||
default:
|
||
std::cout << std::endl;
|
||
|
||
std::string help_message = "Usage: " + std::string(argv[0]) + " [arguments]\n"
|
||
+ "Possible arguments are:\n"
|
||
+ "\t-t, --rx_tcpport <port> : TCP Communication Port with client. \n"
|
||
+ "\t-u, --uid <user id> : Set effective user id if receiver \n"
|
||
+ "\t started with privileges. \n\n";
|
||
|
||
//std::cout << help_message << std::endl;
|
||
throw sls::RuntimeError(help_message);
|
||
}
|
||
}
|
||
|
||
// set effective id if provided
|
||
if (userid != static_cast<uid_t>(-1)) {
|
||
if (geteuid() == userid) {
|
||
FILE_LOG(logINFO) << "Process already has the same Effective UID " << userid;
|
||
} else {
|
||
if (seteuid(userid) != 0) {
|
||
std::ostringstream oss;
|
||
oss << "Could not set Effective UID to " << userid;
|
||
throw sls::RuntimeError(oss.str());
|
||
}
|
||
if(geteuid() != userid) {
|
||
std::ostringstream oss;
|
||
oss << "Could not set Effective UID to " << userid << ". Got " << geteuid();
|
||
throw sls::RuntimeError(oss.str());
|
||
}
|
||
FILE_LOG(logINFO) << "Process Effective UID changed to " << userid;
|
||
}
|
||
}
|
||
|
||
// might throw an exception
|
||
tcpipInterface = sls::make_unique<ClientInterface>(tcpip_port_no);
|
||
}
|
||
|
||
|
||
Receiver::Receiver(int tcpip_port_no)
|
||
{
|
||
// might throw an exception
|
||
tcpipInterface = sls::make_unique<ClientInterface>(tcpip_port_no);
|
||
}
|
||
|
||
int64_t Receiver::getReceiverVersion(){
|
||
return tcpipInterface->getReceiverVersion();
|
||
}
|
||
|
||
|
||
void Receiver::registerCallBackStartAcquisition(int (*func)(
|
||
std::string, std::string, uint64_t, uint32_t, void*),void *arg){
|
||
tcpipInterface->registerCallBackStartAcquisition(func,arg);
|
||
}
|
||
|
||
|
||
void Receiver::registerCallBackAcquisitionFinished(
|
||
void (*func)(uint64_t, void*),void *arg){
|
||
tcpipInterface->registerCallBackAcquisitionFinished(func,arg);
|
||
}
|
||
|
||
|
||
void Receiver::registerCallBackRawDataReady(void (*func)(char*,
|
||
char*, uint32_t, void*),void *arg){
|
||
tcpipInterface->registerCallBackRawDataReady(func,arg);
|
||
}
|
||
|
||
|
||
void Receiver::registerCallBackRawDataModifyReady(void (*func)(char*,
|
||
char*, uint32_t &, void*),void *arg){
|
||
tcpipInterface->registerCallBackRawDataModifyReady(func,arg);
|
||
}
|
||
|