实现HTTP协议Get、Post和文件上传功能——使用WinHttp接口实现

        在《使用WinHttp接口实现HTTP协议Get、Post和文件上传功能》一文中,我已经比较详细地讲解了如何使用WinHttp接口实现各种协议。在最近的代码梳理中,我觉得Post和文件上传模块可以得到简化,于是几乎重写了这两个功能的代码。因为Get、Post和文件上传功能的基础(父)类基本没有改动,函数调用的流程也基本没有变化,所以本文我将重点讲解修改点。(转载请指明出于breaksoftware的csdn博客)

        首先我修改了接口的字符集。之前我都是使用UNICODE作为接口参数类型,其中一个原因是Windows提倡UNICODE编码,其次是因为WinHttp接口只提供了UNICODE接口函数。而我在本次修改中,将字符集改成UTF8。因为在网络传输方便,UTF8格式才是主流。于是为了使用WinHttp接口,我提供了一个A版本的转换层——工程中WinhttpA.h。

        其次,我增强了Post接口。《使用WinHttp接口实现HTTP协议Get、Post和文件上传功能》的读者和我讨论了很多Post协议,让我感觉非常有必要重视起该功能。本文我们将着重讲解Post的实现和测试。

        再次,我将Post的实现和文件上传功能的实现合二为一。因为两者代码非常相似,其实在原理方面也是很相似的。

        最后,我使用前一篇博文中介绍的IMemFileOperation接口,重新定义了Post和文件上传功能的参数定义。因为IMemFileOperation的特性,我们可以上传文件,或者上传一片内存值,或者上传文件中的内容,而这些操作是相同的。

        Get请求没什么好说的了,我们主要关注Post和文件上传。

        一般情况下,我们遇到的是“我们需要向http://www.xxx.com:8080/yyyy/zzz地址Post数据”。其中的“数据”是我们问题的重点。可能很多人认为Post请求就是将所有参数都Post到服务器,其实不然。打个比方,比如我们要求对http://www.xxxx.com/post?a=b&c=d地址Post一个数据e=f,我们并不是将"a=b&c=d&e=f"Post到服务器,而只是"e=f"Post过去,"a=b&c=d"还是按Get的方式发送。于是我对上一版的设计做了改良,废掉了ParseParams函数,简化了设计,但是要求用户传进来的URL中不包含需要Post过去的数据——需要Post的数据通过SetPostParam方法传递进来。我们想把重点发到这种发送分离的实现上:

 

	if ( !WinHttpCrackUrlA_( m_strUrl, strHost, strPath, strExt, nPort ) ) {
		break;
	}

	m_hSession = WinHttpOpenA( m_strAgent.empty() ? NULL : m_strAgent.c_str(), WINHTTP_ACCESS_TYPE_DEFAULT_PROXY, WINHTTP_NO_PROXY_NAME, WINHTTP_NO_PROXY_BYPASS, 0 ); 
	if ( NULL == m_hSession ) {
		break;
	}

	if ( FALSE == WinHttpSetTimeouts(m_hSession, m_nResolveTimeout, m_nConnectTimeout, m_nSendTimeout, m_nSendTimeout) ) {
		break;
	}

	m_hConnect = WinHttpConnectA( m_hSession, strHost.c_str(), nPort, 0 );
	if ( NULL == m_hConnect ) {
		break;
	}

	m_strRequestData = strPath + strExt;

        主要关注最后一行,我将URL路径和URL参数放到m_strRequestData里。之后

