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

Driver: Micro-optimize pending buffer handling.

Browse Source
This commit is contained in:
Nodir Temirkhodjaev 2017-11-21 16:00:24 +05:00
parent ecda75b9e2
commit 83d199acba
2 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/driver/fortbuf.c
View File

@ -54,12 +54,14 @@ fort_buffer_blocked_write (PFORT_BUFFER buf, UINT32 remote_ip, UINT32 pid,
KeAcquireSpinLock(&buf->lock, &irq); KeAcquireSpinLock(&buf->lock, &irq);
/* Try to directly write to pending client */ /* Try to directly write to pending client */
if (buf->irp != NULL) { if (buf->out_len) {
const UINT32 out_top = buf->out_top; const UINT32 out_top = buf->out_top;
UINT32 new_top = out_top + len; UINT32 new_top = out_top + len;
/* Is it time to flush logs? */ /* Is it time to flush logs? */
if (buf->out_len - new_top < FORT_LOG_BLOCKED_SIZE_MAX) { if (buf->out_len - new_top < FORT_LOG_BLOCKED_SIZE_MAX) {
buf->out_len = 0;
*irp = buf->irp; *irp = buf->irp;
buf->irp = NULL; buf->irp = NULL;
@ -105,7 +107,7 @@ fort_buffer_xmove (PFORT_BUFFER buf, PIRP irp, PVOID out, ULONG out_len,
*info = buf_top = buf->top; *info = buf_top = buf->top;
if (!buf_top) { if (!buf_top) {
if (buf->irp != NULL) { if (buf->out_len) {
status = STATUS_INSUFFICIENT_RESOURCES; status = STATUS_INSUFFICIENT_RESOURCES;
} else if (out_len < FORT_LOG_BLOCKED_SIZE_MAX) { } else if (out_len < FORT_LOG_BLOCKED_SIZE_MAX) {
status = STATUS_BUFFER_TOO_SMALL; status = STATUS_BUFFER_TOO_SMALL;
@ -143,6 +145,7 @@ fort_buffer_cancel_pending (PFORT_BUFFER buf, PIRP irp, ULONG_PTR *info)
KeAcquireSpinLock(&buf->lock, &irq); KeAcquireSpinLock(&buf->lock, &irq);
if (irp == buf->irp) { if (irp == buf->irp) {
buf->irp = NULL; buf->irp = NULL;
buf->out_len = 0;
if (buf->out_top) { if (buf->out_top) {
*info = buf->out_top; *info = buf->out_top;

13
src/driver/forttmr.c
View File

@ -12,23 +12,24 @@ static void
fort_timer_callback (PKDPC dpc, PFORT_BUFFER buf, PVOID arg1, PVOID arg2) fort_timer_callback (PKDPC dpc, PFORT_BUFFER buf, PVOID arg1, PVOID arg2)
{ {
PIRP irp = NULL; PIRP irp = NULL;
ULONG info = 0; ULONG info;
UNUSED(dpc); UNUSED(dpc);
UNUSED(arg1); UNUSED(arg1);
UNUSED(arg2); UNUSED(arg2);
KeAcquireSpinLockAtDpcLevel(&buf->lock); KeAcquireSpinLockAtDpcLevel(&buf->lock);
if (buf->irp != NULL && buf->out_top) { info = buf->out_top;
if (info) {
buf->out_top = 0;
buf->out_len = 0;
irp = buf->irp; irp = buf->irp;
buf->irp = NULL; buf->irp = NULL;
info = buf->out_top;
buf->out_top = 0;
} }
KeReleaseSpinLockFromDpcLevel(&buf->lock); KeReleaseSpinLockFromDpcLevel(&buf->lock);
if (irp != NULL) { if (info) {
fort_request_complete_info(irp, STATUS_SUCCESS, info); fort_request_complete_info(irp, STATUS_SUCCESS, info);
} }
} }