introduce parameter to set affinity CPU list

Author: TheSableCZ

ipfixprobe.cpp

@@ -698,6 +698,7 @@ int run(int argc, char *argv[])
    conf.pkt_bufsize = parser.m_pkt_bufsize;
    conf.max_pkts = parser.m_max_pkts;
 
+   set_thread_details(pthread_self(), "", parser.m_cpu_mask);
 
    try {
       if (process_plugin_args(conf, parser)) {

ipfixprobe.hpp

@@ -90,6 +90,7 @@ class IpfixprobeOptParser : public OptionsParser {
    bool m_help;
    std::string m_help_str;
    bool m_version;
+   std::vector<int> m_cpu_mask;
 
    IpfixprobeOptParser() : OptionsParser("ipfixprobe", "flow exporter supporting various custom IPFIX elements"),
                            m_pid(""), m_appfs_mount_point(""), m_daemon(false),
@@ -170,6 +171,18 @@ class IpfixprobeOptParser : public OptionsParser {
           m_version = true;
           return true;
       }, OptionFlags::NoArgument);
+      register_option("-C", "--cpus", "CPU_LIST", "Set global CPU mask for main thread and subthreads", [this](const char *arg) {
+         try {
+            std::stringstream ss(arg);
+            std::string tmp;
+            while (std::getline(ss, tmp, ',')) {
+               m_cpu_mask.emplace_back(str2num<uint16_t>(tmp));
+            }
+            return true;
+         } catch (std::invalid_argument &e) {
+            return false;
+         }
+      });
    }
 };