mirrors/fort
1
0
Fork 0
mirror of https://github.com/tnodir/fort synced 2024-11-15 07:38:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Driver: Minor refactor

Browse Source
This commit is contained in:
Nodir Temirkhodjaev 2023-04-14 12:03:37 +03:00
parent edf0ae45d8
commit af927793b2
4 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/driver/fortcnf.c
View File

@ -59,10 +59,8 @@ static PFORT_CONF_EXE_NODE fort_conf_ref_exe_find_node(
(PFORT_CONF_EXE_NODE) tommy_hashdyn_bucket(&conf_ref->exe_map, path_hash); (PFORT_CONF_EXE_NODE) tommy_hashdyn_bucket(&conf_ref->exe_map, path_hash);
while (node != NULL) { while (node != NULL) {
if (node->path_hash == path_hash if (fort_conf_app_exe_equal(node->app_entry, path, path_len))
&& fort_conf_app_exe_equal(node->app_entry, path, path_len)) {
return node; return node;
}
node = node->next; node = node->next;
} }

5
src/driver/fortcout.c
View File

@ -953,16 +953,15 @@ FORT_API void NTAPI fort_callout_timer(void)
PFORT_BUFFER buf = &fort_device()->buffer; PFORT_BUFFER buf = &fort_device()->buffer;
PFORT_STAT stat = &fort_device()->stat; PFORT_STAT stat = &fort_device()->stat;
KLOCK_QUEUE_HANDLE buf_lock_queue;
KLOCK_QUEUE_HANDLE stat_lock_queue;
PIRP irp = NULL; PIRP irp = NULL;
ULONG_PTR info; ULONG_PTR info;
/* Lock buffer */ /* Lock buffer */
KLOCK_QUEUE_HANDLE buf_lock_queue;
fort_buffer_dpc_begin(buf, &buf_lock_queue); fort_buffer_dpc_begin(buf, &buf_lock_queue);
/* Lock stat */ /* Lock stat */
KLOCK_QUEUE_HANDLE stat_lock_queue;
fort_stat_dpc_begin(stat, &stat_lock_queue); fort_stat_dpc_begin(stat, &stat_lock_queue);
/* Get current Unix time */ /* Get current Unix time */

3
src/driver/fortps.c
View File

@ -470,9 +470,8 @@ static PFORT_PSNODE fort_pstree_find_proc_hash(
PFORT_PSNODE node = (PFORT_PSNODE) tommy_hashdyn_bucket(&ps_tree->procs_map, pid_hash); PFORT_PSNODE node = (PFORT_PSNODE) tommy_hashdyn_bucket(&ps_tree->procs_map, pid_hash);
while (node != NULL) { while (node != NULL) {
if (node->process_id == processId) { if (node->process_id == processId)
return node; return node;
}
node = node->next; node = node->next;
} }

6
src/driver/fortstat.c
View File

@ -51,7 +51,7 @@ static PFORT_STAT_PROC fort_stat_proc_get(PFORT_STAT stat, UINT32 process_id, to
PFORT_STAT_PROC proc = (PFORT_STAT_PROC) tommy_hashdyn_bucket(&stat->procs_map, proc_hash); PFORT_STAT_PROC proc = (PFORT_STAT_PROC) tommy_hashdyn_bucket(&stat->procs_map, proc_hash);
while (proc != NULL) { while (proc != NULL) {
if (proc->proc_hash == proc_hash && proc->process_id == process_id) if (proc->process_id == process_id)
return proc; return proc;
proc = proc->next; proc = proc->next;
@ -276,7 +276,7 @@ static PFORT_FLOW fort_flow_get(PFORT_STAT stat, UINT64 flow_id, tommy_key_t flo
PFORT_FLOW flow = (PFORT_FLOW) tommy_hashdyn_bucket(&stat->flows_map, flow_hash); PFORT_FLOW flow = (PFORT_FLOW) tommy_hashdyn_bucket(&stat->flows_map, flow_hash);
while (flow != NULL) { while (flow != NULL) {
if (flow->flow_hash == flow_hash && flow->flow_id == flow_id) if (flow->flow_id == flow_id)
return flow; return flow;
flow = flow->next; flow = flow->next;
@ -404,7 +404,7 @@ FORT_API void fort_stat_close_flows(PFORT_STAT stat)
if ((old_stat_flags & FORT_STAT_CLOSED) == 0) { if ((old_stat_flags & FORT_STAT_CLOSED) == 0) {
fort_stat_flags_set(stat, FORT_STAT_CLOSED, TRUE); fort_stat_flags_set(stat, FORT_STAT_CLOSED, TRUE);
InterlockedAdd(&stat->flow_closing_count, (LONG) stat->flows_map.count); InterlockedAdd(&stat->flow_closing_count, (LONG) tommy_hashdyn_count(&stat->flows_map));
} }
} }
KeReleaseInStackQueuedSpinLock(&lock_queue); KeReleaseInStackQueuedSpinLock(&lock_queue);