logging.h

Go to the documentation of this file.

// SPDX-License-Identifier: GPL-2.0-only
#ifndef LOGGING_H_
#define LOGGING_H_
#include <bpf/vmlinux.h>
#include <bpf/bpf_helpers.h>
 
#include "logging_types.h"
#include "shared_rust_types.h"
 
#define DEFAULT_RINGBUF_SIZE (256 * 1024)
 
struct log_ringbuf {
    __uint(type, BPF_MAP_TYPE_RINGBUF);
    __uint(max_entries, DEFAULT_RINGBUF_SIZE);
};
 
extern u32                log_level;
extern struct log_ringbuf log_ringbuf;
 
static inline void *log_buf(enum LogLevel level, enum LogReason reason,
                            enum EventType type, size_t size,
                            unsigned long pol_id)
{
    if (level > log_level)
        return NULL;
 
    void *log = bpf_ringbuf_reserve(&log_ringbuf, size, 0);
    if (!log) {
        bpf_printk("seabee log_buf: unable to reserve from ringbuf of size %lu",
                   size);
        return NULL;
    }
 
    struct log_hdr *hdr = (struct log_hdr *)log;
    hdr->level          = level;
    hdr->type           = type;
    hdr->reason         = reason;
    // kernel tgid == user space process id
    // kernel pid == user space thread id
    u64 pid_tgid        = bpf_get_current_pid_tgid();
    hdr->pid            = pid_tgid >> 32;
    hdr->tid            = pid_tgid & 0xFFFFFFFF;
    hdr->uid            = bpf_get_current_uid_gid() & 0xFFFFFFFF;
    hdr->pol_id         = pol_id;
    bpf_get_current_comm(hdr->comm, sizeof(hdr->comm));
    return log;
}
 
static inline void log_generic(enum LogLevel level, enum LogReason reason,
                               enum EventType type, unsigned int pol_id)
{
    void *log = log_buf(level, reason, type, sizeof(struct log_hdr), pol_id);
    if (log) {
        bpf_ringbuf_submit(log, 0);
    }
}
 
static inline void log_generic_msg(enum LogLevel level, enum LogReason reason,
                                   const char *fmt, __u64 *data, __u32 data_len)
{
    struct generic_msg_log *log;
    log = log_buf(level, reason, EVENT_TYPE_MSG, sizeof(*log), NO_POL_ID);
    if (log) {
        long ret = bpf_snprintf((char *)log->msg, sizeof(log->msg), fmt, data,
                                data_len);
        if (ret < 0) {
            bpf_printk("Error: log_generic_msg: bpf_snprintf failed");
        }
        bpf_ringbuf_submit(log, 0);
    }
}
 
#endif // LOGGING_H_