khanat-opennel-code/code/nelns/login_system/nel_launcher_windows_ext/WebDlg.cpp

295 lines
7 KiB
C++

// WebDlg.cpp : implementation file
//
#include "stdafx.h"
#include "nel_launcher.h"
#include "nel_launcherDlg.h"
#include "WebDlg.h"
#include "LoginDlg.h"
#include <comdef.h>
#include <mshtml.h>
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// CWebDlg dialog
CWebDlg::CWebDlg(CWnd* pParent /*=NULL*/)
: CDialog(CWebDlg::IDD, pParent)
{
//{{AFX_DATA_INIT(CWebDlg)
// NOTE: the ClassWizard will add member initialization here
//}}AFX_DATA_INIT
}
void CWebDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CWebDlg)
DDX_Control(pDX, IDC_EXPLORER1, m_explore);
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CWebDlg, CDialog)
//{{AFX_MSG_MAP(CWebDlg)
ON_WM_SIZE()
ON_WM_CTLCOLOR()
ON_WM_ERASEBKGND()
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CWebDlg message handlers
BEGIN_EVENTSINK_MAP(CWebDlg, CDialog)
//{{AFX_EVENTSINK_MAP(CWebDlg)
ON_EVENT(CWebDlg, IDC_EXPLORER1, 250 /* BeforeNavigate2 */, OnBeforeNavigate2, VTS_DISPATCH VTS_PVARIANT VTS_PVARIANT VTS_PVARIANT VTS_PVARIANT VTS_PVARIANT VTS_PBOOL)
ON_EVENT(CWebDlg, IDC_EXPLORER1, 259 /* DocumentComplete */, OnDocumentComplete, VTS_DISPATCH VTS_PVARIANT)
ON_EVENT(CWebDlg, IDC_EXPLORER1, 252 /* NavigateComplete2 */, OnNavigateComplete2Explorer1, VTS_DISPATCH VTS_PVARIANT)
//}}AFX_EVENTSINK_MAP
END_EVENTSINK_MAP()
void CWebDlg::OnBeforeNavigate2(LPDISPATCH pDisp, VARIANT FAR* URL, VARIANT FAR* Flags, VARIANT FAR* TargetFrameName, VARIANT FAR* PostData, VARIANT FAR* Headers, BOOL FAR* Cancel)
{
/* CRect r;
GetClientRect(&r);
ClientToScreen(&r);
m_dlgLoading.MoveWindow(r.left+BROWSER_X, r.top+BROWSER_Y, BROWSER_W, BROWSER_H);
m_dlgLoading.ShowWindow(SW_SHOW);*/
// set the user agent to nel_launcher changing the header
CString csHeader(Headers->bstrVal);
if(csHeader.IsEmpty())
{
IWebBrowser2 *pBrowser;
LPDISPATCH pWebDisp;
pDisp->QueryInterface(IID_IWebBrowser2, (void**) &pBrowser);
pBrowser->get_Container(&pWebDisp);
BSTR bstr = SysAllocString(L"User-Agent: nel_launcher\r\n");
Headers->bstrVal = bstr;
pBrowser->Navigate2(URL, Flags, TargetFrameName, PostData, Headers);
if (!pWebDisp)
(*Cancel) = true;
if (pWebDisp)
pWebDisp->Release();
if (pBrowser)
pBrowser->Release();
SysFreeString(bstr);
return;
}
}
VARIANT CWebDlg::ExecuteScript(CString csCode, CString csLanguage)
{
COleVariant varRet;
IHTMLDocument2* pHTMLDocument2;
IHTMLWindow2* pHTMLWindow2;
LPDISPATCH lpDispatch;
lpDispatch = m_explore.GetDocument();
BSTR bstrCode = _bstr_t((const char *)csCode);
BSTR bstrLang = _bstr_t((const char *)csLanguage);
HRESULT hr;
if(lpDispatch)
{
hr = lpDispatch->QueryInterface(IID_IHTMLDocument2, (void**)&pHTMLDocument2);
if(hr == S_OK)
{
hr = pHTMLDocument2->get_parentWindow(&pHTMLWindow2);
if(hr == S_OK)
{
pHTMLWindow2->execScript(bstrCode, bstrLang, &varRet);
pHTMLWindow2->Release();
}
pHTMLDocument2->Release();
}
lpDispatch->Release();
}
return varRet;
}
VARIANT CWebDlg::ExecuteJavascript(CString csCode)
{
return ExecuteScript(csCode, "Javascript");
}
void CWebDlg::OpenUrl(CString csUrl)
{
m_csUrl = csUrl;
m_explore.Navigate(csUrl, NULL, NULL, NULL, NULL);
}
void CWebDlg::OpenUrl(const std::string &url)
{
m_csUrl = url.c_str();
m_explore.Navigate(url.c_str(), NULL, NULL, NULL, NULL);
}
void CWebDlg::OnDocumentComplete(LPDISPATCH pDisp, VARIANT FAR* URL)
{
IHTMLDocument2* pHTMLDocument2;
LPDISPATCH lpDispatch;
lpDispatch = m_explore.GetDocument();
if (lpDispatch)
{
HRESULT hr;
hr = lpDispatch->QueryInterface(IID_IHTMLDocument2, (LPVOID*) &pHTMLDocument2);
lpDispatch->Release();
if (FAILED(hr))
return;
if (pHTMLDocument2 == NULL)
return;
IHTMLElement* pBody;
hr = pHTMLDocument2->get_body(&pBody);
if (FAILED(hr))
return;
if (pBody == NULL)
return;
BSTR bstr;
pBody->get_innerHTML(&bstr);
CString csSourceCode(bstr);
CString csAction;
CString csComment;
SysFreeString(bstr);
pBody->Release();
if(csSourceCode.Find("newlogin") != -1)
{
// Authentication not ok !
APP.m_bAuthGame = FALSE;
APP.m_bAuthWeb = FALSE;
((CNel_launcherDlg*)AfxGetMainWnd())->m_wndTabs.DestroyWindow();
DestroyWindow();
((CNel_launcherDlg*)AfxGetMainWnd())->Login();
return;
}
// ACE: BADDDDDD!!!! the good web page can contains 404 in the cookie because the cookie contains any number we want
/* else if(csSourceCode.Find("404") != -1)
{
// Authentication lost, try to redo the authentication
CLoginDlg dlg;
dlg.AuthWeb();
OpenUrl(m_csUrl);
return;
}
*/ int iFindAction1 = csSourceCode.Find("<!--nel");
int iFindAction2;
int iFindComment1;
int iFindComment2;
if(iFindAction1 != -1)
{
iFindAction1 = csSourceCode.Find('"', iFindAction1);
if(iFindAction1 != -1)
{
iFindAction2 = csSourceCode.Find('"', iFindAction1+1);
if(iFindAction2 != -1)
{
csAction = csSourceCode.Mid(iFindAction1+1, iFindAction2 - iFindAction1-1);
iFindComment1 = iFindAction2 + 2;
if(iFindComment1 != -1)
{
iFindComment2 = csSourceCode.Find("-->", iFindAction1);
if(iFindComment2 != -1)
{
csComment = csSourceCode.Mid(iFindComment1, iFindComment2 - iFindComment1);
csComment.TrimRight();
csComment.TrimLeft();
if(csAction == "launch")
((CNel_launcherDlg*)GetParent())->launch(csComment);
else if(csAction == "patch")
((CNel_launcherDlg*)GetParent())->patch(csComment);
}
}
}
}
}
}
// m_dlgLoading.ShowWindow(SW_HIDE);
}
BOOL CWebDlg::OnInitDialog()
{
CDialog::OnInitDialog();
CRect r;
m_brushBG.CreateSolidBrush(RGB(0, 16, 0));
GetClientRect(&r);
GetDlgItem(IDC_EXPLORER1)->MoveWindow(-2, -2, r.Width()-48, r.Height());
// m_dlgLoading.Create(IDD_LOADING_PAGE, this);
// m_dlgLoading.ShowWindow(SW_HIDE);
// ClientToScreen(&r);
// m_dlgLoading.MoveWindow(r.left+BROWSER_X, r.top+BROWSER_Y, BROWSER_W, BROWSER_H);
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
}
void CWebDlg::OnSize(UINT nType, int cx, int cy)
{
CDialog::OnSize(nType, cx, cy);
}
HBRUSH CWebDlg::OnCtlColor(CDC* pDC, CWnd* pWnd, UINT nCtlColor)
{
HBRUSH hbr = CDialog::OnCtlColor(pDC, pWnd, nCtlColor);
if(nCtlColor == CTLCOLOR_DLG)
{
hbr = m_brushBG;
pDC->SetBkColor(RGB(0, 16, 0));
pDC->SetBkMode(OPAQUE);
}
return hbr;
}
void CWebDlg::OnNavigateComplete2Explorer1(LPDISPATCH pDisp, VARIANT FAR* URL)
{
}
BOOL CWebDlg::OnEraseBkgnd(CDC* pDC)
{
CBrush backBrush(RGB(0, 16, 0));
CBrush* pOldBrush = pDC->SelectObject(&backBrush);
CRect r;
pDC->GetClipBox(&r); // Erase the area needed.
pDC->PatBlt(r.left, r.top, r.Width(), r.Height(), PATCOPY);
pDC->SelectObject(pOldBrush);
return TRUE;
}