mtpd_server_impl.h

// Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef MTPD_SERVER_IMPL_H_
#define MTPD_SERVER_IMPL_H_

#include <string>
#include <vector>

#include <base/basictypes.h>
#include <base/compiler_specific.h>

#include "mtpd/device_event_delegate.h"
#include "mtpd/device_manager.h"
#include "mtpd/file_entry.h"
#include "mtpd_server/mtpd_server.h"

namespace mtpd {

class DeviceManager;

// The D-bus server for the mtpd daemon.
//
// Example Usage:
//
// DBus::Connection server_conn = DBus::Connection::SystemBus();
// server_conn.request_name("org.chromium.Mtpd");
// MtpdServer* server = new(std::nothrow) MtpdServer(server_conn, &manager);
//
// At this point the server should be attached to the main loop.

class MtpdServer : public org::chromium::Mtpd_adaptor,
                   public DBus::ObjectAdaptor,
                   public DeviceEventDelegate {
 public:
  explicit MtpdServer(DBus::Connection& connection);
  virtual ~MtpdServer();

  // org::chromium::Mtpd_adaptor implementation.
  virtual std::vector<std::string> EnumerateStorage(
      DBus::Error &error) OVERRIDE;
  virtual DBusMTPStorage GetStorageInfo(const std::string& storageName,
                                        DBus::Error &error) OVERRIDE;
  virtual std::string OpenStorage(const std::string& storageName,
                                  const std::string& mode,
                                  DBus::Error &error) OVERRIDE;
  virtual void CloseStorage(const std::string& handle,
                            DBus::Error &error) OVERRIDE;
  virtual DBusFileEntries ReadDirectoryByPath(const std::string& handle,
                                              const std::string& filePath,
                                              DBus::Error &error) OVERRIDE;
  virtual DBusFileEntries ReadDirectoryById(const std::string& handle,
                                            const uint32_t& fileId,
                                            DBus::Error &error) OVERRIDE;
  virtual std::vector<uint8_t> ReadFileByPath(
      const std::string& handle,
      const std::string& filePath,
      DBus::Error &error) OVERRIDE;
  virtual std::vector<uint8_t> ReadFileById(
      const std::string& handle,
      const uint32_t& fileId,
      DBus::Error &error) OVERRIDE;
  virtual bool IsAlive(DBus::Error &error) OVERRIDE;

  // DeviceEventDelegate implementation.
  virtual void StorageAttached(const std::string& storage_name) OVERRIDE;
  virtual void StorageDetached(const std::string& storage_name) OVERRIDE;

  // Returns a file descriptor for monitoring device events.
  int GetDeviceEventDescriptor() const;

  // Processes the available device events.
  void ProcessDeviceEvents();

 private:
  DeviceManager device_manager_;

  DISALLOW_COPY_AND_ASSIGN(MtpdServer);
};

}  // namespace mtpd

#endif  // MTPD_SERVER_IMPL_H_