w3c.cpp

基于w3c的网络开发用例

		if(usp<strlen(szurl)){ // find username and find password
			unsigned long ssp=nsp;
			while(strlen(szurl)>0 && npos<strlen(szurl) && strncmp((szurl+ssp), ":", 1))
				++ssp;

			if(ssp<usp){// find
				strncpy(szuser, szurl+nsp, ssp-nsp);
				szuser[ssp-nsp+1]='\0';
				strncpy(szpassword, szurl+ssp+1, usp-ssp-1);
				szpassword[usp-ssp]='\0';
			}

			nsp=npos=usp+1;
		}

		bflag=false;
		while(strlen(szurl)>0 && npos<strlen(szurl) && strncmp((szurl+npos), "/", 1))
				++npos;

		unsigned long nf=nsp;

		for(;nf<=npos;nf++){
			if(!strncmp((szurl+nf), ":", 1)){ // find PORT
				bflag=true;
				break;
			}
		}

		if(bflag){
			char sztmp[1024]="\0";
			strncpy(sztmp, (szurl+nf+1), npos-nf);
			nport=atol(sztmp);
			strncpy(szaddress, (szurl+nsp), nf-nsp);
		}else if(!strcmp(szprotocol,"https")){
			nport=INTERNET_DEFAULT_HTTPS_PORT;
			strncpy(szaddress, (szurl+nsp), npos-nsp);
		}else if(!strcmp(szprotocol, "ftp")){
			nport=INTERNET_DEFAULT_FTP_PORT;
			strncpy(szaddress, (szurl+nsp), npos-nsp);
		}else {
			nport=INTERNET_DEFAULT_HTTP_PORT;
			strncpy(szaddress, (szurl+nsp), npos-nsp);
		}

		if(npos<strlen(szurl)){ // find URI
			strncpy(szuri, (szurl+npos), strlen(szurl)-npos);
		}else{
			szuri[0]='/';
			szuri[1]='\0';
		}

		return;
	}

	// Asynchronized www client
	bool AsyncW3Client::Connect(const char *szaddress,
													INTERNET_STATUS_CALLBACK lpfn,
													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,
									lpfn,
									((szuser || (!szuser && strlen(szus)==0))? szuser : szus),
									((szpassword || (!szpassword && strlen(szpw)==0))? szpassword : szpw),
									wt, szagent);
	}

	bool AsyncW3Client::Connect(	const char *szaddress,
														long nport,
														INTERNET_STATUS_CALLBACK lpfn,
														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
														INTERNET_FLAG_ASYNC); // flags
			if(!_hOpen || _hOpen==INVALID_HANDLE_VALUE)
				throw "open internet failed...";	

			INTERNET_STATUS_CALLBACK pc=::InternetSetStatusCallback(_hOpen, lpfn);
			//if(!pc || pc==INTERNET_INVALID_STATUS_CALLBACK )
			//	throw "set status call-back functino failed...";
			
			switch(t) {
			case w3ftp:
				_hConnection=::InternetConnect(	_hOpen,
																		szaddress, (nport? nport: INTERNET_DEFAULT_FTP_PORT),
																		szuser, szpassword,
																		INTERNET_SERVICE_FTP,
																		INTERNET_FLAG_PASSIVE,
																		reinterpret_cast<unsigned long>(this));
				break;
			case w3http:
				_hConnection=::InternetConnect(	_hOpen,
																		szaddress, (nport? nport: INTERNET_DEFAULT_HTTP_PORT),
																		szuser, szpassword,
																		INTERNET_SERVICE_HTTP,
																		0,
																		reinterpret_cast<unsigned long>(this));
				break;
			case w3https:
				_hConnection=::InternetConnect(	_hOpen,
																		szaddress, (nport? nport: INTERNET_DEFAULT_HTTPS_PORT),
																		szuser, szpassword,
																		INTERNET_SERVICE_HTTP,
																		0,
																		reinterpret_cast<unsigned long>(this));
				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)
				::InternetCloseHandle(_hConnection);
			__w3cexcept(szaddress, nport, t, szm);
		}catch(...){
			r=false;
			if(_hOpen || _hOpen!=INVALID_HANDLE_VALUE)
				::CloseHandle(_hOpen);
			if(_hConnection || _hConnection!=INVALID_HANDLE_VALUE)
				::CloseHandle(_hConnection);
			__w3cexcept(szaddress, nport, t, "unknown exception...");
		}

		return r;
	}

	bool AsyncW3Client::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),
																reinterpret_cast<unsigned long>(this)); // 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
			&& 
			::GetLastError()!=ERROR_IO_PENDING
			){		
			throw "asynchronized request failed...";
		}

		return true;
	}

	bool AsyncW3Client::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 |
																INTERNET_FLAG_NO_CACHE_WRITE |
																INTERNET_FLAG_FORMS_SUBMIT |
																(_t==w3https? INTERNET_FLAG_SECURE:0),
																reinterpret_cast<unsigned long>(this)); // 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...";
			}
		}

		// GET POST ARGUMENTS
		buf=reinterpret_cast<unsigned char*>(::HeapAlloc(::GetProcessHeap(), HEAP_ZERO_MEMORY, GetPostArgumentsLength()+1));
		len=GetPostData(buf, GetPostArgumentsLength()+1);

		// SEND REQUEST WITH HttpSendRequestEx and InternetWriteFile
		static INTERNET_BUFFERS InternetBufferIn={0};
		InternetBufferIn.dwStructSize=sizeof(INTERNET_BUFFERS);
		InternetBufferIn.Next=NULL;
		InternetBufferIn.lpcszHeader=szContentType;
		InternetBufferIn.dwHeadersLength=strlen(szContentType);
		InternetBufferIn.lpvBuffer=buf;
		InternetBufferIn.dwBufferLength=len;
			
		if(!::HttpSendRequestEx(_hRequest, &InternetBufferIn, NULL, HSR_INITIATE, reinterpret_cast<unsigned long>(this))
			&&
			::GetLastError()!=ERROR_IO_PENDING
			){
			// free
			::HeapFree(::GetProcessHeap(), 0, buf);	
			throw "request failed";
		}

		WaitCompleteRequest();

		if(!::HttpEndRequest(_hRequest, NULL, HSR_ASYNC | HSR_INITIATE, reinterpret_cast<unsigned long>(this))
			&&
			::GetLastError()!=ERROR_IO_PENDING
			){
			// free
			::HeapFree(::GetProcessHeap(), 0, buf);
			throw "request failed";
		}		

		::HeapFree(::GetProcessHeap(), 0, buf);

		return true;
	}

	bool AsyncW3Client::RequestPost2(const char *szuri, const char *szref /*=NULL*/){
		static LPCTSTR szAcceptType=__HTTP_ACCEPT_TYPE;
		static LPCTSTR szContentType="Content-Type: multipart/form-data; boundary=--MULTI-PARTS-FORM-DATA-BOUNDARY\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 |
																INTERNET_FLAG_NO_CACHE_WRITE |
																INTERNET_FLAG_FORMS_SUBMIT |
																(_t==w3https? INTERNET_FLAG_SECURE:0),
																reinterpret_cast<unsigned long>(this)); // 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...";		
		
		if(!::HttpAddRequestHeaders( _hRequest, szContentType, strlen(szContentType), HTTP_ADDREQ_FLAG_ADD_IF_NEW))
			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...";
			}
		}

		// build multi-parts/form-data
		len=AllocMultiPartsFormData(buf, "--MULTI-PARTS-FORM-DATA-BOUNDARY");

		// ADD HEADER CONTENT LENGTH
		char szcl[__DEFAULT_BUF_SIZE]="\0";
		sprintf(szcl, "Content-Length: %d\r\n", len);

		if(!::HttpAddRequestHeaders( _hRequest, szcl, strlen(szcl), HTTP_ADDREQ_FLAG_ADD_IF_NEW))
			throw "additional header failed...";		

		// SEND REQUEST WITH HttpSendRequestEx and InternetWriteFile
		static INTERNET_BUFFERS InternetBufferIn={0};
		InternetBufferIn.dwStructSize=sizeof(INTERNET_BUFFERS);
		InternetBufferIn.Next=NULL;	
		InternetBufferIn.lpvBuffer=buf;
		InternetBufferIn.dwBufferLength=len;
			
		if(!::HttpSendRequestEx(_hRequest, &InternetBufferIn, NULL, HSR_INITIATE, reinterpret_cast<unsigned long>(this))
			&&
			::GetLastError()!=ERROR_IO_PENDING
			){
			// free
			FreeMultiPartsFormData(buf);
			throw "request failed";
		}

		WaitCompleteRequest();

		if(!::HttpEndRequest(_hRequest, NULL, HSR_ASYNC | HSR_INITIATE, reinterpret_cast<unsigned long>(this))
			&&
			::GetLastError()!=ERROR_IO_PENDING
			){
			// free
			FreeMultiPartsFormData(buf);
			throw "request failed";
		}		

		// free multi-parts/form-data
		FreeMultiPartsFormData(buf);

		return true;
	}

	void AsyncW3Client::SetCompleteRequest(){ ::SetEvent(_hCompleteRequestEvent); }
	bool AsyncW3Client::WaitCompleteRequest(unsigned long ntime /*=INFINITE*/ ){
		if(!_hCompleteRequestEvent || _hCompleteRequestEvent==INVALID_HANDLE_VALUE)
			false;
		return ::WaitForSingleObject(_hCompleteRequestEvent, ntime)==WAIT_OBJECT_0? true:false;
	}