wait.c

这个是一个开源项目, 有能力的人可以一起来写

/*
 * $Id: wait.c,v 1.2 2006/06/23 11:44:52 vfrolov Exp $
 *
 * Copyright (c) 2004-2006 Vyacheslav Frolov
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 *
 * $Log: wait.c,v $
 * Revision 1.2  2006/06/23 11:44:52  vfrolov
 * Mass replacement pDevExt by pIoPort
 *
 * Revision 1.1  2005/01/26 12:18:54  vfrolov
 * Initial revision
 *
 *
 */

#include "precomp.h"

NTSTATUS StartIrpWaitOnMask(
    IN PC0C_IO_PORT pIoPort,
    IN PLIST_ENTRY pQueueToComplete)
{
  UNREFERENCED_PARAMETER(pQueueToComplete);

  if (!pIoPort->waitMask)
    return STATUS_INVALID_PARAMETER;

  if (pIoPort->eventMask) {
    PIRP pIrp;

    pIrp = pIoPort->irpQueues[C0C_QUEUE_WAIT].pCurrent;

    *((PULONG)pIrp->AssociatedIrp.SystemBuffer) = pIoPort->eventMask;
    pIoPort->eventMask = 0;
    pIrp->IoStatus.Information = sizeof(ULONG);
    return STATUS_SUCCESS;
  }

  return STATUS_PENDING;
}

NTSTATUS FdoPortWaitOnMask(
    IN PC0C_IO_PORT pIoPort,
    IN PIRP pIrp,
    IN PIO_STACK_LOCATION pIrpStack)
{
  if (pIrpStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(ULONG))
    return STATUS_BUFFER_TOO_SMALL;

  return FdoPortStartIrp(pIoPort, pIrp, C0C_QUEUE_WAIT, StartIrpWaitOnMask);
}

NTSTATUS FdoPortSetWaitMask(
    IN PC0C_IO_PORT pIoPort,
    IN PIRP pIrp,
    IN PIO_STACK_LOCATION pIrpStack)
{
  LIST_ENTRY queueToComplete;
  KIRQL oldIrql;
  PULONG pSysBuf;

  if (pIrpStack->Parameters.DeviceIoControl.InputBufferLength < sizeof(ULONG))
    return STATUS_BUFFER_TOO_SMALL;

  pSysBuf = (PULONG)pIrp->AssociatedIrp.SystemBuffer;

  if (*pSysBuf & ~(
      SERIAL_EV_RXCHAR   |
      SERIAL_EV_RXFLAG   |
      SERIAL_EV_TXEMPTY  |
      SERIAL_EV_CTS      |
      SERIAL_EV_DSR      |
      SERIAL_EV_RLSD     |
      SERIAL_EV_BREAK    |
      SERIAL_EV_ERR      |
      SERIAL_EV_RING     |
      SERIAL_EV_PERR     |
      SERIAL_EV_RX80FULL |
      SERIAL_EV_EVENT1   |
      SERIAL_EV_EVENT2
      ))
    return STATUS_INVALID_PARAMETER;

  InitializeListHead(&queueToComplete);

  KeAcquireSpinLock(pIoPort->pIoLock, &oldIrql);
  FdoPortIo(
      C0C_IO_TYPE_WAIT_COMPLETE,
      &pIoPort->eventMask,
      pIoPort,
      &pIoPort->irpQueues[C0C_QUEUE_WAIT],
      &queueToComplete);

  pIoPort->waitMask = *pSysBuf;
  pIoPort->eventMask = 0;

  KeReleaseSpinLock(pIoPort->pIoLock, oldIrql);
  FdoPortCompleteQueue(&queueToComplete);

  return STATUS_SUCCESS;
}

NTSTATUS FdoPortGetWaitMask(
    IN PC0C_IO_PORT pIoPort,
    IN PIRP pIrp,
    IN PIO_STACK_LOCATION pIrpStack)
{
  KIRQL oldIrql;
  PULONG pSysBuf;

  if (pIrpStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(ULONG))
    return STATUS_BUFFER_TOO_SMALL;

  pSysBuf = (PULONG)pIrp->AssociatedIrp.SystemBuffer;

  KeAcquireSpinLock(pIoPort->pIoLock, &oldIrql);
  *pSysBuf = pIoPort->waitMask;
  KeReleaseSpinLock(pIoPort->pIoLock, oldIrql);

  pIrp->IoStatus.Information = sizeof(ULONG);

  return STATUS_SUCCESS;
}

VOID WaitComplete(
    IN PC0C_IO_PORT pIoPort,
    PLIST_ENTRY pQueueToComplete)
{
  if (pIoPort->eventMask) {
    FdoPortIo(
        C0C_IO_TYPE_WAIT_COMPLETE,
        &pIoPort->eventMask,
        pIoPort,
        &pIoPort->irpQueues[C0C_QUEUE_WAIT],
        pQueueToComplete);
  }
}