mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1404 lines
41 KiB
1404 lines
41 KiB
#include <pch.h>
|
|
#pragma hdrstop
|
|
|
|
#include <wininet.h>
|
|
#include "ssdptypes.h"
|
|
#include "notify.h"
|
|
#include "ssdpnetwork.h"
|
|
#include "ssdpfunc.h"
|
|
#include "ssdpsrv.h"
|
|
#include "event.h"
|
|
#include "ncinet.h"
|
|
#include "ncbase.h"
|
|
#include "upthread.h"
|
|
#include "rundown.h"
|
|
|
|
#define NOTIFY_RESULT_SIZE 2
|
|
static const CHAR c_szDeadNts[] = "upnp:dead";
|
|
|
|
#if DBG
|
|
const DWORD c_csecTimeout = 60 * 3; // 3 minute timeout (debug)
|
|
#else
|
|
const DWORD c_csecTimeout = 60 * 30; // 30 minute timeout
|
|
#endif
|
|
|
|
const TCHAR c_szSubscribe[] = TEXT("SUBSCRIBE");
|
|
const TCHAR c_szUnSubscribe[] = TEXT("UNSUBSCRIBE");
|
|
extern const TCHAR c_szHttpVersion[] = TEXT("HTTP/1.1");
|
|
const TCHAR c_szSubsHeaderFmt[] = TEXT("NT: upnp:event\r\n"
|
|
"Callback: <http://%s:%d/%s>\r\n"
|
|
"Timeout: Second-%d\r\n\r\n");
|
|
const TCHAR c_szReSubsHeaderFmt[] = TEXT("SID: %s\r\nTimeout: Second-%d\r\n");
|
|
const TCHAR c_szUnSubsHeaderFmt[] = TEXT("SID: %s\r\n");
|
|
const TCHAR c_szServer[] = TEXT("Server");
|
|
const TCHAR c_szTimeout[] = TEXT("Timeout");
|
|
const TCHAR c_szSid[] = TEXT("SID");
|
|
|
|
// URI and port of our listener
|
|
const TCHAR c_szNotifyUri[] = TEXT("notify");
|
|
const DWORD c_nPort = 5000;
|
|
|
|
#if DBG
|
|
const DWORD c_csecDefaultTimeout = 60 * 1; // 1 minute is default
|
|
// subscription timeout (debug)
|
|
#else
|
|
const DWORD c_csecDefaultTimeout = 60 * 10; // 10 minutes is default
|
|
// subscription timeout
|
|
#endif
|
|
|
|
|
|
HRESULT HrSendSubscriptionRequest(HINTERNET hin,
|
|
LPCTSTR szUrl,
|
|
LPCTSTR szSid,
|
|
DWORD *pcsecTimeout,
|
|
LPTSTR *pszSidOut,
|
|
ESSR_TYPE essrt);
|
|
|
|
CSsdpPendingNotification::CSsdpPendingNotification()
|
|
{
|
|
ZeroMemory(&m_ssdpRequest, sizeof(m_ssdpRequest));
|
|
}
|
|
|
|
CSsdpPendingNotification::~CSsdpPendingNotification()
|
|
{
|
|
FreeSsdpRequest(&m_ssdpRequest);
|
|
}
|
|
|
|
HRESULT CSsdpPendingNotification::HrInitialize(const SSDP_REQUEST * pRequest)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if(!CopySsdpRequest(&m_ssdpRequest, pRequest))
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpPendingNotification::HrInitialize");
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CSsdpPendingNotification::HrGetRequest(SSDP_REQUEST * pRequest)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if(!CopySsdpRequest(pRequest, &m_ssdpRequest))
|
|
{
|
|
hr = E_FAIL;
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpPendingNotification::HrGetRequest");
|
|
return hr;
|
|
}
|
|
|
|
CSsdpNotifyRequest::CSsdpNotifyRequest() : m_timer(*this), m_hNotifySemaphore(INVALID_HANDLE_VALUE)
|
|
{
|
|
}
|
|
|
|
CSsdpNotifyRequest::~CSsdpNotifyRequest()
|
|
{
|
|
}
|
|
|
|
void CSsdpNotifyRequest::OnRundown(CSsdpNotifyRequest * pNotify)
|
|
{
|
|
CSsdpNotifyRequestManager::Instance().OnRundown(pNotify);
|
|
}
|
|
|
|
class CNotifyRequestTimerAction : public CWorkItem
|
|
{
|
|
public:
|
|
static HRESULT HrCreate(
|
|
CSsdpNotifyRequest * pRequest,
|
|
DWORD dwSecTimeout,
|
|
const CUString & strSid,
|
|
const CUString & strUrl);
|
|
private:
|
|
CNotifyRequestTimerAction(CSsdpNotifyRequest * pRequest, DWORD dwSecTimeout);
|
|
~CNotifyRequestTimerAction();
|
|
CNotifyRequestTimerAction(const CNotifyRequestTimerAction &);
|
|
CNotifyRequestTimerAction & operator=(const CNotifyRequestTimerAction &);
|
|
|
|
DWORD DwRun();
|
|
HRESULT HrIntialize(
|
|
const CUString & strSid,
|
|
const CUString & strUrl);
|
|
|
|
CSsdpNotifyRequest * m_pRequest;
|
|
DWORD m_dwSecTimeout;
|
|
CUString m_strSid;
|
|
CUString m_strUrl;
|
|
};
|
|
|
|
CNotifyRequestTimerAction::CNotifyRequestTimerAction(CSsdpNotifyRequest * pRequest, DWORD dwSecTimeout)
|
|
: m_pRequest(pRequest), m_dwSecTimeout(dwSecTimeout)
|
|
{
|
|
}
|
|
|
|
CNotifyRequestTimerAction::~CNotifyRequestTimerAction()
|
|
{
|
|
}
|
|
|
|
HRESULT CNotifyRequestTimerAction::HrCreate(
|
|
CSsdpNotifyRequest * pRequest,
|
|
DWORD dwSecTimeout,
|
|
const CUString & strSid,
|
|
const CUString & strUrl)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
CNotifyRequestTimerAction * pAction = new CNotifyRequestTimerAction(pRequest, dwSecTimeout);
|
|
if(!pAction)
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = pAction->HrIntialize(strSid, strUrl);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = pAction->HrStart(TRUE);
|
|
}
|
|
if(FAILED(hr))
|
|
{
|
|
delete pAction;
|
|
}
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CNotifyRequestTimerAction::HrCreate");
|
|
return hr;
|
|
}
|
|
|
|
DWORD CNotifyRequestTimerAction::DwRun()
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
char * szSid = NULL;
|
|
char * szUrl = NULL;
|
|
|
|
hr = m_strSid.HrGetMultiByteWithAlloc(&szSid);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = m_strUrl.HrGetMultiByteWithAlloc(&szUrl);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = HrSendSubscriptionRequest(g_hInetSess, szUrl, szSid, &m_dwSecTimeout,
|
|
NULL, SSR_RESUBSCRIBE);
|
|
}
|
|
}
|
|
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = CSsdpNotifyRequestManager::Instance().HrRestartClientResubscribeTimer(
|
|
m_pRequest, m_dwSecTimeout);
|
|
}
|
|
else
|
|
{
|
|
// Failed to send a re-subscribe. We now need to notify clients that
|
|
// a subscription has been lost. Compose a pending notification to
|
|
// let them know this fact.
|
|
//
|
|
SSDP_REQUEST ssdpRequest;
|
|
ZeroMemory(&ssdpRequest, sizeof(ssdpRequest));
|
|
InitializeSsdpRequest(&ssdpRequest);
|
|
ssdpRequest.Headers[GENA_SID] = szSid;
|
|
szSid = NULL;
|
|
hr = HrCopyString(c_szDeadNts, &ssdpRequest.Headers[SSDP_NTS]);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = CSsdpNotifyRequestManager::Instance().HrCheckListNotifyForEvent(&ssdpRequest);
|
|
}
|
|
FreeSsdpRequest(&ssdpRequest);
|
|
}
|
|
|
|
delete [] szSid;
|
|
delete [] szUrl;
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CNotifyRequestTimerAction::DwRun");
|
|
return 0;
|
|
}
|
|
|
|
HRESULT CNotifyRequestTimerAction::HrIntialize(
|
|
const CUString & strSid,
|
|
const CUString & strUrl)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
hr = m_strSid.HrAssign(strSid);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = m_strUrl.HrAssign(strUrl);
|
|
}
|
|
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CNotifyRequestTimerAction::HrIntialize");
|
|
return hr;
|
|
}
|
|
|
|
void CSsdpNotifyRequest::TimerFired()
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
hr = CNotifyRequestTimerAction::HrCreate(this, this->m_dwSecTimeout, this->m_strSid, this->m_strUrl);
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequest::TimerFired");
|
|
}
|
|
|
|
BOOL CSsdpNotifyRequest::TimerTryToLock()
|
|
{
|
|
return m_critSec.FTryEnter();
|
|
}
|
|
|
|
void CSsdpNotifyRequest::TimerUnlock()
|
|
{
|
|
m_critSec.Leave();
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequest::HrRestartClientResubscribeTimer(
|
|
DWORD dwSecTimeout)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
CLock lock(m_critSec);
|
|
|
|
m_dwSecTimeout = dwSecTimeout;
|
|
|
|
// Do 65% of timeout in milliseconds
|
|
DWORD dwTimeoutInMillis = (dwSecTimeout * 65 * 1000) / 100;
|
|
|
|
hr = m_timer.HrResetTimer(dwTimeoutInMillis);
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequest::HrRestartClientResubscribeTimer");
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequest::HrInitializeAlive(
|
|
const char * szNT,
|
|
HANDLE hNotifySemaphore)
|
|
{
|
|
if(!szNT)
|
|
{
|
|
return E_POINTER;
|
|
}
|
|
HRESULT hr = S_OK;
|
|
|
|
m_nt = NOTIFY_ALIVE;
|
|
hr = m_strNT.HrAssign(szNT);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
m_hNotifySemaphore = hNotifySemaphore;
|
|
m_dwSecTimeout = 0;
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequest::HrInitializeAlive");
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequest::HrInitializePropChange(
|
|
const char * szUrl,
|
|
HANDLE hNotifySemaphore)
|
|
{
|
|
if(!szUrl || !*szUrl)
|
|
{
|
|
return E_POINTER;
|
|
}
|
|
HRESULT hr = S_OK;
|
|
|
|
m_nt = NOTIFY_PROP_CHANGE;
|
|
hr = m_strUrl.HrAssign(szUrl);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
m_hNotifySemaphore = hNotifySemaphore;
|
|
m_dwSecTimeout = 0;
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequest::HrInitializePropChange");
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequest::HrSendPropChangeSubscription(SSDP_REGISTER_INFO ** ppRegisterInfo)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
*ppRegisterInfo = NULL;
|
|
|
|
char * szUrl = NULL;
|
|
char * szSid = NULL;
|
|
hr = m_strUrl.HrGetMultiByteWithAlloc(&szUrl);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
TraceTag(ttidEvents, "CSsdpNotifyRequest::HrSendPropChangeSubscription(this=%x) - About to call HrSendSubscriptionRequest", this);
|
|
|
|
if (!g_hInetSess)
|
|
{
|
|
g_hInetSess = HinInternetOpenA("Mozilla/4.0 (compatible; UPnP/1.0; Windows NT/5.1)",
|
|
INTERNET_OPEN_TYPE_DIRECT,
|
|
NULL, NULL, 0);
|
|
}
|
|
|
|
if (g_hInetSess)
|
|
{
|
|
hr = HrSendSubscriptionRequest(g_hInetSess,
|
|
szUrl,
|
|
NULL,
|
|
&m_dwSecTimeout,
|
|
&szSid,
|
|
SSR_SUBSCRIBE);
|
|
}
|
|
else
|
|
{
|
|
hr = E_UNEXPECTED;
|
|
TraceTag(ttidEvents, "CSsdpNotifyRequest::HrSendPropChangeSubscription - HinInternetOpenA failed!");
|
|
}
|
|
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
TraceTag(ttidEvents, "CSsdpNotifyRequest::HrSendPropChangeSubscription(this=%x) - Called HrSendSubscriptionRequest - SID:%s", this, szSid);
|
|
|
|
hr = m_strSid.HrAssign(szSid);
|
|
delete [] szSid;
|
|
}
|
|
delete [] szUrl;
|
|
}
|
|
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
CLock lock(m_critSec);
|
|
|
|
// Do 65% of timeout in milliseconds
|
|
DWORD dwTimeoutInMillis = (m_dwSecTimeout * 65 * 1000) / 100;
|
|
|
|
hr = m_timer.HrSetTimer(dwTimeoutInMillis);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
SSDP_REGISTER_INFO * pRegisterInfo = new SSDP_REGISTER_INFO;
|
|
if(!pRegisterInfo)
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
pRegisterInfo->csecTimeout = m_dwSecTimeout;
|
|
hr = m_strSid.HrGetMultiByteWithAlloc(&pRegisterInfo->szSid);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
*ppRegisterInfo = pRegisterInfo;
|
|
}
|
|
if(FAILED(hr))
|
|
{
|
|
delete pRegisterInfo;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequest::HrSendPropChangeSubscription");
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequest::HrShutdown()
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
CLock lock(m_critSec);
|
|
if(NOTIFY_PROP_CHANGE == m_nt)
|
|
{
|
|
hr = m_timer.HrDelete(INVALID_HANDLE_VALUE);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
char * szUrl = NULL;
|
|
char * szSid = NULL;
|
|
hr = m_strUrl.HrGetMultiByteWithAlloc(&szUrl);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = m_strSid.HrGetMultiByteWithAlloc(&szSid);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = HrSendSubscriptionRequest(g_hInetSess,
|
|
szUrl,
|
|
szSid,
|
|
&m_dwSecTimeout,
|
|
NULL,
|
|
SSR_UNSUBSCRIBE);
|
|
delete [] szSid;
|
|
}
|
|
delete [] szUrl;
|
|
}
|
|
}
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequest::HrShutdown");
|
|
return hr;
|
|
}
|
|
|
|
BOOL CSsdpNotifyRequest::FIsMatchBySemaphore(HANDLE hNotifySemaphore)
|
|
{
|
|
return m_hNotifySemaphore == hNotifySemaphore;
|
|
}
|
|
|
|
BOOL CSsdpNotifyRequest::FIsMatchingEvent(const SSDP_REQUEST * pRequest)
|
|
{
|
|
BOOL bMatch = FALSE;
|
|
|
|
BOOL bIsPossibleMatch = pRequest->Headers[SSDP_NTS] &&
|
|
!lstrcmpiA(pRequest->Headers[SSDP_NTS], "upnp:propchange");
|
|
|
|
bIsPossibleMatch = bIsPossibleMatch && pRequest->Headers[CONTENT_TYPE] &&
|
|
!_strnicmp(pRequest->Headers[CONTENT_TYPE], "text/xml", strlen("text/xml"));
|
|
|
|
bIsPossibleMatch = bIsPossibleMatch && (NOTIFY_PROP_CHANGE == m_nt);
|
|
|
|
if(bIsPossibleMatch)
|
|
{
|
|
// Ensure that we have a valid SID before letting this continue.
|
|
// We enter this critsec so that we will wait until a potential
|
|
// subscription has been sent and a SID received for it before this
|
|
// code continues.
|
|
//
|
|
if(m_strSid.GetLength() && pRequest->Headers[GENA_SID])
|
|
{
|
|
char * szSid = NULL;
|
|
HRESULT hr = m_strSid.HrGetMultiByteWithAlloc(&szSid);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
bMatch = !lstrcmpiA(pRequest->Headers[GENA_SID], szSid);
|
|
delete [] szSid;
|
|
}
|
|
}
|
|
}
|
|
|
|
return bMatch;
|
|
}
|
|
|
|
BOOL CSsdpNotifyRequest::FIsMatchingAliveOrByebye(const SSDP_REQUEST * pRequest)
|
|
{
|
|
BOOL bMatch = FALSE;
|
|
|
|
Assert(!lstrcmpiA(pRequest->Headers[SSDP_NTS], "ssdp:alive") ||
|
|
!lstrcmpiA(pRequest->Headers[SSDP_NTS], "ssdp:byebye"));
|
|
|
|
if(NOTIFY_ALIVE == m_nt)
|
|
{
|
|
char * szNT = NULL;
|
|
HRESULT hr = m_strNT.HrGetMultiByteWithAlloc(&szNT);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
bMatch = !lstrcmpA(szNT, pRequest->Headers[SSDP_NT]);
|
|
delete [] szNT;
|
|
}
|
|
}
|
|
|
|
return bMatch;
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequest::HrQueuePendingNotification(const SSDP_REQUEST * pRequest)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
#if DBG
|
|
|
|
if(SSDP_NOTIFY == pRequest->Method && pRequest->Headers[GENA_SID] && pRequest->Headers[GENA_SEQ])
|
|
{
|
|
TraceTag(ttidEvents,
|
|
"CSsdpNotifyRequest::HrQueuePendingNotification - Event notification for SID:%s",
|
|
pRequest->Headers[GENA_SID]);
|
|
}
|
|
|
|
#endif // DBG
|
|
|
|
CLock lock(m_critSec);
|
|
|
|
PendingNotificationList pendingNotificationList;
|
|
hr = pendingNotificationList.HrPushFrontDefault();
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = pendingNotificationList.Front().HrInitialize(pRequest);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
m_pendingNotificationList.Append(pendingNotificationList);
|
|
TraceTag(ttidEvents, "CSsdpNotifyRequest::HrQueuePendingNotification - releasing semaphore %x", m_hNotifySemaphore);
|
|
|
|
if(!ReleaseSemaphore(m_hNotifySemaphore, 1, NULL))
|
|
{
|
|
hr = HrFromLastWin32Error();
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequest::HrQueuePendingNotification");
|
|
return hr;
|
|
}
|
|
|
|
BOOL CSsdpNotifyRequest::FIsPendingNotification()
|
|
{
|
|
CLock lock(m_critSec);
|
|
|
|
return !m_pendingNotificationList.IsEmpty();
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequest::HrRetreivePendingNotification(MessageList ** ppSvcList)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
CLock lock(m_critSec);
|
|
|
|
// I am just going to return one item
|
|
MessageList * pSvcList = new MessageList;
|
|
if(!pSvcList)
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
pSvcList->list = new SSDP_REQUEST;
|
|
pSvcList->size = 1;
|
|
if(!pSvcList->list)
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
else
|
|
{
|
|
ZeroMemory(pSvcList->list, sizeof(SSDP_REQUEST));
|
|
}
|
|
}
|
|
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
if(m_pendingNotificationList.IsEmpty())
|
|
{
|
|
hr = S_FALSE;
|
|
TraceTag(ttidEvents, "CSsdpNotifyRequest::HrRetreivePendingNotification - no pending notifications!");
|
|
}
|
|
if(S_OK == hr)
|
|
{
|
|
PendingNotificationList pendingNotificationList;
|
|
PendingNotificationList::Iterator iter;
|
|
if(S_OK == m_pendingNotificationList.GetIterator(iter))
|
|
{
|
|
iter.HrMoveToList(pendingNotificationList);
|
|
hr = pendingNotificationList.Front().HrGetRequest(pSvcList->list);
|
|
}
|
|
}
|
|
}
|
|
|
|
if(FAILED(hr))
|
|
{
|
|
if(pSvcList)
|
|
{
|
|
delete pSvcList->list;
|
|
pSvcList->list = NULL;
|
|
pSvcList->size = 0;
|
|
}
|
|
}
|
|
|
|
*ppSvcList = pSvcList;
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequest::HrRetreivePendingNotification");
|
|
return hr;
|
|
}
|
|
|
|
CSsdpNotifyRequestManager CSsdpNotifyRequestManager::s_instance;
|
|
|
|
CSsdpNotifyRequestManager::CSsdpNotifyRequestManager() : m_unTimestamp(0)
|
|
{
|
|
m_hEventTimestamp = CreateEvent(NULL, TRUE, FALSE, NULL);
|
|
}
|
|
|
|
CSsdpNotifyRequestManager::~CSsdpNotifyRequestManager()
|
|
{
|
|
CloseHandle(m_hEventTimestamp);
|
|
}
|
|
|
|
CSsdpNotifyRequestManager & CSsdpNotifyRequestManager::Instance()
|
|
{
|
|
return s_instance;
|
|
}
|
|
|
|
void CSsdpNotifyRequestManager::OnRundown(CSsdpNotifyRequest * pNotify)
|
|
{
|
|
HrRemoveInternal(TRUE, NULL, pNotify);
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequestManager::HrCreateAliveNotifyRequest(
|
|
PCONTEXT_HANDLE_TYPE * ppContextHandle,
|
|
const char * szNT,
|
|
HANDLE hNotifySemaphore)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
CLock lock(m_critSecAliveList);
|
|
|
|
*ppContextHandle = NULL;
|
|
|
|
CSsdpNotifyRequest * pRequest = NULL;
|
|
|
|
NotifyRequestList notifyRequestList;
|
|
hr = notifyRequestList.HrPushFrontDefault();
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = notifyRequestList.Front().HrInitializeAlive(szNT, hNotifySemaphore);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
m_aliveList.Prepend(notifyRequestList);
|
|
*ppContextHandle = &m_aliveList.Front();
|
|
pRequest = &m_aliveList.Front();
|
|
}
|
|
}
|
|
if(SUCCEEDED(hr) && pRequest)
|
|
{
|
|
hr = CSsdpRundownSupport::Instance().HrAddRundownItem(pRequest);
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequestManager::HrCreateAliveNotifyRequest");
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequestManager::HrCreatePropChangeNotifyRequest(
|
|
PCONTEXT_HANDLE_TYPE * ppContextHandle,
|
|
const char * szUrl,
|
|
HANDLE hNotifySemaphore,
|
|
SSDP_REGISTER_INFO ** ppRegisterInfo)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
*ppContextHandle = NULL;
|
|
|
|
CSsdpNotifyRequest * pRequest = NULL;
|
|
|
|
NotifyRequestList notifyRequestList;
|
|
hr = notifyRequestList.HrPushFrontDefault();
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = notifyRequestList.Front().HrInitializePropChange(szUrl, hNotifySemaphore);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
__int64 unTimestamp = 0;
|
|
{
|
|
CLock lock(m_critSecTimestampList);
|
|
hr = m_timestampList.HrPushFront(m_unTimestamp);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
unTimestamp = m_unTimestamp;
|
|
++m_unTimestamp;
|
|
}
|
|
}
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
hr = notifyRequestList.Front().HrSendPropChangeSubscription(ppRegisterInfo);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
CLock lock(m_critSecPropChangeList);
|
|
m_propChangeList.Prepend(notifyRequestList);
|
|
*ppContextHandle = &m_propChangeList.Front();
|
|
pRequest = &m_propChangeList.Front();
|
|
}
|
|
CLock lock(m_critSecTimestampList);
|
|
TimestampList::Iterator iter;
|
|
if(S_OK == m_timestampList.GetIterator(iter))
|
|
{
|
|
__int64 * pun = NULL;
|
|
while(S_OK == iter.HrGetItem(&pun))
|
|
{
|
|
if(*pun == unTimestamp)
|
|
{
|
|
iter.HrErase();
|
|
break;
|
|
}
|
|
if(S_OK != iter.HrNext())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if(!PulseEvent(m_hEventTimestamp))
|
|
{
|
|
hr = HrFromLastWin32Error();
|
|
TraceHr(ttidError, FAL, hr, FALSE, "CSsdpNotifyRequestManager::HrCreatePropChangeNotifyRequest - PulseEvent failed!");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if(SUCCEEDED(hr) && pRequest)
|
|
{
|
|
hr = CSsdpRundownSupport::Instance().HrAddRundownItem(pRequest);
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequestManager::HrCreatePropChangeNotifyRequest");
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequestManager::HrRemoveNotifyRequest(
|
|
HANDLE hNotifySemaphore)
|
|
{
|
|
return HrRemoveInternal(FALSE, hNotifySemaphore, NULL);
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequestManager::HrRemoveNotifyRequestByPointer(
|
|
CSsdpNotifyRequest * pRequest)
|
|
{
|
|
return HrRemoveInternal(FALSE, NULL, pRequest);
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequestManager::HrCheckListNotifyForEvent(const SSDP_REQUEST * pRequest)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
TraceTag(ttidEvents, "CSsdpNotifyRequestManager::HrCheckListNotifyForEvent");
|
|
|
|
__int64 unTimestamp = 0;
|
|
|
|
{
|
|
CLock lock(m_critSecTimestampList);
|
|
unTimestamp = m_unTimestamp;
|
|
}
|
|
|
|
bool bFound = FALSE;
|
|
|
|
while(true)
|
|
{
|
|
{
|
|
CLock lock(m_critSecPropChangeList);
|
|
|
|
NotifyRequestList::Iterator iter;
|
|
if(S_OK == m_propChangeList.GetIterator(iter))
|
|
{
|
|
CSsdpNotifyRequest * pNotifyIter = NULL;
|
|
while(S_OK == iter.HrGetItem(&pNotifyIter))
|
|
{
|
|
if(pNotifyIter->FIsMatchingEvent(pRequest))
|
|
{
|
|
hr = pNotifyIter->HrQueuePendingNotification(pRequest);
|
|
bFound = TRUE;
|
|
break;
|
|
}
|
|
if(S_OK != iter.HrNext())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if(bFound)
|
|
{
|
|
break;
|
|
}
|
|
|
|
{
|
|
BOOL bAllOlder = TRUE;
|
|
|
|
CLock lock(m_critSecTimestampList);
|
|
TimestampList::Iterator iter;
|
|
if(S_OK == m_timestampList.GetIterator(iter))
|
|
{
|
|
__int64 * pun = NULL;
|
|
while(S_OK == iter.HrGetItem(&pun))
|
|
{
|
|
if(*pun < unTimestamp)
|
|
{
|
|
bAllOlder = FALSE;
|
|
break;
|
|
}
|
|
if(S_OK != iter.HrNext())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if(bAllOlder)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
WaitForSingleObject(m_hEventTimestamp, 2000);
|
|
}
|
|
|
|
#if DBG
|
|
|
|
if(!bFound)
|
|
{
|
|
TraceTag(ttidEvents, "CSsdpNotifyRequestManager::HrCheckListNotifyForEvent - not found! SID:%s",
|
|
pRequest->Headers[GENA_SID] ? pRequest->Headers[GENA_SID] : "<Unknown>");
|
|
}
|
|
|
|
#endif // DBG
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequestManager::HrCheckListNotifyForEvent");
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequestManager::HrCheckListNotifyForAliveOrByebye(const SSDP_REQUEST * pRequest)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
CLock lock(m_critSecAliveList);
|
|
|
|
NotifyRequestList::Iterator iter;
|
|
if(S_OK == m_aliveList.GetIterator(iter))
|
|
{
|
|
CSsdpNotifyRequest * pNotifyIter = NULL;
|
|
while(S_OK == iter.HrGetItem(&pNotifyIter))
|
|
{
|
|
if(pNotifyIter->FIsMatchingAliveOrByebye(pRequest))
|
|
{
|
|
hr = pNotifyIter->HrQueuePendingNotification(pRequest);
|
|
if(FAILED(hr))
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
if(S_OK != iter.HrNext())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequestManager::HrCheckListNotifyForAliveOrByebye");
|
|
return hr;
|
|
}
|
|
|
|
BOOL CSsdpNotifyRequestManager::FIsAliveOrByebyeInListNotify(const SSDP_REQUEST * pRequest)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
CLock lock(m_critSecAliveList);
|
|
BOOL bRet = FALSE;
|
|
|
|
NotifyRequestList::Iterator iter;
|
|
if(S_OK == m_aliveList.GetIterator(iter))
|
|
{
|
|
CSsdpNotifyRequest * pNotifyIter = NULL;
|
|
while(S_OK == iter.HrGetItem(&pNotifyIter))
|
|
{
|
|
if(pNotifyIter->FIsMatchingAliveOrByebye(pRequest))
|
|
{
|
|
bRet = TRUE;
|
|
break;
|
|
}
|
|
if(S_OK != iter.HrNext())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequestManager::FIsAliveOrByebyeInListNotify");
|
|
return bRet;
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequestManager::HrRetreivePendingNotification(
|
|
HANDLE hNotifySemaphore,
|
|
MessageList ** ppSvcList)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
BOOL bFound = FALSE;
|
|
|
|
{
|
|
CLock lock(m_critSecAliveList);
|
|
NotifyRequestList::Iterator iter;
|
|
if(S_OK == m_aliveList.GetIterator(iter))
|
|
{
|
|
CSsdpNotifyRequest * pNotifyIter = NULL;
|
|
while(S_OK == iter.HrGetItem(&pNotifyIter))
|
|
{
|
|
if(pNotifyIter->FIsMatchBySemaphore(hNotifySemaphore) && pNotifyIter->FIsPendingNotification())
|
|
{
|
|
hr = pNotifyIter->HrRetreivePendingNotification(ppSvcList);
|
|
bFound = TRUE;
|
|
break;
|
|
}
|
|
if(S_OK != iter.HrNext())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if(!bFound)
|
|
{
|
|
CLock lock(m_critSecPropChangeList);
|
|
NotifyRequestList::Iterator iter;
|
|
if(S_OK == m_propChangeList.GetIterator(iter))
|
|
{
|
|
CSsdpNotifyRequest * pNotifyIter = NULL;
|
|
while(S_OK == iter.HrGetItem(&pNotifyIter))
|
|
{
|
|
if(pNotifyIter->FIsMatchBySemaphore(hNotifySemaphore) && pNotifyIter->FIsPendingNotification())
|
|
{
|
|
hr = pNotifyIter->HrRetreivePendingNotification(ppSvcList);
|
|
bFound = TRUE;
|
|
break;
|
|
}
|
|
if(S_OK != iter.HrNext())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// See if this is a bogus release
|
|
if(!bFound)
|
|
{
|
|
*ppSvcList = new MessageList;
|
|
if(!*ppSvcList)
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
(*ppSvcList)->size = 0;
|
|
(*ppSvcList)->list = NULL;
|
|
}
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequestManager::HrRetreivePendingNotification");
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequestManager::HrRestartClientResubscribeTimer(
|
|
CSsdpNotifyRequest * pRequest,
|
|
DWORD dwSecTimeout)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
CLock lock(m_critSecPropChangeList);
|
|
|
|
NotifyRequestList::Iterator iter;
|
|
if(S_OK == m_propChangeList.GetIterator(iter))
|
|
{
|
|
CSsdpNotifyRequest * pNotifyIter = NULL;
|
|
while(S_OK == iter.HrGetItem(&pNotifyIter))
|
|
{
|
|
if(pNotifyIter == pRequest)
|
|
{
|
|
hr = pNotifyIter->HrRestartClientResubscribeTimer(dwSecTimeout);
|
|
break;
|
|
}
|
|
if(S_OK != iter.HrNext())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequestManager::HrRestartClientResubscribeTimer");
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CSsdpNotifyRequestManager::HrRemoveInternal(BOOL bRundown, HANDLE hNotifySemaphore, CSsdpNotifyRequest * pRequest)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
NotifyRequestList notifyRequestListRemove;
|
|
|
|
{
|
|
CLock lock(m_critSecAliveList);
|
|
|
|
NotifyRequestList::Iterator iter;
|
|
if(S_OK == m_aliveList.GetIterator(iter))
|
|
{
|
|
CSsdpNotifyRequest * pNotifyIter = NULL;
|
|
while(S_OK == iter.HrGetItem(&pNotifyIter))
|
|
{
|
|
Assert(!hNotifySemaphore == !!pRequest); // One or the other
|
|
if((hNotifySemaphore && pNotifyIter->FIsMatchBySemaphore(hNotifySemaphore)) ||
|
|
(pRequest && pNotifyIter == pRequest))
|
|
{
|
|
if(S_OK != iter.HrMoveToList(notifyRequestListRemove))
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
if(S_OK != iter.HrNext())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
{
|
|
CLock lock(m_critSecPropChangeList);
|
|
|
|
NotifyRequestList::Iterator iter;
|
|
if(S_OK == m_propChangeList.GetIterator(iter))
|
|
{
|
|
CSsdpNotifyRequest * pNotifyIter = NULL;
|
|
while(S_OK == iter.HrGetItem(&pNotifyIter))
|
|
{
|
|
Assert(!hNotifySemaphore == !!pRequest); // One or the other
|
|
if((hNotifySemaphore && pNotifyIter->FIsMatchBySemaphore(hNotifySemaphore)) ||
|
|
(pRequest && pNotifyIter == pRequest))
|
|
{
|
|
if(S_OK != iter.HrMoveToList(notifyRequestListRemove))
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
if(S_OK != iter.HrNext())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Delete the items outside of the lock
|
|
{
|
|
NotifyRequestList::Iterator iter;
|
|
if(S_OK == notifyRequestListRemove.GetIterator(iter))
|
|
{
|
|
CSsdpNotifyRequest * pNotifyIter = NULL;
|
|
while(S_OK == iter.HrGetItem(&pNotifyIter))
|
|
{
|
|
if(!bRundown)
|
|
{
|
|
CSsdpRundownSupport::Instance().RemoveRundownItem(pNotifyIter);
|
|
}
|
|
hr = pNotifyIter->HrShutdown();
|
|
if(S_OK != iter.HrNext())
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TraceHr(ttidEvents, FAL, hr, FALSE, "CSsdpNotifyRequestManager::HrRemoveInternal");
|
|
return hr;
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: DwParseTime
|
|
//
|
|
// Purpose: Parses the Timeout header for a subscription
|
|
//
|
|
// Arguments:
|
|
// szTime [in] Timeout value in the format defined by RFC2518
|
|
//
|
|
// Returns: Timeout value in SECONDS
|
|
//
|
|
// Author: danielwe 13 Oct 1999
|
|
//
|
|
// Notes: NYI
|
|
//
|
|
DWORD DwParseTime(LPCTSTR szTime)
|
|
{
|
|
TCHAR szDigits[64];
|
|
const TCHAR c_szTimeout[] = TEXT("Second-");
|
|
const INT c_cchTimeout = lstrlen(c_szTimeout);
|
|
DWORD iDigit = 0;
|
|
|
|
if (szTime && (lstrlen(szTime) > c_cchTimeout))
|
|
{
|
|
if (!_strnicmp(szTime, c_szTimeout, c_cchTimeout))
|
|
{
|
|
DWORD dwDigits;
|
|
|
|
// Ok we know we have at least "Timeout-x" now
|
|
szTime += c_cchTimeout;
|
|
|
|
*szDigits = 0;
|
|
|
|
while (isdigit(*szTime) && (iDigit < sizeof(szDigits) - 1))
|
|
{
|
|
// Copy the digits into the buffer
|
|
szDigits[iDigit++] = *szTime++;
|
|
}
|
|
szDigits[iDigit] = TEXT('\0');
|
|
|
|
dwDigits = _tcstoul(szDigits, NULL, 10);
|
|
|
|
if (dwDigits)
|
|
{
|
|
return dwDigits;
|
|
}
|
|
else
|
|
{
|
|
return c_csecDefaultTimeout;
|
|
}
|
|
}
|
|
}
|
|
|
|
TraceTag(ttidEvents, "DwParseTime: Invalid timeout header %s. Returning "
|
|
"default timeout of %d", szTime ? szTime : "<NULL>",
|
|
c_csecDefaultTimeout);
|
|
|
|
return c_csecDefaultTimeout;
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: HrQueryHeader
|
|
//
|
|
// Purpose: Helper function to query a header from an HTTP response
|
|
//
|
|
// Arguments:
|
|
// hinR [in] Handle to request
|
|
// szHeader [in] Header to query
|
|
// pszValue [out] Returns header data
|
|
//
|
|
// Returns: S_OK if success, or ERROR_INTERNET_* error if failed
|
|
//
|
|
// Author: danielwe 18 Oct 1999
|
|
//
|
|
// Notes: Caller should free pszValue when done with it
|
|
//
|
|
HRESULT HrQueryHeader(HINTERNET hinR, LPCTSTR szHeader, LPTSTR *pszValue)
|
|
{
|
|
DWORD cbBuffer = 0;
|
|
HRESULT hr = S_OK;
|
|
|
|
// First get the header length
|
|
//
|
|
hr = HrHttpQueryInfo(hinR, HTTP_QUERY_CUSTOM, (LPTSTR)szHeader,
|
|
&cbBuffer, 0);
|
|
if (HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER) == hr)
|
|
{
|
|
DWORD cbBuffLen = 0;
|
|
cbBuffLen = _tcslen(szHeader);
|
|
cbBuffer = ( cbBuffer > cbBuffLen ) ? cbBuffer : cbBuffLen ;
|
|
|
|
*pszValue = (LPTSTR) malloc(cbBuffer + sizeof(TCHAR));
|
|
if (*pszValue)
|
|
{
|
|
lstrcpy(*pszValue, szHeader);
|
|
hr = HrHttpQueryInfo(hinR, HTTP_QUERY_CUSTOM, *pszValue, &cbBuffer, 0);
|
|
}
|
|
else
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
AssertSz(FAILED(hr), "First call to HttpQueryInfo must fail!");
|
|
}
|
|
|
|
TraceError("HrQueryHeader", hr);
|
|
return hr;
|
|
}
|
|
|
|
HRESULT HrHttpQueryStatusCode(HINTERNET hinR, DWORD *pdwStatus)
|
|
{
|
|
HRESULT hr;
|
|
DWORD cbBuf = sizeof(*pdwStatus);
|
|
Assert(pdwStatus);
|
|
|
|
*pdwStatus = 0;
|
|
|
|
hr = HrHttpQueryInfo(hinR, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER,
|
|
pdwStatus, &cbBuf, NULL);
|
|
|
|
TraceError("HrHttpQueryStatusCode", hr);
|
|
return hr;
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: HrSendSubscriptionRequest
|
|
//
|
|
// Purpose: Sends a SUBSCRIBE request based on the data contained within
|
|
// pRequest.
|
|
//
|
|
// Arguments:
|
|
// hin [in] Handle to internet session returned from InternetOpen
|
|
// pRequest [in] Pointer to SSDP_NOTIFY_REQUEST containing information
|
|
// about the subscription that needs to be sent
|
|
// essrt [in] Can be one of:
|
|
// SSR_SUBSCRIBE - sends a subscription request
|
|
// SSR_RESUBSCRIBE - sends a re-subscription request
|
|
// SSR_UNSUBSCRIBE - sends an unsubscription request
|
|
//
|
|
// Returns: S_OK if success, or ERROR_INTERNET_* error if failed
|
|
//
|
|
// Author: danielwe 18 Oct 1999
|
|
//
|
|
// Notes:
|
|
//
|
|
HRESULT HrSendSubscriptionRequest(HINTERNET hin,
|
|
LPCTSTR szUrl,
|
|
LPCTSTR szSid,
|
|
DWORD *pcsecTimeout,
|
|
LPTSTR *pszSidOut,
|
|
ESSR_TYPE essrt)
|
|
{
|
|
HINTERNET hinC;
|
|
INTERNET_PORT ipPort;
|
|
URL_COMPONENTS urlComp = {0};
|
|
TCHAR szHostName[INTERNET_MAX_HOST_NAME_LENGTH];
|
|
TCHAR szUrlPath[INTERNET_MAX_URL_LENGTH];
|
|
HRESULT hr = S_OK;
|
|
|
|
Assert(pcsecTimeout);
|
|
|
|
// Parse the server name out of the URL
|
|
//
|
|
urlComp.dwStructSize = sizeof(URL_COMPONENTS);
|
|
urlComp.lpszHostName = (LPTSTR) &szHostName;
|
|
urlComp.dwHostNameLength = INTERNET_MAX_HOST_NAME_LENGTH;
|
|
urlComp.lpszUrlPath = (LPTSTR) &szUrlPath;
|
|
urlComp.dwUrlPathLength = INTERNET_MAX_URL_LENGTH;
|
|
|
|
hr = HrInternetCrackUrlA(szUrl, 0, 0, &urlComp);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hinC = HinInternetConnectA(hin, szHostName, urlComp.nPort, NULL, NULL,
|
|
INTERNET_SERVICE_HTTP, 0, 0);
|
|
if (hinC)
|
|
{
|
|
HINTERNET hinR;
|
|
LPCTSTR szMethod = c_szSubscribe;
|
|
|
|
if (SSR_UNSUBSCRIBE == essrt)
|
|
{
|
|
szMethod = c_szUnSubscribe;
|
|
}
|
|
|
|
hinR = HinHttpOpenRequestA(hinC, szMethod, szUrlPath,
|
|
c_szHttpVersion, NULL, NULL, 0, 0);
|
|
if (hinR)
|
|
{
|
|
TCHAR szHeaders[1024];
|
|
|
|
if (SSR_RESUBSCRIBE == essrt)
|
|
{
|
|
wsprintf(szHeaders, c_szReSubsHeaderFmt, szSid,
|
|
*pcsecTimeout);
|
|
}
|
|
else if (SSR_SUBSCRIBE == essrt)
|
|
{
|
|
SOCKADDR_IN sockIn;
|
|
ULONG nAddr;
|
|
|
|
ZeroMemory(&sockIn, sizeof(SOCKADDR_IN));
|
|
hr = GetIpAddress(szHostName, &sockIn);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
nAddr = sockIn.sin_addr.s_addr;
|
|
wsprintf(szHeaders, c_szSubsHeaderFmt, INET_NTOA(nAddr),
|
|
c_nPort, c_szNotifyUri, c_csecTimeout);
|
|
}
|
|
}
|
|
else if (SSR_UNSUBSCRIBE == essrt)
|
|
{
|
|
wsprintf(szHeaders, c_szUnSubsHeaderFmt, szSid);
|
|
}
|
|
|
|
TraceTag(ttidEvents, "Sending request to %s/%s:%d",
|
|
szHostName, szUrlPath, urlComp.nPort);
|
|
TraceTag(ttidEvents, "Sending %s request:",
|
|
essrt == SSR_SUBSCRIBE ? "subscription" :
|
|
essrt == SSR_RESUBSCRIBE ? "re-subscription" :
|
|
essrt == SSR_UNSUBSCRIBE ? "unsubscribe" : "Unknown!");
|
|
TraceTag(ttidEvents, "-----------------------------");
|
|
TraceTag(ttidEvents, "%s", szHeaders);
|
|
TraceTag(ttidEvents, "-----------------------------");
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = HrHttpSendRequestA(hinR, szHeaders, -1, NULL, 0);
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
LPTSTR szTimeout = NULL;
|
|
LPTSTR szServer = NULL;
|
|
DWORD dwStatus;
|
|
|
|
hr = HrHttpQueryStatusCode(hinR, &dwStatus);
|
|
if (SUCCEEDED(hr) &&
|
|
(dwStatus == HTTP_STATUS_OK) &&
|
|
(SSR_UNSUBSCRIBE != essrt))
|
|
{
|
|
// First get the server header. We don't care what's
|
|
// in it, just that it was present
|
|
//
|
|
hr = HrQueryHeader(hinR, c_szServer, &szServer);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
TraceTag(ttidEvents, "Server header contained: %s",
|
|
szServer);
|
|
|
|
// Get the SID and Timeout headers from the response
|
|
//
|
|
hr = HrQueryHeader(hinR, c_szTimeout, &szTimeout);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
*pcsecTimeout = DwParseTime(szTimeout);
|
|
|
|
TraceTag(ttidEvents, "Subscribe response has %d for "
|
|
"Timeout", *pcsecTimeout);
|
|
|
|
if (essrt == SSR_RESUBSCRIBE)
|
|
{
|
|
LPTSTR szSidNew = NULL;
|
|
|
|
Assert(szSid);
|
|
|
|
// Handle re-subscription response
|
|
|
|
hr = HrQueryHeader(hinR, c_szSid, &szSidNew);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
AssertSz(!lstrcmpi(szSidNew, szSid),
|
|
"SID from re-subscribe response isn't "
|
|
"the same!");
|
|
|
|
TraceTag(ttidEvents, "Resubscribe response has %s for "
|
|
"SID", szSidNew);
|
|
|
|
free(szSidNew);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Handle subscription response
|
|
|
|
Assert(pszSidOut);
|
|
|
|
hr = HrQueryHeader(hinR, c_szSid, pszSidOut);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
TraceTag(ttidEvents, "Subscribe response has %s for "
|
|
"SID", *pszSidOut);
|
|
|
|
}
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
TraceTag(ttidEvents, "Didn't get SID header from "
|
|
"subscription response!");
|
|
}
|
|
|
|
free(szTimeout);
|
|
}
|
|
|
|
free(szServer);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (SUCCEEDED(hr) && dwStatus != HTTP_STATUS_OK)
|
|
{
|
|
TraceTag(ttidError, "HrSendSubscriptionRequest - Received the following error code (%d)", dwStatus);
|
|
hr = E_FAIL;
|
|
}
|
|
}
|
|
}
|
|
|
|
HrInternetCloseHandle(hinR);
|
|
}
|
|
|
|
HrInternetCloseHandle(hinC);
|
|
}
|
|
else
|
|
{
|
|
hr = HrFromLastWin32Error();
|
|
}
|
|
}
|
|
|
|
TraceError("HrSendSubscriptionRequest", hr);
|
|
return hr;
|
|
}
|