Gecko [ www.omxrelocation.in ]

Name	Size	Permission
common	[ DIR ]	drwxr-xr-x
syscall_hooks	[ DIR ]	drwxr-xr-x
transport	[ DIR ]	drwxr-xr-x
Kbuild	3.71 KB	-rw-r--r--
Makefile	2.2 KB	-rw-r--r--
compat.c	5.74 KB	-rw-r--r--
compat.h	6.94 KB	-rw-r--r--
debug.h	3.56 KB	-rw-r--r--
dkms.conf	146 B	-rw-r--r--
memory.h	529 B	-rw-r--r--
module.c	1.86 KB	-rw-r--r--
module_ref.h	421 B	-rw-r--r--
module_rundown_protection.c	3.6 KB	-rw-r--r--
module_rundown_protection.h	743 B	-rw-r--r--
rundown_protection.c	4.2 KB	-rw-r--r--
rundown_protection.h	2.83 KB	-rw-r--r--
stringify.h	261 B	-rw-r--r--
task_info_map.c	19.07 KB	-rw-r--r--
task_info_map.h	1.92 KB	-rw-r--r--
tracepoints.c	3.02 KB	-rw-r--r--
tracepoints.h	299 B	-rw-r--r--

Code Editor : rundown_protection.h

/**
 @file     rundown_protection.h
 @brief    Rundown Protection implementation for Linux kernel
 @details  Copyright (c) 2022 Acronis International GmbH
 @author   Denis Kopyrin (Denis.Kopyrin@acronis.com)
 @since    $Id: $
 */

#pragma once

#ifdef KERNEL_MOCK
#include <mock/mock.h>
#endif
#include <linux/atomic.h>
#include <linux/types.h>	// bool, [u]int(8|16|32|64)_t, atomics
#include <linux/wait.h>		// wait_event*(), wake_up*()

typedef void (*rundown_protection_finalizer_t)(void*);

typedef struct {
	atomic64_t ref_cnt;
	wait_queue_head_t wait_queue;
	rundown_protection_finalizer_t finalizer;
	void *finalizer_ctx;
} rundown_wait_block_t;

typedef struct {
	atomic64_t data;
	rundown_wait_block_t wait_block;
} simple_rundown_protection_t;

// Initializes the rundown protection.
// If 'finalizer' is set, it can be called spuriously as rundown protection is lock-free.
void simple_rundown_protection_init(simple_rundown_protection_t *srp, rundown_protection_finalizer_t finalizer, void *finalizer_ctx, bool ready);

// Thread-safely locks/Unlocks rundown protection, if lock returns 'false', do NOT call unlock and immediately leave the hook
bool simple_rundown_protection_lock(simple_rundown_protection_t *srp);
void simple_rundown_protection_unlock(simple_rundown_protection_t *srp);

// Controls for the rundown protection, NOT thread-safe. Use under the mutex.
// Rundowns the rundown protection. May call 'finalizer' spuriously.
void simple_rundown_protection_set_rundown_active(simple_rundown_protection_t *srp);
// Waits for the rundown protection to have all locks call unlock.
void simple_rundown_protection_wait_for_rundown(simple_rundown_protection_t *srp);
bool simple_rundown_protection_wait_for_rundown_timeout(simple_rundown_protection_t *srp, unsigned long timeout_jiffies);
// Enables the rundown protection allowing lock to return 'true'.
void simple_rundown_protection_set_ready(simple_rundown_protection_t *srp);
// Fetches the pending lock count from the wait block. If it is equal to 0, all hooks are exited.
// Returns reasonable values only after 'set_rundown_active' is called.
int64_t simple_rundown_protection_get_pending_count(simple_rundown_protection_t *srp);

#define rundown_protection_t simple_rundown_protection_t
#define rundown_protection_init simple_rundown_protection_init
#define rundown_protection_lock simple_rundown_protection_lock
#define rundown_protection_unlock simple_rundown_protection_unlock
#define rundown_protection_set_rundown_active simple_rundown_protection_set_rundown_active
#define rundown_protection_wait_for_rundown simple_rundown_protection_wait_for_rundown
#define rundown_protection_wait_for_rundown_timeout simple_rundown_protection_wait_for_rundown_timeout
#define rundown_protection_set_ready simple_rundown_protection_set_ready
#define rundown_protection_get_pending_count simple_rundown_protection_get_pending_count

${this.title}

Code Editor : rundown_protection.h