Gecko [ www.thearyasamaj.org ]

Name	Size	Permission
common	[ DIR ]	drwxr-xr-x
ftrace_hooks	[ DIR ]	drwxr-xr-x
lsm_hooks	[ DIR ]	drwxr-xr-x
syscall_hooks	[ DIR ]	drwxr-xr-x
transport	[ DIR ]	drwxr-xr-x
Kbuild	8.88 KB	-rw-r--r--
Makefile	2.23 KB	-rw-r--r--
compat.c	6.18 KB	-rw-r--r--
compat.h	10.89 KB	-rw-r--r--
debug.h	3.56 KB	-rw-r--r--
dkms.conf	146 B	-rw-r--r--
file_contexts.c	41.04 KB	-rw-r--r--
file_contexts.h	2.51 KB	-rw-r--r--
file_contexts_priv.h	4.92 KB	-rw-r--r--
file_handle_tools.h	2.15 KB	-rw-r--r--
file_key_tools.h	869 B	-rw-r--r--
file_path_tools.h	2.09 KB	-rw-r--r--
hashtable_compat.h	2.45 KB	-rw-r--r--
hook_trampoline_common.h	4.2 KB	-rw-r--r--
interval_tree.h	779 B	-rw-r--r--
memory.h	1.37 KB	-rw-r--r--
module.c	1.86 KB	-rw-r--r--
module_ref.h	421 B	-rw-r--r--
module_rundown_protection.c	3.64 KB	-rw-r--r--
module_rundown_protection.h	743 B	-rw-r--r--
path_tools.h	4.42 KB	-rw-r--r--
rundown_protection.c	4.2 KB	-rw-r--r--
rundown_protection.h	2.83 KB	-rw-r--r--
si_common.h	3.93 KB	-rw-r--r--
si_fp_properties.h	858 B	-rw-r--r--
si_fp_properties_x.h	15.41 KB	-rw-r--r--
si_fp_value_types.h	515 B	-rw-r--r--
si_fp_value_types_x.h	1.25 KB	-rw-r--r--
si_size.h	4.15 KB	-rw-r--r--
si_templates.h	2.29 KB	-rw-r--r--
si_writer.h	6.48 KB	-rw-r--r--
si_writer_common.h	6.39 KB	-rw-r--r--
stringify.h	261 B	-rw-r--r--
task_info_map.c	16.45 KB	-rw-r--r--
task_info_map.h	6.24 KB	-rw-r--r--
task_tools.h	1.34 KB	-rw-r--r--
tracepoints.c	3.58 KB	-rw-r--r--
tracepoints.h	299 B	-rw-r--r--
write_protection.h	1.04 KB	-rw-r--r--

Code Editor : file_contexts_priv.h

/**
@file     file_contexts_priv.h
@brief    Store the opened file contexts(inode + pid), private struct definitions
@details  Copyright (c) 2024 Acronis International GmbH
@author   Bruce Wang (bruce.wang@acronis.com)
@since    $Id: $
*/

#include <linux/list.h>
#include <linux/rcupdate.h>
#include <linux/spinlock.h>
#include <linux/types.h>
#include <linux/version.h>
#include <linux/rbtree.h>

#include "file_contexts.h"
#include "hashtable_compat.h"
#include "transport.h"

#define FILE_CONTEXT_SMALL_TABLE_SIZE_BITS 6                                              // 64
#define FILE_CONTEXT_SMALL_TABLE_MAX_SIZE (1 << (FILE_CONTEXT_SMALL_TABLE_SIZE_BITS - 1)) // 32
#define FILE_CONTEXT_SMALL_TABLE_LRU_CLEAN_SIZE (10)
#define FILE_CONTEXT_BIG_TABLE_SIZE_BITS 15                                       // 32768
#define FILE_CONTEXT_BIG_TBALE_SIZE (1 << (FILE_CONTEXT_BIG_TABLE_SIZE_BITS - 1)) // 16384
#define FILE_CONTEXT_BIG_TABLE_LRU_CLEAN_SIZE (100)
#define FILE_CONTEXT_EXPIRE_TIME_MS (180 * 1000)

