/* 本程序的主要源代码来自MSDN网站, 笔者只是做了一些改进祥桐裤和注释! 另外需要注意的是在Build之前,必须加入ws2_32.lib库文件,否则会提示"error LNK2001:"的错误！*/
Version 1.1 修改记录:
<1>解决了socket阻塞的问题,从而能够正确地处理超时的请求!
<2>增加了由用户控制发送ICMP包的数目的功能(即命令的第二个参数)
<3>增加了对ping结果的统计功能.
#pragma pack(4)
#include "winsock2.h"
#include "stdlib.h"
#include "stdio.h"
#define ICMP_ECHO 8
#define ICMP_ECHOREPLY 0
#define ICMP_MIN 8 // minimum 8 byte icmp packet (just header)
/* The IP header */
typedef struct iphdr {
unsigned int h_len:4// length of the header
unsigned int version:4// Version of IP
unsigned char tos// Type of service
unsigned short total_len// total length of the packet
unsigned short ident// unique identifier
unsigned short frag_and_flags// flags
unsigned char ttl
unsigned char proto// protocol (TCP, UDP etc)
unsigned short checksum// IP checksum
unsigned int sourceIP
unsigned int destIP
}IpHeader
//
// ICMP header
//
typedef struct icmphdr {
BYTE i_type
BYTE i_code/* type sub code */
USHORT i_cksum
USHORT i_id
USHORT i_seq
/* This is not the std header, but we reserve space for time */
ULONG timestamp
}IcmpHeader
#define STATUS_FAILED 0xFFFF
#define DEF_PACKET_SIZE32
#define DEF_PACKET_NUMBER 4/* 发送数据报的个数 */
#define MAX_PACKET 1024
#define xmalloc(s) HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,(s))
#define xfree(p) HeapFree (GetProcessHeap(),0,(p))
void fill_icmp_data(char *, int)
USHORT checksum(USHORT *, int)
int decode_resp(char *,int ,struct sockaddr_in *)
void Usage(char *progname){
fprintf(stderr,"Usage:\n")
fprintf(stderr,"%s [number of packets] [data_size]\n",progname)
fprintf(stderr,"datasize can be up to 1Kb\n")
ExitProcess(STATUS_FAILED)
}
int main(int argc, char **argv){
WSADATA wsaData
SOCKET sockRaw
struct sockaddr_in dest,from
struct hostent * hp
int bread,datasize,times
int fromlen = sizeof(from)
int timeout = 1000
int statistic = 0 /* 用于统计结果 */
char *dest_ip
char *icmp_data
char *recvbuf
unsigned int addr=0
USHORT seq_no = 0
if (WSAStartup(MAKEWORD(2,1),&wsaData) != 0){
fprintf(stderr,"WSAStartup failed: %d\n",GetLastError())
ExitProcess(STATUS_FAILED)
}
if (argc <2 ) {
Usage(argv[0])
}
sockRaw = WSASocket(AF_INET,SOCK_RAW,IPPROTO_ICMP,NULL, 0,WSA_FLAG_OVERLAPPED)
//
//注：为了使用发送接收超时设置(即设置SO_RCVTIMEO, SO_SNDTIMEO)，
//必须将标志位设为WSA_FLAG_OVERLAPPED !
//
if (sockRaw == INVALID_SOCKET) {
fprintf(stderr,"WSASocket() failed: %d\n",WSAGetLastError())
ExitProcess(STATUS_FAILED)
}
bread = setsockopt(sockRaw,SOL_SOCKET,SO_RCVTIMEO,(char*)&timeout,
sizeof(timeout))
if(bread == SOCKET_ERROR) {
fprintf(stderr,"failed to set recv timeout: %d\n",WSAGetLastError())
ExitProcess(STATUS_FAILED)
}
timeout = 1000
bread = setsockopt(sockRaw,SOL_SOCKET,SO_SNDTIMEO,(char*)&timeout,
sizeof(timeout))
if(bread == SOCKET_ERROR) {
fprintf(stderr,"failed to set send timeout: %d\n",WSAGetLastError())
ExitProcess(STATUS_FAILED)
}
memset(&dest,0,sizeof(dest))
hp = gethostbyname(argv[1])
if (!hp){
addr = inet_addr(argv[1])
}
if ((!hp) &&(addr == INADDR_NONE) ) {
fprintf(stderr,"Unable to resolve %s\n",argv[1])
ExitProcess(STATUS_FAILED)
}
if (hp != NULL)
memcpy(&(dest.sin_addr),hp->h_addr,hp->h_length)
else
dest.sin_addr.s_addr = addr
if (hp)
dest.sin_family = hp->h_addrtype
else
dest.sin_family = AF_INET
dest_ip = inet_ntoa(dest.sin_addr)
//
// atoi函数原型是: int atoi( const char *string )
// The return value is 0 if the input cannot be converted to an integer !
//
if(argc>2)
{
times=atoi(argv[2])
if(times == 0)
times=DEF_PACKET_NUMBER
}
else
times=DEF_PACKET_NUMBER
if (argc >3)
{
datasize = atoi(argv[3])
if (datasize == 0)
datasize = DEF_PACKET_SIZE
if (datasize >1024) /* 用户给出的数据包大小太大 */
{
fprintf(stderr,"WARNING : data_size is too large !\n")
datasize = DEF_PACKET_SIZE
}
}
else
datasize = DEF_PACKET_SIZE
datasize += sizeof(IcmpHeader)
icmp_data = (char*)xmalloc(MAX_PACKET)
recvbuf = (char*)xmalloc(MAX_PACKET)
if (!icmp_data) {
fprintf(stderr,"HeapAlloc failed %d\n",GetLastError())
ExitProcess(STATUS_FAILED)
}
memset(icmp_data,0,MAX_PACKET)
fill_icmp_data(icmp_data,datasize)
//
//显示提示信息
//
fprintf(stdout,"\nPinging %s ....\n\n",dest_ip)
for(int i=0i{
int bwrote
((IcmpHeader*)icmp_data)->i_cksum = 0
((IcmpHeader*)icmp_data)->timestamp = GetTickCount()
((IcmpHeader*)icmp_data)->i_seq = seq_no++
((IcmpHeader*)icmp_data)->i_cksum = checksum((USHORT*)icmp_data,datasize)
bwrote = sendto(sockRaw,icmp_data,datasize,0,(struct sockaddr*)&dest,sizeof(dest))
if (bwrote == SOCKET_ERROR){
if (WSAGetLastError() == WSAETIMEDOUT) {
printf("Request timed out.\n")
continue
}
fprintf(stderr,"sendto failed: %d\n",WSAGetLastError())
ExitProcess(STATUS_FAILED)
}
if (bwrote <datasize ) {
fprintf(stdout,"Wrote %d bytes\n",bwrote)
}
bread = recvfrom(sockRaw,recvbuf,MAX_PACKET,0,(struct sockaddr*)&from,&fromlen)
if (bread == SOCKET_ERROR){
if (WSAGetLastError() == WSAETIMEDOUT) {
printf("Request timed out.\n")
continue
}
fprintf(stderr,"recvfrom failed: %d\n",WSAGetLastError())
ExitProcess(STATUS_FAILED)
}
if(!decode_resp(recvbuf,bread,&from))
statistic++/* 成功接收的数目++ */
Sleep(1000)
}
/*
Display the statistic result
*/
fprintf(stdout,"\nPing statistics for %s \n",dest_ip)
fprintf(stdout,"Packets: Sent = %d,Received = %d, Lost = %d (%2.0f%% loss)\n",times,
statistic,(times-statistic),(float)(times-statistic)/times*100)
WSACleanup()
return 0
}
/*
The response is an IP packet. We must decode the IP header to locate
the ICMP data
*/
int decode_resp(char *buf, int bytes,struct sockaddr_in *from) {
IpHeader *iphdr
IcmpHeader *icmphdr
unsigned short iphdrlen
iphdr = (IpHeader *)buf
iphdrlen = (iphdr->h_len) * 4 // number of 32-bit words *4 = bytes
if (bytes <iphdrlen + ICMP_MIN) {
printf("Too few bytes from %s\n",inet_ntoa(from->sin_addr))
}
icmphdr = (IcmpHeader*)(buf + iphdrlen)
if (icmphdr->i_type != ICMP_ECHOREPLY) {
fprintf(stderr,"non-echo type %d recvd\n",icmphdr->i_type)
return 1
}
if (icmphdr->i_id != (USHORT)GetCurrentProcessId()) {
fprintf(stderr,"someone else''s packet!\n")
return 1
}
printf("%d bytes from %s:",bytes, inet_ntoa(from->sin_addr))
printf(" icmp_seq = %d. ",icmphdr->i_seq)
printf(" time: %d ms ",GetTickCount()-icmphdr->timestamp)
printf("\n")
return 0
}
USHORT checksum(USHORT *buffer, int size) {
unsigned long cksum=0
while(size >1) {
cksum+=*buffer++
size -=sizeof(USHORT)
}
if(size) {
cksum += *(UCHAR*)buffer
}
cksum = (cksum >>16) + (cksum &0xffff)
cksum += (cksum >>16)
return (USHORT)(~cksum)
}
/*
Helper function to fill in various stuff in our ICMP request.
*/
void fill_icmp_data(char * icmp_data, int datasize){
IcmpHeader *icmp_hdr
char *datapart
icmp_hdr = (IcmpHeader*)icmp_data
icmp_hdr->i_type = ICMP_ECHO
icmp_hdr->i_code = 0
icmp_hdr->i_id = (USHORT)GetCurrentProcessId()
icmp_hdr->i_cksum = 0
icmp_hdr->i_seq = 0
datapart = icmp_data + sizeof(IcmpHeader)
//
// Place some junk in the buffer.
//
memset(datapart,''E'', datasize - sizeof(IcmpHeader))
}
/******************* 附: ping命令执行时显示的画面 ****************\
* C:\Documents and settings\houzhijiang>ping 236.56.54.12 * * *
* Pinging 236.56.54.12 with 32 bytes of data: *
* *
* Request timed out.*
* Request timed out.*
* Request timed out.*
* Request timed out.*
* *
* Ping statistics for 236.56.54.12:*
* Packets: Sent = 4, Received = 0, Lost = 4 (100% loss),*
* *
\**************************************************************/
/**************************************************************\
* C:\Documents and Settings\houzhijiang>ping 127.0.0.1*
* *
* Pinging 127.0.0.1 with 32 bytes of data: *
* *
* Reply from 127.0.0.1: bytes=32 time<1ms TTL=128 *
* Reply from 127.0.0.1: bytes=32 time<1ms TTL=128 *
* Reply from 127.0.0.1: bytes=32 time<1ms TTL=128 *
* Reply from 127.0.0.1: bytes=32 time<1ms TTL=128 *
* *
* Ping statistics for 127.0.0.1: *
* Packets: Sent = 4, Received = 4, Lost = 0 (0% loss), *
* Approximate round trip times in milli-seconds: *
* Minimum = 0ms, Maximum = 0ms, Average = 0ms *
* *
\**************************************************************/
利用VB的Shell执行PING命令，将PING的输出重定向到文件 c:\r.txt，然后读取c:\r.txt文件显示运行结果。
由于VB中的Shell命令是异步执行的，即调用Shell后，没等Shell执行完毕，程序就继续执行下一条语句。为此，程序使用了系统API来判断Shell是否结束。
1）窗体及控件
2）代码
Option ExplicitPrivate Declare Function GetExitCodeProcess Lib "kernel32" ( _
ByVal hProcess As Long, _
lpExitCode As Long) As Long
Private Declare Function CloseHandle Lib "kernel32" ( _
ByVal hObject As Long) As Long
Private Declare Function OpenProcess Lib "kernel32" ( _
ByVal dwDesiredAccess As Long, _
ByVal bInheritHandle As Long, _
ByVal dwProcessId As Long) As Long
Const PROCESS_QUERY_INFORMATION = &H400
Const STILL_ALIVE = &H103
Private Sub Command1_Click()
蔽悄高 If Trim(Text1.Text) = "" Then
MsgBox "请输入域名或IP地址", vbInformation + vbOKOnly
Text1.SetFocus
Exit Sub
End If
'命令执行期间禁用命令按钮
Command1.Enabled = False
'调用Shell执行Ping，执行结果重定向到C:\r.txt中
Dim pid As Long
pid = Shell("cmd.exe /C Ping " & Text1.Text & " > c:\r.txt", vbHide)
' 提示
Text2.Text = "正在执行Ping " & Text1.Text & " ..."
'等待Shell执行结束
Dim hProc As Long
hProc = OpenProcess(PROCESS_QUERY_INFORMATION, 0, pid)
Dim ExitCode As Long
Do
Call 运轮GetExitCodeProcess(hProc, ExitCode)
DoEvents
Loop While ExitCode = STILL_ALIVE
'清空，准备显示结果
Text2.Text = ""
'打开 C:\r.txt文件
Open "c:\r.txt" For Input 宏尺As #1
Dim strLine As String
Do Until EOF(1)
Line Input #1, strLine
'显示执行结果
Text2.Text = Text2.Text & strLine & vbNewLine
Loop
'关闭文件
Close #1
'删除C:\r.txt
On Error Resume Next
Kill "c:\r.txt"
On Error GoTo 0
'使能命令按钮
Command1.Enabled = True
End Sub
Private Sub Form_Load()
Text1.Text = ""
Text2.Text = ""
End Sub
3）运行结果
Ping baidu.com 正在执行中 ....
Ping baidu.com执行结果
欢迎分享，转载请注明来源：内存溢出
微信扫一扫
支付宝扫一扫
评论列表（0条）