cWinHttp.h

Go to the documentation of this file.

//
// Wrap HTTP, Proxy and SSL into a single API.
// Similar to OpenSSL and CInternetSession for MFC.
//
 
#ifndef _INC_cWinHttp_H
#define _INC_cWinHttp_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "../Net/cNetAddr.h"
#include "../Http/cHttpCommon.h"
#include "GrayCore/include/cFile.h"
 
#ifdef _WIN32
//#define USE_WinHttp   // use Winhttp.dll and Winhttp.lib
#endif
 
#ifdef USE_WinHttp
#include <WinHttp.h>
 
namespace GrayLib
{
        class cWinHttpSession;
        class cWinHttpConnection;
 
        class GRAYLIB_LINK cWinHttpBase // : public cHandlePtr<>
        {
 
        private:
                HINTERNET m_h;  
 
        protected:
                cWinHttpBase()
                        : m_h(INVALID_HANDLE_VALUE)
                {
                }
                ~cWinHttpBase()
                {
                }
                void Close();
 
                void AttachHandle(HINTERNET h)
                {
                        if (m_h != h)
                        {
                                Close();
                                m_h = h;
                        }
                }
 
        public:
                bool isValidHandle() const
                {
                        return m_h != HANDLE_NULL && m_h != INVALID_HANDLE_VALUE ;
                }
                HINTERNET get_Handle() const
                {
                        return m_h;
                }
                virtual cWinHttpSession* GetSession() const = 0;
        };
 
        //**************************************************************************
 
        class GRAYLIB_LINK cWinHttpRequest
                : public cWinHttpBase
                , public cRefBase // , public cStream
        {
                typedef cWinHttpBase SUPER_t;
        public:
                cWinHttpRequest(cWinHttpConnection* pConnection, HTTPVERB_TYPE nVerb, LPCWSTR pstrObjectName,
                        IN LPCWSTR pwszReferrer, IN LPCWSTR FAR * ppwszAcceptTypes, DWORD dwFlags = 0);
                virtual cWinHttpSession* GetSession() const;
 
                bool SendRequest();
                bool ReceiveResponse();
 
        public:
                cWinHttpConnection* m_pConnection;      // my parent connection.
        };
        typedef cRefPtr<cWinHttpRequest> cWinHttpRequestPtr;
 
        //**************************************************************************
 
        class GRAYLIB_LINK cWinHttpConnection
                : public cWinHttpBase
                , public cRefBase
        {
                // Base class for a Managed connection to the Internet.
 
                typedef cWinHttpBase SUPER_t;
        public:
                cWinHttpConnection(cWinHttpSession* pSession,
                        LPCWSTR pstrServer, NET_PORT_t nPort = k_NET_PORT_INVALID,
                        DWORD dwFlags = 0);
                ~cWinHttpConnection();
 
                cWinHttpRequestPtr OpenRequest(HTTPVERB_TYPE nVerb, LPCWSTR pstrObjectName, IN LPCWSTR pwszReferrer, IN LPCWSTR FAR * ppwszAcceptTypes, DWORD dwFlags = 0);
 
                virtual cWinHttpSession* GetSession() const
                {
                        // MFC like
                        return m_pSession;      // get my parent.
                }
                CString GetServerName() const
                {
                        // MFC like
                        return m_strServerName;
                }
 
        protected:
                cWinHttpSession* m_pSession;    // my parent session.
                CString m_strServerName;        // the server URL I'm connected to.
                NET_PORT_t m_wPort;
        };
 
        typedef cRefPtr<cWinHttpConnection> cWinHttpConnectionPtr;
 
        //**************************************************************************
 
        class GRAYLIB_LINK cWinHttpSession : public cWinHttpBase
        {
                // A session can contain connections which then can be queried for files.
                // All status callbacks come via the cWinHttpSession.
                // Add Normal sockets and UDP to go across the available proxy/tunnel.
                typedef cWinHttpBase SUPER_t;
                friend class cWinHttpRequest;
                friend class cWinHttpSession;
 
        public:
                /* explicit */ cWinHttpSession(LPCWSTR pstrAgent = nullptr,
                        DWORD dwAccessType = WINHTTP_ACCESS_TYPE_DEFAULT_PROXY,
                        IN LPCWSTR pwszProxyName = WINHTTP_NO_PROXY_NAME,
                        IN LPCWSTR pwszProxyBypass = WINHTTP_NO_PROXY_BYPASS,
                        IN DWORD dwFlags = 0);
                ~cWinHttpSession();
 
                cWinHttpConnectionPtr GetHttpConnection(LPCWSTR pstrServer,
                        NET_PORT_t nPort = k_NET_PORT_INVALID, DWORD dwFlags = 0);
 
                bool EnableStatusCallback(bool bEnable = true);
 
                virtual cWinHttpSession* GetSession() const
                {
                        return const_cast<cWinHttpSession*>(this);
                }
 
                // Overridables
                virtual void OnStatusCallback(cWinHttpBase* pContext, DWORD dwInternetStatus,
                        LPVOID lpvStatusInformation, DWORD dwStatusInformationLength);
 
                UNITTEST2_PREDEF(cWinHttp );
 
        protected:
                // WINHTTP_STATUS_CALLBACK
                static void CALLBACK OnStatusCallbackStatic(IN HINTERNET hInternet,
                        IN DWORD_PTR dwContext,
                        IN DWORD dwInternetStatus,
                        IN LPVOID lpvStatusInformation OPTIONAL,
                        IN DWORD dwStatusInformationLength);
 
        public:
                bool m_bCallbackEnabled;
        };
 
        class GRAYLIB_LINK cWinHttpCopier
                : public cWinHttpSession
                , public IFileCopier
        {
                typedef cWinHttpSession SUPER_t;
        public:
                cWinHttpCopier();
                ~cWinHttpCopier();
 
                // IFileCopier
                virtual cStringA get_ConnectName() const override;
                virtual HRESULT RequestFile(const FILECHAR_t* pszSrcName, const FILECHAR_t* pszDestPath, IStreamProgressCallback* pProgress, FILE_SIZE_t nOffsetStart, FILE_SIZE_t* pnRequestSizeEst) override;
                virtual HRESULT SendFile(const FILECHAR_t* pszSrcPath, const FILECHAR_t* pszDestName, IStreamProgressCallback* pProgress, FILE_SIZE_t nOffsetStart, FILE_SIZE_t nSizet) override;
 
                virtual void OnStatusCallback(cWinHttpBase* pContext, DWORD dwInternetStatus,
                        LPVOID lpvStatusInformation, DWORD dwStatusInformationLength);
        };
};
#endif
#endif