gecko-dev/ipc/netd/Netd.h

82 строки
1.8 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 "nsISupportsImpl.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
{
protected:
virtual ~NetdConsumer() { }
public:
NS_INLINE_DECL_REFCOUNTING(NetdConsumer)
virtual void MessageReceived(NetdCommand* aMessage) = 0;
};
class NetdWriteTask : public Runnable
{
NS_IMETHOD Run();
};
class NetdClient : public MessageLoopForIO::LineWatcher
{
virtual ~NetdClient();
public:
NS_INLINE_DECL_REFCOUNTING(NetdClient)
typedef std::queue<NetdCommand*> NetdCommandQueue;
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__