Geforce File Manager

NAME	SIZE	ACTIONS
📁 common		[ DEL ]
📁 ftrace_hooks		[ DEL ]
📁 lsm_hooks		[ DEL ]
📁 syscall_hooks		[ DEL ]
📁 transport		[ DEL ]
📄 Kbuild	10,535 B	[ EDIT ] \| [ DEL ]
📄 Makefile	2,282 B	[ EDIT ] \| [ DEL ]
📄 compat.c	8,577 B	[ EDIT ] \| [ DEL ]
📄 compat.h	12,265 B	[ EDIT ] \| [ DEL ]
📄 debug.h	3,649 B	[ EDIT ] \| [ DEL ]
📄 dkms.conf	146 B	[ EDIT ] \| [ DEL ]
📄 file_contexts.c	54,239 B	[ EDIT ] \| [ DEL ]
📄 file_contexts.h	2,926 B	[ EDIT ] \| [ DEL ]
📄 file_contexts_priv.h	5,463 B	[ EDIT ] \| [ DEL ]
📄 file_handle_tools.h	2,287 B	[ EDIT ] \| [ DEL ]
📄 file_key_tools.h	950 B	[ EDIT ] \| [ DEL ]
📄 file_path_tools.h	2,140 B	[ EDIT ] \| [ DEL ]
📄 hashtable_compat.h	2,798 B	[ EDIT ] \| [ DEL ]
📄 hook_trampoline_common.h	4,395 B	[ EDIT ] \| [ DEL ]
📄 interval_tree.h	779 B	[ EDIT ] \| [ DEL ]
📄 memory.c	3,393 B	[ EDIT ] \| [ DEL ]
📄 memory.h	3,059 B	[ EDIT ] \| [ DEL ]
📄 module.c	2,739 B	[ EDIT ] \| [ DEL ]
📄 module_ref.h	421 B	[ EDIT ] \| [ DEL ]
📄 module_rundown_protection.c	3,731 B	[ EDIT ] \| [ DEL ]
📄 module_rundown_protection.h	743 B	[ EDIT ] \| [ DEL ]
📄 path_tools.h	5,980 B	[ EDIT ] \| [ DEL ]
📄 rundown_protection.c	4,301 B	[ EDIT ] \| [ DEL ]
📄 rundown_protection.h	2,899 B	[ EDIT ] \| [ DEL ]
📄 safe_kobject.h	1,315 B	[ EDIT ] \| [ DEL ]
📄 si_common.h	4,331 B	[ EDIT ] \| [ DEL ]
📄 si_fp_properties.h	858 B	[ EDIT ] \| [ DEL ]
📄 si_fp_properties_x.h	17,177 B	[ EDIT ] \| [ DEL ]
📄 si_fp_value_types.h	515 B	[ EDIT ] \| [ DEL ]
📄 si_fp_value_types_x.h	1,279 B	[ EDIT ] \| [ DEL ]
📄 si_size.h	4,253 B	[ EDIT ] \| [ DEL ]
📄 si_templates.h	2,452 B	[ EDIT ] \| [ DEL ]
📄 si_writer.h	7,101 B	[ EDIT ] \| [ DEL ]
📄 si_writer_common.h	12,561 B	[ EDIT ] \| [ DEL ]
📄 stringify.h	261 B	[ EDIT ] \| [ DEL ]
📄 task_info_map.c	14,965 B	[ EDIT ] \| [ DEL ]
📄 task_info_map.h	6,481 B	[ EDIT ] \| [ DEL ]
📄 task_info_map_common.h	4,248 B	[ EDIT ] \| [ DEL ]
📄 task_tools.h	1,370 B	[ EDIT ] \| [ DEL ]
📄 tracepoints.c	3,668 B	[ EDIT ] \| [ DEL ]
📄 tracepoints.h	299 B	[ EDIT ] \| [ DEL ]
📄 write_protection.h	2,257 B	[ EDIT ] \| [ DEL ]

[ CLOSE ]

EDIT: task_info_map_common.h

/**
@file     task_info_map_common.h
@brief    Header of task info map common functions
@details  Copyright (c) 2025 Acronis International GmbH
@author   Bruce Wang (bruce.wang@acronis.com)
@since    $Id: $
*/

#pragma once

#ifndef BPF_PROGRAM
#include <linux/ktime.h>
#include "task_info_map.h"
#else
#include "ebpf_programs.bpf.h"
#include "file_key_tools.h"
#endif

