windows-server-2003/net/ias/protocol/radius/packetreceiver.h

//#--------------------------------------------------------------
//
//  File:       packetreceiver.h
//
//  Synopsis:   This file holds the declarations of the
//            CPacketReceiver class
//
//
//  History:     9/23/97  MKarki Created
//
//    Copyright (C) 1997-98 Microsoft Corporation
//    All rights reserved.
//
//----------------------------------------------------------------
#ifndef _PACKETRECEIVER_H_
#define _PACKETRECEIVER_H_

#include "mempool.h"
#include "packetio.h"
#include "packetradius.h"
#include "prevalidator.h"
#include "reportevent.h"
#include "clients.h"
#include "sockevt.h"
#include "radcommon.h"
#include "regex.h"

class CHashMD5;

class CHashHmacMD5;

class CDictionary;

class CPacketReceiver : public CPacketIo
{
public:

    //
    //  initializes the CPacketReceiver class object
    //
   BOOL Init (
         /*[in]*/   CDictionary   *pCDictionary,
         /*[in]*/   CPreValidator *pCPreValidator,
           /*[in]*/ CHashMD5      *pCHashMD5,
           /*[in]*/ CHashHmacMD5  *pCHashHmacMD5,
           /*[in]*/ CClients      *pCClients,
           /*[in]*/ CReportEvent  *pCReportEvent
      );

    //
    //  start processing of the new packet received
    //
   HRESULT ReceivePacket (
             /*[in]*/   PBYTE           pInBuffer,
             /*[in]*/   DWORD           dwSize,
             /*[in]*/   DWORD           dwIPaddress,
             /*[in]*/   WORD            wPort,
             /*[in]*/   SOCKET          sock,
             /*[in]*/   PORTTYPE        portType
             );

   // The receiver thread worker routine. Returns 'true' if the function should
   // be called again because the worker was unable to schedule a replacement.
   // The caller should continue to call WorkerRoutine until it returns false.
   bool WorkerRoutine(DWORD dwInfo) throw ();

    //
    //  initate the processing of inbound data
    //
    BOOL StartProcessing (
                    fd_set&  AuthSet,
                    fd_set&  AcctSet
                    );

    //
    //  stop processing of inbound data
    //
    BOOL StopProcessing (
                    VOID
                    );

    //
    //  constructor
    //
   CPacketReceiver(VOID);

    //
    //  destructor
    //
   virtual ~CPacketReceiver(VOID);

private:
   // State passed to the receiver thread.
   struct ReceiverCallback : IAS_CALLBACK
   {
      CPacketReceiver* self;
      DWORD dwInfo;
   };

   // Thread start routine for the receiver thread.
   static void WINAPI CallbackRoutine(IAS_CALLBACK* context) throw ();

    BOOL  StartThreadIfNeeded (
                /*[in]*/    DWORD dwHandle
                );

    void ProcessInvalidPacketSize(
                                    /*in*/ DWORD dwInfo,
                                    /*in*/ const void* pBuffer,
                                    /*in*/ DWORD address
                                 );


    BSTR pingPattern;
    RegularExpression regexp;

    CPreValidator   *m_pCPreValidator;

    CHashMD5        *m_pCHashMD5;

    CHashHmacMD5    *m_pCHashHmacMD5;

    CDictionary     *m_pCDictionary;

    CClients        *m_pCClients;

    CReportEvent    *m_pCReportEvent;

    //
    //   memory pool for UDP in request
    //
    memory_pool <MAX_PACKET_SIZE, task_allocator> m_InBufferPool;

    //
    // socket sets
    //
    fd_set          m_AuthSet;
    fd_set          m_AcctSet;

    // Used for knocking threads out of select.
    SocketEvent m_AuthEvent;
    SocketEvent m_AcctEvent;
};

#endif //   infndef _PACKETRECEIVER_H_