Source code

Revision control

Line Code
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
/* 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 nsDashboard_h__
#define nsDashboard_h__

#include "mozilla/Mutex.h"
#include "mozilla/net/DashboardTypes.h"
#include "nsIDashboard.h"
#include "nsIDashboardEventNotifier.h"
#include "nsIDNSListener.h"
#include "nsIServiceManager.h"
#include "nsITimer.h"
#include "nsITransport.h"

class nsIDNSService;

namespace mozilla {
namespace net {

class SocketData;
class HttpData;
class DnsData;
class WebSocketRequest;
class ConnectionData;
class RcwnData;

class Dashboard final : public nsIDashboard, public nsIDashboardEventNotifier {
 public:
  NS_DECL_THREADSAFE_ISUPPORTS
  NS_DECL_NSIDASHBOARD
  NS_DECL_NSIDASHBOARDEVENTNOTIFIER

  Dashboard();
  static const char *GetErrorString(nsresult rv);
  nsresult GetConnectionStatus(ConnectionData *aConnectionData);

 private:
  struct LogData {
    LogData(nsCString host, uint32_t serial, bool encryption)
        : mHost(host),
          mSerial(serial),
          mMsgSent(0),
          mMsgReceived(0),
          mSizeSent(0),
          mSizeReceived(0),
          mEncrypted(encryption) {}
    nsCString mHost;
    uint32_t mSerial;
    uint32_t mMsgSent;
    uint32_t mMsgReceived;
    uint64_t mSizeSent;
    uint64_t mSizeReceived;
    bool mEncrypted;
    bool operator==(const LogData &a) const {
      return mHost.Equals(a.mHost) && (mSerial == a.mSerial);
    }
  };

  struct WebSocketData {
    WebSocketData() : lock("Dashboard.webSocketData") {}
    uint32_t IndexOf(const nsCString &hostname, uint32_t mSerial) {
      LogData temp(hostname, mSerial, false);
      return data.IndexOf(temp);
    }
    nsTArray<LogData> data;
    mozilla::Mutex lock;
  };

  bool mEnableLogging;
  WebSocketData mWs;

 private:
  virtual ~Dashboard() = default;

  nsresult GetSocketsDispatch(SocketData *);
  nsresult GetHttpDispatch(HttpData *);
  nsresult GetDnsInfoDispatch(DnsData *);
  nsresult TestNewConnection(ConnectionData *);

  /* Helper methods that pass the JSON to the callback function. */
  nsresult GetSockets(SocketData *);
  nsresult GetHttpConnections(HttpData *);
  nsresult GetDNSCacheEntries(DnsData *);
  nsresult GetWebSocketConnections(WebSocketRequest *);
  nsresult GetRcwnData(RcwnData *);

  nsCOMPtr<nsIDNSService> mDnsService;
};

}  // namespace net
}  // namespace mozilla

#endif  // nsDashboard_h__