gecko-dev/ipc/netd/Netd.h

79 строки
1.9 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/. */
#ifndef mozilla_system_netd_h__
#define mozilla_system_netd_h__
#include "mozilla/RefPtr.h"
#include "nsAutoPtr.h"
#include "base/message_loop.h"
#include "mozilla/FileUtils.h"
#define MAX_COMMAND_SIZE 4096
namespace mozilla {
namespace ipc {
/*
* Represents raw data going to or coming from the Netd socket.
*/
struct NetdCommand
{
uint8_t mData[MAX_COMMAND_SIZE];
// Number of octets in mData.
size_t mSize;
};
class NetdConsumer : public mozilla::RefCounted<NetdConsumer>
{
public:
MOZ_DECLARE_REFCOUNTED_TYPENAME(NetdConsumer)
virtual ~NetdConsumer() { }
virtual void MessageReceived(NetdCommand* aMessage) = 0;
};
class NetdWriteTask : public Task
{
virtual void Run();
};
class NetdClient : public MessageLoopForIO::LineWatcher,
public RefCounted<NetdClient>
{
public:
MOZ_DECLARE_REFCOUNTED_TYPENAME(NetdClient)
typedef std::queue<NetdCommand*> NetdCommandQueue;
NetdClient();
virtual ~NetdClient();
static void Start();
static void SendNetdCommandIOThread(NetdCommand* aMessage);
private:
void WriteNetdCommand();
virtual void OnError();
virtual void OnLineRead(int aFd, nsDependentCSubstring& aMessage);
virtual void OnFileCanWriteWithoutBlocking(int aFd);
bool OpenSocket();
MessageLoopForIO *mIOLoop;
MessageLoopForIO::FileDescriptorWatcher mReadWatcher;
MessageLoopForIO::FileDescriptorWatcher mWriteWatcher;
ScopedClose mSocket;
NetdCommandQueue mOutgoingQ;
nsAutoPtr<NetdCommand> mCurrentNetdCommand;
size_t mCurrentWriteOffset;
size_t mReConnectTimes;
};
void StartNetd(NetdConsumer *);
void StopNetd();
void SendNetdCommand(NetdCommand *);
} // namespace ipc
} // namespace mozilla
#endif // mozilla_system_netd_h__