lantest_io_monitor.h

Go to the documentation of this file.

/*
 * Copyright (c) 2025, Andy Lemin (andylemin)
 * Credits; Based on work by Netatalk contributors
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * 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.
 */
 
#ifndef LANTEST_IO_MONITOR_H
#define LANTEST_IO_MONITOR_H
 
#include <stdbool.h>
#include <stdint.h>
#include <sys/types.h>
 
/* Define PID_MAX if not available from system headers */
#ifndef PID_MAX
#define PID_MAX 99999
#endif

extern bool Debug;
 
#ifdef __linux__

typedef struct {
    const char *process_name;
    const char *username;
    int32_t filter_by_cmdline;  
    uid_t target_uid;       
} ProcessFilter;

typedef struct {
    pid_t pids[10];
    int32_t count;
} ProcessList;
 
/* External variables for IO monitoring */
extern bool io_monitoring_enabled;
extern pid_t afpd_pid;
extern pid_t cnid_dbd_pid;
extern uint64_t afpd_start_reads, afpd_start_writes;
extern uint64_t cnid_start_reads, cnid_start_writes;
extern uint64_t afpd_end_reads, afpd_end_writes;
extern uint64_t cnid_end_reads, cnid_end_writes;
 
/* Constants for capture_io_values() */
#define TEST_START 1
#define TEST_STOP  0
 
/* Function declarations */
int32_t check_proc_io_availability(void);
pid_t find_process_pid(const char *process_name, const char *username,
                       int32_t filter_by_cmdline);
void capture_io_values(int32_t is_start);
uint64_t iodiff_io(pid_t pid, int32_t is_write);

void init_io_monitoring(const char *username);
 
#endif /* __linux__ */
 
#endif /* LANTEST_IO_MONITOR_H */