|
Ниже приводится полный исходный текст программы Qping. #include <winsock.h>
#define HOST_NAME "www.sources.ru"// Любое доменное имя
#define WINSOCK_VERSION 0x0101// Версия винсока 1.1
#define NO_FLAGS 0// Флаги не указываем
// В RFC 792 определены значения сообщений ICMP
#define ICMP_ECHO 8// An ICMP echo message
#define ICMP_ECHOREPLY 0// An ICMP echo reply message
#defineICMP_HEADERSIZE 8
struct ip // Структура заголовка IP
{
BYTE ip_verlen;// Version and header length
BYTE ip_tos;// Type of service
WORD ip_len;// Total packet length
UINT ip_id;// Datagram identification
WORD ip_fragoff;// Fragment offset
BYTE ip_ttl;// Time to live
BYTE ip_proto;// Protocol
UINT ip_chksum;// Checksum
IN_ADDR ip_src_addr;// Source address
IN_ADDR ip_dst_addr;// Destination address
BYTE ip_data[1];// Variable length data area
};
struct icmp// Структура заголовка ICMP
{
BYTE icmp_type;// Type of message
BYTE icmp_code;// Type "sub code" (zero for echos)
WORD icmp_cksum;// 1's complement checksum
HINSTANCE icmp_id;// Unique ID (the instance handle)
WORD icmp_seq;// Tracks multiple pings
BYTE icmp_data[1];// The start of optional data
};
char szPingBuffer[100];
HINSTANCE hInstance;
WORD InternetChksum(LPWORD lpwIcmpData, WORD wDataLength)
{
longlSum;// Store the summation
WORDwOddByte;// Left over byte from the summation
WORDwAnswer;// The 1's complement checksum
lSum = 0L;
while (wDataLength > 1)
{
lSum += *lpwIcmpData++;
wDataLength -= 2;
}
// Handle the odd byte if necessary and make sure the top half is zero
if (wDataLength == 1)
{
wOddByte = 0;
*((LPBYTE) &wOddByte) = *(LPBYTE)lpwIcmpData;// One byte only
lSum += wOddByte;
}
// Add back the carry outs from the 16 bits to the low 16 bits
lSum = (lSum >> 16) + (lSum & 0xffff);// Add high-16 to low-16
lSum += (lSum >> 16);// Add carry
wAnswer = (WORD)~lSum;// 1's complement, then truncate
// to 16 bits
return(wAnswer);
}
BOOL DoPingOperation(HANDLE hInstance)
{
// Локальные переменные
int iPacketSize;// размер ICMP-пакета
int iHostAddrLength;// Длина адреса сетевого компьютера
int iIPHeadLength;// Длина заголовка IP-датаграммы
int iReceivedBytes;// Количество принятых байтов
int iSentBytes;// Количество посланных байтов
int nProtocol;// Номер протокола ICMP
int iSocketError;// Значение кода ошибки
PDWORD pdwTimeStamp;// Счетчик "тиков" при передаче
DWORD dwReturnTime;// Счетчик "тиков" при приеме
DWORD dwRoundTrip;// Счетчик "тиков" среднего времени пробега
// Структуры, описанные в WINSOCK.H
SOCKADDR_INsockAddrLocal;// Структуры адреса сокета
SOCKADDR_INsockAddrHost;//
SOCKET hSocket;// Дескриптор сокета
LPHOSTENT lpHostEntry;// Структура данных с
// информацией о сетевом компьютере
LPPROTOENT lpProtocolEntry;// Структура данных с информацией о протоколе
BYTE IcmpSendPacket[1024];// Буфер для посылаемых данных
BYTE IcmpRecvPacket[4096];// Буфер для принимаемых данных
struct icmp *pIcmpHeader;// Указатель на структуру ICMP
struct ip *pIpHeader;// Указатель на структуру-заголовок IP
LPSTR lpszHostName;// Указатель на удаленный сервер времени
lpszHostName = HOST_NAME;
if ((lpHostEntry = gethostbyname(HOST_NAME)) == NULL) {
wsprintf(szPingBuffer, "Could not get %s IP address.",(LPSTR)lpszHostName);
return(FALSE);
}
sockAddrLocal.sin_family = AF_INET;
sockAddrLocal.sin_addr = *((LPIN_ADDR) *lpHostEntry->h_addr_list);
// В случае простого сокета, мы должны указывать протокол
if ((lpProtocolEntry = getprotobyname("icmp")) == NULL)
nProtocol = IPPROTO_ICMP;
else
nProtocol = lpProtocolEntry->p_proto;
// Создаем простой сокет и указываем ICMP в качестве протокола
if ((hSocket = socket(PF_INET, SOCK_RAW, nProtocol)) == INVALID_SOCKET)
{
wsprintf(szPingBuffer, "Could not create a RAW socket.");
return(FALSE);
}
pIcmpHeader = (struct icmp *) IcmpSendPacket;// Point at the data area
pIcmpHeader->icmp_type = ICMP_ECHO;// then fill in the data.
pIcmpHeader->icmp_code = 0;// Use the Sockman instance
pIcmpHeader->icmp_id = hInstance;// handle as a unique ID.
pIcmpHeader->icmp_seq = 0;// It's important to reset
pIcmpHeader->icmp_cksum = 0;// the checksum to zero.
//Значение счетчика "тиков" располагается в необязательной области данных
pdwTimeStamp = (PDWORD)&IcmpSendPacket[ICMP_HEADERSIZE];
*pdwTimeStamp = GetTickCount();
iPacketSize = ICMP_HEADERSIZE + sizeof(DWORD);
pIcmpHeader->icmp_cksum = InternetChksum((LPWORD)pIcmpHeader, iPacketSize);
if (pIcmpHeader->icmp_cksum !=0 )
{
iSentBytes = sendto(hSocket, (LPSTR) IcmpSendPacket, iPacketSize,
NO_FLAGS, (LPSOCKADDR) &sockAddrLocal, sizeof(sockAddrLocal));
if (iSentBytes == SOCKET_ERROR) {
closesocket(hSocket);
wsprintf(szPingBuffer,
"The sendto() function returned a socket error.");
return(FALSE);
}
if (iSentBytes != iPacketSize){
closesocket(hSocket);
wsprintf(szPingBuffer,
"Wrong number of bytes sent: %d", iSentBytes);
return(FALSE);
}
iHostAddrLength = sizeof(sockAddrHost);
iReceivedBytes = recvfrom(hSocket, (LPSTR) IcmpRecvPacket,
sizeof(IcmpRecvPacket), NO_FLAGS, (LPSOCKADDR) &sockAddrHost,
&iHostAddrLength);
}
else {
closesocket(hSocket);
wsprintf(szPingBuffer, "Checksum computation error! Result was zero!");
return(FALSE);
}
closesocket(hSocket);
if (iReceivedBytes == SOCKET_ERROR) {
iSocketError = WSAGetLastError();
if (iSocketError == 10004) {
wsprintf(szPingBuffer,
"Ping operation for %s was cancelled.",
(LPSTR)lpszHostName);
dwRoundTrip = 0;
return(TRUE);
}
else {
wsprintf(szPingBuffer,
"Socket Error from recvfrom(): %d", iSocketError);
return(FALSE);
}
}
dwReturnTime = GetTickCount();
dwRoundTrip = dwReturnTime - *pdwTimeStamp;
// Указываем на IP-заголовок принятого пакета
pIpHeader = (struct ip *)IcmpRecvPacket;
// Извлекаем биты 4-7 и преобразуем количество З2-битных слов в количество байтов
iIPHeadLength = (pIpHeader->ip_verlen >> 4) << 2;
// Проверяем длину, чтобы удостовериться, что ICMP-заголовок принят
if (iReceivedBytes < iIPHeadLength + ICMP_HEADERSIZE) {
wsprintf(szPingBuffer, "Received packet was too short.");
return(FALSE);
}
// Указываем на ICMP-сообщение, следующее сразу за IP-заголовком
pIcmpHeader = (struct icmp *) (IcmpRecvPacket + iIPHeadLength);
// Проверяем, что мы приняли именно "эхо"-ответ
if (pIcmpHeader->icmp_type != ICMP_ECHOREPLY) {
wsprintf(szPingBuffer,
"Received packet was not an echo reply to your ping.");
return(FALSE);
}
// Проверяем, принадлежит ли этот пакет нашей программе
if (pIcmpHeader->icmp_id != (HINSTANCE)hInstance) {
wsprintf(szPingBuffer,
"Received packet was not sent by this program.");
return(FALSE);
}
// Да, этот пакет был послан нашей программой. Обратите
// внимание на IP-адрес и имя удаленного компьютера,
// пославшего "эхо"-ответ
lstrcpy(lpszHostName, (LPSTR)lpHostEntry->h_name);
wsprintf(szPingBuffer,
"Round-trip travel time to %s [%s] was %d milliseconds.",
(LPSTR)lpszHostName, (LPSTR)inet_ntoa(sockAddrHost.sin_addr),
dwRoundTrip);
return(TRUE);
}
void main ()
{
WSADATA wsaData;
WSAStartup(WINSOCK_VERSION, &wsaData);
DoPingOperation(hInstance);
MessageBox(NULL, szPingBuffer, "www.sources.ru", MB_OK|MB_ICONSTOP);
WSACleanup();
} |