#define FILE_CONTEXT_CHUNK_SIZE 4096
#define FILE_CONTEXT_CHUNK_LOWER_BOUND(N) ((uint64_t)(N / FILE_CONTEXT_CHUNK_SIZE) * FILE_CONTEXT_CHUNK_SIZE)
#define FILE_CONTEXT_CHUNK_UPPER_BOUND(N) ((uint64_t)((N + FILE_CONTEXT_CHUNK_SIZE - 1) / FILE_CONTEXT_CHUNK_SIZE) * FILE_CONTEXT_CHUNK_SIZE)

#define FILE_CONTEXT_MAX_FILE_SIZE (100 * 1024 * 1024)

/*
 size of file_context_tables_t: 786576 Bytes => 0.75MB
 size of interval_node_t: 72 Bytes
 size of file_context_rw_node_t: 192 Bytes
 size of file_context_open_file_node_t: 720 Bytes
 size of file_context_open_process_node_t: 104 Bytes

In the scenario that all tables are full:
    2 *32768 * (sizeof(file_context_rw_node_t) + 32(estimated) * sizeof(interval_node_t)) +
    32768 * (sizeof(file_context_open_file_node_t) + 32 * sizeof(file_context_open_process_node_t)) +
    sizeof(file_context_tables_t)
    = 283 MB
*/

// Node in file_context_common_table_t
// this struct should always be the first in the assembled struct
typedef struct
{
    // node in file_context_common_table_t->hashtable
    struct hlist_node hash_node;
    uint64_t key;
    // node in file_context_common_table_t->lru_list
    bool lru_list_node_inserted;
    struct list_head lru_list_node;
    // node in temporary delete list
    struct list_head del_list_node;
    unsigned long last_access_time;
    atomic_t ref_count;
    struct rcu_head rcu;
    void (*pre_free_func)(void *);
} file_context_common_node_t;

// Abstruction of tables
typedef struct
{
    struct hlist_head *hashtable;
    struct list_head lru_list;
    unsigned int size;
    uint8_t hashbits;
    unsigned int max_size;
    unsigned short clean_count;
    spinlock_t spinlock;
} file_context_common_table_t;

// A small hashtable struct to store file_context_open_process_node_t, the key is pid
typedef struct
{
    DECLARE_HASHTABLE(hashtable, FILE_CONTEXT_SMALL_TABLE_SIZE_BITS);
    file_context_common_table_t common_table;
} file_context_small_table_t;

// A big hashtable struct to store file_context_open_file_node_t, file_context_rw_node_t, the key is inode ptr
typedef struct
{
    DECLARE_HASHTABLE(hashtable, FILE_CONTEXT_BIG_TABLE_SIZE_BITS);
    file_context_common_table_t common_table;
} file_context_big_table_t;

// A dummy struct for abstruction
typedef struct
{
    file_context_common_node_t common_node;
} dummy_node_t;

// Node in process_table
typedef struct
{
    file_context_common_node_t common_node;
    file_context_open_process_t data;
} file_context_open_process_node_t;

// Node in open_table
typedef struct
{
    file_context_common_node_t common_node;
    // Key: pid, Value: flags
    file_context_small_table_t process_table;
    file_context_key_t key;
    file_context_open_file_t data;
} file_context_open_file_node_t;

typedef struct
{
    uint64_t low;
    uint64_t high;
    struct rb_node rb;
    // node in temporary delete list
    struct list_head del_list_node;
    struct list_head stack_node;
} interval_node_t;

// Node in read_table/write_table
typedef struct
{
    file_context_common_node_t common_node;
    // data fields
    file_context_key_t key;
    file_context_rw_t data;
} file_context_rw_node_t;

typedef struct
{
    uint64_t transport_id;

// TODO: ref_count is excessive, it is going to be used very rarely in a normal run
    // A better idea would be using rcu to acquire the file_context_tables_t pointer
    // and avoid using reference counts altogether
    atomic_t ref_count;
    // Key: inode.ptr, Value: processes
    file_context_big_table_t open_table;
    // Key: inode.ptr, Value: intervals
    file_context_big_table_t read_table;
    // Key: inode.ptr, Value: intervals
    file_context_big_table_t write_table;

struct rcu_head rcu;
} file_context_tables_t;

typedef struct
{
    spinlock_t writer_lock;
    // file_context_tables_t is memory costly, malloc it only when new transport is opened
    // file_context_tables_t pointer is RCU protected
    file_context_tables_t* tables[MAX_TRANSPORT_SIZE];
} file_context_manager_t;

${this.title}

Code Editor : file_contexts_priv.h