加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
mainwindow.cpp 35.25 KB
一键复制 编辑 原始数据 按行查看 历史
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982
#include "mainwindow.h"
#include "ui_mainwindow.h"
#include "waterfallwindow.h"
#ifdef Q_OS_WIN32
#include "WinInet.h"
#include "winnetwk.h"
#include "windows.h"
#include "proxy/rywinhttp.h"
#endif
#ifdef Q_OS_MAC
#include "proxy/proxyautoconfig.h"
#include <CoreFoundation/CoreFoundation.h>
#include <SystemConfiguration/SystemConfiguration.h>
#include <SystemConfiguration/SCDynamicStoreCopySpecific.h>
#include <SystemConfiguration/SCNetworkConfiguration.h>
#endif
#ifdef Q_OS_WIN
#include "zlib/zlib.h"
#else
#include <zlib.h>
#endif
#include <quazip/quazip.h>
#include <quazip/quazipfile.h>
#include "rytablesortfilterproxymodel.h"
#include "ryupdatechecker.h"
#include <QWebFrame>
extern QString version;
extern QString appPath;
QByteArray gzipDecompress(QByteArray data){
if (data.size() <= 4) {
qWarning("gUncompress: Input data is truncated");
return QByteArray();
}
QByteArray result;
int ret;
z_stream strm;
static const int CHUNK_SIZE = 1024;
char out[CHUNK_SIZE];
/* allocate inflate state */
strm.zalloc = Z_NULL;
strm.zfree = Z_NULL;
strm.opaque = Z_NULL;
strm.avail_in = data.size();
strm.next_in = (Bytef*)(data.data());
ret = inflateInit2(&strm, 15 + 32); // gzip decoding
if (ret != Z_OK)
return QByteArray();
// run inflate()
do {
strm.avail_out = CHUNK_SIZE;
strm.next_out = (Bytef*)(out);
ret = inflate(&strm, Z_NO_FLUSH);
Q_ASSERT(ret != Z_STREAM_ERROR); // state not clobbered
switch (ret) {
case Z_NEED_DICT:
ret = Z_DATA_ERROR; // and fall through
case Z_DATA_ERROR:
case Z_MEM_ERROR:
(void)inflateEnd(&strm);
return QByteArray();
}
result.append(out, CHUNK_SIZE - strm.avail_out);
} while (strm.avail_out == 0);
// clean up and return
inflateEnd(&strm);
return result;
}
#ifdef Q_OS_WIN
bool setProxy(const QSettings& proxySetting){
// 感谢maconel的帮助!
QString proxyAutoConfigURL = proxySetting.value("AutoConfigURL","0").toString();
int enable = proxySetting.value("ProxyEnable").toInt();
QString proxyString =proxySetting.value("ProxyServer").toString();
// init options
INTERNET_PER_CONN_OPTION_LIST connOptionList;
INTERNET_PER_CONN_OPTION connOptions[5];
memset(&connOptionList, 0, sizeof(connOptionList));
memset(&connOptions, 0, sizeof(connOptions));
DWORD size = 0;
size = sizeof(connOptionList);
connOptions[0].dwOption = INTERNET_PER_CONN_FLAGS;
connOptions[1].dwOption = INTERNET_PER_CONN_PROXY_SERVER;
connOptions[2].dwOption = INTERNET_PER_CONN_PROXY_BYPASS;
connOptions[3].dwOption = INTERNET_PER_CONN_AUTOCONFIG_URL;
connOptions[4].dwOption = INTERNET_PER_CONN_AUTODISCOVERY_FLAGS;
connOptionList.dwSize = sizeof(connOptionList);
connOptionList.pszConnection = NULL;
connOptionList.dwOptionCount = sizeof(connOptions) / sizeof(connOptions[0]);
connOptionList.dwOptionError = 0;
connOptionList.pOptions = connOptions;
// direct
if(0 == enable){
connOptions[0].Value.dwValue |= PROXY_TYPE_DIRECT;
}
// auto proxy config
if(proxyAutoConfigURL != "0"){
connOptions[0].Value.dwValue |= PROXY_TYPE_AUTO_PROXY_URL;
connOptions[3].Value.pszValue = (WCHAR*)proxyAutoConfigURL.toStdWString().data();
}
// TODO: auto detect
//connOptions[0].Value.dwValue |= PROXY_TYPE_AUTO_DETECT;
if(1 == enable){
connOptions[0].Value.dwValue |= PROXY_TYPE_PROXY;
connOptions[1].Value.pszValue = (WCHAR*)proxyString.toStdWString().data();
//TODO pass domains
//connOptions[2].Value.pszValue = L"localhost;127.0.0.1;<local>";
}
if (!InternetSetOption(NULL, INTERNET_OPTION_PER_CONNECTION_OPTION, &connOptionList, size)){
qDebug()<<QString("fail %1").arg(GetLastError());
return false;
}
InternetSetOption(NULL, INTERNET_OPTION_SETTINGS_CHANGED, NULL, NULL);
InternetSetOption(NULL, INTERNET_OPTION_REFRESH, NULL, NULL);
return true;
}
#endif
#ifdef Q_OS_MAC
const char * myCFStringCStringPtr(CFStringRef aString,CFStringEncoding encoding,bool* needFree = 0){
if(needFree!=0){
*needFree = false;
}
if(!aString){
qDebug()<<"myCFStringCStringPtr source is NULL";
return NULL;
}
const char* ret = CFStringGetCStringPtr(aString,encoding);
if(ret == NULL){
CFIndex length = CFStringGetLength(aString);
CFIndex maxSize =
CFStringGetMaximumSizeForEncoding(length,
kCFStringEncodingUTF8);
char *buffer = (char *)malloc(maxSize);
if (CFStringGetCString(aString, buffer, maxSize,
kCFStringEncodingUTF8)) {
if(needFree != 0){
*needFree = true;
}
return buffer;
}
free(buffer);
return NULL;
}
return ret;
}
QString getServiceName(){
// get primary serverId
char *serviceName = NULL;
CFStringRef primaryServiceId=NULL;
SCDynamicStoreRef dynamicStoreDomainState = SCDynamicStoreCreate(NULL,
CFSTR("Rythem"),
NULL,
NULL);
// get primary interface
if (dynamicStoreDomainState) {
long n = (long)CFStringGetLength(kSCDynamicStoreDomainState);
n += CFStringGetLength(kSCCompNetwork);
n += CFStringGetLength(kSCCompGlobal);
n += CFStringGetLength(kSCEntNetIPv4);
n += 3;// for three '/'
char *netIp4KeyStr = (char*)malloc(sizeof(char)*(n+1));
bool stateNeedFree = false;
const char *domainState = myCFStringCStringPtr(kSCDynamicStoreDomainState, kCFStringEncodingUTF8,&stateNeedFree);
bool networkNeedFree = false;
const char *network = myCFStringCStringPtr(kSCCompNetwork, kCFStringEncodingUTF8,&networkNeedFree);
bool globalNeedFree = false;
const char *global = myCFStringCStringPtr(kSCCompGlobal, kCFStringEncodingUTF8,&globalNeedFree);
bool ipv4NeedFree = false;
const char *ipv4 = myCFStringCStringPtr(kSCEntNetIPv4, kCFStringEncodingUTF8,&ipv4NeedFree);
sprintf(netIp4KeyStr, "%s/%s/%s/%s",
domainState,
network,
global,
ipv4);
if(stateNeedFree){
free((char*)domainState);
}
if(networkNeedFree){
free((char*)network);
}
if(globalNeedFree){
free((char*)global);
}
if(ipv4NeedFree){
free((char*)ipv4);
}
CFStringRef netIpv4KeyCFStr = CFStringCreateWithCString(NULL, netIp4KeyStr, kCFStringEncodingUTF8);
CFPropertyListRef netIpv4List =SCDynamicStoreCopyValue(dynamicStoreDomainState, netIpv4KeyCFStr);
if (netIpv4List) {
primaryServiceId = (CFStringRef)CFDictionaryGetValue((CFDictionaryRef)netIpv4List, kSCDynamicStorePropNetPrimaryService);
if(primaryServiceId){
CFRetain(primaryServiceId);
}
CFRelease(netIpv4List);
}
CFRelease(netIpv4KeyCFStr);
free(netIp4KeyStr);
}
if(!primaryServiceId){
qDebug()<<"no service name found";
return QString();
}
CFShow(primaryServiceId);
// Preferences.plist
SCPreferencesRef prefs = SCPreferencesCreate(NULL, CFSTR("SystemConfiguration"), NULL);
CFArrayRef services = (CFArrayRef)SCNetworkServiceCopyAll(prefs);
for(CFIndex i=CFArrayGetCount(services)-1;i>=0;--i){
SCNetworkServiceRef service = (SCNetworkServiceRef)CFArrayGetValueAtIndex(services,i);
CFTypeID typeID = SCNetworkServiceGetTypeID();
unsigned long tmp = (unsigned long)typeID;
qDebug()<<QString("%1").arg((unsigned long)SCNetworkReachabilityGetTypeID());
CFStringRef serviceId = SCNetworkServiceGetServiceID(service);
CFShow(serviceId);
CFShow(primaryServiceId);
if( CFStringCompare(primaryServiceId,serviceId,0) == kCFCompareEqualTo ){
CFStringRef name = SCNetworkServiceGetName(service);// Wi-Fi "Bluetooth DUN" etc.
if(name){
unsigned long l = CFStringGetLength(name) + 1;
CFShow(name);
serviceName = (char*)malloc(l*sizeof(char));
bool needFree = false;
const char* theName = myCFStringCStringPtr(name, kCFStringEncodingUTF8,&needFree);
sprintf(serviceName, "%s",theName);
if(needFree){
free((char *)theName);
}
}
//break;
}else{
CFStringRef name = SCNetworkServiceGetName(service);
if(name)
CFShow(name);
else
qDebug()<<"no name";
}
}
CFRelease(primaryServiceId);
if(!serviceName){
qDebug()<<"not primary value";
return QString();
}
QString result(serviceName);
qDebug()<<"primary serviceName:"<<result;
free(serviceName);
return result;
}
bool execScript(const QString &script){
QString service = getServiceName();
if(service == NULL || service.isEmpty()){
// TODO 提示
return false;
}
QProcess process;
qDebug()<<"script"<<script.arg(service);
QStringList args = script.arg(service).split(" ");
process.start(script.arg(service));
QEventLoop eventLoop;
eventLoop.connect(&process,SIGNAL(finished(int)),&eventLoop,SLOT(quit()));
eventLoop.exec();
qDebug()<<"output:\n"<<QString(process.readAllStandardOutput())<<QString(process.readAllStandardError());
return process.exitCode() == 0;
}
bool setProxyState(bool enable){
return execScript(QString("./setupproxy setHttpAndHttpsProxyState %1 %2").arg("%1",(enable?"on":"off")));
}
bool setAutoProxyUrl(const QString& url){
return execScript(QString("./setupproxy setAutoProxyUrl %1 %2").arg("%1",url));
}
bool setProxyForService(const QString& host, const int port){
return execScript(QString("./setupproxy setHttpAndHttpsProxy %1 %2 %3").arg(QString("%1"),host,QString::number(port)));
}
bool disableAutoProxyAndSetProxyForService(const QString& host, const int port){
bool a = execScript(QString("./setupproxy setAutoProxyState %1 off"));
return a & setProxyForService(host,port);
}
bool execProxysetting(const QString &pacUrl=QString()){
QProcess process;
QString service = getServiceName();
if(service == NULL || service.isEmpty()){
// TODO 提示
return false;
}
QString script = "proxysetting";
if(pacUrl.isEmpty()){
script = QString(appPath+"/proxysetting --disablepac %1").arg(service);
}else{
script = QString(appPath+"/proxysetting --setpac %1 %2").arg(service,pacUrl);
}
qDebug()<<script;
process.start(script);
QEventLoop eventLoop;
eventLoop.connect(&process,SIGNAL(error(QProcess::ProcessError)),&eventLoop,SLOT(quit()));
eventLoop.connect(&process,SIGNAL(finished(int)),&eventLoop,SLOT(quit()));
eventLoop.exec();
qDebug()<<"output:\n"<<QString(process.readAllStandardOutput())<<QString(process.readAllStandardError())
<<QString("\nretcode:%1\n").arg(process.exitCode());
return process.exitCode() == 0;
}
bool setPAC(const QString& url){
return execProxysetting(url);
//return execScript(QString("networksetup -setautoproxyurl %1 %2").arg("%1",url));
}
bool setMyPAC(){
return setPAC("http://127.0.0.1:8889/rythem_pac");
//return setPAC(QString("file://local%1/rythem_pac").arg(appPath));
}
bool disableMyPac(){
return execProxysetting();
//return execScript(QString("networksetup -setautoproxystate %1 off"));
}
#endif
// RyJsBridge
RyJsBridge::RyJsBridge(){
}
QString RyJsBridge::doAction(int action, const QString msg, quint64 groupId){
qDebug()<<"doAction "<<QString::number(action)<< msg<<"groupId"<<QString::number(groupId);
RyRuleManager *manager = RyRuleManager::instance();
QSharedPointer<RyRuleProject> pro;
QSharedPointer<RyRuleGroup> group;
QSharedPointer<RyRule> rule;
manager->startWriteConfigTimer();
switch(action){
case 0://add local group
group = manager->addGroupToLocalProject(msg);//暂时只允许添加到本地project
//emit ruleChanged(0,"success");
if(!group.isNull()){
qDebug()<<group->toJSON();
return group->toJSON();
}
break;
case 1://add rule to group
rule = manager->addRuleToGroup(msg,groupId);
if(!rule.isNull()){
//qDebug()<<QString::number(rule->ruleId());
//return QString::number(rule->ruleId());
return rule->toJSON();
}
break;
case 2://add remote project
pro = manager->addRemoteProject(msg);
if(!pro.isNull()){
emit ruleChanged(2,pro->toJson());
return pro->toJson();
}
break;
case 3://update group
manager->updateRuleGroup(msg,groupId);
break;
case 4://update rule
manager->updateRule(msg,groupId);
break;
case 5://remove group
manager->removeGroup(msg.toULongLong());
break;
case 6://remove rule (6,ruleId,groupId)
manager->removeRule(msg.toULongLong(),groupId);
break;
case 7://import local config
pro = manager->addLocalProject(msg);
if(!pro.isNull()){
emit ruleChanged(2,pro->toJson());
return pro->toJson();
}
break;
}
return "";
}
QString RyJsBridge::getFile(){
return QFileDialog::getOpenFileName();
}
QString RyJsBridge::getFileList(){
return QFileDialog::getOpenFileNames().join("\n");
}
QString RyJsBridge::getDir(){
return QFileDialog::getExistingDirectory();
}
QString RyJsBridge::getConfigs(){
RyRuleManager *manager = RyRuleManager::instance();
QString s = manager->toJson();
s.remove('\r');
s.remove('\n');
s.remove('\t');
return s;
}
// end of RyJsBridge
MainWindow::MainWindow(QWidget *parent) :
QMainWindow(parent),
ui(new Ui::MainWindow),
_isUsingCapture(false),
isFirstTimeToggle(true)
#ifdef Q_OS_WIN
,proxySetting("\\HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings\\",QSettings::NativeFormat)
#endif
{
ui->setupUi(this);
pipeTableModel = new RyTableModel(this);
sortFilterProxyModel = new RyTableSortFilterProxyModel(this);
ui->tableView->setSortingEnabled(true);
ui->tableView->setModel(sortFilterProxyModel);
sortFilterProxyModel->setDynamicSortFilter(true);
sortFilterProxyModel->setSourceModel(pipeTableModel);
ui->tableView->setSelectionBehavior(QAbstractItemView::SelectRows);
ui->tableView->setColumnWidth(0,30);
ui->tableView->setColumnWidth(1,30);
ui->tableView->setColumnWidth(2,50);
ui->tableView->setColumnWidth(3,50);
ui->tableView->setColumnWidth(7,50);
ui->tableView->verticalHeader()->setDefaultSectionSize(20);
ui->tableView->verticalHeader()->hide();
_jsBridge = new RyJsBridge();
QWebSettings::globalSettings()->setAttribute(QWebSettings::DeveloperExtrasEnabled, true);
addJsObject();
connect(ui->webView->page()->mainFrame(),SIGNAL(javaScriptWindowObjectCleared()),this,SLOT(addJsObject()));
//修复proxy服务器过完启动导致配置页面空白的问题 延迟一秒加载
QTimer::singleShot(1000, this, SLOT(loadConfigPage()));
// should use slot to do this
//ui->composer->setupProxy(RyProxyServer::instance()->serverAddress().toString(),
// RyProxyServer::instance()->serverPort());
ui->composer->setupProxy("127.0.0.1",
8889);
_itemSelectModel = ui->tableView->selectionModel();
connect(ui->tableView,SIGNAL(doubleClicked(QModelIndex)),SLOT(onItemDoubleClicked(QModelIndex)));
//connect(_itemSelectModel,SIGNAL(currentChanged(QModelIndex,QModelIndex)),SLOT(onSelectionChange(QModelIndex)));
connect(_itemSelectModel,SIGNAL(selectionChanged(QItemSelection,QItemSelection)),SLOT(onSelectionChange(QItemSelection,QItemSelection)));
//ui->tableView->setItemDelegate();
createMenus();
//toggleCapture();
connect(ui->ActionCapture,SIGNAL(triggered()),SLOT(toggleCapture()));
connect(ui->actionRemoveAll,SIGNAL(triggered()),this,SLOT(onActionRemoveAll()));
connect(ui->actionWaterfall, SIGNAL(triggered()), this, SLOT(onWaterfallActionTriggered()));
checker = new RyUpdateChecker(this);
QTimer timer;
timer.singleShot(1000,this,SLOT(checkNewVersion()));
//checkNewVersion();
#ifdef Q_OS_MAC
if(appPath.startsWith("/Volumes")){
ui->ActionCapture->setEnabled(false);
ui->ActionCapture->setToolTip("to ENABLE this.Pls move Rythem to /Applications directory");
QMessageBox::information(this,
tr("-"),
tr("Please drag to Applications dir first \n\n otherwise creat replace rule will cause crash on MacOS 10.8 (Mountain Lion)"),
QMessageBox::Ok);
}
#endif
// show local ip address in statusbar
// TODO refresh after network changed
QList<QHostAddress> addresses = QNetworkInterface::allAddresses();
QString statusBarStr = tr("Port:8889 Address: ");//TODO
foreach(const QHostAddress &address , addresses){
qDebug()<<address.toString();
if(address == QHostAddress::LocalHost || address.isNull()
)continue;
qDebug()<<address.protocol()<<address.toString();
if(address.protocol() != QAbstractSocket::IPv4Protocol )continue;
qDebug()<<address.toString();
statusBarStr = statusBarStr.append(address.toString()+" ");
}
statusBarStr = statusBarStr.append(tr(" --By AlloyTeam::iptton"));
statusBarStr = statusBarStr.append(tr(" version:")).append(CURRENT_VERSION);
ui->statusBar->showMessage(statusBarStr);
connect(ui->edit_filter,SIGNAL(textChanged(QString)),this,SLOT(onFilterTextChanged(QString)));
}
void MainWindow::onFilterTextChanged(QString filterText){
sortFilterProxyModel->setFilterText(filterText);
}
MainWindow::~MainWindow()
{
RyProxyServer* server = RyProxyServer::instance();
server->close();
delete _jsBridge;
delete ui;
}
void MainWindow::checkNewVersion(){
#ifdef Q_OS_WIN
//toggleCapture();
#endif
checker->check();
}
void MainWindow::createMenus(){
_fileMenu = menuBar()->addMenu(tr("&File"));
_importSessionsAct = _fileMenu->addAction(tr("&import session..."));
_filterNoImagesAct = _fileMenu->addAction(tr("hide image requests"));
_filterNoImagesAct->setCheckable(true);
_filterNo304sAct = _fileMenu->addAction(tr("hide 304s"));
_filterNo304sAct->setCheckable(true);
_filterShowMatchOnlyAct = _fileMenu->addAction(tr("show matching sessions only"));
_filterShowMatchOnlyAct->setCheckable(true);
_hideConnectTunnelAct = _fileMenu->addAction(tr("hide connect tunnels"));
_hideConnectTunnelAct->setCheckable(true);
_requestLimitMenu = menuBar()->addMenu(tr("limit max save request"));
_limitGroup = new QActionGroup(this);
_setMaxRequest15Act = _limitGroup->addAction("15");
_setMaxRequest15Act->setData(QVariant(15));
_setMaxRequest15Act->setCheckable(true);
_setMaxRequest30Act = _limitGroup->addAction("30");
_setMaxRequest30Act->setData(QVariant(30));
_setMaxRequest30Act->setCheckable(true);
_setMaxRequest30Act->setChecked(true);
_setMaxRequest100Act = _limitGroup->addAction("100");
_setMaxRequest100Act->setData(QVariant(100));
_setMaxRequest100Act->setCheckable(true);
_setMaxRequest500Act = _limitGroup->addAction("500");
_setMaxRequest500Act->setData(QVariant(500));
_setMaxRequest500Act->setCheckable(true);
_requestLimitMenu->addAction(_setMaxRequest15Act);
_requestLimitMenu->addAction(_setMaxRequest30Act);
_requestLimitMenu->addAction(_setMaxRequest100Act);
_requestLimitMenu->addAction(_setMaxRequest500Act);
QAction* devAction = _limitGroup->addAction("2");
devAction->setData(QVariant(2));
devAction->setCheckable(true);
_requestLimitMenu->addAction(devAction);
connect(_fileMenu,SIGNAL(triggered(QAction*)),SLOT(onAction(QAction*)));
connect(_requestLimitMenu,SIGNAL(triggered(QAction*)),SLOT(onAction(QAction*)));
connect(_limitGroup,SIGNAL(triggered(QAction*)),SLOT(onAction(QAction*)));
}
void MainWindow::importSessions(){
QFileDialog dialog;
QString fileName = dialog.getOpenFileName(this,tr("select file to open"),"","Archiev Files(*.saz *.zip)");
QuaZip zip(fileName);
if(!zip.open(QuaZip::mdUnzip)){
qDebug()<<"cannot open "<<fileName;
return;
}
QuaZipFileInfo info;
QuaZipFile file(&zip);
RyPipeData_ptr pipeData;
QString name;
for (bool more = zip.goToFirstFile(); more; more = zip.goToNextFile()) {
if (!zip.getCurrentFileInfo(&info)) {
qWarning("testRead(): getCurrentFileInfo(): %d\n", zip.getZipError());
return;
}
if (!file.open(QIODevice::ReadOnly)) {
qWarning("testRead(): file.open(): %d", file.getZipError());
return;
}
name = file.getActualFileName();
QByteArray ba = file.readAll();
if(name.endsWith("_c.txt")){
//pipeData.clear();
pipeData = RyPipeData_ptr(new RyPipeData(0,0));
pipeData->isImported = true;
pipeData->parseRequest(&ba);
onNewPipe(pipeData);
}else if(name.endsWith("_s.txt")){
pipeData->parseResponse(&ba);
onPipeUpdate(pipeData);
pipeData.clear();
}
if (file.getZipError() != UNZ_OK) {
qWarning("testRead(): file.getFileName(): %d", file.getZipError());
return ;
}
file.close();
if (file.getZipError() != UNZ_OK) {
qWarning("testRead(): file.close(): %d", file.getZipError());
return ;
}
}
zip.close();
}
void MainWindow::onPipeUpdate(RyPipeData_ptr pipeData){
//qDebug()<<"updated "<<pipeData->responseStatus;
//pipeTableModel->updateItem(pipeData);
sortFilterProxyModel->updateItem(pipeData);
}
void MainWindow::onNewPipe(RyPipeData_ptr pipeData){
//pipeTableModel
//pipeTableModel->addItem(pipeData);
sortFilterProxyModel->addItem(pipeData);
}
void MainWindow::onSelectionChange(QItemSelection selected,QItemSelection){
if(selected.indexes().isEmpty()){
ui->actionWaterfall->setEnabled(false);
}else{
if(ui->toolTabs->currentWidget() == ui->inspectorTab){
QModelIndex index = selected.indexes().first();
onItemDoubleClicked(index);
}
ui->actionWaterfall->setEnabled(true);
}
}
void MainWindow::onItemDoubleClicked(QModelIndex topLeft){
RyPipeData_ptr data = sortFilterProxyModel->getItem(topLeft);
if(data.isNull()){
qDebug()<<QString("isNull %1").arg(topLeft.row());
return;
}
ui->toolTabs->setCurrentWidget(ui->inspectorTab);
ui->requestInspectorTabs->setCurrentWidget(ui->requestInspectorTextview);
// TODO: test
ui->requestTextEdit->setPlainText(data->requestHeaderRawData() +"\r\n\r\n"+data->requestBodyRawData() );
QByteArray encoding("UTF-8");
QString contentType = data->getResponseHeader("Content-Type");
int encodingIndex = contentType.indexOf("charset=");
if(encodingIndex!=-1){
encoding.clear();
encoding.append(contentType.mid(encodingIndex + 8));
}
bool isEncrypted = !data->getResponseHeader("Content-Encoding").isEmpty();
QByteArray decrypedData =
(data->isResponseChunked()?
data->responseBodyRawDataUnChunked()
:data->responseBodyRawData());
if(isEncrypted){
decrypedData = gzipDecompress(decrypedData);
}
//TODO: gif
//QMovie *movie = ui->label->movie();
//if(movie){
// delete movie;
//}
if(data->getResponseHeader("Content-Type").toLower().indexOf("image")!=-1){
if(!decrypedData.isEmpty()){
if(data->getResponseHeader("Content-Type").toLower().indexOf("gif")!=-1){
//QBuffer *data = new QBuffer(&decrypedData);
//QMovie *movie = new QMovie(data);
//ui->label->setMovie(movie);
//TODO: start and delete movie..
}else{
QPixmap pixmap;
pixmap.loadFromData(decrypedData);
ui->label->setPixmap(pixmap);
}
ui->responseInspectorTabs->setCurrentWidget(ui->responseInspectorImageView);
}else{
ui->responseInspectorTabs->setCurrentWidget(ui->responseInspectorTextView);
ui->label->clear();
}
}else{
ui->label->clear();
ui->responseInspectorTabs->setCurrentWidget(ui->responseInspectorTextView);
}
// show in textview
//TODO for gbk
ui->responseTextEdit->setPlainText(QString((
data->responseHeaderRawData()
+"\r\n\r\n"
+ decrypedData
).data())
);
data.clear();
}
void MainWindow::onMessageFromOtherInstance(){
setWindowState( (windowState() & ~Qt::WindowMinimized) | Qt::WindowActive);
raise();
activateWindow();
}
void MainWindow::onWaterfallActionTriggered(){
QModelIndexList list = ui->tableView->selectionModel()->selectedRows();
QList<RyPipeData_ptr> pipes;
QListIterator<QModelIndex> it(list);
while(it.hasNext()){
QModelIndex i = it.next();
RyPipeData_ptr data = sortFilterProxyModel->getItem(i);
pipes.append(data);
}
if(pipes.length()){
//deleteLater would be call when win close, don't worry about the memory
WaterfallWindow *win = new WaterfallWindow();
win->setPipeData(pipes);
win->show();
}
else{
QMessageBox box;
box.setText("Please select some network request first");
box.exec();
}
}
void MainWindow::toggleProxy(){
QMutexLocker locker(&proxyMutex);
#ifdef Q_OS_MAC
bool setProxySuccess = false;
if(_isUsingCapture){
if(_previousProxyInfo.isUsingPac == "1"){
setProxySuccess = setPAC(_previousProxyInfo.pacUrl);
}else{
setProxySuccess = disableMyPac();
}
}else{
if(isFirstTimeToggle){
isFirstTimeToggle = false;
CFDictionaryRef proxies = SCDynamicStoreCopyProxies(NULL);
//CFShow(proxies);
int isPacEnabled = 0;
//CFStringRef CFPacUrl = (CFStringRef)CFDictionaryGetValue(proxies, kSCPropNetProxiesProxyAutoConfigURLString);
//if(CFPacUrl){
// _previousProxyInfo.pacUrl = QString::fromUtf8( myCFStringCStringPtr(CFPacUrl,kCFStringEncodingUTF8) );
//}
if (proxies){
CFNumberRef tmpRef;
//kSCPropNetProxiesHTTPSProxy
if ((tmpRef = (CFNumberRef)CFDictionaryGetValue(proxies, kSCPropNetProxiesProxyAutoConfigEnable))){
CFNumberRef pacEnabled;
CFNumberGetValue(tmpRef, CFNumberGetType(tmpRef), &pacEnabled);
//if(!pacEnabled){
// CFNumberGetValue(tmpRef, kCFNumberSInt32Type, &pacEnabled);
//}
if (pacEnabled){
isPacEnabled = 1;
_previousProxyInfo.isUsingPac = QString("1");
CFStringRef CFPacUrl = (CFStringRef)CFDictionaryGetValue(proxies, kSCPropNetProxiesProxyAutoConfigURLString);
_previousProxyInfo.pacUrl = QString::fromUtf8( myCFStringCStringPtr(CFPacUrl,kCFStringEncodingUTF8) );
}
}
}
//qDebug()<<QString("pacEnabeld %1").arg(isPacEnabled);
if(isPacEnabled){
qDebug()<<_previousProxyInfo.pacUrl;
ProxyAutoConfig::instance()->setConfigByUrl(_previousProxyInfo.pacUrl);
}else{
CFNumberRef httpEnabled = (CFNumberRef)CFDictionaryGetValue(proxies, kSCPropNetProxiesHTTPEnable);
CFNumberRef httpsEnabled = (CFNumberRef)CFDictionaryGetValue(proxies, kSCPropNetProxiesHTTPSEnable);
int tmp;
if (httpEnabled && CFNumberGetValue(httpEnabled, kCFNumberIntType, &tmp) && tmp){
CFStringRef host = (CFStringRef)CFDictionaryGetValue(proxies, kSCPropNetProxiesHTTPProxy);
CFNumberRef port = (CFNumberRef)CFDictionaryGetValue(proxies, kSCPropNetProxiesHTTPPort);
QString hostQ = QString::fromUtf8( myCFStringCStringPtr(host,kCFStringEncodingUTF8) );
UInt64 portQ=20;
CFNumberGetValue(port, kCFNumberSInt64Type, &portQ);
QString proxyStr = QString("Proxy %1:%2").arg(hostQ).arg(portQ);
//qDebug()<<proxyStr;
ProxyAutoConfig::instance()->setHttpProxy(proxyStr);
}
if (httpsEnabled && CFNumberGetValue(httpsEnabled, kCFNumberIntType, &tmp) && tmp){
CFStringRef host = (CFStringRef)CFDictionaryGetValue(proxies, kSCPropNetProxiesHTTPSProxy);
CFNumberRef port = (CFNumberRef)CFDictionaryGetValue(proxies, kSCPropNetProxiesHTTPSPort);
QString hostQ = QString::fromUtf8( myCFStringCStringPtr(host,kCFStringEncodingUTF8) );
UInt64 portQ=20;
CFNumberGetValue(port, kCFNumberSInt64Type, &portQ);
QString proxyStr = QString("Proxy %1:%2").arg(hostQ).arg(portQ);
ProxyAutoConfig::instance()->setHttpProxy(proxyStr);
}
}
CFRelease(proxies);
}
setProxySuccess = setMyPAC();
}
if(!setProxySuccess){// hack..
_isUsingCapture = !_isUsingCapture;
}
#endif
#ifdef Q_WS_WIN32
RyWinHttp::init();
if(_isUsingCapture){
ui->ActionCapture->setText(tr("start capture"));
proxySetting.setValue("ProxyEnable",_previousProxyInfo.enable);
if(_previousProxyInfo.enable != 0){
proxySetting.setValue("ProxyServer",_previousProxyInfo.proxyString);
}
if( _previousProxyInfo.isUsingPac != "0"){
proxySetting.setValue("AutoConfigURL",_previousProxyInfo.isUsingPac);
}
}else{
ui->ActionCapture->setText(tr("stop capture"));
_previousProxyInfo.isUsingPac = proxySetting.value("AutoConfigURL","0").toString();
_previousProxyInfo.enable = proxySetting.value("ProxyEnable").toInt();
_previousProxyInfo.proxyString =proxySetting.value("ProxyServer").toString();
QString proxyServer="127.0.0.1:8889";
proxySetting.remove("AutoConfigURL");
proxySetting.setValue("ProxyEnable",QVariant(1));
proxySetting.setValue("ProxyServer",proxyServer);
}
if(!setProxy(proxySetting)){// hack..
_isUsingCapture = !_isUsingCapture;
}
#endif
_isUsingCapture = !_isUsingCapture;
if(_isUsingCapture){
ui->ActionCapture->setChecked(true);
ui->ActionCapture->setText(tr("stop capture"));
ui->ActionCapture->setToolTip(tr("stop capture"));
}else{
ui->ActionCapture->setChecked(false);
ui->ActionCapture->setText(tr("start capture"));
ui->ActionCapture->setToolTip(tr("start capture"));
}
}
void MainWindow::toggleCapture(){
ui->ActionCapture->setEnabled(false);
toggleProxy();
ui->ActionCapture->setEnabled(true);
}
void MainWindow::closeEvent(QCloseEvent *event){
if(_isUsingCapture){
toggleCapture();
}
QMainWindow::closeEvent(event);
qDebug()<<"close event";
qApp->quit();
}
void MainWindow::contextMenuEvent(QContextMenuEvent *event){
event->accept();
}
void MainWindow::addJsObject(){
ui->webView->page()->mainFrame()->addToJavaScriptWindowObject(QString("App"),_jsBridge);
}
void MainWindow::onAction(QAction *action){
if(action == _filterNoImagesAct){
if(action->isChecked()){
sortFilterProxyModel->setFilter(sortFilterProxyModel->filter() | RyTableSortFilterProxyModel::NoImageFilter);
}else{
sortFilterProxyModel->setFilter(sortFilterProxyModel->filter() & (~RyTableSortFilterProxyModel::NoImageFilter));
}
}else if(action == _filterNo304sAct){
if(action->isChecked()){
sortFilterProxyModel->setFilter(sortFilterProxyModel->filter() | RyTableSortFilterProxyModel::No304Filter);
}else{
sortFilterProxyModel->setFilter(sortFilterProxyModel->filter() & (~RyTableSortFilterProxyModel::No304Filter));
}
}else if(action == _importSessionsAct){
importSessions();
}else if(action == _filterShowMatchOnlyAct){
if(action->isChecked()){
sortFilterProxyModel->setFilter(sortFilterProxyModel->filter() | RyTableSortFilterProxyModel::OnlyMatchingFilter);
}else{
sortFilterProxyModel->setFilter(sortFilterProxyModel->filter() & (~RyTableSortFilterProxyModel::OnlyMatchingFilter));
}
}else if(action == _hideConnectTunnelAct){
if(action->isChecked()){
sortFilterProxyModel->setFilter(sortFilterProxyModel->filter() | RyTableSortFilterProxyModel::HideTunnelFilter);
}else{
sortFilterProxyModel->setFilter(sortFilterProxyModel->filter() & (~RyTableSortFilterProxyModel::HideTunnelFilter));
}
}else if(action->actionGroup() == _limitGroup){
QVariant data = action->data();
bool ok;
int limit = data.toInt(&ok);
qDebug()<<"action = group "<<(ok?"ok":"not ok");
if(ok){
this->sortFilterProxyModel->setMaxRequestSize(limit);
}
}
}
void MainWindow::on_actionLongCache_triggered(){
RyRuleManager::instance()->toggleLongCache();
}
void MainWindow::onActionRemoveAll(){
pipeTableModel->removeAllItem();
sortFilterProxyModel->removeAllItem();
ui->actionWaterfall->setEnabled(false);
}
void MainWindow::on_actionDebug_triggered(){
qDebug()<<RyRuleManager::instance()->toJson(true);
}
void MainWindow::on_actionCheckNew_triggered(){
checkNewVersion();
}
void MainWindow::loadConfigPage(){
ui->webView->load(QUrl("http://127.0.0.1:8889/manager/RythemManagerUI/rules.html"));
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化