Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-03 23:28:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Reformat the world

Browse source
This commit is contained in:
Tim Wojtulewicz 2021-09-16 15:35:39 -07:00
parent 194cb24547
commit b2f171ec69
714 changed files with 35149 additions and 35203 deletions
Download patch file Download diff file Expand all files Collapse all files

45
src/iosource/Manager.h
View file

@ -2,20 +2,21 @@
#pragma once
#include "zeek/zeek-config.h"
#include <map>
#include <string>
#include <vector>
#include <map>
#include "zeek/iosource/IOSource.h"
#include "zeek/Flare.h"
#include "zeek/iosource/IOSource.h"
#include "zeek/zeek-config.h"
struct timespec;
struct kevent;
namespace zeek {
namespace iosource {
namespace zeek
{
namespace iosource
{
class PktSrc;
class PktDumper;
@ -24,7 +25,8 @@ class PktDumper;
* Manager class for IO sources. This handles all of the polling of sources
* in the main loop.
*/
class Manager {
class Manager
{
public:
/**
* Constructor.
@ -60,24 +62,24 @@ public:
* Returns the number of registered and still active sources,
* excluding those that are registered as \a dont_count.
*/
int Size() const { return sources.size() - dont_counts; }
int Size() const { return sources.size() - dont_counts; }
/**
* Returns total number of sources including dont_counts;
*/
int TotalSize() const { return sources.size(); }
int TotalSize() const { return sources.size(); }
/**
* Returns the registered PktSrc. If not source is registered yet,
* returns a nullptr.
*/
PktSrc* GetPktSrc() const { return pkt_src; }
PktSrc* GetPktSrc() const { return pkt_src; }
/**
* Terminate all processing immediately by removing all sources (and
* therefore now returning a Size() of zero).
*/
void Terminate() { RemoveAll(); }
void Terminate() { RemoveAll(); }
/**
* Opens a new packet source.
@ -97,7 +99,7 @@ public:
* @param path The file name to dump into.
*
* @param append True to append if \a path already exists.
*
*
* @return The new packet dumper, or null if an error occured.
*/
PktDumper* OpenPktDumper(const std::string& path, bool append);
@ -132,7 +134,6 @@ public:
void Wakeup(const std::string& where);
private:
/**
* Calls the appropriate poll method to gather a set of IOSources that are
* ready for processing.
@ -160,7 +161,8 @@ private:
void RemoveAll();
class WakeupHandler final : public IOSource {
class WakeupHandler final : public IOSource
{
public:
WakeupHandler();
~WakeupHandler();
@ -175,18 +177,19 @@ private:
// IOSource API methods
void Process() override;
const char* Tag() override { return "WakeupHandler"; }
double GetNextTimeout() override { return -1; }
const char* Tag() override { return "WakeupHandler"; }
double GetNextTimeout() override { return -1; }
private:
zeek::detail::Flare flare;
};
struct Source {
struct Source
{
IOSource* src = nullptr;
bool dont_count = false;
bool manage_lifetime = false;
};
};
using SourceList = std::vector<Source*>;
SourceList sources;
@ -208,10 +211,10 @@ private:
// This is only used for the output of the call to kqueue in FindReadySources().
// The actual events are stored as part of the queue.
std::vector<struct kevent> events;
};
};
} // namespace iosource
} // namespace iosource
extern iosource::Manager* iosource_mgr;
} // namespace zeek
} // namespace zeek