100 lines
2.6 KiB
C
100 lines
2.6 KiB
C
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
//
|
||
|
|
// Copyright (c) 1998, Microsoft Corp. All rights reserved.
|
||
|
|
//
|
||
|
|
// FILE
|
||
|
|
//
|
||
|
|
// VSASink.h
|
||
|
|
//
|
||
|
|
// SYNOPSIS
|
||
|
|
//
|
||
|
|
// This file declares the class VSASink.
|
||
|
|
//
|
||
|
|
// MODIFICATION HISTORY
|
||
|
|
//
|
||
|
|
// 03/07/1998 Original version.
|
||
|
|
//
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
#ifndef _VSASINK_H_
|
||
|
|
#define _VSASINK_H_
|
||
|
|
#if _MSC_VER >= 1000
|
||
|
|
#pragma once
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include <iaspolcy.h>
|
||
|
|
#include <nocopy.h>
|
||
|
|
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
//
|
||
|
|
// CLASS
|
||
|
|
//
|
||
|
|
// SubVSA
|
||
|
|
//
|
||
|
|
// DESCRIPTION
|
||
|
|
//
|
||
|
|
// Encapsulates the information for converting an IAS Attribute into a
|
||
|
|
// RADIUS vendor specific attribute.
|
||
|
|
//
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
class SubVSA
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
DWORD vendorID; // The Vendor-ID in host order.
|
||
|
|
BYTE vendorType; // The Vendor-Type code.
|
||
|
|
PIASATTRIBUTE attr; // Protocol-independent representation.
|
||
|
|
|
||
|
|
// When sorting, we want to group by vendor ID and flags. We don't care
|
||
|
|
// about type.
|
||
|
|
bool operator<(const SubVSA& s) const throw ()
|
||
|
|
{
|
||
|
|
return vendorID == s.vendorID ? attr->dwFlags < s.attr->dwFlags
|
||
|
|
: vendorID < s.vendorID;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
//
|
||
|
|
// CLASS
|
||
|
|
//
|
||
|
|
// VSASink
|
||
|
|
//
|
||
|
|
// DESCRIPTION
|
||
|
|
//
|
||
|
|
// This class converts a Request object into a sink for SubVSA's. Sub-VSA's
|
||
|
|
// inserted into the sink are converted into IASAttributes's and inserted
|
||
|
|
// into the request object.
|
||
|
|
//
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
class VSASink
|
||
|
|
: NonCopyable
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
// Construct a sink from a request.
|
||
|
|
explicit VSASink(IAttributesRaw* request) throw ();
|
||
|
|
|
||
|
|
// Insert a SubVSA into the sink.
|
||
|
|
VSASink& operator<<(const SubVSA& vsa);
|
||
|
|
|
||
|
|
// Flush the sink. This should be called after all SubVSA's have been
|
||
|
|
// inserted to ensure that everything has been inserted into the request.
|
||
|
|
void flush();
|
||
|
|
|
||
|
|
protected:
|
||
|
|
enum
|
||
|
|
{
|
||
|
|
NO_VENDOR = 0, // Indicates a 'blank' vendor ID.
|
||
|
|
MAX_SUBVSA_LENGTH = 249, // Maximum length of a sub-VSA.
|
||
|
|
MAX_VSA_LENGTH = 253 // Maximum length of a consolidated VSA.
|
||
|
|
};
|
||
|
|
|
||
|
|
CComPtr<IAttributesRaw> raw; // The request object being wrapped.
|
||
|
|
BYTE buffer[MAX_VSA_LENGTH]; // Buffer used for building VSA's.
|
||
|
|
size_t bufferLength; // Number of bytes in the buffer.
|
||
|
|
DWORD currentVendor; // Vendor being processed.
|
||
|
|
DWORD currentFlags; // Flags for current VSA.
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // _VSASINK_H_
|