1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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/. */
#ifndef nsHttpRequestHead_h__
#define nsHttpRequestHead_h__
#include "nsHttp.h"
#include "nsHttpHeaderArray.h"
#include "nsString.h"
#include "mozilla/RecursiveMutex.h"
class nsIHttpHeaderVisitor;
// This needs to be forward declared here so we can include only this header
// without also including PHttpChannelParams.h
namespace IPC {
template <typename>
struct ParamTraits;
} // namespace IPC
namespace mozilla {
namespace net {
//-----------------------------------------------------------------------------
// nsHttpRequestHead represents the request line and headers from an HTTP
// request.
//-----------------------------------------------------------------------------
class nsHttpRequestHead {
public:
nsHttpRequestHead();
explicit nsHttpRequestHead(const nsHttpRequestHead& aRequestHead);
~nsHttpRequestHead();
nsHttpRequestHead& operator=(const nsHttpRequestHead& aRequestHead);
// The following function is only used in HttpChannelParent to avoid
// copying headers. If you use it be careful to do it only under
// nsHttpRequestHead lock!!!
const nsHttpHeaderArray& Headers() const;
void Enter() { mRecursiveMutex.Lock(); }
void Exit() { mRecursiveMutex.Unlock(); }
void SetHeaders(const nsHttpHeaderArray& aHeaders);
void SetMethod(const nsACString& method);
void SetVersion(HttpVersion version);
void SetRequestURI(const nsACString& s);
void SetPath(const nsACString& s);
uint32_t HeaderCount();
// Using this function it is possible to itereate through all headers
// automatically under one lock.
[[nodiscard]] nsresult VisitHeaders(
nsIHttpHeaderVisitor* visitor,
nsHttpHeaderArray::VisitorFilter filter = nsHttpHeaderArray::eFilterAll);
void Method(nsACString& aMethod);
HttpVersion Version();
void RequestURI(nsACString& RequestURI);
void Path(nsACString& aPath);
void SetHTTPS(bool val);
bool IsHTTPS();
void SetOrigin(const nsACString& scheme, const nsACString& host,
int32_t port);
void Origin(nsACString& aOrigin);
[[nodiscard]] nsresult SetHeader(const nsACString& h, const nsACString& v,
bool m = false);
[[nodiscard]] nsresult SetHeader(nsHttpAtom h, const nsACString& v,
bool m = false);
[[nodiscard]] nsresult SetHeader(nsHttpAtom h, const nsACString& v, bool m,
nsHttpHeaderArray::HeaderVariety variety);
[[nodiscard]] nsresult SetEmptyHeader(const nsACString& h);
[[nodiscard]] nsresult GetHeader(nsHttpAtom h, nsACString& v);
[[nodiscard]] nsresult ClearHeader(nsHttpAtom h);
void ClearHeaders();
bool HasHeaderValue(nsHttpAtom h, const char* v);
// This function returns true if header is set even if it is an empty
// header.
bool HasHeader(nsHttpAtom h);
void Flatten(nsACString&, bool pruneProxyHeaders = false);
// Don't allow duplicate values
[[nodiscard]] nsresult SetHeaderOnce(nsHttpAtom h, const char* v,
bool merge = false);
bool IsSafeMethod();
enum ParsedMethodType {
kMethod_Custom,
kMethod_Get,
kMethod_Post,
kMethod_Options,
kMethod_Connect,
kMethod_Head,
kMethod_Put,
kMethod_Trace
};
static void ParseMethod(const nsCString& aRawMethod,
ParsedMethodType& aParsedMethod);
ParsedMethodType ParsedMethod();
bool EqualsMethod(ParsedMethodType aType);
bool IsGet() { return EqualsMethod(kMethod_Get); }
bool IsPost() { return EqualsMethod(kMethod_Post); }
bool IsOptions() { return EqualsMethod(kMethod_Options); }
bool IsConnect() { return EqualsMethod(kMethod_Connect); }
bool IsHead() { return EqualsMethod(kMethod_Head); }
bool IsPut() { return EqualsMethod(kMethod_Put); }
bool IsTrace() { return EqualsMethod(kMethod_Trace); }
void ParseHeaderSet(const char* buffer);
private:
// All members must be copy-constructable and assignable
nsHttpHeaderArray mHeaders;
nsCString mMethod;
HttpVersion mVersion;
// mRequestURI and mPath are strings instead of an nsIURI
// because this is used off the main thread
nsCString mRequestURI;
nsCString mPath;
nsCString mOrigin;
ParsedMethodType mParsedMethod;
bool mHTTPS;
// We are using RecursiveMutex instead of a Mutex because VisitHeader
// function calls nsIHttpHeaderVisitor::VisitHeader while under lock.
RecursiveMutex mRecursiveMutex;
// During VisitHeader we sould not allow cal to SetHeader.
bool mInVisitHeaders;
friend struct IPC::ParamTraits<nsHttpRequestHead>;
};
} // namespace net
} // namespace mozilla
#endif // nsHttpRequestHead_h__
|
