src/testclient/main.cpp

   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
  33 #include <cec.h>
  34
  35 #include <cstdio>
  36 #include <fcntl.h>
  37 #include <iostream>
  38 #include <fstream>
  39 #include <string>
  40 #include <sstream>
  41 #include "../lib/platform/threads.h"
  42 #include "../lib/util/StdString.h"
  43
  44 using namespace CEC;
  45 using namespace std;
  46
  47 #define CEC_TEST_CLIENT_VERSION 8
  48
  49 #include <cecloader.h>
  50
  51 int        g_cecLogLevel = CEC_LOG_ALL;
  52 int        g_iLogicalAddress = CECDEVICE_PLAYBACKDEVICE1;
  53 ofstream   g_logOutput;
  54 bool       g_bShortLog = false;
  55 CStdString g_strPort;
  56
  57 inline bool HexStrToInt(const std::string& data, uint8_t& value)
  58 {
  59   int iTmp(0);
  60   if (sscanf(data.c_str(), "%x", &iTmp) == 1)
  61   {
  62     if (iTmp > 256)
  63       value = 255;
  64           else if (iTmp < 0)
  65       value = 0;
  66     else
  67       value = (uint8_t) iTmp;
  68
  69     return true;
  70   }
  71
  72   return false;
  73 }
  74
  75 //get the first word (separated by whitespace) from string data and place that in word
  76 //then remove that word from string data
  77 bool GetWord(string& data, string& word)
  78 {
  79   stringstream datastream(data);
  80   string end;
  81
  82   datastream >> word;
  83   if (datastream.fail())
  84   {
  85     data.clear();
  86     return false;
  87   }
  88
  89   size_t pos = data.find(word) + word.length();
  90
  91   if (pos >= data.length())
  92   {
  93     data.clear();
  94     return true;
  95   }
  96
  97   data = data.substr(pos);
  98
  99   datastream.clear();
 100   datastream.str(data);
 101
 102   datastream >> end;
 103   if (datastream.fail())
 104     data.clear();
 105
 106   return true;
 107 }
 108
 109 void flush_log(ICECAdapter *cecParser)
 110 {
 111   cec_log_message message;
 112   while (cecParser && cecParser->GetNextLogMessage(&message))
 113   {
 114     if ((message.level & g_cecLogLevel) == message.level)
 115     {
 116       CStdString strLevel;
 117       switch (message.level)
 118       {
 119       case CEC_LOG_ERROR:
 120         strLevel = "ERROR:   ";
 121         break;
 122       case CEC_LOG_WARNING:
 123         strLevel = "WARNING: ";
 124         break;
 125       case CEC_LOG_NOTICE:
 126         strLevel = "NOTICE:  ";
 127         break;
 128       case CEC_LOG_TRAFFIC:
 129         strLevel = "TRAFFIC: ";
 130         break;
 131       case CEC_LOG_DEBUG:
 132         strLevel = "DEBUG:   ";
 133         break;
 134       default:
 135         break;
 136       }
 137
 138       CStdString strFullLog;
 139       strFullLog.Format("%s[%16lld]\t%s", strLevel.c_str(), message.time, message.message);
 140       cout << strFullLog.c_str() << endl;
 141
 142       if (g_logOutput.is_open())
 143       {
 144         if (g_bShortLog)
 145           g_logOutput << message.message << endl;
 146         else
 147           g_logOutput << strFullLog.c_str() << endl;
 148       }
 149     }
 150   }
 151 }
 152
 153 void list_devices(ICECAdapter *parser)
 154 {
 155   cec_adapter *devices = new cec_adapter[10];
 156   uint8_t iDevicesFound = parser->FindAdapters(devices, 10, NULL);
 157   if (iDevicesFound <= 0)
 158   {
 159     cout << "Found devices: NONE" << endl;
 160   }
 161   else
 162   {
 163     CStdString strLog;
 164     strLog.Format("Found devices: %d", iDevicesFound);
 165     cout << strLog.c_str() << endl;
 166     for (unsigned int iDevicePtr = 0; iDevicePtr < iDevicesFound; iDevicePtr++)
 167     {
 168       CStdString strDevice;
 169       strDevice.Format("device:        %d\npath:          %s\ncom port:      %s", iDevicePtr + 1, devices[iDevicePtr].path, devices[iDevicePtr].comm);
 170       cout << endl << strDevice.c_str() << endl;
 171     }
 172   }
 173 }
 174
 175 void show_help(const char* strExec)
 176 {
 177   cout << endl <<
 178       strExec << " {-h|--help|-l|--list-devices|[COM PORT]}" << endl <<
 179       endl <<
 180       "parameters:" << endl <<
 181       "  -h --help                   Shows this help text" << endl <<
 182       "  -l --list-devices           List all devices on this system" << endl <<
 183       "  -la --logical-address {a}   The logical address to use." << endl <<
 184       "  -f --log-file {file}        Writes all libCEC log message to a file" << endl <<
 185       "  -sf --short-log-file {file} Writes all libCEC log message without timestamps" << endl <<
 186       "                              and log levels to a file." << endl <<
 187       "  -d --log-level {level}      Sets the log level. See cectypes.h for values." << endl <<
 188       "  [COM PORT]                  The com port to connect to. If no COM" << endl <<
 189       "                              port is given, the client tries to connect to the" << endl <<
 190       "                              first device that is detected." << endl <<
 191       endl <<
 192       "Type 'h' or 'help' and press enter after starting the client to display all " << endl <<
 193       "available commands" << endl;
 194 }
 195
 196 void show_console_help(void)
 197 {
 198   cout << endl <<
 199   "================================================================================" << endl <<
 200   "Available commands:" << endl <<
 201   endl <<
 202   "tx {bytes}                transfer bytes over the CEC line." << endl <<
 203   "txn {bytes}               transfer bytes and don't wait for an ACK reply." << endl <<
 204   "[tx 40 00 FF 11 22 33]    sends bytes 0x40 0x00 0xFF 0x11 0x22 0x33" << endl <<
 205   endl <<
 206   "on {address}              power on the device with the given logical address." << endl <<
 207   "[on 5]                    power on a connected audio system" << endl <<
 208   endl <<
 209   "standby {address}         put the device with the given address in standby mode." << endl <<
 210   "[standby 0]               powers off the TV" << endl <<
 211   endl <<
 212   "la {logical_address}      change the logical address of the CEC adapter." << endl <<
 213   "[la 4]                    logical address 4" << endl <<
 214   endl <<
 215   "pa {physical_address}     change the physical address of the CEC adapter." << endl <<
 216   "[pa 10 00]                physical address 1.0.0.0" << endl <<
 217   endl <<
 218   "osd {addr} {string}       set OSD message on the specified device." << endl <<
 219   "[osd 0 Test Message]      displays 'Test Message' on the TV" << endl <<
 220   endl <<
 221   "ver {addr}                get the CEC version of the specified device." << endl <<
 222   "[ver 0]                   get the CEC version of the TV" << endl <<
 223   endl <<
 224   "lang {addr                get the menu language of the specified device." << endl <<
 225   "[lang 0]                  get the menu language of the TV" << endl <<
 226   endl <<
 227   "[mon] {1|0}               enable or disable CEC bus monitoring." << endl <<
 228   "[log] {1 - 31}            change the log level. see cectypes.h for values." << endl <<
 229   "[ping]                    send a ping command to the CEC adapter." << endl <<
 230   "[bl]                      to let the adapter enter the bootloader, to upgrade" << endl <<
 231   "                          the flash rom." << endl <<
 232   "[r]                       reconnect to the CEC adapter." << endl <<
 233   "[h] or [help]             show this help." << endl <<
 234   "[q] or [quit]             to quit the CEC test client and switch off all" << endl <<
 235   "                          connected CEC devices." << endl <<
 236   "================================================================================" << endl;
 237 }
 238
 239 int main (int argc, char *argv[])
 240 {
 241   ICECAdapter *parser = LoadLibCec("CECTester");
 242   if (!parser || parser->GetMinVersion() > CEC_TEST_CLIENT_VERSION)
 243   {
 244 #ifdef __WINDOWS__
 245     cout << "Cannot load libcec.dll" << endl;
 246 #else
 247     cout << "Cannot load libcec.so" << endl;
 248 #endif
 249     return 1;
 250   }
 251   CStdString strLog;
 252   strLog.Format("CEC Parser created - libcec version %d", parser->GetLibVersion());
 253   cout << strLog.c_str() << endl;
 254
 255   //make stdin non-blocking
 256 #ifndef __WINDOWS__
 257   int flags = fcntl(0, F_GETFL, 0);
 258   flags |= O_NONBLOCK;
 259   fcntl(0, F_SETFL, flags);
 260 #endif
 261
 262   int iArgPtr = 1;
 263   while (iArgPtr < argc)
 264   {
 265     if (argc >= iArgPtr + 1)
 266     {
 267       if (!strcmp(argv[iArgPtr], "-f") ||
 268           !strcmp(argv[iArgPtr], "--log-file") ||
 269           !strcmp(argv[iArgPtr], "-sf") ||
 270           !strcmp(argv[iArgPtr], "--short-log-file"))
 271       {
 272         if (argc >= iArgPtr + 2)
 273         {
 274           g_logOutput.open(argv[iArgPtr + 1]);
 275           g_bShortLog = (!strcmp(argv[iArgPtr], "-sf") || !strcmp(argv[iArgPtr], "--short-log-file"));
 276           iArgPtr += 2;
 277         }
 278         else
 279         {
 280           cout << "== skipped log-file parameter: no file given ==" << endl;
 281           ++iArgPtr;
 282         }
 283       }
 284       else if (!strcmp(argv[iArgPtr], "-d") ||
 285           !strcmp(argv[iArgPtr], "--log-level"))
 286       {
 287         if (argc >= iArgPtr + 2)
 288         {
 289           int iNewLevel = atoi(argv[iArgPtr + 1]);
 290           if (iNewLevel >= CEC_LOG_ERROR && iNewLevel <= CEC_LOG_ALL)
 291           {
 292             g_cecLogLevel = iNewLevel;
 293             cout << "log level set to " << argv[iArgPtr + 1] << endl;
 294           }
 295           else
 296           {
 297             cout << "== skipped log-level parameter: invalid level '" << argv[iArgPtr + 1] << "' ==" << endl;
 298           }
 299           iArgPtr += 2;
 300         }
 301         else
 302         {
 303           cout << "== skipped log-level parameter: no level given ==" << endl;
 304           ++iArgPtr;
 305         }
 306       }
 307       else if (!strcmp(argv[iArgPtr], "-la") ||
 308                !strcmp(argv[iArgPtr], "--logical-address"))
 309       {
 310         if (argc >= iArgPtr + 2)
 311         {
 312           int iNewAddress = atoi(argv[iArgPtr + 1]);
 313           if (iNewAddress >= 0 && iNewAddress <= 15)
 314           {
 315             g_iLogicalAddress = iNewAddress;
 316             cout << "logical address set to " << argv[iArgPtr + 1] << endl;
 317           }
 318           else
 319           {
 320             cout << "== skipped logical-address parameter: invalid address '" << argv[iArgPtr + 1] << "' ==" << endl;
 321           }
 322           iArgPtr += 2;
 323         }
 324         else
 325         {
 326           cout << "== skipped logical-address parameter: no address given ==" << endl;
 327           ++iArgPtr;
 328         }
 329       }
 330       else if (!strcmp(argv[iArgPtr], "--list-devices") ||
 331                !strcmp(argv[iArgPtr], "-l"))
 332       {
 333         list_devices(parser);
 334         UnloadLibCec(parser);
 335         return 0;
 336       }
 337       else if (!strcmp(argv[iArgPtr], "--help") ||
 338                !strcmp(argv[iArgPtr], "-h"))
 339       {
 340         show_help(argv[0]);
 341         UnloadLibCec(parser);
 342         return 0;
 343       }
 344       else
 345       {
 346         g_strPort = argv[iArgPtr++];
 347       }
 348     }
 349   }
 350
 351   if (g_strPort.IsEmpty())
 352   {
 353     cout << "no serial port given. trying autodetect: ";
 354     cec_adapter devices[10];
 355     uint8_t iDevicesFound = parser->FindAdapters(devices, 10, NULL);
 356     if (iDevicesFound <= 0)
 357     {
 358       cout << "FAILED" << endl;
 359       UnloadLibCec(parser);
 360       return 1;
 361     }
 362     else
 363     {
 364       cout << endl << " path:     " << devices[0].path << endl <<
 365           " com port: " << devices[0].comm << endl << endl;
 366       g_strPort = devices[0].comm;
 367     }
 368   }
 369
 370   parser->SetLogicalAddress((cec_logical_address) g_iLogicalAddress);
 371
 372   if (!parser->Open(g_strPort.c_str()))
 373   {
 374     cout << "unable to open the device on port " << g_strPort << endl;
 375     flush_log(parser);
 376     UnloadLibCec(parser);
 377     return 1;
 378   }
 379
 380   cout << "cec device opened" << endl;
 381
 382   parser->PowerOnDevices(CECDEVICE_TV);
 383   flush_log(parser);
 384
 385   parser->SetActiveView();
 386   flush_log(parser);
 387
 388   bool bContinue(true);
 389   cout << "waiting for input" << endl;
 390   while (bContinue)
 391   {
 392     flush_log(parser);
 393
 394     string input;
 395     getline(cin, input);
 396     cin.clear();
 397
 398     if (!input.empty())
 399     {
 400       string command;
 401       if (GetWord(input, command))
 402       {
 403         if (command == "tx" || command == "txn")
 404         {
 405           string strvalue;
 406           uint8_t ivalue;
 407           cec_command bytes;
 408           bytes.clear();
 409
 410           while (GetWord(input, strvalue) && HexStrToInt(strvalue, ivalue))
 411             bytes.push_back(ivalue);
 412
 413           if (command == "txn")
 414             bytes.ack_timeout = 0;
 415
 416           parser->Transmit(bytes);
 417         }
 418         else if (command == "on")
 419         {
 420           string strValue;
 421           uint8_t iValue = 0;
 422           if (GetWord(input, strValue) && HexStrToInt(strValue, iValue) && iValue <= 0xF)
 423           {
 424             parser->PowerOnDevices((cec_logical_address) iValue);
 425           }
 426           else
 427           {
 428             cout << "invalid destination" << endl;
 429           }
 430         }
 431         else if (command == "standby")
 432         {
 433           string strValue;
 434           uint8_t iValue = 0;
 435           if (GetWord(input, strValue) && HexStrToInt(strValue, iValue) && iValue <= 0xF)
 436           {
 437             parser->StandbyDevices((cec_logical_address) iValue);
 438           }
 439           else
 440           {
 441             cout << "invalid destination" << endl;
 442           }
 443         }
 444         else if (command == "la")
 445         {
 446           string strvalue;
 447           if (GetWord(input, strvalue))
 448           {
 449             parser->SetLogicalAddress((cec_logical_address) atoi(strvalue.c_str()));
 450           }
 451         }
 452         else if (command == "pa")
 453         {
 454           string strB1, strB2;
 455           uint8_t iB1, iB2;
 456           if (GetWord(input, strB1) && HexStrToInt(strB1, iB1) &&
 457               GetWord(input, strB2) && HexStrToInt(strB2, iB2))
 458           {
 459             uint16_t iPhysicalAddress = ((uint16_t)iB1 << 8) + iB2;
 460             parser->SetPhysicalAddress(iPhysicalAddress);
 461           }
 462         }
 463         else if (command == "osd")
 464         {
 465           bool bFirstWord(false);
 466           string strAddr, strMessage, strWord;
 467           uint8_t iAddr;
 468           if (GetWord(input, strAddr) && HexStrToInt(strAddr, iAddr) && iAddr < 0xF)
 469           {
 470             while (GetWord(input, strWord))
 471             {
 472               if (bFirstWord)
 473               {
 474                 bFirstWord = false;
 475                 strMessage.append(" ");
 476               }
 477               strMessage.append(strWord);
 478             }
 479             parser->SetOSDString((cec_logical_address) iAddr, CEC_DISPLAY_CONTROL_DISPLAY_FOR_DEFAULT_TIME, strMessage.c_str());
 480           }
 481         }
 482         else if (command == "ping")
 483         {
 484           parser->PingAdapter();
 485         }
 486         else if (command == "mon")
 487         {
 488           CStdString strEnable;
 489           if (GetWord(input, strEnable) && (strEnable.Equals("0") || strEnable.Equals("1")))
 490           {
 491             parser->SwitchMonitoring(strEnable.Equals("1"));
 492           }
 493         }
 494         else if (command == "bl")
 495         {
 496           parser->StartBootloader();
 497         }
 498         else if (command == "lang")
 499         {
 500           CStdString strDev;
 501           if (GetWord(input, strDev))
 502           {
 503             int iDev = atoi(strDev);
 504             if (iDev >= 0 && iDev < 15)
 505             {
 506               CStdString strLog;
 507               cec_menu_language language;
 508               if (parser->GetDeviceMenuLanguage((cec_logical_address) iDev, &language))
 509                 strLog.Format("menu language '%s'", language.language);
 510               else
 511                 strLog = "failed!";
 512               cout << strLog.c_str() << endl;
 513             }
 514           }
 515         }
 516         else if (command == "ver")
 517         {
 518           CStdString strDev;
 519           if (GetWord(input, strDev))
 520           {
 521             int iDev = atoi(strDev);
 522             if (iDev >= 0 && iDev < 15)
 523             {
 524               cec_version iVersion = parser->GetDeviceCecVersion((cec_logical_address) iDev);
 525               switch (iVersion)
 526               {
 527               case CEC_VERSION_1_2:
 528                 cout << "CEC version 1.2" << endl;
 529                 break;
 530               case CEC_VERSION_1_2A:
 531                 cout << "CEC version 1.2a" << endl;
 532                 break;
 533               case CEC_VERSION_1_3:
 534                 cout << "CEC version 1.3" << endl;
 535                 break;
 536               case CEC_VERSION_1_3A:
 537                 cout << "CEC version 1.3a" << endl;
 538                 break;
 539               default:
 540                 cout << "unknown CEC version" << endl;
 541                 break;
 542               }
 543             }
 544           }
 545         }
 546         else if (command == "r")
 547         {
 548           cout << "closing the connection" << endl;
 549           parser->Close();
 550           flush_log(parser);
 551
 552           cout << "opening a new connection" << endl;
 553           parser->Open(g_strPort.c_str());
 554           flush_log(parser);
 555
 556           cout << "setting active view" << endl;
 557           parser->SetActiveView();
 558         }
 559         else if (command == "h" || command == "help")
 560         {
 561           show_console_help();
 562         }
 563         else if (command == "q" || command == "quit")
 564         {
 565           bContinue = false;
 566         }
 567         else if (command == "log")
 568         {
 569           CStdString strLevel;
 570           if (GetWord(input, strLevel))
 571           {
 572             int iNewLevel = atoi(strLevel);
 573             if (iNewLevel >= CEC_LOG_ERROR && iNewLevel <= CEC_LOG_ALL)
 574             {
 575               g_cecLogLevel = iNewLevel;
 576               cout << "log level changed to " << strLevel.c_str() << endl;
 577             }
 578           }
 579         }
 580       }
 581       if (bContinue)
 582         cout << "waiting for input" << endl;
 583     }
 584     CCondition::Sleep(50);
 585   }
 586
 587   parser->StandbyDevices(CECDEVICE_BROADCAST);
 588   parser->Close();
 589   flush_log(parser);
 590   UnloadLibCec(parser);
 591
 592   if (g_logOutput.is_open())
 593     g_logOutput.close();
 594
 595   return 0;
 596 }