VOID CHttpRequestByWinHttp::TransmiteDataToServerByPost()
{
	BOOL bSuc = FALSE;
	do {
		m_hRequest = WinHttpOpenRequestA(m_hConnect, "Post",
				m_strRequestData.c_str(), NULL, WINHTTP_NO_REFERER, WINHTTP_DEFAULT_ACCEPT_TYPES, 0);
		if ( NULL == m_hRequest ) {
			break;
		}

        这样,我们便将不需要Post的数据发送了过去。

        现在我们再探讨下需要Post过去的数据。首先我们需要明确下数据的来源:

  • 内存中的数据
  • 文件中的数据

        不管数据来源于何处,它都可以成为待Post过去的数据或者待上传的文件的内容。于是我们借用上一篇博文中的IMemFileOperation接口,定义Post的数据的格式。

typedef struct _FMParam_ {
    std::string strkey;
    ToolsInterface::LPIMemFileOperation value;
    bool postasfile;
    struct FileInfo {
        char szfilename[128];
    };

    struct MemInfo{
        bool bMulti;
    };

    union {
        FileInfo fileinfo;
        MemInfo meminfo;
    };
}FMParam, *PFMParam;

typedef std::vector FMParams;
typedef FMParams::iterator FMParamsIter;
typedef FMParams::const_iterator FMParamsCIter;

        不管是Post数据还是要上传文件,协议中都需要key的存在。strkey是数据的key。value字段只是一个指针,它是指向一个文件还是内存。已经没有关系了,因为之后我们将使用统一的接口去访问它。postasfile字段是标志该参数是否以文件内容的形式Post上去。这儿需要特别说明下,postasfile和value是内存还是文件是没有关系的。因为value只是指向了数据内容,至于内容上传到服务器是作为文件的内容还是只是普通Post的数据值是由postasfile决定的。如果postasfile为真,则FileInfo将被利用到。因为它标记了内容上传到服务器后,服务器上保存的文件名。如果postasfile为假,则我们需要考虑下数据是作为普通数据post,还是作为MultiPart数据Post。这个就取决于MemInfo中的字段了。至于什么是MultiPart类型,可以简单参考《使用WinHttp接口实现HTTP协议Get、Post和文件上传功能》后半部分关于文件上传的讨论。

        对于待上传的数据,之前设计改框架时,框架提供了GetData方法,让继承类提供数据。因为数据存在延续性,所以导致继承类的书写很麻烦——需要记录已经上传了哪些数据。这个版本我将这个设计做了修改,基类暴露一个发送方法,让继承类在需要的时候调用基类的方法,从而不需要基类记录过程的状态。于是以前一大坨代码被简化到如下几行:

DWORD dwUserDataLength = GetUserDataSize();
if ( FALSE == WinHttpSendRequest( m_hRequest, WINHTTP_NO_ADDITIONAL_HEADERS, 0, WINHTTP_NO_REQUEST_DATA, 0, dwUserDataLength, 0)) {
	break;
}

DWORD dwSendUserDataLength = SendUserData();
bSuc = (dwUserDataLength == dwSendUserDataLength) ? TRUE : FALSE;

        通过GetUserDataSize我们将获得待Post过去的数据的大小。然后调用SendUserData发送数据,返回发送了的数据的大小。通过对比两者大小得知是否整个操作是否成功。

       现在我们再看下发送数据的具体实现,首先我们看下一些固定要写死的字段的申明

#define BOUNDARYPART "--MULTI-PARTS-FORM-DATA-BOUNDARY"

#define PARTRETURN  "\r\n"
#define PARTDISPFD  "Content-Disposition:form-data;"
#define PARTNAME    "name"
#define PARTEQUATE  "="
#define PARTQUOTES  "\""
#define PARTSPLIT   "&"
#define PARTSEMICOLON   ";"
#define PARTFILENAME    "filename"
#define PARTTYPEOCT "Content-Type:application/octet-stream"

        读过《使用WinHttp接口实现HTTP协议Get、Post和文件上传功能》的朋友应该记得其中有很多繁杂的数据格式化。之前我们讲过,我们需要先获得待Post的数据大小,再发送数据。这意味着繁杂的数据格式化需要做两次。如果以后需要对其中某个发送数据格式化做修改,那么相应的计算数据长度的方法也要做修改。这是非常不利于维护的。于是,我将两者合为一个函数,通过参数判断是需要计算还是需要发送。这样以后修改发送数据时,只要修改一处,降低了维护的成本和难度。

    DWORD CHttpTransByPost::SendUserData() {
        return SendOrCalcData();
    }

    DWORD CHttpTransByPost::GetUserDataSize() {
        return SendOrCalcData(FALSE);
    }

    DWORD CHttpTransByPost::SendOrCalcData( BOOL bSend /*= TRUE*/ ) {
        DWORD dwsize = 0;
        for (FMParamsCIter it = m_PostParam.begin(); it != m_PostParam.end(); it++) {
            dwsize += SendData(*it, bSend);
        }
        if (!m_strBlockEnd.empty()) {
            dwsize += DataToServer(m_strBlockEnd.c_str(), m_strBlockEnd.length(), bSend);
        }   
        return dwsize;
    }

        在SendOrCalcData的最后,我们判断m_strBlockEnd是否为空,如果不为空,则我们将BlockEnd格式化数据发送过去,告诉服务器MultiPart数据发送结束。如果为空,则代表此次发送数据不需要按MultiPart形式发送。至于是否需要MultiPart,以及其各种格式化则是在下面的代码中判断

BOOL CHttpTransByPost::ModifyRequestHeader( HINTERNET hRequest ) {
        bool bMulti = false;
        for (FMParamsCIter it = m_PostParam.begin(); it != m_PostParam.end(); it++) {
            if (it->postasfile) {
                bMulti = true;
                break;
            }
            else {
                bMulti = it->meminfo.bMulti;
                if (bMulti) {
                    break;
                }
            }
        }

		if (bMulti) {
			m_strBlockStart = "--";
			m_strBlockStart += BOUNDARYPART;
			m_strBlockStart += "\r\n";

			m_strBlockEnd =  "\r\n--";
			m_strBlockEnd += BOUNDARYPART;
			m_strBlockEnd +=  "--\r\n";

			m_strNewHeader = "Content-Type: multipart/form-data; boundary=";
			m_strNewHeader += BOUNDARYPART;
			m_strNewHeader += "\r\n";
		}
		else {
			m_strNewHeader = "Content-Type:application/x-www-form-urlencoded";
			m_strNewHeader += "\r\n";
		}

		::WinHttpAddRequestHeadersA(hRequest, m_strNewHeader.c_str(), 
			m_strNewHeader.length(), WINHTTP_ADDREQ_FLAG_ADD | WINHTTP_ADDREQ_FLAG_REPLACE) ;
		return AddUserRequestHeader(hRequest);
	}

        最后我们将注意力集中到发送(计算)数据的函数SendData上。

    DWORD CHttpTransByPost::SendData(const FMParam& postparam, BOOL bSend /*= TRUE*/ ) {
        DWORD dwsize = 0;
        postparam.value->MFSeek(0, SEEK_SET);
        if (postparam.postasfile) {
            dwsize = SendFileData(postparam, bSend);
        }
        else {
            dwsize = SendMemData(postparam, bSend);
        }
        return dwsize;
    }

        首先,我们使用MFSeek将文件(内存)的指针置到起始处。然后再通过postasfile决定是按文件的形式发送还是按内存的形式发送。

    DWORD CHttpTransByPost::SendFileData(const FMParam& postparam, BOOL bSend) {
        DWORD dwsize = 0;
        dwsize += DataToServer(PARTRETURN, strlen(PARTRETURN), bSend);
        dwsize += DataToServer(m_strBlockStart.c_str(), m_strBlockStart.length(), bSend);
        dwsize += DataToServer(PARTDISPFD, strlen(PARTDISPFD), bSend);
        dwsize += DataToServer(PARTNAME, strlen(PARTNAME), bSend);
        dwsize += DataToServer(PARTEQUATE, strlen(PARTEQUATE), bSend);
        dwsize += DataToServer(PARTQUOTES, strlen(PARTQUOTES), bSend);
        dwsize += DataToServer(postparam.strkey.c_str(), postparam.strkey.length(), bSend);
        dwsize += DataToServer(PARTQUOTES, strlen(PARTQUOTES), bSend);
        dwsize += DataToServer(PARTSEMICOLON, strlen(PARTSEMICOLON), bSend);
        dwsize += DataToServer(PARTFILENAME, strlen(PARTFILENAME), bSend);
        dwsize += DataToServer(PARTEQUATE, strlen(PARTEQUATE), bSend);
        dwsize += DataToServer(PARTQUOTES, strlen(PARTQUOTES), bSend);
        dwsize += DataToServer(postparam.fileinfo.szfilename, strlen(postparam.fileinfo.szfilename), bSend);
        dwsize += DataToServer(PARTQUOTES, strlen(PARTQUOTES), bSend);
        dwsize += DataToServer(PARTRETURN, strlen(PARTRETURN), bSend);
        dwsize += DataToServer(PARTTYPEOCT, strlen(PARTTYPEOCT), bSend);
        dwsize += DataToServer(PARTRETURN, strlen(PARTRETURN), bSend);
        dwsize += DataToServer(PARTRETURN, strlen(PARTRETURN), bSend);
        while(!postparam.value->MFEof()) {
            char buffer[1024] = {0};
            size_t size = postparam.value->MFRead(buffer, 1, 1024);
            dwsize += DataToServer(buffer, size, bSend);
        }
        return dwsize;
    }

        以文件内容形式发送的代码如上。我们关注下最后几行,MFRead读取内容,然后发送(计算)数据。

    DWORD CHttpTransByPost::SendMemData(const FMParam& postparam, BOOL bSend) {
        DWORD dwsize = 0;
        if (postparam.meminfo.bMulti) {
            dwsize += DataToServer(PARTRETURN, strlen(PARTRETURN), bSend);
            dwsize += DataToServer(m_strBlockStart.c_str(), m_strBlockStart.length(), bSend);
            dwsize += DataToServer(PARTDISPFD, strlen(PARTDISPFD), bSend);
            dwsize += DataToServer(PARTNAME, strlen(PARTNAME), bSend);
            dwsize += DataToServer(PARTEQUATE, strlen(PARTEQUATE), bSend);
            dwsize += DataToServer(PARTQUOTES, strlen(PARTQUOTES), bSend);
            dwsize += DataToServer(postparam.strkey.c_str(), postparam.strkey.length(), bSend);
            dwsize += DataToServer(PARTQUOTES, strlen(PARTQUOTES), bSend);
            dwsize += DataToServer(PARTRETURN, strlen(PARTRETURN), bSend);
	        while(!postparam.value->MFEof()) {
		        char buffer[1024] = {0};
		        size_t size = postparam.value->MFRead(buffer, 1, 1024);
                dwsize += DataToServer(buffer, size, bSend);
	        }
        }
        else {
            dwsize += DataToServer(PARTSPLIT, strlen(PARTSPLIT), bSend);
            dwsize += DataToServer(postparam.strkey.c_str(), postparam.strkey.length(), bSend);
            dwsize += DataToServer(PARTEQUATE, strlen(PARTEQUATE), bSend);
            while(!postparam.value->MFEof()) {
                char buffer[1024] = {0};
                size_t size = postparam.value->MFRead(buffer, 1, 1024);
                dwsize += DataToServer(buffer, size, bSend);
            }
        }

        return dwsize;
    }

        以上是发送普通Post数据的方法。其中分为是否需要以MultiiPart形式发送,还是以普通形式发送。MultiPart形式之前已经说过,而普通Post数据形式则是无约束的。我将该数据时拼装成Name1=Value1&Name2=Value2的形式发送的。

        对于MultiParg类型的Post,我们使用WireShark截取发送包

实现HTTP协议Get、Post和文件上传功能——使用WinHttp接口实现_第1张图片

        发送普通Post数据的WireShark截包为

实现HTTP协议Get、Post和文件上传功能——使用WinHttp接口实现_第2张图片

        最后我们看下使用的例子

    HttpRequestFM::CHttpTransByPost* p = new HttpRequestFM::CHttpTransByPost();
    ToolsInterface::IMemFileOperation* pMemOp = new MemFileOperation::CMemOperation();
	
    p->SetOperation(pMemOp);
    p->SetProcessCallBack(ProcssCallback);
    p->SetUrl(BIGFILEURL);
    
    FMParams params;

    FMParam param1;
    param1.postasfile = false;
    param1.strkey = "key1";
    param1.meminfo.bMulti = false;
    MemFileOperation::CMemOperation mem1("value1", strlen("value1"));
    param1.value = &mem1;
    params.push_back(param1);

    FMParam param2;
    param2.postasfile = false;
    param2.strkey = "key2";
    param2.meminfo.bMulti = true;
    //sprintf_s(param2.fileinfo.szfilename, sizeof(param2.fileinfo.szfilename), "2.bin");
    MemFileOperation::CFileOperation file2("F:/2.bin");
    param2.value = &file2;
    params.push_back(param2);
    
    FMParam param3;
    param3.strkey = "key3";
    //param3.meminfo.bMulti = true;
    param3.postasfile = true;
    sprintf_s(param3.fileinfo.szfilename, sizeof(param3.fileinfo.szfilename), "3.bin");
    MemFileOperation::CFileOperation file3("F:/1.bin");
    param3.value = &file3;
    params.push_back(param3);

    p->SetPostParam(params);
    p->Start();

        param1是以普通Post数据格式传输的参数;param2的value是从F:/2.bin文件中读取的,但是其只是MultiPart形式上传的数据,而非上传文件。param3是要求上传文件F:/1.bin文件到服务器上为3.bin。

        通过不同的组合,我们可以同时上传多个文件。比如我们将上例中的param2做稍微的修改,即可以将其对应的文件上传至服务器,实现同时上传多个文件的功能。

        工程源码链接:http://pan.baidu.com/s/1i3eUnMt 密码:hfro

你可能感兴趣的:(网络通信)