summaryrefslogtreecommitdiffstats
path: root/toolkit/components/viaduct/ViaductRequest.h
blob: 8e5963c236c3e4040721e00b6fa9e015bb6f23be (plain)
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
/* 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 https://mozilla.org/MPL/2.0/. */

#ifndef mozilla_ViaductRequest_h
#define mozilla_ViaductRequest_h

#include "mozilla/Monitor.h"
#include "mozilla/fetch_msg_types.pb.h"
#include "mozilla/Viaduct.h"

#include "nsCOMPtr.h"
#include "nsIChannel.h"
#include "nsIChannelEventSink.h"
#include "nsIStreamListener.h"
#include "nsITimer.h"

namespace mozilla {

// A mapping of the ByteBuffer repr(C) Rust struct.
struct ViaductByteBuffer {
  int64_t len;
  uint8_t* data;
};

class ViaductRequest final : public nsIStreamListener,
                             public nsITimerCallback,
                             public nsINamed,
                             public nsIChannelEventSink {
 public:
  NS_DECL_THREADSAFE_ISUPPORTS
  NS_DECL_NSIREQUESTOBSERVER
  NS_DECL_NSISTREAMLISTENER
  NS_DECL_NSITIMERCALLBACK
  NS_DECL_NSINAMED
  NS_DECL_NSICHANNELEVENTSINK

  ViaductRequest()
      : mDone(false), mChannel(nullptr), mMonitor("ViaductRequest") {}
  ViaductByteBuffer MakeRequest(ViaductByteBuffer reqBuf);

 private:
  nsresult LaunchRequest(appservices::httpconfig::protobuf::Request&);
  void ClearTimers();
  void NotifyMonitor();
  bool mDone;
  nsCOMPtr<nsIChannel> mChannel;
  nsCString mBodyBuffer;
  nsCOMPtr<nsITimer> mConnectTimeoutTimer;
  nsCOMPtr<nsITimer> mReadTimeoutTimer;
  appservices::httpconfig::protobuf::Response mResponse;
  Monitor mMonitor MOZ_UNANNOTATED;
  ~ViaductRequest();
};

};  // namespace mozilla

#endif  // mozilla_ViaductRequest_h