cec: cleaned up command line param parsing
[deb_libcec.git] / src / testclient / main.cpp
CommitLineData
abbca718
LOK
1/*
2 * This file is part of the libCEC(R) library.
3 *
4 * libCEC(R) is Copyright (C) 2011 Pulse-Eight Limited. All rights reserved.
5 * libCEC(R) is an original work, containing original code.
6 *
7 * libCEC(R) is a trademark of Pulse-Eight Limited.
8 *
9 * This program is dual-licensed; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
18 *
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
22 *
23 *
24 * Alternatively, you can license this library under a commercial license,
25 * please contact Pulse-Eight Licensing for more information.
26 *
27 * For more information contact:
28 * Pulse-Eight Licensing <license@pulse-eight.com>
29 * http://www.pulse-eight.com/
30 * http://www.pulse-eight.net/
31 */
32
761dcc45 33#include <cec.h>
acec5f48 34
abbca718
LOK
35#include <cstdio>
36#include <fcntl.h>
37#include <iostream>
e6d2161b 38#include <fstream>
abbca718 39#include <string>
b9187cc6 40#include <sstream>
acec5f48
LOK
41#include "../lib/platform/threads.h"
42#include "../lib/util/StdString.h"
abbca718
LOK
43
44using namespace CEC;
45using namespace std;
46
8b7e5ff6 47#define CEC_TEST_CLIENT_VERSION 8
abbca718 48
761dcc45 49#include <cecloader.h>
8bc45476 50
3d0dca16
LOK
51int g_cecLogLevel = CEC_LOG_ALL;
52ofstream g_logOutput;
53bool g_bShortLog = false;
54CStdString g_strPort;
b9187cc6 55
8bca69de 56inline bool HexStrToInt(const std::string& data, uint8_t& value)
b9187cc6 57{
8bca69de
LOK
58 int iTmp(0);
59 if (sscanf(data.c_str(), "%x", &iTmp) == 1)
60 {
61 if (iTmp > 256)
62 value = 255;
63 else if (iTmp < 0)
64 value = 0;
65 else
66 value = (uint8_t) iTmp;
b9187cc6 67
8bca69de
LOK
68 return true;
69 }
70
71 return false;
72}
b9187cc6
LOK
73
74//get the first word (separated by whitespace) from string data and place that in word
75//then remove that word from string data
76bool GetWord(string& data, string& word)
77{
78 stringstream datastream(data);
79 string end;
80
81 datastream >> word;
82 if (datastream.fail())
83 {
84 data.clear();
85 return false;
86 }
87
88 size_t pos = data.find(word) + word.length();
89
90 if (pos >= data.length())
91 {
92 data.clear();
93 return true;
94 }
95
96 data = data.substr(pos);
97
98 datastream.clear();
99 datastream.str(data);
100
101 datastream >> end;
102 if (datastream.fail())
103 data.clear();
104
105 return true;
106}
107
2abe74eb 108void flush_log(ICECAdapter *cecParser)
abbca718
LOK
109{
110 cec_log_message message;
111 while (cecParser && cecParser->GetNextLogMessage(&message))
112 {
bdd433cb 113 if ((message.level & g_cecLogLevel) == message.level)
abbca718 114 {
e6d2161b 115 CStdString strLevel;
bdd433cb
LOK
116 switch (message.level)
117 {
118 case CEC_LOG_ERROR:
e6d2161b 119 strLevel = "ERROR: ";
bdd433cb
LOK
120 break;
121 case CEC_LOG_WARNING:
e6d2161b 122 strLevel = "WARNING: ";
bdd433cb
LOK
123 break;
124 case CEC_LOG_NOTICE:
e6d2161b 125 strLevel = "NOTICE: ";
bdd433cb
LOK
126 break;
127 case CEC_LOG_TRAFFIC:
e6d2161b 128 strLevel = "TRAFFIC: ";
bdd433cb
LOK
129 break;
130 case CEC_LOG_DEBUG:
e6d2161b 131 strLevel = "DEBUG: ";
bdd433cb
LOK
132 break;
133 default:
134 break;
135 }
40339852 136
8bc45476
LOK
137 CStdString strFullLog;
138 strFullLog.Format("%s[%16lld]\t%s", strLevel.c_str(), message.time, message.message);
139 cout << strFullLog.c_str() << endl;
e6d2161b
LOK
140
141 if (g_logOutput.is_open())
8bc45476
LOK
142 {
143 if (g_bShortLog)
144 g_logOutput << message.message << endl;
145 else
146 g_logOutput << strFullLog.c_str() << endl;
147 }
bdd433cb 148 }
abbca718
LOK
149 }
150}
151
2abe74eb 152void list_devices(ICECAdapter *parser)
abbca718 153{
25701fa6
LOK
154 cec_adapter *devices = new cec_adapter[10];
155 uint8_t iDevicesFound = parser->FindAdapters(devices, 10, NULL);
abbca718
LOK
156 if (iDevicesFound <= 0)
157 {
25701fa6 158 cout << "Found devices: NONE" << endl;
abbca718
LOK
159 }
160 else
161 {
25701fa6
LOK
162 CStdString strLog;
163 strLog.Format("Found devices: %d", iDevicesFound);
164 cout << strLog.c_str() << endl;
165 for (unsigned int iDevicePtr = 0; iDevicePtr < iDevicesFound; iDevicePtr++)
abbca718
LOK
166 {
167 CStdString strDevice;
25701fa6 168 strDevice.Format("device: %d\npath: %s\ncom port: %s", iDevicePtr + 1, devices[iDevicePtr].path, devices[iDevicePtr].comm);
abbca718
LOK
169 cout << endl << strDevice.c_str() << endl;
170 }
171 }
172}
173
174void show_help(const char* strExec)
175{
176 cout << endl <<
177 strExec << " {-h|--help|-l|--list-devices|[COM PORT]}" << endl <<
178 endl <<
179 "parameters:" << endl <<
8bc45476
LOK
180 " -h --help Shows this help text" << endl <<
181 " -l --list-devices List all devices on this system" << endl <<
182 " -f --log-file {file} Writes all libCEC log message to a file" << endl <<
183 " -sf --short-log-file {file} Writes all libCEC log message without timestamps" << endl <<
184 " and log levels to a file." << endl <<
185 " [COM PORT] The com port to connect to. If no COM" << endl <<
186 " port is given, the client tries to connect to the" << endl <<
187 " first device that is detected." << endl <<
825ddb96 188 endl <<
8bc45476
LOK
189 "Type 'h' or 'help' and press enter after starting the client to display all " << endl <<
190 "available commands" << endl;
825ddb96
LOK
191}
192
193void show_console_help(void)
194{
195 cout << endl <<
196 "================================================================================" << endl <<
197 "Available commands:" << endl <<
198 endl <<
199 "tx {bytes} transfer bytes over the CEC line." << endl <<
faa6a23b 200 "txn {bytes} transfer bytes and don't wait for an ACK reply." << endl <<
825ddb96
LOK
201 "[tx 40 00 FF 11 22 33] sends bytes 0x40 0x00 0xFF 0x11 0x22 0x33" << endl <<
202 endl <<
88f45c9b
LOK
203 "on {address} power on the device with the given logical address." << endl <<
204 "[on 5] power on a connected audio system" << endl <<
205 endl <<
206 "standby {address} put the device with the given address in standby mode." << endl <<
207 "[standby 0] powers off the TV" << endl <<
208 endl <<
825ddb96
LOK
209 "la {logical_address} change the logical address of the CEC adapter." << endl <<
210 "[la 4] logical address 4" << endl <<
211 endl <<
a424ebac
LOK
212 "pa {physical_address} change the physical address of the CEC adapter." << endl <<
213 "[pa 10 00] physical address 1.0.0.0" << endl <<
214 endl <<
1969b140
LOK
215 "osd {addr} {string} set OSD message on the specified device." << endl <<
216 "[osd 0 Test Message] displays 'Test Message' on the TV" << endl <<
217 endl <<
8b7e5ff6 218 "[mon] {1|0} enable or disable CEC bus monitoring." << endl <<
bdd433cb 219 "[log] {1 - 31} change the log level. see cectypes.h for values." << endl <<
825ddb96 220 "[ping] send a ping command to the CEC adapter." << endl <<
88f45c9b
LOK
221 "[bl] to let the adapter enter the bootloader, to upgrade" << endl <<
222 " the flash rom." << endl <<
223 "[r] reconnect to the CEC adapter." << endl <<
825ddb96 224 "[h] or [help] show this help." << endl <<
88f45c9b
LOK
225 "[q] or [quit] to quit the CEC test client and switch off all" << endl <<
226 " connected CEC devices." << endl <<
825ddb96 227 "================================================================================" << endl;
abbca718
LOK
228}
229
230int main (int argc, char *argv[])
231{
d54c6c91 232 ICECAdapter *parser = LoadLibCec("CECTester");
acec5f48 233 if (!parser || parser->GetMinVersion() > CEC_TEST_CLIENT_VERSION)
abbca718 234 {
acec5f48
LOK
235#ifdef __WINDOWS__
236 cout << "Cannot load libcec.dll" << endl;
237#else
238 cout << "Cannot load libcec.so" << endl;
239#endif
abbca718
LOK
240 return 1;
241 }
242 CStdString strLog;
243 strLog.Format("CEC Parser created - libcec version %d", parser->GetLibVersion());
244 cout << strLog.c_str() << endl;
245
246 //make stdin non-blocking
247#ifndef __WINDOWS__
248 int flags = fcntl(0, F_GETFL, 0);
249 flags |= O_NONBLOCK;
250 fcntl(0, F_SETFL, flags);
251#endif
252
3d0dca16
LOK
253 int iArgPtr = 1;
254 while (iArgPtr < argc)
e6d2161b 255 {
3d0dca16 256 if (argc >= iArgPtr + 1)
e6d2161b 257 {
3d0dca16
LOK
258 if (!strcmp(argv[iArgPtr], "-f") ||
259 !strcmp(argv[iArgPtr], "--log-file") ||
260 !strcmp(argv[iArgPtr], "-sf") ||
261 !strcmp(argv[iArgPtr], "--short-log-file"))
262 {
263 if (argc >= iArgPtr + 2)
264 {
265 g_logOutput.open(argv[iArgPtr + 1]);
266 g_bShortLog = (!strcmp(argv[iArgPtr], "-sf") || !strcmp(argv[iArgPtr], "--short-log-file"));
267 iArgPtr += 2;
268 }
269 else
270 {
271 ++iArgPtr;
272 }
273 }
274 else if (!strcmp(argv[iArgPtr], "--list-devices") ||
275 !strcmp(argv[iArgPtr], "-l"))
276 {
277 list_devices(parser);
278 UnloadLibCec(parser);
279 return 0;
280 }
281 else if (!strcmp(argv[iArgPtr], "--help") ||
282 !strcmp(argv[iArgPtr], "-h"))
283 {
284 show_help(argv[0]);
285 UnloadLibCec(parser);
286 return 0;
287 }
288 else
289 {
290 g_strPort = argv[iArgPtr++];
291 }
e6d2161b
LOK
292 }
293 }
294
3d0dca16 295 if (g_strPort.IsEmpty())
abbca718
LOK
296 {
297 cout << "no serial port given. trying autodetect: ";
25701fa6
LOK
298 cec_adapter devices[10];
299 uint8_t iDevicesFound = parser->FindAdapters(devices, 10, NULL);
abbca718
LOK
300 if (iDevicesFound <= 0)
301 {
302 cout << "FAILED" << endl;
303 UnloadLibCec(parser);
304 return 1;
305 }
306 else
307 {
308 cout << endl << " path: " << devices[0].path << endl <<
3d0dca16
LOK
309 " com port: " << devices[0].comm << endl << endl;
310 g_strPort = devices[0].comm;
abbca718
LOK
311 }
312 }
e6d2161b 313
3d0dca16 314 if (!parser->Open(g_strPort.c_str()))
abbca718 315 {
3d0dca16 316 cout << "unable to open the device on port " << g_strPort << endl;
abbca718
LOK
317 flush_log(parser);
318 UnloadLibCec(parser);
319 return 1;
320 }
321
322 cout << "cec device opened" << endl;
abbca718 323
8829d291 324 parser->PowerOnDevices(CECDEVICE_TV);
abbca718
LOK
325 flush_log(parser);
326
327 parser->SetActiveView();
328 flush_log(parser);
329
330 bool bContinue(true);
331 cout << "waiting for input" << endl;
332 while (bContinue)
333 {
334 flush_log(parser);
335
336 string input;
337 getline(cin, input);
338 cin.clear();
339
340 if (!input.empty())
341 {
342 string command;
343 if (GetWord(input, command))
344 {
faa6a23b 345 if (command == "tx" || command == "txn")
abbca718
LOK
346 {
347 string strvalue;
8bca69de 348 uint8_t ivalue;
9dee1670 349 cec_command bytes;
25701fa6
LOK
350 bytes.clear();
351
abbca718 352 while (GetWord(input, strvalue) && HexStrToInt(strvalue, ivalue))
b2da2768 353 bytes.push_back(ivalue);
abbca718 354
faa6a23b 355 parser->Transmit(bytes, command == "tx");
abbca718 356 }
88f45c9b
LOK
357 else if (command == "on")
358 {
359 string strValue;
360 uint8_t iValue = 0;
361 if (GetWord(input, strValue) && HexStrToInt(strValue, iValue) && iValue <= 0xF)
362 {
363 parser->PowerOnDevices((cec_logical_address) iValue);
364 }
365 else
366 {
367 cout << "invalid destination" << endl;
368 }
369 }
370 else if (command == "standby")
371 {
372 string strValue;
373 uint8_t iValue = 0;
374 if (GetWord(input, strValue) && HexStrToInt(strValue, iValue) && iValue <= 0xF)
375 {
376 parser->StandbyDevices((cec_logical_address) iValue);
377 }
378 else
379 {
380 cout << "invalid destination" << endl;
381 }
382 }
6dfe9213
LOK
383 else if (command == "la")
384 {
385 string strvalue;
6dfe9213
LOK
386 if (GetWord(input, strvalue))
387 {
388 parser->SetLogicalAddress((cec_logical_address) atoi(strvalue.c_str()));
abbca718
LOK
389 }
390 }
a424ebac
LOK
391 else if (command == "pa")
392 {
393 string strB1, strB2;
394 uint8_t iB1, iB2;
395 if (GetWord(input, strB1) && HexStrToInt(strB1, iB1) &&
396 GetWord(input, strB2) && HexStrToInt(strB2, iB2))
397 {
398 uint16_t iPhysicalAddress = ((uint16_t)iB1 << 8) + iB2;
399 parser->SetPhysicalAddress(iPhysicalAddress);
400 }
401 }
1969b140
LOK
402 else if (command == "osd")
403 {
404 bool bFirstWord(false);
405 string strAddr, strMessage, strWord;
406 uint8_t iAddr;
407 if (GetWord(input, strAddr) && HexStrToInt(strAddr, iAddr) && iAddr < 0xF)
408 {
409 while (GetWord(input, strWord))
410 {
411 if (bFirstWord)
412 {
413 bFirstWord = false;
414 strMessage.append(" ");
415 }
416 strMessage.append(strWord);
417 }
418 parser->SetOSDString((cec_logical_address) iAddr, CEC_DISPLAY_CONTROL_DISPLAY_FOR_DEFAULT_TIME, strMessage.c_str());
419 }
420 }
abbca718
LOK
421 else if (command == "ping")
422 {
2abe74eb 423 parser->PingAdapter();
abbca718 424 }
8b7e5ff6
LOK
425 else if (command == "mon")
426 {
427 CStdString strEnable;
428 if (GetWord(input, strEnable) && (strEnable.Equals("0") || strEnable.Equals("1")))
429 {
430 parser->SwitchMonitoring(strEnable.Equals("1"));
431 }
432 }
abbca718
LOK
433 else if (command == "bl")
434 {
435 parser->StartBootloader();
436 }
12027dbe
LOK
437 else if (command == "r")
438 {
25701fa6 439 cout << "closing the connection" << endl;
12027dbe 440 parser->Close();
25701fa6
LOK
441 flush_log(parser);
442
443 cout << "opening a new connection" << endl;
3d0dca16 444 parser->Open(g_strPort.c_str());
25701fa6
LOK
445 flush_log(parser);
446
447 cout << "setting active view" << endl;
12027dbe
LOK
448 parser->SetActiveView();
449 }
825ddb96
LOK
450 else if (command == "h" || command == "help")
451 {
452 show_console_help();
453 }
abbca718
LOK
454 else if (command == "q" || command == "quit")
455 {
456 bContinue = false;
457 }
bdd433cb
LOK
458 else if (command == "log")
459 {
460 CStdString strLevel;
461 if (GetWord(input, strLevel))
462 {
463 int iNewLevel = atoi(strLevel);
464 if (iNewLevel >= CEC_LOG_ERROR && iNewLevel <= CEC_LOG_ALL)
465 {
466 g_cecLogLevel = iNewLevel;
467 cout << "log level changed to " << strLevel.c_str() << endl;
468 }
469 }
470 }
abbca718
LOK
471 }
472 if (bContinue)
473 cout << "waiting for input" << endl;
474 }
475 CCondition::Sleep(50);
476 }
477
2abe74eb 478 parser->StandbyDevices(CECDEVICE_BROADCAST);
5f39c4d8 479 parser->Close();
abbca718
LOK
480 flush_log(parser);
481 UnloadLibCec(parser);
e6d2161b
LOK
482
483 if (g_logOutput.is_open())
484 g_logOutput.close();
485
abbca718
LOK
486 return 0;
487}