counters.h

Go to the documentation of this file.

/* Copyright (C) 2007-2025 Open Information Security Foundation
 *
 * You can copy, redistribute or modify this Program under the terms of
 * the GNU General Public License version 2 as published by the Free
 * Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * version 2 along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */
 
/**
 * \file
 *
 * \author Anoop Saldanha <anoopsaldanha@gmail.com>
 * \author Victor Julien <victor@inliniac.net>
 */
 
#ifndef SURICATA_COUNTERS_H
#define SURICATA_COUNTERS_H
 
#include "threads.h"
 
typedef struct StatsCounterId {
    uint16_t id;
} StatsCounterId;
 
typedef struct StatsCounterAvgId {
    uint16_t id;
} StatsCounterAvgId;
 
typedef struct StatsCounterMaxId {
    uint16_t id;
} StatsCounterMaxId;
 
typedef struct StatsCounterGlobalId {
    uint16_t id;
} StatsCounterGlobalId;
 
/**
 * \brief Container to hold the counter variable
 */
typedef struct StatsCounter_ {
    int type;
 
    /* local id for this counter in this thread */
    uint16_t id;
 
    /* global id, used in output */
    uint16_t gid;
 
    /* when using type STATS_TYPE_Q_FUNC this function is called once
     * to get the counter value, regardless of how many threads there are. */
    uint64_t (*Func)(void);
 
    /* name of the counter */
    const char *name;
    const char *short_name;
 
    /* the next perfcounter for this tv's tm instance */
    struct StatsCounter_ *next;
} StatsCounter;
 
/**
 * \brief counter type for local (private) increments.
 * For AVG counters we use 2 to track values and updates.
 */
typedef struct StatsLocalCounter_ {
    int64_t v;
} StatsLocalCounter;
 
/**
 * \brief Stats Context for a ThreadVars instance
 */
typedef struct StatsPublicThreadContext_ {
    /* flag set by the wakeup thread, to inform the client threads to sync */
    SC_ATOMIC_DECLARE(bool, sync_now);
 
    /* pointer to the head of a list of counters assigned under this context */
    StatsCounter *head;
 
    /* holds the total no of counters already assigned for this perf context */
    uint16_t curr_id;
 
    /* array of pointers to the StatsCounters in `head` above, indexed by the per
     * thread counter id.
     * Size is `curr_id + 1` after all counters have been registered.
     * Ownership of counters is with `head` above. */
    const StatsCounter **pc_array;
 
    StatsLocalCounter *copy_of_private;
 
    /* lock to prevent simultaneous access during update_counter/output_stat */
    SCSpinlock lock;
} StatsPublicThreadContext;
 
/**
 * \brief used to hold the private version of the counters registered
 */
typedef struct StatsPrivateThreadContext_ {
    /* points to the array holding local counters */
    StatsLocalCounter *head;
 
    /* size of head array in elements */
    uint32_t size;
 
    int initialized;
} StatsPrivateThreadContext;
 
typedef struct StatsThreadContext_ {
    StatsPublicThreadContext pub;
    StatsPrivateThreadContext priv;
} StatsThreadContext;
 
/* the initialization functions */
void StatsInit(void);
void StatsSetupPostConfigPreOutput(void);
void StatsSetupPostConfigPostOutput(void);
void StatsSpawnThreads(void);
void StatsRegisterTests(void);
bool StatsEnabled(void);
 
/* functions used to free the resources allotted by the Stats API */
void StatsReleaseResources(void);
 
/* counter registration functions */
StatsCounterId StatsRegisterCounter(const char *, StatsThreadContext *);
StatsCounterAvgId StatsRegisterAvgCounter(const char *, StatsThreadContext *);
StatsCounterMaxId StatsRegisterMaxCounter(const char *, StatsThreadContext *);
StatsCounterGlobalId StatsRegisterGlobalCounter(const char *cname, uint64_t (*Func)(void));
 
/* functions used to update local counter values */
void StatsCounterAddI64(StatsThreadContext *, StatsCounterId, int64_t);
void StatsCounterSetI64(StatsThreadContext *, StatsCounterId, int64_t);
void StatsCounterIncr(StatsThreadContext *, StatsCounterId);
void StatsCounterDecr(StatsThreadContext *, StatsCounterId);
 
void StatsCounterMaxUpdateI64(StatsThreadContext *, StatsCounterMaxId id, int64_t x);
void StatsCounterAvgAddI64(StatsThreadContext *, StatsCounterAvgId id, int64_t x);
 
/* utility functions */
int64_t StatsCounterGetLocalValue(StatsThreadContext *, StatsCounterId);
void StatsThreadInit(StatsThreadContext *);
int StatsSetupPrivate(StatsThreadContext *, const char *);
void StatsThreadCleanup(StatsThreadContext *);
 
void StatsSyncCounters(StatsThreadContext *);
void StatsSyncCountersIfSignalled(StatsThreadContext *);
 
#ifdef BUILD_UNIX_SOCKET
TmEcode StatsOutputCounterSocket(json_t *cmd,
                                 json_t *answer, void *data);
#endif
 
#endif /* SURICATA_COUNTERS_H */