iptest.cpp File Reference

#include <iostream>
#include <list>
#include <string>
#include <sstream>
#include <stdio.h>
#include <windows.h>
#include <winsock2.h>
#include <ddk/tdi.h>
#include <ddk/tdikrnl.h>
#include <ddk/tdiinfo.h>
#include <ddk/ndis.h>
#include <titypes.h>
#include <ip.h>
#include <tcp.h>
#include <receive.h>
#include <lan.h>
#include <routines.h>

Include dependency graph for iptest.cpp:

Go to the source code of this file.

Classes
class	SocketObject
class	TCPSocketObject

Macros
#define	MAX_DG_SIZE   16384
#define	STRINGIFY(x)   #x

Typedefs
typedef unsigned short	u_short

Functions
VOID	ExpInitLookasideLists ()
void	display_row (char *data, int off, int len)
void	connect_complete (void *context, NTSTATUS status, unsigned long count)
void	receive_complete (void *context, NTSTATUS status, unsigned long count)
UINT	TdiAddressSizeFromType (UINT AddressType)
NTSTATUS	TdiBuildNullConnectionInfoInPlace (PTDI_CONNECTION_INFORMATION ConnInfo, ULONG Type)
NTSTATUS	TdiBuildNullConnectionInfo (PTDI_CONNECTION_INFORMATION *ConnectionInfo, ULONG Type)
UINT	TaLengthOfTransportAddress (PTRANSPORT_ADDRESS Addr)
NTSTATUS	TdiBuildConnectionInfoInPlace (PTDI_CONNECTION_INFORMATION ConnectionInfo, PTA_ADDRESS Address)
NTSTATUS	TdiBuildConnectionInfo (PTDI_CONNECTION_INFORMATION *ConnectionInfo, PTA_ADDRESS Address)
VOID	SendPacket (PVOID Context, PNDIS_PACKET NdisPacket, UINT Offset, PVOID LinkAddress, USHORT Type)
int	main (int argc, char **argv)

Variables
std::list< std::string >	output_packets
DWORD	DebugTraceLevel = 0x7fffffff
PVOID	GlobalBufferPool
PVOID	GlobalPacketPool
char	hwaddr [6] = { 0x08, 0x00, 0x20, 0x0b, 0xb7, 0xbb }
char	hdr [14]

Macro Definition Documentation

MAX_DG_SIZE

#define MAX_DG_SIZE   16384

Definition at line 29 of file iptest.cpp.

STRINGIFY

#define STRINGIFY

(

x

)

#x

Definition at line 37 of file iptest.cpp.

Typedef Documentation

u_short

typedef unsigned short u_short

Definition at line 6 of file iptest.cpp.

Function Documentation

connect_complete()

void connect_complete	(	void *	context,
		NTSTATUS	status,
		unsigned long	count
	)

Definition at line 56 of file iptest.cpp.

                                                                             {
    printf( "Connection: status %x\n", status );
}

Referenced by TCPSocketObject::TCPSocketObject().

display_row()

void display_row	(	char *	data,
		int	off,
		int	len
	)

Definition at line 39 of file iptest.cpp.

                                                 {
    int i;
 
    printf( "%08x:", off );
    for( i = off; i < len && i < off + 16; i++ ) {
    printf( " %02x", data[i] & 0xff );
    }
 
    printf( " -- " );
 
    for( i = off; i < len && i < off + 16; i++ ) {
    printf( "%c", (data[i] >= ' ') ? data[i] : '.' );
    }
 
    printf( "\n" );
}

Referenced by receive_complete(), and SendPacket().

ExpInitLookasideLists()

VOID ExpInitLookasideLists

(

)

Referenced by ExpInitSystemPhase0(), and main().

main()

int main	(	int argc	,
		char **	argv
	)

