bool CSteamProto::Relogin()
{
ptrA token(getStringA("TokenSecret"));
if (mir_strlen(token) <= 0)
return false;
HttpRequest *request = new LogonRequest(token);
HttpResponse *response = request->Send(m_hNetlibUser);
bool success = false;
if (CheckResponse(response))
{
JSONROOT root(response->pData);
if (root != NULL) {
JSONNode *node = json_get(root, "error");
ptrT error(json_as_string(node));
if (!mir_tstrcmpi(error, _T("OK")))
{
node = json_get(root, "umqid");
setString("UMQID", ptrA(mir_u2a(ptrT(json_as_string(node)))));
node = json_get(root, "message");
setDword("MessageID", json_as_int(node));
success = true;
}
}
}
delete request;
delete response;
return success;
}
void CDropbox::DestroyAcceessToken()
{
HttpRequest *request = new HttpRequest(hNetlibUser, REQUEST_POST, DROPBOX_API_URL "/disable_access_token");
mir_ptr<NETLIBHTTPREQUEST> response(request->Send());
delete request;
MCONTACT hContact = CDropbox::GetDefaultContact();
db_unset(NULL, MODULE, "TokenSecret");
if (hContact)
{
if (db_get_w(hContact, MODULE, "Status", ID_STATUS_ONLINE) == ID_STATUS_ONLINE)
db_set_w(hContact, MODULE, "Status", ID_STATUS_OFFLINE);
}
}
void CDropbox::RequestAccountInfo()
{
HttpRequest *request = new HttpRequest(hNetlibUser, REQUEST_GET, DROPBOX_API_URL "/account/info");
request->AddBearerAuthHeader(db_get_sa(NULL, MODULE, "TokenSecret"));
mir_ptr<NETLIBHTTPREQUEST> response(request->Send());
delete request;
MCONTACT hContact = CDropbox::GetDefaultContact();
if (response && response->resultCode == HTTP_STATUS_OK)
{
JSONROOT root(response->pData);
if (root)
{
JSONNODE *node = json_get(root, "referral_link");
if (node)
{
ptrW referral_link = ptrW(json_as_string(node));
db_set_ws(hContact, MODULE, "Homepage", referral_link);
}
node = json_get(root, "display_name");
if (node)
{
ptrW display_name = ptrW(json_as_string(node));
wchar_t *sep = wcsrchr(display_name, L' ');
if (sep)
{
db_set_ws(hContact, MODULE, "LastName", sep + 1);
display_name[wcslen(display_name) - wcslen(sep)] = '\0';
db_set_ws(hContact, MODULE, "FirstName", display_name);
}
else
{
db_set_ws(hContact, MODULE, "FirstName", display_name);
db_unset(hContact, MODULE, "LastName");
}
}
node = json_get(root, "country");
if (node)
{
ptrW isocodeW(json_as_string(node));
ptrA isocode(mir_u2a(isocodeW));
if (!strlen(isocode))
db_unset(hContact, MODULE, "Country");
else
{
char *country = (char *)CallService(MS_UTILS_GETCOUNTRYBYISOCODE, (WPARAM)isocode, 0);
db_set_s(hContact, MODULE, "Country", country);
}
}
node = json_get(root, "quota_info");
JSONNODE *nroot = json_as_node(node);
if (nroot)
{
node = json_get(nroot, "shared");
if (node)
db_set_dw(hContact, MODULE, "SharedQuota", json_as_int(node));
node = json_get(nroot, "normal");
if (node)
db_set_dw(hContact, MODULE, "NormalQuota", json_as_int(node));
node = json_get(nroot, "quota");
if (node)
db_set_dw(hContact, MODULE, "TotalQuota", json_as_int(node));
}
}
}
HandleHttpResponseError(hNetlibUser, response);
}
UINT CDropbox::RequestAcceessTokenAsync(void *owner, void* param)
{
HWND hwndDlg = (HWND)param;
CDropbox *instance = (CDropbox*)owner;
EnableWindow(GetDlgItem(hwndDlg, IDC_AUTHORIZE), FALSE);
SetDlgItemText(hwndDlg, IDC_AUTH_STATUS, TranslateT("in process..."));
if (instance->HasAccessToken())
instance->DestroyAcceessToken();
char requestToken[128];
GetDlgItemTextA(hwndDlg, IDC_REQUEST_CODE, requestToken, SIZEOF(requestToken));
char data[1024];
mir_snprintf(
data,
SIZEOF(data),
"grant_type=authorization_code&code=%s",
requestToken);
HttpRequest *request = new HttpRequest(instance->hNetlibUser, REQUEST_POST, DROPBOX_API_URL "/oauth2/token");
request->AddBasicAuthHeader(DROPBOX_API_KEY, DROPBOX_API_SECRET);
request->AddHeader("Content-Type", "application/x-www-form-urlencoded");
request->pData = mir_strdup(data);
request->dataLength = (int)strlen(data);
mir_ptr<NETLIBHTTPREQUEST> response(request->Send());
delete request;
MCONTACT hContact = instance->GetDefaultContact();
if (response)
{
JSONROOT root(response->pData);
if (root)
{
if (response->resultCode == HTTP_STATUS_OK)
{
JSONNODE *node = json_get(root, "access_token");
ptrA access_token = ptrA(mir_u2a(json_as_string(node)));
db_set_s(NULL, MODULE, "TokenSecret", access_token);
if (hContact)
{
if (db_get_w(hContact, MODULE, "Status", ID_STATUS_OFFLINE) == ID_STATUS_OFFLINE)
db_set_w(hContact, MODULE, "Status", ID_STATUS_ONLINE);
}
instance->RequestAccountInfo();
if (hwndDlg)
SetDlgItemText(hwndDlg, IDC_AUTH_STATUS, TranslateT("you have been authorized"));
/*else
ShowNotification(TranslateT("you have been authorized"), MB_ICONINFORMATION);*/
}
else
{
JSONNODE *node = json_get(root, "error_description");
ptrW error_description(json_as_string(node));
if (hwndDlg)
SetDlgItemText(hwndDlg, IDC_AUTH_STATUS, error_description);
/*else
ShowNotification((wchar_t*)error_description, MB_ICONERROR);*/
}
}
}
else
{
if (hwndDlg)
SetDlgItemText(hwndDlg, IDC_AUTH_STATUS, TranslateT("server does not respond"));
HandleHttpResponseError(instance->hNetlibUser, response);
}
SetDlgItemTextA(hwndDlg, IDC_REQUEST_CODE, "");
return 0;
}
bool
Application::GetResource (const Uri *resourceBase, const Uri *uri,
NotifyFunc notify_cb, EventHandler write_cb,
DownloaderAccessPolicy policy, HttpRequest::Options options,
Cancellable *cancellable, gpointer user_data)
{
if (!uri) {
g_warning ("Passing a null uri to Application::GetResource");
if (notify_cb)
notify_cb (NotifyFailed, GPOINTER_TO_INT(NULL), user_data);
return false;
}
if (get_resource_cb && !uri->IsAbsolute ()) {
ManagedStreamCallbacks stream;
if (!Uri::IsNullOrEmpty (uri)) {
stream = get_resource_cb (resourceBase, uri);
} else {
memset (&stream, 0, sizeof (stream));
}
if (stream.handle) {
if (notify_cb)
notify_cb (NotifyStarted, GPOINTER_TO_INT(NULL), user_data);
if (write_cb) {
char buf[4096];
int offset = 0;
int nread;
if (stream.CanSeek (stream.handle))
stream.Seek (stream.handle, 0, 0);
HttpRequestWriteEventArgs *args = new HttpRequestWriteEventArgs (NULL, 0, 0);
do {
if ((nread = stream.Read (stream.handle, buf, 0, sizeof (buf))) <= 0)
break;
args->SetData (buf);
args->SetOffset (offset);
args->SetCount (nread);
write_cb (this, args, user_data);
offset += nread;
} while (true);
args->unref ();
}
if (notify_cb)
notify_cb (NotifyCompleted, GPOINTER_TO_INT(NULL), user_data);
stream.Close (stream.handle);
return true;
}
}
#if 0
// FIXME: drt 171 and 173 expect this to fail simply because the uri
// begins with a '/', but other drts (like 238) depend on this
// working. I give up.
if (!uri->isAbsolute && uri->path && uri->path[0] == '/') {
if (notify_cb)
notify_cb (NotifyFailed, NULL, user_data);
return false;
}
#endif
//no get_resource_cb or empty stream
HttpRequest *request;
if (!(request = GetDeployment ()->CreateHttpRequest (options))) {
if (notify_cb)
notify_cb (NotifyFailed, GPOINTER_TO_INT(NULL), user_data);
return false;
}
NotifyCtx *ctx = g_new (NotifyCtx, 1);
ctx->user_data = user_data;
ctx->notify_cb = notify_cb;
ctx->write_cb = write_cb;
ctx->request = request;
if (notify_cb) {
request->AddHandler (HttpRequest::ProgressChangedEvent, application_downloader_progress_changed, ctx);
request->AddHandler (HttpRequest::StartedEvent, application_downloader_started, ctx);
}
if (cancellable) {
cancellable->SetCancelFuncAndData (application_downloader_abort, request, ctx);
}
request->AddHandler (HttpRequest::WriteEvent, application_downloader_write, ctx);
request->AddHandler (HttpRequest::StoppedEvent, application_downloader_stopped, ctx);
request->Open ("GET", uri, resourceBase, policy);
request->Send ();
return true;
}
