worker-thread.c File Reference

#include "btrfs_drv.h"

Include dependency graph for worker-thread.c:

Go to the source code of this file.

Classes
struct	job_info

Functions
NTSTATUS	do_read_job (PIRP Irp)
NTSTATUS	do_write_job (device_extension *Vcb, PIRP Irp)
	_Function_class_ (WORKER_THREAD_ROUTINE)
bool	add_thread_job (device_extension *Vcb, PIRP Irp)

Function Documentation

_Function_class_()

_Function_class_

(

WORKER_THREAD_ROUTINE

)

Definition at line 95 of file worker-thread.c.

                                            {
    job_info* ji = context;
    PIO_STACK_LOCATION IrpSp = ji->Irp ? IoGetCurrentIrpStackLocation(ji->Irp) : NULL;

    if (IrpSp->MajorFunction == IRP_MJ_READ) {
        do_read_job(ji->Irp);
    } else if (IrpSp->MajorFunction == IRP_MJ_WRITE) {
        do_write_job(ji->Vcb, ji->Irp);
    }

    ExFreePool(ji);
}

add_thread_job()

bool add_thread_job	(	device_extension *	Vcb,
		PIRP	Irp
	)

Definition at line 109 of file worker-thread.c.

                                                     {
    job_info* ji;

    ji = ExAllocatePoolWithTag(NonPagedPool, sizeof(job_info), ALLOC_TAG);
    if (!ji) {
        ERR("out of memory\n");
        return false;
    }

    ji->Vcb = Vcb;
    ji->Irp = Irp;

    if (!Irp->MdlAddress) {
        PMDL Mdl;
        LOCK_OPERATION op;
        ULONG len;
        PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(Irp);

        if (IrpSp->MajorFunction == IRP_MJ_READ) {
            op = IoWriteAccess;
            len = IrpSp->Parameters.Read.Length;
        } else if (IrpSp->MajorFunction == IRP_MJ_WRITE) {
            op = IoReadAccess;
            len = IrpSp->Parameters.Write.Length;
        } else {
            ERR("unexpected major function %u\n", IrpSp->MajorFunction);
            ExFreePool(ji);
            return false;
        }

        Mdl = IoAllocateMdl(Irp->UserBuffer, len, false, false, Irp);

        if (!Mdl) {
            ERR("out of memory\n");
            ExFreePool(ji);
            return false;
        }

        _SEH2_TRY {
            MmProbeAndLockPages(Mdl, Irp->RequestorMode, op);
        } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) {
            ERR("MmProbeAndLockPages raised status %08lx\n", _SEH2_GetExceptionCode());

            IoFreeMdl(Mdl);
            Irp->MdlAddress = NULL;
            ExFreePool(ji);

            _SEH2_YIELD(return FALSE);
        } _SEH2_END;
    }

    ExInitializeWorkItem(&ji->item, do_job, ji);
    ExQueueWorkItem(&ji->item, DelayedWorkQueue);

    return true;
}

Referenced by _Dispatch_type_().

do_read_job()

NTSTATUS do_read_job

(

PIRP

Irp

)

Definition at line 26 of file worker-thread.c.

                               {
    NTSTATUS Status;
    ULONG bytes_read;
    bool top_level = is_top_level(Irp);
    PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(Irp);
    PFILE_OBJECT FileObject = IrpSp->FileObject;
    fcb* fcb = FileObject->FsContext;
    bool acquired_fcb_lock = false;

    Irp->IoStatus.Information = 0;

    if (!ExIsResourceAcquiredSharedLite(fcb->Header.Resource)) {
        ExAcquireResourceSharedLite(fcb->Header.Resource, true);
        acquired_fcb_lock = true;
    }

    _SEH2_TRY {
        Status = do_read(Irp, true, &bytes_read);
    } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) {
        Status = _SEH2_GetExceptionCode();
    } _SEH2_END;

    if (acquired_fcb_lock)
        ExReleaseResourceLite(fcb->Header.Resource);

    if (!NT_SUCCESS(Status))
        ERR("do_read returned %08lx\n", Status);

    Irp->IoStatus.Status = Status;

    TRACE("read %Iu bytes\n", Irp->IoStatus.Information);

    IoCompleteRequest(Irp, IO_NO_INCREMENT);

    if (top_level)
        IoSetTopLevelIrp(NULL);

    TRACE("returning %08lx\n", Status);

    return Status;
}

Referenced by _Dispatch_type_(), and _Function_class_().

do_write_job()

NTSTATUS do_write_job	(	device_extension *	Vcb,
		PIRP	Irp
	)

Definition at line 68 of file worker-thread.c.

                                                       {
    bool top_level = is_top_level(Irp);
    NTSTATUS Status;

    _SEH2_TRY {
        Status = write_file(Vcb, Irp, true, true);
    } _SEH2_EXCEPT (EXCEPTION_EXECUTE_HANDLER) {
        Status = _SEH2_GetExceptionCode();
    } _SEH2_END;

    if (!NT_SUCCESS(Status))
        ERR("write_file returned %08lx\n", Status);

    Irp->IoStatus.Status = Status;

    TRACE("wrote %Iu bytes\n", Irp->IoStatus.Information);

    IoCompleteRequest(Irp, IO_NO_INCREMENT);

    if (top_level)
        IoSetTopLevelIrp(NULL);

    TRACE("returning %08lx\n", Status);

    return Status;
}

Referenced by _Dispatch_type_(), and _Function_class_().