Definition at line 282 of file iptest.cpp.

                                  {
    int asock = INVALID_SOCKET, selret, dgrecv, fromsize, err, port = 5001;
    int bytes, adapter_id, mtu, speed;
    char datagram[MAX_DG_SIZE];
    struct fd_set readf;
    struct timeval tv;
    struct sockaddr_in addr_from = { AF_INET }, addr_to;
    std::string word, cmdin, host;
    std::list<std::string>::iterator i;
    WSADATA wsadata;
    NTSTATUS Status;
    UNICODE_STRING RegistryUnicodePath;
    PCONNECTION_ENDPOINT Connection;
    PIP_INTERFACE Interface;
    IP_PACKET IPPacket;
    LLIP_BIND_INFO BindInfo;
    SocketObject *S = NULL;
 
    RtlInitUnicodeString
    ( &RegistryUnicodePath,
      L"\\SYSTEM\\CurrentControlSet\\Services"
      L"\\Tcpip" );
 
    ExpInitLookasideLists();
 
    WSAStartup( 0x101, &wsadata );
 
    if( argc > 1 ) port = atoi(argv[1]);
 
    IPStartup( &RegistryUnicodePath );
 
    BindInfo.Context = NULL;
    BindInfo.HeaderSize = sizeof(ETH_HEADER);
    BindInfo.MTU = 1500; /* MTU for ethernet */
    BindInfo.Address = (PUCHAR)hwaddr;
    BindInfo.AddressLength = sizeof(hwaddr);
    BindInfo.Transmit = SendPacket;
 
    IPCreateInterface( &BindInfo );
 
    asock = socket( AF_INET, SOCK_DGRAM, 0 );
 
    addr_from.sin_port = htons( port );
 
    if( bind( asock, (struct sockaddr *)&addr_from, sizeof( addr_from ) ) ) {
    printf( "Bind error\n" );
    return 0;
    }
 
    while( true ) {
    FD_ZERO( &readf );
    FD_SET( asock, &readf );
    tv.tv_sec = 0;
    tv.tv_usec = 10000;
    selret = select( asock + 1, &readf, NULL, NULL, &tv );
 
    if( FD_ISSET( asock, &readf ) ) {
        fromsize = sizeof( addr_from );
        dgrecv = recvfrom( asock, datagram, sizeof(datagram), 0,
                   (struct sockaddr *)&addr_from, &fromsize );
 
        if( datagram[0] == 'C' && datagram[1] == 'M' &&
        datagram[2] == 'D' && datagram[3] == ' ' ) {
        int theport, bytes, recvret, off, bytin;
        struct sockaddr_in nam;
        std::string faddr, word;
        std::istringstream
            cmdin( std::string( datagram + 4, dgrecv - 4 ) );
 
        cmdin >> word;
 
/* UDP Section */
        if( word == "udpsocket" ) {
/* TCP Section */
        } else if( word == "tcpsocket" ) {
            cmdin >> host >> port;
            S = new TCPSocketObject( host, port, &Status );
            fprintf( stderr, "Socket: Result %x\n", Status );
        } else if( word == "close" ) {
            TCPClose( Connection );
            TCPFreeConnectionEndpoint( Connection );
        } else if( word == "type" ) {
            std::string therest = &cmdin.str()[word.size()];
            char* p = &therest[0];
            p += strspn ( p, " \t" );
            char* src = p;
            char* dst = p;
            while ( *src )
            {
            char c = *src++;
            if ( c == '\r' || c == '\n' ) break;
            if ( c == '\\' )
            {
                c = *src++;
                switch ( c )
                {
                case 'b': c = '\b'; break;
                case 'n': c = '\n'; break;
                case 'r': c = '\r'; break;
                case 't': c = '\t'; break;
                case 'v': c = '\v'; break;
                }
            }
            *dst++ = c;
            }
            *dst = '\0';
            if( S )
            err = S->send( p, strlen(p), &bytes, NULL );
            if( err > 0 ) { bytin = err; err = 0; }
 
            if( err )
            fprintf ( stderr, "OskitTCPConnect: error %d\n",
                  err );
            else {
            printf ( "wrote %d bytes\n", bytin );
            }
        } else if( word == "send" ) {
            off = 0;
            while( cmdin >> word ) {
            datagram[off++] =
                atoi( (std::string("0x") + word).c_str() );
            }
 
            if( (err = S->send( datagram, off, &bytin, NULL )) != 0 ) {
            fprintf( stderr, "OskitTCPConnect: error %d\n", err );
            } else {
            printf( "wrote %d bytes\n", bytin );
            }
        } else if( word == "recv" ) {
            cmdin >> bytes;
 
            if( (err = S->recv( datagram,
                    sizeof(datagram),
                    &bytes,
                    NULL )) != 0 ) {
            fprintf( stderr, "OskitTCPRecv: error %d\n", err );
            }
 
/* Misc section */
        } else if( word == "end" ) {
            return 0;
        }
        } else if( dgrecv > 14 ) {
        addr_to = addr_from;
 
        if( datagram[12] == 8 && datagram[13] == 6 ) {
            /* Answer arp query */
            char laddr[4];
            /* Mark patch as to the previous sender */
            memcpy( datagram + 32, datagram + 6, 6 );
            memcpy( datagram, datagram + 6, 6 );
            /* Mark packet as from us */
            memcpy( datagram + 22, hwaddr, 6 );
            memcpy( datagram + 6, hwaddr, 6 );
            /* Swap inet addresses */
            memcpy( laddr, datagram + 28, 4 );
            memcpy( datagram + 28, datagram + 38, 4 );
            memcpy( datagram + 38, laddr, 4 );
            /* Set reply opcode */
            datagram[21] = 2;
 
            err = sendto( asock, datagram, dgrecv, 0,
                  (struct sockaddr *)&addr_to,
                  sizeof(addr_to) );
 
            if( err != 0 )
            printf( "sendto: %d\n", err );
        } else {
            memcpy( hdr, datagram + 6, 6 );
            memcpy( hdr + 6, datagram, 6 );
            memcpy( hdr + 12, datagram + 12, 2 );
            IPPacket.Header = datagram;
            IPPacket.Data = datagram + 14;
            IPPacket.TotalSize = dgrecv;
            IPReceive( Interface, &IPPacket );
        }
        }
    }
 
    IPTimeout(NULL, NULL, NULL, NULL);
 
    for( i = output_packets.begin(); i != output_packets.end(); i++ ) {
        err = sendto( asock, i->c_str(), i->size(), 0,
              (struct sockaddr *)&addr_to, sizeof(addr_to) );
 
        fprintf( stderr, "** SENDING PACKET %d bytes **\n", i->size() );
 
        if( err != 0 )
        printf( "sendto: %d\n", err );
    }
 
    output_packets.clear();
    }
}

