在《使用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<FMParam> 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截取发送包
发送普通Post数据的WireShark截包为
最后我们看下使用的例子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做稍微的修改,即可以将其对应的文件上传至服务器,实现同时上传多个文件的功能。