/****************************************************************************
|
*
|
* Copyright (C) 2000-2001 RealNetworks, Inc. All rights reserved.
|
*
|
* This program is free software. It may be distributed under the terms
|
* in the file LICENSE, found in the top level of the source distribution.
|
*
|
*/
|
|
#ifndef _RTSPMSG_H
|
#define _RTSPMSG_H
|
|
#include "types.h"
|
#include "str.h"
|
#include "tlist.h"
|
#include "TypeDef.h"
|
#include <strings.h>
|
#include "MD5Calc.h"
|
#include "tool.h"
|
|
// These must correspond with s_pVerbs indices
|
enum RtspVerb {
|
VERB_NONE,
|
VERB_ANNOUNCE,
|
VERB_DESCRIBE,
|
VERB_GETPARAM,
|
VERB_OPTIONS,
|
VERB_PAUSE,
|
VERB_PLAY,
|
VERB_RECORD,
|
VERB_REDIRECT,
|
VERB_SETUP,
|
VERB_SETPARAM,
|
VERB_TEARDOWN,
|
// extensions here
|
VERB_LAST
|
};
|
|
enum RtspMsgType {
|
RTSP_TYPE_NONE,
|
RTSP_TYPE_REQUEST,
|
RTSP_TYPE_RESPONSE,
|
RTSP_TYPE_LAST
|
};
|
|
class CAuthenticator: public CMD5Calc
|
{
|
public:
|
CAuthenticator(void);
|
~CAuthenticator(void);
|
|
void reset();
|
void setRealmAndNonce(char const* realm, char const* nonce);
|
void setUsernameAndPassword(char const* username, char const* password, bool passwordIsMD5 = false);
|
void reclaimDigestResponse(char const* responseStr);
|
|
char const* realm() const { return fRealm; }
|
char const* nonce() const { return fNonce; }
|
char const* username() const { return fUsername; }
|
char const* password() const { return fPassword; }
|
|
char const* computeDigestResponse(char *cmd, char *url);
|
private:
|
void resetRealmAndNonce();
|
void resetUsernameAndPassword();
|
void assignRealmAndNonce(char const* realm, char const* nonce);
|
void assignUsernameAndPassword(char const* username, char const* password, bool passwordIsMD5);
|
|
private:
|
char *fRealm;
|
char *fNonce;
|
char *fUsername;
|
char *fPassword;
|
bool fPasswordIsMD5;
|
};
|
|
class CUserData
|
{
|
public:
|
CUserData(void)
|
{
|
HostIp[0] = '\0';
|
Port = 0;
|
SockFd = 0;
|
ServerPort = 0;
|
MulticastIp[0] = '\0';
|
MulticastPort = 0;
|
RtpSSrc = 0;
|
|
AuthRealm[0] = '\0';
|
AuthNonce[0] = '\0';
|
}
|
~CUserData(void) {}
|
|
//Remote host ip and port
|
char HostIp[16+1];
|
UINT16 Port;
|
CString TransID;
|
UINT16 ServerPort;
|
|
char MulticastIp[16+1];
|
UINT16 MulticastPort;
|
unsigned int RtpSSrc;
|
|
int SockFd;
|
|
char AuthRealm[64+1];
|
char AuthNonce[64+1];
|
};
|
|
class CRtspHdr
|
{
|
private:
|
CRtspHdr(void);
|
|
public:
|
CRtspHdr(const CString& strKey);
|
CRtspHdr(const CString& strKey, const CString& strVal);
|
|
const CString& GetKey(void) const;
|
const CString& GetVal(void) const;
|
void SetVal(const CString& strVal);
|
|
protected:
|
CString m_strKey;
|
CString m_strVal;
|
};
|
typedef TDoubleList<CRtspHdr*> CRtspHdrList;
|
|
class CRtspMsg
|
{
|
private:
|
bool operator==(const CRtspMsg& other) const;
|
bool operator!=(const CRtspMsg& other) const;
|
|
public:
|
CRtspMsg(void);
|
CRtspMsg(const CRtspMsg& other);
|
virtual ~CRtspMsg(void);
|
|
const CRtspMsg& operator=(const CRtspMsg& other);
|
|
operator CPByte(void) const;
|
Byte operator[](UINT32 nPos) const;
|
|
virtual RtspMsgType GetType(void) const;
|
|
// Total header length for key/val pairs (incl. ": " and CRLF)
|
// but NOT separator CRLF
|
size_t GetHdrLen(void) const;
|
size_t GetBufLen(void) const;
|
|
Byte GetAt(UINT32 nPos) const;
|
void SetAt(UINT32 nPos, Byte by);
|
|
void GetRtspVer(UINT32* puMajor, UINT32* puMinor) const;
|
void SetRtspVer(UINT32 uMajor, UINT32 uMinor);
|
|
size_t GetHdrCount(void) const;
|
CString GetHdr(const CString& strKey) const;
|
CRtspHdr* GetHdr(UINT32 nIndex) const;
|
void SetHdr(const CString& strKey, const CString& strVal);
|
void SetHdr(const CRtspHdr& hdrNew);
|
|
PByte GetBuf(void) const;
|
void SetBuf(CPByte buf, size_t nLen);
|
|
public:
|
void SetRemoteHost(const char FromIP[], UINT16 FromPort)
|
{
|
snprintf(mUserData.HostIp, 16, FromIP);
|
mUserData.Port = FromPort;
|
}
|
void SetSockFd(int sockfd)
|
{
|
mUserData.SockFd = sockfd;
|
}
|
|
void SetRealmAndNonce(char *realm, char *nonce)
|
{
|
if (realm != NULL && strlen(realm) > 0)
|
{
|
strcpy(mUserData.AuthRealm, realm);
|
}
|
if (nonce != NULL && strlen(nonce) > 0)
|
{
|
strcpy(mUserData.AuthNonce, nonce);
|
}
|
}
|
|
CUserData mUserData;
|
protected:
|
UINT32 m_nRtspVer; // RTSP version (hiword.loword)
|
UINT32 m_nSeq;
|
CRtspHdrList m_listHdrs;
|
size_t m_nBufLen;
|
PByte m_pbuf;
|
};
|
typedef TDoubleList<CRtspMsg*> CRtspMsgList;
|
|
class CRtspRequestMsg : public CRtspMsg
|
{
|
public:
|
class CUrlParam
|
{
|
public:
|
CUrlParam(void)
|
{
|
DevAor[0] = '\0';
|
Type = -1;
|
BeginTime[0] = '\0';
|
EndTime[0] = '\0';
|
StreamID = -1;
|
SessionID[0] = '\0';
|
StreamType = 0;
|
}
|
~CUrlParam(void) {}
|
|
//Get parameter to save by different keywords from input line
|
void LineTo(char Line[])
|
{
|
if (0 == strlen(Line) )
|
return;
|
|
char* p = Line;
|
if (0 == strncmp(p, "DevAor", 6) )
|
{
|
p += strlen("DevAor");
|
while ( (*p != '\0') && ( ('=' == *p) || (' ' == *p) ) )
|
p++;
|
if (*p != '\0')
|
{
|
strncpy2(DevAor, p, SIP_AOR_MAX_SIZE);
|
}
|
}
|
else if (0 == strncasecmp(p, "Type", 4) )
|
{
|
p += strlen("Type");
|
while ( (*p != '\0') && ( ('=' == *p) || (' ' == *p) ) )
|
p++;
|
if (*p != '\0')
|
{
|
Type = atoi(p);
|
}
|
}
|
else if (0 == strncasecmp(p, "sessionid", 9))
|
{
|
p += strlen("sessionid");
|
while ( (*p != '\0') && ( ('=' == *p) || (' ' == *p) ) )
|
p++;
|
if (*p != '\0')
|
{
|
strncpy2(SessionID, p, HTTP_URL_MAX_LEN);
|
}
|
}
|
else if (0 == strncasecmp(p, "BeginTime", 9) )
|
{
|
p += strlen("BeginTime");
|
while ( (*p != '\0') && ( ('=' == *p) || (' ' == *p) ) )
|
p++;
|
if (*p != '\0')
|
{
|
strncpy2(BeginTime, p, DATE_TIME_MAX_LEN);
|
//Delete '.' from time string
|
p = BeginTime;
|
while ( (*p != '\0') && (*p != '.') )
|
p++;
|
if ('.' == *p)
|
*p = 0x20; //0x2E = ' '
|
}
|
}
|
else if (0 == strncasecmp(p, "EndTime", 7) )
|
{
|
p += strlen("EndTime");
|
while ( (*p != '\0') && ( ('=' == *p) || (' ' == *p) ) )
|
p++;
|
if (*p != '\0')
|
{
|
strncpy2(EndTime, p, DATE_TIME_MAX_LEN);
|
//Delete '.' from time string
|
p = EndTime;
|
while ( (*p != '\0') && (*p != '.') )
|
p++;
|
if ('.' == *p)
|
*p = 0x20; //0x2E = ' '
|
}
|
}
|
else if (0 == strncasecmp(p, "trackID", 7) )
|
{
|
p += strlen("trackID");
|
while ( (*p != '\0') && ( ('=' == *p) || (' ' == *p) ) )
|
p++;
|
if (*p != '\0')
|
{
|
StreamID = atoi(p);
|
}
|
}
|
else if (0 == strncasecmp(p, "StreamType", 10))
|
{
|
p += strlen("StreamType");
|
while ( (*p != '\0') && ( ('=' == *p) || (' ' == *p) ) )
|
p++;
|
if (*p != '\0')
|
{
|
StreamType = atoi(p);
|
}
|
}
|
else if (0 == strncasecmp(p, "PlayType", 8))
|
{
|
p += strlen("PlayType");
|
while ( (*p != '\0') && ( ('=' == *p) || (' ' == *p) ) )
|
p++;
|
if (*p != '\0')
|
{
|
Type = atoi(p);
|
}
|
}
|
else
|
{
|
printf("%s: Failed to check keywords Line<%s>! \n", __FUNCTION__, Line);
|
}
|
}
|
|
public:
|
char DevAor[SIP_AOR_MAX_SIZE+1];
|
int Type;
|
char BeginTime[DATE_TIME_MAX_LEN+1];
|
char EndTime[DATE_TIME_MAX_LEN+1];
|
int StreamID;
|
char SessionID[HTTP_URL_MAX_LEN+1];
|
int StreamType;
|
};
|
|
public:
|
CRtspRequestMsg(void);
|
CRtspRequestMsg(const CRtspRequestMsg& other);
|
virtual ~CRtspRequestMsg(void);
|
|
const CRtspRequestMsg& operator=(const CRtspRequestMsg& other);
|
|
virtual RtspMsgType GetType(void) const;
|
|
RtspVerb GetVerb(void) const;
|
CPCHAR GetVerbStr(void) const;
|
void SetVerb(RtspVerb verb);
|
void SetVerb(CPCHAR szVerb);
|
|
CPCHAR GetUrl(void) const;
|
void SetUrl(const CString& strUrl);
|
void GetFileName(char filename[64]);
|
|
public:
|
void ParserUri(void);
|
SOCKETTYPE_E ParserClientStreamTransType(void);
|
bool ParserClientAddr(char * ClientAddr);
|
UINT16 ParserClientPort(void);
|
bool ParserClientMulticast(void);
|
UINT16 ParserClientMulticastPort(void);
|
|
CUrlParam UrlParam;
|
|
protected:
|
enum RtspVerb m_verb;
|
CString m_strUrl;
|
};
|
typedef TDoubleList<CRtspRequestMsg*> CRtspRequestMsgList;
|
|
class CRtspResponseMsg : public CRtspMsg
|
{
|
public:
|
CRtspResponseMsg(void);
|
CRtspResponseMsg(const CRtspResponseMsg& other);
|
virtual ~CRtspResponseMsg(void);
|
|
const CRtspResponseMsg& operator=(const CRtspResponseMsg& other);
|
|
virtual RtspMsgType GetType(void) const;
|
|
UINT32 GetStatusCode(void) const;
|
const CString& GetStatusMsg(void) const;
|
void SetStatus(UINT32 nCode, CPCHAR szMsg = NULL);
|
|
protected:
|
UINT32 m_nCode;
|
CString m_strStatusMsg;
|
};
|
typedef TDoubleList<CRtspResponseMsg*> CRtspResponseMsgList;
|
|
#endif //ndef _RTSPMSG_H
|
