PRB: The 1394DIAG Sample Does Not Function Correctly (280077)

RESOLUTION

NTSTATUS
t1394_BusReset(
    IN PDEVICE_OBJECT   DeviceObject,
    IN PIRP             Irp,
    IN ULONG            fulFlags
    )
{
    NTSTATUS            ntStatus = STATUS_SUCCESS;
    PDEVICE_EXTENSION   deviceExtension = DeviceObject->DeviceExtension;
    PIRB                pIrb;
    PIRP	        newIrp;
    BOOLEAN             allocNewIrp = FALSE;
    KEVENT              Event;
    IO_STATUS_BLOCK     ioStatus;
	
    ENTER("t1394_BusReset");

    TRACE(TL_TRACE, ("fulFlags = 0x%x\n", fulFlags));

    // If this is a UserMode request, create a newIrp so that the request
    // is issued from KernelMode.
	
    if (Irp->RequestorMode == UserMode) {
         newIrp = IoBuildDeviceIoControlRequest (IOCTL_1394_CLASS,
                               deviceExtension->StackDeviceObject, 
	                       NULL, 0, NULL, 0, TRUE, &Event,
                               &ioStatus);

         if (!newIrp) {
             TRACE(TL_ERROR, ("Failed to allocate newIrp!\n"));
             TRAP;
      
             ntStatus = STATUS_INSUFFICIENT_RESOURCES;
             goto Exit_BusReset;            
         }
         allocNewIrp = TRUE;
    }
   	
    pIrb = ExAllocatePool(NonPagedPool, sizeof(IRB));

    if (!pIrb) {

        TRACE(TL_ERROR, ("Failed to allocate pIrb!\n"));
        TRAP;

        ntStatus = STATUS_INSUFFICIENT_RESOURCES;
        goto Exit_BusReset;
    } // if

    RtlZeroMemory (pIrb, sizeof (IRB));
    pIrb->FunctionNumber = REQUEST_BUS_RESET;
    pIrb->Flags = 0;
    pIrb->u.BusReset.fulFlags = fulFlags;

    // 
    // If you allocated this IRP, submit it asynchronously and wait for its
    // completion event to be signaled, otherwise, submit it synchronously.
    // 
    if (allocNewIrp) {
        KeInitializeEvent (&Event, NotificationEvent, FALSE);
        ntStatus = t1394_SubmitIrpAsync(deviceExtension->StackDeviceObject, 
                                        newIrp, pIrb);

        if (ntStatus == STATUS_PENDING) {
            KeWaitForSingleObject (&Event, Executive, KernelMode, FALSE,
                                   NULL); 
            ntStatus = ioStatus.Status;
        }
    }
    else {
        ntStatus = t1394_SubmitIrpSynch(deviceExtension->StackDeviceObject,
                                        Irp, pIrb);
    }
    
    if (!NT_SUCCESS(ntStatus)) {

        TRACE(TL_ERROR, ("SubmitIrpSync failed = 0x%x\n", ntStatus));
        TRAP;
    }

    ExFreePool(pIrb);

Exit_BusReset:

    if (allocNewIrp) 
        Irp->IoStatus = ioStatus;
        
    EXIT("t1394_BusReset", ntStatus);
    return(ntStatus);