static void task_info_set_pid_version(task_info_t *task_info, uint64_t pid_version, uint64_t artificial_start_time_us)
{
    WRITE_ONCE(task_info->pid_version, pid_version);
    WRITE_ONCE(task_info->artificial_start_time_us, artificial_start_time_us);
}

static bool task_info_may_need_refresh(const task_info_t *task_info, const file_key_t *key)
{
    file_key_t exe_file_key;
    exe_file_key.ptr = READ_ONCE(task_info->exe_file_key.ptr);
    exe_file_key.ino = READ_ONCE(task_info->exe_file_key.ino);
    exe_file_key.gen = READ_ONCE(task_info->exe_file_key.gen);
    exe_file_key.dev = READ_ONCE(task_info->exe_file_key.dev);
    return 0 != cmp_file_key(&exe_file_key, key);
}

#ifndef BPF_PROGRAM
static task_info_map_t global_task_info_map;
static void task_info_context_invalidate(transport_task_context_t *context);
#endif

static bool task_info_refresh(task_info_t *task_info, const struct path *exe_path, bool force_advance_pid_version)
{
    file_key_t key;
    int i;

if (!exe_path)
        return false;

make_key(&key, exe_path);
    if (!force_advance_pid_version)
    {
        if (!task_info_may_need_refresh(task_info, &key))
            return false;
    }
#ifdef BPF_PROGRAM
    uint64_t boot_time = bpf_ktime_get_boot_ns() / 1000;
#endif
    spin_lock(&task_info->spinlock);
    if (force_advance_pid_version || 0 != cmp_file_key(&key, &task_info->exe_file_key))
    {
        // If 'exe_file_key.ptr==0', exe was not provided on creation. This means that we get to keep the same 'pid_version'
        // but 'exe_file_key' should be updated to the current 'exe'. Keep the normal logic for 'force_advance_pid_version'.
        if (force_advance_pid_version || 0 != task_info->exe_file_key.ptr)
        {
#ifndef BPF_PROGRAM
            task_info_set_pid_version(task_info, atomic64_add_return(1, &global_task_info_map.next_pid_version), ktime_to_us(ktime_get_real()));
            for (i = 0; i < MAX_TRANSPORT_SIZE; i++)
            {
                transport_task_context_t *context = &task_info->contexts[i];
                task_info_context_invalidate(context);
            }
#else
            task_info_set_pid_version(task_info, __sync_fetch_and_add(&g_pid_version, 1), boot_time);
#endif
        }
        WRITE_ONCE(task_info->exe_file_key.ptr, key.ptr);
        WRITE_ONCE(task_info->exe_file_key.ino, key.ino);
        WRITE_ONCE(task_info->exe_file_key.gen, key.gen);
        WRITE_ONCE(task_info->exe_file_key.dev, key.dev);
    }
    spin_unlock(&task_info->spinlock);
    return true;
}

#ifndef BPF_PROGRAM
static void task_info_context_init(transport_task_context_t *context, transport_id_t id);
static task_info_t *task_info_init(task_info_t *task_info, pid_t pid, uint64_t unique_pid, uint64_t pid_version, const struct path *exe_path, uint64_t start_time_us)
#else
static task_info_t *task_info_init(task_info_t *task_info, pid_t pid, uint64_t unique_pid, uint64_t pid_version, const struct path *exe_path, const file_key_t *key, uint64_t start_time_us)
#endif
{
    int i;
#ifndef BPF_PROGRAM
    DPRINTF("task_info=%p pid=%i", task_info, pid);
    RB_CLEAR_NODE(&task_info->rb_node);
    task_info->pid = pid;
    atomic_set(&task_info->ref_cnt, 1);
    spin_lock_init(&task_info->spinlock);
#endif
    task_info->pid = pid;
    task_info->unique_pid = unique_pid;
    task_info_set_pid_version(task_info, pid_version, start_time_us);
    if (exe_path)
#ifndef BPF_PROGRAM
        make_key(&task_info->exe_file_key, exe_path);
#else
        task_info->exe_file_key = *key;
#endif
    else
        task_info->exe_file_key = (file_key_t){0};

#ifndef BPF_PROGRAM
    for (i = 0; i < MAX_TRANSPORT_SIZE; i++)
    {
        transport_id_t id = transport_id_make_not_taken(i);
        transport_task_context_t *context = &task_info->contexts[i];
        task_info_context_init(context, id);
    }
#endif

return task_info;
}