#include "w3c.h"
#include <cstring>

using namespace std;

#define __HTTP_VERB_GET	"GET"
#define __HTTP_VERB_POST "POST"
#define __HTTP_ACCEPT_TYPE "*/*"
#define __HTTP_ACCEPT "Accept: */*\r\n"

#define __DEFAULT_BUF_SIZE 1024

	void __w3cexcept(const char *szaddress, long nport, W3Client::w3t t, const char *szmsg){
#ifdef _DEBUG
		string sztmp;
		sztmp+="[ ";
		switch(t) {
		case W3Client::w3http:
			sztmp+="http://";
			break;
		case W3Client::w3https:
			sztmp+="https://";
			break;
		case W3Client::w3ftp:
			sztmp+="ftp://";
			break;
		}
		sztmp+=szaddress;
		sztmp+=":";
		char szp[10]="\0";
		sprintf(szp, "%d", nport);
		sztmp+=szp;
		sztmp+=" ] ";
		sztmp+=szmsg;

		::OutputDebugString(sztmp.c_str());
		
		DWORD err=::GetLastError();
		LPVOID     lpMsgBuffer;
		DWORD dwRet=FormatMessage( FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE,
																GetModuleHandle("wininet.dll"),
																err,
																MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
																reinterpret_cast<LPTSTR>(&lpMsgBuffer),
																0,
																NULL);

		char szmsg2[1024]="\0";
		sprintf(szmsg2, "[%d, 0x%08x] %s", err, err, reinterpret_cast<LPTSTR>(lpMsgBuffer));
		OutputDebugString(szmsg2);
		::LocalFree(lpMsgBuffer);
#endif		
	}

	void __w3cexcept(const char *szaddress, long nport, W3Client::w3t t, const char *szuri, const char *szmsg){
#ifdef _DEBUG
		string sztmp;
		sztmp+="[ ";
		switch(t) {
		case W3Client::w3http:
			sztmp+="http://";
			break;
		case W3Client::w3https:
			sztmp+="https://";
			break;
		case W3Client::w3ftp:
			sztmp+="ftp://";
			break;
		}
		sztmp+=szaddress;
		sztmp+=":";
		char szp[10]="\0";
		sprintf(szp, "%d", nport);		
		sztmp+=szp;
		sztmp+=szuri;
		sztmp+=" ] ";
		sztmp+=szmsg;

		::OutputDebugString(sztmp.c_str());

		LPVOID lpMsgBuffer;
		DWORD err=::GetLastError();
		DWORD dwRet=FormatMessage( FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE,
																GetModuleHandle("wininet.dll"),
																err,
																MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
																reinterpret_cast<LPTSTR>(&lpMsgBuffer),
																0,
																NULL);
		char szmsg2[1024]="\0";
		sprintf(szmsg2, "[%d, 0x%08x] %s", err, err, reinterpret_cast<LPTSTR>(lpMsgBuffer));
		OutputDebugString(szmsg2);
		::LocalFree(lpMsgBuffer);
#endif	
	}

	// W3Client : synchronized www client
	bool W3Client::Connect(const char *szaddress,
											const char *szuser /*=NULL*/,
											const char *szpassword /*=NULL*/,
											const char *szagent /*=__W3_DEFAULT_AGENT*/){
		
		char szp[__DEFAULT_BUF_SIZE]="\0", szus[__DEFAULT_BUF_SIZE]="\0", szpw[__DEFAULT_BUF_SIZE]="\0";
		char sza[__DEFAULT_BUF_SIZE]="\0", szuri[__DEFAULT_BUF_SIZE]="\0";
		unsigned long port=0;

		__w3curlparse(szaddress, szp, szus, szpw, sza, port, szuri);

		w3t wt;

		if(!strncmp(szp, "https", 5))
			wt=w3https;
		else if(!strncmp(szp, "http", 4))
			wt=w3http;
		else if(!strncmp(szp, "ftp", 3))
			wt=w3ftp;

		_szuri=szuri;
		
		return Connect(sza, port,
									((szuser || (!szuser && strlen(szus)==0))? szuser : szus),
									((szpassword || (!szpassword && strlen(szpw)==0))? szpassword : szpw),
									wt, szagent);
	}

	bool W3Client::Connect(	const char *szaddress,
											long nport,
											const char *szuser /*=NULL*/,
											const char *szpassword /*=NULL*/,											
											w3t t /*=w3http*/,											
											const char *szagent /*=__W3_DEFAULT_AGENT*/){
		
		bool r=true;

		try{
			_hOpen=::InternetOpen( szagent, // agent
														INTERNET_OPEN_TYPE_PRECONFIG, // access type
														NULL, // proxy
														NULL, // proxy by pass
														0); // flags
			if(!_hOpen || _hOpen==INVALID_HANDLE_VALUE)
				throw "open internet failed...";

			switch(t) {
			case w3ftp:
				_hConnection=::InternetConnect(	_hOpen,
																		szaddress, (nport? nport: INTERNET_DEFAULT_FTP_PORT),
																		szuser, szpassword,
																		INTERNET_SERVICE_FTP,
																		INTERNET_FLAG_PASSIVE,
																		NULL);
				break;
			case w3http:
				_hConnection=::InternetConnect(	_hOpen,
																		szaddress, (nport? nport: INTERNET_DEFAULT_HTTP_PORT),
																		szuser, szpassword,
																		INTERNET_SERVICE_HTTP,
																		0,
																		NULL);
				break;
			case w3https:
				_hConnection=::InternetConnect(	_hOpen,
																		szaddress, (nport? nport: INTERNET_DEFAULT_HTTPS_PORT),
																		szuser, szpassword,
																		INTERNET_SERVICE_HTTP,
																		0,
																		NULL);
				break;
			}

			if(!_hConnection || _hConnection==INVALID_HANDLE_VALUE)
				throw "connect failed...";

			if(::InternetAttemptConnect(NULL)!=ERROR_SUCCESS)
				throw "connect failed...";

			_szaddress=szaddress;
			if(!nport){
				switch(t) {
				case w3ftp:
					_nport=INTERNET_DEFAULT_FTP_PORT;
					break;
				case w3http:
					_nport=INTERNET_DEFAULT_HTTP_PORT;
					break;
				case w3https:
					_nport=INTERNET_DEFAULT_HTTPS_PORT;
					break;				
				}
			}else
				_nport=nport;
			_t=t;

			if(szuser)
				_szuser=szuser;

			if(szpassword)
				_szpassword=szpassword;

			InitializeCookies();
			InitializePostArguments();			
			
		}catch(const char *szm){
			r=false;
			if(_hOpen || _hOpen!=INVALID_HANDLE_VALUE)
				::InternetCloseHandle(_hOpen);
			if(_hConnection || _hConnection!=INVALID_HANDLE_VALUE)
				::CloseHandle(_hConnection);
			__w3cexcept(szaddress, nport, t, szm);
		}catch(...){
			r=false;
			if(_hOpen || _hOpen!=INVALID_HANDLE_VALUE)
				::InternetCloseHandle(_hOpen);
			if(_hConnection || _hConnection!=INVALID_HANDLE_VALUE)
				::CloseHandle(_hConnection);
			__w3cexcept(szaddress, nport, t, "unknown exception...");
		}

		return r;
	}

	void W3Client::Close(){
		if(_hRequest)
			::InternetCloseHandle(_hRequest);

		if(_hConnection)
			::InternetCloseHandle(_hConnection);

		if(_hOpen)
			::InternetCloseHandle(_hOpen);

		return;
	}

	bool W3Client::Request(const char *szuri, w3m m, const char *szref /*=NULL*/){

		bool bflag=false;
		try{
			switch(m) {
			case reqPost:
				bflag=RequestPost(szuri, szref);
				break;
			case reqPostMultipartsFormdata:
				bflag=RequestPost2(szuri, szref);
				break;
			case reqGet:
				bflag=RequestGet(szuri, szref);
				break;
			}

			if(bflag){
				_szuri=szuri;
			}else{
				::InternetCloseHandle(_hRequest);
				_hRequest=NULL;
			}

		}catch(const char *szm){
			::InternetCloseHandle(_hRequest);
			_hRequest=NULL;
			__w3cexcept(_szaddress.c_str(), _nport, _t, szuri, szm);
		}catch(...){
			::InternetCloseHandle(_hRequest);
			_hRequest=NULL;			
			__w3cexcept(_szaddress.c_str(), _nport, _t, szuri, "unknown exception...");
		}

		return bflag;
	}

	bool W3Client::RequestGet(const char *szuri, const char *szref /*=NULL*/){

		static LPCTSTR szAcceptType=TEXT(__HTTP_ACCEPT_TYPE);

		if(!_hConnection || _hConnection==INVALID_HANDLE_VALUE)
			throw "handle not opened...";

		_hRequest=::HttpOpenRequest( _hConnection,
																__HTTP_VERB_GET, // HTTP Verb
																szuri, // Object Name
																HTTP_VERSION, // Version
																szref, // Reference
																&szAcceptType, // Accept Type
																INTERNET_FLAG_NO_CACHE_WRITE | (_t==w3https? INTERNET_FLAG_SECURE:0),
																NULL); // context call-back point

		if(!_hRequest || _hRequest==INVALID_HANDLE_VALUE)
			throw "request failed...";

		// REPLACE HEADER
		if(!::HttpAddRequestHeaders( _hRequest,
														__HTTP_ACCEPT, strlen(__HTTP_ACCEPT),														
														HTTP_ADDREQ_FLAG_REPLACE))
			throw "additional header failed...";

		// COOKIE ADD
		if(_listcookies.size()>0){
			string szurl;
			switch(_t) {
			case w3http:
				szurl="http://";
				break;
			case w3https:
				szurl="https://";
				break;
			}
			szurl+=_szaddress.c_str();
			if(!((_t==w3http && _nport==INTERNET_DEFAULT_HTTP_PORT) || (_t==w3https && _nport==INTERNET_DEFAULT_HTTPS_PORT))){
				char tmp[10]="\0";
				sprintf(tmp, ":%d", _nport);
				szurl+=tmp;
			}
			szurl+=szuri;

			for(list<HTTP_COOKIE*>::iterator it=_listcookies.begin(); it!=_listcookies.end(); it++){
				HTTP_COOKIE *pc=reinterpret_cast<HTTP_COOKIE*>(*it);
				if(!::InternetSetCookie(szurl.c_str(), pc->name.c_str(), pc->value.c_str()))
					throw "add cookie failed...";
			}
		}

		// SEND REQUEST
		if(!::HttpSendRequest( _hRequest,	// handle by returned HttpOpenRequest
												NULL, // additional HTTP header
												0, // additional HTTP header length
												NULL, // additional data in HTTP Post or HTTP Put
												0) // additional data length
												)
			throw "request failed...";

		return true;
	}

	void W3Client::InitializePostArguments(){

		if(_listargs.size()>0){
			for(list<HTTP_ARG*>::iterator it=_listargs.begin(); it!=_listargs.end(); it++){
				HTTP_ARG *p=reinterpret_cast<HTTP_ARG*>(*it);
				delete p;
			}
			_listargs.clear();
		}
		return;
	}

	void W3Client::AddPostArgument(const char *szname, const int nvalue){
		HTTP_ARG *pa=new HTTP_ARG(szname, nvalue);
		_listargs.push_back(pa);
		return;
	}

	void W3Client::AddPostArgument(const char *szname, const long nvalue){
		HTTP_ARG *pa=new HTTP_ARG(szname, nvalue);
		_listargs.push_back(pa);
		return;
	}

	void W3Client::AddPostArgument(const char *szname, const float nvalue){
		HTTP_ARG *pa=new HTTP_ARG(szname, nvalue);
		_listargs.push_back(pa);
		return;
	}

	void W3Client::AddPostArgument(const char *szname, const double nvalue){
		HTTP_ARG *pa=new HTTP_ARG(szname, nvalue);
		_listargs.push_back(pa);
		return;
	}

	void W3Client::AddPostArgument(const char *szname, const char *szvalue, bool bfile /*=false*/){
		HTTP_ARG *pa=new HTTP_ARG(szname, szvalue, bfile);
		_listargs.push_back(pa);
		return;
	}

	void W3Client::InitializeCookies(){
		if(_listcookies.size()>0){
			for(list<HTTP_COOKIE*>::iterator it=_listcookies.begin(); it!=_listcookies.end(); it++){
				HTTP_COOKIE *p=reinterpret_cast<HTTP_COOKIE*>(*it);
				delete p;
			}
			_listcookies.clear();
		}
		return;
	}

		void W3Client::AddCookie(const char *szname, const char *szvalue){
		HTTP_COOKIE *pc=new HTTP_COOKIE(szname, szvalue);
		_listcookies.push_back(pc);

		return;
	}

	void W3Client::AddCookie(const char *szname, const int value){
		HTTP_COOKIE *pc=new HTTP_COOKIE(szname, value);
		_listcookies.push_back(pc);
		return;
	}

	void W3Client::AddCookie(const char *szname, const long value){
		HTTP_COOKIE *pc=new HTTP_COOKIE(szname, value);
		_listcookies.push_back(pc);
		return;
	}

	void W3Client::AddCookie(const char *szname, const float value){
		HTTP_COOKIE *pc=new HTTP_COOKIE(szname, value);
		_listcookies.push_back(pc);
		return;
	}

	void W3Client::AddCookie(const char *szname, const double value){
		HTTP_COOKIE *pc=new HTTP_COOKIE(szname, value);
		_listcookies.push_back(pc);
		return;
	}


	unsigned long W3Client::GetPostArgumentsLength(){
		unsigned long len=0;
		if(_listargs.size()>0){
			for(list<HTTP_ARG*>::iterator it=_listargs.begin(); it!=_listargs.end(); it++){
				HTTP_ARG *p=reinterpret_cast<HTTP_ARG*>(*it);
				len+=p->length()+1;
			}
		}
		return len? len-1: 0;
	}

	unsigned long W3Client::GetPostData(unsigned char *buf, unsigned long len){

		unsigned long l=0;
		if(len>GetPostArgumentsLength() && _listargs.size()>0){
			for(list<HTTP_ARG*>::iterator it=_listargs.begin(); it!=_listargs.end(); it++){
				HTTP_ARG *p=reinterpret_cast<HTTP_ARG*>(*it);
				p->dump(buf+l, p->length());
				buf[l+p->length()]='&';
				l+=p->length()+1;
			}
			buf[l-1]='\0';
		}
		return l? l-1: 0;
	}

	bool W3Client::RequestPost(const char *szuri, const char *szref /*=NULL*/){

		static LPCTSTR szAcceptType=__HTTP_ACCEPT_TYPE;	
		static LPCTSTR szContentType="Content-Type: application/x-www-form-urlencoded\r\n";

		unsigned char *buf=NULL;
		unsigned long len=0;

		if(!_hConnection || _hConnection==INVALID_HANDLE_VALUE)
			throw "handle not opened...";
			
		
		_hRequest=::HttpOpenRequest( _hConnection,
																__HTTP_VERB_POST, // HTTP Verb
																szuri, // Object Name
																HTTP_VERSION, // Version
																szref, // Reference
																&szAcceptType, // Accept Type															
																INTERNET_FLAG_KEEP_CONNECTION |