mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-30 21:55:31 +00:00
353 lines
10 KiB
C++
353 lines
10 KiB
C++
/* -*- Mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; tab-width: 40 -*- */
|
|
/* vim: set ts=2 et sw=2 tw=80: */
|
|
/* 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 "BluetoothDaemonSocketInterface.h"
|
|
#include "BluetoothSocketMessageWatcher.h"
|
|
#include "nsXULAppAPI.h"
|
|
#include "mozilla/unused.h"
|
|
|
|
BEGIN_BLUETOOTH_NAMESPACE
|
|
|
|
//
|
|
// Socket module
|
|
//
|
|
|
|
const int BluetoothDaemonSocketModule::MAX_NUM_CLIENTS = 1;
|
|
|
|
// Commands
|
|
//
|
|
|
|
nsresult
|
|
BluetoothDaemonSocketModule::ListenCmd(BluetoothSocketType aType,
|
|
const nsAString& aServiceName,
|
|
const uint8_t aServiceUuid[16],
|
|
int aChannel, bool aEncrypt,
|
|
bool aAuth,
|
|
BluetoothSocketResultHandler* aRes)
|
|
{
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
nsAutoPtr<BluetoothDaemonPDU> pdu(new BluetoothDaemonPDU(0x02, 0x01, 0));
|
|
|
|
nsresult rv = PackPDU(
|
|
aType,
|
|
PackConversion<nsAString, BluetoothServiceName>(aServiceName),
|
|
PackArray<uint8_t>(aServiceUuid, 16),
|
|
PackConversion<int, int32_t>(aChannel),
|
|
SocketFlags(aEncrypt, aAuth), *pdu);
|
|
if (NS_FAILED(rv)) {
|
|
return rv;
|
|
}
|
|
rv = Send(pdu, aRes);
|
|
if (NS_FAILED(rv)) {
|
|
return rv;
|
|
}
|
|
unused << pdu.forget();
|
|
return rv;
|
|
}
|
|
|
|
nsresult
|
|
BluetoothDaemonSocketModule::ConnectCmd(const nsAString& aBdAddr,
|
|
BluetoothSocketType aType,
|
|
const uint8_t aUuid[16],
|
|
int aChannel, bool aEncrypt,
|
|
bool aAuth,
|
|
BluetoothSocketResultHandler* aRes)
|
|
{
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
nsAutoPtr<BluetoothDaemonPDU> pdu(new BluetoothDaemonPDU(0x02, 0x02, 0));
|
|
|
|
nsresult rv = PackPDU(
|
|
PackConversion<nsAString, BluetoothAddress>(aBdAddr),
|
|
aType,
|
|
PackArray<uint8_t>(aUuid, 16),
|
|
PackConversion<int, int32_t>(aChannel),
|
|
SocketFlags(aEncrypt, aAuth), *pdu);
|
|
if (NS_FAILED(rv)) {
|
|
return rv;
|
|
}
|
|
rv = Send(pdu, aRes);
|
|
if (NS_FAILED(rv)) {
|
|
return rv;
|
|
}
|
|
unused << pdu.forget();
|
|
return rv;
|
|
}
|
|
|
|
/* |DeleteTask| deletes a class instance on the I/O thread
|
|
*/
|
|
template <typename T>
|
|
class DeleteTask MOZ_FINAL : public Task
|
|
{
|
|
public:
|
|
DeleteTask(T* aPtr)
|
|
: mPtr(aPtr)
|
|
{ }
|
|
|
|
void Run() MOZ_OVERRIDE
|
|
{
|
|
mPtr = nullptr;
|
|
}
|
|
|
|
private:
|
|
nsAutoPtr<T> mPtr;
|
|
};
|
|
|
|
/* |AcceptWatcher| specializes SocketMessageWatcher for Accept
|
|
* operations by reading the socket messages from Bluedroid and
|
|
* forwarding the received client socket to the resource handler.
|
|
* The first message is received immediately. When there's a new
|
|
* connection, Bluedroid sends the 2nd message with the socket
|
|
* info and socket file descriptor.
|
|
*/
|
|
class BluetoothDaemonSocketModule::AcceptWatcher MOZ_FINAL
|
|
: public SocketMessageWatcher
|
|
{
|
|
public:
|
|
AcceptWatcher(int aFd, BluetoothSocketResultHandler* aRes)
|
|
: SocketMessageWatcher(aFd, aRes)
|
|
{ }
|
|
|
|
void Proceed(BluetoothStatus aStatus) MOZ_OVERRIDE
|
|
{
|
|
if (aStatus == STATUS_SUCCESS) {
|
|
IntStringIntResultRunnable::Dispatch(
|
|
GetResultHandler(), &BluetoothSocketResultHandler::Accept,
|
|
ConstantInitOp3<int, nsString, int>(GetClientFd(), GetBdAddress(),
|
|
GetConnectionStatus()));
|
|
} else {
|
|
ErrorRunnable::Dispatch(GetResultHandler(),
|
|
&BluetoothSocketResultHandler::OnError,
|
|
ConstantInitOp1<BluetoothStatus>(aStatus));
|
|
}
|
|
|
|
MessageLoopForIO::current()->PostTask(
|
|
FROM_HERE, new DeleteTask<AcceptWatcher>(this));
|
|
}
|
|
};
|
|
|
|
nsresult
|
|
BluetoothDaemonSocketModule::AcceptCmd(int aFd,
|
|
BluetoothSocketResultHandler* aRes)
|
|
{
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
/* receive Bluedroid's socket-setup messages and client fd */
|
|
Task* t = new SocketMessageWatcherTask(new AcceptWatcher(aFd, aRes));
|
|
XRE_GetIOMessageLoop()->PostTask(FROM_HERE, t);
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
BluetoothDaemonSocketModule::CloseCmd(BluetoothSocketResultHandler* aRes)
|
|
{
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
/* stop the watcher corresponding to |aRes| */
|
|
Task* t = new DeleteSocketMessageWatcherTask(aRes);
|
|
XRE_GetIOMessageLoop()->PostTask(FROM_HERE, t);
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
void
|
|
BluetoothDaemonSocketModule::HandleSvc(const BluetoothDaemonPDUHeader& aHeader,
|
|
BluetoothDaemonPDU& aPDU,
|
|
void* aUserData)
|
|
{
|
|
static void (BluetoothDaemonSocketModule::* const HandleRsp[])(
|
|
const BluetoothDaemonPDUHeader&,
|
|
BluetoothDaemonPDU&,
|
|
BluetoothSocketResultHandler*) = {
|
|
INIT_ARRAY_AT(0x00, &BluetoothDaemonSocketModule::ErrorRsp),
|
|
INIT_ARRAY_AT(0x01, &BluetoothDaemonSocketModule::ListenRsp),
|
|
INIT_ARRAY_AT(0x02, &BluetoothDaemonSocketModule::ConnectRsp),
|
|
};
|
|
|
|
if (NS_WARN_IF(MOZ_ARRAY_LENGTH(HandleRsp) <= aHeader.mOpcode) ||
|
|
NS_WARN_IF(!HandleRsp[aHeader.mOpcode])) {
|
|
return;
|
|
}
|
|
|
|
nsRefPtr<BluetoothSocketResultHandler> res =
|
|
already_AddRefed<BluetoothSocketResultHandler>(
|
|
static_cast<BluetoothSocketResultHandler*>(aUserData));
|
|
|
|
if (!res) {
|
|
return; // Return early if no result handler has been set
|
|
}
|
|
|
|
(this->*(HandleRsp[aHeader.mOpcode]))(aHeader, aPDU, res);
|
|
}
|
|
|
|
nsresult
|
|
BluetoothDaemonSocketModule::Send(BluetoothDaemonPDU* aPDU,
|
|
BluetoothSocketResultHandler* aRes)
|
|
{
|
|
aRes->AddRef(); // Keep reference for response
|
|
return Send(aPDU, static_cast<void*>(aRes));
|
|
}
|
|
|
|
uint8_t
|
|
BluetoothDaemonSocketModule::SocketFlags(bool aEncrypt, bool aAuth)
|
|
{
|
|
return (0x01 * aEncrypt) | (0x02 * aAuth);
|
|
}
|
|
|
|
// Responses
|
|
//
|
|
|
|
void
|
|
BluetoothDaemonSocketModule::ErrorRsp(const BluetoothDaemonPDUHeader& aHeader,
|
|
BluetoothDaemonPDU& aPDU,
|
|
BluetoothSocketResultHandler* aRes)
|
|
{
|
|
ErrorRunnable::Dispatch(
|
|
aRes, &BluetoothSocketResultHandler::OnError, UnpackPDUInitOp(aPDU));
|
|
}
|
|
|
|
class BluetoothDaemonSocketModule::ListenInitOp MOZ_FINAL : private PDUInitOp
|
|
{
|
|
public:
|
|
ListenInitOp(BluetoothDaemonPDU& aPDU)
|
|
: PDUInitOp(aPDU)
|
|
{ }
|
|
|
|
nsresult
|
|
operator () (int& aArg1) const
|
|
{
|
|
BluetoothDaemonPDU& pdu = GetPDU();
|
|
|
|
aArg1 = pdu.AcquireFd();
|
|
|
|
if (NS_WARN_IF(aArg1 < 0)) {
|
|
return NS_ERROR_ILLEGAL_VALUE;
|
|
}
|
|
WarnAboutTrailingData();
|
|
return NS_OK;
|
|
}
|
|
};
|
|
|
|
void
|
|
BluetoothDaemonSocketModule::ListenRsp(const BluetoothDaemonPDUHeader& aHeader,
|
|
BluetoothDaemonPDU& aPDU,
|
|
BluetoothSocketResultHandler* aRes)
|
|
{
|
|
IntResultRunnable::Dispatch(
|
|
aRes, &BluetoothSocketResultHandler::Listen, ListenInitOp(aPDU));
|
|
}
|
|
|
|
/* |ConnectWatcher| specializes SocketMessageWatcher for
|
|
* connect operations by reading the socket messages from
|
|
* Bluedroid and forwarding the connected socket to the
|
|
* resource handler.
|
|
*/
|
|
class BluetoothDaemonSocketModule::ConnectWatcher MOZ_FINAL
|
|
: public SocketMessageWatcher
|
|
{
|
|
public:
|
|
ConnectWatcher(int aFd, BluetoothSocketResultHandler* aRes)
|
|
: SocketMessageWatcher(aFd, aRes)
|
|
{ }
|
|
|
|
void Proceed(BluetoothStatus aStatus) MOZ_OVERRIDE
|
|
{
|
|
if (aStatus == STATUS_SUCCESS) {
|
|
IntStringIntResultRunnable::Dispatch(
|
|
GetResultHandler(), &BluetoothSocketResultHandler::Connect,
|
|
ConstantInitOp3<int, nsString, int>(GetFd(), GetBdAddress(),
|
|
GetConnectionStatus()));
|
|
} else {
|
|
ErrorRunnable::Dispatch(GetResultHandler(),
|
|
&BluetoothSocketResultHandler::OnError,
|
|
ConstantInitOp1<BluetoothStatus>(aStatus));
|
|
}
|
|
|
|
MessageLoopForIO::current()->PostTask(
|
|
FROM_HERE, new DeleteTask<ConnectWatcher>(this));
|
|
}
|
|
};
|
|
|
|
void
|
|
BluetoothDaemonSocketModule::ConnectRsp(const BluetoothDaemonPDUHeader& aHeader,
|
|
BluetoothDaemonPDU& aPDU,
|
|
BluetoothSocketResultHandler* aRes)
|
|
{
|
|
/* the file descriptor is attached in the PDU's ancillary data */
|
|
int fd = aPDU.AcquireFd();
|
|
if (fd < 0) {
|
|
ErrorRunnable::Dispatch(aRes, &BluetoothSocketResultHandler::OnError,
|
|
ConstantInitOp1<BluetoothStatus>(STATUS_FAIL));
|
|
return;
|
|
}
|
|
|
|
/* receive Bluedroid's socket-setup messages */
|
|
Task* t = new SocketMessageWatcherTask(new ConnectWatcher(fd, aRes));
|
|
XRE_GetIOMessageLoop()->PostTask(FROM_HERE, t);
|
|
}
|
|
|
|
//
|
|
// Socket interface
|
|
//
|
|
|
|
BluetoothDaemonSocketInterface::BluetoothDaemonSocketInterface(
|
|
BluetoothDaemonSocketModule* aModule)
|
|
: mModule(aModule)
|
|
{
|
|
MOZ_ASSERT(mModule);
|
|
}
|
|
|
|
BluetoothDaemonSocketInterface::~BluetoothDaemonSocketInterface()
|
|
{ }
|
|
|
|
void
|
|
BluetoothDaemonSocketInterface::Listen(BluetoothSocketType aType,
|
|
const nsAString& aServiceName,
|
|
const uint8_t aServiceUuid[16],
|
|
int aChannel, bool aEncrypt,
|
|
bool aAuth,
|
|
BluetoothSocketResultHandler* aRes)
|
|
{
|
|
MOZ_ASSERT(mModule);
|
|
|
|
mModule->ListenCmd(aType, aServiceName, aServiceUuid, aChannel,
|
|
aEncrypt, aAuth, aRes);
|
|
}
|
|
|
|
void
|
|
BluetoothDaemonSocketInterface::Connect(const nsAString& aBdAddr,
|
|
BluetoothSocketType aType,
|
|
const uint8_t aUuid[16],
|
|
int aChannel, bool aEncrypt,
|
|
bool aAuth,
|
|
BluetoothSocketResultHandler* aRes)
|
|
{
|
|
MOZ_ASSERT(mModule);
|
|
|
|
mModule->ConnectCmd(aBdAddr, aType, aUuid, aChannel, aEncrypt, aAuth, aRes);
|
|
}
|
|
|
|
void
|
|
BluetoothDaemonSocketInterface::Accept(int aFd,
|
|
BluetoothSocketResultHandler* aRes)
|
|
{
|
|
MOZ_ASSERT(mModule);
|
|
|
|
mModule->AcceptCmd(aFd, aRes);
|
|
}
|
|
|
|
void
|
|
BluetoothDaemonSocketInterface::Close(BluetoothSocketResultHandler* aRes)
|
|
{
|
|
MOZ_ASSERT(mModule);
|
|
|
|
mModule->CloseCmd(aRes);
|
|
}
|
|
|
|
END_BLUETOOTH_NAMESPACE
|