forked from mirrors/gecko-dev
70ba2956d1
MozReview-Commit-ID: 4hsrKYlV0ER --HG-- extra : transplant_source : %0BF%BB%02%B5%C4%D9H%DC%07%FC%BD%BDQ7%5Dr%AD%19%F4
408 lines
10 KiB
C++
408 lines
10 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
// HttpLog.h should generally be included first
|
|
#include "HttpLog.h"
|
|
|
|
#ifdef MOZ_CRASHREPORTER
|
|
#include "nsExceptionHandler.h"
|
|
#endif
|
|
#include "nsHttpRequestHead.h"
|
|
#include "nsIHttpHeaderVisitor.h"
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// nsHttpRequestHead
|
|
//-----------------------------------------------------------------------------
|
|
|
|
namespace mozilla {
|
|
namespace net {
|
|
|
|
#ifdef MOZ_CRASHREPORTER
|
|
|
|
void nsHttpRequestHead::DbgReentrantMonitorAutoEnter::DbgCheck(bool aIn)
|
|
{
|
|
nsHttpAtom header;
|
|
if (!mInstance.mAnnotated && mInstance.mHeaders.Count() &&
|
|
!mInstance.mHeaders.PeekHeaderAt(0, header)) {
|
|
nsAutoCString str;
|
|
str.Append(nsPrintfCString("%s %s", aIn ? "in" : "out", mFunc));
|
|
// Output the content of the array header and the first nsEntry.
|
|
const uint8_t* p = reinterpret_cast<uint8_t*>
|
|
(mInstance.mHeaders.mHeaders.Elements()) - sizeof(nsTArrayHeader);
|
|
for (int i = 0; i < 28; ++i, ++p) {
|
|
str.Append(nsPrintfCString(" %02x", *p));
|
|
}
|
|
CrashReporter::AnnotateCrashReport(
|
|
NS_LITERAL_CSTRING("InvalidHttpHeaderArray"), str);
|
|
// Make sure we annotate only when we found it is invalid at the first
|
|
// time.
|
|
mInstance.mAnnotated = true;
|
|
}
|
|
}
|
|
|
|
#define ReentrantMonitorAutoEnter DbgReentrantMonitorAutoEnter
|
|
#define mon(x) mon(*this, __func__)
|
|
|
|
#endif
|
|
|
|
nsHttpRequestHead::nsHttpRequestHead()
|
|
: mMethod(NS_LITERAL_CSTRING("GET"))
|
|
, mVersion(NS_HTTP_VERSION_1_1)
|
|
, mParsedMethod(kMethod_Get)
|
|
, mHTTPS(false)
|
|
, mReentrantMonitor("nsHttpRequestHead.mReentrantMonitor")
|
|
, mInVisitHeaders(false)
|
|
#ifdef MOZ_CRASHREPORTER
|
|
, mAnnotated(false)
|
|
#endif
|
|
{
|
|
MOZ_COUNT_CTOR(nsHttpRequestHead);
|
|
}
|
|
|
|
nsHttpRequestHead::~nsHttpRequestHead()
|
|
{
|
|
MOZ_COUNT_DTOR(nsHttpRequestHead);
|
|
}
|
|
|
|
// Don't use this function. It is only used by HttpChannelParent to avoid
|
|
// copying of request headers!!!
|
|
const nsHttpHeaderArray &
|
|
nsHttpRequestHead::Headers() const
|
|
{
|
|
nsHttpRequestHead &curr = const_cast<nsHttpRequestHead&>(*this);
|
|
curr.mReentrantMonitor.AssertCurrentThreadIn();
|
|
return mHeaders;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::SetHeaders(const nsHttpHeaderArray& aHeaders)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
mHeaders = aHeaders;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::SetVersion(nsHttpVersion version)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
mVersion = version;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::SetRequestURI(const nsCSubstring &s)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
mRequestURI = s;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::SetPath(const nsCSubstring &s)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
mPath = s;
|
|
}
|
|
|
|
uint32_t
|
|
nsHttpRequestHead::HeaderCount()
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
return mHeaders.Count();
|
|
}
|
|
|
|
nsresult
|
|
nsHttpRequestHead::VisitHeaders(nsIHttpHeaderVisitor *visitor,
|
|
nsHttpHeaderArray::VisitorFilter filter /* = nsHttpHeaderArray::eFilterAll*/)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
mInVisitHeaders = true;
|
|
nsresult rv = mHeaders.VisitHeaders(visitor, filter);
|
|
mInVisitHeaders = false;
|
|
return rv;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::Method(nsACString &aMethod)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
aMethod = mMethod;
|
|
}
|
|
|
|
nsHttpVersion
|
|
nsHttpRequestHead::Version()
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
return mVersion;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::RequestURI(nsACString &aRequestURI)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
aRequestURI = mRequestURI;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::Path(nsACString &aPath)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
aPath = mPath.IsEmpty() ? mRequestURI : mPath;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::SetHTTPS(bool val)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
mHTTPS = val;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::Origin(nsACString &aOrigin)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
aOrigin = mOrigin;
|
|
}
|
|
|
|
nsresult
|
|
nsHttpRequestHead::SetHeader(nsHttpAtom h, const nsACString &v,
|
|
bool m /*= false*/)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
|
|
if (mInVisitHeaders) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
return mHeaders.SetHeader(h, v, m,
|
|
nsHttpHeaderArray::eVarietyRequestOverride);
|
|
}
|
|
|
|
nsresult
|
|
nsHttpRequestHead::SetHeader(nsHttpAtom h, const nsACString &v, bool m,
|
|
nsHttpHeaderArray::HeaderVariety variety)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
|
|
if (mInVisitHeaders) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
return mHeaders.SetHeader(h, v, m, variety);
|
|
}
|
|
|
|
nsresult
|
|
nsHttpRequestHead::SetEmptyHeader(nsHttpAtom h)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
|
|
if (mInVisitHeaders) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
return mHeaders.SetEmptyHeader(h,
|
|
nsHttpHeaderArray::eVarietyRequestOverride);
|
|
}
|
|
|
|
nsresult
|
|
nsHttpRequestHead::GetHeader(nsHttpAtom h, nsACString &v)
|
|
{
|
|
v.Truncate();
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
return mHeaders.GetHeader(h, v);
|
|
}
|
|
|
|
nsresult
|
|
nsHttpRequestHead::ClearHeader(nsHttpAtom h)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
|
|
if (mInVisitHeaders) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
mHeaders.ClearHeader(h);
|
|
return NS_OK;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::ClearHeaders()
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
|
|
if (mInVisitHeaders) {
|
|
return;
|
|
}
|
|
|
|
mHeaders.Clear();
|
|
}
|
|
|
|
bool
|
|
nsHttpRequestHead::HasHeader(nsHttpAtom h)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
return mHeaders.HasHeader(h);
|
|
}
|
|
|
|
bool
|
|
nsHttpRequestHead::HasHeaderValue(nsHttpAtom h, const char *v)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
return mHeaders.HasHeaderValue(h, v);
|
|
}
|
|
|
|
nsresult
|
|
nsHttpRequestHead::SetHeaderOnce(nsHttpAtom h, const char *v,
|
|
bool merge /*= false */)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
|
|
if (mInVisitHeaders) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
if (!merge || !mHeaders.HasHeaderValue(h, v)) {
|
|
return mHeaders.SetHeader(h, nsDependentCString(v), merge,
|
|
nsHttpHeaderArray::eVarietyRequestOverride);
|
|
}
|
|
return NS_OK;
|
|
}
|
|
|
|
nsHttpRequestHead::ParsedMethodType
|
|
nsHttpRequestHead::ParsedMethod()
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
return mParsedMethod;
|
|
}
|
|
|
|
bool
|
|
nsHttpRequestHead::EqualsMethod(ParsedMethodType aType)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
return mParsedMethod == aType;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::ParseHeaderSet(const char *buffer)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
nsHttpAtom hdr;
|
|
nsAutoCString val;
|
|
while (buffer) {
|
|
const char *eof = strchr(buffer, '\r');
|
|
if (!eof) {
|
|
break;
|
|
}
|
|
if (NS_SUCCEEDED(nsHttpHeaderArray::ParseHeaderLine(
|
|
nsDependentCSubstring(buffer, eof - buffer),
|
|
&hdr,
|
|
&val))) {
|
|
|
|
mHeaders.SetHeaderFromNet(hdr, val, false);
|
|
}
|
|
buffer = eof + 1;
|
|
if (*buffer == '\n') {
|
|
buffer++;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool
|
|
nsHttpRequestHead::IsHTTPS()
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
return mHTTPS;
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::SetMethod(const nsACString &method)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
mParsedMethod = kMethod_Custom;
|
|
mMethod = method;
|
|
if (!strcmp(mMethod.get(), "GET")) {
|
|
mParsedMethod = kMethod_Get;
|
|
} else if (!strcmp(mMethod.get(), "POST")) {
|
|
mParsedMethod = kMethod_Post;
|
|
} else if (!strcmp(mMethod.get(), "OPTIONS")) {
|
|
mParsedMethod = kMethod_Options;
|
|
} else if (!strcmp(mMethod.get(), "CONNECT")) {
|
|
mParsedMethod = kMethod_Connect;
|
|
} else if (!strcmp(mMethod.get(), "HEAD")) {
|
|
mParsedMethod = kMethod_Head;
|
|
} else if (!strcmp(mMethod.get(), "PUT")) {
|
|
mParsedMethod = kMethod_Put;
|
|
} else if (!strcmp(mMethod.get(), "TRACE")) {
|
|
mParsedMethod = kMethod_Trace;
|
|
}
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::SetOrigin(const nsACString &scheme, const nsACString &host,
|
|
int32_t port)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
mOrigin.Assign(scheme);
|
|
mOrigin.Append(NS_LITERAL_CSTRING("://"));
|
|
mOrigin.Append(host);
|
|
if (port >= 0) {
|
|
mOrigin.Append(NS_LITERAL_CSTRING(":"));
|
|
mOrigin.AppendInt(port);
|
|
}
|
|
}
|
|
|
|
bool
|
|
nsHttpRequestHead::IsSafeMethod()
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
// This code will need to be extended for new safe methods, otherwise
|
|
// they'll default to "not safe".
|
|
if ((mParsedMethod == kMethod_Get) || (mParsedMethod == kMethod_Head) ||
|
|
(mParsedMethod == kMethod_Options) || (mParsedMethod == kMethod_Trace)
|
|
) {
|
|
return true;
|
|
}
|
|
|
|
if (mParsedMethod != kMethod_Custom) {
|
|
return false;
|
|
}
|
|
|
|
return (!strcmp(mMethod.get(), "PROPFIND") ||
|
|
!strcmp(mMethod.get(), "REPORT") ||
|
|
!strcmp(mMethod.get(), "SEARCH"));
|
|
}
|
|
|
|
void
|
|
nsHttpRequestHead::Flatten(nsACString &buf, bool pruneProxyHeaders)
|
|
{
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
// note: the first append is intentional.
|
|
|
|
buf.Append(mMethod);
|
|
buf.Append(' ');
|
|
buf.Append(mRequestURI);
|
|
buf.AppendLiteral(" HTTP/");
|
|
|
|
switch (mVersion) {
|
|
case NS_HTTP_VERSION_1_1:
|
|
buf.AppendLiteral("1.1");
|
|
break;
|
|
case NS_HTTP_VERSION_0_9:
|
|
buf.AppendLiteral("0.9");
|
|
break;
|
|
default:
|
|
buf.AppendLiteral("1.0");
|
|
}
|
|
|
|
buf.AppendLiteral("\r\n");
|
|
|
|
mHeaders.Flatten(buf, pruneProxyHeaders, false);
|
|
}
|
|
|
|
#ifdef MOZ_CRASHREPORTER
|
|
#undef ReentrantMonitorAutoEnter
|
|
#undef mon
|
|
#endif
|
|
|
|
} // namespace net
|
|
} // namespace mozilla
|