用C/C++程式碼檢測ip能否ping通（配合awk和system可以做到批量檢測）

       遇到一個小需求， 快速搞定。 來看看用C/C++程式碼檢測ip能否ping通：

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

using namespace std;

string getCmdResult(const string &strCmd)  // 這個是獲取命令執行的結果， 類似於system, 之前我已經說過了
{  
    char buf[10240] = {0};  
    FILE *pf = NULL;  
      
    if( (pf = popen(strCmd.c_str(), "r")) == NULL )  
    {  
        return "";  
    }  
  
    string strResult;  
    while(fgets(buf, sizeof buf, pf))  
    {  
        strResult += buf;  
    }  
      
    pclose(pf);  
  
    unsigned int iSize =  strResult.size();  
    if(iSize > 0 && strResult[iSize - 1] == '\n')  // linux  
    {  
        strResult = strResult.substr(0, iSize - 1);  
    }  
  
    return strResult;  
}  


int main(int argc, char *argv[])
{
	if(argc != 2)
	{
		cout << "no" << endl;
		return -1;
	}

	string strCmd = "ping " + string(argv[1]) + " -w 1";

	string strRe = getCmdResult(strCmd);

	if(strRe.find("received") != string::npos && strRe.find(", 0 received") == string::npos)
	{
		cout << "ipok:" + string(argv[1]) << endl;
	}
	else
	{
		cout << argv[1] << endl;
	}

	return 0;
}

      測試一下：

ubuntu@VM-0-13-ubuntu:~$ ./a.out 
no
ubuntu@VM-0-13-ubuntu:~$ ./a.out 1.1.1.1
1.1.1.1
ubuntu@VM-0-13-ubuntu:~$ ./a.out 172.16.0.13
ipok:172.16.0.13
ubuntu@VM-0-13-ubuntu:~$ ./a.out www.baidu.com
ipok:www.baidu.com
ubuntu@VM-0-13-ubuntu:~$ 

      如上ping測試的超時時間是1s, 自己可以改。  另外， 如果有a.txt檔案， 每行一個ip, 怎麼知道哪些ip能否ping通呢？ awk和system搞起吧， 我們已經說過了：

ubuntu@VM-0-13-ubuntu:~$ cat a.txt 
1.1.1.1
www.baidu.com
www.qq.com
ubuntu@VM-0-13-ubuntu:~$ 
ubuntu@VM-0-13-ubuntu:~$ 
ubuntu@VM-0-13-ubuntu:~$ 
ubuntu@VM-0-13-ubuntu:~$ awk '{cmd="./a.out " $1; system(cmd)}' a.txt
1.1.1.1
ipok:www.baidu.com
ipok:www.qq.com
ubuntu@VM-0-13-ubuntu:~$ 

      可見 1.1.1.1 ping不通， 其餘的可以ping通。

     上面用awk和system有個問題：如果ip過多， 則必須等到所有ip檢測完畢後， 才知道最後的結果。 也就是說， 並不是處理完一個ip後， 就立即能看到結果的。怎麼辦呢？可以寫程式逐行讀取檔案來搞起， 看下：

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fstream>
#include <string>
using namespace std;

string getCmdResult(const string &strCmd)  
{  
    char buf[10240] = {0};  
    FILE *pf = NULL;  
      
    if( (pf = popen(strCmd.c_str(), "r")) == NULL )  
    {  
        return "";  
    }  
  
    string strResult;  
    while(fgets(buf, sizeof buf, pf))  
    {  
        strResult += buf;  
    }  
      
    pclose(pf);  
  
    unsigned int iSize =  strResult.size();  
    if(iSize > 0 && strResult[iSize - 1] == '\n')  // linux  
    {  
        strResult = strResult.substr(0, iSize - 1);  
    }  
  
    return strResult;  
}  

string ipCheck(const string &ip)
{
	string strCmd = "ping " + ip + " -w 1";
	string strRe = getCmdResult(strCmd);
	if((strRe.find("received") != string::npos && strRe.find(", 0 received") == string::npos))
	{
		return "ipok:" + string(ip);
	}
	else
	{
		return ip;
	}
}

int main(int argc, char *argv[])   // ./a.out a.txt b.txt
{
	if(argc != 3)
	{
		cout << "error" << endl;
		return -1;
	}

	string strCmd = "rm -rf " + string(argv[2]);
	system(strCmd.c_str());

	strCmd = "wc -l " + string(argv[1]) + "| awk '{print $1}'"; // 獲取檔案行數
	string strNumLine = getCmdResult(strCmd);


	ifstream in(argv[1]);
	string filename;
	string line;

	unsigned int i = 0;
	if(in) // 有該檔案
	{
		while (getline (in, line)) // line中不包括每行的換行符
		{
			// 這裡最好做ip格式判斷
			string strResult = ipCheck(line);
			strCmd = "echo " + strResult +  " >> " + string(argv[2]) ;
			cout  << strCmd << endl;
			system(strCmd.c_str());
		}
	}
	else // 沒有該檔案
	{
		cout <<"no such file" << endl;
	}

	return 0;
}

      看下結果：

ubuntu@VM-0-13-ubuntu:~/tmp_test$ ls
a.txt  test.cpp
ubuntu@VM-0-13-ubuntu:~/tmp_test$ 
ubuntu@VM-0-13-ubuntu:~/tmp_test$ 
ubuntu@VM-0-13-ubuntu:~/tmp_test$ cat a.txt 
1.1.1.1
2.2.2.2
www.baidu.com
3.3.3.3
4.4.4.4
ubuntu@VM-0-13-ubuntu:~/tmp_test$ 
ubuntu@VM-0-13-ubuntu:~/tmp_test$ 
ubuntu@VM-0-13-ubuntu:~/tmp_test$ g++ test.cpp
ubuntu@VM-0-13-ubuntu:~/tmp_test$ 
ubuntu@VM-0-13-ubuntu:~/tmp_test$ 
ubuntu@VM-0-13-ubuntu:~/tmp_test$ ./a.out a.txt b.txt
echo 1.1.1.1 >> b.txt
echo 2.2.2.2 >> b.txt
echo ipok:www.baidu.com >> b.txt
echo 3.3.3.3 >> b.txt
echo 4.4.4.4 >> b.txt
ubuntu@VM-0-13-ubuntu:~/tmp_test$ 
ubuntu@VM-0-13-ubuntu:~/tmp_test$ 
ubuntu@VM-0-13-ubuntu:~/tmp_test$ 
ubuntu@VM-0-13-ubuntu:~/tmp_test$ 
ubuntu@VM-0-13-ubuntu:~/tmp_test$ 
ubuntu@VM-0-13-ubuntu:~/tmp_test$ cat b.txt 
1.1.1.1
2.2.2.2
ipok:www.baidu.com
3.3.3.3
4.4.4.4
ubuntu@VM-0-13-ubuntu:~/tmp_test$