summaryrefslogtreecommitdiffstats
path: root/remoting/host/simple_host_process.cc
blob: 79db1069a68fd1bf336df2697968ab2d8f8183bc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
// Copyright (c) 2010 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
//
// This is an application of a minimal host process in a Chromoting
// system. It serves the purpose of gluing different pieces together
// to make a functional host process for testing.
//
// It peforms the following functionality:
// 1. Connect to the GTalk network and register the machine as a host.
// 2. Accepts connection through libjingle.
// 3. Receive mouse / keyboard events through libjingle.
// 4. Sends screen capture through libjingle.

#include <iostream>
#include <string>

#include "build/build_config.h"

#if defined(OS_POSIX)
#include <termios.h>
#endif  // defined (OS_POSIX)

#include "base/at_exit.h"
#include "base/waitable_event.h"
#include "remoting/host/capturer_fake.h"
#include "remoting/host/encoder_verbatim.h"
#include "remoting/host/simple_host.h"

#if defined(OS_WIN)
#include "remoting/host/capturer_gdi.h"
#include "remoting/host/event_executor_win.h"
#elif defined(OS_LINUX)
#include "remoting/host/capturer_linux.h"
#include "remoting/host/event_executor_linux.h"
#elif defined(OS_MACOSX)
#include "remoting/host/capturer_mac.h"
#include "remoting/host/event_executor_mac.h"
#endif

void SetConsoleEcho(bool on) {
#if defined(OS_WIN)
  HANDLE hIn = GetStdHandle(STD_INPUT_HANDLE);
  if ((hIn == INVALID_HANDLE_VALUE) || (hIn == NULL))
    return;

  DWORD mode;
  if (!GetConsoleMode(hIn, &mode))
    return;

  if (on) {
    mode = mode | ENABLE_ECHO_INPUT;
  } else {
    mode = mode & ~ENABLE_ECHO_INPUT;
  }

  SetConsoleMode(hIn, mode);
#elif defined(OS_POSIX)
  struct termios settings;
  tcgetattr(STDIN_FILENO, &settings);
  if (on) {
    settings.c_lflag |= ECHO;
  } else {
    settings.c_lflag &= ~ECHO;
  }
  tcsetattr(STDIN_FILENO, TCSANOW, &settings);
#endif  //  defined(OS_WIN)
}

int main(int argc, char** argv) {
  base::AtExitManager exit_manager;

  std::string username;
  std::string auth_token;

  // Check the argument to see if we should use a fake capturer and encoder.
  bool fake = false;
  // True if the JID was specified on the command line.
  bool has_jid = false;
  // True if the auth token was specified on the command line.
  bool has_auth = false;

  for (int i = 1; i < argc; i++) {
    std::string arg = argv[i];
    if (arg == "--fake") {
      fake = true;
    } else if (arg == "--jid") {
      if (++i >= argc) {
        std::cerr << "Expected JID to follow --jid option" << std::endl;
        return 1;
      }
      has_jid = true;
      username = argv[i];
    } else if (arg == "--auth") {
      if (++i >= argc) {
        std::cerr << "Expected auth token to follow --auth option" << std::endl;
        return 1;
      }
      has_auth = true;
      auth_token = argv[i];
    }
  }

  // Prompt user for username and auth token.
  if (!has_jid) {
    std::cout << "JID: ";
    std::cin >> username;
  }
  if (!has_auth) {
    std::cout << "Auth Token: ";
    std::cin >> auth_token;
  }

  scoped_ptr<remoting::Capturer> capturer;
  scoped_ptr<remoting::Encoder> encoder;
  scoped_ptr<remoting::EventExecutor> executor;
#if defined(OS_WIN)
  capturer.reset(new remoting::CapturerGdi());
  executor.reset(new remoting::EventExecutorWin());
#elif defined(OS_LINUX)
  capturer.reset(new remoting::CapturerLinux());
  executor.reset(new remoting::EventExecutorLinux());
#elif defined(OS_MACOSX)
  capturer.reset(new remoting::CapturerMac());
  executor.reset(new remoting::EventExecutorMac());
#endif
  encoder.reset(new remoting::EncoderVerbatim());

  if (fake) {
    // Inject a fake capturer.
    capturer.reset(new remoting::CapturerFake());
  }

  // Construct a simple host with username and auth_token.
  // TODO(hclam): Allow the host to load saved credentials.
  base::WaitableEvent host_done(false, false);
  scoped_refptr<remoting::SimpleHost> host
      = new remoting::SimpleHost(username, auth_token,
                                 capturer.release(),
                                 encoder.release(),
                                 executor.release(),
                                 &host_done);
  host->Run();
  host_done.Wait();
  return 0;
}