windows-server-2003/ds/security/protocols/kerberos/parser/asn1integer.cxx

//=============================================================================
//
//  MODULE: ASN1Integer.cxx
//
//  Description:
//
//  Implementation of ASN.1 integer parsing logic
//
//  Modification History
//
//  Mark Pustilnik              Date: 06/08/02 - created
//
//=============================================================================

#include "ASN1Parser.hxx"

//
// Retrieve the value of an ASN.1-encoded integer
//

DWORD
ASN1ParserInteger::GetValue(
    IN OUT ASN1FRAME * Frame,
    OUT ASN1VALUE * Value
    )
{
    DWORD dw;
    ASN1FRAME FrameIn = *Frame;
    DWORD DLength = DataLength( Frame );
    ULPBYTE Address = Frame->Address;

    if ( *Frame->Address != BuildDescriptor(
                                ctUniversal,
                                pcPrimitive,
                                (BYTE)utInteger ))
    {
        dw = ERROR_INVALID_USER_BUFFER;
        goto Cleanup;
    }

    //
    // Skip over the the descriptor
    //

    Address++;

    //
    // Skip over the length header
    //

    Address += HeaderLength( Address );

    Value->Address = Address;
    Value->Length = DLength;
    Value->ut = utInteger;
    Value->dw = 0;

    for ( ULONG i = 0; i < DLength; i++ )
    {
        Value->dw <<= 8;
        Value->dw += *Address;
        Address++;
    }

    //
    // Mask off the bits we don't care about by making use of the bitmask
    //

    Value->dw &= m_BitMask;

    Frame->Address = Address;

    dw = ERROR_SUCCESS;

Cleanup:

    if ( dw != ERROR_SUCCESS )
    {
        *Frame = FrameIn;
    }

    return dw;
}