receive_complete()

void receive_complete	(	void *	context,
		NTSTATUS	status,
		unsigned long	count
	)

Definition at line 60 of file iptest.cpp.

                                                                             {
    printf( "Receive: status %s (bytes %d)\n", status, count );
    if( !status && count ) {
    for( int off = 0; off < count; off += 16 ) {
        display_row( (char *)context, off, count );
    }
    printf( "\n" );
    }
}

SendPacket()

VOID SendPacket	(	PVOID	Context,
		PNDIS_PACKET	NdisPacket,
		UINT	Offset,
		PVOID	LinkAddress,
		USHORT	Type
	)

Definition at line 247 of file iptest.cpp.

                       {
    PCHAR DataOut;
    PUCHAR Addr = (PUCHAR)LinkAddress;
    UINT Size;
    std::string output_packet;
 
    printf( "Sending packet: %02x:%02x:%02x:%02x:%02x:%02x\n",
        Addr[0], Addr[1], Addr[2], Addr[3], Addr[4], Addr[5] );
 
    GetDataPtr( NdisPacket, Offset, &DataOut, &Size );
    for( int off = 0; off < Size; off += 16 ) {
    display_row( DataOut, off, Size );
    }
    printf( "\n" );
 
    output_packet += std::string( hwaddr, sizeof(hwaddr) );
    output_packet += std::string( (char *)LinkAddress, sizeof(hwaddr) );
    output_packet += (char)(Type >> 8);
    output_packet += (char)Type;
    output_packet += std::string( DataOut + Offset, Size - Offset );
 
    output_packets.push_back( output_packet );
}

Referenced by KdpSendWaitContinue(), and main().

TaLengthOfTransportAddress()

UINT TaLengthOfTransportAddress

