summaryrefslogtreecommitdiff
path: root/netwerk/base/Dashboard.h
blob: 4e893c15d489b96dee1e25e699e9f2ea0d1db4e7 (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
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
/* 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 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(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();

    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 *);

    nsCOMPtr<nsIDNSService> mDnsService;
};

} // namespace net
} // namespace mozilla

#endif // nsDashboard_h__