128 lines
2.9 KiB
C++
128 lines
2.9 KiB
C++
#include "NetworkWrapper.h"
|
|
#include "json.hpp"
|
|
#include <vector>
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <fstream>
|
|
using namespace std;
|
|
using namespace nlohmann;
|
|
|
|
#define USER_AGENT "Mozilla/5.0 (Windows NT 6.3; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.79 Safari/537.36"
|
|
|
|
string vastr;
|
|
string unstr;
|
|
vector<string> errdic;
|
|
|
|
/// Return: -1 Failed, 0 URL broken, 1 URL ok, 2 URL unknown
|
|
int CheckURL(const string& url)
|
|
{
|
|
string turl=url;
|
|
string content;
|
|
|
|
while(true)
|
|
{
|
|
HTTPConnection c;
|
|
c.setUserAgent(USER_AGENT);
|
|
c.setURL(turl);
|
|
c.setDataOutputBuffer(nullptr,0);
|
|
c.setSSLVerifyHost(false);
|
|
c.setSSLVerifyPeer(false);
|
|
|
|
//cout<<"Performing on "<<turl<<endl;
|
|
|
|
if(c.perform()!=0)
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
//cout<<"ResponseCode: "<<c.getResponseCode()<<endl;
|
|
|
|
if(c.getResponseCode()!=200)
|
|
{
|
|
if(c.getResponseCode()==302)
|
|
{
|
|
turl=c.getRedirectURL();
|
|
//cout<<"Redirect to "<<turl<<endl;
|
|
}
|
|
else return -1;
|
|
}
|
|
else
|
|
{
|
|
content=string((char*)c.getDataOutputBuffer(),c.getDataOutputBufferLength());
|
|
break;
|
|
}
|
|
}
|
|
|
|
/*
|
|
do {
|
|
ofstream ofs("out.txt");
|
|
ofs<<content<<endl;
|
|
}while(0);
|
|
//*/
|
|
|
|
if(content.find(vastr)!=string::npos)
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
if(content.find(unstr)!=string::npos)
|
|
{
|
|
return 2;
|
|
}
|
|
|
|
for(auto& s:errdic)
|
|
{
|
|
if(content.find(s)!=string::npos)
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
/// ...
|
|
return 3;
|
|
}
|
|
|
|
int main(int argc,char** argv)
|
|
{
|
|
do{
|
|
ifstream ifs("strings_utf8.json");
|
|
string str,tmp;
|
|
while(getline(ifs,tmp)) str.append(tmp);
|
|
json j=json::parse(str);
|
|
vastr=j["ValidContent"];
|
|
unstr=j["UnknownContent"];
|
|
for(auto& s:j["ErrorContent"])
|
|
{
|
|
errdic.push_back(s["content"].get<string>());
|
|
}
|
|
}while(0);
|
|
|
|
if(argc==1)
|
|
{
|
|
while(true)
|
|
{
|
|
cout<<"Please input URL to check:"<<endl;
|
|
string url;
|
|
if(getline(cin,url))
|
|
{
|
|
cout<< [&](){switch(CheckURL(url)){
|
|
case -1:return "Internal Error";
|
|
case 0:return "URL Broken";
|
|
case 1:return "URL OK";
|
|
case 2:return "URL Unknown Status";
|
|
default:return "Wrong Status";}}() << endl;
|
|
}
|
|
else break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
cout<< [&](){switch(CheckURL(argv[1])){
|
|
case -1:return "Internal Error";
|
|
case 0:return "URL Broken";
|
|
case 1:return "URL OK";
|
|
case 2:return "URL Unknown Status";
|
|
default:return "Wrong Status";}}() << endl;
|
|
}
|
|
}
|