(

PTRANSPORT_ADDRESS

Addr

)

Definition at line 156 of file iptest.cpp.

                                                           {
    UINT AddrLen = 2 * sizeof( ULONG ) + Addr->Address[0].AddressLength;
    printf("AddrLen %x\n", AddrLen);
    return AddrLen;
}

TdiAddressSizeFromType()

UINT TdiAddressSizeFromType

(

UINT

AddressType

)

Definition at line 79 of file iptest.cpp.

                                                {
    switch( AddressType ) {
    case TDI_ADDRESS_TYPE_IP:
    return sizeof(TA_IP_ADDRESS);
    default:
    KeBugCheck( 0 );
    }
    return 0;
}

Referenced by TdiBuildNullConnectionInfo(), and TdiBuildNullConnectionInfoInPlace().

TdiBuildConnectionInfo()

NTSTATUS TdiBuildConnectionInfo	(	PTDI_CONNECTION_INFORMATION *	ConnectionInfo,
		PTA_ADDRESS	Address
	)

Definition at line 176 of file iptest.cpp.

                        {
  NTSTATUS Status = TdiBuildNullConnectionInfo( ConnectionInfo,
                        Address->AddressType );
 
  if( NT_SUCCESS(Status) )
      TdiBuildConnectionInfoInPlace( *ConnectionInfo, Address );
 
  return Status;
}

Referenced by TCPSocketObject::TCPSocketObject().

TdiBuildConnectionInfoInPlace()

NTSTATUS TdiBuildConnectionInfoInPlace	(	PTDI_CONNECTION_INFORMATION	ConnectionInfo,
		PTA_ADDRESS	Address
	)

Definition at line 163 of file iptest.cpp.

                        {
    NTSTATUS Status = STATUS_SUCCESS;
 
    RtlCopyMemory( ConnectionInfo->RemoteAddress,
           Address,
           ConnectionInfo->RemoteAddressLength );
 
    return Status;
}

Referenced by TdiBuildConnectionInfo().

TdiBuildNullConnectionInfo()

NTSTATUS TdiBuildNullConnectionInfo	(	PTDI_CONNECTION_INFORMATION *	ConnectionInfo,
		ULONG	Type
	)

Definition at line 116 of file iptest.cpp.

{
  PTDI_CONNECTION_INFORMATION ConnInfo;
  ULONG TdiAddressSize;
  NTSTATUS Status;
 
  TdiAddressSize = TdiAddressSizeFromType(Type);
 
  ConnInfo = (PTDI_CONNECTION_INFORMATION)
    ExAllocatePool(NonPagedPool,
           sizeof(TDI_CONNECTION_INFORMATION) +
           TdiAddressSize);
  if (!ConnInfo)
    return STATUS_INSUFFICIENT_RESOURCES;
 
  Status = TdiBuildNullConnectionInfoInPlace( ConnInfo, Type );
 
  if (!NT_SUCCESS(Status))
      ExFreePool( ConnInfo );
  else
      *ConnectionInfo = ConnInfo;
 
  ConnInfo->RemoteAddress = (PTA_ADDRESS)&ConnInfo[1];
  ConnInfo->RemoteAddressLength = TdiAddressSize;
 
  return Status;
}

Referenced by TdiBuildConnectionInfo().

TdiBuildNullConnectionInfoInPlace()

NTSTATUS TdiBuildNullConnectionInfoInPlace	(	PTDI_CONNECTION_INFORMATION	ConnInfo,
		ULONG	Type
	)

Definition at line 89 of file iptest.cpp.

{
  ULONG TdiAddressSize;
 
  TdiAddressSize = TdiAddressSizeFromType(Type);
 
  RtlZeroMemory(ConnInfo,
        sizeof(TDI_CONNECTION_INFORMATION) +
        TdiAddressSize);
 
  ConnInfo->OptionsLength = sizeof(ULONG);
  ConnInfo->RemoteAddressLength = 0;
  ConnInfo->RemoteAddress = NULL;
 
  return STATUS_SUCCESS;
}

Referenced by TdiBuildNullConnectionInfo().

Variable Documentation

