82 lines
1.8 KiB
C++
82 lines
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 Task
|
|
{
|
|
virtual void 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__
|