Tcpportping

最近因工作需要，研究批量tcp port ping。在网上看到一个测试单个IP的脚本，遂将脚本改造了一下，实现批量的tcpportping。这个脚本ping目标IP和端口6次，去掉最大值，然后取5次的平均值。

共3个文件：Tcportping.py、tcpportping.c和tcpportping.so（根据tcpportping.c生成）
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
    
[root@***** ]# more tcpportping.c
/* tcpportping.c */
#include
#include
#include
#include
#include
#include
#include
    
/* count time functions */
static double mytime(void)
{
    struct timeval tv;
    if (gettimeofday(&tv, NULL) == -1)
        return 0.0;
    
    return (double)tv.tv_usec + (double)tv.tv_sec * 1000;
}
    
static PyObject *                                 /* returns object */
tcpping(PyObject *self, PyObject *args )
{
    struct  sockaddr_in addr;
    struct  hostent *hp;
    double  time;
    char    *host = NULL;
    int     fd;
    int     port, timeout;
    
    if (!PyArg_ParseTuple(args, "sii", &host, &port, &timeout))  /* convert Python -> C */
        return NULL;                              /* null=raise exception */
    
    if ((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
        return Py_BuildValue("d", -1.0);        /* convert C -> Python */
    }
    
    bzero((char *)&addr, sizeof(addr));
    if ((hp = gethostbyname(host)) == NULL) {
        return Py_BuildValue("d", -2.0);        /* convert C -> Python */
    }
    bcopy(hp->h_addr, &addr.sin_addr, hp->h_length);
    addr.sin_family = AF_INET;
    addr.sin_port = htons(port);
    
    struct timeval tv;
    
    tv.tv_sec = 0;
    tv.tv_usec = timeout * 1000;
    
    double stime = mytime();
    if (connect(fd, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
        return Py_BuildValue("d", -3.0);        /* convert C -> Python */
    }
    fd_set read, write;
    FD_ZERO(&read);
    FD_ZERO(&write);
    
    FD_SET(fd, &read);
    FD_SET(fd, &write);
    
    if (select(fd + 1, &read, &write, NULL, &tv) == 0) {
        close(fd);
        return Py_BuildValue("d", -4.0);        /* convert C -> Python */
    }
    
    double etime = mytime();
    time = etime - stime;
    if (!FD_ISSET(fd, &read) && !FD_ISSET(fd, &write)) {
        close(fd);
        return Py_BuildValue("d", -4.0);        /* convert C -> Python */
    }
    close(fd);
    return Py_BuildValue("d", time/1000);        /* convert C -> Python */
}
    
/* registration table  */
static struct PyMethodDef portping_methods[] = {
    {"tcpping", tcpping, METH_VARARGS},       /* method name, C func ptr, always-tuple */
    {NULL, NULL}                   /* end of table marker */
};
    
/* module initializer */
void inittcpportping( )                       /* called on first import */
{                                      /* name matters if loaded dynamically */
    (void) Py_InitModule("tcpportping", portping_methods);   /* mod name, table ptr */
}
 
 
 
 
 
# 生成tcpportping.so
[root@**** ]#  gcc -fpic -shared tcpportping.c -o tcpportping.so
 
# 主程序内容
[root@***** ~]# more Tcportping.py
#!/usr/bin/python2
 
import tcpportping
iplist=file('/root/iplist.txt')
 
ip=[]
port=[]
domain=[]
location=[]
for line in iplist.readlines():
        new_line=line.split()
        ip.append(new_line[0])
        port.append(new_line[1])
        domain.append(new_line[2])
        location.append(new_line[3])
 
#print ip
#print port
for i in range(len(ip)):
        Delay = [tcpportping.tcpping(ip[(i)], int(port[(i)]), 100),]
        Delay.append(tcpportping.tcpping(ip[(i)], int(port[(i)]), 100))
        Delay.append(tcpportping.tcpping(ip[(i)], int(port[(i)]), 100))
        Delay.append(tcpportping.tcpping(ip[(i)], int(port[(i)]), 100))
        Delay.append(tcpportping.tcpping(ip[(i)], int(port[(i)]), 100))
        Delay.append(tcpportping.tcpping(ip[(i)], int(port[(i)]), 100))
#       print Delay
        Delay.remove(max(Delay))
#       print Delay
        t = reduce(lambda x,y:x+y,Delay)/5
#       print t
        print domain[(i)]+","+location[(i)]+","+ip[(i)]+","+port[(i)]+","+str(t)
         
# 从文件中读取要ping的目的IP和端口，格式为：
 
111.161.60.55   2099      英雄联盟           天津联通      
125.39.200.43   2099      英雄联盟           天津联通      
125.39.200.44   8443      英雄联盟           天津联通
 
# 执行后输出的结果：也可以修改print格式成任何格式。
[root@nxttBC ~]# ./Tcportping.py
英雄联盟,天津联通,111.161.60.55,2099,21.585
英雄联盟,天津联通,125.39.200.43,2099,22.1576
英雄联盟,天津联通,125.39.200.44,8443,21.7886
 
# 也可以将结果重定向保存为文件