ssh_client/src/ssh_plugin.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 SSH_PLUGIN_H
#define SSH_PLUGIN_H

#include <string>
#include <map>

#include "ppapi/cpp/completion_callback.h"
#include "ppapi/cpp/instance.h"
#include "ppapi/cpp/var.h"

#include "json/value.h"

#include "pthread_helpers.h"
#include "file_system.h"

class SshPluginInstance : public pp::Instance,
                          public OutputInterface {
 public:
  explicit SshPluginInstance(PP_Instance instance);
  virtual ~SshPluginInstance();

  virtual void HandleMessage(const pp::Var& message_data);

  static SshPluginInstance* GetInstance() { return instance_; }
  pp::Core* core() { return core_; }
  pthread_t openssh_thread() { return openssh_thread_; }

  // Implements OutputInterface.
  virtual bool OpenFile(int fd, const char* name, int mode,
                        InputInterface* stream);
  virtual bool OpenSocket(int fd, const char* host, uint16_t port,
                          InputInterface* stream);
  virtual bool Write(int fd, const char* data, size_t size);
  virtual bool Read(int fd, size_t size);
  virtual bool Close(int fd);
  virtual size_t GetWriteWindow();
  virtual void SendExitCode(int error);

 private:
  typedef std::map<int, InputInterface*> InputStreams;

  void StartSession(const Json::Value& args);
  void OnOpen(const Json::Value& args);
  void OnRead(const Json::Value& args);
  void OnWriteAcknowledge(const Json::Value& args);
  void OnClose(const Json::Value& args);
  void OnReadReady(const Json::Value& args);
  void OnResize(const Json::Value& args);
  void OnExitAcknowledge(const Json::Value& args);

  void SessionThreadImpl();
  static void* SessionThread(void* arg);

  void Invoke(const std::string& function, const Json::Value& args);
  void InvokeJS(const std::string& function, const Json::Value& args);

  void PrintLog(const std::string& msg);
  void PrintLogImpl(int32_t result, const std::string& msg);

  void SendExitCodeImpl(int32_t result, int error);

  static SshPluginInstance* instance_;

  pp::Core* core_;
  pthread_t openssh_thread_;
  Json::Value session_args_;
  pp::CompletionCallbackFactory<SshPluginInstance> factory_;
  InputStreams streams_;
  FileSystem file_system_;

  DISALLOW_COPY_AND_ASSIGN(SshPluginInstance);
};

#endif  // SSH_PLUGIN_H