1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#ifndef TOOLS_ANDROID_FORWARDER2_DAEMON_H_
6#define TOOLS_ANDROID_FORWARDER2_DAEMON_H_
7
8#include <string>
9
10#include "base/basictypes.h"
11#include "base/memory/scoped_ptr.h"
12
13namespace forwarder2 {
14
15class Socket;
16
17// Provides a way to spawn a daemon and communicate with it.
18class Daemon {
19 public:
20  // Callback used by the daemon to shutdown properly. See pipe_notifier.h for
21  // more details.
22  typedef int (*GetExitNotifierFDCallback)();
23
24  class ClientDelegate {
25   public:
26    virtual ~ClientDelegate() {}
27
28    // Called after the daemon is ready to receive commands.
29    virtual void OnDaemonReady(Socket* daemon_socket) = 0;
30  };
31
32  class ServerDelegate {
33   public:
34    virtual ~ServerDelegate() {}
35
36    // Called after the daemon bound its Unix Domain Socket. This can be used to
37    // setup signal handlers or perform global initialization.
38    virtual void Init() = 0;
39
40    virtual void OnClientConnected(scoped_ptr<Socket> client_socket) = 0;
41  };
42
43  // |identifier| should be a unique string identifier. It is used to
44  // bind/connect the underlying Unix Domain Socket.
45  // Note that this class does not take ownership of |client_delegate| and
46  // |server_delegate|.
47  Daemon(const std::string& log_file_path,
48         const std::string& identifier,
49         ClientDelegate* client_delegate,
50         ServerDelegate* server_delegate,
51         GetExitNotifierFDCallback get_exit_fd_callback);
52
53  ~Daemon();
54
55  // Returns whether the daemon was successfully spawned. Note that this does
56  // not necessarily mean that the current process was forked in case the daemon
57  // is already running.
58  bool SpawnIfNeeded();
59
60  // Kills the daemon and blocks until it exited. Returns whether it succeeded.
61  bool Kill();
62
63 private:
64  const std::string log_file_path_;
65  const std::string identifier_;
66  ClientDelegate* const client_delegate_;
67  ServerDelegate* const server_delegate_;
68  const GetExitNotifierFDCallback get_exit_fd_callback_;
69
70  DISALLOW_COPY_AND_ASSIGN(Daemon);
71};
72
73}  // namespace forwarder2
74
75#endif  // TOOLS_ANDROID_FORWARDER2_DAEMON_H_
76