mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1177 lines
22 KiB
1177 lines
22 KiB
/*++
|
|
|
|
Copyright (c) 1996-2001 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
rralloc.c
|
|
|
|
Abstract:
|
|
|
|
Domain Name System (DNS) Library
|
|
|
|
Resource record allocation \ creation routines.
|
|
|
|
Author:
|
|
|
|
Jim Gilroy (jamesg) January, 1997
|
|
|
|
Environment:
|
|
|
|
User Mode - Win32
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
|
|
|
|
#include "local.h"
|
|
|
|
#define SET_FLAGS(Flags, value) \
|
|
( *(PWORD)&(Flags) = value )
|
|
|
|
|
|
|
|
PDNS_RECORD
|
|
WINAPI
|
|
Dns_AllocateRecord(
|
|
IN WORD wBufferLength
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Allocate record structure.
|
|
|
|
Arguments:
|
|
|
|
wBufferLength - desired buffer length (beyond structure header)
|
|
|
|
Return Value:
|
|
|
|
Ptr to message buffer.
|
|
NULL on error.
|
|
|
|
--*/
|
|
{
|
|
PDNS_RECORD prr;
|
|
|
|
prr = ALLOCATE_HEAP( SIZEOF_DNS_RECORD_HEADER + wBufferLength );
|
|
if ( prr == NULL )
|
|
{
|
|
SetLastError( DNS_ERROR_NO_MEMORY );
|
|
return( NULL );
|
|
}
|
|
RtlZeroMemory(
|
|
prr,
|
|
SIZEOF_DNS_RECORD_HEADER );
|
|
|
|
// as first cut, set datalength to buffer length
|
|
|
|
prr->wDataLength = wBufferLength;
|
|
return( prr );
|
|
}
|
|
|
|
|
|
|
|
VOID
|
|
WINAPI
|
|
Dns_RecordFree(
|
|
IN OUT PDNS_RECORD pRecord
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Free a record
|
|
|
|
Arguments:
|
|
|
|
pRecord -- record list to free
|
|
|
|
Return Value:
|
|
|
|
None.
|
|
|
|
--*/
|
|
{
|
|
DNSDBG( HEAP, ( "Dns_RecordFree( %p )\n", pRecord ));
|
|
|
|
// handle NULL for convenience
|
|
|
|
if ( !pRecord )
|
|
{
|
|
return;
|
|
}
|
|
|
|
// free owner name?
|
|
|
|
if ( FLAG_FreeOwner( pRecord ) )
|
|
{
|
|
FREE_HEAP( pRecord->pName );
|
|
}
|
|
|
|
//
|
|
// free data -- but only if flag set
|
|
//
|
|
// note: even if we fix copy functions to do atomic
|
|
// allocations, we'll still have to have free to
|
|
// handle RPC allocations
|
|
// (unless we very cleverly, treated RPC as flat blob, then
|
|
// did fix up (to offsets before and afterward)
|
|
//
|
|
|
|
if ( FLAG_FreeData( pRecord ) )
|
|
{
|
|
switch( pRecord->wType )
|
|
{
|
|
case DNS_TYPE_A:
|
|
break;
|
|
|
|
case DNS_TYPE_PTR:
|
|
case DNS_TYPE_NS:
|
|
case DNS_TYPE_CNAME:
|
|
case DNS_TYPE_MB:
|
|
case DNS_TYPE_MD:
|
|
case DNS_TYPE_MF:
|
|
case DNS_TYPE_MG:
|
|
case DNS_TYPE_MR:
|
|
|
|
if ( pRecord->Data.PTR.pNameHost )
|
|
{
|
|
FREE_HEAP( pRecord->Data.PTR.pNameHost );
|
|
}
|
|
break;
|
|
|
|
case DNS_TYPE_SOA:
|
|
|
|
if ( pRecord->Data.SOA.pNamePrimaryServer )
|
|
{
|
|
FREE_HEAP( pRecord->Data.SOA.pNamePrimaryServer );
|
|
}
|
|
if ( pRecord->Data.SOA.pNameAdministrator )
|
|
{
|
|
FREE_HEAP( pRecord->Data.SOA.pNameAdministrator );
|
|
}
|
|
break;
|
|
|
|
case DNS_TYPE_MINFO:
|
|
case DNS_TYPE_RP:
|
|
|
|
if ( pRecord->Data.MINFO.pNameMailbox )
|
|
{
|
|
FREE_HEAP( pRecord->Data.MINFO.pNameMailbox );
|
|
}
|
|
if ( pRecord->Data.MINFO.pNameErrorsMailbox )
|
|
{
|
|
FREE_HEAP( pRecord->Data.MINFO.pNameErrorsMailbox );
|
|
}
|
|
break;
|
|
|
|
case DNS_TYPE_MX:
|
|
case DNS_TYPE_AFSDB:
|
|
case DNS_TYPE_RT:
|
|
|
|
if ( pRecord->Data.MX.pNameExchange )
|
|
{
|
|
FREE_HEAP( pRecord->Data.MX.pNameExchange );
|
|
}
|
|
break;
|
|
|
|
case DNS_TYPE_HINFO:
|
|
case DNS_TYPE_ISDN:
|
|
case DNS_TYPE_TEXT:
|
|
case DNS_TYPE_X25:
|
|
|
|
{
|
|
DWORD iter;
|
|
DWORD count = pRecord->Data.TXT.dwStringCount;
|
|
|
|
for ( iter = 0; iter < count; iter++ )
|
|
{
|
|
if ( pRecord->Data.TXT.pStringArray[iter] )
|
|
{
|
|
FREE_HEAP( pRecord->Data.TXT.pStringArray[iter] );
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
case DNS_TYPE_SRV:
|
|
|
|
if ( pRecord->Data.SRV.pNameTarget )
|
|
{
|
|
FREE_HEAP( pRecord->Data.SRV.pNameTarget );
|
|
}
|
|
break;
|
|
|
|
case DNS_TYPE_WINSR:
|
|
|
|
if ( pRecord->Data.WINSR.pNameResultDomain )
|
|
{
|
|
FREE_HEAP( pRecord->Data.WINSR.pNameResultDomain );
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
// other types -- A, AAAA, ATMA, WINS, NULL,
|
|
// have no internal pointers
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
// for catching heap problems
|
|
|
|
pRecord->pNext = DNS_BAD_PTR;
|
|
pRecord->pName = DNS_BAD_PTR;
|
|
|
|
FREE_HEAP( pRecord );
|
|
}
|
|
|
|
|
|
|
|
VOID
|
|
WINAPI
|
|
Dns_RecordListFree(
|
|
IN OUT PDNS_RECORD pRecord
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Free list of records.
|
|
|
|
Arguments:
|
|
|
|
pRecord -- record list to free
|
|
|
|
Return Value:
|
|
|
|
None.
|
|
|
|
--*/
|
|
{
|
|
PDNS_RECORD pnext;
|
|
|
|
DNSDBG( TRACE, (
|
|
"Dns_RecordListFree( %p )\n",
|
|
pRecord ));
|
|
|
|
//
|
|
// loop through and free every RR in list
|
|
//
|
|
|
|
while ( pRecord )
|
|
{
|
|
pnext = pRecord->pNext;
|
|
|
|
Dns_RecordFree( pRecord );
|
|
|
|
pRecord = pnext;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
VOID
|
|
WINAPI
|
|
Dns_RecordListFreeEx(
|
|
IN OUT PDNS_RECORD pRecord,
|
|
IN BOOL fFreeOwner
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Free list of records.
|
|
|
|
DCR: RecordListFreeEx (no free owner option) is probably useless
|
|
|
|
Note: owner name is freed ONLY when indicated by flag;
|
|
other ptrs are considered to be either
|
|
1) internal as when records read from wire or copied
|
|
2) external and to be freed by record creator
|
|
|
|
Arguments:
|
|
|
|
pRecord -- record list to free
|
|
|
|
fFreeOwner -- flag indicating owner name should be freed
|
|
|
|
Return Value:
|
|
|
|
None.
|
|
|
|
--*/
|
|
{
|
|
PDNS_RECORD pnext;
|
|
|
|
DNSDBG( TRACE, (
|
|
"Dns_RecordListFreeEx( %p, %d )\n",
|
|
pRecord,
|
|
fFreeOwner ));
|
|
|
|
//
|
|
// loop through and free every RR in list
|
|
//
|
|
|
|
while ( pRecord )
|
|
{
|
|
pnext = pRecord->pNext;
|
|
|
|
// free owner name?
|
|
// - if "FreeOwner" flag NOT set, then don't free
|
|
|
|
if ( !fFreeOwner )
|
|
{
|
|
FLAG_FreeOwner( pRecord ) = FALSE;
|
|
}
|
|
|
|
// free record
|
|
|
|
Dns_RecordFree( pRecord );
|
|
|
|
pRecord = pnext;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
//
|
|
// Special record type creation routines
|
|
//
|
|
|
|
PDNS_RECORD
|
|
CreateRecordBasic(
|
|
IN PDNS_NAME pOwnerName,
|
|
IN BOOL fCopyName,
|
|
IN WORD wType,
|
|
IN WORD wDataLength,
|
|
IN DWORD AllocLength,
|
|
IN DWORD Ttl,
|
|
IN DNS_CHARSET NameCharSet,
|
|
IN DNS_CHARSET RecordCharSet
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Create record of arbitary type.
|
|
|
|
Helper function to wrap up
|
|
- record alloc
|
|
- name alloc
|
|
- basic setup
|
|
|
|
Arguments:
|
|
|
|
pOwnerName -- owner name
|
|
|
|
fCopyName -- TRUE - make copy of owner name
|
|
FALSE - use directly
|
|
|
|
wType -- type
|
|
|
|
AllocLength -- allocaction length, including any imbedded data
|
|
|
|
wDataLength -- data length to set
|
|
|
|
Ttl -- TTL
|
|
|
|
NameCharSet -- character set of name
|
|
|
|
RecordCharSet -- character set for resulting record
|
|
|
|
Return Value:
|
|
|
|
Ptr to PTR record.
|
|
NULL on error.
|
|
|
|
--*/
|
|
{
|
|
PDNS_RECORD precord;
|
|
PDNS_RECORD prr;
|
|
PCHAR pname;
|
|
DWORD bufLength;
|
|
|
|
//
|
|
// alloc record
|
|
//
|
|
|
|
prr = Dns_AllocateRecord( (WORD)AllocLength );
|
|
if ( !prr )
|
|
{
|
|
return( NULL );
|
|
}
|
|
|
|
//
|
|
// copy owner name
|
|
//
|
|
|
|
if ( fCopyName && pOwnerName )
|
|
{
|
|
pname = Dns_NameCopyAllocate(
|
|
pOwnerName,
|
|
0, // length unknown
|
|
NameCharSet,
|
|
RecordCharSet );
|
|
if ( !pname )
|
|
{
|
|
FREE_HEAP( prr );
|
|
return( NULL );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
pname = pOwnerName;
|
|
}
|
|
|
|
//
|
|
// set fields
|
|
// - name, type and charset
|
|
// - TTL, section left zero
|
|
// - FreeData is specifically off
|
|
//
|
|
|
|
prr->pName = pname;
|
|
prr->wType = wType;
|
|
prr->wDataLength = wDataLength;
|
|
SET_FREE_OWNER(prr);
|
|
prr->Flags.S.CharSet = RecordCharSet;
|
|
prr->dwTtl = Ttl;
|
|
|
|
return( prr );
|
|
}
|
|
|
|
|
|
|
|
PDNS_RECORD
|
|
Dns_CreateFlatRecord(
|
|
IN PDNS_NAME pOwnerName,
|
|
IN WORD wType,
|
|
IN PCHAR pData,
|
|
IN DWORD DataLength,
|
|
IN DWORD Ttl,
|
|
IN DNS_CHARSET NameCharSet,
|
|
IN DNS_CHARSET RecordCharSet
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Create flat record.
|
|
|
|
Arguments:
|
|
|
|
pOwnerName -- owner name
|
|
|
|
wType -- record type
|
|
|
|
pData -- ptr to data for record
|
|
|
|
DataLength -- length (in bytes) of data
|
|
|
|
Ttl -- TTL
|
|
|
|
NameCharSet -- character set of name
|
|
|
|
RecordCharSet -- character set for resulting record
|
|
|
|
Return Value:
|
|
|
|
Ptr to PTR record.
|
|
NULL on error.
|
|
|
|
--*/
|
|
{
|
|
PDNS_RECORD prr;
|
|
|
|
//
|
|
// determine record size
|
|
// - record buffer will include hostname
|
|
//
|
|
|
|
prr = CreateRecordBasic(
|
|
pOwnerName,
|
|
TRUE, // copy name
|
|
wType,
|
|
(WORD) DataLength, // datalength
|
|
DataLength, // alloc datalength
|
|
Ttl,
|
|
NameCharSet,
|
|
RecordCharSet );
|
|
if ( !prr )
|
|
{
|
|
return( NULL );
|
|
}
|
|
|
|
//
|
|
// copy in data
|
|
//
|
|
|
|
RtlCopyMemory(
|
|
(PBYTE) &prr->Data,
|
|
pData,
|
|
DataLength );
|
|
|
|
return( prr );
|
|
}
|
|
|
|
|
|
|
|
PDNS_RECORD
|
|
Dns_CreatePtrTypeRecord(
|
|
IN PDNS_NAME pOwnerName,
|
|
IN BOOL fCopyName,
|
|
IN PDNS_NAME pTargetName,
|
|
IN WORD wType,
|
|
IN DWORD Ttl,
|
|
IN DNS_CHARSET NameCharSet,
|
|
IN DNS_CHARSET RecordCharSet
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Create PTR type (single-indirection) record.
|
|
|
|
This can be used to create any "PTR-type" record:
|
|
PTR, CNAME, NS, etc.
|
|
|
|
Arguments:
|
|
|
|
pOwnerName -- owner name
|
|
|
|
fCopyName -- TRUE - make copy of owner name
|
|
FALSE - use directly
|
|
|
|
pTargetName -- target name
|
|
|
|
Ttl -- TTL
|
|
|
|
NameCharSet -- character set of name
|
|
|
|
RecordCharSet -- character set for resulting record
|
|
|
|
Return Value:
|
|
|
|
Ptr to PTR record.
|
|
NULL on error.
|
|
|
|
--*/
|
|
{
|
|
PDNS_RECORD precord;
|
|
PDNS_RECORD prr;
|
|
PCHAR pname;
|
|
DWORD bufLength;
|
|
|
|
//
|
|
// determine record size
|
|
// - record buffer will include hostname
|
|
//
|
|
|
|
bufLength = Dns_GetBufferLengthForNameCopy(
|
|
pTargetName,
|
|
0, // length unknown
|
|
NameCharSet,
|
|
RecordCharSet );
|
|
if ( !bufLength )
|
|
{
|
|
return( NULL );
|
|
}
|
|
|
|
//
|
|
// create record
|
|
//
|
|
|
|
prr = CreateRecordBasic(
|
|
pOwnerName,
|
|
fCopyName,
|
|
wType,
|
|
sizeof(DNS_PTR_DATA), // data length
|
|
(sizeof(DNS_PTR_DATA) + bufLength), // alloc length
|
|
Ttl,
|
|
NameCharSet,
|
|
RecordCharSet );
|
|
if ( !prr )
|
|
{
|
|
return( NULL );
|
|
}
|
|
|
|
//
|
|
// write target name into buffer, immediately following PTR data struct
|
|
//
|
|
|
|
prr->Data.PTR.pNameHost = (PCHAR)&prr->Data + sizeof(DNS_PTR_DATA);
|
|
|
|
Dns_NameCopy(
|
|
prr->Data.PTR.pNameHost,
|
|
NULL,
|
|
pTargetName,
|
|
0,
|
|
NameCharSet,
|
|
RecordCharSet
|
|
);
|
|
|
|
return( prr );
|
|
}
|
|
|
|
|
|
|
|
PDNS_RECORD
|
|
Dns_CreatePtrRecordEx(
|
|
IN PIP_UNION pIp,
|
|
IN PDNS_NAME pszHostName,
|
|
IN DWORD Ttl,
|
|
IN DNS_CHARSET NameCharSet,
|
|
IN DNS_CHARSET RecordCharSet
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Create PTR record from IP address and hostname.
|
|
|
|
Arguments:
|
|
|
|
pIp -- IP union (IP4 or IP6)
|
|
|
|
pszHostName -- host name, FULL FQDN
|
|
|
|
Ttl -- TTL
|
|
|
|
NameCharSet -- character set of name
|
|
|
|
RecordCharSet -- character set for resulting record
|
|
|
|
Return Value:
|
|
|
|
Ptr to PTR record.
|
|
NULL on error.
|
|
|
|
--*/
|
|
{
|
|
PCHAR pname;
|
|
|
|
//
|
|
// create reverse lookup name
|
|
// - note this is external allocation
|
|
//
|
|
|
|
if ( IPUNION_IS_IP4( pIp ) )
|
|
{
|
|
IP4_ADDRESS ip = IPUNION_GET_IP4(pIp);
|
|
|
|
if ( RecordCharSet == DnsCharSetUnicode )
|
|
{
|
|
pname = (PCHAR) Dns_Ip4AddressToReverseNameAlloc_W( ip );
|
|
}
|
|
else
|
|
{
|
|
pname = Dns_Ip4AddressToReverseNameAlloc_A( ip );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
IP6_ADDRESS ip = IPUNION_GET_IP6(pIp);
|
|
|
|
if ( RecordCharSet == DnsCharSetUnicode )
|
|
{
|
|
pname = (PCHAR) Dns_Ip6AddressToReverseNameAlloc_W( ip );
|
|
}
|
|
else
|
|
{
|
|
pname = Dns_Ip6AddressToReverseNameAlloc_A( ip );
|
|
}
|
|
}
|
|
|
|
if ( !pname )
|
|
{
|
|
return( NULL );
|
|
}
|
|
|
|
//
|
|
// build record
|
|
//
|
|
|
|
return Dns_CreatePtrTypeRecord(
|
|
pname,
|
|
FALSE, // don't copy IP
|
|
pszHostName, // target name
|
|
DNS_TYPE_PTR,
|
|
Ttl,
|
|
NameCharSet,
|
|
RecordCharSet );
|
|
}
|
|
|
|
|
|
|
|
PDNS_RECORD
|
|
Dns_CreatePtrRecordExEx(
|
|
IN PIP_UNION pIp,
|
|
IN PSTR pszHostName,
|
|
IN PSTR pszDomainName,
|
|
IN DWORD Ttl,
|
|
IN DNS_CHARSET NameCharSet,
|
|
IN DNS_CHARSET RecordCharSet
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Create PTR record from hostname and domain name.
|
|
|
|
Helper function for DHCP registrations when hostname
|
|
and domain name are separate and both required.
|
|
|
|
Arguments:
|
|
|
|
pIp -- IP union (IP4 or IP6)
|
|
|
|
pszHostName -- host name (single label)
|
|
|
|
pszDomainName -- domain name
|
|
|
|
Ttl -- TTL
|
|
|
|
NameCharSet -- character set of name
|
|
|
|
RecordCharSet -- character set for resulting record
|
|
|
|
Return Value:
|
|
|
|
Ptr to PTR record.
|
|
NULL on error.
|
|
|
|
--*/
|
|
{
|
|
WCHAR nameBuffer[ DNS_MAX_NAME_BUFFER_LENGTH ];
|
|
|
|
DNSDBG( TRACE, (
|
|
"Dns_CreatePtrRecordExEx()\n" ));
|
|
|
|
//
|
|
// build appended name
|
|
//
|
|
// DCR: could require just host name and check that
|
|
// either domain exists or hostname is full
|
|
//
|
|
|
|
if ( !pszHostName || !pszDomainName )
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
if ( NameCharSet != DnsCharSetUnicode )
|
|
{
|
|
if ( ! Dns_NameAppend_A(
|
|
(PCHAR) nameBuffer,
|
|
DNS_MAX_NAME_BUFFER_LENGTH,
|
|
pszHostName,
|
|
pszDomainName ) )
|
|
{
|
|
DNS_ASSERT( FALSE );
|
|
return NULL;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if ( ! Dns_NameAppend_W(
|
|
(PWCHAR) nameBuffer,
|
|
DNS_MAX_NAME_BUFFER_LENGTH,
|
|
(PWSTR) pszHostName,
|
|
(PWSTR) pszDomainName ) )
|
|
{
|
|
DNS_ASSERT( FALSE );
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
//
|
|
// build record
|
|
//
|
|
|
|
return Dns_CreatePtrRecordEx(
|
|
pIp,
|
|
(PCHAR) nameBuffer,
|
|
Ttl,
|
|
NameCharSet,
|
|
RecordCharSet
|
|
);
|
|
}
|
|
|
|
|
|
|
|
PDNS_RECORD
|
|
Dns_CreateARecord(
|
|
IN PDNS_NAME pOwnerName,
|
|
IN IP_ADDRESS Ip4Addr,
|
|
IN DWORD Ttl,
|
|
IN DNS_CHARSET NameCharSet,
|
|
IN DNS_CHARSET RecordCharSet
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Create A record.
|
|
|
|
Arguments:
|
|
|
|
pOwnerName -- owner name
|
|
|
|
Ip4Addr -- IP address
|
|
|
|
Ttl -- TTL
|
|
|
|
NameCharSet -- character set of name
|
|
|
|
RecordCharSet -- character set for resulting record
|
|
|
|
Return Value:
|
|
|
|
Ptr to PTR record.
|
|
NULL on error.
|
|
|
|
--*/
|
|
{
|
|
PDNS_RECORD prr;
|
|
|
|
//
|
|
// determine record size
|
|
// - record buffer will include hostname
|
|
//
|
|
|
|
prr = CreateRecordBasic(
|
|
pOwnerName,
|
|
TRUE, // copy name
|
|
DNS_TYPE_A,
|
|
sizeof(DNS_A_DATA),
|
|
sizeof(DNS_A_DATA),
|
|
Ttl,
|
|
NameCharSet,
|
|
RecordCharSet );
|
|
if ( !prr )
|
|
{
|
|
return( NULL );
|
|
}
|
|
|
|
//
|
|
// set IP
|
|
//
|
|
|
|
prr->Data.A.IpAddress = Ip4Addr;
|
|
|
|
return( prr );
|
|
}
|
|
|
|
|
|
|
|
PDNS_RECORD
|
|
Dns_CreateAAAARecord(
|
|
IN PDNS_NAME pOwnerName,
|
|
IN IP6_ADDRESS Ip6Addr,
|
|
IN DWORD Ttl,
|
|
IN DNS_CHARSET NameCharSet,
|
|
IN DNS_CHARSET RecordCharSet
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Create A record.
|
|
|
|
Arguments:
|
|
|
|
pOwnerName -- owner name
|
|
|
|
Ip6Addr -- IP6 address
|
|
|
|
Ttl -- TTL
|
|
|
|
NameCharSet -- character set of name
|
|
|
|
RecordCharSet -- character set for resulting record
|
|
|
|
Return Value:
|
|
|
|
Ptr to PTR record.
|
|
NULL on error.
|
|
|
|
--*/
|
|
{
|
|
PDNS_RECORD prr;
|
|
|
|
//
|
|
// determine record size
|
|
// - record buffer will include hostname
|
|
//
|
|
|
|
prr = CreateRecordBasic(
|
|
pOwnerName,
|
|
TRUE, // copy name
|
|
DNS_TYPE_AAAA,
|
|
sizeof(DNS_AAAA_DATA),
|
|
sizeof(DNS_AAAA_DATA),
|
|
Ttl,
|
|
NameCharSet,
|
|
RecordCharSet );
|
|
if ( !prr )
|
|
{
|
|
return( NULL );
|
|
}
|
|
|
|
//
|
|
// set IP
|
|
//
|
|
|
|
prr->Data.AAAA.Ip6Address = Ip6Addr;
|
|
|
|
return( prr );
|
|
}
|
|
|
|
|
|
|
|
PDNS_RECORD
|
|
Dns_CreateForwardRecord(
|
|
IN PDNS_NAME pOwnerName,
|
|
IN PIP_UNION pIp,
|
|
IN DWORD Ttl,
|
|
IN DNS_CHARSET NameCharSet,
|
|
IN DNS_CHARSET RecordCharSet
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Create forward lookup record.
|
|
|
|
This is just a shim to avoid duplicating selection logic.
|
|
|
|
Arguments:
|
|
|
|
pOwnerName -- owner name
|
|
|
|
pIp -- IP address union
|
|
|
|
Ttl -- TTL
|
|
|
|
NameCharSet -- character set of name
|
|
|
|
RecordCharSet -- character set for resulting record
|
|
|
|
Return Value:
|
|
|
|
Ptr to PTR record.
|
|
NULL on error.
|
|
|
|
--*/
|
|
{
|
|
//
|
|
// build desired type
|
|
//
|
|
// DCR: must add\choose A6
|
|
//
|
|
|
|
if ( IPUNION_IS_IP4( pIp ) )
|
|
{
|
|
return Dns_CreateARecord(
|
|
pOwnerName,
|
|
IPUNION_GET_IP4(pIp),
|
|
Ttl,
|
|
NameCharSet,
|
|
RecordCharSet );
|
|
}
|
|
else
|
|
{
|
|
return Dns_CreateAAAARecord(
|
|
pOwnerName,
|
|
IPUNION_GET_IP6(pIp),
|
|
Ttl,
|
|
NameCharSet,
|
|
RecordCharSet );
|
|
}
|
|
}
|
|
|
|
|
|
|
|
PDNS_RECORD
|
|
Dns_CreateForwardRecordForSockaddr(
|
|
IN PDNS_NAME pOwnerName,
|
|
IN PSOCKADDR pSockaddr,
|
|
IN DWORD Ttl,
|
|
IN DNS_CHARSET NameCharSet,
|
|
IN DNS_CHARSET RecordCharSet
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Create forward lookup record.
|
|
|
|
This is just a shim to avoid duplicating selection logic.
|
|
|
|
Arguments:
|
|
|
|
pOwnerName -- owner name
|
|
|
|
pSockaddr -- ptr to sockaddr
|
|
|
|
Ttl -- TTL
|
|
|
|
NameCharSet -- character set of name
|
|
|
|
RecordCharSet -- character set for resulting record
|
|
|
|
Return Value:
|
|
|
|
Ptr to PTR record.
|
|
NULL on error.
|
|
|
|
--*/
|
|
{
|
|
PFAMILY_INFO pinfo;
|
|
|
|
DNSDBG( TRACE, (
|
|
"Dns_CreateForwardRecordForSockaddr()\n" ));
|
|
|
|
pinfo = FamilyInfo_GetForSockaddr( pSockaddr );
|
|
if ( !pinfo )
|
|
{
|
|
SetLastError( ERROR_INVALID_DATA );
|
|
return NULL;
|
|
}
|
|
|
|
//
|
|
// build flat record of desired type
|
|
//
|
|
|
|
return Dns_CreateFlatRecord(
|
|
pOwnerName,
|
|
pinfo->DnsType,
|
|
(PBYTE)pSockaddr + pinfo->OffsetToAddrInSockaddr,
|
|
pinfo->LengthAddr,
|
|
Ttl,
|
|
NameCharSet,
|
|
RecordCharSet );
|
|
}
|
|
|
|
|
|
|
|
PDNS_RECORD
|
|
Dns_CreateRecordForIpString_W(
|
|
IN PCWSTR pwsName,
|
|
IN WORD wType,
|
|
IN DWORD Ttl
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Create record for IP string query.
|
|
|
|
Arguments:
|
|
|
|
pwsName -- name that may be IP string query
|
|
|
|
wType -- type of query
|
|
|
|
Return Value:
|
|
|
|
Ptr to record for query, if query name\type is IP.
|
|
NULL if query not for IP.
|
|
|
|
--*/
|
|
{
|
|
IP_UNION ipUnion;
|
|
PDNS_RECORD prr;
|
|
|
|
|
|
DNSDBG( TRACE, (
|
|
"\nDns_CreateRecordForIpString( %S, wType = %d )\n",
|
|
pwsName,
|
|
wType ));
|
|
|
|
if ( !pwsName )
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
//
|
|
// support A or AAAA queries for IP strings
|
|
// - IP4 strings must be in w.x.y.z form otherwise
|
|
// we convert the all numeric names also
|
|
//
|
|
// DCR: need A6 support for direct query
|
|
//
|
|
|
|
if ( wType == DNS_TYPE_A )
|
|
{
|
|
IP4_ADDRESS ip4;
|
|
PCWSTR pdot;
|
|
DWORD count;
|
|
|
|
if ( ! Dns_Ip4StringToAddress_W(
|
|
& ip4,
|
|
(PWSTR) pwsName ) )
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
// verify three dot form w.x.y.z
|
|
|
|
pdot = pwsName;
|
|
count = 3;
|
|
while ( count-- )
|
|
{
|
|
pdot = wcschr( pdot, L'.' );
|
|
if ( !pdot || !*++pdot )
|
|
{
|
|
return( NULL );
|
|
}
|
|
}
|
|
|
|
IPUNION_SET_IP4( &ipUnion, ip4 );
|
|
}
|
|
else if ( wType == DNS_TYPE_AAAA )
|
|
{
|
|
IP6_ADDRESS ip6;
|
|
|
|
if ( ! Dns_Ip6StringToAddress_W(
|
|
& ip6,
|
|
(PWSTR) pwsName ) )
|
|
{
|
|
return NULL;
|
|
}
|
|
IPUNION_SET_IP6( &ipUnion, ip6 );
|
|
}
|
|
else
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
//
|
|
// name is IP string -- build record
|
|
//
|
|
|
|
prr = Dns_CreateForwardRecord(
|
|
(PDNS_NAME) pwsName,
|
|
& ipUnion,
|
|
Ttl,
|
|
DnsCharSetUnicode,
|
|
DnsCharSetUnicode );
|
|
|
|
DNSDBG( TRACE, (
|
|
"Create record %p for IP string %S type %d.\n",
|
|
prr,
|
|
pwsName,
|
|
wType ));
|
|
|
|
return prr;
|
|
}
|
|
|
|
//
|
|
// End rralloc.c
|
|
//
|