7 Apr
2007
7 Apr
'07
3:09 p.m.
Author: gedmurphy
Date: Sat Apr 7 19:09:57 2007
New Revision: 26277
URL: http://svn.reactos.org/svn/reactos?rev=26277&view=rev
Log:
start to redo tracert
Modified:
trunk/reactos/base/applications/network/tracert/tracert.c
trunk/reactos/base/applications/network/tracert/tracert.h
Modified: trunk/reactos/base/applications/network/tracert/tracert.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/base/applications/network/…
==============================================================================
--- trunk/reactos/base/applications/network/tracert/tracert.c (original)
+++ trunk/reactos/base/applications/network/tracert/tracert.c Sat Apr 7 19:09:57 2007
@@ -1,690 +1,36 @@
-/*
- * ReactOS Win32 Applications
- * Copyright (C) 2005 ReactOS Team
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
- */
-/*
- * COPYRIGHT: See COPYING in the top level directory
- * PROJECT: ReactOS traceroute utility
- * FILE: apps/utils/net/tracert/tracert.c
- * PURPOSE: trace a packets route through a network
- * PROGRAMMERS: Ged Murphy (gedmurphy(a)gmail.com)
- * REVISIONS:
- * GM 03/05/05 Created
+ /*
+ * PROJECT: ReactOS trace route utility
+ * LICENSE: GPL - See COPYING in the top level directory
+ * FILE: base/applications/network/tracert.c
+ * PURPOSE: Trace network paths through networks
+ * COPYRIGHT: Copyright 2006 - 2007 Ged Murphy <gedmurphy(a)reactos.org>
*
*/
-
-#include <winsock2.h>
-#include <tchar.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <ws2tcpip.h>
-#include <string.h>
-#include <time.h>
#include "tracert.h"
-#define WIN32_LEAN_AND_MEAN
-
-#ifdef DBG
-#undef DBG
-#endif
-
-/*
- * globals
- */
-SOCKET icmpSock; // socket descriptor
-SOCKADDR_IN source, dest; // source and destination address info
-ECHO_REPLY_HEADER sendpacket; // ICMP echo packet
-IPv4_HEADER recvpacket; // return reveive packet
-
-BOOL bUsePerformanceCounter; // whether to use the high res performance counter
-LARGE_INTEGER TicksPerMs; // number of millisecs in relation to proc freq
-LARGE_INTEGER TicksPerUs; // number of microsecs in relation to proc freq
-LONGLONG lTimeStart; // send packet, timer start
-LONGLONG lTimeEnd; // receive packet, timer end
+//#define TRACERT_DBG
CHAR cHostname[256]; // target hostname
CHAR cDestIP[18]; // target IP
-/*
- * command line options
- */
-BOOL bResolveAddresses = TRUE; // -d MS ping defaults to true.
-INT iMaxHops = 30; // -h Max number of hops before trace ends
-INT iHostList; // -j @UNIMPLEMENTED@
-INT iTimeOut = 2000; // -w time before packet times out
-
-
-/*
- *
- * Parse command line parameters and set any options
- *
- */
-static BOOL ParseCmdline(int argc, char* argv[])
-{
- int i;
-
- if (argc < 2)
- {
- Usage();
- return FALSE;
- }
-
- for (i = 1; i < argc; i++)
- {
- if (argv[i][0] == '-')
- {
- switch (argv[i][1])
- {
- case 'd': bResolveAddresses = FALSE;
- break;
- case 'h': sscanf(argv[i+1], "%d", &iMaxHops);
- break;
- case 'j': break; /* @unimplemented@ */
- case 'w': sscanf(argv[i+1], "%d", &iTimeOut);
- break;
- default:
- _tprintf(_T("%s is not a valid option.\n"), argv[i]);
- Usage();
- return FALSE;
- }
- }
- else
- /* copy target address */
- strncpy(cHostname, argv[i], 255);
- }
-
- return TRUE;
-}
-
-
-
-/*
- *
- * Driver function, controls the traceroute program
- *
- */
-static INT Driver(VOID)
-{
-
- INT iHopCount = 1; // hop counter. default max is 30
- USHORT iSeqNum = 0; // initialise packet sequence number
- INT iTTL = 1; // set initial packet TTL to 1
- BOOL bFoundTarget = FALSE; // Have we reached our destination yet
- BOOL bAwaitPacket; // indicates whether we have recieved a good packet
- INT iDecRes; // DecodeResponse return value
- INT iRecieveReturn; // RecieveReturn return value
- INT iNameInfoRet; // getnameinfo return value
- INT iPacketSize = PACKET_SIZE; // packet size
- WORD wHeaderLen; // header length
- PECHO_REPLY_HEADER icmphdr;
-
-
- //temps for getting host name
- CHAR cHost[256];
- CHAR cServ[256];
- CHAR *ip;
-
- /* setup winsock */
- WSADATA wsaData;
-
- /* check for winsock 2 */
- if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
- {
-#ifdef DBG
- _tprintf(_T("WSAStartup failed.\n"));
-#endif /* DBG */
- exit(1);
- }
-
- /* establish what timing method we can use */
- SetupTimingMethod();
-
- /* setup target info */
- ResolveHostname();
-
- /* print standard tracing info to screen */
- _tprintf(_T("\nTracing route to %s [%s]\n"), cHostname, cDestIP);
- _tprintf(_T("over a maximum of %d hop"), iMaxHops);
- iMaxHops > 1 ? _tprintf(_T("s:\n\n")) :
_tprintf(_T(":\n\n"));
-
- /* run until we hit either max hops, or we recieve 3 echo replys */
- while ((iHopCount <= iMaxHops) && (bFoundTarget != TRUE))
- {
- INT i;
-
- _tprintf(_T("%3d "), iHopCount);
- /* run 3 pings for each hop */
- for (i=0; i<3; i++)
- {
- if (Setup(iTTL) != TRUE)
- {
-#ifdef DBG
- _tprintf(_T("error in Setup()\n"));
-#endif /* DBG */
- WSACleanup();
- exit(1);
- }
- PreparePacket(iPacketSize, iSeqNum);
- if (SendPacket(iPacketSize) != SOCKET_ERROR)
- {
- /* loop until we get a good packet */
- bAwaitPacket = TRUE;
- while (bAwaitPacket)
- {
- /* Receive replies until we either get a successful
- * read, or a fatal error occurs. */
- if ((iRecieveReturn = ReceivePacket(iPacketSize)) < 0)
- {
- /* check the sequence number in the packet
- * if it's bad, complain and wait for another packet
- * , otherwise break */
- wHeaderLen = recvpacket.h_len * 4;
- icmphdr = (ECHO_REPLY_HEADER *)((char*)&recvpacket +
wHeaderLen);
- if (icmphdr->icmpheader.seq != iSeqNum)
- {
- _tprintf(_T("bad sequence number!\n"));
- continue;
- }
- else
- break;
- }
-
- /* if RecievePacket timed out we don't bother decoding */
- if (iRecieveReturn != 1)
- {
- iDecRes = DecodeResponse(iPacketSize);
-
- switch (iDecRes)
- {
- case 0 : bAwaitPacket = FALSE; /* time exceeded */
- break;
- case 1 : bAwaitPacket = FALSE; /* echo reply */
- break;
- case 2 : bAwaitPacket = FALSE; /* destination unreachable */
- break;
-#ifdef DBG
- case -1 :
- _tprintf(_T("recieved foreign
packet\n"));
- break;
- case -2 :
- _tprintf(_T("error in
DecodeResponse\n"));
- break;
- case -3 :
- _tprintf(_T("unknown ICMP packet\n"));
- break;
-#endif /* DBG */
- default : break;
- }
- }
- else
- /* packet timed out. Don't wait for it again */
- bAwaitPacket = FALSE;
- }
- }
-
- iSeqNum++;
- _tprintf(_T(" "));
- }
-
- if(bResolveAddresses)
- {
- /* gethostbyaddr() and getnameinfo() are
- * unimplemented in ROS at present.
- * Alex has advised he will be implementing getnameinfo.
- * I've used that for the time being for testing in Windows*/
-
- //ip = inet_addr(inet_ntoa(source.sin_addr));
- //host = gethostbyaddr((char *)&ip, 4, 0);
-
- ip = inet_ntoa(source.sin_addr);
-
- iNameInfoRet = getnameinfo((SOCKADDR *)&source,
- sizeof(SOCKADDR),
- cHost,
- 256,
- cServ,
- 256,
- NI_NUMERICSERV);
- if (iNameInfoRet == 0)
- {
- /* if IP address resolved to a hostname,
- * print the IP address after it */
- if (lstrcmpA(cHost, ip) != 0)
- _tprintf(_T("%s [%s]"), cHost, ip);
- else
- _tprintf(_T("%s"), cHost);
- }
- else
- {
- _tprintf(_T("error: %d"), WSAGetLastError());
-#ifdef DBG
- _tprintf(_T(" getnameinfo failed: %d"), iNameInfoRet);
-#endif /* DBG */
- }
-
- }
- else
- _tprintf(_T("%s"), inet_ntoa(source.sin_addr));
-
- _tprintf(_T("\n"));
-
- /* check if we've arrived at the target */
- if (strcmp(cDestIP, inet_ntoa(source.sin_addr)) == 0)
- bFoundTarget = TRUE;
- else
- {
- iTTL++;
- iHopCount++;
- Sleep(500);
- }
- }
- _tprintf(_T("\nTrace complete.\n"));
- WSACleanup();
-
- return 0;
-}
-
-
-/*
- * Establish if performance counters are available and
- * set up timing figures in relation to processor frequency.
- * If performance counters are not available, we'll be using
- * gettickcount, so set the figures to 1
- *
- */
-static VOID SetupTimingMethod(VOID)
-{
- LARGE_INTEGER PerformanceCounterFrequency;
-
- /* check if performance counters are available */
- bUsePerformanceCounter =
QueryPerformanceFrequency(&PerformanceCounterFrequency);
- if (bUsePerformanceCounter)
- {
- /* restrict execution to first processor on SMP systems */
- if (SetThreadAffinityMask(GetCurrentThread(), 1) == 0)
- bUsePerformanceCounter = FALSE;
-
- TicksPerMs.QuadPart = PerformanceCounterFrequency.QuadPart / 1000;
- TicksPerUs.QuadPart = PerformanceCounterFrequency.QuadPart / 1000000;
- }
-
- if (!bUsePerformanceCounter)
- {
- TicksPerMs.QuadPart = 1;
- TicksPerUs.QuadPart = 1;
- }
-}
-
-
-/*
- *
- * Check for a hostname or dotted deciamal for our target.
- * If we have a hostname, resolve to an IP and store it, else
- * just store the target IP address. Also set up other key
- * SOCKADDR_IN members needed for the connection.
- *
- */
-static VOID ResolveHostname(VOID)
-{
- HOSTENT *hp;
- ULONG addr;
-
- memset(&dest, 0, sizeof(dest));
-
- addr = inet_addr(cHostname);
- /* if address is not a dotted decimal */
- if (addr == INADDR_NONE)
- {
- hp = gethostbyname(cHostname);
- if (hp != 0)
- {
- memcpy(&dest.sin_addr, hp->h_addr, hp->h_length);
- //dest.sin_addr = *((struct in_addr *)hp->h_addr);
- dest.sin_family = hp->h_addrtype;
- }
- else
- {
- _tprintf(_T("Unable to resolve target system name %s.\n"),
cHostname);
- WSACleanup();
- exit(1);
- }
- }
- else
- {
- dest.sin_addr.s_addr = addr;
- dest.sin_family = AF_INET;
- }
- /* copy destination IP address into a string */
- strcpy(cDestIP, inet_ntoa(dest.sin_addr));
-}
-
-
-
-/*
- *
- * Create our socket which will be used for sending and recieving,
- * Socket Type is raw, Protocol is ICMP. Also set the TTL value which will be
- * set in the outgoing IP packet.
- *
- */
-static INT Setup(INT iTTL)
-{
- INT iSockRet;
-
- /* create raw socket */
- icmpSock = WSASocket(AF_INET, SOCK_RAW, IPPROTO_ICMP, 0, 0, 0);
- if (icmpSock == INVALID_SOCKET)
- {
- _tprintf(_T("Could not create socket : %d.\n"), WSAGetLastError());
- if (WSAGetLastError() == WSAEACCES)
- {
- _tprintf(_T("\n\nYou must be an administrator to run this
program!\n\n"));
- WSACleanup();
- exit(1);
- }
- return FALSE;
- }
-
- /* setup for TTL */
- iSockRet = setsockopt(icmpSock, IPPROTO_IP, IP_TTL, (const char *)&iTTL,
sizeof(iTTL));
- if (iSockRet == SOCKET_ERROR)
- {
- _tprintf(_T("TTL setsockopt failed : %d. \n"), WSAGetLastError());
- return FALSE;
- }
-
- return TRUE;
-}
-
-
-
-/*
- * Prepare the ICMP echo request packet for sending.
- * Calculate the packet checksum
- *
- */
-static VOID PreparePacket(INT iPacketSize, USHORT iSeqNum)
-{
- /* assemble ICMP echo request packet */
- sendpacket.icmpheader.type = ECHO_REQUEST;
- sendpacket.icmpheader.code = 0;
- sendpacket.icmpheader.checksum = 0;
- sendpacket.icmpheader.id = (USHORT)GetCurrentProcessId();
- sendpacket.icmpheader.seq = iSeqNum;
-
- /* calculate checksum of packet */
- sendpacket.icmpheader.checksum = CheckSum((PUSHORT)&sendpacket,
sizeof(ICMP_HEADER) + iPacketSize);
-}
-
-
-
-/*
- *
- * Get the system time and send the ICMP packet to the destination
- * address.
- *
- */
-static INT SendPacket(INT datasize)
-{
- INT iSockRet;
- INT iPacketSize;
-
- iPacketSize = sizeof(ECHO_REPLY_HEADER) + datasize;
-
-#ifdef DBG
- _tprintf(_T("\nsending packet of %d bytes\n"), iPacketSize);
-#endif /* DBG */
-
- /* get time packet was sent */
- lTimeStart = GetTime();
-
- iSockRet = sendto(icmpSock, //socket
- (char *)&sendpacket, //buffer
- iPacketSize, //size of buffer
- 0, //flags
- (SOCKADDR *)&dest, //destination
- sizeof(dest)); //address length
-
- if (iSockRet == SOCKET_ERROR)
- {
- if (WSAGetLastError() == WSAEACCES)
- {
- _tprintf(_T("\n\nYou must be an administrator to run this
program!\n\n"));
- WSACleanup();
- exit(1);
- }
- else
- {
-#ifdef DBG
- _tprintf(_T("sendto failed %d\n"), WSAGetLastError());
-#endif /* DBG */
- return FALSE;
- }
- }
-#ifdef DBG
- _tprintf(_T("sent %d bytes\n"), iSockRet);
-#endif /* DBG */
-
- /* return number of bytes sent */
- return iSockRet;
-}
-
-
-
-/*
- *
- * Set up a timeout value and put the socket in a select poll.
- * Wait until we recieve an IPv4 reply packet in reply to the ICMP
- * echo request packet and get the time the packet was recieved.
- * If we don't recieve a packet, do some checking to establish why.
- *
- */
-static INT ReceivePacket(INT datasize)
-{
- TIMEVAL timeVal;
- FD_SET readFDS;
- int iSockRet = 0, iSelRet;
- int iFromLen;
- int iPacketSize;
-
- /* allow for a larger recv buffer to store ICMP TTL
- * exceed, IP header and orginal ICMP request */
- iPacketSize = MAX_REC_SIZE + datasize;
-
- iFromLen = sizeof(source);
-
-#ifdef DBG
- _tprintf(_T("receiving packet. Available buffer, %d bytes\n"),
iPacketSize);
-#endif /* DBG */
-
- /* monitor icmpSock for incomming connections */
- FD_ZERO(&readFDS);
- FD_SET(icmpSock, &readFDS);
-
- /* set timeout values */
- timeVal.tv_sec = iTimeOut / 1000;
- timeVal.tv_usec = iTimeOut % 1000;
-
- iSelRet = select(0, &readFDS, NULL, NULL, &timeVal);
-
- if ((iSelRet != SOCKET_ERROR) && (iSelRet != 0))
- {
- iSockRet = recvfrom(icmpSock, //socket
- (char *)&recvpacket, //buffer
- iPacketSize, //size of buffer
- 0, //flags
- (SOCKADDR *)&source, //source address
- &iFromLen); //pointer to address length
- /* get time packet was recieved */
- lTimeEnd = GetTime();
- /* if socket timed out */
- }
- else if (iSelRet == 0)
- {
- _tprintf(_T(" * "));
- return 1;
- }
- else if (iSelRet == SOCKET_ERROR)
- {
- _tprintf(_T("select() failed in sendPacket() %d\n"),
WSAGetLastError());
- return -1;
- }
-
-
- if (iSockRet == SOCKET_ERROR)
- {
- _tprintf(_T("recvfrom failed: %d\n"), WSAGetLastError());
- return -2;
- }
-#ifdef DBG
- else
- _tprintf(_T("reveived %d bytes\n"), iSockRet);
-#endif /* DBG */
-
- return 0;
-}
-
-
-
-/*
- *
- * Cast the IPv4 packet to an echo reply and to a TTL exceed.
- * Check the 'type' field to establish what was recieved, and
- * ensure the packet is related to the originating process.
- * It all is well, print the time taken for the round trip.
- *
- */
-static INT DecodeResponse(INT iPacketSize)
-{
- unsigned short header_len = recvpacket.h_len * 4;
- /* cast the recieved packet into an ECHO reply and a TTL Exceed so we can check the
ID*/
- ECHO_REPLY_HEADER *IcmpHdr = (ECHO_REPLY_HEADER *)((char*)&recvpacket +
header_len);
- TTL_EXCEED_HEADER *TTLExceedHdr = (TTL_EXCEED_HEADER *)((char *)&recvpacket +
header_len);
-
- /* Make sure the reply is ok */
- if (iPacketSize < header_len + ICMP_MIN_SIZE)
- {
- _tprintf(_T("too few bytes from %s\n"), inet_ntoa(dest.sin_addr));
- return -2;
- }
-
- switch (IcmpHdr->icmpheader.type)
- {
- case TTL_EXCEEDED :
- if (TTLExceedHdr->OrigIcmpHeader.id != (USHORT)GetCurrentProcessId())
- {
- /* FIXME */
- /* we've picked up a packet not related to this process
- * probably from another local program. We ignore it */
-#ifdef DGB
- _tprintf(_T("header id, process id %d"),
TTLExceedHdr->OrigIcmpHeader.id, GetCurrentProcessId());
-#endif /* DBG */
- //_tprintf(_T("oops ");
- return -1;
- }
- _tprintf(_T("%3Ld ms"), (lTimeEnd - lTimeStart) /
TicksPerMs.QuadPart);
- return 0;
- case ECHO_REPLY :
- if (IcmpHdr->icmpheader.id != (USHORT)GetCurrentProcessId())
- {
- /* FIXME */
- /* we've picked up a packet not related to this process
- * probably from another local program. We ignore it */
-#ifdef DGB
- _tprintf(_T("\nPicked up wrong packet. icmpheader.id = %d and
process id = %d"), IcmpHdr->icmpheader.id, GetCurrentProcessId());
-#endif /* DBG */
- //_tprintf(_T("oops ");
- return -1;
- }
- _tprintf(_T("%3Ld ms"), (lTimeEnd - lTimeStart) /
TicksPerMs.QuadPart);
- return 1;
- case DEST_UNREACHABLE :
- _tprintf(_T(" * "));
- return 2;
- default :
- /* unknown ICMP packet */
- return -3;
- }
-}
-
-
-/*
- *
- * Get the system time using preformance counters if available,
- * otherwise fall back to GetTickCount()
- *
- */
-
-static LONGLONG GetTime(VOID)
-{
- LARGE_INTEGER Time;
-
- if (bUsePerformanceCounter)
- {
- if (QueryPerformanceCounter(&Time) == 0)
- {
- Time.u.LowPart = (DWORD)GetTickCount();
- Time.u.HighPart = 0;
- return (LONGLONG)Time.u.LowPart;
- }
- }
- else
- {
- Time.u.LowPart = (DWORD)GetTickCount();
- Time.u.HighPart = 0;
- return (LONGLONG)Time.u.LowPart;
- }
- return Time.QuadPart;
-}
-
-
-/*
- *
- * Calculate packet checksum.
- *
- */
-static WORD CheckSum(PUSHORT data, UINT size)
-{
- DWORD dwSum = 0;
-
- while (size > 1)
- {
- dwSum += *data++;
- size -= sizeof(USHORT);
- }
-
- if (size)
- dwSum += *(UCHAR*)data;
-
- dwSum = (dwSum >> 16) + (dwSum & 0xFFFF);
- dwSum += (dwSum >> 16);
-
- return (USHORT)(~dwSum);
-}
-
-
-/*
- *
- * print program usage to screen
- *
- */
-static VOID Usage(VOID)
+static VOID
+DebugPrint(LPTSTR lpString, ...)
+{
+#ifdef TRACERT_DBG
+ va_list args;
+ va_start(args, lpString);
+ _vtprintf(lpString, args);
+ va_end(args);
+#else
+ UNREFERENCED_PARAMETER(lpString);
+#endif
+}
+
+
+static VOID
+Usage(VOID)
{
_tprintf(_T("\nUsage: tracert [-d] [-h maximum_hops] [-j host-list] [-w timeout]
target_name\n\n"
"Options:\n"
@@ -693,7 +39,6 @@
" -j host-list Loose source route along host-list.\n"
" -w timeout Wait timeout milliseconds for each
reply.\n\n"));
- /* temp notes to stop user questions until getnameinfo/gethostbyaddr and getsockopt
are implemented */
_tprintf(_T("NOTES\n-----\n"
"- Setting TTL values is not currently supported in ReactOS, so the trace
will\n"
" jump straight to the destination. This feature will be implemented
soon.\n"
@@ -703,17 +48,637 @@
}
-
-/*
- *
- * Program entry point
- *
- */
-int main(int argc, char* argv[])
-{
- if (!ParseCmdline(argc, argv)) return -1;
-
- Driver();
+static BOOL
+ParseCmdline(int argc,
+ LPCTSTR argv[],
+ PAPPINFO pInfo)
+{
+ INT i;
+
+ if (argc < 2)
+ {
+ Usage();
+ return FALSE;
+ }
+ else
+ {
+ for (i = 1; i < argc; i++)
+ {
+ if (argv[i][0] == _T('-'))
+ {
+ switch (argv[i][1])
+ {
+ case _T('d'):
+ pInfo->bResolveAddresses = FALSE;
+ break;
+
+ case _T('h'):
+ _stscanf(argv[i+1], _T("%d"),
&pInfo->iMaxHops);
+ break;
+
+ case _T('j'):
+ _tprintf(_T("-j is not yet implemented.\n"));
+ break;
+
+ case _T('w'):
+ _stscanf(argv[i+1], _T("%d"),
&pInfo->iTimeOut);
+ break;
+
+ default:
+ {
+ _tprintf(_T("%s is not a valid option.\n"), argv[i]);
+ Usage();
+ return FALSE;
+ }
+ }
+ }
+ else
+ /* copy target address */
+ _tcsncpy(cHostname, argv[i], 255);
+ }
+ }
+
+ return TRUE;
+}
+
+
+static WORD
+CheckSum(PUSHORT data,
+ UINT size)
+{
+ DWORD dwSum = 0;
+
+ while (size > 1)
+ {
+ dwSum += *data++;
+ size -= sizeof(USHORT);
+ }
+
+ if (size)
+ dwSum += *(UCHAR*)data;
+
+ dwSum = (dwSum >> 16) + (dwSum & 0xFFFF);
+ dwSum += (dwSum >> 16);
+
+ return (USHORT)(~dwSum);
+}
+
+
+static VOID
+SetupTimingMethod(PAPPINFO pInfo)
+{
+ LARGE_INTEGER PerformanceCounterFrequency;
+
+ /* check if performance counters are available */
+ pInfo->bUsePerformanceCounter =
QueryPerformanceFrequency(&PerformanceCounterFrequency);
+
+ if (pInfo->bUsePerformanceCounter)
+ {
+ /* restrict execution to first processor on SMP systems */
+ if (SetThreadAffinityMask(GetCurrentThread(), 1) == 0)
+ pInfo->bUsePerformanceCounter = FALSE;
+
+ pInfo->TicksPerMs.QuadPart = PerformanceCounterFrequency.QuadPart / 1000;
+ pInfo->TicksPerUs.QuadPart = PerformanceCounterFrequency.QuadPart / 1000000;
+ }
+ else
+ {
+ pInfo->TicksPerMs.QuadPart = 1;
+ pInfo->TicksPerUs.QuadPart = 1;
+ }
+}
+
+
+static BOOL
+ResolveHostname(PAPPINFO pInfo)
+{
+ HOSTENT *hp;
+ ULONG addr;
+
+ ZeroMemory(&pInfo->dest, sizeof(pInfo->dest));
+
+ /* if address is not a dotted decimal */
+ if ((addr = inet_addr(cHostname))== INADDR_NONE)
+ {
+ if ((hp = gethostbyname(cHostname)) != 0)
+ {
+ //CopyMemory(&pInfo->dest.sin_addr, hp->h_addr, hp->h_length);
+ pInfo->dest.sin_addr = *((struct in_addr *)hp->h_addr);
+ pInfo->dest.sin_family = hp->h_addrtype;
+ }
+ else
+ {
+ _tprintf(_T("Unable to resolve target system name %s.\n"),
cHostname);
+ return FALSE;
+ }
+ }
+ else
+ {
+ pInfo->dest.sin_addr.s_addr = addr;
+ pInfo->dest.sin_family = AF_INET;
+ }
+
+ _tcscpy(cDestIP, inet_ntoa(pInfo->dest.sin_addr));
+
+ return TRUE;
+}
+
+
+static LONGLONG
+GetTime(PAPPINFO pInfo)
+{
+ LARGE_INTEGER Time;
+
+ /* Get the system time using preformance counters if available */
+ if (pInfo->bUsePerformanceCounter)
+ {
+ if (QueryPerformanceCounter(&Time))
+ {
+ return Time.QuadPart;
+ }
+ }
+
+ /* otherwise fall back to GetTickCount */
+ Time.u.LowPart = (DWORD)GetTickCount();
+ Time.u.HighPart = 0;
+
+ return (LONGLONG)Time.u.LowPart;
+}
+
+
+static BOOL
+SetTTL(SOCKET sock,
+ INT iTTL)
+{
+ if (setsockopt(sock,
+ IPPROTO_IP,
+ IP_TTL,
+ (const char *)&iTTL,
+ sizeof(iTTL)) == SOCKET_ERROR)
+ {
+ DebugPrint(_T("TTL setsockopt failed : %d. \n"), WSAGetLastError());
+ return FALSE;
+ }
+
+ return TRUE;
+}
+
+
+static BOOL
+CreateSocket(PAPPINFO pInfo)
+{
+ pInfo->icmpSock = WSASocket(AF_INET,
+ SOCK_RAW,
+ IPPROTO_ICMP,
+ 0,
+ 0,
+ 0);
+
+ if (pInfo->icmpSock == INVALID_SOCKET)
+ {
+ INT err = WSAGetLastError();
+ DebugPrint(_T("Could not create socket : %d.\n"), err);
+
+ if (err == WSAEACCES)
+ {
+ _tprintf(_T("\n\nYou must have access to raw sockets (admin) to run this
program!\n\n"));
+ }
+
+ return FALSE;
+ }
+
+ return TRUE;
+}
+
+
+static VOID
+PreparePacket(PAPPINFO pInfo,
+ USHORT iSeqNum)
+{
+ /* assemble ICMP echo request packet */
+ pInfo->SendPacket->icmpheader.type = ECHO_REQUEST;
+ pInfo->SendPacket->icmpheader.code = 0;
+ pInfo->SendPacket->icmpheader.checksum = 0;
+ pInfo->SendPacket->icmpheader.id = (USHORT)GetCurrentProcessId();
+ pInfo->SendPacket->icmpheader.seq = iSeqNum;
+
+ /* calculate checksum of packet */
+ pInfo->SendPacket->icmpheader.checksum =
CheckSum((PUSHORT)&pInfo->SendPacket,
+ sizeof(ICMP_HEADER) +
PACKET_SIZE);
+}
+
+
+static INT
+SendPacket(PAPPINFO pInfo)
+{
+ INT iSockRet;
+
+ DebugPrint(_T("\nsending packet of %d bytes... "), PACKET_SIZE);
+
+ /* get time packet was sent */
+ pInfo->lTimeStart = GetTime(pInfo);
+
+ iSockRet = sendto(pInfo->icmpSock, //socket
+ (char *)pInfo->SendPacket, //buffer
+ PACKET_SIZE, //size of buffer
+ 0, //flags
+ (SOCKADDR *)&pInfo->dest, //destination
+ sizeof(pInfo->dest)); //address length
+
+ if (iSockRet == SOCKET_ERROR)
+ {
+ if (WSAGetLastError() == WSAEACCES)
+ {
+ /* FIXME: Is this correct? */
+ _tprintf(_T("\n\nYou must be an administrator to run this
program!\n\n"));
+ WSACleanup();
+ HeapFree(GetProcessHeap(), 0, pInfo);
+ exit(-1);
+ }
+ else
+ {
+ DebugPrint(_T("sendto failed %d\n"), WSAGetLastError());
+ }
+ }
+ else
+ {
+ DebugPrint(_T("sent %d bytes\n"), iSockRet);
+ }
+
+ return iSockRet;
+}
+
+
+static BOOL
+ReceivePacket(PAPPINFO pInfo)
+{
+ TIMEVAL timeVal;
+ FD_SET readFDS;
+ INT iSockRet = 0, iSelRet;
+ INT iFromLen;
+ BOOL bRet = FALSE;
+
+ iFromLen = sizeof(pInfo->source);
+
+ DebugPrint(_T("Receiving packet. Available buffer, %d bytes... "),
MAX_PING_PACKET_SIZE);
+
+ /* monitor icmpSock for incomming connections */
+ FD_ZERO(&readFDS);
+ FD_SET(pInfo->icmpSock, &readFDS);
+
+ /* set timeout values */
+ timeVal.tv_sec = pInfo->iTimeOut / 1000;
+ timeVal.tv_usec = pInfo->iTimeOut % 1000;
+
+ iSelRet = select(0,
+ &readFDS,
+ NULL,
+ NULL,
+ &timeVal);
+
+ if (iSelRet == SOCKET_ERROR)
+ {
+ DebugPrint(_T("select() failed in sendPacket() %d\n"),
WSAGetLastError());
+ }
+ else if (iSelRet == 0) /* if socket timed out */
+ {
+ _tprintf(_T(" * "));
+ }
+ else if ((iSelRet != SOCKET_ERROR) && (iSelRet != 0))
+ {
+ iSockRet = recvfrom(pInfo->icmpSock, // socket
+ (char *)pInfo->RecvPacket, // buffer
+ MAX_PING_PACKET_SIZE, // size of buffer
+ 0, // flags
+ (SOCKADDR *)&pInfo->source, // source address
+ &iFromLen); // address length
+
+ if (iSockRet != SOCKET_ERROR)
+ {
+ /* get time packet was recieved */
+ pInfo->lTimeEnd = GetTime(pInfo);
+ DebugPrint(_T("reveived %d bytes\n"), iSockRet);
+ bRet = TRUE;
+ }
+ else
+ {
+ DebugPrint(_T("recvfrom failed: %d\n"), WSAGetLastError());
+ }
+ }
+
+ return bRet;
+}
+
+
+static INT
+DecodeResponse(PAPPINFO pInfo)
+{
+ unsigned short header_len = pInfo->RecvPacket->h_len * 4;
+
+ /* cast the recieved packet into an ECHO reply and a TTL Exceed and check the ID*/
+ ECHO_REPLY_HEADER *IcmpHdr = (ECHO_REPLY_HEADER *)((char*)pInfo->RecvPacket +
header_len);
+ TTL_EXCEED_HEADER *TTLExceedHdr = (TTL_EXCEED_HEADER *)((char *)pInfo->RecvPacket
+ header_len);
+
+ /* Make sure the reply is ok */
+ if (PACKET_SIZE < header_len + ICMP_MIN_SIZE)
+ {
+ DebugPrint(_T("too few bytes from %s\n"),
inet_ntoa(pInfo->dest.sin_addr));
+ return -2;
+ }
+
+ switch (IcmpHdr->icmpheader.type)
+ {
+ case TTL_EXCEEDED :
+ if (TTLExceedHdr->OrigIcmpHeader.id != (USHORT)GetCurrentProcessId())
+ {
+ /* FIXME: our network stack shouldn't allow this... */
+ /* we've picked up a packet not related to this process probably from
another local program. We ignore it */
+ DebugPrint(_T("Rouge packet: header id, process id %d"),
TTLExceedHdr->OrigIcmpHeader.id, GetCurrentProcessId());
+ return -1;
+ }
+ _tprintf(_T("%3Ld ms"), (pInfo->lTimeEnd -
pInfo->lTimeStart) / pInfo->TicksPerMs.QuadPart);
+ return 0;
+
+ case ECHO_REPLY :
+ if (IcmpHdr->icmpheader.id != (USHORT)GetCurrentProcessId())
+ {
+ /* FIXME: our network stack shouldn't allow this... */
+ /* we've picked up a packet not related to this process probably from
another local program. We ignore it */
+ DebugPrint(_T("Rouge packet: header id %d, process id
%d"), IcmpHdr->icmpheader.id, GetCurrentProcessId());
+ return -1;
+ }
+ _tprintf(_T("%3Ld ms"), (pInfo->lTimeEnd -
pInfo->lTimeStart) / pInfo->TicksPerMs.QuadPart);
+ return 1;
+
+ case DEST_UNREACHABLE :
+ _tprintf(_T(" * "));
+ return 2;
+ }
return 0;
}
+
+
+static BOOL
+AllocateBuffers(PAPPINFO pInfo)
+{
+ pInfo->SendPacket = (PECHO_REPLY_HEADER)HeapAlloc(GetProcessHeap(),
+ 0,
+ sizeof(ECHO_REPLY_HEADER) +
PACKET_SIZE);
+ if (!pInfo->SendPacket)
+ return FALSE;
+
+ pInfo->RecvPacket = (PIPv4_HEADER)HeapAlloc(GetProcessHeap(),
+ 0,
+ sizeof(IPv4_HEADER) + PACKET_SIZE);
+ if (!pInfo->RecvPacket)
+ {
+ HeapFree(GetProcessHeap(),
+ 0,
+ pInfo->SendPacket);
+
+ return FALSE;
+ }
+
+ return TRUE;
+}
+
+
+static INT
+Driver(PAPPINFO pInfo)
+{
+ INT iHopCount = 1; // hop counter. default max is 30
+ BOOL bFoundTarget = FALSE; // Have we reached our destination yet
+ INT iRecieveReturn; // RecieveReturn return value
+ PECHO_REPLY_HEADER icmphdr;
+ INT iTTL = 1;
+
+ INT ret = -1;
+
+ //temps for getting host name
+ CHAR cHost[256];
+ CHAR cServ[256];
+ CHAR *ip;
+
+ SetupTimingMethod(pInfo);
+
+ if (AllocateBuffers(pInfo) &&
+ ResolveHostname(pInfo) &&
+ CreateSocket(pInfo))
+ {
+ /* print tracing info to screen */
+ _tprintf(_T("\nTracing route to %s [%s]\n"), cHostname, cDestIP);
+ _tprintf(_T("over a maximum of %d hop"), pInfo->iMaxHops);
+ pInfo->iMaxHops > 1 ? _tprintf(_T("s:\n\n")) :
_tprintf(_T(":\n\n"));
+
+ /* run until we hit either max hops, or find the target */
+ while ((iHopCount <= pInfo->iMaxHops) &&
+ (bFoundTarget != TRUE))
+ {
+ USHORT iSeqNum = 0;
+ INT i;
+
+ _tprintf(_T("%3d "), iHopCount);
+
+ /* run 3 pings for each hop */
+ for (i = 0; i < 3; i++)
+ {
+ if (SetTTL(pInfo->icmpSock, iTTL) != TRUE)
+ {
+ DebugPrint(_T("error in Setup()\n"));
+ return ret;
+ }
+
+ PreparePacket(pInfo, iSeqNum);
+
+ if (SendPacket(pInfo) != SOCKET_ERROR)
+ {
+ BOOL bAwaitPacket = FALSE; // indicates whether we have recieved a
good packet
+
+ do
+ {
+ /* Receive replies until we get a successful read, or a fatal
error */
+ if ((iRecieveReturn = ReceivePacket(pInfo)) < 0)
+ {
+ /* FIXME: consider moving this into RecievePacket */
+ /* check the seq num in the packet, if it's bad wait for
another */
+ WORD hdrLen = pInfo->RecvPacket->h_len * 4;
+ icmphdr = (ECHO_REPLY_HEADER
*)((char*)&pInfo->RecvPacket + hdrLen);
+ if (icmphdr->icmpheader.seq != iSeqNum)
+ {
+ _tprintf(_T("bad sequence number!\n"));
+ continue;
+ }
+ }
+
+ if (iRecieveReturn)
+ {
+ DecodeResponse(pInfo);
+ }
+ else
+ /* packet timed out. Don't wait for it again */
+ bAwaitPacket = FALSE;
+
+ } while (bAwaitPacket);
+ }
+
+ iSeqNum++;
+ _tprintf(_T(" "));
+ }
+
+ if(pInfo->bResolveAddresses)
+ {
+ INT iNameInfoRet; // getnameinfo return value
+ /* gethostbyaddr() and getnameinfo() are
+ * unimplemented in ROS at present.
+ * Alex has advised he will be implementing getnameinfo.
+ * I've used that for the time being for testing in Windows*/
+
+ //ip = inet_addr(inet_ntoa(source.sin_addr));
+ //host = gethostbyaddr((char *)&ip, 4, 0);
+
+ ip = inet_ntoa(pInfo->source.sin_addr);
+
+ iNameInfoRet = getnameinfo((SOCKADDR *)&pInfo->source,
+ sizeof(SOCKADDR),
+ cHost,
+ 256,
+ cServ,
+ 256,
+ NI_NUMERICSERV);
+ if (iNameInfoRet == 0)
+ {
+ /* if IP address resolved to a hostname,
+ * print the IP address after it */
+ if (lstrcmpA(cHost, ip) != 0)
+ _tprintf(_T("%s [%s]"), cHost, ip);
+ else
+ _tprintf(_T("%s"), cHost);
+ }
+ else
+ {
+ DebugPrint(_T("error: %d"), WSAGetLastError());
+ DebugPrint(_T(" getnameinfo failed: %d"), iNameInfoRet);
+ }
+
+ }
+ else
+ _tprintf(_T("%s"), inet_ntoa(pInfo->source.sin_addr));
+
+ _tprintf(_T("\n"));
+
+ /* check if we've arrived at the target */
+ if (strcmp(cDestIP, inet_ntoa(pInfo->source.sin_addr)) == 0)
+ bFoundTarget = TRUE;
+ else
+ {
+ iTTL++;
+ iHopCount++;
+ Sleep(500);
+ }
+ }
+ _tprintf(_T("\nTrace complete.\n"));
+ ret = 0;
+ }
+
+ return ret;
+}
+
+
+static VOID
+Cleanup(PAPPINFO pInfo)
+{
+ if (pInfo->icmpSock)
+ closesocket(pInfo->icmpSock);
+
+ WSACleanup();
+
+ if (pInfo->SendPacket)
+ HeapFree(GetProcessHeap(),
+ 0,
+ pInfo->SendPacket);
+
+ if (pInfo->SendPacket)
+ HeapFree(GetProcessHeap(),
+ 0,
+ pInfo->RecvPacket);
+}
+
+
+#if defined(_UNICODE) && defined(__GNUC__)
+static
+#endif
+int _tmain(int argc, LPCTSTR argv[])
+{
+ PAPPINFO pInfo;
+ WSADATA wsaData;
+ int ret = -1;
+
+ pInfo = (PAPPINFO)HeapAlloc(GetProcessHeap(),
+ HEAP_ZERO_MEMORY,
+ sizeof(APPINFO));
+ if (pInfo)
+ {
+ pInfo->bResolveAddresses = TRUE;
+ pInfo->iMaxHops = 30;
+ pInfo->iTimeOut = 1000;
+
+ if (ParseCmdline(argc, argv, pInfo))
+ {
+ if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
+ {
+ DebugPrint(_T("WSAStartup failed.\n"));
+ }
+ else
+ {
+ ret = Driver(pInfo);
+ Cleanup(pInfo);
+ }
+ }
+
+ HeapFree(GetProcessHeap(),
+ 0,
+ pInfo);
+ }
+
+ return ret;
+}
+
+
+#if defined(_UNICODE) && defined(__GNUC__)
+/* HACK - MINGW HAS NO OFFICIAL SUPPORT FOR wmain()!!! */
+int main( int argc, char **argv )
+{
+ WCHAR **argvW;
+ int i, j, Ret = 1;
+
+ if ((argvW = malloc(argc * sizeof(WCHAR*))))
+ {
+ /* convert the arguments */
+ for (i = 0, j = 0; i < argc; i++)
+ {
+ if (!(argvW[i] = malloc((strlen(argv[i]) + 1) * sizeof(WCHAR))))
+ {
+ j++;
+ }
+ swprintf(argvW[i], L"%hs", argv[i]);
+ }
+
+ if (j == 0)
+ {
+ /* no error converting the parameters, call wmain() */
+ Ret = wmain(argc, (LPCTSTR *)argvW);
+ }
+
+ /* free the arguments */
+ for (i = 0; i < argc; i++)
+ {
+ if (argvW[i])
+ free(argvW[i]);
+ }
+ free(argvW);
+ }
+
+ return Ret;
+}
+#endif
Modified: trunk/reactos/base/applications/network/tracert/tracert.h
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/base/applications/network/…
==============================================================================
--- trunk/reactos/base/applications/network/tracert/tracert.h (original)
+++ trunk/reactos/base/applications/network/tracert/tracert.h Sat Apr 7 19:09:57 2007
@@ -1,24 +1,24 @@
-/*
- * COPYRIGHT: See COPYING in the top level directory
- * PROJECT: ReactOS traceroute utility
- * FILE: apps/utils/net/tracert/tracert.h
- * PURPOSE: trace a packets route through a network
- * PROGRAMMERS: Ged Murphy (gedmurphy(a)gmail.com)
- * REVISIONS:
- * GM 03/05/05 Created
- */
+#define WIN32_LEAN_AND_MEAN
+#include <winsock2.h>
+#include <tchar.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <ws2tcpip.h>
+#include <string.h>
+#include <time.h>
#define ECHO_REPLY 0
#define DEST_UNREACHABLE 3
#define ECHO_REQUEST 8
#define TTL_EXCEEDED 11
+#define MAX_PING_PACKET_SIZE 1024
+#define MAX_PING_DATA_SIZE (MAX_PING_PACKET_SIZE + sizeof(IPv4Header)
+#define PACKET_SIZE 32
#define ICMP_MIN_SIZE 8
-#define ICMP_MAX_SIZE 65535
-#define PACKET_SIZE 32
+
/* we need this for packets which have the 'dont fragment'
- * bit set, as they can get quite large otherwise
- * (I've seen some reach 182 bytes */
+ * bit set, as they can get quite large otherwise */
#define MAX_REC_SIZE 200
/* pack the structures */
@@ -65,19 +65,24 @@
struct ICMPHeader OrigIcmpHeader;
} TTL_EXCEED_HEADER, *PTTL_EXCEED_HEADER;
-/* return to normal */
#include <poppack.h>
-/* function definitions */
-static BOOL ParseCmdline(int argc, char* argv[]);
-static INT Driver(void);
-static INT Setup(INT ttl);
-static VOID SetupTimingMethod(void);
-static VOID ResolveHostname(void);
-static VOID PreparePacket(INT packetSize, USHORT seqNum);
-static INT SendPacket(INT datasize);
-static INT ReceivePacket(INT datasize);
-static INT DecodeResponse(INT packetSize);
-static LONGLONG GetTime(void);
-static WORD CheckSum(PUSHORT data, UINT size);
-static VOID Usage(void);
+typedef struct _APPINFO
+{
+ SOCKET icmpSock; // socket descriptor
+ SOCKADDR_IN source, dest; // source and destination address info
+ PECHO_REPLY_HEADER SendPacket; // ICMP echo packet
+ PIPv4_HEADER RecvPacket; // return reveive packet
+
+ BOOL bUsePerformanceCounter; // whether to use the high res performance counter
+ LARGE_INTEGER TicksPerMs; // number of millisecs in relation to proc freq
+ LARGE_INTEGER TicksPerUs; // number of microsecs in relation to proc freq
+ LONGLONG lTimeStart; // send packet, timer start
+ LONGLONG lTimeEnd; // receive packet, timer end
+
+ BOOL bResolveAddresses; // -d MS ping defaults to true.
+ INT iMaxHops; // -h Max number of hops before trace ends
+ INT iHostList; // -j Source route
+ INT iTimeOut; // -w time before packet times out
+
+} APPINFO, *PAPPINFO;