Tcpportping
最近因工作需要，研究批量tcp port ping。在网上看到一个测试单个IP的脚本，遂将脚本改造了一下，实现批量的tcpportping。这个脚本ping目标IP和端口6次，去掉最大值，然后取5次的平均值。
共3个文件：Tcportping.py、tcpportping.c和tcpportping.so（根据tcpportping.c生成）；
需要安装python-devel
[root@***** ]# more tcpportping.c
/* tcpportping.c */
#include </usr/include/python2.6/Python.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <sys/time.h>
/* 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
# 也可以将结果重定向保存为文件