DebugTraceLevel

DWORD DebugTraceLevel = 0x7fffffff

Definition at line 26 of file iptest.cpp.

GlobalBufferPool

PVOID GlobalBufferPool

Definition at line 27 of file iptest.cpp.

GlobalPacketPool

PVOID GlobalPacketPool

Definition at line 27 of file iptest.cpp.

hdr

char hdr[14]

Initial value:

= { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
         0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
         0x08, 0x00 }

Definition at line 33 of file iptest.cpp.

Referenced by addref_object(), AdvProcDetailsDlgProc(), AdvProcDriverDlgProc(), AdvPropGeneralDlgProc(), alloc_async_task(), alloc_handle(), APPINFO_Destroy(), APPINFO_QueryOption(), APPINFO_SetOption(), async_exec_query(), async_read_file_proc(), AsyncFtpCreateDirectoryProc(), AsyncFtpDeleteFileProc(), AsyncFtpFindFirstFileProc(), AsyncFtpGetCurrentDirectoryProc(), AsyncFtpGetFileProc(), AsyncFtpOpenFileProc(), AsyncFtpPutFileProc(), AsyncFtpRemoveDirectoryProc(), AsyncFtpRenameFileProc(), AsyncFtpSetCurrentDirectoryProc(), AsyncHttpEndRequestProc(), AsyncHttpSendRequestProc(), AsyncInternetOpenUrlProc(), cert_mgr_advanced_dlg_proc(), cert_mgr_dlg_proc(), cert_properties_general_dlg_proc(), chmc_pmgl_add(), codeview_process_info(), COMBOEX_ComboWndProc(), COMBOEX_NotifyItem(), ComboExTestOnNotify(), connect_destroy(), connect_query_option(), create_bitmap_file(), create_dib(), create_storage_from_def(), CRYPT_AsnDecodeRsaPrivKey(), CRYPT_AsnDecodeRsaPubKey(), CRYPT_AsnEncodeRsaPubKey(), CRYPT_findPropID(), CRYPT_ReadContextProp(), CRYPT_ReadSerializedElement(), CRYPT_SerializeStoreElement(), CryptSIPRetrieveSubjectGuid(), emfdc_create_palette(), EMFDRV_CreatePalette(), EventDetailsCtrl(), export_file_dlg_proc(), export_finish_dlg_proc(), export_format_dlg_proc(), export_password_dlg_proc(), export_private_key_dlg_proc(), export_welcome_dlg_proc(), find_smbios_entry(), findPropID(), free_handle(), FTP_SendCommand(), FTP_SendCommandA(), FTPFILE_Destroy(), FTPFILE_LockRequestFile(), FTPFILE_QueryDataAvailable(), FTPFILE_QueryOption(), FTPFILE_ReadFile(), FTPFILE_WriteFile(), FTPFINDNEXT_AsyncFindNextFileProc(), FTPFINDNEXT_Destroy(), FTPFINDNEXT_FindNextFileW(), FTPFINDNEXT_QueryOption(), FTPSESSION_CloseConnection(), FTPSESSION_Destroy(), FTPSESSION_QueryOption(), get_baseboard_string(), get_bios_string(), get_compsysproduct_string(), get_compsysproduct_uuid(), get_systemenclosure_chassistypes(), get_systemenclosure_lockpresent(), get_systemenclosure_string(), getICMPStats(), grab_object(), GroupGeneralPageProc(), HeaderTestWndProc(), hierarchy_dlg_proc(), HTTP_ProcessHeader(), HTTPREQ_CloseConnection(), HTTPREQ_Destroy(), HTTPREQ_LockRequestFile(), HTTPREQ_QueryDataAvailable(), HTTPREQ_QueryOption(), HTTPREQ_ReadFile(), HTTPREQ_SetOption(), HTTPREQ_WriteFile(), HTTPSESSION_Destroy(), HTTPSESSION_QueryOption(), HTTPSESSION_SetOption(), I_RpcReceive(), I_RpcSend(), iewnd_OnNotify(), ImageAddCertificate(), ImageEnumerateCertificates(), IMAGEHLP_GetSectionOffset(), IMAGEHLP_ReportCodeSections(), IMAGEHLP_ReportImportSection(), import_file_dlg_proc(), import_finish_dlg_proc(), import_store_dlg_proc(), import_welcome_dlg_proc(), INET_QueryOption(), INTERNET_SendCallback(), InternetFindNextFileW(), InternetLockRequestFile(), InternetQueryDataAvailable(), InternetQueryOptionA(), InternetQueryOptionW(), InternetReadFile(), InternetReadFileExA(), InternetReadFileExW(), IPersistStream_fnLoad(), IPersistStream_fnSave(), LISTVIEW_MouseHover(), LISTVIEW_MouseMove(), main(), match_profile(), modMessage(), module_find_cb(), MONTHCAL_Notify(), MsiSIPGetSignedDataMsg(), nav_page_proc(), OLEPictureImpl_LoadEnhMetafile(), CLayerUIPropPage::OnClickNotify(), CSysPagerWnd::OnCustomDraw(), CSysPagerWnd::OnGetInfoTip(), CTrayNotifyWnd::OnRealign(), CNotifyToolbar::OnTooltipShow(), CMenuToolbarBase::OnWinEvent(), CAddressEditBox::OnWinEvent(), PAGER_Notify(), PAGER_SendConvertedNotify(), parent_wnd_notify(), parent_wnd_proc(), pe_load_dbg_file(), process_bind_packet(), process_bind_packet_no_send(), process_header(), process_request_packet(), PropertyStorage_MakeHeader(), PropertyStorage_MakeSectionHdr(), PropertyStorage_ReadFromStream(), PropertyStorage_ReadHeaderFromStream(), PropertyStorage_ReadSectionHeaderFromStream(), PropertyStorage_WriteHeadersToStream(), query_option(), release_object(), request_destroy(), request_query_option(), request_set_option(), RpcAssoc_BindConnection(), rpcrt4_conn_secure_packet(), RPCRT4_default_secure_packet(), rpcrt4_http_prepare_in_pipe(), rpcrt4_http_prepare_out_pipe(), rpcrt4_http_read_http_packet(), RPCRT4_io_thread(), RPCRT4_IsValidHttpPacket(), rpcrt4_ncacn_http_receive_fragment(), RPCRT4_process_packet(), RPCRT4_ValidateCommonHeader(), select_cert_dlg_proc(), send_callback(), send_notify(), session_destroy(), session_query_option(), session_set_option(), set_option(), show_cert_dialog(), show_edit_cert_properties_dialog(), show_export_ui(), show_import_ui(), START_TEST(), test_convert(), test_data_cache_save(), test_decodeRsaPrivateKey(), test_decodeRsaPublicKey(), test_encodeRsaPublicKey(), test_fragmentsize(), test_header_proc(), test_ip_pktinfo(), test_notify_parent_proc(), test_notify_proc(), test_prepareheader(), testAddSerialized(), testRegStore(), TOOLBAR_Cust_AddButton(), TOOLBAR_Cust_MoveButton(), TOOLBAR_Cust_RemoveButton(), TOOLBAR_LButtonUp(), TOOLBAR_Restore(), TOOLTIPS_Hide(), TOOLTIPS_Show(), TOOLTIPS_TrackHide(), TREEVIEW_SendRealNotify(), TREEVIEW_SendSimpleNotify(), UserGeneralPageProc(), UserMembershipPageProc(), UserProfilePageProc(), WinHttpCloseHandle(), WinHttpGetDefaultProxyConfiguration(), WinHttpGetIEProxyConfigForCurrentUser(), WinHttpQueryOption(), WinHttpSetDefaultProxyConfiguration(), WinHttpSetOption(), WinHttpSetStatusCallback(), WinHttpSetTimeouts(), and WndProc().

hwaddr

char hwaddr[6] = { 0x08, 0x00, 0x20, 0x0b, 0xb7, 0xbb }

Definition at line 31 of file iptest.cpp.

Referenced by main(), and SendPacket().

output_packets

std::list<std::string> output_packets

Definition at line 25 of file iptest.cpp.

Referenced by main(), and SendPacket().