summaryrefslogtreecommitdiff
path: root/protocols/Twitter/src/utility.h
blob: bc9456127406088d4c082b3d6bc2a7243d1e92ef (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
/*
Copyright © 2009 Jim Porter

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#pragma once

#include "common.h"
#include "http.h"
#include "twitter.h"

template<typename T>
void CreateProtoService(const char *module,const char *service,
	int (__cdecl T::*serviceProc)(WPARAM,LPARAM),T *self)
{
	char temp[MAX_PATH*2];

	mir_snprintf(temp,sizeof(temp),"%s%s",module,service);
	CreateServiceFunctionObj(temp,( MIRANDASERVICEOBJ )*(void**)&serviceProc, self );
}

template<typename T>
void HookProtoEvent(const char* evt, int (__cdecl T::*eventProc)(WPARAM,LPARAM), T *self)
{
	::HookEventObj(evt,(MIRANDAHOOKOBJ)*(void**)&eventProc,self);
}

template<typename T>
HANDLE ForkThreadEx(void (__cdecl T::*thread)(void*),T *self,void *data = 0)
{
	return reinterpret_cast<HANDLE>( mir_forkthreadowner(
		(pThreadFuncOwner)*(void**)&thread,self,data,0));
}

template<typename T>
void ForkThread(void (__cdecl T::*thread)(void*),T *self,void *data = 0)
{
	CloseHandle(ForkThreadEx(thread,self,data));
}

std::string b64encode(const std::string &s);

class mir_twitter : public twitter
{
public:
	mir_twitter() : twitter(), handle_(NULL), httpPOST_(NULL) {}
	void set_handle(HANDLE h)
	{
		handle_ = h;
	}

	// OAuthWebRequest used for all OAuth related queries
	//
	// consumerKey and consumerSecret - must be provided for every call, they identify the application
	// oauthToken and oauthTokenSecret - need to be provided for every call, except for the first token request before authorizing
	// pin - only used during authorization, when the user enters the PIN they received from the twitter website
	std::wstring OAuthWebRequestSubmit( 
		const std::wstring& url, 
		const std::wstring& httpMethod, 
		const OAuthParameters *postData,
		const std::wstring& consumerKey, 
		const std::wstring& consumerSecret, 
		const std::wstring& oauthToken = L"", 
		const std::wstring& oauthTokenSecret = L"", 
		const std::wstring& pin = L""
		);

	std::wstring OAuthWebRequestSubmit( 
		const OAuthParameters& parameters, 
		const std::wstring& url 
		);

	std::wstring UrlGetQuery( const std::wstring& url );

	OAuthParameters BuildSignedOAuthParameters( const OAuthParameters& requestParameters, 
								   const std::wstring& url, 
								   const std::wstring& httpMethod, 
								   const OAuthParameters *postData,
								   const std::wstring& consumerKey, 
								   const std::wstring& consumerSecret,
								   const std::wstring& requestToken, 
								   const std::wstring& requestTokenSecret, 
								   const std::wstring& pin );

	std::wstring BuildQueryString( const OAuthParameters &parameters ) ;
	std::wstring OAuthConcatenateRequestElements( const std::wstring& httpMethod, std::wstring url, const std::wstring& parameters );
	std::map<std::wstring, std::wstring> CrackURL(std::wstring );
	std::wstring brook_httpsend(std::wstring, std::wstring, std::wstring, std::wstring);
	void Disconnect(void) { if (httpPOST_) Netlib_CloseHandle(httpPOST_); httpPOST_ = NULL; }
	std::wstring OAuthNormalizeUrl( const std::wstring& url );
	std::wstring OAuthNormalizeRequestParameters( const OAuthParameters& requestParameters );
	OAuthParameters ParseQueryString( const std::wstring& url );

	std::wstring OAuthCreateNonce();
	std::wstring OAuthCreateTimestamp();
	std::string HMACSHA1( const std::string& keyBytes, const std::string& data );
	std::wstring Base64String( const std::string& hash );
	std::wstring OAuthCreateSignature( const std::wstring& signatureBase, const std::wstring& consumerSecret, const std::wstring& requestTokenSecret );

protected:
	http::response slurp(const std::string &,http::method, OAuthParameters );
	INT_PTR LOG(const char *fmt,...);
	INT_PTR WLOG(const char* first, const std::wstring last);

	HANDLE httpPOST_;
	HANDLE handle_;
};

inline void mbcs_to_tcs(UINT code_page,const char *mbstr,TCHAR *tstr,int tlen)
{

	MultiByteToWideChar(code_page,0,mbstr,-1,tstr,tlen);

}

inline void wcs_to_tcs(UINT code_page,const wchar_t *wstr,TCHAR *tstr,int tlen)
{

	wcsncpy(tstr,wstr,tlen);

}

class ScopedLock
{
public:
	ScopedLock(HANDLE h) : handle_(h)
	{
		WaitForSingleObject(handle_,INFINITE);
	}
	~ScopedLock()
	{
		if(handle_)
			ReleaseMutex(handle_);
	}

	void Unlock()
	{
		ReleaseMutex(handle_);
		handle_ = 0;
	}
private:
	HANDLE handle_;
};

int ext_to_format(const std::string &ext);
bool save_url(HANDLE hNetlib,const std::string &url,const std::tstring &filename);