Emilio Cobos Álvarez
abbeff4fd9
Autogenerated with:
cp dom/.clang-format netwerk/ && mach format netwerk/**.{cpp,h,mm}
Manual fixes:
* mozurl/cbindgen.toml (including headers earlier)
* NetlinkService.h (forward-declaring a necessary struct)
* nsIWebSocketChannel.idl (forward-declaring OriginAttributes).
* nsAsyncRedirectVerifyHelper.h (forward-declaring nsIEventTarget)
* CapsuleEncoder.h gets a missing Capsule forward-decl.
* WebSocketConnectionBase.h gets some missing includes.
* HappyEyeballsConnectionAttempt gets some missing includes.
* nsUDPSocket missed nsSocketTransportService2.h for OnSocketThread().
* HttpInfo missed a forward declaration.
* MockHttpAuth missed an nsCOMPtr include.
* nsHttpAuthManager missed a forward declaration for nsIHttpAuthCache.
* WebTransportSessionProxy was using mozilla::Mutex in a member without
including it.
* win32 fixes to keep include order because windows headers suck.
* nsInputStreamPump needs an Atomics.h include.
Differential Revision: https://phabricator.services.mozilla.com/D311696
138 lines
4.3 KiB
C++
138 lines
4.3 KiB
C++
/* 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/. */
|
|
|
|
#include "NetworkDataCountLayer.h"
|
|
|
|
#include "nsSocketTransportService2.h"
|
|
#include "prio.h"
|
|
#include "prmem.h"
|
|
|
|
namespace mozilla {
|
|
namespace net {
|
|
|
|
static PRDescIdentity sNetworkDataCountLayerIdentity;
|
|
static PRIOMethods sNetworkDataCountLayerMethods;
|
|
static PRIOMethods* sNetworkDataCountLayerMethodsPtr = nullptr;
|
|
|
|
class NetworkDataCountSecret {
|
|
public:
|
|
NetworkDataCountSecret() = default;
|
|
|
|
uint64_t mSentBytes = 0;
|
|
uint64_t mReceivedBytes = 0;
|
|
};
|
|
|
|
static PRInt32 NetworkDataCountSend(PRFileDesc* fd, const void* buf,
|
|
PRInt32 amount, PRIntn flags,
|
|
PRIntervalTime timeout) {
|
|
MOZ_RELEASE_ASSERT(fd->identity == sNetworkDataCountLayerIdentity);
|
|
|
|
NetworkDataCountSecret* secret =
|
|
reinterpret_cast<NetworkDataCountSecret*>(fd->secret);
|
|
|
|
PRInt32 rv =
|
|
(fd->lower->methods->send)(fd->lower, buf, amount, flags, timeout);
|
|
if (rv > 0) {
|
|
secret->mSentBytes += rv;
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
static PRInt32 NetworkDataCountWrite(PRFileDesc* fd, const void* buf,
|
|
PRInt32 amount) {
|
|
return NetworkDataCountSend(fd, buf, amount, 0, PR_INTERVAL_NO_WAIT);
|
|
}
|
|
|
|
static PRInt32 NetworkDataCountRecv(PRFileDesc* fd, void* buf, PRInt32 amount,
|
|
PRIntn flags, PRIntervalTime timeout) {
|
|
MOZ_RELEASE_ASSERT(fd->identity == sNetworkDataCountLayerIdentity);
|
|
|
|
NetworkDataCountSecret* secret =
|
|
reinterpret_cast<NetworkDataCountSecret*>(fd->secret);
|
|
|
|
PRInt32 rv =
|
|
(fd->lower->methods->recv)(fd->lower, buf, amount, flags, timeout);
|
|
if (rv > 0) {
|
|
secret->mReceivedBytes += rv;
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
static PRInt32 NetworkDataCountRead(PRFileDesc* fd, void* buf, PRInt32 amount) {
|
|
return NetworkDataCountRecv(fd, buf, amount, 0, PR_INTERVAL_NO_WAIT);
|
|
}
|
|
|
|
static PRStatus NetworkDataCountClose(PRFileDesc* fd) {
|
|
if (!fd) {
|
|
return PR_FAILURE;
|
|
}
|
|
|
|
PRFileDesc* layer = PR_PopIOLayer(fd, PR_TOP_IO_LAYER);
|
|
|
|
MOZ_RELEASE_ASSERT(layer && layer->identity == sNetworkDataCountLayerIdentity,
|
|
"NetworkDataCount Layer not on top of stack");
|
|
|
|
NetworkDataCountSecret* secret =
|
|
reinterpret_cast<NetworkDataCountSecret*>(layer->secret);
|
|
layer->secret = nullptr;
|
|
layer->dtor(layer);
|
|
delete secret;
|
|
return fd->methods->close(fd);
|
|
}
|
|
|
|
nsresult AttachNetworkDataCountLayer(PRFileDesc* fd) {
|
|
if (!sNetworkDataCountLayerMethodsPtr) {
|
|
sNetworkDataCountLayerIdentity =
|
|
PR_GetUniqueIdentity("NetworkDataCount Layer");
|
|
sNetworkDataCountLayerMethods = *PR_GetDefaultIOMethods();
|
|
sNetworkDataCountLayerMethods.send = NetworkDataCountSend;
|
|
sNetworkDataCountLayerMethods.write = NetworkDataCountWrite;
|
|
sNetworkDataCountLayerMethods.recv = NetworkDataCountRecv;
|
|
sNetworkDataCountLayerMethods.read = NetworkDataCountRead;
|
|
sNetworkDataCountLayerMethods.close = NetworkDataCountClose;
|
|
sNetworkDataCountLayerMethodsPtr = &sNetworkDataCountLayerMethods;
|
|
}
|
|
|
|
PRFileDesc* layer = PR_CreateIOLayerStub(sNetworkDataCountLayerIdentity,
|
|
sNetworkDataCountLayerMethodsPtr);
|
|
|
|
if (!layer) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
NetworkDataCountSecret* secret = new NetworkDataCountSecret();
|
|
|
|
layer->secret = reinterpret_cast<PRFilePrivate*>(secret);
|
|
|
|
PRStatus status = PR_PushIOLayer(fd, PR_NSPR_IO_LAYER, layer);
|
|
|
|
if (status == PR_FAILURE) {
|
|
delete secret;
|
|
PR_Free(layer); // PR_CreateIOLayerStub() uses PR_Malloc().
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
return NS_OK;
|
|
}
|
|
|
|
void NetworkDataCountSent(PRFileDesc* fd, uint64_t& sentBytes) {
|
|
PRFileDesc* ndcFd = PR_GetIdentitiesLayer(fd, sNetworkDataCountLayerIdentity);
|
|
MOZ_RELEASE_ASSERT(ndcFd);
|
|
|
|
NetworkDataCountSecret* secret =
|
|
reinterpret_cast<NetworkDataCountSecret*>(ndcFd->secret);
|
|
sentBytes = secret->mSentBytes;
|
|
}
|
|
|
|
void NetworkDataCountReceived(PRFileDesc* fd, uint64_t& receivedBytes) {
|
|
PRFileDesc* ndcFd = PR_GetIdentitiesLayer(fd, sNetworkDataCountLayerIdentity);
|
|
MOZ_RELEASE_ASSERT(ndcFd);
|
|
|
|
NetworkDataCountSecret* secret =
|
|
reinterpret_cast<NetworkDataCountSecret*>(ndcFd->secret);
|
|
receivedBytes = secret->mReceivedBytes;
|
|
}
|
|
|
|
} // namespace net
|
|
} // namespace mozilla
|