1 /*
2 * libjingle
3 * Copyright 2004--2005, Google Inc.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are met:
7 *
8 * 1. Redistributions of source code must retain the above copyright notice,
9 * this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright notice,
11 * this list of conditions and the following disclaimer in the documentation
12 * and/or other materials provided with the distribution.
13 * 3. The name of the author may not be used to endorse or promote products
14 * derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28 #include <time.h>
29 #include <iomanip>
30 #include <cstdio>
31 #include <cstring>
32 #include <vector>
33 #include "talk/base/logging.h"
34 #include "talk/base/flags.h"
35 #include "talk/base/pathutils.h"
36 #include "talk/base/stream.h"
37 #include "talk/base/ssladapter.h"
38 #include "talk/base/win32socketserver.h"
39 #include "talk/p2p/base/constants.h"
40 #include "talk/xmpp/xmppclientsettings.h"
41 #include "talk/examples/login/xmppthread.h"
42 #include "talk/examples/login/xmppauth.h"
43 #include "talk/examples/login/xmpppump.h"
44 #include "talk/examples/call/callclient.h"
45 #include "talk/examples/call/console.h"
46 #include "talk/session/phone/filemediaengine.h"
47 #include "talk/session/phone/mediasessionclient.h"
48
49 class DebugLog : public sigslot::has_slots<> {
50 public:
DebugLog()51 DebugLog() :
52 debug_input_buf_(NULL), debug_input_len_(0), debug_input_alloc_(0),
53 debug_output_buf_(NULL), debug_output_len_(0), debug_output_alloc_(0),
54 censor_password_(false)
55 {}
56 char * debug_input_buf_;
57 int debug_input_len_;
58 int debug_input_alloc_;
59 char * debug_output_buf_;
60 int debug_output_len_;
61 int debug_output_alloc_;
62 bool censor_password_;
63
Input(const char * data,int len)64 void Input(const char * data, int len) {
65 if (debug_input_len_ + len > debug_input_alloc_) {
66 char * old_buf = debug_input_buf_;
67 debug_input_alloc_ = 4096;
68 while (debug_input_alloc_ < debug_input_len_ + len) {
69 debug_input_alloc_ *= 2;
70 }
71 debug_input_buf_ = new char[debug_input_alloc_];
72 memcpy(debug_input_buf_, old_buf, debug_input_len_);
73 delete[] old_buf;
74 }
75 memcpy(debug_input_buf_ + debug_input_len_, data, len);
76 debug_input_len_ += len;
77 DebugPrint(debug_input_buf_, &debug_input_len_, false);
78 }
79
Output(const char * data,int len)80 void Output(const char * data, int len) {
81 if (debug_output_len_ + len > debug_output_alloc_) {
82 char * old_buf = debug_output_buf_;
83 debug_output_alloc_ = 4096;
84 while (debug_output_alloc_ < debug_output_len_ + len) {
85 debug_output_alloc_ *= 2;
86 }
87 debug_output_buf_ = new char[debug_output_alloc_];
88 memcpy(debug_output_buf_, old_buf, debug_output_len_);
89 delete[] old_buf;
90 }
91 memcpy(debug_output_buf_ + debug_output_len_, data, len);
92 debug_output_len_ += len;
93 DebugPrint(debug_output_buf_, &debug_output_len_, true);
94 }
95
IsAuthTag(const char * str,size_t len)96 static bool IsAuthTag(const char * str, size_t len) {
97 if (str[0] == '<' && str[1] == 'a' &&
98 str[2] == 'u' &&
99 str[3] == 't' &&
100 str[4] == 'h' &&
101 str[5] <= ' ') {
102 std::string tag(str, len);
103
104 if (tag.find("mechanism") != std::string::npos)
105 return true;
106 }
107 return false;
108 }
109
DebugPrint(char * buf,int * plen,bool output)110 void DebugPrint(char * buf, int * plen, bool output) {
111 int len = *plen;
112 if (len > 0) {
113 time_t tim = time(NULL);
114 struct tm * now = localtime(&tim);
115 char *time_string = asctime(now);
116 if (time_string) {
117 size_t time_len = strlen(time_string);
118 if (time_len > 0) {
119 time_string[time_len-1] = 0; // trim off terminating \n
120 }
121 }
122 LOG(INFO) << (output ? "SEND >>>>>>>>>>>>>>>>" : "RECV <<<<<<<<<<<<<<<<")
123 << " : " << time_string;
124
125 bool indent;
126 int start = 0, nest = 3;
127 for (int i = 0; i < len; i += 1) {
128 if (buf[i] == '>') {
129 if ((i > 0) && (buf[i-1] == '/')) {
130 indent = false;
131 } else if ((start + 1 < len) && (buf[start + 1] == '/')) {
132 indent = false;
133 nest -= 2;
134 } else {
135 indent = true;
136 }
137
138 // Output a tag
139 LOG(INFO) << std::setw(nest) << " "
140 << std::string(buf + start, i + 1 - start);
141
142 if (indent)
143 nest += 2;
144
145 // Note if it's a PLAIN auth tag
146 if (IsAuthTag(buf + start, i + 1 - start)) {
147 censor_password_ = true;
148 }
149
150 // incr
151 start = i + 1;
152 }
153
154 if (buf[i] == '<' && start < i) {
155 if (censor_password_) {
156 LOG(INFO) << std::setw(nest) << " " << "## TEXT REMOVED ##";
157 censor_password_ = false;
158 } else {
159 LOG(INFO) << std::setw(nest) << " "
160 << std::string(buf + start, i - start);
161 }
162 start = i;
163 }
164 }
165 len = len - start;
166 memcpy(buf, buf + start, len);
167 *plen = len;
168 }
169 }
170 };
171
172 static DebugLog debug_log_;
173 static const int DEFAULT_PORT = 5222;
174
175
CreateFileMediaEngine(const char * voice_in,const char * voice_out,const char * video_in,const char * video_out)176 cricket::MediaEngine* CreateFileMediaEngine(const char* voice_in,
177 const char* voice_out,
178 const char* video_in,
179 const char* video_out) {
180 cricket::FileMediaEngine* file_media_engine = new cricket::FileMediaEngine;
181 // Set the RTP dump file names.
182 if (voice_in) {
183 file_media_engine->set_voice_input_filename(voice_in);
184 }
185 if (voice_out) {
186 file_media_engine->set_voice_output_filename(voice_out);
187 }
188 if (video_in) {
189 file_media_engine->set_video_input_filename(video_in);
190 }
191 if (video_out) {
192 file_media_engine->set_video_output_filename(video_out);
193 }
194
195 // Set voice and video codecs. TODO: The codecs actually depend on
196 // the the input voice and video streams.
197 std::vector<cricket::AudioCodec> voice_codecs;
198 voice_codecs.push_back(
199 cricket::AudioCodec(9, "G722", 16000, 0, 1, 0));
200 file_media_engine->set_voice_codecs(voice_codecs);
201 std::vector<cricket::VideoCodec> video_codecs;
202 video_codecs.push_back(
203 cricket::VideoCodec(97, "H264", 320, 240, 30, 0));
204 file_media_engine->set_video_codecs(video_codecs);
205
206 return file_media_engine;
207 }
208
main(int argc,char ** argv)209 int main(int argc, char **argv) {
210 // This app has three threads. The main thread will run the XMPP client,
211 // which will print to the screen in its own thread. A second thread
212 // will get input from the console, parse it, and pass the appropriate
213 // message back to the XMPP client's thread. A third thread is used
214 // by MediaSessionClient as its worker thread.
215
216 // define options
217 DEFINE_bool(a, false, "Turn on auto accept.");
218 DEFINE_bool(d, false, "Turn on debugging.");
219 DEFINE_string(
220 protocol, "hybrid",
221 "Initial signaling protocol to use: jingle, gingle, or hybrid.");
222 DEFINE_string(
223 secure, "disable",
224 "Disable or enable encryption: disable, enable, require.");
225 DEFINE_bool(testserver, false, "Use test server");
226 DEFINE_bool(plainserver, false, "Turn off tls and allow plain password.");
227 DEFINE_int(portallocator, 0, "Filter out unwanted connection types.");
228 DEFINE_string(filterhost, NULL, "Filter out the host from all candidates.");
229 DEFINE_string(pmuc, "groupchat.google.com", "The persistant muc domain.");
230 DEFINE_string(s, "talk.google.com", "The connection server to use.");
231 DEFINE_string(voiceinput, NULL, "RTP dump file for voice input.");
232 DEFINE_string(voiceoutput, NULL, "RTP dump file for voice output.");
233 DEFINE_string(videoinput, NULL, "RTP dump file for video input.");
234 DEFINE_string(videooutput, NULL, "RTP dump file for video output.");
235 DEFINE_bool(help, false, "Prints this message");
236
237 // parse options
238 FlagList::SetFlagsFromCommandLine(&argc, argv, true);
239 if (FLAG_help) {
240 FlagList::Print(NULL, false);
241 return 0;
242 }
243
244 bool auto_accept = FLAG_a;
245 bool debug = FLAG_d;
246 std::string protocol = FLAG_protocol;
247 bool test_server = FLAG_testserver;
248 bool plain_server = FLAG_plainserver;
249 int32 portallocator_flags = FLAG_portallocator;
250 std::string pmuc_domain = FLAG_pmuc;
251 std::string server = FLAG_s;
252 std::string secure = FLAG_secure;
253
254 cricket::SignalingProtocol initial_protocol = cricket::PROTOCOL_HYBRID;
255 if (protocol == "jingle") {
256 initial_protocol = cricket::PROTOCOL_JINGLE;
257 } else if (protocol == "gingle") {
258 initial_protocol = cricket::PROTOCOL_GINGLE;
259 } else if (protocol == "hybrid") {
260 initial_protocol = cricket::PROTOCOL_HYBRID;
261 } else {
262 printf("Invalid protocol. Must be jingle, gingle, or hybrid.\n");
263 return 1;
264 }
265
266 cricket::SecureMediaPolicy secure_policy = cricket::SEC_DISABLED;
267 if (secure == "disable") {
268 secure_policy = cricket::SEC_DISABLED;
269 } else if (secure == "enable") {
270 secure_policy = cricket::SEC_ENABLED;
271 } else if (secure == "require") {
272 secure_policy = cricket::SEC_REQUIRED;
273 } else {
274 printf("Invalid encryption. Must be enable, disable, or require.\n");
275 return 1;
276 }
277
278 // parse username and password, if present
279 buzz::Jid jid;
280 std::string username;
281 talk_base::InsecureCryptStringImpl pass;
282 if (argc > 1) {
283 username = argv[1];
284 if (argc > 2) {
285 pass.password() = argv[2];
286 }
287 }
288
289 if (debug)
290 talk_base::LogMessage::LogToDebug(talk_base::LS_VERBOSE);
291
292 if (username.empty()) {
293 std::cout << "JID: ";
294 std::cin >> username;
295 }
296 if (username.find('@') == std::string::npos) {
297 username.append("@localhost");
298 }
299 jid = buzz::Jid(username);
300 if (!jid.IsValid() || jid.node() == "") {
301 printf("Invalid JID. JIDs should be in the form user@domain\n");
302 return 1;
303 }
304 if (pass.password().empty() && !test_server) {
305 Console::SetEcho(false);
306 std::cout << "Password: ";
307 std::cin >> pass.password();
308 Console::SetEcho(true);
309 std::cout << std::endl;
310 }
311
312 buzz::XmppClientSettings xcs;
313 xcs.set_user(jid.node());
314 xcs.set_resource("call");
315 xcs.set_host(jid.domain());
316 xcs.set_use_tls(!test_server);
317
318 if (plain_server) {
319 xcs.set_use_tls(false);
320 xcs.set_allow_plain(true);
321 }
322 if (test_server) {
323 pass.password() = jid.node();
324 xcs.set_allow_plain(true);
325 }
326 xcs.set_pass(talk_base::CryptString(pass));
327
328 std::string host;
329 int port;
330
331 int colon = server.find(':');
332 if (colon == -1) {
333 host = server;
334 port = DEFAULT_PORT;
335 } else {
336 host = server.substr(0, colon);
337 port = atoi(server.substr(colon + 1).c_str());
338 }
339
340 xcs.set_server(talk_base::SocketAddress(host, port));
341 printf("Logging in to %s as %s\n", server.c_str(), jid.Str().c_str());
342
343 talk_base::InitializeSSL();
344
345
346 #if WIN32
347 // Need to pump messages on our main thread on Windows.
348 talk_base::Win32Thread w32_thread;
349 talk_base::ThreadManager::SetCurrent(&w32_thread);
350 #endif
351 talk_base::Thread* main_thread = talk_base::Thread::Current();
352
353 XmppPump pump;
354 CallClient *client = new CallClient(pump.client());
355
356 if (FLAG_voiceinput || FLAG_voiceoutput ||
357 FLAG_videoinput || FLAG_videooutput) {
358 // If any dump file is specified, we use FileMediaEngine.
359 cricket::MediaEngine* engine = CreateFileMediaEngine(FLAG_voiceinput,
360 FLAG_voiceoutput,
361 FLAG_videoinput,
362 FLAG_videooutput);
363 // The engine will be released by the client later.
364 client->SetMediaEngine(engine);
365 }
366
367 Console *console = new Console(main_thread, client);
368 client->SetConsole(console);
369 client->SetAutoAccept(auto_accept);
370 client->SetPmucDomain(pmuc_domain);
371 client->SetPortAllocatorFlags(portallocator_flags);
372 client->SetAllowLocalIps(true);
373 client->SetInitialProtocol(initial_protocol);
374 client->SetSecurePolicy(secure_policy);
375 console->Start();
376
377 if (debug) {
378 pump.client()->SignalLogInput.connect(&debug_log_, &DebugLog::Input);
379 pump.client()->SignalLogOutput.connect(&debug_log_, &DebugLog::Output);
380 }
381
382 pump.DoLogin(xcs, new XmppSocket(true), NULL);
383 main_thread->Run();
384 pump.DoDisconnect();
385
386 console->Stop();
387 delete console;
388 delete client;
389
390 return 0;
391 }
392