73 lines
2.0 KiB
C++
73 lines
2.0 KiB
C++
#define PY_SSIZE_T_CLEAN
|
|
#include <Python.h>
|
|
|
|
#include "proxytun/settings.cpp"
|
|
#include "proxytun/proxytun.cpp"
|
|
#include "classes/netfilter.cpp"
|
|
#include <syncstream>
|
|
#include <iostream>
|
|
#include <stdexcept>
|
|
#include <cstdlib>
|
|
|
|
using namespace std;
|
|
using namespace Firegex::PyProxy;
|
|
using Firegex::NfQueue::MultiThreadQueue;
|
|
|
|
ssize_t read_check(int __fd, void *__buf, size_t __nbytes){
|
|
ssize_t bytes = read(__fd, __buf, __nbytes);
|
|
if (bytes == 0){
|
|
cerr << "[fatal] [updater] read() returned EOF" << endl;
|
|
throw invalid_argument("read() returned EOF");
|
|
}
|
|
if (bytes < 0){
|
|
cerr << "[fatal] [updater] read() returned an error" << bytes << endl;
|
|
throw invalid_argument("read() returned an error");
|
|
}
|
|
return bytes;
|
|
}
|
|
|
|
void config_updater (){
|
|
while (true){
|
|
uint32_t code_size;
|
|
read_check(STDIN_FILENO, &code_size, 4);
|
|
vector<uint8_t> code(code_size);
|
|
read_check(STDIN_FILENO, code.data(), code_size);
|
|
cerr << "[info] [updater] Updating configuration" << endl;
|
|
try{
|
|
config.reset(new PyCodeConfig(code));
|
|
cerr << "[info] [updater] Config update done" << endl;
|
|
osyncstream(cout) << "ACK OK" << endl;
|
|
}catch(const std::exception& e){
|
|
cerr << "[error] [updater] Failed to build new configuration!" << endl;
|
|
osyncstream(cout) << "ACK FAIL " << e.what() << endl;
|
|
}
|
|
}
|
|
}
|
|
|
|
int main(int argc, char *argv[]){
|
|
|
|
Py_Initialize();
|
|
atexit(Py_Finalize);
|
|
|
|
if (freopen(nullptr, "rb", stdin) == nullptr){ // We need to read from stdin binary data
|
|
cerr << "[fatal] [main] Failed to reopen stdin in binary mode" << endl;
|
|
return 1;
|
|
}
|
|
int n_of_threads = 1;
|
|
char * n_threads_str = getenv("NTHREADS");
|
|
if (n_threads_str != nullptr) n_of_threads = ::atoi(n_threads_str);
|
|
if(n_of_threads <= 0) n_of_threads = 1;
|
|
|
|
config.reset(new PyCodeConfig());
|
|
MultiThreadQueue<PyProxyQueue> queue(n_of_threads);
|
|
|
|
osyncstream(cout) << "QUEUE " << queue.queue_num() << endl;
|
|
cerr << "[info] [main] Queue: " << queue.queue_num() << " threads assigned: " << n_of_threads << endl;
|
|
|
|
thread qthr([&](){
|
|
queue.start();
|
|
});
|
|
config_updater();
|
|
qthr.join();
|
|
}
|