1
0

resultWebAuth.cpp 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842
  1. #include "./resultWebAuth.h"
  2. #include "./common.h"
  3. #include "./commandWebAuth.h"
  4. #include "./loginData.h"
  5. #include "./dataAddress.h"
  6. #include "./loginCredentials.h"
  7. #include "./loginBox.h"
  8. #include "./loginProvider.h"
  9. #include "./loginStatus.h"
  10. #include "./externalMngr.h"
  11. #include "./browserWindow.h"
  12. #include "../resource.h"
  13. #include "../api.h"
  14. #include "../api_auth.h"
  15. #include "../../omBrowser/obj_ombrowser.h"
  16. #include "../../omBrowser/ifc_ombrowserwndenum.h"
  17. #include "../../omBrowser/ifc_ombrowserwndmngr.h"
  18. #include "../../omBrowser/browserUiCommon.h"
  19. #include "../../winamp/jsapi.h"
  20. #include "../../winamp/jsapi_CallbackParameters.h"
  21. #include <api/service/waservicefactory.h>
  22. #include <strsafe.h>
  23. #define SERVICE_ID 369
  24. #define DISPTABLE_CLASS LoginResultWebAuth
  25. DISPTABLE_BEGIN()
  26. DISPENTRY_ADD(DISPID_LOGINCOMPLETE, L"loginComplete", OnLoginComplete)
  27. DISPENTRY_ADD(DISPID_GETPAGERECT, L"getPageRect", OnGetPageRect)
  28. DISPENTRY_ADD(DISPID_GETBOXRECT, L"getBoxRect", OnGetBoxRect)
  29. DISPENTRY_ADD(DISPID_SETSTATUS, L"setStatus", OnSetStatus)
  30. DISPENTRY_ADD(DISPID_NAVIGATE, L"navigate", OnNavigate)
  31. DISPENTRY_ADD(DISPID_GETSTRING, L"getString", OnGetString)
  32. DISPTABLE_END
  33. #undef DISPTABLE_CLASS
  34. LoginResultWebAuth::LoginResultWebAuth(obj_ombrowser *pManager, LPCWSTR pszTargetUrl, LoginData *pInput, Callback fnCallback, void *pUser)
  35. : ref(1), browserMngr(pManager), external(NULL), targetUrl(NULL), input(pInput), callback(fnCallback), user(pUser),
  36. credentials(NULL), authCode(AUTH_NOT_AUTHORIZED), hView(NULL), completed(NULL),
  37. dispId(DISPID_UNKNOWN), connectionVerified(FALSE), readyUrl(NULL)
  38. {
  39. InitializeCriticalSection(&lock);
  40. targetUrl = LoginBox_CopyString(pszTargetUrl);
  41. input->AddRef();
  42. browserMngr->AddRef();
  43. if(SUCCEEDED(ExternalManager::CreateInstance(&external)))
  44. external->AddDispatch(L"loginBox", this, &dispId);
  45. }
  46. LoginResultWebAuth::~LoginResultWebAuth()
  47. {
  48. hView = NULL;
  49. Finish();
  50. if (NULL != completed)
  51. {
  52. CloseHandle(completed);
  53. }
  54. LoginBox_FreeString(targetUrl);
  55. input->Release();
  56. if (NULL != credentials)
  57. credentials->Release();
  58. LoginBox_FreeString(readyUrl);
  59. DeleteCriticalSection(&lock);
  60. }
  61. HRESULT LoginResultWebAuth::InitBrowserManager(obj_ombrowser **browserMngr)
  62. {
  63. HRESULT hr;
  64. waServiceFactory *sf = WASABI_API_SVC->service_getServiceByGuid(OBJ_OmBrowser);
  65. if (NULL == sf) return E_UNEXPECTED;
  66. *browserMngr = (obj_ombrowser*)sf->getInterface();
  67. if (NULL == browserMngr)
  68. hr = E_UNEXPECTED;
  69. else
  70. {
  71. HWND hWinamp = (NULL != WASABI_API_WINAMP) ? WASABI_API_WINAMP->GetMainWindow() : NULL;
  72. if (NULL == hWinamp || FAILED((*browserMngr)->Initialize(NULL, hWinamp)))
  73. {
  74. (*browserMngr)->Release();
  75. hr = E_FAIL;
  76. }
  77. else
  78. {
  79. hr = S_OK;
  80. }
  81. }
  82. sf->Release();
  83. return hr;
  84. }
  85. HRESULT LoginResultWebAuth::CreateInstance(LPCWSTR targetUrl, LoginData *input, Callback callback, void *user, LoginResultWebAuth **instance)
  86. {
  87. if (NULL == instance) return E_POINTER;
  88. *instance = NULL;
  89. if (NULL == targetUrl || NULL == input)
  90. return E_INVALIDARG;
  91. obj_ombrowser *browserMngr;
  92. if (FAILED(InitBrowserManager(&browserMngr)))
  93. return E_FAIL;
  94. HRESULT hr;
  95. LoginResultWebAuth *result = new LoginResultWebAuth(browserMngr, targetUrl, input, callback, user);
  96. if (NULL == result)
  97. hr = E_OUTOFMEMORY;
  98. else
  99. {
  100. if (DISPID_UNKNOWN == result->dispId)
  101. hr = E_FAIL;
  102. else
  103. hr = result->Start();
  104. if (FAILED(hr))
  105. {
  106. result->Release();
  107. result = NULL;
  108. }
  109. else
  110. {
  111. *instance = result;
  112. }
  113. }
  114. browserMngr->Release();
  115. return hr;
  116. }
  117. STDMETHODIMP_(ULONG) LoginResultWebAuth::AddRef(void)
  118. {
  119. return InterlockedIncrement((LONG*)&ref);
  120. }
  121. STDMETHODIMP_(ULONG) LoginResultWebAuth::Release(void)
  122. {
  123. if (0 == ref)
  124. return ref;
  125. LONG r = InterlockedDecrement((LONG*)&ref);
  126. if (0 == r)
  127. delete(this);
  128. return r;
  129. }
  130. STDMETHODIMP LoginResultWebAuth::QueryInterface(REFIID riid, PVOID *ppvObject)
  131. {
  132. if (NULL == ppvObject)
  133. return E_POINTER;
  134. if (IsEqualIID(riid, IFC_OmService))
  135. *ppvObject = static_cast<ifc_omservice*>(this);
  136. else if (IsEqualIID(riid, IFC_OmServiceCommand))
  137. *ppvObject = static_cast<ifc_omservicecommand*>(this);
  138. else if (IsEqualIID(riid, LCUID_WEBAUTH))
  139. *ppvObject = static_cast<LoginResultWebAuth*>(this);
  140. else if (IsEqualIID(riid, IID_IDispatch))
  141. *ppvObject = static_cast<IDispatch*>(this);
  142. else if (IsEqualIID(riid, IID_IUnknown))
  143. *ppvObject = static_cast<IUnknown*>(reinterpret_cast<LoginResult*>(this));
  144. else
  145. {
  146. *ppvObject = NULL;
  147. return E_NOINTERFACE;
  148. }
  149. if (NULL == *ppvObject)
  150. return E_UNEXPECTED;
  151. AddRef();
  152. return S_OK;
  153. }
  154. size_t LoginResultWebAuth::Wasabi_AddRef()
  155. {
  156. return AddRef();
  157. }
  158. size_t LoginResultWebAuth::Wasabi_Release()
  159. {
  160. return Release();
  161. }
  162. int LoginResultWebAuth::Wasabi_QueryInterface(GUID iid, void **object)
  163. {
  164. return QueryInterface(iid, object);
  165. }
  166. HRESULT LoginResultWebAuth::Start()
  167. {
  168. input->SetStatus(MAKEINTRESOURCE(IDS_STATUS_INITIALIZING));
  169. HWND hParent = input->GetPage();
  170. UINT viewStyle = NBCS_DISABLECONTEXTMENU | NBCS_DIALOGMODE | NBCS_POPUPOWNER |
  171. NBCS_DISABLEFULLSCREEN | NBCS_DISABLEHOSTCSS | NBCS_NOTOOLBAR | NBCS_NOSTATUSBAR;
  172. HRESULT hr = browserMngr->CreateView(this, hParent, NULL, viewStyle, &hView);
  173. if (FAILED(hr))
  174. return hr;
  175. if (FALSE == BrowserWindow_Attach(hView, this))
  176. {
  177. DestroyWindow(hView);
  178. hView = NULL;
  179. hr = E_FAIL;
  180. }
  181. else
  182. {
  183. SetWindowPos(hView, HWND_BOTTOM, 0, 0, 0, 0, SWP_NOACTIVATE);
  184. }
  185. return hr;
  186. }
  187. HRESULT LoginResultWebAuth::Finish()
  188. {
  189. EnterCriticalSection(&lock);
  190. if (NULL != external)
  191. {
  192. if (DISPID_UNKNOWN != dispId)
  193. {
  194. external->DeleteMemberByDispID(dispId);
  195. dispId = DISPID_UNKNOWN;
  196. }
  197. external->Release();
  198. external = NULL;
  199. }
  200. // try to destroy windows before browserMngr->Finish() (this will let them to lock less)
  201. if (NULL != hView)
  202. {
  203. DestroyWindow(hView);
  204. hView = NULL;
  205. }
  206. if (NULL != browserMngr)
  207. {
  208. ifc_ombrowserwndmngr *windowMngr;
  209. if (SUCCEEDED(browserMngr->QueryInterface(IFC_OmBrowserWindowManager, (void**)&windowMngr)))
  210. {
  211. ifc_ombrowserwndenum *windowEnum;
  212. if (SUCCEEDED(windowMngr->Enumerate(NULL, NULL, &windowEnum)))
  213. {
  214. HWND hBrowser;
  215. while(S_OK == windowEnum->Next(1, &hBrowser, NULL))
  216. {
  217. if (FALSE != DestroyWindow(hBrowser) && hBrowser == hView)
  218. hView = NULL;
  219. }
  220. windowEnum->Release();
  221. }
  222. windowMngr->Release();
  223. }
  224. }
  225. if (NULL != browserMngr)
  226. {
  227. browserMngr->Finish();
  228. browserMngr->Release();
  229. browserMngr = NULL;
  230. }
  231. LeaveCriticalSection(&lock);
  232. return S_OK;
  233. }
  234. void LoginResultWebAuth::NotifyComplete()
  235. {
  236. Callback callbackCopy(NULL);
  237. HANDLE completedCopy(NULL);
  238. INT statusId;
  239. switch(authCode)
  240. {
  241. case AUTH_SUCCESS: statusId = IDS_STATUS_SUCCEEDED; break;
  242. case AUTH_SECURID: statusId = IDS_STATUS_PASSCODE_REQUIRED; break;
  243. case AUTH_ABORT: statusId = IDS_STATUS_ABORTED; break;
  244. default: statusId = IDS_STATUS_FAILED; break;
  245. }
  246. input->SetStatus(MAKEINTRESOURCE(statusId));
  247. EnterCriticalSection(&lock);
  248. callbackCopy = callback;
  249. if (NULL == completed || FALSE == DuplicateHandle(GetCurrentProcess(), completed,
  250. GetCurrentProcess(), &completedCopy, 0, FALSE, DUPLICATE_SAME_ACCESS))
  251. {
  252. completedCopy = NULL;
  253. }
  254. LeaveCriticalSection(&lock);
  255. if (NULL != completedCopy)
  256. {
  257. SetEvent(completedCopy);
  258. CloseHandle(completedCopy);
  259. }
  260. if (NULL != callbackCopy)
  261. callbackCopy(this);
  262. }
  263. HRESULT LoginResultWebAuth::GetWaitHandle(HANDLE *handle)
  264. {
  265. if (NULL == handle)
  266. return E_POINTER;
  267. HRESULT hr = S_OK;
  268. EnterCriticalSection(&lock);
  269. if (NULL == completed)
  270. {
  271. completed = CreateEvent(NULL, TRUE, (S_OK == IsCompleted()), NULL);
  272. if (NULL == completed)
  273. {
  274. *handle = NULL;
  275. DWORD error = GetLastError();
  276. hr = HRESULT_FROM_WIN32(error);
  277. }
  278. }
  279. if (SUCCEEDED(hr) &&
  280. FALSE == DuplicateHandle(GetCurrentProcess(), completed,
  281. GetCurrentProcess(), handle, 0, FALSE, DUPLICATE_SAME_ACCESS))
  282. {
  283. *handle = NULL;
  284. DWORD error = GetLastError();
  285. hr = HRESULT_FROM_WIN32(error);
  286. }
  287. LeaveCriticalSection(&lock);
  288. return hr;
  289. }
  290. HRESULT LoginResultWebAuth::GetUser(void **pUser)
  291. {
  292. if (NULL == pUser) return E_POINTER;
  293. *pUser = user;
  294. return S_OK;
  295. }
  296. HRESULT LoginResultWebAuth::RequestAbort(BOOL fDrop)
  297. {
  298. HRESULT hr;
  299. EnterCriticalSection(&lock);
  300. authCode = AUTH_ABORT;
  301. if (NULL == browserMngr)
  302. hr = S_FALSE;
  303. else
  304. {
  305. input->SetStatus(MAKEINTRESOURCE(IDS_STATUS_ABORTING));
  306. Finish();
  307. hr = S_OK;
  308. }
  309. if (NULL != credentials)
  310. {
  311. credentials->Release();
  312. credentials = NULL;
  313. }
  314. LeaveCriticalSection(&lock);
  315. if (S_OK == hr)
  316. NotifyComplete();
  317. return hr;
  318. }
  319. HRESULT LoginResultWebAuth::IsCompleted()
  320. {
  321. return (NULL == browserMngr) ? S_OK : S_FALSE;
  322. }
  323. HRESULT LoginResultWebAuth::IsAborting()
  324. {
  325. return E_NOTIMPL;
  326. }
  327. HRESULT LoginResultWebAuth::GetLoginData(LoginData **loginData)
  328. {
  329. if (NULL == loginData) return E_POINTER;
  330. *loginData = input;
  331. if (NULL != input)
  332. input->AddRef();
  333. return S_OK;
  334. }
  335. STDMETHODIMP_(void) LoginResultWebAuth::Event_BrowserReady(HWND hBrowser)
  336. {
  337. input->SetStatus(MAKEINTRESOURCE(IDS_STATUS_CONNECTING));
  338. }
  339. STDMETHODIMP_(void) LoginResultWebAuth::Event_DocumentReady(HWND hBrowser)
  340. {
  341. BOOL validateDocument = TRUE;
  342. EnterCriticalSection(&lock);
  343. if (NULL != readyUrl)
  344. {
  345. if (CSTR_EQUAL == CompareString(CSTR_INVARIANT, NORM_IGNORECASE, readyUrl, -1, L"loginbox:finish", -1))
  346. {
  347. if (FAILED(BrowserWindow_QueueApc(hView, 0L)))
  348. {
  349. Finish();
  350. NotifyComplete();
  351. }
  352. return;
  353. }
  354. connectionVerified = FALSE;
  355. BrowserControl_Navigate(hBrowser, NAVIGATE_STOP, FALSE);
  356. if (FALSE != BrowserControl_Navigate(hBrowser, readyUrl, TRUE))
  357. validateDocument = FALSE;
  358. LoginBox_FreeString(readyUrl);
  359. readyUrl = NULL;
  360. }
  361. LeaveCriticalSection(&lock);
  362. if (FALSE != validateDocument && FALSE == connectionVerified)
  363. {
  364. // time to kill it
  365. authCode = AUTH_404;
  366. Finish();
  367. NotifyComplete();
  368. }
  369. }
  370. STDMETHODIMP_(void) LoginResultWebAuth::Event_BrowserClosing(HWND hBrowser)
  371. {
  372. }
  373. STDMETHODIMP_(void) LoginResultWebAuth::Event_InvokeApc(HWND hBrowser, LPARAM param)
  374. {
  375. Finish();
  376. NotifyComplete();
  377. }
  378. unsigned int LoginResultWebAuth::GetId()
  379. {
  380. return SERVICE_ID;
  381. }
  382. HRESULT LoginResultWebAuth::GetName(wchar_t *pszBuffer, int cchBufferMax)
  383. {
  384. if (NULL == pszBuffer)
  385. return E_POINTER;
  386. WCHAR szName[128] = {0}, szTemplate[256] = {0};
  387. WASABI_API_LNGSTRINGW_BUF(IDS_WEBAUTH_CAPTION_TEMPLATE, szTemplate, ARRAYSIZE(szTemplate));
  388. LoginProvider *loginProvider;
  389. if (SUCCEEDED(input->GetProvider(&loginProvider)) && NULL != loginProvider)
  390. {
  391. loginProvider->GetName(szName, ARRAYSIZE(szName));
  392. loginProvider->Release();
  393. }
  394. return StringCchPrintf(pszBuffer, cchBufferMax, szTemplate, szName);
  395. }
  396. HRESULT LoginResultWebAuth::GetUrl(wchar_t *pszBuffer, int cchBufferMax)
  397. {
  398. input->SetStatus(MAKEINTRESOURCE(IDS_STATUS_CONNECTING));
  399. LPWSTR cursor;
  400. size_t remaining;
  401. HRESULT hr = StringCchCopyEx(pszBuffer, cchBufferMax, targetUrl, &cursor, &remaining, STRSAFE_IGNORE_NULLS);
  402. if (FAILED(hr)) return hr;
  403. LoginDataAddress *inputAddress;
  404. if (SUCCEEDED(input->QueryInterface(IID_LoginDataAddress, (void**)&inputAddress)))
  405. {
  406. LPCWSTR pszAddress = inputAddress->GetAddress();
  407. if (NULL != pszAddress && L'\0' != *pszAddress)
  408. {
  409. LPWSTR paramBegin = cursor;
  410. while (L'?' != *paramBegin && paramBegin != pszBuffer)
  411. paramBegin = CharPrev(pszBuffer, paramBegin);
  412. WCHAR separator = (L'?' == *paramBegin) ? L'&' : L'?';
  413. hr = StringCchPrintfEx(cursor, remaining, &cursor, &remaining, 0, L"%cuserUrl=%s", separator, pszAddress);
  414. }
  415. inputAddress->Release();
  416. }
  417. return hr;
  418. }
  419. HRESULT LoginResultWebAuth::GetExternal(IDispatch **ppDispatch)
  420. {
  421. if (NULL == ppDispatch)
  422. return E_POINTER;
  423. *ppDispatch = external;
  424. if (NULL != external)
  425. external->AddRef();
  426. return S_OK;
  427. }
  428. HRESULT LoginResultWebAuth::QueryState(HWND hBrowser, const GUID *commandGroup, UINT commandId)
  429. {
  430. if (NULL == commandGroup)
  431. return E_NOTIMPL;
  432. if (IsEqualGUID(*commandGroup, CMDGROUP_NAVIGATION))
  433. {
  434. switch(commandId)
  435. {
  436. case NAVCOMMAND_HOME:
  437. return CMDSTATE_DISABLED;
  438. }
  439. }
  440. return E_NOTIMPL;
  441. }
  442. HRESULT LoginResultWebAuth::Exec(HWND hBrowser, const GUID *commandGroup, UINT commandId, ULONG_PTR commandArg)
  443. {
  444. return E_NOTIMPL;
  445. }
  446. HRESULT LoginResultWebAuth::DispParamsToCredentials(WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, UINT FAR *puArgErr, HRESULT *errorEx, LPWSTR *finishUrl)
  447. {
  448. if (NULL != finishUrl)
  449. *finishUrl = NULL;
  450. JSAPI_VERIFY_METHOD(wFlags);
  451. JSAPI_VERIFY_PARAMCOUNT_OPTIONAL(pdispparams, 5, 6);
  452. BSTR username;
  453. BSTR token;
  454. BSTR sessionKey;
  455. __time64_t expire;
  456. JSAPI_GETUNSIGNED_AS_NUMBER(authCode, pdispparams, 1, puArgErr);
  457. JSAPI_GETSTRING(token, pdispparams, 2, puArgErr);
  458. JSAPI_GETSTRING(sessionKey, pdispparams, 3, puArgErr);
  459. JSAPI_GETUNSIGNED_AS_NUMBER(expire, pdispparams, 4, puArgErr);
  460. JSAPI_GETSTRING(username, pdispparams, 5, puArgErr);
  461. expire += _time64(0);
  462. if (NULL != finishUrl && pdispparams->cArgs > 5)
  463. {
  464. VARIANTARG *arg = &pdispparams->rgvarg[pdispparams->cArgs - 6];
  465. if (NULL != arg && VT_BSTR == arg->vt)
  466. *finishUrl = LoginBox_CopyString(arg->bstrVal);
  467. }
  468. if (NULL != credentials)
  469. {
  470. credentials->Release();
  471. credentials = NULL;
  472. }
  473. if (AUTH_SUCCESS == authCode)
  474. {
  475. GUID realm;
  476. HRESULT hr = input->GetRealm(&realm);
  477. if (SUCCEEDED(hr))
  478. {
  479. LPSTR sessionAnsi(NULL), tokenAnsi(NULL);
  480. if (FAILED(LoginBox_WideCharToMultiByte(CP_UTF8, 0, sessionKey, -1, NULL, NULL, &sessionAnsi)) ||
  481. FAILED(LoginBox_WideCharToMultiByte(CP_UTF8, 0, token, -1, NULL, NULL, &tokenAnsi)))
  482. {
  483. hr = E_FAIL;
  484. }
  485. if (SUCCEEDED(hr))
  486. hr = LoginCredentials::CreateInstance(&realm, username, sessionAnsi, tokenAnsi, expire, &credentials);
  487. LoginBox_FreeAnsiString(sessionAnsi);
  488. LoginBox_FreeAnsiString(tokenAnsi);
  489. }
  490. if (FAILED(hr))
  491. {
  492. if (NULL != errorEx)
  493. *errorEx = hr;
  494. }
  495. }
  496. return S_OK;
  497. }
  498. HRESULT LoginResultWebAuth::OnLoginComplete(WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, unsigned int FAR *puArgErr)
  499. {
  500. JSAPI_INIT_RESULT(pvarResult, VT_BOOL);
  501. HRESULT hr, errorEx;
  502. BOOL finishLogin = TRUE;
  503. EnterCriticalSection(&lock);
  504. LPWSTR finishUrl;
  505. hr = DispParamsToCredentials(wFlags, pdispparams, pvarResult, puArgErr, &errorEx, &finishUrl);
  506. JSAPI_SET_RESULT(pvarResult, boolVal, (SUCCEEDED(hr) && SUCCEEDED(errorEx)) ? VARIANT_TRUE : VARIANT_FALSE);
  507. if (NULL != finishUrl && L'\0' != *finishUrl)
  508. {
  509. LoginBox_FreeString(readyUrl);
  510. readyUrl = LoginBox_CopyString(L"loginbox:finish");
  511. if (NULL != readyUrl)
  512. {
  513. if (FALSE == hView || FALSE == BrowserControl_Navigate(hView, finishUrl, FALSE))
  514. {
  515. LoginBox_FreeString(readyUrl);
  516. readyUrl = NULL;
  517. }
  518. else
  519. finishLogin = FALSE;
  520. }
  521. }
  522. LoginBox_FreeString(finishUrl);
  523. LeaveCriticalSection(&lock);
  524. if (FALSE != finishLogin)
  525. {
  526. if (FAILED(BrowserWindow_QueueApc(hView, 0L)))
  527. {
  528. Finish();
  529. NotifyComplete();
  530. }
  531. }
  532. return hr;
  533. }
  534. HRESULT LoginResultWebAuth::OnGetPageRect(WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, unsigned int FAR *puArgErr)
  535. {
  536. JSAPI_VERIFY_METHOD(wFlags);
  537. JSAPI_INIT_RESULT(pvarResult, VT_DISPATCH);
  538. RECT rect;
  539. EnterCriticalSection(&lock);
  540. HWND hPage = input->GetPage();
  541. if (NULL == hPage || FALSE == GetWindowRect(hPage, &rect))
  542. SetRectEmpty(&rect);
  543. LeaveCriticalSection(&lock);
  544. JSAPI::CallbackParameters *params = new JSAPI::CallbackParameters;
  545. params->AddLong(L"x", rect.left);
  546. params->AddLong(L"y", rect.top);
  547. params->AddLong(L"cx", rect.right - rect.left);
  548. params->AddLong(L"cy", rect.bottom - rect.top);
  549. V_DISPATCH(pvarResult) = params;
  550. return S_OK;
  551. }
  552. HRESULT LoginResultWebAuth::OnGetBoxRect(WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, unsigned int FAR *puArgErr)
  553. {
  554. JSAPI_VERIFY_METHOD(wFlags);
  555. JSAPI_INIT_RESULT(pvarResult, VT_DISPATCH);
  556. RECT rect;
  557. EnterCriticalSection(&lock);
  558. HWND hBox = input->GetLoginbox();
  559. if (NULL == hBox || FALSE == GetWindowRect(hBox, &rect))
  560. SetRectEmpty(&rect);
  561. LeaveCriticalSection(&lock);
  562. if (NULL != pvarResult)
  563. {
  564. JSAPI::CallbackParameters *params = new JSAPI::CallbackParameters;
  565. params->AddLong(L"x", rect.left);
  566. params->AddLong(L"y", rect.top);
  567. params->AddLong(L"cx", rect.right - rect.left);
  568. params->AddLong(L"cy", rect.bottom - rect.top);
  569. V_DISPATCH(pvarResult) = params;
  570. }
  571. return S_OK;
  572. }
  573. HRESULT LoginResultWebAuth::OnSetStatus(WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, unsigned int FAR *puArgErr)
  574. {
  575. connectionVerified = TRUE;
  576. JSAPI_VERIFY_METHOD(wFlags);
  577. JSAPI_VERIFY_PARAMCOUNT(pdispparams, 1);
  578. JSAPI_INIT_RESULT(pvarResult, VT_BOOL);
  579. LPCWSTR pszStatus = NULL;
  580. INT paramIndex = pdispparams->cArgs - 1;
  581. VARIANTARG *param = &pdispparams->rgvarg[paramIndex];
  582. switch(param->vt)
  583. {
  584. case VT_I4:
  585. pszStatus = (MAKEINTRESOURCE(param->lVal));
  586. break;
  587. case VT_BSTR:
  588. pszStatus = param->bstrVal;
  589. break;
  590. default:
  591. if (puArgErr) *puArgErr = paramIndex;
  592. return DISP_E_TYPEMISMATCH;
  593. }
  594. HRESULT hr = input->SetStatus(pszStatus);
  595. JSAPI_SET_RESULT(pvarResult, boolVal, (SUCCEEDED(hr) ? VARIANT_TRUE : VARIANT_FALSE));
  596. return S_OK;
  597. }
  598. HRESULT LoginResultWebAuth::OnNavigate(WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, unsigned int FAR *puArgErr)
  599. {
  600. JSAPI_VERIFY_METHOD(wFlags);
  601. JSAPI_VERIFY_PARAMCOUNT_OPTIONAL(pdispparams, 1, 3);
  602. JSAPI_INIT_RESULT(pvarResult, VT_BOOL);
  603. BSTR targetUrl = JSAPI_PARAM_OPTIONAL(pdispparams, 1, bstrVal, NULL);
  604. BSTR callbackUrl = JSAPI_PARAM_OPTIONAL(pdispparams, 2, bstrVal, NULL);
  605. BSTR message = JSAPI_PARAM_OPTIONAL(pdispparams, 3, bstrVal, NULL);
  606. HRESULT hr;
  607. if (NULL == targetUrl || L'\0' == *targetUrl ||
  608. NULL == callbackUrl || L'\0' == *callbackUrl)
  609. {
  610. hr = E_INVALIDARG;
  611. }
  612. else
  613. {
  614. EnterCriticalSection(&lock);
  615. LoginBox_FreeString(readyUrl);
  616. readyUrl = LoginBox_CopyString(callbackUrl);
  617. if (NULL == readyUrl)
  618. hr = E_OUTOFMEMORY;
  619. else
  620. {
  621. connectionVerified = FALSE;
  622. if (NULL == hView || FALSE == BrowserControl_Navigate(hView, targetUrl, FALSE))
  623. {
  624. LoginBox_FreeString(readyUrl);
  625. readyUrl = NULL;
  626. hr = E_FAIL;
  627. }
  628. else
  629. hr = S_OK;
  630. }
  631. LeaveCriticalSection(&lock);
  632. }
  633. JSAPI_SET_RESULT(pvarResult, boolVal, (SUCCEEDED(hr) ? VARIANT_TRUE : VARIANT_FALSE));
  634. return S_OK;
  635. }
  636. HRESULT LoginResultWebAuth::OnGetString(WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, unsigned int FAR *puArgErr)
  637. {
  638. if (0 == ((DISPATCH_METHOD |DISPATCH_PROPERTYGET) &wFlags))
  639. return DISP_E_MEMBERNOTFOUND;
  640. JSAPI_VERIFY_PARAMCOUNT(pdispparams, 1);
  641. JSAPI_INIT_RESULT(pvarResult, VT_BSTR);
  642. UINT stringId;
  643. JSAPI_GETUNSIGNED_AS_NUMBER(stringId, pdispparams, 1, puArgErr);
  644. WCHAR szBuffer[4096] = {0};
  645. BSTR string = SysAllocString(WASABI_API_LNGSTRINGW_BUF(stringId, szBuffer, ARRAYSIZE(szBuffer)));
  646. JSAPI_SET_RESULT(pvarResult, bstrVal, string);
  647. return S_OK;
  648. }
  649. HRESULT LoginResultWebAuth::GetResult(INT *pAuthCode, LoginCredentials **ppCredentials)
  650. {
  651. if (S_OK != IsCompleted())
  652. return E_PENDING;
  653. EnterCriticalSection(&lock);
  654. if (NULL != pAuthCode)
  655. *pAuthCode = authCode;
  656. if (NULL != ppCredentials)
  657. {
  658. *ppCredentials = credentials;
  659. if (NULL != credentials)
  660. credentials->AddRef();
  661. }
  662. LeaveCriticalSection(&lock);
  663. return S_OK;
  664. }
  665. #define CBCLASS LoginResultWebAuth
  666. START_MULTIPATCH;
  667. START_PATCH(MPIID_OMSVC)
  668. M_CB(MPIID_OMSVC, ifc_omservice, ADDREF, Wasabi_AddRef);
  669. M_CB(MPIID_OMSVC, ifc_omservice, RELEASE, Wasabi_Release);
  670. M_CB(MPIID_OMSVC, ifc_omservice, QUERYINTERFACE, Wasabi_QueryInterface);
  671. M_CB(MPIID_OMSVC, ifc_omservice, API_GETID, GetId);
  672. M_CB(MPIID_OMSVC, ifc_omservice, API_GETNAME, GetName);
  673. M_CB(MPIID_OMSVC, ifc_omservice, API_GETURL, GetUrl);
  674. M_CB(MPIID_OMSVC, ifc_omservice, API_GETEXTERNAL, GetExternal);
  675. NEXT_PATCH(MPIID_OMSVCCOMMAND)
  676. M_CB(MPIID_OMSVCCOMMAND, ifc_omservicecommand, ADDREF, Wasabi_AddRef);
  677. M_CB(MPIID_OMSVCCOMMAND, ifc_omservicecommand, RELEASE, Wasabi_Release);
  678. M_CB(MPIID_OMSVCCOMMAND, ifc_omservicecommand, QUERYINTERFACE, Wasabi_QueryInterface);
  679. M_CB(MPIID_OMSVCCOMMAND, ifc_omservicecommand, API_QUERYSTATE, QueryState);
  680. M_CB(MPIID_OMSVCCOMMAND, ifc_omservicecommand, API_EXEC, Exec);
  681. END_PATCH
  682. END_MULTIPATCH;
  683. #undef CBCLASS